1# Copyright (C) 2009 Kevin Ollivier All rights reserved. 2# 3# Redistribution and use in source and binary forms, with or without 4# modification, are permitted provided that the following conditions 5# are met: 6# 1. Redistributions of source code must retain the above copyright 7# notice, this list of conditions and the following disclaimer. 8# 2. Redistributions in binary form must reproduce the above copyright 9# notice, this list of conditions and the following disclaimer in the 10# documentation and/or other materials provided with the distribution. 11# 12# THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY 13# EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 14# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 15# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE COMPUTER, INC. OR 16# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 17# EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 18# PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 19# PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY 20# OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 21# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 22# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 23# 24# This module is for code where we override waf's default behavior or extend waf 25 26import os 27import subprocess 28import sys 29 30import Utils 31 32# version of exec_command that handles Windows command lines longer than 32000 chars 33def exec_command(s, **kw): 34 filename = '' 35 if sys.platform.startswith('win') and len(' '.join(s)) > 32000: 36 import tempfile 37 (fd, filename) = tempfile.mkstemp() 38 t = [] 39 for i in s: 40 if i.find(" ") != -1: 41 i = '"%s"' % i 42 t.append(i) 43 44 t1 = t[1:] 45 # Fix for LNK1170 error 46 filename_str = '@' + filename 47 filename2_str = None 48 if len(' '.join(s)) > 131070: 49 t2 = t[len(t) / 2:] 50 t1 = t[1:len(t) / 2] 51 (fd2, filename2) = tempfile.mkstemp() 52 os.write(fd2, ' '.join(t2)) 53 os.close(fd2) 54 filename2_str = '@' + filename2 55 os.write(fd, ' '.join(t1)) 56 os.close(fd) 57 58 s = [s[0]] 59 s.append(filename_str) 60 if filename2_str: 61 s.append(filename2_str) 62 63 if 'log' in kw: 64 kw['stdout'] = kw['stderr'] = kw['log'] 65 del(kw['log']) 66 kw['shell'] = isinstance(s, str) 67 68 def cleanup(): 69 try: 70 if os.path.exists(filename): 71 os.remove(filename) 72 except: 73 pass 74 75 try: 76 proc = subprocess.Popen(s, **kw) 77 result = proc.wait() 78 cleanup() 79 return result 80 81 except OSError: 82 cleanup() 83 raise 84 85Utils.exec_command = exec_command 86 87# Better performing h_file to keep hashing from consuming lots of time 88import stat 89def h_file(filename): 90 st = os.stat(filename) 91 if stat.S_ISDIR(st[stat.ST_MODE]): raise IOError('not a file') 92 m = Utils.md5() 93 m.update(str(st.st_mtime)) 94 m.update(str(st.st_size)) 95 m.update(filename) 96 return m.digest() 97 98Utils.h_file = h_file 99