* [PATCH v4] trace-cmd: Stop recording when processes traced with -P exit
@ 2020-05-14 16:42 Bijan Tabatabai
2020-05-14 22:43 ` Steven Rostedt
0 siblings, 1 reply; 2+ messages in thread
From: Bijan Tabatabai @ 2020-05-14 16:42 UTC (permalink / raw)
To: rostedt; +Cc: linux-trace-devel, Bijan Tabatabai
From: Bijan Tabatabai <bijan311@yahoo.com>
When the -F flag is used in trace-cmd record, trace-cmd stops recording
when the executable it is tracing terminates.
This patch makes the -P flag act similarly.
Signed-off-by: Bijan Tabatabai <bijan311@yahoo.com>
---
v4 fixes a bug where trace-cmd would stop tracing after the processes in
the first instance exited.
---
tracecmd/include/trace-local.h | 2 +
tracecmd/trace-record.c | 91 +++++++++++++++++++++++++++++++++-
2 files changed, 92 insertions(+), 1 deletion(-)
diff --git a/tracecmd/include/trace-local.h b/tracecmd/include/trace-local.h
index 4c6a63d..5d58550 100644
--- a/tracecmd/include/trace-local.h
+++ b/tracecmd/include/trace-local.h
@@ -215,9 +215,11 @@ struct buffer_instance {
struct opt_list *options;
struct filter_pids *filter_pids;
+ struct filter_pids *process_pids;
char *common_pid_filter;
int nr_filter_pids;
int len_filter_pids;
+ int nr_process_pids;
bool ptrace_child;
int have_set_event_pid;
diff --git a/tracecmd/trace-record.c b/tracecmd/trace-record.c
index d0619ba..cb1634d 100644
--- a/tracecmd/trace-record.c
+++ b/tracecmd/trace-record.c
@@ -16,6 +16,7 @@
#include <sys/time.h>
#include <sys/wait.h>
#include <sys/socket.h>
+#include <sys/syscall.h>
#include <sys/utsname.h>
#ifndef NO_PTRACE
#include <sys/ptrace.h>
@@ -33,6 +34,7 @@
#include <errno.h>
#include <limits.h>
#include <libgen.h>
+#include <poll.h>
#include <pwd.h>
#include <grp.h>
#ifdef VSOCK
@@ -1230,6 +1232,81 @@ static pid_t trace_waitpid(enum trace_type type, pid_t pid, int *status, int opt
trace_stream_read(pids, recorder_threads, &tv);
} while (1);
}
+
+#ifndef __NR_pidfd_open
+#define __NR_pidfd_open 434
+#endif
+
+static int pidfd_open(pid_t pid, unsigned int flags) {
+ return syscall(__NR_pidfd_open, pid, flags);
+}
+
+static int trace_waitpidfd(id_t pidfd) {
+ struct pollfd pollfd;
+
+ pollfd.fd = pidfd;
+ pollfd.events = POLLIN;
+
+ while (!finished) {
+ int ret = poll(&pollfd, 1, -1);
+ /* If waitid was interrupted, keep waiting */
+ if (ret < 0 && errno == EINTR)
+ continue;
+ else if (ret < 0)
+ return 1;
+ else
+ break;
+ }
+
+ return 0;
+}
+
+static int trace_wait_for_processes(struct buffer_instance *instance) {
+ int ret = 0;
+ int nr_fds = 0;
+ int i;
+ int *pidfds;
+ struct filter_pids *pid;
+
+ pidfds = malloc(sizeof(int) * instance->nr_process_pids);
+ if (!pidfds)
+ return 1;
+
+ for (pid = instance->process_pids;
+ pid && instance->nr_process_pids;
+ pid = pid->next) {
+ if (pid->exclude) {
+ instance->nr_process_pids--;
+ continue;
+ }
+ pidfds[nr_fds] = pidfd_open(pid->pid, 0);
+
+ /* If the pid doesn't exist, the process has probably exited */
+ if (pidfds[nr_fds] < 0 && errno == ESRCH) {
+ instance->nr_process_pids--;
+ continue;
+ } else if (pidfds[nr_fds] < 0) {
+ ret = 1;
+ goto out;
+ }
+
+ nr_fds++;
+ instance->nr_process_pids--;
+ }
+
+ for (i = 0; i < nr_fds; i++) {
+ if (trace_waitpidfd(pidfds[i])) {
+ ret = 1;
+ goto out;
+ }
+ }
+
+out:
+ for (i = 0; i < nr_fds; i++)
+ close(pidfds[i]);
+ free(pidfds);
+ return ret;
+}
#ifndef NO_PTRACE
/**
@@ -5866,7 +5943,9 @@ static void parse_record_options(int argc,
fpids_count += add_filter_pid(ctx->instance,
atoi(pid), 0);
pid = strtok_r(NULL, ",", &sav);
+ ctx->instance->nr_process_pids++;
}
+ ctx->instance->process_pids = ctx->instance->filter_pids;
free(pids);
break;
case 'c':
@@ -6345,6 +6424,7 @@ static void record_trace(int argc, char **argv,
tracecmd_msg_wait_close(ctx->instance->msg_handle);
} else {
bool pwait = false;
+ bool wait_indefinitely = false;
update_task_filter();
tracecmd_enable_tracing();
@@ -6361,7 +6441,16 @@ static void record_trace(int argc, char **argv,
}
/* sleep till we are woken with Ctrl^C */
printf("Hit Ctrl^C to stop recording\n");
- while (!finished)
+ for_all_instances(instance) {
+ /* If an instance is not tracing individual processes
+ * or there is an error while waiting for a process to
+ * exit, fallback to waiting indefinitely.
+ */
+ if (!instance->nr_process_pids
+ || trace_wait_for_processes(instance))
+ wait_indefinitely = true;
+ }
+ while (!finished && wait_indefinitely)
trace_or_sleep(type, pwait);
}
--
2.25.1
^ permalink raw reply related [flat|nested] 2+ messages in thread
* Re: [PATCH v4] trace-cmd: Stop recording when processes traced with -P exit
2020-05-14 16:42 [PATCH v4] trace-cmd: Stop recording when processes traced with -P exit Bijan Tabatabai
@ 2020-05-14 22:43 ` Steven Rostedt
0 siblings, 0 replies; 2+ messages in thread
From: Steven Rostedt @ 2020-05-14 22:43 UTC (permalink / raw)
To: Bijan Tabatabai; +Cc: linux-trace-devel, Bijan Tabatabai
On Thu, 14 May 2020 11:42:44 -0500
Bijan Tabatabai <bijan311@gmail.com> wrote:
> From: Bijan Tabatabai <bijan311@yahoo.com>
>
> When the -F flag is used in trace-cmd record, trace-cmd stops recording
> when the executable it is tracing terminates.
> This patch makes the -P flag act similarly.
>
> Signed-off-by: Bijan Tabatabai <bijan311@yahoo.com>
> ---
> v4 fixes a bug where trace-cmd would stop tracing after the processes in
> the first instance exited.
> ---
>
Looks good, but there's one small nit.
> @@ -6361,7 +6441,16 @@ static void record_trace(int argc, char **argv,
> }
> /* sleep till we are woken with Ctrl^C */
> printf("Hit Ctrl^C to stop recording\n");
> - while (!finished)
> + for_all_instances(instance) {
> + /* If an instance is not tracing individual processes
> + * or there is an error while waiting for a process to
> + * exit, fallback to waiting indefinitely.
> + */
> + if (!instance->nr_process_pids
> + || trace_wait_for_processes(instance))
My preference is:
if (!instance->nr_process_pids ||
trace_wait_for_processes(instance))
But I can make this change.
Thanks!
-- Steve
> + wait_indefinitely = true;
> + }
> + while (!finished && wait_indefinitely)
> trace_or_sleep(type, pwait);
> }
>
^ permalink raw reply [flat|nested] 2+ messages in thread
end of thread, other threads:[~2020-05-14 22:43 UTC | newest]
Thread overview: 2+ messages (download: mbox.gz / follow: Atom feed)
-- links below jump to the message on this page --
2020-05-14 16:42 [PATCH v4] trace-cmd: Stop recording when processes traced with -P exit Bijan Tabatabai
2020-05-14 22:43 ` Steven Rostedt
This is an external index of several public inboxes,
see mirroring instructions on how to clone and mirror
all data and code used by this external index.