console: Use map_sysmem() for the pre-relocation console

[people/ms/u-boot.git] / common / console.c

diff --git a/common/console.c b/common/console.c
index 60e7a94a56da4d62a2ffdef98312badc16351047..762d5f291c2284b3b0860ed4249fc332ecdd30e4 100644 (file)
--- a/common/console.c
+++ b/common/console.c
@@ -11,6 +11,7 @@
 #include <stdarg.h>
 #include <iomux.h>
 #include <malloc.h>
+#include <mapmem.h>
 #include <os.h>
 #include <serial.h>
 #include <stdio_dev.h>
@@ -416,9 +417,13 @@ int tstc(void)
 
 static void pre_console_putc(const char c)
 {
-       char *buffer = (char *)CONFIG_PRE_CON_BUF_ADDR;
+       char *buffer;
+
+       buffer = map_sysmem(CONFIG_PRE_CON_BUF_ADDR, CONFIG_PRE_CON_BUF_SZ);
 
        buffer[CIRC_BUF_IDX(gd->precon_buf_idx++)] = c;
+
+       unmap_sysmem(buffer);
 }
 
 static void pre_console_puts(const char *s)
@@ -430,14 +435,16 @@ static void pre_console_puts(const char *s)
 static void print_pre_console_buffer(int flushpoint)
 {
        unsigned long in = 0, out = 0;
-       char *buf_in = (char *)CONFIG_PRE_CON_BUF_ADDR;
        char buf_out[CONFIG_PRE_CON_BUF_SZ + 1];
+       char *buf_in;
 
+       buf_in = map_sysmem(CONFIG_PRE_CON_BUF_ADDR, CONFIG_PRE_CON_BUF_SZ);
        if (gd->precon_buf_idx > CONFIG_PRE_CON_BUF_SZ)
                in = gd->precon_buf_idx - CONFIG_PRE_CON_BUF_SZ;
 
        while (in < gd->precon_buf_idx)
                buf_out[out++] = buf_in[CIRC_BUF_IDX(in++)];
+       unmap_sysmem(buf_in);
 
        buf_out[out] = 0;