/* $NetBSD: apmlabel.c,v 1.1.1.1 2007/03/05 23:06:53 dillo Exp $ */ /* * Copyright (C) 1998 Wolfgang Solfrank. * Copyright (C) 1998 TooLs GmbH. * All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions * are met: * 1. Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * 2. Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * 3. All advertising materials mentioning features or use of this software * must display the following acknowledgement: * This product includes software developed by TooLs GmbH. * 4. The name of TooLs GmbH may not be used to endorse or promote products * derived from this software without specific prior written permission. * * THIS SOFTWARE IS PROVIDED BY TOOLS GMBH ``AS IS'' AND ANY EXPRESS OR * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. * IN NO EVENT SHALL TOOLS GMBH BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */ #include #ifndef lint __RCSID("$NetBSD: apmlabel.c,v 1.1.1.1 2007/03/05 23:06:53 dillo Exp $"); #endif /* not lint */ #include #include #include #include #include #include #include #include #include #include #define FSTYPENAMES #include #include #include #include "dkcksum.h" #include "extern.h" int main(int, char **); void usage(void); void getlabel(int); void setlabel(int, int); int getparts(int, int); struct apple_drvr_map *convert_drvr_map(unsigned char *); struct apple_part_map_entry *convert_part_map_entry(unsigned char *); struct disklabel label; void getlabel(int sd) { if (ioctl(sd, DIOCGDINFO, &label) < 0) { perror("get label"); exit(1); } /* * Some ports seem to not set the number of partitions * correctly, albeit they seem to set the raw partition ok! */ if (label.d_npartitions <= getrawpartition()) label.d_npartitions = getrawpartition() + 1; } void setlabel(int sd, int doraw) { int one = 1; label.d_checksum = 0; label.d_checksum = dkcksum(&label); if (ioctl(sd, doraw ? DIOCWDINFO : DIOCSDINFO, &label) < 0) { perror("set label"); exit(1); } if (!doraw) /* If we haven't written to the disk, don't discard on close */ ioctl(sd, DIOCKLABEL, &one); } int getparts(int sd, int verbose) { unsigned char buf[DEV_BSIZE]; struct apple_drvr_map *drvr; struct apple_part_map_entry *part; struct partition npe; uint16_t blksize, blkcnt, partcnt; int i, j, unused, changed; uint64_t temp; changed = 0; if (lseek(sd, 0, SEEK_SET) == -1) { perror("seek drvr map"); exit(1); } if ((i=read(sd, buf, sizeof(buf))) != DEV_BSIZE) { perror("read drvr map"); exit(1); } drvr = convert_drvr_map(buf); if (drvr->sbSig != APPLE_DRVR_MAP_MAGIC) return (changed); blksize = drvr->sbBlockSize; blkcnt = drvr->sbBlkCount; partcnt = 1; for (i = 0; i < partcnt; i++) { if (lseek(sd, (i+1)*blksize, SEEK_SET) == -1) { perror("seek part"); exit(1); } if (read(sd, buf, sizeof(buf)) != DEV_BSIZE) { perror("read part"); exit(1); } part = convert_part_map_entry(buf); if (part->pmSig != APPLE_PART_MAP_ENTRY_MAGIC) return (changed); if (i == 0) partcnt = part->pmMapBlkCnt; /* XXX: consistency checks? */ memset((void *)&npe, 0, sizeof(npe)); if (strcasecmp((char *)part->pmPartType, APPLE_PART_TYPE_MAC) == 0 || strcasecmp((char *)part->pmPartType, "Apple_HFSX") == 0) npe.p_fstype = FS_HFS; else if (strcasecmp((char *)part->pmPartType, "Apple_UFS") == 0) { npe.p_fstype = FS_APPLEUFS; npe.p_size = 16384; /* XXX */ npe.p_fsize = 1024; npe.p_frag = 8; npe.p_cpg = 16; } else continue; temp = (uint64_t)part->pmDataCnt * (uint64_t)blksize; if (temp % label.d_secsize != 0) { warnx("partition size not multiple of sector size" ", skipping"); continue; } npe.p_size = temp / label.d_secsize; temp = (uint64_t)(part->pmPyPartStart + part->pmLgDataStart) * (uint64_t)blksize; if (temp % label.d_secsize != 0) { warnx("partition offset not multiple of sector size" ", skipping"); continue; } npe.p_offset = temp / label.d_secsize; /* find existing entry, or first free slot */ unused = -1; /* flag as no free slot */ if (verbose) printf( "Found %s partition; size %u (%u MB), offset %u\n", fstypenames[npe.p_fstype], npe.p_size, npe.p_size / 2048, npe.p_offset); for (j = 0; j < label.d_npartitions; j++) { struct partition *lpe; if (j == RAW_PART) continue; lpe = &label.d_partitions[j]; if (lpe->p_size == npe.p_size && lpe->p_offset == npe.p_offset #ifdef notyet && (lpe->p_fstype == npe.p_fstype || lpe->p_fstype == FS_UNUSED) */ #endif ) { if (verbose) printf( " skipping existing %s partition at slot %c.\n", fstypenames[lpe->p_fstype], j + 'a'); unused = -2; /* flag as existing */ break; } if (unused == -1 && lpe->p_size == 0 && lpe->p_fstype == FS_UNUSED) unused = j; } if (unused == -2) continue; /* entry exists, skip... */ if (unused == -1) { if (label.d_npartitions < MAXPARTITIONS) { unused = label.d_npartitions; label.d_npartitions++; } else { printf( " WARNING: no slots free for %s partition.\n", fstypenames[npe.p_fstype]); continue; } } if (verbose) printf(" adding %s partition to slot %c.\n", fstypenames[npe.p_fstype], unused + 'a'); changed++; label.d_partitions[unused] = npe; } return (changed); } struct apple_drvr_map * convert_drvr_map(unsigned char *buf) { struct apple_drvr_map *drvr; int i; drvr = (struct apple_drvr_map *)buf; BE16TOH(drvr->sbSig); BE16TOH(drvr->sbBlockSize); BE32TOH(drvr->sbBlkCount); BE16TOH(drvr->sbDevType); BE16TOH(drvr->sbDevID); BE32TOH(drvr->sbData); BE16TOH(drvr->sbDrvrCount); for (i=0; isb_dd[i].descBlock); BE16TOH(drvr->sb_dd[i].descSize); BE16TOH(drvr->sb_dd[i].descType); } return drvr; } struct apple_part_map_entry * convert_part_map_entry(unsigned char *buf) { struct apple_part_map_entry *part; part = (struct apple_part_map_entry *)buf; BE16TOH(part->pmSig); BE16TOH(part->pmSigPad); BE32TOH(part->pmMapBlkCnt); BE32TOH(part->pmPyPartStart); BE32TOH(part->pmPartBlkCnt); BE32TOH(part->pmLgDataStart); BE32TOH(part->pmDataCnt); BE32TOH(part->pmPartStatus); BE32TOH(part->pmLgBootStart); BE32TOH(part->pmBootSize); BE32TOH(part->pmBootLoad); BE32TOH(part->pmBootLoad2); BE32TOH(part->pmBootEntry); BE32TOH(part->pmBootEntry2); BE32TOH(part->pmBootCksum); return part; } void usage(void) { fprintf(stderr, "usage: %s [-fqrw] rawdisk\n", getprogname()); exit(1); } int main(int argc, char **argv) { int sd, ch, changed; char name[MAXPATHLEN]; int force; /* force label update */ int raw; /* update on-disk label as well */ int verbose; /* verbose output */ int write_it; /* update in-core label if changed */ force = 0; raw = 0; verbose = 1; write_it = 0; while ((ch = getopt(argc, argv, "fqrw")) != -1) { switch (ch) { case 'f': force = 1; break; case 'q': verbose = 0; break; case 'r': raw = 1; break; case 'w': write_it = 1; break; default: usage(); } } argc -= optind; argv += optind; if (argc != 1) usage(); if ((sd = opendisk(argv[0], write_it ? O_RDWR : O_RDONLY, name, (size_t)MAXPATHLEN, 1)) < 0) { perror(argv[0]); exit(1); } getlabel(sd); changed = getparts(sd, verbose); if (verbose) { putchar('\n'); showpartitions(stdout, &label, 0); putchar('\n'); } if (write_it) { if (! changed && ! force) printf("No change; not updating disk label.\n"); else { if (verbose) printf("Updating in-core %sdisk label.\n", raw ? "and on-disk " : ""); setlabel(sd, raw); } } else { printf("Not updating disk label.\n"); } close(sd); return (0); }