scripts/gcc-wrapper.py

   1 #! /usr/bin/env python
   2 # -*- coding: utf-8 -*-
   3
   4 # Copyright (c) 2011-2012, The Linux Foundation. All rights reserved.
   5 #
   6 # Redistribution and use in source and binary forms, with or without
   7 # modification, are permitted provided that the following conditions are met:
   8 #     * Redistributions of source code must retain the above copyright
   9 #       notice, this list of conditions and the following disclaimer.
  10 #     * Redistributions in binary form must reproduce the above copyright
  11 #       notice, this list of conditions and the following disclaimer in the
  12 #       documentation and/or other materials provided with the distribution.
  13 #     * Neither the name of The Linux Foundation nor
  14 #       the names of its contributors may be used to endorse or promote
  15 #       products derived from this software without specific prior written
  16 #       permission.
  17 #
  18 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  19 # AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  20 # IMPLIED WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  21 # NON-INFRINGEMENT ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR
  22 # CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  23 # EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  24 # PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
  25 # OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
  26 # WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
  27 # OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
  28 # ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  29
  30 # Invoke gcc, looking for warnings, and causing a failure if there are
  31 # non-whitelisted warnings.
  32
  33 import errno
  34 import re
  35 import os
  36 import sys
  37 import subprocess
  38
  39 # Note that gcc uses unicode, which may depend on the locale.  TODO:
  40 # force LANG to be set to en_US.UTF-8 to get consistent warnings.
  41
  42 allowed_warnings = set([
  43     "return_address.c:63",
  44     "hid-appleir.c:347",
  45     "hid-magicmouse.c:580",
  46     "hid-ntrig.c:1026",
  47     "core.c:1334",
  48     "menu.c:129",
  49     "bus.c:318", # FIXME
  50  ])
  51
  52 # Capture the name of the object file, can find it.
  53 ofile = None
  54
  55 warning_re = re.compile(r'''(.*/|)([^/]+\.[a-z]+:\d+):(\d+:)? warning:''')
  56 def interpret_warning(line):
  57     """Decode the message from gcc.  The messages we care about have a filename, and a warning"""
  58     line = line.rstrip('\n')
  59     m = warning_re.match(line)
  60     if m and m.group(2) not in allowed_warnings:
  61         print "error, forbidden warning:", m.group(2)
  62
  63         # If there is a warning, remove any object if it exists.
  64         if ofile:
  65             try:
  66                 os.remove(ofile)
  67             except OSError:
  68                 pass
  69         sys.exit(1)
  70
  71 def run_gcc():
  72     args = sys.argv[1:]
  73     # Look for -o
  74     try:
  75         i = args.index('-o')
  76         global ofile
  77         ofile = args[i+1]
  78     except (ValueError, IndexError):
  79         pass
  80
  81     compiler = sys.argv[0]
  82
  83     try:
  84         proc = subprocess.Popen(args, stderr=subprocess.PIPE)
  85         for line in proc.stderr:
  86             print line,
  87             interpret_warning(line)
  88
  89         result = proc.wait()
  90     except OSError as e:
  91         result = e.errno
  92         if result == errno.ENOENT:
  93             print args[0] + ':',e.strerror
  94             print 'Is your PATH set correctly?'
  95         else:
  96             print ' '.join(args), str(e)
  97
  98     return result
  99
 100 if __name__ == '__main__':
 101     status = run_gcc()
 102     sys.exit(status)