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