1 | /*␊ |
2 | * Supplemental ramdisk functions for the multiboot ramdisk driver.␊ |
3 | * Copyright 2009 Tamas Kosarszky. All rights reserved.␊ |
4 | *␊ |
5 | */␊ |
6 | ␊ |
7 | #include "libsaio.h"␊ |
8 | #include "bootstruct.h"␊ |
9 | #include "multiboot.h"␊ |
10 | #include "ramdisk.h"␊ |
11 | ␊ |
12 | struct multiboot_info * gRAMDiskMI = NULL;␊ |
13 | ␊ |
14 | config_file_t ramdiskConfig;␉␉ ␊ |
15 | ␊ |
16 | // gRAMDiskVolume holds the bvr for the mounted ramdisk image.␊ |
17 | BVRef gRAMDiskVolume = NULL;␊ |
18 | bool gRAMDiskBTAliased = false;␊ |
19 | char gRAMDiskFile[512];␊ |
20 | ␊ |
21 | long (*LoadExtraDrivers_p)(FileLoadDrivers_t FileLoadDrivers_p) = NULL;␊ |
22 | int (*p_ramdiskReadBytes)( int biosdev, unsigned int blkno,␊ |
23 | ␉␉␉␉␉␉ unsigned int byteoff,␊ |
24 | ␉␉␉␉␉␉ unsigned int byteCount, void * buffer ) = NULL;␊ |
25 | int (*p_get_ramdisk_info)(int biosdev, struct driveInfo *dip) = NULL;␊ |
26 | ␊ |
27 | int multibootRamdiskReadBytes( int biosdev, unsigned int blkno,␊ |
28 | ␉␉␉␉␉␉␉ unsigned int byteoff,␊ |
29 | ␉␉␉␉␉␉␉ unsigned int byteCount, void * buffer );␊ |
30 | int multiboot_get_ramdisk_info(int biosdev, struct driveInfo *dip);␊ |
31 | static long multiboot_LoadExtraDrivers(FileLoadDrivers_t FileLoadDrivers_p);␊ |
32 | ␊ |
33 | ␊ |
34 | // Notify OS X that a ramdisk has been setup. XNU with attach this to /dev/md0 (WARNING : md0Ramdisk MUST BE EXTERN FOR NBI COMPTATIBILITY)␊ |
35 | void md0Ramdisk()␊ |
36 | {␊ |
37 | ␉RAMDiskParam ramdiskPtr;␊ |
38 | ␉char filename[512];␊ |
39 | ␉const char* override_filename = 0;␊ |
40 | ␉int fh = -1;␊ |
41 | ␉int len;␊ |
42 | ␉␊ |
43 | ␉if(getValueForKey(kMD0Image, &override_filename, &len, ␊ |
44 | ␉␉␉␉␉ DEFAULT_BOOT_CONFIG))␊ |
45 | ␉{␊ |
46 | ␉␉// Use user specified md0 file␊ |
47 | ␉␉snprintf(filename, sizeof(filename) ,"%s", override_filename);␊ |
48 | ␉␉fh = open(filename, 0);␊ |
49 | ␉␉␊ |
50 | ␉␉if(fh < 0)␊ |
51 | ␉␉{␊ |
52 | ␉␉␉snprintf(filename, sizeof(filename) ,"bt(0,0)/Extra/%s", override_filename);␊ |
53 | ␉␉␉fh = open(filename, 0);␊ |
54 | ␉␉␉␊ |
55 | ␉␉␉if(fh < 0)␊ |
56 | ␉␉␉{␊ |
57 | ␉␉␉␉snprintf(filename, sizeof(filename) ,"rd(0,0)/Extra/%s", override_filename);␊ |
58 | ␉␉␉␉fh = open(filename, 0);␊ |
59 | ␉␉␉␉␊ |
60 | ␉␉␉␉if(fh < 0)␊ |
61 | ␉␉␉␉{␊ |
62 | ␉␉␉␉␉snprintf(filename, sizeof(filename) ,"/Extra/%s", override_filename);␊ |
63 | ␉␉␉␉␉fh = open(filename, 0);␊ |
64 | ␉␉␉␉}␊ |
65 | ␉␉␉}␊ |
66 | ␉␉}␊ |
67 | ␉}␊ |
68 | ␉␊ |
69 | ␉if(fh < 0)␊ |
70 | ␉{␊ |
71 | ␉␉snprintf(filename, sizeof(filename) ,"bt(0,0)/Extra/Postboot.img");␊ |
72 | ␉␉fh = open(filename, 0);␊ |
73 | ␉␉␊ |
74 | ␉␉if(fh < 0)␊ |
75 | ␉␉{␊ |
76 | ␉␉␉snprintf(filename, sizeof(filename) ,"rd(0,0)/Extra/Postboot.img");␊ |
77 | ␉␉␉fh = open(filename, 0);␊ |
78 | ␉␉␉␊ |
79 | ␉␉␉if(fh < 0)␊ |
80 | ␉␉␉{␊ |
81 | ␉␉␉␉snprintf(filename, sizeof(filename) ,"/Extra/Postboot.img");␉// Check /Extra if not in rd(0,0)␊ |
82 | ␉␉␉␉fh = open(filename, 0);␊ |
83 | ␉␉␉}␊ |
84 | ␉␉}␊ |
85 | ␉}␊ |
86 | ␉␊ |
87 | ␉if (fh >= 0)␊ |
88 | ␉{␊ |
89 | ␉␉verbose("Enabling ramdisk %s\n", filename);␊ |
90 | ␉␉␊ |
91 | ␉␉ramdiskPtr.size = file_size(fh);␊ |
92 | ␉␉ramdiskPtr.base = AllocateKernelMemory(ramdiskPtr.size);␊ |
93 | ␉␉␊ |
94 | ␉␉if(ramdiskPtr.size && ramdiskPtr.base)␊ |
95 | ␉␉{␊ |
96 | ␉␉␉// Read new ramdisk image contents in kernel memory.␊ |
97 | ␉␉␉if (read(fh, (char*) ramdiskPtr.base, ramdiskPtr.size) == ramdiskPtr.size)␊ |
98 | ␉␉␉{␉␉␉␉␊ |
99 | ␉␉␉␉␊ |
100 | AllocateMemoryRange("RAMDisk", ramdiskPtr.base, ramdiskPtr.size);␊ |
101 | ␉␉␉␉␊ |
102 | ␉␉␉␉Node* node = DT__FindNode("/chosen/memory-map", false);␊ |
103 | ␉␉␉␉if(node != NULL)␊ |
104 | ␉␉␉␉{␊ |
105 | ␉␉␉␉␉DT__AddProperty(node, "RAMDisk", sizeof(RAMDiskParam), (void*)&ramdiskPtr);␉␉␊ |
106 | ␉␉␉␉}␊ |
107 | ␉␉␉␉else␊ |
108 | ␉␉␉␉{␊ |
109 | ␉␉␉␉␉verbose("Unable to notify Mac OS X of the ramdisk %s.\n", filename);␊ |
110 | ␉␉␉␉}␊ |
111 | ␉␉␉}␊ |
112 | ␉␉␉else␊ |
113 | ␉␉␉{␊ |
114 | ␉␉␉␉verbose("Unable to read md0 image %s.\n", filename);␊ |
115 | ␉␉␉}␉␉␉␊ |
116 | ␉␉}␊ |
117 | ␉␉else␊ |
118 | ␉␉{␊ |
119 | ␉␉␉verbose("md0 image %s is empty.\n", filename);␊ |
120 | ␉␉}␊ |
121 | ␉␉␊ |
122 | ␉␉close(fh);␊ |
123 | ␉␉␊ |
124 | ␉}␊ |
125 | }␊ |
126 | ␊ |
127 | void umountRAMDisk(void)␊ |
128 | {␊ |
129 | ␉if (gRAMDiskMI != NULL)␊ |
130 | ␉{␊ |
131 | ␉␉// Release ramdisk BVRef and DiskBVMap.␊ |
132 | ␉␉struct DiskBVMap *oldMap = diskResetBootVolumes(0x100);␊ |
133 | if (oldMap == NULL)␊ |
134 | {␊ |
135 | return;␊ |
136 | }␊ |
137 | ␉␉CacheReset();␊ |
138 | ␉␉diskFreeMap(oldMap);␊ |
139 | ␉␉␊ |
140 | ␉␉// Free multiboot info and module structures.␊ |
141 | ␉␉if ((void *)gRAMDiskMI->mi_mods_addr != NULL) free((void *)gRAMDiskMI->mi_mods_addr);␊ |
142 | ␉␉if (gRAMDiskMI != NULL) free(gRAMDiskMI);␊ |
143 | ␉␉␊ |
144 | ␉␉// Reset multiboot structures.␊ |
145 | ␉␉gRAMDiskMI = NULL;␊ |
146 | ␉␉␊ |
147 | ␉␉*gRAMDiskFile = '\0';␊ |
148 | ␉␉␊ |
149 | ␉␉// Release ramdisk driver hooks.␊ |
150 | ␉␉p_get_ramdisk_info = NULL;␊ |
151 | ␉␉p_ramdiskReadBytes = NULL;␊ |
152 | ␉␉LoadExtraDrivers_p = NULL;␊ |
153 | ␉␉␊ |
154 | ␉␉// Reset ramdisk bvr␊ |
155 | ␉␉gRAMDiskVolume = NULL;␊ |
156 | ␉␉printf("\nunmounting: done");␊ |
157 | ␉}␊ |
158 | }␊ |
159 | ␊ |
160 | int mountRAMDisk(const char * param)␊ |
161 | {␊ |
162 | ␉int fh = 0, ramDiskSize;␊ |
163 | ␉int error = 0;␊ |
164 | ␉␊ |
165 | ␉// Get file handle for ramdisk file.␊ |
166 | ␉fh = open(param, 0);␊ |
167 | ␉if (fh != -1)␊ |
168 | ␉{␊ |
169 | ␉␉printf("\nreading ramdisk image: %s\n", param);␊ |
170 | ␉␉␊ |
171 | ␉␉ramDiskSize = file_size(fh);␊ |
172 | ␉␉if (ramDiskSize > 0)␊ |
173 | ␉␉{␊ |
174 | ␉␉␉// Unmount previously mounted image if exists.␊ |
175 | ␉␉␉umountRAMDisk();␊ |
176 | ␉␉␉␊ |
177 | ␉␉␉// Read new ramdisk image contents into PREBOOT_DATA area.␊ |
178 | ␉␉␉if (read(fh, (char *)PREBOOT_DATA, ramDiskSize) != ramDiskSize) error = -1;␊ |
179 | ␉␉}␊ |
180 | ␉␉else error = -1;␊ |
181 | ␉␉␊ |
182 | ␉␉close(fh);␊ |
183 | ␉}␊ |
184 | ␉else error = -1;␊ |
185 | ␉␊ |
186 | ␉if (error == 0)␊ |
187 | ␉{␊ |
188 | ␉␉// Save filename in gRAMDiskFile to display information.␊ |
189 | ␉␉strlcpy(gRAMDiskFile, param, sizeof(gRAMDiskFile));␊ |
190 | ␉␉␊ |
191 | ␉␉// Set gRAMDiskMI as well for the multiboot ramdisk driver hook.␊ |
192 | ␉␉gRAMDiskMI = malloc(sizeof(multiboot_info));␊ |
193 | ␉␉␊ |
194 | ␉␉struct multiboot_module * ramdisk_module = malloc(sizeof(multiboot_module));␊ |
195 | ␉␉␊ |
196 | ␉␉// Fill in multiboot info and module structures.␊ |
197 | ␉␉if (gRAMDiskMI != NULL && ramdisk_module != NULL)␊ |
198 | ␉␉{␊ |
199 | ␉␉␉bzero(gRAMDiskMI, sizeof(multiboot_info));␊ |
200 | ␉␉␉bzero(ramdisk_module, sizeof(multiboot_module));␊ |
201 | ␉␉␉␊ |
202 | ␉␉␉gRAMDiskMI->mi_mods_count = 1;␊ |
203 | ␉␉␉gRAMDiskMI->mi_mods_addr = (uint32_t)ramdisk_module;␊ |
204 | ␉␉␉ramdisk_module->mm_mod_start = PREBOOT_DATA;␊ |
205 | ␉␉␉ramdisk_module->mm_mod_end = PREBOOT_DATA + ramDiskSize;␊ |
206 | ␉␉␉␊ |
207 | ␉␉␉// Set ramdisk driver hooks.␊ |
208 | ␉␉␉p_get_ramdisk_info = &multiboot_get_ramdisk_info;␊ |
209 | ␉␉␉p_ramdiskReadBytes = &multibootRamdiskReadBytes;␊ |
210 | ␉␉␉LoadExtraDrivers_p = &multiboot_LoadExtraDrivers;␊ |
211 | ␉␉␉␊ |
212 | ␉␉␉int partCount; // unused␊ |
213 | ␉␉␉// Save bvr of the mounted image.␊ |
214 | ␉␉␉gRAMDiskVolume = diskScanBootVolumes(0x100, &partCount);␊ |
215 | ␉␉␉if(gRAMDiskVolume == NULL)␊ |
216 | ␉␉␉{␊ |
217 | ␉␉␉␉umountRAMDisk();␊ |
218 | ␉␉␉␉printf("\nRamdisk contains no partitions.\n");␊ |
219 | ␉␉␉␉pause();␊ |
220 | ␊ |
221 | ␉␉␉␉// error = -1; // ??␊ |
222 | ␉␉␉␉␊ |
223 | ␉␉␉}␊ |
224 | ␉␉␉else␊ |
225 | ␉␉␉{␊ |
226 | ␉␉␉␉char dirSpec[128];␊ |
227 | ␉␉␉␉␊ |
228 | ␉␉␉␉// Reading ramdisk configuration.␊ |
229 | ␉␉␉␉strlcpy(dirSpec, RAMDISKCONFIG_FILENAME, sizeof(dirSpec));␊ |
230 | ␉␉␉␉␊ |
231 | ␉␉␉␉if (loadConfigFile(dirSpec, &ramdiskConfig) == 0)␊ |
232 | ␉␉␉␉{␊ |
233 | ␉␉␉␉␉getBoolForKey("BTAlias", &gRAMDiskBTAliased, &ramdiskConfig);␊ |
234 | ␉␉␉␉}␊ |
235 | ␉␉␉␉else␊ |
236 | ␉␉␉␉{␊ |
237 | ␉␉␉␉␉verbose("\nno ramdisk config...\n");␊ |
238 | ␉␉␉␉}␊ |
239 | ␉␉␉␉␊ |
240 | ␉␉␉␉printf("\nmounting: done");␊ |
241 | ␉␉␉}␊ |
242 | ␉␉}␊ |
243 | else error = -1;␊ |
244 | ␊ |
245 | if (error == -1) {␊ |
246 | if (ramdisk_module)␊ |
247 | free(ramdisk_module);␊ |
248 | else if (gRAMDiskMI)␊ |
249 | free(gRAMDiskMI);␊ |
250 | } ␊ |
251 | ␉} ␊ |
252 | ␊ |
253 | ␉return error;␊ |
254 | }␊ |
255 | ␊ |
256 | void setRAMDiskBTHook(bool mode)␊ |
257 | {␊ |
258 | ␉gRAMDiskBTAliased = mode;␊ |
259 | ␉if (mode)␊ |
260 | ␉{␊ |
261 | ␉␉printf("\nEnabled bt(0,0) alias.");␊ |
262 | ␉}␊ |
263 | ␉else␊ |
264 | ␉{␊ |
265 | ␉␉printf("\nDisabled bt(0,0) alias.");␊ |
266 | ␉}␊ |
267 | }␊ |
268 | ␊ |
269 | void showInfoRAMDisk(void)␊ |
270 | {␊ |
271 | ␉int len;␊ |
272 | ␉const char *val;␊ |
273 | ␉␊ |
274 | ␉if (gRAMDiskMI != NULL)␊ |
275 | ␉{␊ |
276 | ␉␉struct multiboot_module * ramdisk_module = (void *)gRAMDiskMI->mi_mods_addr;␊ |
277 | ␉␉␊ |
278 | ␉␉printf("\nfile: %s %d", gRAMDiskFile,␊ |
279 | ␉␉␉ ramdisk_module->mm_mod_end - ramdisk_module->mm_mod_start);␊ |
280 | ␉␉printf("\nalias: %s", gRAMDiskBTAliased ? "enabled" : "disabled");␊ |
281 | ␉␉␊ |
282 | ␉␉// Display ramdisk information if available.␊ |
283 | ␉␉if (getValueForKey("Info", &val, &len, &ramdiskConfig))␊ |
284 | ␉␉{␊ |
285 | ␉␉␉printf("\ninfo: %s", val);␊ |
286 | ␉␉}␊ |
287 | ␉␉else␊ |
288 | ␉␉{␊ |
289 | ␉␉␉printf("\nramdisk info not available.");␊ |
290 | ␉␉}␊ |
291 | ␉}␊ |
292 | ␉else␊ |
293 | ␉{␊ |
294 | ␉␉printf("\nNo ramdisk mounted.");␊ |
295 | ␉}␊ |
296 | }␊ |
297 | ␊ |
298 | int loadPrebootRAMDisk(void)␊ |
299 | {␊ |
300 | ␉mountRAMDisk("bt(0,0)/Extra/Preboot.dmg");␊ |
301 | ␉if (gRAMDiskMI != NULL)␊ |
302 | ␉{␊ |
303 | ␉␉printf("\n");␊ |
304 | ␉␉return 0;␊ |
305 | ␉}␊ |
306 | ␉else␊ |
307 | ␉{␊ |
308 | ␉␉return -1;␊ |
309 | ␉}␊ |
310 | }␊ |
311 | ␊ |
312 | void processRAMDiskCommand(char ** argPtr, const char * cmd)␊ |
313 | {␊ |
314 | ␉char * ptr = *argPtr;␊ |
315 | ␉char param[1024];␊ |
316 | ␉getNextArg(&ptr, param);␊ |
317 | ␉␊ |
318 | ␉if (strncmp(cmd, "m",sizeof("m")) == 0)␊ |
319 | ␉{␊ |
320 | ␉␉mountRAMDisk(param);␊ |
321 | ␉␉sleep(2);␊ |
322 | ␉}␊ |
323 | ␉else if (strncmp(cmd, "u", sizeof("u")) == 0)␊ |
324 | ␉{␊ |
325 | ␉␉umountRAMDisk();␊ |
326 | ␉␉sleep(2);␊ |
327 | ␉}␊ |
328 | ␉else if (strncmp(cmd, "e", sizeof("e")) == 0)␊ |
329 | ␉{␊ |
330 | ␉␉setRAMDiskBTHook(true);␊ |
331 | ␉␉sleep(2);␊ |
332 | ␉}␊ |
333 | ␉else if (strncmp(cmd, "d", sizeof("d")) == 0)␊ |
334 | ␉{␊ |
335 | ␉␉setRAMDiskBTHook(false);␊ |
336 | ␉␉sleep(2);␊ |
337 | ␉}␊ |
338 | ␉else if (strncmp(cmd, "i", sizeof("i")) == 0)␊ |
339 | ␉{␊ |
340 | ␉␉setActiveDisplayPage(1);␊ |
341 | ␉␉clearScreenRows(0, 24);␊ |
342 | ␉␉setCursorPosition(0, 0, 1);␊ |
343 | ␉␉showInfoRAMDisk();␊ |
344 | ␉␉printf("\n\nPress any key to continue.\n");␊ |
345 | ␉␉getc();␊ |
346 | ␉␉setActiveDisplayPage(0);␊ |
347 | ␉}␊ |
348 | ␉else␊ |
349 | ␉{␊ |
350 | ␉␉setActiveDisplayPage(1);␊ |
351 | ␉␉clearScreenRows(0, 24);␊ |
352 | ␉␉setCursorPosition(0, 0, 1);␊ |
353 | ␉␉printf("\nusage:\n");␊ |
354 | ␉␉printf("\n?rd i - display ramdisk information");␊ |
355 | ␉␉printf("\n?rd m <filename> - mount ramdisk image\n?rd u - unmount ramdisk image");␊ |
356 | ␉␉printf("\n?rd e - enable bt(0,0) alias\n?rd d - disable bt(0,0) alias");␊ |
357 | ␉␉printf("\n\nPress any key to continue.\n");␊ |
358 | ␉␉getc();␊ |
359 | ␉␉setActiveDisplayPage(0);␊ |
360 | ␉}␊ |
361 | }␊ |
362 | ␊ |
363 | ///////////////////////////////////////////////////////////////////////////␊ |
364 | // Ramdisk multiboot support␊ |
365 | ␊ |
366 | int multibootRamdiskReadBytes( int biosdev, unsigned int blkno,␊ |
367 | ␉␉␉␉␉␉␉ unsigned int byteoff,␊ |
368 | ␉␉␉␉␉␉␉ unsigned int byteCount, void * buffer )␊ |
369 | {␊ |
370 | int module_count = gRAMDiskMI->mi_mods_count;␊ |
371 | struct multiboot_module *modules = (void*)gRAMDiskMI->mi_mods_addr;␊ |
372 | if(biosdev < 0x100)␊ |
373 | return -1;␊ |
374 | if(biosdev >= (0x100 + module_count))␊ |
375 | return -1;␊ |
376 | struct multiboot_module *module = modules + (biosdev - 0x100);␊ |
377 | ␉␊ |
378 | void *p_initrd = (void*)module->mm_mod_start;␊ |
379 | bcopy(p_initrd + blkno*512 + byteoff, buffer, byteCount);␊ |
380 | return 0;␊ |
381 | }␊ |
382 | ␊ |
383 | int multiboot_get_ramdisk_info(int biosdev, struct driveInfo *dip)␊ |
384 | {␊ |
385 | int module_count = gRAMDiskMI->mi_mods_count;␊ |
386 | struct multiboot_module *modules = (void*)gRAMDiskMI->mi_mods_addr;␊ |
387 | if(biosdev < 0x100)␊ |
388 | return -1;␊ |
389 | if(biosdev >= (0x100 + module_count))␊ |
390 | return -1;␊ |
391 | struct multiboot_module *module = modules + (biosdev - 0x100);␊ |
392 | dip->biosdev = biosdev;␊ |
393 | dip->uses_ebios = true;␉// XXX aserebln uses_ebios isn't a boolean at all␊ |
394 | dip->di.params.phys_sectors = (module->mm_mod_end - module->mm_mod_start + 511) / 512;␊ |
395 | dip->valid = true;␊ |
396 | return 0;␊ |
397 | }␊ |
398 | ␊ |
399 | static long multiboot_LoadExtraDrivers(FileLoadDrivers_t FileLoadDrivers_p)␊ |
400 | {␊ |
401 | char extensionsSpec[1024];␊ |
402 | int ramdiskUnit;␊ |
403 | long ret, flags, cachetime;␊ |
404 | ␉␊ |
405 | for(ramdiskUnit = 0; ramdiskUnit < gRAMDiskMI->mi_mods_count; ++ramdiskUnit)␊ |
406 | {␊ |
407 | int partCount; // unused␊ |
408 | BVRef ramdiskChain = diskScanBootVolumes(0x100 + ramdiskUnit, &partCount);␊ |
409 | if(ramdiskChain == NULL)␊ |
410 | {␊ |
411 | verbose("Ramdisk contains no partitions\n");␊ |
412 | continue;␊ |
413 | }␊ |
414 | for(; ramdiskChain != NULL; ramdiskChain = ramdiskChain->next)␊ |
415 | {␊ |
416 | snprintf(extensionsSpec, sizeof(extensionsSpec) ,"rd(%d,%d)/Extra/", ramdiskUnit, ramdiskChain->part_no);␊ |
417 | ␊ |
418 | ret = GetFileInfo(NULL, extensionsSpec, &flags, &cachetime);␊ |
419 | ␊ |
420 | if (!ret) continue;␊ |
421 | ␊ |
422 | if (((flags & kFileTypeMask) != kFileTypeDirectory)) continue;␊ |
423 | ␊ |
424 | ret = FileLoadDrivers_p(extensionsSpec, sizeof(extensionsSpec), 0 /* this is a kext root dir, not a kext with plugins */);␊ |
425 | if(ret != 0)␊ |
426 | {␊ |
427 | verbose("FileLoadDrivers failed on a ramdisk\n");␊ |
428 | return ret;␊ |
429 | }␊ |
430 | }␊ |
431 | }␊ |
432 | return 0;␊ |
433 | }␊ |
434 | |