libltrace.c revision 602330f458e8049ec07685c4e02f0368821a2d2c
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 51f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedessignal_exit(int sig) { 52f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes exiting = 1; 53f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes debug(1, "Received interrupt signal; exiting..."); 54f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes signal(SIGINT, SIG_IGN); 55f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes signal(SIGTERM, SIG_IGN); 56f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes signal(SIGALRM, signal_alarm); 57602330f458e8049ec07685c4e02f0368821a2d2cPetr Machata //alarm(1); 58f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes} 59f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 60f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedesstatic void 61f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedesnormal_exit(void) { 62f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes output_line(0, 0); 63f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes if (options.summary) { 64f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes show_summary(); 65f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 66f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes if (options.output) { 67f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes fclose(options.output); 68f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes options.output = NULL; 69f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 70f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes} 71f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 72f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedesvoid 73f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedesltrace_init(int argc, char **argv) { 74f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes struct opt_p_t *opt_p_tmp; 75f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 76f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes atexit(normal_exit); 77f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes signal(SIGINT, signal_exit); /* Detach processes when interrupted */ 78f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes signal(SIGTERM, signal_exit); /* ... or killed */ 79f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 80f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes argv = process_options(argc, argv); 81f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes while (opt_F) { 82f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes /* If filename begins with ~, expand it to the user's home */ 83f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes /* directory. This does not correctly handle ~yoda, but that */ 84f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes /* isn't as bad as it seems because the shell will normally */ 85f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes /* be doing the expansion for us; only the hardcoded */ 86f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes /* ~/.ltrace.conf should ever use this code. */ 87f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes if (opt_F->filename[0] == '~') { 88f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes char path[PATH_MAX]; 89f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes char *home_dir = getenv("HOME"); 90f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes if (home_dir) { 91f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes strncpy(path, home_dir, PATH_MAX - 1); 92f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes path[PATH_MAX - 1] = '\0'; 93f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes strncat(path, opt_F->filename + 1, 94f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes PATH_MAX - strlen(path) - 1); 95f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes read_config_file(path); 96f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 97f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } else { 98f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes read_config_file(opt_F->filename); 99f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 100f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes opt_F = opt_F->next; 101f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 102f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes if (opt_e) { 103f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes struct opt_e_t *tmp = opt_e; 104f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes while (tmp) { 105f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes debug(1, "Option -e: %s\n", tmp->name); 106f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes tmp = tmp->next; 107f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 108f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 109f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes if (command) { 110c7585b60235268e1f62bd91c5f040ef6df6e0ef3Petr Machata open_program(command, execute_program(command, argv), 0); 111f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 112f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes opt_p_tmp = opt_p; 113f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes while (opt_p_tmp) { 1148d1b92ba755f6d6229f5e230fc43d958b13836f8Juan Cespedes open_pid(opt_p_tmp->pid); 115f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes opt_p_tmp = opt_p_tmp->next; 116f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 117f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes} 118f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 11961da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedesstatic int num_ltrace_callbacks[EVENT_MAX]; 12061da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedesstatic callback_func * ltrace_callbacks[EVENT_MAX]; 12140dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedes 12240dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedesvoid 12361da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedesltrace_add_callback(callback_func func, Event_type type) { 12461da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes ltrace_callbacks[type] = realloc(ltrace_callbacks[type], (num_ltrace_callbacks[type]+1)*sizeof(callback_func)); 12561da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes ltrace_callbacks[type][num_ltrace_callbacks[type]++] = func; 12661da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes} 12740dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedes 12861da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedesstatic void 12961da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedesdispatch_callbacks(Event * ev) { 13061da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes int i; 13161da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes /* Ignoring case 1: signal into a dying tracer */ 13261da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes if (ev->type==EVENT_SIGNAL && 13361da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes exiting && ev->e_un.signum == SIGSTOP) { 13461da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes return; 13561da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes } 13661da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes /* Ignoring case 2: process being born before a clone event */ 13761da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes if (ev->proc && ev->proc->state == STATE_IGNORED) { 13861da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes return; 13961da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes } 14061da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes for (i=0; i<num_ltrace_callbacks[ev->type]; i++) { 14161da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes ltrace_callbacks[ev->type][i](ev); 14240dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedes } 14340dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedes} 14440dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedes 145f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedesvoid 146f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedesltrace_main(void) { 14740dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedes Event * ev; 148f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes while (1) { 14940dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedes ev = next_event(); 15061da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes dispatch_callbacks(ev); 15103192f890af95bf9abf5ea1c3dc0998f6b12550dJuan Cespedes handle_event(ev); 152f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 153f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes} 154