194078ecce3a103c28457e6f90f1e5b0dacc61146Petr Machata/* 294078ecce3a103c28457e6f90f1e5b0dacc61146Petr Machata * This file is part of ltrace. 3673ff510953b65b844a58478aa434120f457c014Petr Machata * Copyright (C) 2011,2012,2013 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> 27c00837c2928da53a3515b107399b742ea157e78aPetr Machata#include <limits.h> 28c00837c2928da53a3515b107399b742ea157e78aPetr Machata#include <locale.h> 293ed2a4227d5f2a8d44f3c6ffd704a9d2e867c591Petr Machata#include <signal.h> 30f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes#include <stdio.h> 31f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes#include <stdlib.h> 32f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes#include <string.h> 333ed2a4227d5f2a8d44f3c6ffd704a9d2e867c591Petr Machata#include <unistd.h> 34f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 358a730f3e5844b86386032e93f91aa3df814ad9c3Petr Machata#include "backend.h" 36f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes#include "common.h" 37366c2f46d844f040458df9b7e35fc3b8527ed2d3Petr Machata#include "proc.h" 38e6c00ae8228be812c7361b5e193779a81aa00cd8Petr Machata#include "prototype.h" 398a730f3e5844b86386032e93f91aa3df814ad9c3Petr Machata#include "read_config_file.h" 408a730f3e5844b86386032e93f91aa3df814ad9c3Petr Machata#include "summary.h" 41f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 42f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedeschar *command = NULL; 43f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 44f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedesint exiting = 0; /* =1 if a SIGINT or SIGTERM has been received */ 45f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 462b46cfc1127d390eddd9593fe5ce5399c1f68130Petr Machatastatic enum callback_status 47929bd57ca202fd2f2e8485ebf65d683e664f67b5Petr Machatastop_non_p_processes(struct process *proc, void *data) 48cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata{ 49cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata int stop = 1; 50f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 51cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata struct opt_p_t *it; 52cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata for (it = opt_p; it != NULL; it = it->next) { 53929bd57ca202fd2f2e8485ebf65d683e664f67b5Petr Machata struct process *p_proc = pid2proc(it->pid); 54cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata if (p_proc == NULL) { 55cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata printf("stop_non_p_processes: %d terminated?\n", it->pid); 56cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata continue; 57f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 589a5420c82a2fd81681572a2e3859ea1671c3bdedPetr Machata if (p_proc == proc || p_proc->leader == proc->leader) { 59cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata stop = 0; 60cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata break; 61cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata } 62cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata } 63cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata 64cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata if (stop) { 65cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata debug(2, "Sending SIGSTOP to process %u", proc->pid); 66cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata kill(proc->pid, SIGSTOP); 67f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 68cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata 692b46cfc1127d390eddd9593fe5ce5399c1f68130Petr Machata return CBS_CONT; 70cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata} 71cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata 72cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machatastatic void 73cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machatasignal_alarm(int sig) { 74cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata signal(SIGALRM, SIG_DFL); 75cebb88491e68bdf4d466b03970c9ca77f65615f8Petr Machata each_process(NULL, &stop_non_p_processes, NULL); 76f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes} 77f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 78f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedesstatic void 79ffe4cd25089680daf1bd1ec0114d177ec3e0cf95Petr Machatasignal_exit(int sig) 80ffe4cd25089680daf1bd1ec0114d177ec3e0cf95Petr Machata{ 81ffe4cd25089680daf1bd1ec0114d177ec3e0cf95Petr Machata if (exiting != 0) 82ffe4cd25089680daf1bd1ec0114d177ec3e0cf95Petr Machata return; 83ffe4cd25089680daf1bd1ec0114d177ec3e0cf95Petr Machata 84ffe4cd25089680daf1bd1ec0114d177ec3e0cf95Petr Machata exiting = 1 + !!os_ltrace_exiting_sighandler(); 85ffe4cd25089680daf1bd1ec0114d177ec3e0cf95Petr Machata 86f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes signal(SIGINT, SIG_IGN); 87f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes signal(SIGTERM, SIG_IGN); 88f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes signal(SIGALRM, signal_alarm); 89602330f458e8049ec07685c4e02f0368821a2d2cPetr Machata //alarm(1); 90f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes} 91f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 92f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedesstatic void 93efc9436b1ac30e73347696cf3fcfc9ad3f60772fPetr Machatanormal_exit(void) 94efc9436b1ac30e73347696cf3fcfc9ad3f60772fPetr Machata{ 9593eb0ee606dc4425cc2f1019f5e549acf77f40b0Petr Machata if (options.summary) 96f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes show_summary(); 97f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes if (options.output) { 98f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes fclose(options.output); 99f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes options.output = NULL; 100f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 101f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes} 102f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 103f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedesvoid 104c00837c2928da53a3515b107399b742ea157e78aPetr Machataltrace_init(int argc, char **argv) 105c00837c2928da53a3515b107399b742ea157e78aPetr Machata{ 106c00837c2928da53a3515b107399b742ea157e78aPetr Machata setlocale(LC_ALL, ""); 107c00837c2928da53a3515b107399b742ea157e78aPetr Machata 108f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes struct opt_p_t *opt_p_tmp; 109f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 110f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes atexit(normal_exit); 111f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes signal(SIGINT, signal_exit); /* Detach processes when interrupted */ 112f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes signal(SIGTERM, signal_exit); /* ... or killed */ 113f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 114f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes argv = process_options(argc, argv); 115f83053d927d0e7a488b1077136eea7611a022553Petr Machata init_global_config(); 1168a98e6f756c7e6a4171e7dee563f0efbb28b9f33Petr Machata 117f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes if (command) { 11802bd9eca0b439b6950ab72ba64b67181e5bf0fbdPetr Machata /* Check that the binary ABI is supported before 11902bd9eca0b439b6950ab72ba64b67181e5bf0fbdPetr Machata * calling execute_program. */ 1200ba3c5eee259b77e3883e40c4d0cd2fab5b03ff3Petr Machata { 1210ba3c5eee259b77e3883e40c4d0cd2fab5b03ff3Petr Machata struct ltelf lte; 1220ba3c5eee259b77e3883e40c4d0cd2fab5b03ff3Petr Machata if (ltelf_init(<e, command) == 0) 1230ba3c5eee259b77e3883e40c4d0cd2fab5b03ff3Petr Machata ltelf_destroy(<e); 1240ba3c5eee259b77e3883e40c4d0cd2fab5b03ff3Petr Machata else 1250ba3c5eee259b77e3883e40c4d0cd2fab5b03ff3Petr Machata exit(EXIT_FAILURE); 1260ba3c5eee259b77e3883e40c4d0cd2fab5b03ff3Petr Machata } 12702bd9eca0b439b6950ab72ba64b67181e5bf0fbdPetr Machata 128c805c624d4cd23674bbc18f9d0f97c5d8dcdff7cPetr Machata pid_t pid = execute_program(command, argv); 129929bd57ca202fd2f2e8485ebf65d683e664f67b5Petr Machata struct process *proc = open_program(command, pid); 130cc0e1e4b83d69441cc5f61ea87eda5458ee9fae3Petr Machata if (proc == NULL) { 131cc0e1e4b83d69441cc5f61ea87eda5458ee9fae3Petr Machata fprintf(stderr, "couldn't open program '%s': %s\n", 132cc0e1e4b83d69441cc5f61ea87eda5458ee9fae3Petr Machata command, strerror(errno)); 133cc0e1e4b83d69441cc5f61ea87eda5458ee9fae3Petr Machata exit(EXIT_FAILURE); 134cc0e1e4b83d69441cc5f61ea87eda5458ee9fae3Petr Machata } 13579a47bafb099516cea0475317576f3b6a7bf82fbPetr Machata 13679a47bafb099516cea0475317576f3b6a7bf82fbPetr Machata trace_set_options(proc); 137c805c624d4cd23674bbc18f9d0f97c5d8dcdff7cPetr Machata continue_process(pid); 138f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 139f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes opt_p_tmp = opt_p; 140f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes while (opt_p_tmp) { 1418d1b92ba755f6d6229f5e230fc43d958b13836f8Juan Cespedes open_pid(opt_p_tmp->pid); 142f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes opt_p_tmp = opt_p_tmp->next; 143f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 144f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes} 145f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes 14661da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedesstatic int num_ltrace_callbacks[EVENT_MAX]; 14761da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedesstatic callback_func * ltrace_callbacks[EVENT_MAX]; 14840dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedes 14940dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedesvoid 15061da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedesltrace_add_callback(callback_func func, Event_type type) { 15161da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes ltrace_callbacks[type] = realloc(ltrace_callbacks[type], (num_ltrace_callbacks[type]+1)*sizeof(callback_func)); 15261da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes ltrace_callbacks[type][num_ltrace_callbacks[type]++] = func; 15361da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes} 15440dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedes 15561da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedesstatic void 15661da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedesdispatch_callbacks(Event * ev) { 15761da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes int i; 15861da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes /* Ignoring case 1: signal into a dying tracer */ 15961da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes if (ev->type==EVENT_SIGNAL && 16061da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes exiting && ev->e_un.signum == SIGSTOP) { 16161da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes return; 16261da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes } 16361da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes /* Ignoring case 2: process being born before a clone event */ 16461da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes if (ev->proc && ev->proc->state == STATE_IGNORED) { 16561da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes return; 16661da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes } 16761da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes for (i=0; i<num_ltrace_callbacks[ev->type]; i++) { 16861da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes ltrace_callbacks[ev->type][i](ev); 16940dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedes } 17040dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedes} 17140dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedes 172f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedesvoid 173f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedesltrace_main(void) { 17440dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedes Event * ev; 175f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes while (1) { 17640dc6353a443f2eed64fcada3bcf305e56855b15Juan Cespedes ev = next_event(); 17761da33723c5fb09762e38bd39a26ee15d62ffebcJuan Cespedes dispatch_callbacks(ev); 17803192f890af95bf9abf5ea1c3dc0998f6b12550dJuan Cespedes handle_event(ev); 179f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes } 180f728123bd75a65a6a1536e198c3c30719e494e71Juan Cespedes} 181