Chameleon

Chameleon Svn Source Tree

Root/branches/cparm/i386/modules/RamDiskLoader/ramdisk.c

1/*␊
2 * Supplemental ramdisk functions for the multiboot ramdisk driver.␊
3 * Copyright 2009 Tamas Kosarszky. All rights reserved.␊
4 *␊
5 */␊
6␊
7#include "libsaio.h"␊
8#include "bootstruct.h"␊
9#include "multiboot.h"␊
10#include "ramdisk.h"␊
11␊
12struct multiboot_info * gRAMDiskMI = NULL;␊
13␊
14config_file_t ramdiskConfig;␉␉ ␊
15␊
16// gRAMDiskVolume holds the bvr for the mounted ramdisk image.␊
17BVRef gRAMDiskVolume = NULL;␊
18bool gRAMDiskBTAliased = false;␊
19char gRAMDiskFile[512];␊
20␊
21long (*LoadExtraDrivers_p)(FileLoadDrivers_t FileLoadDrivers_p) = NULL;␊
22int (*p_ramdiskReadBytes)( int biosdev, unsigned int blkno,␊
23␉␉␉␉␉␉ unsigned int byteoff,␊
24␉␉␉␉␉␉ unsigned int byteCount, void * buffer ) = NULL;␊
25int (*p_get_ramdisk_info)(int biosdev, struct driveInfo *dip) = NULL;␊
26␊
27int multibootRamdiskReadBytes( int biosdev, unsigned int blkno,␊
28␉␉␉␉␉␉␉ unsigned int byteoff,␊
29␉␉␉␉␉␉␉ unsigned int byteCount, void * buffer );␊
30int multiboot_get_ramdisk_info(int biosdev, struct driveInfo *dip);␊
31static long multiboot_LoadExtraDrivers(FileLoadDrivers_t FileLoadDrivers_p);␊
32␊
33␊
34// Notify OS X that a ramdisk has been setup. XNU with attach this to /dev/md0 (WARNING : md0Ramdisk MUST BE EXTERN FOR NBI COMPTATIBILITY)␊
35void md0Ramdisk()␊
36{␊
37␉RAMDiskParam ramdiskPtr;␊
38␉char filename[512];␊
39␉const char* override_filename = 0;␊
40␉int fh = -1;␊
41␉int len;␊
42␉␊
43␉if(getValueForKey(kMD0Image, &override_filename, &len, ␊
44␉␉␉␉␉ DEFAULT_BOOT_CONFIG))␊
45␉{␊
46␉␉// Use user specified md0 file␊
47␉␉snprintf(filename, sizeof(filename) ,"%s", override_filename);␊
48␉␉fh = open(filename, 0);␊
49␉␉␊
50␉␉if(fh < 0)␊
51␉␉{␊
52␉␉␉snprintf(filename, sizeof(filename) ,"bt(0,0)/Extra/%s", override_filename);␊
53␉␉␉fh = open(filename, 0);␊
54␉␉␉␊
55␉␉␉if(fh < 0)␊
56␉␉␉{␊
57␉␉␉␉snprintf(filename, sizeof(filename) ,"rd(0,0)/Extra/%s", override_filename);␊
58␉␉␉␉fh = open(filename, 0);␊
59␉␉␉␉␊
60␉␉␉␉if(fh < 0)␊
61␉␉␉␉{␊
62␉␉␉␉␉snprintf(filename, sizeof(filename) ,"/Extra/%s", override_filename);␊
63␉␉␉␉␉fh = open(filename, 0);␊
64␉␉␉␉}␊
65␉␉␉}␊
66␉␉}␊
67␉}␊
68␉␊
69␉if(fh < 0)␊
70␉{␊
71␉␉snprintf(filename, sizeof(filename) ,"bt(0,0)/Extra/Postboot.img");␊
72␉␉fh = open(filename, 0);␊
73␉␉␊
74␉␉if(fh < 0)␊
75␉␉{␊
76␉␉␉snprintf(filename, sizeof(filename) ,"rd(0,0)/Extra/Postboot.img");␊
77␉␉␉fh = open(filename, 0);␊
78␉␉␉␊
79␉␉␉if(fh < 0)␊
80␉␉␉{␊
81␉␉␉␉snprintf(filename, sizeof(filename) ,"/Extra/Postboot.img");␉// Check /Extra if not in rd(0,0)␊
82␉␉␉␉fh = open(filename, 0);␊
83␉␉␉}␊
84␉␉}␊
85␉}␊
86␉␊
87␉if (fh >= 0)␊
88␉{␊
89␉␉verbose("Enabling ramdisk %s\n", filename);␊
90␉␉␊
91␉␉ramdiskPtr.size = file_size(fh);␊
92␉␉ramdiskPtr.base = AllocateKernelMemory(ramdiskPtr.size);␊
93␉␉␊
94␉␉if(ramdiskPtr.size && ramdiskPtr.base)␊
95␉␉{␊
96␉␉␉// Read new ramdisk image contents in kernel memory.␊
97␉␉␉if (read(fh, (char*) ramdiskPtr.base, ramdiskPtr.size) == ramdiskPtr.size)␊
98␉␉␉{␉␉␉␉␊
99␉␉␉␉␊
100 AllocateMemoryRange("RAMDisk", ramdiskPtr.base, ramdiskPtr.size);␊
101␉␉␉␉␊
102␉␉␉␉Node* node = DT__FindNode("/chosen/memory-map", false);␊
103␉␉␉␉if(node != NULL)␊
104␉␉␉␉{␊
105␉␉␉␉␉DT__AddProperty(node, "RAMDisk", sizeof(RAMDiskParam), (void*)&ramdiskPtr);␉␉␊
106␉␉␉␉}␊
107␉␉␉␉else␊
108␉␉␉␉{␊
109␉␉␉␉␉verbose("Unable to notify Mac OS X of the ramdisk %s.\n", filename);␊
110␉␉␉␉}␊
111␉␉␉}␊
112␉␉␉else␊
113␉␉␉{␊
114␉␉␉␉verbose("Unable to read md0 image %s.\n", filename);␊
115␉␉␉}␉␉␉␊
116␉␉}␊
117␉␉else␊
118␉␉{␊
119␉␉␉verbose("md0 image %s is empty.\n", filename);␊
120␉␉}␊
121␉␉␊
122␉␉close(fh);␊
123␉␉␊
124␉}␊
125}␊
126␊
127void umountRAMDisk(void)␊
128{␊
129␉if (gRAMDiskMI != NULL)␊
130␉{␊
131␉␉// Release ramdisk BVRef and DiskBVMap.␊
132␉␉struct DiskBVMap *oldMap = diskResetBootVolumes(0x100);␊
133 if (oldMap == NULL)␊
134 {␊
135 return;␊
136 }␊
137␉␉CacheReset();␊
138␉␉diskFreeMap(oldMap);␊
139␉␉␊
140␉␉// Free multiboot info and module structures.␊
141␉␉if ((void *)gRAMDiskMI->mi_mods_addr != NULL) free((void *)gRAMDiskMI->mi_mods_addr);␊
142␉␉if (gRAMDiskMI != NULL) free(gRAMDiskMI);␊
143␉␉␊
144␉␉// Reset multiboot structures.␊
145␉␉gRAMDiskMI = NULL;␊
146␉␉␊
147␉␉*gRAMDiskFile = '\0';␊
148␉␉␊
149␉␉// Release ramdisk driver hooks.␊
150␉␉p_get_ramdisk_info = NULL;␊
151␉␉p_ramdiskReadBytes = NULL;␊
152␉␉LoadExtraDrivers_p = NULL;␊
153␉␉␊
154␉␉// Reset ramdisk bvr␊
155␉␉gRAMDiskVolume = NULL;␊
156␉␉printf("\nunmounting: done");␊
157␉}␊
158}␊
159␊
160int mountRAMDisk(const char * param)␊
161{␊
162␉int fh = 0, ramDiskSize;␊
163␉int error = 0;␊
164␉␊
165␉// Get file handle for ramdisk file.␊
166␉fh = open(param, 0);␊
167␉if (fh != -1)␊
168␉{␊
169␉␉printf("\nreading ramdisk image: %s\n", param);␊
170␉␉␊
171␉␉ramDiskSize = file_size(fh);␊
172␉␉if (ramDiskSize > 0)␊
173␉␉{␊
174␉␉␉// Unmount previously mounted image if exists.␊
175␉␉␉umountRAMDisk();␊
176␉␉␉␊
177␉␉␉// Read new ramdisk image contents into PREBOOT_DATA area.␊
178␉␉␉if (read(fh, (char *)PREBOOT_DATA, ramDiskSize) != ramDiskSize) error = -1;␊
179␉␉}␊
180␉␉else error = -1;␊
181␉␉␊
182␉␉close(fh);␊
183␉}␊
184␉else error = -1;␊
185␉␊
186␉if (error == 0)␊
187␉{␊
188␉␉// Save filename in gRAMDiskFile to display information.␊
189␉␉strlcpy(gRAMDiskFile, param, sizeof(gRAMDiskFile));␊
190␉␉␊
191␉␉// Set gRAMDiskMI as well for the multiboot ramdisk driver hook.␊
192␉␉gRAMDiskMI = malloc(sizeof(multiboot_info));␊
193␉␉␊
194␉␉struct multiboot_module * ramdisk_module = malloc(sizeof(multiboot_module));␊
195␉␉␊
196␉␉// Fill in multiboot info and module structures.␊
197␉␉if (gRAMDiskMI != NULL && ramdisk_module != NULL)␊
198␉␉{␊
199␉␉␉bzero(gRAMDiskMI, sizeof(multiboot_info));␊
200␉␉␉bzero(ramdisk_module, sizeof(multiboot_module));␊
201␉␉␉␊
202␉␉␉gRAMDiskMI->mi_mods_count = 1;␊
203␉␉␉gRAMDiskMI->mi_mods_addr = (uint32_t)ramdisk_module;␊
204␉␉␉ramdisk_module->mm_mod_start = PREBOOT_DATA;␊
205␉␉␉ramdisk_module->mm_mod_end = PREBOOT_DATA + ramDiskSize;␊
206␉␉␉␊
207␉␉␉// Set ramdisk driver hooks.␊
208␉␉␉p_get_ramdisk_info = &multiboot_get_ramdisk_info;␊
209␉␉␉p_ramdiskReadBytes = &multibootRamdiskReadBytes;␊
210␉␉␉LoadExtraDrivers_p = &multiboot_LoadExtraDrivers;␊
211␉␉␉␊
212␉␉␉int partCount; // unused␊
213␉␉␉// Save bvr of the mounted image.␊
214␉␉␉gRAMDiskVolume = diskScanBootVolumes(0x100, &partCount);␊
215␉␉␉if(gRAMDiskVolume == NULL)␊
216␉␉␉{␊
217␉␉␉␉umountRAMDisk();␊
218␉␉␉␉printf("\nRamdisk contains no partitions.\n");␊
219␉␉␉␉pause();␊
220 ␊
221␉␉␉␉// error = -1; // ??␊
222␉␉␉␉␊
223␉␉␉}␊
224␉␉␉else␊
225␉␉␉{␊
226␉␉␉␉char dirSpec[128];␊
227␉␉␉␉␊
228␉␉␉␉// Reading ramdisk configuration.␊
229␉␉␉␉strlcpy(dirSpec, RAMDISKCONFIG_FILENAME, sizeof(dirSpec));␊
230␉␉␉␉␊
231␉␉␉␉if (loadConfigFile(dirSpec, &ramdiskConfig) == 0)␊
232␉␉␉␉{␊
233␉␉␉␉␉getBoolForKey("BTAlias", &gRAMDiskBTAliased, &ramdiskConfig);␊
234␉␉␉␉}␊
235␉␉␉␉else␊
236␉␉␉␉{␊
237␉␉␉␉␉verbose("\nno ramdisk config...\n");␊
238␉␉␉␉}␊
239␉␉␉␉␊
240␉␉␉␉printf("\nmounting: done");␊
241␉␉␉}␊
242␉␉}␊
243 else error = -1;␊
244 ␊
245 if (error == -1) {␊
246 if (ramdisk_module)␊
247 free(ramdisk_module);␊
248 else if (gRAMDiskMI)␊
249 free(gRAMDiskMI);␊
250 } ␊
251␉} ␊
252 ␊
253␉return error;␊
254}␊
255␊
256void setRAMDiskBTHook(bool mode)␊
257{␊
258␉gRAMDiskBTAliased = mode;␊
259␉if (mode)␊
260␉{␊
261␉␉printf("\nEnabled bt(0,0) alias.");␊
262␉}␊
263␉else␊
264␉{␊
265␉␉printf("\nDisabled bt(0,0) alias.");␊
266␉}␊
267}␊
268␊
269void showInfoRAMDisk(void)␊
270{␊
271␉int len;␊
272␉const char *val;␊
273␉␊
274␉if (gRAMDiskMI != NULL)␊
275␉{␊
276␉␉struct multiboot_module * ramdisk_module = (void *)gRAMDiskMI->mi_mods_addr;␊
277␉␉␊
278␉␉printf("\nfile: %s %d", gRAMDiskFile,␊
279␉␉␉ ramdisk_module->mm_mod_end - ramdisk_module->mm_mod_start);␊
280␉␉printf("\nalias: %s", gRAMDiskBTAliased ? "enabled" : "disabled");␊
281␉␉␊
282␉␉// Display ramdisk information if available.␊
283␉␉if (getValueForKey("Info", &val, &len, &ramdiskConfig))␊
284␉␉{␊
285␉␉␉printf("\ninfo: %s", val);␊
286␉␉}␊
287␉␉else␊
288␉␉{␊
289␉␉␉printf("\nramdisk info not available.");␊
290␉␉}␊
291␉}␊
292␉else␊
293␉{␊
294␉␉printf("\nNo ramdisk mounted.");␊
295␉}␊
296}␊
297␊
298int loadPrebootRAMDisk(void)␊
299{␊
300␉mountRAMDisk("bt(0,0)/Extra/Preboot.dmg");␊
301␉if (gRAMDiskMI != NULL)␊
302␉{␊
303␉␉printf("\n");␊
304␉␉return 0;␊
305␉}␊
306␉else␊
307␉{␊
308␉␉return -1;␊
309␉}␊
310}␊
311␊
312void processRAMDiskCommand(char ** argPtr, const char * cmd)␊
313{␊
314␉char * ptr = *argPtr;␊
315␉char param[1024];␊
316␉getNextArg(&ptr, param);␊
317␉␊
318␉if (strncmp(cmd, "m",sizeof("m")) == 0)␊
319␉{␊
320␉␉mountRAMDisk(param);␊
321␉␉sleep(2);␊
322␉}␊
323␉else if (strncmp(cmd, "u", sizeof("u")) == 0)␊
324␉{␊
325␉␉umountRAMDisk();␊
326␉␉sleep(2);␊
327␉}␊
328␉else if (strncmp(cmd, "e", sizeof("e")) == 0)␊
329␉{␊
330␉␉setRAMDiskBTHook(true);␊
331␉␉sleep(2);␊
332␉}␊
333␉else if (strncmp(cmd, "d", sizeof("d")) == 0)␊
334␉{␊
335␉␉setRAMDiskBTHook(false);␊
336␉␉sleep(2);␊
337␉}␊
338␉else if (strncmp(cmd, "i", sizeof("i")) == 0)␊
339␉{␊
340␉␉setActiveDisplayPage(1);␊
341␉␉clearScreenRows(0, 24);␊
342␉␉setCursorPosition(0, 0, 1);␊
343␉␉showInfoRAMDisk();␊
344␉␉printf("\n\nPress any key to continue.\n");␊
345␉␉getc();␊
346␉␉setActiveDisplayPage(0);␊
347␉}␊
348␉else␊
349␉{␊
350␉␉setActiveDisplayPage(1);␊
351␉␉clearScreenRows(0, 24);␊
352␉␉setCursorPosition(0, 0, 1);␊
353␉␉printf("\nusage:\n");␊
354␉␉printf("\n?rd i - display ramdisk information");␊
355␉␉printf("\n?rd m <filename> - mount ramdisk image\n?rd u - unmount ramdisk image");␊
356␉␉printf("\n?rd e - enable bt(0,0) alias\n?rd d - disable bt(0,0) alias");␊
357␉␉printf("\n\nPress any key to continue.\n");␊
358␉␉getc();␊
359␉␉setActiveDisplayPage(0);␊
360␉}␊
361}␊
362␊
363///////////////////////////////////////////////////////////////////////////␊
364// Ramdisk multiboot support␊
365␊
366int multibootRamdiskReadBytes( int biosdev, unsigned int blkno,␊
367␉␉␉␉␉␉␉ unsigned int byteoff,␊
368␉␉␉␉␉␉␉ unsigned int byteCount, void * buffer )␊
369{␊
370 int module_count = gRAMDiskMI->mi_mods_count;␊
371 struct multiboot_module *modules = (void*)gRAMDiskMI->mi_mods_addr;␊
372 if(biosdev < 0x100)␊
373 return -1;␊
374 if(biosdev >= (0x100 + module_count))␊
375 return -1;␊
376 struct multiboot_module *module = modules + (biosdev - 0x100);␊
377␉␊
378 void *p_initrd = (void*)module->mm_mod_start;␊
379 bcopy(p_initrd + blkno*512 + byteoff, buffer, byteCount);␊
380 return 0;␊
381}␊
382␊
383int multiboot_get_ramdisk_info(int biosdev, struct driveInfo *dip)␊
384{␊
385 int module_count = gRAMDiskMI->mi_mods_count;␊
386 struct multiboot_module *modules = (void*)gRAMDiskMI->mi_mods_addr;␊
387 if(biosdev < 0x100)␊
388 return -1;␊
389 if(biosdev >= (0x100 + module_count))␊
390 return -1;␊
391 struct multiboot_module *module = modules + (biosdev - 0x100);␊
392 dip->biosdev = biosdev;␊
393 dip->uses_ebios = true;␉// XXX aserebln uses_ebios isn't a boolean at all␊
394 dip->di.params.phys_sectors = (module->mm_mod_end - module->mm_mod_start + 511) / 512;␊
395 dip->valid = true;␊
396 return 0;␊
397}␊
398␊
399static long multiboot_LoadExtraDrivers(FileLoadDrivers_t FileLoadDrivers_p)␊
400{␊
401 char extensionsSpec[1024];␊
402 int ramdiskUnit;␊
403 long ret, flags, cachetime;␊
404␉␊
405 for(ramdiskUnit = 0; ramdiskUnit < gRAMDiskMI->mi_mods_count; ++ramdiskUnit)␊
406 {␊
407 int partCount; // unused␊
408 BVRef ramdiskChain = diskScanBootVolumes(0x100 + ramdiskUnit, &partCount);␊
409 if(ramdiskChain == NULL)␊
410 {␊
411 verbose("Ramdisk contains no partitions\n");␊
412 continue;␊
413 }␊
414 for(; ramdiskChain != NULL; ramdiskChain = ramdiskChain->next)␊
415 {␊
416 snprintf(extensionsSpec, sizeof(extensionsSpec) ,"rd(%d,%d)/Extra/", ramdiskUnit, ramdiskChain->part_no);␊
417 ␊
418 ret = GetFileInfo(NULL, extensionsSpec, &flags, &cachetime);␊
419 ␊
420 if (!ret) continue;␊
421 ␊
422 if (((flags & kFileTypeMask) != kFileTypeDirectory)) continue;␊
423 ␊
424 ret = FileLoadDrivers_p(extensionsSpec, sizeof(extensionsSpec), 0 /* this is a kext root dir, not a kext with plugins */);␊
425 if(ret != 0)␊
426 {␊
427 verbose("FileLoadDrivers failed on a ramdisk\n");␊
428 return ret;␊
429 }␊
430 }␊
431 }␊
432 return 0;␊
433}␊
434

Archive Download this file

Revision: 2182