scutil.tproj/dictionary.c

   1 /*
   2  * Copyright (c) 2000-2004, 2009 Apple Inc. All rights reserved.
   3  *
   4  * @APPLE_LICENSE_HEADER_START@
   5  *
   6  * This file contains Original Code and/or Modifications of Original Code
   7  * as defined in and that are subject to the Apple Public Source License
   8  * Version 2.0 (the 'License'). You may not use this file except in
   9  * compliance with the License. Please obtain a copy of the License at
  10  * http://www.opensource.apple.com/apsl/ and read it before using this
  11  * file.
  12  *
  13  * The Original Code and all software distributed under the License are
  14  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  15  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  16  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
  18  * Please see the License for the specific language governing rights and
  19  * limitations under the License.
  20  *
  21  * @APPLE_LICENSE_HEADER_END@
  22  */
  23
  24 /*
  25  * Modification History
  26  *
  27  * June 1, 2001                 Allan Nathanson <ajn@apple.com>
  28  * - public API conversion
  29  *
  30  * November 9, 2000             Allan Nathanson <ajn@apple.com>
  31  * - initial revision
  32  */
  33
  34 #include "scutil.h"
  35 #include "dictionary.h"
  36
  37
  38 //#include <stdlib.h>
  39 //#include <limits.h>
  40
  41
  42 __private_extern__
  43 void
  44 do_dictInit(int argc, char **argv)
  45 {
  46         if (value != NULL) {
  47                 CFRelease(value);
  48         }
  49
  50         value = CFDictionaryCreateMutable(NULL
  51                                          ,0
  52                                          ,&kCFTypeDictionaryKeyCallBacks
  53                                          ,&kCFTypeDictionaryValueCallBacks
  54                                          );
  55
  56         return;
  57 }
  58
  59
  60 __private_extern__
  61 void
  62 do_dictShow(int argc, char **argv)
  63 {
  64         if (value == NULL) {
  65                 SCPrint(TRUE, stdout, CFSTR("d.show: dictionary must be initialized.\n"));
  66                 return;
  67         }
  68
  69         SCPrint(TRUE, stdout, CFSTR("%@\n"), value);
  70
  71         return;
  72 }
  73
  74
  75 __private_extern__
  76 void
  77 do_dictSetKey(int argc, char **argv)
  78 {
  79         CFMutableArrayRef       array           = NULL;
  80         Boolean                 doArray         = FALSE;
  81         Boolean                 doBoolean       = FALSE;
  82         Boolean                 doNumeric       = FALSE;
  83         CFStringRef             key;
  84         CFMutableDictionaryRef  newValue;
  85         CFTypeRef               val             = NULL;
  86
  87         if (value == NULL) {
  88                 SCPrint(TRUE, stdout, CFSTR("d.add: dictionary must be initialized.\n"));
  89                 return;
  90         }
  91
  92         if (!isA_CFDictionary(value)) {
  93                 SCPrint(TRUE, stdout, CFSTR("d.add: data (fetched from configuration server) is not a dictionary.\n"));
  94                 return;
  95         }
  96
  97         key = CFStringCreateWithCString(NULL, argv[0], kCFStringEncodingUTF8);
  98         argv++; argc--;
  99
 100         while (argc > 0) {
 101                 if (strcmp(argv[0], "*") == 0) {
 102                         /* if array requested */
 103                         doArray = TRUE;
 104                 } else if (strcmp(argv[0], "-") == 0) {
 105                         /* if string values requested */
 106                 } else if (strcmp(argv[0], "?") == 0) {
 107                         /* if boolean values requested */
 108                         doBoolean = TRUE;
 109                 } else if (strcmp(argv[0], "#") == 0) {
 110                         /* if numeric values requested */
 111                         doNumeric = TRUE;
 112                 } else {
 113                         /* it's not a special flag */
 114                         break;
 115                 }
 116                 argv++; argc--;
 117         }
 118
 119         if (argc > 1) {
 120                 doArray = TRUE;
 121         } else if (!doArray && (argc == 0)) {
 122                 SCPrint(TRUE, stdout, CFSTR("d.add: no values.\n"));
 123                 CFRelease(key);
 124                 return;
 125         }
 126
 127         if (doArray) {
 128                 array = CFArrayCreateMutable(NULL, 0, &kCFTypeArrayCallBacks);
 129         }
 130
 131         while (argc > 0) {
 132                 if (doBoolean) {
 133                         if         ((strcasecmp(argv[0], "true") == 0) ||
 134                                     (strcasecmp(argv[0], "t"   ) == 0) ||
 135                                     (strcasecmp(argv[0], "yes" ) == 0) ||
 136                                     (strcasecmp(argv[0], "y"   ) == 0) ||
 137                                     (strcmp    (argv[0], "1"   ) == 0)) {
 138                                 val = CFRetain(kCFBooleanTrue);
 139                         } else if ((strcasecmp(argv[0], "false") == 0) ||
 140                                    (strcasecmp(argv[0], "f"    ) == 0) ||
 141                                    (strcasecmp(argv[0], "no"   ) == 0) ||
 142                                    (strcasecmp(argv[0], "n"    ) == 0) ||
 143                                    (strcmp    (argv[0], "0"    ) == 0)) {
 144                                 val = CFRetain(kCFBooleanFalse);
 145                         } else {
 146                                 SCPrint(TRUE, stdout, CFSTR("d.add: invalid data.\n"));
 147                                 if (doArray) CFRelease(array);
 148                                 CFRelease(key);
 149                                 return;
 150                         }
 151                 } else if (doNumeric) {
 152                         int     intValue;
 153
 154                         if (sscanf(argv[0], "%d", &intValue) == 1) {
 155                                 val = CFNumberCreate(NULL, kCFNumberIntType, &intValue);
 156                         } else {
 157                                 SCPrint(TRUE, stdout, CFSTR("d.add: invalid data.\n"));
 158                                 if (doArray) CFRelease(array);
 159                                 CFRelease(key);
 160                                 return;
 161                         }
 162                 } else {
 163                         val = (CFPropertyListRef)CFStringCreateWithCString(NULL, argv[0], kCFStringEncodingUTF8);
 164                 }
 165
 166                 if (doArray) {
 167                         CFArrayAppendValue(array, val);
 168                         CFRelease(val);
 169                         argv++; argc--;
 170                 } else {
 171                         break;
 172                 }
 173         }
 174
 175         newValue = CFDictionaryCreateMutableCopy(NULL, 0, value);
 176         if (doArray) {
 177                 CFDictionarySetValue(newValue, key, array);
 178                 CFRelease(array);
 179         } else {
 180                 CFDictionarySetValue(newValue, key, val);
 181                 CFRelease(val);
 182         }
 183         CFRelease(key);
 184         CFRelease(value);
 185         value = newValue;
 186
 187         return;
 188 }
 189
 190
 191 __private_extern__
 192 void
 193 do_dictRemoveKey(int argc, char **argv)
 194 {
 195         CFStringRef             key;
 196         CFMutableDictionaryRef  val;
 197
 198         if (value == NULL) {
 199                 SCPrint(TRUE, stdout, CFSTR("d.remove: dictionary must be initialized.\n"));
 200                 return;
 201         }
 202
 203         if (!isA_CFDictionary(value)) {
 204                 SCPrint(TRUE, stdout, CFSTR("d.remove: data (fetched from configuration server) is not a dictionary.\n"));
 205                 return;
 206         }
 207
 208         val = CFDictionaryCreateMutableCopy(NULL, 0, value);
 209         CFRelease(value);
 210         value = val;
 211
 212         key = CFStringCreateWithCString(NULL, argv[0], kCFStringEncodingUTF8);
 213         CFDictionaryRemoveValue((CFMutableDictionaryRef)value, key);
 214         CFRelease(key);
 215
 216         return;
 217 }