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

Archive Download this file

Revision: 1023