libltrace.c revision 94078ecce3a103c28457e6f90f1e5b0dacc61146
194078ecce3a103c28457e6f90f1e5b0dacc61146Petr Machata/* 294078ecce3a103c28457e6f90f1e5b0dacc61146Petr Machata * This file is part of ltrace. 394078ecce3a103c28457e6f90f1e5b0dacc61146Petr Machata * Copyright (C) 2011 Petr Machata, Red Hat Inc. 494078ecce3a103c28457e6f90f1e5b0dacc61146Petr Machata * Copyright (C) 2009 Juan Cespedes 594078ecce3a103c28457e6f90f1e5b0dacc61146Petr Machata * 694078ecce3a103c28457e6f90f1e5b0dacc61146Petr Machata * This program is free software; you can redistribute it and/or 794078ecce3a103c28457e6f90f1e5b0dacc61146Petr Machata * modify it under the terms of the GNU General Public License as 894078ecce3a103c28457e6f90f1e5b0dacc61146Petr Machata * published by the Free Software Foundation; either version 2 of the 994078ecce3a103c28457e6f90f1e5b0dacc61146Petr Machata * License, or (at your option) any later version. 1094078ecce3a103c28457e6f90f1e5b0dacc61146Petr Machata * 1194078ecce3a103c28457e6f90f1e5b0dacc61146Petr Machata * This program is distributed in the hope that it will be useful, but 1294078ecce3a103c28457e6f90f1e5b0dacc61146Petr Machata * WITHOUT ANY WARRANTY; without even the implied warranty of 1394078ecce3a103c28457e6f90f1e5b0dacc61146Petr Machata * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 1494078ecce3a103c28457e6f90f1e5b0dacc61146Petr Machata * General Public License for more details. 1594078ecce3a103c28457e6f90f1e5b0dacc61146Petr Machata * 1694078ecce3a103c28457e6f90f1e5b0dacc61146Petr Machata * You should have received a copy of the GNU General Public License 1794078ecce3a103c28457e6f90f1e5b0dacc61146Petr Machata * along with this program; if not, write to the Free Software 1894078ecce3a103c28457e6f90f1e5b0dacc61146Petr Machata * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 1994078ecce3a103c28457e6f90f1e5b0dacc61146Petr Machata * 02110-1301 USA 2094078ecce3a103c28457e6f90f1e5b0dacc61146Petr Machata */ 2194078ecce3a103c28457e6f90f1e5b0dacc61146Petr Machata 22f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes#include "config.h" 23f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 243ed2a4227d5f2a8d44f3c6ffd704a9d2e867c591Petr Machata#include <sys/param.h> 253ed2a4227d5f2a8d44f3c6ffd704a9d2e867c591Petr Machata#include <sys/wait.h> 263ed2a4227d5f2a8d44f3c6ffd704a9d2e867c591Petr Machata#include <errno.h> 273ed2a4227d5f2a8d44f3c6ffd704a9d2e867c591Petr Machata#include <signal.h> 28f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes#include <stdio.h> 29f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes#include <stdlib.h> 30f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes#include <string.h> 313ed2a4227d5f2a8d44f3c6ffd704a9d2e867c591Petr Machata#include <unistd.h> 32f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 33f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes#include "common.h" 34366c2f46d844f040458df9b7e35fc3b8527ed2d3Petr Machata#include "proc.h" 3594078ecce3a103c28457e6f90f1e5b0dacc61146Petr Machata#include "read_config_file.h" 36f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 37f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedeschar *command = NULL; 38f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 39f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedesint exiting = 0; /* =1 if a SIGINT or SIGTERM has been received */ 40f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 412b46cfc1127d390eddd9593fe5ce5399c1f68130Petr Machatastatic enum callback_status 422b46cfc1127d390eddd9593fe5ce5399c1f68130Petr Machatastop_non_p_processes(Process *proc, void *data) 43cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata{ 44cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata int stop = 1; 45f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 46cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata struct opt_p_t *it; 47cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata for (it = opt_p; it != NULL; it = it->next) { 48cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata Process * p_proc = pid2proc(it->pid); 49cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata if (p_proc == NULL) { 50cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata printf("stop_non_p_processes: %d terminated?\n", it->pid); 51cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata continue; 52f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 539a5420c82a2fd81681572a2e3859ea1671c3bdedPetr Machata if (p_proc == proc || p_proc->leader == proc->leader) { 54cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata stop = 0; 55cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata break; 56cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata } 57cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata } 58cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata 59cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata if (stop) { 60cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata debug(2, "Sending SIGSTOP to process %u", proc->pid); 61cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata kill(proc->pid, SIGSTOP); 62f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 63cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata 642b46cfc1127d390eddd9593fe5ce5399c1f68130Petr Machata return CBS_CONT; 65cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata} 66cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata 67cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machatastatic void 68cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machatasignal_alarm(int sig) { 69cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata signal(SIGALRM, SIG_DFL); 70cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata each_process(NULL, &stop_non_p_processes, NULL); 71f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes} 72f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 73f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedesstatic void 74ffe4cd25089680daf1bd1ec0114d177ec3e0cf95Petr Machatasignal_exit(int sig) 75ffe4cd25089680daf1bd1ec0114d177ec3e0cf95Petr Machata{ 76f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes debug(1, "Received interrupt signal; exiting..."); 77ffe4cd25089680daf1bd1ec0114d177ec3e0cf95Petr Machata if (exiting != 0) 78ffe4cd25089680daf1bd1ec0114d177ec3e0cf95Petr Machata return; 79ffe4cd25089680daf1bd1ec0114d177ec3e0cf95Petr Machata 80ffe4cd25089680daf1bd1ec0114d177ec3e0cf95Petr Machata exiting = 1 + !!os_ltrace_exiting_sighandler(); 81ffe4cd25089680daf1bd1ec0114d177ec3e0cf95Petr Machata 82f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes signal(SIGINT, SIG_IGN); 83f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes signal(SIGTERM, SIG_IGN); 84f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes signal(SIGALRM, signal_alarm); 85602330f458e8049ec07685c4e02f0368821a2d2cPetr Machata //alarm(1); 86f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes} 87f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 88f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedesstatic void 89f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedesnormal_exit(void) { 90f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes output_line(0, 0); 91f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes if (options.summary) { 92f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes show_summary(); 93f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 94f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes if (options.output) { 95f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes fclose(options.output); 96f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes options.output = NULL; 97f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 98f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes} 99f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 100f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedesvoid 101f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedesltrace_init(int argc, char **argv) { 102f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes struct opt_p_t *opt_p_tmp; 103f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 104f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes atexit(normal_exit); 105f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes signal(SIGINT, signal_exit); /* Detach processes when interrupted */ 106f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes signal(SIGTERM, signal_exit); /* ... or killed */ 107f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 108f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes argv = process_options(argc, argv); 109f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes while (opt_F) { 110f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes /* If filename begins with ~, expand it to the user's home */ 111f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes /* directory. This does not correctly handle ~yoda, but that */ 112f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes /* isn't as bad as it seems because the shell will normally */ 113f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes /* be doing the expansion for us; only the hardcoded */ 114f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes /* ~/.ltrace.conf should ever use this code. */ 115f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes if (opt_F->filename[0] == '~') { 116f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes char path[PATH_MAX]; 117f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes char *home_dir = getenv("HOME"); 118f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes if (home_dir) { 119f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes strncpy(path, home_dir, PATH_MAX - 1); 120f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes path[PATH_MAX - 1] = '\0'; 121f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes strncat(path, opt_F->filename + 1, 122f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes PATH_MAX - strlen(path) - 1); 123f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes read_config_file(path); 124f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 125f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } else { 126f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes read_config_file(opt_F->filename); 127f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 128f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes opt_F = opt_F->next; 129f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 130f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes if (command) { 13102bd9eca0b439b6950ab72ba64b67181e5bf0fbdPetr Machata /* Check that the binary ABI is supported before 13202bd9eca0b439b6950ab72ba64b67181e5bf0fbdPetr Machata * calling execute_program. */ 13302bd9eca0b439b6950ab72ba64b67181e5bf0fbdPetr Machata struct ltelf lte = {}; 13402bd9eca0b439b6950ab72ba64b67181e5bf0fbdPetr Machata open_elf(<e, command); 13502bd9eca0b439b6950ab72ba64b67181e5bf0fbdPetr Machata 136c805c624d4cd23674bbc18f9d0f97c5d8dcdff7cPetr Machata pid_t pid = execute_program(command, argv); 13775934ad3b30790f6a892069576d0790a351ef101Petr Machata struct Process *proc = open_program(command, pid); 138cc0e1e4b83d69441cc5f61ea87eda5458ee9fae3Petr Machata if (proc == NULL) { 139cc0e1e4b83d69441cc5f61ea87eda5458ee9fae3Petr Machata fprintf(stderr, "couldn't open program '%s': %s\n", 140cc0e1e4b83d69441cc5f61ea87eda5458ee9fae3Petr Machata command, strerror(errno)); 141cc0e1e4b83d69441cc5f61ea87eda5458ee9fae3Petr Machata exit(EXIT_FAILURE); 142cc0e1e4b83d69441cc5f61ea87eda5458ee9fae3Petr Machata } 14379a47bafb099516cea0475317576f3b6a7bf82fbPetr Machata 14479a47bafb099516cea0475317576f3b6a7bf82fbPetr Machata trace_set_options(proc); 145c805c624d4cd23674bbc18f9d0f97c5d8dcdff7cPetr Machata continue_process(pid); 146f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 147f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes opt_p_tmp = opt_p; 148f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes while (opt_p_tmp) { 1498d1b92ba755f6d6229f5e230fc43d958b13836f8Juan Cespedes open_pid(opt_p_tmp->pid); 150f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes opt_p_tmp = opt_p_tmp->next; 151f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 152f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes} 153f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 15461da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedesstatic int num_ltrace_callbacks[EVENT_MAX]; 15561da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedesstatic callback_func * ltrace_callbacks[EVENT_MAX]; 15640dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedes 15740dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedesvoid 15861da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedesltrace_add_callback(callback_func func, Event_type type) { 15961da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes ltrace_callbacks[type] = realloc(ltrace_callbacks[type], (num_ltrace_callbacks[type]+1)*sizeof(callback_func)); 16061da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes ltrace_callbacks[type][num_ltrace_callbacks[type]++] = func; 16161da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes} 16240dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedes 16361da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedesstatic void 16461da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedesdispatch_callbacks(Event * ev) { 16561da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes int i; 16661da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes /* Ignoring case 1: signal into a dying tracer */ 16761da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes if (ev->type==EVENT_SIGNAL && 16861da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes exiting && ev->e_un.signum == SIGSTOP) { 16961da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes return; 17061da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes } 17161da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes /* Ignoring case 2: process being born before a clone event */ 17261da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes if (ev->proc && ev->proc->state == STATE_IGNORED) { 17361da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes return; 17461da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes } 17561da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes for (i=0; i<num_ltrace_callbacks[ev->type]; i++) { 17661da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes ltrace_callbacks[ev->type][i](ev); 17740dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedes } 17840dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedes} 17940dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedes 180f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedesvoid 181f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedesltrace_main(void) { 18240dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedes Event * ev; 183f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes while (1) { 18440dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedes ev = next_event(); 18561da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes dispatch_callbacks(ev); 18603192f890af95bf9abf5ea1c3dc0998f6b12550dJuan Cespedes handle_event(ev); 187f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 188f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes} 189