xref: /netbsd-src/sbin/newfs_udf/newfs_udf.c (revision a7e090f70e491979434963c9a27df4020fe0a18b)
1 /* $NetBSD: newfs_udf.c,v 1.8 2009/09/17 10:37:28 reinoud Exp $ */
2 
3 /*
4  * Copyright (c) 2006, 2008 Reinoud Zandijk
5  * All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer.
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  *
16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
17  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
18  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
19  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
20  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
21  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
22  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
23  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
25  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26  *
27  */
28 
29 /*
30  * TODO
31  * - implement metadata formatting for BD-R
32  * - implement support for a read-only companion partition?
33  */
34 
35 #define _EXPOSE_MMC
36 #if 0
37 # define DEBUG
38 #endif
39 
40 #include <stdio.h>
41 #include <stdlib.h>
42 #include <dirent.h>
43 #include <inttypes.h>
44 #include <stdint.h>
45 #include <string.h>
46 #include <errno.h>
47 #include <fcntl.h>
48 #include <unistd.h>
49 #include <util.h>
50 #include <time.h>
51 #include <assert.h>
52 #include <err.h>
53 
54 #include <sys/ioctl.h>
55 #include <sys/stat.h>
56 #include <sys/types.h>
57 #include <sys/cdio.h>
58 #include <sys/disklabel.h>
59 #include <sys/dkio.h>
60 #include <sys/param.h>
61 #include <sys/queue.h>
62 
63 #include <fs/udf/ecma167-udf.h>
64 #include <fs/udf/udf_mount.h>
65 
66 #include "mountprog.h"
67 #include "udf_create.h"
68 
69 /* general settings */
70 #define UDF_512_TRACK	0	/* NOT recommended */
71 #define UDF_META_PERC  20	/* picked */
72 
73 
74 /* prototypes */
75 int newfs_udf(int argc, char **argv);
76 static void usage(void) __attribute__((__noreturn__));
77 
78 int udf_derive_format(int req_en, int req_dis, int force);
79 int udf_proces_names(void);
80 int udf_do_newfs(void);
81 
82 /* Identifying myself */
83 #define APP_NAME		"*NetBSD newfs"
84 #define APP_VERSION_MAIN	0
85 #define APP_VERSION_SUB		3
86 #define IMPL_NAME		"*NetBSD userland UDF"
87 
88 
89 /* global variables describing disc and format requests */
90 int	 fd;				/* device: file descriptor */
91 char	*dev;				/* device: name		   */
92 struct mmc_discinfo mmc_discinfo;	/* device: disc info	   */
93 
94 char	*format_str;			/* format: string representation */
95 int	 format_flags;			/* format: attribute flags	 */
96 int	 media_accesstype;		/* derived from current mmc cap  */
97 int	 check_surface;			/* for rewritables               */
98 
99 int	 wrtrack_skew;
100 int	 meta_perc = UDF_META_PERC;
101 float	 meta_fract = (float) UDF_META_PERC / 100.0;
102 
103 
104 /* shared structure between udf_create.c users */
105 struct udf_create_context context;
106 struct udf_disclayout     layout;
107 
108 
109 /* queue for temporary storage of sectors to be written out */
110 struct wrsect {
111 	uint32_t  sectornr;
112 	uint8_t	 *sector_data;
113 	TAILQ_ENTRY(wrsect) next;
114 };
115 
116 /* write queue and track blocking skew */
117 TAILQ_HEAD(wrsect_list, wrsect) write_queue;
118 
119 
120 /* --------------------------------------------------------------------- */
121 
122 /*
123  * write queue implementation
124  */
125 
126 static int
127 udf_write_sector(void *sector, uint32_t location)
128 {
129 	struct wrsect *pos, *seekpos;
130 
131 
132 	/* search location */
133 	TAILQ_FOREACH_REVERSE(seekpos, &write_queue, wrsect_list, next) {
134 		if (seekpos->sectornr <= location)
135 			break;
136 	}
137 	if ((seekpos == NULL) || (seekpos->sectornr != location)) {
138 		pos = calloc(1, sizeof(struct wrsect));
139 		if (pos == NULL)
140 			return ENOMEM;
141 		/* allocate space for copy of sector data */
142 		pos->sector_data = calloc(1, context.sector_size);
143 		if (pos->sector_data == NULL)
144 			return ENOMEM;
145 		pos->sectornr = location;
146 
147 		if (seekpos) {
148 			TAILQ_INSERT_AFTER(&write_queue, seekpos, pos, next);
149 		} else {
150 			TAILQ_INSERT_HEAD(&write_queue, pos, next);
151 		}
152 	} else {
153 		pos = seekpos;
154 	}
155 	memcpy(pos->sector_data, sector, context.sector_size);
156 
157 	return 0;
158 }
159 
160 
161 /*
162  * Now all write requests are queued in the TAILQ, write them out to the
163  * disc/file image. Special care needs to be taken for devices that are only
164  * strict overwritable i.e. only in packet size chunks
165  *
166  * XXX support for growing vnd?
167  */
168 
169 static int
170 writeout_write_queue(void)
171 {
172 	struct wrsect *pos;
173 	uint64_t offset;
174 	uint32_t line_len, line_offset;
175 	uint32_t line_start, new_line_start, relpos;
176 	uint32_t blockingnr;
177 	uint8_t *linebuf, *adr;
178 
179 	blockingnr  = layout.blockingnr;
180 	line_len    = blockingnr   * context.sector_size;
181 	line_offset = wrtrack_skew * context.sector_size;
182 
183 	linebuf     = malloc(line_len);
184 	if (linebuf == NULL)
185 		return ENOMEM;
186 
187 	pos = TAILQ_FIRST(&write_queue);
188 	bzero(linebuf, line_len);
189 
190 	/*
191 	 * Always writing out in whole lines now; this is slightly wastefull
192 	 * on logical overwrite volumes but it reduces complexity and the loss
193 	 * is near zero compared to disc size.
194 	 */
195 	line_start = (pos->sectornr - wrtrack_skew) / blockingnr;
196 	TAILQ_FOREACH(pos, &write_queue, next) {
197 		new_line_start = (pos->sectornr - wrtrack_skew) / blockingnr;
198 		if (new_line_start != line_start) {
199 			/* write out */
200 			offset = (uint64_t) line_start * line_len + line_offset;
201 #ifdef DEBUG
202 			printf("WRITEOUT %08"PRIu64" + %02d -- "
203 				"[%08"PRIu64"..%08"PRIu64"]\n",
204 				offset / context.sector_size, blockingnr,
205 				offset / context.sector_size,
206 				offset / context.sector_size + blockingnr-1);
207 #endif
208 			if (pwrite(fd, linebuf, line_len, offset) < 0) {
209 				perror("Writing failed");
210 				return errno;
211 			}
212 			line_start = new_line_start;
213 			bzero(linebuf, line_len);
214 		}
215 
216 		relpos = (pos->sectornr - wrtrack_skew) % blockingnr;
217 		adr = linebuf + relpos * context.sector_size;
218 		memcpy(adr, pos->sector_data, context.sector_size);
219 	}
220 	/* writeout last chunk */
221 	offset = (uint64_t) line_start * line_len + line_offset;
222 #ifdef DEBUG
223 	printf("WRITEOUT %08"PRIu64" + %02d -- [%08"PRIu64"..%08"PRIu64"]\n",
224 		offset / context.sector_size, blockingnr,
225 		offset / context.sector_size,
226 		offset / context.sector_size + blockingnr-1);
227 #endif
228 	if (pwrite(fd, linebuf, line_len, offset) < 0) {
229 		perror("Writing failed");
230 		return errno;
231 	}
232 
233 	/* success */
234 	return 0;
235 }
236 
237 /* --------------------------------------------------------------------- */
238 
239 /*
240  * mmc_discinfo and mmc_trackinfo readers modified from origional in udf main
241  * code in sys/fs/udf/
242  */
243 
244 #ifdef DEBUG
245 static void
246 udf_dump_discinfo(struct mmc_discinfo *di)
247 {
248 	char bits[128];
249 
250 	printf("Device/media info  :\n");
251 	printf("\tMMC profile        0x%02x\n", di->mmc_profile);
252 	printf("\tderived class      %d\n", di->mmc_class);
253 	printf("\tsector size        %d\n", di->sector_size);
254 	printf("\tdisc state         %d\n", di->disc_state);
255 	printf("\tlast ses state     %d\n", di->last_session_state);
256 	printf("\tbg format state    %d\n", di->bg_format_state);
257 	printf("\tfrst track         %d\n", di->first_track);
258 	printf("\tfst on last ses    %d\n", di->first_track_last_session);
259 	printf("\tlst on last ses    %d\n", di->last_track_last_session);
260 	printf("\tlink block penalty %d\n", di->link_block_penalty);
261 	snprintb(bits, sizeof(bits), MMC_DFLAGS_FLAGBITS, (uint64_t) di->disc_flags);
262 	printf("\tdisc flags         %s\n", bits);
263 	printf("\tdisc id            %x\n", di->disc_id);
264 	printf("\tdisc barcode       %"PRIx64"\n", di->disc_barcode);
265 
266 	printf("\tnum sessions       %d\n", di->num_sessions);
267 	printf("\tnum tracks         %d\n", di->num_tracks);
268 
269 	snprintb(bits, sizeof(bits), MMC_CAP_FLAGBITS, di->mmc_cur);
270 	printf("\tcapabilities cur   %s\n", bits);
271 	snprintb(bits, sizeof(bits), MMC_CAP_FLAGBITS, di->mmc_cap);
272 	printf("\tcapabilities cap   %s\n", bits);
273 	printf("\n");
274 	printf("\tlast_possible_lba  %d\n", di->last_possible_lba);
275 	printf("\n");
276 }
277 #else
278 #define udf_dump_discinfo(a);
279 #endif
280 
281 /* --------------------------------------------------------------------- */
282 
283 static int
284 udf_update_discinfo(struct mmc_discinfo *di)
285 {
286 	struct disklabel  disklab;
287 	struct partition *dp;
288 	struct stat st;
289 	int partnr, error;
290 
291 	memset(di, 0, sizeof(struct mmc_discinfo));
292 
293 	/* check if we're on a MMC capable device, i.e. CD/DVD */
294 	error = ioctl(fd, MMCGETDISCINFO, di);
295 	if (error == 0)
296 		return 0;
297 
298 	/*
299 	 * disc partition support; note we can't use DIOCGPART in userland so
300 	 * get disc label and use the stat info to get the partition number.
301 	 */
302 	if (ioctl(fd, DIOCGDINFO, &disklab) == -1) {
303 		/* failed to get disclabel! */
304 		perror("disklabel");
305 		return errno;
306 	}
307 
308 	/* get disk partition it refers to */
309 	fstat(fd, &st);
310 	partnr = DISKPART(st.st_rdev);
311 	dp = &disklab.d_partitions[partnr];
312 
313 	/* set up a disc info profile for partitions */
314 	di->mmc_profile		= 0x01;	/* disc type */
315 	di->mmc_class		= MMC_CLASS_DISC;
316 	di->disc_state		= MMC_STATE_CLOSED;
317 	di->last_session_state	= MMC_STATE_CLOSED;
318 	di->bg_format_state	= MMC_BGFSTATE_COMPLETED;
319 	di->link_block_penalty	= 0;
320 
321 	di->mmc_cur     = MMC_CAP_RECORDABLE | MMC_CAP_REWRITABLE |
322 		MMC_CAP_ZEROLINKBLK | MMC_CAP_HW_DEFECTFREE;
323 	di->mmc_cap    = di->mmc_cur;
324 	di->disc_flags = MMC_DFLAGS_UNRESTRICTED;
325 
326 	/* TODO problem with last_possible_lba on resizable VND; request */
327 	if (dp->p_size == 0) {
328 		perror("faulty disklabel partition returned, check label\n");
329 		return EIO;
330 	}
331 	di->last_possible_lba = dp->p_size - 1;
332 	di->sector_size       = disklab.d_secsize;
333 
334 	di->num_sessions = 1;
335 	di->num_tracks   = 1;
336 
337 	di->first_track  = 1;
338 	di->first_track_last_session = di->last_track_last_session = 1;
339 
340 	return 0;
341 }
342 
343 
344 static int
345 udf_update_trackinfo(struct mmc_discinfo *di, struct mmc_trackinfo *ti)
346 {
347 	int error, class;
348 
349 	class = di->mmc_class;
350 	if (class != MMC_CLASS_DISC) {
351 		/* tracknr specified in struct ti */
352 		error = ioctl(fd, MMCGETTRACKINFO, ti);
353 		return error;
354 	}
355 
356 	/* discs partition support */
357 	if (ti->tracknr != 1)
358 		return EIO;
359 
360 	/* create fake ti (TODO check for resized vnds) */
361 	ti->sessionnr  = 1;
362 
363 	ti->track_mode = 0;	/* XXX */
364 	ti->data_mode  = 0;	/* XXX */
365 	ti->flags = MMC_TRACKINFO_LRA_VALID | MMC_TRACKINFO_NWA_VALID;
366 
367 	ti->track_start    = 0;
368 	ti->packet_size    = 1;
369 
370 	/* TODO support for resizable vnd */
371 	ti->track_size    = di->last_possible_lba;
372 	ti->next_writable = di->last_possible_lba;
373 	ti->last_recorded = ti->next_writable;
374 	ti->free_blocks   = 0;
375 
376 	return 0;
377 }
378 
379 
380 static int
381 udf_setup_writeparams(struct mmc_discinfo *di)
382 {
383 	struct mmc_writeparams mmc_writeparams;
384 	int error;
385 
386 	if (di->mmc_class == MMC_CLASS_DISC)
387 		return 0;
388 
389 	/*
390 	 * only CD burning normally needs setting up, but other disc types
391 	 * might need other settings to be made. The MMC framework will set up
392 	 * the nessisary recording parameters according to the disc
393 	 * characteristics read in. Modifications can be made in the discinfo
394 	 * structure passed to change the nature of the disc.
395 	 */
396 	memset(&mmc_writeparams, 0, sizeof(struct mmc_writeparams));
397 	mmc_writeparams.mmc_class  = di->mmc_class;
398 	mmc_writeparams.mmc_cur    = di->mmc_cur;
399 
400 	/*
401 	 * UDF dictates first track to determine track mode for the whole
402 	 * disc. [UDF 1.50/6.10.1.1, UDF 1.50/6.10.2.1]
403 	 * To prevent problems with a `reserved' track in front we start with
404 	 * the 2nd track and if that is not valid, go for the 1st.
405 	 */
406 	mmc_writeparams.tracknr = 2;
407 	mmc_writeparams.data_mode  = MMC_DATAMODE_DEFAULT;	/* XA disc */
408 	mmc_writeparams.track_mode = MMC_TRACKMODE_DEFAULT;	/* data */
409 
410 	error = ioctl(fd, MMCSETUPWRITEPARAMS, &mmc_writeparams);
411 	if (error) {
412 		mmc_writeparams.tracknr = 1;
413 		error = ioctl(fd, MMCSETUPWRITEPARAMS, &mmc_writeparams);
414 	}
415 	return error;
416 }
417 
418 
419 static void
420 udf_synchronise_caches(void)
421 {
422 	struct mmc_op mmc_op;
423 
424 	bzero(&mmc_op, sizeof(struct mmc_op));
425 	mmc_op.operation = MMC_OP_SYNCHRONISECACHE;
426 
427 	/* this device might not know this ioct, so just be ignorant */
428 	(void) ioctl(fd, MMCOP, &mmc_op);
429 }
430 
431 /* --------------------------------------------------------------------- */
432 
433 static int
434 udf_write_dscr_phys(union dscrptr *dscr, uint32_t location,
435 	uint32_t sects)
436 {
437 	uint32_t phys, cnt;
438 	uint8_t *bpos;
439 	int error;
440 
441 	dscr->tag.tag_loc = udf_rw32(location);
442 	(void) udf_validate_tag_and_crc_sums(dscr);
443 
444 	for (cnt = 0; cnt < sects; cnt++) {
445 		bpos  = (uint8_t *) dscr;
446 		bpos += context.sector_size * cnt;
447 
448 		phys = location + cnt;
449 		error = udf_write_sector(bpos, phys);
450 		if (error)
451 			return error;
452 	}
453 	return 0;
454 }
455 
456 
457 static int
458 udf_write_dscr_virt(union dscrptr *dscr, uint32_t location, uint32_t vpart,
459 	uint32_t sects)
460 {
461 	struct file_entry *fe;
462 	struct extfile_entry *efe;
463 	struct extattrhdr_desc *extattrhdr;
464 	uint32_t phys, cnt;
465 	uint8_t *bpos;
466 	int error;
467 
468 	extattrhdr = NULL;
469 	if (udf_rw16(dscr->tag.id) == TAGID_FENTRY) {
470 		fe = (struct file_entry *) dscr;
471 		if (udf_rw32(fe->l_ea) > 0)
472 			extattrhdr = (struct extattrhdr_desc *) fe->data;
473 	}
474 	if (udf_rw16(dscr->tag.id) == TAGID_EXTFENTRY) {
475 		efe = (struct extfile_entry *) dscr;
476 		if (udf_rw32(efe->l_ea) > 0)
477 			extattrhdr = (struct extattrhdr_desc *) efe->data;
478 	}
479 	if (extattrhdr) {
480 		extattrhdr->tag.tag_loc = udf_rw32(location);
481 		udf_validate_tag_and_crc_sums((union dscrptr *) extattrhdr);
482 	}
483 
484 	dscr->tag.tag_loc = udf_rw32(location);
485 	udf_validate_tag_and_crc_sums(dscr);
486 
487 	for (cnt = 0; cnt < sects; cnt++) {
488 		bpos  = (uint8_t *) dscr;
489 		bpos += context.sector_size * cnt;
490 
491 		/* NOTE linear mapping assumed in the ranges used */
492 		phys = context.vtop_offset[vpart] + location + cnt;
493 
494 		error = udf_write_sector(bpos, phys);
495 		if (error)
496 			return error;
497 	}
498 	return 0;
499 }
500 
501 /* --------------------------------------------------------------------- */
502 
503 /*
504  * udf_derive_format derives the format_flags from the disc's mmc_discinfo.
505  * The resulting flags uniquely define a disc format. Note there are at least
506  * 7 distinct format types defined in UDF.
507  */
508 
509 #define UDF_VERSION(a) \
510 	(((a) == 0x100) || ((a) == 0x102) || ((a) == 0x150) || ((a) == 0x200) || \
511 	 ((a) == 0x201) || ((a) == 0x250) || ((a) == 0x260))
512 
513 int
514 udf_derive_format(int req_enable, int req_disable, int force)
515 {
516 	/* disc writability, formatted, appendable */
517 	if ((mmc_discinfo.mmc_cur & MMC_CAP_RECORDABLE) == 0) {
518 		(void)printf("Can't newfs readonly device\n");
519 		return EROFS;
520 	}
521 	if (mmc_discinfo.mmc_cur & MMC_CAP_SEQUENTIAL) {
522 		/* sequentials need sessions appended */
523 		if (mmc_discinfo.disc_state == MMC_STATE_CLOSED) {
524 			(void)printf("Can't append session to a closed disc\n");
525 			return EROFS;
526 		}
527 		if ((mmc_discinfo.disc_state != MMC_STATE_EMPTY) && !force) {
528 			(void)printf("Disc not empty! Use -F to force "
529 			    "initialisation\n");
530 			return EROFS;
531 		}
532 	} else {
533 		/* check if disc (being) formatted or has been started on */
534 		if (mmc_discinfo.disc_state == MMC_STATE_EMPTY) {
535 			(void)printf("Disc is not formatted\n");
536 			return EROFS;
537 		}
538 	}
539 
540 	/* determine UDF format */
541 	format_flags = 0;
542 	if (mmc_discinfo.mmc_cur & MMC_CAP_REWRITABLE) {
543 		/* all rewritable media */
544 		format_flags |= FORMAT_REWRITABLE;
545 		if (context.min_udf >= 0x0250) {
546 			/* standard dictates meta as default */
547 			format_flags |= FORMAT_META;
548 		}
549 
550 		if ((mmc_discinfo.mmc_cur & MMC_CAP_HW_DEFECTFREE) == 0) {
551 			/* sparables for defect management */
552 			if (context.min_udf >= 0x150)
553 				format_flags |= FORMAT_SPARABLE;
554 		}
555 	} else {
556 		/* all once recordable media */
557 		format_flags |= FORMAT_WRITEONCE;
558 		if (mmc_discinfo.mmc_cur & MMC_CAP_SEQUENTIAL) {
559 			format_flags |= FORMAT_SEQUENTIAL;
560 
561 			if (mmc_discinfo.mmc_cur & MMC_CAP_PSEUDOOVERWRITE) {
562 				/* logical overwritable */
563 				format_flags |= FORMAT_LOW;
564 			} else {
565 				/* have to use VAT for overwriting */
566 				format_flags |= FORMAT_VAT;
567 			}
568 		} else {
569 			/* rare WORM devices, but BluRay has one, strat4096 */
570 			format_flags |= FORMAT_WORM;
571 		}
572 	}
573 
574 	/* enable/disable requests */
575 	if (req_disable & FORMAT_META) {
576 		format_flags &= ~FORMAT_META;
577 		req_disable  &= ~FORMAT_META;
578 	}
579 	if (req_disable || req_enable) {
580 		(void)printf("Internal error\n");
581 		(void)printf("\tunrecognised enable/disable req.\n");
582 		return EIO;
583 	}
584 	if ((format_flags && FORMAT_VAT) && UDF_512_TRACK)
585 		format_flags |= FORMAT_TRACK512;
586 
587 	/* determine partition/media access type */
588 	media_accesstype = UDF_ACCESSTYPE_NOT_SPECIFIED;
589 	if (mmc_discinfo.mmc_cur & MMC_CAP_REWRITABLE) {
590 		media_accesstype = UDF_ACCESSTYPE_OVERWRITABLE;
591 		if (mmc_discinfo.mmc_cur & MMC_CAP_ERASABLE)
592 			media_accesstype = UDF_ACCESSTYPE_REWRITEABLE;
593 	} else {
594 		/* all once recordable media */
595 		media_accesstype = UDF_ACCESSTYPE_WRITE_ONCE;
596 	}
597 	if (mmc_discinfo.mmc_cur & MMC_CAP_PSEUDOOVERWRITE)
598 		media_accesstype = UDF_ACCESSTYPE_PSEUDO_OVERWITE;
599 
600 	/* adjust minimum version limits */
601 	if (format_flags & FORMAT_VAT)
602 		context.min_udf = MAX(context.min_udf, 0x0150);
603 	if (format_flags & FORMAT_SPARABLE)
604 		context.min_udf = MAX(context.min_udf, 0x0150);
605 	if (format_flags & FORMAT_META)
606 		context.min_udf = MAX(context.min_udf, 0x0250);
607 	if (format_flags & FORMAT_LOW)
608 		context.min_udf = MAX(context.min_udf, 0x0260);
609 
610 	/* adjust maximum version limits not to tease or break things */
611 	if (!(format_flags & FORMAT_META) && (context.max_udf > 0x200))
612 		context.max_udf = 0x201;
613 
614 	if ((format_flags & (FORMAT_VAT | FORMAT_SPARABLE)) == 0)
615 		if (context.max_udf <= 0x150)
616 			context.min_udf = 0x102;
617 
618 	/* limit Ecma 167 descriptor if possible/needed */
619 	context.dscrver = 3;
620 	if ((context.min_udf < 0x200) || (context.max_udf < 0x200)) {
621 		context.dscrver = 2;
622 		context.max_udf = 0x150;	/* last version < 0x200 */
623 	}
624 
625 	/* is it possible ? */
626 	if (context.min_udf > context.max_udf) {
627 		(void)printf("Initialisation prohibited by specified maximum "
628 		    "UDF version 0x%04x. Minimum version required 0x%04x\n",
629 		    context.max_udf, context.min_udf);
630 		return EPERM;
631 	}
632 
633 	if (!UDF_VERSION(context.min_udf) || !UDF_VERSION(context.max_udf)) {
634 		printf("Choose UDF version numbers from "
635 			"0x102, 0x150, 0x200, 0x201, 0x250 and 0x260\n");
636 		printf("Default version is 0x201\n");
637 		return EPERM;
638 	}
639 
640 	return 0;
641 }
642 
643 #undef UDF_VERSION
644 
645 
646 /* --------------------------------------------------------------------- */
647 
648 int
649 udf_proces_names(void)
650 {
651 	uint32_t primary_nr;
652 	uint64_t volset_nr;
653 
654 	if (context.logvol_name == NULL)
655 		context.logvol_name = strdup("anonymous");
656 	if (context.primary_name == NULL) {
657 		if (mmc_discinfo.disc_flags & MMC_DFLAGS_DISCIDVALID) {
658 			primary_nr = mmc_discinfo.disc_id;
659 		} else {
660 			primary_nr = (uint32_t) random();
661 		}
662 		context.primary_name = calloc(32, 1);
663 		sprintf(context.primary_name, "%08"PRIx32, primary_nr);
664 	}
665 	if (context.volset_name == NULL) {
666 		if (mmc_discinfo.disc_flags & MMC_DFLAGS_BARCODEVALID) {
667 			volset_nr = mmc_discinfo.disc_barcode;
668 		} else {
669 			volset_nr  =  (uint32_t) random();
670 			volset_nr |= ((uint64_t) random()) << 32;
671 		}
672 		context.volset_name = calloc(128,1);
673 		sprintf(context.volset_name, "%016"PRIx64, volset_nr);
674 	}
675 	if (context.fileset_name == NULL)
676 		context.fileset_name = strdup("anonymous");
677 
678 	/* check passed/created identifiers */
679 	if (strlen(context.logvol_name)  > 128) {
680 		(void)printf("Logical volume name too long\n");
681 		return EINVAL;
682 	}
683 	if (strlen(context.primary_name) >  32) {
684 		(void)printf("Primary volume name too long\n");
685 		return EINVAL;
686 	}
687 	if (strlen(context.volset_name)  > 128) {
688 		(void)printf("Volume set name too long\n");
689 		return EINVAL;
690 	}
691 	if (strlen(context.fileset_name) > 32) {
692 		(void)printf("Fileset name too long\n");
693 		return EINVAL;
694 	}
695 
696 	/* signal all OK */
697 	return 0;
698 }
699 
700 /* --------------------------------------------------------------------- */
701 
702 static int
703 udf_prepare_disc(void)
704 {
705 	struct mmc_trackinfo ti;
706 	struct mmc_op        op;
707 	int tracknr, error;
708 
709 	/* If the last track is damaged, repair it */
710 	ti.tracknr = mmc_discinfo.last_track_last_session;
711 	error = udf_update_trackinfo(&mmc_discinfo, &ti);
712 	if (error)
713 		return error;
714 
715 	if (ti.flags & MMC_TRACKINFO_DAMAGED) {
716 		/*
717 		 * Need to repair last track before anything can be done.
718 		 * this is an optional command, so ignore its error but report
719 		 * warning.
720 		 */
721 		memset(&op, 0, sizeof(op));
722 		op.operation   = MMC_OP_REPAIRTRACK;
723 		op.mmc_profile = mmc_discinfo.mmc_profile;
724 		op.tracknr     = ti.tracknr;
725 		error = ioctl(fd, MMCOP, &op);
726 
727 		if (error)
728 			(void)printf("Drive can't explicitly repair last "
729 				"damaged track, but it might autorepair\n");
730 	}
731 	/* last track (if any) might not be damaged now, operations are ok now */
732 
733 	/* setup write parameters from discinfo */
734 	error = udf_setup_writeparams(&mmc_discinfo);
735 	if (error)
736 		return error;
737 
738 	/* if the drive is not sequential, we're done */
739 	if ((mmc_discinfo.mmc_cur & MMC_CAP_SEQUENTIAL) == 0)
740 		return 0;
741 
742 #ifdef notyet
743 	/* if last track is not the reserved but an empty track, unreserve it */
744 	if (ti.flags & MMC_TRACKINFO_BLANK) {
745 		if (ti.flags & MMC_TRACKINFO_RESERVED == 0) {
746 			memset(&op, 0, sizeof(op));
747 			op.operation   = MMC_OP_UNRESERVETRACK;
748 			op.mmc_profile = mmc_discinfo.mmc_profile;
749 			op.tracknr     = ti.tracknr;
750 			error = ioctl(fd, MMCOP, &op);
751 			if (error)
752 				return error;
753 
754 			/* update discinfo since it changed by the operation */
755 			error = udf_update_discinfo(&mmc_discinfo);
756 			if (error)
757 				return error;
758 		}
759 	}
760 #endif
761 
762 	/* close the last session if its still open */
763 	if (mmc_discinfo.last_session_state == MMC_STATE_INCOMPLETE) {
764 		printf("Closing last open session if present\n");
765 		/* close all associated tracks */
766 		tracknr = mmc_discinfo.first_track_last_session;
767 		while (tracknr <= mmc_discinfo.last_track_last_session) {
768 			ti.tracknr = tracknr;
769 			error = udf_update_trackinfo(&mmc_discinfo, &ti);
770 			if (error)
771 				return error;
772 			printf("\tClosing open track %d\n", tracknr);
773 			memset(&op, 0, sizeof(op));
774 			op.operation   = MMC_OP_CLOSETRACK;
775 			op.mmc_profile = mmc_discinfo.mmc_profile;
776 			op.tracknr     = tracknr;
777 			error = ioctl(fd, MMCOP, &op);
778 			if (error)
779 				return error;
780 			tracknr ++;
781 		}
782 		printf("Closing session\n");
783 		memset(&op, 0, sizeof(op));
784 		op.operation   = MMC_OP_CLOSESESSION;
785 		op.mmc_profile = mmc_discinfo.mmc_profile;
786 		op.sessionnr   = mmc_discinfo.num_sessions;
787 		error = ioctl(fd, MMCOP, &op);
788 		if (error)
789 			return error;
790 
791 		/* update discinfo since it changed by the operations */
792 		error = udf_update_discinfo(&mmc_discinfo);
793 		if (error)
794 			return error;
795 	}
796 
797 	if (format_flags & FORMAT_TRACK512) {
798 		/* get last track again */
799 		ti.tracknr = mmc_discinfo.last_track_last_session;
800 		error = udf_update_trackinfo(&mmc_discinfo, &ti);
801 		if (error)
802 			return error;
803 
804 		/* Split up the space at 512 for iso cd9660 hooking */
805 		memset(&op, 0, sizeof(op));
806 		op.operation   = MMC_OP_RESERVETRACK_NWA;	/* UPTO nwa */
807 		op.mmc_profile = mmc_discinfo.mmc_profile;
808 		op.extent      = 512;				/* size */
809 		error = ioctl(fd, MMCOP, &op);
810 		if (error)
811 			return error;
812 	}
813 
814 	return 0;
815 }
816 
817 /* --------------------------------------------------------------------- */
818 
819 static int
820 udf_surface_check(void)
821 {
822 	uint32_t loc, block_bytes;
823 	uint32_t sector_size, blockingnr, bpos;
824 	uint8_t *buffer;
825 	int error, num_errors;
826 
827 	sector_size = context.sector_size;
828 	blockingnr  = layout.blockingnr;
829 
830 	block_bytes = layout.blockingnr * sector_size;
831 	if ((buffer = malloc(block_bytes)) == NULL)
832 		return ENOMEM;
833 
834 	/* set all one to not kill Flash memory? */
835 	for (bpos = 0; bpos < block_bytes; bpos++)
836 		buffer[bpos] = 0x00;
837 
838 	printf("\nChecking disc surface : phase 1 - writing\n");
839 	num_errors = 0;
840 	loc = layout.first_lba;
841 	while (loc <= layout.last_lba) {
842 		/* write blockingnr sectors */
843 		error = pwrite(fd, buffer, block_bytes, loc*sector_size);
844 		printf("   %08d + %d (%02d %%)\r", loc, blockingnr,
845 			(int)((100.0 * loc)/layout.last_lba));
846 		fflush(stdout);
847 		if (error == -1) {
848 			/* block is bad */
849 			printf("BAD block at %08d + %d         \n",
850 				loc, layout.blockingnr);
851 			if ((error = udf_register_bad_block(loc)))
852 				return error;
853 			num_errors ++;
854 		}
855 		loc += layout.blockingnr;
856 	}
857 
858 	printf("\nChecking disc surface : phase 2 - reading\n");
859 	num_errors = 0;
860 	loc = layout.first_lba;
861 	while (loc <= layout.last_lba) {
862 		/* read blockingnr sectors */
863 		error = pread(fd, buffer, block_bytes, loc*sector_size);
864 		printf("   %08d + %d (%02d %%)\r", loc, blockingnr,
865 			(int)((100.0 * loc)/layout.last_lba));
866 		fflush(stdout);
867 		if (error == -1) {
868 			/* block is bad */
869 			printf("BAD block at %08d + %d         \n",
870 				loc, layout.blockingnr);
871 			if ((error = udf_register_bad_block(loc)))
872 				return error;
873 			num_errors ++;
874 		}
875 		loc += layout.blockingnr;
876 	}
877 	printf("Scan complete : %d bad blocks found\n", num_errors);
878 	free(buffer);
879 
880 	return 0;
881 }
882 
883 /* --------------------------------------------------------------------- */
884 
885 static int
886 udf_write_iso9660_vrs(void)
887 {
888 	struct vrs_desc *iso9660_vrs_desc;
889 	uint32_t pos;
890 	int error, cnt, dpos;
891 
892 	/* create ISO/Ecma-167 identification descriptors */
893 	if ((iso9660_vrs_desc = calloc(1, context.sector_size)) == NULL)
894 		return ENOMEM;
895 
896 	/*
897 	 * All UDF formats should have their ISO/Ecma-167 descriptors written
898 	 * except when not possible due to track reservation in the case of
899 	 * VAT
900 	 */
901 	if ((format_flags & FORMAT_TRACK512) == 0) {
902 		dpos = (2048 + context.sector_size - 1) / context.sector_size;
903 
904 		/* wipe at least 6 times 2048 byte `sectors' */
905 		for (cnt = 0; cnt < 6 *dpos; cnt++) {
906 			pos = layout.iso9660_vrs + cnt;
907 			if ((error = udf_write_sector(iso9660_vrs_desc, pos)))
908 				return error;
909 		}
910 
911 		/* common VRS fields in all written out ISO descriptors */
912 		iso9660_vrs_desc->struct_type = 0;
913 		iso9660_vrs_desc->version     = 1;
914 		pos = layout.iso9660_vrs;
915 
916 		/* BEA01, NSR[23], TEA01 */
917 		memcpy(iso9660_vrs_desc->identifier, "BEA01", 5);
918 		if ((error = udf_write_sector(iso9660_vrs_desc, pos)))
919 			return error;
920 		pos += dpos;
921 
922 		if (context.dscrver == 2)
923 			memcpy(iso9660_vrs_desc->identifier, "NSR02", 5);
924 		else
925 			memcpy(iso9660_vrs_desc->identifier, "NSR03", 5);
926 		;
927 		if ((error = udf_write_sector(iso9660_vrs_desc, pos)))
928 			return error;
929 		pos += dpos;
930 
931 		memcpy(iso9660_vrs_desc->identifier, "TEA01", 5);
932 		if ((error = udf_write_sector(iso9660_vrs_desc, pos)))
933 			return error;
934 	}
935 
936 	/* return success */
937 	return 0;
938 }
939 
940 
941 /* --------------------------------------------------------------------- */
942 
943 /*
944  * Main function that creates and writes out disc contents based on the
945  * format_flags's that uniquely define the type of disc to create.
946  */
947 
948 int
949 udf_do_newfs(void)
950 {
951 	union dscrptr *zero_dscr;
952 	union dscrptr *terminator_dscr;
953 	union dscrptr *root_dscr;
954 	union dscrptr *vat_dscr;
955 	union dscrptr *dscr;
956 	struct mmc_trackinfo ti;
957 	uint32_t sparable_blocks;
958 	uint32_t sector_size, blockingnr;
959 	uint32_t cnt, loc, len;
960 	int sectcopy;
961 	int error, integrity_type;
962 	int data_part, metadata_part;
963 
964 	/* init */
965 	sector_size = mmc_discinfo.sector_size;
966 
967 	/* determine span/size */
968 	ti.tracknr = mmc_discinfo.first_track_last_session;
969 	error = udf_update_trackinfo(&mmc_discinfo, &ti);
970 	if (error)
971 		return error;
972 
973 	if (mmc_discinfo.sector_size < context.sector_size) {
974 		fprintf(stderr, "Impossible to format: sectorsize too small\n");
975 		return EIO;
976 	}
977 	context.sector_size = sector_size;
978 
979 	/* determine blockingnr */
980 	blockingnr = ti.packet_size;
981 	if (blockingnr <= 1) {
982 		/* paranoia on blockingnr */
983 		switch (mmc_discinfo.mmc_profile) {
984 		case 0x09 : /* CD-R    */
985 		case 0x0a : /* CD-RW   */
986 			blockingnr = 32;	/* UDF requirement */
987 			break;
988 		case 0x11 : /* DVD-R (DL) */
989 		case 0x1b : /* DVD+R      */
990 		case 0x2b : /* DVD+R Dual layer */
991 		case 0x13 : /* DVD-RW restricted overwrite */
992 		case 0x14 : /* DVD-RW sequential */
993 			blockingnr = 16;	/* SCSI definition */
994 			break;
995 		case 0x41 : /* BD-R Sequential recording (SRM) */
996 		case 0x51 : /* HD DVD-R   */
997 			blockingnr = 32;	/* SCSI definition */
998 			break;
999 		default:
1000 			break;
1001 		}
1002 
1003 	}
1004 	if (blockingnr <= 0) {
1005 		printf("Can't fixup blockingnumber for device "
1006 			"type %d\n", mmc_discinfo.mmc_profile);
1007 
1008 		printf("Device is not returning valid blocking"
1009 			" number and media type is unknown.\n");
1010 
1011 		return EINVAL;
1012 	}
1013 
1014 	/* setup sector writeout queue's */
1015 	TAILQ_INIT(&write_queue);
1016 	wrtrack_skew = ti.track_start % blockingnr;
1017 
1018 	if (mmc_discinfo.mmc_class == MMC_CLASS_CD) {
1019 		/* not too much for CD-RW, still 20MiB */
1020 		sparable_blocks = 32;
1021 	} else {
1022 		/* take a value for DVD*RW mainly, BD is `defect free' */
1023 		sparable_blocks = 512;
1024 	}
1025 
1026 	/* get layout */
1027 	error = udf_calculate_disc_layout(format_flags, context.min_udf,
1028 		wrtrack_skew,
1029 		ti.track_start, mmc_discinfo.last_possible_lba,
1030 		sector_size, blockingnr, sparable_blocks,
1031 		meta_fract);
1032 
1033 	/* cache partition for we need it often */
1034 	data_part     = context.data_part;
1035 	metadata_part = context.metadata_part;
1036 
1037 	/* Create sparing table descriptor if applicable */
1038 	if (format_flags & FORMAT_SPARABLE) {
1039 		if ((error = udf_create_sparing_tabled()))
1040 			return error;
1041 
1042 		if (check_surface) {
1043 			if ((error = udf_surface_check()))
1044 				return error;
1045 		}
1046 	}
1047 
1048 	/* Create a generic terminator descriptor */
1049 	terminator_dscr = calloc(1, sector_size);
1050 	if (terminator_dscr == NULL)
1051 		return ENOMEM;
1052 	udf_create_terminator(terminator_dscr, 0);
1053 
1054 	/*
1055 	 * Start with wipeout of VRS1 upto start of partition. This allows
1056 	 * formatting for sequentials with the track reservation and it
1057 	 * cleans old rubbish on rewritables. For sequentuals without the
1058 	 * track reservation all is wiped from track start.
1059 	 */
1060 	if ((zero_dscr = calloc(1, context.sector_size)) == NULL)
1061 		return ENOMEM;
1062 
1063 	loc = (format_flags & FORMAT_TRACK512) ? layout.vds1 : ti.track_start;
1064 	for (; loc < layout.part_start_lba; loc++) {
1065 		if ((error = udf_write_sector(zero_dscr, loc)))
1066 			return error;
1067 	}
1068 
1069 	/* Create anchors */
1070 	for (cnt = 0; cnt < 3; cnt++) {
1071 		if ((error = udf_create_anchor(cnt)))
1072 			return error;
1073 	}
1074 
1075 	/*
1076 	 * Create the two Volume Descriptor Sets (VDS) each containing the
1077 	 * following descriptors : primary volume, partition space,
1078 	 * unallocated space, logical volume, implementation use and the
1079 	 * terminator
1080 	 */
1081 
1082 	/* start of volume recognision sequence building */
1083 	context.vds_seq = 0;
1084 
1085 	/* Create primary volume descriptor */
1086 	if ((error = udf_create_primaryd()))
1087 		return error;
1088 
1089 	/* Create partition descriptor */
1090 	if ((error = udf_create_partitiond(context.data_part, media_accesstype)))
1091 		return error;
1092 
1093 	/* Create unallocated space descriptor */
1094 	if ((error = udf_create_unalloc_spaced()))
1095 		return error;
1096 
1097 	/* Create logical volume descriptor */
1098 	if ((error = udf_create_logical_dscr(format_flags)))
1099 		return error;
1100 
1101 	/* Create implementation use descriptor */
1102 	/* TODO input of fields 1,2,3 and passing them */
1103 	if ((error = udf_create_impvold(NULL, NULL, NULL)))
1104 		return error;
1105 
1106 	/* write out what we've created so far */
1107 
1108 	/* writeout iso9660 vrs */
1109 	if ((error = udf_write_iso9660_vrs()))
1110 		return error;
1111 
1112 	/* Writeout anchors */
1113 	for (cnt = 0; cnt < 3; cnt++) {
1114 		dscr = (union dscrptr *) context.anchors[cnt];
1115 		loc  = layout.anchors[cnt];
1116 		if ((error = udf_write_dscr_phys(dscr, loc, 1)))
1117 			return error;
1118 
1119 		/* sequential media has only one anchor */
1120 		if (format_flags & FORMAT_SEQUENTIAL)
1121 			break;
1122 	}
1123 
1124 	/* write out main and secondary VRS */
1125 	for (sectcopy = 1; sectcopy <= 2; sectcopy++) {
1126 		loc = (sectcopy == 1) ? layout.vds1 : layout.vds2;
1127 
1128 		/* primary volume descriptor */
1129 		dscr = (union dscrptr *) context.primary_vol;
1130 		error = udf_write_dscr_phys(dscr, loc, 1);
1131 		if (error)
1132 			return error;
1133 		loc++;
1134 
1135 		/* partition descriptor(s) */
1136 		for (cnt = 0; cnt < UDF_PARTITIONS; cnt++) {
1137 			dscr = (union dscrptr *) context.partitions[cnt];
1138 			if (dscr) {
1139 				error = udf_write_dscr_phys(dscr, loc, 1);
1140 				if (error)
1141 					return error;
1142 				loc++;
1143 			}
1144 		}
1145 
1146 		/* unallocated space descriptor */
1147 		dscr = (union dscrptr *) context.unallocated;
1148 		error = udf_write_dscr_phys(dscr, loc, 1);
1149 		if (error)
1150 			return error;
1151 		loc++;
1152 
1153 		/* logical volume descriptor */
1154 		dscr = (union dscrptr *) context.logical_vol;
1155 		error = udf_write_dscr_phys(dscr, loc, 1);
1156 		if (error)
1157 			return error;
1158 		loc++;
1159 
1160 		/* implementation use descriptor */
1161 		dscr = (union dscrptr *) context.implementation;
1162 		error = udf_write_dscr_phys(dscr, loc, 1);
1163 		if (error)
1164 			return error;
1165 		loc++;
1166 
1167 		/* terminator descriptor */
1168 		error = udf_write_dscr_phys(terminator_dscr, loc, 1);
1169 		if (error)
1170 			return error;
1171 		loc++;
1172 	}
1173 
1174 	/* writeout the two sparable table descriptors (if needed) */
1175 	if (format_flags & FORMAT_SPARABLE) {
1176 		for (sectcopy = 1; sectcopy <= 2; sectcopy++) {
1177 			loc  = (sectcopy == 1) ? layout.spt_1 : layout.spt_2;
1178 			dscr = (union dscrptr *) context.sparing_table;
1179 			len  = layout.sparing_table_dscr_lbas;
1180 
1181 			/* writeout */
1182 			error = udf_write_dscr_phys(dscr, loc, len);
1183 			if (error)
1184 				return error;
1185 		}
1186 	}
1187 
1188 	/*
1189 	 * Create unallocated space bitmap descriptor. Sequential recorded
1190 	 * media report their own free/used space; no free/used space tables
1191 	 * should be recorded for these.
1192 	 */
1193 	if ((format_flags & FORMAT_SEQUENTIAL) == 0) {
1194 		error = udf_create_space_bitmap(
1195 				layout.alloc_bitmap_dscr_size,
1196 				layout.part_size_lba,
1197 				&context.part_unalloc_bits[data_part]);
1198 		if (error)
1199 			return error;
1200 		/* TODO: freed space bitmap if applicable */
1201 
1202 		/* mark space allocated for the unallocated space bitmap */
1203 		udf_mark_allocated(layout.unalloc_space, data_part,
1204 			layout.alloc_bitmap_dscr_size);
1205 	}
1206 
1207 	/*
1208 	 * Create metadata partition file entries and allocate and init their
1209 	 * space and free space maps.
1210 	 */
1211 	if (format_flags & FORMAT_META) {
1212 		error = udf_create_space_bitmap(
1213 				layout.meta_bitmap_dscr_size,
1214 				layout.meta_part_size_lba,
1215 				&context.part_unalloc_bits[metadata_part]);
1216 		if (error)
1217 			return error;
1218 
1219 		error = udf_create_meta_files();
1220 		if (error)
1221 			return error;
1222 
1223 		/* mark space allocated for meta partition and its bitmap */
1224 		udf_mark_allocated(layout.meta_file,   data_part, 1);
1225 		udf_mark_allocated(layout.meta_mirror, data_part, 1);
1226 		udf_mark_allocated(layout.meta_bitmap, data_part, 1);
1227 		udf_mark_allocated(layout.meta_part_start_lba, data_part,
1228 			layout.meta_part_size_lba);
1229 
1230 		/* mark space allocated for the unallocated space bitmap */
1231 		udf_mark_allocated(layout.meta_bitmap_space, data_part,
1232 			layout.meta_bitmap_dscr_size);
1233 	}
1234 
1235 	/* create logical volume integrity descriptor */
1236 	context.num_files = 0;
1237 	context.num_directories = 0;
1238 	integrity_type = UDF_INTEGRITY_OPEN;
1239 	if ((error = udf_create_lvintd(integrity_type)))
1240 		return error;
1241 
1242 	/* create FSD */
1243 	if ((error = udf_create_fsd()))
1244 		return error;
1245 	udf_mark_allocated(layout.fsd, metadata_part, 1);
1246 
1247 	/* create root directory */
1248 	assert(context.unique_id == 0x10);
1249 	context.unique_id = 0;
1250 	if ((error = udf_create_new_rootdir(&root_dscr)))
1251 		return error;
1252 	udf_mark_allocated(layout.rootdir, metadata_part, 1);
1253 
1254 	/* writeout FSD + rootdir */
1255 	dscr = (union dscrptr *) context.fileset_desc;
1256 	error = udf_write_dscr_virt(dscr, layout.fsd, metadata_part, 1);
1257 	if (error)
1258 		return error;
1259 
1260 	error = udf_write_dscr_virt(root_dscr, layout.rootdir, metadata_part, 1);
1261 	if (error)
1262 		return error;
1263 
1264 	/* writeout initial open integrity sequence + terminator */
1265 	loc = layout.lvis;
1266 	dscr = (union dscrptr *) context.logvol_integrity;
1267 	error = udf_write_dscr_phys(dscr, loc, 1);
1268 	if (error)
1269 		return error;
1270 	loc++;
1271 	error = udf_write_dscr_phys(terminator_dscr, loc, 1);
1272 	if (error)
1273 		return error;
1274 
1275 
1276 	/* XXX the place to add more files */
1277 
1278 
1279 	if ((format_flags & FORMAT_SEQUENTIAL) == 0) {
1280 		/* update lvint and mark it closed */
1281 		udf_update_lvintd(UDF_INTEGRITY_CLOSED);
1282 
1283 		/* overwrite initial terminator */
1284 		loc = layout.lvis+1;
1285 		dscr = (union dscrptr *) context.logvol_integrity;
1286 		error = udf_write_dscr_phys(dscr, loc, 1);
1287 		if (error)
1288 			return error;
1289 		loc++;
1290 
1291 		/* mark end of integrity desciptor sequence again */
1292 		error = udf_write_dscr_phys(terminator_dscr, loc, 1);
1293 		if (error)
1294 			return error;
1295 	}
1296 
1297 	/* write out unallocated space bitmap on non sequential media */
1298 	if ((format_flags & FORMAT_SEQUENTIAL) == 0) {
1299 		/* writeout unallocated space bitmap */
1300 		loc  = layout.unalloc_space;
1301 		dscr = (union dscrptr *) (context.part_unalloc_bits[data_part]);
1302 		len  = layout.alloc_bitmap_dscr_size;
1303 		error = udf_write_dscr_virt(dscr, loc, data_part, len);
1304 		if (error)
1305 			return error;
1306 	}
1307 
1308 	if (format_flags & FORMAT_META) {
1309 		loc = layout.meta_file;
1310 		dscr = (union dscrptr *) context.meta_file;
1311 		error = udf_write_dscr_virt(dscr, loc, data_part, 1);
1312 		if (error)
1313 			return error;
1314 
1315 		loc = layout.meta_mirror;
1316 		dscr = (union dscrptr *) context.meta_mirror;
1317 		error = udf_write_dscr_virt(dscr, loc, data_part, 1);
1318 		if (error)
1319 			return error;
1320 
1321 		loc = layout.meta_bitmap;
1322 		dscr = (union dscrptr *) context.meta_bitmap;
1323 		error = udf_write_dscr_virt(dscr, loc, data_part, 1);
1324 		if (error)
1325 			return error;
1326 
1327 		/* writeout unallocated space bitmap */
1328 		loc  = layout.meta_bitmap_space;
1329 		dscr = (union dscrptr *) (context.part_unalloc_bits[metadata_part]);
1330 		len  = layout.meta_bitmap_dscr_size;
1331 		error = udf_write_dscr_virt(dscr, loc, data_part, len);
1332 		if (error)
1333 			return error;
1334 	}
1335 
1336 	/* create a VAT and account for FSD+root */
1337 	vat_dscr = NULL;
1338 	if (format_flags & FORMAT_VAT) {
1339 		/* update lvint to reflect the newest values (no writeout) */
1340 		udf_update_lvintd(UDF_INTEGRITY_CLOSED);
1341 
1342 		error = udf_create_new_VAT(&vat_dscr);
1343 		if (error)
1344 			return error;
1345 
1346 		loc = layout.vat;
1347 		error = udf_write_dscr_virt(vat_dscr, loc, metadata_part, 1);
1348 		if (error)
1349 			return error;
1350 	}
1351 
1352 	/* write out sectors */
1353 	if ((error = writeout_write_queue()))
1354 		return error;
1355 
1356 	/* done */
1357 	return 0;
1358 }
1359 
1360 /* --------------------------------------------------------------------- */
1361 
1362 /* version can be specified as 0xabc or a.bc */
1363 static int
1364 parse_udfversion(const char *pos, uint32_t *version) {
1365 	int hex = 0;
1366 	char c1, c2, c3, c4;
1367 
1368 	*version = 0;
1369 	if (*pos == '0') {
1370 		pos++;
1371 		/* expect hex format */
1372 		hex = 1;
1373 		if (*pos++ != 'x')
1374 			return 1;
1375 	}
1376 
1377 	c1 = *pos++;
1378 	if (c1 < '0' || c1 > '9')
1379 		return 1;
1380 	c1 -= '0';
1381 
1382 	c2 = *pos++;
1383 	if (!hex) {
1384 		if (c2 != '.')
1385 			return 1;
1386 		c2 = *pos++;
1387 	}
1388 	if (c2 < '0' || c2 > '9')
1389 		return 1;
1390 	c2 -= '0';
1391 
1392 	c3 = *pos++;
1393 	if (c3 < '0' || c3 > '9')
1394 		return 1;
1395 	c3 -= '0';
1396 
1397 	c4 = *pos++;
1398 	if (c4 != 0)
1399 		return 1;
1400 
1401 	*version = c1 * 0x100 + c2 * 0x10 + c3;
1402 	return 0;
1403 }
1404 
1405 
1406 static int
1407 a_udf_version(const char *s, const char *id_type)
1408 {
1409 	uint32_t version;
1410 
1411 	if (parse_udfversion(s, &version))
1412 		errx(1, "unknown %s id %s; specify as hex or float", id_type, s);
1413 	return version;
1414 }
1415 
1416 /* --------------------------------------------------------------------- */
1417 
1418 static void
1419 usage(void)
1420 {
1421 	(void)fprintf(stderr, "Usage: %s [-cFM] [-L loglabel] "
1422 	    "[-P discid] [-S setlabel] [-s size] [-p perc] "
1423 	    "[-t gmtoff] [-v min_udf] [-V max_udf] special\n", getprogname());
1424 	exit(EXIT_FAILURE);
1425 }
1426 
1427 
1428 int
1429 main(int argc, char **argv)
1430 {
1431 	struct tm *tm;
1432 	struct stat st;
1433 	time_t now;
1434 	char  scrap[255];
1435 	int ch, req_enable, req_disable, force;
1436 	int error;
1437 
1438 	setprogname(argv[0]);
1439 
1440 	/* initialise */
1441 	format_str    = strdup("");
1442 	req_enable    = req_disable = 0;
1443 	format_flags  = FORMAT_INVALID;
1444 	force         = 0;
1445 	check_surface = 0;
1446 
1447 	srandom((unsigned long) time(NULL));
1448 	udf_init_create_context();
1449 	context.app_name  = APP_NAME;
1450 	context.impl_name = IMPL_NAME;
1451 	context.app_version_main = APP_VERSION_MAIN;
1452 	context.app_version_sub  = APP_VERSION_SUB;
1453 
1454 	/* minimum and maximum UDF versions we advise */
1455 	context.min_udf = 0x201;
1456 	context.max_udf = 0x201;
1457 
1458 	/* use user's time zone as default */
1459 	(void)time(&now);
1460 	tm = localtime(&now);
1461 	context.gmtoff = tm->tm_gmtoff;
1462 
1463 	/* process options */
1464 	while ((ch = getopt(argc, argv, "cFL:Mp:P:s:S:t:v:V:")) != -1) {
1465 		switch (ch) {
1466 		case 'c' :
1467 			check_surface = 1;
1468 			break;
1469 		case 'F' :
1470 			force = 1;
1471 			break;
1472 		case 'L' :
1473 			if (context.logvol_name) free(context.logvol_name);
1474 			context.logvol_name = strdup(optarg);
1475 			break;
1476 		case 'M' :
1477 			req_disable |= FORMAT_META;
1478 			break;
1479 		case 'p' :
1480 			meta_perc = a_num(optarg, "meta_perc");
1481 			/* limit to `sensible` values */
1482 			meta_perc = MIN(meta_perc, 99);
1483 			meta_perc = MAX(meta_perc, 1);
1484 			meta_fract = (float) meta_perc/100.0;
1485 			break;
1486 		case 'v' :
1487 			context.min_udf = a_udf_version(optarg, "min_udf");
1488 			if (context.min_udf > context.max_udf)
1489 				context.max_udf = context.min_udf;
1490 			break;
1491 		case 'V' :
1492 			context.max_udf = a_udf_version(optarg, "max_udf");
1493 			if (context.min_udf > context.max_udf)
1494 				context.min_udf = context.max_udf;
1495 			break;
1496 		case 'P' :
1497 			context.primary_name = strdup(optarg);
1498 			break;
1499 		case 's' :
1500 			/* TODO size argument; recordable emulation */
1501 			break;
1502 		case 'S' :
1503 			if (context.volset_name) free(context.volset_name);
1504 			context.volset_name = strdup(optarg);
1505 			break;
1506 		case 't' :
1507 			/* time zone overide */
1508 			context.gmtoff = a_num(optarg, "gmtoff");
1509 			break;
1510 		default  :
1511 			usage();
1512 			/* NOTREACHED */
1513 		}
1514 	}
1515 
1516 	if (optind + 1 != argc)
1517 		usage();
1518 
1519 	/* get device and directory specifier */
1520 	dev = argv[optind];
1521 
1522 	/* open device */
1523 	if ((fd = open(dev, O_RDWR, 0)) == -1) {
1524 		perror("can't open device");
1525 		return EXIT_FAILURE;
1526 	}
1527 
1528 	/* stat the device */
1529 	if (fstat(fd, &st) != 0) {
1530 		perror("can't stat the device");
1531 		close(fd);
1532 		return EXIT_FAILURE;
1533 	}
1534 
1535 	/* Formatting can only be done on raw devices */
1536 	if (!S_ISCHR(st.st_mode)) {
1537 		printf("%s is not a raw device\n", dev);
1538 		close(fd);
1539 		return EXIT_FAILURE;
1540 	}
1541 
1542 	/* just in case something went wrong, synchronise the drive's cache */
1543 	udf_synchronise_caches();
1544 
1545 	/* get disc information */
1546 	error = udf_update_discinfo(&mmc_discinfo);
1547 	if (error) {
1548 		perror("can't retrieve discinfo");
1549 		close(fd);
1550 		return EXIT_FAILURE;
1551 	}
1552 
1553 	/* derive disc identifiers when not specified and check given */
1554 	error = udf_proces_names();
1555 	if (error) {
1556 		/* error message has been printed */
1557 		close(fd);
1558 		return EXIT_FAILURE;
1559 	}
1560 
1561 	/* derive newfs disc format from disc profile */
1562 	error = udf_derive_format(req_enable, req_disable, force);
1563 	if (error)  {
1564 		/* error message has been printed */
1565 		close(fd);
1566 		return EXIT_FAILURE;
1567 	}
1568 
1569 	udf_dump_discinfo(&mmc_discinfo);
1570 	printf("Formatting disc compatible with UDF version %x to %x\n\n",
1571 		context.min_udf, context.max_udf);
1572 	(void)snprintb(scrap, sizeof(scrap), FORMAT_FLAGBITS,
1573 	    (uint64_t) format_flags);
1574 	printf("UDF properties       %s\n", scrap);
1575 	printf("Volume set          `%s'\n", context.volset_name);
1576 	printf("Primary volume      `%s`\n", context.primary_name);
1577 	printf("Logical volume      `%s`\n", context.logvol_name);
1578 	if (format_flags & FORMAT_META)
1579 		printf("Metadata percentage  %d %%\n", meta_perc);
1580 	printf("\n");
1581 
1582 	/* prepare disc if nessisary (recordables mainly) */
1583 	error = udf_prepare_disc();
1584 	if (error) {
1585 		perror("preparing disc failed");
1586 		close(fd);
1587 		return EXIT_FAILURE;
1588 	};
1589 
1590 	/* set up administration */
1591 	error = udf_do_newfs();
1592 
1593 	/* in any case, synchronise the drive's cache to prevent lockups */
1594 	udf_synchronise_caches();
1595 
1596 	close(fd);
1597 	if (error)
1598 		return EXIT_FAILURE;
1599 
1600 	return EXIT_SUCCESS;
1601 }
1602 
1603 /* --------------------------------------------------------------------- */
1604 
1605