Fix compile on Windows swarming bot

infra/bots/common.py

 #!/usr/bin/env python
 #
 # Copyright 2016 Google Inc.
 #
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 
 import contextlib
 import math
 import os
+import psutil
 import shutil
 import socket
 import subprocess
 import sys
 import time
 import urllib2
 
 from flavor import android_flavor
 from flavor import chromeos_flavor
 from flavor import cmake_flavor
@@ skipping 128 matching lines @@
 
     Assumes that CWD is the directory containing this file.
     """
     self.name = bot_name
     self.skia_dir = os.path.abspath(os.path.join(
         os.path.dirname(os.path.realpath(__file__)),
         os.pardir, os.pardir))
     self.swarm_out_dir = swarm_out_dir
     os.chdir(self.skia_dir)
     self.build_dir = os.path.abspath(os.path.join(self.skia_dir, os.pardir))
+    self.infrabots_dir = os.path.join(self.skia_dir, 'infra', 'bots')
+    self.home_dir = os.path.expanduser('~')
+
     self.spec = self.get_bot_spec(bot_name)
     self.bot_cfg = self.spec['builder_cfg']
     self.out_dir = os.path.join(os.pardir, 'out')
     self.configuration = self.spec['configuration']
     self.default_env = {
+      'CHROME_HEADLESS': '1',
       'SKIA_OUT': self.out_dir,
       'BUILDTYPE': self.configuration,
       'PATH': os.environ['PATH'],
     }
+    if 'Win' in self.bot_cfg['os']:
+      self.default_env['SystemRoot'] = 'C:\\Windows'
+      self.default_env['TEMP'] = os.path.join(
+          self.home_dir, 'AppData', 'Local', 'Temp')
+      self.default_env['TMP'] = self.default_env['TEMP']
     self.default_env.update(self.spec['env'])
     self.build_targets = [str(t) for t in self.spec['build_targets']]
     self.is_trybot = self.bot_cfg['is_trybot']
     self.upload_dm_results = self.spec['upload_dm_results']
     self.upload_perf_results = self.spec['upload_perf_results']
     self.perf_data_dir = os.path.join(self.swarm_out_dir, 'perfdata',
                                       self.name, 'data')
     self.resource_dir = os.path.join(self.skia_dir, 'resources')
     self.images_dir = os.path.join(self.build_dir, 'images')
     self.local_skp_dir = os.path.join(self.build_dir, 'playback', 'skps')
@@ skipping 65 matching lines @@
   def compile_steps(self):
     for t in self.build_targets:
       self.flavor.compile(t)
     dst = os.path.join(self.swarm_out_dir, 'out', self.configuration)
     os.makedirs(dst)
     for f in BUILD_PRODUCTS_WHITELIST:
       path = os.path.join(self.out_dir, self.configuration, f)
       if os.path.exists(path):
         print 'Copying build product %s' % path
         shutil.copy(path, dst)
+    self.cleanup()
 
   def _run_once(self, fn, *args, **kwargs):
     if not fn.__name__ in self._already_ran:
       self._already_ran[fn.__name__] = True
       fn(*args, **kwargs)
 
   def install(self):
     """Copy the required executables and files to the device."""
     self.device_dirs = self.flavor.get_device_dirs()
 
@@ skipping 112 matching lines @@
     # See skia:2789.
     if ('Valgrind' in self.name and
         self.bot_cfg.get('cpu_or_gpu') == 'GPU'):
       abandonGpuContext = list(args)
       abandonGpuContext.append('--abandonGpuContext')
       self.flavor.run(abandonGpuContext)
       preAbandonGpuContext = list(args)
       preAbandonGpuContext.append('--preAbandonGpuContext')
       self.flavor.run(preAbandonGpuContext)
 
-    self.flavor.cleanup_steps()
+    self.cleanup()
 
   def perf_steps(self, got_revision, master_name, slave_name, build_number,
                  issue=None, patchset=None):
     """Run Skia benchmarks."""
     self.build_number = build_number
     self.got_revision = got_revision
     self.master_name = master_name
     self.slave_name = slave_name
     self._run_once(self.install)
     if self.upload_perf_results:
@@ skipping 58 matching lines @@
       abandonGpuContext.extend(['--abandonGpuContext', '--nocpu'])
       self.flavor.run(abandonGpuContext, env=self.default_env)
 
     # Copy results to host.
     if self.upload_perf_results:
       if not os.path.exists(self.perf_data_dir):
         os.makedirs(self.perf_data_dir)
       self.flavor.copy_directory_contents_to_host(
           self.device_dirs.perf_data_dir, self.perf_data_dir)
 
+    self.cleanup()
+
+  def cleanup(self):
+    if sys.platform == 'win32':
+      # Kill mspdbsrv.exe, which tends to hang around after the build finishes.
+      for p in psutil.process_iter():
+        try:
+          if p.name == 'mspdbsrv.exe':
+            p.kill()
+        except psutil._error.AccessDenied:
+          pass
     self.flavor.cleanup_steps()