• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1#!/usr/bin/env python
2#===- lib/dfsan/scripts/build-libc-list.py ---------------------------------===#
3#
4#                     The LLVM Compiler Infrastructure
5#
6# This file is distributed under the University of Illinois Open Source
7# License. See LICENSE.TXT for details.
8#
9#===------------------------------------------------------------------------===#
10# The purpose of this script is to identify every function symbol in a set of
11# libraries (in this case, libc and libgcc) so that they can be marked as
12# uninstrumented, thus allowing the instrumentation pass to treat calls to those
13# functions correctly.
14
15import os
16import subprocess
17import sys
18from optparse import OptionParser
19
20def defined_function_list(object):
21  functions = []
22  readelf_proc = subprocess.Popen(['readelf', '-s', '-W', object],
23                                  stdout=subprocess.PIPE)
24  readelf = readelf_proc.communicate()[0].split('\n')
25  if readelf_proc.returncode != 0:
26    raise subprocess.CalledProcessError(readelf_proc.returncode, 'readelf')
27  for line in readelf:
28    if (line[31:35] == 'FUNC' or line[31:36] == 'IFUNC') and \
29       line[39:44] != 'LOCAL' and \
30       line[55:58] != 'UND':
31      function_name = line[59:].split('@')[0]
32      functions.append(function_name)
33  return functions
34
35p = OptionParser()
36
37p.add_option('--libc-dso-path', metavar='PATH',
38             help='path to libc DSO directory',
39             default='/lib/x86_64-linux-gnu')
40p.add_option('--libc-archive-path', metavar='PATH',
41             help='path to libc archive directory',
42             default='/usr/lib/x86_64-linux-gnu')
43
44p.add_option('--libgcc-dso-path', metavar='PATH',
45             help='path to libgcc DSO directory',
46             default='/lib/x86_64-linux-gnu')
47p.add_option('--libgcc-archive-path', metavar='PATH',
48             help='path to libgcc archive directory',
49             default='/usr/lib/gcc/x86_64-linux-gnu/4.6')
50
51p.add_option('--with-libstdcxx', action='store_true',
52             dest='with_libstdcxx',
53             help='include libstdc++ in the list (inadvisable)')
54p.add_option('--libstdcxx-dso-path', metavar='PATH',
55             help='path to libstdc++ DSO directory',
56             default='/usr/lib/x86_64-linux-gnu')
57
58(options, args) = p.parse_args()
59
60libs = [os.path.join(options.libc_dso_path, name) for name in
61        ['ld-linux-x86-64.so.2',
62         'libanl.so.1',
63         'libBrokenLocale.so.1',
64         'libcidn.so.1',
65         'libcrypt.so.1',
66         'libc.so.6',
67         'libdl.so.2',
68         'libm.so.6',
69         'libnsl.so.1',
70         'libpthread.so.0',
71         'libresolv.so.2',
72         'librt.so.1',
73         'libthread_db.so.1',
74         'libutil.so.1']]
75libs += [os.path.join(options.libc_archive_path, name) for name in
76         ['libc_nonshared.a',
77          'libpthread_nonshared.a']]
78
79libs.append(os.path.join(options.libgcc_dso_path, 'libgcc_s.so.1'))
80libs.append(os.path.join(options.libgcc_archive_path, 'libgcc.a'))
81
82if options.with_libstdcxx:
83  libs.append(os.path.join(options.libstdcxx_dso_path, 'libstdc++.so.6'))
84
85functions = []
86for l in libs:
87  if os.path.exists(l):
88    functions += defined_function_list(l)
89  else:
90    print >> sys.stderr, 'warning: library %s not found' % l
91
92functions = list(set(functions))
93functions.sort()
94
95for f in functions:
96  print 'fun:%s=uninstrumented' % f
97