Merge branch 'master' of ssh://sources.redhat.com/git/systemtap

1 files changed, 30 insertions, 0 deletions

diff --git a/main.cxx b/main.cxx
index 794a5891..37c5b135 100644
--- a/main.cxx
+++ b/main.cxx
@@ -425,6 +425,36 @@ main (int argc, char * const argv [])
   if (s_tc != NULL)
     s.tapset_compile_coverage = true;
 
+  const char* s_kr = getenv ("SYSTEMTAP_RELEASE");
+  if (s_kr != NULL) {
+    if (s_kr[0] == '/') // fully specified path
+      {
+        s.kernel_build_tree = s_kr;
+        string version_file_name = s.kernel_build_tree + "/include/config/kernel.release";
+        // The file include/config/kernel.release within the
+        // build tree is used to pull out the version information
+        ifstream version_file (version_file_name.c_str());
+        if (version_file.fail ())
+          {
+            cerr << "Missing " << version_file_name << endl;
+            exit(1);
+          }
+        else
+          {
+            char c;
+            s.kernel_release = "";
+            while (version_file.get(c) && c != '\n')
+              s.kernel_release.push_back(c);
+          }
+      }
+    else
+      {
+        s.kernel_release = string (s_kr);
+        s.kernel_build_tree = "/lib/modules/" + s.kernel_release + "/build";
+      }
+  }
+
+
   while (true)
     {
       int long_opt;