* add initrd support

2 files changed, 4 insertions, 2 deletions

diff --git a/bootstub.c b/bootstub.c
index f028c6d..a24939b 100644
--- a/bootstub.c
+++ b/bootstub.c
@@ -75,7 +75,9 @@ static void setup_boot_params(struct boot_params *bp, struct setup_header *sh)
 	bp->hdr.cmd_line_ptr = CMDLINE_OFFSET;
 	bp->hdr.cmdline_size = strnlen((const char *)CMDLINE_OFFSET,256);
 	bp->hdr.ramdisk_size = *(u32 *)INITRD_SIZE_OFFSET;
-	bp->hdr.ramdisk_image = BZIMAGE_OFFSET + *(u32 *)BZIMAGE_SIZE_OFFSET + *(u32 *)INITRD_SIZE_OFFSET;
+	bp->hdr.ramdisk_image = (bp->alt_mem_k*1024 - bp->hdr.ramdisk_size) & ~(0x0f);
+	memcpy((u8*)bp->hdr.ramdisk_image, (u8 *)BZIMAGE_OFFSET + *(u32 *)BZIMAGE_SIZE_OFFSET + *(u32 *)INITRD_SIZE_OFFSET,
+		bp->hdr.ramdisk_size);
 }
 
 static int get_32bit_entry(unsigned char *ptr)
diff --git a/head.S b/head.S
index 0dab98d..e242d48 100644
--- a/head.S
+++ b/head.S
@@ -10,7 +10,7 @@
 
 /* When bootstub get control, the memory map in DRAM is like:
  *		~			~
- * 0x102000	|	initrd		| initrd will move to high memory otherwise 
+ * 0x102000	|	initrd		| initrd will move to highest memory otherwise 
  *+bzImage size +-----------------------+ bzImage uncompressing will destory it
  *		|	bzImage		|
  * 0x102000	+-----------------------+