Chameleon

Chameleon Svn Source Tree

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

Archive Download this file

Revision: 520