Implement the EmbeddedTool class in valgrind_test.py to allow running tools that are embedded int...

tools/valgrind/valgrind_test.py

 #!/usr/bin/python
 # Copyright (c) 2011 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 # valgrind_test.py
 
 """Runs an exe through Valgrind and puts the intermediate files in a
 directory.
 """
@@ skipping 942 matching lines @@
       ret = cmd2.Main(args, check_sanity)
     else:
       logging.info("No reports, skipping RaceVerifier second pass")
     logging.info("Please see " + self.MORE_INFO_URL + " for more information " +
                  "on RaceVerifier")
     return ret
 
   def Run(self, args, module):
    return self.Main(args, False)
 
+class EmbeddedTool(BaseTool):
+  """Abstract class for tools embedded directly into the test binary."""
+  def __init__(self):
+    super(EmbeddedTool, self).__init__()
+    self._env = {}
+
+  def ToolCommand(self):
+    """Basically just the args of the script."""
+    return self._args
+
+  def NeedsPipes(self):
+    """True iff the tool needs to chain several subprocesses."""
+    raise NotImplementedError, "This method should be implemented " \
+                               "in the tool-specific subclass"
+
+  def Execute(self):
+    """Executes the app to be tested."""
+    for var in self._env:
+      self.PutEnvAndLog(var, self._env[var])
+    proc = self.ToolCommand()
+    logging.info('starting execution...')
+    if self.NeedsPipes():
+      return common.RunSubprocessChain(proc, self._timeout)
+    else:
+      return common.RunSubprocess(proc, self._timeout)
+
+  def PutEnvAndLog(self, env_name, env_value):
+    """Sets the env var |env_name| to |env_value| and writes to logging.info.
+    """
+    os.putenv(env_name, env_value)
+    logging.info('export %s=%s', env_name, env_value)
+
+
+class Asan(EmbeddedTool):
+  def __init__(self):
+    super(Asan, self).__init__()
+    self._timeout = 1200
+
+  def ToolName(self):
+    return "asan"
+
+  def NeedsPipes(self):
+    return True
+
+  def ToolCommand(self):
+    procs = [self._args]
+    procs.append([os.path.join(self._source_dir, "third_party", "asan",
+                              "scripts", "asan_symbolize.py")])
+    procs.append(["c++filt"])
+    return procs
+
+  def Analyze(sels, unused_check_sanity):
+    return 0
 
 class ToolFactory:
   def Create(self, tool_name):
     if tool_name == "memcheck":
       return Memcheck()
     if tool_name == "tsan":
       if common.IsWindows():
         return ThreadSanitizerWindows()
       else:
         return ThreadSanitizerPosix()
     if tool_name == "drmemory":
       return DrMemory()
     if tool_name == "tsan_rv":
       return RaceVerifier()
+    if tool_name == "asan":
+      return Asan()
     try:
       platform_name = common.PlatformNames()[0]
     except common.NotImplementedError:
       platform_name = sys.platform + "(Unknown)"
     raise RuntimeError, "Unknown tool (tool=%s, platform=%s)" % (tool_name,
                                                                  platform_name)
 
 def CreateTool(tool):
   return ToolFactory().Create(tool)
 
 if __name__ == '__main__':
   logging.error(sys.argv[0] + " can not be run from command line")
   sys.exit(1)