summaryrefslogtreecommitdiffstats
path: root/main.cxx
diff options
context:
space:
mode:
Diffstat (limited to 'main.cxx')
-rw-r--r--main.cxx267
1 files changed, 198 insertions, 69 deletions
diff --git a/main.cxx b/main.cxx
index a31d53fa..ca2f4dd3 100644
--- a/main.cxx
+++ b/main.cxx
@@ -1,4 +1,4 @@
-// systemtap translator driver
+// systemtap translator/driver
// Copyright (C) 2005 Red Hat Inc.
//
// This file is part of systemtap, and is free software. You can
@@ -11,38 +11,67 @@
#include "parse.h"
#include "elaborate.h"
#include "translate.h"
+#include "buildrun.h"
#include <iostream>
#include <fstream>
#include <sstream>
+#include <cerrno>
extern "C" {
#include <glob.h>
#include <unistd.h>
+#include <sys/utsname.h>
+#include <time.h>
}
using namespace std;
-bool verbose = false;
-void usage ()
+void
+usage (systemtap_session& s)
{
- cerr << "SystemTap translator "
- << "(version " << VERSION << " built " << DATE << ")" << endl;
- cerr << "Copyright (C) 2005 Red Hat, Inc." << endl;
- cerr << "This is free software; see the source for copying conditions." << endl;
- cerr << endl;
- cerr << "Usage: stap [options] FILE [ARGS ...] Run script in file." << endl;
- cerr << " or: stap [options] - [ARGS ...] Run script on stdin." << endl;
- cerr << " or: stap [options] -e SCRIPT [ARGS ...] Run given script." << endl;
- cerr << endl;
- cerr << "Arguments:" << endl;
- cerr << " -- No more options after this" << endl;
- cerr << " -v verbose" << endl;
- cerr << " -p NUM Stop after pass NUM 1-3" << endl;
- cerr << " (parse, elaborate, translate)" << endl;
- cerr << " -I DIR Look in DIR for additional .stp script files." << endl;
- cerr << " -o FILE Send output to file instead of stdout." << endl;
+ cerr
+ << "SystemTap translator "
+ << "(version " << VERSION << " built " << DATE << ")" << endl
+ << "Copyright (C) 2005 Red Hat, Inc." << endl
+ << "This is free software; see the source for copying conditions."
+ << endl
+ << endl
+ << "Usage: stap [options] FILE [ARGS ...] Run script in file."
+ << endl
+ << " or: stap [options] - [ARGS ...] Run script on stdin."
+ << endl
+ << " or: stap [options] -e SCRIPT [ARGS ...] Run given script."
+ << endl
+ << endl
+ << "Arguments:" << endl
+ << " -- no more options after this" << endl
+ << " -v verbose" << (s.verbose ? " [set]" : "")
+ << endl
+ << " -t test mode" << (s.test_mode ? " [set]" : "")
+ << endl
+ << " -p NUM stop after pass NUM 1-5" << endl
+ << " (parse, elaborate, translate, compile, run)" << endl
+ << " -I DIR look in DIR for additional .stp script files";
+ if (s.include_path.size() == 0)
+ cerr << endl;
+ else
+ cerr << ", instead of" << endl;
+ for (unsigned i=0; i<s.include_path.size(); i++)
+ cerr << " " << s.include_path[i] << endl;
+ cerr
+ << " -R DIR look in DIR for runtime, instead of "
+ << s.runtime_path
+ << endl
+ << " -r RELEASE use kernel RELEASE, instead of "
+ << s.kernel_release
+ << endl
+ << " -m MODULE set probe module name, insetad of "
+ << s.module_name
+ << endl
+ << " -o FILE send output to file instead of stdout" << endl
+ << " -k keep temporary directory" << endl;
// XXX: other options:
// -s: safe mode
// -d: dump safety-related external references
@@ -51,57 +80,97 @@ void usage ()
}
+// little utility function
+
+template <typename T>
+static string
+stringify(T t)
+{
+ ostringstream s;
+ s << t;
+ return s.str ();
+}
+
+
int
main (int argc, char * const argv [])
{
- int last_pass = 3; // -p NUM
string cmdline_script; // -e PROGRAM
string script_file; // FILE
bool have_script = false;
- string output_file; // -o FILE
- vector<string> include_path; // -I DIR
- vector<string> args; // ARGS
+ string output_file = "-"; // -o FILE
+
+ // Initialize defaults
+ systemtap_session s;
+ struct utsname buf;
+ (void) uname (& buf);
+ s.kernel_release = string (buf.release);
+ s.verbose = false;
+ s.test_mode = false;
+ s.last_pass = 5;
+ s.runtime_path = "/usr/share/systemtap/runtime"; // XXX
+ s.module_name = "stap_" + stringify(getuid()) + "_" + stringify(time(0));
+ s.keep_tmpdir = false;
+
while (true)
{
- int grc = getopt (argc, argv, "vp:I:e:o:");
+ int grc = getopt (argc, argv, "vp:I:e:o:tR:r:m:k");
if (grc < 0)
break;
switch (grc)
{
case 'v':
- verbose = true;
+ s.verbose = true;
break;
case 'p':
- last_pass = atoi (optarg);
- if (last_pass < 1 || last_pass > 3)
+ s.last_pass = atoi (optarg);
+ if (s.last_pass < 1 || s.last_pass > 5)
{
cerr << "Invalid pass number." << endl;
- usage ();
+ usage (s);
}
break;
case 'I':
- include_path.push_back (string (optarg));
+ s.include_path.push_back (string (optarg));
break;
case 'e':
if (have_script)
- usage ();
+ usage (s);
cmdline_script = string (optarg);
have_script = true;
break;
case 'o':
- if (output_file != "")
- usage ();
output_file = string (optarg);
break;
+ case 't':
+ s.test_mode = true;
+ break;
+
+ case 'R':
+ s.runtime_path = string (optarg);
+ break;
+
+ case 'm':
+ s.module_name = string (optarg);
+ break;
+
+ case 'r':
+ s.kernel_release = string (optarg);
+ break;
+
+ case 'k':
+ s.keep_tmpdir = true;
+ break;
+
case '?':
case 'h':
default:
- usage ();
+ usage (s);
}
}
@@ -113,21 +182,32 @@ main (int argc, char * const argv [])
have_script = true;
}
else
- args.push_back (string (argv[i]));
+ s.args.push_back (string (argv[i]));
}
// need a user file
if (! have_script)
- usage();
+ usage(s);
+
+ int rc = 0;
// arguments parsed; get down to business
- int rc = 0;
- systemtap_session s;
- if (output_file != "")
- s.op = new translator_output (output_file);
- else
- s.op = new translator_output (cout);
+ // Create a temporary directory to build within.
+ // Be careful with this, as "s.tmpdir" is "rm -rf"'d at the end.
+ {
+ char tmpdirt[] = "/tmp/stapXXXXXX";
+ const char* tmpdir = mkdtemp (tmpdirt);
+ if (! tmpdir)
+ {
+ const char* e = strerror (errno);
+ cerr << "mkdtemp (\"" << tmpdir << "\"): " << e << endl;
+ s.tmpdir = "";
+ rc = 1;
+ }
+ else
+ s.tmpdir = tmpdir;
+ }
// PASS 1a: PARSING USER SCRIPT
@@ -147,10 +227,10 @@ main (int argc, char * const argv [])
rc ++;
// PASS 1b: PARSING LIBRARY SCRIPTS
- for (unsigned i=0; i<include_path.size(); i++)
+ for (unsigned i=0; i<s.include_path.size(); i++)
{
glob_t globbuf;
- string dir = include_path[i] + "/*.stp";
+ string dir = s.include_path[i] + "/*.stp";
int r = glob(dir.c_str (), 0, NULL, & globbuf);
if (r == GLOB_NOSPACE || r == GLOB_ABORTED)
rc ++;
@@ -168,70 +248,119 @@ main (int argc, char * const argv [])
globfree (& globbuf);
}
- if (last_pass == 1 && rc == 0)
+ if (rc == 0 && s.last_pass == 1)
{
- s.op->line() << "# parse tree dump";
- s.user_file->print (s.op->newline());
+ cout << "# parse tree dump" << endl;
+ s.user_file->print (cout);
+ cout << endl;
for (unsigned i=0; i<s.library_files.size(); i++)
- s.library_files[i]->print (s.op->newline());
+ {
+ s.library_files[i]->print (cout);
+ cout << endl;
+ }
}
+
// PASS 2: ELABORATION
- if (rc == 0 && last_pass > 1)
+ if (rc == 0 && s.last_pass > 1)
rc = semantic_pass (s);
- if (last_pass == 2 && rc == 0)
+
+ if (rc == 0 && s.last_pass == 2)
{
if (s.globals.size() > 0)
- s.op->line() << "# globals";
+ cout << "# globals" << endl;
for (unsigned i=0; i<s.globals.size(); i++)
{
vardecl* v = s.globals[i];
- v->printsig (s.op->newline());
+ v->printsig (cout);
+ cout << endl;
}
if (s.functions.size() > 0)
- s.op->newline() << "# functions";
+ cout << "# functions" << endl;
for (unsigned i=0; i<s.functions.size(); i++)
{
functiondecl* f = s.functions[i];
- f->printsig (s.op->newline());
- s.op->indent(1);
+ f->printsig (cout);
+ cout << endl;
if (f->locals.size() > 0)
- s.op->newline(1) << "# locals";
+ cout << " # locals" << endl;
for (unsigned j=0; j<f->locals.size(); j++)
{
vardecl* v = f->locals[j];
- v->printsig (s.op->newline());
+ cout << " ";
+ v->printsig (cout);
+ cout << endl;
}
- s.op->indent(-1);
}
if (s.probes.size() > 0)
- s.op->newline() << "# probes";
+ cout << "# probes" << endl;
for (unsigned i=0; i<s.probes.size(); i++)
{
derived_probe* p = s.probes[i];
- p->printsig (s.op->newline());
- s.op->indent(1);
+ p->printsig (cout);
+ cout << endl;
if (p->locals.size() > 0)
- s.op->newline() << "# locals";
+ cout << " # locals" << endl;
for (unsigned j=0; j<p->locals.size(); j++)
{
vardecl* v = p->locals[j];
- v->printsig (s.op->newline());
+ cout << " ";
+ v->printsig (cout);
+ cout << endl;
}
- s.op->indent(-1);
}
-
- s.op->newline();
}
+
// PASS 3: TRANSLATION
- if (rc == 0 && last_pass > 2)
- rc = translate_pass (s);
+ if (rc == 0 && s.last_pass > 2)
+ {
+ s.translated_source = string(s.tmpdir) + "/" + s.module_name + ".c";
+ rc = translate_pass (s);
+ }
+
+ if (rc == 0 && s.last_pass == 3)
+ {
+ ifstream i (s.translated_source.c_str());
+ cout << i.rdbuf();
+ }
+
+ // PASS 4: COMPILATION
+ if (rc == 0 && s.last_pass > 3)
+ {
+ rc = compile_pass (s);
+ }
+
+ // PASS 5: RUN
+ if (rc == 0 && s.last_pass > 4)
+ {
+ rc = run_pass (s);
+ }
+
+ // Pull out saved output
+ if (output_file != "-")
+ s.op = new translator_output (output_file);
+ else
+ s.op = new translator_output (cout);
+
- s.op->line() << endl;
- delete s.op;
+ // Clean up temporary directory. Obviously, be careful with this.
+ if (s.tmpdir == "")
+ ; // do nothing
+ else
+ {
+ if (s.keep_tmpdir)
+ cerr << "Keeping temporary directory \"" << s.tmpdir << "\"" << endl;
+ else
+ {
+ string cleanupcmd = "/bin/rm -rf ";
+ cleanupcmd += s.tmpdir;
+ if (s.verbose) cerr << "Running " << cleanupcmd << endl;
+ (void) system (cleanupcmd.c_str());
+ }
+ }
return rc;
}
generated by cgit (git 2.34.1) at 2025-08-03 00:06:01 +0000