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