2 * The new sysinstall program.
4 * This is probably the last program in the `sysinstall' line - the next
5 * generation being essentially a complete rewrite.
10 * Jordan Hubbard. All rights reserved.
12 * Redistribution and use in source and binary forms, with or without
13 * modification, are permitted provided that the following conditions
15 * 1. Redistributions of source code must retain the above copyright
16 * notice, this list of conditions and the following disclaimer,
17 * verbatim and that no modifications are made prior to this
19 * 2. Redistributions in binary form must reproduce the above copyright
20 * notice, this list of conditions and the following disclaimer in the
21 * documentation and/or other materials provided with the distribution.
23 * THIS SOFTWARE IS PROVIDED BY JORDAN HUBBARD ``AS IS'' AND
24 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
25 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
26 * ARE DISCLAIMED. IN NO EVENT SHALL JORDAN HUBBARD OR HIS PETS BE LIABLE
27 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
28 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
29 * OR SERVICES; LOSS OF USE, DATA, LIFE OR PROFITS; OR BUSINESS INTERRUPTION)
30 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
31 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
32 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
37 #include "sysinstall.h"
38 #include <sys/param.h>
39 #include <sys/mount.h>
47 unsigned int SrcDists;
48 unsigned int XOrgDists;
49 unsigned int KernelDists;
51 enum _disttype { DT_TARBALL, DT_SUBDIST, DT_PACKAGE };
53 typedef struct _dist {
55 unsigned int *my_mask;
57 enum _disttype my_type;
59 char *my_string; /* DT_TARBALL & DT_PACKAGE */
60 struct _dist *my_dist; /* DT_SUBDIST */
64 static Distribution KernelDistTable[];
65 static Distribution SrcDistTable[];
66 static Distribution XOrgDistTable[];
68 #define DTE_TARBALL(name, mask, flag, directory) \
69 { name, mask, DIST_ ## flag, DT_TARBALL, { directory } }
70 #define DTE_PACKAGE(name, mask, flag, package) \
71 { name, mask, DIST_ ## flag, DT_PACKAGE, { package } }
72 #define DTE_SUBDIST(name, mask, flag, subdist) \
73 { name, mask, DIST_ ## flag, DT_SUBDIST, { .my_dist = subdist } }
74 #define DTE_END { NULL, NULL, 0, 0, { NULL } }
76 #define BASE_DIST (&DistTable[0])
78 /* The top-level distribution categories */
79 static Distribution DistTable[] = {
80 DTE_TARBALL("base", &Dists, BASE, "/"),
81 DTE_SUBDIST("kernels", &Dists, KERNEL, KernelDistTable),
82 DTE_TARBALL("doc", &Dists, DOC, "/"),
83 DTE_TARBALL("games", &Dists, GAMES, "/"),
84 DTE_TARBALL("manpages", &Dists, MANPAGES, "/"),
85 DTE_TARBALL("catpages", &Dists, CATPAGES, "/"),
86 DTE_TARBALL("proflibs", &Dists, PROFLIBS, "/"),
87 DTE_TARBALL("dict", &Dists, DICT, "/"),
88 DTE_TARBALL("info", &Dists, INFO, "/"),
90 DTE_TARBALL("lib32", &Dists, LIB32, "/"),
92 DTE_SUBDIST("src", &Dists, SRC, SrcDistTable),
93 DTE_TARBALL("ports", &Dists, PORTS, "/usr"),
94 DTE_TARBALL("local", &Dists, LOCAL, "/"),
95 DTE_PACKAGE("X.Org", &Dists, XORG, "xorg"),
99 /* The kernel distributions */
100 static Distribution KernelDistTable[] = {
101 DTE_TARBALL("GENERIC", &KernelDists, KERNEL_GENERIC, "/boot"),
103 DTE_TARBALL("SMP", &KernelDists, KERNEL_SMP, "/boot"),
108 /* The /usr/src distribution */
109 static Distribution SrcDistTable[] = {
110 DTE_TARBALL("sbase", &SrcDists, SRC_BASE, "/usr/src"),
111 DTE_TARBALL("scddl", &SrcDists, SRC_CDDL, "/usr/src"),
112 DTE_TARBALL("scontrib", &SrcDists, SRC_CONTRIB, "/usr/src"),
113 DTE_TARBALL("scrypto", &SrcDists, SRC_SCRYPTO, "/usr/src"),
114 DTE_TARBALL("sgnu", &SrcDists, SRC_GNU, "/usr/src"),
115 DTE_TARBALL("setc", &SrcDists, SRC_ETC, "/usr/src"),
116 DTE_TARBALL("sgames", &SrcDists, SRC_GAMES, "/usr/src"),
117 DTE_TARBALL("sinclude", &SrcDists, SRC_INCLUDE, "/usr/src"),
118 DTE_TARBALL("skrb5", &SrcDists, SRC_SKERBEROS5, "/usr/src"),
119 DTE_TARBALL("slib", &SrcDists, SRC_LIB, "/usr/src"),
120 DTE_TARBALL("slibexec", &SrcDists, SRC_LIBEXEC, "/usr/src"),
121 DTE_TARBALL("srelease", &SrcDists, SRC_RELEASE, "/usr/src"),
122 DTE_TARBALL("sbin", &SrcDists, SRC_BIN, "/usr/src"),
123 DTE_TARBALL("ssecure", &SrcDists, SRC_SSECURE, "/usr/src"),
124 DTE_TARBALL("ssbin", &SrcDists, SRC_SBIN, "/usr/src"),
125 DTE_TARBALL("sshare", &SrcDists, SRC_SHARE, "/usr/src"),
126 DTE_TARBALL("ssys", &SrcDists, SRC_SYS, "/usr/src"),
127 DTE_TARBALL("subin", &SrcDists, SRC_UBIN, "/usr/src"),
128 DTE_TARBALL("susbin", &SrcDists, SRC_USBIN, "/usr/src"),
129 DTE_TARBALL("stools", &SrcDists, SRC_TOOLS, "/usr/src"),
130 DTE_TARBALL("srescue", &SrcDists, SRC_RESCUE, "/usr/src"),
134 static int distMaybeSetPorts(dialogMenuItem *self);
137 distVerifyFlags(void)
144 Dists |= DIST_KERNEL;
146 msgDebug("Dist Masks: Dists: %0x, Srcs: %0x Kernels: %0x\n", Dists,
147 SrcDists, KernelDists);
148 msgDebug("XServer: %0x\n", XOrgDists);
153 distReset(dialogMenuItem *self)
159 return DITEM_SUCCESS | DITEM_REDRAW;
163 distConfig(dialogMenuItem *self)
169 if ((cp = variable_get(VAR_DIST_MAIN)) != NULL)
172 if ((cp = variable_get(VAR_DIST_SRC)) != NULL)
175 if ((cp = variable_get(VAR_DIST_X11)) != NULL)
176 XOrgDists = atoi(cp);
178 if ((cp = variable_get(VAR_DIST_KERNEL)) != NULL)
179 KernelDists = atoi(cp);
182 return DITEM_SUCCESS | DITEM_REDRAW;
189 XOrgDists = DIST_XORG_ALL;
190 return DITEM_SUCCESS;
197 /* select default kernel based on deduced cpu count */
198 return NCpus > 1 ? DIST_KERNEL_SMP : DIST_KERNEL_GENERIC;
200 return DIST_KERNEL_GENERIC;
205 distSetDeveloper(dialogMenuItem *self)
210 Dists = _DIST_DEVELOPER;
211 SrcDists = DIST_SRC_ALL;
212 KernelDists = selectKernel();
213 i = distMaybeSetPorts(self);
219 distSetXDeveloper(dialogMenuItem *self)
223 i = distSetDeveloper(self);
230 distSetKernDeveloper(dialogMenuItem *self)
235 Dists = _DIST_DEVELOPER;
236 SrcDists = DIST_SRC_SYS | DIST_SRC_BASE;
237 KernelDists = selectKernel();
238 i = distMaybeSetPorts(self);
244 distSetXKernDeveloper(dialogMenuItem *self)
248 i = distSetKernDeveloper(self);
255 distSetUser(dialogMenuItem *self)
261 KernelDists = selectKernel();
262 i = distMaybeSetPorts(self);
268 distSetXUser(dialogMenuItem *self)
272 i = distSetUser(self);
279 distSetMinimum(dialogMenuItem *self)
282 Dists = DIST_BASE | DIST_KERNEL;
283 KernelDists = selectKernel();
285 return DITEM_SUCCESS | DITEM_REDRAW;
289 distSetEverything(dialogMenuItem *self)
294 SrcDists = DIST_SRC_ALL;
295 XOrgDists = DIST_XORG_ALL;
296 KernelDists = DIST_KERNEL_ALL;
297 i = distMaybeSetPorts(self);
299 return i | DITEM_REDRAW;
303 distMaybeSetPorts(dialogMenuItem *self)
305 dialog_clear_norefresh();
306 if (!msgYesNo("Would you like to install the FreeBSD ports collection?\n\n"
307 "This will give you ready access to over 17,000 ported software packages,\n"
308 "at a cost of around 400MB of disk space when \"clean\" and possibly\n"
309 "much more than that when a lot of the distribution tarballs are loaded\n"
310 "(unless you have the extra discs available from a FreeBSD CD/DVD distribution\n"
311 "and can mount them on /cdrom, in which case this is far less of a problem).\n\n"
312 "The ports collection is a very valuable resource and well worth having\n"
313 "on your /usr partition, so it is advisable to say Yes to this option.\n\n"
314 "For more information on the ports collection & the latest ports, visit:\n"
315 " http://www.freebsd.org/ports\n"))
318 Dists &= ~DIST_PORTS;
319 return DITEM_SUCCESS | DITEM_RESTORE;
323 distSetByName(Distribution *dist, char *name)
325 int i, status = FALSE;
327 /* Loop through current set */
328 for (i = 0; dist[i].my_name; i++) {
329 switch (dist[i].my_type) {
332 if (!strcmp(dist[i].my_name, name)) {
333 *(dist[i].my_mask) |= dist[i].my_bit;
338 if (distSetByName(dist[i].my_data.my_dist, name)) {
349 distUnsetByName(Distribution *dist, char *name)
351 int i, status = FALSE;
353 /* Loop through current set */
354 for (i = 0; dist[i].my_name; i++) {
355 switch (dist[i].my_type) {
358 if (!strcmp(dist[i].my_name, name)) {
359 *(dist[i].my_mask) &= ~(dist[i].my_bit);
364 if (distUnsetByName(dist[i].my_data.my_dist, name)) {
373 /* Just for the dispatch stuff */
375 distSetCustom(dialogMenuItem *self)
377 char *cp, *cp2, *tmp;
379 if (!(tmp = variable_get(VAR_DISTS))) {
380 msgDebug("distSetCustom() called without %s variable set.\n", VAR_DISTS);
381 return DITEM_FAILURE;
384 cp = alloca(strlen(tmp) + 1);
386 msgFatal("Couldn't alloca() %d bytes!\n", (int)(strlen(tmp) + 1));
389 if ((cp2 = index(cp, ' ')) != NULL)
391 if (!distSetByName(DistTable, cp))
392 msgDebug("distSetCustom: Warning, no such release \"%s\"\n", cp);
396 return DITEM_SUCCESS;
399 /* Just for the dispatch stuff */
401 distUnsetCustom(dialogMenuItem *self)
403 char *cp, *cp2, *tmp;
405 if (!(tmp = variable_get(VAR_DISTS))) {
406 msgDebug("distUnsetCustom() called without %s variable set.\n", VAR_DISTS);
407 return DITEM_FAILURE;
410 cp = alloca(strlen(tmp) + 1);
412 msgFatal("Couldn't alloca() %d bytes!\n", (int)(strlen(tmp) + 1));
415 if ((cp2 = index(cp, ' ')) != NULL)
417 if (!distUnsetByName(DistTable, cp))
418 msgDebug("distUnsetCustom: Warning, no such release \"%s\"\n", cp);
421 return DITEM_SUCCESS;
425 distSetSrc(dialogMenuItem *self)
429 dialog_clear_norefresh();
430 if (!dmenuOpenSimple(&MenuSrcDistributions, FALSE))
435 return i | DITEM_RESTORE;
439 distSetKernel(dialogMenuItem *self)
443 dialog_clear_norefresh();
444 if (!dmenuOpenSimple(&MenuKernelDistributions, FALSE))
449 return i | DITEM_RESTORE;
452 static Boolean got_intr = FALSE;
454 /* timeout handler */
458 msgDebug("User generated interrupt.\n");
463 check_for_interrupt(void)
473 * translate distribution filename to lower case
474 * as doTARBALL does in release/Makefile
477 translateDist(char trdist[PATH_MAX], const char *dist)
482 * translate distribution filename to lower case
483 * as doTARBALL does in release/Makefile
485 for (j = 0; j < PATH_MAX-1 && dist[j] != '\0'; j++)
486 trdist[j] = tolower(dist[j]);
491 * Try to get distribution as multiple pieces, locating and parsing an
492 * info file which tells us how many we need for this distribution.
495 distExtractTarball(char *path, char *dist, char *my_dir, int is_base)
497 char *buf = NULL, trdist[PATH_MAX], fname[PATH_MAX];
498 struct timeval start, stop;
499 int j, status, total, intr;
500 int cpid, zpid, fd2, chunk, numchunks;
501 properties dist_attr = NULL;
505 translateDist(trdist, dist);
507 msgDebug("%s: path \"%s\" dist \"%s\" trdist \"%s\" "
508 "my_dir \"%s\" %sis_base\n",
509 __func__, path, dist, trdist, my_dir, is_base ? "" : "!");
513 snprintf(fname, sizeof (fname), "%s/%s.inf", path, trdist);
516 fp = DEVICE_GET(mediaDevice, fname, TRUE);
517 intr = check_for_interrupt();
518 if (fp == (FILE *)IO_ERROR || intr || !mediaDevice) {
520 msgDebug("%s: fname %s fp: %p, intr: %d mediaDevice: %p\n",
521 __func__, fname, fp, intr, mediaDevice);
522 /* Hard error, can't continue */
523 if (!msgYesNo("Unable to open %s: %s.\nReinitialize media?",
524 fname, !intr ? "I/O error." : "User interrupt.")) {
525 DEVICE_SHUTDOWN(mediaDevice);
526 if (!DEVICE_INIT(mediaDevice))
531 } else if (fp == NULL) {
532 /* No attributes file, so try as a single file. */
533 snprintf(fname, sizeof(fname), "%s/%s.%s", path, trdist,
534 USE_GZIP ? "tgz" : "tbz");
536 msgDebug("%s: fp is NULL (1) fname: %s\n", __func__, fname);
538 * Passing TRUE as 3rd parm to get routine makes this a "probing"
539 * get, for which errors are not considered too significant.
542 fp = DEVICE_GET(mediaDevice, fname, TRUE);
543 intr = check_for_interrupt();
544 if (fp == (FILE *)IO_ERROR || intr || !mediaDevice) {
546 msgDebug("%s: fname %s fp: %p, intr: %d mediaDevice: %p\n",
547 __func__, fname, fp, intr, mediaDevice);
548 /* Hard error, can't continue */
549 msgConfirm("Unable to open %s: %s", fname,
550 !intr ? "I/O error" : "User interrupt");
551 DEVICE_SHUTDOWN(mediaDevice);
552 if (!DEVICE_INIT(mediaDevice))
555 } else if (fp != NULL) {
556 char *dir = root_bias(my_dir);
558 dialog_clear_norefresh();
559 msgNotify("Extracting %s into %s directory...", dist, dir);
560 status = mediaExtractDist(dir, dist, fp);
565 msgDebug("%s: fp is NULL (2) fname %s\n", __func__, fname);
571 msgDebug("Parsing attributes file for distribution %s\n", dist);
573 dist_attr = properties_read(fileno(fp));
574 intr = check_for_interrupt();
575 if (intr || !dist_attr) {
577 msgDebug("%s: intr %d dist_attr %p\n", __func__, intr, dist_attr);
578 msgConfirm("Cannot parse information file for the %s distribution: %s\n"
579 "Please verify that your media is valid and try again.",
580 dist, !intr ? "I/O error" : "User interrupt");
582 tmp = property_find(dist_attr, "Pieces");
584 numchunks = strtol(tmp, 0, 0);
589 msgDebug("%s: numchunks is zero\n", __func__);
594 msgDebug("Attempting to extract distribution from %u chunks.\n",
598 (void)gettimeofday(&start, (struct timezone *)NULL);
600 /* We have one or more chunks, initialize unpackers... */
601 mediaExtractDistBegin(root_bias(my_dir), &fd2, &zpid, &cpid);
603 /* And go for all the chunks */
604 dialog_clear_norefresh();
605 for (chunk = 0; chunk < numchunks; chunk++) {
606 int n, retval, last_msg, chunksize, realsize;
612 snprintf(fname, sizeof(fname), "cksum.%c%c", (chunk / 26) + 'a',
614 tmp = property_find(dist_attr, fname);
617 tmp = index(tmp, ' ');
618 chunksize = strtol(tmp, 0, 0);
620 snprintf(fname, sizeof(fname), "%s/%s.%c%c", path, trdist, (chunk / 26) + 'a',
623 msgDebug("trying for piece %d of %d: %s\n", chunk + 1, numchunks,
625 fp = DEVICE_GET(mediaDevice, fname, FALSE);
626 intr = check_for_interrupt();
627 /* XXX: this can't work if we get an I/O error */
628 if (fp <= (FILE *)NULL || intr) {
630 msgConfirm("Failed to find %s on this media. Reinitializing media.", fname);
632 msgConfirm("Failed to retreive piece file %s.\n"
633 "%s: Reinitializing media.",
634 fname, !intr ? "I/O error" : "User interrupt");
635 DEVICE_SHUTDOWN(mediaDevice);
636 if (!DEVICE_INIT(mediaDevice))
642 snprintf(prompt, sizeof(prompt), "Extracting %s into %s directory...",
643 dist, root_bias(my_dir));
644 dialog_gauge("Progress", prompt, 8, 15, 6, 50,
645 (chunk + 1) * 100 / numchunks);
647 buf = safe_realloc(buf, chunksize);
652 n = fread(buf + realsize, 1, BUFSIZ, fp);
653 if (check_for_interrupt()) {
654 msgConfirm("Media read error: User interrupt.");
662 /* Print statistics about how we're doing */
663 (void) gettimeofday(&stop, (struct timezone *)0);
664 stop.tv_sec = stop.tv_sec - start.tv_sec;
665 stop.tv_usec = stop.tv_usec - start.tv_usec;
666 if (stop.tv_usec < 0)
667 stop.tv_sec--, stop.tv_usec += 1000000;
668 seconds = stop.tv_sec + (stop.tv_usec / 1000000.0);
672 if (seconds != last_msg) {
674 msgInfo("%10d bytes read from %s dist, chunk %2d of %2d @ %.1f KBytes/sec.",
675 total, dist, chunk + 1, numchunks,
676 (total / seconds) / 1000.0);
681 if (!chunksize || (realsize == chunksize)) {
682 /* No substitution necessary */
683 retval = write(fd2, buf, realsize);
684 if (retval != realsize) {
686 dialog_clear_norefresh();
687 msgConfirm("Write failure on transfer! (wrote %d bytes of %d bytes)", retval, realsize);
691 for (j = 0; j < realsize; j++) {
692 /* On finding CRLF, skip the CR; don't exceed end of buffer. */
693 if ((buf[j] != 0x0d) || (j == total - 1) || (buf[j + 1] != 0x0a)) {
694 retval = write(fd2, buf + j, 1);
697 dialog_clear_norefresh();
698 msgConfirm("Write failure on transfer! (wrote %d bytes of %d bytes)", j, chunksize);
710 properties_free(dist_attr);
713 status = mediaExtractDistEnd(zpid, cpid);
715 (void)mediaExtractDistEnd(zpid, cpid);
722 distExtract(char *parent, Distribution *me)
726 WINDOW *w = savescr();
727 struct sigaction old, new;
731 msgDebug("distExtract: parent: %s, me: %s\n", parent ? parent : "(none)", me->my_name);
733 /* Make ^C fake a sudden timeout */
734 new.sa_handler = handle_intr;
736 (void)sigemptyset(&new.sa_mask);
737 dialog_clear_norefresh();
738 dialog_msgbox("Please Wait", "Extracting all requested distributions...", -1, -1, 0);
739 sigaction(SIGINT, &new, &old);
741 /* Loop through to see if we're in our parent's plans */
742 for (i = 0; me[i].my_name; i++) {
743 dist = me[i].my_name;
744 path = parent ? parent : dist;
746 /* If our bit isn't set, go to the next */
747 if (!(me[i].my_bit & *(me[i].my_mask)))
750 switch (me[i].my_type) {
752 /* Recurse if we actually have a sub-distribution */
753 status = distExtract(dist, me[i].my_data.my_dist);
755 dialog_clear_norefresh();
756 msgConfirm("Unable to transfer all components of the %s distribution.\n"
757 "You may wish to switch media types and try again.\n",
762 dialog_clear_norefresh();
763 msgNotify("Installing %s distribution...", dist);
764 status = (package_add(me[i].my_data.my_string) == DITEM_SUCCESS);
766 dialog_clear_norefresh();
769 status = distExtractTarball(path, dist, me[i].my_data.my_string,
770 &me[i] == BASE_DIST);
772 dialog_clear_norefresh();
773 if (me[i].my_bit != DIST_LOCAL) {
774 status = msgYesNo("Unable to transfer the %s distribution from\n%s.\n\n"
775 "Do you want to try to retrieve it again?",
776 me[i].my_name, mediaDevice->name);
786 * If extract was successful, remove ourselves from further
790 *(me[i].my_mask) &= ~(me[i].my_bit);
792 sigaction(SIGINT, &old, NULL); /* Restore signal handler */
798 printSelected(char *buf, int selected, Distribution *me, int *col)
802 /* Loop through to see if we're in our parent's plans */
803 for (i = 0; me[i].my_name; i++) {
805 /* If our bit isn't set, go to the next */
806 if (!(me[i].my_bit & selected))
809 *col += strlen(me[i].my_name);
814 sprintf(&buf[strlen(buf)], " %s", me[i].my_name);
816 /* Recurse if have a sub-distribution */
817 if (me[i].my_type == DT_SUBDIST)
818 printSelected(buf, *(me[i].my_mask), me[i].my_data.my_dist, col);
823 distExtractAll(dialogMenuItem *self)
825 int old_dists, old_kernel, retries = 0, status = DITEM_SUCCESS;
831 if (!dmenuOpenSimple(&MenuSubDistributions, FALSE) || !Dists)
832 return DITEM_FAILURE;
835 if (!mediaVerify() || !DEVICE_INIT(mediaDevice))
836 return DITEM_FAILURE;
839 old_kernel = KernelDists;
842 dialog_clear_norefresh();
844 msgNotify("Attempting to install all selected distributions..");
846 /* Try for 3 times around the loop, then give up. */
847 while (Dists && ++retries < 3)
848 distExtract(NULL, DistTable);
850 dialog_clear_norefresh();
851 /* Only do base fixup if base dist was successfully extracted */
852 if ((old_dists & DIST_BASE) && !(Dists & DIST_BASE))
853 status |= installFixupBase(self);
854 /* Only do kernel fixup if kernel dist was successfully extracted */
855 if ((old_dists & DIST_KERNEL) && !(Dists & DIST_KERNEL))
856 status |= installFixupKernel(self, old_kernel);
858 /* Clear any local dist flags now */
859 Dists &= ~DIST_LOCAL;
865 dialog_clear_norefresh();
866 printSelected(buf, Dists, DistTable, &col);
867 dialog_clear_norefresh();
869 msgConfirm("Couldn't extract the following distributions. This may\n"
870 "be because they were not available on the installation\n"
871 "media you've chosen:\n\n\t%s", buf);