third_party/waf/waflib/extras/syms.py

   1 #! /usr/bin/env python
   2 # encoding: utf-8
   3
   4 """
   5 This tool supports the export_symbols_regex to export the symbols in a shared library.
   6 by default, all symbols are exported by gcc, and nothing by msvc.
   7 to use the tool, do something like:
   8
   9 def build(ctx):
  10         ctx(features='c cshlib syms', source='a.c b.c', export_symbols_regex='mylib_.*', target='testlib')
  11
  12 only the symbols starting with 'mylib_' will be exported.
  13 """
  14
  15 import re
  16 from waflib.Context import STDOUT
  17 from waflib.Task import Task
  18 from waflib.Errors import WafError
  19 from waflib.TaskGen import feature, after_method
  20
  21 class gen_sym(Task):
  22         def run(self):
  23                 obj = self.inputs[0]
  24                 kw = {}
  25
  26                 reg = getattr(self.generator, 'export_symbols_regex', '.+?')
  27                 if 'msvc' in (self.env.CC_NAME, self.env.CXX_NAME):
  28                         re_nm = re.compile(r'External\s+\|\s+_(?P<symbol>%s)\b' % reg)
  29                         cmd = (self.env.DUMPBIN or ['dumpbin']) + ['/symbols', obj.abspath()]
  30                 else:
  31                         if self.env.DEST_BINFMT == 'pe': #gcc uses nm, and has a preceding _ on windows
  32                                 re_nm = re.compile(r'(T|D)\s+_(?P<symbol>%s)\b' % reg)
  33                         elif self.env.DEST_BINFMT=='mac-o':
  34                                 re_nm=re.compile(r'(T|D)\s+(?P<symbol>_?%s)\b' % reg)
  35                         else:
  36                                 re_nm = re.compile(r'(T|D)\s+(?P<symbol>%s)\b' % reg)
  37                         cmd = (self.env.NM or ['nm']) + ['-g', obj.abspath()]
  38                 syms = [m.group('symbol') for m in re_nm.finditer(self.generator.bld.cmd_and_log(cmd, quiet=STDOUT, **kw))]
  39                 self.outputs[0].write('%r' % syms)
  40
  41 class compile_sym(Task):
  42         def run(self):
  43                 syms = {}
  44                 for x in self.inputs:
  45                         slist = eval(x.read())
  46                         for s in slist:
  47                                 syms[s] = 1
  48                 lsyms = list(syms.keys())
  49                 lsyms.sort()
  50                 if self.env.DEST_BINFMT == 'pe':
  51                         self.outputs[0].write('EXPORTS\n' + '\n'.join(lsyms))
  52                 elif self.env.DEST_BINFMT == 'elf':
  53                         self.outputs[0].write('{ global:\n' + ';\n'.join(lsyms) + ";\nlocal: *; };\n")
  54                 elif self.env.DEST_BINFMT=='mac-o':
  55                         self.outputs[0].write('\n'.join(lsyms) + '\n')
  56                 else:
  57                         raise WafError('NotImplemented')
  58
  59 @feature('syms')
  60 @after_method('process_source', 'process_use', 'apply_link', 'process_uselib_local', 'propagate_uselib_vars')
  61 def do_the_symbol_stuff(self):
  62         def_node = self.path.find_or_declare(getattr(self, 'sym_file', self.target + '.def'))
  63         compiled_tasks = getattr(self, 'compiled_tasks', None)
  64         if compiled_tasks:
  65                 ins = [x.outputs[0] for x in compiled_tasks]
  66                 self.gen_sym_tasks = [self.create_task('gen_sym', x, x.change_ext('.%d.sym' % self.idx)) for x in ins]
  67                 self.create_task('compile_sym', [x.outputs[0] for x in self.gen_sym_tasks], def_node)
  68
  69         link_task = getattr(self, 'link_task', None)
  70         if link_task:
  71                 self.link_task.dep_nodes.append(def_node)
  72
  73                 if 'msvc' in (self.env.CC_NAME, self.env.CXX_NAME):
  74                         self.link_task.env.append_value('LINKFLAGS', ['/def:' + def_node.bldpath()])
  75                 elif self.env.DEST_BINFMT == 'pe':
  76                         # gcc on windows takes *.def as an additional input
  77                         self.link_task.inputs.append(def_node)
  78                 elif self.env.DEST_BINFMT == 'elf':
  79                         self.link_task.env.append_value('LINKFLAGS', ['-Wl,-version-script', '-Wl,' + def_node.bldpath()])
  80                 elif self.env.DEST_BINFMT=='mac-o':
  81                         self.link_task.env.append_value('LINKFLAGS',['-Wl,-exported_symbols_list,' + def_node.bldpath()])
  82                 else:
  83                         raise WafError('NotImplemented')
  84