1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
|
#include <stdio.h>
#include <unistd.h>
#include <dirent.h>
#include <signal.h>
#include <errno.h>
#include "uftrace.h"
#include "utils/utils.h"
#include "utils/fstack.h"
#include "utils/kernel.h"
#include "libmcount/mcount.h"
static char *tmp_dirname;
static void cleanup_tempdir(void)
{
if (!tmp_dirname)
return;
remove_directory(tmp_dirname);
tmp_dirname = NULL;
}
static void reset_live_opts(struct opts *opts)
{
/* this is needed to set display_depth at replay */
live_disabled = opts->disabled;
/*
* These options are handled in record and no need to do it in
* replay again.
*/
opts->filter = NULL;
opts->depth = MCOUNT_DEFAULT_DEPTH;
opts->disabled = false;
opts->no_event = false;
}
static void sigsegv_handler(int sig)
{
pr_warn("Segmentation fault\n");
cleanup_tempdir();
raise(sig);
}
static bool can_skip_replay(struct opts *opts, int record_result)
{
if (opts->nop)
return true;
return false;
}
static void setup_child_environ(struct opts *opts)
{
char *old_preload, *libpath;
#ifdef INSTALL_LIB_PATH
if (!opts->lib_path) {
char *envbuf = getenv("LD_LIBRARY_PATH");
if (envbuf) {
envbuf = xstrdup(envbuf);
libpath = strjoin(envbuf, INSTALL_LIB_PATH, ":");
setenv("LD_LIBRARY_PATH", libpath, 1);
free(libpath);
}
else {
setenv("LD_LIBRARY_PATH", INSTALL_LIB_PATH, 1);
}
}
#endif
libpath = get_libmcount_path(opts);
if (libpath == NULL)
pr_err_ns("cannot found libmcount.so\n");
old_preload = getenv("LD_PRELOAD");
if (old_preload) {
size_t len = strlen(libpath) + strlen(old_preload) + 2;
char *preload = xmalloc(len);
snprintf(preload, len, "%s:%s", libpath, old_preload);
setenv("LD_PRELOAD", preload, 1);
free(preload);
}
else
setenv("LD_PRELOAD", libpath, 1);
}
int command_live(int argc, char *argv[], struct opts *opts)
{
char template[32] = "/tmp/uftrace-live-XXXXXX";
int fd = mkstemp(template);
struct sigaction sa = {
.sa_flags = SA_RESETHAND,
};
int ret;
if (fd < 0)
pr_err("cannot create temp name");
close(fd);
unlink(template);
tmp_dirname = template;
atexit(cleanup_tempdir);
sa.sa_handler = sigsegv_handler;
sigfillset(&sa.sa_mask);
sigaction(SIGSEGV, &sa, NULL);
opts->dirname = template;
if (opts->list_event) {
if (geteuid() == 0)
list_kernel_events();
if (fork() == 0) {
setup_child_environ(opts);
setenv("UFTRACE_LIST_EVENT", "1", 1);
execv(opts->exename, argv);
abort();
}
return 0;
}
ret = command_record(argc, argv, opts);
if (!can_skip_replay(opts, ret)) {
int ret2;
reset_live_opts(opts);
pr_dbg("live-record finished.. \n");
if (opts->report) {
pr_out("#\n# uftrace report\n#\n");
ret2 = command_report(argc, argv, opts);
if (ret == UFTRACE_EXIT_SUCCESS)
ret = ret2;
pr_out("\n#\n# uftrace replay\n#\n");
}
pr_dbg("start live-replaying...\n");
ret2 = command_replay(argc, argv, opts);
if (ret == UFTRACE_EXIT_SUCCESS)
ret = ret2;
}
cleanup_tempdir();
return ret;
}
|