/* test_plugin.c -- simple linker plugin test Copyright (C) 2008-2015 Free Software Foundation, Inc. Written by Cary Coutant . This file is part of gold. This program is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; either version 3 of the License, or (at your option) any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with this program; if not, write to the Free Software Foundation, Inc., 51 Franklin Street - Fifth Floor, Boston, MA 02110-1301, USA. */ #ifdef HAVE_CONFIG_H #include "config.h" #endif #include #include #include #include "plugin-api.h" struct claimed_file { const char* name; void* handle; int nsyms; struct ld_plugin_symbol* syms; struct claimed_file* next; }; struct sym_info { int size; char* type; char* bind; char* vis; char* sect; char* name; }; static struct claimed_file* first_claimed_file = NULL; static struct claimed_file* last_claimed_file = NULL; static ld_plugin_register_claim_file register_claim_file_hook = NULL; static ld_plugin_register_all_symbols_read register_all_symbols_read_hook = NULL; static ld_plugin_register_cleanup register_cleanup_hook = NULL; static ld_plugin_add_symbols add_symbols = NULL; static ld_plugin_get_symbols get_symbols = NULL; static ld_plugin_get_symbols get_symbols_v2 = NULL; static ld_plugin_add_input_file add_input_file = NULL; static ld_plugin_message message = NULL; static ld_plugin_get_input_file get_input_file = NULL; static ld_plugin_release_input_file release_input_file = NULL; static ld_plugin_get_input_section_count get_input_section_count = NULL; static ld_plugin_get_input_section_type get_input_section_type = NULL; static ld_plugin_get_input_section_name get_input_section_name = NULL; static ld_plugin_get_input_section_contents get_input_section_contents = NULL; static ld_plugin_update_section_order update_section_order = NULL; static ld_plugin_allow_section_ordering allow_section_ordering = NULL; #define MAXOPTS 10 static const char *opts[MAXOPTS]; static int nopts = 0; enum ld_plugin_status onload(struct ld_plugin_tv *tv); enum ld_plugin_status claim_file_hook(const struct ld_plugin_input_file *file, int *claimed); enum ld_plugin_status all_symbols_read_hook(void); enum ld_plugin_status cleanup_hook(void); static void parse_readelf_line(char*, struct sym_info*); enum ld_plugin_status onload(struct ld_plugin_tv *tv) { struct ld_plugin_tv *entry; int api_version = 0; int gold_version = 0; int i; for (entry = tv; entry->tv_tag != LDPT_NULL; ++entry) { switch (entry->tv_tag) { case LDPT_API_VERSION: api_version = entry->tv_u.tv_val; break; case LDPT_GOLD_VERSION: gold_version = entry->tv_u.tv_val; break; case LDPT_LINKER_OUTPUT: break; case LDPT_OPTION: if (nopts < MAXOPTS) opts[nopts++] = entry->tv_u.tv_string; break; case LDPT_REGISTER_CLAIM_FILE_HOOK: register_claim_file_hook = entry->tv_u.tv_register_claim_file; break; case LDPT_REGISTER_ALL_SYMBOLS_READ_HOOK: register_all_symbols_read_hook = entry->tv_u.tv_register_all_symbols_read; break; case LDPT_REGISTER_CLEANUP_HOOK: register_cleanup_hook = entry->tv_u.tv_register_cleanup; break; case LDPT_ADD_SYMBOLS: add_symbols = entry->tv_u.tv_add_symbols; break; case LDPT_GET_SYMBOLS: get_symbols = entry->tv_u.tv_get_symbols; break; case LDPT_GET_SYMBOLS_V2: get_symbols_v2 = entry->tv_u.tv_get_symbols; break; case LDPT_ADD_INPUT_FILE: add_input_file = entry->tv_u.tv_add_input_file; break; case LDPT_MESSAGE: message = entry->tv_u.tv_message; break; case LDPT_GET_INPUT_FILE: get_input_file = entry->tv_u.tv_get_input_file; break; case LDPT_RELEASE_INPUT_FILE: release_input_file = entry->tv_u.tv_release_input_file; break; case LDPT_GET_INPUT_SECTION_COUNT: get_input_section_count = *entry->tv_u.tv_get_input_section_count; break; case LDPT_GET_INPUT_SECTION_TYPE: get_input_section_type = *entry->tv_u.tv_get_input_section_type; break; case LDPT_GET_INPUT_SECTION_NAME: get_input_section_name = *entry->tv_u.tv_get_input_section_name; break; case LDPT_GET_INPUT_SECTION_CONTENTS: get_input_section_contents = *entry->tv_u.tv_get_input_section_contents; break; case LDPT_UPDATE_SECTION_ORDER: update_section_order = *entry->tv_u.tv_update_section_order; break; case LDPT_ALLOW_SECTION_ORDERING: allow_section_ordering = *entry->tv_u.tv_allow_section_ordering; break; default: break; } } if (message == NULL) { fprintf(stderr, "tv_message interface missing\n"); return LDPS_ERR; } if (register_claim_file_hook == NULL) { fprintf(stderr, "tv_register_claim_file_hook interface missing\n"); return LDPS_ERR; } if (register_all_symbols_read_hook == NULL) { fprintf(stderr, "tv_register_all_symbols_read_hook interface missing\n"); return LDPS_ERR; } if (register_cleanup_hook == NULL) { fprintf(stderr, "tv_register_cleanup_hook interface missing\n"); return LDPS_ERR; } (*message)(LDPL_INFO, "API version: %d", api_version); (*message)(LDPL_INFO, "gold version: %d", gold_version); for (i = 0; i < nopts; ++i) (*message)(LDPL_INFO, "option: %s", opts[i]); if ((*register_claim_file_hook)(claim_file_hook) != LDPS_OK) { (*message)(LDPL_ERROR, "error registering claim file hook"); return LDPS_ERR; } if ((*register_all_symbols_read_hook)(all_symbols_read_hook) != LDPS_OK) { (*message)(LDPL_ERROR, "error registering all symbols read hook"); return LDPS_ERR; } if ((*register_cleanup_hook)(cleanup_hook) != LDPS_OK) { (*message)(LDPL_ERROR, "error registering cleanup hook"); return LDPS_ERR; } if (get_input_section_count == NULL) { fprintf(stderr, "tv_get_input_section_count interface missing\n"); return LDPS_ERR; } if (get_input_section_type == NULL) { fprintf(stderr, "tv_get_input_section_type interface missing\n"); return LDPS_ERR; } if (get_input_section_name == NULL) { fprintf(stderr, "tv_get_input_section_name interface missing\n"); return LDPS_ERR; } if (get_input_section_contents == NULL) { fprintf(stderr, "tv_get_input_section_contents interface missing\n"); return LDPS_ERR; } if (update_section_order == NULL) { fprintf(stderr, "tv_update_section_order interface missing\n"); return LDPS_ERR; } if (allow_section_ordering == NULL) { fprintf(stderr, "tv_allow_section_ordering interface missing\n"); return LDPS_ERR; } return LDPS_OK; } enum ld_plugin_status claim_file_hook (const struct ld_plugin_input_file* file, int* claimed) { int len; off_t end_offset; char buf[160]; struct claimed_file* claimed_file; struct ld_plugin_symbol* syms; int nsyms = 0; int maxsyms = 0; FILE* irfile; struct sym_info info; int weak; int def; int vis; int is_comdat; int i; (*message)(LDPL_INFO, "%s: claim file hook called (offset = %ld, size = %ld)", file->name, (long)file->offset, (long)file->filesize); /* Look for the beginning of output from readelf -s. */ irfile = fdopen(file->fd, "r"); (void)fseek(irfile, file->offset, SEEK_SET); end_offset = file->offset + file->filesize; len = fread(buf, 1, 13, irfile); if (len < 13 || strncmp(buf, "\nSymbol table", 13) != 0) return LDPS_OK; /* Skip the two header lines. */ (void) fgets(buf, sizeof(buf), irfile); (void) fgets(buf, sizeof(buf), irfile); if (add_symbols == NULL) { fprintf(stderr, "tv_add_symbols interface missing\n"); return LDPS_ERR; } /* Parse the output from readelf. The columns are: Index Value Size Type Binding Visibility Section Name. */ syms = (struct ld_plugin_symbol*)malloc(sizeof(struct ld_plugin_symbol) * 8); if (syms == NULL) return LDPS_ERR; maxsyms = 8; while (ftell(irfile) < end_offset && fgets(buf, sizeof(buf), irfile) != NULL) { parse_readelf_line(buf, &info); /* Ignore local symbols. */ if (strncmp(info.bind, "LOCAL", 5) == 0) continue; weak = strncmp(info.bind, "WEAK", 4) == 0; if (strncmp(info.sect, "UND", 3) == 0) def = weak ? LDPK_WEAKUNDEF : LDPK_UNDEF; else if (strncmp(info.sect, "COM", 3) == 0) def = LDPK_COMMON; else def = weak ? LDPK_WEAKDEF : LDPK_DEF; if (strncmp(info.vis, "INTERNAL", 8) == 0) vis = LDPV_INTERNAL; else if (strncmp(info.vis, "HIDDEN", 6) == 0) vis = LDPV_HIDDEN; else if (strncmp(info.vis, "PROTECTED", 9) == 0) vis = LDPV_PROTECTED; else vis = LDPV_DEFAULT; /* If the symbol is listed in the options list, special-case it as a comdat symbol. */ is_comdat = 0; for (i = 0; i < nopts; ++i) { if (info.name != NULL && strcmp(info.name, opts[i]) == 0) { is_comdat = 1; break; } } if (nsyms >= maxsyms) { syms = (struct ld_plugin_symbol*) realloc(syms, sizeof(struct ld_plugin_symbol) * maxsyms * 2); if (syms == NULL) return LDPS_ERR; maxsyms *= 2; } if (info.name == NULL) syms[nsyms].name = NULL; else { len = strlen(info.name); syms[nsyms].name = malloc(len + 1); strncpy(syms[nsyms].name, info.name, len + 1); } syms[nsyms].version = NULL; syms[nsyms].def = def; syms[nsyms].visibility = vis; syms[nsyms].size = info.size; syms[nsyms].comdat_key = is_comdat ? syms[nsyms].name : NULL; syms[nsyms].resolution = LDPR_UNKNOWN; ++nsyms; } claimed_file = (struct claimed_file*) malloc(sizeof(struct claimed_file)); if (claimed_file == NULL) return LDPS_ERR; claimed_file->name = file->name; claimed_file->handle = file->handle; claimed_file->nsyms = nsyms; claimed_file->syms = syms; claimed_file->next = NULL; if (last_claimed_file == NULL) first_claimed_file = claimed_file; else last_claimed_file->next = claimed_file; last_claimed_file = claimed_file; (*message)(LDPL_INFO, "%s: claiming file, adding %d symbols", file->name, nsyms); if (nsyms > 0) (*add_symbols)(file->handle, nsyms, syms); *claimed = 1; return LDPS_OK; } enum ld_plugin_status all_symbols_read_hook(void) { int i; const char* res; struct claimed_file* claimed_file; struct ld_plugin_input_file file; FILE* irfile; off_t end_offset; struct sym_info info; int len; char buf[160]; char* p; const char* filename; (*message)(LDPL_INFO, "all symbols read hook called"); if (get_symbols_v2 == NULL) { fprintf(stderr, "tv_get_symbols (v2) interface missing\n"); return LDPS_ERR; } for (claimed_file = first_claimed_file; claimed_file != NULL; claimed_file = claimed_file->next) { (*get_symbols_v2)(claimed_file->handle, claimed_file->nsyms, claimed_file->syms); for (i = 0; i < claimed_file->nsyms; ++i) { switch (claimed_file->syms[i].resolution) { case LDPR_UNKNOWN: res = "UNKNOWN"; break; case LDPR_UNDEF: res = "UNDEF"; break; case LDPR_PREVAILING_DEF: res = "PREVAILING_DEF_REG"; break; case LDPR_PREVAILING_DEF_IRONLY: res = "PREVAILING_DEF_IRONLY"; break; case LDPR_PREVAILING_DEF_IRONLY_EXP: res = "PREVAILING_DEF_IRONLY_EXP"; break; case LDPR_PREEMPTED_REG: res = "PREEMPTED_REG"; break; case LDPR_PREEMPTED_IR: res = "PREEMPTED_IR"; break; case LDPR_RESOLVED_IR: res = "RESOLVED_IR"; break; case LDPR_RESOLVED_EXEC: res = "RESOLVED_EXEC"; break; case LDPR_RESOLVED_DYN: res = "RESOLVED_DYN"; break; default: res = "?"; break; } (*message)(LDPL_INFO, "%s: %s: %s", claimed_file->name, claimed_file->syms[i].name, res); } } if (add_input_file == NULL) { fprintf(stderr, "tv_add_input_file interface missing\n"); return LDPS_ERR; } if (get_input_file == NULL) { fprintf(stderr, "tv_get_input_file interface missing\n"); return LDPS_ERR; } if (release_input_file == NULL) { fprintf(stderr, "tv_release_input_file interface missing\n"); return LDPS_ERR; } for (claimed_file = first_claimed_file; claimed_file != NULL; claimed_file = claimed_file->next) { (*get_input_file) (claimed_file->handle, &file); /* Look for the beginning of output from readelf -s. */ irfile = fdopen(file.fd, "r"); (void)fseek(irfile, file.offset, SEEK_SET); end_offset = file.offset + file.filesize; len = fread(buf, 1, 13, irfile); if (len < 13 || strncmp(buf, "\nSymbol table", 13) != 0) { fprintf(stderr, "%s: can't re-read original input file\n", claimed_file->name); return LDPS_ERR; } /* Skip the two header lines. */ (void) fgets(buf, sizeof(buf), irfile); (void) fgets(buf, sizeof(buf), irfile); filename = NULL; while (ftell(irfile) < end_offset && fgets(buf, sizeof(buf), irfile) != NULL) { parse_readelf_line(buf, &info); /* Look for file name. */ if (strncmp(info.type, "FILE", 4) == 0) { len = strlen(info.name); p = malloc(len + 1); strncpy(p, info.name, len + 1); filename = p; break; } } (*release_input_file) (claimed_file->handle); if (filename == NULL) filename = claimed_file->name; if (claimed_file->nsyms == 0) continue; if (strlen(filename) >= sizeof(buf)) { (*message)(LDPL_FATAL, "%s: filename too long", filename); return LDPS_ERR; } strcpy(buf, filename); p = strrchr(buf, '.'); if (p == NULL || (strcmp(p, ".syms") != 0 && strcmp(p, ".c") != 0 && strcmp(p, ".cc") != 0)) { (*message)(LDPL_FATAL, "%s: filename has unknown suffix", filename); return LDPS_ERR; } p[1] = 'o'; p[2] = '\0'; (*message)(LDPL_INFO, "%s: adding new input file", buf); (*add_input_file)(buf); } return LDPS_OK; } enum ld_plugin_status cleanup_hook(void) { (*message)(LDPL_INFO, "cleanup hook called"); return LDPS_OK; } static void parse_readelf_line(char* p, struct sym_info* info) { int len; p += strspn(p, " "); /* Index field. */ p += strcspn(p, " "); p += strspn(p, " "); /* Value field. */ p += strcspn(p, " "); p += strspn(p, " "); /* Size field. */ info->size = atoi(p); p += strcspn(p, " "); p += strspn(p, " "); /* Type field. */ info->type = p; p += strcspn(p, " "); p += strspn(p, " "); /* Binding field. */ info->bind = p; p += strcspn(p, " "); p += strspn(p, " "); /* Visibility field. */ info->vis = p; p += strcspn(p, " "); p += strspn(p, " "); if (*p == '[') { /* Skip st_other. */ p += strcspn(p, "]"); p += strspn(p, "] "); } /* Section field. */ info->sect = p; p += strcspn(p, " "); p += strspn(p, " "); /* Name field. */ /* FIXME: Look for version. */ len = strlen(p); if (len == 0) p = NULL; else if (p[len-1] == '\n') p[--len] = '\0'; info->name = p; }