1 /* $NetBSD: main.c,v 1.13 2009/06/11 08:12:00 wiz Exp $ */ 2 3 /*- 4 * Copyright (c) 2008 The NetBSD Foundation, Inc. 5 * All rights reserved. 6 * 7 * Redistribution and use in source and binary forms, with or without 8 * modification, are permitted provided that the following conditions 9 * are met: 10 * 1. Redistributions of source code must retain the above copyright 11 * notice, this list of conditions and the following disclaimer. 12 * 2. Redistributions in binary form must reproduce the above copyright 13 * notice, this list of conditions and the following disclaimer in the 14 * documentation and/or other materials provided with the distribution. 15 * 16 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS 17 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 18 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 19 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS 20 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 21 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 22 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 24 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 26 * POSSIBILITY OF SUCH DAMAGE. 27 */ 28 29 #include <sys/cdefs.h> 30 #ifndef lint 31 __RCSID("$NetBSD: main.c,v 1.13 2009/06/11 08:12:00 wiz Exp $"); 32 #endif /* !lint */ 33 34 #include <sys/module.h> 35 #include <sys/queue.h> 36 37 #include <assert.h> 38 #include <stdbool.h> 39 #include <stdio.h> 40 #include <stdlib.h> 41 #include <string.h> 42 #include <unistd.h> 43 #include <err.h> 44 45 #include <prop/proplib.h> 46 47 int main(int, char **); 48 static void parse_bool_param(prop_dictionary_t, const char *, 49 const char *); 50 static void parse_int_param(prop_dictionary_t, const char *, 51 const char *); 52 static void parse_param(prop_dictionary_t, const char *, 53 void (*)(prop_dictionary_t, const char *, 54 const char *)); 55 static void parse_string_param(prop_dictionary_t, const char *, 56 const char *); 57 static void usage(void) __dead; 58 static void merge_dicts(prop_dictionary_t, const prop_dictionary_t); 59 60 int 61 main(int argc, char **argv) 62 { 63 SIMPLEQ_HEAD(del_head, del_item) del_head; 64 modctl_load_t cmdargs; 65 prop_dictionary_t ext_props, props; 66 bool del_props, merge_props, output_props; 67 const char *ext_file; 68 char *propsstr; 69 int ch; 70 int flags; 71 72 struct del_item { 73 SIMPLEQ_ENTRY(del_item) del_items; 74 const char *del_key; 75 } *delp; 76 77 SIMPLEQ_INIT(&del_head); 78 ext_file = NULL; 79 ext_props = NULL; 80 props = prop_dictionary_create(); 81 del_props = merge_props = output_props = false; 82 flags = 0; 83 84 while ((ch = getopt(argc, argv, "Pb:d:fi:m:ps:")) != -1) { 85 switch (ch) { 86 case 'P': 87 flags |= MODCTL_NO_PROP; 88 break; 89 case 'b': 90 parse_param(props, optarg, parse_bool_param); 91 break; 92 93 case 'd': 94 del_props = true; 95 delp = malloc(sizeof(struct del_item)); 96 if (delp == NULL) 97 errx(EXIT_FAILURE, "Out of memory"); 98 delp->del_key = optarg; 99 SIMPLEQ_INSERT_TAIL(&del_head, delp, del_items); 100 break; 101 102 case 'f': 103 flags |= MODCTL_LOAD_FORCE; 104 break; 105 106 case 'i': 107 parse_param(props, optarg, parse_int_param); 108 break; 109 110 case 'm': 111 merge_props = true; 112 ext_file = optarg; 113 break; 114 115 case 'p': 116 output_props = true; 117 break; 118 119 case 's': 120 parse_param(props, optarg, parse_string_param); 121 break; 122 123 default: 124 usage(); 125 /* NOTREACHED */ 126 } 127 } 128 129 argc -= optind; 130 argv += optind; 131 132 propsstr = prop_dictionary_externalize(props); 133 if (propsstr == NULL) 134 errx(EXIT_FAILURE, "Failed to process properties"); 135 136 if (output_props) { 137 if (merge_props) { 138 ext_props = 139 prop_dictionary_internalize_from_file(ext_file); 140 if (ext_props == NULL) { 141 errx(EXIT_FAILURE, "Failed to read existing " 142 "property list"); 143 } 144 145 free(propsstr); 146 merge_dicts(ext_props, props); 147 148 if (del_props) 149 SIMPLEQ_FOREACH(delp, &del_head, del_items) 150 prop_dictionary_remove(ext_props, 151 delp->del_key); 152 153 propsstr = prop_dictionary_externalize(ext_props); 154 if (propsstr == NULL) 155 errx(EXIT_FAILURE, "Failed to process " 156 "properties"); 157 } 158 159 fputs(propsstr, stdout); 160 } else { 161 if (argc != 1) 162 usage(); 163 cmdargs.ml_filename = argv[0]; 164 cmdargs.ml_flags = flags; 165 cmdargs.ml_props = propsstr; 166 cmdargs.ml_propslen = strlen(propsstr); 167 168 if (modctl(MODCTL_LOAD, &cmdargs)) { 169 err(EXIT_FAILURE, NULL); 170 } 171 } 172 173 free(propsstr); 174 prop_object_release(props); 175 176 exit(EXIT_SUCCESS); 177 } 178 179 static void 180 parse_bool_param(prop_dictionary_t props, const char *name, 181 const char *value) 182 { 183 bool boolvalue; 184 185 assert(name != NULL); 186 assert(value != NULL); 187 188 if (strcasecmp(value, "1") == 0 || 189 strcasecmp(value, "true") == 0 || 190 strcasecmp(value, "yes") == 0) 191 boolvalue = true; 192 else if (strcasecmp(value, "0") == 0 || 193 strcasecmp(value, "false") == 0 || 194 strcasecmp(value, "no") == 0) 195 boolvalue = false; 196 else 197 errx(EXIT_FAILURE, "Invalid boolean value `%s'", value); 198 199 prop_dictionary_set(props, name, prop_bool_create(boolvalue)); 200 } 201 202 static void 203 parse_int_param(prop_dictionary_t props, const char *name, 204 const char *value) 205 { 206 int64_t intvalue; 207 208 assert(name != NULL); 209 assert(value != NULL); 210 211 if (dehumanize_number(value, &intvalue) != 0) 212 err(EXIT_FAILURE, "Invalid integer value `%s'", value); 213 214 prop_dictionary_set(props, name, 215 prop_number_create_integer(intvalue)); 216 } 217 218 static void 219 parse_param(prop_dictionary_t props, const char *origstr, 220 void (*fmt_handler)(prop_dictionary_t, const char *, const char *)) 221 { 222 char *name, *value; 223 224 name = strdup(origstr); 225 226 value = strchr(name, '='); 227 if (value == NULL) { 228 free(name); 229 errx(EXIT_FAILURE, "Invalid parameter `%s'", origstr); 230 } 231 *value++ = '\0'; 232 233 fmt_handler(props, name, value); 234 235 free(name); 236 } 237 238 static void 239 parse_string_param(prop_dictionary_t props, const char *name, 240 const char *value) 241 { 242 243 assert(name != NULL); 244 assert(value != NULL); 245 246 prop_dictionary_set(props, name, prop_string_create_cstring(value)); 247 } 248 249 static void 250 usage(void) 251 { 252 253 (void)fprintf(stderr, 254 "Usage: %s [-fP] [-b var=boolean] [-i var=integer] " 255 "[-s var=string] module\n" 256 " %s -p [-b var=boolean] [-d var] [-i var=integer] " 257 "[-m plist]\n [-s var=string]\n", 258 getprogname(), getprogname()); 259 exit(EXIT_FAILURE); 260 } 261 262 static void 263 merge_dicts(prop_dictionary_t existing_dict, const prop_dictionary_t new_dict) 264 { 265 prop_dictionary_keysym_t props_keysym; 266 prop_object_iterator_t props_iter; 267 prop_object_t props_obj; 268 const char *props_key; 269 270 props_iter = prop_dictionary_iterator(new_dict); 271 if (props_iter == NULL) { 272 errx(EXIT_FAILURE, "Failed to iterate new property list"); 273 } 274 275 while ((props_obj = prop_object_iterator_next(props_iter)) != NULL) { 276 props_keysym = (prop_dictionary_keysym_t)props_obj; 277 props_key = prop_dictionary_keysym_cstring_nocopy(props_keysym); 278 props_obj = prop_dictionary_get_keysym(new_dict, props_keysym); 279 if ((props_obj == NULL) || !prop_dictionary_set(existing_dict, 280 props_key, props_obj)) { 281 errx(EXIT_FAILURE, "Failed to copy " 282 "existing property list"); 283 } 284 } 285 prop_object_iterator_release(props_iter); 286 287 return; 288 } 289