LCOM1920-T5G03

#include <lcom/lcf.h>

#include "uart.h"

#include "queue.h"

#include "errors.h"

#define UART_BITRATE                            115200

#define UART_WAIT                               20 //microseconds

#define UART_RBR                                0

#define UART_THR                                0

#define UART_IER                                1

#define UART_IIR                                2

#define UART_FCR                                2

#define UART_LCR                                3

#define UART_MCR                                4

#define UART_LSR                                5

#define UART_MSR                                6

#define UART_SR                                 7

#define UART_DLL                                0

#define UART_DLM                                1

/// LCR

#define UART_BITS_PER_CHAR_POS                  0

#define UART_STOP_BITS_POS                      2

#define UART_PARITY_POS                         3

#define UART_BREAK_CONTROL_POS                  6

#define UART_DLAB_POS                           7

#define UART_BITS_PER_CHAR                      (BIT(0) | BIT(1))

#define UART_STOP_BITS                          (BIT(2))

#define UART_PARITY                             (BIT(3) | BIT(4) | BIT(5))

#define UART_BREAK_CONTROL                      (BIT(6))

#define UART_DLAB                               (BIT(7))

#define UART_GET_BITS_PER_CHAR(n)               (((n)&UART_BITS_PER_CHAR) + 5)

#define UART_GET_STOP_BITS(n)                   (((n)&UART_STOP_BITS)? 2 : 1)

#define UART_GET_PARITY(n)                      (((n)&UART_PARITY       )>>UART_PARITY_POS       )

#define UART_GET_BREAK_CONTROL(n)               (((n)&UART_BREAK_CONTROL)>>UART_BREAK_CONTROL_POS)

#define UART_GET_DLAB(n)                        (((n)&UART_DLAB         )>>UART_DLAB_POS         )

#define UART_GET_DIV_LATCH(m,l)                 ((m)<<8 | (l))

#define UART_GET_DLL(n)                         ((n)&0xFF)

#define UART_GET_DLM(n)                         (((n)>>8)&0xFF)

/// IER

#define UART_INT_EN_RX_POS                      0

#define UART_INT_EN_TX_POS                      1

#define UART_INT_EN_RECEIVER_LINE_STAT_POS      2

#define UART_INT_EN_MODEM_STAT_POS              3

#define UART_INT_EN_RX                          (BIT(0))

#define UART_INT_EN_TX                          (BIT(1))

#define UART_INT_EN_RECEIVER_LINE_STAT          (BIT(2))

#define UART_INT_EN_MODEM_STAT                  (BIT(3))

#define UART_GET_INT_EN_RX(n)                   (((n)&UART_INT_EN_RX                )>>UART_INT_EN_RX_POS                )

#define UART_GET_INT_EN_TX(n)                   (((n)&UART_INT_EN_TX                )>>UART_INT_EN_TX_POS                )

#define UART_GET_INT_EN_RECEIVER_LINE_STAT(n)   (((n)&UART_INT_EN_RECEIVER_LINE_STAT)>>UART_INT_EN_RECEIVER_LINE_STAT_POS)

#define UART_GET_INT_EN_MODEM_STAT(n)           (((n)&UART_INT_EN_MODEM_STAT        )>>UART_INT_EN_MODEM_STAT_POS        )

/// LSR

#define UART_RECEIVER_READY_POS                 0

#define UART_OVERRUN_ERROR_POS                  1

#define UART_PARITY_ERROR_POS                   2

#define UART_FRAMING_ERROR_POS                  3

#define UART_TRANSMITTER_EMPTY_POS              5

#define UART_RECEIVER_READY                     (BIT(0))

#define UART_OVERRUN_ERROR                      (BIT(1))

#define UART_PARITY_ERROR                       (BIT(2))

#define UART_FRAMING_ERROR                      (BIT(3))

#define UART_TRANSMITTER_EMPTY                  (BIT(5))

#define UART_GET_RECEIVER_READY(n)              (((n)&UART_RECEIVER_READY           )>>UART_RECEIVER_READY_POS           )

#define UART_GET_OVERRUN_ERROR                  (((n)&UART_OVERRUN_ERROR            )>>UART_OVERRUN_ERROR_POS            )

#define UART_GET_PARITY_ERROR                   (((n)&UART_PARITY_ERROR             )>>UART_PARITY_ERROR_POS             )

#define UART_GET_FRAMING_ERROR                  (((n)&UART_FRAMING_ERROR            )>>UART_FRAMING_ERROR_POS            )

