Home | History | Annotate | Line # | Download | only in apmlabel
apmlabel.c revision 1.2 
      1 /*	$NetBSD: apmlabel.c,v 1.2 2011/08/27 16:10:51 joerg Exp $	*/
      2 
      3 /*
      4  * Copyright (C) 1998 Wolfgang Solfrank.
      5  * Copyright (C) 1998 TooLs GmbH.
      6  * All rights reserved.
      7  *
      8  * Redistribution and use in source and binary forms, with or without
      9  * modification, are permitted provided that the following conditions
     10  * are met:
     11  * 1. Redistributions of source code must retain the above copyright
     12  *    notice, this list of conditions and the following disclaimer.
     13  * 2. Redistributions in binary form must reproduce the above copyright
     14  *    notice, this list of conditions and the following disclaimer in the
     15  *    documentation and/or other materials provided with the distribution.
     16  * 3. All advertising materials mentioning features or use of this software
     17  *    must display the following acknowledgement:
     18  *	This product includes software developed by TooLs GmbH.
     19  * 4. The name of TooLs GmbH may not be used to endorse or promote products
     20  *    derived from this software without specific prior written permission.
     21  *
     22  * THIS SOFTWARE IS PROVIDED BY TOOLS GMBH ``AS IS'' AND ANY EXPRESS OR
     23  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
     24  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
     25  * IN NO EVENT SHALL TOOLS GMBH BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
     26  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
     27  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
     28  * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
     29  * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
     30  * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
     31  * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     32  */
     33 
     34 #include <sys/cdefs.h>
     35 #ifndef lint
     36 __RCSID("$NetBSD: apmlabel.c,v 1.2 2011/08/27 16:10:51 joerg Exp $");
     37 #endif /* not lint */
     38 
     39 #include <stdio.h>
     40 #include <err.h>
     41 #include <errno.h>
     42 #include <fcntl.h>
     43 #include <limits.h>
     44 #include <stdlib.h>
     45 #include <string.h>
     46 #include <unistd.h>
     47 #include <util.h>
     48 
     49 #include <sys/param.h>
     50 #define FSTYPENAMES
     51 #include <sys/disklabel.h>
     52 #include <sys/bootblock.h>
     53 #include <sys/ioctl.h>
     54 
     55 #include "dkcksum.h"
     56 #include "extern.h"
     57 
     58 __dead static void	usage(void);
     59 static void	getlabel(int);
     60 static void	setlabel(int, int);
     61 static int	getparts(int, int);
     62 static struct apple_drvr_map *convert_drvr_map(unsigned char *);
     63 static struct apple_part_map_entry *convert_part_map_entry(unsigned char *);
     64 
     65 static struct disklabel label;
     66 
     67 static void
     68 getlabel(int sd)
     69 {
     70 
     71 	if (ioctl(sd, DIOCGDINFO, &label) < 0) {
     72 		perror("get label");
     73 		exit(1);
     74 	}
     75 	/*
     76 	 * Some ports seem to not set the number of partitions
     77 	 * correctly, albeit they seem to set the raw partition ok!
     78 	 */
     79 	if (label.d_npartitions <= getrawpartition())
     80 		label.d_npartitions = getrawpartition() + 1;
     81 }
     82 
     83 static void
     84 setlabel(int sd, int doraw)
     85 {
     86 	int one = 1;
     87 
     88 	label.d_checksum = 0;
     89 	label.d_checksum = dkcksum(&label);
     90 	if (ioctl(sd, doraw ? DIOCWDINFO : DIOCSDINFO, &label) < 0) {
     91 		perror("set label");
     92 		exit(1);
     93 	}
     94 	if (!doraw)
     95 		/* If we haven't written to the disk, don't discard on close */
     96 		ioctl(sd, DIOCKLABEL, &one);
     97 
     98 }
     99 
    100 static int
    101 getparts(int sd, int verbose)
    102 {
    103 	unsigned char		buf[DEV_BSIZE];
    104 	struct apple_drvr_map	*drvr;
    105 	struct apple_part_map_entry *part;
    106 	struct partition	npe;
    107 	uint16_t		blksize, blkcnt, partcnt;
    108 	int			i, j, unused, changed;
    109 	uint64_t		temp;
    110 
    111 	changed = 0;
    112 
    113 	if (lseek(sd, 0, SEEK_SET) == -1) {
    114 		perror("seek drvr map");
    115 		exit(1);
    116 	}
    117 	if ((i=read(sd, buf, sizeof(buf))) != DEV_BSIZE) {
    118 		perror("read drvr map");
    119 		exit(1);
    120 	}
    121 	drvr = convert_drvr_map(buf);
    122 
    123 	if (drvr->sbSig != APPLE_DRVR_MAP_MAGIC)
    124 		return (changed);
    125 	blksize = drvr->sbBlockSize;
    126 	blkcnt = drvr->sbBlkCount;
    127 
    128 	partcnt = 1;
    129 
    130 	for (i = 0; i < partcnt; i++) {
    131 		if (lseek(sd, (i+1)*blksize, SEEK_SET) == -1) {
    132 			perror("seek part");
    133 			exit(1);
    134 		}
    135 		if (read(sd, buf, sizeof(buf)) != DEV_BSIZE) {
    136 			perror("read part");
    137 			exit(1);
    138 		}
    139 
    140 		part = convert_part_map_entry(buf);
    141 
    142 		if (part->pmSig != APPLE_PART_MAP_ENTRY_MAGIC)
    143 			return (changed);
    144 		if (i == 0)
    145 			partcnt = part->pmMapBlkCnt;
    146 		/* XXX: consistency checks? */
    147 
    148 		memset((void *)&npe, 0, sizeof(npe));
    149 
    150 		if (strcasecmp((char *)part->pmPartType,
    151 			APPLE_PART_TYPE_MAC) == 0
    152 		    || strcasecmp((char *)part->pmPartType, "Apple_HFSX") == 0)
    153 			npe.p_fstype = FS_HFS;
    154 		else if (strcasecmp((char *)part->pmPartType,
    155 			     "Apple_UFS") == 0) {
    156 			npe.p_fstype = FS_APPLEUFS;
    157 			npe.p_size = 16384;	/* XXX */
    158 			npe.p_fsize = 1024;
    159 			npe.p_frag = 8;
    160 			npe.p_cpg = 16;
    161 		}
    162 		else
    163 			continue;
    164 
    165 		temp = (uint64_t)part->pmDataCnt * (uint64_t)blksize;
    166 		if (temp % label.d_secsize != 0) {
    167 		    warnx("partition size not multiple of sector size"
    168 			  ", skipping");
    169 		    continue;
    170 		}
    171 		npe.p_size = temp / label.d_secsize;
    172 		temp = (uint64_t)(part->pmPyPartStart + part->pmLgDataStart)
    173 		    * (uint64_t)blksize;
    174 		if (temp % label.d_secsize != 0) {
    175 		    warnx("partition offset not multiple of sector size"
    176 			  ", skipping");
    177 		    continue;
    178 		}
    179 		npe.p_offset = temp / label.d_secsize;
    180 
    181 				/* find existing entry, or first free slot */
    182 		unused = -1;	/* flag as no free slot */
    183 		if (verbose)
    184 			printf(
    185 			    "Found %s partition; size %u (%u MB), offset %u\n",
    186 			    fstypenames[npe.p_fstype],
    187 			    npe.p_size, npe.p_size / 2048, npe.p_offset);
    188 		for (j = 0; j < label.d_npartitions; j++) {
    189 			struct partition *lpe;
    190 
    191 			if (j == RAW_PART)
    192 				continue;
    193 			lpe = &label.d_partitions[j];
    194 			if (lpe->p_size == npe.p_size &&
    195 			    lpe->p_offset == npe.p_offset
    196 #ifdef notyet
    197 			    && (lpe->p_fstype == npe.p_fstype ||
    198 			     lpe->p_fstype == FS_UNUSED) */
    199 #endif
    200 			     ) {
    201 				if (verbose)
    202 					printf(
    203 			    "  skipping existing %s partition at slot %c.\n",
    204 					    fstypenames[lpe->p_fstype],
    205 					    j + 'a');
    206 				unused = -2;	/* flag as existing */
    207 				break;
    208 			}
    209 			if (unused == -1 && lpe->p_size == 0 &&
    210 			    lpe->p_fstype == FS_UNUSED)
    211 				unused = j;
    212 		}
    213 		if (unused == -2)
    214 			continue;	/* entry exists, skip... */
    215 		if (unused == -1) {
    216 			if (label.d_npartitions < MAXPARTITIONS) {
    217 				unused = label.d_npartitions;
    218 				label.d_npartitions++;
    219 			} else {
    220 				printf(
    221 				"  WARNING: no slots free for %s partition.\n",
    222 				    fstypenames[npe.p_fstype]);
    223 				continue;
    224 			}
    225 		}
    226 
    227 		if (verbose)
    228 			printf("  adding %s partition to slot %c.\n",
    229 			    fstypenames[npe.p_fstype], unused + 'a');
    230 		changed++;
    231 		label.d_partitions[unused] = npe;
    232 	}
    233 
    234 	return (changed);
    235 }
    236 
    237 static struct apple_drvr_map *
    238 convert_drvr_map(unsigned char *buf)
    239 {
    240 	struct apple_drvr_map *drvr;
    241 	int i;
    242 
    243 	drvr = (struct apple_drvr_map *)buf;
    244 
    245 	BE16TOH(drvr->sbSig);
    246 	BE16TOH(drvr->sbBlockSize);
    247 	BE32TOH(drvr->sbBlkCount);
    248 	BE16TOH(drvr->sbDevType);
    249 	BE16TOH(drvr->sbDevID);
    250 	BE32TOH(drvr->sbData);
    251 	BE16TOH(drvr->sbDrvrCount);
    252 	for (i=0; i<APPLE_DRVR_MAP_MAX_DESCRIPTORS; i++) {
    253 		BE32TOH(drvr->sb_dd[i].descBlock);
    254 		BE16TOH(drvr->sb_dd[i].descSize);
    255 		BE16TOH(drvr->sb_dd[i].descType);
    256 	}
    257 
    258 	return drvr;
    259 }
    260 
    261 static struct apple_part_map_entry *
    262 convert_part_map_entry(unsigned char *buf)
    263 {
    264 	struct apple_part_map_entry *part;
    265 
    266 	part = (struct apple_part_map_entry *)buf;
    267 
    268 	BE16TOH(part->pmSig);
    269 	BE16TOH(part->pmSigPad);
    270 	BE32TOH(part->pmMapBlkCnt);
    271 	BE32TOH(part->pmPyPartStart);
    272 	BE32TOH(part->pmPartBlkCnt);
    273 	BE32TOH(part->pmLgDataStart);
    274 	BE32TOH(part->pmDataCnt);
    275 	BE32TOH(part->pmPartStatus);
    276 	BE32TOH(part->pmLgBootStart);
    277 	BE32TOH(part->pmBootSize);
    278 	BE32TOH(part->pmBootLoad);
    279 	BE32TOH(part->pmBootLoad2);
    280 	BE32TOH(part->pmBootEntry);
    281 	BE32TOH(part->pmBootEntry2);
    282 	BE32TOH(part->pmBootCksum);
    283 
    284 	return part;
    285 }
    286 
    287 static void
    288 usage(void)
    289 {
    290 	fprintf(stderr, "usage: %s [-fqrw] rawdisk\n",
    291 	    getprogname());
    292 	exit(1);
    293 }
    294 
    295 
    296 int
    297 main(int argc, char **argv)
    298 {
    299 	int	sd, ch, changed;
    300 	char	name[MAXPATHLEN];
    301 	int	force;			/* force label update */
    302 	int	raw;			/* update on-disk label as well */
    303 	int	verbose;		/* verbose output */
    304 	int	write_it;		/* update in-core label if changed */
    305 
    306 	force = 0;
    307 	raw = 0;
    308 	verbose = 1;
    309 	write_it = 0;
    310 	while ((ch = getopt(argc, argv, "fqrw")) != -1) {
    311 		switch (ch) {
    312 		case 'f':
    313 			force = 1;
    314 			break;
    315 		case 'q':
    316 			verbose = 0;
    317 			break;
    318 		case 'r':
    319 			raw = 1;
    320 			break;
    321 		case 'w':
    322 			write_it = 1;
    323 			break;
    324 		default:
    325 			usage();
    326 		}
    327 	}
    328 	argc -= optind;
    329 	argv += optind;
    330 	if (argc != 1)
    331 		usage();
    332 
    333 	if ((sd = opendisk(argv[0], write_it ? O_RDWR : O_RDONLY, name,
    334 	    (size_t)MAXPATHLEN, 1)) < 0) {
    335 		perror(argv[0]);
    336 		exit(1);
    337 	}
    338 	getlabel(sd);
    339 	changed = getparts(sd, verbose);
    340 
    341 	if (verbose) {
    342 		putchar('\n');
    343 		showpartitions(stdout, &label, 0);
    344 		putchar('\n');
    345 	}
    346 	if (write_it) {
    347 		if (! changed && ! force)
    348 			printf("No change; not updating disk label.\n");
    349 		else {
    350 			if (verbose)
    351 				printf("Updating in-core %sdisk label.\n",
    352 				    raw ? "and on-disk " : "");
    353 			setlabel(sd, raw);
    354 		}
    355 	} else {
    356 		printf("Not updating disk label.\n");
    357 	}
    358 	close(sd);
    359 	return (0);
    360 }
    361