1#!/usr/bin/env python 2 3"""CaptureCmd - A generic tool for capturing information about the 4invocations of another program. 5 6Usage 7-- 81. Move the original tool to a safe known location. 9 102. Link CaptureCmd to the original tool's location. 11 123. Define CAPTURE_CMD_PROGRAM to the known location of the original 13tool; this must be an absolute path. 14 154. Define CAPTURE_CMD_DIR to a directory to write invocation 16information to. 17""" 18 19import hashlib 20import os 21import sys 22import time 23 24def saveCaptureData(prefix, dir, object): 25 string = repr(object) + '\n' 26 key = hashlib.sha1(string).hexdigest() 27 path = os.path.join(dir, 28 prefix + key) 29 if not os.path.exists(path): 30 f = open(path, 'wb') 31 f.write(string) 32 f.close() 33 return prefix + key 34 35def main(): 36 program = os.getenv('CAPTURE_CMD_PROGRAM') 37 dir = os.getenv('CAPTURE_CMD_DIR') 38 fallback = os.getenv('CAPTURE_CMD_FALLBACK') 39 if not program: 40 raise ValueError('CAPTURE_CMD_PROGRAM is not defined!') 41 if not dir: 42 raise ValueError('CAPTURE_CMD_DIR is not defined!') 43 44 # Make the output directory if it doesn't already exist. 45 if not os.path.exists(dir): 46 os.mkdir(dir, 0700) 47 48 # Get keys for various data. 49 env = os.environ.items() 50 env.sort() 51 envKey = saveCaptureData('env-', dir, env) 52 cwdKey = saveCaptureData('cwd-', dir, os.getcwd()) 53 argvKey = saveCaptureData('argv-', dir, sys.argv) 54 entry = (time.time(), envKey, cwdKey, argvKey) 55 saveCaptureData('cmd-', dir, entry) 56 57 if fallback: 58 pid = os.fork() 59 if not pid: 60 os.execv(program, sys.argv) 61 os._exit(1) 62 else: 63 res = os.waitpid(pid, 0) 64 if not res: 65 os.execv(fallback, sys.argv) 66 os._exit(1) 67 os._exit(res) 68 else: 69 os.execv(program, sys.argv) 70 os._exit(1) 71 72if __name__ == '__main__': 73 main() 74