[NETLINK] genetlink: Fix bugs spotted by Andrew Morton.
[linux-2.6/linux-acpi-2.6/ibm-acpi-2.6.git] / scripts / mod / modpost.c
blobf70ff13d481834b14ceb926879a6a3a2e99e1894
1 /* Postprocess module symbol versions
3 * Copyright 2003 Kai Germaschewski
4 * Copyright 2002-2004 Rusty Russell, IBM Corporation
6 * Based in part on module-init-tools/depmod.c,file2alias
8 * This software may be used and distributed according to the terms
9 * of the GNU General Public License, incorporated herein by reference.
11 * Usage: modpost vmlinux module1.o module2.o ...
14 #include <ctype.h>
15 #include "modpost.h"
17 /* Are we using CONFIG_MODVERSIONS? */
18 int modversions = 0;
19 /* Warn about undefined symbols? (do so if we have vmlinux) */
20 int have_vmlinux = 0;
21 /* Is CONFIG_MODULE_SRCVERSION_ALL set? */
22 static int all_versions = 0;
24 void
25 fatal(const char *fmt, ...)
27 va_list arglist;
29 fprintf(stderr, "FATAL: ");
31 va_start(arglist, fmt);
32 vfprintf(stderr, fmt, arglist);
33 va_end(arglist);
35 exit(1);
38 void
39 warn(const char *fmt, ...)
41 va_list arglist;
43 fprintf(stderr, "WARNING: ");
45 va_start(arglist, fmt);
46 vfprintf(stderr, fmt, arglist);
47 va_end(arglist);
50 void *do_nofail(void *ptr, const char *expr)
52 if (!ptr) {
53 fatal("modpost: Memory allocation failure: %s.\n", expr);
55 return ptr;
58 /* A list of all modules we processed */
60 static struct module *modules;
62 struct module *
63 find_module(char *modname)
65 struct module *mod;
67 for (mod = modules; mod; mod = mod->next)
68 if (strcmp(mod->name, modname) == 0)
69 break;
70 return mod;
73 struct module *
74 new_module(char *modname)
76 struct module *mod;
77 char *p, *s;
79 mod = NOFAIL(malloc(sizeof(*mod)));
80 memset(mod, 0, sizeof(*mod));
81 p = NOFAIL(strdup(modname));
83 /* strip trailing .o */
84 if ((s = strrchr(p, '.')) != NULL)
85 if (strcmp(s, ".o") == 0)
86 *s = '\0';
88 /* add to list */
89 mod->name = p;
90 mod->next = modules;
91 modules = mod;
93 return mod;
96 /* A hash of all exported symbols,
97 * struct symbol is also used for lists of unresolved symbols */
99 #define SYMBOL_HASH_SIZE 1024
101 struct symbol {
102 struct symbol *next;
103 struct module *module;
104 unsigned int crc;
105 int crc_valid;
106 unsigned int weak:1;
107 char name[0];
110 static struct symbol *symbolhash[SYMBOL_HASH_SIZE];
112 /* This is based on the hash agorithm from gdbm, via tdb */
113 static inline unsigned int tdb_hash(const char *name)
115 unsigned value; /* Used to compute the hash value. */
116 unsigned i; /* Used to cycle through random values. */
118 /* Set the initial value from the key size. */
119 for (value = 0x238F13AF * strlen(name), i=0; name[i]; i++)
120 value = (value + (((unsigned char *)name)[i] << (i*5 % 24)));
122 return (1103515243 * value + 12345);
125 /* Allocate a new symbols for use in the hash of exported symbols or
126 * the list of unresolved symbols per module */
128 struct symbol *
129 alloc_symbol(const char *name, unsigned int weak, struct symbol *next)
131 struct symbol *s = NOFAIL(malloc(sizeof(*s) + strlen(name) + 1));
133 memset(s, 0, sizeof(*s));
134 strcpy(s->name, name);
135 s->weak = weak;
136 s->next = next;
137 return s;
140 /* For the hash of exported symbols */
142 void
143 new_symbol(const char *name, struct module *module, unsigned int *crc)
145 unsigned int hash;
146 struct symbol *new;
148 hash = tdb_hash(name) % SYMBOL_HASH_SIZE;
149 new = symbolhash[hash] = alloc_symbol(name, 0, symbolhash[hash]);
150 new->module = module;
151 if (crc) {
152 new->crc = *crc;
153 new->crc_valid = 1;
157 struct symbol *
158 find_symbol(const char *name)
160 struct symbol *s;
162 /* For our purposes, .foo matches foo. PPC64 needs this. */
163 if (name[0] == '.')
164 name++;
166 for (s = symbolhash[tdb_hash(name) % SYMBOL_HASH_SIZE]; s; s=s->next) {
167 if (strcmp(s->name, name) == 0)
168 return s;
170 return NULL;
173 /* Add an exported symbol - it may have already been added without a
174 * CRC, in this case just update the CRC */
175 void
176 add_exported_symbol(const char *name, struct module *module, unsigned int *crc)
178 struct symbol *s = find_symbol(name);
180 if (!s) {
181 new_symbol(name, module, crc);
182 return;
184 if (crc) {
185 s->crc = *crc;
186 s->crc_valid = 1;
190 void *
191 grab_file(const char *filename, unsigned long *size)
193 struct stat st;
194 void *map;
195 int fd;
197 fd = open(filename, O_RDONLY);
198 if (fd < 0 || fstat(fd, &st) != 0)
199 return NULL;
201 *size = st.st_size;
202 map = mmap(NULL, *size, PROT_READ|PROT_WRITE, MAP_PRIVATE, fd, 0);
203 close(fd);
205 if (map == MAP_FAILED)
206 return NULL;
207 return map;
211 Return a copy of the next line in a mmap'ed file.
212 spaces in the beginning of the line is trimmed away.
213 Return a pointer to a static buffer.
215 char*
216 get_next_line(unsigned long *pos, void *file, unsigned long size)
218 static char line[4096];
219 int skip = 1;
220 size_t len = 0;
221 signed char *p = (signed char *)file + *pos;
222 char *s = line;
224 for (; *pos < size ; (*pos)++)
226 if (skip && isspace(*p)) {
227 p++;
228 continue;
230 skip = 0;
231 if (*p != '\n' && (*pos < size)) {
232 len++;
233 *s++ = *p++;
234 if (len > 4095)
235 break; /* Too long, stop */
236 } else {
237 /* End of string */
238 *s = '\0';
239 return line;
242 /* End of buffer */
243 return NULL;
246 void
247 release_file(void *file, unsigned long size)
249 munmap(file, size);
252 void
253 parse_elf(struct elf_info *info, const char *filename)
255 unsigned int i;
256 Elf_Ehdr *hdr = info->hdr;
257 Elf_Shdr *sechdrs;
258 Elf_Sym *sym;
260 hdr = grab_file(filename, &info->size);
261 if (!hdr) {
262 perror(filename);
263 abort();
265 info->hdr = hdr;
266 if (info->size < sizeof(*hdr))
267 goto truncated;
269 /* Fix endianness in ELF header */
270 hdr->e_shoff = TO_NATIVE(hdr->e_shoff);
271 hdr->e_shstrndx = TO_NATIVE(hdr->e_shstrndx);
272 hdr->e_shnum = TO_NATIVE(hdr->e_shnum);
273 hdr->e_machine = TO_NATIVE(hdr->e_machine);
274 sechdrs = (void *)hdr + hdr->e_shoff;
275 info->sechdrs = sechdrs;
277 /* Fix endianness in section headers */
278 for (i = 0; i < hdr->e_shnum; i++) {
279 sechdrs[i].sh_type = TO_NATIVE(sechdrs[i].sh_type);
280 sechdrs[i].sh_offset = TO_NATIVE(sechdrs[i].sh_offset);
281 sechdrs[i].sh_size = TO_NATIVE(sechdrs[i].sh_size);
282 sechdrs[i].sh_link = TO_NATIVE(sechdrs[i].sh_link);
283 sechdrs[i].sh_name = TO_NATIVE(sechdrs[i].sh_name);
285 /* Find symbol table. */
286 for (i = 1; i < hdr->e_shnum; i++) {
287 const char *secstrings
288 = (void *)hdr + sechdrs[hdr->e_shstrndx].sh_offset;
290 if (sechdrs[i].sh_offset > info->size)
291 goto truncated;
292 if (strcmp(secstrings+sechdrs[i].sh_name, ".modinfo") == 0) {
293 info->modinfo = (void *)hdr + sechdrs[i].sh_offset;
294 info->modinfo_len = sechdrs[i].sh_size;
296 if (sechdrs[i].sh_type != SHT_SYMTAB)
297 continue;
299 info->symtab_start = (void *)hdr + sechdrs[i].sh_offset;
300 info->symtab_stop = (void *)hdr + sechdrs[i].sh_offset
301 + sechdrs[i].sh_size;
302 info->strtab = (void *)hdr +
303 sechdrs[sechdrs[i].sh_link].sh_offset;
305 if (!info->symtab_start) {
306 fprintf(stderr, "modpost: %s no symtab?\n", filename);
307 abort();
309 /* Fix endianness in symbols */
310 for (sym = info->symtab_start; sym < info->symtab_stop; sym++) {
311 sym->st_shndx = TO_NATIVE(sym->st_shndx);
312 sym->st_name = TO_NATIVE(sym->st_name);
313 sym->st_value = TO_NATIVE(sym->st_value);
314 sym->st_size = TO_NATIVE(sym->st_size);
316 return;
318 truncated:
319 fprintf(stderr, "modpost: %s is truncated.\n", filename);
320 abort();
323 void
324 parse_elf_finish(struct elf_info *info)
326 release_file(info->hdr, info->size);
329 #define CRC_PFX "__crc_"
330 #define KSYMTAB_PFX "__ksymtab_"
332 void
333 handle_modversions(struct module *mod, struct elf_info *info,
334 Elf_Sym *sym, const char *symname)
336 unsigned int crc;
338 switch (sym->st_shndx) {
339 case SHN_COMMON:
340 fprintf(stderr, "*** Warning: \"%s\" [%s] is COMMON symbol\n",
341 symname, mod->name);
342 break;
343 case SHN_ABS:
344 /* CRC'd symbol */
345 if (memcmp(symname, CRC_PFX, strlen(CRC_PFX)) == 0) {
346 crc = (unsigned int) sym->st_value;
347 add_exported_symbol(symname + strlen(CRC_PFX),
348 mod, &crc);
350 break;
351 case SHN_UNDEF:
352 /* undefined symbol */
353 if (ELF_ST_BIND(sym->st_info) != STB_GLOBAL &&
354 ELF_ST_BIND(sym->st_info) != STB_WEAK)
355 break;
356 /* ignore global offset table */
357 if (strcmp(symname, "_GLOBAL_OFFSET_TABLE_") == 0)
358 break;
359 /* ignore __this_module, it will be resolved shortly */
360 if (strcmp(symname, MODULE_SYMBOL_PREFIX "__this_module") == 0)
361 break;
362 /* cope with newer glibc (2.3.4 or higher) STT_ definition in elf.h */
363 #if defined(STT_REGISTER) || defined(STT_SPARC_REGISTER)
364 /* add compatibility with older glibc */
365 #ifndef STT_SPARC_REGISTER
366 #define STT_SPARC_REGISTER STT_REGISTER
367 #endif
368 if (info->hdr->e_machine == EM_SPARC ||
369 info->hdr->e_machine == EM_SPARCV9) {
370 /* Ignore register directives. */
371 if (ELF_ST_TYPE(sym->st_info) == STT_SPARC_REGISTER)
372 break;
373 if (symname[0] == '.') {
374 char *munged = strdup(symname);
375 munged[0] = '_';
376 munged[1] = toupper(munged[1]);
377 symname = munged;
380 #endif
382 if (memcmp(symname, MODULE_SYMBOL_PREFIX,
383 strlen(MODULE_SYMBOL_PREFIX)) == 0)
384 mod->unres = alloc_symbol(symname +
385 strlen(MODULE_SYMBOL_PREFIX),
386 ELF_ST_BIND(sym->st_info) == STB_WEAK,
387 mod->unres);
388 break;
389 default:
390 /* All exported symbols */
391 if (memcmp(symname, KSYMTAB_PFX, strlen(KSYMTAB_PFX)) == 0) {
392 add_exported_symbol(symname + strlen(KSYMTAB_PFX),
393 mod, NULL);
395 if (strcmp(symname, MODULE_SYMBOL_PREFIX "init_module") == 0)
396 mod->has_init = 1;
397 if (strcmp(symname, MODULE_SYMBOL_PREFIX "cleanup_module") == 0)
398 mod->has_cleanup = 1;
399 break;
404 is_vmlinux(const char *modname)
406 const char *myname;
408 if ((myname = strrchr(modname, '/')))
409 myname++;
410 else
411 myname = modname;
413 return strcmp(myname, "vmlinux") == 0;
416 /* Parse tag=value strings from .modinfo section */
417 static char *next_string(char *string, unsigned long *secsize)
419 /* Skip non-zero chars */
420 while (string[0]) {
421 string++;
422 if ((*secsize)-- <= 1)
423 return NULL;
426 /* Skip any zero padding. */
427 while (!string[0]) {
428 string++;
429 if ((*secsize)-- <= 1)
430 return NULL;
432 return string;
435 static char *get_modinfo(void *modinfo, unsigned long modinfo_len,
436 const char *tag)
438 char *p;
439 unsigned int taglen = strlen(tag);
440 unsigned long size = modinfo_len;
442 for (p = modinfo; p; p = next_string(p, &size)) {
443 if (strncmp(p, tag, taglen) == 0 && p[taglen] == '=')
444 return p + taglen + 1;
446 return NULL;
449 void
450 read_symbols(char *modname)
452 const char *symname;
453 char *version;
454 struct module *mod;
455 struct elf_info info = { };
456 Elf_Sym *sym;
458 parse_elf(&info, modname);
460 mod = new_module(modname);
462 /* When there's no vmlinux, don't print warnings about
463 * unresolved symbols (since there'll be too many ;) */
464 if (is_vmlinux(modname)) {
465 unsigned int fake_crc = 0;
466 have_vmlinux = 1;
467 add_exported_symbol("struct_module", mod, &fake_crc);
468 mod->skip = 1;
471 for (sym = info.symtab_start; sym < info.symtab_stop; sym++) {
472 symname = info.strtab + sym->st_name;
474 handle_modversions(mod, &info, sym, symname);
475 handle_moddevtable(mod, &info, sym, symname);
478 version = get_modinfo(info.modinfo, info.modinfo_len, "version");
479 if (version)
480 maybe_frob_rcs_version(modname, version, info.modinfo,
481 version - (char *)info.hdr);
482 if (version || (all_versions && !is_vmlinux(modname)))
483 get_src_version(modname, mod->srcversion,
484 sizeof(mod->srcversion)-1);
486 parse_elf_finish(&info);
488 /* Our trick to get versioning for struct_module - it's
489 * never passed as an argument to an exported function, so
490 * the automatic versioning doesn't pick it up, but it's really
491 * important anyhow */
492 if (modversions)
493 mod->unres = alloc_symbol("struct_module", 0, mod->unres);
496 #define SZ 500
498 /* We first write the generated file into memory using the
499 * following helper, then compare to the file on disk and
500 * only update the later if anything changed */
502 void __attribute__((format(printf, 2, 3)))
503 buf_printf(struct buffer *buf, const char *fmt, ...)
505 char tmp[SZ];
506 int len;
507 va_list ap;
509 va_start(ap, fmt);
510 len = vsnprintf(tmp, SZ, fmt, ap);
511 if (buf->size - buf->pos < len + 1) {
512 buf->size += 128;
513 buf->p = realloc(buf->p, buf->size);
515 strncpy(buf->p + buf->pos, tmp, len + 1);
516 buf->pos += len;
517 va_end(ap);
520 void
521 buf_write(struct buffer *buf, const char *s, int len)
523 if (buf->size - buf->pos < len) {
524 buf->size += len;
525 buf->p = realloc(buf->p, buf->size);
527 strncpy(buf->p + buf->pos, s, len);
528 buf->pos += len;
531 /* Header for the generated file */
533 void
534 add_header(struct buffer *b, struct module *mod)
536 buf_printf(b, "#include <linux/module.h>\n");
537 buf_printf(b, "#include <linux/vermagic.h>\n");
538 buf_printf(b, "#include <linux/compiler.h>\n");
539 buf_printf(b, "\n");
540 buf_printf(b, "MODULE_INFO(vermagic, VERMAGIC_STRING);\n");
541 buf_printf(b, "\n");
542 buf_printf(b, "struct module __this_module\n");
543 buf_printf(b, "__attribute__((section(\".gnu.linkonce.this_module\"))) = {\n");
544 buf_printf(b, " .name = KBUILD_MODNAME,\n");
545 if (mod->has_init)
546 buf_printf(b, " .init = init_module,\n");
547 if (mod->has_cleanup)
548 buf_printf(b, "#ifdef CONFIG_MODULE_UNLOAD\n"
549 " .exit = cleanup_module,\n"
550 "#endif\n");
551 buf_printf(b, "};\n");
554 /* Record CRCs for unresolved symbols */
556 void
557 add_versions(struct buffer *b, struct module *mod)
559 struct symbol *s, *exp;
561 for (s = mod->unres; s; s = s->next) {
562 exp = find_symbol(s->name);
563 if (!exp || exp->module == mod) {
564 if (have_vmlinux && !s->weak)
565 fprintf(stderr, "*** Warning: \"%s\" [%s.ko] "
566 "undefined!\n", s->name, mod->name);
567 continue;
569 s->module = exp->module;
570 s->crc_valid = exp->crc_valid;
571 s->crc = exp->crc;
574 if (!modversions)
575 return;
577 buf_printf(b, "\n");
578 buf_printf(b, "static const struct modversion_info ____versions[]\n");
579 buf_printf(b, "__attribute_used__\n");
580 buf_printf(b, "__attribute__((section(\"__versions\"))) = {\n");
582 for (s = mod->unres; s; s = s->next) {
583 if (!s->module) {
584 continue;
586 if (!s->crc_valid) {
587 fprintf(stderr, "*** Warning: \"%s\" [%s.ko] "
588 "has no CRC!\n",
589 s->name, mod->name);
590 continue;
592 buf_printf(b, "\t{ %#8x, \"%s\" },\n", s->crc, s->name);
595 buf_printf(b, "};\n");
598 void
599 add_depends(struct buffer *b, struct module *mod, struct module *modules)
601 struct symbol *s;
602 struct module *m;
603 int first = 1;
605 for (m = modules; m; m = m->next) {
606 m->seen = is_vmlinux(m->name);
609 buf_printf(b, "\n");
610 buf_printf(b, "static const char __module_depends[]\n");
611 buf_printf(b, "__attribute_used__\n");
612 buf_printf(b, "__attribute__((section(\".modinfo\"))) =\n");
613 buf_printf(b, "\"depends=");
614 for (s = mod->unres; s; s = s->next) {
615 if (!s->module)
616 continue;
618 if (s->module->seen)
619 continue;
621 s->module->seen = 1;
622 buf_printf(b, "%s%s", first ? "" : ",",
623 strrchr(s->module->name, '/') + 1);
624 first = 0;
626 buf_printf(b, "\";\n");
629 void
630 add_srcversion(struct buffer *b, struct module *mod)
632 if (mod->srcversion[0]) {
633 buf_printf(b, "\n");
634 buf_printf(b, "MODULE_INFO(srcversion, \"%s\");\n",
635 mod->srcversion);
639 void
640 write_if_changed(struct buffer *b, const char *fname)
642 char *tmp;
643 FILE *file;
644 struct stat st;
646 file = fopen(fname, "r");
647 if (!file)
648 goto write;
650 if (fstat(fileno(file), &st) < 0)
651 goto close_write;
653 if (st.st_size != b->pos)
654 goto close_write;
656 tmp = NOFAIL(malloc(b->pos));
657 if (fread(tmp, 1, b->pos, file) != b->pos)
658 goto free_write;
660 if (memcmp(tmp, b->p, b->pos) != 0)
661 goto free_write;
663 free(tmp);
664 fclose(file);
665 return;
667 free_write:
668 free(tmp);
669 close_write:
670 fclose(file);
671 write:
672 file = fopen(fname, "w");
673 if (!file) {
674 perror(fname);
675 exit(1);
677 if (fwrite(b->p, 1, b->pos, file) != b->pos) {
678 perror(fname);
679 exit(1);
681 fclose(file);
684 void
685 read_dump(const char *fname)
687 unsigned long size, pos = 0;
688 void *file = grab_file(fname, &size);
689 char *line;
691 if (!file)
692 /* No symbol versions, silently ignore */
693 return;
695 while ((line = get_next_line(&pos, file, size))) {
696 char *symname, *modname, *d;
697 unsigned int crc;
698 struct module *mod;
700 if (!(symname = strchr(line, '\t')))
701 goto fail;
702 *symname++ = '\0';
703 if (!(modname = strchr(symname, '\t')))
704 goto fail;
705 *modname++ = '\0';
706 if (strchr(modname, '\t'))
707 goto fail;
708 crc = strtoul(line, &d, 16);
709 if (*symname == '\0' || *modname == '\0' || *d != '\0')
710 goto fail;
712 if (!(mod = find_module(modname))) {
713 if (is_vmlinux(modname)) {
714 have_vmlinux = 1;
716 mod = new_module(NOFAIL(strdup(modname)));
717 mod->skip = 1;
719 add_exported_symbol(symname, mod, &crc);
721 return;
722 fail:
723 fatal("parse error in symbol dump file\n");
726 void
727 write_dump(const char *fname)
729 struct buffer buf = { };
730 struct symbol *symbol;
731 int n;
733 for (n = 0; n < SYMBOL_HASH_SIZE ; n++) {
734 symbol = symbolhash[n];
735 while (symbol) {
736 symbol = symbol->next;
740 for (n = 0; n < SYMBOL_HASH_SIZE ; n++) {
741 symbol = symbolhash[n];
742 while (symbol) {
743 buf_printf(&buf, "0x%08x\t%s\t%s\n", symbol->crc,
744 symbol->name, symbol->module->name);
745 symbol = symbol->next;
748 write_if_changed(&buf, fname);
752 main(int argc, char **argv)
754 struct module *mod;
755 struct buffer buf = { };
756 char fname[SZ];
757 char *dump_read = NULL, *dump_write = NULL;
758 int opt;
760 while ((opt = getopt(argc, argv, "i:mo:a")) != -1) {
761 switch(opt) {
762 case 'i':
763 dump_read = optarg;
764 break;
765 case 'm':
766 modversions = 1;
767 break;
768 case 'o':
769 dump_write = optarg;
770 break;
771 case 'a':
772 all_versions = 1;
773 break;
774 default:
775 exit(1);
779 if (dump_read)
780 read_dump(dump_read);
782 while (optind < argc) {
783 read_symbols(argv[optind++]);
786 for (mod = modules; mod; mod = mod->next) {
787 if (mod->skip)
788 continue;
790 buf.pos = 0;
792 add_header(&buf, mod);
793 add_versions(&buf, mod);
794 add_depends(&buf, mod, modules);
795 add_moddevtable(&buf, mod);
796 add_srcversion(&buf, mod);
798 sprintf(fname, "%s.mod.c", mod->name);
799 write_if_changed(&buf, fname);
802 if (dump_write)
803 write_dump(dump_write);
805 return 0;