configmenu.c revision 1.7
1 /* $NetBSD: configmenu.c,v 1.7 2019/06/20 00:43:55 christos Exp $ */ 2 3 /*- 4 * Copyright (c) 2012 The NetBSD Foundation, Inc. 5 * All rights reserved. 6 * 7 * This code is derived from software contributed to The NetBSD Foundation 8 * by Jeffrey C. Rizzo 9 * 10 * Redistribution and use in source and binary forms, with or without 11 * modification, are permitted provided that the following conditions 12 * are met: 13 * 1. Redistributions of source code must retain the above copyright 14 * notice, this list of conditions and the following disclaimer. 15 * 2. 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 distribution. 18 * 19 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS 20 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 21 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 22 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS 23 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 24 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 25 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 26 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 27 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 28 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 29 * POSSIBILITY OF SUCH DAMAGE. 30 */ 31 32 /* configmenu.c -- post-installation system configuration menu. */ 33 34 #include <stdio.h> 35 #include <curses.h> 36 #include <unistd.h> 37 #include <errno.h> 38 #include "defs.h" 39 #include "msg_defs.h" 40 #include "menu_defs.h" 41 42 43 static int set_network(struct menudesc*, void *); 44 static int set_timezone_menu(struct menudesc *, void *); 45 static int set_root_shell(struct menudesc *, void *); 46 static int change_root_password(struct menudesc *, void *); 47 static int add_new_user(struct menudesc *, void *); 48 static int set_binpkg(struct menudesc *, void *); 49 static int set_pkgsrc(struct menudesc *, void *); 50 static void config_list_init(void); 51 static void get_rootsh(void); 52 static int toggle_rcvar(struct menudesc *, void *); 53 static void configmenu_hdr(struct menudesc *, void *); 54 static int check_root_password(void); 55 56 char pkgpath[STRSIZE]; 57 char pkgsrcpath[STRSIZE]; 58 59 extern const char *tz_default; 60 61 enum { 62 CONFIGOPT_NETCONF, 63 CONFIGOPT_TZ, 64 CONFIGOPT_ROOTSH, 65 CONFIGOPT_ROOTPW, 66 CONFIGOPT_BINPKG, 67 CONFIGOPT_PKGSRC, 68 CONFIGOPT_SSHD, 69 CONFIGOPT_NTPD, 70 CONFIGOPT_NTPDATE, 71 CONFIGOPT_MDNSD, 72 CONFIGOPT_XDM, 73 CONFIGOPT_CGD, 74 CONFIGOPT_LVM, 75 CONFIGOPT_RAIDFRAME, 76 CONFIGOPT_ADDUSER, 77 CONFIGOPT_LAST 78 }; 79 80 typedef struct configinfo { 81 const char *optname; 82 uint opt; 83 const char *rcvar; 84 int (*action)(struct menudesc *, void *); 85 const char *setting; 86 } configinfo; 87 88 89 configinfo config_list[] = { 90 {MSG_Configure_network, CONFIGOPT_NETCONF, NULL, set_network, MSG_configure}, 91 {MSG_timezone, CONFIGOPT_TZ, NULL, set_timezone_menu, NULL}, 92 {MSG_Root_shell, CONFIGOPT_ROOTSH, NULL, set_root_shell, NULL}, 93 {MSG_change_rootpw, CONFIGOPT_ROOTPW, NULL, change_root_password, MSG_change}, 94 {MSG_enable_binpkg, CONFIGOPT_BINPKG, NULL, set_binpkg, MSG_install}, 95 {MSG_get_pkgsrc, CONFIGOPT_PKGSRC, NULL, set_pkgsrc, MSG_install}, 96 {MSG_enable_sshd, CONFIGOPT_SSHD, "sshd", toggle_rcvar, NULL}, 97 {MSG_enable_ntpd, CONFIGOPT_NTPD, "ntpd", toggle_rcvar, NULL}, 98 {MSG_run_ntpdate, CONFIGOPT_NTPDATE, "ntpdate", toggle_rcvar, NULL}, 99 {MSG_enable_mdnsd, CONFIGOPT_MDNSD, "mdnsd", toggle_rcvar, NULL}, 100 {MSG_enable_xdm, CONFIGOPT_XDM, "xdm", toggle_rcvar, NULL}, 101 {MSG_enable_cgd, CONFIGOPT_CGD, "cgd", toggle_rcvar, NULL}, 102 {MSG_enable_lvm, CONFIGOPT_LVM, "lvm", toggle_rcvar, NULL}, 103 {MSG_enable_raid, CONFIGOPT_RAIDFRAME, "raidframe", toggle_rcvar, NULL}, 104 {MSG_add_a_user, CONFIGOPT_ADDUSER, NULL, add_new_user, ""}, 105 {NULL, CONFIGOPT_LAST, NULL, NULL, NULL} 106 }; 107 108 static void 109 config_list_init(void) 110 { 111 int i; 112 113 for (i=0; i < CONFIGOPT_LAST; i++) { 114 switch (i) { 115 case CONFIGOPT_TZ: 116 get_tz_default(); 117 config_list[CONFIGOPT_TZ].setting = tz_default; 118 break; 119 case CONFIGOPT_ROOTSH: 120 get_rootsh(); 121 break; 122 case CONFIGOPT_ROOTPW: 123 if (check_root_password()) 124 config_list[i].setting = MSG_password_set; 125 else 126 config_list[i].setting = MSG_empty; 127 break; 128 default: 129 if (config_list[i].rcvar != NULL) { 130 if (check_rcvar(config_list[i].rcvar)) 131 config_list[i].setting = MSG_YES; 132 else 133 config_list[i].setting = MSG_NO; 134 } 135 break; 136 } 137 } 138 } 139 140 static void 141 get_rootsh(void) 142 { 143 static char *buf = NULL; 144 145 if (buf != NULL) 146 free(buf); 147 148 if (target_already_root()) 149 collect(T_OUTPUT, &buf, 150 "/usr/bin/awk -F: '$1==\"root\" { print $NF; exit }'" 151 " /etc/passwd"); 152 else 153 collect(T_OUTPUT, &buf, 154 "chroot %s /usr/bin/awk -F: '$1==\"root\" { print $NF; exit }'" 155 " /etc/passwd",target_prefix()); 156 157 config_list[CONFIGOPT_ROOTSH].setting = (const char *)buf; 158 } 159 160 static void 161 set_config(menudesc *menu, int opt, void *arg) 162 { 163 configinfo **configp = arg; 164 configinfo *config = configp[opt]; 165 const char *optname, *setting; 166 167 optname = config->optname; 168 setting = msg_string(config->setting); 169 170 wprintw(menu->mw, "%-50s %-10s", msg_string(optname), setting); 171 } 172 173 static int 174 init_config_menu(configinfo *conf, menu_ent *me, configinfo **ce) 175 { 176 int opt; 177 int configopts; 178 179 for (configopts = 0; ; conf++) { 180 opt = conf->opt; 181 if (opt == CONFIGOPT_LAST) 182 break; 183 *ce = conf; 184 me->opt_menu = OPT_NOMENU; 185 me->opt_flags = 0; 186 me->opt_name = NULL; /* NULL so set_config will draw */ 187 me->opt_exp_name = NULL; 188 me->opt_action = conf->action; 189 configopts++; 190 ce++; 191 me++; 192 } 193 194 return configopts; 195 } 196 197 static int 198 /*ARGSUSED*/ 199 set_timezone_menu(struct menudesc *menu, void *arg) 200 { 201 configinfo **confp = arg; 202 set_timezone(); 203 get_tz_default(); 204 confp[menu->cursel]->setting = tz_default; 205 return 0; 206 } 207 208 static int 209 set_root_shell(struct menudesc *menu, void *arg) 210 { 211 configinfo **confp = arg; 212 213 process_menu(MENU_rootsh, &confp[menu->cursel]->setting); 214 if (run_program(RUN_PROGRESS | RUN_CHROOT, 215 "chpass -s %s root", confp[menu->cursel]->setting) != 0) 216 confp[menu->cursel]->setting = MSG_failed; 217 return 0; 218 } 219 220 static int 221 set_network(struct menudesc *menu, void *arg) 222 { 223 network_up = 0; 224 if (config_network()) 225 mnt_net_config(); 226 return 0; 227 } 228 229 static int 230 check_root_password(void) 231 { 232 char *buf; 233 int rval; 234 235 if (target_already_root()) 236 collect(T_OUTPUT, &buf, "getent passwd root | cut -d: -f2"); 237 else 238 collect(T_OUTPUT, &buf, "chroot %s getent passwd root | " 239 "chroot %s cut -d: -f2", 240 target_prefix(), target_prefix()); 241 242 if (logfp) 243 fprintf(logfp,"buf %s strlen(buf) %zu\n", buf, strlen(buf)); 244 245 if (strlen(buf) <= 1) /* newline */ 246 rval = 0; 247 else 248 rval = 1; 249 free(buf); 250 return rval; 251 } 252 253 static int 254 add_new_user(struct menudesc *menu, void *arg) 255 { 256 char username[STRSIZE] = ""; 257 int inwheel=0; 258 259 msg_prompt(MSG_addusername, NULL, username, sizeof username -1); 260 if (strlen(username) == 0) 261 return 0; 262 inwheel = ask_yesno(MSG_addusertowheel); 263 ushell = "/bin/csh"; 264 process_menu(MENU_usersh, NULL); 265 if (inwheel) 266 run_program(RUN_PROGRESS | RUN_CHROOT, 267 "/usr/sbin/useradd -m -s %s -G wheel %s", 268 ushell, username); 269 else 270 run_program(RUN_PROGRESS | RUN_CHROOT, 271 "/usr/sbin/useradd -m -s %s %s", ushell, username); 272 run_program(RUN_DISPLAY | RUN_PROGRESS | RUN_CHROOT, 273 "passwd -l %s", username); 274 return 0; 275 } 276 277 static int 278 change_root_password(struct menudesc *menu, void *arg) 279 { 280 configinfo **confp = arg; 281 282 msg_display(MSG_rootpw); 283 if (ask_yesno(NULL)) { 284 if (run_program(RUN_DISPLAY | RUN_PROGRESS | RUN_CHROOT, 285 "passwd -l root") == 0) 286 confp[menu->cursel]->setting = MSG_password_set; 287 else 288 confp[menu->cursel]->setting = MSG_failed; 289 } 290 return 0; 291 } 292 293 static int 294 set_binpkg(struct menudesc *menu, void *arg) 295 { 296 configinfo **confp = arg; 297 char additional_pkgs[STRSIZE] = {0}; 298 char pattern[STRSIZE]; 299 int allok = 0; 300 arg_rv parm; 301 302 do { 303 parm.rv = -1; 304 parm.arg = additional_pkgs; 305 process_menu(MENU_binpkg, &parm); 306 if (parm.rv == SET_SKIP) { 307 confp[menu->cursel]->setting = MSG_abandoned; 308 return 0; 309 } 310 311 make_url(pkgpath, &pkg, pkg_dir); 312 if (run_program(RUN_DISPLAY | RUN_PROGRESS | RUN_CHROOT, 313 "pkg_add %s/pkgin", pkgpath) == 0) { 314 allok = 1; 315 } 316 } while (allok == 0); 317 318 /* configure pkgin to use $pkgpath as a repository */ 319 snprintf(pattern, STRSIZE, "s,^[^#].*$,%s,", pkgpath); 320 replace("/usr/pkg/etc/pkgin/repositories.conf", pattern); 321 322 run_program(RUN_DISPLAY | RUN_PROGRESS | RUN_CHROOT, 323 "/usr/pkg/bin/pkgin -y update"); 324 325 if (strlen(additional_pkgs) > 0) 326 run_program(RUN_DISPLAY | RUN_PROGRESS | RUN_CHROOT, 327 "/usr/pkg/bin/pkgin -y install %s", additional_pkgs); 328 329 hit_enter_to_continue(MSG_binpkg_installed, NULL); 330 331 confp[menu->cursel]->setting = MSG_DONE; 332 return 0; 333 } 334 335 static int 336 set_pkgsrc(struct menudesc *menu, void *arg) 337 { 338 configinfo **confp = arg; 339 distinfo dist; 340 341 dist.name = "pkgsrc"; 342 dist.set = SET_PKGSRC; 343 dist.desc = "source for 3rd-party packages"; 344 dist.marker_file = NULL; 345 346 int status = SET_RETRY; 347 348 do { 349 status = get_pkgsrc(); 350 if (status == SET_OK) { 351 status = extract_file(&dist, 0); 352 continue; 353 } else if (status == SET_SKIP) { 354 confp[menu->cursel]->setting = MSG_abandoned; 355 return 0; 356 } 357 if (!ask_yesno(MSG_retry_pkgsrc_network)) { 358 confp[menu->cursel]->setting = MSG_abandoned; 359 return 1; 360 } 361 } 362 while (status == SET_RETRY); 363 364 confp[menu->cursel]->setting = MSG_DONE; 365 return 0; 366 } 367 368 static int 369 toggle_rcvar(struct menudesc *menu, void *arg) 370 { 371 configinfo **confp = arg; 372 int s; 373 const char *setting, *varname; 374 char pattern[STRSIZE]; 375 char buf[STRSIZE]; 376 char *cp; 377 int found = 0; 378 FILE *fp; 379 380 varname = confp[menu->cursel]->rcvar; 381 382 s = check_rcvar(varname); 383 384 /* we're toggling, so invert the sense */ 385 if (s) { 386 confp[menu->cursel]->setting = MSG_NO; 387 setting = "NO"; 388 } else { 389 confp[menu->cursel]->setting = MSG_YES; 390 setting = "YES"; 391 } 392 393 if (!(fp = fopen(target_expand("/etc/rc.conf"), "r"))) { 394 msg_fmt_display(MSG_openfail, "%s%s", 395 target_expand("/etc/rc.conf"), strerror(errno)); 396 hit_enter_to_continue(NULL, NULL); 397 return 0; 398 } 399 400 while (fgets(buf, sizeof buf, fp) != NULL) { 401 cp = buf + strspn(buf, " \t"); /* Skip initial spaces */ 402 if (strncmp(cp, varname, strlen(varname)) == 0) { 403 cp += strlen(varname); 404 if (*cp != '=') 405 continue; 406 buf[strlen(buf) - 1] = 0; 407 snprintf(pattern, sizeof pattern, 408 "s,^%s$,%s=%s,", 409 buf, varname, setting); 410 found = 1; 411 break; 412 } 413 } 414 415 fclose(fp); 416 417 if (!found) { 418 add_rc_conf("%s=%s\n", varname, setting); 419 if (logfp) { 420 fprintf(logfp, "adding %s=%s\n", varname, setting); 421 fflush(logfp); 422 } 423 } else { 424 if (logfp) { 425 fprintf(logfp, "replacement pattern is %s\n", pattern); 426 fflush(logfp); 427 } 428 replace("/etc/rc.conf", pattern); 429 } 430 431 return 0; 432 } 433 434 static void 435 configmenu_hdr(struct menudesc *menu, void *arg) 436 { 437 msg_display(MSG_configmenu); 438 } 439 440 void 441 do_configmenu(struct install_partition_desc *install) 442 { 443 int menu_no; 444 int opts; 445 menu_ent me[CONFIGOPT_LAST]; 446 configinfo *ce[CONFIGOPT_LAST]; 447 448 memset(me, 0, sizeof(me)); 449 450 /* if the target isn't mounted already, figure it out. */ 451 if (install != NULL && target_mounted() == 0) { 452 partman_go = 0; 453 if (find_disks(msg_string(MSG_configure_prior)) < 0) 454 return; 455 456 if (mount_disks(install) != 0) 457 return; 458 } 459 460 config_list_init(); 461 make_url(pkgpath, &pkg, pkg_dir); 462 opts = init_config_menu(config_list, me, ce); 463 464 wrefresh(curscr); 465 wmove(stdscr, 0, 0); 466 wclear(stdscr); 467 wrefresh(stdscr); 468 469 menu_no = new_menu(NULL, me, opts, 0, -4, 0, 70, 470 MC_SCROLL | MC_NOBOX | MC_DFLTEXIT, 471 configmenu_hdr, set_config, NULL, "XXX Help String", 472 MSG_doneconfig); 473 474 process_menu(menu_no, ce); 475 free_menu(menu_no); 476 } 477
Indexes created Fri Oct 03 02:09:57 GMT 2025