3 * Copyright (c) 2009, Jouni Malinen <j@w1.fi>
5 * This software may be distributed under the terms of the BSD license.
6 * See README for more details.
16 static struct dl_list active_references =
17 { &active_references, &active_references };
22 #define DMGL_PARAMS (1 << 0)
23 #define DMGL_ANSI (1 << 1)
25 static char *prg_fname = NULL;
26 static bfd *cached_abfd = NULL;
27 static asymbol **syms = NULL;
29 static void get_prg_fname(void)
31 char exe[50], fname[512];
33 os_snprintf(exe, sizeof(exe) - 1, "/proc/%u/exe", getpid());
34 len = readlink(exe, fname, sizeof(fname) - 1);
35 if (len < 0 || len >= (int) sizeof(fname)) {
36 wpa_printf(MSG_ERROR, "readlink: %s", strerror(errno));
40 prg_fname = strdup(fname);
44 static bfd * open_bfd(const char *fname)
49 abfd = bfd_openr(prg_fname, NULL);
51 wpa_printf(MSG_INFO, "bfd_openr failed");
55 if (bfd_check_format(abfd, bfd_archive)) {
56 wpa_printf(MSG_INFO, "bfd_check_format failed");
61 if (!bfd_check_format_matches(abfd, bfd_object, &matching)) {
62 wpa_printf(MSG_INFO, "bfd_check_format_matches failed");
72 static void read_syms(bfd *abfd)
74 long storage, symcount;
75 bfd_boolean dynamic = FALSE;
80 if (!(bfd_get_file_flags(abfd) & HAS_SYMS)) {
81 wpa_printf(MSG_INFO, "No symbols");
85 storage = bfd_get_symtab_upper_bound(abfd);
87 storage = bfd_get_dynamic_symtab_upper_bound(abfd);
91 wpa_printf(MSG_INFO, "Unknown symtab upper bound");
95 syms = malloc(storage);
97 wpa_printf(MSG_INFO, "Failed to allocate memory for symtab "
98 "(%ld bytes)", storage);
102 symcount = bfd_canonicalize_dynamic_symtab(abfd, syms);
104 symcount = bfd_canonicalize_symtab(abfd, syms);
106 wpa_printf(MSG_INFO, "Failed to canonicalize %ssymtab",
107 dynamic ? "dynamic " : "");
118 const char *filename;
119 const char *function;
124 static void find_addr_sect(bfd *abfd, asection *section, void *obj)
126 struct bfd_data *data = obj;
133 if (!(bfd_get_section_vma(abfd, section)))
136 vma = bfd_get_section_vma(abfd, section);
140 size = bfd_get_section_size(section);
141 if (data->pc >= vma + size)
144 data->found = bfd_find_nearest_line(abfd, section, syms,
152 static void wpa_trace_bfd_addr(void *pc)
154 bfd *abfd = cached_abfd;
155 struct bfd_data data;
158 const char *filename;
163 data.pc = (bfd_hostptr_t) pc;
165 bfd_map_over_sections(abfd, find_addr_sect, &data);
172 aname = bfd_demangle(abfd, data.function,
173 DMGL_ANSI | DMGL_PARAMS);
174 name = aname ? aname : data.function;
175 filename = data.filename;
177 char *end = os_strrchr(filename, '/');
179 while (*filename && *filename == prg_fname[i] &&
185 wpa_printf(MSG_INFO, " %s() %s:%u",
186 name, filename, data.line);
190 data.found = bfd_find_inliner_info(abfd, &data.filename,
191 &data.function, &data.line);
192 } while (data.found);
196 static const char * wpa_trace_bfd_addr2func(void *pc)
198 bfd *abfd = cached_abfd;
199 struct bfd_data data;
204 data.pc = (bfd_hostptr_t) pc;
206 bfd_map_over_sections(abfd, find_addr_sect, &data);
211 return data.function;
215 static void wpa_trace_bfd_init(void)
224 cached_abfd = open_bfd(prg_fname);
226 wpa_printf(MSG_INFO, "Failed to open bfd");
231 read_syms(cached_abfd);
233 wpa_printf(MSG_INFO, "Failed to read symbols");
239 void wpa_trace_dump_funcname(const char *title, void *pc)
241 wpa_printf(MSG_INFO, "WPA_TRACE: %s: %p", title, pc);
242 wpa_trace_bfd_init();
243 wpa_trace_bfd_addr(pc);
247 size_t wpa_trace_calling_func(const char *buf[], size_t len)
250 void *btrace_res[WPA_TRACE_LEN];
256 if (len > WPA_TRACE_LEN)
259 wpa_trace_bfd_init();
264 btrace_num = backtrace(btrace_res, len);
268 for (i = 0; i < btrace_num; i++) {
269 struct bfd_data data;
271 data.pc = (bfd_hostptr_t) btrace_res[i];
273 bfd_map_over_sections(abfd, find_addr_sect, &data);
278 os_strcmp(data.function, __func__) != 0)) {
279 buf[pos++] = data.function;
284 data.found = bfd_find_inliner_info(abfd, &data.filename,
293 #else /* WPA_TRACE_BFD */
295 #define wpa_trace_bfd_init() do { } while (0)
296 #define wpa_trace_bfd_addr(pc) do { } while (0)
297 #define wpa_trace_bfd_addr2func(pc) NULL
299 #endif /* WPA_TRACE_BFD */
301 void wpa_trace_dump_func(const char *title, void **btrace, int btrace_num)
305 enum { TRACE_HEAD, TRACE_RELEVANT, TRACE_TAIL } state;
307 wpa_trace_bfd_init();
308 wpa_printf(MSG_INFO, "WPA_TRACE: %s - START", title);
309 sym = backtrace_symbols(btrace, btrace_num);
311 for (i = 0; i < btrace_num; i++) {
312 const char *func = wpa_trace_bfd_addr2func(btrace[i]);
313 if (state == TRACE_HEAD && func &&
314 (os_strcmp(func, "wpa_trace_add_ref_func") == 0 ||
315 os_strcmp(func, "wpa_trace_check_ref") == 0 ||
316 os_strcmp(func, "wpa_trace_show") == 0))
318 if (state == TRACE_TAIL && sym && sym[i] &&
319 os_strstr(sym[i], "__libc_start_main"))
321 if (state == TRACE_HEAD)
322 state = TRACE_RELEVANT;
324 wpa_printf(MSG_INFO, "[%d]: %s", i, sym[i]);
326 wpa_printf(MSG_INFO, "[%d]: ?? [%p]", i, btrace[i]);
327 wpa_trace_bfd_addr(btrace[i]);
328 if (state == TRACE_RELEVANT && func &&
329 os_strcmp(func, "main") == 0)
333 wpa_printf(MSG_INFO, "WPA_TRACE: %s - END", title);
337 void wpa_trace_show(const char *title)
342 wpa_trace_record(&info);
343 wpa_trace_dump(title, &info);
347 void wpa_trace_add_ref_func(struct wpa_trace_ref *ref, const void *addr)
352 wpa_trace_record(ref);
353 dl_list_add(&active_references, &ref->list);
357 void wpa_trace_check_ref(const void *addr)
359 struct wpa_trace_ref *ref;
360 dl_list_for_each(ref, &active_references, struct wpa_trace_ref, list) {
361 if (addr != ref->addr)
363 wpa_trace_show("Freeing referenced memory");
364 wpa_trace_dump("Reference registration", ref);
369 #endif /* WPA_TRACE */