Chameleon

Chameleon Svn Source Tree

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

Source at commit 575 created 13 years 6 months ago.
By meklort, Ramdisk 10.x extension loading
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#ifdef UNUSED␊
115␉␉gMI =␊
116#endif␊
117␉␉gRAMDiskMI = NULL;␊
118␉␉*gRAMDiskFile = '\0';␊
119␊
120␉␉// Release ramdisk driver hooks.␊
121␉␉p_get_ramdisk_info = NULL;␊
122␉␉p_ramdiskReadBytes = NULL;␊
123␊
124␉␉// Reset ramdisk bvr␊
125␉␉gRAMDiskVolume = NULL;␊
126␉␉printf("\nunmounting: done");␊
127␉}␊
128}␊
129␊
130int mountRAMDisk(const char * param)␊
131{␊
132␉int fh = 0, ramDiskSize;␊
133␉int error = 0;␊
134␊
135␉// Get file handle for ramdisk file.␊
136␉fh = open(param, 0);␊
137␉if (fh != -1)␊
138␉{␊
139␉␉printf("\nreading ramdisk image: %s", param);␊
140␊
141␉␉ramDiskSize = file_size(fh);␊
142␉␉if (ramDiskSize > 0)␊
143␉␉{␊
144␉␉␉// Unmount previously mounted image if exists.␊
145␉␉␉umountRAMDisk();␊
146␊
147␉␉␉// Read new ramdisk image contents into PREBOOT_DATA area.␊
148␉␉␉if (read(fh, (char *)PREBOOT_DATA, ramDiskSize) != ramDiskSize) error = -1;␊
149␉␉}␊
150␉␉else error = -1;␊
151␊
152␉␉close(fh);␊
153␉}␊
154␉else error = -1;␊
155␊
156␉if (error == 0)␊
157␉{␊
158␉␉// Save filename in gRAMDiskFile to display information.␊
159␉␉strcpy(gRAMDiskFile, param);␊
160␊
161␉␉// Set gMI as well for the multiboot ramdisk driver hook.␊
162#ifdef UNUSED␊
163␉␉gMI =␊
164#endif␊
165␉␉gRAMDiskMI = malloc(sizeof(multiboot_info));␊
166␉␉struct multiboot_module * ramdisk_module = malloc(sizeof(multiboot_module));␊
167␊
168␉␉// Fill in multiboot info and module structures.␊
169␉␉if (gRAMDiskMI != NULL && ramdisk_module != NULL)␊
170␉␉{␊
171␉␉␉gRAMDiskMI->mi_mods_count = 1;␊
172␉␉␉gRAMDiskMI->mi_mods_addr = (uint32_t)ramdisk_module;␊
173␉␉␉ramdisk_module->mm_mod_start = PREBOOT_DATA;␊
174␉␉␉ramdisk_module->mm_mod_end = PREBOOT_DATA + ramDiskSize;␊
175␊
176#ifdef UNUSED␊
177␉␉␉// Set ramdisk driver hooks.␊
178␉␉␉p_get_ramdisk_info = &multiboot_get_ramdisk_info;␊
179␉␉␉p_ramdiskReadBytes = &multibootRamdiskReadBytes;␊
180#endif␊
181␉␉␉int partCount; // unused␊
182␉␉␉// Save bvr of the mounted image.␊
183␉␉␉gRAMDiskVolume = diskScanBootVolumes(0x100, &partCount);␊
184␉␉␉if(gRAMDiskVolume == NULL)␊
185␉␉␉{␊
186␉␉␉␉umountRAMDisk();␊
187␉␉␉␉printf("\nRamdisk contains no partitions.");␊
188␉␉␉}␊
189␉␉␉else␊
190␉␉␉{␊
191␉␉␉␉char dirSpec[128];␊
192␊
193␉␉␉␉// Reading ramdisk configuration.␊
194␉␉␉␉strcpy(dirSpec, RAMDISKCONFIG_FILENAME);␊
195␊
196␉␉␉␉if (loadConfigFile(dirSpec, &bootInfo->ramdiskConfig) == 0)␊
197␉␉␉␉{␊
198␉␉␉␉␉getBoolForKey("BTAlias", &gRAMDiskBTAliased, &bootInfo->ramdiskConfig);␊
199␉␉␉␉}␊
200␉␉␉␉else␊
201␉␉␉␉{␊
202␉␉␉␉␉printf("\nno ramdisk config...\n");␊
203␉␉␉␉}␊
204␊
205␉␉␉␉printf("\nmounting: done");␊
206␉␉␉}␊
207␉␉}␊
208␉}␊
209␊
210␉return error;␊
211}␊
212␊
213void setRAMDiskBTHook(bool mode)␊
214{␊
215␉gRAMDiskBTAliased = mode;␊
216␉if (mode)␊
217␉{␊
218␉␉printf("\nEnabled bt(0,0) alias.");␊
219␉}␊
220␉else␊
221␉{␊
222␉␉printf("\nDisabled bt(0,0) alias.");␊
223␉}␊
224}␊
225␊
226void showInfoRAMDisk(void)␊
227{␊
228␉int len;␊
229␉const char *val;␊
230␊
231␉if (gRAMDiskMI != NULL)␊
232␉{␊
233␉␉struct multiboot_module * ramdisk_module = (void *)gRAMDiskMI->mi_mods_addr;␊
234␊
235␉␉printf("\nfile: %s %d", gRAMDiskFile,␊
236␉␉ramdisk_module->mm_mod_end - ramdisk_module->mm_mod_start);␊
237␉␉printf("\nalias: %s", gRAMDiskBTAliased ? "enabled" : "disabled");␊
238␊
239␉␉// Display ramdisk information if available.␊
240␉␉if (getValueForKey("Info", &val, &len, &bootInfo->ramdiskConfig))␊
241␉␉{␊
242␉␉␉printf("\ninfo: %s", val);␊
243␉␉}␊
244␉␉else␊
245␉␉{␊
246␉␉␉printf("\nramdisk info not available.");␊
247␉␉}␊
248␉}␊
249␉else␊
250␉{␊
251␉␉printf("\nNo ramdisk mounted.");␊
252␉}␊
253}␊
254␊
255int loadPrebootRAMDisk()␊
256{␊
257␉mountRAMDisk("bt(0,0)/Extra/Preboot.dmg");␊
258␉if (gRAMDiskMI != NULL)␊
259␉{␊
260␉␉printf("\n");␊
261␉␉return 0;␊
262␉}␊
263␉else␊
264␉{␊
265␉␉return -1;␊
266␉}␊
267}␊
268␊
269void processRAMDiskCommand(char ** argPtr, const char * cmd)␊
270{␊
271␉char * ptr = *argPtr;␊
272␉char param[1024];␊
273␉getNextArg(&ptr, param);␊
274␊
275␉if (strcmp(cmd, "m") == 0)␊
276␉{␊
277␉␉mountRAMDisk(param);␊
278␉␉sleep(2);␊
279␉}␊
280␉else if (strcmp(cmd, "u") == 0)␊
281␉{␊
282␉␉umountRAMDisk();␊
283␉␉sleep(2);␊
284␉}␊
285␉else if (strcmp(cmd, "e") == 0)␊
286␉{␊
287␉␉setRAMDiskBTHook(true);␊
288␉␉sleep(2);␊
289␉}␊
290␉else if (strcmp(cmd, "d") == 0)␊
291␉{␊
292␉␉setRAMDiskBTHook(false);␊
293␉␉sleep(2);␊
294␉}␊
295␉else if (strcmp(cmd, "i") == 0)␊
296␉{␊
297␉␉setActiveDisplayPage(1);␊
298␉␉clearScreenRows(0, 24);␊
299␉␉setCursorPosition(0, 0, 1);␊
300␉␉showInfoRAMDisk();␊
301␉␉printf("\n\nPress any key to continue.\n");␊
302␉␉getc();␊
303␉␉setActiveDisplayPage(0);␊
304␉}␊
305␉else␊
306␉{␊
307␉␉setActiveDisplayPage(1);␊
308␉␉clearScreenRows(0, 24);␊
309␉␉setCursorPosition(0, 0, 1);␊
310␉␉printf("\nusage:\n");␊
311␉␉printf("\n?rd i - display ramdisk information");␊
312␉␉printf("\n?rd m <filename> - mount ramdisk image\n?rd u - unmount ramdisk image");␊
313␉␉printf("\n?rd e - enable bt(0,0) alias\n?rd d - disable bt(0,0) alias");␊
314␉␉printf("\n\nPress any key to continue.\n");␊
315␉␉getc();␊
316␉␉setActiveDisplayPage(0);␊
317␉}␊
318}␊
319#endif␊
320

Archive Download this file

Revision: 575