#include <common.h>
 #include <asm/io.h>
 #include <asm/jz4740.h>
+#include <serial.h>
+#include <linux/compiler.h>
 
 /*
  * serial_init - initialize a channel
  */
 struct jz4740_uart *uart = (struct jz4740_uart *)CONFIG_SYS_UART_BASE;
 
-int serial_init(void)
+static int jz_serial_init(void)
 {
        /* Disable port interrupts while changing hardware */
        writeb(0, &uart->dlhr_ier);
        return 0;
 }
 
-void serial_setbrg(void)
+static void jz_serial_setbrg(void)
 {
        u32 baud_div, tmp;
 
        writeb(tmp, &uart->lcr);
 }
 
-int serial_tstc(void)
+static int jz_serial_tstc(void)
 {
        if (readb(&uart->lsr) & UART_LSR_DR)
                return 1;
        return 0;
 }
 
-void serial_putc(const char c)
+static void jz_serial_putc(const char c)
 {
        if (c == '\n')
                serial_putc('\r');
        writeb((u8)c, &uart->rbr_thr_dllr);
 }
 
-int serial_getc(void)
+static int jz_serial_getc(void)
 {
        while (!serial_tstc())
                ;
        return readb(&uart->rbr_thr_dllr);
 }
 
-void serial_puts(const char *s)
+static void jz_serial_puts(const char *s)
 {
        while (*s)
                serial_putc(*s++);
 }
+
+#ifdef CONFIG_SERIAL_MULTI
+static struct serial_device jz_serial_drv = {
+       .name   = "jz_serial",
+       .start  = jz_serial_init,
+       .stop   = NULL,
+       .setbrg = jz_serial_setbrg,
+       .putc   = jz_serial_putc,
+       .puts   = jz_serial_puts,
+       .getc   = jz_serial_getc,
+       .tstc   = jz_serial_tstc,
+};
+
+void jz_serial_initialize(void)
+{
+       serial_register(&jz_serial_drv);
+}
+
+__weak struct serial_device *default_serial_console(void)
+{
+       return &jz_serial_drv;
+}
+#else
+int serial_init(void)
+{
+       return jz_serial_init();
+}
+
+void serial_setbrg(void)
+{
+       jz_serial_setbrg();
+}
+
+void serial_putc(const char c)
+{
+       jz_serial_putc(c);
+}
+
+void serial_puts(const char *s)
+{
+       jz_serial_puts(s);
+}
+
+int serial_getc(void)
+{
+       return jz_serial_getc();
+}
+
+int serial_tstc(void)
+{
+       return jz_serial_tstc();
+}
+#endif
 
 serial_initfunc(uartlite_serial_initialize);
 serial_initfunc(au1x00_serial_initialize);
 serial_initfunc(asc_serial_initialize);
+serial_initfunc(jz_serial_initialize);
 
 void serial_register(struct serial_device *dev)
 {
        zynq_serial_initalize();
        au1x00_serial_initialize();
        asc_serial_initialize();
+       jz_serial_initialize();
 
        serial_assign(default_serial_console()->name);
 }