Chameleon

Chameleon Svn Source Tree

Root/branches/ErmaC/Enoch/i386/libsaio/stringTable.c

1/*
2 * Copyright (c) 1999-2003 Apple Computer, Inc. All rights reserved.
3 *
4 * @APPLE_LICENSE_HEADER_START@
5 *
6 * Portions Copyright (c) 1999-2003 Apple Computer, Inc. All Rights
7 * Reserved. This file contains Original Code and/or Modifications of
8 * Original Code as defined in and that are subject to the Apple Public
9 * Source License Version 2.0 (the "License"). You may not use this file
10 * except in compliance with the License. Please obtain a copy of the
11 * License at http://www.apple.com/publicsource and read it before using
12 * this file.
13 *
14 * The Original Code and all software distributed under the License are
15 * distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY KIND, EITHER
16 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
17 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE OR NON- INFRINGEMENT. Please see the
19 * License for the specific language governing rights and limitations
20 * under the License.
21 *
22 * @APPLE_LICENSE_HEADER_END@
23 */
24/*
25 * Copyright 1993 NeXT, Inc.
26 * All rights reserved.
27 */
28
29#include "bootstruct.h"
30#include "libsaio.h"
31#include "boot.h"
32#include "xml.h"
33
34extern char *Language;
35extern char *LoadableFamilies;
36
37bool sysConfigValid;
38
39#if UNUSED
40
41/*
42 * Compare a string to a key with quoted characters
43 */
44static inline int keyncmp(const char *str, const char *key, int n)
45{
46 int c;
47 while (n--) {
48c = *key++;
49if (c == '\\') {
50 switch(c = *key++) {
51 case 'n':
52c = '\n';
53break;
54 case 'r':
55c = '\r';
56break;
57 case 't':
58c = '\t';
59break;
60 default:
61break;
62 }
63} else if (c == '\"') {
64 // Premature end of key
65 return 1;
66}
67if (c != *str++) {
68 return 1;
69}
70 }
71 return 0;
72}
73
74static void eatThru(char val, const char **table_p)
75{
76register const char *table = *table_p;
77register bool found = false;
78
79while (*table && !found)
80{
81if (*table == '\\') table += 2;
82else
83{
84if (*table == val) found = true;
85table++;
86}
87}
88*table_p = table;
89}
90
91/* Remove key and its associated value from the table. */
92bool removeKeyFromTable(const char *key, char *table)
93{
94 register int len;
95 register char *tab;
96 char *buf;
97
98 len = strlen(key);
99 tab = (char *)table;
100 buf = (char *)malloc(len + 3);
101
102 snprintf(buf, len + 3,"\"%s\"", key);
103 len = strlen(buf);
104
105 while(*tab) {
106 if(strncmp(buf, tab, len) == 0) {
107 char c;
108
109 while((c = *(tab + len)) != ';') {
110 if(c == 0) {
111 len = -1;
112 goto out;
113 }
114 len++;
115 }
116 len++;
117 if(*(tab + len) == '\n') len++;
118 goto out;
119 }
120 tab++;
121 }
122 len = -1;
123out:
124 free(buf);
125
126 if(len == -1) return false;
127
128 while((*tab = *(tab + len))) {
129 tab++;
130 }
131
132 return true;
133}
134
135char *newStringFromList(char **list, int *size)
136{
137char *begin = *list, *end;
138char *newstr;
139int newsize = *size;
140int bufsize;
141
142while (*begin && newsize && isspace(*begin)) {
143begin++;
144newsize--;
145}
146end = begin;
147while (*end && newsize && !isspace(*end)) {
148end++;
149newsize--;
150}
151if (begin == end) {
152return 0;
153}
154bufsize = end - begin + 1;
155newstr = malloc(bufsize);
156if (!newstr)
157{
158return 0;
159}
160strlcpy(newstr, begin, bufsize);
161*list = end;
162*size = newsize;
163return newstr;
164}
165
166#endif
167
168/*
169 * compress == compress escaped characters to one character
170 */
171int stringLength(const char *table, int compress)
172{
173int ret = 0;
174
175while (*table) {
176if (*table == '\\') {
177table += 2;
178ret += 1 + (compress ? 0 : 1);
179}
180else
181{
182if (*table == '\"')
183{
184return ret;
185}
186ret++;
187table++;
188}
189}
190return ret;
191}
192
193
194bool getValueForConfigTableKey(config_file_t *config, const char *key, const char **val, int *size)
195{
196if (config->dictionary != 0 ) {
197// Look up key in XML dictionary
198TagPtr value;
199value = XMLGetProperty(config->dictionary, key);
200if (value != 0) {
201if (value->type != kTagTypeString) {
202error("Non-string tag '%s' found in config file\n", key);
203return false;
204}
205*val = value->string;
206*size = strlen(value->string);
207return true;
208}
209} else {
210
211// Legacy plist-style table
212
213}
214
215return false;
216}
217
218#if UNUSED
219
220/*
221 * Returns a new malloc'ed string if one is found
222 * in the string table matching 'key'. Also translates
223 * \n escapes in the string.
224 */
225char *newStringForStringTableKey(char *table, char *key, config_file_t *config)
226{
227const char *val;
228char *newstr, *p;
229int size;
230
231if (getValueForConfigTableKey(config, key, &val, &size))
232{
233newstr = (char *)malloc(size+1);
234if (!newstr)
235{
236return 0;
237}
238for (p = newstr; size; size--, p++, val++) {
239if ((*p = *val) == '\\') {
240switch (*++val) {
241case 'r':
242*p = '\r';
243break;
244case 'n':
245*p = '\n';
246break;
247case 't':
248*p = '\t';
249break;
250default:
251*p = *val;
252break;
253}
254size--;
255}
256}
257*p = '\0';
258return newstr;
259} else {
260return 0;
261}
262}
263
264#endif
265
266char *newStringForKey(char *key, config_file_t *config)
267{
268const char *val;
269char *newstr;
270int size;
271
272if (getValueForKey(key, &val, &size, config) && size) {
273newstr = (char *)malloc(size + 1);
274if (!newstr)
275{
276return 0;
277}
278strlcpy(newstr, val, size + 1);
279return newstr;
280} else {
281return 0;
282}
283}
284
285/* parse a command line
286 * in the form: [<argument> ...] [<option>=<value> ...]
287 * both <option> and <value> must be either composed of
288 * non-whitespace characters, or enclosed in quotes.
289 */
290static const char *getToken(const char *line, const char **begin, int *len)
291{
292if (*line == '\"') {
293*begin = ++line;
294while (*line && *line != '\"') {
295line++;
296}
297*len = line++ - *begin;
298} else {
299*begin = line;
300while (*line && !isspace(*line) && *line != '=') {
301line++;
302}
303*len = line - *begin;
304}
305return line;
306}
307
308bool getValueForBootKey(const char *line, const char *match, const char **matchval, int *len)
309{
310const char *key, *value;
311int key_len, value_len;
312bool retval = false;
313
314while (*line) {
315/* look for keyword or argument */
316while (isspace(*line))
317{
318line++;
319}
320
321/* now look for '=' or whitespace */
322line = getToken(line, &key, &key_len);
323/* line now points to '=' or space */
324if (*line && !isspace(*line)) {
325line = getToken(++line, &value, &value_len);
326} else {
327value = line;
328value_len = 0;
329}
330if ((strlen(match) == key_len) && strncmp(match, key, key_len) == 0)
331{
332// create a new string
333char* newstr = malloc(value_len + 1);
334strncpy(newstr, value, value_len);
335newstr[value_len] = 0;
336
337*matchval = newstr;
338*len = value_len;
339retval = true;
340/* Continue to look for this key; last one wins. */
341}
342}
343
344return retval;
345}
346
347/* Return NULL if no option has been successfully retrieved, or the string otherwise */
348const char *getStringForKey(const char *key, config_file_t *config)
349{
350static const char *value = 0;
351int len = 0;
352if (!getValueForKey(key, &value, &len, config))
353{
354value = 0;
355}
356return value;
357}
358
359/* Returns TRUE if a value was found, FALSE otherwise.
360 * The boolean value of the key is stored in 'val'.
361 */
362bool getBoolForKey(const char *key, bool *result_val, config_file_t *config)
363{
364const char *key_val;
365int size;
366
367// looking for real boolean (<true/> or <false/>)
368// if is a boolean tag, return immediately
369TagPtr entry = XMLGetProperty(config->dictionary,key);
370
371if(XMLIsBoolean(entry))
372{
373int value = XMLCastBoolean(entry);
374if (value) {
375*result_val = true;
376return true;
377}
378else
379{
380*result_val = false;
381return false;
382}
383}
384
385// check if is a boolean as "string" (Yes/No)
386// (IMHO this should be deprecated soon)
387if (getValueForKey(key, &key_val, &size, config))
388{
389if ((size >= 1) && (key_val[0] == 'Y' || key_val[0] == 'y'))
390{
391*result_val = true;
392}
393else
394{
395*result_val = false;
396}
397return true;
398}
399return false;
400}
401
402bool getIntForKey(const char *key, int *value, config_file_t *config)
403{
404const char *val;
405int size, sum;
406bool negative = false;
407
408if (getValueForKey(key, &val, &size, config))
409{
410if (size)
411{
412if (*val == '-')
413{
414negative = true;
415val++;
416size--;
417}
418
419for (sum = 0; size > 0; size--)
420{
421if (*val < '0' || *val > '9')
422{
423return false;
424}
425
426sum = (sum * 10) + (*val++ - '0');
427}
428
429if (negative)
430{
431sum = -sum;
432}
433*value = sum;
434return true;
435}
436}
437return false;
438}
439/*
440 *
441 */
442bool getDimensionForKey( const char *key, unsigned int *value, config_file_t *config, unsigned int dimension_max, unsigned int object_size )
443{
444const char *val;
445
446int size = 0;
447int sum = 0;
448
449bool negative = false;
450bool percentage = false;
451
452if (getValueForKey(key, &val, &size, config))
453{
454if ( size )
455{
456if (*val == '-')
457{
458negative = true;
459val++;
460size--;
461}
462
463if (val[size-1] == '%')
464{
465percentage = true;
466size--;
467}
468
469// convert string to integer
470for (sum = 0; size > 0; size--)
471{
472if (*val < '0' || *val > '9')
473{
474return false;
475}
476
477sum = (sum * 10) + (*val++ - '0');
478}
479
480if (percentage)
481{
482sum = ( dimension_max * sum ) / 100;
483}
484
485// calculate offset from opposite origin
486if (negative)
487{
488sum = ( ( dimension_max - object_size ) - sum );
489}
490
491}
492else
493{
494
495// null value calculate center
496sum = ( dimension_max - object_size ) / 2;
497
498}
499
500*value = (uint16_t) sum;
501return true;
502}
503
504// key not found
505return false;
506}
507
508/*
509 *get color value from plist format #RRGGBB
510 */
511bool getColorForKey( const char *key, unsigned int *value, config_file_t *config )
512{
513const char *val;
514int size;
515
516if (getValueForKey(key, &val, &size, config))
517{
518if (*val == '#') {
519val++;
520*value = strtol(val, NULL, 16);
521return true;
522}
523}
524return false;
525}
526
527bool getValueForKey( const char *key, const char **val, int *size, config_file_t *config )
528{
529const char *overrideVal;
530int overrideSize;
531bool override, ret;
532
533if (getValueForBootKey(bootArgs->CommandLine, key, val, size))
534{
535return true;
536}
537
538ret = getValueForConfigTableKey(config, key, val, size);
539
540// Try to find alternate keys in bootInfo->chameleonConfig (if config can be overriden)
541// and prefer its values with the exceptions for
542// "Kernel"="mach_kernel" and "Kernel Flags"="".
543
544if (config->canOverride)
545{
546if (getValueForConfigTableKey(&bootInfo->chameleonConfig, key, &overrideVal, &overrideSize))
547{
548override = true;
549
550// NOTE: Values are defined by apple as being in com.apple.Boot.plist
551// kHelperRootUUIDKey, kKernelArchKey, kMKextCacheKey, kKernelCacheKey, kKernelNameKey, kKernelFlagsKey
552if (ret && (strcmp(key, kKernelNameKey) == 0) && (overrideSize == 0))
553{
554override = false;
555}
556
557if (ret && (strcmp(key, kKernelFlagsKey) == 0) && (overrideSize == 0))
558{
559override = false;
560}
561
562if (override)
563{
564*val = overrideVal;
565*size = overrideSize;
566ret = true;
567}
568}
569}
570return ret;
571}
572
573#if UNUSED
574void printSystemConfig(char *p1)
575{
576char *p2 = p1, tmp;
577
578while (*p1 != '\0')
579{
580while (*p2 != '\0' && *p2 != '\n')
581{
582p2++;
583}
584tmp = *p2;
585*p2 = '\0';
586printf("%s\n", p1);
587*p2 = tmp;
588if (tmp == '\0')
589{
590break;
591}
592p1 = ++p2;
593 }
594}
595#endif
596
597//==========================================================================
598// ParseXMLFile
599// Modifies the input buffer.
600// Expects to see one dictionary in the XML file.
601// Puts the first dictionary it finds in the
602// tag pointer and returns 0, or returns -1 if not found
603// (and does not modify dict pointer).
604// Prints an error message if there is a parsing error.
605//
606int ParseXMLFile( char *buffer, TagPtr *dict )
607{
608long length, pos;
609TagPtr tag = 0;
610pos = 0;
611char *configBuffer;
612
613length = strlen(buffer) + 1;
614
615configBuffer = malloc(strlen(buffer)+1);
616if (!configBuffer)
617{
618return -1;
619}
620strlcpy(configBuffer, buffer, length );
621
622while (1)
623{
624length = XMLParseNextTag(configBuffer + pos, &tag);
625if (length == -1)
626{
627break;
628}
629
630pos += length;
631
632if (tag == 0)
633{
634continue;
635}
636if (tag->type == kTagTypeDict)
637{
638break;
639}
640
641XMLFreeTag(tag);
642}
643free(configBuffer);
644if (length < 0) {
645error ("Error parsing plist file\n");
646return -1;
647}
648*dict = tag;
649return 0;
650}
651
652/* loadConfigFile
653 *
654 * Returns 0 - successful.
655 * -1 - unsuccesful.
656 */
657int loadConfigFile (const char *configFile, config_file_t *config)
658{
659int fd, count;
660
661if ((fd = open_bvdev("bt(0,0)", configFile, 0)) < 0)
662{
663return -1;
664}
665// read file
666count = read(fd, config->plist, IO_CONFIG_DATA_SIZE);
667close(fd);
668
669// build xml dictionary
670ParseXMLFile(config->plist, &config->dictionary);
671return 0;
672}
673
674/* loadSystemConfig
675 *
676 * Returns 0 - successful.
677 * -1 - unsuccesful.
678 */
679int loadSystemConfig(config_file_t *config)
680{
681// Micky1979, the order is important
682char *dirspec[] = {
683"/com.apple.recovery.boot/com.apple.Boot.plist",// OS X Recovery
684"/OS X Install Data/com.apple.Boot.plist",// OS X Upgrade (10.8+)
685"/Mac OS X Install Data/com.apple.Boot.plist",// OS X Upgrade (Lion 10.7)
686"/.IABootFiles/com.apple.Boot.plist",// OS X Installer
687"/Library/Preferences/SystemConfiguration/com.apple.Boot.plist"// (Installed System or Installer)
688};
689
690int i, fd, count, ret=-1;
691
692for(i = 0; i< sizeof(dirspec)/sizeof(dirspec[0]); i++)
693{
694if ((fd = open(dirspec[i], 0)) >= 0)
695{
696// read file
697count = read(fd, config->plist, IO_CONFIG_DATA_SIZE);
698close(fd);
699
700// build xml dictionary
701ParseXMLFile(config->plist, &config->dictionary);
702sysConfigValid = true;
703ret=0;
704
705break;
706}
707}
708
709if(ret == -1) ret = loadHelperConfig(config);
710
711// Always enable canOverride flag (for SystemConfig)
712config->canOverride = true;
713
714return ret;
715}
716
717/* loadChameleonConfig
718 *
719 * Returns 0 - successful.
720 * -1 - unsuccesful.
721 */
722int loadChameleonConfig(config_file_t *config, BVRef chain)
723{
724char *dirspec[] = {
725"/Extra/org.chameleon.Boot.plist",
726"/Extra/com.apple.Boot.plist", /* DEPRECATED */
727};
728
729int i;
730
731for(i = 0; i< sizeof(dirspec)/sizeof(dirspec[0]); i++)
732{
733if ( loadChameleonConfigForDevice(config, "rd(0,0)", dirspec[i]) == 0 )
734{
735return 0;
736}
737
738if ( loadChameleonConfigForDevice(config, "", dirspec[i]) == 0 )
739{
740return 0;
741}
742
743if ( loadChameleonConfigForDevice(config, "bt(0,0)", dirspec[i]) == 0 )
744{
745return 0;
746}
747BVRef bvr;
748for ( bvr = chain; bvr; bvr = bvr->next ) /* C99 Error */
749{
750char device[256];
751getDeviceDescription(bvr, device);
752
753if ( loadChameleonConfigForDevice(config, device, dirspec[i]) == 0 )
754{
755return 0;
756}
757}
758}
759return -1;
760}
761
762/* loadChameleonConfigForDevice
763 *
764 * Returns 0 - successful.
765 * -1 - unsuccesful.
766 */
767int loadChameleonConfigForDevice(config_file_t *config, const char *device, const char *path)
768{
769char full_path[1024];
770int fd;
771
772snprintf(full_path, sizeof(full_path), "%s%s", device, path);
773
774if ((fd = open(full_path, 0)) >= 0)
775{
776// Check for depreciated file names and annoy the user about it.
777if(strstr(full_path, "com.apple.Boot.plist")) {
778printf("%s is deprecated.\n", full_path);
779full_path[strlen(full_path) - strlen("com.apple.Boot.plist")] = 0;
780printf("Please use the file %sorg.chameleon.Boot.plist instead.\n", full_path);
781pause();
782}
783// read file
784read(fd, config->plist, IO_CONFIG_DATA_SIZE);
785close(fd);
786
787// build xml dictionary
788ParseXMLFile(config->plist, &config->dictionary);
789sysConfigValid = true;
790return 0;
791}
792return -1;
793}
794
795/* loadHelperConfig
796 *
797 * Returns 0 - successful.
798 * -1 - unsuccesful.
799 */
800int loadHelperConfig(config_file_t *config)
801{
802int rfd, pfd, sfd, count, ret=-1, fixedsize;
803
804char *dirspec[] = {
805"/com.apple.boot.P/Library/Preferences/SystemConfiguration/com.apple.Boot.plist",
806"/com.apple.boot.R/Library/Preferences/SystemConfiguration/com.apple.Boot.plist",
807"/com.apple.boot.S/Library/Preferences/SystemConfiguration/com.apple.Boot.plist"
808};
809
810// This is a simple rock - paper scissors algo. R beats S, P beats R, S beats P
811// If all three, S is used for now. This should be changed to something else (say, timestamp?)
812
813pfd = open(dirspec[0], 0);
814if(pfd >= 0)// com.apple.boot.P exists
815{
816
817sfd = open(dirspec[2], 0); // com.apple.boot.S takes precidence if it also exists
818if(sfd >= 0)
819{
820// Use sfd
821fixedsize = MIN(file_size(sfd), IO_CONFIG_DATA_SIZE);
822count = read(sfd, config->plist, fixedsize);
823close(sfd);
824close(pfd);
825if (count != fixedsize) return -1;
826
827// build xml dictionary
828ParseXMLFile(config->plist, &config->dictionary);
829sysConfigValid = true;
830ret=0;
831
832}
833else
834{
835// used pfd
836fixedsize = MIN(file_size(pfd), IO_CONFIG_DATA_SIZE);
837count = read(pfd, config->plist, fixedsize);
838close(pfd);
839if (count != fixedsize) return -1;
840
841// build xml dictionary
842ParseXMLFile(config->plist, &config->dictionary);
843sysConfigValid = true;
844ret = 0;
845}
846
847}
848else
849{
850rfd = open(dirspec[1], 0); // com.apple.boot.R exists
851if(rfd >= 0)
852{
853pfd = open(dirspec[2], 0); // com.apple.boot.P takes recidence if it exists
854if(pfd >= 0)
855{
856// use sfd
857fixedsize = MIN(file_size(pfd), IO_CONFIG_DATA_SIZE);
858count = read(pfd, config->plist, fixedsize);
859close(pfd);
860close(rfd);
861if (count != fixedsize) return -1;
862
863// build xml dictionary
864ParseXMLFile(config->plist, &config->dictionary);
865sysConfigValid = true;
866ret = 0;
867
868}
869else
870{
871// use rfd
872fixedsize = MIN(file_size(rfd), IO_CONFIG_DATA_SIZE);
873count = read(rfd, config->plist, fixedsize);
874close(rfd);
875if (count != fixedsize) return -1;
876
877// build xml dictionary
878ParseXMLFile(config->plist, &config->dictionary);
879sysConfigValid = true;
880ret = 0;
881
882}
883
884}
885else
886{
887sfd = open(dirspec[2], 0); // com.apple.boot.S exists, but nothing else does
888if(sfd >= 0)
889{
890// use sfd
891fixedsize = MIN(file_size(sfd), IO_CONFIG_DATA_SIZE);
892count = read(sfd, config->plist, fixedsize);
893close(sfd);
894if (count != fixedsize) return -1;
895
896// build xml dictionary
897ParseXMLFile(config->plist, &config->dictionary);
898sysConfigValid = true;
899ret = 0;
900
901}
902}
903
904}
905
906return ret;
907}
908
909char *newString(const char *oldString)
910{
911if ( oldString )
912{
913return strcpy(malloc(strlen(oldString)+1), oldString);
914}
915else
916{
917return NULL;
918}
919}
920
921/*
922 * Extracts the next argument from the command line, double quotes are allowed here.
923 */
924char *getNextArg(char **argPtr, char *val)
925{
926char * ptr = *argPtr;
927const char * strStart;
928int len = 0;
929bool isQuoted = false;
930
931*val = '\0';
932
933// Scan for the next non-whitespace character.
934while ( *ptr && (*ptr == ' ' || *ptr == '=') )
935{
936ptr++;
937}
938
939strStart = ptr;
940
941// Skip the leading double quote character.
942if (*ptr == '\"')
943{
944isQuoted = true;
945ptr++;
946strStart++;
947}
948
949// Scan for the argument terminator character.
950// This can be either a NULL character - in case we reach the end of the string,
951// a double quote in case of quoted argument,
952// or a whitespace character (' ' or '=') for non-quoted argument.
953while (*ptr && !( (isQuoted && (*ptr == '\"')) ||
954 (!isQuoted && (*ptr == ' ' || *ptr == '=')) )
955 ) {
956ptr++;
957}
958
959len = ptr - strStart;
960
961// Skip the closing double quote character and adjust
962// the starting pointer for the next getNextArg call.
963if (*ptr && isQuoted && *ptr == '\"') {
964ptr++;
965}
966
967// Copy the extracted argument to val.
968strncat(val, strStart, len);
969
970// Set command line pointer.
971*argPtr = ptr;
972
973return ptr;
974}
975

Archive Download this file

Revision: 2866