#define UART_GET_TRANSMITTER_EMPTY(n)           (((n)&UART_TRANSMITTER_EMPTY        )>>UART_TRANSMITTER_EMPTY_POS        )

/// IIR

#define UART_INT_PEND_POS                       1

#define UART_INT_PEND                           (BIT(3)|BIT(2)|BIT(1))

#define UART_GET_IF_INT_PEND(n)                 (!((n)&1))

typedef enum {

    uart_int_receiver_line_stat = (         BIT(1) | BIT(0)),

    uart_int_rx                 = (         BIT(1)         ),

    uart_int_char_timeout_fifo  = (BIT(2) | BIT(1)         ),

    uart_int_tx                 = (                  BIT(0)),

    uart_int_modem_stat         = (0)

} uart_int_code;

#define UART_GET_INT_PEND(n)                    ((uart_int_code)(((n)&UART_INT_PEND)>>UART_INT_PEND_POS))

int (subscribe_uart_interrupt)(uint8_t interrupt_bit, int *interrupt_id) {

    if (interrupt_id == NULL) return 1;

    *interrupt_id = interrupt_bit;

    return (sys_irqsetpolicy(COM1_IRQ, IRQ_REENABLE | IRQ_EXCLUSIVE, interrupt_id));

}

static void uart_parse_config(uart_config *config){

    /// LCR

    config->bits_per_char          = UART_GET_BITS_PER_CHAR     (config->lcr);

    config->stop_bits              = UART_GET_STOP_BITS         (config->lcr);

    config->parity                 = UART_GET_PARITY            (config->lcr); if((config->parity & BIT(0)) == 0) config->parity = uart_parity_none;

    config->break_control          = UART_GET_BREAK_CONTROL     (config->lcr);

    config->dlab                   = UART_GET_DLAB              (config->lcr);

    /// IER

    config->received_data_int      = UART_GET_INT_EN_RX                (config->ier);

    config->transmitter_empty_int  = UART_GET_INT_EN_TX                (config->ier);

    config->receiver_line_stat_int = UART_GET_INT_EN_RECEIVER_LINE_STAT(config->ier);

    config->modem_stat_int         = UART_GET_INT_EN_MODEM_STAT        (config->ier);

    /// DIV LATCH

    config->divisor_latch          = UART_GET_DIV_LATCH(config->dlm, config->dll);

}

static int uart_get_lcr(int base_addr, uint8_t *p){

    return util_sys_inb(base_addr+UART_LCR, p);

}

static int uart_set_lcr(int base_addr, uint8_t config){

    if(sys_outb(base_addr+UART_LCR, config)) return WRITE_ERROR;

    return SUCCESS;

}

static int uart_get_lsr(int base_addr, uint8_t *p){

    return util_sys_inb(base_addr+UART_LSR, p);

}

static int uart_get_iir(int base_addr, uint8_t *p){

    return util_sys_inb(base_addr+UART_IIR, p);

}

static int uart_enable_divisor_latch(int base_addr){

    int ret = SUCCESS;

    uint8_t conf; if((ret = uart_get_lcr(base_addr, &conf))) return ret;

    return uart_set_lcr(base_addr, conf | UART_DLAB);

}

static int uart_disable_divisor_latch(int base_addr){

    int ret = SUCCESS;

    uint8_t conf; if((ret = uart_get_lcr(base_addr, &conf))) return ret;

    return uart_set_lcr(base_addr, conf & (~UART_DLAB));

}

static int uart_get_ier(int base_addr, uint8_t *p){

    int ret;

    if((ret = uart_disable_divisor_latch(base_addr))) return ret;

    return util_sys_inb(base_addr+UART_IER, p);

}

static int uart_set_ier(int base_addr, uint8_t n){

    int ret;

    if((ret = uart_disable_divisor_latch(base_addr))) return ret;

    if(sys_outb(base_addr+UART_IER, n)) return WRITE_ERROR;

    return SUCCESS;

}

int uart_get_config(int base_addr, uart_config *config){

    int ret = SUCCESS;

    config->base_addr = base_addr;

    if((ret = uart_get_lcr(base_addr, &config->lcr))) return ret;

    if((ret = uart_get_ier(base_addr, &config->ier))) return ret;

    if((ret = uart_enable_divisor_latch (base_addr))) return ret;

    if((ret = util_sys_inb(base_addr+UART_DLL, &config->dll   ))) return ret;

    if((ret = util_sys_inb(base_addr+UART_DLM, &config->dlm   ))) return ret;

    if((ret = uart_disable_divisor_latch(base_addr))) return ret;

    uart_parse_config(config);

    return ret;

}

