libltrace.c revision ffe4cd25089680daf1bd1ec0114d177ec3e0cf95
1f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes#include "config.h" 2f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 3f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes#include <stdio.h> 4f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes#include <stdlib.h> 5f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes#include <unistd.h> 6f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes#include <string.h> 7f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes#include <errno.h> 8f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes#include <sys/param.h> 9f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes#include <signal.h> 10f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes#include <sys/wait.h> 11f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 12f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes#include "common.h" 13f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 14f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedeschar *command = NULL; 15f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 16f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedesint exiting = 0; /* =1 if a SIGINT or SIGTERM has been received */ 17f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 18cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machatastatic enum pcb_status 19cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machatastop_non_p_processes (Process * proc, void * data) 20cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata{ 21cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata int stop = 1; 22f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 23cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata struct opt_p_t *it; 24cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata for (it = opt_p; it != NULL; it = it->next) { 25cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata Process * p_proc = pid2proc(it->pid); 26cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata if (p_proc == NULL) { 27cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata printf("stop_non_p_processes: %d terminated?\n", it->pid); 28cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata continue; 29f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 309a5420c82a2fd81681572a2e3859ea1671c3bdedPetr Machata if (p_proc == proc || p_proc->leader == proc->leader) { 31cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata stop = 0; 32cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata break; 33cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata } 34cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata } 35cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata 36cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata if (stop) { 37cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata debug(2, "Sending SIGSTOP to process %u", proc->pid); 38cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata kill(proc->pid, SIGSTOP); 39f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 40cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata 41cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata return pcb_cont; 42cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata} 43cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata 44cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machatastatic void 45cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machatasignal_alarm(int sig) { 46cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata signal(SIGALRM, SIG_DFL); 47cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata each_process(NULL, &stop_non_p_processes, NULL); 48f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes} 49f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 50f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedesstatic void 51ffe4cd25089680daf1bd1ec0114d177ec3e0cf95Petr Machatasignal_exit(int sig) 52ffe4cd25089680daf1bd1ec0114d177ec3e0cf95Petr Machata{ 53f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes debug(1, "Received interrupt signal; exiting..."); 54ffe4cd25089680daf1bd1ec0114d177ec3e0cf95Petr Machata if (exiting != 0) 55ffe4cd25089680daf1bd1ec0114d177ec3e0cf95Petr Machata return; 56ffe4cd25089680daf1bd1ec0114d177ec3e0cf95Petr Machata 57ffe4cd25089680daf1bd1ec0114d177ec3e0cf95Petr Machata exiting = 1 + !!os_ltrace_exiting_sighandler(); 58ffe4cd25089680daf1bd1ec0114d177ec3e0cf95Petr Machata 59f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes signal(SIGINT, SIG_IGN); 60f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes signal(SIGTERM, SIG_IGN); 61f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes signal(SIGALRM, signal_alarm); 62602330f458e8049ec07685c4e02f0368821a2d2cPetr Machata //alarm(1); 63f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes} 64f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 65f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedesstatic void 66f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedesnormal_exit(void) { 67f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes output_line(0, 0); 68f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes if (options.summary) { 69f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes show_summary(); 70f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 71f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes if (options.output) { 72f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes fclose(options.output); 73f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes options.output = NULL; 74f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 75f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes} 76f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 77f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedesvoid 78f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedesltrace_init(int argc, char **argv) { 79f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes struct opt_p_t *opt_p_tmp; 80f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 81f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes atexit(normal_exit); 82f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes signal(SIGINT, signal_exit); /* Detach processes when interrupted */ 83f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes signal(SIGTERM, signal_exit); /* ... or killed */ 84f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 85f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes argv = process_options(argc, argv); 86f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes while (opt_F) { 87f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes /* If filename begins with ~, expand it to the user's home */ 88f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes /* directory. This does not correctly handle ~yoda, but that */ 89f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes /* isn't as bad as it seems because the shell will normally */ 90f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes /* be doing the expansion for us; only the hardcoded */ 91f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes /* ~/.ltrace.conf should ever use this code. */ 92f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes if (opt_F->filename[0] == '~') { 93f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes char path[PATH_MAX]; 94f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes char *home_dir = getenv("HOME"); 95f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes if (home_dir) { 96f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes strncpy(path, home_dir, PATH_MAX - 1); 97f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes path[PATH_MAX - 1] = '\0'; 98f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes strncat(path, opt_F->filename + 1, 99f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes PATH_MAX - strlen(path) - 1); 100f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes read_config_file(path); 101f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 102f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } else { 103f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes read_config_file(opt_F->filename); 104f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 105f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes opt_F = opt_F->next; 106f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 107f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes if (opt_e) { 108f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes struct opt_e_t *tmp = opt_e; 109f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes while (tmp) { 110f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes debug(1, "Option -e: %s\n", tmp->name); 111f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes tmp = tmp->next; 112f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 113f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 114f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes if (command) { 11502bd9eca0b439b6950ab72ba64b67181e5bf0fbdPetr Machata /* Check that the binary ABI is supported before 11602bd9eca0b439b6950ab72ba64b67181e5bf0fbdPetr Machata * calling execute_program. */ 11702bd9eca0b439b6950ab72ba64b67181e5bf0fbdPetr Machata struct ltelf lte = {}; 11802bd9eca0b439b6950ab72ba64b67181e5bf0fbdPetr Machata open_elf(<e, command); 11902bd9eca0b439b6950ab72ba64b67181e5bf0fbdPetr Machata 120c7585b60235268e1f62bd91c5f040ef6df6e0ef3Petr Machata open_program(command, execute_program(command, argv), 0); 121f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 122f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes opt_p_tmp = opt_p; 123f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes while (opt_p_tmp) { 1248d1b92ba755f6d6229f5e230fc43d958b13836f8Juan Cespedes open_pid(opt_p_tmp->pid); 125f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes opt_p_tmp = opt_p_tmp->next; 126f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 127f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes} 128f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 12961da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedesstatic int num_ltrace_callbacks[EVENT_MAX]; 13061da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedesstatic callback_func * ltrace_callbacks[EVENT_MAX]; 13140dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedes 13240dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedesvoid 13361da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedesltrace_add_callback(callback_func func, Event_type type) { 13461da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes ltrace_callbacks[type] = realloc(ltrace_callbacks[type], (num_ltrace_callbacks[type]+1)*sizeof(callback_func)); 13561da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes ltrace_callbacks[type][num_ltrace_callbacks[type]++] = func; 13661da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes} 13740dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedes 13861da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedesstatic void 13961da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedesdispatch_callbacks(Event * ev) { 14061da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes int i; 14161da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes /* Ignoring case 1: signal into a dying tracer */ 14261da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes if (ev->type==EVENT_SIGNAL && 14361da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes exiting && ev->e_un.signum == SIGSTOP) { 14461da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes return; 14561da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes } 14661da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes /* Ignoring case 2: process being born before a clone event */ 14761da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes if (ev->proc && ev->proc->state == STATE_IGNORED) { 14861da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes return; 14961da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes } 15061da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes for (i=0; i<num_ltrace_callbacks[ev->type]; i++) { 15161da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes ltrace_callbacks[ev->type][i](ev); 15240dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedes } 15340dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedes} 15440dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedes 155f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedesvoid 156f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedesltrace_main(void) { 15740dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedes Event * ev; 158f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes while (1) { 15940dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedes ev = next_event(); 16061da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes dispatch_callbacks(ev); 16103192f890af95bf9abf5ea1c3dc0998f6b12550dJuan Cespedes handle_event(ev); 162f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 163f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes} 164