Home | History | Annotate | Line # | Download | only in libdwarf
      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