]> git.saurik.com Git - apple/xnu.git/blobdiff - pexpert/gen/bootargs.c
xnu-4903.231.4.tar.gz
[apple/xnu.git] / pexpert / gen / bootargs.c
index 6ca4fa102ab25a9afd0e105f5dba9ef944c56139..754513a5cb09e9fae8e68c99efa079bcfb7247ef 100644 (file)
@@ -1,8 +1,8 @@
 /*
- * Copyright (c) 2000-2008 Apple Inc. All rights reserved.
+ * Copyright (c) 2000-2016 Apple Inc. All rights reserved.
  *
  * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
- * 
+ *
  * This file contains Original Code and/or Modifications of Original Code
  * as defined in and that are subject to the Apple Public Source License
  * Version 2.0 (the 'License'). You may not use this file except in
  * unlawful or unlicensed copies of an Apple operating system, or to
  * circumvent, violate, or enable the circumvention or violation of, any
  * terms of an Apple operating system software license agreement.
- * 
+ *
  * Please obtain a copy of the License at
  * http://www.opensource.apple.com/apsl/ and read it before using this file.
- * 
+ *
  * The Original Code and all software distributed under the License are
  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
  * Please see the License for the specific language governing rights and
  * limitations under the License.
- * 
+ *
  * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
  */
 #include <pexpert/pexpert.h>
 #include <pexpert/device_tree.h>
 
+typedef boolean_t (*argsep_func_t) (char c);
+
 static boolean_t isargsep( char c);
-#if !CONFIG_EMBEDDED
+static boolean_t israngesep( char c);
+#ifndef CONFIG_EMBEDDED
 static int argstrcpy(char *from, char *to);
-#endif 
+#endif
 static int argstrcpy2(char *from,char *to, unsigned maxlen);
-static int argnumcpy(int val, void *to, unsigned maxlen);
-static int getval(char *s, int *val);
+static int argnumcpy(long long val, void *to, unsigned maxlen);
+static int getval(char *s, long long *val, argsep_func_t issep, boolean_t skip_equal_sign);
+boolean_t get_range_bounds(char * c, int64_t * lower, int64_t * upper);
 
 extern int IODTGetDefault(const char *key, void *infoAddr, unsigned int infoSize);
 
