contrib/elftoolchain/libdwarf/libdwarf_str.c

   1 /*-
   2  * Copyright (c) 2009,2010 Kai Wang
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  *
  14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  24  * SUCH DAMAGE.
  25  */
  26
  27 #include "_libdwarf.h"
  28
  29 ELFTC_VCSID("$Id: libdwarf_str.c 2070 2011-10-27 03:05:32Z jkoshy $");
  30
  31 #define _INIT_DWARF_STRTAB_SIZE 1024
  32
  33 int
  34 _dwarf_strtab_add(Dwarf_Debug dbg, char *string, uint64_t *off,
  35     Dwarf_Error *error)
  36 {
  37         size_t len;
  38
  39         assert(dbg != NULL && string != NULL);
  40
  41         len = strlen(string) + 1;
  42         while (dbg->dbg_strtab_size + len > dbg->dbg_strtab_cap) {
  43                 dbg->dbg_strtab_cap *= 2;
  44                 dbg->dbg_strtab = realloc(dbg->dbg_strtab,
  45                     (size_t) dbg->dbg_strtab_cap);
  46                 if (dbg->dbg_strtab == NULL) {
  47                         DWARF_SET_ERROR(dbg, error, DW_DLE_MEMORY);
  48                         return (DW_DLE_MEMORY);
  49                 }
  50         }
  51
  52         if (off != NULL)
  53                 *off = dbg->dbg_strtab_size;
  54
  55         strncpy(&dbg->dbg_strtab[dbg->dbg_strtab_size], string, len - 1);
  56         dbg->dbg_strtab_size += len;
  57         dbg->dbg_strtab[dbg->dbg_strtab_size - 1] = '\0';
  58
  59         return (DW_DLE_NONE);
  60 }
  61
  62 char *
  63 _dwarf_strtab_get_table(Dwarf_Debug dbg)
  64 {
  65
  66         assert(dbg != NULL);
  67
  68         return (dbg->dbg_strtab);
  69 }
  70
  71 int
  72 _dwarf_strtab_init(Dwarf_Debug dbg, Dwarf_Error *error)
  73 {
  74         Dwarf_Section *ds;
  75
  76         assert(dbg != NULL);
  77
  78         if (dbg->dbg_mode == DW_DLC_READ || dbg->dbg_mode == DW_DLC_RDWR) {
  79                 ds = _dwarf_find_section(dbg, ".debug_str");
  80                 if (ds == NULL) {
  81                         dbg->dbg_strtab = NULL;
  82                         dbg->dbg_strtab_cap = dbg->dbg_strtab_size = 0;
  83                         return (DW_DLE_NONE);
  84                 }
  85
  86                 dbg->dbg_strtab_cap = dbg->dbg_strtab_size = ds->ds_size;
  87
  88                 if (dbg->dbg_mode == DW_DLC_RDWR) {
  89                         if ((dbg->dbg_strtab = malloc((size_t) ds->ds_size)) ==
  90                             NULL) {
  91                                 DWARF_SET_ERROR(dbg, error, DW_DLE_MEMORY);
  92                                 return (DW_DLE_MEMORY);
  93                         }
  94                         memcpy(dbg->dbg_strtab, ds->ds_data, ds->ds_size);
  95                 } else
  96                         dbg->dbg_strtab = (char *) ds->ds_data;
  97         } else {
  98                 /* DW_DLC_WRITE */
  99
 100                 dbg->dbg_strtab_cap = _INIT_DWARF_STRTAB_SIZE;
 101                 dbg->dbg_strtab_size = 0;
 102
 103                 if ((dbg->dbg_strtab = malloc((size_t) dbg->dbg_strtab_cap)) ==
 104                     NULL) {
 105                         DWARF_SET_ERROR(dbg, error, DW_DLE_MEMORY);
 106                         return (DW_DLE_MEMORY);
 107                 }
 108
 109                 dbg->dbg_strtab[0] = '\0';
 110         }
 111
 112         return (DW_DLE_NONE);
 113 }
 114
 115 void
 116 _dwarf_strtab_cleanup(Dwarf_Debug dbg)
 117 {
 118
 119         assert(dbg != NULL);
 120
 121         if (dbg->dbg_mode == DW_DLC_RDWR || dbg->dbg_mode == DW_DLC_WRITE)
 122                 free(dbg->dbg_strtab);
 123 }
 124
 125 int
 126 _dwarf_strtab_gen(Dwarf_P_Debug dbg, Dwarf_Error *error)
 127 {
 128         Dwarf_P_Section ds;
 129         int ret;
 130
 131         assert(dbg != NULL);
 132
 133         if ((ret = _dwarf_section_init(dbg, &ds, ".debug_str", 0, error)) !=
 134             DW_DLE_NONE)
 135                 return (ret);
 136
 137         if (dbg->dbg_strtab_size > ds->ds_cap) {
 138                 ds->ds_data = realloc(ds->ds_data,
 139                     (size_t) dbg->dbg_strtab_size);
 140                 if (ds->ds_data == NULL) {
 141                         _dwarf_section_free(dbg, &ds);
 142                         DWARF_SET_ERROR(dbg, error, DW_DLE_MEMORY);
 143                         return (DW_DLE_MEMORY);
 144                 }
 145                 ds->ds_cap = dbg->dbg_strtab_size;
 146         }
 147
 148         memcpy(ds->ds_data, dbg->dbg_strtab, dbg->dbg_strtab_size);
 149         ds->ds_size = dbg->dbg_strtab_size;
 150
 151         /*
 152          * Inform application the creation of .debug_str ELF section.
 153          * Note that .debug_str use a different format than usual ELF
 154          * string table, so it should not have SHT_STRTAB as its type.
 155          */
 156         ret = _dwarf_section_callback(dbg, ds, SHT_PROGBITS, 0, 0, 0, error);
 157
 158         return (ret);
 159 }