buildtools/wafsamba/samba_utils.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 errno
   5 import os, sys, re, fnmatch, shlex, inspect
   6 from optparse import SUPPRESS_HELP
   7 from waflib import Build, Options, Utils, Task, Logs, Configure, Errors, Context
   8 from waflib import Scripting
   9 from waflib.TaskGen import feature, before, after
  10 from waflib.Configure import ConfigurationContext
  11 from waflib.Logs import debug
  12 from waflib import ConfigSet
  13 from waflib.Build import CACHE_SUFFIX
  14
  15 # TODO: make this a --option
  16 LIB_PATH="shared"
  17
  18
  19 PY3 = sys.version_info[0] == 3
  20
  21 if PY3:
  22
  23     # helper function to get a string from a variable that maybe 'str' or
  24     # 'bytes' if 'bytes' then it is decoded using 'utf8'. If 'str' is passed
  25     # it is returned unchanged
  26     # Using this function is PY2/PY3 code should ensure in most cases
  27     # the PY2 code runs unchanged in PY2 whereas the code in PY3 possibly
  28     # decodes the variable (see PY2 implementation of this function below)
  29     def get_string(bytesorstring):
  30         tmp = bytesorstring
  31         if isinstance(bytesorstring, bytes):
  32             tmp = bytesorstring.decode('utf8')
  33         elif not isinstance(bytesorstring, str):
  34             raise ValueError('Expected byte of string for %s:%s' % (type(bytesorstring), bytesorstring))
  35         return tmp
  36
  37 else:
  38
  39     # Helper function to return string.
  40     # if 'str' or 'unicode' passed in they are returned unchanged
  41     # otherwise an exception is generated
  42     # Using this function is PY2/PY3 code should ensure in most cases
  43     # the PY2 code runs unchanged in PY2 whereas the code in PY3 possibly
  44     # decodes the variable (see PY3 implementation of this function above)
  45     def get_string(bytesorstring):
  46         tmp = bytesorstring
  47         if not(isinstance(bytesorstring, str) or isinstance(bytesorstring, unicode)):
  48             raise ValueError('Expected str or unicode for %s:%s' % (type(bytesorstring), bytesorstring))
  49         return tmp
  50
  51 # sigh, python octal constants are a mess
  52 MODE_644 = int('644', 8)
  53 MODE_744 = int('744', 8)
  54 MODE_755 = int('755', 8)
  55 MODE_777 = int('777', 8)
  56
  57 def conf(f):
  58     # override in order to propagate the argument "mandatory"
  59     def fun(*k, **kw):
  60         mandatory = True
  61         if 'mandatory' in kw:
  62             mandatory = kw['mandatory']
  63             del kw['mandatory']
  64
  65         try:
  66             return f(*k, **kw)
  67         except Errors.ConfigurationError:
  68             if mandatory:
  69                 raise
  70
  71     fun.__name__ = f.__name__
  72     if 'mandatory' in inspect.getsource(f):
  73         fun = f
  74
  75     setattr(Configure.ConfigurationContext, f.__name__, fun)
  76     setattr(Build.BuildContext, f.__name__, fun)
  77     return f
  78 Configure.conf = conf
  79 Configure.conftest = conf
  80
  81 @conf
  82 def SET_TARGET_TYPE(ctx, target, value):
  83     '''set the target type of a target'''
  84     cache = LOCAL_CACHE(ctx, 'TARGET_TYPE')
  85     if target in cache and cache[target] != 'EMPTY':
  86         Logs.error("ERROR: Target '%s' in directory %s re-defined as %s - was %s" % (target, ctx.path.abspath(), value, cache[target]))
  87         sys.exit(1)
  88     LOCAL_CACHE_SET(ctx, 'TARGET_TYPE', target, value)
  89     debug("task_gen: Target '%s' created of type '%s' in %s" % (target, value, ctx.path.abspath()))
  90     return True
  91
  92
  93 def GET_TARGET_TYPE(ctx, target):
  94     '''get target type from cache'''
  95     cache = LOCAL_CACHE(ctx, 'TARGET_TYPE')
  96     if not target in cache:
  97         return None
  98     return cache[target]
  99
 100
 101 def ADD_LD_LIBRARY_PATH(path):
 102     '''add something to LD_LIBRARY_PATH'''
 103     if 'LD_LIBRARY_PATH' in os.environ:
 104         oldpath = os.environ['LD_LIBRARY_PATH']
 105     else:
 106         oldpath = ''
 107     newpath = oldpath.split(':')
 108     if not path in newpath:
 109         newpath.append(path)
 110         os.environ['LD_LIBRARY_PATH'] = ':'.join(newpath)
 111
 112
 113 def needs_private_lib(bld, target):
 114     '''return True if a target links to a private library'''
 115     for lib in getattr(target, "final_libs", []):
 116         t = bld.get_tgen_by_name(lib)
 117         if t and getattr(t, 'private_library', False):
 118             return True
 119     return False
 120
 121
 122 def install_rpath(target):
 123     '''the rpath value for installation'''
 124     bld = target.bld
 125     bld.env['RPATH'] = []
 126     ret = set()
 127     if bld.env.RPATH_ON_INSTALL:
 128         ret.add(bld.EXPAND_VARIABLES(bld.env.LIBDIR))
 129     if bld.env.RPATH_ON_INSTALL_PRIVATE and needs_private_lib(bld, target):
 130         ret.add(bld.EXPAND_VARIABLES(bld.env.PRIVATELIBDIR))
 131     return list(ret)
 132
 133
 134 def build_rpath(bld):
 135     '''the rpath value for build'''
 136     rpaths = [os.path.normpath('%s/%s' % (bld.env.BUILD_DIRECTORY, d)) for d in ("shared", "shared/private")]
 137     bld.env['RPATH'] = []
 138     if bld.env.RPATH_ON_BUILD:
 139         return rpaths
 140     for rpath in rpaths:
 141         ADD_LD_LIBRARY_PATH(rpath)
 142     return []
 143
 144
 145 @conf
 146 def LOCAL_CACHE(ctx, name):
 147     '''return a named build cache dictionary, used to store
 148        state inside other functions'''
 149     if name in ctx.env:
 150         return ctx.env[name]
 151     ctx.env[name] = {}
 152     return ctx.env[name]
 153
 154
 155 @conf
 156 def LOCAL_CACHE_SET(ctx, cachename, key, value):
 157     '''set a value in a local cache'''
 158     cache = LOCAL_CACHE(ctx, cachename)
 159     cache[key] = value
 160
 161
 162 @conf
 163 def ASSERT(ctx, expression, msg):
 164     '''a build assert call'''
 165     if not expression:
 166         raise Errors.WafError("ERROR: %s\n" % msg)
 167 Build.BuildContext.ASSERT = ASSERT
 168
 169
 170 def SUBDIR(bld, subdir, list):
 171     '''create a list of files by pre-pending each with a subdir name'''
 172     ret = ''
 173     for l in TO_LIST(list):
 174         ret = ret + os.path.normpath(os.path.join(subdir, l)) + ' '
 175     return ret
 176 Build.BuildContext.SUBDIR = SUBDIR
 177
 178
 179 def dict_concat(d1, d2):
 180     '''concatenate two dictionaries d1 += d2'''
 181     for t in d2:
 182         if t not in d1:
 183             d1[t] = d2[t]
 184
 185 def ADD_COMMAND(opt, name, function):
 186     '''add a new top level command to waf'''
 187     Context.g_module.__dict__[name] = function
 188     opt.name = function
 189 Options.OptionsContext.ADD_COMMAND = ADD_COMMAND
 190
 191
 192 @feature('c', 'cc', 'cshlib', 'cprogram')
 193 @before('apply_core','exec_rule')
 194 def process_depends_on(self):
 195     '''The new depends_on attribute for build rules
 196        allow us to specify a dependency on output from
 197        a source generation rule'''
 198     if getattr(self , 'depends_on', None):
 199         lst = self.to_list(self.depends_on)
 200         for x in lst:
 201             y = self.bld.get_tgen_by_name(x)
 202             self.bld.ASSERT(y is not None, "Failed to find dependency %s of %s" % (x, self.name))
 203             y.post()
 204             if getattr(y, 'more_includes', None):
 205                   self.includes += " " + y.more_includes
 206
 207
 208 def unique_list(seq):
 209     '''return a uniquified list in the same order as the existing list'''
 210     seen = {}
 211     result = []
 212     for item in seq:
 213         if item in seen: continue
 214         seen[item] = True
 215         result.append(item)
 216     return result
 217
 218
 219 def TO_LIST(str, delimiter=None):
 220     '''Split a list, preserving quoted strings and existing lists'''
 221     if str is None:
 222         return []
 223     if isinstance(str, list):
 224         # we need to return a new independent list...
 225         return list(str)
 226     if len(str) == 0:
 227         return []
 228     lst = str.split(delimiter)
 229     # the string may have had quotes in it, now we
 230     # check if we did have quotes, and use the slower shlex
 231     # if we need to
 232     for e in lst:
 233         if e[0] == '"':
 234             return shlex.split(str)
 235     return lst
 236
 237
 238 def subst_vars_error(string, env):
 239     '''substitute vars, throw an error if a variable is not defined'''
 240     lst = re.split(r'(\$\{\w+\})', string)
 241     out = []
 242     for v in lst:
 243         if re.match(r'\$\{\w+\}', v):
 244             vname = v[2:-1]
 245             if not vname in env:
 246                 raise KeyError("Failed to find variable %s in %s in env %s <%s>" % (vname, string, env.__class__, str(env)))
 247             v = env[vname]
 248             if isinstance(v, list):
 249                 v = ' '.join(v)
 250         out.append(v)
 251     return ''.join(out)
 252
 253
 254 @conf
 255 def SUBST_ENV_VAR(ctx, varname):
 256     '''Substitute an environment variable for any embedded variables'''
 257     return subst_vars_error(ctx.env[varname], ctx.env)
 258 Build.BuildContext.SUBST_ENV_VAR = SUBST_ENV_VAR
 259
 260
 261 def recursive_dirlist(dir, relbase, pattern=None):
 262     '''recursive directory list'''
 263     ret = []
 264     for f in os.listdir(dir):
 265         f2 = dir + '/' + f
 266         if os.path.isdir(f2):
 267             ret.extend(recursive_dirlist(f2, relbase))
 268         else:
 269             if pattern and not fnmatch.fnmatch(f, pattern):
 270                 continue
 271             ret.append(os.path.relpath(f2, relbase))
 272     return ret
 273
 274
 275 def symlink(src, dst, force=True):
 276     """Can create symlink by force"""
 277     try:
 278         os.symlink(src, dst)
 279     except OSError as exc:
 280         if exc.errno == errno.EEXIST and force:
 281             os.remove(dst)
 282             os.symlink(src, dst)
 283         else:
 284             raise
 285
 286
 287 def mkdir_p(dir):
 288     '''like mkdir -p'''
 289     if not dir:
 290         return
 291     if dir.endswith("/"):
 292         mkdir_p(dir[:-1])
 293         return
 294     if os.path.isdir(dir):
 295         return
 296     mkdir_p(os.path.dirname(dir))
 297     os.mkdir(dir)
 298
 299
 300 def SUBST_VARS_RECURSIVE(string, env):
 301     '''recursively expand variables'''
 302     if string is None:
 303         return string
 304     limit=100
 305     while (string.find('${') != -1 and limit > 0):
 306         string = subst_vars_error(string, env)
 307         limit -= 1
 308     return string
 309
 310
 311 @conf
 312 def EXPAND_VARIABLES(ctx, varstr, vars=None):
 313     '''expand variables from a user supplied dictionary
 314
 315     This is most useful when you pass vars=locals() to expand
 316     all your local variables in strings
 317     '''
 318
 319     if isinstance(varstr, list):
 320         ret = []
 321         for s in varstr:
 322             ret.append(EXPAND_VARIABLES(ctx, s, vars=vars))
 323         return ret
 324
 325     if not isinstance(varstr, str):
 326         return varstr
 327
 328     env = ConfigSet.ConfigSet()
 329     ret = varstr
 330     # substitute on user supplied dict if available
 331     if vars is not None:
 332         for v in vars.keys():
 333             env[v] = vars[v]
 334         ret = SUBST_VARS_RECURSIVE(ret, env)
 335
 336     # if anything left, subst on the environment as well
 337     if ret.find('${') != -1:
 338         ret = SUBST_VARS_RECURSIVE(ret, ctx.env)
 339     # make sure there is nothing left. Also check for the common
 340     # typo of $( instead of ${
 341     if ret.find('${') != -1 or ret.find('$(') != -1:
 342         Logs.error('Failed to substitute all variables in varstr=%s' % ret)
 343         sys.exit(1)
 344     return ret
 345 Build.BuildContext.EXPAND_VARIABLES = EXPAND_VARIABLES
 346
 347
 348 def RUN_COMMAND(cmd,
 349                 env=None,
 350                 shell=False):
 351     '''run a external command, return exit code or signal'''
 352     if env:
 353         cmd = SUBST_VARS_RECURSIVE(cmd, env)
 354
 355     status = os.system(cmd)
 356     if os.WIFEXITED(status):
 357         return os.WEXITSTATUS(status)
 358     if os.WIFSIGNALED(status):
 359         return - os.WTERMSIG(status)
 360     Logs.error("Unknown exit reason %d for command: %s" % (status, cmd))
 361     return -1
 362
 363
 364 def RUN_PYTHON_TESTS(testfiles, pythonpath=None, extra_env=None):
 365     env = LOAD_ENVIRONMENT()
 366     if pythonpath is None:
 367         pythonpath = os.path.join(Context.g_module.out, 'python')
 368     result = 0
 369     for interp in env.python_interpreters:
 370         if not isinstance(interp, str):
 371             interp = ' '.join(interp)
 372         for testfile in testfiles:
 373             cmd = "PYTHONPATH=%s %s %s" % (pythonpath, interp, testfile)
 374             if extra_env:
 375                 for key, value in extra_env.items():
 376                     cmd = "%s=%s %s" % (key, value, cmd)
 377             print('Running Python test with %s: %s' % (interp, testfile))
 378             ret = RUN_COMMAND(cmd)
 379             if ret:
 380                 print('Python test failed: %s' % cmd)
 381                 result = ret
 382     return result
 383
 384
 385 # make sure we have md5. some systems don't have it
 386 try:
 387     from hashlib import md5
 388     # Even if hashlib.md5 exists, it may be unusable.
 389     # Try to use MD5 function. In FIPS mode this will cause an exception
 390     # and we'll get to the replacement code
 391     foo = md5(b'abcd')
 392 except:
 393     try:
 394         import md5
 395         # repeat the same check here, mere success of import is not enough.
 396         # Try to use MD5 function. In FIPS mode this will cause an exception
 397         foo = md5.md5(b'abcd')
 398     except:
 399         Context.SIG_NIL = hash('abcd')
 400         class replace_md5(object):
 401             def __init__(self):
 402                 self.val = None
 403             def update(self, val):
 404                 self.val = hash((self.val, val))
 405             def digest(self):
 406                 return str(self.val)
 407             def hexdigest(self):
 408                 return self.digest().encode('hex')
 409         def replace_h_file(filename):
 410             f = open(filename, 'rb')
 411             m = replace_md5()
 412             while (filename):
 413                 filename = f.read(100000)
 414                 m.update(filename)
 415             f.close()
 416             return m.digest()
 417         Utils.md5 = replace_md5
 418         Task.md5 = replace_md5
 419         Utils.h_file = replace_h_file
 420
 421
 422 def LOAD_ENVIRONMENT():
 423     '''load the configuration environment, allowing access to env vars
 424        from new commands'''
 425     env = ConfigSet.ConfigSet()
 426     try:
 427         p = os.path.join(Context.g_module.out, 'c4che/default'+CACHE_SUFFIX)
 428         env.load(p)
 429     except (OSError, IOError):
 430         pass
 431     return env
 432
 433
 434 def IS_NEWER(bld, file1, file2):
 435     '''return True if file1 is newer than file2'''
 436     curdir = bld.path.abspath()
 437     t1 = os.stat(os.path.join(curdir, file1)).st_mtime
 438     t2 = os.stat(os.path.join(curdir, file2)).st_mtime
 439     return t1 > t2
 440 Build.BuildContext.IS_NEWER = IS_NEWER
 441
 442
 443 @conf
 444 def RECURSE(ctx, directory):
 445     '''recurse into a directory, relative to the curdir or top level'''
 446     try:
 447         visited_dirs = ctx.visited_dirs
 448     except AttributeError:
 449         visited_dirs = ctx.visited_dirs = set()
 450     d = os.path.join(ctx.path.abspath(), directory)
 451     if os.path.exists(d):
 452         abspath = os.path.abspath(d)
 453     else:
 454         abspath = os.path.abspath(os.path.join(Context.g_module.top, directory))
 455     ctxclass = ctx.__class__.__name__
 456     key = ctxclass + ':' + abspath
 457     if key in visited_dirs:
 458         # already done it
 459         return
 460     visited_dirs.add(key)
 461     relpath = os.path.relpath(abspath, ctx.path.abspath())
 462     if ctxclass in ['OptionsContext',
 463                     'ConfigurationContext',
 464                     'BuildContext',
 465                     'CleanContext',
 466                     'InstallContext',
 467                     'UninstallContext',
 468                     'ListContext']:
 469         return ctx.recurse(relpath)
 470     if 'waflib.extras.compat15' in sys.modules:
 471         return ctx.recurse(relpath)
 472     raise Errors.WafError('Unknown RECURSE context class: {}'.format(ctxclass))
 473 Options.OptionsContext.RECURSE = RECURSE
 474 Build.BuildContext.RECURSE = RECURSE
 475
 476
 477 def CHECK_MAKEFLAGS(options):
 478     '''check for MAKEFLAGS environment variable in case we are being
 479     called from a Makefile try to honor a few make command line flags'''
 480     if not 'WAF_MAKE' in os.environ:
 481         return
 482     makeflags = os.environ.get('MAKEFLAGS')
 483     if makeflags is None:
 484         makeflags = ""
 485     jobs_set = False
 486     jobs = None
 487     # we need to use shlex.split to cope with the escaping of spaces
 488     # in makeflags
 489     for opt in shlex.split(makeflags):
 490         # options can come either as -x or as x
 491         if opt[0:2] == 'V=':
 492             options.verbose = Logs.verbose = int(opt[2:])
 493             if Logs.verbose > 0:
 494                 Logs.zones = ['runner']
 495             if Logs.verbose > 2:
 496                 Logs.zones = ['*']
 497         elif opt[0].isupper() and opt.find('=') != -1:
 498             # this allows us to set waf options on the make command line
 499             # for example, if you do "make FOO=blah", then we set the
 500             # option 'FOO' in Options.options, to blah. If you look in wafsamba/wscript
 501             # you will see that the command line accessible options have their dest=
 502             # set to uppercase, to allow for passing of options from make in this way
 503             # this is also how "make test TESTS=testpattern" works, and
 504             # "make VERBOSE=1" as well as things like "make SYMBOLCHECK=1"
 505             loc = opt.find('=')
 506             setattr(options, opt[0:loc], opt[loc+1:])
 507         elif opt[0] != '-':
 508             for v in opt:
 509                 if re.search(r'j[0-9]*$', v):
 510                     jobs_set = True
 511                     jobs = opt.strip('j')
 512                 elif v == 'k':
 513                     options.keep = True
 514         elif re.search(r'-j[0-9]*$', opt):
 515             jobs_set = True
 516             jobs = opt.strip('-j')
 517         elif opt == '-k':
 518             options.keep = True
 519     if not jobs_set:
 520         # default to one job
 521         options.jobs = 1
 522     elif jobs_set and jobs:
 523         options.jobs = int(jobs)
 524
 525 waflib_options_parse_cmd_args = Options.OptionsContext.parse_cmd_args
 526 def wafsamba_options_parse_cmd_args(self, _args=None, cwd=None, allow_unknown=False):
 527     (options, commands, envvars) = \
 528         waflib_options_parse_cmd_args(self,
 529                                       _args=_args,
 530                                       cwd=cwd,
 531                                       allow_unknown=allow_unknown)
 532     CHECK_MAKEFLAGS(options)
 533     if options.jobs == 1:
 534         #
 535         # waflib.Runner.Parallel processes jobs inline if the possible number
 536         # of jobs is just 1. But (at least in waf <= 2.0.12) it still calls
 537         # create a waflib.Runner.Spawner() which creates a single
 538         # waflib.Runner.Consumer() thread that tries to process jobs from the
 539         # queue.
 540         #
 541         # This has strange effects, which are not noticed typically,
 542         # but at least on AIX python has broken threading and fails
 543         # in random ways.
 544         #
 545         # So we just add a dummy Spawner class.
 546         class NoOpSpawner(object):
 547             def __init__(self, master):
 548                 return
 549         from waflib import Runner
 550         Runner.Spawner = NoOpSpawner
 551     return options, commands, envvars
 552 Options.OptionsContext.parse_cmd_args = wafsamba_options_parse_cmd_args
 553
 554 option_groups = {}
 555
 556 def option_group(opt, name):
 557     '''find or create an option group'''
 558     global option_groups
 559     if name in option_groups:
 560         return option_groups[name]
 561     gr = opt.add_option_group(name)
 562     option_groups[name] = gr
 563     return gr
 564 Options.OptionsContext.option_group = option_group
 565
 566
 567 def save_file(filename, contents, create_dir=False):
 568     '''save data to a file'''
 569     if create_dir:
 570         mkdir_p(os.path.dirname(filename))
 571     try:
 572         f = open(filename, 'w')
 573         f.write(contents)
 574         f.close()
 575     except:
 576         return False
 577     return True
 578
 579
 580 def load_file(filename):
 581     '''return contents of a file'''
 582     try:
 583         f = open(filename, 'r')
 584         r = f.read()
 585         f.close()
 586     except:
 587         return None
 588     return r
 589
 590
 591 def reconfigure(ctx):
 592     '''rerun configure if necessary'''
 593     if not os.path.exists(os.environ.get('WAFLOCK', '.lock-wscript')):
 594         raise Errors.WafError('configure has not been run')
 595     import samba_wildcard
 596     bld = samba_wildcard.fake_build_environment()
 597     Configure.autoconfig = True
 598     Scripting.check_configured(bld)
 599
 600
 601 def map_shlib_extension(ctx, name, python=False):
 602     '''map a filename with a shared library extension of .so to the real shlib name'''
 603     if name is None:
 604         return None
 605     if name[-1:].isdigit():
 606         # some libraries have specified versions in the wscript rule
 607         return name
 608     (root1, ext1) = os.path.splitext(name)
 609     if python:
 610         return ctx.env.pyext_PATTERN % root1
 611     else:
 612         (root2, ext2) = os.path.splitext(ctx.env.cshlib_PATTERN)
 613     return root1+ext2
 614 Build.BuildContext.map_shlib_extension = map_shlib_extension
 615
 616 def apply_pattern(filename, pattern):
 617     '''apply a filename pattern to a filename that may have a directory component'''
 618     dirname = os.path.dirname(filename)
 619     if not dirname:
 620         return pattern % filename
 621     basename = os.path.basename(filename)
 622     return os.path.join(dirname, pattern % basename)
 623
 624 def make_libname(ctx, name, nolibprefix=False, version=None, python=False):
 625     """make a library filename
 626          Options:
 627               nolibprefix: don't include the lib prefix
 628               version    : add a version number
 629               python     : if we should use python module name conventions"""
 630
 631     if python:
 632         libname = apply_pattern(name, ctx.env.pyext_PATTERN)
 633     else:
 634         libname = apply_pattern(name, ctx.env.cshlib_PATTERN)
 635     if nolibprefix and libname[0:3] == 'lib':
 636         libname = libname[3:]
 637     if version:
 638         if version[0] == '.':
 639             version = version[1:]
 640         (root, ext) = os.path.splitext(libname)
 641         if ext == ".dylib":
 642             # special case - version goes before the prefix
 643             libname = "%s.%s%s" % (root, version, ext)
 644         else:
 645             libname = "%s%s.%s" % (root, ext, version)
 646     return libname
 647 Build.BuildContext.make_libname = make_libname
 648
 649
 650 def get_tgt_list(bld):
 651     '''return a list of build objects for samba'''
 652
 653     targets = LOCAL_CACHE(bld, 'TARGET_TYPE')
 654
 655     # build a list of task generators we are interested in
 656     tgt_list = []
 657     for tgt in targets:
 658         type = targets[tgt]
 659         if not type in ['SUBSYSTEM', 'BUILTIN', 'MODULE', 'BINARY', 'LIBRARY', 'PLUGIN', 'ASN1', 'PYTHON']:
 660             continue
 661         t = bld.get_tgen_by_name(tgt)
 662         if t is None:
 663             Logs.error("Target %s of type %s has no task generator" % (tgt, type))
 664             sys.exit(1)
 665         tgt_list.append(t)
 666     return tgt_list
 667
 668 from waflib.Context import WSCRIPT_FILE
 669 def PROCESS_SEPARATE_RULE(self, rule):
 670     ''' cause waf to process additional script based on `rule'.
 671         You should have file named wscript_<stage>_rule in the current directory
 672         where stage is either 'configure' or 'build'
 673     '''
 674     stage = ''
 675     if isinstance(self, Configure.ConfigurationContext):
 676         stage = 'configure'
 677     elif isinstance(self, Build.BuildContext):
 678         stage = 'build'
 679     file_path = os.path.join(self.path.abspath(), WSCRIPT_FILE+'_'+stage+'_'+rule)
 680     node = self.root.find_node(file_path)
 681     if node:
 682         try:
 683             cache = self.recurse_cache
 684         except AttributeError:
 685             cache = self.recurse_cache = {}
 686         if node not in cache:
 687             cache[node] = True
 688             self.pre_recurse(node)
 689             try:
 690                 function_code = node.read('r', None)
 691                 exec(compile(function_code, node.abspath(), 'exec'), self.exec_dict)
 692             finally:
 693                 self.post_recurse(node)
 694
 695 Build.BuildContext.PROCESS_SEPARATE_RULE = PROCESS_SEPARATE_RULE
 696 ConfigurationContext.PROCESS_SEPARATE_RULE = PROCESS_SEPARATE_RULE
 697
 698 def AD_DC_BUILD_IS_ENABLED(self):
 699     if self.CONFIG_SET('AD_DC_BUILD_IS_ENABLED'):
 700         return True
 701     return False
 702
 703 Build.BuildContext.AD_DC_BUILD_IS_ENABLED = AD_DC_BUILD_IS_ENABLED
 704
 705 @feature('cprogram', 'cshlib', 'cstaticlib')
 706 @after('apply_lib_vars')
 707 @before('apply_obj_vars')
 708 def samba_before_apply_obj_vars(self):
 709     """before apply_obj_vars for uselib, this removes the standard paths"""
 710
 711     def is_standard_libpath(env, path):
 712         normalized_path = os.path.normpath(path)
 713         for _path in env.STANDARD_LIBPATH:
 714             if _path == normalized_path:
 715                 return True
 716         return False
 717
 718     v = self.env
 719
 720     for i in v['RPATH']:
 721         if is_standard_libpath(v, i):
 722             v['RPATH'].remove(i)
 723
 724     for i in v['LIBPATH']:
 725         if is_standard_libpath(v, i):
 726             v['LIBPATH'].remove(i)
 727
 728 # Samba options are mostly on by default (administrators and packagers
 729 # specify features to remove, not add), which is why default=True
 730
 731 def samba_add_onoff_option(opt, option, help=(), dest=None, default=True,
 732                            with_name="with", without_name="without"):
 733     if default is None:
 734         default_str = "auto"
 735     elif default is True:
 736         default_str = "yes"
 737     elif default is False:
 738         default_str = "no"
 739     else:
 740         default_str = str(default)
 741
 742     if help == ():
 743         help = ("Build with %s support (default=%s)" % (option, default_str))
 744     if dest is None:
 745         dest = "with_%s" % option.replace('-', '_')
 746
 747     with_val = "--%s-%s" % (with_name, option)
 748     without_val = "--%s-%s" % (without_name, option)
 749
 750     opt.add_option(with_val, help=help, action="store_true", dest=dest,
 751                    default=default)
 752     opt.add_option(without_val, help=SUPPRESS_HELP, action="store_false",
 753                    dest=dest)
 754 Options.OptionsContext.samba_add_onoff_option = samba_add_onoff_option