]> code.delx.au - refind/blob - refind/config.c
Added new "scan_delay" feature.
[refind] / refind / config.c
1 /*
2 * refit/config.c
3 * Configuration file functions
4 *
5 * Copyright (c) 2006 Christoph Pfisterer
6 * All rights reserved.
7 *
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions are
10 * met:
11 *
12 * * Redistributions of source code must retain the above copyright
13 * notice, this list of conditions and the following disclaimer.
14 *
15 * * Redistributions in binary form must reproduce the above copyright
16 * notice, this list of conditions and the following disclaimer in the
17 * documentation and/or other materials provided with the
18 * distribution.
19 *
20 * * Neither the name of Christoph Pfisterer nor the names of the
21 * contributors may be used to endorse or promote products derived
22 * from this software without specific prior written permission.
23 *
24 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
25 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
26 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
27 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
28 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
29 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
30 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
31 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
32 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
33 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
34 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
35 */
36
37 /*
38 * Modifications copyright (c) 2012 Roderick W. Smith
39 *
40 * Modifications distributed under the terms of the GNU General Public
41 * License (GPL) version 3 (GPLv3), a copy of which must be distributed
42 * with this source code or binaries made from it.
43 *
44 */
45
46 #include "global.h"
47 #include "lib.h"
48 #include "icns.h"
49 #include "menu.h"
50 #include "config.h"
51 #include "screen.h"
52 #include "../include/refit_call_wrapper.h"
53
54 // constants
55
56 #define CONFIG_FILE_NAME L"refind.conf"
57 #define LINUX_OPTIONS_FILENAMES L"refind_linux.conf,refind-linux.conf"
58 #define MAXCONFIGFILESIZE (128*1024)
59
60 #define ENCODING_ISO8859_1 (0)
61 #define ENCODING_UTF8 (1)
62 #define ENCODING_UTF16_LE (2)
63
64 static REFIT_MENU_ENTRY MenuEntryReturn = { L"Return to Main Menu", TAG_RETURN, 0, 0, 0, NULL, NULL, NULL };
65
66 //
67 // read a file into a buffer
68 //
69
70 static EFI_STATUS ReadFile(IN EFI_FILE_HANDLE BaseDir, CHAR16 *FileName, REFIT_FILE *File)
71 {
72 EFI_STATUS Status;
73 EFI_FILE_HANDLE FileHandle;
74 EFI_FILE_INFO *FileInfo;
75 UINT64 ReadSize;
76
77 File->Buffer = NULL;
78 File->BufferSize = 0;
79
80 // read the file, allocating a buffer on the way
81 Status = refit_call5_wrapper(BaseDir->Open, BaseDir, &FileHandle, FileName, EFI_FILE_MODE_READ, 0);
82 if (CheckError(Status, L"while loading the configuration file"))
83 return Status;
84
85 FileInfo = LibFileInfo(FileHandle);
86 if (FileInfo == NULL) {
87 // TODO: print and register the error
88 refit_call1_wrapper(FileHandle->Close, FileHandle);
89 return EFI_LOAD_ERROR;
90 }
91 ReadSize = FileInfo->FileSize;
92 if (ReadSize > MAXCONFIGFILESIZE)
93 ReadSize = MAXCONFIGFILESIZE;
94 FreePool(FileInfo);
95
96 File->BufferSize = (UINTN)ReadSize; // was limited to a few K before, so this is safe
97 File->Buffer = AllocatePool(File->BufferSize);
98 Status = refit_call3_wrapper(FileHandle->Read, FileHandle, &File->BufferSize, File->Buffer);
99 if (CheckError(Status, L"while loading the configuration file")) {
100 FreePool(File->Buffer);
101 File->Buffer = NULL;
102 refit_call1_wrapper(FileHandle->Close, FileHandle);
103 return Status;
104 }
105 Status = refit_call1_wrapper(FileHandle->Close, FileHandle);
106
107 // setup for reading
108 File->Current8Ptr = (CHAR8 *)File->Buffer;
109 File->End8Ptr = File->Current8Ptr + File->BufferSize;
110 File->Current16Ptr = (CHAR16 *)File->Buffer;
111 File->End16Ptr = File->Current16Ptr + (File->BufferSize >> 1);
112
113 // detect encoding
114 File->Encoding = ENCODING_ISO8859_1; // default: 1:1 translation of CHAR8 to CHAR16
115 if (File->BufferSize >= 4) {
116 if (File->Buffer[0] == 0xFF && File->Buffer[1] == 0xFE) {
117 // BOM in UTF-16 little endian (or UTF-32 little endian)
118 File->Encoding = ENCODING_UTF16_LE; // use CHAR16 as is
119 File->Current16Ptr++;
120 } else if (File->Buffer[0] == 0xEF && File->Buffer[1] == 0xBB && File->Buffer[2] == 0xBF) {
121 // BOM in UTF-8
122 File->Encoding = ENCODING_UTF8; // translate from UTF-8 to UTF-16
123 File->Current8Ptr += 3;
124 } else if (File->Buffer[1] == 0 && File->Buffer[3] == 0) {
125 File->Encoding = ENCODING_UTF16_LE; // use CHAR16 as is
126 }
127 // TODO: detect other encodings as they are implemented
128 }
129
130 return EFI_SUCCESS;
131 }
132
133 //
134 // get a single line of text from a file
135 //
136
137 static CHAR16 *ReadLine(REFIT_FILE *File)
138 {
139 CHAR16 *Line, *q;
140 UINTN LineLength;
141
142 if (File->Buffer == NULL)
143 return NULL;
144
145 if (File->Encoding == ENCODING_ISO8859_1 || File->Encoding == ENCODING_UTF8) {
146
147 CHAR8 *p, *LineStart, *LineEnd;
148
149 p = File->Current8Ptr;
150 if (p >= File->End8Ptr)
151 return NULL;
152
153 LineStart = p;
154 for (; p < File->End8Ptr; p++)
155 if (*p == 13 || *p == 10)
156 break;
157 LineEnd = p;
158 for (; p < File->End8Ptr; p++)
159 if (*p != 13 && *p != 10)
160 break;
161 File->Current8Ptr = p;
162
163 LineLength = (UINTN)(LineEnd - LineStart) + 1;
164 Line = AllocatePool(LineLength * sizeof(CHAR16));
165 if (Line == NULL)
166 return NULL;
167
168 q = Line;
169 if (File->Encoding == ENCODING_ISO8859_1) {
170 for (p = LineStart; p < LineEnd; )
171 *q++ = *p++;
172 } else if (File->Encoding == ENCODING_UTF8) {
173 // TODO: actually handle UTF-8
174 for (p = LineStart; p < LineEnd; )
175 *q++ = *p++;
176 }
177 *q = 0;
178
179 } else if (File->Encoding == ENCODING_UTF16_LE) {
180
181 CHAR16 *p, *LineStart, *LineEnd;
182
183 p = File->Current16Ptr;
184 if (p >= File->End16Ptr)
185 return NULL;
186
187 LineStart = p;
188 for (; p < File->End16Ptr; p++)
189 if (*p == 13 || *p == 10)
190 break;
191 LineEnd = p;
192 for (; p < File->End16Ptr; p++)
193 if (*p != 13 && *p != 10)
194 break;
195 File->Current16Ptr = p;
196
197 LineLength = (UINTN)(LineEnd - LineStart) + 1;
198 Line = AllocatePool(LineLength * sizeof(CHAR16));
199 if (Line == NULL)
200 return NULL;
201
202 for (p = LineStart, q = Line; p < LineEnd; )
203 *q++ = *p++;
204 *q = 0;
205
206 } else
207 return NULL; // unsupported encoding
208
209 return Line;
210 }
211
212 //
213 // get a line of tokens from a file
214 //
215
216 UINTN ReadTokenLine(IN REFIT_FILE *File, OUT CHAR16 ***TokenList)
217 {
218 BOOLEAN LineFinished, IsQuoted = FALSE;
219 CHAR16 *Line, *Token, *p;
220 UINTN TokenCount = 0;
221
222 *TokenList = NULL;
223
224 while (TokenCount == 0) {
225 Line = ReadLine(File);
226 if (Line == NULL)
227 return(0);
228
229 p = Line;
230 LineFinished = FALSE;
231 while (!LineFinished) {
232 // skip whitespace
233 while ((*p == ' ' || *p == '\t' || *p == '=' || *p == ',') && !IsQuoted)
234 p++;
235 if (*p == 0 || *p == '#')
236 break;
237
238 if (*p == '"') {
239 IsQuoted = !IsQuoted;
240 p++;
241 } // if
242 Token = p;
243
244 // find end of token
245 while (*p && *p != '"' && ((*p != ' ' && *p != '\t' && *p != '=' && *p != '#' && *p != ',') || IsQuoted)) {
246 if ((*p == '/') && !IsQuoted) // Switch Unix-style to DOS-style directory separators
247 *p = '\\';
248 p++;
249 } // if
250 if (*p == '"')
251 IsQuoted = !IsQuoted;
252 if (*p == 0 || *p == '#')
253 LineFinished = TRUE;
254 *p++ = 0;
255
256 AddListElement((VOID ***)TokenList, &TokenCount, (VOID *)StrDuplicate(Token));
257 }
258
259 FreePool(Line);
260 }
261 return (TokenCount);
262 } /* ReadTokenLine() */
263
264 VOID FreeTokenLine(IN OUT CHAR16 ***TokenList, IN OUT UINTN *TokenCount)
265 {
266 // TODO: also free the items
267 FreeList((VOID ***)TokenList, TokenCount);
268 }
269
270 // handle a parameter with a single integer argument
271 static VOID HandleInt(IN CHAR16 **TokenList, IN UINTN TokenCount, OUT UINTN *Value)
272 {
273 if (TokenCount == 2)
274 *Value = Atoi(TokenList[1]);
275 }
276
277 // handle a parameter with a single string argument
278 static VOID HandleString(IN CHAR16 **TokenList, IN UINTN TokenCount, OUT CHAR16 **Target) {
279 if (TokenCount == 2) {
280 if (*Target != NULL)
281 FreePool(*Target);
282 *Target = StrDuplicate(TokenList[1]);
283 } // if
284 } // static VOID HandleString()
285
286 // Handle a parameter with a series of string arguments, to be added to a comma-delimited
287 // list. Passes each token through the CleanUpPathNameSlashes() function to ensure
288 // consistency in subsequent comparisons of filenames.
289 static VOID HandleStrings(IN CHAR16 **TokenList, IN UINTN TokenCount, OUT CHAR16 **Target) {
290 UINTN i;
291
292 if (*Target != NULL) {
293 FreePool(*Target);
294 *Target = NULL;
295 } // if
296 for (i = 1; i < TokenCount; i++) {
297 CleanUpPathNameSlashes(TokenList[i]);
298 MergeStrings(Target, TokenList[i], L',');
299 }
300 } // static VOID HandleStrings()
301
302 // read config file
303 VOID ReadConfig(VOID)
304 {
305 EFI_STATUS Status;
306 REFIT_FILE File;
307 CHAR16 **TokenList;
308 CHAR16 *FlagName;
309 UINTN TokenCount, i;
310
311 if (!FileExists(SelfDir, CONFIG_FILE_NAME)) {
312 Print(L"Configuration file missing!\n");
313 return;
314 }
315
316 Status = ReadFile(SelfDir, CONFIG_FILE_NAME, &File);
317 if (EFI_ERROR(Status))
318 return;
319
320 for (;;) {
321 TokenCount = ReadTokenLine(&File, &TokenList);
322 if (TokenCount == 0)
323 break;
324
325 if (StriCmp(TokenList[0], L"timeout") == 0) {
326 HandleInt(TokenList, TokenCount, &(GlobalConfig.Timeout));
327
328 } else if (StriCmp(TokenList[0], L"hideui") == 0) {
329 for (i = 1; i < TokenCount; i++) {
330 FlagName = TokenList[i];
331 if (StriCmp(FlagName, L"banner") == 0) {
332 GlobalConfig.HideUIFlags |= HIDEUI_FLAG_BANNER;
333 } else if (StriCmp(FlagName, L"label") == 0) {
334 GlobalConfig.HideUIFlags |= HIDEUI_FLAG_LABEL;
335 } else if (StriCmp(FlagName, L"singleuser") == 0) {
336 GlobalConfig.HideUIFlags |= HIDEUI_FLAG_SINGLEUSER;
337 } else if (StriCmp(FlagName, L"hwtest") == 0) {
338 GlobalConfig.HideUIFlags |= HIDEUI_FLAG_HWTEST;
339 } else if (StriCmp(FlagName, L"arrows") == 0) {
340 GlobalConfig.HideUIFlags |= HIDEUI_FLAG_ARROWS;
341 } else if (StriCmp(FlagName, L"all") == 0) {
342 GlobalConfig.HideUIFlags = HIDEUI_ALL;
343 } else {
344 Print(L" unknown hideui flag: '%s'\n", FlagName);
345 }
346 }
347
348 } else if (StriCmp(TokenList[0], L"icons_dir") == 0) {
349 HandleString(TokenList, TokenCount, &(GlobalConfig.IconsDir));
350
351 } else if (StriCmp(TokenList[0], L"scanfor") == 0) {
352 for (i = 0; i < NUM_SCAN_OPTIONS; i++) {
353 if (i < TokenCount)
354 GlobalConfig.ScanFor[i] = TokenList[i][0];
355 else
356 GlobalConfig.ScanFor[i] = ' ';
357 }
358
359 } else if ((StriCmp(TokenList[0], L"scan_delay") == 0) && (TokenCount == 2)) {
360 HandleInt(TokenList, TokenCount, &(GlobalConfig.ScanDelay));
361
362 } else if (StriCmp(TokenList[0], L"also_scan_dirs") == 0) {
363 HandleStrings(TokenList, TokenCount, &(GlobalConfig.AlsoScan));
364
365 } else if ((StriCmp(TokenList[0], L"don't_scan_dirs") == 0) || (StriCmp(TokenList[0], L"dont_scan_dirs") == 0)) {
366 HandleStrings(TokenList, TokenCount, &(GlobalConfig.DontScan));
367
368 } else if (StriCmp(TokenList[0], L"scan_driver_dirs") == 0) {
369 HandleStrings(TokenList, TokenCount, &(GlobalConfig.DriverDirs));
370
371 } else if (StriCmp(TokenList[0], L"showtools") == 0) {
372 SetMem(GlobalConfig.ShowTools, NUM_TOOLS * sizeof(UINTN), 0);
373 for (i = 1; (i < TokenCount) && (i < NUM_TOOLS); i++) {
374 FlagName = TokenList[i];
375 if (StriCmp(FlagName, L"shell") == 0) {
376 GlobalConfig.ShowTools[i - 1] = TAG_SHELL;
377 } else if (StriCmp(FlagName, L"gptsync") == 0) {
378 GlobalConfig.ShowTools[i - 1] = TAG_GPTSYNC;
379 } else if (StriCmp(FlagName, L"about") == 0) {
380 GlobalConfig.ShowTools[i - 1] = TAG_ABOUT;
381 } else if (StriCmp(FlagName, L"exit") == 0) {
382 GlobalConfig.ShowTools[i - 1] = TAG_EXIT;
383 } else if (StriCmp(FlagName, L"reboot") == 0) {
384 GlobalConfig.ShowTools[i - 1] = TAG_REBOOT;
385 } else if (StriCmp(FlagName, L"shutdown") == 0) {
386 GlobalConfig.ShowTools[i - 1] = TAG_SHUTDOWN;
387 } else {
388 Print(L" unknown showtools flag: '%s'\n", FlagName);
389 }
390 } // showtools options
391
392 } else if (StriCmp(TokenList[0], L"banner") == 0) {
393 HandleString(TokenList, TokenCount, &(GlobalConfig.BannerFileName));
394
395 } else if (StriCmp(TokenList[0], L"selection_small") == 0) {
396 HandleString(TokenList, TokenCount, &(GlobalConfig.SelectionSmallFileName));
397
398 } else if (StriCmp(TokenList[0], L"selection_big") == 0) {
399 HandleString(TokenList, TokenCount, &(GlobalConfig.SelectionBigFileName));
400
401 } else if (StriCmp(TokenList[0], L"default_selection") == 0) {
402 HandleString(TokenList, TokenCount, &(GlobalConfig.DefaultSelection));
403
404 } else if (StriCmp(TokenList[0], L"textonly") == 0) {
405 GlobalConfig.TextOnly = TRUE;
406
407 } else if ((StriCmp(TokenList[0], L"resolution") == 0) && (TokenCount == 3)) {
408 GlobalConfig.RequestedScreenWidth = Atoi(TokenList[1]);
409 GlobalConfig.RequestedScreenHeight = Atoi(TokenList[2]);
410
411 } else if (StriCmp(TokenList[0], L"use_graphics_for") == 0) {
412 GlobalConfig.GraphicsFor = 0;
413 for (i = 1; i < TokenCount; i++) {
414 if (StriCmp(TokenList[i], L"osx") == 0) {
415 GlobalConfig.GraphicsFor |= GRAPHICS_FOR_OSX;
416 } else if (StriCmp(TokenList[i], L"linux") == 0) {
417 GlobalConfig.GraphicsFor |= GRAPHICS_FOR_LINUX;
418 } else if (StriCmp(TokenList[i], L"elilo") == 0) {
419 GlobalConfig.GraphicsFor |= GRAPHICS_FOR_ELILO;
420 } else if (StriCmp(TokenList[i], L"grub") == 0) {
421 GlobalConfig.GraphicsFor |= GRAPHICS_FOR_GRUB;
422 } else if (StriCmp(TokenList[i], L"windows") == 0) {
423 GlobalConfig.GraphicsFor |= GRAPHICS_FOR_WINDOWS;
424 }
425 } // for (graphics_on tokens)
426
427 } else if (StriCmp(TokenList[0], L"scan_all_linux_kernels") == 0) {
428 GlobalConfig.ScanAllLinux = TRUE;
429
430 } else if (StriCmp(TokenList[0], L"max_tags") == 0) {
431 HandleInt(TokenList, TokenCount, &(GlobalConfig.MaxTags));
432 }
433
434 FreeTokenLine(&TokenList, &TokenCount);
435 }
436 FreePool(File.Buffer);
437 } /* VOID ReadConfig() */
438
439 static VOID AddSubmenu(LOADER_ENTRY *Entry, REFIT_FILE *File, REFIT_VOLUME *Volume, CHAR16 *Title) {
440 REFIT_MENU_SCREEN *SubScreen;
441 LOADER_ENTRY *SubEntry;
442 UINTN TokenCount;
443 CHAR16 **TokenList;
444
445 SubScreen = InitializeSubScreen(Entry);
446
447 // Set defaults for the new entry; will be modified based on lines read from the config. file....
448 SubEntry = InitializeLoaderEntry(Entry);
449
450 if ((SubEntry == NULL) || (SubScreen == NULL))
451 return;
452 SubEntry->me.Title = StrDuplicate(Title);
453
454 while (((TokenCount = ReadTokenLine(File, &TokenList)) > 0) && (StriCmp(TokenList[0], L"}") != 0)) {
455
456 if ((StriCmp(TokenList[0], L"loader") == 0) && (TokenCount > 1)) { // set the boot loader filename
457 if (SubEntry->LoaderPath != NULL)
458 FreePool(SubEntry->LoaderPath);
459 SubEntry->LoaderPath = StrDuplicate(TokenList[1]);
460 SubEntry->DevicePath = FileDevicePath(Volume->DeviceHandle, SubEntry->LoaderPath);
461
462 } else if (StriCmp(TokenList[0], L"initrd") == 0) {
463 if (SubEntry->InitrdPath != NULL)
464 FreePool(SubEntry->InitrdPath);
465 SubEntry->InitrdPath = NULL;
466 if (TokenCount > 1) {
467 SubEntry->InitrdPath = StrDuplicate(TokenList[1]);
468 }
469
470 } else if (StriCmp(TokenList[0], L"options") == 0) {
471 if (SubEntry->LoadOptions != NULL)
472 FreePool(SubEntry->LoadOptions);
473 SubEntry->LoadOptions = NULL;
474 if (TokenCount > 1) {
475 SubEntry->LoadOptions = StrDuplicate(TokenList[1]);
476 } // if/else
477
478 } else if ((StriCmp(TokenList[0], L"add_options") == 0) && (TokenCount > 1)) {
479 MergeStrings(&SubEntry->LoadOptions, TokenList[1], L' ');
480
481 } else if ((StriCmp(TokenList[0], L"graphics") == 0) && (TokenCount > 1)) {
482 SubEntry->UseGraphicsMode = (StriCmp(TokenList[1], L"on") == 0);
483
484 } else if (StriCmp(TokenList[0], L"disabled") == 0) {
485 SubEntry->Enabled = FALSE;
486 } // ief/elseif
487
488 FreeTokenLine(&TokenList, &TokenCount);
489 } // while()
490
491 if (SubEntry->InitrdPath != NULL) {
492 MergeStrings(&SubEntry->LoadOptions, L"initrd=", L' ');
493 MergeStrings(&SubEntry->LoadOptions, SubEntry->InitrdPath, 0);
494 FreePool(SubEntry->InitrdPath);
495 SubEntry->InitrdPath = NULL;
496 } // if
497 if (SubEntry->Enabled == TRUE) {
498 AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry);
499 }
500 Entry->me.SubScreen = SubScreen;
501 } // VOID AddSubmenu()
502
503 // Finds a volume with the specified Identifier (a volume label or a number
504 // followed by a colon, for the moment). If found, sets *Volume to point to
505 // that volume. If not, leaves it unchanged.
506 // Returns TRUE if a match was found, FALSE if not.
507 static BOOLEAN FindVolume(REFIT_VOLUME **Volume, CHAR16 *Identifier) {
508 UINTN i = 0, CountedVolumes = 0;
509 INTN Number = -1;
510 BOOLEAN Found = FALSE;
511
512 if ((StrLen(Identifier) >= 2) && (Identifier[StrLen(Identifier) - 1] == L':') &&
513 (Identifier[0] >= L'0') && (Identifier[0] <= L'9')) {
514 Number = (INTN) Atoi(Identifier);
515 }
516 while ((i < VolumesCount) && (!Found)) {
517 if (Number >= 0) { // User specified a volume by number
518 if (Volumes[i]->IsReadable) {
519 if (CountedVolumes == Number) {
520 *Volume = Volumes[i];
521 Found = TRUE;
522 }
523 CountedVolumes++;
524 } // if
525 } else { // User specified a volume by label
526 if (StriCmp(Identifier, Volumes[i]->VolName) == 0) {
527 *Volume = Volumes[i];
528 Found = TRUE;
529 } // if
530 } // if/else
531 i++;
532 } // while()
533 return (Found);
534 } // static VOID FindVolume()
535
536 // Adds the options from a SINGLE refind.conf stanza to a new loader entry and returns
537 // that entry. The calling function is then responsible for adding the entry to the
538 // list of entries.
539 static LOADER_ENTRY * AddStanzaEntries(REFIT_FILE *File, REFIT_VOLUME *Volume, CHAR16 *Title) {
540 CHAR16 **TokenList;
541 UINTN TokenCount;
542 LOADER_ENTRY *Entry;
543 BOOLEAN DefaultsSet = FALSE, AddedSubmenu = FALSE;
544 REFIT_VOLUME *CurrentVolume = Volume;
545
546 // prepare the menu entry
547 Entry = InitializeLoaderEntry(NULL);
548 if (Entry == NULL)
549 return NULL;
550
551 Entry->Title = StrDuplicate(Title);
552 Entry->me.Title = AllocateZeroPool(256 * sizeof(CHAR16));
553 SPrint(Entry->me.Title, 255, L"Boot %s from %s", (Title != NULL) ? Title : L"Unknown", CurrentVolume->VolName);
554 Entry->me.Row = 0;
555 Entry->me.BadgeImage = CurrentVolume->VolBadgeImage;
556 Entry->VolName = CurrentVolume->VolName;
557
558 // Parse the config file to add options for a single stanza, terminating when the token
559 // is "}" or when the end of file is reached.
560 while (((TokenCount = ReadTokenLine(File, &TokenList)) > 0) && (StriCmp(TokenList[0], L"}") != 0)) {
561 if ((StriCmp(TokenList[0], L"loader") == 0) && (TokenCount > 1)) { // set the boot loader filename
562 Entry->LoaderPath = StrDuplicate(TokenList[1]);
563 Entry->DevicePath = FileDevicePath(CurrentVolume->DeviceHandle, Entry->LoaderPath);
564 SetLoaderDefaults(Entry, TokenList[1], CurrentVolume);
565 FreePool(Entry->LoadOptions);
566 Entry->LoadOptions = NULL; // Discard default options, if any
567 DefaultsSet = TRUE;
568 } else if ((StriCmp(TokenList[0], L"volume") == 0) && (TokenCount > 1)) {
569 if (FindVolume(&CurrentVolume, TokenList[1])) {
570 FreePool(Entry->me.Title);
571 Entry->me.Title = AllocateZeroPool(256 * sizeof(CHAR16));
572 SPrint(Entry->me.Title, 255, L"Boot %s from %s", (Title != NULL) ? Title : L"Unknown", CurrentVolume->VolName);
573 Entry->me.BadgeImage = CurrentVolume->VolBadgeImage;
574 Entry->VolName = CurrentVolume->VolName;
575 } // if match found
576 } else if ((StriCmp(TokenList[0], L"icon") == 0) && (TokenCount > 1)) {
577 FreePool(Entry->me.Image);
578 Entry->me.Image = LoadIcns(CurrentVolume->RootDir, TokenList[1], 128);
579 if (Entry->me.Image == NULL) {
580 Entry->me.Image = DummyImage(128);
581 }
582 } else if ((StriCmp(TokenList[0], L"initrd") == 0) && (TokenCount > 1)) {
583 if (Entry->InitrdPath)
584 FreePool(Entry->InitrdPath);
585 Entry->InitrdPath = StrDuplicate(TokenList[1]);
586 } else if ((StriCmp(TokenList[0], L"options") == 0) && (TokenCount > 1)) {
587 if (Entry->LoadOptions)
588 FreePool(Entry->LoadOptions);
589 Entry->LoadOptions = StrDuplicate(TokenList[1]);
590 } else if ((StriCmp(TokenList[0], L"ostype") == 0) && (TokenCount > 1)) {
591 if (TokenCount > 1) {
592 Entry->OSType = TokenList[1][0];
593 }
594 } else if ((StriCmp(TokenList[0], L"graphics") == 0) && (TokenCount > 1)) {
595 Entry->UseGraphicsMode = (StriCmp(TokenList[1], L"on") == 0);
596 } else if (StriCmp(TokenList[0], L"disabled") == 0) {
597 Entry->Enabled = FALSE;
598 } else if ((StriCmp(TokenList[0], L"submenuentry") == 0) && (TokenCount > 1)) {
599 AddSubmenu(Entry, File, CurrentVolume, TokenList[1]);
600 AddedSubmenu = TRUE;
601 } // set options to pass to the loader program
602 FreeTokenLine(&TokenList, &TokenCount);
603 } // while()
604
605 if (AddedSubmenu)
606 AddMenuEntry(Entry->me.SubScreen, &MenuEntryReturn);
607
608 if (Entry->InitrdPath) {
609 MergeStrings(&Entry->LoadOptions, L"initrd=", L' ');
610 MergeStrings(&Entry->LoadOptions, Entry->InitrdPath, 0);
611 FreePool(Entry->InitrdPath);
612 Entry->InitrdPath = NULL;
613 } // if
614
615 if (!DefaultsSet)
616 SetLoaderDefaults(Entry, L"\\EFI\\BOOT\\nemo.efi", CurrentVolume); // user included no entry; use bogus one
617
618 return(Entry);
619 } // static VOID AddStanzaEntries()
620
621 // Read the user-configured loaders file, refind_loaders.conf, and add or delete
622 // entries based on the contents of that file....
623 VOID ScanUserConfigured(VOID)
624 {
625 EFI_STATUS Status;
626 REFIT_FILE File;
627 REFIT_VOLUME *Volume;
628 CHAR16 **TokenList;
629 CHAR16 *Title = NULL;
630 UINTN TokenCount;
631 LOADER_ENTRY *Entry;
632
633 if (FileExists(SelfDir, CONFIG_FILE_NAME)) {
634 Status = ReadFile(SelfDir, CONFIG_FILE_NAME, &File);
635 if (EFI_ERROR(Status))
636 return;
637
638 Volume = SelfVolume;
639
640 while ((TokenCount = ReadTokenLine(&File, &TokenList)) > 0) {
641 if ((StriCmp(TokenList[0], L"menuentry") == 0) && (TokenCount > 1)) {
642 Title = StrDuplicate(TokenList[1]);
643 Entry = AddStanzaEntries(&File, Volume, TokenList[1]);
644 if (Entry->Enabled) {
645 if (Entry->me.SubScreen == NULL)
646 GenerateSubScreen(Entry, Volume);
647 AddPreparedLoaderEntry(Entry);
648 } else {
649 FreePool(Entry);
650 } // if/else
651 FreePool(Title);
652 } // if
653 FreeTokenLine(&TokenList, &TokenCount);
654 } // while()
655 } // if()
656 } // VOID ScanUserConfigured()
657
658 // Read a Linux kernel options file for a Linux boot loader into memory. The LoaderPath
659 // and Volume variables identify the location of the options file, but not its name --
660 // you pass this function the filename of the Linux kernel, initial RAM disk, or other
661 // file in the target directory, and this function finds the file with a name in the
662 // comma-delimited list of names specified by LINUX_OPTIONS_FILENAMES within that
663 // directory and loads it. This function tries multiple files because I originally
664 // used the filename linux.conf, but close on the heels of that decision, the Linux
665 // kernel developers decided to use that name for a similar purpose, but with a
666 // different file format. Thus, I'm migrating rEFInd to use the name refind_linux.conf,
667 // but I want a migration period in which both names are used.
668 //
669 // The return value is a pointer to the REFIT_FILE handle for the file, or NULL if
670 // it wasn't found.
671 REFIT_FILE * ReadLinuxOptionsFile(IN CHAR16 *LoaderPath, IN REFIT_VOLUME *Volume) {
672 CHAR16 *OptionsFilename, *FullFilename;
673 BOOLEAN GoOn = TRUE;
674 UINTN i = 0;
675 REFIT_FILE *File = NULL;
676 EFI_STATUS Status;
677
678 do {
679 OptionsFilename = FindCommaDelimited(LINUX_OPTIONS_FILENAMES, i++);
680 FullFilename = FindPath(LoaderPath);
681 if ((OptionsFilename != NULL) && (FullFilename != NULL)) {
682 MergeStrings(&FullFilename, OptionsFilename, '\\');
683 if (FileExists(Volume->RootDir, FullFilename)) {
684 File = AllocateZeroPool(sizeof(REFIT_FILE));
685 Status = ReadFile(Volume->RootDir, FullFilename, File);
686 GoOn = FALSE;
687 if (CheckError(Status, L"while loading the Linux options file")) {
688 if (File != NULL)
689 FreePool(File);
690 File = NULL;
691 GoOn = TRUE;
692 } // if error
693 } // if file exists
694 } else { // a filename string is NULL
695 GoOn = FALSE;
696 } // if/else
697 if (OptionsFilename != NULL)
698 FreePool(OptionsFilename);
699 if (FullFilename != NULL)
700 FreePool(FullFilename);
701 OptionsFilename = FullFilename = NULL;
702 } while (GoOn);
703 return (File);
704 } // static REFIT_FILE * ReadLinuxOptionsFile()
705
706 // Retrieve a single line of options from a Linux kernel options file
707 CHAR16 * GetFirstOptionsFromFile(IN CHAR16 *LoaderPath, IN REFIT_VOLUME *Volume) {
708 UINTN TokenCount;
709 CHAR16 *Options = NULL;
710 CHAR16 **TokenList;
711 REFIT_FILE *File;
712
713 File = ReadLinuxOptionsFile(LoaderPath, Volume);
714 if (File != NULL) {
715 TokenCount = ReadTokenLine(File, &TokenList);
716 if (TokenCount > 1)
717 Options = StrDuplicate(TokenList[1]);
718 FreeTokenLine(&TokenList, &TokenCount);
719 FreePool(File);
720 }
721 return Options;
722 } // static CHAR16 * GetOptionsFile()
723