9a38900d5fc6df67b618326adfb18b3ba5a37541
[samba.git] / buildtools / wafsamba / wafsamba.py
1 # a waf tool to add autoconf-like macros to the configure section
2 # and for SAMBA_ macros for building libraries, binaries etc
3
4 import Build, os, sys, Options, Task, Utils, cc, TaskGen, fnmatch, re, shutil, Logs, Constants
5 from Configure import conf
6 from Logs import debug
7 from samba_utils import SUBST_VARS_RECURSIVE
8 TaskGen.task_gen.apply_verif = Utils.nada
9
10 # bring in the other samba modules
11 from samba_optimisation import *
12 from samba_utils import *
13 from samba_version import *
14 from samba_autoconf import *
15 from samba_patterns import *
16 from samba_pidl import *
17 from samba_autoproto import *
18 from samba_python import *
19 from samba_perl import *
20 from samba_deps import *
21 from samba_bundled import *
22 from samba_third_party import *
23 import samba_install
24 import samba_conftests
25 import samba_abi
26 import samba_headers
27 import tru64cc
28 import irixcc
29 import hpuxcc
30 import generic_cc
31 import samba_dist
32 import samba_wildcard
33 import stale_files
34 import symbols
35 import pkgconfig
36 import configure_file
37
38 # some systems have broken threading in python
39 if os.environ.get('WAF_NOTHREADS') == '1':
40     import nothreads
41
42 LIB_PATH="shared"
43
44 os.environ['PYTHONUNBUFFERED'] = '1'
45
46
47 if Constants.HEXVERSION < 0x105019:
48     Logs.error('''
49 Please use the version of waf that comes with Samba, not
50 a system installed version. See http://wiki.samba.org/index.php/Waf
51 for details.
52
53 Alternatively, please run ./configure and make as usual. That will
54 call the right version of waf.''')
55     sys.exit(1)
56
57
58 @conf
59 def SAMBA_BUILD_ENV(conf):
60     '''create the samba build environment'''
61     conf.env.BUILD_DIRECTORY = conf.blddir
62     mkdir_p(os.path.join(conf.blddir, LIB_PATH))
63     mkdir_p(os.path.join(conf.blddir, LIB_PATH, "private"))
64     mkdir_p(os.path.join(conf.blddir, "modules"))
65     mkdir_p(os.path.join(conf.blddir, 'python/samba/dcerpc'))
66     # this allows all of the bin/shared and bin/python targets
67     # to be expressed in terms of build directory paths
68     mkdir_p(os.path.join(conf.blddir, 'default'))
69     for (source, target) in [('shared', 'shared'), ('modules', 'modules'), ('python', 'python_modules')]:
70         link_target = os.path.join(conf.blddir, 'default/' + target)
71         if not os.path.lexists(link_target):
72             os.symlink('../' + source, link_target)
73
74     # get perl to put the blib files in the build directory
75     blib_bld = os.path.join(conf.blddir, 'default/pidl/blib')
76     blib_src = os.path.join(conf.srcdir, 'pidl/blib')
77     mkdir_p(blib_bld + '/man1')
78     mkdir_p(blib_bld + '/man3')
79     if os.path.islink(blib_src):
80         os.unlink(blib_src)
81     elif os.path.exists(blib_src):
82         shutil.rmtree(blib_src)
83
84
85 def ADD_INIT_FUNCTION(bld, subsystem, target, init_function):
86     '''add an init_function to the list for a subsystem'''
87     if init_function is None:
88         return
89     bld.ASSERT(subsystem is not None, "You must specify a subsystem for init_function '%s'" % init_function)
90     cache = LOCAL_CACHE(bld, 'INIT_FUNCTIONS')
91     if not subsystem in cache:
92         cache[subsystem] = []
93     cache[subsystem].append( { 'TARGET':target, 'INIT_FUNCTION':init_function } )
94 Build.BuildContext.ADD_INIT_FUNCTION = ADD_INIT_FUNCTION
95
96
97 def generate_empty_file(task):
98     target_fname = installed_location=task.outputs[0].bldpath(task.env)
99     target_file = open(installed_location, 'w')
100     target_file.close()
101     return 0
102
103 #################################################################
104 def SAMBA_LIBRARY(bld, libname, source,
105                   deps='',
106                   public_deps='',
107                   includes='',
108                   public_headers=None,
109                   public_headers_install=True,
110                   header_path=None,
111                   pc_files=None,
112                   vnum=None,
113                   soname=None,
114                   cflags='',
115                   ldflags='',
116                   external_library=False,
117                   realname=None,
118                   keep_underscore=False,
119                   autoproto=None,
120                   autoproto_extra_source='',
121                   group='main',
122                   depends_on='',
123                   local_include=True,
124                   global_include=True,
125                   vars=None,
126                   subdir=None,
127                   install_path=None,
128                   install=True,
129                   pyembed=False,
130                   pyext=False,
131                   target_type='LIBRARY',
132                   bundled_extension=False,
133                   bundled_name=None,
134                   link_name=None,
135                   abi_directory=None,
136                   abi_match=None,
137                   hide_symbols=False,
138                   manpages=None,
139                   private_library=False,
140                   grouping_library=False,
141                   allow_undefined_symbols=False,
142                   allow_warnings=False,
143                   enabled=True):
144     '''define a Samba library'''
145
146     if LIB_MUST_BE_PRIVATE(bld, libname):
147         private_library=True
148
149     if not enabled:
150         SET_TARGET_TYPE(bld, libname, 'DISABLED')
151         return
152
153     source = bld.EXPAND_VARIABLES(source, vars=vars)
154     if subdir:
155         source = bld.SUBDIR(subdir, source)
156
157     # remember empty libraries, so we can strip the dependencies
158     if ((source == '') or (source == [])):
159         if deps == '' and public_deps == '':
160             SET_TARGET_TYPE(bld, libname, 'EMPTY')
161             return
162         empty_c = libname + '.empty.c'
163         bld.SAMBA_GENERATOR('%s_empty_c' % libname,
164                             rule=generate_empty_file,
165                             target=empty_c)
166         source=empty_c
167
168     if BUILTIN_LIBRARY(bld, libname):
169         obj_target = libname
170     else:
171         obj_target = libname + '.objlist'
172
173     if group == 'libraries':
174         subsystem_group = 'main'
175     else:
176         subsystem_group = group
177
178     # first create a target for building the object files for this library
179     # by separating in this way, we avoid recompiling the C files
180     # separately for the install library and the build library
181     bld.SAMBA_SUBSYSTEM(obj_target,
182                         source         = source,
183                         deps           = deps,
184                         public_deps    = public_deps,
185                         includes       = includes,
186                         public_headers = public_headers,
187                         public_headers_install = public_headers_install,
188                         header_path    = header_path,
189                         cflags         = cflags,
190                         group          = subsystem_group,
191                         autoproto      = autoproto,
192                         autoproto_extra_source=autoproto_extra_source,
193                         depends_on     = depends_on,
194                         hide_symbols   = hide_symbols,
195                         allow_warnings = allow_warnings,
196                         pyembed        = pyembed,
197                         pyext          = pyext,
198                         local_include  = local_include,
199                         global_include = global_include)
200
201     if BUILTIN_LIBRARY(bld, libname):
202         return
203
204     if not SET_TARGET_TYPE(bld, libname, target_type):
205         return
206
207     # the library itself will depend on that object target
208     deps += ' ' + public_deps
209     deps = TO_LIST(deps)
210     deps.append(obj_target)
211
212     realname = bld.map_shlib_extension(realname, python=(target_type=='PYTHON'))
213     link_name = bld.map_shlib_extension(link_name, python=(target_type=='PYTHON'))
214
215     # we don't want any public libraries without version numbers
216     if (not private_library and target_type != 'PYTHON' and not realname):
217         if vnum is None and soname is None:
218             raise Utils.WafError("public library '%s' must have a vnum" %
219                     libname)
220         if pc_files is None:
221             raise Utils.WafError("public library '%s' must have pkg-config file" %
222                        libname)
223         if public_headers is None:
224             raise Utils.WafError("public library '%s' must have header files" %
225                        libname)
226
227     if bundled_name is not None:
228         pass
229     elif target_type == 'PYTHON' or realname or not private_library:
230         if keep_underscore:
231             bundled_name = libname
232         else:
233             bundled_name = libname.replace('_', '-')
234     else:
235         assert (private_library == True and realname is None)
236         if abi_directory or vnum or soname:
237             bundled_extension=True
238         bundled_name = PRIVATE_NAME(bld, libname.replace('_', '-'),
239                                     bundled_extension, private_library)
240
241     ldflags = TO_LIST(ldflags)
242
243     features = 'cc cshlib symlink_lib install_lib'
244     if pyext:
245         features += ' pyext'
246     if pyembed:
247         features += ' pyembed'
248
249     if abi_directory:
250         features += ' abi_check'
251
252     vscript = None
253     if bld.env.HAVE_LD_VERSION_SCRIPT:
254         if private_library:
255             version = "%s_%s" % (Utils.g_module.APPNAME, Utils.g_module.VERSION)
256         elif vnum:
257             version = "%s_%s" % (libname, vnum)
258         else:
259             version = None
260         if version:
261             vscript = "%s.vscript" % libname
262             bld.ABI_VSCRIPT(libname, abi_directory, version, vscript,
263                             abi_match)
264             fullname = apply_pattern(bundled_name, bld.env.shlib_PATTERN)
265             fullpath = bld.path.find_or_declare(fullname)
266             vscriptpath = bld.path.find_or_declare(vscript)
267             if not fullpath:
268                 raise Utils.WafError("unable to find fullpath for %s" % fullname)
269             if not vscriptpath:
270                 raise Utils.WafError("unable to find vscript path for %s" % vscript)
271             bld.add_manual_dependency(fullpath, vscriptpath)
272             if Options.is_install:
273                 # also make the .inst file depend on the vscript
274                 instname = apply_pattern(bundled_name + '.inst', bld.env.shlib_PATTERN)
275                 bld.add_manual_dependency(bld.path.find_or_declare(instname), bld.path.find_or_declare(vscript))
276             vscript = os.path.join(bld.path.abspath(bld.env), vscript)
277
278     bld.SET_BUILD_GROUP(group)
279     t = bld(
280         features        = features,
281         source          = [],
282         target          = bundled_name,
283         depends_on      = depends_on,
284         samba_ldflags   = ldflags,
285         samba_deps      = deps,
286         samba_includes  = includes,
287         version_script  = vscript,
288         local_include   = local_include,
289         global_include  = global_include,
290         vnum            = vnum,
291         soname          = soname,
292         install_path    = None,
293         samba_inst_path = install_path,
294         name            = libname,
295         samba_realname  = realname,
296         samba_install   = install,
297         abi_directory   = "%s/%s" % (bld.path.abspath(), abi_directory),
298         abi_match       = abi_match,
299         private_library = private_library,
300         grouping_library=grouping_library,
301         allow_undefined_symbols=allow_undefined_symbols
302         )
303
304     if realname and not link_name:
305         link_name = 'shared/%s' % realname
306
307     if link_name:
308         t.link_name = link_name
309
310     if pc_files is not None and not private_library:
311         bld.PKG_CONFIG_FILES(pc_files, vnum=vnum)
312
313     if (manpages is not None and 'XSLTPROC_MANPAGES' in bld.env and
314         bld.env['XSLTPROC_MANPAGES']):
315         bld.MANPAGES(manpages, install)
316
317
318 Build.BuildContext.SAMBA_LIBRARY = SAMBA_LIBRARY
319
320
321 #################################################################
322 def SAMBA_BINARY(bld, binname, source,
323                  deps='',
324                  includes='',
325                  public_headers=None,
326                  header_path=None,
327                  modules=None,
328                  ldflags=None,
329                  cflags='',
330                  autoproto=None,
331                  use_hostcc=False,
332                  use_global_deps=True,
333                  compiler=None,
334                  group='main',
335                  manpages=None,
336                  local_include=True,
337                  global_include=True,
338                  subsystem_name=None,
339                  pyembed=False,
340                  vars=None,
341                  subdir=None,
342                  install=True,
343                  install_path=None,
344                  enabled=True):
345     '''define a Samba binary'''
346
347     if not enabled:
348         SET_TARGET_TYPE(bld, binname, 'DISABLED')
349         return
350
351     if not SET_TARGET_TYPE(bld, binname, 'BINARY'):
352         return
353
354     features = 'cc cprogram symlink_bin install_bin'
355     if pyembed:
356         features += ' pyembed'
357
358     obj_target = binname + '.objlist'
359
360     source = bld.EXPAND_VARIABLES(source, vars=vars)
361     if subdir:
362         source = bld.SUBDIR(subdir, source)
363     source = unique_list(TO_LIST(source))
364
365     if group == 'binaries':
366         subsystem_group = 'main'
367     else:
368         subsystem_group = group
369
370     # only specify PIE flags for binaries
371     pie_cflags = cflags
372     pie_ldflags = TO_LIST(ldflags)
373     if bld.env['ENABLE_PIE'] is True:
374         pie_cflags += ' -fPIE'
375         pie_ldflags.extend(TO_LIST('-pie'))
376     if bld.env['ENABLE_RELRO'] is True:
377         pie_ldflags.extend(TO_LIST('-Wl,-z,relro,-z,now'))
378
379     # first create a target for building the object files for this binary
380     # by separating in this way, we avoid recompiling the C files
381     # separately for the install binary and the build binary
382     bld.SAMBA_SUBSYSTEM(obj_target,
383                         source         = source,
384                         deps           = deps,
385                         includes       = includes,
386                         cflags         = pie_cflags,
387                         group          = subsystem_group,
388                         autoproto      = autoproto,
389                         subsystem_name = subsystem_name,
390                         local_include  = local_include,
391                         global_include = global_include,
392                         use_hostcc     = use_hostcc,
393                         pyext          = pyembed,
394                         use_global_deps= use_global_deps)
395
396     bld.SET_BUILD_GROUP(group)
397
398     # the binary itself will depend on that object target
399     deps = TO_LIST(deps)
400     deps.append(obj_target)
401
402     t = bld(
403         features       = features,
404         source         = [],
405         target         = binname,
406         samba_deps     = deps,
407         samba_includes = includes,
408         local_include  = local_include,
409         global_include = global_include,
410         samba_modules  = modules,
411         top            = True,
412         samba_subsystem= subsystem_name,
413         install_path   = None,
414         samba_inst_path= install_path,
415         samba_install  = install,
416         samba_ldflags  = pie_ldflags
417         )
418
419     if manpages is not None and 'XSLTPROC_MANPAGES' in bld.env and bld.env['XSLTPROC_MANPAGES']:
420         bld.MANPAGES(manpages, install)
421
422 Build.BuildContext.SAMBA_BINARY = SAMBA_BINARY
423
424
425 #################################################################
426 def SAMBA_MODULE(bld, modname, source,
427                  deps='',
428                  includes='',
429                  subsystem=None,
430                  init_function=None,
431                  module_init_name='samba_init_module',
432                  autoproto=None,
433                  autoproto_extra_source='',
434                  cflags='',
435                  internal_module=True,
436                  local_include=True,
437                  global_include=True,
438                  vars=None,
439                  subdir=None,
440                  enabled=True,
441                  pyembed=False,
442                  manpages=None,
443                  allow_undefined_symbols=False,
444                  allow_warnings=False
445                  ):
446     '''define a Samba module.'''
447
448     bld.ASSERT(subsystem, "You must specify a subsystem for SAMBA_MODULE(%s)" % modname)
449
450     source = bld.EXPAND_VARIABLES(source, vars=vars)
451     if subdir:
452         source = bld.SUBDIR(subdir, source)
453
454     if internal_module or BUILTIN_LIBRARY(bld, modname):
455         # Do not create modules for disabled subsystems
456         if GET_TARGET_TYPE(bld, subsystem) == 'DISABLED':
457             return
458         bld.SAMBA_SUBSYSTEM(modname, source,
459                     deps=deps,
460                     includes=includes,
461                     autoproto=autoproto,
462                     autoproto_extra_source=autoproto_extra_source,
463                     cflags=cflags,
464                     local_include=local_include,
465                     global_include=global_include,
466                     allow_warnings=allow_warnings,
467                     enabled=enabled)
468
469         bld.ADD_INIT_FUNCTION(subsystem, modname, init_function)
470         return
471
472     if not enabled:
473         SET_TARGET_TYPE(bld, modname, 'DISABLED')
474         return
475
476     # Do not create modules for disabled subsystems
477     if GET_TARGET_TYPE(bld, subsystem) == 'DISABLED':
478         return
479
480     obj_target = modname + '.objlist'
481
482     realname = modname
483     deps += ' ' + subsystem
484     while realname.startswith("lib"+subsystem+"_"):
485         realname = realname[len("lib"+subsystem+"_"):]
486     while realname.startswith(subsystem+"_"):
487         realname = realname[len(subsystem+"_"):]
488
489     realname = bld.make_libname(realname)
490     while realname.startswith("lib"):
491         realname = realname[len("lib"):]
492
493     build_link_name = "modules/%s/%s" % (subsystem, realname)
494
495     if init_function:
496         cflags += " -D%s=%s" % (init_function, module_init_name)
497
498     bld.SAMBA_LIBRARY(modname,
499                       source,
500                       deps=deps,
501                       includes=includes,
502                       cflags=cflags,
503                       realname = realname,
504                       autoproto = autoproto,
505                       local_include=local_include,
506                       global_include=global_include,
507                       vars=vars,
508                       link_name=build_link_name,
509                       install_path="${MODULESDIR}/%s" % subsystem,
510                       pyembed=pyembed,
511                       manpages=manpages,
512                       allow_undefined_symbols=allow_undefined_symbols,
513                       allow_warnings=allow_warnings
514                       )
515
516
517 Build.BuildContext.SAMBA_MODULE = SAMBA_MODULE
518
519
520 #################################################################
521 def SAMBA_SUBSYSTEM(bld, modname, source,
522                     deps='',
523                     public_deps='',
524                     includes='',
525                     public_headers=None,
526                     public_headers_install=True,
527                     header_path=None,
528                     cflags='',
529                     cflags_end=None,
530                     group='main',
531                     init_function_sentinel=None,
532                     autoproto=None,
533                     autoproto_extra_source='',
534                     depends_on='',
535                     local_include=True,
536                     local_include_first=True,
537                     global_include=True,
538                     subsystem_name=None,
539                     enabled=True,
540                     use_hostcc=False,
541                     use_global_deps=True,
542                     vars=None,
543                     subdir=None,
544                     hide_symbols=False,
545                     allow_warnings=False,
546                     pyext=False,
547                     pyembed=False):
548     '''define a Samba subsystem'''
549
550     if not enabled:
551         SET_TARGET_TYPE(bld, modname, 'DISABLED')
552         return
553
554     # remember empty subsystems, so we can strip the dependencies
555     if ((source == '') or (source == [])):
556         if deps == '' and public_deps == '':
557             SET_TARGET_TYPE(bld, modname, 'EMPTY')
558             return
559         empty_c = modname + '.empty.c'
560         bld.SAMBA_GENERATOR('%s_empty_c' % modname,
561                             rule=generate_empty_file,
562                             target=empty_c)
563         source=empty_c
564
565     if not SET_TARGET_TYPE(bld, modname, 'SUBSYSTEM'):
566         return
567
568     source = bld.EXPAND_VARIABLES(source, vars=vars)
569     if subdir:
570         source = bld.SUBDIR(subdir, source)
571     source = unique_list(TO_LIST(source))
572
573     deps += ' ' + public_deps
574
575     bld.SET_BUILD_GROUP(group)
576
577     features = 'cc'
578     if pyext:
579         features += ' pyext'
580     if pyembed:
581         features += ' pyembed'
582
583     t = bld(
584         features       = features,
585         source         = source,
586         target         = modname,
587         samba_cflags   = CURRENT_CFLAGS(bld, modname, cflags,
588                                         allow_warnings=allow_warnings,
589                                         hide_symbols=hide_symbols),
590         depends_on     = depends_on,
591         samba_deps     = TO_LIST(deps),
592         samba_includes = includes,
593         local_include  = local_include,
594         local_include_first  = local_include_first,
595         global_include = global_include,
596         samba_subsystem= subsystem_name,
597         samba_use_hostcc = use_hostcc,
598         samba_use_global_deps = use_global_deps,
599         )
600
601     if cflags_end is not None:
602         t.samba_cflags.extend(TO_LIST(cflags_end))
603
604     if autoproto is not None:
605         bld.SAMBA_AUTOPROTO(autoproto, source + TO_LIST(autoproto_extra_source))
606     if public_headers is not None:
607         bld.PUBLIC_HEADERS(public_headers, header_path=header_path,
608                            public_headers_install=public_headers_install)
609     return t
610
611
612 Build.BuildContext.SAMBA_SUBSYSTEM = SAMBA_SUBSYSTEM
613
614
615 def SAMBA_GENERATOR(bld, name, rule, source='', target='',
616                     group='generators', enabled=True,
617                     public_headers=None,
618                     public_headers_install=True,
619                     header_path=None,
620                     vars=None,
621                     dep_vars=[],
622                     always=False):
623     '''A generic source generator target'''
624
625     if not SET_TARGET_TYPE(bld, name, 'GENERATOR'):
626         return
627
628     if not enabled:
629         return
630
631     dep_vars.append('ruledeps')
632     dep_vars.append('SAMBA_GENERATOR_VARS')
633
634     bld.SET_BUILD_GROUP(group)
635     t = bld(
636         rule=rule,
637         source=bld.EXPAND_VARIABLES(source, vars=vars),
638         target=target,
639         shell=isinstance(rule, str),
640         on_results=True,
641         before='cc',
642         ext_out='.c',
643         samba_type='GENERATOR',
644         dep_vars = dep_vars,
645         name=name)
646
647     if vars is None:
648         vars = {}
649     t.env.SAMBA_GENERATOR_VARS = vars
650
651     if always:
652         t.always = True
653
654     if public_headers is not None:
655         bld.PUBLIC_HEADERS(public_headers, header_path=header_path,
656                            public_headers_install=public_headers_install)
657     return t
658 Build.BuildContext.SAMBA_GENERATOR = SAMBA_GENERATOR
659
660
661
662 @runonce
663 def SETUP_BUILD_GROUPS(bld):
664     '''setup build groups used to ensure that the different build
665     phases happen consecutively'''
666     bld.p_ln = bld.srcnode # we do want to see all targets!
667     bld.env['USING_BUILD_GROUPS'] = True
668     bld.add_group('setup')
669     bld.add_group('build_compiler_source')
670     bld.add_group('vscripts')
671     bld.add_group('base_libraries')
672     bld.add_group('generators')
673     bld.add_group('compiler_prototypes')
674     bld.add_group('compiler_libraries')
675     bld.add_group('build_compilers')
676     bld.add_group('build_source')
677     bld.add_group('prototypes')
678     bld.add_group('headers')
679     bld.add_group('main')
680     bld.add_group('symbolcheck')
681     bld.add_group('syslibcheck')
682     bld.add_group('final')
683 Build.BuildContext.SETUP_BUILD_GROUPS = SETUP_BUILD_GROUPS
684
685
686 def SET_BUILD_GROUP(bld, group):
687     '''set the current build group'''
688     if not 'USING_BUILD_GROUPS' in bld.env:
689         return
690     bld.set_group(group)
691 Build.BuildContext.SET_BUILD_GROUP = SET_BUILD_GROUP
692
693
694
695 @conf
696 def ENABLE_TIMESTAMP_DEPENDENCIES(conf):
697     """use timestamps instead of file contents for deps
698     this currently doesn't work"""
699     def h_file(filename):
700         import stat
701         st = os.stat(filename)
702         if stat.S_ISDIR(st[stat.ST_MODE]): raise IOError('not a file')
703         m = Utils.md5()
704         m.update(str(st.st_mtime))
705         m.update(str(st.st_size))
706         m.update(filename)
707         return m.digest()
708     Utils.h_file = h_file
709
710
711 def SAMBA_SCRIPT(bld, name, pattern, installdir, installname=None):
712     '''used to copy scripts from the source tree into the build directory
713        for use by selftest'''
714
715     source = bld.path.ant_glob(pattern)
716
717     bld.SET_BUILD_GROUP('build_source')
718     for s in TO_LIST(source):
719         iname = s
720         if installname is not None:
721             iname = installname
722         target = os.path.join(installdir, iname)
723         tgtdir = os.path.dirname(os.path.join(bld.srcnode.abspath(bld.env), '..', target))
724         mkdir_p(tgtdir)
725         link_src = os.path.normpath(os.path.join(bld.curdir, s))
726         link_dst = os.path.join(tgtdir, os.path.basename(iname))
727         if os.path.islink(link_dst) and os.readlink(link_dst) == link_src:
728             continue
729         if os.path.exists(link_dst):
730             os.unlink(link_dst)
731         Logs.info("symlink: %s -> %s/%s" % (s, installdir, iname))
732         os.symlink(link_src, link_dst)
733 Build.BuildContext.SAMBA_SCRIPT = SAMBA_SCRIPT
734
735
736 def copy_and_fix_python_path(task):
737     pattern='sys.path.insert(0, "bin/python")'
738     if task.env["PYTHONARCHDIR"] in sys.path and task.env["PYTHONDIR"] in sys.path:
739         replacement = ""
740     elif task.env["PYTHONARCHDIR"] == task.env["PYTHONDIR"]:
741         replacement="""sys.path.insert(0, "%s")""" % task.env["PYTHONDIR"]
742     else:
743         replacement="""sys.path.insert(0, "%s")
744 sys.path.insert(1, "%s")""" % (task.env["PYTHONARCHDIR"], task.env["PYTHONDIR"])
745
746     if task.env["PYTHON"][0] == "/":
747         replacement_shebang = "#!%s\n" % task.env["PYTHON"]
748     else:
749         replacement_shebang = "#!/usr/bin/env %s\n" % task.env["PYTHON"]
750
751     installed_location=task.outputs[0].bldpath(task.env)
752     source_file = open(task.inputs[0].srcpath(task.env))
753     installed_file = open(installed_location, 'w')
754     lineno = 0
755     for line in source_file:
756         newline = line
757         if (lineno == 0 and task.env["PYTHON_SPECIFIED"] is True and
758                 line[:2] == "#!"):
759             newline = replacement_shebang
760         elif pattern in line:
761             newline = line.replace(pattern, replacement)
762         installed_file.write(newline)
763         lineno = lineno + 1
764     installed_file.close()
765     os.chmod(installed_location, 0755)
766     return 0
767
768 def copy_and_fix_perl_path(task):
769     pattern='use lib "$RealBin/lib";'
770
771     replacement = ""
772     if not task.env["PERL_LIB_INSTALL_DIR"] in task.env["PERL_INC"]:
773          replacement = 'use lib "%s";' % task.env["PERL_LIB_INSTALL_DIR"]
774
775     if task.env["PERL"][0] == "/":
776         replacement_shebang = "#!%s\n" % task.env["PERL"]
777     else:
778         replacement_shebang = "#!/usr/bin/env %s\n" % task.env["PERL"]
779
780     installed_location=task.outputs[0].bldpath(task.env)
781     source_file = open(task.inputs[0].srcpath(task.env))
782     installed_file = open(installed_location, 'w')
783     lineno = 0
784     for line in source_file:
785         newline = line
786         if lineno == 0 and task.env["PERL_SPECIFIED"] == True and line[:2] == "#!":
787             newline = replacement_shebang
788         elif pattern in line:
789             newline = line.replace(pattern, replacement)
790         installed_file.write(newline)
791         lineno = lineno + 1
792     installed_file.close()
793     os.chmod(installed_location, 0755)
794     return 0
795
796
797 def install_file(bld, destdir, file, chmod=MODE_644, flat=False,
798                  python_fixup=False, perl_fixup=False,
799                  destname=None, base_name=None):
800     '''install a file'''
801     destdir = bld.EXPAND_VARIABLES(destdir)
802     if not destname:
803         destname = file
804         if flat:
805             destname = os.path.basename(destname)
806     dest = os.path.join(destdir, destname)
807     if python_fixup:
808         # fix the path python will use to find Samba modules
809         inst_file = file + '.inst'
810         bld.SAMBA_GENERATOR('python_%s' % destname,
811                             rule=copy_and_fix_python_path,
812                             dep_vars=["PYTHON","PYTHON_SPECIFIED","PYTHONDIR","PYTHONARCHDIR"],
813                             source=file,
814                             target=inst_file)
815         file = inst_file
816     if perl_fixup:
817         # fix the path perl will use to find Samba modules
818         inst_file = file + '.inst'
819         bld.SAMBA_GENERATOR('perl_%s' % destname,
820                             rule=copy_and_fix_perl_path,
821                             dep_vars=["PERL","PERL_SPECIFIED","PERL_LIB_INSTALL_DIR"],
822                             source=file,
823                             target=inst_file)
824         file = inst_file
825     if base_name:
826         file = os.path.join(base_name, file)
827     bld.install_as(dest, file, chmod=chmod)
828
829
830 def INSTALL_FILES(bld, destdir, files, chmod=MODE_644, flat=False,
831                   python_fixup=False, perl_fixup=False,
832                   destname=None, base_name=None):
833     '''install a set of files'''
834     for f in TO_LIST(files):
835         install_file(bld, destdir, f, chmod=chmod, flat=flat,
836                      python_fixup=python_fixup, perl_fixup=perl_fixup,
837                      destname=destname, base_name=base_name)
838 Build.BuildContext.INSTALL_FILES = INSTALL_FILES
839
840
841 def INSTALL_WILDCARD(bld, destdir, pattern, chmod=MODE_644, flat=False,
842                      python_fixup=False, exclude=None, trim_path=None):
843     '''install a set of files matching a wildcard pattern'''
844     files=TO_LIST(bld.path.ant_glob(pattern))
845     if trim_path:
846         files2 = []
847         for f in files:
848             files2.append(os_path_relpath(f, trim_path))
849         files = files2
850
851     if exclude:
852         for f in files[:]:
853             if fnmatch.fnmatch(f, exclude):
854                 files.remove(f)
855     INSTALL_FILES(bld, destdir, files, chmod=chmod, flat=flat,
856                   python_fixup=python_fixup, base_name=trim_path)
857 Build.BuildContext.INSTALL_WILDCARD = INSTALL_WILDCARD
858
859
860 def INSTALL_DIRS(bld, destdir, dirs):
861     '''install a set of directories'''
862     destdir = bld.EXPAND_VARIABLES(destdir)
863     dirs = bld.EXPAND_VARIABLES(dirs)
864     for d in TO_LIST(dirs):
865         bld.install_dir(os.path.join(destdir, d))
866 Build.BuildContext.INSTALL_DIRS = INSTALL_DIRS
867
868
869 def MANPAGES(bld, manpages, install):
870     '''build and install manual pages'''
871     bld.env.MAN_XSL = 'http://docbook.sourceforge.net/release/xsl/current/manpages/docbook.xsl'
872     for m in manpages.split():
873         source = m + '.xml'
874         bld.SAMBA_GENERATOR(m,
875                             source=source,
876                             target=m,
877                             group='final',
878                             rule='${XSLTPROC} --xinclude -o ${TGT} --nonet ${MAN_XSL} ${SRC}'
879                             )
880         if install:
881             bld.INSTALL_FILES('${MANDIR}/man%s' % m[-1], m, flat=True)
882 Build.BuildContext.MANPAGES = MANPAGES
883
884 def SAMBAMANPAGES(bld, manpages, extra_source=None):
885     '''build and install manual pages'''
886     bld.env.SAMBA_EXPAND_XSL = bld.srcnode.abspath() + '/docs-xml/xslt/expand-sambadoc.xsl'
887     bld.env.SAMBA_MAN_XSL = bld.srcnode.abspath() + '/docs-xml/xslt/man.xsl'
888     bld.env.SAMBA_CATALOGS = 'file:///etc/xml/catalog file:///usr/local/share/xml/catalog file://' + bld.srcnode.abspath() + '/bin/default/docs-xml/build/catalog.xml'
889
890     for m in manpages.split():
891         source = m + '.xml'
892         if extra_source is not None:
893             source = [source, extra_source]
894         bld.SAMBA_GENERATOR(m,
895                             source=source,
896                             target=m,
897                             group='final',
898                             rule='''XML_CATALOG_FILES="${SAMBA_CATALOGS}"
899                                     export XML_CATALOG_FILES
900                                     ${XSLTPROC} --xinclude --stringparam noreference 0 -o ${TGT}.xml --nonet ${SAMBA_EXPAND_XSL} ${SRC[0].abspath(env)}
901                                     ${XSLTPROC} --nonet -o ${TGT} ${SAMBA_MAN_XSL} ${TGT}.xml'''
902                             )
903         bld.INSTALL_FILES('${MANDIR}/man%s' % m[-1], m, flat=True)
904 Build.BuildContext.SAMBAMANPAGES = SAMBAMANPAGES
905
906 #############################################################
907 # give a nicer display when building different types of files
908 def progress_display(self, msg, fname):
909     col1 = Logs.colors(self.color)
910     col2 = Logs.colors.NORMAL
911     total = self.position[1]
912     n = len(str(total))
913     fs = '[%%%dd/%%%dd] %s %%s%%s%%s\n' % (n, n, msg)
914     return fs % (self.position[0], self.position[1], col1, fname, col2)
915
916 def link_display(self):
917     if Options.options.progress_bar != 0:
918         return Task.Task.old_display(self)
919     fname = self.outputs[0].bldpath(self.env)
920     return progress_display(self, 'Linking', fname)
921 Task.TaskBase.classes['cc_link'].display = link_display
922
923 def samba_display(self):
924     if Options.options.progress_bar != 0:
925         return Task.Task.old_display(self)
926
927     targets    = LOCAL_CACHE(self, 'TARGET_TYPE')
928     if self.name in targets:
929         target_type = targets[self.name]
930         type_map = { 'GENERATOR' : 'Generating',
931                      'PROTOTYPE' : 'Generating'
932                      }
933         if target_type in type_map:
934             return progress_display(self, type_map[target_type], self.name)
935
936     if len(self.inputs) == 0:
937         return Task.Task.old_display(self)
938
939     fname = self.inputs[0].bldpath(self.env)
940     if fname[0:3] == '../':
941         fname = fname[3:]
942     ext_loc = fname.rfind('.')
943     if ext_loc == -1:
944         return Task.Task.old_display(self)
945     ext = fname[ext_loc:]
946
947     ext_map = { '.idl' : 'Compiling IDL',
948                 '.et'  : 'Compiling ERRTABLE',
949                 '.asn1': 'Compiling ASN1',
950                 '.c'   : 'Compiling' }
951     if ext in ext_map:
952         return progress_display(self, ext_map[ext], fname)
953     return Task.Task.old_display(self)
954
955 Task.TaskBase.classes['Task'].old_display = Task.TaskBase.classes['Task'].display
956 Task.TaskBase.classes['Task'].display = samba_display
957
958
959 @after('apply_link')
960 @feature('cshlib')
961 def apply_bundle_remove_dynamiclib_patch(self):
962     if self.env['MACBUNDLE'] or getattr(self,'mac_bundle',False):
963         if not getattr(self,'vnum',None):
964             try:
965                 self.env['LINKFLAGS'].remove('-dynamiclib')
966                 self.env['LINKFLAGS'].remove('-single_module')
967             except ValueError:
968                 pass