selftest/selftest.py

   1 #!/usr/bin/python -u
   2 # Bootstrap Samba and run a number of tests against it.
   3 # Copyright (C) 2005-2012 Jelmer Vernooij <jelmer@samba.org>
   4 # Copyright (C) 2007-2009 Stefan Metzmacher <metze@samba.org>
   5
   6 # This program is free software; you can redistribute it and/or modify
   7 # it under the terms of the GNU General Public License as published by
   8 # the Free Software Foundation; either version 3 of the License, or
   9 # (at your option) any later version.
  10
  11 # This program is distributed in the hope that it will be useful,
  12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 # GNU General Public License for more details.
  15
  16 # You should have received a copy of the GNU General Public License
  17 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18
  19 import atexit
  20 from cStringIO import StringIO
  21 import os
  22 import sys
  23 import signal
  24 import subprocess
  25 from samba import subunit
  26 import traceback
  27 import warnings
  28
  29 import optparse
  30
  31 sys.path.insert(0, os.path.dirname(os.path.dirname(__file__)))
  32
  33 from selftest import (
  34     socket_wrapper,
  35     subunithelper,
  36     testlist,
  37     )
  38 from selftest.client import write_clientconf
  39 from selftest.run import (
  40     expand_command_list,
  41     expand_command_run,
  42     exported_envvars_str,
  43     now,
  44     run_testsuite_command,
  45     )
  46 from selftest.target import (
  47     EnvironmentManager,
  48     NoneTarget,
  49     UnsupportedEnvironment,
  50     )
  51
  52 includes = ()
  53 excludes = ()
  54
  55 def read_excludes(fn):
  56     excludes.extend(testlist.read_test_regexes(fn))
  57
  58 def read_includes(fn):
  59     includes.extend(testlist.read_test_regexes(fn))
  60
  61 parser = optparse.OptionParser("TEST-REGEXES")
  62 parser.add_option("--target", type="choice", choices=["samba", "samba3", "none"], default="samba", help="Samba version to target")
  63 parser.add_option("--quick", help="run quick overall test", action="store_true", default=False)
  64 parser.add_option("--list", help="list available tests", action="store_true", default=False)
  65 parser.add_option("--socket-wrapper", help="enable socket wrapper", action="store_true", default=False)
  66 parser.add_option("--socket-wrapper-pcap", help="save traffic to pcap directories", type="str")
  67 parser.add_option("--socket-wrapper-keep-pcap", help="keep all pcap files, not just those for tests that failed", action="store_true", default=False)
  68 parser.add_option("--one", help="abort when the first test fails", action="store_true", default=False)
  69 parser.add_option("--exclude", action="callback", help="Add file to exclude files", callback=read_excludes)
  70 parser.add_option("--include", action="callback", help="Add file to include files", callback=read_includes)
  71 parser.add_option("--testenv", help="run a shell in the requested test environment", action="store_true", default=False)
  72 parser.add_option("--resetup-environment", help="Re-setup environment", action="store_true", default=False)
  73 parser.add_option("--load-list", help="Load list of tests to load from a file", type=str)
  74 parser.add_option("--prefix", help="prefix to run tests in", type=str, default="./st")
  75 parser.add_option("--srcdir", type=str, default=".", help="source directory")
  76 parser.add_option("--bindir", type=str, default="./bin", help="binaries directory")
  77 parser.add_option("--testlist", type=str, action="append", help="file to read available tests from")
  78 parser.add_option("--ldap", help="back samba onto specified ldap server", choices=["openldap", "fedora-ds"], type="choice")
  79
  80 opts, args = parser.parse_args()
  81
  82 subunit_ops = subunithelper.SubunitOps(sys.stdout)
  83
  84 def handle_signal(sig, frame):
  85     sys.stderr.write("Exiting early because of signal %s.\n" % sig)
  86     sys.exit(1)
  87
  88 for sig in (signal.SIGINT, signal.SIGQUIT, signal.SIGTERM, signal.SIGPIPE):
  89     signal.signal(sig, handle_signal)
  90
  91 def skip(name):
  92     return testlist.find_in_list(excludes, name)
  93
  94 def setup_pcap(name):
  95     if (not opts.socket_wrapper_pcap or
  96         not os.environ.get("SOCKET_WRAPPER_PCAP_DIR")):
  97         return
  98
  99     fname = "".join([x for x in name if x.isalnum() or x == '-'])
 100
 101     pcap_file = os.path.join(
 102         os.environ["SOCKET_WRAPPER_PCAP_DIR"], "%s.pcap" % fname)
 103
 104     socket_wrapper.setup_pcap(pcap_file)
 105     return pcap_file
 106
 107
 108 def cleanup_pcap(pcap_file, exit_code):
 109     if not opts.socket_wrapper_pcap:
 110         return
 111     if opts.socket_wrapper_keep_pcap:
 112         return
 113     if exitcode == 0:
 114         return
 115     if pcap_file is None:
 116         return
 117
 118     os.unlink(pcap_file)
 119
 120
 121 def run_testsuite(name, cmd, subunit_ops, env=None):
 122     """Run a single testsuite.
 123
 124     :param env: Environment to run in
 125     :param name: Name of the testsuite
 126     :param cmd: Name of the (fully expanded) command to run
 127     :return: exitcode of the command
 128     """
 129     pcap_file = setup_pcap(name)
 130
 131     exitcode = run_testsuite_command(name, cmd, subunit_ops, env)
 132     if exitcode is None:
 133         sys.exit(1)
 134
 135     cleanup_pcap(pcap_file, exitcode)
 136
 137     if not opts.socket_wrapper_keep_pcap and pcap_file is not None:
 138         sys.stdout.write("PCAP FILE: %s\n" % pcap_file)
 139
 140     if exitcode != 0 and opts.one:
 141         sys.exit(1)
 142
 143     return exitcode
 144
 145
 146 if opts.list and opts.testenv:
 147     sys.stderr.write("--list and --testenv are mutually exclusive\n")
 148     sys.exit(1)
 149
 150 tests = args
 151
 152 # quick hack to disable rpc validation when using valgrind - it is way too slow
 153 if not os.environ.get("VALGRIND"):
 154     os.environ["VALIDATE"] = "validate"
 155     os.environ["MALLOC_CHECK_"] = "3"
 156
 157 # make all our python scripts unbuffered
 158 os.environ["PYTHONUNBUFFERED"] = "1"
 159
 160 bindir_abs = os.path.abspath(opts.bindir)
 161
 162 # Backwards compatibility:
 163 if os.environ.get("TEST_LDAP") == "yes":
 164     if os.environ.get("FEDORA_DS_ROOT"):
 165         ldap = "fedora-ds"
 166     else:
 167         ldap = "openldap"
 168
 169 torture_maxtime = int(os.getenv("TORTURE_MAXTIME", "1200"))
 170 if opts.ldap:
 171     # LDAP is slow
 172     torture_maxtime *= 2
 173
 174 prefix = os.path.normpath(opts.prefix)
 175
 176 # Ensure we have the test prefix around.
 177 #
 178 # We need restrictive permissions on this as some subdirectories in this tree
 179 # will have wider permissions (ie 0777) and this would allow other users on the
 180 # host to subvert the test process.
 181 if not os.path.isdir(prefix):
 182     os.mkdir(prefix, 0700)
 183 else:
 184     os.chmod(prefix, 0700)
 185
 186 prefix_abs = os.path.abspath(prefix)
 187 tmpdir_abs = os.path.abspath(os.path.join(prefix_abs, "tmp"))
 188 if not os.path.isdir(tmpdir_abs):
 189     os.mkdir(tmpdir_abs, 0777)
 190
 191 srcdir_abs = os.path.abspath(opts.srcdir)
 192
 193 if prefix_abs == "/":
 194     raise Exception("using '/' as absolute prefix is a bad idea")
 195
 196 os.environ["PREFIX"] = prefix
 197 os.environ["KRB5CCNAME"] = os.path.join(prefix, "krb5ticket")
 198 os.environ["PREFIX_ABS"] = prefix_abs
 199 os.environ["SRCDIR"] = opts.srcdir
 200 os.environ["SRCDIR_ABS"] = srcdir_abs
 201 os.environ["BINDIR"] = bindir_abs
 202
 203 tls_enabled = not opts.quick
 204 if tls_enabled:
 205     os.environ["TLS_ENABLED"] = "yes"
 206 else:
 207     os.environ["TLS_ENABLED"] = "no"
 208
 209 def prefix_pathvar(name, newpath):
 210     if name in os.environ:
 211         os.environ[name] = "%s:%s" % (newpath, os.environ[name])
 212     else:
 213         os.environ[name] = newpath
 214 prefix_pathvar("PKG_CONFIG_PATH", os.path.join(bindir_abs, "pkgconfig"))
 215 prefix_pathvar("PYTHONPATH", os.path.join(bindir_abs, "python"))
 216
 217 if opts.socket_wrapper_keep_pcap:
 218     # Socket wrapper keep pcap implies socket wrapper pcap
 219     opts.socket_wrapper_pcap = True
 220
 221 if opts.socket_wrapper_pcap:
 222     # Socket wrapper pcap implies socket wrapper
 223     opts.socket_wrapper = True
 224
 225 if opts.socket_wrapper:
 226     socket_wrapper_dir = socket_wrapper.setup_dir(os.path.join(prefix_abs, "w"), opts.socket_wrapper_pcap)
 227     sys.stdout.write("SOCKET_WRAPPER_DIR=%s\n" % socket_wrapper_dir)
 228 elif not opts.list:
 229     if os.getuid() != 0:
 230         warnings.warn("not using socket wrapper, but also not running as root. Will not be able to listen on proper ports")
 231
 232 testenv_default = "none"
 233
 234 # After this many seconds, the server will self-terminate.  All tests
 235 # must terminate in this time, and testenv will only stay alive this
 236 # long
 237
 238 if os.environ.get("SMBD_MAXTIME", ""):
 239     server_maxtime = int(os.environ["SMBD_MAXTIME"])
 240 else:
 241     server_maxtime = 7500
 242
 243
 244 def has_socket_wrapper(bindir):
 245     """Check if Samba has been built with socket wrapper support.
 246     """
 247     f = StringIO()
 248     subprocess.check_call([os.path.join(bindir, "smbd"), "-b"], stdout=f)
 249     for l in f.readlines():
 250         if "SOCKET_WRAPPER" in l:
 251             return True
 252     return False
 253
 254
 255 if not opts.list:
 256     if opts.target == "samba":
 257         if opts.socket_wrapper and not has_socket_wrapper(opts.bindir):
 258             sys.stderr.write("You must include --enable-socket-wrapper when compiling Samba in order to execute 'make test'.  Exiting....\n")
 259             sys.exit(1)
 260         testenv_default = "ad_dc_ntvfs"
 261         from selftest.target.samba import Samba
 262         target = Samba(opts.bindir, ldap, opts.srcdir, server_maxtime)
 263     elif opts.target == "samba3":
 264         if opts.socket_wrapper and not has_socket_wrapper(opts.bindir):
 265             sys.stderr.write("You must include --enable-socket-wrapper when compiling Samba in order to execute 'make test'.  Exiting....\n")
 266             sys.exit(1)
 267         testenv_default = "member"
 268         from selftest.target.samba3 import Samba3
 269         target = Samba3(opts.bindir, srcdir_abs, server_maxtime)
 270     elif opts.target == "none":
 271         testenv_default = "none"
 272         target = NoneTarget()
 273
 274     env_manager = EnvironmentManager(target)
 275     atexit.register(env_manager.teardown_all)
 276
 277 interfaces = ",".join([
 278     "127.0.0.11/8",
 279     "127.0.0.12/8",
 280     "127.0.0.13/8",
 281     "127.0.0.14/8",
 282     "127.0.0.15/8",
 283     "127.0.0.16/8"])
 284
 285 clientdir = os.path.join(prefix_abs, "client")
 286
 287 conffile = os.path.join(clientdir, "client.conf")
 288 os.environ["SMB_CONF_PATH"] = conffile
 289
 290 todo = []
 291
 292 if not opts.testlist:
 293     sys.stderr.write("No testlists specified\n")
 294     sys.exit(1)
 295
 296 os.environ["SELFTEST_PREFIX"] = prefix_abs
 297 os.environ["SELFTEST_TMPDIR"] = tmpdir_abs
 298 os.environ["TEST_DATA_PREFIX"] = tmpdir_abs
 299 if opts.socket_wrapper:
 300     os.environ["SELFTEST_INTERFACES"] = interfaces
 301 else:
 302     os.environ["SELFTEST_INTERFACES"] = ""
 303 if opts.quick:
 304     os.environ["SELFTEST_QUICK"] = "1"
 305 else:
 306     os.environ["SELFTEST_QUICK"] = ""
 307 os.environ["SELFTEST_MAXTIME"] = str(torture_maxtime)
 308
 309
 310 available = []
 311 for fn in opts.testlist:
 312     for testsuite in testlist.read_testlist_file(fn):
 313         if not testlist.should_run_test(tests, testsuite):
 314             continue
 315         name = testsuite[0]
 316         if (includes is not None and
 317             testlist.find_in_list(includes, name) is not None):
 318             continue
 319         available.append(testsuite)
 320
 321 if opts.load_list:
 322     restricted_mgr = testlist.RestrictedTestManager.from_path(opts.load_list)
 323 else:
 324     restricted_mgr = None
 325
 326 for testsuite in available:
 327     name = testsuite[0]
 328     skipreason = skip(name)
 329     if restricted_mgr is not None:
 330         match = restricted_mgr.should_run_testsuite(name)
 331         if match == []:
 332             continue
 333     else:
 334         match = None
 335     if skipreason is not None:
 336         if not opts.list:
 337             subunit_ops.skip_testsuite(name, skipreason)
 338     else:
 339         todo.append(testsuite + (match,))
 340
 341 if restricted_mgr is not None:
 342     for name in restricted_mgr.iter_unused():
 343         sys.stdout.write("No test or testsuite found matching %s\n" % name)
 344 if todo == []:
 345     sys.stderr.write("No tests to run\n")
 346     sys.exit(1)
 347
 348 suitestotal = len(todo)
 349
 350 if not opts.list:
 351     subunit_ops.progress(suitestotal, subunit.PROGRESS_SET)
 352     subunit_ops.time(now())
 353
 354 exported_envvars = [
 355     # domain stuff
 356     "DOMAIN",
 357     "REALM",
 358
 359     # domain controller stuff
 360     "DC_SERVER",
 361     "DC_SERVER_IP",
 362     "DC_NETBIOSNAME",
 363     "DC_NETBIOSALIAS",
 364
 365     # domain member
 366     "MEMBER_SERVER",
 367     "MEMBER_SERVER_IP",
 368     "MEMBER_NETBIOSNAME",
 369     "MEMBER_NETBIOSALIAS",
 370
 371     # rpc proxy controller stuff
 372     "RPC_PROXY_SERVER",
 373     "RPC_PROXY_SERVER_IP",
 374     "RPC_PROXY_NETBIOSNAME",
 375     "RPC_PROXY_NETBIOSALIAS",
 376
 377     # domain controller stuff for Vampired DC
 378     "VAMPIRE_DC_SERVER",
 379     "VAMPIRE_DC_SERVER_IP",
 380     "VAMPIRE_DC_NETBIOSNAME",
 381     "VAMPIRE_DC_NETBIOSALIAS",
 382
 383     # domain controller stuff for Vampired DC
 384     "PROMOTED_DC_SERVER",
 385     "PROMOTED_DC_SERVER_IP",
 386     "PROMOTED_DC_NETBIOSNAME",
 387     "PROMOTED_DC_NETBIOSALIAS",
 388
 389     # server stuff
 390     "SERVER",
 391     "SERVER_IP",
 392     "NETBIOSNAME",
 393     "NETBIOSALIAS",
 394
 395     # user stuff
 396     "USERNAME",
 397     "USERID",
 398     "PASSWORD",
 399     "DC_USERNAME",
 400     "DC_PASSWORD",
 401
 402     # misc stuff
 403     "KRB5_CONFIG",
 404     "WINBINDD_SOCKET_DIR",
 405     "WINBINDD_PRIV_PIPE_DIR",
 406     "NMBD_SOCKET_DIR",
 407     "LOCAL_PATH"
 408 ]
 409
 410 def switch_env(name, prefix):
 411     if ":" in name:
 412         (envname, option) = name.split(":", 1)
 413     else:
 414         envname = name
 415         option = "client"
 416
 417     env = env_manager.setup_env(envname, prefix)
 418
 419     testenv_vars = env.get_vars()
 420
 421     if option == "local":
 422         socket_wrapper.set_default_iface(testenv_vars["SOCKET_WRAPPER_DEFAULT_IFACE"])
 423         os.environ["SMB_CONF_PATH"] = testenv_vars["SERVERCONFFILE"]
 424     elif option == "client":
 425         socket_wrapper.set_default_iface(11)
 426         write_clientconf(conffile, clientdir, testenv_vars)
 427         os.environ["SMB_CONF_PATH"] = conffile
 428     else:
 429         raise Exception("Unknown option[%s] for envname[%s]" % (option,
 430             envname))
 431
 432     for name in exported_envvars:
 433         if name in testenv_vars:
 434             os.environ[name] = testenv_vars[name]
 435         elif name in os.environ:
 436             del os.environ[name]
 437
 438     return env
 439
 440 # This 'global' file needs to be empty when we start
 441 dns_host_file_path = os.path.join(prefix_abs, "dns_host_file")
 442 if os.path.exists(dns_host_file_path):
 443     os.unlink(dns_host_file_path)
 444
 445 if opts.testenv:
 446     testenv_name = os.environ.get("SELFTEST_TESTENV", testenv_default)
 447
 448     env = switch_env(testenv_name, prefix)
 449     testenv_vars = env.get_vars()
 450
 451     os.environ["PIDDIR"] = testenv_vars["PIDDIR"]
 452     os.environ["ENVNAME"] = testenv_name
 453
 454     envvarstr = exported_envvars_str(testenv_vars, exported_envvars)
 455
 456     term = os.environ.get("TERMINAL", "xterm -e")
 457     cmd = """'echo -e "
 458 Welcome to the Samba4 Test environment '%(testenv_name)'
 459
 460 This matches the client environment used in make test
 461 server is pid `cat \$PIDDIR/samba.pid`
 462
 463 Some useful environment variables:
 464 TORTURE_OPTIONS=\$TORTURE_OPTIONS
 465 SMB_CONF_PATH=\$SMB_CONF_PATH
 466
 467 $envvarstr
 468 \" && LD_LIBRARY_PATH=%(LD_LIBRARY_PATH)s $(SHELL)'""" % {
 469         "testenv_name": testenv_name,
 470         "LD_LIBRARY_PATH": os.environ["LD_LIBRARY_PATH"]}
 471     subprocess.call(term + ' ' + cmd, shell=True)
 472     env_manager.teardown_env(testenv_name)
 473 elif opts.list:
 474     for (name, envname, cmd, supports_loadfile, supports_idlist, subtests) in todo:
 475         cmd = expand_command_list(cmd)
 476         if cmd is None:
 477             warnings.warn("Unable to list tests in %s" % name)
 478             continue
 479
 480         exitcode = subprocess.call(cmd, shell=True)
 481
 482         if exitcode != 0:
 483             sys.stderr.write("%s exited with exit code %s\n" % (cmd, exitcode))
 484             sys.exit(1)
 485 else:
 486     for (name, envname, cmd, supports_loadfile, supports_idlist, subtests) in todo:
 487         try:
 488             env = switch_env(envname, prefix)
 489         except UnsupportedEnvironment:
 490             subunit_ops.start_testsuite(name)
 491             subunit_ops.end_testsuite(name, "skip",
 492                 "environment %s is unknown in this test backend - skipping" % envname)
 493             continue
 494         except Exception, e:
 495             subunit_ops.start_testsuite(name)
 496             traceback.print_exc()
 497             subunit_ops.end_testsuite(name, "error",
 498                 "unable to set up environment %s: %s" % (envname, e))
 499             continue
 500
 501         cmd, tmpf = expand_command_run(cmd, supports_loadfile, supports_idlist,
 502             subtests)
 503
 504         run_testsuite(name, cmd, subunit_ops, env=env)
 505
 506         if tmpf is not None:
 507             os.remove(tmpf)
 508
 509         if opts.resetup_environment:
 510             env_manager.teardown_env(envname)
 511     env_manager.teardown_all()
 512
 513 sys.stdout.write("\n")
 514
 515 # if there were any valgrind failures, show them
 516 for fn in os.listdir(prefix):
 517     if fn.startswith("valgrind.log"):
 518         sys.stdout.write("VALGRIND FAILURE\n")
 519         f = open(os.path.join(prefix, fn), 'r')
 520         try:
 521             sys.stdout.write(f.read())
 522         finally:
 523             f.close()
 524
 525 sys.exit(0)