gen/disklabel.c

   1 /*
   2  * Copyright (c) 1999, 2008 Apple Inc. All rights reserved.
   3  *
   4  * @APPLE_LICENSE_HEADER_START@
   5  *
   6  * This file contains Original Code and/or Modifications of Original Code
   7  * as defined in and that are subject to the Apple Public Source License
   8  * Version 2.0 (the 'License'). You may not use this file except in
   9  * compliance with the License. Please obtain a copy of the License at
  10  * http://www.opensource.apple.com/apsl/ and read it before using this
  11  * file.
  12  *
  13  * The Original Code and all software distributed under the License are
  14  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  15  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  16  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
  18  * Please see the License for the specific language governing rights and
  19  * limitations under the License.
  20  *
  21  * @APPLE_LICENSE_HEADER_END@
  22  */
  23 /*
  24  * Copyright (c) 1983, 1987, 1993
  25  *      The Regents of the University of California.  All rights reserved.
  26  *
  27  * Redistribution and use in source and binary forms, with or without
  28  * modification, are permitted provided that the following conditions
  29  * are met:
  30  * 1. Redistributions of source code must retain the above copyright
  31  *    notice, this list of conditions and the following disclaimer.
  32  * 2. Redistributions in binary form must reproduce the above copyright
  33  *    notice, this list of conditions and the following disclaimer in the
  34  *    documentation and/or other materials provided with the distribution.
  35  * 3. All advertising materials mentioning features or use of this software
  36  *    must display the following acknowledgement:
  37  *      This product includes software developed by the University of
  38  *      California, Berkeley and its contributors.
  39  * 4. Neither the name of the University nor the names of its contributors
  40  *    may be used to endorse or promote products derived from this software
  41  *    without specific prior written permission.
  42  *
  43  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  44  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  45  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  46  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  47  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  48  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  49  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  50  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  51  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  52  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  53  * SUCH DAMAGE.
  54  */
  55
  56
  57 #include <sys/param.h>
  58 #include <sys/time.h>
  59 #define DKTYPENAMES
  60 #include <sys/disklabel.h>
  61 /* from ufs/ffs/fs.h */
  62 #define BBSIZE          8192
  63 #define SBSIZE          8192
  64
  65 #include <errno.h>
  66 #include <fcntl.h>
  67 #include <stdio.h>
  68 #include <stdlib.h>
  69 #include <string.h>
  70 #include <unistd.h>
  71 #include <ctype.h>
  72
  73 #pragma clang diagnostic push
  74 #pragma clang diagnostic ignored "-Wcomma"
  75
  76 #ifdef unused
  77 static int      error(int);
  78 #endif // unused
  79 static int      gettype(const char *, const char **);
  80
  81 struct disklabel *
  82 getdiskbyname(const char *name)
  83 {
  84         static struct disklabel *dp = NULL;
  85         struct partition *pp;
  86         char    *buf;
  87         char    *db_array[2] = { _PATH_DISKTAB, 0 };
  88         char    *cp, *cq;       /* can't be register */
  89         char    p, max, psize[3], pbsize[3],
  90                 pfsize[3], poffset[3], ptype[3];
  91         u_int32_t *dx;
  92
  93         if (cgetent(&buf, db_array, (char *) name) < 0)
  94                 return NULL;
  95
  96         if (dp == NULL) {
  97                 dp = malloc(sizeof(struct disklabel));
  98                 if (dp == NULL) {
  99                         return NULL;
 100                 }
 101         }
 102         memset(dp, 0, sizeof(struct disklabel));
 103
 104         /*
 105          * typename
 106          */
 107         cq = dp->d_typename;
 108         cp = buf;
 109         while (cq < dp->d_typename + sizeof(dp->d_typename) - 1 &&
 110             (*cq = *cp) && *cq != '|' && *cq != ':')
 111                 cq++, cp++;
 112         *cq = '\0';
 113         /*
 114          * boot name (optional)  xxboot, bootxx
 115          */
 116         cgetstr(buf, "b0", &dp->d_boot0);
 117         cgetstr(buf, "b1", &dp->d_boot1);
 118
 119         if (cgetstr(buf, "ty", &cq) > 0 && strcmp(cq, "removable") == 0)
 120                 dp->d_flags |= D_REMOVABLE;
 121         else  if (cq && strcmp(cq, "simulated") == 0)
 122                 dp->d_flags |= D_RAMDISK;
 123         if (cgetcap(buf, "sf", ':') != NULL)
 124                 dp->d_flags |= D_BADSECT;
 125
 126 #define getnumdflt(field, dname, dflt) \
 127         { long f; (field) = (typeof(field))((cgetnum(buf, dname, &f) == -1) ? (dflt) : f); }
 128
 129         getnumdflt(dp->d_secsize, "se", DEV_BSIZE);
 130         cgetnum(buf, "nt",(long *) &dp->d_ntracks);
 131         cgetnum(buf, "ns",(long *) &dp->d_nsectors);
 132         cgetnum(buf, "nc",(long *) &dp->d_ncylinders);
 133
 134         if (cgetstr(buf, "dt", &cq) > 0)
 135                 dp->d_type = gettype(cq, dktypenames);
 136         else
 137                 getnumdflt(dp->d_type, "dt", 0);
 138         getnumdflt(dp->d_secpercyl, "sc", dp->d_nsectors * dp->d_ntracks);
 139         getnumdflt(dp->d_secperunit, "su", dp->d_secpercyl * dp->d_ncylinders);
 140         getnumdflt(dp->d_rpm, "rm", 3600);
 141         getnumdflt(dp->d_interleave, "il", 1);
 142         getnumdflt(dp->d_trackskew, "sk", 0);
 143         getnumdflt(dp->d_cylskew, "cs", 0);
 144         getnumdflt(dp->d_headswitch, "hs", 0);
 145         getnumdflt(dp->d_trkseek, "ts", 0);
 146         getnumdflt(dp->d_bbsize, "bs", BBSIZE);
 147         getnumdflt(dp->d_sbsize, "sb", SBSIZE);
 148         strcpy(psize, "px");
 149         strcpy(pbsize, "bx");
 150         strcpy(pfsize, "fx");
 151         strcpy(poffset, "ox");
 152         strcpy(ptype, "tx");
 153         max = 'a' - 1;
 154         pp = &dp->d_partitions[0];
 155         for (p = 'a'; p < 'a' + MAXPARTITIONS; p++, pp++) {
 156                 psize[1] = pbsize[1] = pfsize[1] = poffset[1] = ptype[1] = p;
 157                 if (cgetnum(buf, psize,(long *) &pp->p_size) == -1)
 158                         pp->p_size = 0;
 159                 else {
 160                         cgetnum(buf, poffset, (long *) &pp->p_offset);
 161                         getnumdflt(pp->p_fsize, pfsize, 0);
 162                         if (pp->p_fsize) {
 163                                 long bsize;
 164
 165                                 if (cgetnum(buf, pbsize, &bsize) == 0)
 166                                         pp->p_frag = bsize / pp->p_fsize;
 167                                 else
 168                                         pp->p_frag = 8;
 169                         }
 170                         getnumdflt(pp->p_fstype, ptype, 0);
 171                         if (pp->p_fstype == 0 && cgetstr(buf, ptype, &cq) > 0)
 172                                 pp->p_fstype = gettype(cq, fstypenames);
 173                         max = p;
 174                 }
 175         }
 176         dp->d_npartitions = max + 1 - 'a';
 177         (void)strcpy(psize, "dx");
 178         dx = dp->d_drivedata;
 179         for (p = '0'; p < '0' + NDDATA; p++, dx++) {
 180                 psize[1] = p;
 181                 getnumdflt(*dx, psize, 0);
 182         }
 183         dp->d_magic = DISKMAGIC;
 184         dp->d_magic2 = DISKMAGIC;
 185         free(buf);
 186         return (dp);
 187 }
 188
 189 static int
 190 gettype(const char *t, const char **names)
 191 {
 192         const char **nm;
 193
 194         for (nm = names; *nm; nm++)
 195                 if (strcasecmp(t, *nm) == 0)
 196                         return (int)(nm - names);
 197         if (isdigit(*t))
 198                 return (atoi(t));
 199         return (0);
 200 }
 201
 202 #ifdef unused
 203 static int
 204 error(err)
 205         int err;
 206 {
 207         char *p;
 208
 209         (void)write(STDERR_FILENO, "disktab: ", 9);
 210         (void)write(STDERR_FILENO, _PATH_DISKTAB, sizeof(_PATH_DISKTAB) - 1);
 211         (void)write(STDERR_FILENO, ": ", 2);
 212         p = strerror(err);
 213         (void)write(STDERR_FILENO, p, strlen(p));
 214         (void)write(STDERR_FILENO, "\n", 1);
 215 }
 216 #endif // unused
 217 #pragma clang diagnostic pop
 218