blob: 3673a9b0a9a39033bb64254189d4f4d0d23067f8 (
plain)
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
|
// build/run probes
// Copyright (C) 2005, 2006 Red Hat Inc.
//
// This file is part of systemtap, and is free software. You can
// redistribute it and/or modify it under the terms of the GNU General
// Public License (GPL); either version 2, or (at your option) any
// later version.
#include "config.h"
#include "buildrun.h"
#include "session.h"
#include "util.h"
#include <fstream>
#include <sstream>
extern "C" {
#include "signal.h"
#include <sys/wait.h>
#include <pwd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <string.h>
#include <errno.h>
}
using namespace std;
int
compile_pass (systemtap_session& s)
{
// fill in a quick Makefile
string makefile_nm = s.tmpdir + "/Makefile";
ofstream o (makefile_nm.c_str());
int rc = 0;
// Create makefile
for (unsigned i=0; i<s.macros.size(); i++)
o << "CFLAGS += -D " << lex_cast_qstring(s.macros[i]) << endl;
if (s.verbose > 2)
o << "CFLAGS += -ftime-report -Q" << endl;
o << "CFLAGS += -freorder-blocks" << endl; // improve on -Os
// o << "CFLAGS += -fno-unit-at-a-time" << endl;
// Assumes linux 2.6 kbuild
o << "CFLAGS += -Wno-unused -Werror" << endl;
o << "CFLAGS += -I\"" << s.runtime_path << "\"" << endl;
// XXX: this may help ppc toc overflow
// o << "CFLAGS := $(subst -Os,-O2,$(CFLAGS)) -fminimal-toc" << endl;
o << "obj-m := " << s.module_name << ".o" << endl;
o.close ();
// Generate module directory pathname and make sure it exists.
string module_dir = string("/lib/modules/")
+ s.kernel_release + "/build";
struct stat st;
rc = stat(module_dir.c_str(), &st);
if (rc != 0)
{
clog << "Module directory " << module_dir << " check failed: "
<< strerror(errno) << endl
<< "Make sure kernel devel is installed." << endl;
return rc;
}
// Run make
string make_cmd = string("make")
+ string (" -C \"") + module_dir + string("\"");
make_cmd += string(" M=\"") + s.tmpdir + string("\" modules");
if (s.verbose > 1)
make_cmd += " V=1";
else
make_cmd += " -s >/dev/null 2>&1";
if (s.verbose > 1) clog << "Running " << make_cmd << endl;
rc = system (make_cmd.c_str());
return rc;
}
int
run_pass (systemtap_session& s)
{
int rc = 0;
struct passwd *pw = getpwuid(getuid());
string username = string(pw->pw_name);
// for now, just spawn staprun
string staprun_cmd = string("sudo ")
+ string(BINDIR) + "/staprun "
+ (s.verbose>1 ? "" : "-q ")
+ "-u " + username + " "
+ (s.output_file.empty() ? "" : "-o " + s.output_file + " ");
staprun_cmd += "-d " + stringify(getpid()) + " ";
if (s.cmd != "")
staprun_cmd += "-c \"" + s.cmd + "\" ";
if (s.target_pid)
staprun_cmd += "-t " + stringify(s.target_pid) + " ";
if (s.buffer_size)
staprun_cmd += "-b " + stringify(s.buffer_size) + " ";
staprun_cmd += s.tmpdir + "/" + s.module_name + ".ko";
if (s.verbose>1) clog << "Running " << staprun_cmd << endl;
signal (SIGHUP, SIG_IGN);
signal (SIGINT, SIG_IGN);
rc = system (staprun_cmd.c_str ());
return rc;
}
|
