objcommon.c

   1 /*  Copyright (C) 2008  Jeffrey Brian Arnold <jbarnold@mit.edu>
   2  *
   3  *  This program is free software; you can redistribute it and/or modify
   4  *  it under the terms of the GNU General Public License, version 2.
   5  *
   6  *  This program is distributed in the hope that it will be useful,
   7  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
   8  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   9  *  GNU General Public License for more details.
  10  *
  11  *  You should have received a copy of the GNU General Public License
  12  *  along with this program; if not, write to the Free Software
  13  *  Foundation, Inc., 51 Franklin Street - Fifth Floor, Boston, MA
  14  *  02110-1301, USA.
  15  */
  16
  17 #include "objcommon.h"
  18
  19 void vec_do_reserve(void **data, size_t *mem_size, size_t new_size)
  20 {
  21         if (new_size > *mem_size || new_size * 2 < *mem_size) {
  22                 if (new_size < *mem_size * 2)
  23                         new_size = *mem_size * 2;
  24                 *data = realloc(*data, new_size);
  25                 assert(new_size == 0 || *data != NULL);
  26                 *mem_size = new_size;
  27         }
  28 }
  29
  30 void get_syms(bfd *abfd, struct asymbolp_vec *syms)
  31 {
  32         long storage_needed = bfd_get_symtab_upper_bound(abfd);
  33         if (storage_needed == 0)
  34                 return;
  35         assert(storage_needed >= 0);
  36
  37         vec_init(syms);
  38         vec_reserve(syms, storage_needed);
  39         vec_resize(syms, bfd_canonicalize_symtab(abfd, syms->data));
  40         assert(syms->size >= 0);
  41 }
  42
  43 struct supersect *fetch_supersect(bfd *abfd, asection *sect,
  44                                   struct asymbolp_vec *syms)
  45 {
  46         static struct supersect *supersects = NULL;
  47
  48         struct supersect *ss;
  49         for (ss = supersects; ss != NULL; ss = ss->next) {
  50                 if (strcmp(sect->name, ss->name) == 0 && ss->parent == abfd)
  51                         return ss;
  52         }
  53
  54         struct supersect *new = malloc(sizeof(*new));
  55         new->parent = abfd;
  56         new->name = malloc(strlen(sect->name) + 1);
  57         strcpy(new->name, sect->name);
  58         new->next = supersects;
  59         supersects = new;
  60
  61         vec_init(&new->contents);
  62         vec_resize(&new->contents, bfd_get_section_size(sect));
  63         assert(bfd_get_section_contents
  64                (abfd, sect, new->contents.data, 0, new->contents.size));
  65         new->alignment = 1 << bfd_get_section_alignment(abfd, sect);
  66
  67         vec_init(&new->relocs);
  68         vec_reserve(&new->relocs, bfd_get_reloc_upper_bound(abfd, sect));
  69         vec_resize(&new->relocs,
  70                    bfd_canonicalize_reloc(abfd, sect, new->relocs.data,
  71                                           syms->data));
  72         assert(new->relocs.size >= 0);
  73
  74         return new;
  75 }
  76
  77 int label_offset(const char *sym_name)
  78 {
  79         int i;
  80         for (i = 0;
  81              sym_name[i] != 0 && sym_name[i + 1] != 0 && sym_name[i + 2] != 0
  82              && sym_name[i + 3] != 0; i++) {
  83                 if (sym_name[i] == '_' && sym_name[i + 1] == '_'
  84                     && sym_name[i + 2] == '_' && sym_name[i + 3] == '_')
  85                         return i + 4;
  86         }
  87         return -1;
  88 }
  89
  90 const char *only_label(const char *sym_name)
  91 {
  92         int offset = label_offset(sym_name);
  93         if (offset == -1)
  94                 return NULL;
  95         return &sym_name[offset];
  96 }
  97
  98 const char *dup_wolabel(const char *sym_name)
  99 {
 100         int offset, entire_strlen, label_strlen, new_strlen;
 101         char *newstr;
 102
 103         offset = label_offset(sym_name);
 104         if (offset == -1)
 105                 label_strlen = 0;
 106         else
 107                 label_strlen = strlen(&sym_name[offset]) + strlen("____");
 108
 109         entire_strlen = strlen(sym_name);
 110         new_strlen = entire_strlen - label_strlen;
 111         newstr = malloc(new_strlen + 1);
 112         memcpy(newstr, sym_name, new_strlen);
 113         newstr[new_strlen] = 0;
 114         return newstr;
 115 }