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 | ␊ |
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 will 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(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 | ␊ |
84 | void 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 | ␊ |
111 | int 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 | ␊ |
190 | void 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 | ␊ |
203 | void 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 | ␊ |
232 | int 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 | ␊ |
247 | void 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();␊ |
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();␊ |
295 | ␉␉setActiveDisplayPage(0);␊ |
296 | ␉}␊ |
297 | }␊ |
298 | |