Chameleon

Chameleon Svn Source Tree

Root/branches/meklort/i386/boot2/ramdisk.c

Source at commit 577 created 13 years 6 months ago.
By meklort, Reenabled multiboot code (ramdisk support)
1/*␊
2 * Supplemental ramdisk functions for the multiboot ramdisk driver.␊
3 * Copyright 2009 Tamas Kosarszky. All rights reserved.␊
4 *␊
5 */␊
6␊
7#include "boot.h"␊
8#include "bootstruct.h"␊
9#include "multiboot.h"␊
10#include "ramdisk.h"␊
11␊
12#ifndef OPTION_ROM␊
13struct multiboot_info * gRAMDiskMI = NULL;␊
14␊
15// gRAMDiskVolume holds the bvr for the mounted ramdisk image.␊
16BVRef gRAMDiskVolume = NULL;␊
17bool gRAMDiskBTAliased = false;␊
18char gRAMDiskFile[512];␊
19#endif␊
20␊
21// Notify OS X that a ramdisk has been setup. XNU with attach this to /dev/md0␊
22void md0Ramdisk()␊
23{␊
24␉RAMDiskParam ramdiskPtr;␊
25␉char filename[512];␊
26␉const char* override_filename = 0;␊
27␉int fh = -1;␊
28␉int len;␊
29␉␊
30␉if(getValueForKey(kMD0Image, &override_filename, &len, ␊
31␉␉␉␉ &bootInfo->bootConfig))␊
32␉{␊
33␉␉// Use user specified md0 file␊
34␉␉sprintf(filename, "%s", override_filename);␊
35␉␉fh = open(filename, 0);␊
36␉␉␊
37␉␉if(fh < 0)␊
38␉␉{␊
39␉␉␉sprintf(filename, "rd(0,0)/Extra/%s", override_filename);␊
40␉␉␉fh = open(filename, 0);␊
41␊
42␉␉␉if(fh < 0)␊
43␉␉␉{␊
44␉␉␉␉sprintf(filename, "/Extra/%s", override_filename);␊
45␉␉␉␉fh = open(filename, 0);␊
46␉␉␉}␊
47␉␉}␉␉␊
48␉}␊
49␊
50␉if(fh < 0)␊
51␉{␊
52␉␉sprintf(filename, "rd(0,0)/Extra/Postboot.img");␊
53␉␉fh = open(filename, 0);␊
54␊
55␉␉if(fh < 0)␊
56␉␉{␊
57␉␉␉sprintf(filename, "/Extra/Postboot.img");␉// Check /Extra if not in rd(0,0)␊
58␉␉␉fh = open(filename, 0);␊
59␉␉}␊
60␉}␉␉␊
61␊
62␉if (fh >= 0)␊
63␉{␊
64␉␉verbose("Enabling ramdisk %s\n", filename);␊
65␊
66␉␉ramdiskPtr.size = file_size(fh);␊
67␉␉ramdiskPtr.base = AllocateKernelMemory(ramdiskPtr.size);␊
68␊
69␉␉if(ramdiskPtr.size && ramdiskPtr.base)␊
70␉␉{␊
71␉␉␉// Read new ramdisk image contents in kernel memory.␊
72␉␉␉if (read(fh, (char*) ramdiskPtr.base, ramdiskPtr.size) == ramdiskPtr.size)␊
73␉␉␉{␊
74␉␉␉␉AllocateMemoryRange("RAMDisk", ramdiskPtr.base, ramdiskPtr.size, kBootDriverTypeInvalid);␊
75␉␉␉␉Node* node = DT__FindNode("/chosen/memory-map", false);␊
76␉␉␉␉if(node != NULL)␊
77␉␉␉␉{␊
78␉␉␉␉␉DT__AddProperty(node, "RAMDisk", sizeof(RAMDiskParam), (void*)&ramdiskPtr);␉␉␊
79␉␉␉␉}␊
80␉␉␉␉else␊
81␉␉␉␉{␊
82␉␉␉␉␉verbose("Unable to notify Mac OS X of the ramdisk %s.\n", filename);␊
83␉␉␉␉}␊
84␉␉␉}␊
85␉␉␉else␊
86␉␉␉{␊
87␉␉␉␉verbose("Unable to read md0 image %s.\n", filename);␊
88␉␉␉}␉␉␉␊
89␉␉}␊
90␉␉else␊
91␉␉{␊
92␉␉␉verbose("md0 image %s is empty.\n", filename);␊
93␉␉}␊
94␊
95␉␉close(fh);␊
96␊
97␉}␊
98}␊
99#ifndef OPTION_ROM␊
100void umountRAMDisk()␊
101{␊
102␉if (gRAMDiskMI != NULL)␊
103␉{␊
104␉␉// Release ramdisk BVRef and DiskBVMap.␊
105␉␉struct DiskBVMap *oldMap = diskResetBootVolumes(0x100);␊
106␉␉CacheReset();␊
107␉␉diskFreeMap(oldMap);␊
108␊
109␉␉// Free multiboot info and module structures.␊
110␉␉if ((void *)gRAMDiskMI->mi_mods_addr != NULL) free((void *)gRAMDiskMI->mi_mods_addr);␊
111␉␉if (gRAMDiskMI != NULL) free(gRAMDiskMI);␊
112␊
113␉␉// Reset multiboot structures.␊
114␉␉gRAMDiskMI = NULL;␊
115␉␉//#ifdef UNUSED␊
116␉␉gMI = gRAMDiskMI;␊
117␉␉//#endif␊
118␉␉␊
119␉␉*gRAMDiskFile = '\0';␊
120␊
121␉␉// Release ramdisk driver hooks.␊
122␉␉p_get_ramdisk_info = NULL;␊
123␉␉p_ramdiskReadBytes = NULL;␊
124␊
125␉␉// Reset ramdisk bvr␊
126␉␉gRAMDiskVolume = NULL;␊
127␉␉printf("\nunmounting: done");␊
128␉}␊
129}␊
130␊
131int mountRAMDisk(const char * param)␊
132{␊
133␉int fh = 0, ramDiskSize;␊
134␉int error = 0;␊
135␊
136␉// Get file handle for ramdisk file.␊
137␉fh = open(param, 0);␊
138␉if (fh != -1)␊
139␉{␊
140␉␉verbose("\nreading ramdisk image: %s", param);␊
141␊
142␉␉ramDiskSize = file_size(fh);␊
143␉␉if (ramDiskSize > 0)␊
144␉␉{␊
145␉␉␉// Unmount previously mounted image if exists.␊
146␉␉␉umountRAMDisk();␊
147␊
148␉␉␉// Read new ramdisk image contents into PREBOOT_DATA area.␊
149␉␉␉if (read(fh, (char *)PREBOOT_DATA, ramDiskSize) != ramDiskSize) error = -1;␊
150␉␉}␊
151␉␉else error = -1;␊
152␊
153␉␉close(fh);␊
154␉}␊
155␉else error = -1;␊
156␊
157␉if (error == 0)␊
158␉{␊
159␉␉// Save filename in gRAMDiskFile to display information.␊
160␉␉strcpy(gRAMDiskFile, param);␊
161␊
162␉␉// Set gMI as well for the multiboot ramdisk driver hook.␊
163␉␉gRAMDiskMI = malloc(sizeof(multiboot_info));␊
164␉␉//#ifdef UNUSED␊
165␉␉gMI = gRAMDiskMI;␊
166␉␉//#endif␊
167␉␉struct multiboot_module * ramdisk_module = malloc(sizeof(multiboot_module));␊
168␊
169␉␉// Fill in multiboot info and module structures.␊
170␉␉if (gRAMDiskMI != NULL && ramdisk_module != NULL)␊
171␉␉{␊
172␉␉␉gRAMDiskMI->mi_mods_count = 1;␊
173␉␉␉gRAMDiskMI->mi_mods_addr = (uint32_t)ramdisk_module;␊
174␉␉␉ramdisk_module->mm_mod_start = PREBOOT_DATA;␊
175␉␉␉ramdisk_module->mm_mod_end = PREBOOT_DATA + ramDiskSize;␊
176␊
177␉␉␉//#ifdef UNUSED␊
178␉␉␉// Set ramdisk driver hooks.␊
179␉␉␉p_get_ramdisk_info = &multiboot_get_ramdisk_info;␊
180␉␉␉p_ramdiskReadBytes = &multibootRamdiskReadBytes;␊
181␉␉␉//#endif␊
182␉␉␉int partCount; // unused␊
183␉␉␉// Save bvr of the mounted image.␊
184␉␉␉gRAMDiskVolume = diskScanBootVolumes(0x100, &partCount);␊
185␉␉␉if(gRAMDiskVolume == NULL)␊
186␉␉␉{␊
187␉␉␉␉umountRAMDisk();␊
188␉␉␉␉printf("\nRamdisk contains no partitions.\n");␊
189␉␉␉␉pause();␊
190␊
191␉␉␉}␊
192␉␉␉else␊
193␉␉␉{␊
194␉␉␉␉char dirSpec[128];␊
195␊
196␉␉␉␉// Reading ramdisk configuration.␊
197␉␉␉␉strcpy(dirSpec, RAMDISKCONFIG_FILENAME);␊
198␊
199␉␉␉␉if (loadConfigFile(dirSpec, &bootInfo->ramdiskConfig) == 0)␊
200␉␉␉␉{␊
201␉␉␉␉␉getBoolForKey("BTAlias", &gRAMDiskBTAliased, &bootInfo->ramdiskConfig);␊
202␉␉␉␉}␊
203␉␉␉␉else␊
204␉␉␉␉{␊
205␉␉␉␉␉verbose("\nno ramdisk config...\n");␊
206␉␉␉␉}␊
207␊
208␉␉␉␉verbose("\nmounting: done");␊
209␉␉␉}␊
210␉␉}␊
211␉}␊
212␉return error;␊
213}␊
214␊
215void setRAMDiskBTHook(bool mode)␊
216{␊
217␉gRAMDiskBTAliased = mode;␊
218␉if (mode)␊
219␉{␊
220␉␉printf("\nEnabled bt(0,0) alias.");␊
221␉}␊
222␉else␊
223␉{␊
224␉␉printf("\nDisabled bt(0,0) alias.");␊
225␉}␊
226}␊
227␊
228void showInfoRAMDisk(void)␊
229{␊
230␉int len;␊
231␉const char *val;␊
232␊
233␉if (gRAMDiskMI != NULL)␊
234␉{␊
235␉␉struct multiboot_module * ramdisk_module = (void *)gRAMDiskMI->mi_mods_addr;␊
236␊
237␉␉printf("\nfile: %s %d", gRAMDiskFile,␊
238␉␉ramdisk_module->mm_mod_end - ramdisk_module->mm_mod_start);␊
239␉␉printf("\nalias: %s", gRAMDiskBTAliased ? "enabled" : "disabled");␊
240␊
241␉␉// Display ramdisk information if available.␊
242␉␉if (getValueForKey("Info", &val, &len, &bootInfo->ramdiskConfig))␊
243␉␉{␊
244␉␉␉printf("\ninfo: %s", val);␊
245␉␉}␊
246␉␉else␊
247␉␉{␊
248␉␉␉printf("\nramdisk info not available.");␊
249␉␉}␊
250␉}␊
251␉else␊
252␉{␊
253␉␉printf("\nNo ramdisk mounted.");␊
254␉}␊
255}␊
256␊
257int loadPrebootRAMDisk()␊
258{␊
259␉mountRAMDisk("bt(0,0)/Extra/Preboot.dmg");␊
260␉if (gRAMDiskMI != NULL)␊
261␉{␊
262␉␉printf("\n");␊
263␉␉return 0;␊
264␉}␊
265␉else␊
266␉{␊
267␉␉return -1;␊
268␉}␊
269}␊
270␊
271void processRAMDiskCommand(char ** argPtr, const char * cmd)␊
272{␊
273␉char * ptr = *argPtr;␊
274␉char param[1024];␊
275␉getNextArg(&ptr, param);␊
276␊
277␉if (strcmp(cmd, "m") == 0)␊
278␉{␊
279␉␉mountRAMDisk(param);␊
280␉␉sleep(2);␊
281␉}␊
282␉else if (strcmp(cmd, "u") == 0)␊
283␉{␊
284␉␉umountRAMDisk();␊
285␉␉sleep(2);␊
286␉}␊
287␉else if (strcmp(cmd, "e") == 0)␊
288␉{␊
289␉␉setRAMDiskBTHook(true);␊
290␉␉sleep(2);␊
291␉}␊
292␉else if (strcmp(cmd, "d") == 0)␊
293␉{␊
294␉␉setRAMDiskBTHook(false);␊
295␉␉sleep(2);␊
296␉}␊
297␉else if (strcmp(cmd, "i") == 0)␊
298␉{␊
299␉␉setActiveDisplayPage(1);␊
300␉␉clearScreenRows(0, 24);␊
301␉␉setCursorPosition(0, 0, 1);␊
302␉␉showInfoRAMDisk();␊
303␉␉printf("\n\nPress any key to continue.\n");␊
304␉␉getc();␊
305␉␉setActiveDisplayPage(0);␊
306␉}␊
307␉else␊
308␉{␊
309␉␉setActiveDisplayPage(1);␊
310␉␉clearScreenRows(0, 24);␊
311␉␉setCursorPosition(0, 0, 1);␊
312␉␉printf("\nusage:\n");␊
313␉␉printf("\n?rd i - display ramdisk information");␊
314␉␉printf("\n?rd m <filename> - mount ramdisk image\n?rd u - unmount ramdisk image");␊
315␉␉printf("\n?rd e - enable bt(0,0) alias\n?rd d - disable bt(0,0) alias");␊
316␉␉printf("\n\nPress any key to continue.\n");␊
317␉␉getc();␊
318␉␉setActiveDisplayPage(0);␊
319␉}␊
320}␊
321#endif␊
322

Archive Download this file

Revision: 577