void uart_print_config(uart_config config){

    printf("%s configuration:\n", (config.base_addr == COM1_ADDR ? "COM1" : "COM2"));

    printf("\tLCR = 0x%X: %d bits per char\t %d stop bits\t", config.lcr, config.bits_per_char, config.stop_bits);

    if((config.parity&BIT(0)) == 0) printf("NO parity\n");

    else switch(config.parity){

        case uart_parity_odd : printf("ODD parity\n"     ); break;

        case uart_parity_even: printf("EVEN parity\n"    ); break;

        case uart_parity_par1: printf("parity bit is 1\n"); break;

        case uart_parity_par0: printf("parity bit is 0\n"); break;

        default              : printf("invalid\n"        ); break;

    }

    printf("\tDLM = 0x%02X DLL=0x%02X: bitrate = %d bps\n", config.dlm, config.dll, UART_BITRATE/config.divisor_latch);

    printf("\tIER = 0x%02X: Rx interrupts: %s\tTx interrupts: %s\n", config.ier,

        (config.received_data_int     ? "ENABLED":"DISABLED"),

        (config.transmitter_empty_int ? "ENABLED":"DISABLED"));

}

int uart_set_bits_per_character(int base_addr, uint8_t bits_per_char){

    if(bits_per_char < 5 || bits_per_char > 8) return INVALID_ARG;

    int ret = SUCCESS;

    bits_per_char = (bits_per_char-5)&0x3;

    uint8_t conf; if((ret = uart_get_lcr(base_addr, &conf))) return ret;

    conf = (conf & (~UART_BITS_PER_CHAR)) | bits_per_char;

    return uart_set_lcr(base_addr, conf);

}

int uart_set_stop_bits(int base_addr, uint8_t stop){

    if(stop != 1 && stop != 2) return INVALID_ARG;

    int ret = SUCCESS;

    stop -= 1;

    stop = (stop&1)<<2;

    uint8_t conf; if((ret = uart_get_lcr(base_addr, &conf))) return ret;

    conf = (conf & (~UART_STOP_BITS)) | stop;

    return uart_set_lcr(base_addr, conf);

}

int uart_set_parity(int base_addr, uart_parity par){

    int ret = SUCCESS;

    uint8_t parity = par << 3;

    uint8_t conf; if((ret = uart_get_lcr(base_addr, &conf))) return ret;

    conf = (conf & (~UART_PARITY)) | parity;

    return uart_set_lcr(base_addr, conf);

}

int uart_set_bit_rate(int base_addr, float bit_rate){

    int ret = SUCCESS;

    uint16_t latch = UART_BITRATE/bit_rate;

    uint8_t dll = UART_GET_DLL(latch);

    uint8_t dlm = UART_GET_DLM(latch);

    if((ret = uart_enable_divisor_latch(base_addr))) return ret;

    if(sys_outb(base_addr+UART_DLL, dll)) return WRITE_ERROR;

    if(sys_outb(base_addr+UART_DLM, dlm)) return WRITE_ERROR;

    if((ret = uart_disable_divisor_latch(base_addr))) return ret;

    return SUCCESS;

}

static int uart_get_char(int base_addr, uint8_t *p){

    int ret;

    if((ret = uart_disable_divisor_latch(base_addr))) return ret;

    return util_sys_inb(base_addr+UART_RBR, p);

}

static int uart_send_char(int base_addr, uint8_t c){

    int ret;

    if((ret = uart_disable_divisor_latch(base_addr))) return ret;

    if(sys_outb(base_addr+UART_THR, c)) return WRITE_ERROR;

    return SUCCESS;

}

static int uart_receiver_ready(int base_addr){

    uint8_t lsr;

    if(uart_get_lsr(base_addr, &lsr)) return false;

    return UART_GET_RECEIVER_READY(lsr);

}

static int uart_transmitter_empty(int base_addr){

    uint8_t lsr;

    if(uart_get_lsr(base_addr, &lsr)) return false;

    return UART_GET_TRANSMITTER_EMPTY(lsr);

}

int uart_enable_int_rx(int base_addr){

    int ret;

    uint8_t ier;

    if((ret = uart_get_ier(base_addr, &ier))) return ret;

    ier |= UART_INT_EN_RX;

    return uart_set_ier(base_addr, ier);

}

