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         if (sect->userdata != NULL)
  47                 return sect->userdata;
  48
  49         struct supersect *new = malloc(sizeof(*new));
  50         sect->userdata = new;
  51         new->parent = abfd;
  52         new->name = malloc(strlen(sect->name) + 1);
  53         strcpy(new->name, sect->name);
  54         new->flags = bfd_get_section_flags(abfd, sect);
  55
  56         vec_init(&new->contents);
  57         vec_resize(&new->contents, bfd_get_section_size(sect));
  58         assert(bfd_get_section_contents
  59                (abfd, sect, new->contents.data, 0, new->contents.size));
  60         new->alignment = bfd_get_section_alignment(abfd, sect);
  61
  62         vec_init(&new->relocs);
  63         vec_reserve(&new->relocs, bfd_get_reloc_upper_bound(abfd, sect));
  64         vec_resize(&new->relocs,
  65                    bfd_canonicalize_reloc(abfd, sect, new->relocs.data,
  66                                           syms->data));
  67         assert(new->relocs.size >= 0);
  68         vec_init(&new->new_relocs);
  69
  70         return new;
  71 }
  72
  73 struct supersect *new_supersects = NULL;
  74
  75 struct supersect *new_supersect(char *name)
  76 {
  77         struct supersect *ss;
  78         for (ss = new_supersects; ss != NULL; ss = ss->next) {
  79                 if (strcmp(name, ss->name) == 0)
  80                         return ss;
  81         }
  82
  83         struct supersect *new = malloc(sizeof(*new));
  84         new->parent = NULL;
  85         new->name = name;
  86         new->next = new_supersects;
  87         new_supersects = new;
  88         new->flags = SEC_ALLOC | SEC_HAS_CONTENTS | SEC_RELOC;
  89
  90         vec_init(&new->contents);
  91         new->alignment = 0;
  92         vec_init(&new->relocs);
  93         vec_init(&new->new_relocs);
  94
  95         return new;
  96 }
  97
  98 void *sect_do_grow(struct supersect *ss, size_t n, size_t size, int alignment)
  99 {
 100         if (ss->alignment < ffs(alignment) - 1)
 101                 ss->alignment = ffs(alignment) - 1;
 102         int pad = ss->contents.size - align(ss->contents.size, alignment);
 103         memset(vec_grow(&ss->contents, pad), 0, pad);
 104         return vec_grow(&ss->contents, n * size);
 105 }
 106
 107 int label_offset(const char *sym_name)
 108 {
 109         int i;
 110         for (i = 0;
 111              sym_name[i] != 0 && sym_name[i + 1] != 0 && sym_name[i + 2] != 0
 112              && sym_name[i + 3] != 0; i++) {
 113                 if (sym_name[i] == '_' && sym_name[i + 1] == '_'
 114                     && sym_name[i + 2] == '_' && sym_name[i + 3] == '_')
 115                         return i + 4;
 116         }
 117         return -1;
 118 }
 119
 120 const char *only_label(const char *sym_name)
 121 {
 122         int offset = label_offset(sym_name);
 123         if (offset == -1)
 124                 return NULL;
 125         return &sym_name[offset];
 126 }
 127
 128 const char *dup_wolabel(const char *sym_name)
 129 {
 130         int offset, entire_strlen, label_strlen, new_strlen;
 131         char *newstr;
 132
 133         offset = label_offset(sym_name);
 134         if (offset == -1)
 135                 label_strlen = 0;
 136         else
 137                 label_strlen = strlen(&sym_name[offset]) + strlen("____");
 138
 139         entire_strlen = strlen(sym_name);
 140         new_strlen = entire_strlen - label_strlen;
 141         newstr = malloc(new_strlen + 1);
 142         memcpy(newstr, sym_name, new_strlen);
 143         newstr[new_strlen] = 0;
 144         return newstr;
 145 }