scripts/gcc-wrapper.py

   1 #! /usr/bin/env python2
   2 # -*- coding: utf-8 -*-
   3
   4 # Copyright (c) 2011-2016, 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     "core.c:144",
  44     "inet_connection_sock.c:430",
  45     "inet_connection_sock.c:467",
  46     "inet6_connection_sock.c:89",
  47  ])
  48
  49 # Capture the name of the object file, can find it.
  50 ofile = None
  51
  52 warning_re = re.compile(r'''(.*/|)([^/]+\.[a-z]+:\d+):(\d+:)? warning:''')
  53 def interpret_warning(line):
  54     """Decode the message from gcc.  The messages we care about have a filename, and a warning"""
  55     line = line.rstrip('\n')
  56     m = warning_re.match(line)
  57     if m and m.group(2) not in allowed_warnings:
  58         print "error, forbidden warning:", m.group(2)
  59
  60         # If there is a warning, remove any object if it exists.
  61         if ofile:
  62             try:
  63                 os.remove(ofile)
  64             except OSError:
  65                 pass
  66         sys.exit(1)
  67
  68 def run_gcc():
  69     args = sys.argv[1:]
  70     # Look for -o
  71     try:
  72         i = args.index('-o')
  73         global ofile
  74         ofile = args[i+1]
  75     except (ValueError, IndexError):
  76         pass
  77
  78     compiler = sys.argv[0]
  79
  80     try:
  81         proc = subprocess.Popen(args, stderr=subprocess.PIPE)
  82         for line in proc.stderr:
  83             print line,
  84             interpret_warning(line)
  85
  86         result = proc.wait()
  87     except OSError as e:
  88         result = e.errno
  89         if result == errno.ENOENT:
  90             print args[0] + ':',e.strerror
  91             print 'Is your PATH set correctly?'
  92         else:
  93             print ' '.join(args), str(e)
  94
  95     return result
  96
  97 if __name__ == '__main__':
  98     status = run_gcc()
  99     sys.exit(status)