d0af9f27b4d650ea10f94e23c4001b31c6c6b473
[metze/samba/wip.git] / buildtools / wafsamba / samba_pidl.py
1 # waf build tool for building IDL files with pidl
2
3 from TaskGen import before
4 import Build, os
5 from samba_utils import *
6
7 def SAMBA_PIDL(bld, pname, source,
8                options='',
9                output_dir='.',
10                symlink=False,
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"):
64         if isinstance(bld.CONFIG_GET("CPP"), list):
65             cpp = "CPP=%s" % bld.CONFIG_GET("CPP")[0]
66         else:
67             cpp = "CPP=%s" % bld.CONFIG_GET("CPP")
68
69     if bld.CONFIG_SET("CC"):
70         if isinstance(bld.CONFIG_GET("CC"), list):
71             cc = "CC=%s" % bld.CONFIG_GET("CC")[0]
72         else:
73             cc = "CC=%s" % bld.CONFIG_GET("CC")
74
75     t = bld(rule='cd .. && %s %s ${PERL} "${PIDL}" --quiet ${OPTIONS} --outputdir ${OUTPUTDIR} -- "${SRC[0].abspath(env)}"' % (cpp, cc),
76             ext_out    = '.c',
77             before     = 'cc',
78             on_results = True,
79             shell      = True,
80             source     = source,
81             target     = out_files,
82             name       = name,
83             samba_type = 'PIDL')
84
85     # prime the list of nodes we are dependent on with the cached pidl sources
86     t.allnodes = pidl_src_nodes
87
88     t.env.PIDL = os.path.join(bld.srcnode.abspath(), 'pidl/pidl')
89     t.env.OPTIONS = TO_LIST(options)
90
91     # this rather convoluted set of path calculations is to cope with the possibility
92     # that gen_ndr is a symlink into the source tree. By doing this for the source3
93     # gen_ndr directory we end up generating identical output in gen_ndr for the old
94     # build system and the new one. That makes keeping things in sync much easier.
95     # eventually we should drop the gen_ndr files in git, but in the meanwhile this works
96
97     found_dir = bld.path.find_dir(output_dir)
98     if not 'abspath' in dir(found_dir):
99         Logs.error('Unable to find pidl output directory %s' %
100                    os.path.normpath(os.path.join(bld.curdir, output_dir)))
101         sys.exit(1)
102
103     outdir = bld.path.find_dir(output_dir).abspath(t.env)
104
105     if symlink and not os.path.lexists(outdir):
106         link_source = os.path.normpath(os.path.join(bld.curdir,output_dir))
107         os.symlink(link_source, outdir)
108
109     real_outputdir = os.path.realpath(outdir)
110     t.env.OUTPUTDIR = os_path_relpath(real_outputdir, os.path.dirname(bld.env.BUILD_DIRECTORY))
111
112     if generate_tables and table_header_idx is not None:
113         pidl_headers = LOCAL_CACHE(bld, 'PIDL_HEADERS')
114         pidl_headers[name] = [bld.path.find_or_declare(out_files[table_header_idx])]
115
116     t.more_includes = '#' + bld.path.relpath_gen(bld.srcnode)
117 Build.BuildContext.SAMBA_PIDL = SAMBA_PIDL
118
119
120 def SAMBA_PIDL_LIST(bld, name, source,
121                     options='',
122                     output_dir='.',
123                     symlink=False,
124                     generate_tables=True):
125     '''A wrapper for building a set of IDL files'''
126     for p in TO_LIST(source):
127         bld.SAMBA_PIDL(name, p, options=options, output_dir=output_dir, symlink=symlink, generate_tables=generate_tables)
128 Build.BuildContext.SAMBA_PIDL_LIST = SAMBA_PIDL_LIST
129
130
131 #################################################################
132 # the rule for generating the NDR tables
133 from TaskGen import feature, before
134 @feature('collect')
135 @before('exec_rule')
136 def collect(self):
137     pidl_headers = LOCAL_CACHE(self.bld, 'PIDL_HEADERS')
138     for (name, hd) in pidl_headers.items():
139         y = self.bld.name_to_obj(name, self.env)
140         self.bld.ASSERT(y is not None, 'Failed to find PIDL header %s' % name)
141         y.post()
142         for node in hd:
143             self.bld.ASSERT(node is not None, 'Got None as build node generating PIDL table for %s' % name)
144             self.source += " " + node.relpath_gen(self.path)
145
146
147 def SAMBA_PIDL_TABLES(bld, name, target):
148     '''generate the pidl NDR tables file'''
149     headers = bld.env.PIDL_HEADERS
150     bld.SET_BUILD_GROUP('main')
151     t = bld(
152             features = 'collect',
153             rule     = '${PERL} ${SRC} --output ${TGT} | sed "s|default/||" > ${TGT}',
154             ext_out  = '.c',
155             before   = 'cc',
156             on_results = True,
157             shell    = True,
158             source   = '../../librpc/tables.pl',
159             target   = target,
160             name     = name)
161     t.env.LIBRPC = os.path.join(bld.srcnode.abspath(), 'librpc')
162 Build.BuildContext.SAMBA_PIDL_TABLES = SAMBA_PIDL_TABLES
163