int uart_disable_int_rx(int base_addr){

    int ret;

    uint8_t ier;

    if((ret = uart_get_ier(base_addr, &ier))) return ret;

    ier &= ~UART_INT_EN_RX;

    return uart_set_ier(base_addr, ier);

}

int uart_enable_int_tx(int base_addr){

    int ret;

    uint8_t ier;

    if((ret = uart_get_ier(base_addr, &ier))) return ret;

    ier |= UART_INT_EN_TX;

    return uart_set_ier(base_addr, ier);

}

int uart_disable_int_tx(int base_addr){

    int ret;

    uint8_t ier;

    if((ret = uart_get_ier(base_addr, &ier))) return ret;

    ier &= ~UART_INT_EN_TX;

    return uart_set_ier(base_addr, ier);

}

/// NCTP

#define NCTP_START      0x80

#define NCTP_END        0xFF

#define NCTP_OK         0xFF

#define NCTP_NOK        0x00

queue_t *out = NULL;

queue_t *in  = NULL;

void (*process)(const uint8_t*, const size_t);

int nctp_init(void (*proc_func)(const uint8_t*, const size_t)){

    out = queue_ctor(); if(out == NULL) return NULL_PTR;

    in  = queue_ctor(); if(in  == NULL) return NULL_PTR;

    process = proc_func;

    return SUCCESS;

}

int nctp_free(void){

    while(!queue_empty(out)){

        free(queue_top(out));

        queue_pop(out);

    }

    while(!queue_empty(in)){

        free(queue_top(in));

        queue_pop(in);

    }

    return SUCCESS;

}

int nctp_send(size_t num, uint8_t* ptr[], size_t sz[]){

    {

        size_t cnt = 0;

        for(size_t i = 0; i < num; ++i){

            cnt += sz[i];

            if(cnt > queue_max_size) return TRANS_REFUSED;

        }

    }

    int ret;

    uint8_t *tmp;

    tmp = malloc(sizeof(uint8_t)); *tmp = NCTP_START; queue_push(out, tmp);

    for(size_t i = 0; i < num; ++i){

        uint8_t *p = ptr[i]; size_t s = sz[i];

        for(size_t j = 0; j < s; ++j, ++p){

            tmp = malloc(sizeof(uint8_t)); *tmp = *p; queue_push(out, tmp);

        }

    }

    tmp = malloc(sizeof(uint8_t)); *tmp = NCTP_END; queue_push(out, tmp);

    if(uart_transmitter_empty(COM1_ADDR)){

        if((ret = uart_send_char(COM1_ADDR, *(uint8_t*)queue_top(out)))) return ret;

        queue_pop(out);

    }

    return SUCCESS;

}

static int nctp_transmit(void){

    if(!queue_empty(out)){

        int ret = uart_send_char(COM1_ADDR, *(uint8_t*)queue_top(out));

        queue_pop(out);

        return ret;

    }else return SUCCESS;

}

static void nctp_process_received(){

    free(queue_top(in)); queue_pop(in);

    size_t sz = 1024; uint8_t *p = malloc(sz*sizeof(uint8_t));

    size_t i = 0;

    while(*(uint8_t*)queue_top(in) != NCTP_END){

        //printf("%c\n", *(uint8_t*)queue_top(in));

        p[i++] = *(uint8_t*)queue_top(in);

        free(queue_top(in)); queue_pop(in);

        if(i >= sz) p = realloc(p, sz=2*sz);

    }

    free(queue_top(in)); queue_pop(in);

    process(p, i);

    free(p);

}

static int nctp_receive(void){

    int ret;

    uint8_t c;

    int num_ends = 0;

    while(uart_receiver_ready(COM1_ADDR)){

        if((ret = uart_get_char(COM1_ADDR, &c))) return ret;

        uint8_t *tmp = malloc(sizeof(uint8_t)); *tmp = c;

        queue_push(in, tmp);

        if(c == NCTP_END) ++num_ends;

    }

    while(num_ends-- > 0) nctp_process_received();

    return SUCCESS;

}

int nctp_ih_err = SUCCESS;

void nctp_ih(void){

    uint8_t iir;

    if((nctp_ih_err = uart_get_iir(COM1_ADDR, &iir))) return;

    if(UART_GET_IF_INT_PEND(iir)){

        switch(UART_GET_INT_PEND(iir)){

            case uart_int_rx: nctp_receive (); break;

            case uart_int_tx: nctp_transmit(); break;

            default: break;

        }

    }

}