util.c revision 1.22 1 1.22 martin /* $NetBSD: util.c,v 1.22 2019/06/12 06:20:18 martin Exp $ */
2 1.1 dholland
3 1.1 dholland /*
4 1.1 dholland * Copyright 1997 Piermont Information Systems Inc.
5 1.1 dholland * All rights reserved.
6 1.1 dholland *
7 1.1 dholland * Written by Philip A. Nelson for Piermont Information Systems Inc.
8 1.1 dholland *
9 1.1 dholland * Redistribution and use in source and binary forms, with or without
10 1.1 dholland * modification, are permitted provided that the following conditions
11 1.1 dholland * are met:
12 1.1 dholland * 1. Redistributions of source code must retain the above copyright
13 1.1 dholland * notice, this list of conditions and the following disclaimer.
14 1.1 dholland * 2. Redistributions in binary form must reproduce the above copyright
15 1.1 dholland * notice, this list of conditions and the following disclaimer in the
16 1.1 dholland * documentation and/or other materials provided with the distribution.
17 1.1 dholland * 3. The name of Piermont Information Systems Inc. may not be used to endorse
18 1.1 dholland * or promote products derived from this software without specific prior
19 1.1 dholland * written permission.
20 1.1 dholland *
21 1.1 dholland * THIS SOFTWARE IS PROVIDED BY PIERMONT INFORMATION SYSTEMS INC. ``AS IS''
22 1.1 dholland * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 1.1 dholland * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 1.1 dholland * ARE DISCLAIMED. IN NO EVENT SHALL PIERMONT INFORMATION SYSTEMS INC. BE
25 1.1 dholland * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
26 1.1 dholland * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
27 1.1 dholland * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
28 1.1 dholland * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
29 1.1 dholland * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
30 1.1 dholland * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
31 1.1 dholland * THE POSSIBILITY OF SUCH DAMAGE.
32 1.1 dholland *
33 1.1 dholland */
34 1.1 dholland
35 1.1 dholland /* util.c -- routines that don't really fit anywhere else... */
36 1.1 dholland
37 1.17 martin #include <assert.h>
38 1.17 martin #include <inttypes.h>
39 1.1 dholland #include <stdio.h>
40 1.1 dholland #include <stdarg.h>
41 1.1 dholland #include <string.h>
42 1.1 dholland #include <unistd.h>
43 1.1 dholland #include <sys/mount.h>
44 1.1 dholland #include <sys/dkio.h>
45 1.1 dholland #include <sys/ioctl.h>
46 1.1 dholland #include <sys/types.h>
47 1.1 dholland #include <sys/param.h>
48 1.1 dholland #include <sys/sysctl.h>
49 1.1 dholland #include <sys/stat.h>
50 1.1 dholland #include <sys/statvfs.h>
51 1.1 dholland #include <isofs/cd9660/iso.h>
52 1.1 dholland #include <curses.h>
53 1.1 dholland #include <err.h>
54 1.1 dholland #include <errno.h>
55 1.1 dholland #include <dirent.h>
56 1.1 dholland #include <util.h>
57 1.1 dholland #include "defs.h"
58 1.1 dholland #include "md.h"
59 1.22 martin #include "defsizes.h"
60 1.1 dholland #include "msg_defs.h"
61 1.1 dholland #include "menu_defs.h"
62 1.1 dholland
63 1.1 dholland #define MAX_CD_DEVS 256 /* how many cd drives do we expect to attach */
64 1.1 dholland #define ISO_BLKSIZE ISO_DEFAULT_BLOCK_SIZE
65 1.1 dholland
66 1.1 dholland static const char *msg_yes, *msg_no, *msg_all, *msg_some, *msg_none;
67 1.1 dholland static int select_menu_width;
68 1.1 dholland
69 1.11 christos static uint8_t set_status[SET_GROUP_END];
70 1.1 dholland #define SET_VALID 0x01
71 1.1 dholland #define SET_SELECTED 0x02
72 1.1 dholland #define SET_SKIPPED 0x04
73 1.1 dholland #define SET_INSTALLED 0x08
74 1.1 dholland
75 1.1 dholland struct tarstats {
76 1.1 dholland int nselected;
77 1.1 dholland int nfound;
78 1.1 dholland int nnotfound;
79 1.1 dholland int nerror;
80 1.1 dholland int nsuccess;
81 1.1 dholland int nskipped;
82 1.1 dholland } tarstats;
83 1.1 dholland
84 1.1 dholland distinfo dist_list[] = {
85 1.1 dholland #ifdef SET_KERNEL_1_NAME
86 1.12 martin {SET_KERNEL_1_NAME, SET_KERNEL_1, false, MSG_set_kernel_1, NULL},
87 1.1 dholland #endif
88 1.1 dholland #ifdef SET_KERNEL_2_NAME
89 1.12 martin {SET_KERNEL_2_NAME, SET_KERNEL_2, false, MSG_set_kernel_2, NULL},
90 1.1 dholland #endif
91 1.1 dholland #ifdef SET_KERNEL_3_NAME
92 1.12 martin {SET_KERNEL_3_NAME, SET_KERNEL_3, false, MSG_set_kernel_3, NULL},
93 1.1 dholland #endif
94 1.1 dholland #ifdef SET_KERNEL_4_NAME
95 1.12 martin {SET_KERNEL_4_NAME, SET_KERNEL_4, false, MSG_set_kernel_4, NULL},
96 1.1 dholland #endif
97 1.1 dholland #ifdef SET_KERNEL_5_NAME
98 1.12 martin {SET_KERNEL_5_NAME, SET_KERNEL_5, false, MSG_set_kernel_5, NULL},
99 1.1 dholland #endif
100 1.1 dholland #ifdef SET_KERNEL_6_NAME
101 1.12 martin {SET_KERNEL_6_NAME, SET_KERNEL_6, false, MSG_set_kernel_6, NULL},
102 1.1 dholland #endif
103 1.1 dholland #ifdef SET_KERNEL_7_NAME
104 1.12 martin {SET_KERNEL_7_NAME, SET_KERNEL_7, false, MSG_set_kernel_7, NULL},
105 1.1 dholland #endif
106 1.1 dholland #ifdef SET_KERNEL_8_NAME
107 1.12 martin {SET_KERNEL_8_NAME, SET_KERNEL_8, false, MSG_set_kernel_8, NULL},
108 1.1 dholland #endif
109 1.1 dholland #ifdef SET_KERNEL_9_NAME
110 1.12 martin {SET_KERNEL_9_NAME, SET_KERNEL_9, false, MSG_set_kernel_9, NULL},
111 1.1 dholland #endif
112 1.1 dholland
113 1.12 martin {"modules", SET_MODULES, false, MSG_set_modules, NULL},
114 1.12 martin {"base", SET_BASE, false, MSG_set_base, NULL},
115 1.12 martin {"etc", SET_ETC, false, MSG_set_system, NULL},
116 1.12 martin {"comp", SET_COMPILER, false, MSG_set_compiler, NULL},
117 1.12 martin {"games", SET_GAMES, false, MSG_set_games, NULL},
118 1.12 martin {"man", SET_MAN_PAGES, false, MSG_set_man_pages, NULL},
119 1.12 martin {"misc", SET_MISC, false, MSG_set_misc, NULL},
120 1.12 martin {"tests", SET_TESTS, false, MSG_set_tests, NULL},
121 1.12 martin {"text", SET_TEXT_TOOLS, false, MSG_set_text_tools, NULL},
122 1.12 martin
123 1.12 martin {NULL, SET_GROUP, false, MSG_set_X11, NULL},
124 1.12 martin {"xbase", SET_X11_BASE, false, MSG_set_X11_base, NULL},
125 1.12 martin {"xcomp", SET_X11_PROG, false, MSG_set_X11_prog, NULL},
126 1.12 martin {"xetc", SET_X11_ETC, false, MSG_set_X11_etc, NULL},
127 1.12 martin {"xfont", SET_X11_FONTS, false, MSG_set_X11_fonts, NULL},
128 1.12 martin {"xserver", SET_X11_SERVERS, false, MSG_set_X11_servers, NULL},
129 1.12 martin {NULL, SET_GROUP_END, false, NULL, NULL},
130 1.1 dholland
131 1.1 dholland #ifdef SET_MD_1_NAME
132 1.12 martin {SET_MD_1_NAME, SET_MD_1, false, MSG_set_md_1, NULL},
133 1.1 dholland #endif
134 1.1 dholland #ifdef SET_MD_2_NAME
135 1.12 martin {SET_MD_2_NAME, SET_MD_2, false, MSG_set_md_2, NULL},
136 1.1 dholland #endif
137 1.1 dholland #ifdef SET_MD_3_NAME
138 1.12 martin {SET_MD_3_NAME, SET_MD_3, false, MSG_set_md_3, NULL},
139 1.1 dholland #endif
140 1.1 dholland #ifdef SET_MD_4_NAME
141 1.12 martin {SET_MD_4_NAME, SET_MD_4, false, MSG_set_md_4, NULL},
142 1.1 dholland #endif
143 1.1 dholland
144 1.12 martin {NULL, SET_GROUP, true, MSG_set_source, NULL},
145 1.12 martin {"syssrc", SET_SYSSRC, true, MSG_set_syssrc, NULL},
146 1.12 martin {"src", SET_SRC, true, MSG_set_src, NULL},
147 1.12 martin {"sharesrc", SET_SHARESRC, true, MSG_set_sharesrc, NULL},
148 1.12 martin {"gnusrc", SET_GNUSRC, true, MSG_set_gnusrc, NULL},
149 1.12 martin {"xsrc", SET_XSRC, true, MSG_set_xsrc, NULL},
150 1.12 martin {"debug", SET_DEBUG, false, MSG_set_debug, NULL},
151 1.12 martin {"xdebug", SET_X11_DEBUG, false, MSG_set_xdebug, NULL},
152 1.12 martin {NULL, SET_GROUP_END, false, NULL, NULL},
153 1.1 dholland
154 1.12 martin {NULL, SET_LAST, false, NULL, NULL},
155 1.1 dholland };
156 1.1 dholland
157 1.1 dholland #define MAX_CD_INFOS 16 /* how many media can be found? */
158 1.1 dholland struct cd_info {
159 1.1 dholland char device_name[16];
160 1.1 dholland char menu[100];
161 1.1 dholland };
162 1.1 dholland static struct cd_info cds[MAX_CD_INFOS];
163 1.1 dholland
164 1.22 martin /* flags whether to offer the respective options (depending on helper
165 1.22 martin programs available on install media */
166 1.22 martin int have_raid, have_vnd, have_cgd, have_lvm, have_gpt, have_dk;
167 1.22 martin
168 1.1 dholland /*
169 1.1 dholland * local prototypes
170 1.1 dholland */
171 1.1 dholland
172 1.1 dholland static int check_for(unsigned int mode, const char *pathname);
173 1.1 dholland static int get_iso9660_volname(int dev, int sess, char *volname);
174 1.1 dholland static int get_available_cds(void);
175 1.22 martin static int binary_available(const char *prog);
176 1.1 dholland
177 1.1 dholland void
178 1.1 dholland init_set_status(int flags)
179 1.1 dholland {
180 1.1 dholland static const uint8_t sets_valid[] = {MD_SETS_VALID};
181 1.1 dholland static const uint8_t sets_selected_full[] = {MD_SETS_SELECTED};
182 1.1 dholland static const uint8_t sets_selected_minimal[] = {MD_SETS_SELECTED_MINIMAL};
183 1.1 dholland static const uint8_t sets_selected_nox[] = {MD_SETS_SELECTED_NOX};
184 1.1 dholland static const uint8_t *sets_selected;
185 1.1 dholland unsigned int nelem_selected;
186 1.1 dholland unsigned int i, len;
187 1.1 dholland const char *longest;
188 1.1 dholland
189 1.1 dholland if (flags & SFLAG_MINIMAL) {
190 1.1 dholland sets_selected = sets_selected_minimal;
191 1.22 martin nelem_selected = __arraycount(sets_selected_minimal);
192 1.1 dholland } else if (flags & SFLAG_NOX) {
193 1.1 dholland sets_selected = sets_selected_nox;
194 1.22 martin nelem_selected = __arraycount(sets_selected_nox);
195 1.1 dholland } else {
196 1.1 dholland sets_selected = sets_selected_full;
197 1.22 martin nelem_selected = __arraycount(sets_selected_full);
198 1.1 dholland }
199 1.1 dholland
200 1.22 martin for (i = 0; i < __arraycount(sets_valid); i++)
201 1.1 dholland set_status[sets_valid[i]] = SET_VALID;
202 1.1 dholland for (i = 0; i < nelem_selected; i++)
203 1.1 dholland set_status[sets_selected[i]] |= SET_SELECTED;
204 1.1 dholland
205 1.1 dholland set_status[SET_GROUP] = SET_VALID;
206 1.1 dholland
207 1.1 dholland /* Lookup some strings we need lots of times */
208 1.1 dholland msg_yes = msg_string(MSG_Yes);
209 1.1 dholland msg_no = msg_string(MSG_No);
210 1.1 dholland msg_all = msg_string(MSG_All);
211 1.1 dholland msg_some = msg_string(MSG_Some);
212 1.1 dholland msg_none = msg_string(MSG_None);
213 1.1 dholland
214 1.1 dholland /* Find longest and use it to determine width of selection menu */
215 1.1 dholland len = strlen(msg_no); longest = msg_no;
216 1.1 dholland i = strlen(msg_yes); if (i > len) {len = i; longest = msg_yes; }
217 1.1 dholland i = strlen(msg_all); if (i > len) {len = i; longest = msg_all; }
218 1.1 dholland i = strlen(msg_some); if (i > len) {len = i; longest = msg_some; }
219 1.1 dholland i = strlen(msg_none); if (i > len) {len = i; longest = msg_none; }
220 1.22 martin select_menu_width = snprintf(NULL, 0, "%-30s %s", "", longest);
221 1.1 dholland
222 1.1 dholland /* Give the md code a chance to choose the right kernel, etc. */
223 1.1 dholland md_init_set_status(flags);
224 1.1 dholland }
225 1.1 dholland
226 1.1 dholland int
227 1.1 dholland dir_exists_p(const char *path)
228 1.1 dholland {
229 1.1 dholland
230 1.1 dholland return file_mode_match(path, S_IFDIR);
231 1.1 dholland }
232 1.1 dholland
233 1.1 dholland int
234 1.1 dholland file_exists_p(const char *path)
235 1.1 dholland {
236 1.1 dholland
237 1.1 dholland return file_mode_match(path, S_IFREG);
238 1.1 dholland }
239 1.1 dholland
240 1.1 dholland int
241 1.1 dholland file_mode_match(const char *path, unsigned int mode)
242 1.1 dholland {
243 1.1 dholland struct stat st;
244 1.1 dholland
245 1.1 dholland return (stat(path, &st) == 0 && (st.st_mode & S_IFMT) == mode);
246 1.1 dholland }
247 1.1 dholland
248 1.22 martin /* return ram size in MB */
249 1.22 martin uint64_t
250 1.1 dholland get_ramsize(void)
251 1.1 dholland {
252 1.1 dholland uint64_t ramsize;
253 1.1 dholland size_t len = sizeof ramsize;
254 1.1 dholland int mib[2] = {CTL_HW, HW_PHYSMEM64};
255 1.1 dholland
256 1.1 dholland sysctl(mib, 2, &ramsize, &len, NULL, 0);
257 1.1 dholland
258 1.1 dholland /* Find out how many Megs ... round up. */
259 1.1 dholland return (ramsize + MEG - 1) / MEG;
260 1.1 dholland }
261 1.1 dholland
262 1.1 dholland void
263 1.1 dholland run_makedev(void)
264 1.1 dholland {
265 1.1 dholland char *owd;
266 1.1 dholland
267 1.1 dholland msg_display_add("\n\n");
268 1.1 dholland msg_display_add(MSG_makedev);
269 1.1 dholland
270 1.1 dholland owd = getcwd(NULL, 0);
271 1.1 dholland
272 1.1 dholland /* make /dev, in case the user didn't extract it. */
273 1.1 dholland make_target_dir("/dev");
274 1.1 dholland target_chdir_or_die("/dev");
275 1.1 dholland run_program(0, "/bin/sh MAKEDEV all");
276 1.1 dholland
277 1.1 dholland chdir(owd);
278 1.1 dholland free(owd);
279 1.1 dholland }
280 1.1 dholland
281 1.1 dholland /*
282 1.1 dholland * Performs in-place replacement of a set of patterns in a file that lives
283 1.1 dholland * inside the installed system. The patterns must be separated by a semicolon.
284 1.1 dholland * For example:
285 1.1 dholland *
286 1.1 dholland * replace("/etc/some-file.conf", "s/prop1=NO/prop1=YES/;s/foo/bar/");
287 1.1 dholland */
288 1.1 dholland void
289 1.1 dholland replace(const char *path, const char *patterns, ...)
290 1.1 dholland {
291 1.1 dholland char *spatterns;
292 1.1 dholland va_list ap;
293 1.1 dholland
294 1.1 dholland va_start(ap, patterns);
295 1.1 dholland vasprintf(&spatterns, patterns, ap);
296 1.1 dholland va_end(ap);
297 1.1 dholland if (spatterns == NULL)
298 1.1 dholland err(1, "vasprintf(&spatterns, \"%s\", ...)", patterns);
299 1.1 dholland
300 1.1 dholland run_program(RUN_CHROOT, "sed -an -e '%s;H;$!d;g;w %s' %s", spatterns,
301 1.1 dholland path, path);
302 1.1 dholland
303 1.1 dholland free(spatterns);
304 1.1 dholland }
305 1.1 dholland
306 1.1 dholland static int
307 1.1 dholland floppy_fetch(const char *set_name)
308 1.1 dholland {
309 1.1 dholland char post[4];
310 1.1 dholland msg errmsg;
311 1.1 dholland int menu;
312 1.1 dholland int status;
313 1.1 dholland const char *write_mode = ">";
314 1.1 dholland
315 1.1 dholland strcpy(post, "aa");
316 1.1 dholland
317 1.1 dholland errmsg = "";
318 1.1 dholland menu = MENU_fdok;
319 1.1 dholland for (;;) {
320 1.1 dholland umount_mnt2();
321 1.1 dholland msg_display(errmsg);
322 1.1 dholland msg_display_add(MSG_fdmount, set_name, post);
323 1.1 dholland process_menu(menu, &status);
324 1.1 dholland if (status != SET_CONTINUE)
325 1.1 dholland return status;
326 1.1 dholland menu = MENU_fdremount;
327 1.1 dholland errmsg = MSG_fdremount;
328 1.1 dholland if (run_program(0, "/sbin/mount -r -t %s %s /mnt2",
329 1.1 dholland fd_type, fd_dev))
330 1.1 dholland continue;
331 1.1 dholland mnt2_mounted = 1;
332 1.1 dholland errmsg = MSG_fdnotfound;
333 1.1 dholland
334 1.1 dholland /* Display this because it might take a while.... */
335 1.1 dholland if (run_program(RUN_DISPLAY,
336 1.1 dholland "sh -c '/bin/cat /mnt2/%s.%s %s %s/%s/%s%s'",
337 1.1 dholland set_name, post, write_mode,
338 1.12 martin target_prefix(), xfer_dir, set_name,
339 1.12 martin set_postfix(set_name)))
340 1.1 dholland /* XXX: a read error will give a corrupt file! */
341 1.1 dholland continue;
342 1.1 dholland
343 1.1 dholland /* We got that file, advance to next fragment */
344 1.1 dholland if (post[1] < 'z')
345 1.1 dholland post[1]++;
346 1.1 dholland else
347 1.1 dholland post[1] = 'a', post[0]++;
348 1.1 dholland write_mode = ">>";
349 1.1 dholland errmsg = "";
350 1.1 dholland menu = MENU_fdok;
351 1.1 dholland }
352 1.1 dholland }
353 1.1 dholland
354 1.1 dholland /*
355 1.1 dholland * Load files from floppy. Requires a /mnt2 directory for mounting them.
356 1.1 dholland */
357 1.1 dholland int
358 1.1 dholland get_via_floppy(void)
359 1.1 dholland {
360 1.6 martin int rv = -1;
361 1.6 martin
362 1.6 martin process_menu(MENU_floppysource, &rv);
363 1.6 martin if (rv == SET_RETRY)
364 1.2 martin return SET_RETRY;
365 1.1 dholland
366 1.1 dholland fetch_fn = floppy_fetch;
367 1.1 dholland
368 1.1 dholland /* Set ext_dir for absolute path. */
369 1.1 dholland snprintf(ext_dir_bin, sizeof ext_dir_bin, "%s/%s", target_prefix(), xfer_dir);
370 1.1 dholland snprintf(ext_dir_src, sizeof ext_dir_src, "%s/%s", target_prefix(), xfer_dir);
371 1.1 dholland
372 1.1 dholland return SET_OK;
373 1.1 dholland }
374 1.1 dholland
375 1.1 dholland /*
376 1.1 dholland * Get the volume name of a ISO9660 file system
377 1.1 dholland */
378 1.1 dholland static int
379 1.1 dholland get_iso9660_volname(int dev, int sess, char *volname)
380 1.1 dholland {
381 1.1 dholland int blkno, error, last;
382 1.1 dholland char buf[ISO_BLKSIZE];
383 1.1 dholland struct iso_volume_descriptor *vd = NULL;
384 1.1 dholland struct iso_primary_descriptor *pd = NULL;
385 1.1 dholland
386 1.1 dholland for (blkno = sess+16; blkno < sess+16+100; blkno++) {
387 1.1 dholland error = pread(dev, buf, ISO_BLKSIZE, blkno*ISO_BLKSIZE);
388 1.1 dholland if (error == -1)
389 1.1 dholland return -1;
390 1.1 dholland vd = (struct iso_volume_descriptor *)&buf;
391 1.1 dholland if (memcmp(vd->id, ISO_STANDARD_ID, sizeof(vd->id)) != 0)
392 1.1 dholland return -1;
393 1.1 dholland if (isonum_711((const unsigned char *)&vd->type)
394 1.1 dholland == ISO_VD_PRIMARY) {
395 1.1 dholland pd = (struct iso_primary_descriptor*)buf;
396 1.1 dholland strncpy(volname, pd->volume_id, sizeof pd->volume_id);
397 1.1 dholland last = sizeof pd->volume_id-1;
398 1.1 dholland while (last >= 0
399 1.1 dholland && (volname[last] == ' ' || volname[last] == 0))
400 1.1 dholland last--;
401 1.1 dholland volname[last+1] = 0;
402 1.1 dholland return 0;
403 1.1 dholland }
404 1.1 dholland }
405 1.1 dholland return -1;
406 1.1 dholland }
407 1.1 dholland
408 1.1 dholland /*
409 1.15 martin * Local state while iterating CDs and collecting volumes
410 1.15 martin */
411 1.15 martin struct get_available_cds_state {
412 1.15 martin struct cd_info *info;
413 1.15 martin size_t count;
414 1.15 martin };
415 1.15 martin
416 1.15 martin /*
417 1.15 martin * Callback function: if this is a CD, enumerate all volumes on it
418 1.1 dholland */
419 1.15 martin static bool
420 1.15 martin get_available_cds_helper(void *arg, const char *device)
421 1.1 dholland {
422 1.15 martin struct get_available_cds_state *state = arg;
423 1.15 martin char dname[16], volname[80];
424 1.1 dholland struct disklabel label;
425 1.15 martin int part, dev, error, sess, ready;
426 1.15 martin
427 1.16 martin if (!is_cdrom_device(device, false))
428 1.15 martin return true;
429 1.15 martin
430 1.15 martin sprintf(dname, "/dev/r%s%c", device, 'a'+RAW_PART);
431 1.15 martin dev = open(dname, O_RDONLY, 0);
432 1.15 martin if (dev == -1)
433 1.15 martin return true;
434 1.15 martin
435 1.15 martin ready = 0;
436 1.15 martin error = ioctl(dev, DIOCTUR, &ready);
437 1.15 martin if (error != 0 || ready == 0) {
438 1.15 martin close(dev);
439 1.15 martin return true;
440 1.15 martin }
441 1.15 martin error = ioctl(dev, DIOCGDINFO, &label);
442 1.15 martin close(dev);
443 1.15 martin if (error != 0)
444 1.15 martin return true;
445 1.1 dholland
446 1.15 martin for (part = 0; part < label.d_npartitions; part++) {
447 1.15 martin
448 1.15 martin if (label.d_partitions[part].p_fstype == FS_UNUSED
449 1.15 martin || label.d_partitions[part].p_size == 0)
450 1.15 martin continue;
451 1.15 martin
452 1.15 martin if (label.d_partitions[part].p_fstype == FS_ISO9660) {
453 1.15 martin sess = label.d_partitions[part].p_cdsession;
454 1.15 martin sprintf(dname, "/dev/r%s%c", device, 'a'+part);
455 1.14 martin dev = open(dname, O_RDONLY, 0);
456 1.14 martin if (dev == -1)
457 1.14 martin continue;
458 1.15 martin error = get_iso9660_volname(dev, sess, volname);
459 1.15 martin close(dev);
460 1.15 martin if (error)
461 1.15 martin continue;
462 1.15 martin sprintf(state->info->device_name,
463 1.15 martin "%s%c", device, 'a'+part);
464 1.15 martin sprintf(state->info->menu, "%s (%s)",
465 1.15 martin state->info->device_name, volname);
466 1.15 martin } else {
467 1.15 martin /*
468 1.15 martin * All install CDs use partition
469 1.15 martin * a for the sets.
470 1.15 martin */
471 1.15 martin if (part > 0)
472 1.14 martin continue;
473 1.15 martin sprintf(state->info->device_name,
474 1.15 martin "%s%c", device, 'a'+part);
475 1.15 martin strcpy(state->info->menu, state->info->device_name);
476 1.1 dholland }
477 1.15 martin state->info++;
478 1.15 martin if (++state->count >= MAX_CD_INFOS)
479 1.15 martin return false;
480 1.1 dholland }
481 1.15 martin
482 1.15 martin return true;
483 1.15 martin }
484 1.15 martin
485 1.15 martin /*
486 1.15 martin * Get a list of all available CD media (not drives!), return
487 1.15 martin * the number of entries collected.
488 1.15 martin */
489 1.15 martin static int
490 1.15 martin get_available_cds(void)
491 1.15 martin {
492 1.15 martin struct get_available_cds_state data;
493 1.15 martin
494 1.15 martin data.info = cds;
495 1.15 martin data.count = 0;
496 1.15 martin
497 1.15 martin enumerate_disks(&data, get_available_cds_helper);
498 1.15 martin
499 1.15 martin return data.count;
500 1.1 dholland }
501 1.1 dholland
502 1.1 dholland static int
503 1.1 dholland cd_has_sets(void)
504 1.1 dholland {
505 1.1 dholland /* Mount it */
506 1.1 dholland if (run_program(RUN_SILENT, "/sbin/mount -rt cd9660 /dev/%s /mnt2",
507 1.1 dholland cdrom_dev) != 0)
508 1.1 dholland return 0;
509 1.1 dholland
510 1.1 dholland mnt2_mounted = 1;
511 1.1 dholland
512 1.1 dholland snprintf(ext_dir_bin, sizeof ext_dir_bin, "%s/%s", "/mnt2", set_dir_bin);
513 1.1 dholland snprintf(ext_dir_src, sizeof ext_dir_src, "%s/%s", "/mnt2", set_dir_src);
514 1.1 dholland return dir_exists_p(ext_dir_bin);
515 1.1 dholland }
516 1.1 dholland
517 1.1 dholland /*
518 1.1 dholland * Check whether we can remove the boot media.
519 1.1 dholland * If it is not a local filesystem, return -1.
520 1.1 dholland * If we can not decide for sure (can not tell MD content from plain ffs
521 1.1 dholland * on hard disk, for example), return 0.
522 1.1 dholland * If it is a CD/DVD, return 1.
523 1.1 dholland */
524 1.1 dholland int
525 1.1 dholland boot_media_still_needed(void)
526 1.1 dholland {
527 1.1 dholland struct statvfs sb;
528 1.1 dholland
529 1.1 dholland if (statvfs("/", &sb) == 0) {
530 1.1 dholland if (!(sb.f_flag & ST_LOCAL))
531 1.1 dholland return -1;
532 1.1 dholland if (strcmp(sb.f_fstypename, MOUNT_CD9660) == 0
533 1.1 dholland || strcmp(sb.f_fstypename, MOUNT_UDF) == 0)
534 1.1 dholland return 1;
535 1.1 dholland }
536 1.1 dholland
537 1.1 dholland return 0;
538 1.1 dholland }
539 1.1 dholland
540 1.1 dholland /*
541 1.1 dholland * Get from a CDROM distribution.
542 1.1 dholland * Also used on "installation using bootable install media"
543 1.1 dholland * as the default option in the "distmedium" menu.
544 1.1 dholland */
545 1.1 dholland int
546 1.1 dholland get_via_cdrom(void)
547 1.1 dholland {
548 1.1 dholland menu_ent cd_menu[MAX_CD_INFOS];
549 1.1 dholland struct stat sb;
550 1.6 martin int rv, num_cds, menu_cd, i, selected_cd = 0;
551 1.1 dholland bool silent = false;
552 1.1 dholland int mib[2];
553 1.1 dholland char rootdev[SSTRSIZE] = "";
554 1.1 dholland size_t varlen;
555 1.1 dholland
556 1.1 dholland /* If root is not md(4) and we have set dir, skip this step. */
557 1.1 dholland mib[0] = CTL_KERN;
558 1.1 dholland mib[1] = KERN_ROOT_DEVICE;
559 1.1 dholland varlen = sizeof(rootdev);
560 1.1 dholland (void)sysctl(mib, 2, rootdev, &varlen, NULL, 0);
561 1.1 dholland if (stat(set_dir_bin, &sb) == 0 && S_ISDIR(sb.st_mode) &&
562 1.1 dholland strncmp("md", rootdev, 2) != 0) {
563 1.1 dholland strlcpy(ext_dir_bin, set_dir_bin, sizeof ext_dir_bin);
564 1.1 dholland strlcpy(ext_dir_src, set_dir_src, sizeof ext_dir_src);
565 1.1 dholland return SET_OK;
566 1.1 dholland }
567 1.1 dholland
568 1.1 dholland num_cds = get_available_cds();
569 1.1 dholland if (num_cds <= 0) {
570 1.1 dholland silent = true;
571 1.1 dholland } else if (num_cds == 1) {
572 1.1 dholland /* single CD found, check for sets on it */
573 1.1 dholland strcpy(cdrom_dev, cds[0].device_name);
574 1.1 dholland if (cd_has_sets())
575 1.1 dholland return SET_OK;
576 1.1 dholland } else {
577 1.1 dholland for (i = 0; i< num_cds; i++) {
578 1.1 dholland cd_menu[i].opt_name = cds[i].menu;
579 1.22 martin cd_menu[i].opt_exp_name = NULL;
580 1.1 dholland cd_menu[i].opt_menu = OPT_NOMENU;
581 1.1 dholland cd_menu[i].opt_flags = OPT_EXIT;
582 1.2 martin cd_menu[i].opt_action = set_menu_select;
583 1.1 dholland }
584 1.1 dholland /* create a menu offering available choices */
585 1.1 dholland menu_cd = new_menu(MSG_Available_cds,
586 1.1 dholland cd_menu, num_cds, -1, 4, 0, 0,
587 1.1 dholland MC_SCROLL | MC_NOEXITOPT,
588 1.1 dholland NULL, NULL, NULL, NULL, NULL);
589 1.1 dholland if (menu_cd == -1)
590 1.1 dholland return SET_RETRY;
591 1.1 dholland msg_display(MSG_ask_cd);
592 1.1 dholland process_menu(menu_cd, &selected_cd);
593 1.1 dholland free_menu(menu_cd);
594 1.1 dholland strcpy(cdrom_dev, cds[selected_cd].device_name);
595 1.1 dholland if (cd_has_sets())
596 1.1 dholland return SET_OK;
597 1.1 dholland }
598 1.1 dholland
599 1.1 dholland if (silent)
600 1.1 dholland msg_display("");
601 1.1 dholland else {
602 1.1 dholland umount_mnt2();
603 1.22 martin hit_enter_to_continue(MSG_cd_path_not_found, NULL);
604 1.1 dholland }
605 1.1 dholland
606 1.1 dholland /* ask for paths on the CD */
607 1.6 martin rv = -1;
608 1.6 martin process_menu(MENU_cdromsource, &rv);
609 1.6 martin if (rv == SET_RETRY)
610 1.2 martin return SET_RETRY;
611 1.1 dholland
612 1.1 dholland if (cd_has_sets())
613 1.1 dholland return SET_OK;
614 1.1 dholland
615 1.1 dholland return SET_RETRY;
616 1.1 dholland }
617 1.1 dholland
618 1.1 dholland
619 1.1 dholland /*
620 1.1 dholland * Get from a pathname inside an unmounted local filesystem
621 1.1 dholland * (e.g., where sets were preloaded onto a local DOS partition)
622 1.1 dholland */
623 1.1 dholland int
624 1.1 dholland get_via_localfs(void)
625 1.1 dholland {
626 1.6 martin int rv = -1;
627 1.6 martin
628 1.1 dholland /* Get device, filesystem, and filepath */
629 1.6 martin process_menu (MENU_localfssource, &rv);
630 1.6 martin if (rv == SET_RETRY)
631 1.2 martin return SET_RETRY;
632 1.1 dholland
633 1.1 dholland /* Mount it */
634 1.1 dholland if (run_program(0, "/sbin/mount -rt %s /dev/%s /mnt2",
635 1.1 dholland localfs_fs, localfs_dev))
636 1.1 dholland return SET_RETRY;
637 1.1 dholland
638 1.1 dholland mnt2_mounted = 1;
639 1.1 dholland
640 1.1 dholland snprintf(ext_dir_bin, sizeof ext_dir_bin, "%s/%s/%s",
641 1.1 dholland "/mnt2", localfs_dir, set_dir_bin);
642 1.1 dholland snprintf(ext_dir_src, sizeof ext_dir_src, "%s/%s/%s",
643 1.1 dholland "/mnt2", localfs_dir, set_dir_src);
644 1.1 dholland
645 1.1 dholland return SET_OK;
646 1.1 dholland }
647 1.1 dholland
648 1.1 dholland /*
649 1.1 dholland * Get from an already-mounted pathname.
650 1.1 dholland */
651 1.1 dholland
652 1.1 dholland int
653 1.1 dholland get_via_localdir(void)
654 1.1 dholland {
655 1.6 martin int rv = -1;
656 1.6 martin
657 1.1 dholland /* Get filepath */
658 1.6 martin process_menu(MENU_localdirsource, &rv);
659 1.6 martin if (rv == SET_RETRY)
660 1.2 martin return SET_RETRY;
661 1.1 dholland
662 1.1 dholland /*
663 1.1 dholland * We have to have an absolute path ('cos pax runs in a
664 1.1 dholland * different directory), make it so.
665 1.1 dholland */
666 1.1 dholland snprintf(ext_dir_bin, sizeof ext_dir_bin, "/%s/%s", localfs_dir, set_dir_bin);
667 1.1 dholland snprintf(ext_dir_src, sizeof ext_dir_src, "/%s/%s", localfs_dir, set_dir_src);
668 1.1 dholland
669 1.1 dholland return SET_OK;
670 1.1 dholland }
671 1.1 dholland
672 1.1 dholland
673 1.1 dholland /*
674 1.1 dholland * Support for custom distribution fetches / unpacks.
675 1.1 dholland */
676 1.1 dholland
677 1.1 dholland unsigned int
678 1.1 dholland set_X11_selected(void)
679 1.1 dholland {
680 1.1 dholland int i;
681 1.1 dholland
682 1.1 dholland for (i = SET_X11_FIRST; ++i < SET_X11_LAST;)
683 1.1 dholland if (set_status[i] & SET_SELECTED)
684 1.1 dholland return 1;
685 1.1 dholland return 0;
686 1.1 dholland }
687 1.1 dholland
688 1.1 dholland unsigned int
689 1.1 dholland get_kernel_set(void)
690 1.1 dholland {
691 1.1 dholland int i;
692 1.1 dholland
693 1.1 dholland for (i = SET_KERNEL_FIRST; ++i < SET_KERNEL_LAST;)
694 1.1 dholland if (set_status[i] & SET_SELECTED)
695 1.1 dholland return i;
696 1.1 dholland return SET_NONE;
697 1.1 dholland }
698 1.1 dholland
699 1.1 dholland void
700 1.1 dholland set_kernel_set(unsigned int kernel_set)
701 1.1 dholland {
702 1.1 dholland int i;
703 1.1 dholland
704 1.1 dholland /* only one kernel set is allowed */
705 1.1 dholland for (i = SET_KERNEL_FIRST; ++i < SET_KERNEL_LAST;)
706 1.1 dholland set_status[i] &= ~SET_SELECTED;
707 1.1 dholland set_status[kernel_set] |= SET_SELECTED;
708 1.1 dholland }
709 1.1 dholland
710 1.1 dholland static int
711 1.1 dholland set_toggle(menudesc *menu, void *arg)
712 1.1 dholland {
713 1.1 dholland distinfo **distp = arg;
714 1.1 dholland int set = distp[menu->cursel]->set;
715 1.1 dholland
716 1.1 dholland if (set > SET_KERNEL_FIRST && set < SET_KERNEL_LAST &&
717 1.1 dholland !(set_status[set] & SET_SELECTED))
718 1.1 dholland set_kernel_set(set);
719 1.1 dholland else
720 1.1 dholland set_status[set] ^= SET_SELECTED;
721 1.1 dholland return 0;
722 1.1 dholland }
723 1.1 dholland
724 1.1 dholland static int
725 1.1 dholland set_all_none(menudesc *menu, void *arg, int set, int clr)
726 1.1 dholland {
727 1.1 dholland distinfo **distp = arg;
728 1.1 dholland distinfo *dist = *distp;
729 1.1 dholland int nested;
730 1.1 dholland
731 1.1 dholland for (nested = 0; dist->set != SET_GROUP_END || nested--; dist++) {
732 1.1 dholland if (dist->set == SET_GROUP) {
733 1.1 dholland nested++;
734 1.1 dholland continue;
735 1.1 dholland }
736 1.1 dholland set_status[dist->set] = (set_status[dist->set] & ~clr) | set;
737 1.1 dholland }
738 1.1 dholland return 0;
739 1.1 dholland }
740 1.1 dholland
741 1.1 dholland static int
742 1.1 dholland set_all(menudesc *menu, void *arg)
743 1.1 dholland {
744 1.1 dholland return set_all_none(menu, arg, SET_SELECTED, 0);
745 1.1 dholland }
746 1.1 dholland
747 1.1 dholland static int
748 1.1 dholland set_none(menudesc *menu, void *arg)
749 1.1 dholland {
750 1.1 dholland return set_all_none(menu, arg, 0, SET_SELECTED);
751 1.1 dholland }
752 1.1 dholland
753 1.1 dholland static void
754 1.1 dholland set_label(menudesc *menu, int opt, void *arg)
755 1.1 dholland {
756 1.1 dholland distinfo **distp = arg;
757 1.1 dholland distinfo *dist = distp[opt];
758 1.1 dholland const char *selected;
759 1.1 dholland const char *desc;
760 1.1 dholland int nested;
761 1.1 dholland
762 1.1 dholland desc = dist->desc;
763 1.1 dholland
764 1.1 dholland if (dist->set != SET_GROUP)
765 1.1 dholland selected = set_status[dist->set] & SET_SELECTED ? msg_yes : msg_no;
766 1.1 dholland else {
767 1.1 dholland /* sub menu - display None/Some/All */
768 1.1 dholland nested = 0;
769 1.1 dholland selected = "unknown";
770 1.1 dholland while ((++dist)->set != SET_GROUP_END || nested--) {
771 1.1 dholland if (dist->set == SET_GROUP) {
772 1.1 dholland nested++;
773 1.1 dholland continue;
774 1.1 dholland }
775 1.1 dholland if (!(set_status[dist->set] & SET_VALID))
776 1.1 dholland continue;
777 1.1 dholland if (set_status[dist->set] & SET_SELECTED) {
778 1.1 dholland if (selected == msg_none) {
779 1.1 dholland selected = msg_some;
780 1.1 dholland break;
781 1.1 dholland }
782 1.1 dholland selected = msg_all;
783 1.1 dholland } else {
784 1.1 dholland if (selected == msg_all) {
785 1.1 dholland selected = msg_some;
786 1.1 dholland break;
787 1.1 dholland }
788 1.1 dholland selected = msg_none;
789 1.1 dholland }
790 1.1 dholland }
791 1.1 dholland }
792 1.1 dholland
793 1.22 martin wprintw(menu->mw, "%-30s %s", msg_string(desc), selected);
794 1.1 dholland }
795 1.1 dholland
796 1.1 dholland static int set_sublist(menudesc *menu, void *arg);
797 1.1 dholland
798 1.1 dholland static int
799 1.1 dholland initialise_set_menu(distinfo *dist, menu_ent *me, distinfo **de, int all_none)
800 1.1 dholland {
801 1.1 dholland int set;
802 1.1 dholland int sets;
803 1.1 dholland int nested;
804 1.1 dholland
805 1.1 dholland for (sets = 0; ; dist++) {
806 1.1 dholland set = dist->set;
807 1.1 dholland if (set == SET_LAST || set == SET_GROUP_END)
808 1.1 dholland break;
809 1.1 dholland if (!(set_status[set] & SET_VALID))
810 1.1 dholland continue;
811 1.1 dholland *de = dist;
812 1.1 dholland me->opt_menu = OPT_NOMENU;
813 1.1 dholland me->opt_flags = 0;
814 1.1 dholland me->opt_name = NULL;
815 1.22 martin me->opt_exp_name = NULL;
816 1.1 dholland if (set != SET_GROUP)
817 1.1 dholland me->opt_action = set_toggle;
818 1.1 dholland else {
819 1.1 dholland /* Collapse sublist */
820 1.1 dholland nested = 0;
821 1.1 dholland while ((++dist)->set != SET_GROUP_END || nested--) {
822 1.1 dholland if (dist->set == SET_GROUP)
823 1.1 dholland nested++;
824 1.1 dholland }
825 1.1 dholland me->opt_action = set_sublist;
826 1.1 dholland }
827 1.1 dholland sets++;
828 1.1 dholland de++;
829 1.1 dholland me++;
830 1.1 dholland }
831 1.1 dholland
832 1.1 dholland if (all_none) {
833 1.1 dholland me->opt_menu = OPT_NOMENU;
834 1.1 dholland me->opt_flags = 0;
835 1.1 dholland me->opt_name = MSG_select_all;
836 1.22 martin me->opt_exp_name = NULL;
837 1.1 dholland me->opt_action = set_all;
838 1.1 dholland me++;
839 1.1 dholland me->opt_menu = OPT_NOMENU;
840 1.1 dholland me->opt_flags = 0;
841 1.1 dholland me->opt_name = MSG_select_none;
842 1.22 martin me->opt_exp_name = NULL;
843 1.1 dholland me->opt_action = set_none;
844 1.1 dholland sets += 2;
845 1.1 dholland }
846 1.1 dholland
847 1.1 dholland return sets;
848 1.1 dholland }
849 1.1 dholland
850 1.1 dholland static int
851 1.1 dholland set_sublist(menudesc *menu, void *arg)
852 1.1 dholland {
853 1.1 dholland distinfo *de[SET_LAST];
854 1.1 dholland menu_ent me[SET_LAST];
855 1.1 dholland distinfo **dist = arg;
856 1.1 dholland int menu_no;
857 1.1 dholland int sets;
858 1.1 dholland
859 1.1 dholland sets = initialise_set_menu(dist[menu->cursel] + 1, me, de, 1);
860 1.1 dholland
861 1.1 dholland menu_no = new_menu(NULL, me, sets, 20, 10, 0, select_menu_width,
862 1.1 dholland MC_SUBMENU | MC_SCROLL | MC_DFLTEXIT,
863 1.1 dholland NULL, set_label, NULL, NULL,
864 1.1 dholland MSG_install_selected_sets);
865 1.1 dholland
866 1.1 dholland process_menu(menu_no, de);
867 1.1 dholland free_menu(menu_no);
868 1.1 dholland
869 1.1 dholland return 0;
870 1.1 dholland }
871 1.1 dholland
872 1.1 dholland void
873 1.1 dholland customise_sets(void)
874 1.1 dholland {
875 1.1 dholland distinfo *de[SET_LAST];
876 1.1 dholland menu_ent me[SET_LAST];
877 1.1 dholland int sets;
878 1.1 dholland int menu_no;
879 1.1 dholland
880 1.1 dholland msg_display(MSG_cur_distsets);
881 1.1 dholland msg_table_add(MSG_cur_distsets_header);
882 1.1 dholland
883 1.1 dholland sets = initialise_set_menu(dist_list, me, de, 0);
884 1.1 dholland
885 1.1 dholland menu_no = new_menu(NULL, me, sets, 0, 5, 0, select_menu_width,
886 1.1 dholland MC_SCROLL | MC_NOBOX | MC_DFLTEXIT | MC_NOCLEAR,
887 1.1 dholland NULL, set_label, NULL, NULL,
888 1.1 dholland MSG_install_selected_sets);
889 1.1 dholland
890 1.1 dholland process_menu(menu_no, de);
891 1.1 dholland free_menu(menu_no);
892 1.1 dholland }
893 1.1 dholland
894 1.1 dholland /*
895 1.1 dholland * Extract_file **REQUIRES** an absolute path in ext_dir. Any code
896 1.1 dholland * that sets up xfer_dir for use by extract_file needs to put in the
897 1.1 dholland * full path name to the directory.
898 1.1 dholland */
899 1.1 dholland
900 1.1 dholland int
901 1.1 dholland extract_file(distinfo *dist, int update)
902 1.1 dholland {
903 1.1 dholland char path[STRSIZE];
904 1.1 dholland char *owd;
905 1.1 dholland int rval;
906 1.1 dholland
907 1.1 dholland /* If we might need to tidy up, ensure directory exists */
908 1.1 dholland if (fetch_fn != NULL)
909 1.1 dholland make_target_dir(xfer_dir);
910 1.1 dholland
911 1.1 dholland (void)snprintf(path, sizeof path, "%s/%s%s",
912 1.12 martin ext_dir_for_set(dist->name), dist->name, set_postfix(dist->name));
913 1.1 dholland
914 1.1 dholland owd = getcwd(NULL, 0);
915 1.1 dholland
916 1.1 dholland /* Do we need to fetch the file now? */
917 1.1 dholland if (fetch_fn != NULL) {
918 1.1 dholland rval = fetch_fn(dist->name);
919 1.1 dholland if (rval != SET_OK)
920 1.1 dholland return rval;
921 1.1 dholland }
922 1.1 dholland
923 1.1 dholland /* check tarfile exists */
924 1.1 dholland if (!file_exists_p(path)) {
925 1.1 dholland
926 1.1 dholland #ifdef SUPPORT_8_3_SOURCE_FILESYSTEM
927 1.1 dholland /*
928 1.1 dholland * Update path to use dist->name truncated to the first eight
929 1.1 dholland * characters and check again
930 1.1 dholland */
931 1.1 dholland (void)snprintf(path, sizeof path, "%s/%.8s%.4s", /* 4 as includes '.' */
932 1.12 martin ext_dir_for_set(dist->name), dist->name, set_postfix(dist->name));
933 1.1 dholland if (!file_exists_p(path)) {
934 1.1 dholland #endif /* SUPPORT_8_3_SOURCE_FILESYSTEM */
935 1.1 dholland
936 1.1 dholland tarstats.nnotfound++;
937 1.1 dholland
938 1.22 martin hit_enter_to_continue(MSG_notarfile, NULL);
939 1.1 dholland return SET_RETRY;
940 1.1 dholland }
941 1.1 dholland #ifdef SUPPORT_8_3_SOURCE_FILESYSTEM
942 1.1 dholland }
943 1.1 dholland #endif /* SUPPORT_8_3_SOURCE_FILESYSTEM */
944 1.1 dholland
945 1.1 dholland tarstats.nfound++;
946 1.1 dholland /* cd to the target root. */
947 1.1 dholland if (update && (dist->set == SET_ETC || dist->set == SET_X11_ETC)) {
948 1.1 dholland make_target_dir("/.sysinst");
949 1.1 dholland target_chdir_or_die("/.sysinst");
950 1.1 dholland } else if (dist->set == SET_PKGSRC)
951 1.1 dholland target_chdir_or_die("/usr");
952 1.1 dholland else
953 1.1 dholland target_chdir_or_die("/");
954 1.1 dholland
955 1.1 dholland /*
956 1.1 dholland * /usr/X11R7/lib/X11/xkb/symbols/pc was a directory in 5.0
957 1.1 dholland * but is a file in 5.1 and beyond, so on upgrades we need to
958 1.1 dholland * delete it before extracting the xbase set.
959 1.1 dholland */
960 1.1 dholland if (update && dist->set == SET_X11_BASE)
961 1.1 dholland run_program(0, "rm -rf usr/X11R7/lib/X11/xkb/symbols/pc");
962 1.1 dholland
963 1.1 dholland /* now extract set files into "./". */
964 1.1 dholland rval = run_program(RUN_DISPLAY | RUN_PROGRESS,
965 1.1 dholland "progress -zf %s tar --chroot -xhepf -", path);
966 1.1 dholland
967 1.1 dholland chdir(owd);
968 1.1 dholland free(owd);
969 1.1 dholland
970 1.1 dholland /* Check rval for errors and give warning. */
971 1.1 dholland if (rval != 0) {
972 1.1 dholland tarstats.nerror++;
973 1.1 dholland msg_display(MSG_tarerror, path);
974 1.22 martin hit_enter_to_continue(NULL, NULL);
975 1.1 dholland return SET_RETRY;
976 1.1 dholland }
977 1.1 dholland
978 1.1 dholland if (fetch_fn != NULL && clean_xfer_dir) {
979 1.1 dholland run_program(0, "rm %s", path);
980 1.1 dholland /* Plausibly we should unlink an empty xfer_dir as well */
981 1.1 dholland }
982 1.1 dholland
983 1.1 dholland set_status[dist->set] |= SET_INSTALLED;
984 1.1 dholland tarstats.nsuccess++;
985 1.1 dholland return SET_OK;
986 1.1 dholland }
987 1.1 dholland
988 1.1 dholland static void
989 1.1 dholland skip_set(distinfo *dist, int skip_type)
990 1.1 dholland {
991 1.1 dholland int nested;
992 1.1 dholland int set;
993 1.1 dholland
994 1.1 dholland nested = 0;
995 1.1 dholland while ((++dist)->set != SET_GROUP_END || nested--) {
996 1.1 dholland set = dist->set;
997 1.1 dholland if (set == SET_GROUP) {
998 1.1 dholland nested++;
999 1.1 dholland continue;
1000 1.1 dholland }
1001 1.1 dholland if (set == SET_LAST)
1002 1.1 dholland break;
1003 1.1 dholland if (set_status[set] == (SET_SELECTED | SET_VALID))
1004 1.1 dholland set_status[set] |= SET_SKIPPED;
1005 1.1 dholland tarstats.nskipped++;
1006 1.1 dholland }
1007 1.1 dholland }
1008 1.1 dholland
1009 1.1 dholland /*
1010 1.1 dholland * Get and unpack the distribution.
1011 1.1 dholland * Show success_msg if installation completes.
1012 1.1 dholland * Otherwise show failure_msg and wait for the user to ack it before continuing.
1013 1.1 dholland * success_msg and failure_msg must both be 0-adic messages.
1014 1.1 dholland */
1015 1.1 dholland int
1016 1.1 dholland get_and_unpack_sets(int update, msg setupdone_msg, msg success_msg, msg failure_msg)
1017 1.1 dholland {
1018 1.1 dholland distinfo *dist;
1019 1.1 dholland int status;
1020 1.1 dholland int set;
1021 1.1 dholland
1022 1.1 dholland /* Ensure mountpoint for distribution files exists in current root. */
1023 1.1 dholland (void)mkdir("/mnt2", S_IRWXU| S_IRGRP|S_IXGRP | S_IROTH|S_IXOTH);
1024 1.1 dholland if (script)
1025 1.1 dholland (void)fprintf(script, "mkdir -m 755 /mnt2\n");
1026 1.1 dholland
1027 1.1 dholland /* reset failure/success counters */
1028 1.1 dholland memset(&tarstats, 0, sizeof(tarstats));
1029 1.1 dholland
1030 1.1 dholland /* Find out which files to "get" if we get files. */
1031 1.1 dholland
1032 1.1 dholland /* Accurately count selected sets */
1033 1.1 dholland for (dist = dist_list; (set = dist->set) != SET_LAST; dist++) {
1034 1.11 christos if (dist->name == NULL)
1035 1.11 christos continue;
1036 1.1 dholland if ((set_status[set] & (SET_VALID | SET_SELECTED))
1037 1.1 dholland == (SET_VALID | SET_SELECTED))
1038 1.1 dholland tarstats.nselected++;
1039 1.1 dholland }
1040 1.1 dholland
1041 1.1 dholland status = SET_RETRY;
1042 1.11 christos for (dist = dist_list; (set = dist->set) != SET_LAST; dist++) {
1043 1.1 dholland if (dist->name == NULL)
1044 1.1 dholland continue;
1045 1.1 dholland if (set_status[set] != (SET_VALID | SET_SELECTED))
1046 1.1 dholland continue;
1047 1.1 dholland
1048 1.1 dholland if (status != SET_OK) {
1049 1.1 dholland /* This might force a redraw.... */
1050 1.1 dholland clearok(curscr, 1);
1051 1.1 dholland touchwin(stdscr);
1052 1.1 dholland wrefresh(stdscr);
1053 1.1 dholland /* Sort out the location of the set files */
1054 1.1 dholland do {
1055 1.1 dholland umount_mnt2();
1056 1.1 dholland msg_display(MSG_distmedium, tarstats.nselected,
1057 1.1 dholland tarstats.nsuccess + tarstats.nskipped,
1058 1.1 dholland dist->name);
1059 1.1 dholland fetch_fn = NULL;
1060 1.1 dholland process_menu(MENU_distmedium, &status);
1061 1.1 dholland } while (status == SET_RETRY);
1062 1.1 dholland
1063 1.1 dholland if (status == SET_SKIP) {
1064 1.1 dholland set_status[set] |= SET_SKIPPED;
1065 1.1 dholland tarstats.nskipped++;
1066 1.1 dholland continue;
1067 1.1 dholland }
1068 1.1 dholland if (status == SET_SKIP_GROUP) {
1069 1.1 dholland skip_set(dist, status);
1070 1.1 dholland continue;
1071 1.1 dholland }
1072 1.1 dholland if (status != SET_OK) {
1073 1.22 martin hit_enter_to_continue(failure_msg, NULL);
1074 1.1 dholland return 1;
1075 1.1 dholland }
1076 1.1 dholland }
1077 1.1 dholland
1078 1.1 dholland /* Try to extract this set */
1079 1.1 dholland status = extract_file(dist, update);
1080 1.1 dholland if (status == SET_RETRY)
1081 1.1 dholland dist--;
1082 1.1 dholland }
1083 1.1 dholland
1084 1.1 dholland if (tarstats.nerror == 0 && tarstats.nsuccess == tarstats.nselected) {
1085 1.1 dholland msg_display(MSG_endtarok);
1086 1.1 dholland /* Give user a chance to see the success message */
1087 1.1 dholland sleep(1);
1088 1.1 dholland } else {
1089 1.1 dholland /* We encountered errors. Let the user know. */
1090 1.1 dholland msg_display(MSG_endtar,
1091 1.1 dholland tarstats.nselected, tarstats.nnotfound, tarstats.nskipped,
1092 1.1 dholland tarstats.nfound, tarstats.nsuccess, tarstats.nerror);
1093 1.22 martin hit_enter_to_continue(NULL, NULL);
1094 1.1 dholland }
1095 1.1 dholland
1096 1.1 dholland /*
1097 1.1 dholland * postinstall needs to be run after extracting all sets, because
1098 1.1 dholland * otherwise /var/db/obsolete will only have current information
1099 1.1 dholland * from the base, comp, and etc sets.
1100 1.1 dholland */
1101 1.1 dholland if (update && (set_status[SET_ETC] & SET_INSTALLED)) {
1102 1.1 dholland int oldsendmail;
1103 1.1 dholland oldsendmail = run_program(RUN_DISPLAY | RUN_CHROOT |
1104 1.1 dholland RUN_ERROR_OK | RUN_PROGRESS,
1105 1.1 dholland "/usr/sbin/postinstall -s /.sysinst -d / check mailerconf");
1106 1.1 dholland if (oldsendmail == 1) {
1107 1.1 dholland msg_display(MSG_oldsendmail);
1108 1.6 martin if (ask_yesno(NULL)) {
1109 1.1 dholland run_program(RUN_DISPLAY | RUN_CHROOT,
1110 1.1 dholland "/usr/sbin/postinstall -s /.sysinst -d / fix mailerconf");
1111 1.1 dholland }
1112 1.1 dholland }
1113 1.1 dholland run_program(RUN_DISPLAY | RUN_CHROOT,
1114 1.1 dholland "/usr/sbin/postinstall -s /.sysinst -d / fix");
1115 1.3 tls
1116 1.3 tls /* Don't discard the system's old entropy if any */
1117 1.3 tls run_program(RUN_CHROOT | RUN_SILENT,
1118 1.3 tls "/etc/rc.d/random_seed start");
1119 1.1 dholland }
1120 1.1 dholland
1121 1.1 dholland /* Configure the system */
1122 1.1 dholland if (set_status[SET_BASE] & SET_INSTALLED)
1123 1.1 dholland run_makedev();
1124 1.1 dholland
1125 1.1 dholland if (!update) {
1126 1.3 tls struct stat sb1, sb2;
1127 1.3 tls
1128 1.4 martin if (stat(target_expand("/"), &sb1) == 0
1129 1.4 martin && stat(target_expand("/var"), &sb2) == 0
1130 1.4 martin && sb1.st_dev != sb2.st_dev) {
1131 1.3 tls add_rc_conf("random_file=/etc/entropy-file\n");
1132 1.3 tls if (target_file_exists_p("/boot.cfg")) {
1133 1.3 tls run_program(RUN_CHROOT|RUN_FATAL,
1134 1.3 tls "sh -c 'sed -e s./var/db/./etc/. "
1135 1.3 tls "< /boot.cfg "
1136 1.3 tls "> /tmp/boot.cfg.tmp'");
1137 1.3 tls mv_within_target_or_die("/tmp/boot.cfg.tmp",
1138 1.3 tls "/boot.cfg");
1139 1.3 tls }
1140 1.3 tls }
1141 1.3 tls
1142 1.3 tls /* Save keyboard type */
1143 1.1 dholland save_kb_encoding();
1144 1.1 dholland
1145 1.1 dholland /* Other configuration. */
1146 1.1 dholland mnt_net_config();
1147 1.1 dholland }
1148 1.1 dholland
1149 1.1 dholland /* Mounted dist dir? */
1150 1.1 dholland umount_mnt2();
1151 1.1 dholland
1152 1.3 tls /* Save entropy -- on some systems it's ~all we'll ever get */
1153 1.3 tls run_program(RUN_DISPLAY | RUN_CHROOT | RUN_FATAL | RUN_PROGRESS,
1154 1.3 tls "/etc/rc.d/random_seed stop");
1155 1.1 dholland /* Install/Upgrade complete ... reboot or exit to script */
1156 1.22 martin hit_enter_to_continue(success_msg, NULL);
1157 1.1 dholland return 0;
1158 1.1 dholland }
1159 1.1 dholland
1160 1.1 dholland void
1161 1.1 dholland umount_mnt2(void)
1162 1.1 dholland {
1163 1.1 dholland if (!mnt2_mounted)
1164 1.1 dholland return;
1165 1.1 dholland run_program(RUN_SILENT, "/sbin/umount /mnt2");
1166 1.1 dholland mnt2_mounted = 0;
1167 1.1 dholland }
1168 1.1 dholland
1169 1.1 dholland
1170 1.1 dholland /*
1171 1.1 dholland * Do a quick sanity check that the target can reboot.
1172 1.1 dholland * return 1 if everything OK, 0 if there is a problem.
1173 1.1 dholland * Uses a table of files we expect to find after a base install/upgrade.
1174 1.1 dholland */
1175 1.1 dholland
1176 1.1 dholland /* test flag and pathname to check for after unpacking. */
1177 1.1 dholland struct check_table { unsigned int mode; const char *path;} checks[] = {
1178 1.1 dholland { S_IFREG, "/netbsd" },
1179 1.1 dholland { S_IFDIR, "/etc" },
1180 1.1 dholland { S_IFREG, "/etc/fstab" },
1181 1.1 dholland { S_IFREG, "/sbin/init" },
1182 1.1 dholland { S_IFREG, "/bin/sh" },
1183 1.1 dholland { S_IFREG, "/etc/rc" },
1184 1.1 dholland { S_IFREG, "/etc/rc.subr" },
1185 1.1 dholland { S_IFREG, "/etc/rc.conf" },
1186 1.1 dholland { S_IFDIR, "/dev" },
1187 1.1 dholland { S_IFCHR, "/dev/console" },
1188 1.1 dholland /* XXX check for rootdev in target /dev? */
1189 1.1 dholland { S_IFREG, "/sbin/fsck" },
1190 1.1 dholland { S_IFREG, "/sbin/fsck_ffs" },
1191 1.1 dholland { S_IFREG, "/sbin/mount" },
1192 1.1 dholland { S_IFREG, "/sbin/mount_ffs" },
1193 1.1 dholland { S_IFREG, "/sbin/mount_nfs" },
1194 1.1 dholland #if defined(DEBUG) || defined(DEBUG_CHECK)
1195 1.1 dholland { S_IFREG, "/foo/bar" }, /* bad entry to exercise warning */
1196 1.1 dholland #endif
1197 1.1 dholland { 0, 0 }
1198 1.1 dholland
1199 1.1 dholland };
1200 1.1 dholland
1201 1.1 dholland /*
1202 1.1 dholland * Check target for a single file.
1203 1.1 dholland */
1204 1.1 dholland static int
1205 1.1 dholland check_for(unsigned int mode, const char *pathname)
1206 1.1 dholland {
1207 1.1 dholland int found;
1208 1.1 dholland
1209 1.1 dholland found = (target_test(mode, pathname) == 0);
1210 1.1 dholland if (found == 0)
1211 1.1 dholland msg_display(MSG_rootmissing, pathname);
1212 1.1 dholland return found;
1213 1.1 dholland }
1214 1.1 dholland
1215 1.1 dholland /*
1216 1.1 dholland * Check that all the files in check_table are present in the
1217 1.1 dholland * target root. Warn if not found.
1218 1.1 dholland */
1219 1.1 dholland int
1220 1.1 dholland sanity_check(void)
1221 1.1 dholland {
1222 1.1 dholland int target_ok = 1;
1223 1.1 dholland struct check_table *p;
1224 1.1 dholland
1225 1.1 dholland for (p = checks; p->path; p++) {
1226 1.1 dholland target_ok = target_ok && check_for(p->mode, p->path);
1227 1.1 dholland }
1228 1.1 dholland if (target_ok)
1229 1.1 dholland return 0;
1230 1.1 dholland
1231 1.1 dholland /* Uh, oh. Something's missing. */
1232 1.22 martin hit_enter_to_continue(MSG_badroot, NULL);
1233 1.1 dholland return 1;
1234 1.1 dholland }
1235 1.1 dholland
1236 1.1 dholland /*
1237 1.1 dholland * Some globals to pass things back from callbacks
1238 1.1 dholland */
1239 1.1 dholland static char zoneinfo_dir[STRSIZE];
1240 1.1 dholland static int zonerootlen;
1241 1.1 dholland static char *tz_selected; /* timezonename (relative to share/zoneinfo */
1242 1.1 dholland const char *tz_default; /* UTC, or whatever /etc/localtime points to */
1243 1.1 dholland static char tz_env[STRSIZE];
1244 1.1 dholland static int save_cursel, save_topline;
1245 1.1 dholland
1246 1.1 dholland /*
1247 1.1 dholland * Callback from timezone menu
1248 1.1 dholland */
1249 1.1 dholland static int
1250 1.1 dholland set_tz_select(menudesc *m, void *arg)
1251 1.1 dholland {
1252 1.1 dholland time_t t;
1253 1.1 dholland char *new;
1254 1.5 christos struct tm *tm;
1255 1.1 dholland
1256 1.1 dholland if (m && strcmp(tz_selected, m->opts[m->cursel].opt_name) != 0) {
1257 1.1 dholland /* Change the displayed timezone */
1258 1.1 dholland new = strdup(m->opts[m->cursel].opt_name);
1259 1.1 dholland if (new == NULL)
1260 1.1 dholland return 0;
1261 1.1 dholland free(tz_selected);
1262 1.1 dholland tz_selected = new;
1263 1.1 dholland snprintf(tz_env, sizeof tz_env, "%.*s%s",
1264 1.1 dholland zonerootlen, zoneinfo_dir, tz_selected);
1265 1.1 dholland setenv("TZ", tz_env, 1);
1266 1.1 dholland }
1267 1.1 dholland if (m)
1268 1.1 dholland /* Warp curser to 'Exit' line on menu */
1269 1.1 dholland m->cursel = -1;
1270 1.1 dholland
1271 1.1 dholland /* Update displayed time */
1272 1.1 dholland t = time(NULL);
1273 1.5 christos tm = localtime(&t);
1274 1.1 dholland msg_display(MSG_choose_timezone,
1275 1.5 christos tz_default, tz_selected, safectime(&t), tm ? tm->tm_zone :
1276 1.5 christos "?");
1277 1.1 dholland return 0;
1278 1.1 dholland }
1279 1.1 dholland
1280 1.1 dholland static int
1281 1.1 dholland set_tz_back(menudesc *m, void *arg)
1282 1.1 dholland {
1283 1.1 dholland
1284 1.1 dholland zoneinfo_dir[zonerootlen] = 0;
1285 1.1 dholland m->cursel = save_cursel;
1286 1.1 dholland m->topline = save_topline;
1287 1.1 dholland return 0;
1288 1.1 dholland }
1289 1.1 dholland
1290 1.1 dholland static int
1291 1.1 dholland set_tz_dir(menudesc *m, void *arg)
1292 1.1 dholland {
1293 1.1 dholland
1294 1.1 dholland strlcpy(zoneinfo_dir + zonerootlen, m->opts[m->cursel].opt_name,
1295 1.1 dholland sizeof zoneinfo_dir - zonerootlen);
1296 1.1 dholland save_cursel = m->cursel;
1297 1.1 dholland save_topline = m->topline;
1298 1.1 dholland m->cursel = 0;
1299 1.1 dholland m->topline = 0;
1300 1.1 dholland return 0;
1301 1.1 dholland }
1302 1.1 dholland
1303 1.1 dholland /*
1304 1.1 dholland * Alarm-handler to update example-display
1305 1.1 dholland */
1306 1.1 dholland static void
1307 1.1 dholland /*ARGSUSED*/
1308 1.1 dholland timezone_sig(int sig)
1309 1.1 dholland {
1310 1.1 dholland
1311 1.1 dholland set_tz_select(NULL, NULL);
1312 1.1 dholland alarm(60);
1313 1.1 dholland }
1314 1.1 dholland
1315 1.1 dholland static int
1316 1.1 dholland tz_sort(const void *a, const void *b)
1317 1.1 dholland {
1318 1.1 dholland return strcmp(((const menu_ent *)a)->opt_name, ((const menu_ent *)b)->opt_name);
1319 1.1 dholland }
1320 1.1 dholland
1321 1.1 dholland static void
1322 1.1 dholland tzm_set_names(menudesc *m, void *arg)
1323 1.1 dholland {
1324 1.1 dholland DIR *dir;
1325 1.1 dholland struct dirent *dp;
1326 1.1 dholland static int nfiles;
1327 1.1 dholland static int maxfiles = 32;
1328 1.1 dholland static menu_ent *tz_menu;
1329 1.1 dholland static char **tz_names;
1330 1.1 dholland void *p;
1331 1.1 dholland int maxfname;
1332 1.1 dholland char *fp;
1333 1.1 dholland struct stat sb;
1334 1.1 dholland
1335 1.1 dholland if (tz_menu == NULL)
1336 1.1 dholland tz_menu = malloc(maxfiles * sizeof *tz_menu);
1337 1.1 dholland if (tz_names == NULL)
1338 1.1 dholland tz_names = malloc(maxfiles * sizeof *tz_names);
1339 1.1 dholland if (tz_menu == NULL || tz_names == NULL)
1340 1.1 dholland return; /* error - skip timezone setting */
1341 1.1 dholland while (nfiles > 0)
1342 1.1 dholland free(tz_names[--nfiles]);
1343 1.1 dholland
1344 1.1 dholland dir = opendir(zoneinfo_dir);
1345 1.1 dholland fp = strchr(zoneinfo_dir, 0);
1346 1.1 dholland if (fp != zoneinfo_dir + zonerootlen) {
1347 1.1 dholland tz_names[0] = 0;
1348 1.1 dholland tz_menu[0].opt_name = msg_string(MSG_tz_back);
1349 1.22 martin tz_menu[0].opt_exp_name = NULL;
1350 1.1 dholland tz_menu[0].opt_menu = OPT_NOMENU;
1351 1.1 dholland tz_menu[0].opt_flags = 0;
1352 1.1 dholland tz_menu[0].opt_action = set_tz_back;
1353 1.1 dholland nfiles = 1;
1354 1.1 dholland }
1355 1.1 dholland maxfname = zoneinfo_dir + sizeof zoneinfo_dir - fp - 1;
1356 1.1 dholland if (dir != NULL) {
1357 1.1 dholland while ((dp = readdir(dir)) != NULL) {
1358 1.1 dholland if (dp->d_namlen > maxfname || dp->d_name[0] == '.')
1359 1.1 dholland continue;
1360 1.1 dholland strlcpy(fp, dp->d_name, maxfname);
1361 1.1 dholland if (stat(zoneinfo_dir, &sb) == -1)
1362 1.1 dholland continue;
1363 1.1 dholland if (nfiles >= maxfiles) {
1364 1.1 dholland p = realloc(tz_menu, 2 * maxfiles * sizeof *tz_menu);
1365 1.1 dholland if (p == NULL)
1366 1.1 dholland break;
1367 1.1 dholland tz_menu = p;
1368 1.1 dholland p = realloc(tz_names, 2 * maxfiles * sizeof *tz_names);
1369 1.1 dholland if (p == NULL)
1370 1.1 dholland break;
1371 1.1 dholland tz_names = p;
1372 1.1 dholland maxfiles *= 2;
1373 1.1 dholland }
1374 1.1 dholland if (S_ISREG(sb.st_mode))
1375 1.1 dholland tz_menu[nfiles].opt_action = set_tz_select;
1376 1.1 dholland else if (S_ISDIR(sb.st_mode)) {
1377 1.1 dholland tz_menu[nfiles].opt_action = set_tz_dir;
1378 1.1 dholland strlcat(fp, "/",
1379 1.1 dholland sizeof(zoneinfo_dir) - (fp - zoneinfo_dir));
1380 1.1 dholland } else
1381 1.1 dholland continue;
1382 1.1 dholland tz_names[nfiles] = strdup(zoneinfo_dir + zonerootlen);
1383 1.1 dholland tz_menu[nfiles].opt_name = tz_names[nfiles];
1384 1.22 martin tz_menu[nfiles].opt_exp_name = NULL;
1385 1.1 dholland tz_menu[nfiles].opt_menu = OPT_NOMENU;
1386 1.1 dholland tz_menu[nfiles].opt_flags = 0;
1387 1.1 dholland nfiles++;
1388 1.1 dholland }
1389 1.1 dholland closedir(dir);
1390 1.1 dholland }
1391 1.1 dholland *fp = 0;
1392 1.1 dholland
1393 1.1 dholland m->opts = tz_menu;
1394 1.1 dholland m->numopts = nfiles;
1395 1.1 dholland qsort(tz_menu, nfiles, sizeof *tz_menu, tz_sort);
1396 1.1 dholland }
1397 1.1 dholland
1398 1.1 dholland void
1399 1.1 dholland get_tz_default(void)
1400 1.1 dholland {
1401 1.1 dholland char localtime_link[STRSIZE];
1402 1.1 dholland static char localtime_target[STRSIZE];
1403 1.1 dholland int rc;
1404 1.1 dholland
1405 1.1 dholland strlcpy(localtime_link, target_expand("/etc/localtime"),
1406 1.1 dholland sizeof localtime_link);
1407 1.1 dholland
1408 1.1 dholland /* Add sanity check that /mnt/usr/share/zoneinfo contains
1409 1.1 dholland * something useful
1410 1.1 dholland */
1411 1.1 dholland
1412 1.1 dholland rc = readlink(localtime_link, localtime_target,
1413 1.1 dholland sizeof(localtime_target) - 1);
1414 1.1 dholland if (rc < 0) {
1415 1.1 dholland /* error, default to UTC */
1416 1.1 dholland tz_default = "UTC";
1417 1.1 dholland } else {
1418 1.1 dholland localtime_target[rc] = '\0';
1419 1.1 dholland tz_default = strchr(strstr(localtime_target, "zoneinfo"), '/') + 1;
1420 1.1 dholland }
1421 1.1 dholland }
1422 1.1 dholland
1423 1.1 dholland /*
1424 1.1 dholland * Choose from the files in usr/share/zoneinfo and set etc/localtime
1425 1.1 dholland */
1426 1.1 dholland int
1427 1.1 dholland set_timezone(void)
1428 1.1 dholland {
1429 1.1 dholland char localtime_link[STRSIZE];
1430 1.1 dholland char localtime_target[STRSIZE];
1431 1.1 dholland time_t t;
1432 1.5 christos struct tm *tm;
1433 1.1 dholland int menu_no;
1434 1.1 dholland
1435 1.1 dholland strlcpy(zoneinfo_dir, target_expand("/usr/share/zoneinfo/"),
1436 1.1 dholland sizeof zoneinfo_dir - 1);
1437 1.1 dholland zonerootlen = strlen(zoneinfo_dir);
1438 1.1 dholland
1439 1.1 dholland get_tz_default();
1440 1.1 dholland
1441 1.1 dholland tz_selected = strdup(tz_default);
1442 1.1 dholland snprintf(tz_env, sizeof(tz_env), "%s%s", zoneinfo_dir, tz_selected);
1443 1.1 dholland setenv("TZ", tz_env, 1);
1444 1.1 dholland t = time(NULL);
1445 1.5 christos tm = localtime(&t);
1446 1.1 dholland msg_display(MSG_choose_timezone,
1447 1.5 christos tz_default, tz_selected, safectime(&t), tm ? tm->tm_zone :
1448 1.5 christos "?");
1449 1.1 dholland
1450 1.1 dholland signal(SIGALRM, timezone_sig);
1451 1.1 dholland alarm(60);
1452 1.1 dholland
1453 1.1 dholland menu_no = new_menu(NULL, NULL, 14, 23, 9,
1454 1.1 dholland 12, 32, MC_ALWAYS_SCROLL | MC_NOSHORTCUT,
1455 1.1 dholland tzm_set_names, NULL, NULL,
1456 1.1 dholland "\nPlease consult the install documents.", NULL);
1457 1.1 dholland if (menu_no < 0)
1458 1.1 dholland goto done; /* error - skip timezone setting */
1459 1.1 dholland
1460 1.1 dholland process_menu(menu_no, NULL);
1461 1.1 dholland
1462 1.1 dholland free_menu(menu_no);
1463 1.1 dholland
1464 1.1 dholland signal(SIGALRM, SIG_IGN);
1465 1.1 dholland
1466 1.1 dholland snprintf(localtime_target, sizeof(localtime_target),
1467 1.1 dholland "/usr/share/zoneinfo/%s", tz_selected);
1468 1.1 dholland strlcpy(localtime_link, target_expand("/etc/localtime"),
1469 1.1 dholland sizeof localtime_link);
1470 1.1 dholland unlink(localtime_link);
1471 1.1 dholland symlink(localtime_target, localtime_link);
1472 1.1 dholland
1473 1.1 dholland done:
1474 1.1 dholland return 1;
1475 1.1 dholland }
1476 1.1 dholland
1477 1.1 dholland void
1478 1.1 dholland scripting_vfprintf(FILE *f, const char *fmt, va_list ap)
1479 1.1 dholland {
1480 1.1 dholland
1481 1.1 dholland if (f)
1482 1.1 dholland (void)vfprintf(f, fmt, ap);
1483 1.1 dholland if (script)
1484 1.1 dholland (void)vfprintf(script, fmt, ap);
1485 1.1 dholland }
1486 1.1 dholland
1487 1.1 dholland void
1488 1.1 dholland scripting_fprintf(FILE *f, const char *fmt, ...)
1489 1.1 dholland {
1490 1.1 dholland va_list ap;
1491 1.1 dholland
1492 1.1 dholland va_start(ap, fmt);
1493 1.1 dholland scripting_vfprintf(f, fmt, ap);
1494 1.1 dholland va_end(ap);
1495 1.1 dholland }
1496 1.1 dholland
1497 1.1 dholland void
1498 1.1 dholland add_rc_conf(const char *fmt, ...)
1499 1.1 dholland {
1500 1.1 dholland FILE *f;
1501 1.1 dholland va_list ap;
1502 1.1 dholland
1503 1.1 dholland va_start(ap, fmt);
1504 1.1 dholland f = target_fopen("/etc/rc.conf", "a");
1505 1.1 dholland if (f != 0) {
1506 1.1 dholland scripting_fprintf(NULL, "cat <<EOF >>%s/etc/rc.conf\n",
1507 1.1 dholland target_prefix());
1508 1.1 dholland scripting_vfprintf(f, fmt, ap);
1509 1.1 dholland fclose(f);
1510 1.1 dholland scripting_fprintf(NULL, "EOF\n");
1511 1.1 dholland }
1512 1.1 dholland va_end(ap);
1513 1.1 dholland }
1514 1.1 dholland
1515 1.1 dholland int
1516 1.1 dholland del_rc_conf(const char *value)
1517 1.1 dholland {
1518 1.1 dholland FILE *fp, *nfp;
1519 1.1 dholland char buf[4096]; /* Ridiculously high, but should be enough in any way */
1520 1.1 dholland char *rcconf, *tempname = NULL, *bakname = NULL;
1521 1.1 dholland char *cp;
1522 1.1 dholland int done = 0;
1523 1.1 dholland int fd;
1524 1.1 dholland int retval = 0;
1525 1.1 dholland
1526 1.1 dholland /* The paths might seem strange, but using /tmp would require copy instead
1527 1.1 dholland * of rename operations. */
1528 1.1 dholland if (asprintf(&rcconf, "%s", target_expand("/etc/rc.conf")) < 0
1529 1.1 dholland || asprintf(&tempname, "%s", target_expand("/etc/rc.conf.tmp.XXXXXX")) < 0
1530 1.1 dholland || asprintf(&bakname, "%s", target_expand("/etc/rc.conf.bak.XXXXXX")) < 0) {
1531 1.1 dholland if (rcconf)
1532 1.1 dholland free(rcconf);
1533 1.1 dholland if (tempname)
1534 1.1 dholland free(tempname);
1535 1.1 dholland msg_display(MSG_rcconf_delete_failed, value);
1536 1.22 martin hit_enter_to_continue(NULL, NULL);
1537 1.1 dholland return -1;
1538 1.1 dholland }
1539 1.1 dholland
1540 1.1 dholland if ((fd = mkstemp(bakname)) < 0) {
1541 1.1 dholland msg_display(MSG_rcconf_delete_failed, value);
1542 1.22 martin hit_enter_to_continue(NULL, NULL);
1543 1.1 dholland return -1;
1544 1.1 dholland }
1545 1.1 dholland close(fd);
1546 1.1 dholland
1547 1.1 dholland if (!(fp = fopen(rcconf, "r+")) || (fd = mkstemp(tempname)) < 0) {
1548 1.1 dholland if (fp)
1549 1.1 dholland fclose(fp);
1550 1.1 dholland msg_display(MSG_rcconf_delete_failed, value);
1551 1.22 martin hit_enter_to_continue(NULL, NULL);
1552 1.1 dholland return -1;
1553 1.1 dholland }
1554 1.1 dholland
1555 1.1 dholland nfp = fdopen(fd, "w");
1556 1.1 dholland if (!nfp) {
1557 1.1 dholland fclose(fp);
1558 1.1 dholland close(fd);
1559 1.1 dholland msg_display(MSG_rcconf_delete_failed, value);
1560 1.22 martin hit_enter_to_continue(NULL, NULL);
1561 1.1 dholland return -1;
1562 1.1 dholland }
1563 1.1 dholland
1564 1.1 dholland while (fgets(buf, sizeof buf, fp) != NULL) {
1565 1.1 dholland
1566 1.1 dholland cp = buf + strspn(buf, " \t"); /* Skip initial spaces */
1567 1.1 dholland if (strncmp(cp, value, strlen(value)) == 0) {
1568 1.1 dholland cp += strlen(value);
1569 1.1 dholland if (*cp != '=')
1570 1.1 dholland scripting_fprintf(nfp, "%s", buf);
1571 1.1 dholland else
1572 1.1 dholland done = 1;
1573 1.1 dholland } else {
1574 1.1 dholland scripting_fprintf(nfp, "%s", buf);
1575 1.1 dholland }
1576 1.1 dholland }
1577 1.1 dholland fclose(fp);
1578 1.1 dholland fclose(nfp);
1579 1.1 dholland
1580 1.1 dholland if (done) {
1581 1.1 dholland if (rename(rcconf, bakname)) {
1582 1.1 dholland msg_display(MSG_rcconf_backup_failed);
1583 1.6 martin if (!ask_noyes(NULL)) {
1584 1.1 dholland retval = -1;
1585 1.1 dholland goto done;
1586 1.1 dholland }
1587 1.1 dholland }
1588 1.1 dholland
1589 1.1 dholland if (rename(tempname, rcconf)) {
1590 1.1 dholland if (rename(bakname, rcconf)) {
1591 1.22 martin hit_enter_to_continue(MSG_rcconf_restore_failed,
1592 1.22 martin NULL);
1593 1.1 dholland } else {
1594 1.22 martin hit_enter_to_continue(MSG_rcconf_delete_failed,
1595 1.22 martin NULL);
1596 1.1 dholland }
1597 1.1 dholland } else {
1598 1.1 dholland (void)unlink(bakname);
1599 1.1 dholland }
1600 1.1 dholland }
1601 1.1 dholland
1602 1.1 dholland done:
1603 1.1 dholland (void)unlink(tempname);
1604 1.1 dholland free(rcconf);
1605 1.1 dholland free(tempname);
1606 1.1 dholland free(bakname);
1607 1.1 dholland return retval;
1608 1.1 dholland }
1609 1.1 dholland
1610 1.1 dholland void
1611 1.1 dholland add_sysctl_conf(const char *fmt, ...)
1612 1.1 dholland {
1613 1.1 dholland FILE *f;
1614 1.1 dholland va_list ap;
1615 1.1 dholland
1616 1.1 dholland va_start(ap, fmt);
1617 1.1 dholland f = target_fopen("/etc/sysctl.conf", "a");
1618 1.1 dholland if (f != 0) {
1619 1.1 dholland scripting_fprintf(NULL, "cat <<EOF >>%s/etc/sysctl.conf\n",
1620 1.1 dholland target_prefix());
1621 1.1 dholland scripting_vfprintf(f, fmt, ap);
1622 1.1 dholland fclose(f);
1623 1.1 dholland scripting_fprintf(NULL, "EOF\n");
1624 1.1 dholland }
1625 1.1 dholland va_end(ap);
1626 1.1 dholland }
1627 1.1 dholland
1628 1.1 dholland void
1629 1.1 dholland enable_rc_conf(void)
1630 1.1 dholland {
1631 1.1 dholland
1632 1.1 dholland replace("/etc/rc.conf", "s/^rc_configured=NO/rc_configured=YES/");
1633 1.1 dholland }
1634 1.1 dholland
1635 1.1 dholland int
1636 1.1 dholland check_lfs_progs(void)
1637 1.1 dholland {
1638 1.1 dholland
1639 1.1 dholland #ifndef NO_LFS
1640 1.2 martin return binary_available("fsck_lfs") && binary_available("mount_lfs")
1641 1.2 martin && binary_available("newfs_lfs");
1642 1.1 dholland #else
1643 1.1 dholland return 0;
1644 1.1 dholland #endif
1645 1.1 dholland }
1646 1.1 dholland
1647 1.1 dholland int
1648 1.1 dholland set_is_source(const char *set_name) {
1649 1.1 dholland int len = strlen(set_name);
1650 1.1 dholland return len >= 3 && memcmp(set_name + len - 3, "src", 3) == 0;
1651 1.1 dholland }
1652 1.1 dholland
1653 1.1 dholland const char *
1654 1.1 dholland set_dir_for_set(const char *set_name) {
1655 1.1 dholland if (strcmp(set_name, "pkgsrc") == 0)
1656 1.1 dholland return pkgsrc_dir;
1657 1.1 dholland return set_is_source(set_name) ? set_dir_src : set_dir_bin;
1658 1.1 dholland }
1659 1.1 dholland
1660 1.1 dholland const char *
1661 1.1 dholland ext_dir_for_set(const char *set_name) {
1662 1.1 dholland if (strcmp(set_name, "pkgsrc") == 0)
1663 1.1 dholland return ext_dir_pkgsrc;
1664 1.1 dholland return set_is_source(set_name) ? ext_dir_src : ext_dir_bin;
1665 1.1 dholland }
1666 1.1 dholland
1667 1.2 martin void
1668 1.2 martin do_coloring(unsigned int fg, unsigned int bg) {
1669 1.2 martin if (bg > COLOR_WHITE)
1670 1.2 martin bg = COLOR_BLUE;
1671 1.2 martin if (fg > COLOR_WHITE)
1672 1.2 martin fg = COLOR_WHITE;
1673 1.2 martin if (fg != bg && has_colors()) {
1674 1.2 martin init_pair(1, fg, bg);
1675 1.2 martin wbkgd(stdscr, COLOR_PAIR(1));
1676 1.2 martin wbkgd(mainwin, COLOR_PAIR(1));
1677 1.2 martin wattrset(stdscr, COLOR_PAIR(1));
1678 1.2 martin wattrset(mainwin, COLOR_PAIR(1));
1679 1.2 martin }
1680 1.2 martin /* redraw screen */
1681 1.2 martin touchwin(stdscr);
1682 1.2 martin touchwin(mainwin);
1683 1.2 martin wrefresh(stdscr);
1684 1.2 martin wrefresh(mainwin);
1685 1.2 martin return;
1686 1.2 martin }
1687 1.2 martin
1688 1.2 martin int
1689 1.2 martin set_menu_select(menudesc *m, void *arg)
1690 1.2 martin {
1691 1.2 martin *(int *)arg = m->cursel;
1692 1.2 martin return 1;
1693 1.2 martin }
1694 1.2 martin
1695 1.2 martin /*
1696 1.2 martin * check wether a binary is available somewhere in PATH,
1697 1.2 martin * return 1 if found, 0 if not.
1698 1.2 martin */
1699 1.22 martin static int
1700 1.2 martin binary_available(const char *prog)
1701 1.2 martin {
1702 1.9 kamil char *p, tmp[MAXPATHLEN], *path = getenv("PATH"), *opath;
1703 1.2 martin
1704 1.2 martin if (path == NULL)
1705 1.2 martin return access(prog, X_OK) == 0;
1706 1.2 martin path = strdup(path);
1707 1.2 martin if (path == NULL)
1708 1.2 martin return 0;
1709 1.9 kamil
1710 1.9 kamil opath = path;
1711 1.9 kamil
1712 1.2 martin while ((p = strsep(&path, ":")) != NULL) {
1713 1.2 martin if (strlcpy(tmp, p, MAXPATHLEN) >= MAXPATHLEN)
1714 1.2 martin continue;
1715 1.2 martin if (strlcat(tmp, "/", MAXPATHLEN) >= MAXPATHLEN)
1716 1.2 martin continue;
1717 1.2 martin if (strlcat(tmp, prog, MAXPATHLEN) >= MAXPATHLEN)
1718 1.2 martin continue;
1719 1.2 martin if (access(tmp, X_OK) == 0) {
1720 1.9 kamil free(opath);
1721 1.2 martin return 1;
1722 1.2 martin }
1723 1.2 martin }
1724 1.9 kamil free(opath);
1725 1.2 martin return 0;
1726 1.2 martin }
1727 1.2 martin
1728 1.5 christos const char *
1729 1.5 christos safectime(time_t *t)
1730 1.5 christos {
1731 1.5 christos const char *s = ctime(t);
1732 1.5 christos if (s != NULL)
1733 1.5 christos return s;
1734 1.5 christos
1735 1.5 christos // Debugging.
1736 1.5 christos fprintf(stderr, "Can't convert to localtime 0x%jx (%s)\n",
1737 1.5 christos (intmax_t)*t, strerror(errno));
1738 1.5 christos /*123456789012345678901234*/
1739 1.5 christos return "preposterous clock time\n";
1740 1.5 christos }
1741 1.6 martin
1742 1.6 martin int
1743 1.7 martin ask_yesno(const char* msgtxt)
1744 1.6 martin {
1745 1.6 martin arg_rv p;
1746 1.6 martin
1747 1.8 joerg p.arg = __UNCONST(msgtxt);
1748 1.6 martin p.rv = -1;
1749 1.6 martin
1750 1.6 martin process_menu(MENU_yesno, &p);
1751 1.6 martin return p.rv;
1752 1.6 martin }
1753 1.6 martin
1754 1.6 martin int
1755 1.7 martin ask_noyes(const char *msgtxt)
1756 1.6 martin {
1757 1.6 martin arg_rv p;
1758 1.6 martin
1759 1.8 joerg p.arg = __UNCONST(msgtxt);
1760 1.6 martin p.rv = -1;
1761 1.6 martin
1762 1.6 martin process_menu(MENU_noyes, &p);
1763 1.6 martin return p.rv;
1764 1.6 martin }
1765 1.12 martin
1766 1.22 martin int
1767 1.22 martin ask_reedit(const struct disk_partitions *parts)
1768 1.22 martin {
1769 1.22 martin const char *args[2];
1770 1.22 martin arg_rep_int arg;
1771 1.22 martin
1772 1.22 martin args[0] = msg_string(parts->pscheme->name);
1773 1.22 martin args[1] = msg_string(parts->pscheme->short_name);
1774 1.22 martin arg.args.argv = args;
1775 1.22 martin arg.args.argc = 2;
1776 1.22 martin arg.rv = 0;
1777 1.22 martin process_menu(MENU_reedit, &arg);
1778 1.22 martin
1779 1.22 martin return arg.rv;
1780 1.22 martin }
1781 1.22 martin
1782 1.12 martin bool
1783 1.12 martin use_tgz_for_set(const char *set_name)
1784 1.12 martin {
1785 1.12 martin const struct distinfo *dist;
1786 1.12 martin
1787 1.12 martin for (dist = dist_list; dist->set != SET_LAST; dist++) {
1788 1.12 martin if (dist->name == NULL)
1789 1.12 martin continue;
1790 1.12 martin if (strcmp(set_name, dist->name) == 0)
1791 1.12 martin return dist->force_tgz;
1792 1.12 martin }
1793 1.12 martin
1794 1.13 martin return true;
1795 1.12 martin }
1796 1.12 martin
1797 1.12 martin /* Return the postfix used for a given set */
1798 1.13 martin const char *
1799 1.13 martin set_postfix(const char *set_name)
1800 1.12 martin {
1801 1.12 martin return use_tgz_for_set(set_name) ? dist_tgz_postfix : dist_postfix;
1802 1.12 martin }
1803 1.12 martin
1804 1.17 martin /*
1805 1.18 martin * Replace positional arguments (encoded as $0 .. $N) in the string
1806 1.18 martin * passed by the contents of the passed argument array.
1807 1.18 martin * Caller must free() the result string.
1808 1.17 martin */
1809 1.18 martin char*
1810 1.18 martin str_arg_subst(const char *src, size_t argc, const char **argv)
1811 1.17 martin {
1812 1.18 martin const char *p, *last;
1813 1.17 martin char *out, *t;
1814 1.17 martin size_t len;
1815 1.17 martin
1816 1.17 martin len = strlen(src);
1817 1.17 martin for (p = strchr(src, '$'); p; p = strchr(p+1, '$')) {
1818 1.17 martin char *endp = NULL;
1819 1.17 martin size_t n;
1820 1.17 martin int e;
1821 1.17 martin
1822 1.17 martin /* $ followed by a correct numeric position? */
1823 1.17 martin n = strtou(p+1, &endp, 10, 0, INT_MAX, &e);
1824 1.17 martin if ((e == 0 || e == ENOTSUP) && n < argc) {
1825 1.18 martin len += strlen(argv[n]);
1826 1.17 martin len -= endp-p;
1827 1.17 martin p = endp-1;
1828 1.17 martin }
1829 1.17 martin }
1830 1.17 martin
1831 1.17 martin out = malloc(len+1);
1832 1.18 martin if (out == NULL)
1833 1.18 martin return NULL;
1834 1.17 martin
1835 1.17 martin t = out;
1836 1.17 martin for (last = src, p = strchr(src, '$'); p; p = strchr(p+1, '$')) {
1837 1.17 martin char *endp = NULL;
1838 1.17 martin size_t n;
1839 1.17 martin int e;
1840 1.17 martin
1841 1.17 martin /* $ followed by a correct numeric position? */
1842 1.17 martin n = strtou(p+1, &endp, 10, 0, INT_MAX, &e);
1843 1.17 martin if ((e == 0 || e == ENOTSUP) && n < argc) {
1844 1.17 martin size_t l = p-last;
1845 1.17 martin memcpy(t, last, l);
1846 1.17 martin t += l;
1847 1.18 martin strcpy(t, argv[n]);
1848 1.18 martin t += strlen(argv[n]);
1849 1.17 martin last = endp;
1850 1.17 martin }
1851 1.17 martin }
1852 1.17 martin if (*last) {
1853 1.17 martin strcpy(t, last);
1854 1.17 martin t += strlen(last);
1855 1.17 martin } else {
1856 1.17 martin *t = 0;
1857 1.17 martin }
1858 1.17 martin assert((size_t)(t-out) == len);
1859 1.17 martin
1860 1.18 martin return out;
1861 1.18 martin }
1862 1.18 martin
1863 1.18 martin /*
1864 1.22 martin * Does this string have any positional args that need expanding?
1865 1.22 martin */
1866 1.22 martin bool
1867 1.22 martin needs_expanding(const char *src, size_t argc)
1868 1.22 martin {
1869 1.22 martin const char *p;
1870 1.22 martin
1871 1.22 martin for (p = strchr(src, '$'); p; p = strchr(p+1, '$')) {
1872 1.22 martin char *endp = NULL;
1873 1.22 martin size_t n;
1874 1.22 martin int e;
1875 1.22 martin
1876 1.22 martin /* $ followed by a correct numeric position? */
1877 1.22 martin n = strtou(p+1, &endp, 10, 0, INT_MAX, &e);
1878 1.22 martin if ((e == 0 || e == ENOTSUP) && n < argc)
1879 1.22 martin return true;
1880 1.22 martin }
1881 1.22 martin return false;
1882 1.22 martin }
1883 1.22 martin
1884 1.22 martin /*
1885 1.18 martin * Replace positional arguments (encoded as $0 .. $N) in the
1886 1.22 martin * message by the strings passed as ... and call outfunc
1887 1.22 martin * with the result.
1888 1.18 martin */
1889 1.22 martin static void
1890 1.22 martin msg_display_subst_internal(void (*outfunc)(msg,...), const char *master,
1891 1.22 martin size_t argc, va_list ap)
1892 1.18 martin {
1893 1.18 martin const char **args, **arg;
1894 1.18 martin char *out;
1895 1.18 martin
1896 1.18 martin args = malloc(sizeof(const char *)*argc);
1897 1.18 martin if (args == NULL)
1898 1.18 martin return;
1899 1.18 martin
1900 1.18 martin arg = args;
1901 1.18 martin for (size_t i = 0; i < argc; i++)
1902 1.18 martin *arg++ = va_arg(ap, const char*);
1903 1.17 martin
1904 1.18 martin out = str_arg_subst(msg_string(master), argc, args);
1905 1.18 martin if (out != NULL) {
1906 1.22 martin outfunc(out);
1907 1.18 martin free(out);
1908 1.18 martin }
1909 1.17 martin free(args);
1910 1.17 martin }
1911 1.17 martin
1912 1.22 martin /*
1913 1.22 martin * Replace positional arguments (encoded as $0 .. $N) in the
1914 1.22 martin * message by the strings passed as ...
1915 1.22 martin */
1916 1.22 martin void
1917 1.22 martin msg_display_subst(const char *master, size_t argc, ...)
1918 1.22 martin {
1919 1.22 martin va_list ap;
1920 1.22 martin
1921 1.22 martin va_start(ap, argc);
1922 1.22 martin msg_display_subst_internal(msg_display, master, argc, ap);
1923 1.22 martin va_end(ap);
1924 1.22 martin }
1925 1.22 martin
1926 1.22 martin /*
1927 1.22 martin * Same as above, but add to message instead of starting a new one
1928 1.22 martin */
1929 1.22 martin void
1930 1.22 martin msg_display_add_subst(const char *master, size_t argc, ...)
1931 1.22 martin {
1932 1.22 martin va_list ap;
1933 1.22 martin
1934 1.22 martin va_start(ap, argc);
1935 1.22 martin msg_display_subst_internal(msg_display_add, master, argc, ap);
1936 1.22 martin va_end(ap);
1937 1.22 martin }
1938 1.22 martin
1939 1.22 martin /* initialize have_* variables */
1940 1.22 martin void
1941 1.22 martin check_available_binaries()
1942 1.22 martin {
1943 1.22 martin static int did_test = false;
1944 1.22 martin
1945 1.22 martin if (did_test) return;
1946 1.22 martin did_test = 1;
1947 1.22 martin
1948 1.22 martin have_raid = binary_available("raidctl");
1949 1.22 martin have_vnd = binary_available("vndconfig");
1950 1.22 martin have_cgd = binary_available("cgdconfig");
1951 1.22 martin have_lvm = binary_available("lvm");
1952 1.22 martin have_gpt = binary_available("gpt");
1953 1.22 martin have_dk = binary_available("dkctl");
1954 1.22 martin }
1955 1.22 martin
1956 1.22 martin /*
1957 1.22 martin * Wait for enter and immediately clear the screen after user response
1958 1.22 martin * (in case some longer action follows, so the user has instant feedback)
1959 1.22 martin */
1960 1.22 martin void
1961 1.22 martin hit_enter_to_continue(const char *prompt, const char *title)
1962 1.22 martin {
1963 1.22 martin if (prompt != NULL)
1964 1.22 martin msg_display(prompt);
1965 1.22 martin process_menu(MENU_ok, __UNCONST(title));
1966 1.22 martin msg_clear();
1967 1.22 martin wrefresh(mainwin);
1968 1.22 martin }
1969 1.22 martin
1970 1.22 martin /*
1971 1.22 martin * On auto pilot:
1972 1.22 martin * convert an existing set of partitions ot a list of part_usage_info
1973 1.22 martin * so that we "want" exactly what is there already.
1974 1.22 martin */
1975 1.22 martin static bool
1976 1.22 martin usage_info_list_from_parts(struct part_usage_info **list, size_t *count,
1977 1.22 martin struct disk_partitions *parts)
1978 1.22 martin {
1979 1.22 martin struct disk_part_info info;
1980 1.22 martin part_id pno;
1981 1.22 martin size_t no, num;
1982 1.22 martin
1983 1.22 martin num = 0;
1984 1.22 martin for (pno = 0; pno < parts->num_part; pno++) {
1985 1.22 martin if (!parts->pscheme->get_part_info(parts, pno, &info))
1986 1.22 martin continue;
1987 1.22 martin num++;
1988 1.22 martin }
1989 1.22 martin
1990 1.22 martin *list = calloc(num, sizeof(**list));
1991 1.22 martin if (*list == NULL)
1992 1.22 martin return false;
1993 1.22 martin
1994 1.22 martin *count = num;
1995 1.22 martin for (no = pno = 0; pno < parts->num_part && no < num; pno++) {
1996 1.22 martin if (!parts->pscheme->get_part_info(parts, pno, &info))
1997 1.22 martin continue;
1998 1.22 martin (*list)[no].size = info.size;
1999 1.22 martin if (info.last_mounted != NULL && *info.last_mounted != 0) {
2000 1.22 martin strlcpy((*list)[no].mount, info.last_mounted,
2001 1.22 martin sizeof((*list)[no].mount));
2002 1.22 martin (*list)[no].instflags |= PUIINST_MOUNT;
2003 1.22 martin }
2004 1.22 martin (*list)[no].parts = parts;
2005 1.22 martin (*list)[no].cur_part_id = pno;
2006 1.22 martin (*list)[no].cur_start = info.start;
2007 1.22 martin (*list)[no].cur_flags = info.flags;
2008 1.22 martin (*list)[no].type = info.nat_type->generic_ptype;
2009 1.22 martin if ((*list)[no].type == PT_swap) {
2010 1.22 martin (*list)[no].fs_type = FS_SWAP;
2011 1.22 martin (*list)[no].fs_version = 0;
2012 1.22 martin } else {
2013 1.22 martin (*list)[no].fs_type = info.fs_type;
2014 1.22 martin (*list)[no].fs_version = info.fs_sub_type;
2015 1.22 martin }
2016 1.22 martin no++;
2017 1.22 martin }
2018 1.22 martin return true;
2019 1.22 martin }
2020 1.22 martin
2021 1.22 martin bool
2022 1.22 martin usage_set_from_parts(struct partition_usage_set *wanted,
2023 1.22 martin struct disk_partitions *parts)
2024 1.22 martin {
2025 1.22 martin memset(wanted, 0, sizeof(*wanted));
2026 1.22 martin wanted->parts = parts;
2027 1.22 martin
2028 1.22 martin return usage_info_list_from_parts(&wanted->infos, &wanted->num, parts);
2029 1.22 martin }
2030 1.22 martin
2031 1.22 martin bool
2032 1.22 martin install_desc_from_parts(struct install_partition_desc *install,
2033 1.22 martin struct disk_partitions *parts)
2034 1.22 martin {
2035 1.22 martin memset(install, 0, sizeof(*install));
2036 1.22 martin
2037 1.22 martin return usage_info_list_from_parts(&install->infos, &install->num,
2038 1.22 martin parts);
2039 1.22 martin }
2040 1.22 martin
2041 1.22 martin void
2042 1.22 martin free_usage_set(struct partition_usage_set *wanted)
2043 1.22 martin {
2044 1.22 martin free(wanted->menu_opts);
2045 1.22 martin free(wanted->infos);
2046 1.22 martin }
2047 1.22 martin
2048 1.22 martin void
2049 1.22 martin free_install_desc(struct install_partition_desc *install)
2050 1.22 martin {
2051 1.22 martin free(install->infos);
2052 1.22 martin }
2053 1.22 martin
2054