Chameleon

Chameleon Svn Source Tree

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

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

Archive Download this file

Revision: 535