| /* |
| * Copyright (c) 2015-2018, ARM Limited and Contributors. All rights reserved. |
| * |
| * SPDX-License-Identifier: BSD-3-Clause |
| */ |
| |
| #include <asm_macros.S> |
| #include <assert_macros.S> |
| #include <console.h> |
| |
| .globl console_register |
| .globl console_unregister |
| .globl console_is_registered |
| .globl console_set_scope |
| .globl console_switch_state |
| .globl console_putc |
| .globl console_getc |
| .globl console_flush |
| |
| /* |
| * The console list pointer is in the data section and not in |
| * .bss even though it is zero-init. In particular, this allows |
| * the console functions to start using this variable before |
| * the runtime memory is initialized for images which do not |
| * need to copy the .data section from ROM to RAM. |
| */ |
| .section .data.console_list ; .align 2 |
| console_list: .word 0x0 |
| .section .data.console_state ; .align 0 |
| console_state: .byte CONSOLE_FLAG_BOOT |
| |
| /* ----------------------------------------------- |
| * int console_register(console_t *console) |
| * Function to insert a new console structure into |
| * the console list. Should usually be called by |
| * console_<driver>_register implementations. The |
| * data structure passed will be taken over by the |
| * console framework and *MUST* be allocated in |
| * persistent memory (e.g. the data section). |
| * In : r0 - address of console_t structure |
| * Out: r0 - Always 1 (for easier tail calling) |
| * Clobber list: r0, r1 |
| * ----------------------------------------------- |
| */ |
| func console_register |
| push {r6, lr} |
| #if ENABLE_ASSERTIONS |
| /* Assert that r0 isn't a NULL pointer */ |
| cmp r0, #0 |
| ASM_ASSERT(ne) |
| /* Assert that the struct isn't in the stack */ |
| ldr r1, =__STACKS_START__ |
| cmp r0, r1 |
| blo not_on_stack |
| ldr r1, =__STACKS_END__ |
| cmp r0, r1 |
| ASM_ASSERT(hs) |
| not_on_stack: |
| /* Assert that this struct isn't in the list */ |
| mov r1, r0 /* Preserve r0 and lr */ |
| bl console_is_registered |
| cmp r0, #0 |
| ASM_ASSERT(eq) |
| mov r0, r1 |
| #endif /* ENABLE_ASSERTIONS */ |
| ldr r6, =console_list |
| ldr r1, [r6] /* R1 = first struct in list */ |
| str r0, [r6] /* list head = new console */ |
| str r1, [r0, #CONSOLE_T_NEXT] /* new console next ptr = R1 */ |
| mov r0, #1 |
| pop {r6, pc} |
| endfunc console_register |
| |
| /* ----------------------------------------------- |
| * int console_unregister(console_t *console) |
| * Function to find a specific console in the list |
| * of currently active consoles and remove it. |
| * In: r0 - address of console_t struct to remove |
| * Out: r0 - removed address, or NULL if not found |
| * Clobber list: r0, r1 |
| * ----------------------------------------------- |
| */ |
| func console_unregister |
| #if ENABLE_ASSERTIONS |
| /* Assert that r0 isn't a NULL pointer */ |
| cmp r0, #0 |
| ASM_ASSERT(ne) |
| #endif /* ENABLE_ASSERTIONS */ |
| push {r6} |
| ldr r6, =console_list /* R6 = ptr to first struct */ |
| ldr r1, [r6] /* R1 = first struct */ |
| |
| unregister_loop: |
| cmp r1, #0 |
| beq unregister_not_found |
| cmp r0, r1 |
| beq unregister_found |
| ldr r6, [r6] /* R6 = next ptr of struct */ |
| ldr r1, [r6] /* R1 = next struct */ |
| b unregister_loop |
| |
| unregister_found: |
| ldr r1, [r1] /* R1 = next struct */ |
| str r1, [r6] /* prev->next = cur->next */ |
| pop {r6} |
| bx lr |
| |
| unregister_not_found: |
| mov r0, #0 /* return NULL if not found */ |
| pop {r6} |
| bx lr |
| endfunc console_unregister |
| |
| /* ----------------------------------------------- |
| * int console_is_registered(console_t *console) |
| * Function to detect if a specific console is |
| * registered or not. |
| * In: r0 - address of console_t struct to remove |
| * Out: r0 - 1 if it is registered, 0 if not. |
| * Clobber list: r0 |
| * ----------------------------------------------- |
| */ |
| func console_is_registered |
| #if ENABLE_ASSERTIONS |
| /* Assert that r0 isn't a NULL pointer */ |
| cmp r0, #0 |
| ASM_ASSERT(ne) |
| #endif /* ENABLE_ASSERTIONS */ |
| push {r6} |
| ldr r6, =console_list |
| ldr r6, [r6] /* R6 = first console struct */ |
| check_registered_loop: |
| cmp r6, #0 /* Check if end of list */ |
| beq console_not_registered |
| cmp r0, r6 /* Check if the pointers are different */ |
| beq console_registered |
| ldr r6, [r6, #CONSOLE_T_NEXT] /* Get pointer to next struct */ |
| b check_registered_loop |
| console_not_registered: |
| mov r0, #0 |
| pop {r6} |
| bx lr |
| console_registered: |
| mov r0, #1 |
| pop {r6} |
| bx lr |
| endfunc console_is_registered |
| |
| /* ----------------------------------------------- |
| * void console_switch_state(unsigned int new_state) |
| * Function to switch the current console state. |
| * The console state determines which of the |
| * registered consoles are actually used at a time. |
| * In : r0 - global console state to move to |
| * Clobber list: r0, r1 |
| * ----------------------------------------------- |
| */ |
| func console_switch_state |
| ldr r1, =console_state |
| strb r0, [r1] |
| bx lr |
| endfunc console_switch_state |
| |
| /* ----------------------------------------------- |
| * void console_set_scope(console_t *console, |
| * unsigned int scope) |
| * Function to update the states that a given console |
| * may be active in. |
| * In : r0 - pointer to console_t struct |
| * : r1 - new active state mask |
| * Clobber list: r0, r1, r2 |
| * ----------------------------------------------- |
| */ |
| func console_set_scope |
| #if ENABLE_ASSERTIONS |
| ands r2, r1, #~CONSOLE_FLAG_SCOPE_MASK |
| ASM_ASSERT(eq) |
| #endif /* ENABLE_ASSERTIONS */ |
| ldr r2, [r0, #CONSOLE_T_FLAGS] |
| and r2, r2, #~CONSOLE_FLAG_SCOPE_MASK |
| orr r2, r2, r1 |
| str r2, [r0, #CONSOLE_T_FLAGS] |
| bx lr |
| endfunc console_set_scope |
| |
| /* --------------------------------------------- |
| * int console_putc(int c) |
| * Function to output a character. Calls all |
| * active console's putc() handlers in succession. |
| * In : r0 - character to be printed |
| * Out: r0 - printed character on success, or < 0 |
| if at least one console had an error |
| * Clobber list : r0, r1, r2 |
| * --------------------------------------------- |
| */ |
| func console_putc |
| push {r4-r6, lr} |
| mov r5, #ERROR_NO_VALID_CONSOLE /* R5 = current return value */ |
| mov r4, r0 /* R4 = character to print */ |
| ldr r6, =console_list |
| ldr r6, [r6] /* R6 = first console struct */ |
| |
| putc_loop: |
| cmp r6, #0 |
| beq putc_done |
| ldr r1, =console_state |
| ldrb r1, [r1] |
| ldr r2, [r6, #CONSOLE_T_FLAGS] |
| tst r1, r2 |
| beq putc_continue |
| ldr r2, [r6, #CONSOLE_T_PUTC] |
| cmp r2, #0 |
| beq putc_continue |
| mov r0, r4 |
| mov r1, r6 |
| blx r2 |
| cmp r5, #ERROR_NO_VALID_CONSOLE /* update R5 if it's NOVALID */ |
| cmpne r0, #0 /* else update it if R0 < 0 */ |
| movlt r5, r0 |
| putc_continue: |
| ldr r6, [r6] /* R6 = next struct */ |
| b putc_loop |
| |
| putc_done: |
| mov r0, r5 |
| pop {r4-r6, pc} |
| endfunc console_putc |
| |
| /* --------------------------------------------- |
| * int console_getc(void) |
| * Function to get a character from any console. |
| * Keeps looping through all consoles' getc() |
| * handlers until one of them returns a |
| * character, then stops iterating and returns |
| * that character to the caller. Will stop looping |
| * if all active consoles report real errors |
| * (other than just not having a char available). |
| * Out : r0 - read character, or < 0 on error |
| * Clobber list : r0, r1 |
| * --------------------------------------------- |
| */ |
| func console_getc |
| push {r5-r6, lr} |
| getc_try_again: |
| mov r5, #ERROR_NO_VALID_CONSOLE /* R5 = current return value */ |
| ldr r6, =console_list |
| ldr r6, [r6] /* R6 = first console struct */ |
| cmp r6, #0 |
| bne getc_loop |
| mov r0, r5 /* If no consoles registered */ |
| pop {r5-r6, pc} /* return immediately. */ |
| |
| getc_loop: |
| ldr r0, =console_state |
| ldrb r0, [r0] |
| ldr r1, [r6, #CONSOLE_T_FLAGS] |
| tst r0, r1 |
| beq getc_continue |
| ldr r1, [r6, #CONSOLE_T_GETC] |
| cmp r1, #0 |
| beq getc_continue |
| mov r0, r6 |
| blx r1 |
| cmp r0, #0 /* if R0 >= 0: return */ |
| bge getc_found |
| cmp r5, #ERROR_NO_PENDING_CHAR /* may update R5 (NOCHAR has */ |
| movne r5, r0 /* precedence vs real errors) */ |
| getc_continue: |
| ldr r6, [r6] /* R6 = next struct */ |
| cmp r6, #0 |
| bne getc_loop |
| cmp r5, #ERROR_NO_PENDING_CHAR /* Keep scanning if at least */ |
| beq getc_try_again /* one console returns NOCHAR */ |
| mov r0, r5 |
| |
| getc_found: |
| pop {r5-r6, pc} |
| endfunc console_getc |
| |
| /* --------------------------------------------- |
| * int console_flush(void) |
| * Function to force a write of all buffered |
| * data that hasn't been output. Calls all |
| * console's flush() handlers in succession. |
| * Out: r0 - 0 on success, < 0 if at least one error |
| * Clobber list : r0, r1, r2 |
| * --------------------------------------------- |
| */ |
| func console_flush |
| push {r5-r6, lr} |
| mov r5, #ERROR_NO_VALID_CONSOLE /* R5 = current return value */ |
| ldr r6, =console_list |
| ldr r6, [r6] /* R6 = first console struct */ |
| |
| flush_loop: |
| cmp r6, #0 |
| beq flush_done |
| ldr r1, =console_state |
| ldrb r1, [r1] |
| ldr r2, [r6, #CONSOLE_T_FLAGS] |
| tst r1, r2 |
| beq flush_continue |
| ldr r1, [r6, #CONSOLE_T_FLUSH] |
| cmp r1, #0 |
| beq flush_continue |
| mov r0, r6 |
| blx r1 |
| cmp r5, #ERROR_NO_VALID_CONSOLE /* update R5 if it's NOVALID */ |
| cmpne r0, #0 /* else update it if R0 < 0 */ |
| movlt r5, r0 |
| flush_continue: |
| ldr r6, [r6] /* R6 = next struct */ |
| b flush_loop |
| |
| flush_done: |
| mov r0, r5 |
| pop {r5-r6, pc} |
| endfunc console_flush |