UART: Added a flag to disable stdio traces
[simavr] / simavr / sim / avr_uart.c
1 /*
2         avr_uart.c
3
4         Handles UART access
5         Right now just handle "write" to the serial port at any speed
6         and printf to the console when '\n' is written.
7
8         Copyright 2008, 2009 Michel Pollet <buserror@gmail.com>
9
10         This file is part of simavr.
11
12         simavr is free software: you can redistribute it and/or modify
13         it under the terms of the GNU General Public License as published by
14         the Free Software Foundation, either version 3 of the License, or
15         (at your option) any later version.
16
17         simavr is distributed in the hope that it will be useful,
18         but WITHOUT ANY WARRANTY; without even the implied warranty of
19         MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
20         GNU General Public License for more details.
21
22         You should have received a copy of the GNU General Public License
23         along with simavr.  If not, see <http://www.gnu.org/licenses/>.
24  */
25
26 #include <stdio.h>
27 #include <unistd.h>
28 #include "avr_uart.h"
29 #include "sim_hex.h"
30
31 DEFINE_FIFO(uint8_t, uart_fifo, 64);
32
33 static avr_cycle_count_t avr_uart_txc_raise(struct avr_t * avr, avr_cycle_count_t when, void * param)
34 {
35         avr_uart_t * p = (avr_uart_t *)param;
36         if (avr_regbit_get(avr, p->txen)) {
37                 // if the interrupts are not used, still raised the UDRE and TXC flaga
38                 avr_raise_interrupt(avr, &p->udrc);
39                 avr_raise_interrupt(avr, &p->txc);
40         }
41         return 0;
42 }
43
44 static avr_cycle_count_t avr_uart_rxc_raise(struct avr_t * avr, avr_cycle_count_t when, void * param)
45 {
46         avr_uart_t * p = (avr_uart_t *)param;
47         if (avr_regbit_get(avr, p->rxen))
48                 avr_raise_interrupt(avr, &p->rxc);
49         return 0;
50 }
51
52 static uint8_t avr_uart_rxc_read(struct avr_t * avr, avr_io_addr_t addr, void * param)
53 {
54         avr_uart_t * p = (avr_uart_t *)param;
55         uint8_t v = avr_core_watch_read(avr, addr);
56
57         //static uint8_t old = 0xff; if (v != old) printf("UCSRA read %02x\n", v); old = v;
58         //
59         // if RX is enabled, and there is nothing to read, and
60         // the AVR core is reading this register, it's probably
61         // to pool the RXC TXC flag and spinloop
62         // so here we introduce a usleep to make it a bit lighter
63         // on CPU and let data arrive
64         //
65         uint8_t ri = !avr_regbit_get(avr, p->rxen) || !avr_regbit_get(avr, p->rxc.raised);
66         uint8_t ti = !avr_regbit_get(avr, p->txen) || !avr_regbit_get(avr, p->txc.raised);
67
68         if (p->flags & AVR_UART_FLAG_POOL_SLEEP) {
69
70                 if (ri && ti)
71                         usleep(1);
72         }
73         // if reception is idle and the fifo is empty, tell whomever there is room
74         if (avr_regbit_get(avr, p->rxen) && uart_fifo_isempty(&p->input))
75                 avr_raise_irq(p->io.irq + UART_IRQ_OUT_XON, 1);
76
77         return v;
78 }
79
80 static uint8_t avr_uart_read(struct avr_t * avr, avr_io_addr_t addr, void * param)
81 {
82         avr_uart_t * p = (avr_uart_t *)param;
83
84         // clear the rxc bit in case the code is using pooling
85         avr_regbit_clear(avr, p->rxc.raised);
86
87         if (!avr_regbit_get(avr, p->rxen)) {
88                 avr->data[addr] = 0;
89                 // made to trigger potential watchpoints
90                 avr_core_watch_read(avr, addr);
91                 return 0;
92         }
93         uint8_t v = uart_fifo_read(&p->input);
94
95         //printf("UART read %02x %s\n", v, uart_fifo_isempty(&p->input) ? "EMPTY!" : "");
96         avr->data[addr] = v;
97         // made to trigger potential watchpoints
98         v = avr_core_watch_read(avr, addr);
99
100         if (!uart_fifo_isempty(&p->input))
101                 avr_cycle_timer_register_usec(avr, 100, avr_uart_rxc_raise, p); // should be uart speed dependent
102
103         return v;
104 }
105
106 static void avr_uart_baud_write(struct avr_t * avr, avr_io_addr_t addr, uint8_t v, void * param)
107 {
108         avr_uart_t * p = (avr_uart_t *)param;
109         avr_core_watch_write(avr, addr, v);
110         uint32_t val = avr->data[p->r_ubrrl] | (avr->data[p->r_ubrrh] << 8);
111         uint32_t baud = avr->frequency / (val+1);
112         if (avr_regbit_get(avr, p->u2x))
113                 baud /= 8;
114         else
115                 baud /= 16;
116         printf("UART-%c configured to %04x = %d baud\n", p->name, val, baud);
117 }
118
119 static void avr_uart_write(struct avr_t * avr, avr_io_addr_t addr, uint8_t v, void * param)
120 {
121         avr_uart_t * p = (avr_uart_t *)param;
122
123         if (addr == p->r_udr) {
124                 avr_core_watch_write(avr, addr, v);
125
126                 avr_regbit_clear(avr, p->udrc.raised);
127                 avr_cycle_timer_register_usec(avr, 100, avr_uart_txc_raise, p); // should be uart speed dependent
128
129                 if (p->flags & AVR_UART_FLAG_STDIO) {
130                         static char buf[128];
131                         static int l = 0;
132                         buf[l++] = v < ' ' ? '.' : v;
133                         buf[l] = 0;
134                         if (v == '\n' || l == 127) {
135                                 l = 0;
136                                 printf("\e[32m%s\e[0m\n", buf);
137                         }
138                 }
139         //      printf("UDR%c(%02x) = %02x\n", p->name, addr, v);
140                 // tell other modules we are "outputing" a byte
141                 if (avr_regbit_get(avr, p->txen))
142                         avr_raise_irq(p->io.irq + UART_IRQ_OUTPUT, v);
143         } else {
144                 // get the bits before the write
145                 uint8_t udre = avr_regbit_get(avr, p->udrc.raised);
146                 uint8_t txc = avr_regbit_get(avr, p->txc.raised);
147
148                 avr_core_watch_write(avr, addr, v);
149
150                 // if writing one to a one, clear bit
151                 if (udre && avr_regbit_get(avr, p->udrc.raised))
152                         avr_regbit_clear(avr, p->udrc.raised);
153                 if (txc && avr_regbit_get(avr, p->txc.raised))
154                         avr_regbit_clear(avr, p->txc.raised);
155         }
156 }
157
158 static void avr_uart_irq_input(struct avr_irq_t * irq, uint32_t value, void * param)
159 {
160         avr_uart_t * p = (avr_uart_t *)param;
161         avr_t * avr = p->io.avr;
162
163         // check to see fi receiver is enabled
164         if (!avr_regbit_get(avr, p->rxen))
165                 return;
166
167         if (uart_fifo_isempty(&p->input))
168                 avr_cycle_timer_register_usec(avr, 100, avr_uart_rxc_raise, p); // should be uart speed dependent
169         uart_fifo_write(&p->input, value); // add to fifo
170
171 //      printf("UART IRQ in %02x (%d/%d) %s\n", value, p->input.read, p->input.write, uart_fifo_isfull(&p->input) ? "FULL!!" : "");
172
173         if (uart_fifo_isfull(&p->input))
174                 avr_raise_irq(p->io.irq + UART_IRQ_OUT_XOFF, 1);
175 }
176
177
178 void avr_uart_reset(struct avr_io_t *io)
179 {
180         avr_uart_t * p = (avr_uart_t *)io;
181         avr_t * avr = p->io.avr;
182         avr_regbit_set(avr, p->udrc.raised);
183         avr_irq_register_notify(p->io.irq + UART_IRQ_INPUT, avr_uart_irq_input, p);
184         avr_cycle_timer_cancel(avr, avr_uart_rxc_raise, p);
185         avr_cycle_timer_cancel(avr, avr_uart_txc_raise, p);
186         uart_fifo_reset(&p->input);
187
188         // DEBUG allow printf without fidding with enabling the uart
189         avr_regbit_set(avr, p->txen);
190
191 }
192
193 #define AVR_IOCTL_UART_SET_FLAGS(_name) AVR_IOCTL_DEF('u','a','s',(_name))
194 #define AVR_IOCTL_UART_GET_FLAGS(_name) AVR_IOCTL_DEF('u','a','g',(_name))
195
196 static int avr_uart_ioctl(struct avr_io_t * port, uint32_t ctl, void * io_param)
197 {
198         avr_uart_t * p = (avr_uart_t *)port;
199         int res = -1;
200
201         if (!io_param)
202                 return res;
203
204         if (ctl == AVR_IOCTL_UART_SET_FLAGS(p->name)) {
205                 p->flags = *(uint32_t*)io_param;
206                 res = 0;
207         }
208         if (ctl == AVR_IOCTL_UART_GET_FLAGS(p->name)) {
209                 *(uint32_t*)io_param = p->flags;
210                 res = 0;
211         }
212
213         return res;
214 }
215
216
217 static  avr_io_t        _io = {
218         .kind = "uart",
219         .reset = avr_uart_reset,
220         .ioctl = avr_uart_ioctl,
221 };
222
223 void avr_uart_init(avr_t * avr, avr_uart_t * p)
224 {
225         p->io = _io;
226
227 //      printf("%s UART%c UDR=%02x\n", __FUNCTION__, p->name, p->r_udr);
228
229         // allocate this module's IRQ
230         p->io.irq_count = UART_IRQ_COUNT;
231         p->io.irq = avr_alloc_irq(0, p->io.irq_count);
232         p->io.irq_ioctl_get = AVR_IOCTL_UART_GETIRQ(p->name);
233
234         p->flags = AVR_UART_FLAG_POOL_SLEEP|AVR_UART_FLAG_STDIO;
235
236         avr_register_io(avr, &p->io);
237         avr_register_vector(avr, &p->rxc);
238         avr_register_vector(avr, &p->txc);
239         avr_register_vector(avr, &p->udrc);
240
241         avr_register_io_write(avr, p->r_udr, avr_uart_write, p);
242         avr_register_io_read(avr, p->r_udr, avr_uart_read, p);
243         // monitor code that reads the rxc flag, and delay it a bit
244         avr_register_io_read(avr, p->rxc.raised.reg, avr_uart_rxc_read, p);
245
246         avr_register_io_write(avr, p->r_ucsra, avr_uart_write, p);
247         avr_register_io_write(avr, p->r_ubrrl, avr_uart_baud_write, p);
248 }
249