1 files changed, 121 insertions, 0 deletions
diff --git a/scripts/kprobes_test/run_module.py b/scripts/kprobes_test/run_module.py
new file mode 100755
index 00000000..74465d94
--- /dev/null
+++ b/scripts/kprobes_test/run_module.py
@@ -0,0 +1,121 @@
+#!/usr/bin/python
+
+# Copyright (C) 2008 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.
+
+import sys
+import os
+import os.path
+import time
+import select
+from config_opts import config_opts
+
+def run_module():
+    # Find the current size of /var/log/messages
+    logfile = '/var/log/messages'
+    print >>sys.stderr, "Getting size of %s..." % logfile
+
+    start_pos = os.path.getsize(logfile)
+    l = open(logfile, 'r')
+    l.seek(start_pos)
+
+    # Insert the module
+    print >>sys.stderr, "Inserting module..."
+    rc = os.system("/sbin/insmod %s" % os.path.join(os.getcwd(),
+                                                    'kprobe_module.ko'))
+    if os.WEXITSTATUS(rc) != 0:
+        # This is actually semi-OK, which is why there is no error
+        # message here, only a notice.  This might mean that every
+        # probe tried cannot be registered.
+        print >>sys.stderr, "Notice: insmod failed"
+    else:
+        # Run the load generate commands (if any).  Note we ignore the
+        # return values, since we don't really care if the commands
+        # succeed or not.
+        if config_opts.has_key('load_cmds'):
+            num_waits = 0
+
+            # Redirect the output to 'run_module.log' by modifying
+            # what stdout points to.
+            old_stdout = os.dup(sys.stdout.fileno())
+            fd = os.open('run_module.log', os.O_CREAT | os.O_WRONLY)
+            os.dup2(fd, sys.stdout.fileno())
+
+            # Run the commands.
+            for cmd in config_opts['load_cmds']:
+                pid = os.spawnvp(os.P_NOWAIT, cmd[0], cmd)
+                num_waits += 1
+            while num_waits > 0:
+                (pid, status) = os.waitpid(-1, 0)
+                num_waits -= 1
+
+            # Restore old value of stdout.
+            os.close(fd)
+            os.dup2(old_stdout, sys.stdout.fileno())
+
+        # Remove the module
+        print >>sys.stderr, "Removing module..."
+        rc = os.system("/sbin/rmmod kprobe_module")
+        if os.WEXITSTATUS(rc) != 0:
+            print >>sys.stderr, "Error: rmmod failed"
+            return -1
+
+    # Now we have to wait until everything is flushed to the logfile
+    f = open(config_opts['probes_result'], 'w')
+    while 1:
+        # Find the ending size of /var/log/messages
+        end_pos = os.path.getsize(logfile)
+
+        if end_pos < start_pos:
+            # The log files have been rotated.  Read any leftover data,
+            # then reopen file.
+            data = l.read()
+            if data:
+                f.write(data)
+
+                # See if we can find 'kprobe_module unloaded' in the
+                # data we just read.
+                if data.find('kprobe_module unloaded') != -1:
+                    break
+        
+            l.close()
+            l = open(logfile, 'r')
+            start_pos = 0
+            continue
+
+        # Try to wait until data is available
+        while 1:
+            try:
+                input, output, exc = select.select([l.fileno()], [], [], 60)
+                break
+            except select.error, err:
+                if err[0] != EINTR:
+                    raise
+
+        # Get the new stuff logged to /var/log/messages
+        data = l.read(end_pos - start_pos + 1)
+        if not data:
+            # ignore EOF
+            time.sleep(2)
+            continue
+
+        # Write results data
+        f.write(data)
+
+        # See if we can find 'kprobe_module unloaded' in the data we
+        # just read.
+        if data.find('kprobe_module unloaded') == -1:
+            start_pos = end_pos
+        else:
+            break
+
+    l.close()
+    f.close()
+    return 0
+
+rc = run_module()
+sys.exit(rc)