kvm_mkdb.tproj/nlist.c

   1 /*
   2  * Copyright (c) 1999 Apple Computer, Inc. All rights reserved.
   3  *
   4  * @APPLE_LICENSE_HEADER_START@
   5  *
   6  * "Portions Copyright (c) 1999 Apple Computer, Inc.  All Rights
   7  * Reserved.  This file contains Original Code and/or Modifications of
   8  * Original Code as defined in and that are subject to the Apple Public
   9  * Source License Version 1.0 (the 'License').  You may not use this file
  10  * except in compliance with the License.  Please obtain a copy of the
  11  * License at http://www.apple.com/publicsource and read it before using
  12  * this file.
  13  *
  14  * The Original Code and all software distributed under the License are
  15  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  16  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  17  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
  18  * FITNESS FOR A PARTICULAR PURPOSE OR NON-INFRINGEMENT.  Please see the
  19  * License for the specific language governing rights and limitations
  20  * under the License."
  21  *
  22  * @APPLE_LICENSE_HEADER_END@
  23  */
  24 /*-
  25  * Copyright (c) 1990, 1993
  26  *      The Regents of the University of California.  All rights reserved.
  27  *
  28  * Redistribution and use in source and binary forms, with or without
  29  * modification, are permitted provided that the following conditions
  30  * are met:
  31  * 1. Redistributions of source code must retain the above copyright
  32  *    notice, this list of conditions and the following disclaimer.
  33  * 2. Redistributions in binary form must reproduce the above copyright
  34  *    notice, this list of conditions and the following disclaimer in the
  35  *    documentation and/or other materials provided with the distribution.
  36  * 3. All advertising materials mentioning features or use of this software
  37  *    must display the following acknowledgement:
  38  *      This product includes software developed by the University of
  39  *      California, Berkeley and its contributors.
  40  * 4. Neither the name of the University nor the names of its contributors
  41  *    may be used to endorse or promote products derived from this software
  42  *    without specific prior written permission.
  43  *
  44  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  45  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  46  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  47  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  48  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  49  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  50  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  51  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  52  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  53  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  54  * SUCH DAMAGE.
  55  */
  56
  57 #ifndef lint
  58 static char sccsid[] = "@(#)nlist.c     8.1 (Berkeley) 6/6/93";
  59 #endif /* not lint */
  60
  61 #include <sys/param.h>
  62
  63 #ifdef __APPLE__
  64 #include <machine/exec.h>
  65 #include <machine/param.h>
  66 #define _AOUT_INCLUDE_
  67 #else
  68 #include <a.out.h>
  69 #endif /* NeXT */
  70 #include <db.h>
  71 #include <err.h>
  72 #include <errno.h>
  73 #include <fcntl.h>
  74 #include <kvm.h>
  75 #include <limits.h>
  76 #include <stdio.h>
  77 #include <stdlib.h>
  78 #include <string.h>
  79 #include <unistd.h>
  80
  81 #include "extern.h"
  82
  83 typedef struct nlist NLIST;
  84 #define _strx   n_un.n_strx
  85 #define _name   n_un.n_name
  86
  87 #define badfmt(str)     errx(1, "%s: %s: %s", kfile, str, strerror(EFTYPE))
  88
  89 static void badread __P((int, char *));
  90
  91 static char *kfile;
  92
  93 void
  94 create_knlist(name, db)
  95         char *name;
  96         DB *db;
  97 {
  98         register int nsyms;
  99         struct exec ebuf;
 100         FILE *fp;
 101         NLIST nbuf;
 102         DBT data, key;
 103         int fd, nr, strsize;
 104         char *strtab, buf[1024];
 105
 106         kfile = name;
 107         if ((fd = open(name, O_RDONLY, 0)) < 0)
 108                 err(1, "%s", name);
 109
 110         /* Read in exec structure. */
 111         nr = read(fd, &ebuf, sizeof(struct exec));
 112         if (nr != sizeof(struct exec))
 113                 badfmt("no exec header");
 114
 115         /* Check magic number and symbol count. */
 116         if (N_BADMAG(ebuf))
 117                 badfmt("bad magic number");
 118         if (!ebuf.a_syms)
 119                 badfmt("stripped");
 120
 121         /* Seek to string table. */
 122         if (lseek(fd, N_STROFF(ebuf), SEEK_SET) == -1)
 123                 badfmt("corrupted string table");
 124
 125         /* Read in the size of the symbol table. */
 126         nr = read(fd, (char *)&strsize, sizeof(strsize));
 127         if (nr != sizeof(strsize))
 128                 badread(nr, "no symbol table");
 129
 130         /* Read in the string table. */
 131         strsize -= sizeof(strsize);
 132         if (!(strtab = malloc(strsize)))
 133                 err(1, NULL);
 134         if ((nr = read(fd, strtab, strsize)) != strsize)
 135                 badread(nr, "corrupted symbol table");
 136
 137         /* Seek to symbol table. */
 138         if (!(fp = fdopen(fd, "r")))
 139                 err(1, "%s", name);
 140         if (fseek(fp, N_SYMOFF(ebuf), SEEK_SET) == -1)
 141                 err(1, "%s", name);
 142
 143         data.data = (u_char *)&nbuf;
 144         data.size = sizeof(NLIST);
 145
 146         /* Read each symbol and enter it into the database. */
 147         nsyms = ebuf.a_syms / sizeof(struct nlist);
 148         while (nsyms--) {
 149                 if (fread((char *)&nbuf, sizeof (NLIST), 1, fp) != 1) {
 150                         if (feof(fp))
 151                                 badfmt("corrupted symbol table");
 152                         err(1, "%s", name);
 153                 }
 154                 if (!nbuf._strx || nbuf.n_type&N_STAB)
 155                         continue;
 156
 157                 key.data = (u_char *)strtab + nbuf._strx - sizeof(long);
 158                 key.size = strlen((char *)key.data);
 159                 if (db->put(db, &key, &data, 0))
 160                         err(1, "record enter");
 161
 162                 if (strcmp((char *)key.data, VRS_SYM) == 0) {
 163                         long cur_off, voff;
 164 #ifndef KERNTEXTOFF
 165 #define KERNTEXTOFF KERNBASE
 166 #endif
 167                         /*
 168                          * Calculate offset relative to a normal (non-kernel)
 169                          * a.out.  KERNTEXTOFF is where the kernel is really
 170                          * loaded; N_TXTADDR is where a normal file is loaded.
 171                          * From there, locate file offset in text or data.
 172                          */
 173                         voff = nbuf.n_value - KERNTEXTOFF + N_TXTADDR(ebuf);
 174                         if ((nbuf.n_type & N_TYPE) == N_TEXT)
 175                                 voff += N_TXTOFF(ebuf) - N_TXTADDR(ebuf);
 176                         else
 177                                 voff += N_DATOFF(ebuf) - N_DATADDR(ebuf);
 178                         cur_off = ftell(fp);
 179                         if (fseek(fp, voff, SEEK_SET) == -1)
 180                                 badfmt("corrupted string table");
 181
 182                         /*
 183                          * Read version string up to, and including newline.
 184                          * This code assumes that a newline terminates the
 185                          * version line.
 186                          */
 187                         if (fgets(buf, sizeof(buf), fp) == NULL)
 188                                 badfmt("corrupted string table");
 189
 190                         key.data = (u_char *)VRS_KEY;
 191                         key.size = sizeof(VRS_KEY) - 1;
 192                         data.data = (u_char *)buf;
 193                         data.size = strlen(buf);
 194                         if (db->put(db, &key, &data, 0))
 195                                 err(1, "record enter");
 196
 197                         /* Restore to original values. */
 198                         data.data = (u_char *)&nbuf;
 199                         data.size = sizeof(NLIST);
 200                         if (fseek(fp, cur_off, SEEK_SET) == -1)
 201                                 badfmt("corrupted string table");
 202                 }
 203         }
 204         (void)fclose(fp);
 205 }
 206
 207 static void
 208 badread(nr, p)
 209         int nr;
 210         char *p;
 211 {
 212         if (nr < 0)
 213                 err(1, "%s", kfile);
 214         badfmt(p);
 215 }