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);␊
49␉␉␊
50␉␉if(fh < 0)␊
51␉␉{␊
52␉␉␉snprintf(filename, sizeof(filename) ,"bt(0,0)/Extra/%s", override_filename);␊
53␉␉␉fh = open(filename);␊
54␉␉␉␊
55␉␉␉if(fh < 0)␊
56␉␉␉{␊
57␉␉␉␉snprintf(filename, sizeof(filename) ,"rd(0,0)/Extra/%s", override_filename);␊
58␉␉␉␉fh = open(filename);␊
59␊
60␉␉␉␉if(fh < 0)␊
61␉␉␉␉{␊
62␉␉␉␉␉snprintf(filename, sizeof(filename) ,"/Extra/%s", override_filename);␊
63␉␉␉␉␉fh = open(filename);␊
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);␊
73␉␉␊
74␉␉if(fh < 0)␊
75␉␉{␊
76␉␉␉snprintf(filename, sizeof(filename) ,"rd(0,0)/Extra/Postboot.img");␊
77␉␉␉fh = open(filename);␊
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);␊
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);␊
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␉␉␉gRAMDiskMI->mi_mods_count = 1;␊
200␉␉␉gRAMDiskMI->mi_mods_addr = (uint32_t)ramdisk_module;␊
201␉␉␉ramdisk_module->mm_mod_start = PREBOOT_DATA;␊
202␉␉␉ramdisk_module->mm_mod_end = PREBOOT_DATA + ramDiskSize;␊
203␊
204␉␉␉// Set ramdisk driver hooks.␊
205␉␉␉p_get_ramdisk_info = &multiboot_get_ramdisk_info;␊
206␉␉␉p_ramdiskReadBytes = &multibootRamdiskReadBytes;␊
207␉␉␉LoadExtraDrivers_p = &multiboot_LoadExtraDrivers;␊
208␊
209␉␉␉int partCount; // unused␊
210␉␉␉// Save bvr of the mounted image.␊
211␉␉␉gRAMDiskVolume = diskScanBootVolumes(0x100, &partCount);␊
212␉␉␉if(gRAMDiskVolume == NULL)␊
213␉␉␉{␊
214␉␉␉␉umountRAMDisk();␊
215␉␉␉␉printf("\nRamdisk contains no partitions.\n");␊
216␉␉␉␉pause();␊
217 ␊
218 // error = -1; // ??␊
219␊
220␉␉␉}␊
221␉␉␉else␊
222␉␉␉{␊
223␉␉␉␉char dirSpec[128];␊
224␊
225␉␉␉␉// Reading ramdisk configuration.␊
226␉␉␉␉strlcpy(dirSpec, RAMDISKCONFIG_FILENAME, sizeof(dirSpec));␊
227␊
228␉␉␉␉if (loadConfigFile(dirSpec, &ramdiskConfig) == 0)␊
229␉␉␉␉{␊
230␉␉␉␉␉getBoolForKey("BTAlias", &gRAMDiskBTAliased, &ramdiskConfig);␊
231␉␉␉␉}␊
232␉␉␉␉else␊
233␉␉␉␉{␊
234␉␉␉␉␉verbose("\nno ramdisk config...\n");␊
235␉␉␉␉}␊
236␊
237␉␉␉␉printf("\nmounting: done");␊
238␉␉␉}␊
239␉␉}␊
240 else error = -1;␊
241 ␊
242 if (error == -1) {␊
243 if (ramdisk_module)␊
244 free(ramdisk_module);␊
245 else if (gRAMDiskMI)␊
246 free(gRAMDiskMI);␊
247 } ␊
248␉} ␊
249 ␊
250␉return error;␊
251}␊
252␊
253void setRAMDiskBTHook(bool mode)␊
254{␊
255␉gRAMDiskBTAliased = mode;␊
256␉if (mode)␊
257␉{␊
258␉␉printf("\nEnabled bt(0,0) alias.");␊
259␉}␊
260␉else␊
261␉{␊
262␉␉printf("\nDisabled bt(0,0) alias.");␊
263␉}␊
264}␊
265␊
266void showInfoRAMDisk(void)␊
267{␊
268␉int len;␊
269␉const char *val;␊
270␊
271␉if (gRAMDiskMI != NULL)␊
272␉{␊
273␉␉struct multiboot_module * ramdisk_module = (void *)gRAMDiskMI->mi_mods_addr;␊
274␊
275␉␉printf("\nfile: %s %d", gRAMDiskFile,␊
276␉␉ramdisk_module->mm_mod_end - ramdisk_module->mm_mod_start);␊
277␉␉printf("\nalias: %s", gRAMDiskBTAliased ? "enabled" : "disabled");␊
278␊
279␉␉// Display ramdisk information if available.␊
280␉␉if (getValueForKey("Info", &val, &len, &ramdiskConfig))␊
281␉␉{␊
282␉␉␉printf("\ninfo: %s", val);␊
283␉␉}␊
284␉␉else␊
285␉␉{␊
286␉␉␉printf("\nramdisk info not available.");␊
287␉␉}␊
288␉}␊
289␉else␊
290␉{␊
291␉␉printf("\nNo ramdisk mounted.");␊
292␉}␊
293}␊
294␊
295int loadPrebootRAMDisk(void)␊
296{␊
297␉mountRAMDisk("bt(0,0)/Extra/Preboot.dmg");␊
298␉if (gRAMDiskMI != NULL)␊
299␉{␊
300␉␉printf("\n");␊
301␉␉return 0;␊
302␉}␊
303␉else␊
304␉{␊
305␉␉return -1;␊
306␉}␊
307}␊
308␊
309void processRAMDiskCommand(char ** argPtr, const char * cmd)␊
310{␊
311␉char * ptr = *argPtr;␊
312␉char param[1024];␊
313␉getNextArg(&ptr, param);␊
314␊
315␉if (strncmp(cmd, "m",sizeof("m")) == 0)␊
316␉{␊
317␉␉mountRAMDisk(param);␊
318␉␉sleep(2);␊
319␉}␊
320␉else if (strncmp(cmd, "u", sizeof("u")) == 0)␊
321␉{␊
322␉␉umountRAMDisk();␊
323␉␉sleep(2);␊
324␉}␊
325␉else if (strncmp(cmd, "e", sizeof("e")) == 0)␊
326␉{␊
327␉␉setRAMDiskBTHook(true);␊
328␉␉sleep(2);␊
329␉}␊
330␉else if (strncmp(cmd, "d", sizeof("d")) == 0)␊
331␉{␊
332␉␉setRAMDiskBTHook(false);␊
333␉␉sleep(2);␊
334␉}␊
335␉else if (strncmp(cmd, "i", sizeof("i")) == 0)␊
336␉{␊
337␉␉setActiveDisplayPage(1);␊
338␉␉clearScreenRows(0, 24);␊
339␉␉setCursorPosition(0, 0, 1);␊
340␉␉showInfoRAMDisk();␊
341␉␉printf("\n\nPress any key to continue.\n");␊
342␉␉getc();␊
343␉␉setActiveDisplayPage(0);␊
344␉}␊
345␉else␊
346␉{␊
347␉␉setActiveDisplayPage(1);␊
348␉␉clearScreenRows(0, 24);␊
349␉␉setCursorPosition(0, 0, 1);␊
350␉␉printf("\nusage:\n");␊
351␉␉printf("\n?rd i - display ramdisk information");␊
352␉␉printf("\n?rd m <filename> - mount ramdisk image\n?rd u - unmount ramdisk image");␊
353␉␉printf("\n?rd e - enable bt(0,0) alias\n?rd d - disable bt(0,0) alias");␊
354␉␉printf("\n\nPress any key to continue.\n");␊
355␉␉getc();␊
356␉␉setActiveDisplayPage(0);␊
357␉}␊
358}␊
359␊
360///////////////////////////////////////////////////////////////////////////␊
361// Ramdisk multiboot support␊
362␊
363int multibootRamdiskReadBytes( int biosdev, unsigned int blkno,␊
364␉␉␉␉␉␉␉ unsigned int byteoff,␊
365␉␉␉␉␉␉␉ unsigned int byteCount, void * buffer )␊
366{␊
367 int module_count = gRAMDiskMI->mi_mods_count;␊
368 struct multiboot_module *modules = (void*)gRAMDiskMI->mi_mods_addr;␊
369 if(biosdev < 0x100)␊
370 return -1;␊
371 if(biosdev >= (0x100 + module_count))␊
372 return -1;␊
373 struct multiboot_module *module = modules + (biosdev - 0x100);␊
374␉␊
375 void *p_initrd = (void*)module->mm_mod_start;␊
376 bcopy(p_initrd + blkno*512 + byteoff, buffer, byteCount);␊
377 return 0;␊
378}␊
379␊
380int multiboot_get_ramdisk_info(int biosdev, struct driveInfo *dip)␊
381{␊
382 int module_count = gRAMDiskMI->mi_mods_count;␊
383 struct multiboot_module *modules = (void*)gRAMDiskMI->mi_mods_addr;␊
384 if(biosdev < 0x100)␊
385 return -1;␊
386 if(biosdev >= (0x100 + module_count))␊
387 return -1;␊
388 struct multiboot_module *module = modules + (biosdev - 0x100);␊
389 dip->biosdev = biosdev;␊
390 dip->uses_ebios = true;␉// XXX aserebln uses_ebios isn't a boolean at all␊
391 dip->di.params.phys_sectors = (module->mm_mod_end - module->mm_mod_start + 511) / 512;␊
392 dip->valid = true;␊
393 return 0;␊
394}␊
395␊
396static long multiboot_LoadExtraDrivers(FileLoadDrivers_t FileLoadDrivers_p)␊
397{␊
398 char extensionsSpec[1024];␊
399 int ramdiskUnit;␊
400 long ret, flags, cachetime;␊
401␉␊
402 for(ramdiskUnit = 0; ramdiskUnit < gRAMDiskMI->mi_mods_count; ++ramdiskUnit)␊
403 {␊
404 int partCount; // unused␊
405 BVRef ramdiskChain = diskScanBootVolumes(0x100 + ramdiskUnit, &partCount);␊
406 if(ramdiskChain == NULL)␊
407 {␊
408 verbose("Ramdisk contains no partitions\n");␊
409 continue;␊
410 }␊
411 for(; ramdiskChain != NULL; ramdiskChain = ramdiskChain->next)␊
412 {␊
413 snprintf(extensionsSpec, sizeof(extensionsSpec) ,"rd(%d,%d)/Extra/", ramdiskUnit, ramdiskChain->part_no);␊
414 ␊
415 ret = GetFileInfo(NULL, extensionsSpec, &flags, &cachetime);␊
416 ␊
417 if (!ret) continue;␊
418 ␊
419 if (((flags & kFileTypeMask) != kFileTypeDirectory)) continue;␊
420 ␊
421 ret = FileLoadDrivers_p(extensionsSpec, sizeof(extensionsSpec), 0 /* this is a kext root dir, not a kext with plugins */);␊
422 if(ret != 0)␊
423 {␊
424 verbose("FileLoadDrivers failed on a ramdisk\n");␊
425 return ret;␊
426 }␊
427 }␊
428 }␊
429 return 0;␊
430}␊
431

Archive Download this file

Revision: 2118