/* $NetBSD: util.c,v 1.50 2000/03/14 22:42:49 fvdl Exp $ */ /* * Copyright 1997 Piermont Information Systems Inc. * All rights reserved. * * Written by Philip A. Nelson for Piermont Information Systems Inc. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions * are met: * 1. Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * 2. Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * 3. All advertising materials mentioning features or use of this software * must display the following acknowledgement: * This product includes software developed for the NetBSD Project by * Piermont Information Systems Inc. * 4. The name of Piermont Information Systems Inc. may not be used to endorse * or promote products derived from this software without specific prior * written permission. * * THIS SOFTWARE IS PROVIDED BY PIERMONT INFORMATION SYSTEMS INC. ``AS IS'' * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE * ARE DISCLAIMED. IN NO EVENT SHALL PIERMONT INFORMATION SYSTEMS INC. BE * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF * THE POSSIBILITY OF SUCH DAMAGE. * */ /* util.c -- routines that don't really fit anywhere else... */ #include #include #include #include #include #include #include #include #include #include "defs.h" #include "md.h" #include "msg_defs.h" #include "menu_defs.h" /* * local prototypes */ struct tarstats { int nselected; int nfound; int nnotfound; int nerror; int nsuccess; int nskipped; } tarstats; int extract_file __P((char *path)); int extract_dist __P((void)); int cleanup_dist __P((const char *path)); int distribution_sets_exist_p __P((const char *path)); static int check_for __P((const char *type, const char *pathname)); /* * XXX these are WAY bogus! */ int dir_exists_p(path) const char *path; { register int result; result = (run_prog(0, 0, NULL, "test -d %s", path) == 0); return (result); } int file_exists_p(path) const char *path; { register int result; result = (run_prog(0, 0, NULL, "test -f %s", path) == 0); return (result); } int distribution_sets_exist_p(path) const char *path; { char buf[STRSIZE]; int result; result = 1; snprintf(buf, STRSIZE, "%s/%s", path, "kern.tgz"); result = result && file_exists_p(buf); snprintf(buf, STRSIZE, "%s/%s", path, "etc.tgz"); result = result && file_exists_p(buf); return(result); } void get_ramsize() { long len = sizeof(long); int mib[2] = {CTL_HW, HW_PHYSMEM}; sysctl(mib, 2, (void *)&ramsize, (size_t *)&len, NULL, 0); /* Find out how many Megs ... round up. */ rammb = (ramsize + MEG - 1) / MEG; } static int asked = 0; void ask_sizemult(cylsize) int cylsize; { current_cylsize = cylsize; /* XXX */ if (!asked) { msg_display(MSG_sizechoice); process_menu(MENU_sizechoice); } asked = 1; } void reask_sizemult(cylsize) int cylsize; { asked = 0; ask_sizemult(cylsize); } void run_makedev() { char *owd; wclear(stdscr); wrefresh(stdscr); msg_display(MSG_makedev); sleep (1); owd = getcwd(NULL,0); /* make /dev, in case the user didn't extract it. */ make_target_dir("/dev"); target_chdir_or_die("/dev"); run_prog(0, 0, NULL, "/bin/sh MAKEDEV all"); chdir(owd); free(owd); } /* * Load files from floppy. Requires a /mnt2 directory for mounting them. */ int get_via_floppy() { char distname[STRSIZE]; char fddev[STRSIZE] = "/dev/fd0a"; char fname[STRSIZE]; char fullname[STRSIZE]; char catcmd[STRSIZE]; distinfo *list; char post[4]; int mounted = 0; int first; struct stat sb; cd_dist_dir("unloading from floppy"); msg_prompt_add(MSG_fddev, fddev, fddev, STRSIZE); list = dist_list; while (list->name) { strcpy(post, ".aa"); snprintf(distname, STRSIZE, "%s%s", list->name, dist_postfix); while (list->getit && strcmp(&post[1],list->fdlast) <= 0) { snprintf(fname, STRSIZE, "%s%s", list->name, post); snprintf(fullname, STRSIZE, "/mnt2/%s", fname); first = 1; while (!mounted || stat(fullname, &sb)) { if (mounted) run_prog(0, 0, NULL, "/sbin/umount /mnt2"); if (first) msg_display(MSG_fdmount, fname); else msg_display(MSG_fdnotfound, fname); process_menu(MENU_fdok); if (!yesno) return 0; while (run_prog(0, 0, NULL, "/sbin/mount -r -t %s %s /mnt2", fdtype, fddev)) { msg_display(MSG_fdremount, fname); process_menu(MENU_fdremount); if (!yesno) return 0; } mounted = 1; first = 0; } sprintf(catcmd, "/bin/cat %s >> %s", fullname, distname); if (logging) (void)fprintf(log, "%s\n", catcmd); if (scripting) (void)fprintf(script, "%s\n", catcmd); do_system(catcmd); if (post[2] < 'z') post[2]++; else post[2] = 'a', post[1]++; } run_prog(0, 0, NULL, "/sbin/umount /mnt2"); mounted = 0; list++; } #ifndef DEBUG chdir("/"); /* back to current real root */ #endif return 1; } /* * Get from a CDROM distribution. */ int get_via_cdrom() { char tmpdir[STRSIZE]; /* * Fill in final default path, similar to ftp path * because we expect the CDROM structure to be the * same as the ftp site. */ strncat(cdrom_dir, machine, STRSIZE - strlen(cdrom_dir)); strncat(cdrom_dir, ftp_prefix, STRSIZE - strlen(cdrom_dir)); /* Get CD-rom device name and path within CD-rom */ process_menu(MENU_cdromsource); again: run_prog(0, 0, NULL, "/sbin/umount /mnt2"); /* Mount it */ if (run_prog(0, 0, NULL, "/sbin/mount -rt cd9660 /dev/%sa /mnt2", cdrom_dev)) { msg_display(MSG_badsetdir, cdrom_dev); process_menu(MENU_cdrombadmount); if (!yesno) return 0; if (!ignorerror) goto again; } snprintf(tmpdir, STRSIZE, "%s/%s", "/mnt2", cdrom_dir); /* Verify distribution files exist. */ if (distribution_sets_exist_p(tmpdir) == 0) { msg_display(MSG_badsetdir, tmpdir); process_menu(MENU_cdrombadmount); if (!yesno) return (0); if (!ignorerror) goto again; } /* return location, don't clean... */ strncpy(ext_dir, tmpdir, STRSIZE); clean_dist_dir = 0; mnt2_mounted = 1; return 1; } /* * Get from a pathname inside an unmounted local filesystem * (e.g., where sets were preloaded onto a local DOS partition) */ int get_via_localfs() { char tmpdir[STRSIZE]; /* Get device, filesystem, and filepath */ process_menu (MENU_localfssource); again: run_prog(0, 0, NULL, "/sbin/umount /mnt2"); /* Mount it */ if (run_prog(0, 0, NULL, "/sbin/mount -rt %s /dev/%s /mnt2", localfs_fs, localfs_dev)) { msg_display(MSG_localfsbadmount, localfs_dir, localfs_dev); process_menu(MENU_localfsbadmount); if (!yesno) return 0; if (!ignorerror) goto again; } snprintf(tmpdir, STRSIZE, "%s/%s", "/mnt2", localfs_dir); /* Verify distribution files exist. */ if (distribution_sets_exist_p(tmpdir) == 0) { msg_display(MSG_badsetdir, tmpdir); process_menu(MENU_localfsbadmount); if (!yesno) return 0; if (!ignorerror) goto again; } /* return location, don't clean... */ strncpy(ext_dir, tmpdir, STRSIZE); clean_dist_dir = 0; mnt2_mounted = 1; return 1; } /* * Get from an already-mounted pathname. */ int get_via_localdir(void) { /* Get device, filesystem, and filepath */ process_menu(MENU_localdirsource); again: /* Complain if not a directory */ if (dir_exists_p(localfs_dir) == 0) { msg_display(MSG_badlocalsetdir, localfs_dir); process_menu(MENU_localdirbad); if (!yesno) return (0); if (!ignorerror) goto again; } /* Verify distribution files exist. */ if (distribution_sets_exist_p(localfs_dir) == 0) { msg_display(MSG_badsetdir, localfs_dir); process_menu(MENU_localdirbad); if (!yesno) return (0); if (!ignorerror) goto again; } /* return location, don't clean... */ strncpy(ext_dir, localfs_dir, STRSIZE); clean_dist_dir = 0; mnt2_mounted = 0; return 1; } void cd_dist_dir(forwhat) char *forwhat; { char *cwd; /* ask user for the mountpoint. */ msg_prompt(MSG_distdir, dist_dir, dist_dir, STRSIZE, forwhat); /* make sure the directory exists. */ make_target_dir(dist_dir); clean_dist_dir = 1; target_chdir_or_die(dist_dir); /* Set ext_dir for absolute path. */ cwd = getcwd(NULL,0); strncpy(ext_dir, cwd, STRSIZE); free (cwd); } /* * Support for custom distribution fetches / unpacks. */ void toggle_getit(num) int num; { dist_list[num].getit ^= 1; } void show_cur_distsets() { distinfo *list; msg_display(MSG_cur_distsets); msg_table_add(MSG_cur_distsets_header); list = dist_list; while (list->name) { msg_table_add(MSG_cur_distsets_row, list->desc, list->getit ? msg_string(MSG_yes) : msg_string(MSG_no)); list++; } } /* Do we want a verbose extract? */ static int verbose = -1; void ask_verbose_dist() { if (verbose < 0) { msg_display(MSG_verboseextract); process_menu(MENU_noyes); verbose = yesno; wclear(stdscr); wrefresh(stdscr); } } int extract_file(path) char *path; { char *owd; int tarexit, rv; owd = getcwd (NULL,0); /* check tarfile exists */ if (!file_exists_p(path)) { tarstats.nnotfound++; msg_display(MSG_notarfile, path); process_menu(MENU_noyes); return (yesno == 0); } tarstats.nfound++; /* cd to the target root. */ target_chdir_or_die("/"); /* now extract set files files into "./". */ tarexit = run_prog(0, 1, NULL, "pax -zr%spe -f %s", verbose ? "v" : "", path); /* Check tarexit for errors and give warning. */ if (tarexit) { tarstats.nerror++; msg_display(MSG_tarerror, path); process_menu(MENU_noyes); rv = (yesno == 0); } else { tarstats.nsuccess++; rv = 0; } chdir(owd); free(owd); return (rv); } /* * Extract_dist **REQUIRES** an absolute path in ext_dir. Any code * that sets up dist_dir for use by extract_dist needs to put in the * full path name to the directory. */ int extract_dist() { char distname[STRSIZE]; char fname[STRSIZE]; distinfo *list; int punt; /* reset failure/success counters */ memset(&tarstats, 0, sizeof(tarstats)); /*endwin();*/ for (punt = 0, list = dist_list; list->name != NULL; list++) { if (list->getit) { tarstats.nselected++; if (punt) { tarstats.nskipped++; continue; } if (cleanup_dist(list->name) == 0) { msg_display(MSG_cleanup_warn); process_menu(MENU_ok); } (void)snprintf(distname, STRSIZE, "%s%s", list->name, dist_postfix); (void)snprintf(fname, STRSIZE, "%s/%s", ext_dir, distname); /* if extraction failed and user aborted, punt. */ punt = extract_file(fname); } } puts(CL); /* XXX */ wclear(stdscr); wrefresh(stdscr); if (tarstats.nerror == 0 && tarstats.nsuccess == tarstats.nselected) { msg_display(MSG_endtarok); process_menu(MENU_ok); return 0; } else { /* We encountered errors. Let the user know. */ msg_display(MSG_endtar, tarstats.nselected, tarstats.nnotfound, tarstats.nskipped, tarstats.nfound, tarstats.nsuccess, tarstats.nerror); process_menu(MENU_ok); return 1; } } /* * Do pre-extract cleanup for set 'name': * open a file named '/dist/_obsolete file', which contain a list of * files to kill from the target. For each file, test if it is present on * the target. Then display the list of files which will be removed, * ask user for confirmation, and process. * Non-empty directories will be renaned to . */ /* definition for a list of files. */ struct filelist { struct filelist *next; char name[MAXPATHLEN]; mode_t type; }; int cleanup_dist(name) const char *name; { char file_path[MAXPATHLEN]; char file_name[MAXPATHLEN]; FILE *list_file; struct filelist *head = NULL; struct filelist *current; int saved_errno; struct stat st; int retval = 1; int needok = 0; snprintf(file_path, MAXPATHLEN, "/dist/%s_obsolete", name); list_file = fopen(file_path, "r"); if (list_file == NULL) { saved_errno = errno; if (logging) fprintf(log, "Open of %s failed: %s\n", file_path, strerror(saved_errno)); if (saved_errno == ENOENT) return 1; msg_display_add(MSG_openfail, name, strerror(saved_errno)); process_menu(MENU_ok); return 0; } while (fgets(file_name, MAXPATHLEN, list_file)) { /* ignore lines that don't begin with '/' */ if (file_name[0] != '/') continue; /* Remove trailing \n if any */ if (file_name[strlen(file_name)-1] == '\n') file_name[strlen(file_name)-1] = '\0'; snprintf(file_path, MAXPATHLEN, "%s%s", target_prefix(), file_name); if (lstat(file_path, &st) != 0) { saved_errno = errno; if (logging) fprintf(log, "stat() of %s failed: %s\n", file_path, strerror(saved_errno)); if (saved_errno == ENOENT) continue; msg_display_add(MSG_statfail, file_path, strerror(saved_errno)); process_menu(MENU_ok); return 0; } if (head == NULL) { head = current = malloc(sizeof(struct filelist)); if (head == NULL) { fprintf(stderr, "out of memory\n"); exit(1); } } else { current->next = malloc(sizeof(struct filelist)); if (head == NULL) { fprintf(stderr, "out of memory\n"); exit(1); } current = current->next; } current->next = NULL; snprintf(current->name, MAXPATHLEN, "%s", file_path); current->type = st.st_mode & S_IFMT; if (logging) fprintf(log, "Adding file %s, type %d to list of " "obsolete file\n", current->name, current->type); } if (head == NULL) return 1; #if 0 /* XXX doesn't work, too many files printed ! */ msg_display(MSG_deleting_files); for (current = head; current != NULL; current = current->next) { if (current->type != S_IFDIR) { /* XXX msg_printf_add going/gone away */ msg_printf_add("%s ", current->name); } } msg_display_add(MSG_deleting_dirs); for (current = head; current != NULL; current = current->next) { if (current->type == S_IFDIR) { /* XXX msg_printf_add going/gone away */ msg_printf_add("%s ", current->name); } } process_menu(MENU_ok); #endif /* first remove files */ for (current = head; current != NULL; current = current->next) { if (current->type == S_IFDIR) continue; if (scripting) (void)fprintf(script, "rm %s\n", current->name); if (unlink(current->name) != 0) { saved_errno = errno; if (logging) fprintf(log, "rm %s failed: %s\n", current->name, strerror(saved_errno)); msg_display_add(MSG_unlink_fail, current->name, strerror(saved_errno)); retval = 0; needok = 1; } } /* now dirs */ for (current = head; current != NULL; current = current->next) { if (current->type != S_IFDIR) continue; if (rmdir(current->name) == 0) { if (scripting) (void)fprintf(script, "rmdir %s\n", current->name); continue; } saved_errno = errno; if (saved_errno == ENOTEMPTY) { if (logging) fprintf(log, "dir %s not empty, " "trying to rename to %s.old\n", current->name, current->name); snprintf(file_path, MAXPATHLEN, "%s.old", current->name); if (scripting) (void)fprintf(script, "mv %s %s\n", current->name, file_path); needok = 1; if (rename(current->name, file_path) != 0) { saved_errno = errno; if (logging) fprintf(log, "mv %s %s failed: %s\n", current->name, file_path, strerror(saved_errno)); msg_display_add(MSG_rename_fail, current->name, file_path, strerror(errno)); retval = 0; } msg_display_add(MSG_renamed_dir, current->name, file_path); } else { /* rmdir error */ if (logging) fprintf(log, "rm %s failed: %s\n", current->name, strerror(saved_errno)); msg_display_add(MSG_unlink_fail, current->name, strerror(saved_errno)); retval = 0; needok = 1; } } if (needok) process_menu(MENU_ok); return retval; } /* * Get and unpack the distribution. * show success_msg if installation completes. Otherwise,, * show failure_msg and wait for the user to ack it before continuing. * success_msg and failure_msg must both be 0-adic messages. */ int get_and_unpack_sets(success_msg, failure_msg) msg success_msg; msg failure_msg; { /* Ensure mountpoint for distribution files exists in current root. */ (void) mkdir("/mnt2", S_IRWXU| S_IRGRP|S_IXGRP | S_IROTH|S_IXOTH); if (scripting) (void)fprintf(script, "mkdir /mnt2\nchmod 755 /mnt2\n"); /* Find out which files to "get" if we get files. */ wclear(stdscr); wrefresh(stdscr); process_menu(MENU_distset); /* ask user whether to do normal or verbose extraction */ ask_verbose_dist(); /* Get the distribution files */ do { got_dist = 0; process_menu(MENU_distmedium); } while (got_dist == -1); if (nodist) return 1; if (got_dist) { /* Extract the distribution, abort on errors. */ if (extract_dist()) return 1; /* Configure the system */ run_makedev(); /* Other configuration. */ mnt_net_config(); /* Clean up dist dir (use absolute path name) */ if (clean_dist_dir) run_prog(0, 0, NULL, "/bin/rm -rf %s", ext_dir); /* Mounted dist dir? */ if (mnt2_mounted) run_prog(0, 0, NULL, "/sbin/umount /mnt2"); /* Install/Upgrade complete ... reboot or exit to script */ msg_display(success_msg); process_menu(MENU_ok); return 0; } msg_display(failure_msg); process_menu(MENU_ok); return 1; } /* * Do a quick sanity check that the target can reboot. * return 1 if everything OK, 0 if there is a problem. * Uses a table of files we expect to find after a base install/upgrade. */ /* test flag and pathname to check for after unpacking. */ struct check_table { const char *testarg; const char *path;} checks[] = { { "-f", "/netbsd" }, { "-d", "/etc" }, { "-f", "/etc/fstab" }, { "-f", "/sbin/init" }, { "-f", "/bin/sh" }, { "-f", "/etc/rc" }, { "-f", "/etc/rc.subr" }, { "-f", "/etc/rc.conf" }, { "-d" "/dev" }, { "-c", "/dev/console" }, /* XXX check for rootdev in target /dev? */ { "-f", "/etc/fstab" }, { "-f", "/sbin/fsck" }, { "-f", "/sbin/fsck_ffs" }, { "-f", "/sbin/mount" }, { "-f", "/sbin/mount_ffs" }, { "-f", "/sbin/mount_nfs" }, #if defined(DEBUG) || defined(DEBUG_CHECK) { "-f", "/foo/bar" }, /* bad entry to exercise warning */ #endif { 0, 0 } }; /* * Check target for a single file. */ static int check_for(type, pathname) const char *type; const char *pathname; { int found; found = (target_test(type, pathname) == 0); if (found == 0) msg_display(MSG_rootmissing, pathname); return found; } /* * Check that all the files in check_table are present in the * target root. Warn if not found. */ int sanity_check() { int target_ok = 1; struct check_table *p; for (p = checks; p->path; p++) { target_ok = target_ok && check_for(p->testarg, p->path); } if (target_ok) return 0; /* Uh, oh. Something's missing. */ msg_display(MSG_badroot); process_menu(MENU_ok); return 1; } /* set reverse to 1 to default to no */ int askyesno(int reverse) { WINDOW *yesnowin; int c, found; yesnowin = subwin(stdscr, 5, 20, getmaxy(stdscr)/2 - 2, getmaxx(stdscr)/2 - 10); if (yesnowin == NULL) { fprintf(stderr, "sysinst: failed to allocate yes/no box\n"); exit(1); } box(yesnowin, '*', '*'); wmove(yesnowin, 2,2); if (reverse) waddstr(yesnowin, "Yes or No: [N]"); else waddstr(yesnowin, "Yes or No: [Y]"); wrefresh(yesnowin); while ((c = getchar())) { if (c == 'y' || c == 'Y') { found = 1; break; } else if (c == 'n' || c == 'N' ) { found = 0; break; } else if (c == '\n' || c == '\r') { if (reverse) found = 0; else found = 1; break; } } wclear(yesnowin); wrefresh(yesnowin); delwin(yesnowin); refresh(); return(found); }