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