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 } 160