label.c revision 1.2.6.2 1 1.2.6.2 tls /* $NetBSD: label.c,v 1.2.6.2 2014/08/20 00:05:13 tls Exp $ */
2 1.2.6.2 tls
3 1.2.6.2 tls /*
4 1.2.6.2 tls * Copyright 1997 Jonathan Stone
5 1.2.6.2 tls * All rights reserved.
6 1.2.6.2 tls *
7 1.2.6.2 tls * Redistribution and use in source and binary forms, with or without
8 1.2.6.2 tls * modification, are permitted provided that the following conditions
9 1.2.6.2 tls * are met:
10 1.2.6.2 tls * 1. Redistributions of source code must retain the above copyright
11 1.2.6.2 tls * notice, this list of conditions and the following disclaimer.
12 1.2.6.2 tls * 2. Redistributions in binary form must reproduce the above copyright
13 1.2.6.2 tls * notice, this list of conditions and the following disclaimer in the
14 1.2.6.2 tls * documentation and/or other materials provided with the distribution.
15 1.2.6.2 tls * 3. All advertising materials mentioning features or use of this software
16 1.2.6.2 tls * must display the following acknowledgement:
17 1.2.6.2 tls * This product includes software developed for the NetBSD Project by
18 1.2.6.2 tls * Jonathan Stone.
19 1.2.6.2 tls * 4. The name of Jonathan Stone may not be used to endorse
20 1.2.6.2 tls * or promote products derived from this software without specific prior
21 1.2.6.2 tls * written permission.
22 1.2.6.2 tls *
23 1.2.6.2 tls * THIS SOFTWARE IS PROVIDED BY JONATHAN STONE ``AS IS''
24 1.2.6.2 tls * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
25 1.2.6.2 tls * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
26 1.2.6.2 tls * ARE DISCLAIMED. IN NO EVENT SHALL PIERMONT INFORMATION SYSTEMS INC. BE
27 1.2.6.2 tls * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
28 1.2.6.2 tls * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
29 1.2.6.2 tls * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
30 1.2.6.2 tls * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
31 1.2.6.2 tls * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
32 1.2.6.2 tls * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
33 1.2.6.2 tls * THE POSSIBILITY OF SUCH DAMAGE.
34 1.2.6.2 tls *
35 1.2.6.2 tls */
36 1.2.6.2 tls
37 1.2.6.2 tls #include <sys/cdefs.h>
38 1.2.6.2 tls #if defined(LIBC_SCCS) && !defined(lint)
39 1.2.6.2 tls __RCSID("$NetBSD: label.c,v 1.2.6.2 2014/08/20 00:05:13 tls Exp $");
40 1.2.6.2 tls #endif
41 1.2.6.2 tls
42 1.2.6.2 tls #include <sys/types.h>
43 1.2.6.2 tls #include <stddef.h>
44 1.2.6.2 tls #include <errno.h>
45 1.2.6.2 tls #include <stdio.h>
46 1.2.6.2 tls #include <fcntl.h>
47 1.2.6.2 tls #include <util.h>
48 1.2.6.2 tls #include <unistd.h>
49 1.2.6.2 tls #include <sys/dkio.h>
50 1.2.6.2 tls #include <sys/param.h>
51 1.2.6.2 tls #include <ufs/ffs/fs.h>
52 1.2.6.2 tls
53 1.2.6.2 tls #include "defs.h"
54 1.2.6.2 tls #include "msg_defs.h"
55 1.2.6.2 tls #include "menu_defs.h"
56 1.2.6.2 tls
57 1.2.6.2 tls struct ptn_menu_info {
58 1.2.6.2 tls int flags;
59 1.2.6.2 tls #define PIF_SHOW_UNUSED 1
60 1.2.6.2 tls };
61 1.2.6.2 tls
62 1.2.6.2 tls /*
63 1.2.6.2 tls * local prototypes
64 1.2.6.2 tls */
65 1.2.6.2 tls static int boringpart(partinfo *, int, int, int);
66 1.2.6.2 tls static uint32_t getpartoff(uint32_t);
67 1.2.6.2 tls static uint32_t getpartsize(uint32_t, uint32_t);
68 1.2.6.2 tls
69 1.2.6.2 tls static int checklabel(partinfo *, int, int, int, int *, int *);
70 1.2.6.2 tls static int atofsb(const char *, uint32_t *, uint32_t *);
71 1.2.6.2 tls
72 1.2.6.2 tls
73 1.2.6.2 tls /*
74 1.2.6.2 tls * Return 1 if partition i in lp should be ignored when checking
75 1.2.6.2 tls * for overlapping partitions.
76 1.2.6.2 tls */
77 1.2.6.2 tls static int
78 1.2.6.2 tls boringpart(partinfo *lp, int i, int rawpart, int bsdpart)
79 1.2.6.2 tls {
80 1.2.6.2 tls
81 1.2.6.2 tls if (i == rawpart || i == bsdpart ||
82 1.2.6.2 tls lp[i].pi_fstype == FS_UNUSED || lp[i].pi_size == 0)
83 1.2.6.2 tls return 1;
84 1.2.6.2 tls return 0;
85 1.2.6.2 tls }
86 1.2.6.2 tls
87 1.2.6.2 tls
88 1.2.6.2 tls
89 1.2.6.2 tls /*
90 1.2.6.2 tls * Check a sysinst label structure for overlapping partitions.
91 1.2.6.2 tls * Returns 0 if no overlapping partition found, nonzero otherwise.
92 1.2.6.2 tls * Sets reference arguments ovly1 and ovly2 to the indices of
93 1.2.6.2 tls * overlapping partitions if any are found.
94 1.2.6.2 tls */
95 1.2.6.2 tls static int
96 1.2.6.2 tls checklabel(partinfo *lp, int nparts, int rawpart, int bsdpart,
97 1.2.6.2 tls int *ovly1, int *ovly2)
98 1.2.6.2 tls {
99 1.2.6.2 tls int i;
100 1.2.6.2 tls int j;
101 1.2.6.2 tls
102 1.2.6.2 tls *ovly1 = -1;
103 1.2.6.2 tls *ovly2 = -1;
104 1.2.6.2 tls
105 1.2.6.2 tls for (i = 0; i < nparts - 1; i ++ ) {
106 1.2.6.2 tls partinfo *ip = &lp[i];
107 1.2.6.2 tls uint32_t istart, istop;
108 1.2.6.2 tls
109 1.2.6.2 tls /* skip unused or reserved partitions */
110 1.2.6.2 tls if (boringpart(lp, i, rawpart, bsdpart))
111 1.2.6.2 tls continue;
112 1.2.6.2 tls
113 1.2.6.2 tls /*
114 1.2.6.2 tls * check succeeding partitions for overlap.
115 1.2.6.2 tls * O(n^2), but n is small (currently <= 16).
116 1.2.6.2 tls */
117 1.2.6.2 tls istart = ip->pi_offset;
118 1.2.6.2 tls istop = istart + ip->pi_size;
119 1.2.6.2 tls
120 1.2.6.2 tls for (j = i+1; j < nparts; j++) {
121 1.2.6.2 tls partinfo *jp = &lp[j];
122 1.2.6.2 tls uint32_t jstart, jstop;
123 1.2.6.2 tls
124 1.2.6.2 tls /* skip unused or reserved partitions */
125 1.2.6.2 tls if (boringpart(lp, j, rawpart, bsdpart))
126 1.2.6.2 tls continue;
127 1.2.6.2 tls
128 1.2.6.2 tls jstart = jp->pi_offset;
129 1.2.6.2 tls jstop = jstart + jp->pi_size;
130 1.2.6.2 tls
131 1.2.6.2 tls /* overlap? */
132 1.2.6.2 tls if ((istart <= jstart && jstart < istop) ||
133 1.2.6.2 tls (jstart <= istart && istart < jstop)) {
134 1.2.6.2 tls *ovly1 = i;
135 1.2.6.2 tls *ovly2 = j;
136 1.2.6.2 tls return (1);
137 1.2.6.2 tls }
138 1.2.6.2 tls }
139 1.2.6.2 tls }
140 1.2.6.2 tls
141 1.2.6.2 tls return (0);
142 1.2.6.2 tls }
143 1.2.6.2 tls
144 1.2.6.2 tls int
145 1.2.6.2 tls checkoverlap(partinfo *lp, int nparts, int rawpart, int bsdpart)
146 1.2.6.2 tls {
147 1.2.6.2 tls int i, j;
148 1.2.6.2 tls if (checklabel(lp, nparts, rawpart, bsdpart, &i, &j)) {
149 1.2.6.2 tls msg_display(MSG_partitions_overlap,'a'+i,'a'+j);
150 1.2.6.2 tls return 1;
151 1.2.6.2 tls }
152 1.2.6.2 tls return 0;
153 1.2.6.2 tls }
154 1.2.6.2 tls
155 1.2.6.2 tls static int
156 1.2.6.2 tls check_one_root(partinfo *lp, int nparts)
157 1.2.6.2 tls {
158 1.2.6.2 tls int part;
159 1.2.6.2 tls int foundroot = 0;
160 1.2.6.2 tls
161 1.2.6.2 tls for (part = 0; part < nparts; lp++, part++) {
162 1.2.6.2 tls #if 0
163 1.2.6.2 tls if (!PI_ISBSDFS(lp))
164 1.2.6.2 tls continue;
165 1.2.6.2 tls #endif
166 1.2.6.2 tls if (!(lp->pi_flags & PIF_MOUNT))
167 1.2.6.2 tls continue;
168 1.2.6.2 tls if (strcmp(lp->pi_mount, "/") != 0)
169 1.2.6.2 tls continue;
170 1.2.6.2 tls if (foundroot)
171 1.2.6.2 tls /* Duplicate */
172 1.2.6.2 tls return 0;
173 1.2.6.2 tls foundroot = 1;
174 1.2.6.2 tls /* Save partition number, a few things need to know it */
175 1.2.6.2 tls pm->rootpart = part;
176 1.2.6.2 tls }
177 1.2.6.2 tls return foundroot;
178 1.2.6.2 tls }
179 1.2.6.2 tls
180 1.2.6.2 tls static int
181 1.2.6.2 tls edit_fs_start(menudesc *m, void *arg)
182 1.2.6.2 tls {
183 1.2.6.2 tls partinfo *p = arg;
184 1.2.6.2 tls uint32_t start, end;
185 1.2.6.2 tls
186 1.2.6.2 tls start = getpartoff(p->pi_offset);
187 1.2.6.2 tls if (p->pi_size != 0) {
188 1.2.6.2 tls /* Try to keep end in the same place */
189 1.2.6.2 tls end = p->pi_offset + p->pi_size;
190 1.2.6.2 tls if (end < start)
191 1.2.6.2 tls p->pi_size = 0;
192 1.2.6.2 tls else
193 1.2.6.2 tls p->pi_size = end - start;
194 1.2.6.2 tls }
195 1.2.6.2 tls p->pi_offset = start;
196 1.2.6.2 tls return 0;
197 1.2.6.2 tls }
198 1.2.6.2 tls
199 1.2.6.2 tls static int
200 1.2.6.2 tls edit_fs_size(menudesc *m, void *arg)
201 1.2.6.2 tls {
202 1.2.6.2 tls partinfo *p = arg;
203 1.2.6.2 tls uint32_t size;
204 1.2.6.2 tls
205 1.2.6.2 tls size = getpartsize(p->pi_offset, p->pi_size);
206 1.2.6.2 tls if (size == ~0u)
207 1.2.6.2 tls size = pm->dlsize - p->pi_offset;
208 1.2.6.2 tls p->pi_size = size;
209 1.2.6.2 tls if (size == 0) {
210 1.2.6.2 tls p->pi_offset = 0;
211 1.2.6.2 tls p->pi_fstype = FS_UNUSED;
212 1.2.6.2 tls }
213 1.2.6.2 tls return 0;
214 1.2.6.2 tls }
215 1.2.6.2 tls
216 1.2.6.2 tls void
217 1.2.6.2 tls set_ptype(partinfo *p, int fstype, int flag)
218 1.2.6.2 tls {
219 1.2.6.2 tls p->pi_flags = (p->pi_flags & ~PIF_FFSv2) | flag;
220 1.2.6.2 tls
221 1.2.6.2 tls if (p->pi_fstype == fstype)
222 1.2.6.2 tls return;
223 1.2.6.2 tls
224 1.2.6.2 tls p->pi_fstype = fstype;
225 1.2.6.2 tls if (fstype == FS_BSDFFS || fstype == FS_BSDLFS) {
226 1.2.6.2 tls p->pi_frag = 8;
227 1.2.6.2 tls /*
228 1.2.6.2 tls * match newfs defaults for fragments size:
229 1.2.6.2 tls * fs size frag size
230 1.2.6.2 tls * < 20 MB 0.5 KB
231 1.2.6.2 tls * < 1000 MB 1 KB
232 1.2.6.2 tls * < 128 GB 2 KB
233 1.2.6.2 tls * >= 128 GB 4 KB
234 1.2.6.2 tls */
235 1.2.6.2 tls /* note pi_size is uint32_t so we have to avoid overflow */
236 1.2.6.2 tls if (p->pi_size < (20 * 1024 * (1024 / 512)))
237 1.2.6.2 tls p->pi_fsize = 512;
238 1.2.6.2 tls else if (p->pi_size < (1000 * 1024 * (1024 / 512)))
239 1.2.6.2 tls p->pi_fsize = 1024;
240 1.2.6.2 tls else if (p->pi_size < (128 * 1024 * 1024 * (1024 / 512)))
241 1.2.6.2 tls p->pi_fsize = 2048;
242 1.2.6.2 tls else
243 1.2.6.2 tls p->pi_fsize = 4096;
244 1.2.6.2 tls } else {
245 1.2.6.2 tls /* zero - fields not used */
246 1.2.6.2 tls p->pi_frag = 0;
247 1.2.6.2 tls p->pi_fsize = 0;
248 1.2.6.2 tls }
249 1.2.6.2 tls }
250 1.2.6.2 tls
251 1.2.6.2 tls void
252 1.2.6.2 tls set_bsize(partinfo *p, int size)
253 1.2.6.2 tls {
254 1.2.6.2 tls int frags, sz;
255 1.2.6.2 tls
256 1.2.6.2 tls sz = p->pi_fsize;
257 1.2.6.2 tls if (sz <= 0)
258 1.2.6.2 tls sz = 512;
259 1.2.6.2 tls frags = size / sz;
260 1.2.6.2 tls if (frags > 8)
261 1.2.6.2 tls frags = 8;
262 1.2.6.2 tls if (frags <= 0)
263 1.2.6.2 tls frags = 1;
264 1.2.6.2 tls
265 1.2.6.2 tls p->pi_frag = frags;
266 1.2.6.2 tls p->pi_fsize = size / frags;
267 1.2.6.2 tls }
268 1.2.6.2 tls
269 1.2.6.2 tls void
270 1.2.6.2 tls set_fsize(partinfo *p, int fsize)
271 1.2.6.2 tls {
272 1.2.6.2 tls int bsz = p->pi_fsize * p->pi_frag;
273 1.2.6.2 tls int frags = bsz / fsize;
274 1.2.6.2 tls
275 1.2.6.2 tls if (frags > 8)
276 1.2.6.2 tls frags = 8;
277 1.2.6.2 tls if (frags <= 0)
278 1.2.6.2 tls frags = 1;
279 1.2.6.2 tls
280 1.2.6.2 tls p->pi_fsize = fsize;
281 1.2.6.2 tls p->pi_frag = frags;
282 1.2.6.2 tls }
283 1.2.6.2 tls
284 1.2.6.2 tls static int
285 1.2.6.2 tls edit_fs_isize(menudesc *m, void *arg)
286 1.2.6.2 tls {
287 1.2.6.2 tls partinfo *p = arg;
288 1.2.6.2 tls char answer[12];
289 1.2.6.2 tls
290 1.2.6.2 tls snprintf(answer, sizeof answer, "%u", p->pi_isize);
291 1.2.6.2 tls msg_prompt_win(MSG_fs_isize, -1, 18, 0, 0,
292 1.2.6.2 tls answer, answer, sizeof answer);
293 1.2.6.2 tls p->pi_isize = atol(answer);
294 1.2.6.2 tls return 0;
295 1.2.6.2 tls }
296 1.2.6.2 tls
297 1.2.6.2 tls
298 1.2.6.2 tls static int
299 1.2.6.2 tls edit_fs_preserve(menudesc *m, void *arg)
300 1.2.6.2 tls {
301 1.2.6.2 tls partinfo *p = arg;
302 1.2.6.2 tls
303 1.2.6.2 tls p->pi_flags ^= PIF_NEWFS;
304 1.2.6.2 tls return 0;
305 1.2.6.2 tls }
306 1.2.6.2 tls
307 1.2.6.2 tls static int
308 1.2.6.2 tls edit_fs_mount(menudesc *m, void *arg)
309 1.2.6.2 tls {
310 1.2.6.2 tls partinfo *p = arg;
311 1.2.6.2 tls
312 1.2.6.2 tls p->pi_flags ^= PIF_MOUNT;
313 1.2.6.2 tls return 0;
314 1.2.6.2 tls }
315 1.2.6.2 tls
316 1.2.6.2 tls static int
317 1.2.6.2 tls edit_fs_mountpt(menudesc *m, void *arg)
318 1.2.6.2 tls {
319 1.2.6.2 tls partinfo *p = arg;
320 1.2.6.2 tls
321 1.2.6.2 tls msg_prompt_win(MSG_mountpoint, -1, 18, 0, 0,
322 1.2.6.2 tls p->pi_mount, p->pi_mount, sizeof p->pi_mount);
323 1.2.6.2 tls
324 1.2.6.2 tls if (p->pi_mount[0] == ' ' || strcmp(p->pi_mount, "none") == 0) {
325 1.2.6.2 tls p->pi_mount[0] = 0;
326 1.2.6.2 tls return 0;
327 1.2.6.2 tls }
328 1.2.6.2 tls
329 1.2.6.2 tls if (p->pi_mount[0] != '/') {
330 1.2.6.2 tls memmove(p->pi_mount + 1, p->pi_mount,
331 1.2.6.2 tls sizeof p->pi_mount - 1);
332 1.2.6.2 tls p->pi_mount[sizeof p->pi_mount - 1] = 0;
333 1.2.6.2 tls p->pi_mount[0] = '/';
334 1.2.6.2 tls }
335 1.2.6.2 tls
336 1.2.6.2 tls return 0;
337 1.2.6.2 tls }
338 1.2.6.2 tls
339 1.2.6.2 tls static int
340 1.2.6.2 tls edit_restore(menudesc *m, void *arg)
341 1.2.6.2 tls {
342 1.2.6.2 tls partinfo *p = arg;
343 1.2.6.2 tls
344 1.2.6.2 tls p->pi_flags |= PIF_RESET;
345 1.2.6.2 tls return 1;
346 1.2.6.2 tls }
347 1.2.6.2 tls
348 1.2.6.2 tls static int
349 1.2.6.2 tls set_fstype(menudesc *m, void *arg)
350 1.2.6.2 tls {
351 1.2.6.2 tls partinfo *p = arg;
352 1.2.6.2 tls
353 1.2.6.2 tls if (m->cursel == FS_UNUSED)
354 1.2.6.2 tls memset(p, 0, sizeof *p);
355 1.2.6.2 tls else
356 1.2.6.2 tls p->pi_fstype = m->cursel;
357 1.2.6.2 tls return 1;
358 1.2.6.2 tls }
359 1.2.6.2 tls
360 1.2.6.2 tls static void
361 1.2.6.2 tls get_fstype(menudesc *m, void *arg)
362 1.2.6.2 tls {
363 1.2.6.2 tls partinfo *p = arg;
364 1.2.6.2 tls
365 1.2.6.2 tls m->cursel = p->pi_fstype;
366 1.2.6.2 tls }
367 1.2.6.2 tls
368 1.2.6.2 tls static void set_ptn_header(menudesc *m, void *arg);
369 1.2.6.2 tls static void set_ptn_label(menudesc *m, int opt, void *arg);
370 1.2.6.2 tls int all_fstype_menu = -1;
371 1.2.6.2 tls
372 1.2.6.2 tls int
373 1.2.6.2 tls edit_ptn(menudesc *menu, void *arg)
374 1.2.6.2 tls {
375 1.2.6.2 tls static menu_ent fs_fields[] = {
376 1.2.6.2 tls #define PTN_MENU_FSKIND 0
377 1.2.6.2 tls {NULL, MENU_selfskind, OPT_SUB, NULL},
378 1.2.6.2 tls #define PTN_MENU_START 1
379 1.2.6.2 tls {NULL, OPT_NOMENU, 0, edit_fs_start},
380 1.2.6.2 tls #define PTN_MENU_SIZE 2
381 1.2.6.2 tls {NULL, OPT_NOMENU, 0, edit_fs_size},
382 1.2.6.2 tls #define PTN_MENU_END 3
383 1.2.6.2 tls {NULL, OPT_NOMENU, OPT_IGNORE, NULL}, /* displays 'end' */
384 1.2.6.2 tls #define PTN_MENU_NEWFS 4
385 1.2.6.2 tls {NULL, OPT_NOMENU, 0, edit_fs_preserve},
386 1.2.6.2 tls #define PTN_MENU_ISIZE 5
387 1.2.6.2 tls {NULL, OPT_NOMENU, 0, edit_fs_isize},
388 1.2.6.2 tls #define PTN_MENU_BSIZE 6
389 1.2.6.2 tls {NULL, MENU_selbsize, OPT_SUB, NULL},
390 1.2.6.2 tls #define PTN_MENU_FSIZE 7
391 1.2.6.2 tls {NULL, MENU_selfsize, OPT_SUB, NULL},
392 1.2.6.2 tls #define PTN_MENU_MOUNT 8
393 1.2.6.2 tls {NULL, OPT_NOMENU, 0, edit_fs_mount},
394 1.2.6.2 tls #define PTN_MENU_MOUNTOPT 9
395 1.2.6.2 tls {NULL, MENU_mountoptions, OPT_SUB, NULL},
396 1.2.6.2 tls #define PTN_MENU_MOUNTPT 10
397 1.2.6.2 tls {NULL, OPT_NOMENU, 0, edit_fs_mountpt},
398 1.2.6.2 tls {MSG_askunits, MENU_sizechoice, OPT_SUB, NULL},
399 1.2.6.2 tls {MSG_restore, OPT_NOMENU, 0, edit_restore},
400 1.2.6.2 tls };
401 1.2.6.2 tls static int fspart_menu = -1;
402 1.2.6.2 tls static menu_ent all_fstypes[FSMAXTYPES];
403 1.2.6.2 tls partinfo *p, p_save;
404 1.2.6.2 tls unsigned int i;
405 1.2.6.2 tls
406 1.2.6.2 tls if (fspart_menu == -1) {
407 1.2.6.2 tls fspart_menu = new_menu(NULL, fs_fields, nelem(fs_fields),
408 1.2.6.2 tls 0, -1, 0, 70,
409 1.2.6.2 tls MC_NOBOX | MC_NOCLEAR | MC_SCROLL,
410 1.2.6.2 tls set_ptn_header, set_ptn_label, NULL,
411 1.2.6.2 tls NULL, MSG_partition_sizes_ok);
412 1.2.6.2 tls }
413 1.2.6.2 tls
414 1.2.6.2 tls if (all_fstype_menu == -1) {
415 1.2.6.2 tls for (i = 0; i < nelem(all_fstypes); i++) {
416 1.2.6.2 tls all_fstypes[i].opt_name = getfslabelname(i);
417 1.2.6.2 tls all_fstypes[i].opt_menu = OPT_NOMENU;
418 1.2.6.2 tls all_fstypes[i].opt_flags = 0;
419 1.2.6.2 tls all_fstypes[i].opt_action = set_fstype;
420 1.2.6.2 tls }
421 1.2.6.2 tls all_fstype_menu = new_menu(MSG_Select_the_type,
422 1.2.6.2 tls all_fstypes, nelem(all_fstypes),
423 1.2.6.2 tls 30, 6, 10, 0, MC_SUBMENU | MC_SCROLL,
424 1.2.6.2 tls get_fstype, NULL, NULL, NULL, MSG_unchanged);
425 1.2.6.2 tls }
426 1.2.6.2 tls
427 1.2.6.2 tls p = pm->bsdlabel + menu->cursel;
428 1.2.6.2 tls p->pi_flags &= ~PIF_RESET;
429 1.2.6.2 tls p_save = *p;
430 1.2.6.2 tls for (;;) {
431 1.2.6.2 tls process_menu(fspart_menu, p);
432 1.2.6.2 tls if (!(p->pi_flags & PIF_RESET))
433 1.2.6.2 tls break;
434 1.2.6.2 tls *p = p_save;
435 1.2.6.2 tls }
436 1.2.6.2 tls
437 1.2.6.2 tls return 0;
438 1.2.6.2 tls }
439 1.2.6.2 tls
440 1.2.6.2 tls static void
441 1.2.6.2 tls set_ptn_header(menudesc *m, void *arg)
442 1.2.6.2 tls {
443 1.2.6.2 tls partinfo *p = arg;
444 1.2.6.2 tls int i;
445 1.2.6.2 tls int t;
446 1.2.6.2 tls
447 1.2.6.2 tls msg_clear();
448 1.2.6.2 tls msg_table_add(MSG_edfspart, 'a' + (p - pm->bsdlabel));
449 1.2.6.2 tls
450 1.2.6.2 tls /* Determine which of the properties can be changed */
451 1.2.6.2 tls for (i = PTN_MENU_START; i <= PTN_MENU_MOUNTPT; i++) {
452 1.2.6.2 tls /* Default to disabled... */
453 1.2.6.2 tls m->opts[i].opt_flags |= OPT_IGNORE;
454 1.2.6.2 tls t = p->pi_fstype;
455 1.2.6.2 tls if (i == PTN_MENU_END)
456 1.2.6.2 tls /* The 'end address' is calculated from the size */
457 1.2.6.2 tls continue;
458 1.2.6.2 tls if (t == FS_UNUSED || (t == FS_SWAP && i > PTN_MENU_END)) {
459 1.2.6.2 tls /* Nothing after 'size' can be set for swap/unused */
460 1.2.6.2 tls p->pi_flags &= ~(PIF_NEWFS | PIF_MOUNT);
461 1.2.6.2 tls p->pi_mount[0] = 0;
462 1.2.6.2 tls continue;
463 1.2.6.2 tls }
464 1.2.6.2 tls if (i == PTN_MENU_NEWFS && t != FS_BSDFFS && t != FS_BSDLFS
465 1.2.6.2 tls && t != FS_APPLEUFS) {
466 1.2.6.2 tls /* Can only newfs UFS and LFS filesystems */
467 1.2.6.2 tls p->pi_flags &= ~PIF_NEWFS;
468 1.2.6.2 tls continue;
469 1.2.6.2 tls }
470 1.2.6.2 tls if (i >= PTN_MENU_ISIZE && i <= PTN_MENU_FSIZE) {
471 1.2.6.2 tls /* Parameters for newfs... */
472 1.2.6.2 tls if (!(p->pi_flags & PIF_NEWFS))
473 1.2.6.2 tls /* Not if we aren't going to run newfs */
474 1.2.6.2 tls continue;
475 1.2.6.2 tls if (t == FS_APPLEUFS && i != PTN_MENU_ISIZE)
476 1.2.6.2 tls /* Can only set # inodes for appleufs */
477 1.2.6.2 tls continue;
478 1.2.6.2 tls if (t == FS_BSDLFS && i != PTN_MENU_BSIZE)
479 1.2.6.2 tls /* LFS doesn't have fragments */
480 1.2.6.2 tls continue;
481 1.2.6.2 tls }
482 1.2.6.2 tls /* Ok: we want this one */
483 1.2.6.2 tls m->opts[i].opt_flags &= ~OPT_IGNORE;
484 1.2.6.2 tls }
485 1.2.6.2 tls }
486 1.2.6.2 tls
487 1.2.6.2 tls static void
488 1.2.6.2 tls disp_sector_count(menudesc *m, msg fmt, uint s)
489 1.2.6.2 tls {
490 1.2.6.2 tls uint ms = MEG / pm->sectorsize;
491 1.2.6.2 tls
492 1.2.6.2 tls wprintw(m->mw, msg_string(fmt),
493 1.2.6.2 tls s / ms, s / pm->dlcylsize, s % pm->dlcylsize ? '*' : ' ', s );
494 1.2.6.2 tls }
495 1.2.6.2 tls
496 1.2.6.2 tls static void
497 1.2.6.2 tls set_ptn_label(menudesc *m, int opt, void *arg)
498 1.2.6.2 tls {
499 1.2.6.2 tls partinfo *p = arg;
500 1.2.6.2 tls const char *c;
501 1.2.6.2 tls
502 1.2.6.2 tls if (m->opts[opt].opt_flags & OPT_IGNORE
503 1.2.6.2 tls && (opt != PTN_MENU_END || p->pi_fstype == FS_UNUSED)) {
504 1.2.6.2 tls wprintw(m->mw, " -");
505 1.2.6.2 tls return;
506 1.2.6.2 tls }
507 1.2.6.2 tls
508 1.2.6.2 tls switch (opt) {
509 1.2.6.2 tls case PTN_MENU_FSKIND:
510 1.2.6.2 tls if (p->pi_fstype == FS_BSDFFS)
511 1.2.6.2 tls if (p->pi_flags & PIF_FFSv2)
512 1.2.6.2 tls c = "FFSv2";
513 1.2.6.2 tls else
514 1.2.6.2 tls c = "FFSv1";
515 1.2.6.2 tls else
516 1.2.6.2 tls c = getfslabelname(p->pi_fstype);
517 1.2.6.2 tls wprintw(m->mw, msg_string(MSG_fstype_fmt), c);
518 1.2.6.2 tls break;
519 1.2.6.2 tls case PTN_MENU_START:
520 1.2.6.2 tls disp_sector_count(m, MSG_start_fmt, p->pi_offset);
521 1.2.6.2 tls break;
522 1.2.6.2 tls case PTN_MENU_SIZE:
523 1.2.6.2 tls disp_sector_count(m, MSG_size_fmt, p->pi_size);
524 1.2.6.2 tls break;
525 1.2.6.2 tls case PTN_MENU_END:
526 1.2.6.2 tls disp_sector_count(m, MSG_end_fmt, p->pi_offset + p->pi_size);
527 1.2.6.2 tls break;
528 1.2.6.2 tls case PTN_MENU_NEWFS:
529 1.2.6.2 tls wprintw(m->mw, msg_string(MSG_newfs_fmt),
530 1.2.6.2 tls msg_string(p->pi_flags & PIF_NEWFS ? MSG_Yes : MSG_No));
531 1.2.6.2 tls break;
532 1.2.6.2 tls case PTN_MENU_ISIZE:
533 1.2.6.2 tls wprintw(m->mw, msg_string(p->pi_isize > 0 ?
534 1.2.6.2 tls MSG_isize_fmt : MSG_isize_fmt_dflt), p->pi_isize);
535 1.2.6.2 tls break;
536 1.2.6.2 tls case PTN_MENU_BSIZE:
537 1.2.6.2 tls wprintw(m->mw, msg_string(MSG_bsize_fmt),
538 1.2.6.2 tls p->pi_fsize * p->pi_frag);
539 1.2.6.2 tls break;
540 1.2.6.2 tls case PTN_MENU_FSIZE:
541 1.2.6.2 tls wprintw(m->mw, msg_string(MSG_fsize_fmt), p->pi_fsize);
542 1.2.6.2 tls break;
543 1.2.6.2 tls case PTN_MENU_MOUNT:
544 1.2.6.2 tls wprintw(m->mw, msg_string(MSG_mount_fmt),
545 1.2.6.2 tls msg_string(p->pi_flags & PIF_MOUNT ? MSG_Yes : MSG_No));
546 1.2.6.2 tls break;
547 1.2.6.2 tls case PTN_MENU_MOUNTOPT:
548 1.2.6.2 tls wprintw(m->mw, "%s", msg_string(MSG_mount_options_fmt));
549 1.2.6.2 tls if (p->pi_flags & PIF_ASYNC)
550 1.2.6.2 tls wprintw(m->mw, "async ");
551 1.2.6.2 tls if (p->pi_flags & PIF_NOATIME)
552 1.2.6.2 tls wprintw(m->mw, "noatime ");
553 1.2.6.2 tls if (p->pi_flags & PIF_NODEV)
554 1.2.6.2 tls wprintw(m->mw, "nodev ");
555 1.2.6.2 tls if (p->pi_flags & PIF_NODEVMTIME)
556 1.2.6.2 tls wprintw(m->mw, "nodevmtime ");
557 1.2.6.2 tls if (p->pi_flags & PIF_NOEXEC)
558 1.2.6.2 tls wprintw(m->mw, "noexec ");
559 1.2.6.2 tls if (p->pi_flags & PIF_NOSUID)
560 1.2.6.2 tls wprintw(m->mw, "nosuid ");
561 1.2.6.2 tls if (p->pi_flags & PIF_LOG)
562 1.2.6.2 tls wprintw(m->mw, "log ");
563 1.2.6.2 tls break;
564 1.2.6.2 tls case PTN_MENU_MOUNTPT:
565 1.2.6.2 tls wprintw(m->mw, msg_string(MSG_mountpt_fmt), p->pi_mount);
566 1.2.6.2 tls break;
567 1.2.6.2 tls }
568 1.2.6.2 tls }
569 1.2.6.2 tls
570 1.2.6.2 tls static int
571 1.2.6.2 tls show_all_unused(menudesc *m, void *arg)
572 1.2.6.2 tls {
573 1.2.6.2 tls struct ptn_menu_info *pi = arg;
574 1.2.6.2 tls
575 1.2.6.2 tls pi->flags |= PIF_SHOW_UNUSED;
576 1.2.6.2 tls return 0;
577 1.2.6.2 tls }
578 1.2.6.2 tls
579 1.2.6.2 tls static void
580 1.2.6.2 tls set_label_texts(menudesc *menu, void *arg)
581 1.2.6.2 tls {
582 1.2.6.2 tls struct ptn_menu_info *pi = arg;
583 1.2.6.2 tls menu_ent *m;
584 1.2.6.2 tls int ptn, show_unused_ptn;
585 1.2.6.2 tls int rawptn = getrawpartition();
586 1.2.6.2 tls int maxpart = getmaxpartitions();
587 1.2.6.2 tls
588 1.2.6.2 tls msg_display(MSG_fspart);
589 1.2.6.2 tls msg_table_add(MSG_fspart_header, multname, multname, multname);
590 1.2.6.2 tls
591 1.2.6.2 tls for (show_unused_ptn = 0, ptn = 0; ptn < maxpart; ptn++) {
592 1.2.6.2 tls m = &menu->opts[ptn];
593 1.2.6.2 tls m->opt_menu = OPT_NOMENU;
594 1.2.6.2 tls m->opt_name = NULL;
595 1.2.6.2 tls m->opt_action = edit_ptn;
596 1.2.6.2 tls if (ptn == rawptn
597 1.2.6.2 tls #ifdef PART_BOOT
598 1.2.6.2 tls || ptn == PART_BOOT
599 1.2.6.2 tls #endif
600 1.2.6.2 tls || ptn == PART_C) {
601 1.2.6.2 tls m->opt_flags = OPT_IGNORE;
602 1.2.6.2 tls } else {
603 1.2.6.2 tls m->opt_flags = 0;
604 1.2.6.2 tls if (pm->bsdlabel[ptn].pi_fstype == FS_UNUSED)
605 1.2.6.2 tls continue;
606 1.2.6.2 tls }
607 1.2.6.2 tls show_unused_ptn = ptn + 2;
608 1.2.6.2 tls }
609 1.2.6.2 tls
610 1.2.6.2 tls if (!(pi->flags & PIF_SHOW_UNUSED) && ptn > show_unused_ptn) {
611 1.2.6.2 tls ptn = show_unused_ptn;
612 1.2.6.2 tls m = &menu->opts[ptn];
613 1.2.6.2 tls m->opt_name = MSG_show_all_unused_partitions;
614 1.2.6.2 tls m->opt_action = show_all_unused;
615 1.2.6.2 tls ptn++;
616 1.2.6.2 tls }
617 1.2.6.2 tls
618 1.2.6.2 tls m = &menu->opts[ptn];
619 1.2.6.2 tls m->opt_menu = MENU_sizechoice;
620 1.2.6.2 tls m->opt_flags = OPT_SUB;
621 1.2.6.2 tls m->opt_action = NULL;
622 1.2.6.2 tls m->opt_name = MSG_askunits;
623 1.2.6.2 tls
624 1.2.6.2 tls menu->numopts = ptn + 1;
625 1.2.6.2 tls }
626 1.2.6.2 tls
627 1.2.6.2 tls /*
628 1.2.6.2 tls * Check a disklabel.
629 1.2.6.2 tls * If there are overlapping active partitions,
630 1.2.6.2 tls * Ask the user if they want to edit the partition or give up.
631 1.2.6.2 tls */
632 1.2.6.2 tls int
633 1.2.6.2 tls edit_and_check_label(partinfo *lp, int nparts, int rawpart, int bsdpart)
634 1.2.6.2 tls {
635 1.2.6.2 tls static struct menu_ent *menu;
636 1.2.6.2 tls static int menu_no = -1;
637 1.2.6.2 tls static struct ptn_menu_info pi;
638 1.2.6.2 tls int maxpart = getmaxpartitions();
639 1.2.6.2 tls
640 1.2.6.2 tls if (menu == NULL) {
641 1.2.6.2 tls menu = malloc((maxpart + 1) * sizeof *menu);
642 1.2.6.2 tls if (!menu)
643 1.2.6.2 tls return 1;
644 1.2.6.2 tls }
645 1.2.6.2 tls
646 1.2.6.2 tls if (menu_no == -1) {
647 1.2.6.2 tls menu_no = new_menu(NULL, menu, maxpart + 1,
648 1.2.6.2 tls 0, -1, maxpart + 2, 74,
649 1.2.6.2 tls MC_ALWAYS_SCROLL | MC_NOBOX | MC_DFLTEXIT,
650 1.2.6.2 tls set_label_texts, fmt_fspart, NULL, NULL,
651 1.2.6.2 tls MSG_partition_sizes_ok);
652 1.2.6.2 tls }
653 1.2.6.2 tls
654 1.2.6.2 tls if (menu_no < 0)
655 1.2.6.2 tls return 1;
656 1.2.6.2 tls
657 1.2.6.2 tls pi.flags = 0;
658 1.2.6.2 tls pm->current_cylsize = pm->dlcylsize;
659 1.2.6.2 tls
660 1.2.6.2 tls for (;;) {
661 1.2.6.2 tls /* first give the user the option to edit the label... */
662 1.2.6.2 tls process_menu(menu_no, &pi);
663 1.2.6.2 tls
664 1.2.6.2 tls /* User thinks the label is OK. */
665 1.2.6.2 tls /* check we have a single root fs */
666 1.2.6.2 tls if (check_one_root(lp, nparts) == 0 && partman_go == 0)
667 1.2.6.2 tls msg_display(MSG_must_be_one_root);
668 1.2.6.2 tls else
669 1.2.6.2 tls /* Check for overlaps */
670 1.2.6.2 tls if (checkoverlap(lp, nparts, rawpart, bsdpart) == 0)
671 1.2.6.2 tls return 1;
672 1.2.6.2 tls
673 1.2.6.2 tls /*XXX ???*/
674 1.2.6.2 tls msg_display_add(MSG_edit_partitions_again);
675 1.2.6.2 tls process_menu(MENU_yesno, NULL);
676 1.2.6.2 tls if (!yesno)
677 1.2.6.2 tls return(0);
678 1.2.6.2 tls }
679 1.2.6.2 tls
680 1.2.6.2 tls /*NOTREACHED*/
681 1.2.6.2 tls }
682 1.2.6.2 tls
683 1.2.6.2 tls /*
684 1.2.6.2 tls * Read a label from disk into a sysinst label structure.
685 1.2.6.2 tls */
686 1.2.6.2 tls int
687 1.2.6.2 tls incorelabel(const char *dkname, partinfo *lp)
688 1.2.6.2 tls {
689 1.2.6.2 tls struct disklabel lab;
690 1.2.6.2 tls int i, maxpart;
691 1.2.6.2 tls struct partition *pp;
692 1.2.6.2 tls int fd;
693 1.2.6.2 tls char buf[64];
694 1.2.6.2 tls
695 1.2.6.2 tls if (get_real_geom(dkname, &lab) == 0)
696 1.2.6.2 tls return -1;
697 1.2.6.2 tls
698 1.2.6.2 tls touchwin(stdscr);
699 1.2.6.2 tls maxpart = getmaxpartitions();
700 1.2.6.2 tls if (maxpart > MAXPARTITIONS)
701 1.2.6.2 tls maxpart = MAXPARTITIONS;
702 1.2.6.2 tls if (maxpart > lab.d_npartitions)
703 1.2.6.2 tls maxpart = lab.d_npartitions;
704 1.2.6.2 tls
705 1.2.6.2 tls /*
706 1.2.6.2 tls * Historically sysinst writes the name to d_typename rather
707 1.2.6.2 tls * than d_diskname. Who knows why, but pull the value back here.
708 1.2.6.2 tls */
709 1.2.6.2 tls if (lab.d_typename[0] != 0 && strcmp(lab.d_typename, "unknown") != 0)
710 1.2.6.2 tls strlcpy(pm->bsddiskname, lab.d_typename, sizeof pm->bsddiskname);
711 1.2.6.2 tls
712 1.2.6.2 tls fd = opendisk(dkname, O_RDONLY, buf, sizeof buf, 0);
713 1.2.6.2 tls pp = &lab.d_partitions[0];
714 1.2.6.2 tls for (i = 0; i < maxpart; lp++, pp++, i++) {
715 1.2.6.2 tls lp->pi_partition = *pp;
716 1.2.6.2 tls if (lp->pi_fstype >= FSMAXTYPES)
717 1.2.6.2 tls lp->pi_fstype = FS_OTHER;
718 1.2.6.2 tls strlcpy(lp->pi_mount, get_last_mounted(fd, pp->p_offset, lp),
719 1.2.6.2 tls sizeof lp->pi_mount);
720 1.2.6.2 tls }
721 1.2.6.2 tls if (fd != -1)
722 1.2.6.2 tls close(fd);
723 1.2.6.2 tls
724 1.2.6.2 tls return 0;
725 1.2.6.2 tls }
726 1.2.6.2 tls
727 1.2.6.2 tls /*
728 1.2.6.2 tls * Try to get 'last mounted on' (or equiv) from fs superblock.
729 1.2.6.2 tls */
730 1.2.6.2 tls const char *
731 1.2.6.2 tls get_last_mounted(int fd, int partstart, partinfo *lp)
732 1.2.6.2 tls {
733 1.2.6.2 tls static char sblk[SBLOCKSIZE]; /* is this enough? */
734 1.2.6.2 tls struct fs *SB = (struct fs *)sblk;
735 1.2.6.2 tls static const int sblocks[] = SBLOCKSEARCH;
736 1.2.6.2 tls const int *sbp;
737 1.2.6.2 tls char *cp;
738 1.2.6.2 tls const char *mnt = NULL;
739 1.2.6.2 tls int len;
740 1.2.6.2 tls
741 1.2.6.2 tls if (fd == -1)
742 1.2.6.2 tls return "";
743 1.2.6.2 tls
744 1.2.6.2 tls /* Check UFS1/2 (and hence LFS) superblock */
745 1.2.6.2 tls for (sbp = sblocks; mnt == NULL && *sbp != -1; sbp++) {
746 1.2.6.2 tls if (pread(fd, sblk, sizeof sblk,
747 1.2.6.2 tls partstart * (off_t)512 + *sbp) != sizeof sblk)
748 1.2.6.2 tls continue;
749 1.2.6.2 tls /* Maybe we should validate the checksum??? */
750 1.2.6.2 tls switch (SB->fs_magic) {
751 1.2.6.2 tls case FS_UFS1_MAGIC:
752 1.2.6.2 tls case FS_UFS1_MAGIC_SWAPPED:
753 1.2.6.2 tls if (!(SB->fs_old_flags & FS_FLAGS_UPDATED)) {
754 1.2.6.2 tls if (*sbp == SBLOCK_UFS1)
755 1.2.6.2 tls mnt = (const char *)SB->fs_fsmnt;
756 1.2.6.2 tls } else {
757 1.2.6.2 tls /* Check we have the main superblock */
758 1.2.6.2 tls if (SB->fs_sblockloc == *sbp)
759 1.2.6.2 tls mnt = (const char *)SB->fs_fsmnt;
760 1.2.6.2 tls }
761 1.2.6.2 tls continue;
762 1.2.6.2 tls case FS_UFS2_MAGIC:
763 1.2.6.2 tls case FS_UFS2_MAGIC_SWAPPED:
764 1.2.6.2 tls /* Check we have the main superblock */
765 1.2.6.2 tls if (SB->fs_sblockloc == *sbp) {
766 1.2.6.2 tls mnt = (const char *)SB->fs_fsmnt;
767 1.2.6.2 tls if (lp != NULL)
768 1.2.6.2 tls lp->pi_flags |= PIF_FFSv2;
769 1.2.6.2 tls }
770 1.2.6.2 tls continue;
771 1.2.6.2 tls }
772 1.2.6.2 tls
773 1.2.6.2 tls #if 0 /* Requires fs/ext2fs/ext2fs.h which collides badly... */
774 1.2.6.2 tls if ((struct ext2fs *)sblk)->e2fs_magic == E2FS_MAGIC) {
775 1.2.6.2 tls mnt = ((struct ext2fs *)sblk)->e2fs_fsmnt;
776 1.2.6.2 tls continue;
777 1.2.6.2 tls }
778 1.2.6.2 tls #endif
779 1.2.6.2 tls if (*sbp != 0)
780 1.2.6.2 tls continue;
781 1.2.6.2 tls
782 1.2.6.2 tls /* If start of partition check for other fs types */
783 1.2.6.2 tls if (sblk[0x42] == 0x29 && memcmp(sblk + 0x52, "FAT", 3) == 0) {
784 1.2.6.2 tls /* Probably a FAT filesystem, report volume name */
785 1.2.6.2 tls cp = strchr(sblk + 0x47, ' ');
786 1.2.6.2 tls if (cp == NULL || cp - (sblk + 0x47) > 11)
787 1.2.6.2 tls cp = sblk + 0x47 + 11;
788 1.2.6.2 tls *cp = 0;
789 1.2.6.2 tls return sblk + 0x47;
790 1.2.6.2 tls }
791 1.2.6.2 tls }
792 1.2.6.2 tls
793 1.2.6.2 tls if (mnt == NULL)
794 1.2.6.2 tls return "";
795 1.2.6.2 tls
796 1.2.6.2 tls /* If sysinst mounted this last then strip prefix */
797 1.2.6.2 tls len = strlen(targetroot_mnt);
798 1.2.6.2 tls if (memcmp(mnt, targetroot_mnt, len) == 0) {
799 1.2.6.2 tls if (mnt[len] == 0)
800 1.2.6.2 tls return "/";
801 1.2.6.2 tls if (mnt[len] == '/')
802 1.2.6.2 tls return mnt + len;
803 1.2.6.2 tls }
804 1.2.6.2 tls return mnt;
805 1.2.6.2 tls #undef SB
806 1.2.6.2 tls }
807 1.2.6.2 tls
808 1.2.6.2 tls /* Ask for a partition offset, check bounds and do the needed roundups */
809 1.2.6.2 tls static uint32_t
810 1.2.6.2 tls getpartoff(uint32_t defpartstart)
811 1.2.6.2 tls {
812 1.2.6.2 tls char defsize[20], isize[20], maxpartc;
813 1.2.6.2 tls uint32_t i;
814 1.2.6.2 tls uint32_t localsizemult;
815 1.2.6.2 tls int partn;
816 1.2.6.2 tls const char *errmsg = "\n";
817 1.2.6.2 tls
818 1.2.6.2 tls maxpartc = 'a' + getmaxpartitions() - 1;
819 1.2.6.2 tls for (;;) {
820 1.2.6.2 tls snprintf(defsize, sizeof defsize, "%d", defpartstart/sizemult);
821 1.2.6.2 tls msg_prompt_win(MSG_label_offset, -1, 13, 70, 9,
822 1.2.6.2 tls (defpartstart > 0) ? defsize : NULL, isize, sizeof isize,
823 1.2.6.2 tls errmsg, maxpartc, maxpartc, multname);
824 1.2.6.2 tls if (strcmp(defsize, isize) == 0)
825 1.2.6.2 tls /* Don't do rounding if default accepted */
826 1.2.6.2 tls return defpartstart;
827 1.2.6.2 tls if (isize[1] == '\0' && isize[0] >= 'a' &&
828 1.2.6.2 tls isize[0] <= maxpartc) {
829 1.2.6.2 tls partn = isize[0] - 'a';
830 1.2.6.2 tls i = pm->bsdlabel[partn].pi_size + pm->bsdlabel[partn].pi_offset;
831 1.2.6.2 tls localsizemult = 1;
832 1.2.6.2 tls } else if (atoi(isize) == -1) {
833 1.2.6.2 tls i = pm->ptstart;
834 1.2.6.2 tls localsizemult = 1;
835 1.2.6.2 tls } else {
836 1.2.6.2 tls if (atofsb(isize, &i, &localsizemult)) {
837 1.2.6.2 tls errmsg = msg_string(MSG_invalid_sector_number);
838 1.2.6.2 tls continue;
839 1.2.6.2 tls }
840 1.2.6.2 tls }
841 1.2.6.2 tls /* round to cylinder size if localsizemult != 1 */
842 1.2.6.2 tls i = NUMSEC(i/localsizemult, localsizemult, pm->dlcylsize);
843 1.2.6.2 tls /* Adjust to start of slice if needed */
844 1.2.6.2 tls if ((i < pm->ptstart && (pm->ptstart - i) < localsizemult) ||
845 1.2.6.2 tls (i > pm->ptstart && (i - pm->ptstart) < localsizemult)) {
846 1.2.6.2 tls i = pm->ptstart;
847 1.2.6.2 tls }
848 1.2.6.2 tls if (i <= pm->dlsize)
849 1.2.6.2 tls break;
850 1.2.6.2 tls errmsg = msg_string(MSG_startoutsidedisk);
851 1.2.6.2 tls }
852 1.2.6.2 tls return i;
853 1.2.6.2 tls }
854 1.2.6.2 tls
855 1.2.6.2 tls
856 1.2.6.2 tls /* Ask for a partition size, check bounds and do the needed roundups */
857 1.2.6.2 tls static uint32_t
858 1.2.6.2 tls getpartsize(uint32_t partstart, uint32_t defpartsize)
859 1.2.6.2 tls {
860 1.2.6.2 tls char dsize[20], isize[20], maxpartc;
861 1.2.6.2 tls const char *errmsg = "\n";
862 1.2.6.2 tls uint32_t i, partend, localsizemult;
863 1.2.6.2 tls uint32_t fsptend = pm->ptstart + pm->ptsize;
864 1.2.6.2 tls int partn;
865 1.2.6.2 tls
866 1.2.6.2 tls maxpartc = 'a' + getmaxpartitions() - 1;
867 1.2.6.2 tls for (;;) {
868 1.2.6.2 tls snprintf(dsize, sizeof dsize, "%d", defpartsize/sizemult);
869 1.2.6.2 tls msg_prompt_win(MSG_label_size, -1, 12, 70, 9,
870 1.2.6.2 tls (defpartsize != 0) ? dsize : 0, isize, sizeof isize,
871 1.2.6.2 tls errmsg, maxpartc, multname);
872 1.2.6.2 tls if (strcmp(isize, dsize) == 0)
873 1.2.6.2 tls return defpartsize;
874 1.2.6.2 tls if (isize[1] == '\0' && isize[0] >= 'a' &&
875 1.2.6.2 tls isize[0] <= maxpartc) {
876 1.2.6.2 tls partn = isize[0] - 'a';
877 1.2.6.2 tls i = pm->bsdlabel[partn].pi_offset - partstart;
878 1.2.6.2 tls localsizemult = 1;
879 1.2.6.2 tls } else if (atoi(isize) == -1) {
880 1.2.6.2 tls i = fsptend - partstart;
881 1.2.6.2 tls localsizemult = 1;
882 1.2.6.2 tls } else {
883 1.2.6.2 tls if (atofsb(isize, &i, &localsizemult)) {
884 1.2.6.2 tls errmsg = msg_string(MSG_invalid_sector_number);
885 1.2.6.2 tls continue;
886 1.2.6.2 tls }
887 1.2.6.2 tls }
888 1.2.6.2 tls /*
889 1.2.6.2 tls * partend is aligned to a cylinder if localsizemult
890 1.2.6.2 tls * is not 1 sector
891 1.2.6.2 tls */
892 1.2.6.2 tls partend = NUMSEC((partstart + i) / localsizemult,
893 1.2.6.2 tls localsizemult, pm->dlcylsize);
894 1.2.6.2 tls /* Align to end-of-disk or end-of-slice if close enough */
895 1.2.6.2 tls if (partend > (pm->dlsize - localsizemult)
896 1.2.6.2 tls && partend < (pm->dlsize + localsizemult))
897 1.2.6.2 tls partend = pm->dlsize;
898 1.2.6.2 tls if (partend > (fsptend - localsizemult)
899 1.2.6.2 tls && partend < (fsptend + localsizemult))
900 1.2.6.2 tls partend = fsptend;
901 1.2.6.2 tls /* sanity checks */
902 1.2.6.2 tls if (partend > pm->dlsize) {
903 1.2.6.2 tls partend = pm->dlsize;
904 1.2.6.2 tls msg_prompt_win(MSG_endoutsidedisk, -1, 13, 70, 6,
905 1.2.6.2 tls NULL, isize, 1,
906 1.2.6.2 tls (partend - partstart) / sizemult, multname);
907 1.2.6.2 tls }
908 1.2.6.2 tls if (partend < partstart)
909 1.2.6.2 tls return 0;
910 1.2.6.2 tls return (partend - partstart);
911 1.2.6.2 tls }
912 1.2.6.2 tls /* NOTREACHED */
913 1.2.6.2 tls }
914 1.2.6.2 tls
915 1.2.6.2 tls /*
916 1.2.6.2 tls * convert a string to a number of sectors, with a possible unit
917 1.2.6.2 tls * 150M = 150 Megabytes
918 1.2.6.2 tls * 2000c = 2000 cylinders
919 1.2.6.2 tls * 150256s = 150256 sectors
920 1.2.6.2 tls * Without units, use the default (sizemult)
921 1.2.6.2 tls * returns the number of sectors, and the unit used (for roundups).
922 1.2.6.2 tls */
923 1.2.6.2 tls
924 1.2.6.2 tls static int
925 1.2.6.2 tls atofsb(const char *str, uint32_t *p_val, uint32_t *localsizemult)
926 1.2.6.2 tls {
927 1.2.6.2 tls int i;
928 1.2.6.2 tls uint32_t val;
929 1.2.6.2 tls
930 1.2.6.2 tls *localsizemult = sizemult;
931 1.2.6.2 tls if (str[0] == '\0') {
932 1.2.6.2 tls return 1;
933 1.2.6.2 tls }
934 1.2.6.2 tls val = 0;
935 1.2.6.2 tls for (i = 0; str[i] != '\0'; i++) {
936 1.2.6.2 tls if (str[i] >= '0' && str[i] <= '9') {
937 1.2.6.2 tls val = val * 10 + str[i] - '0';
938 1.2.6.2 tls continue;
939 1.2.6.2 tls }
940 1.2.6.2 tls if (str[i + 1] != '\0') {
941 1.2.6.2 tls /* A non-digit caracter, not at the end */
942 1.2.6.2 tls return 1;
943 1.2.6.2 tls }
944 1.2.6.2 tls if (str[i] == 'G' || str[i] == 'g') {
945 1.2.6.2 tls val *= 1024;
946 1.2.6.2 tls *localsizemult = MEG / pm->sectorsize;
947 1.2.6.2 tls break;
948 1.2.6.2 tls }
949 1.2.6.2 tls if (str[i] == 'M' || str[i] == 'm') {
950 1.2.6.2 tls *localsizemult = MEG / pm->sectorsize;
951 1.2.6.2 tls break;
952 1.2.6.2 tls }
953 1.2.6.2 tls if (str[i] == 'c') {
954 1.2.6.2 tls *localsizemult = pm->dlcylsize;
955 1.2.6.2 tls break;
956 1.2.6.2 tls }
957 1.2.6.2 tls if (str[i] == 's') {
958 1.2.6.2 tls *localsizemult = 1;
959 1.2.6.2 tls break;
960 1.2.6.2 tls }
961 1.2.6.2 tls /* not a known unit */
962 1.2.6.2 tls return 1;
963 1.2.6.2 tls }
964 1.2.6.2 tls *p_val = val * (*localsizemult);
965 1.2.6.2 tls return 0;
966 1.2.6.2 tls }
967