1 /* $NetBSD: configmenu.c,v 1.21 2025/07/26 15:56:56 martin 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 #if CHECK_ENTROPY 49 static int add_entropy(struct menudesc *, void *); 50 #endif 51 static int set_binpkg(struct menudesc *, void *); 52 static int set_pkgsrc(struct menudesc *, void *); 53 static void config_list_init(void); 54 static void get_rootsh(void); 55 static int toggle_rcvar(struct menudesc *, void *); 56 static int toggle_mdnsd(struct menudesc *, void *); 57 static void configmenu_hdr(struct menudesc *, void *); 58 static int check_root_password(void); 59 static int run_bin_sh(struct menudesc *, void *); 60 61 char pkgpath[STRSIZE]; 62 char pkgsrcpath[STRSIZE]; 63 64 extern const char *tz_default; 65 66 enum { 67 CONFIGOPT_NETCONF, 68 CONFIGOPT_TZ, 69 CONFIGOPT_ROOTSH, 70 CONFIGOPT_ROOTPW, 71 CONFIGOPT_BINPKG, 72 CONFIGOPT_PKGSRC, 73 CONFIGOPT_SSHD, 74 CONFIGOPT_NTPD, 75 CONFIGOPT_NTPDATE, 76 CONFIGOPT_MDNSD, 77 CONFIGOPT_XDM, 78 CONFIGOPT_CGD, 79 CONFIGOPT_LVM, 80 CONFIGOPT_RAIDFRAME, 81 CONFIGOPT_ADDUSER, 82 CONFIGOPT_ADD_ENTROPY, 83 CONFIGOPT_RUN_SH, 84 CONFIGOPT_LAST 85 }; 86 87 typedef struct configinfo { 88 const char *optname; 89 uint opt; 90 const char *rcvar; 91 int (*action)(struct menudesc *, void *); 92 const char *setting; 93 } configinfo; 94 95 96 configinfo config_list[] = { 97 {MSG_Configure_network, CONFIGOPT_NETCONF, NULL, set_network, MSG_configure}, 98 {MSG_timezone, CONFIGOPT_TZ, NULL, set_timezone_menu, NULL}, 99 {MSG_Root_shell, CONFIGOPT_ROOTSH, NULL, set_root_shell, NULL}, 100 {MSG_change_rootpw, CONFIGOPT_ROOTPW, NULL, change_root_password, MSG_change}, 101 {MSG_enable_binpkg, CONFIGOPT_BINPKG, NULL, set_binpkg, MSG_install}, 102 {MSG_get_pkgsrc, CONFIGOPT_PKGSRC, NULL, set_pkgsrc, MSG_install}, 103 {MSG_enable_sshd, CONFIGOPT_SSHD, "sshd", toggle_rcvar, NULL}, 104 {MSG_enable_ntpd, CONFIGOPT_NTPD, "ntpd", toggle_rcvar, NULL}, 105 {MSG_run_ntpdate, CONFIGOPT_NTPDATE, "ntpdate", toggle_rcvar, NULL}, 106 {MSG_enable_mdnsd, CONFIGOPT_MDNSD, "mdnsd", toggle_mdnsd, NULL}, 107 {MSG_enable_xdm, CONFIGOPT_XDM, "xdm", toggle_rcvar, NULL}, 108 {MSG_enable_cgd, CONFIGOPT_CGD, "cgd", toggle_rcvar, NULL}, 109 {MSG_enable_lvm, CONFIGOPT_LVM, "lvm", toggle_rcvar, NULL}, 110 {MSG_enable_raid, CONFIGOPT_RAIDFRAME, "raidframe", toggle_rcvar, NULL}, 111 {MSG_add_a_user, CONFIGOPT_ADDUSER, NULL, add_new_user, ""}, 112 #if CHECK_ENTROPY 113 {MSG_Configure_entropy, CONFIGOPT_ADD_ENTROPY, NULL, add_entropy, ""}, 114 #endif 115 {MSG_Run_bin_sh, CONFIGOPT_RUN_SH, NULL, run_bin_sh, ""}, 116 {NULL, CONFIGOPT_LAST, NULL, NULL, NULL} 117 }; 118 119 static void 120 config_list_init(void) 121 { 122 int i; 123 124 for (i=0; i < CONFIGOPT_LAST; i++) { 125 switch (i) { 126 case CONFIGOPT_TZ: 127 get_tz_default(); 128 config_list[CONFIGOPT_TZ].setting = tz_default; 129 break; 130 case CONFIGOPT_ROOTSH: 131 get_rootsh(); 132 break; 133 case CONFIGOPT_ROOTPW: 134 if (check_root_password()) 135 config_list[i].setting = MSG_password_set; 136 else 137 config_list[i].setting = MSG_empty; 138 break; 139 default: 140 if (config_list[i].rcvar != NULL) { 141 if (check_rcvar(config_list[i].rcvar)) 142 config_list[i].setting = MSG_YES; 143 else 144 config_list[i].setting = MSG_NO; 145 } 146 break; 147 } 148 } 149 } 150 151 static void 152 get_rootsh(void) 153 { 154 static char *buf = NULL; 155 156 if (buf != NULL) 157 free(buf); 158 159 if (target_already_root()) 160 collect(T_OUTPUT, &buf, 161 "/usr/bin/awk -F: '$1==\"root\" { print $NF; exit }'" 162 " /etc/passwd"); 163 else 164 collect(T_OUTPUT, &buf, 165 "chroot %s /usr/bin/awk -F: '$1==\"root\" { print $NF; exit }'" 166 " /etc/passwd",target_prefix()); 167 168 config_list[CONFIGOPT_ROOTSH].setting = (const char *)buf; 169 } 170 171 static void 172 set_config(menudesc *menu, int opt, void *arg) 173 { 174 configinfo **configp = arg; 175 configinfo *config = configp[opt]; 176 const char *optname, *setting; 177 178 optname = config->optname; 179 setting = msg_string(config->setting); 180 181 wprintw(menu->mw, "%-50s %-10s", msg_string(optname), setting); 182 } 183 184 static int 185 init_config_menu(configinfo *conf, menu_ent *me, configinfo **ce) 186 { 187 int opt; 188 int configopts; 189 190 for (configopts = 0; ; conf++) { 191 opt = conf->opt; 192 if (opt == CONFIGOPT_LAST) 193 break; 194 #if CHECK_ENTROPY 195 if (opt == CONFIGOPT_ADD_ENTROPY && entropy_needed() == 0) 196 continue; 197 #endif 198 *ce = conf; 199 memset(me, 0, sizeof(*me)); 200 me->opt_action = conf->action; 201 configopts++; 202 ce++; 203 me++; 204 } 205 206 return configopts; 207 } 208 209 static int 210 /*ARGSUSED*/ 211 set_timezone_menu(struct menudesc *menu, void *arg) 212 { 213 configinfo **confp = arg; 214 set_timezone(); 215 get_tz_default(); 216 confp[menu->cursel]->setting = tz_default; 217 return 0; 218 } 219 220 static int 221 set_root_shell(struct menudesc *menu, void *arg) 222 { 223 configinfo **confp = arg; 224 225 process_menu(MENU_rootsh, &confp[menu->cursel]->setting); 226 if (run_program(RUN_PROGRESS | RUN_CHROOT, 227 "chpass -s %s root", confp[menu->cursel]->setting) != 0) 228 confp[menu->cursel]->setting = MSG_failed; 229 return 0; 230 } 231 232 static int 233 set_network(struct menudesc *menu, void *arg) 234 { 235 network_up = 0; 236 if (config_network(1)) 237 mnt_net_config(); 238 return 0; 239 } 240 241 static int 242 check_root_password(void) 243 { 244 char *buf; 245 int rval; 246 247 if (target_already_root()) 248 collect(T_OUTPUT, &buf, "getent passwd root | cut -d: -f2"); 249 else 250 collect(T_OUTPUT, &buf, "chroot %s getent passwd root | " 251 "chroot %s cut -d: -f2", 252 target_prefix(), target_prefix()); 253 254 if (logfp) 255 fprintf(logfp,"buf %s strlen(buf) %zu\n", buf, strlen(buf)); 256 257 if (strlen(buf) <= 1) /* newline */ 258 rval = 0; 259 else 260 rval = 1; 261 free(buf); 262 return rval; 263 } 264 265 #if CHECK_ENTROPY 266 static int 267 add_entropy(struct menudesc *menu, void *arg) 268 { 269 do_add_entropy(); 270 return 0; 271 } 272 #endif 273 274 static int 275 add_new_user(struct menudesc *menu, void *arg) 276 { 277 char username[STRSIZE] = ""; 278 int inwheel=0; 279 280 msg_prompt(MSG_addusername, NULL, username, sizeof username -1); 281 if (strlen(username) == 0) 282 return 0; 283 inwheel = ask_yesno(MSG_addusertowheel); 284 ushell = "/bin/csh"; 285 process_menu(MENU_usersh, NULL); 286 if (inwheel) 287 run_program(RUN_PROGRESS | RUN_CHROOT, 288 "/usr/sbin/useradd -m -s %s -G wheel %s", 289 ushell, username); 290 else 291 run_program(RUN_PROGRESS | RUN_CHROOT, 292 "/usr/sbin/useradd -m -s %s %s", ushell, username); 293 run_program(RUN_DISPLAY | RUN_PROGRESS | RUN_CHROOT, 294 "passwd -l %s", username); 295 return 0; 296 } 297 298 void 299 root_pw_setup(void) 300 { 301 msg_display(MSG_force_rootpw); 302 run_program(RUN_DISPLAY | RUN_PROGRESS | RUN_CHROOT | RUN_STDSCR, 303 "passwd -l root"); 304 } 305 306 static int 307 change_root_password(struct menudesc *menu, void *arg) 308 { 309 configinfo **confp = arg; 310 311 msg_display(MSG_rootpw); 312 if (ask_yesno(NULL)) { 313 if (run_program(RUN_DISPLAY | RUN_PROGRESS | RUN_CHROOT, 314 "passwd -l root") == 0) 315 confp[menu->cursel]->setting = MSG_password_set; 316 else 317 confp[menu->cursel]->setting = MSG_failed; 318 } 319 return 0; 320 } 321 322 static int 323 set_binpkg(struct menudesc *menu, void *arg) 324 { 325 configinfo **confp = arg; 326 char additional_pkgs[STRSIZE] = {0}; 327 int allok = 0; 328 arg_rv parm; 329 330 if (config_network(0)) 331 mnt_net_config(); 332 333 do { 334 parm.rv = -1; 335 parm.arg = additional_pkgs; 336 process_menu(MENU_binpkg, &parm); 337 if (parm.rv == SET_SKIP) { 338 confp[menu->cursel]->setting = MSG_abandoned; 339 return 0; 340 } 341 342 /* 343 * Make sure we have the TLS certs in a usable state 344 * (if target is a new installation) 345 */ 346 if (pkg.xfer == XFER_HTTPS) 347 run_program(RUN_CHROOT | RUN_SILENT, 348 "/bin/sh /etc/rc.d/certctl_init onestart"); 349 350 make_url(pkgpath, &pkg, pkg_dir); 351 if (run_program(RUN_DISPLAY | RUN_PROGRESS | RUN_CHROOT, 352 "pkg_add %s/pkgin", pkgpath) == 0) { 353 allok = 1; 354 } 355 } while (allok == 0); 356 357 /* configure pkgin to use $pkgpath as a repository */ 358 replace("/usr/pkg/etc/pkgin/repositories.conf", "s,^[^#].*$,%s,", 359 pkgpath); 360 361 run_program(RUN_DISPLAY | RUN_PROGRESS | RUN_CHROOT, 362 "/usr/pkg/bin/pkgin -y update"); 363 364 if (strlen(additional_pkgs) > 0) 365 run_program(RUN_DISPLAY | RUN_PROGRESS | RUN_CHROOT, 366 "/usr/pkg/bin/pkgin -y install %s", additional_pkgs); 367 368 hit_enter_to_continue(MSG_binpkg_installed, NULL); 369 370 confp[menu->cursel]->setting = MSG_DONE; 371 return 0; 372 } 373 374 static int 375 set_pkgsrc(struct menudesc *menu, void *arg) 376 { 377 configinfo **confp = arg; 378 distinfo dist; 379 380 dist.name = "pkgsrc"; 381 dist.set = SET_PKGSRC; 382 dist.desc = "source for 3rd-party packages"; 383 dist.marker_file = NULL; 384 385 int status = SET_RETRY; 386 387 do { 388 status = get_pkgsrc(); 389 if (status == SET_OK) { 390 status = extract_file(&dist, 0); 391 continue; 392 } else if (status == SET_SKIP) { 393 confp[menu->cursel]->setting = MSG_abandoned; 394 return 0; 395 } 396 if (!ask_yesno(MSG_retry_pkgsrc_network)) { 397 confp[menu->cursel]->setting = MSG_abandoned; 398 return 1; 399 } 400 } 401 while (status == SET_RETRY); 402 403 confp[menu->cursel]->setting = MSG_DONE; 404 return 0; 405 } 406 407 static int 408 toggle_rcvar(struct menudesc *menu, void *arg) 409 { 410 configinfo **confp = arg; 411 int s; 412 const char *setting, *varname; 413 char pattern[STRSIZE]; 414 char buf[STRSIZE]; 415 char *cp; 416 int found = 0; 417 FILE *fp; 418 419 varname = confp[menu->cursel]->rcvar; 420 421 s = check_rcvar(varname); 422 423 /* we're toggling, so invert the sense */ 424 if (s) { 425 confp[menu->cursel]->setting = MSG_NO; 426 setting = "NO"; 427 } else { 428 confp[menu->cursel]->setting = MSG_YES; 429 setting = "YES"; 430 } 431 432 if (!(fp = fopen(target_expand("/etc/rc.conf"), "r"))) { 433 msg_fmt_display(MSG_openfail, "%s%s", 434 target_expand("/etc/rc.conf"), strerror(errno)); 435 hit_enter_to_continue(NULL, NULL); 436 return 0; 437 } 438 439 while (fgets(buf, sizeof buf, fp) != NULL) { 440 cp = buf + strspn(buf, " \t"); /* Skip initial spaces */ 441 if (strncmp(cp, varname, strlen(varname)) == 0) { 442 cp += strlen(varname); 443 if (*cp != '=') 444 continue; 445 buf[strlen(buf) - 1] = 0; 446 snprintf(pattern, sizeof pattern, 447 "s,^%s$,%s=%s,", 448 buf, varname, setting); 449 found = 1; 450 break; 451 } 452 } 453 454 fclose(fp); 455 456 if (!found) { 457 add_rc_conf("%s=%s\n", varname, setting); 458 if (logfp) { 459 fprintf(logfp, "adding %s=%s\n", varname, setting); 460 fflush(logfp); 461 } 462 } else { 463 if (logfp) { 464 fprintf(logfp, "replacement pattern is %s\n", pattern); 465 fflush(logfp); 466 } 467 replace("/etc/rc.conf", "%s", pattern); 468 } 469 470 return 0; 471 } 472 473 static int 474 toggle_mdnsd(struct menudesc *menu, void *arg) 475 { 476 configinfo **confp = arg; 477 int s; 478 const char *setting, *varname; 479 480 varname = confp[menu->cursel]->rcvar; 481 482 s = check_rcvar(varname); 483 484 /* we're toggling, so invert the sense */ 485 if (s) { 486 confp[menu->cursel]->setting = MSG_NO; 487 setting = "files dns"; 488 } else { 489 confp[menu->cursel]->setting = MSG_YES; 490 setting = "files multicast_dns dns"; 491 } 492 493 if (logfp) { 494 fprintf(logfp, "setting hosts: %s\n", setting); 495 fflush(logfp); 496 } 497 replace("/etc/nsswitch.conf", "s/^hosts:.*/hosts:\t\t%s/", setting); 498 499 toggle_rcvar(menu, arg); 500 501 return 0; 502 } 503 504 static int 505 run_bin_sh(struct menudesc *menu, void *arg) 506 { 507 endwin(); 508 system("/bin/sh -i -E"); 509 510 return 0; 511 } 512 513 static void 514 configmenu_hdr(struct menudesc *menu, void *arg) 515 { 516 msg_display(MSG_configmenu); 517 } 518 519 void 520 do_configmenu(struct install_partition_desc *install) 521 { 522 int menu_no; 523 int opts; 524 menu_ent me[CONFIGOPT_LAST]; 525 configinfo *ce[CONFIGOPT_LAST]; 526 527 memset(me, 0, sizeof(me)); 528 529 /* if the target isn't mounted already, figure it out. */ 530 if (install != NULL && target_mounted() == 0) { 531 partman_go = 0; 532 if (find_disks(msg_string(MSG_configure_prior), true) < 0) 533 return; 534 535 if (mount_disks(install) != 0) 536 return; 537 } 538 539 config_list_init(); 540 make_url(pkgpath, &pkg, pkg_dir); 541 opts = init_config_menu(config_list, me, ce); 542 543 wrefresh(curscr); 544 wmove(stdscr, 0, 0); 545 wclear(stdscr); 546 wrefresh(stdscr); 547 548 menu_no = new_menu(NULL, me, opts, 0, -4, 0, 70, 549 MC_SCROLL | MC_NOBOX | MC_DFLTEXIT, 550 configmenu_hdr, set_config, NULL, NULL, 551 MSG_doneconfig); 552 553 process_menu(menu_no, ce); 554 free_menu(menu_no); 555 } 556