forked from namhyung/uftrace
-
Notifications
You must be signed in to change notification settings - Fork 0
/
script.c
148 lines (120 loc) · 3.11 KB
/
script.c
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
/*
* Script binding for function entry and exit
*
* Copyright (C) 2017, LG Electronics, Honggyu Kim <hong.gyu.kim@lge.com>
*
* Released under the GPL v2.
*/
/* This should be defined before #include "utils.h" */
#define PR_FMT "script"
#define PR_DOMAIN DBG_SCRIPT
#include <unistd.h>
#include "utils/script.h"
#include "utils/filter.h"
#include "utils/list.h"
#include "utils/utils.h"
#include "utils/script-python.h"
#include "utils/script-luajit.h"
/* This will be set by getenv("UFTRACE_SCRIPT"). */
char *script_str;
enum script_type_t script_lang;
/* The below functions are used both in record time and script command. */
script_uftrace_entry_t script_uftrace_entry;
script_uftrace_exit_t script_uftrace_exit;
script_uftrace_end_t script_uftrace_end;
script_atfork_prepare_t script_atfork_prepare;
struct script_filter_item {
struct list_head list;
struct uftrace_pattern patt;
};
static LIST_HEAD(filters);
enum script_type_t get_script_type(const char *str)
{
char *ext = strrchr(str, '.');
/*
* The given script will be detected by the file suffix.
* As of now, it only handles ".py" suffix for python.
*/
if (!strcmp(ext, ".py"))
return SCRIPT_PYTHON;
else if (!strcmp(ext, ".lua"))
return SCRIPT_LUAJIT;
return SCRIPT_UNKNOWN;
}
void script_add_filter(char *func, enum uftrace_pattern_type ptype)
{
struct script_filter_item *item;
if (func == NULL)
return;
item = xmalloc(sizeof(*item));
init_filter_pattern(ptype, &item->patt, func);
pr_dbg2("add script filter: %s (%s)\n", func,
get_filter_pattern(item->patt.type));
list_add_tail(&item->list, &filters);
}
/* returns 1 on match - script should be run */
int script_match_filter(char *func)
{
struct script_filter_item *item;
/* special case: no filter */
if (list_empty(&filters))
return 1;
list_for_each_entry(item, &filters, list) {
if (match_filter_pattern(&item->patt, func))
return 1;
}
return 0;
}
void script_finish_filter(void)
{
struct script_filter_item *item, *tmp;
list_for_each_entry_safe(item, tmp, &filters, list) {
free_filter_pattern(&item->patt);
free(item);
}
}
int script_init(struct script_info *info, enum uftrace_pattern_type ptype)
{
char *script_pathname = info->name;
pr_dbg2("%s(\"%s\")\n", __func__, script_pathname);
if (access(script_pathname, F_OK) < 0) {
perror(script_pathname);
return -1;
}
script_lang = get_script_type(script_pathname);
switch (script_lang) {
case SCRIPT_PYTHON:
if (script_init_for_python(info, ptype) < 0) {
pr_warn("failed to init python scripting\n");
script_pathname = NULL;
}
break;
case SCRIPT_LUAJIT:
if (script_init_for_luajit(info, ptype) < 0) {
pr_warn("failed to init luajit scripting\n");
script_pathname = NULL;
}
break;
default:
pr_warn("unsupported script type: %s\n", script_pathname);
script_pathname = NULL;
}
if (script_pathname == NULL)
return -1;
return 0;
}
void script_finish(void)
{
pr_dbg2("%s()\n", __func__);
switch (script_lang) {
case SCRIPT_PYTHON:
script_finish_for_python();
break;
case SCRIPT_LUAJIT:
script_finish_for_luajit();
break;
default:
break;
}
script_finish_filter();
}