aboutsummaryrefslogtreecommitdiff
path: root/slip.c
blob: 3b448f4d176d231a344d85e242f885cc5db07233 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
/* SLIP send/recv code, from RFC 1055 */

#include <stdio.h>      /* perror */

#include "slip_internal.h"

/* SLIP special character codes
 */
#define END             0300    /* indicates end of packet */
#define ESC             0333    /* indicates byte stuffing */
#define ESC_END         0334    /* ESC ESC_END means END data byte */
#define ESC_ESC         0335    /* ESC ESC_ESC means ESC data byte */

/* SLIP_SEND: sends a packet of length "len", starting at
 * location "p".
 */
int hal_slip_send(const uint8_t * const ptr, const size_t len)
{
    int i;
    uint8_t *p = (uint8_t *)ptr;

#define check_send_char(c) if (hal_slip_send_char(c) == -1) return perror("write"), -1;

    /* send an initial END character to flush out any data that may
     * have accumulated in the receiver due to line noise
     */
    check_send_char(END);

    /* for each byte in the packet, send the appropriate character
     * sequence
     */
    for (i = 0; i < len; ++i) {
        switch (*p) {
            /* if it's the same code as an END character, we send a
             * special two character code so as not to make the
             * receiver think we sent an END
             */
        case END:
            check_send_char(ESC);
            check_send_char(ESC_END);
            break;

            /* if it's the same code as an ESC character,
             * we send a special two character code so as not
             * to make the receiver think we sent an ESC
             */
        case ESC:
            check_send_char(ESC);
            check_send_char(ESC_ESC);
            break;

            /* otherwise, we just send the character
             */
        default:
            check_send_char(*p);
        }

        p++;
    }

    /* tell the receiver that we're done sending the packet
     */
    check_send_char(END);

    return 0;
#undef check_send_char
}

/* SLIP_RECV: receives a packet into the buffer located at "p".
 *      If more than len bytes are received, the packet will
 *      be truncated.
 *      Returns the number of bytes stored in the buffer.
 */