+
 struct i24 {
        int32_t i24 : 24;
        int32_t _pad : 8;
@@ -47,7 +52,7 @@ struct i24 {
 #define        STR     1
 
 #if !defined(__LP64__) && !defined(__arm__)
-boolean_t 
+boolean_t
 PE_parse_boot_arg(
        const char  *arg_string,
        void            *arg_ptr)
@@ -63,23 +68,24 @@ PE_parse_boot_arg(
 }
 #endif
 
-boolean_t
-PE_parse_boot_argn(
-       const char      *arg_string,
-       void            *arg_ptr,
-       int                     max_len)
+static boolean_t
+PE_parse_boot_argn_internal(
+       const char *arg_string,
+       void *      arg_ptr,
+       int         max_len,
+       boolean_t   force_string)
 {
        char *args;
        char *cp, c;
-       unsigned int i;
-       int val;
+       uintptr_t i;
+       long long val = 0;
        boolean_t arg_boolean;
        boolean_t arg_found;
 
        args = PE_boot_args();
        if (*args == '\0') return FALSE;
 
-#if CONFIG_EMBEDDED
+#ifdef CONFIG_EMBEDDED
        if (max_len == -1) return FALSE;
 #endif
 
@@ -89,7 +95,7 @@ PE_parse_boot_argn(
 
        while (*args)
        {
-               if (*args == '-') 
+               if (*args == '-')
                        arg_boolean = TRUE;
                else
                        arg_boolean = FALSE;
@@ -106,9 +112,17 @@ PE_parse_boot_argn(
                if (strncmp(args, arg_string, i) ||
                    (i!=strlen(arg_string)))
                        goto gotit;
+
                if (arg_boolean) {
-                       argnumcpy(1, arg_ptr, max_len);
-                       arg_found = TRUE;
+                       if (!force_string) {
+                               if (max_len > 0) {
+                                       argnumcpy(1, arg_ptr, max_len);/* max_len of 0 performs no copy at all*/
+                                       arg_found = TRUE;
+                               }
+                               else if (max_len == 0) {        
+                                       arg_found = TRUE;
+                               }
+                       }
                        break;
                } else {
                        while (*cp && isargsep (*cp))
@@ -119,25 +133,41 @@ PE_parse_boot_argn(
                        }
                        if ('_' == *arg_string) /* Force a string copy if the argument name begins with an underscore */
                        {
-                               int hacklen = 17 > max_len ? 17 : max_len;
-                               argstrcpy2 (++cp, (char *)arg_ptr, hacklen - 1); /* Hack - terminate after 16 characters */
-                               arg_found = TRUE;
+                               if (max_len > 0) {
+                                       int hacklen = 17 > max_len ? 17 : max_len;
+                                       argstrcpy2 (++cp, (char *)arg_ptr, hacklen - 1); /* Hack - terminate after 16 characters */
+                                       arg_found = TRUE;
+                               }
+                               else if (max_len == 0) {
+                                       arg_found = TRUE;
+                               }
                                break;
                        }
-                       switch (getval(cp, &val)) 
+                       switch ((force_string && *cp == '=') ? STR : getval(cp, &val, isargsep, FALSE))
                        {
                                case NUM:
-                                       argnumcpy(val, arg_ptr, max_len);
-                                       arg_found = TRUE;
+                                       if (max_len > 0) {
+                                               argnumcpy(val, arg_ptr, max_len);
+                                               arg_found = TRUE;
+                                       }
+                                       else if (max_len == 0) {
+                                               arg_found = TRUE;
+                                       }
                                        break;
                                case STR:
-                                       if(max_len > 0) //max_len of 0 performs no copy at all
-                                               argstrcpy2(++cp, (char *)arg_ptr, max_len - 1);
+                                       if (max_len > 0) {
+                                               argstrcpy2(++cp, (char *)arg_ptr, max_len - 1);/*max_len of 0 performs no copy at all*/
+                                               arg_found = TRUE;
+                                       }
+                                       else if (max_len == 0) {
+                                               arg_found = TRUE;
+                                       }
 #if !CONFIG_EMBEDDED
-                                       else if(max_len == -1) // unreachable on embedded
+                                       else if (max_len == -1) { /* unreachable on embedded */
                                                argstrcpy(++cp, (char *)arg_ptr);
+                                               arg_found = TRUE;
+                                       }
 #endif
-                                       arg_found = TRUE;
                                        break;
                        }
                        goto gotit;
@@ -153,20 +183,46 @@ gotit:
        return(arg_found);
 }
 
+boolean_t
+PE_parse_boot_argn(
+       const char      *arg_string,
+       void            *arg_ptr,
+       int                     max_len)
+{
+       return PE_parse_boot_argn_internal(arg_string, arg_ptr, max_len, FALSE);
+}
+
+boolean_t
+PE_parse_boot_arg_str(
+       const char      *arg_string,
+       char            *arg_ptr,
+       int                     strlen)
+{
+       return PE_parse_boot_argn_internal(arg_string, arg_ptr, strlen, TRUE);
+}
+
 static boolean_t
-isargsep(
-       char c)
+isargsep(char c)
 {
        if (c == ' ' || c == '\0' || c == '\t')
-               return(TRUE);
+               return (TRUE);
+       else
+               return (FALSE);
+}
+
+static boolean_t
+israngesep(char c)
+{
+       if (isargsep(c) || c == '_' || c == ',')
+               return (TRUE);
        else
-               return(FALSE);
+               return (FALSE);
 }
 
 #if !CONFIG_EMBEDDED
 static int
 argstrcpy(
-       char *from, 
+       char *from,
        char *to)
 {
        int i = 0;
@@ -182,7 +238,7 @@ argstrcpy(
 
 static int
 argstrcpy2(
-       char *from, 
+       char *from,
        char *to,
        unsigned maxlen)
 {
@@ -196,7 +252,7 @@ argstrcpy2(
        return(i);
 }
 
-static int argnumcpy(int val, void *to, unsigned maxlen)
+static int argnumcpy(long long val, void *to, unsigned maxlen)
 {
        switch (maxlen) {
                case 0:
@@ -213,6 +269,11 @@ static int argnumcpy(int val, void *to, unsigned maxlen)
                        ((struct i24 *)to)->i24 = val;
                        break;
                case 4:
+                       *(int32_t *)to = val;
+                       break;
+               case 8:
+                       *(int64_t *)to = val;
+                       break;
                default:
                        *(int32_t *)to = val;
                        maxlen = 4;
@@ -224,17 +285,26 @@ static int argnumcpy(int val, void *to, unsigned maxlen)
 
 static int
 getval(
-       char *s, 
-       int *val)
+       char *s,
+       long long *val,
+       argsep_func_t issep,
+       boolean_t skip_equal_sign )
 {
-       unsigned int radix, intval;
-    unsigned char c;
+       unsigned long long radix, intval;
+       unsigned char c;
        int sign = 1;
+       boolean_t has_value = FALSE;
 
        if (*s == '=') {
                s++;
-               if (*s == '-')
-                       sign = -1, s++;
+               has_value = TRUE;
+       }
+
+       if (has_value || skip_equal_sign) {
+               if (*s == '-') {
+                       sign = -1;
+                       s++;
+               }
                intval = *s++-'0';
                radix = 10;
                if (intval == 0) {
@@ -258,7 +328,7 @@ getval(
                                break;
 
                        default:
-                               if (!isargsep(*s))
+                               if (!issep(*s))
                                        return (STR);
                        }
                 } else if (intval >= radix) {
@@ -266,7 +336,7 @@ getval(
                 }
                for(;;) {
                         c = *s++;
-                        if (isargsep(c))
+                        if (issep(c))
                             break;
                         if ((radix <= 10) &&
                             ((c >= '0') && (c <= ('9' - (10 - radix))))) {
@@ -297,7 +367,7 @@ getval(
                        intval *= radix;
                        intval += c;
                }
-                if (!isargsep(c) && !isargsep(*s))
+                if (!issep(c) && !issep(*s))
                     return STR;
                *val = intval * sign;
                return (NUM);
@@ -306,7 +376,7 @@ getval(
        return (NUM);
 }
 
-boolean_t 
+boolean_t
 PE_imgsrc_mount_supported()
 {
        return TRUE;
@@ -353,3 +423,38 @@ PE_get_default(
         */
        return IODTGetDefault(property_name, property_ptr, max_property) ? FALSE : TRUE;
 }
+
+/* function: get_range_bounds
+ * Parse a range string like "1_3,5_20" and return 1,3 as lower and upper.
+ * Note: '_' is separator for bounds integer delimiter and
+ *       ',' is considered as separator for range pair.
+ * returns TRUE when both range values are found
+ */
+boolean_t
+get_range_bounds(char *c, int64_t *lower, int64_t *upper)
+{
+       if (c == NULL || lower == NULL || upper == NULL) {
+               return FALSE;
+       }
+
+       if (NUM != getval(c, lower, israngesep, TRUE)) {
+               return FALSE;
+       }
+
+       while (*c != '\0') {
+               if (*c == '_') {
+                       break;
+               }
+               c++;
+       }
+
+       if (*c == '_') {
+               c++;
+               if (NUM != getval(c, upper, israngesep, TRUE)) {
+                       return FALSE;
+               }
+       } else {
+               return FALSE;
+       }
+       return TRUE;
+}