SAFE public projects git trees. - safe/jmp/linux-2.6/blob - scripts/mod/modpost.c

   1 /* Postprocess module symbol versions
   2  *
   3  * Copyright 2003       Kai Germaschewski
   4  * Copyright 2002-2004  Rusty Russell, IBM Corporation
   5  * Copyright 2006       Sam Ravnborg
   6  * Based in part on module-init-tools/depmod.c,file2alias
   7  *
   8  * This software may be used and distributed according to the terms
   9  * of the GNU General Public License, incorporated herein by reference.
  10  *
  11  * Usage: modpost vmlinux module1.o module2.o ...
  12  */
  13
  14 #include <ctype.h>
  15 #include "modpost.h"
  16 #include "../../include/linux/license.h"
  17
  18 /* Are we using CONFIG_MODVERSIONS? */
  19 int modversions = 0;
  20 /* Warn about undefined symbols? (do so if we have vmlinux) */
  21 int have_vmlinux = 0;
  22 /* Is CONFIG_MODULE_SRCVERSION_ALL set? */
  23 static int all_versions = 0;
  24 /* If we are modposting external module set to 1 */
  25 static int external_module = 0;
  26 /* Only warn about unresolved symbols */
  27 static int warn_unresolved = 0;
  28 /* How a symbol is exported */
  29 enum export {
  30         export_plain,      export_unused,     export_gpl,
  31         export_unused_gpl, export_gpl_future, export_unknown
  32 };
  33
  34 void fatal(const char *fmt, ...)
  35 {
  36         va_list arglist;
  37
  38         fprintf(stderr, "FATAL: ");
  39
  40         va_start(arglist, fmt);
  41         vfprintf(stderr, fmt, arglist);
  42         va_end(arglist);
  43
  44         exit(1);
  45 }
  46
  47 void warn(const char *fmt, ...)
  48 {
  49         va_list arglist;
  50
  51         fprintf(stderr, "WARNING: ");
  52
  53         va_start(arglist, fmt);
  54         vfprintf(stderr, fmt, arglist);
  55         va_end(arglist);
  56 }
  57
  58 void merror(const char *fmt, ...)
  59 {
  60         va_list arglist;
  61
  62         fprintf(stderr, "ERROR: ");
  63
  64         va_start(arglist, fmt);
  65         vfprintf(stderr, fmt, arglist);
  66         va_end(arglist);
  67 }
  68
  69 static int is_vmlinux(const char *modname)
  70 {
  71         const char *myname;
  72
  73         if ((myname = strrchr(modname, '/')))
  74                 myname++;
  75         else
  76                 myname = modname;
  77
  78         return strcmp(myname, "vmlinux") == 0;
  79 }
  80
  81 void *do_nofail(void *ptr, const char *expr)
  82 {
  83         if (!ptr) {
  84                 fatal("modpost: Memory allocation failure: %s.\n", expr);
  85         }
  86         return ptr;
  87 }
  88
  89 /* A list of all modules we processed */
  90
  91 static struct module *modules;
  92
  93 static struct module *find_module(char *modname)
  94 {
  95         struct module *mod;
  96
  97         for (mod = modules; mod; mod = mod->next)
  98                 if (strcmp(mod->name, modname) == 0)
  99                         break;
 100         return mod;
 101 }
 102
 103 static struct module *new_module(char *modname)
 104 {
 105         struct module *mod;
 106         char *p, *s;
 107
 108         mod = NOFAIL(malloc(sizeof(*mod)));
 109         memset(mod, 0, sizeof(*mod));
 110         p = NOFAIL(strdup(modname));
 111
 112         /* strip trailing .o */
 113         if ((s = strrchr(p, '.')) != NULL)
 114                 if (strcmp(s, ".o") == 0)
 115                         *s = '\0';
 116
 117         /* add to list */
 118         mod->name = p;
 119         mod->gpl_compatible = -1;
 120         mod->next = modules;
 121         modules = mod;
 122
 123         return mod;
 124 }
 125
 126 /* A hash of all exported symbols,
 127  * struct symbol is also used for lists of unresolved symbols */
 128
 129 #define SYMBOL_HASH_SIZE 1024
 130
 131 struct symbol {
 132         struct symbol *next;
 133         struct module *module;
 134         unsigned int crc;
 135         int crc_valid;
 136         unsigned int weak:1;
 137         unsigned int vmlinux:1;    /* 1 if symbol is defined in vmlinux */
 138         unsigned int kernel:1;     /* 1 if symbol is from kernel
 139                                     *  (only for external modules) **/
 140         unsigned int preloaded:1;  /* 1 if symbol from Module.symvers */
 141         enum export  export;       /* Type of export */
 142         char name[0];
 143 };
 144
 145 static struct symbol *symbolhash[SYMBOL_HASH_SIZE];
 146
 147 /* This is based on the hash agorithm from gdbm, via tdb */
 148 static inline unsigned int tdb_hash(const char *name)
 149 {
 150         unsigned value; /* Used to compute the hash value.  */
 151         unsigned   i;   /* Used to cycle through random values. */
 152
 153         /* Set the initial value from the key size. */
 154         for (value = 0x238F13AF * strlen(name), i=0; name[i]; i++)
 155                 value = (value + (((unsigned char *)name)[i] << (i*5 % 24)));
 156
 157         return (1103515243 * value + 12345);
 158 }
 159
 160 /**
 161  * Allocate a new symbols for use in the hash of exported symbols or
 162  * the list of unresolved symbols per module
 163  **/
 164 static struct symbol *alloc_symbol(const char *name, unsigned int weak,
 165                                    struct symbol *next)
 166 {
 167         struct symbol *s = NOFAIL(malloc(sizeof(*s) + strlen(name) + 1));
 168
 169         memset(s, 0, sizeof(*s));
 170         strcpy(s->name, name);
 171         s->weak = weak;
 172         s->next = next;
 173         return s;
 174 }
 175
 176 /* For the hash of exported symbols */
 177 static struct symbol *new_symbol(const char *name, struct module *module,
 178                                  enum export export)
 179 {
 180         unsigned int hash;
 181         struct symbol *new;
 182
 183         hash = tdb_hash(name) % SYMBOL_HASH_SIZE;
 184         new = symbolhash[hash] = alloc_symbol(name, 0, symbolhash[hash]);
 185         new->module = module;
 186         new->export = export;
 187         return new;
 188 }
 189
 190 static struct symbol *find_symbol(const char *name)
 191 {
 192         struct symbol *s;
 193
 194         /* For our purposes, .foo matches foo.  PPC64 needs this. */
 195         if (name[0] == '.')
 196                 name++;
 197
 198         for (s = symbolhash[tdb_hash(name) % SYMBOL_HASH_SIZE]; s; s=s->next) {
 199                 if (strcmp(s->name, name) == 0)
 200                         return s;
 201         }
 202         return NULL;
 203 }
 204
 205 static struct {
 206         const char *str;
 207         enum export export;
 208 } export_list[] = {
 209         { .str = "EXPORT_SYMBOL",            .export = export_plain },
 210         { .str = "EXPORT_UNUSED_SYMBOL",     .export = export_unused },
 211         { .str = "EXPORT_SYMBOL_GPL",        .export = export_gpl },
 212         { .str = "EXPORT_UNUSED_SYMBOL_GPL", .export = export_unused_gpl },
 213         { .str = "EXPORT_SYMBOL_GPL_FUTURE", .export = export_gpl_future },
 214         { .str = "(unknown)",                .export = export_unknown },
 215 };
 216
 217
 218 static const char *export_str(enum export ex)
 219 {
 220         return export_list[ex].str;
 221 }
 222
 223 static enum export export_no(const char * s)
 224 {
 225         int i;
 226         if (!s)
 227                 return export_unknown;
 228         for (i = 0; export_list[i].export != export_unknown; i++) {
 229                 if (strcmp(export_list[i].str, s) == 0)
 230                         return export_list[i].export;
 231         }
 232         return export_unknown;
 233 }
 234
 235 static enum export export_from_sec(struct elf_info *elf, Elf_Section sec)
 236 {
 237         if (sec == elf->export_sec)
 238                 return export_plain;
 239         else if (sec == elf->export_unused_sec)
 240                 return export_unused;
 241         else if (sec == elf->export_gpl_sec)
 242                 return export_gpl;
 243         else if (sec == elf->export_unused_gpl_sec)
 244                 return export_unused_gpl;
 245         else if (sec == elf->export_gpl_future_sec)
 246                 return export_gpl_future;
 247         else
 248                 return export_unknown;
 249 }
 250
 251 /**
 252  * Add an exported symbol - it may have already been added without a
 253  * CRC, in this case just update the CRC
 254  **/
 255 static struct symbol *sym_add_exported(const char *name, struct module *mod,
 256                                        enum export export)
 257 {
 258         struct symbol *s = find_symbol(name);
 259
 260         if (!s) {
 261                 s = new_symbol(name, mod, export);
 262         } else {
 263                 if (!s->preloaded) {
 264                         warn("%s: '%s' exported twice. Previous export "
 265                              "was in %s%s\n", mod->name, name,
 266                              s->module->name,
 267                              is_vmlinux(s->module->name) ?"":".ko");
 268                 }
 269         }
 270         s->preloaded = 0;
 271         s->vmlinux   = is_vmlinux(mod->name);
 272         s->kernel    = 0;
 273         s->export    = export;
 274         return s;
 275 }
 276
 277 static void sym_update_crc(const char *name, struct module *mod,
 278                            unsigned int crc, enum export export)
 279 {
 280         struct symbol *s = find_symbol(name);
 281
 282         if (!s)
 283                 s = new_symbol(name, mod, export);
 284         s->crc = crc;
 285         s->crc_valid = 1;
 286 }
 287
 288 void *grab_file(const char *filename, unsigned long *size)
 289 {
 290         struct stat st;
 291         void *map;
 292         int fd;
 293
 294         fd = open(filename, O_RDONLY);
 295         if (fd < 0 || fstat(fd, &st) != 0)
 296                 return NULL;
 297
 298         *size = st.st_size;
 299         map = mmap(NULL, *size, PROT_READ|PROT_WRITE, MAP_PRIVATE, fd, 0);
 300         close(fd);
 301
 302         if (map == MAP_FAILED)
 303                 return NULL;
 304         return map;
 305 }
 306
 307 /**
 308   * Return a copy of the next line in a mmap'ed file.
 309   * spaces in the beginning of the line is trimmed away.
 310   * Return a pointer to a static buffer.
 311   **/
 312 char* get_next_line(unsigned long *pos, void *file, unsigned long size)
 313 {
 314         static char line[4096];
 315         int skip = 1;
 316         size_t len = 0;
 317         signed char *p = (signed char *)file + *pos;
 318         char *s = line;
 319
 320         for (; *pos < size ; (*pos)++)
 321         {
 322                 if (skip && isspace(*p)) {
 323                         p++;
 324                         continue;
 325                 }
 326                 skip = 0;
 327                 if (*p != '\n' && (*pos < size)) {
 328                         len++;
 329                         *s++ = *p++;
 330                         if (len > 4095)
 331                                 break; /* Too long, stop */
 332                 } else {
 333                         /* End of string */
 334                         *s = '\0';
 335                         return line;
 336                 }
 337         }
 338         /* End of buffer */
 339         return NULL;
 340 }
 341
 342 void release_file(void *file, unsigned long size)
 343 {
 344         munmap(file, size);
 345 }
 346
 347 static int parse_elf(struct elf_info *info, const char *filename)
 348 {
 349         unsigned int i;
 350         Elf_Ehdr *hdr;
 351         Elf_Shdr *sechdrs;
 352         Elf_Sym  *sym;
 353
 354         hdr = grab_file(filename, &info->size);
 355         if (!hdr) {
 356                 perror(filename);
 357                 exit(1);
 358         }
 359         info->hdr = hdr;
 360         if (info->size < sizeof(*hdr)) {
 361                 /* file too small, assume this is an empty .o file */
 362                 return 0;
 363         }
 364         /* Is this a valid ELF file? */
 365         if ((hdr->e_ident[EI_MAG0] != ELFMAG0) ||
 366             (hdr->e_ident[EI_MAG1] != ELFMAG1) ||
 367             (hdr->e_ident[EI_MAG2] != ELFMAG2) ||
 368             (hdr->e_ident[EI_MAG3] != ELFMAG3)) {
 369                 /* Not an ELF file - silently ignore it */
 370                 return 0;
 371         }
 372         /* Fix endianness in ELF header */
 373         hdr->e_shoff    = TO_NATIVE(hdr->e_shoff);
 374         hdr->e_shstrndx = TO_NATIVE(hdr->e_shstrndx);
 375         hdr->e_shnum    = TO_NATIVE(hdr->e_shnum);
 376         hdr->e_machine  = TO_NATIVE(hdr->e_machine);
 377         sechdrs = (void *)hdr + hdr->e_shoff;
 378         info->sechdrs = sechdrs;
 379
 380         /* Fix endianness in section headers */
 381         for (i = 0; i < hdr->e_shnum; i++) {
 382                 sechdrs[i].sh_type   = TO_NATIVE(sechdrs[i].sh_type);
 383                 sechdrs[i].sh_offset = TO_NATIVE(sechdrs[i].sh_offset);
 384                 sechdrs[i].sh_size   = TO_NATIVE(sechdrs[i].sh_size);
 385                 sechdrs[i].sh_link   = TO_NATIVE(sechdrs[i].sh_link);
 386                 sechdrs[i].sh_name   = TO_NATIVE(sechdrs[i].sh_name);
 387                 sechdrs[i].sh_info   = TO_NATIVE(sechdrs[i].sh_info);
 388         }
 389         /* Find symbol table. */
 390         for (i = 1; i < hdr->e_shnum; i++) {
 391                 const char *secstrings
 392                         = (void *)hdr + sechdrs[hdr->e_shstrndx].sh_offset;
 393                 const char *secname;
 394
 395                 if (sechdrs[i].sh_offset > info->size) {
 396                         fatal("%s is truncated. sechdrs[i].sh_offset=%u > sizeof(*hrd)=%ul\n", filename, (unsigned int)sechdrs[i].sh_offset, sizeof(*hdr));
 397                         return 0;
 398                 }
 399                 secname = secstrings + sechdrs[i].sh_name;
 400                 if (strcmp(secname, ".modinfo") == 0) {
 401                         info->modinfo = (void *)hdr + sechdrs[i].sh_offset;
 402                         info->modinfo_len = sechdrs[i].sh_size;
 403                 } else if (strcmp(secname, "__ksymtab") == 0)
 404                         info->export_sec = i;
 405                 else if (strcmp(secname, "__ksymtab_unused") == 0)
 406                         info->export_unused_sec = i;
 407                 else if (strcmp(secname, "__ksymtab_gpl") == 0)
 408                         info->export_gpl_sec = i;
 409                 else if (strcmp(secname, "__ksymtab_unused_gpl") == 0)
 410                         info->export_unused_gpl_sec = i;
 411                 else if (strcmp(secname, "__ksymtab_gpl_future") == 0)
 412                         info->export_gpl_future_sec = i;
 413
 414                 if (sechdrs[i].sh_type != SHT_SYMTAB)
 415                         continue;
 416
 417                 info->symtab_start = (void *)hdr + sechdrs[i].sh_offset;
 418                 info->symtab_stop  = (void *)hdr + sechdrs[i].sh_offset
 419                                                  + sechdrs[i].sh_size;
 420                 info->strtab       = (void *)hdr +
 421                                      sechdrs[sechdrs[i].sh_link].sh_offset;
 422         }
 423         if (!info->symtab_start) {
 424                 fatal("%s has no symtab?\n", filename);
 425         }
 426         /* Fix endianness in symbols */
 427         for (sym = info->symtab_start; sym < info->symtab_stop; sym++) {
 428                 sym->st_shndx = TO_NATIVE(sym->st_shndx);
 429                 sym->st_name  = TO_NATIVE(sym->st_name);
 430                 sym->st_value = TO_NATIVE(sym->st_value);
 431                 sym->st_size  = TO_NATIVE(sym->st_size);
 432         }
 433         return 1;
 434 }
 435
 436 static void parse_elf_finish(struct elf_info *info)
 437 {
 438         release_file(info->hdr, info->size);
 439 }
 440
 441 #define CRC_PFX     MODULE_SYMBOL_PREFIX "__crc_"
 442 #define KSYMTAB_PFX MODULE_SYMBOL_PREFIX "__ksymtab_"
 443
 444 static void handle_modversions(struct module *mod, struct elf_info *info,
 445                                Elf_Sym *sym, const char *symname)
 446 {
 447         unsigned int crc;
 448         enum export export = export_from_sec(info, sym->st_shndx);
 449
 450         switch (sym->st_shndx) {
 451         case SHN_COMMON:
 452                 warn("\"%s\" [%s] is COMMON symbol\n", symname, mod->name);
 453                 break;
 454         case SHN_ABS:
 455                 /* CRC'd symbol */
 456                 if (memcmp(symname, CRC_PFX, strlen(CRC_PFX)) == 0) {
 457                         crc = (unsigned int) sym->st_value;
 458                         sym_update_crc(symname + strlen(CRC_PFX), mod, crc,
 459                                         export);
 460                 }
 461                 break;
 462         case SHN_UNDEF:
 463                 /* undefined symbol */
 464                 if (ELF_ST_BIND(sym->st_info) != STB_GLOBAL &&
 465                     ELF_ST_BIND(sym->st_info) != STB_WEAK)
 466                         break;
 467                 /* ignore global offset table */
 468                 if (strcmp(symname, "_GLOBAL_OFFSET_TABLE_") == 0)
 469                         break;
 470                 /* ignore __this_module, it will be resolved shortly */
 471                 if (strcmp(symname, MODULE_SYMBOL_PREFIX "__this_module") == 0)
 472                         break;
 473 /* cope with newer glibc (2.3.4 or higher) STT_ definition in elf.h */
 474 #if defined(STT_REGISTER) || defined(STT_SPARC_REGISTER)
 475 /* add compatibility with older glibc */
 476 #ifndef STT_SPARC_REGISTER
 477 #define STT_SPARC_REGISTER STT_REGISTER
 478 #endif
 479                 if (info->hdr->e_machine == EM_SPARC ||
 480                     info->hdr->e_machine == EM_SPARCV9) {
 481                         /* Ignore register directives. */
 482                         if (ELF_ST_TYPE(sym->st_info) == STT_SPARC_REGISTER)
 483                                 break;
 484                         if (symname[0] == '.') {
 485                                 char *munged = strdup(symname);
 486                                 munged[0] = '_';
 487                                 munged[1] = toupper(munged[1]);
 488                                 symname = munged;
 489                         }
 490                 }
 491 #endif
 492
 493                 if (memcmp(symname, MODULE_SYMBOL_PREFIX,
 494                            strlen(MODULE_SYMBOL_PREFIX)) == 0)
 495                         mod->unres = alloc_symbol(symname +
 496                                                   strlen(MODULE_SYMBOL_PREFIX),
 497                                                   ELF_ST_BIND(sym->st_info) == STB_WEAK,
 498                                                   mod->unres);
 499                 break;
 500         default:
 501                 /* All exported symbols */
 502                 if (memcmp(symname, KSYMTAB_PFX, strlen(KSYMTAB_PFX)) == 0) {
 503                         sym_add_exported(symname + strlen(KSYMTAB_PFX), mod,
 504                                         export);
 505                 }
 506                 if (strcmp(symname, MODULE_SYMBOL_PREFIX "init_module") == 0)
 507                         mod->has_init = 1;
 508                 if (strcmp(symname, MODULE_SYMBOL_PREFIX "cleanup_module") == 0)
 509                         mod->has_cleanup = 1;
 510                 break;
 511         }
 512 }
 513
 514 /**
 515  * Parse tag=value strings from .modinfo section
 516  **/
 517 static char *next_string(char *string, unsigned long *secsize)
 518 {
 519         /* Skip non-zero chars */
 520         while (string[0]) {
 521                 string++;
 522                 if ((*secsize)-- <= 1)
 523                         return NULL;
 524         }
 525
 526         /* Skip any zero padding. */
 527         while (!string[0]) {
 528                 string++;
 529                 if ((*secsize)-- <= 1)
 530                         return NULL;
 531         }
 532         return string;
 533 }
 534
 535 static char *get_next_modinfo(void *modinfo, unsigned long modinfo_len,
 536                               const char *tag, char *info)
 537 {
 538         char *p;
 539         unsigned int taglen = strlen(tag);
 540         unsigned long size = modinfo_len;
 541
 542         if (info) {
 543                 size -= info - (char *)modinfo;
 544                 modinfo = next_string(info, &size);
 545         }
 546
 547         for (p = modinfo; p; p = next_string(p, &size)) {
 548                 if (strncmp(p, tag, taglen) == 0 && p[taglen] == '=')
 549                         return p + taglen + 1;
 550         }
 551         return NULL;
 552 }
 553
 554 static char *get_modinfo(void *modinfo, unsigned long modinfo_len,
 555                          const char *tag)
 556
 557 {
 558         return get_next_modinfo(modinfo, modinfo_len, tag, NULL);
 559 }
 560
 561 /**
 562  * Test if string s ends in string sub
 563  * return 0 if match
 564  **/
 565 static int strrcmp(const char *s, const char *sub)
 566 {
 567         int slen, sublen;
 568
 569         if (!s || !sub)
 570                 return 1;
 571
 572         slen = strlen(s);
 573         sublen = strlen(sub);
 574
 575         if ((slen == 0) || (sublen == 0))
 576                 return 1;
 577
 578         if (sublen > slen)
 579                 return 1;
 580
 581         return memcmp(s + slen - sublen, sub, sublen);
 582 }
 583
 584 /**
 585  * Whitelist to allow certain references to pass with no warning.
 586  *
 587  * Pattern 0:
 588  *   Do not warn if funtion/data are marked with __init_refok/__initdata_refok.
 589  *   The pattern is identified by:
 590  *   fromsec = .text.init.refok | .data.init.refok
 591  *
 592  * Pattern 1:
 593  *   If a module parameter is declared __initdata and permissions=0
 594  *   then this is legal despite the warning generated.
 595  *   We cannot see value of permissions here, so just ignore
 596  *   this pattern.
 597  *   The pattern is identified by:
 598  *   tosec   = .init.data
 599  *   fromsec = .data*
 600  *   atsym   =__param*
 601  *
 602  * Pattern 2:
 603  *   Many drivers utilise a *driver container with references to
 604  *   add, remove, probe functions etc.
 605  *   These functions may often be marked __init and we do not want to
 606  *   warn here.
 607  *   the pattern is identified by:
 608  *   tosec   = .init.text | .exit.text | .init.data
 609  *   fromsec = .data
 610  *   atsym = *driver, *_template, *_sht, *_ops, *_probe, *probe_one, *_console
 611  *
 612  * Pattern 3:
 613  *   Whitelist all references from .pci_fixup* section to .init.text
 614  *   This is part of the PCI init when built-in
 615  *
 616  * Pattern 4:
 617  *   Whitelist all refereces from .text.head to .init.data
 618  *   Whitelist all refereces from .text.head to .init.text
 619  *
 620  * Pattern 5:
 621  *   Some symbols belong to init section but still it is ok to reference
 622  *   these from non-init sections as these symbols don't have any memory
 623  *   allocated for them and symbol address and value are same. So even
 624  *   if init section is freed, its ok to reference those symbols.
 625  *   For ex. symbols marking the init section boundaries.
 626  *   This pattern is identified by
 627  *   refsymname = __init_begin, _sinittext, _einittext
 628  *
 629  * Pattern 7:
 630  *  Logos used in drivers/video/logo reside in __initdata but the
 631  *  funtion that references them are EXPORT_SYMBOL() so cannot be
 632  *  marker __init. So we whitelist them here.
 633  *  The pattern is:
 634  *  tosec      = .init.data
 635  *  fromsec    = .text*
 636  *  refsymname = logo_
 637  *
 638  * Pattern 8:
 639  *  Symbols contained in .paravirtprobe may safely reference .init.text.
 640  *  The pattern is:
 641  *  tosec   = .init.text
 642  *  fromsec  = .paravirtprobe
 643  *
 644  * Pattern 10:
 645  *  ia64 has machvec table for each platform and
 646  *  powerpc has a machine desc table for each platform.
 647  *  It is mixture of function pointers of .init.text and .text.
 648  *  fromsec  = .machvec | .machine.desc
 649  **/
 650 static int secref_whitelist(const char *modname, const char *tosec,
 651                             const char *fromsec, const char *atsym,
 652                             const char *refsymname)
 653 {
 654         int f1 = 1, f2 = 1;
 655         const char **s;
 656         const char *pat2sym[] = {
 657                 "driver",
 658                 "_template", /* scsi uses *_template a lot */
 659                 "_sht",      /* scsi also used *_sht to some extent */
 660                 "_ops",
 661                 "_probe",
 662                 "_probe_one",
 663                 "_console",
 664                 "apic_es7000",
 665                 NULL
 666         };
 667
 668         const char *pat3refsym[] = {
 669                 "__init_begin",
 670                 "_sinittext",
 671                 "_einittext",
 672                 NULL
 673         };
 674
 675         /* Check for pattern 0 */
 676         if ((strcmp(fromsec, ".text.init.refok") == 0) ||
 677             (strcmp(fromsec, ".data.init.refok") == 0))
 678                 return 1;
 679
 680         /* Check for pattern 1 */
 681         if (strcmp(tosec, ".init.data") != 0)
 682                 f1 = 0;
 683         if (strncmp(fromsec, ".data", strlen(".data")) != 0)
 684                 f1 = 0;
 685         if (strncmp(atsym, "__param", strlen("__param")) != 0)
 686                 f1 = 0;
 687
 688         if (f1)
 689                 return f1;
 690
 691         /* Check for pattern 2 */
 692         if ((strcmp(tosec, ".init.text") != 0) &&
 693             (strcmp(tosec, ".exit.text") != 0) &&
 694             (strcmp(tosec, ".init.data") != 0))
 695                 f2 = 0;
 696         if (strcmp(fromsec, ".data") != 0)
 697                 f2 = 0;
 698
 699         for (s = pat2sym; *s; s++)
 700                 if (strrcmp(atsym, *s) == 0)
 701                         f1 = 1;
 702         if (f1 && f2)
 703                 return 1;
 704
 705         /* Check for pattern 3 */
 706         if ((strncmp(fromsec, ".pci_fixup", strlen(".pci_fixup")) == 0) &&
 707             (strcmp(tosec, ".init.text") == 0))
 708         return 1;
 709
 710         /* Check for pattern 4 */
 711         if ((strcmp(fromsec, ".text.head") == 0) &&
 712                 ((strcmp(tosec, ".init.data") == 0) ||
 713                 (strcmp(tosec, ".init.text") == 0)))
 714         return 1;
 715
 716         /* Check for pattern 5 */
 717         for (s = pat3refsym; *s; s++)
 718                 if (strcmp(refsymname, *s) == 0)
 719                         return 1;
 720
 721         /* Check for pattern 7 */
 722         if ((strcmp(tosec, ".init.data") == 0) &&
 723             (strncmp(fromsec, ".text", strlen(".text")) == 0) &&
 724             (strncmp(refsymname, "logo_", strlen("logo_")) == 0))
 725                 return 1;
 726
 727         /* Check for pattern 8 */
 728         if ((strcmp(tosec, ".init.text") == 0) &&
 729             (strcmp(fromsec, ".paravirtprobe") == 0))
 730                 return 1;
 731
 732         /* Check for pattern 10 */
 733         if ((strcmp(fromsec, ".machvec") == 0) ||
 734             (strcmp(fromsec, ".machine.desc") == 0))
 735                 return 1;
 736
 737         return 0;
 738 }
 739
 740 /**
 741  * Find symbol based on relocation record info.
 742  * In some cases the symbol supplied is a valid symbol so
 743  * return refsym. If st_name != 0 we assume this is a valid symbol.
 744  * In other cases the symbol needs to be looked up in the symbol table
 745  * based on section and address.
 746  *  **/
 747 static Elf_Sym *find_elf_symbol(struct elf_info *elf, Elf_Addr addr,
 748                                 Elf_Sym *relsym)
 749 {
 750         Elf_Sym *sym;
 751
 752         if (relsym->st_name != 0)
 753                 return relsym;
 754         for (sym = elf->symtab_start; sym < elf->symtab_stop; sym++) {
 755                 if (sym->st_shndx != relsym->st_shndx)
 756                         continue;
 757                 if (ELF_ST_TYPE(sym->st_info) == STT_SECTION)
 758                         continue;
 759                 if (sym->st_value == addr)
 760                         return sym;
 761         }
 762         return NULL;
 763 }
 764
 765 static inline int is_arm_mapping_symbol(const char *str)
 766 {
 767         return str[0] == '$' && strchr("atd", str[1])
 768                && (str[2] == '\0' || str[2] == '.');
 769 }
 770
 771 /*
 772  * If there's no name there, ignore it; likewise, ignore it if it's
 773  * one of the magic symbols emitted used by current ARM tools.
 774  *
 775  * Otherwise if find_symbols_between() returns those symbols, they'll
 776  * fail the whitelist tests and cause lots of false alarms ... fixable
 777  * only by merging __exit and __init sections into __text, bloating
 778  * the kernel (which is especially evil on embedded platforms).
 779  */
 780 static inline int is_valid_name(struct elf_info *elf, Elf_Sym *sym)
 781 {
 782         const char *name = elf->strtab + sym->st_name;
 783
 784         if (!name || !strlen(name))
 785                 return 0;
 786         return !is_arm_mapping_symbol(name);
 787 }
 788
 789 /*
 790  * Find symbols before or equal addr and after addr - in the section sec.
 791  * If we find two symbols with equal offset prefer one with a valid name.
 792  * The ELF format may have a better way to detect what type of symbol
 793  * it is, but this works for now.
 794  **/
 795 static void find_symbols_between(struct elf_info *elf, Elf_Addr addr,
 796                                  const char *sec,
 797                                  Elf_Sym **before, Elf_Sym **after)
 798 {
 799         Elf_Sym *sym;
 800         Elf_Ehdr *hdr = elf->hdr;
 801         Elf_Addr beforediff = ~0;
 802         Elf_Addr afterdiff = ~0;
 803         const char *secstrings = (void *)hdr +
 804                                  elf->sechdrs[hdr->e_shstrndx].sh_offset;
 805
 806         *before = NULL;
 807         *after = NULL;
 808
 809         for (sym = elf->symtab_start; sym < elf->symtab_stop; sym++) {
 810                 const char *symsec;
 811
 812                 if (sym->st_shndx >= SHN_LORESERVE)
 813                         continue;
 814                 symsec = secstrings + elf->sechdrs[sym->st_shndx].sh_name;
 815                 if (strcmp(symsec, sec) != 0)
 816                         continue;
 817                 if (!is_valid_name(elf, sym))
 818                         continue;
 819                 if (sym->st_value <= addr) {
 820                         if ((addr - sym->st_value) < beforediff) {
 821                                 beforediff = addr - sym->st_value;
 822                                 *before = sym;
 823                         }
 824                         else if ((addr - sym->st_value) == beforediff) {
 825                                 *before = sym;
 826                         }
 827                 }
 828                 else
 829                 {
 830                         if ((sym->st_value - addr) < afterdiff) {
 831                                 afterdiff = sym->st_value - addr;
 832                                 *after = sym;
 833                         }
 834                         else if ((sym->st_value - addr) == afterdiff) {
 835                                 *after = sym;
 836                         }
 837                 }
 838         }
 839 }
 840
 841 /**
 842  * Print a warning about a section mismatch.
 843  * Try to find symbols near it so user can find it.
 844  * Check whitelist before warning - it may be a false positive.
 845  **/
 846 static void warn_sec_mismatch(const char *modname, const char *fromsec,
 847                               struct elf_info *elf, Elf_Sym *sym, Elf_Rela r)
 848 {
 849         const char *refsymname = "";
 850         Elf_Sym *before, *after;
 851         Elf_Sym *refsym;
 852         Elf_Ehdr *hdr = elf->hdr;
 853         Elf_Shdr *sechdrs = elf->sechdrs;
 854         const char *secstrings = (void *)hdr +
 855                                  sechdrs[hdr->e_shstrndx].sh_offset;
 856         const char *secname = secstrings + sechdrs[sym->st_shndx].sh_name;
 857
 858         find_symbols_between(elf, r.r_offset, fromsec, &before, &after);
 859
 860         refsym = find_elf_symbol(elf, r.r_addend, sym);
 861         if (refsym && strlen(elf->strtab + refsym->st_name))
 862                 refsymname = elf->strtab + refsym->st_name;
 863
 864         /* check whitelist - we may ignore it */
 865         if (before &&
 866             secref_whitelist(modname, secname, fromsec,
 867                              elf->strtab + before->st_name, refsymname))
 868                 return;
 869
 870         /* fromsec whitelist - without a valid 'before'
 871          * powerpc has a GOT table in .got2 section */
 872         if (strcmp(fromsec, ".got2") == 0)
 873                 return;
 874
 875         if (before && after) {
 876                 warn("%s(%s+0x%llx): Section mismatch: reference to %s:%s "
 877                      "(between '%s' and '%s')\n",
 878                      modname, fromsec, (unsigned long long)r.r_offset,
 879                      secname, refsymname,
 880                      elf->strtab + before->st_name,
 881                      elf->strtab + after->st_name);
 882         } else if (before) {
 883                 warn("%s(%s+0x%llx): Section mismatch: reference to %s:%s "
 884                      "(after '%s')\n",
 885                      modname, fromsec, (unsigned long long)r.r_offset,
 886                      secname, refsymname,
 887                      elf->strtab + before->st_name);
 888         } else if (after) {
 889                 warn("%s(%s+0x%llx): Section mismatch: reference to %s:%s "
 890                      "before '%s' (at offset -0x%llx)\n",
 891                      modname, fromsec, (unsigned long long)r.r_offset,
 892                      secname, refsymname,
 893                      elf->strtab + after->st_name);
 894         } else {
 895                 warn("%s(%s+0x%llx): Section mismatch: reference to %s:%s\n",
 896                      modname, fromsec, (unsigned long long)r.r_offset,
 897                      secname, refsymname);
 898         }
 899 }
 900
 901 static void addend_386_rel(struct elf_info *elf, int section, Elf_Rela *r)
 902 {
 903         Elf_Shdr *sechdrs = elf->sechdrs;
 904         unsigned int r_typ;
 905         unsigned int *location;
 906
 907         r_typ = ELF_R_TYPE(r->r_info);
 908         location = (void *)elf->hdr +
 909                 sechdrs[sechdrs[section].sh_info].sh_offset + r->r_offset;
 910         switch (r_typ) {
 911         case R_386_32:
 912                 r->r_addend = TO_NATIVE(*location);
 913                 break;
 914         case R_386_PC32:
 915                 r->r_addend = TO_NATIVE(*location) + 4;
 916                 break;
 917         }
 918 }
 919
 920 static void addend_arm_rel(struct elf_info *elf, int section, Elf_Rela *r)
 921 {
 922         Elf_Shdr *sechdrs = elf->sechdrs;
 923         unsigned int r_typ;
 924         unsigned int *location;
 925
 926         r_typ = ELF_R_TYPE(r->r_info);
 927         location = (void *)elf->hdr +
 928                 sechdrs[sechdrs[section].sh_info].sh_offset + r->r_offset;
 929         switch (r_typ) {
 930         case R_ARM_ABS32:
 931                 r->r_addend = TO_NATIVE(*location);
 932                 break;
 933         case R_ARM_PC24:
 934                 r->r_addend = ((TO_NATIVE(*location) & 0x00ffffff) << 2) + 8;
 935                 break;
 936         }
 937 }
 938
 939 static int addend_mips_rel(struct elf_info *elf, int section, Elf_Rela *r)
 940 {
 941         Elf_Shdr *sechdrs = elf->sechdrs;
 942         unsigned int r_typ;
 943         unsigned int *location;
 944         unsigned int inst;
 945
 946         r_typ = ELF_R_TYPE(r->r_info);
 947         if (r_typ == R_MIPS_HI16)
 948                 return 1;       /* skip this */
 949         location = (void *)elf->hdr +
 950                 sechdrs[sechdrs[section].sh_info].sh_offset + r->r_offset;
 951         inst = TO_NATIVE(*location);
 952         switch (r_typ) {
 953         case R_MIPS_LO16:
 954                 r->r_addend = ((inst & 0xffff) ^ 0x8000) - 0x8000;
 955                 break;
 956         case R_MIPS_26:
 957                 r->r_addend = (inst & 0x03ffffff) << 2;
 958                 break;
 959         }
 960         return 0;
 961 }
 962
 963 /**
 964  * A module includes a number of sections that are discarded
 965  * either when loaded or when used as built-in.
 966  * For loaded modules all functions marked __init and all data
 967  * marked __initdata will be discarded when the module has been intialized.
 968  * Likewise for modules used built-in the sections marked __exit
 969  * are discarded because __exit marked function are supposed to be called
 970  * only when a moduel is unloaded which never happes for built-in modules.
 971  * The check_sec_ref() function traverses all relocation records
 972  * to find all references to a section that reference a section that will
 973  * be discarded and warns about it.
 974  **/
 975 static void check_sec_ref(struct module *mod, const char *modname,
 976                           struct elf_info *elf,
 977                           int section(const char*),
 978                           int section_ref_ok(const char *))
 979 {
 980         int i;
 981         Elf_Sym  *sym;
 982         Elf_Ehdr *hdr = elf->hdr;
 983         Elf_Shdr *sechdrs = elf->sechdrs;
 984         const char *secstrings = (void *)hdr +
 985                                  sechdrs[hdr->e_shstrndx].sh_offset;
 986
 987         /* Walk through all sections */
 988         for (i = 0; i < hdr->e_shnum; i++) {
 989                 const char *name = secstrings + sechdrs[i].sh_name;
 990                 const char *secname;
 991                 Elf_Rela r;
 992                 unsigned int r_sym;
 993                 /* We want to process only relocation sections and not .init */
 994                 if (sechdrs[i].sh_type == SHT_RELA) {
 995                         Elf_Rela *rela;
 996                         Elf_Rela *start = (void *)hdr + sechdrs[i].sh_offset;
 997                         Elf_Rela *stop  = (void*)start + sechdrs[i].sh_size;
 998                         name += strlen(".rela");
 999                         if (section_ref_ok(name))
1000                                 continue;
1001
1002                         for (rela = start; rela < stop; rela++) {
1003                                 r.r_offset = TO_NATIVE(rela->r_offset);
1004 #if KERNEL_ELFCLASS == ELFCLASS64
1005                                 if (hdr->e_machine == EM_MIPS) {
1006                                         unsigned int r_typ;
1007                                         r_sym = ELF64_MIPS_R_SYM(rela->r_info);
1008                                         r_sym = TO_NATIVE(r_sym);
1009                                         r_typ = ELF64_MIPS_R_TYPE(rela->r_info);
1010                                         r.r_info = ELF64_R_INFO(r_sym, r_typ);
1011                                 } else {
1012                                         r.r_info = TO_NATIVE(rela->r_info);
1013                                         r_sym = ELF_R_SYM(r.r_info);
1014                                 }
1015 #else
1016                                 r.r_info = TO_NATIVE(rela->r_info);
1017                                 r_sym = ELF_R_SYM(r.r_info);
1018 #endif
1019                                 r.r_addend = TO_NATIVE(rela->r_addend);
1020                                 sym = elf->symtab_start + r_sym;
1021                                 /* Skip special sections */
1022                                 if (sym->st_shndx >= SHN_LORESERVE)
1023                                         continue;
1024
1025                                 secname = secstrings +
1026                                         sechdrs[sym->st_shndx].sh_name;
1027                                 if (section(secname))
1028                                         warn_sec_mismatch(modname, name,
1029                                                           elf, sym, r);
1030                         }
1031                 } else if (sechdrs[i].sh_type == SHT_REL) {
1032                         Elf_Rel *rel;
1033                         Elf_Rel *start = (void *)hdr + sechdrs[i].sh_offset;
1034                         Elf_Rel *stop  = (void*)start + sechdrs[i].sh_size;
1035                         name += strlen(".rel");
1036                         if (section_ref_ok(name))
1037                                 continue;
1038
1039                         for (rel = start; rel < stop; rel++) {
1040                                 r.r_offset = TO_NATIVE(rel->r_offset);
1041 #if KERNEL_ELFCLASS == ELFCLASS64
1042                                 if (hdr->e_machine == EM_MIPS) {
1043                                         unsigned int r_typ;
1044                                         r_sym = ELF64_MIPS_R_SYM(rel->r_info);
1045                                         r_sym = TO_NATIVE(r_sym);
1046                                         r_typ = ELF64_MIPS_R_TYPE(rel->r_info);
1047                                         r.r_info = ELF64_R_INFO(r_sym, r_typ);
1048                                 } else {
1049                                         r.r_info = TO_NATIVE(rel->r_info);
1050                                         r_sym = ELF_R_SYM(r.r_info);
1051                                 }
1052 #else
1053                                 r.r_info = TO_NATIVE(rel->r_info);
1054                                 r_sym = ELF_R_SYM(r.r_info);
1055 #endif
1056                                 r.r_addend = 0;
1057                                 if (hdr->e_machine == EM_386)
1058                                         addend_386_rel(elf, i, &r);
1059                                 else if (hdr->e_machine == EM_ARM)
1060                                         addend_arm_rel(elf, i, &r);
1061                                 else if (hdr->e_machine == EM_MIPS) {
1062                                         if (addend_mips_rel(elf, i, &r))
1063                                                 continue;
1064                                 }
1065                                 sym = elf->symtab_start + r_sym;
1066                                 /* Skip special sections */
1067                                 if (sym->st_shndx >= SHN_LORESERVE)
1068                                         continue;
1069
1070                                 secname = secstrings +
1071                                         sechdrs[sym->st_shndx].sh_name;
1072                                 if (section(secname))
1073                                         warn_sec_mismatch(modname, name,
1074                                                           elf, sym, r);
1075                         }
1076                 }
1077         }
1078 }
1079
1080 /**
1081  * Functions used only during module init is marked __init and is stored in
1082  * a .init.text section. Likewise data is marked __initdata and stored in
1083  * a .init.data section.
1084  * If this section is one of these sections return 1
1085  * See include/linux/init.h for the details
1086  **/
1087 static int init_section(const char *name)
1088 {
1089         if (strcmp(name, ".init") == 0)
1090                 return 1;
1091         if (strncmp(name, ".init.", strlen(".init.")) == 0)
1092                 return 1;
1093         return 0;
1094 }
1095
1096 /**
1097  * Identify sections from which references to a .init section is OK.
1098  *
1099  * Unfortunately references to read only data that referenced .init
1100  * sections had to be excluded. Almost all of these are false
1101  * positives, they are created by gcc. The downside of excluding rodata
1102  * is that there really are some user references from rodata to
1103  * init code, e.g. drivers/video/vgacon.c:
1104  *
1105  * const struct consw vga_con = {
1106  *        con_startup:            vgacon_startup,
1107  *
1108  * where vgacon_startup is __init.  If you want to wade through the false
1109  * positives, take out the check for rodata.
1110  **/
1111 static int init_section_ref_ok(const char *name)
1112 {
1113         const char **s;
1114         /* Absolute section names */
1115         const char *namelist1[] = {
1116                 ".init",
1117                 ".opd",   /* see comment [OPD] at exit_section_ref_ok() */
1118                 ".toc1",  /* used by ppc64 */
1119                 ".stab",
1120                 ".data.rel.ro", /* used by parisc64 */
1121                 ".parainstructions",
1122                 ".text.lock",
1123                 "__bug_table", /* used by powerpc for BUG() */
1124                 ".pci_fixup_header",
1125                 ".pci_fixup_final",
1126                 ".pdr",
1127                 "__param",
1128                 "__ex_table",
1129                 ".fixup",
1130                 ".smp_locks",
1131                 ".plt",  /* seen on ARCH=um build on x86_64. Harmless */
1132                 "__ftr_fixup",          /* powerpc cpu feature fixup */
1133                 "__fw_ftr_fixup",       /* powerpc firmware feature fixup */
1134                 NULL
1135         };
1136         /* Start of section names */
1137         const char *namelist2[] = {
1138                 ".init.",
1139                 ".altinstructions",
1140                 ".eh_frame",
1141                 ".debug",
1142                 ".parainstructions",
1143                 ".rodata",
1144                 NULL
1145         };
1146         /* part of section name */
1147         const char *namelist3 [] = {
1148                 ".unwind",  /* sample: IA_64.unwind.init.text */
1149                 NULL
1150         };
1151
1152         for (s = namelist1; *s; s++)
1153                 if (strcmp(*s, name) == 0)
1154                         return 1;
1155         for (s = namelist2; *s; s++)
1156                 if (strncmp(*s, name, strlen(*s)) == 0)
1157                         return 1;
1158         for (s = namelist3; *s; s++)
1159                 if (strstr(name, *s) != NULL)
1160                         return 1;
1161         if (strrcmp(name, ".init") == 0)
1162                 return 1;
1163         return 0;
1164 }
1165
1166 /*
1167  * Functions used only during module exit is marked __exit and is stored in
1168  * a .exit.text section. Likewise data is marked __exitdata and stored in
1169  * a .exit.data section.
1170  * If this section is one of these sections return 1
1171  * See include/linux/init.h for the details
1172  **/
1173 static int exit_section(const char *name)
1174 {
1175         if (strcmp(name, ".exit.text") == 0)
1176                 return 1;
1177         if (strcmp(name, ".exit.data") == 0)
1178                 return 1;
1179         return 0;
1180
1181 }
1182
1183 /*
1184  * Identify sections from which references to a .exit section is OK.
1185  *
1186  * [OPD] Keith Ownes <kaos@sgi.com> commented:
1187  * For our future {in}sanity, add a comment that this is the ppc .opd
1188  * section, not the ia64 .opd section.
1189  * ia64 .opd should not point to discarded sections.
1190  * [.rodata] like for .init.text we ignore .rodata references -same reason
1191  **/
1192 static int exit_section_ref_ok(const char *name)
1193 {
1194         const char **s;
1195         /* Absolute section names */
1196         const char *namelist1[] = {
1197                 ".exit.text",
1198                 ".exit.data",
1199                 ".init.text",
1200                 ".rodata",
1201                 ".opd", /* See comment [OPD] */
1202                 ".toc1",  /* used by ppc64 */
1203                 ".altinstructions",
1204                 ".pdr",
1205                 "__bug_table", /* used by powerpc for BUG() */
1206                 ".exitcall.exit",
1207                 ".eh_frame",
1208                 ".parainstructions",
1209                 ".stab",
1210                 "__ex_table",
1211                 ".fixup",
1212                 ".smp_locks",
1213                 ".plt",  /* seen on ARCH=um build on x86_64. Harmless */
1214                 NULL
1215         };
1216         /* Start of section names */
1217         const char *namelist2[] = {
1218                 ".debug",
1219                 NULL
1220         };
1221         /* part of section name */
1222         const char *namelist3 [] = {
1223                 ".unwind",  /* Sample: IA_64.unwind.exit.text */
1224                 NULL
1225         };
1226
1227         for (s = namelist1; *s; s++)
1228                 if (strcmp(*s, name) == 0)
1229                         return 1;
1230         for (s = namelist2; *s; s++)
1231                 if (strncmp(*s, name, strlen(*s)) == 0)
1232                         return 1;
1233         for (s = namelist3; *s; s++)
1234                 if (strstr(name, *s) != NULL)
1235                         return 1;
1236         return 0;
1237 }
1238
1239 static void read_symbols(char *modname)
1240 {
1241         const char *symname;
1242         char *version;
1243         char *license;
1244         struct module *mod;
1245         struct elf_info info = { };
1246         Elf_Sym *sym;
1247
1248         if (!parse_elf(&info, modname))
1249                 return;
1250
1251         mod = new_module(modname);
1252
1253         /* When there's no vmlinux, don't print warnings about
1254          * unresolved symbols (since there'll be too many ;) */
1255         if (is_vmlinux(modname)) {
1256                 have_vmlinux = 1;
1257                 mod->skip = 1;
1258         }
1259
1260         license = get_modinfo(info.modinfo, info.modinfo_len, "license");
1261         while (license) {
1262                 if (license_is_gpl_compatible(license))
1263                         mod->gpl_compatible = 1;
1264                 else {
1265                         mod->gpl_compatible = 0;
1266                         break;
1267                 }
1268                 license = get_next_modinfo(info.modinfo, info.modinfo_len,
1269                                            "license", license);
1270         }
1271
1272         for (sym = info.symtab_start; sym < info.symtab_stop; sym++) {
1273                 symname = info.strtab + sym->st_name;
1274
1275                 handle_modversions(mod, &info, sym, symname);
1276                 handle_moddevtable(mod, &info, sym, symname);
1277         }
1278         check_sec_ref(mod, modname, &info, init_section, init_section_ref_ok);
1279         check_sec_ref(mod, modname, &info, exit_section, exit_section_ref_ok);
1280
1281         version = get_modinfo(info.modinfo, info.modinfo_len, "version");
1282         if (version)
1283                 maybe_frob_rcs_version(modname, version, info.modinfo,
1284                                        version - (char *)info.hdr);
1285         if (version || (all_versions && !is_vmlinux(modname)))
1286                 get_src_version(modname, mod->srcversion,
1287                                 sizeof(mod->srcversion)-1);
1288
1289         parse_elf_finish(&info);
1290
1291         /* Our trick to get versioning for struct_module - it's
1292          * never passed as an argument to an exported function, so
1293          * the automatic versioning doesn't pick it up, but it's really
1294          * important anyhow */
1295         if (modversions)
1296                 mod->unres = alloc_symbol("struct_module", 0, mod->unres);
1297 }
1298
1299 #define SZ 500
1300
1301 /* We first write the generated file into memory using the
1302  * following helper, then compare to the file on disk and
1303  * only update the later if anything changed */
1304
1305 void __attribute__((format(printf, 2, 3))) buf_printf(struct buffer *buf,
1306                                                       const char *fmt, ...)
1307 {
1308         char tmp[SZ];
1309         int len;
1310         va_list ap;
1311
1312         va_start(ap, fmt);
1313         len = vsnprintf(tmp, SZ, fmt, ap);
1314         buf_write(buf, tmp, len);
1315         va_end(ap);
1316 }
1317
1318 void buf_write(struct buffer *buf, const char *s, int len)
1319 {
1320         if (buf->size - buf->pos < len) {
1321                 buf->size += len + SZ;
1322                 buf->p = realloc(buf->p, buf->size);
1323         }
1324         strncpy(buf->p + buf->pos, s, len);
1325         buf->pos += len;
1326 }
1327
1328 static void check_for_gpl_usage(enum export exp, const char *m, const char *s)
1329 {
1330         const char *e = is_vmlinux(m) ?"":".ko";
1331
1332         switch (exp) {
1333         case export_gpl:
1334                 fatal("modpost: GPL-incompatible module %s%s "
1335                       "uses GPL-only symbol '%s'\n", m, e, s);
1336                 break;
1337         case export_unused_gpl:
1338                 fatal("modpost: GPL-incompatible module %s%s "
1339                       "uses GPL-only symbol marked UNUSED '%s'\n", m, e, s);
1340                 break;
1341         case export_gpl_future:
1342                 warn("modpost: GPL-incompatible module %s%s "
1343                       "uses future GPL-only symbol '%s'\n", m, e, s);
1344                 break;
1345         case export_plain:
1346         case export_unused:
1347         case export_unknown:
1348                 /* ignore */
1349                 break;
1350         }
1351 }
1352
1353 static void check_for_unused(enum export exp, const char* m, const char* s)
1354 {
1355         const char *e = is_vmlinux(m) ?"":".ko";
1356
1357         switch (exp) {
1358         case export_unused:
1359         case export_unused_gpl:
1360                 warn("modpost: module %s%s "
1361                       "uses symbol '%s' marked UNUSED\n", m, e, s);
1362                 break;
1363         default:
1364                 /* ignore */
1365                 break;
1366         }
1367 }
1368
1369 static void check_exports(struct module *mod)
1370 {
1371         struct symbol *s, *exp;
1372
1373         for (s = mod->unres; s; s = s->next) {
1374                 const char *basename;
1375                 exp = find_symbol(s->name);
1376                 if (!exp || exp->module == mod)
1377                         continue;
1378                 basename = strrchr(mod->name, '/');
1379                 if (basename)
1380                         basename++;
1381                 else
1382                         basename = mod->name;
1383                 if (!mod->gpl_compatible)
1384                         check_for_gpl_usage(exp->export, basename, exp->name);
1385                 check_for_unused(exp->export, basename, exp->name);
1386         }
1387 }
1388
1389 /**
1390  * Header for the generated file
1391  **/
1392 static void add_header(struct buffer *b, struct module *mod)
1393 {
1394         buf_printf(b, "#include <linux/module.h>\n");
1395         buf_printf(b, "#include <linux/vermagic.h>\n");
1396         buf_printf(b, "#include <linux/compiler.h>\n");
1397         buf_printf(b, "\n");
1398         buf_printf(b, "MODULE_INFO(vermagic, VERMAGIC_STRING);\n");
1399         buf_printf(b, "\n");
1400         buf_printf(b, "struct module __this_module\n");
1401         buf_printf(b, "__attribute__((section(\".gnu.linkonce.this_module\"))) = {\n");
1402         buf_printf(b, " .name = KBUILD_MODNAME,\n");
1403         if (mod->has_init)
1404                 buf_printf(b, " .init = init_module,\n");
1405         if (mod->has_cleanup)
1406                 buf_printf(b, "#ifdef CONFIG_MODULE_UNLOAD\n"
1407                               " .exit = cleanup_module,\n"
1408                               "#endif\n");
1409         buf_printf(b, " .arch = MODULE_ARCH_INIT,\n");
1410         buf_printf(b, "};\n");
1411 }
1412
1413 /**
1414  * Record CRCs for unresolved symbols
1415  **/
1416 static int add_versions(struct buffer *b, struct module *mod)
1417 {
1418         struct symbol *s, *exp;
1419         int err = 0;
1420
1421         for (s = mod->unres; s; s = s->next) {
1422                 exp = find_symbol(s->name);
1423                 if (!exp || exp->module == mod) {
1424                         if (have_vmlinux && !s->weak) {
1425                                 if (warn_unresolved) {
1426                                         warn("\"%s\" [%s.ko] undefined!\n",
1427                                              s->name, mod->name);
1428                                 } else {
1429                                         merror("\"%s\" [%s.ko] undefined!\n",
1430                                                   s->name, mod->name);
1431                                         err = 1;
1432                                 }
1433                         }
1434                         continue;
1435                 }
1436                 s->module = exp->module;
1437                 s->crc_valid = exp->crc_valid;
1438                 s->crc = exp->crc;
1439         }
1440
1441         if (!modversions)
1442                 return err;
1443
1444         buf_printf(b, "\n");
1445         buf_printf(b, "static const struct modversion_info ____versions[]\n");
1446         buf_printf(b, "__attribute_used__\n");
1447         buf_printf(b, "__attribute__((section(\"__versions\"))) = {\n");
1448
1449         for (s = mod->unres; s; s = s->next) {
1450                 if (!s->module) {
1451                         continue;
1452                 }
1453                 if (!s->crc_valid) {
1454                         warn("\"%s\" [%s.ko] has no CRC!\n",
1455                                 s->name, mod->name);
1456                         continue;
1457                 }
1458                 buf_printf(b, "\t{ %#8x, \"%s\" },\n", s->crc, s->name);
1459         }
1460
1461         buf_printf(b, "};\n");
1462
1463         return err;
1464 }
1465
1466 static void add_depends(struct buffer *b, struct module *mod,
1467                         struct module *modules)
1468 {
1469         struct symbol *s;
1470         struct module *m;
1471         int first = 1;
1472
1473         for (m = modules; m; m = m->next) {
1474                 m->seen = is_vmlinux(m->name);
1475         }
1476
1477         buf_printf(b, "\n");
1478         buf_printf(b, "static const char __module_depends[]\n");
1479         buf_printf(b, "__attribute_used__\n");
1480         buf_printf(b, "__attribute__((section(\".modinfo\"))) =\n");
1481         buf_printf(b, "\"depends=");
1482         for (s = mod->unres; s; s = s->next) {
1483                 const char *p;
1484                 if (!s->module)
1485                         continue;
1486
1487                 if (s->module->seen)
1488                         continue;
1489
1490                 s->module->seen = 1;
1491                 if ((p = strrchr(s->module->name, '/')) != NULL)
1492                         p++;
1493                 else
1494                         p = s->module->name;
1495                 buf_printf(b, "%s%s", first ? "" : ",", p);
1496                 first = 0;
1497         }
1498         buf_printf(b, "\";\n");
1499 }
1500
1501 static void add_srcversion(struct buffer *b, struct module *mod)
1502 {
1503         if (mod->srcversion[0]) {
1504                 buf_printf(b, "\n");
1505                 buf_printf(b, "MODULE_INFO(srcversion, \"%s\");\n",
1506                            mod->srcversion);
1507         }
1508 }
1509
1510 static void write_if_changed(struct buffer *b, const char *fname)
1511 {
1512         char *tmp;
1513         FILE *file;
1514         struct stat st;
1515
1516         file = fopen(fname, "r");
1517         if (!file)
1518                 goto write;
1519
1520         if (fstat(fileno(file), &st) < 0)
1521                 goto close_write;
1522
1523         if (st.st_size != b->pos)
1524                 goto close_write;
1525
1526         tmp = NOFAIL(malloc(b->pos));
1527         if (fread(tmp, 1, b->pos, file) != b->pos)
1528                 goto free_write;
1529
1530         if (memcmp(tmp, b->p, b->pos) != 0)
1531                 goto free_write;
1532
1533         free(tmp);
1534         fclose(file);
1535         return;
1536
1537  free_write:
1538         free(tmp);
1539  close_write:
1540         fclose(file);
1541  write:
1542         file = fopen(fname, "w");
1543         if (!file) {
1544                 perror(fname);
1545                 exit(1);
1546         }
1547         if (fwrite(b->p, 1, b->pos, file) != b->pos) {
1548                 perror(fname);
1549                 exit(1);
1550         }
1551         fclose(file);
1552 }
1553
1554 /* parse Module.symvers file. line format:
1555  * 0x12345678<tab>symbol<tab>module[[<tab>export]<tab>something]
1556  **/
1557 static void read_dump(const char *fname, unsigned int kernel)
1558 {
1559         unsigned long size, pos = 0;
1560         void *file = grab_file(fname, &size);
1561         char *line;
1562
1563         if (!file)
1564                 /* No symbol versions, silently ignore */
1565                 return;
1566
1567         while ((line = get_next_line(&pos, file, size))) {
1568                 char *symname, *modname, *d, *export, *end;
1569                 unsigned int crc;
1570                 struct module *mod;
1571                 struct symbol *s;
1572
1573                 if (!(symname = strchr(line, '\t')))
1574                         goto fail;
1575                 *symname++ = '\0';
1576                 if (!(modname = strchr(symname, '\t')))
1577                         goto fail;
1578                 *modname++ = '\0';
1579                 if ((export = strchr(modname, '\t')) != NULL)
1580                         *export++ = '\0';
1581                 if (export && ((end = strchr(export, '\t')) != NULL))
1582                         *end = '\0';
1583                 crc = strtoul(line, &d, 16);
1584                 if (*symname == '\0' || *modname == '\0' || *d != '\0')
1585                         goto fail;
1586
1587                 if (!(mod = find_module(modname))) {
1588                         if (is_vmlinux(modname)) {
1589                                 have_vmlinux = 1;
1590                         }
1591                         mod = new_module(NOFAIL(strdup(modname)));
1592                         mod->skip = 1;
1593                 }
1594                 s = sym_add_exported(symname, mod, export_no(export));
1595                 s->kernel    = kernel;
1596                 s->preloaded = 1;
1597                 sym_update_crc(symname, mod, crc, export_no(export));
1598         }
1599         return;
1600 fail:
1601         fatal("parse error in symbol dump file\n");
1602 }
1603
1604 /* For normal builds always dump all symbols.
1605  * For external modules only dump symbols
1606  * that are not read from kernel Module.symvers.
1607  **/
1608 static int dump_sym(struct symbol *sym)
1609 {
1610         if (!external_module)
1611                 return 1;
1612         if (sym->vmlinux || sym->kernel)
1613                 return 0;
1614         return 1;
1615 }
1616
1617 static void write_dump(const char *fname)
1618 {
1619         struct buffer buf = { };
1620         struct symbol *symbol;
1621         int n;
1622
1623         for (n = 0; n < SYMBOL_HASH_SIZE ; n++) {
1624                 symbol = symbolhash[n];
1625                 while (symbol) {
1626                         if (dump_sym(symbol))
1627                                 buf_printf(&buf, "0x%08x\t%s\t%s\t%s\n",
1628                                         symbol->crc, symbol->name,
1629                                         symbol->module->name,
1630                                         export_str(symbol->export));
1631                         symbol = symbol->next;
1632                 }
1633         }
1634         write_if_changed(&buf, fname);
1635 }
1636
1637 int main(int argc, char **argv)
1638 {
1639         struct module *mod;
1640         struct buffer buf = { };
1641         char fname[SZ];
1642         char *kernel_read = NULL, *module_read = NULL;
1643         char *dump_write = NULL;
1644         int opt;
1645         int err;
1646
1647         while ((opt = getopt(argc, argv, "i:I:mo:aw")) != -1) {
1648                 switch(opt) {
1649                         case 'i':
1650                                 kernel_read = optarg;
1651                                 break;
1652                         case 'I':
1653                                 module_read = optarg;
1654                                 external_module = 1;
1655                                 break;
1656                         case 'm':
1657                                 modversions = 1;
1658                                 break;
1659                         case 'o':
1660                                 dump_write = optarg;
1661                                 break;
1662                         case 'a':
1663                                 all_versions = 1;
1664                                 break;
1665                         case 'w':
1666                                 warn_unresolved = 1;
1667                                 break;
1668                         default:
1669                                 exit(1);
1670                 }
1671         }
1672
1673         if (kernel_read)
1674                 read_dump(kernel_read, 1);
1675         if (module_read)
1676                 read_dump(module_read, 0);
1677
1678         while (optind < argc) {
1679                 read_symbols(argv[optind++]);
1680         }
1681
1682         for (mod = modules; mod; mod = mod->next) {
1683                 if (mod->skip)
1684                         continue;
1685                 check_exports(mod);
1686         }
1687
1688         err = 0;
1689
1690         for (mod = modules; mod; mod = mod->next) {
1691                 if (mod->skip)
1692                         continue;
1693
1694                 buf.pos = 0;
1695
1696                 add_header(&buf, mod);
1697                 err |= add_versions(&buf, mod);
1698                 add_depends(&buf, mod, modules);
1699                 add_moddevtable(&buf, mod);
1700                 add_srcversion(&buf, mod);
1701
1702                 sprintf(fname, "%s.mod.c", mod->name);
1703                 write_if_changed(&buf, fname);
1704         }
1705
1706         if (dump_write)
1707                 write_dump(dump_write);
1708
1709         return err;
1710 }