Chameleon

Chameleon Svn Source Tree

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

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

Archive Download this file

Revision: 1146