buildtools/wafsamba/samba_pidl.py

   1 # waf build tool for building IDL files with pidl
   2
   3 import os
   4 import Build
   5 from TaskGen import feature, before
   6 from samba_utils import SET_TARGET_TYPE, TO_LIST, LOCAL_CACHE
   7
   8 def SAMBA_PIDL(bld, pname, source,
   9                options='',
  10                output_dir='.',
  11                generate_tables=True):
  12     '''Build a IDL file using pidl.
  13        This will produce up to 13 output files depending on the options used'''
  14
  15     bname = source[0:-4]; # strip off the .idl suffix
  16     bname = os.path.basename(bname)
  17     name = "%s_%s" % (pname, bname.upper())
  18
  19     if not SET_TARGET_TYPE(bld, name, 'PIDL'):
  20         return
  21
  22     bld.SET_BUILD_GROUP('build_source')
  23
  24     # the output files depend on the options used. Use this dictionary
  25     # to map between the options and the resulting file names
  26     options_map = { '--header'            : '%s.h',
  27                     '--ndr-parser'        : 'ndr_%s.c ndr_%s.h',
  28                     '--samba3-ndr-server' : 'srv_%s.c srv_%s.h',
  29                     '--samba3-ndr-client' : 'cli_%s.c cli_%s.h',
  30                     '--server'            : 'ndr_%s_s.c',
  31                     '--client'            : 'ndr_%s_c.c ndr_%s_c.h',
  32                     '--python'            : 'py_%s.c',
  33                     '--tdr-parser'        : 'tdr_%s.c tdr_%s.h',
  34                     '--dcom-proxy'        : '%s_p.c',
  35                     '--com-header'        : 'com_%s.h'
  36                     }
  37
  38     table_header_idx = None
  39     out_files = []
  40     options_list = TO_LIST(options)
  41
  42     for o in options_list:
  43         if o in options_map:
  44             ofiles = TO_LIST(options_map[o])
  45             for f in ofiles:
  46                 out_files.append(os.path.join(output_dir, f % bname))
  47                 if f == 'ndr_%s.h':
  48                     # remember this one for the tables generation
  49                     table_header_idx = len(out_files) - 1
  50
  51     # depend on the full pidl sources
  52     source = TO_LIST(source)
  53     try:
  54         pidl_src_nodes = bld.pidl_files_cache
  55     except AttributeError:
  56         bld.pidl_files_cache = bld.srcnode.ant_glob('pidl/lib/Parse/**/*.pm', flat=False)
  57         bld.pidl_files_cache.extend(bld.srcnode.ant_glob('pidl', flat=False))
  58         pidl_src_nodes = bld.pidl_files_cache
  59
  60     # the cd .. is needed because pidl currently is sensitive to the directory it is run in
  61     cpp = ""
  62     cc = ""
  63     if bld.CONFIG_SET("CPP") and bld.CONFIG_GET("CPP") != "":
  64         if isinstance(bld.CONFIG_GET("CPP"), list):
  65             cpp = 'CPP="%s"' % " ".join(bld.CONFIG_GET("CPP"))
  66         else:
  67             cpp = 'CPP="%s"' % bld.CONFIG_GET("CPP")
  68
  69     if cpp == "CPP=xlc_r":
  70         cpp = ""
  71
  72
  73     if bld.CONFIG_SET("CC"):
  74         if isinstance(bld.CONFIG_GET("CC"), list):
  75             cc = 'CC="%s"' % " ".join(bld.CONFIG_GET("CC"))
  76         else:
  77             cc = 'CC="%s"' % bld.CONFIG_GET("CC")
  78
  79     t = bld(rule='cd .. && %s %s ${PERL} "${PIDL}" --quiet ${OPTIONS} --outputdir ${OUTPUTDIR} -- "${SRC[0].abspath(env)}"' % (cpp, cc),
  80             ext_out    = '.c',
  81             before     = 'cc',
  82             update_outputs = True,
  83             shell      = True,
  84             source     = source,
  85             target     = out_files,
  86             name       = name,
  87             samba_type = 'PIDL')
  88
  89     # prime the list of nodes we are dependent on with the cached pidl sources
  90     t.allnodes = pidl_src_nodes
  91
  92     t.env.PIDL = os.path.join(bld.srcnode.abspath(), 'pidl/pidl')
  93     t.env.OPTIONS = TO_LIST(options)
  94     t.env.OUTPUTDIR = bld.bldnode.name + '/' + bld.path.find_dir(output_dir).bldpath(t.env)
  95
  96     if generate_tables and table_header_idx is not None:
  97         pidl_headers = LOCAL_CACHE(bld, 'PIDL_HEADERS')
  98         pidl_headers[name] = [bld.path.find_or_declare(out_files[table_header_idx])]
  99
 100     t.more_includes = '#' + bld.path.relpath_gen(bld.srcnode)
 101 Build.BuildContext.SAMBA_PIDL = SAMBA_PIDL
 102
 103
 104 def SAMBA_PIDL_LIST(bld, name, source,
 105                     options='',
 106                     output_dir='.',
 107                     generate_tables=True):
 108     '''A wrapper for building a set of IDL files'''
 109     for p in TO_LIST(source):
 110         bld.SAMBA_PIDL(name, p, options=options, output_dir=output_dir, generate_tables=generate_tables)
 111 Build.BuildContext.SAMBA_PIDL_LIST = SAMBA_PIDL_LIST
 112
 113
 114 #################################################################
 115 # the rule for generating the NDR tables
 116 @feature('collect')
 117 @before('exec_rule')
 118 def collect(self):
 119     pidl_headers = LOCAL_CACHE(self.bld, 'PIDL_HEADERS')
 120     for (name, hd) in pidl_headers.items():
 121         y = self.bld.get_tgen_by_name(name)
 122         self.bld.ASSERT(y is not None, 'Failed to find PIDL header %s' % name)
 123         y.post()
 124         for node in hd:
 125             self.bld.ASSERT(node is not None, 'Got None as build node generating PIDL table for %s' % name)
 126             self.source += " " + node.relpath_gen(self.path)
 127
 128
 129 def SAMBA_PIDL_TABLES(bld, name, target):
 130     '''generate the pidl NDR tables file'''
 131     bld.SET_BUILD_GROUP('main')
 132     t = bld(
 133             features = 'collect',
 134             rule     = '${PERL} ${SRC} --output ${TGT} | sed "s|default/||" > ${TGT}',
 135             ext_out  = '.c',
 136             before   = 'cc',
 137             update_outputs = True,
 138             shell    = True,
 139             source   = '../../librpc/tables.pl',
 140             target   = target,
 141             name     = name)
 142     t.env.LIBRPC = os.path.join(bld.srcnode.abspath(), 'librpc')
 143 Build.BuildContext.SAMBA_PIDL_TABLES = SAMBA_PIDL_TABLES
 144