int hal_slip_recv(uint8_t * const p, const size_t len)
{
    uint8_t c;
    size_t received = 0;

#define check_recv_char(c) if (hal_slip_recv_char(&c) == -1) return perror("read"), -1;

    /* sit in a loop reading bytes until we put together
     * a whole packet.
     * Make sure not to copy them into the packet if we
     * run out of room.
     */
    while (1) {
        /* get a character to process
         */
        check_recv_char(c);

        /* handle bytestuffing if necessary
         */
        switch (c) {

            /* if it's an END character then we're done with
             * the packet
             */
        case END:
            /* a minor optimization: if there is no
             * data in the packet, ignore it. This is
             * meant to avoid bothering IP with all
             * the empty packets generated by the
             * duplicate END characters which are in
             * turn sent to try to detect line noise.
             */
            if (received)
                return received;
            else
                break;

            /* if it's the same code as an ESC character, wait
             * and get another character and then figure out
             * what to store in the packet based on that.
             */
        case ESC:
            check_recv_char(c);

            /* if "c" is not one of these two, then we
             * have a protocol violation.  The best bet
             * seems to be to leave the byte alone and
             * just stuff it into the packet
             */
            switch(c) {
            case ESC_END:
                c = END;
                break;
            case ESC_ESC:
                c = ESC;
                break;
            }

            /* here we fall into the default handler and let
             * it store the character for us
             */
        default:
            if (received < len)
                p[received++] = c;
        }
    }
#undef check_recv_char
}
">buf[rb->widx] = c; if (++rb->widx >= sizeof(rb->buf)) rb->widx = 0; } static ringbuf_t uart_ringbuf; /* current character received from UART */ static uint8_t uart_rx; /* Semaphore to inform uart_cli_read that there's a new character. */ osSemaphoreId uart_sem; osSemaphoreDef(uart_sem); /* Callback for HAL_UART_Receive_DMA(). */ void HAL_UART1_RxCpltCallback(UART_HandleTypeDef *huart) { ringbuf_write_char(&uart_ringbuf, uart_rx); osSemaphoreRelease(uart_sem); HAL_UART_Receive_DMA(huart, &uart_rx, 1); } static void uart_cli_print(struct cli_def *cli __attribute__ ((unused)), const char *buf) { char crlf[] = "\r\n"; uart_send_string2(STM_UART_MGMT, buf); uart_send_string2(STM_UART_MGMT, crlf); } static int uart_cli_read(struct cli_def *cli __attribute__ ((unused)), void *buf, size_t count) { for (int i = 0; i < count; ++i) { while (ringbuf_read_char(&uart_ringbuf, (uint8_t *)(buf + i)) == 0) osSemaphoreWait(uart_sem, osWaitForever); } return count; } static int uart_cli_write(struct cli_def *cli __attribute__ ((unused)), const void *buf, size_t count) { uart_send_bytes(STM_UART_MGMT, (uint8_t *) buf, count); return (int) count; } int control_mgmt_uart_dma_rx(mgmt_cli_dma_state_t state) { if (state == DMA_RX_START) { if (uart_ringbuf.rx_state != DMA_RX_START) { ringbuf_init(&uart_ringbuf); HAL_UART_Receive_DMA(&huart_mgmt, &uart_rx, 1); uart_ringbuf.rx_state = DMA_RX_START; } return 1; } else if (state == DMA_RX_STOP) { if (HAL_UART_DMAStop(&huart_mgmt) != CMSIS_HAL_OK) return 0; uart_ringbuf.rx_state = DMA_RX_STOP; return 1; } return 0; } static int embedded_cli_loop(struct cli_def *cli) { unsigned char c; int n = 0; static struct cli_loop_ctx ctx; memset(&ctx, 0, sizeof(ctx)); ctx.insertmode = 1; cli->state = CLI_STATE_LOGIN; /* start off in unprivileged mode */ cli_set_privilege(cli, PRIVILEGE_UNPRIVILEGED); cli_set_configmode(cli, MODE_EXEC, NULL); cli_error(cli, "%s", cli->banner); while (1) { cli_loop_start_new_command(cli, &ctx); control_mgmt_uart_dma_rx(DMA_RX_START); while (1) { cli_loop_show_prompt(cli, &ctx); n = cli_loop_read_next_char(cli, &ctx, &c); /* cli_print(cli, "Next char: '%c'/%i, ringbuf ridx %i, widx %i", c, (int) c, uart_ringbuf.ridx, RINGBUF_WIDX(uart_ringbuf) */ if (n == CLI_LOOP_CTRL_BREAK) break; if (n == CLI_LOOP_CTRL_CONTINUE) continue; n = cli_loop_process_char(cli, &ctx, c); if (n == CLI_LOOP_CTRL_BREAK) break; if (n == CLI_LOOP_CTRL_CONTINUE) continue; } if (ctx.l < 0) break; /* cli_print(cli, "Process command: '%s'", ctx.cmd); */ n = cli_loop_process_cmd(cli, &ctx); if (n == CLI_LOOP_CTRL_BREAK) break; } return CLI_OK; } static void mgmt_cli_init(struct cli_def *cli) { cli_init(cli); cli_read_callback(cli, uart_cli_read); cli_write_callback(cli, uart_cli_write); cli_print_callback(cli, uart_cli_print); cli_set_banner(cli, "Cryptech Alpha"); cli_set_hostname(cli, "cryptech"); cli_telnet_protocol(cli, 0); } hal_user_t user; static int check_auth(const char *username, const char *password) { hal_client_handle_t client = { -1 }; /* PIN-based login */ if (strcmp(username, "wheel") == 0) user = HAL_USER_WHEEL; else if (strcmp(username, "so") == 0) user = HAL_USER_SO; else if (strcmp(username, "user") == 0) user = HAL_USER_NORMAL; else user = HAL_USER_NONE; if (hal_rpc_login(client, user, password, strlen(password)) == LIBHAL_OK) return CLI_OK; user = HAL_USER_NONE; return CLI_ERROR; } int cli_main(void) { static struct cli_def cli; uart_sem = osSemaphoreCreate(osSemaphore(uart_sem), 0); mgmt_cli_init(&cli); cli_set_auth_callback(&cli, check_auth); /* we don't have any privileged commands at the moment */ cli_unregister_command(&cli, "enable"); configure_cli_fpga(&cli); configure_cli_keystore(&cli); configure_cli_masterkey(&cli); configure_cli_firmware(&cli); configure_cli_bootloader(&cli); configure_cli_misc(&cli); while (1) { embedded_cli_loop(&cli); /* embedded_cli_loop returns when the user enters 'quit' or 'exit' */ cli_print(&cli, "\nLogging out...\n"); user = HAL_USER_NONE; } /*NOTREACHED*/ return -1; }