lib/libc/gen/disklabel.c

   1 /*      $NetBSD: disklabel.c,v 1.33 2005/06/22 21:35:28 he Exp $        */
   2
   3 /*
   4  * Copyright (c) 1983, 1987, 1993
   5  *      The Regents of the University of California.  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  * 3. Neither the name of the University nor the names of its contributors
  16  *    may be used to endorse or promote products derived from this software
  17  *    without specific prior written permission.
  18  *
  19  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  20  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  21  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  22  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  23  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  24  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  25  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  26  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  27  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  28  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  29  * SUCH DAMAGE.
  30  */
  31
  32 #if HAVE_NBTOOL_CONFIG_H
  33 #include "nbtool_config.h"
  34 #endif
  35
  36 #include <sys/cdefs.h>
  37 #if defined(LIBC_SCCS) && !defined(lint)
  38 #if 0
  39 static char sccsid[] = "@(#)disklabel.c 8.2 (Berkeley) 5/3/95";
  40 #else
  41 __RCSID("$NetBSD: disklabel.c,v 1.33 2005/06/22 21:35:28 he Exp $");
  42 #endif
  43 #endif /* LIBC_SCCS and not lint */
  44
  45 #include "namespace.h"
  46 #include <sys/param.h>
  47 #define DKTYPENAMES
  48 #define FSTYPENAMES
  49 #include <ufs/ufs/dinode.h>
  50 #include <ufs/ffs/fs.h>
  51
  52 #if HAVE_NBTOOL_CONFIG_H
  53 #include <nbinclude/sys/disklabel.h>
  54 #include <nbinclude/disktab.h>
  55 #else
  56 #include <sys/disklabel.h>
  57 #include <disktab.h>
  58 #endif /* HAVE_NBTOOL_CONFIG_H */
  59
  60 #include <assert.h>
  61 #include <ctype.h>
  62 #include <errno.h>
  63 #include <fcntl.h>
  64 #include <stdio.h>
  65 #include <stdlib.h>
  66 #include <string.h>
  67 #include <unistd.h>
  68
  69 #ifdef __weak_alias
  70 __weak_alias(getdiskbyname,_getdiskbyname)
  71 #endif
  72
  73 #if 0
  74 static void     error __P((int));
  75 #endif
  76 static int      gettype __P((char *, const char *const *));
  77
  78 static const char *db_array[2] = { _PATH_DISKTAB, 0 };
  79
  80 int
  81 setdisktab(name)
  82         const char *name;
  83 {
  84         if (!name || !*name)
  85                 return -1;
  86
  87         db_array[0] = name;
  88         return 0;
  89 }
  90
  91
  92 struct disklabel *
  93 getdiskbyname(name)
  94         const char *name;
  95 {
  96         static struct   disklabel disk;
  97         struct  disklabel *dp = &disk;
  98         struct partition *pp;
  99         char    *buf;
 100         char    *cp, *cq;       /* can't be */
 101         char    p, max, psize[3], pbsize[3],
 102                 pfsize[3], poffset[3], ptype[3];
 103         u_int32_t *dx;
 104         long f;
 105
 106         _DIAGASSERT(name != NULL);
 107
 108         if (cgetent(&buf, db_array, name) < 0)
 109                 return NULL;
 110
 111         memset(&disk, 0, sizeof(disk));
 112         /*
 113          * typename
 114          */
 115         cq = dp->d_typename;
 116         cp = buf;
 117         while (cq < dp->d_typename + sizeof(dp->d_typename) - 1 &&
 118             (*cq = *cp) && *cq != '|' && *cq != ':')
 119                 cq++, cp++;
 120         *cq = '\0';
 121         /*
 122          * boot name (optional)  xxboot, bootxx
 123          */
 124         cgetstr(buf, "b0", &dp->d_boot0);
 125         cgetstr(buf, "b1", &dp->d_boot1);
 126
 127         if (cgetstr(buf, "ty", &cq) >= 0) {
 128                 if (strcmp(cq, "removable") == 0)
 129                         dp->d_flags |= D_REMOVABLE;
 130                 else if (strcmp(cq, "simulated") == 0)
 131                         dp->d_flags |= D_RAMDISK;
 132                 free(cq);
 133         }
 134         if (cgetcap(buf, "sf", ':') != NULL)
 135                 dp->d_flags |= D_BADSECT;
 136
 137 #define getnumdflt(field, dname, dflt) \
 138     (field) = ((cgetnum(buf, dname, &f) == -1) ? (dflt) : (u_int32_t) f)
 139 #define getnum(field, dname) \
 140         if (cgetnum(buf, dname, &f) != -1) field = (u_int32_t)f
 141
 142         getnumdflt(dp->d_secsize, "se", DEV_BSIZE);
 143         getnum(dp->d_ntracks, "nt");
 144         getnum(dp->d_nsectors, "ns");
 145         getnum(dp->d_ncylinders, "nc");
 146
 147         if (cgetstr(buf, "dt", &cq) >= 0) {
 148                 dp->d_type = gettype(cq, dktypenames);
 149                 free(cq);
 150         } else
 151                 getnumdflt(dp->d_type, "dt", 0);
 152         getnumdflt(dp->d_secpercyl, "sc", dp->d_nsectors * dp->d_ntracks);
 153         getnumdflt(dp->d_secperunit, "su", dp->d_secpercyl * dp->d_ncylinders);
 154         getnumdflt(dp->d_rpm, "rm", 3600);
 155         getnumdflt(dp->d_interleave, "il", 1);
 156         getnumdflt(dp->d_trackskew, "sk", 0);
 157         getnumdflt(dp->d_cylskew, "cs", 0);
 158         getnumdflt(dp->d_headswitch, "hs", 0);
 159         getnumdflt(dp->d_trkseek, "ts", 0);
 160         getnumdflt(dp->d_bbsize, "bs", BBSIZE);
 161         getnumdflt(dp->d_sbsize, "sb", SBLOCKSIZE);
 162         strcpy(psize, "px");    /* XXX: strcpy is safe */
 163         strcpy(pbsize, "bx");   /* XXX: strcpy is safe */
 164         strcpy(pfsize, "fx");   /* XXX: strcpy is safe */
 165         strcpy(poffset, "ox");  /* XXX: strcpy is safe */
 166         strcpy(ptype, "tx");    /* XXX: strcpy is safe */
 167         max = 'a' - 1;
 168         pp = &dp->d_partitions[0];
 169         for (p = 'a'; p < 'a' + MAXPARTITIONS; p++, pp++) {
 170                 long ff;
 171
 172                 psize[1] = pbsize[1] = pfsize[1] = poffset[1] = ptype[1] = p;
 173                 if (cgetnum(buf, psize, &ff) == -1)
 174                         pp->p_size = 0;
 175                 else {
 176                         pp->p_size = (u_int32_t)ff;
 177                         getnum(pp->p_offset, poffset);
 178                         getnumdflt(pp->p_fsize, pfsize, 0);
 179                         if (pp->p_fsize) {
 180                                 long bsize;
 181
 182                                 if (cgetnum(buf, pbsize, &bsize) == -1)
 183                                         pp->p_frag = 8;
 184                                 else
 185                                         pp->p_frag =
 186                                             (u_int8_t)(bsize / pp->p_fsize);
 187                         }
 188                         getnumdflt(pp->p_fstype, ptype, 0);
 189                         if (pp->p_fstype == 0)
 190                                 if (cgetstr(buf, ptype, &cq) >= 0) {
 191                                         pp->p_fstype = gettype(cq, fstypenames);
 192                                         free(cq);
 193                                 }
 194                         max = p;
 195                 }
 196         }
 197         dp->d_npartitions = max + 1 - 'a';
 198         strcpy(psize, "dx");    /* XXX: strcpy is safe */
 199         dx = dp->d_drivedata;
 200         for (p = '0'; p < '0' + NDDATA; p++, dx++) {
 201                 psize[1] = p;
 202                 getnumdflt(*dx, psize, 0);
 203         }
 204         dp->d_magic = DISKMAGIC;
 205         dp->d_magic2 = DISKMAGIC;
 206         free(buf);
 207         return (dp);
 208 }
 209
 210 static int
 211 gettype(t, names)
 212         char *t;
 213         const char *const *names;
 214 {
 215         const char *const *nm;
 216
 217         _DIAGASSERT(t != NULL);
 218         _DIAGASSERT(names != NULL);
 219
 220         for (nm = names; *nm; nm++)
 221                 if (strcasecmp(t, *nm) == 0)
 222                         return (nm - names);
 223         if (isdigit((unsigned char) *t))
 224                 return (atoi(t));
 225         return (0);
 226 }
 227
 228 #if 0
 229 static void
 230 error(err)
 231         int err;
 232 {
 233         char *p;
 234
 235         (void)write(STDERR_FILENO, "disktab: ", 9);
 236         (void)write(STDERR_FILENO, _PATH_DISKTAB, sizeof(_PATH_DISKTAB) - 1);
 237         (void)write(STDERR_FILENO, ": ", 2);
 238         p = strerror(err);
 239         (void)write(STDERR_FILENO, p, strlen(p));
 240         (void)write(STDERR_FILENO, "\n", 1);
 241 }
 242 #endif