Chameleon

Chameleon Svn Source Tree

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

Archive Download this file

Revision: 1146