Merge branch 'audit.b62' of git://git.kernel.org/pub/scm/linux/kernel/git/viro/audit...
[linux-2.6] / drivers / serial / atmel_serial.c
1 /*
2  *  linux/drivers/char/atmel_serial.c
3  *
4  *  Driver for Atmel AT91 / AT32 Serial ports
5  *  Copyright (C) 2003 Rick Bronson
6  *
7  *  Based on drivers/char/serial_sa1100.c, by Deep Blue Solutions Ltd.
8  *  Based on drivers/char/serial.c, by Linus Torvalds, Theodore Ts'o.
9  *
10  *  DMA support added by Chip Coldwell.
11  *
12  * This program 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 2 of the License, or
15  * (at your option) any later version.
16  *
17  * This program 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 this program; if not, write to the Free Software
24  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
25  *
26  */
27 #include <linux/module.h>
28 #include <linux/tty.h>
29 #include <linux/ioport.h>
30 #include <linux/slab.h>
31 #include <linux/init.h>
32 #include <linux/serial.h>
33 #include <linux/clk.h>
34 #include <linux/console.h>
35 #include <linux/sysrq.h>
36 #include <linux/tty_flip.h>
37 #include <linux/platform_device.h>
38 #include <linux/dma-mapping.h>
39 #include <linux/atmel_pdc.h>
40 #include <linux/atmel_serial.h>
41
42 #include <asm/io.h>
43
44 #include <asm/mach/serial_at91.h>
45 #include <mach/board.h>
46
47 #ifdef CONFIG_ARM
48 #include <mach/cpu.h>
49 #include <mach/gpio.h>
50 #endif
51
52 #define PDC_BUFFER_SIZE         512
53 /* Revisit: We should calculate this based on the actual port settings */
54 #define PDC_RX_TIMEOUT          (3 * 10)                /* 3 bytes */
55
56 #if defined(CONFIG_SERIAL_ATMEL_CONSOLE) && defined(CONFIG_MAGIC_SYSRQ)
57 #define SUPPORT_SYSRQ
58 #endif
59
60 #include <linux/serial_core.h>
61
62 #ifdef CONFIG_SERIAL_ATMEL_TTYAT
63
64 /* Use device name ttyAT, major 204 and minor 154-169.  This is necessary if we
65  * should coexist with the 8250 driver, such as if we have an external 16C550
66  * UART. */
67 #define SERIAL_ATMEL_MAJOR      204
68 #define MINOR_START             154
69 #define ATMEL_DEVICENAME        "ttyAT"
70
71 #else
72
73 /* Use device name ttyS, major 4, minor 64-68.  This is the usual serial port
74  * name, but it is legally reserved for the 8250 driver. */
75 #define SERIAL_ATMEL_MAJOR      TTY_MAJOR
76 #define MINOR_START             64
77 #define ATMEL_DEVICENAME        "ttyS"
78
79 #endif
80
81 #define ATMEL_ISR_PASS_LIMIT    256
82
83 /* UART registers. CR is write-only, hence no GET macro */
84 #define UART_PUT_CR(port,v)     __raw_writel(v, (port)->membase + ATMEL_US_CR)
85 #define UART_GET_MR(port)       __raw_readl((port)->membase + ATMEL_US_MR)
86 #define UART_PUT_MR(port,v)     __raw_writel(v, (port)->membase + ATMEL_US_MR)
87 #define UART_PUT_IER(port,v)    __raw_writel(v, (port)->membase + ATMEL_US_IER)
88 #define UART_PUT_IDR(port,v)    __raw_writel(v, (port)->membase + ATMEL_US_IDR)
89 #define UART_GET_IMR(port)      __raw_readl((port)->membase + ATMEL_US_IMR)
90 #define UART_GET_CSR(port)      __raw_readl((port)->membase + ATMEL_US_CSR)
91 #define UART_GET_CHAR(port)     __raw_readl((port)->membase + ATMEL_US_RHR)
92 #define UART_PUT_CHAR(port,v)   __raw_writel(v, (port)->membase + ATMEL_US_THR)
93 #define UART_GET_BRGR(port)     __raw_readl((port)->membase + ATMEL_US_BRGR)
94 #define UART_PUT_BRGR(port,v)   __raw_writel(v, (port)->membase + ATMEL_US_BRGR)
95 #define UART_PUT_RTOR(port,v)   __raw_writel(v, (port)->membase + ATMEL_US_RTOR)
96
97  /* PDC registers */
98 #define UART_PUT_PTCR(port,v)   __raw_writel(v, (port)->membase + ATMEL_PDC_PTCR)
99 #define UART_GET_PTSR(port)     __raw_readl((port)->membase + ATMEL_PDC_PTSR)
100
101 #define UART_PUT_RPR(port,v)    __raw_writel(v, (port)->membase + ATMEL_PDC_RPR)
102 #define UART_GET_RPR(port)      __raw_readl((port)->membase + ATMEL_PDC_RPR)
103 #define UART_PUT_RCR(port,v)    __raw_writel(v, (port)->membase + ATMEL_PDC_RCR)
104 #define UART_PUT_RNPR(port,v)   __raw_writel(v, (port)->membase + ATMEL_PDC_RNPR)
105 #define UART_PUT_RNCR(port,v)   __raw_writel(v, (port)->membase + ATMEL_PDC_RNCR)
106
107 #define UART_PUT_TPR(port,v)    __raw_writel(v, (port)->membase + ATMEL_PDC_TPR)
108 #define UART_PUT_TCR(port,v)    __raw_writel(v, (port)->membase + ATMEL_PDC_TCR)
109 #define UART_GET_TCR(port)      __raw_readl((port)->membase + ATMEL_PDC_TCR)
110
111 static int (*atmel_open_hook)(struct uart_port *);
112 static void (*atmel_close_hook)(struct uart_port *);
113
114 struct atmel_dma_buffer {
115         unsigned char   *buf;
116         dma_addr_t      dma_addr;
117         unsigned int    dma_size;
118         unsigned int    ofs;
119 };
120
121 struct atmel_uart_char {
122         u16             status;
123         u16             ch;
124 };
125
126 #define ATMEL_SERIAL_RINGSIZE 1024
127
128 /*
129  * We wrap our port structure around the generic uart_port.
130  */
131 struct atmel_uart_port {
132         struct uart_port        uart;           /* uart */
133         struct clk              *clk;           /* uart clock */
134         int                     may_wakeup;     /* cached value of device_may_wakeup for times we need to disable it */
135         u32                     backup_imr;     /* IMR saved during suspend */
136         int                     break_active;   /* break being received */
137
138         short                   use_dma_rx;     /* enable PDC receiver */
139         short                   pdc_rx_idx;     /* current PDC RX buffer */
140         struct atmel_dma_buffer pdc_rx[2];      /* PDC receier */
141
142         short                   use_dma_tx;     /* enable PDC transmitter */
143         struct atmel_dma_buffer pdc_tx;         /* PDC transmitter */
144
145         struct tasklet_struct   tasklet;
146         unsigned int            irq_status;
147         unsigned int            irq_status_prev;
148
149         struct circ_buf         rx_ring;
150 };
151
152 static struct atmel_uart_port atmel_ports[ATMEL_MAX_UART];
153
154 #ifdef SUPPORT_SYSRQ
155 static struct console atmel_console;
156 #endif
157
158 static inline struct atmel_uart_port *
159 to_atmel_uart_port(struct uart_port *uart)
160 {
161         return container_of(uart, struct atmel_uart_port, uart);
162 }
163
164 #ifdef CONFIG_SERIAL_ATMEL_PDC
165 static bool atmel_use_dma_rx(struct uart_port *port)
166 {
167         struct atmel_uart_port *atmel_port = to_atmel_uart_port(port);
168
169         return atmel_port->use_dma_rx;
170 }
171
172 static bool atmel_use_dma_tx(struct uart_port *port)
173 {
174         struct atmel_uart_port *atmel_port = to_atmel_uart_port(port);
175
176         return atmel_port->use_dma_tx;
177 }
178 #else
179 static bool atmel_use_dma_rx(struct uart_port *port)
180 {
181         return false;
182 }
183
184 static bool atmel_use_dma_tx(struct uart_port *port)
185 {
186         return false;
187 }
188 #endif
189
190 /*
191  * Return TIOCSER_TEMT when transmitter FIFO and Shift register is empty.
192  */
193 static u_int atmel_tx_empty(struct uart_port *port)
194 {
195         return (UART_GET_CSR(port) & ATMEL_US_TXEMPTY) ? TIOCSER_TEMT : 0;
196 }
197
198 /*
199  * Set state of the modem control output lines
200  */
201 static void atmel_set_mctrl(struct uart_port *port, u_int mctrl)
202 {
203         unsigned int control = 0;
204         unsigned int mode;
205
206 #ifdef CONFIG_ARCH_AT91RM9200
207         if (cpu_is_at91rm9200()) {
208                 /*
209                  * AT91RM9200 Errata #39: RTS0 is not internally connected
210                  * to PA21. We need to drive the pin manually.
211                  */
212                 if (port->mapbase == AT91RM9200_BASE_US0) {
213                         if (mctrl & TIOCM_RTS)
214                                 at91_set_gpio_value(AT91_PIN_PA21, 0);
215                         else
216                                 at91_set_gpio_value(AT91_PIN_PA21, 1);
217                 }
218         }
219 #endif
220
221         if (mctrl & TIOCM_RTS)
222                 control |= ATMEL_US_RTSEN;
223         else
224                 control |= ATMEL_US_RTSDIS;
225
226         if (mctrl & TIOCM_DTR)
227                 control |= ATMEL_US_DTREN;
228         else
229                 control |= ATMEL_US_DTRDIS;
230
231         UART_PUT_CR(port, control);
232
233         /* Local loopback mode? */
234         mode = UART_GET_MR(port) & ~ATMEL_US_CHMODE;
235         if (mctrl & TIOCM_LOOP)
236                 mode |= ATMEL_US_CHMODE_LOC_LOOP;
237         else
238                 mode |= ATMEL_US_CHMODE_NORMAL;
239         UART_PUT_MR(port, mode);
240 }
241
242 /*
243  * Get state of the modem control input lines
244  */
245 static u_int atmel_get_mctrl(struct uart_port *port)
246 {
247         unsigned int status, ret = 0;
248
249         status = UART_GET_CSR(port);
250
251         /*
252          * The control signals are active low.
253          */
254         if (!(status & ATMEL_US_DCD))
255                 ret |= TIOCM_CD;
256         if (!(status & ATMEL_US_CTS))
257                 ret |= TIOCM_CTS;
258         if (!(status & ATMEL_US_DSR))
259                 ret |= TIOCM_DSR;
260         if (!(status & ATMEL_US_RI))
261                 ret |= TIOCM_RI;
262
263         return ret;
264 }
265
266 /*
267  * Stop transmitting.
268  */
269 static void atmel_stop_tx(struct uart_port *port)
270 {
271         if (atmel_use_dma_tx(port)) {
272                 /* disable PDC transmit */
273                 UART_PUT_PTCR(port, ATMEL_PDC_TXTDIS);
274                 UART_PUT_IDR(port, ATMEL_US_ENDTX | ATMEL_US_TXBUFE);
275         } else
276                 UART_PUT_IDR(port, ATMEL_US_TXRDY);
277 }
278
279 /*
280  * Start transmitting.
281  */
282 static void atmel_start_tx(struct uart_port *port)
283 {
284         if (atmel_use_dma_tx(port)) {
285                 if (UART_GET_PTSR(port) & ATMEL_PDC_TXTEN)
286                         /* The transmitter is already running.  Yes, we
287                            really need this.*/
288                         return;
289
290                 UART_PUT_IER(port, ATMEL_US_ENDTX | ATMEL_US_TXBUFE);
291                 /* re-enable PDC transmit */
292                 UART_PUT_PTCR(port, ATMEL_PDC_TXTEN);
293         } else
294                 UART_PUT_IER(port, ATMEL_US_TXRDY);
295 }
296
297 /*
298  * Stop receiving - port is in process of being closed.
299  */
300 static void atmel_stop_rx(struct uart_port *port)
301 {
302         if (atmel_use_dma_rx(port)) {
303                 /* disable PDC receive */
304                 UART_PUT_PTCR(port, ATMEL_PDC_RXTDIS);
305                 UART_PUT_IDR(port, ATMEL_US_ENDRX | ATMEL_US_TIMEOUT);
306         } else
307                 UART_PUT_IDR(port, ATMEL_US_RXRDY);
308 }
309
310 /*
311  * Enable modem status interrupts
312  */
313 static void atmel_enable_ms(struct uart_port *port)
314 {
315         UART_PUT_IER(port, ATMEL_US_RIIC | ATMEL_US_DSRIC
316                         | ATMEL_US_DCDIC | ATMEL_US_CTSIC);
317 }
318
319 /*
320  * Control the transmission of a break signal
321  */
322 static void atmel_break_ctl(struct uart_port *port, int break_state)
323 {
324         if (break_state != 0)
325                 UART_PUT_CR(port, ATMEL_US_STTBRK);     /* start break */
326         else
327                 UART_PUT_CR(port, ATMEL_US_STPBRK);     /* stop break */
328 }
329
330 /*
331  * Stores the incoming character in the ring buffer
332  */
333 static void
334 atmel_buffer_rx_char(struct uart_port *port, unsigned int status,
335                      unsigned int ch)
336 {
337         struct atmel_uart_port *atmel_port = to_atmel_uart_port(port);
338         struct circ_buf *ring = &atmel_port->rx_ring;
339         struct atmel_uart_char *c;
340
341         if (!CIRC_SPACE(ring->head, ring->tail, ATMEL_SERIAL_RINGSIZE))
342                 /* Buffer overflow, ignore char */
343                 return;
344
345         c = &((struct atmel_uart_char *)ring->buf)[ring->head];
346         c->status       = status;
347         c->ch           = ch;
348
349         /* Make sure the character is stored before we update head. */
350         smp_wmb();
351
352         ring->head = (ring->head + 1) & (ATMEL_SERIAL_RINGSIZE - 1);
353 }
354
355 /*
356  * Deal with parity, framing and overrun errors.
357  */
358 static void atmel_pdc_rxerr(struct uart_port *port, unsigned int status)
359 {
360         /* clear error */
361         UART_PUT_CR(port, ATMEL_US_RSTSTA);
362
363         if (status & ATMEL_US_RXBRK) {
364                 /* ignore side-effect */
365                 status &= ~(ATMEL_US_PARE | ATMEL_US_FRAME);
366                 port->icount.brk++;
367         }
368         if (status & ATMEL_US_PARE)
369                 port->icount.parity++;
370         if (status & ATMEL_US_FRAME)
371                 port->icount.frame++;
372         if (status & ATMEL_US_OVRE)
373                 port->icount.overrun++;
374 }
375
376 /*
377  * Characters received (called from interrupt handler)
378  */
379 static void atmel_rx_chars(struct uart_port *port)
380 {
381         struct atmel_uart_port *atmel_port = to_atmel_uart_port(port);
382         unsigned int status, ch;
383
384         status = UART_GET_CSR(port);
385         while (status & ATMEL_US_RXRDY) {
386                 ch = UART_GET_CHAR(port);
387
388                 /*
389                  * note that the error handling code is
390                  * out of the main execution path
391                  */
392                 if (unlikely(status & (ATMEL_US_PARE | ATMEL_US_FRAME
393                                        | ATMEL_US_OVRE | ATMEL_US_RXBRK)
394                              || atmel_port->break_active)) {
395
396                         /* clear error */
397                         UART_PUT_CR(port, ATMEL_US_RSTSTA);
398
399                         if (status & ATMEL_US_RXBRK
400                             && !atmel_port->break_active) {
401                                 atmel_port->break_active = 1;
402                                 UART_PUT_IER(port, ATMEL_US_RXBRK);
403                         } else {
404                                 /*
405                                  * This is either the end-of-break
406                                  * condition or we've received at
407                                  * least one character without RXBRK
408                                  * being set. In both cases, the next
409                                  * RXBRK will indicate start-of-break.
410                                  */
411                                 UART_PUT_IDR(port, ATMEL_US_RXBRK);
412                                 status &= ~ATMEL_US_RXBRK;
413                                 atmel_port->break_active = 0;
414                         }
415                 }
416
417                 atmel_buffer_rx_char(port, status, ch);
418                 status = UART_GET_CSR(port);
419         }
420
421         tasklet_schedule(&atmel_port->tasklet);
422 }
423
424 /*
425  * Transmit characters (called from tasklet with TXRDY interrupt
426  * disabled)
427  */
428 static void atmel_tx_chars(struct uart_port *port)
429 {
430         struct circ_buf *xmit = &port->info->xmit;
431
432         if (port->x_char && UART_GET_CSR(port) & ATMEL_US_TXRDY) {
433                 UART_PUT_CHAR(port, port->x_char);
434                 port->icount.tx++;
435                 port->x_char = 0;
436         }
437         if (uart_circ_empty(xmit) || uart_tx_stopped(port))
438                 return;
439
440         while (UART_GET_CSR(port) & ATMEL_US_TXRDY) {
441                 UART_PUT_CHAR(port, xmit->buf[xmit->tail]);
442                 xmit->tail = (xmit->tail + 1) & (UART_XMIT_SIZE - 1);
443                 port->icount.tx++;
444                 if (uart_circ_empty(xmit))
445                         break;
446         }
447
448         if (uart_circ_chars_pending(xmit) < WAKEUP_CHARS)
449                 uart_write_wakeup(port);
450
451         if (!uart_circ_empty(xmit))
452                 UART_PUT_IER(port, ATMEL_US_TXRDY);
453 }
454
455 /*
456  * receive interrupt handler.
457  */
458 static void
459 atmel_handle_receive(struct uart_port *port, unsigned int pending)
460 {
461         struct atmel_uart_port *atmel_port = to_atmel_uart_port(port);
462
463         if (atmel_use_dma_rx(port)) {
464                 /*
465                  * PDC receive. Just schedule the tasklet and let it
466                  * figure out the details.
467                  *
468                  * TODO: We're not handling error flags correctly at
469                  * the moment.
470                  */
471                 if (pending & (ATMEL_US_ENDRX | ATMEL_US_TIMEOUT)) {
472                         UART_PUT_IDR(port, (ATMEL_US_ENDRX
473                                                 | ATMEL_US_TIMEOUT));
474                         tasklet_schedule(&atmel_port->tasklet);
475                 }
476
477                 if (pending & (ATMEL_US_RXBRK | ATMEL_US_OVRE |
478                                 ATMEL_US_FRAME | ATMEL_US_PARE))
479                         atmel_pdc_rxerr(port, pending);
480         }
481
482         /* Interrupt receive */
483         if (pending & ATMEL_US_RXRDY)
484                 atmel_rx_chars(port);
485         else if (pending & ATMEL_US_RXBRK) {
486                 /*
487                  * End of break detected. If it came along with a
488                  * character, atmel_rx_chars will handle it.
489                  */
490                 UART_PUT_CR(port, ATMEL_US_RSTSTA);
491                 UART_PUT_IDR(port, ATMEL_US_RXBRK);
492                 atmel_port->break_active = 0;
493         }
494 }
495
496 /*
497  * transmit interrupt handler. (Transmit is IRQF_NODELAY safe)
498  */
499 static void
500 atmel_handle_transmit(struct uart_port *port, unsigned int pending)
501 {
502         struct atmel_uart_port *atmel_port = to_atmel_uart_port(port);
503
504         if (atmel_use_dma_tx(port)) {
505                 /* PDC transmit */
506                 if (pending & (ATMEL_US_ENDTX | ATMEL_US_TXBUFE)) {
507                         UART_PUT_IDR(port, ATMEL_US_ENDTX | ATMEL_US_TXBUFE);
508                         tasklet_schedule(&atmel_port->tasklet);
509                 }
510         } else {
511                 /* Interrupt transmit */
512                 if (pending & ATMEL_US_TXRDY) {
513                         UART_PUT_IDR(port, ATMEL_US_TXRDY);
514                         tasklet_schedule(&atmel_port->tasklet);
515                 }
516         }
517 }
518
519 /*
520  * status flags interrupt handler.
521  */
522 static void
523 atmel_handle_status(struct uart_port *port, unsigned int pending,
524                     unsigned int status)
525 {
526         struct atmel_uart_port *atmel_port = to_atmel_uart_port(port);
527
528         if (pending & (ATMEL_US_RIIC | ATMEL_US_DSRIC | ATMEL_US_DCDIC
529                                 | ATMEL_US_CTSIC)) {
530                 atmel_port->irq_status = status;
531                 tasklet_schedule(&atmel_port->tasklet);
532         }
533 }
534
535 /*
536  * Interrupt handler
537  */
538 static irqreturn_t atmel_interrupt(int irq, void *dev_id)
539 {
540         struct uart_port *port = dev_id;
541         unsigned int status, pending, pass_counter = 0;
542
543         do {
544                 status = UART_GET_CSR(port);
545                 pending = status & UART_GET_IMR(port);
546                 if (!pending)
547                         break;
548
549                 atmel_handle_receive(port, pending);
550                 atmel_handle_status(port, pending, status);
551                 atmel_handle_transmit(port, pending);
552         } while (pass_counter++ < ATMEL_ISR_PASS_LIMIT);
553
554         return pass_counter ? IRQ_HANDLED : IRQ_NONE;
555 }
556
557 /*
558  * Called from tasklet with ENDTX and TXBUFE interrupts disabled.
559  */
560 static void atmel_tx_dma(struct uart_port *port)
561 {
562         struct atmel_uart_port *atmel_port = to_atmel_uart_port(port);
563         struct circ_buf *xmit = &port->info->xmit;
564         struct atmel_dma_buffer *pdc = &atmel_port->pdc_tx;
565         int count;
566
567         /* nothing left to transmit? */
568         if (UART_GET_TCR(port))
569                 return;
570
571         xmit->tail += pdc->ofs;
572         xmit->tail &= UART_XMIT_SIZE - 1;
573
574         port->icount.tx += pdc->ofs;
575         pdc->ofs = 0;
576
577         /* more to transmit - setup next transfer */
578
579         /* disable PDC transmit */
580         UART_PUT_PTCR(port, ATMEL_PDC_TXTDIS);
581
582         if (!uart_circ_empty(xmit) && !uart_tx_stopped(port)) {
583                 dma_sync_single_for_device(port->dev,
584                                            pdc->dma_addr,
585                                            pdc->dma_size,
586                                            DMA_TO_DEVICE);
587
588                 count = CIRC_CNT_TO_END(xmit->head, xmit->tail, UART_XMIT_SIZE);
589                 pdc->ofs = count;
590
591                 UART_PUT_TPR(port, pdc->dma_addr + xmit->tail);
592                 UART_PUT_TCR(port, count);
593                 /* re-enable PDC transmit and interrupts */
594                 UART_PUT_PTCR(port, ATMEL_PDC_TXTEN);
595                 UART_PUT_IER(port, ATMEL_US_ENDTX | ATMEL_US_TXBUFE);
596         }
597
598         if (uart_circ_chars_pending(xmit) < WAKEUP_CHARS)
599                 uart_write_wakeup(port);
600 }
601
602 static void atmel_rx_from_ring(struct uart_port *port)
603 {
604         struct atmel_uart_port *atmel_port = to_atmel_uart_port(port);
605         struct circ_buf *ring = &atmel_port->rx_ring;
606         unsigned int flg;
607         unsigned int status;
608
609         while (ring->head != ring->tail) {
610                 struct atmel_uart_char c;
611
612                 /* Make sure c is loaded after head. */
613                 smp_rmb();
614
615                 c = ((struct atmel_uart_char *)ring->buf)[ring->tail];
616
617                 ring->tail = (ring->tail + 1) & (ATMEL_SERIAL_RINGSIZE - 1);
618
619                 port->icount.rx++;
620                 status = c.status;
621                 flg = TTY_NORMAL;
622
623                 /*
624                  * note that the error handling code is
625                  * out of the main execution path
626                  */
627                 if (unlikely(status & (ATMEL_US_PARE | ATMEL_US_FRAME
628                                        | ATMEL_US_OVRE | ATMEL_US_RXBRK))) {
629                         if (status & ATMEL_US_RXBRK) {
630                                 /* ignore side-effect */
631                                 status &= ~(ATMEL_US_PARE | ATMEL_US_FRAME);
632
633                                 port->icount.brk++;
634                                 if (uart_handle_break(port))
635                                         continue;
636                         }
637                         if (status & ATMEL_US_PARE)
638                                 port->icount.parity++;
639                         if (status & ATMEL_US_FRAME)
640                                 port->icount.frame++;
641                         if (status & ATMEL_US_OVRE)
642                                 port->icount.overrun++;
643
644                         status &= port->read_status_mask;
645
646                         if (status & ATMEL_US_RXBRK)
647                                 flg = TTY_BREAK;
648                         else if (status & ATMEL_US_PARE)
649                                 flg = TTY_PARITY;
650                         else if (status & ATMEL_US_FRAME)
651                                 flg = TTY_FRAME;
652                 }
653
654
655                 if (uart_handle_sysrq_char(port, c.ch))
656                         continue;
657
658                 uart_insert_char(port, status, ATMEL_US_OVRE, c.ch, flg);
659         }
660
661         /*
662          * Drop the lock here since it might end up calling
663          * uart_start(), which takes the lock.
664          */
665         spin_unlock(&port->lock);
666         tty_flip_buffer_push(port->info->port.tty);
667         spin_lock(&port->lock);
668 }
669
670 static void atmel_rx_from_dma(struct uart_port *port)
671 {
672         struct atmel_uart_port *atmel_port = to_atmel_uart_port(port);
673         struct tty_struct *tty = port->info->port.tty;
674         struct atmel_dma_buffer *pdc;
675         int rx_idx = atmel_port->pdc_rx_idx;
676         unsigned int head;
677         unsigned int tail;
678         unsigned int count;
679
680         do {
681                 /* Reset the UART timeout early so that we don't miss one */
682                 UART_PUT_CR(port, ATMEL_US_STTTO);
683
684                 pdc = &atmel_port->pdc_rx[rx_idx];
685                 head = UART_GET_RPR(port) - pdc->dma_addr;
686                 tail = pdc->ofs;
687
688                 /* If the PDC has switched buffers, RPR won't contain
689                  * any address within the current buffer. Since head
690                  * is unsigned, we just need a one-way comparison to
691                  * find out.
692                  *
693                  * In this case, we just need to consume the entire
694                  * buffer and resubmit it for DMA. This will clear the
695                  * ENDRX bit as well, so that we can safely re-enable
696                  * all interrupts below.
697                  */
698                 head = min(head, pdc->dma_size);
699
700                 if (likely(head != tail)) {
701                         dma_sync_single_for_cpu(port->dev, pdc->dma_addr,
702                                         pdc->dma_size, DMA_FROM_DEVICE);
703
704                         /*
705                          * head will only wrap around when we recycle
706                          * the DMA buffer, and when that happens, we
707                          * explicitly set tail to 0. So head will
708                          * always be greater than tail.
709                          */
710                         count = head - tail;
711
712                         tty_insert_flip_string(tty, pdc->buf + pdc->ofs, count);
713
714                         dma_sync_single_for_device(port->dev, pdc->dma_addr,
715                                         pdc->dma_size, DMA_FROM_DEVICE);
716
717                         port->icount.rx += count;
718                         pdc->ofs = head;
719                 }
720
721                 /*
722                  * If the current buffer is full, we need to check if
723                  * the next one contains any additional data.
724                  */
725                 if (head >= pdc->dma_size) {
726                         pdc->ofs = 0;
727                         UART_PUT_RNPR(port, pdc->dma_addr);
728                         UART_PUT_RNCR(port, pdc->dma_size);
729
730                         rx_idx = !rx_idx;
731                         atmel_port->pdc_rx_idx = rx_idx;
732                 }
733         } while (head >= pdc->dma_size);
734
735         /*
736          * Drop the lock here since it might end up calling
737          * uart_start(), which takes the lock.
738          */
739         spin_unlock(&port->lock);
740         tty_flip_buffer_push(tty);
741         spin_lock(&port->lock);
742
743         UART_PUT_IER(port, ATMEL_US_ENDRX | ATMEL_US_TIMEOUT);
744 }
745
746 /*
747  * tasklet handling tty stuff outside the interrupt handler.
748  */
749 static void atmel_tasklet_func(unsigned long data)
750 {
751         struct uart_port *port = (struct uart_port *)data;
752         struct atmel_uart_port *atmel_port = to_atmel_uart_port(port);
753         unsigned int status;
754         unsigned int status_change;
755
756         /* The interrupt handler does not take the lock */
757         spin_lock(&port->lock);
758
759         if (atmel_use_dma_tx(port))
760                 atmel_tx_dma(port);
761         else
762                 atmel_tx_chars(port);
763
764         status = atmel_port->irq_status;
765         status_change = status ^ atmel_port->irq_status_prev;
766
767         if (status_change & (ATMEL_US_RI | ATMEL_US_DSR
768                                 | ATMEL_US_DCD | ATMEL_US_CTS)) {
769                 /* TODO: All reads to CSR will clear these interrupts! */
770                 if (status_change & ATMEL_US_RI)
771                         port->icount.rng++;
772                 if (status_change & ATMEL_US_DSR)
773                         port->icount.dsr++;
774                 if (status_change & ATMEL_US_DCD)
775                         uart_handle_dcd_change(port, !(status & ATMEL_US_DCD));
776                 if (status_change & ATMEL_US_CTS)
777                         uart_handle_cts_change(port, !(status & ATMEL_US_CTS));
778
779                 wake_up_interruptible(&port->info->delta_msr_wait);
780
781                 atmel_port->irq_status_prev = status;
782         }
783
784         if (atmel_use_dma_rx(port))
785                 atmel_rx_from_dma(port);
786         else
787                 atmel_rx_from_ring(port);
788
789         spin_unlock(&port->lock);
790 }
791
792 /*
793  * Perform initialization and enable port for reception
794  */
795 static int atmel_startup(struct uart_port *port)
796 {
797         struct atmel_uart_port *atmel_port = to_atmel_uart_port(port);
798         struct tty_struct *tty = port->info->port.tty;
799         int retval;
800
801         /*
802          * Ensure that no interrupts are enabled otherwise when
803          * request_irq() is called we could get stuck trying to
804          * handle an unexpected interrupt
805          */
806         UART_PUT_IDR(port, -1);
807
808         /*
809          * Allocate the IRQ
810          */
811         retval = request_irq(port->irq, atmel_interrupt, IRQF_SHARED,
812                         tty ? tty->name : "atmel_serial", port);
813         if (retval) {
814                 printk("atmel_serial: atmel_startup - Can't get irq\n");
815                 return retval;
816         }
817
818         /*
819          * Initialize DMA (if necessary)
820          */
821         if (atmel_use_dma_rx(port)) {
822                 int i;
823
824                 for (i = 0; i < 2; i++) {
825                         struct atmel_dma_buffer *pdc = &atmel_port->pdc_rx[i];
826
827                         pdc->buf = kmalloc(PDC_BUFFER_SIZE, GFP_KERNEL);
828                         if (pdc->buf == NULL) {
829                                 if (i != 0) {
830                                         dma_unmap_single(port->dev,
831                                                 atmel_port->pdc_rx[0].dma_addr,
832                                                 PDC_BUFFER_SIZE,
833                                                 DMA_FROM_DEVICE);
834                                         kfree(atmel_port->pdc_rx[0].buf);
835                                 }
836                                 free_irq(port->irq, port);
837                                 return -ENOMEM;
838                         }
839                         pdc->dma_addr = dma_map_single(port->dev,
840                                                        pdc->buf,
841                                                        PDC_BUFFER_SIZE,
842                                                        DMA_FROM_DEVICE);
843                         pdc->dma_size = PDC_BUFFER_SIZE;
844                         pdc->ofs = 0;
845                 }
846
847                 atmel_port->pdc_rx_idx = 0;
848
849                 UART_PUT_RPR(port, atmel_port->pdc_rx[0].dma_addr);
850                 UART_PUT_RCR(port, PDC_BUFFER_SIZE);
851
852                 UART_PUT_RNPR(port, atmel_port->pdc_rx[1].dma_addr);
853                 UART_PUT_RNCR(port, PDC_BUFFER_SIZE);
854         }
855         if (atmel_use_dma_tx(port)) {
856                 struct atmel_dma_buffer *pdc = &atmel_port->pdc_tx;
857                 struct circ_buf *xmit = &port->info->xmit;
858
859                 pdc->buf = xmit->buf;
860                 pdc->dma_addr = dma_map_single(port->dev,
861                                                pdc->buf,
862                                                UART_XMIT_SIZE,
863                                                DMA_TO_DEVICE);
864                 pdc->dma_size = UART_XMIT_SIZE;
865                 pdc->ofs = 0;
866         }
867
868         /*
869          * If there is a specific "open" function (to register
870          * control line interrupts)
871          */
872         if (atmel_open_hook) {
873                 retval = atmel_open_hook(port);
874                 if (retval) {
875                         free_irq(port->irq, port);
876                         return retval;
877                 }
878         }
879
880         /* Save current CSR for comparison in atmel_tasklet_func() */
881         atmel_port->irq_status_prev = UART_GET_CSR(port);
882         atmel_port->irq_status = atmel_port->irq_status_prev;
883
884         /*
885          * Finally, enable the serial port
886          */
887         UART_PUT_CR(port, ATMEL_US_RSTSTA | ATMEL_US_RSTRX);
888         /* enable xmit & rcvr */
889         UART_PUT_CR(port, ATMEL_US_TXEN | ATMEL_US_RXEN);
890
891         if (atmel_use_dma_rx(port)) {
892                 /* set UART timeout */
893                 UART_PUT_RTOR(port, PDC_RX_TIMEOUT);
894                 UART_PUT_CR(port, ATMEL_US_STTTO);
895
896                 UART_PUT_IER(port, ATMEL_US_ENDRX | ATMEL_US_TIMEOUT);
897                 /* enable PDC controller */
898                 UART_PUT_PTCR(port, ATMEL_PDC_RXTEN);
899         } else {
900                 /* enable receive only */
901                 UART_PUT_IER(port, ATMEL_US_RXRDY);
902         }
903
904         return 0;
905 }
906
907 /*
908  * Disable the port
909  */
910 static void atmel_shutdown(struct uart_port *port)
911 {
912         struct atmel_uart_port *atmel_port = to_atmel_uart_port(port);
913         /*
914          * Ensure everything is stopped.
915          */
916         atmel_stop_rx(port);
917         atmel_stop_tx(port);
918
919         /*
920          * Shut-down the DMA.
921          */
922         if (atmel_use_dma_rx(port)) {
923                 int i;
924
925                 for (i = 0; i < 2; i++) {
926                         struct atmel_dma_buffer *pdc = &atmel_port->pdc_rx[i];
927
928                         dma_unmap_single(port->dev,
929                                          pdc->dma_addr,
930                                          pdc->dma_size,
931                                          DMA_FROM_DEVICE);
932                         kfree(pdc->buf);
933                 }
934         }
935         if (atmel_use_dma_tx(port)) {
936                 struct atmel_dma_buffer *pdc = &atmel_port->pdc_tx;
937
938                 dma_unmap_single(port->dev,
939                                  pdc->dma_addr,
940                                  pdc->dma_size,
941                                  DMA_TO_DEVICE);
942         }
943
944         /*
945          * Disable all interrupts, port and break condition.
946          */
947         UART_PUT_CR(port, ATMEL_US_RSTSTA);
948         UART_PUT_IDR(port, -1);
949
950         /*
951          * Free the interrupt
952          */
953         free_irq(port->irq, port);
954
955         /*
956          * If there is a specific "close" function (to unregister
957          * control line interrupts)
958          */
959         if (atmel_close_hook)
960                 atmel_close_hook(port);
961 }
962
963 /*
964  * Flush any TX data submitted for DMA. Called when the TX circular
965  * buffer is reset.
966  */
967 static void atmel_flush_buffer(struct uart_port *port)
968 {
969         struct atmel_uart_port *atmel_port = to_atmel_uart_port(port);
970
971         if (atmel_use_dma_tx(port)) {
972                 UART_PUT_TCR(port, 0);
973                 atmel_port->pdc_tx.ofs = 0;
974         }
975 }
976
977 /*
978  * Power / Clock management.
979  */
980 static void atmel_serial_pm(struct uart_port *port, unsigned int state,
981                             unsigned int oldstate)
982 {
983         struct atmel_uart_port *atmel_port = to_atmel_uart_port(port);
984
985         switch (state) {
986         case 0:
987                 /*
988                  * Enable the peripheral clock for this serial port.
989                  * This is called on uart_open() or a resume event.
990                  */
991                 clk_enable(atmel_port->clk);
992
993                 /* re-enable interrupts if we disabled some on suspend */
994                 UART_PUT_IER(port, atmel_port->backup_imr);
995                 break;
996         case 3:
997                 /* Back up the interrupt mask and disable all interrupts */
998                 atmel_port->backup_imr = UART_GET_IMR(port);
999                 UART_PUT_IDR(port, -1);
1000
1001                 /*
1002                  * Disable the peripheral clock for this serial port.
1003                  * This is called on uart_close() or a suspend event.
1004                  */
1005                 clk_disable(atmel_port->clk);
1006                 break;
1007         default:
1008                 printk(KERN_ERR "atmel_serial: unknown pm %d\n", state);
1009         }
1010 }
1011
1012 /*
1013  * Change the port parameters
1014  */
1015 static void atmel_set_termios(struct uart_port *port, struct ktermios *termios,
1016                               struct ktermios *old)
1017 {
1018         unsigned long flags;
1019         unsigned int mode, imr, quot, baud;
1020
1021         /* Get current mode register */
1022         mode = UART_GET_MR(port) & ~(ATMEL_US_USCLKS | ATMEL_US_CHRL
1023                                         | ATMEL_US_NBSTOP | ATMEL_US_PAR
1024                                         | ATMEL_US_USMODE);
1025
1026         baud = uart_get_baud_rate(port, termios, old, 0, port->uartclk / 16);
1027         quot = uart_get_divisor(port, baud);
1028
1029         if (quot > 65535) {     /* BRGR is 16-bit, so switch to slower clock */
1030                 quot /= 8;
1031                 mode |= ATMEL_US_USCLKS_MCK_DIV8;
1032         }
1033
1034         /* byte size */
1035         switch (termios->c_cflag & CSIZE) {
1036         case CS5:
1037                 mode |= ATMEL_US_CHRL_5;
1038                 break;
1039         case CS6:
1040                 mode |= ATMEL_US_CHRL_6;
1041                 break;
1042         case CS7:
1043                 mode |= ATMEL_US_CHRL_7;
1044                 break;
1045         default:
1046                 mode |= ATMEL_US_CHRL_8;
1047                 break;
1048         }
1049
1050         /* stop bits */
1051         if (termios->c_cflag & CSTOPB)
1052                 mode |= ATMEL_US_NBSTOP_2;
1053
1054         /* parity */
1055         if (termios->c_cflag & PARENB) {
1056                 /* Mark or Space parity */
1057                 if (termios->c_cflag & CMSPAR) {
1058                         if (termios->c_cflag & PARODD)
1059                                 mode |= ATMEL_US_PAR_MARK;
1060                         else
1061                                 mode |= ATMEL_US_PAR_SPACE;
1062                 } else if (termios->c_cflag & PARODD)
1063                         mode |= ATMEL_US_PAR_ODD;
1064                 else
1065                         mode |= ATMEL_US_PAR_EVEN;
1066         } else
1067                 mode |= ATMEL_US_PAR_NONE;
1068
1069         /* hardware handshake (RTS/CTS) */
1070         if (termios->c_cflag & CRTSCTS)
1071                 mode |= ATMEL_US_USMODE_HWHS;
1072         else
1073                 mode |= ATMEL_US_USMODE_NORMAL;
1074
1075         spin_lock_irqsave(&port->lock, flags);
1076
1077         port->read_status_mask = ATMEL_US_OVRE;
1078         if (termios->c_iflag & INPCK)
1079                 port->read_status_mask |= (ATMEL_US_FRAME | ATMEL_US_PARE);
1080         if (termios->c_iflag & (BRKINT | PARMRK))
1081                 port->read_status_mask |= ATMEL_US_RXBRK;
1082
1083         if (atmel_use_dma_rx(port))
1084                 /* need to enable error interrupts */
1085                 UART_PUT_IER(port, port->read_status_mask);
1086
1087         /*
1088          * Characters to ignore
1089          */
1090         port->ignore_status_mask = 0;
1091         if (termios->c_iflag & IGNPAR)
1092                 port->ignore_status_mask |= (ATMEL_US_FRAME | ATMEL_US_PARE);
1093         if (termios->c_iflag & IGNBRK) {
1094                 port->ignore_status_mask |= ATMEL_US_RXBRK;
1095                 /*
1096                  * If we're ignoring parity and break indicators,
1097                  * ignore overruns too (for real raw support).
1098                  */
1099                 if (termios->c_iflag & IGNPAR)
1100                         port->ignore_status_mask |= ATMEL_US_OVRE;
1101         }
1102         /* TODO: Ignore all characters if CREAD is set.*/
1103
1104         /* update the per-port timeout */
1105         uart_update_timeout(port, termios->c_cflag, baud);
1106
1107         /* save/disable interrupts and drain transmitter */
1108         imr = UART_GET_IMR(port);
1109         UART_PUT_IDR(port, -1);
1110         while (!(UART_GET_CSR(port) & ATMEL_US_TXEMPTY))
1111                 cpu_relax();
1112
1113         /* disable receiver and transmitter */
1114         UART_PUT_CR(port, ATMEL_US_TXDIS | ATMEL_US_RXDIS);
1115
1116         /* set the parity, stop bits and data size */
1117         UART_PUT_MR(port, mode);
1118
1119         /* set the baud rate */
1120         UART_PUT_BRGR(port, quot);
1121         UART_PUT_CR(port, ATMEL_US_RSTSTA | ATMEL_US_RSTRX);
1122         UART_PUT_CR(port, ATMEL_US_TXEN | ATMEL_US_RXEN);
1123
1124         /* restore interrupts */
1125         UART_PUT_IER(port, imr);
1126
1127         /* CTS flow-control and modem-status interrupts */
1128         if (UART_ENABLE_MS(port, termios->c_cflag))
1129                 port->ops->enable_ms(port);
1130
1131         spin_unlock_irqrestore(&port->lock, flags);
1132 }
1133
1134 /*
1135  * Return string describing the specified port
1136  */
1137 static const char *atmel_type(struct uart_port *port)
1138 {
1139         return (port->type == PORT_ATMEL) ? "ATMEL_SERIAL" : NULL;
1140 }
1141
1142 /*
1143  * Release the memory region(s) being used by 'port'.
1144  */
1145 static void atmel_release_port(struct uart_port *port)
1146 {
1147         struct platform_device *pdev = to_platform_device(port->dev);
1148         int size = pdev->resource[0].end - pdev->resource[0].start + 1;
1149
1150         release_mem_region(port->mapbase, size);
1151
1152         if (port->flags & UPF_IOREMAP) {
1153                 iounmap(port->membase);
1154                 port->membase = NULL;
1155         }
1156 }
1157
1158 /*
1159  * Request the memory region(s) being used by 'port'.
1160  */
1161 static int atmel_request_port(struct uart_port *port)
1162 {
1163         struct platform_device *pdev = to_platform_device(port->dev);
1164         int size = pdev->resource[0].end - pdev->resource[0].start + 1;
1165
1166         if (!request_mem_region(port->mapbase, size, "atmel_serial"))
1167                 return -EBUSY;
1168
1169         if (port->flags & UPF_IOREMAP) {
1170                 port->membase = ioremap(port->mapbase, size);
1171                 if (port->membase == NULL) {
1172                         release_mem_region(port->mapbase, size);
1173                         return -ENOMEM;
1174                 }
1175         }
1176
1177         return 0;
1178 }
1179
1180 /*
1181  * Configure/autoconfigure the port.
1182  */
1183 static void atmel_config_port(struct uart_port *port, int flags)
1184 {
1185         if (flags & UART_CONFIG_TYPE) {
1186                 port->type = PORT_ATMEL;
1187                 atmel_request_port(port);
1188         }
1189 }
1190
1191 /*
1192  * Verify the new serial_struct (for TIOCSSERIAL).
1193  */
1194 static int atmel_verify_port(struct uart_port *port, struct serial_struct *ser)
1195 {
1196         int ret = 0;
1197         if (ser->type != PORT_UNKNOWN && ser->type != PORT_ATMEL)
1198                 ret = -EINVAL;
1199         if (port->irq != ser->irq)
1200                 ret = -EINVAL;
1201         if (ser->io_type != SERIAL_IO_MEM)
1202                 ret = -EINVAL;
1203         if (port->uartclk / 16 != ser->baud_base)
1204                 ret = -EINVAL;
1205         if ((void *)port->mapbase != ser->iomem_base)
1206                 ret = -EINVAL;
1207         if (port->iobase != ser->port)
1208                 ret = -EINVAL;
1209         if (ser->hub6 != 0)
1210                 ret = -EINVAL;
1211         return ret;
1212 }
1213
1214 static struct uart_ops atmel_pops = {
1215         .tx_empty       = atmel_tx_empty,
1216         .set_mctrl      = atmel_set_mctrl,
1217         .get_mctrl      = atmel_get_mctrl,
1218         .stop_tx        = atmel_stop_tx,
1219         .start_tx       = atmel_start_tx,
1220         .stop_rx        = atmel_stop_rx,
1221         .enable_ms      = atmel_enable_ms,
1222         .break_ctl      = atmel_break_ctl,
1223         .startup        = atmel_startup,
1224         .shutdown       = atmel_shutdown,
1225         .flush_buffer   = atmel_flush_buffer,
1226         .set_termios    = atmel_set_termios,
1227         .type           = atmel_type,
1228         .release_port   = atmel_release_port,
1229         .request_port   = atmel_request_port,
1230         .config_port    = atmel_config_port,
1231         .verify_port    = atmel_verify_port,
1232         .pm             = atmel_serial_pm,
1233 };
1234
1235 /*
1236  * Configure the port from the platform device resource info.
1237  */
1238 static void __devinit atmel_init_port(struct atmel_uart_port *atmel_port,
1239                                       struct platform_device *pdev)
1240 {
1241         struct uart_port *port = &atmel_port->uart;
1242         struct atmel_uart_data *data = pdev->dev.platform_data;
1243
1244         port->iotype    = UPIO_MEM;
1245         port->flags     = UPF_BOOT_AUTOCONF;
1246         port->ops       = &atmel_pops;
1247         port->fifosize  = 1;
1248         port->line      = pdev->id;
1249         port->dev       = &pdev->dev;
1250
1251         port->mapbase   = pdev->resource[0].start;
1252         port->irq       = pdev->resource[1].start;
1253
1254         tasklet_init(&atmel_port->tasklet, atmel_tasklet_func,
1255                         (unsigned long)port);
1256
1257         memset(&atmel_port->rx_ring, 0, sizeof(atmel_port->rx_ring));
1258
1259         if (data->regs)
1260                 /* Already mapped by setup code */
1261                 port->membase = data->regs;
1262         else {
1263                 port->flags     |= UPF_IOREMAP;
1264                 port->membase   = NULL;
1265         }
1266
1267         /* for console, the clock could already be configured */
1268         if (!atmel_port->clk) {
1269                 atmel_port->clk = clk_get(&pdev->dev, "usart");
1270                 clk_enable(atmel_port->clk);
1271                 port->uartclk = clk_get_rate(atmel_port->clk);
1272                 clk_disable(atmel_port->clk);
1273                 /* only enable clock when USART is in use */
1274         }
1275
1276         atmel_port->use_dma_rx = data->use_dma_rx;
1277         atmel_port->use_dma_tx = data->use_dma_tx;
1278         if (atmel_use_dma_tx(port))
1279                 port->fifosize = PDC_BUFFER_SIZE;
1280 }
1281
1282 /*
1283  * Register board-specific modem-control line handlers.
1284  */
1285 void __init atmel_register_uart_fns(struct atmel_port_fns *fns)
1286 {
1287         if (fns->enable_ms)
1288                 atmel_pops.enable_ms = fns->enable_ms;
1289         if (fns->get_mctrl)
1290                 atmel_pops.get_mctrl = fns->get_mctrl;
1291         if (fns->set_mctrl)
1292                 atmel_pops.set_mctrl = fns->set_mctrl;
1293         atmel_open_hook         = fns->open;
1294         atmel_close_hook        = fns->close;
1295         atmel_pops.pm           = fns->pm;
1296         atmel_pops.set_wake     = fns->set_wake;
1297 }
1298
1299 #ifdef CONFIG_SERIAL_ATMEL_CONSOLE
1300 static void atmel_console_putchar(struct uart_port *port, int ch)
1301 {
1302         while (!(UART_GET_CSR(port) & ATMEL_US_TXRDY))
1303                 cpu_relax();
1304         UART_PUT_CHAR(port, ch);
1305 }
1306
1307 /*
1308  * Interrupts are disabled on entering
1309  */
1310 static void atmel_console_write(struct console *co, const char *s, u_int count)
1311 {
1312         struct uart_port *port = &atmel_ports[co->index].uart;
1313         unsigned int status, imr;
1314         unsigned int pdc_tx;
1315
1316         /*
1317          * First, save IMR and then disable interrupts
1318          */
1319         imr = UART_GET_IMR(port);
1320         UART_PUT_IDR(port, ATMEL_US_RXRDY | ATMEL_US_TXRDY);
1321
1322         /* Store PDC transmit status and disable it */
1323         pdc_tx = UART_GET_PTSR(port) & ATMEL_PDC_TXTEN;
1324         UART_PUT_PTCR(port, ATMEL_PDC_TXTDIS);
1325
1326         uart_console_write(port, s, count, atmel_console_putchar);
1327
1328         /*
1329          * Finally, wait for transmitter to become empty
1330          * and restore IMR
1331          */
1332         do {
1333                 status = UART_GET_CSR(port);
1334         } while (!(status & ATMEL_US_TXRDY));
1335
1336         /* Restore PDC transmit status */
1337         if (pdc_tx)
1338                 UART_PUT_PTCR(port, ATMEL_PDC_TXTEN);
1339
1340         /* set interrupts back the way they were */
1341         UART_PUT_IER(port, imr);
1342 }
1343
1344 /*
1345  * If the port was already initialised (eg, by a boot loader),
1346  * try to determine the current setup.
1347  */
1348 static void __init atmel_console_get_options(struct uart_port *port, int *baud,
1349                                              int *parity, int *bits)
1350 {
1351         unsigned int mr, quot;
1352
1353         /*
1354          * If the baud rate generator isn't running, the port wasn't
1355          * initialized by the boot loader.
1356          */
1357         quot = UART_GET_BRGR(port) & ATMEL_US_CD;
1358         if (!quot)
1359                 return;
1360
1361         mr = UART_GET_MR(port) & ATMEL_US_CHRL;
1362         if (mr == ATMEL_US_CHRL_8)
1363                 *bits = 8;
1364         else
1365                 *bits = 7;
1366
1367         mr = UART_GET_MR(port) & ATMEL_US_PAR;
1368         if (mr == ATMEL_US_PAR_EVEN)
1369                 *parity = 'e';
1370         else if (mr == ATMEL_US_PAR_ODD)
1371                 *parity = 'o';
1372
1373         /*
1374          * The serial core only rounds down when matching this to a
1375          * supported baud rate. Make sure we don't end up slightly
1376          * lower than one of those, as it would make us fall through
1377          * to a much lower baud rate than we really want.
1378          */
1379         *baud = port->uartclk / (16 * (quot - 1));
1380 }
1381
1382 static int __init atmel_console_setup(struct console *co, char *options)
1383 {
1384         struct uart_port *port = &atmel_ports[co->index].uart;
1385         int baud = 115200;
1386         int bits = 8;
1387         int parity = 'n';
1388         int flow = 'n';
1389
1390         if (port->membase == NULL) {
1391                 /* Port not initialized yet - delay setup */
1392                 return -ENODEV;
1393         }
1394
1395         clk_enable(atmel_ports[co->index].clk);
1396
1397         UART_PUT_IDR(port, -1);
1398         UART_PUT_CR(port, ATMEL_US_RSTSTA | ATMEL_US_RSTRX);
1399         UART_PUT_CR(port, ATMEL_US_TXEN | ATMEL_US_RXEN);
1400
1401         if (options)
1402                 uart_parse_options(options, &baud, &parity, &bits, &flow);
1403         else
1404                 atmel_console_get_options(port, &baud, &parity, &bits);
1405
1406         return uart_set_options(port, co, baud, parity, bits, flow);
1407 }
1408
1409 static struct uart_driver atmel_uart;
1410
1411 static struct console atmel_console = {
1412         .name           = ATMEL_DEVICENAME,
1413         .write          = atmel_console_write,
1414         .device         = uart_console_device,
1415         .setup          = atmel_console_setup,
1416         .flags          = CON_PRINTBUFFER,
1417         .index          = -1,
1418         .data           = &atmel_uart,
1419 };
1420
1421 #define ATMEL_CONSOLE_DEVICE    (&atmel_console)
1422
1423 /*
1424  * Early console initialization (before VM subsystem initialized).
1425  */
1426 static int __init atmel_console_init(void)
1427 {
1428         if (atmel_default_console_device) {
1429                 add_preferred_console(ATMEL_DEVICENAME,
1430                                       atmel_default_console_device->id, NULL);
1431                 atmel_init_port(&atmel_ports[atmel_default_console_device->id],
1432                                 atmel_default_console_device);
1433                 register_console(&atmel_console);
1434         }
1435
1436         return 0;
1437 }
1438
1439 console_initcall(atmel_console_init);
1440
1441 /*
1442  * Late console initialization.
1443  */
1444 static int __init atmel_late_console_init(void)
1445 {
1446         if (atmel_default_console_device
1447             && !(atmel_console.flags & CON_ENABLED))
1448                 register_console(&atmel_console);
1449
1450         return 0;
1451 }
1452
1453 core_initcall(atmel_late_console_init);
1454
1455 static inline bool atmel_is_console_port(struct uart_port *port)
1456 {
1457         return port->cons && port->cons->index == port->line;
1458 }
1459
1460 #else
1461 #define ATMEL_CONSOLE_DEVICE    NULL
1462
1463 static inline bool atmel_is_console_port(struct uart_port *port)
1464 {
1465         return false;
1466 }
1467 #endif
1468
1469 static struct uart_driver atmel_uart = {
1470         .owner          = THIS_MODULE,
1471         .driver_name    = "atmel_serial",
1472         .dev_name       = ATMEL_DEVICENAME,
1473         .major          = SERIAL_ATMEL_MAJOR,
1474         .minor          = MINOR_START,
1475         .nr             = ATMEL_MAX_UART,
1476         .cons           = ATMEL_CONSOLE_DEVICE,
1477 };
1478
1479 #ifdef CONFIG_PM
1480 static bool atmel_serial_clk_will_stop(void)
1481 {
1482 #ifdef CONFIG_ARCH_AT91
1483         return at91_suspend_entering_slow_clock();
1484 #else
1485         return false;
1486 #endif
1487 }
1488
1489 static int atmel_serial_suspend(struct platform_device *pdev,
1490                                 pm_message_t state)
1491 {
1492         struct uart_port *port = platform_get_drvdata(pdev);
1493         struct atmel_uart_port *atmel_port = to_atmel_uart_port(port);
1494
1495         if (atmel_is_console_port(port) && console_suspend_enabled) {
1496                 /* Drain the TX shifter */
1497                 while (!(UART_GET_CSR(port) & ATMEL_US_TXEMPTY))
1498                         cpu_relax();
1499         }
1500
1501         /* we can not wake up if we're running on slow clock */
1502         atmel_port->may_wakeup = device_may_wakeup(&pdev->dev);
1503         if (atmel_serial_clk_will_stop())
1504                 device_set_wakeup_enable(&pdev->dev, 0);
1505
1506         uart_suspend_port(&atmel_uart, port);
1507
1508         return 0;
1509 }
1510
1511 static int atmel_serial_resume(struct platform_device *pdev)
1512 {
1513         struct uart_port *port = platform_get_drvdata(pdev);
1514         struct atmel_uart_port *atmel_port = to_atmel_uart_port(port);
1515
1516         uart_resume_port(&atmel_uart, port);
1517         device_set_wakeup_enable(&pdev->dev, atmel_port->may_wakeup);
1518
1519         return 0;
1520 }
1521 #else
1522 #define atmel_serial_suspend NULL
1523 #define atmel_serial_resume NULL
1524 #endif
1525
1526 static int __devinit atmel_serial_probe(struct platform_device *pdev)
1527 {
1528         struct atmel_uart_port *port;
1529         void *data;
1530         int ret;
1531
1532         BUILD_BUG_ON(!is_power_of_2(ATMEL_SERIAL_RINGSIZE));
1533
1534         port = &atmel_ports[pdev->id];
1535         port->backup_imr = 0;
1536
1537         atmel_init_port(port, pdev);
1538
1539         if (!atmel_use_dma_rx(&port->uart)) {
1540                 ret = -ENOMEM;
1541                 data = kmalloc(sizeof(struct atmel_uart_char)
1542                                 * ATMEL_SERIAL_RINGSIZE, GFP_KERNEL);
1543                 if (!data)
1544                         goto err_alloc_ring;
1545                 port->rx_ring.buf = data;
1546         }
1547
1548         ret = uart_add_one_port(&atmel_uart, &port->uart);
1549         if (ret)
1550                 goto err_add_port;
1551
1552         if (atmel_is_console_port(&port->uart)
1553                         && ATMEL_CONSOLE_DEVICE->flags & CON_ENABLED) {
1554                 /*
1555                  * The serial core enabled the clock for us, so undo
1556                  * the clk_enable() in atmel_console_setup()
1557                  */
1558                 clk_disable(port->clk);
1559         }
1560
1561         device_init_wakeup(&pdev->dev, 1);
1562         platform_set_drvdata(pdev, port);
1563
1564         return 0;
1565
1566 err_add_port:
1567         kfree(port->rx_ring.buf);
1568         port->rx_ring.buf = NULL;
1569 err_alloc_ring:
1570         if (!atmel_is_console_port(&port->uart)) {
1571                 clk_put(port->clk);
1572                 port->clk = NULL;
1573         }
1574
1575         return ret;
1576 }
1577
1578 static int __devexit atmel_serial_remove(struct platform_device *pdev)
1579 {
1580         struct uart_port *port = platform_get_drvdata(pdev);
1581         struct atmel_uart_port *atmel_port = to_atmel_uart_port(port);
1582         int ret = 0;
1583
1584         device_init_wakeup(&pdev->dev, 0);
1585         platform_set_drvdata(pdev, NULL);
1586
1587         ret = uart_remove_one_port(&atmel_uart, port);
1588
1589         tasklet_kill(&atmel_port->tasklet);
1590         kfree(atmel_port->rx_ring.buf);
1591
1592         /* "port" is allocated statically, so we shouldn't free it */
1593
1594         clk_put(atmel_port->clk);
1595
1596         return ret;
1597 }
1598
1599 static struct platform_driver atmel_serial_driver = {
1600         .probe          = atmel_serial_probe,
1601         .remove         = __devexit_p(atmel_serial_remove),
1602         .suspend        = atmel_serial_suspend,
1603         .resume         = atmel_serial_resume,
1604         .driver         = {
1605                 .name   = "atmel_usart",
1606                 .owner  = THIS_MODULE,
1607         },
1608 };
1609
1610 static int __init atmel_serial_init(void)
1611 {
1612         int ret;
1613
1614         ret = uart_register_driver(&atmel_uart);
1615         if (ret)
1616                 return ret;
1617
1618         ret = platform_driver_register(&atmel_serial_driver);
1619         if (ret)
1620                 uart_unregister_driver(&atmel_uart);
1621
1622         return ret;
1623 }
1624
1625 static void __exit atmel_serial_exit(void)
1626 {
1627         platform_driver_unregister(&atmel_serial_driver);
1628         uart_unregister_driver(&atmel_uart);
1629 }
1630
1631 module_init(atmel_serial_init);
1632 module_exit(atmel_serial_exit);
1633
1634 MODULE_AUTHOR("Rick Bronson");
1635 MODULE_DESCRIPTION("Atmel AT91 / AT32 serial port driver");
1636 MODULE_LICENSE("GPL");
1637 MODULE_ALIAS("platform:atmel_usart");