#include <ext4fs.h>
 #include <fat.h>
 #include <fs.h>
+#include <asm/io.h>
 
 DECLARE_GLOBAL_DATA_PTR;
 
        return -1;
 }
 
-static inline int fs_read_unsupported(const char *filename, ulong addr,
+static inline int fs_read_unsupported(const char *filename, void *buf,
                                      int offset, int len)
 {
        return -1;
 
 #define fs_ls_fat file_fat_ls
 
-static int fs_read_fat(const char *filename, ulong addr, int offset, int len)
+static int fs_read_fat(const char *filename, void *buf, int offset, int len)
 {
        int len_read;
 
-       len_read = file_fat_read_at(filename, offset,
-                                   (unsigned char *)addr, len);
+       len_read = file_fat_read_at(filename, offset, buf, len);
        if (len_read == -1) {
                printf("** Unable to read file %s **\n", filename);
                return -1;
 
 #define fs_ls_ext ext4fs_ls
 
-static int fs_read_ext(const char *filename, ulong addr, int offset, int len)
+static int fs_read_ext(const char *filename, void *buf, int offset, int len)
 {
        int file_len;
        int len_read;
        if (len == 0)
                len = file_len;
 
-       len_read = ext4fs_read((char *)addr, len);
+       len_read = ext4fs_read(buf, len);
        ext4fs_close();
 
        if (len_read != len) {
        int (*probe)(block_dev_desc_t *fs_dev_desc,
                     disk_partition_t *fs_partition);
        int (*ls)(const char *dirname);
-       int (*read)(const char *filename, ulong addr, int offset, int len);
+       int (*read)(const char *filename, void *buf, int offset, int len);
        void (*close)(void);
 };
 
 int fs_read(const char *filename, ulong addr, int offset, int len)
 {
        struct fstype_info *info = fs_get_info(fs_type);
+       void *buf;
        int ret;
 
-       ret = info->read(filename, addr, offset, len);
+       /*
+        * We don't actually know how many bytes are being read, since len==0
+        * means read the whole file.
+        */
+       buf = map_sysmem(addr, len);
+       ret = info->read(filename, buf, offset, len);
+       unmap_sysmem(buf);
 
        /* If we requested a specific number of bytes, check we got it */
        if (ret >= 0 && len && ret != len) {