dsdb/pytest/ldap: use idiomatic 'e' for exceptions
[samba.git] / selftest / selftest.pl
index 64fb9f85d3a5b60f1c3a249e30e4f604a3cdd83d..228bb29ecfe9891ce24effed0a9a4180bb1bca78 100755 (executable)
@@ -28,6 +28,7 @@ use lib "$RealBin";
 use Subunit;
 use SocketWrapper;
 use target::Samba;
+use Time::HiRes qw(time);
 
 eval {
 require Time::HiRes;
@@ -59,6 +60,7 @@ my $opt_libnss_wrapper_so_path = "";
 my $opt_libresolv_wrapper_so_path = "";
 my $opt_libsocket_wrapper_so_path = "";
 my $opt_libuid_wrapper_so_path = "";
+my $opt_libasan_so_path = "";
 my $opt_use_dns_faking = 0;
 my @testlists = ();
 
@@ -85,7 +87,18 @@ sub find_in_list($$)
 
 sub skip
 {
-       my ($name) = @_;
+       my ($name, $envname) = @_;
+       my ($env_basename, $env_localpart) = split(/:/, $envname);
+
+       if ($opt_target eq "samba3" && $Samba::ENV_NEEDS_AD_DC{$env_basename}) {
+               return "environment $envname is disabled as this build does not include an AD DC";
+       }
+
+       if (@opt_include_env && !(grep {$_ eq $env_basename} @opt_include_env)) {
+               return "environment $envname is disabled (via --include-env command line option) in this test run - skipping";
+       } elsif (@opt_exclude_env && grep {$_ eq $env_basename} @opt_exclude_env) {
+               return "environment $envname is disabled (via --exclude-env command line option) in this test run - skipping";
+       }
 
        return find_in_list(\@excludes, $name);
 }
@@ -139,9 +152,9 @@ sub run_testsuite($$$$$)
 
        Subunit::start_testsuite($name);
        Subunit::progress_push();
-       Subunit::report_time(time());
+       Subunit::report_time();
        system($cmd);
-       Subunit::report_time(time());
+       Subunit::report_time();
        Subunit::progress_pop();
 
        if ($? == -1) {
@@ -213,6 +226,7 @@ Preload cwrap:
  --resolv_wrapper_so_path=FILE the resolv_wrapper library to preload
  --socket_wrapper_so_path=FILE the socket_wrapper library to preload
  --uid_wrapper_so_path=FILE the uid_wrapper library to preload
+ --asan_so_path=FILE the asan library to preload
 
 DNS:
   --use-dns-faking          Fake DNS entries rather than talking to our
@@ -263,6 +277,7 @@ my $result = GetOptions (
                'resolv_wrapper_so_path=s' => \$opt_libresolv_wrapper_so_path,
                'socket_wrapper_so_path=s' => \$opt_libsocket_wrapper_so_path,
                'uid_wrapper_so_path=s' => \$opt_libuid_wrapper_so_path,
+               'asan_so_path=s' => \$opt_libasan_so_path,
                'use-dns-faking' => \$opt_use_dns_faking
            );
 
@@ -286,6 +301,9 @@ unless (defined($ENV{VALGRIND})) {
 # make all our python scripts unbuffered
 $ENV{PYTHONUNBUFFERED} = 1;
 
+# do not depend on the users setup
+$ENV{TZ} = "UTC";
+
 my $bindir_abs = abs_path($bindir);
 
 # Backwards compatibility:
@@ -366,6 +384,14 @@ if ($opt_socket_wrapper_pcap) {
 
 my $ld_preload = $ENV{LD_PRELOAD};
 
+if ($opt_libasan_so_path) {
+       if ($ld_preload) {
+               $ld_preload = "$ld_preload:$opt_libasan_so_path";
+       } else {
+               $ld_preload = "$opt_libasan_so_path";
+       }
+}
+
 if ($opt_libnss_wrapper_so_path) {
        if ($ld_preload) {
                $ld_preload = "$ld_preload:$opt_libnss_wrapper_so_path";
@@ -404,6 +430,9 @@ print "LD_PRELOAD=$ENV{LD_PRELOAD}\n";
 # Enable uid_wrapper globally
 $ENV{UID_WRAPPER} = 1;
 
+# We are already hitting the limit, so double it.
+$ENV{NSS_WRAPPER_MAX_HOSTENTS} = 200;
+
 # Disable RTLD_DEEPBIND hack for Samba bind dlz module
 #
 # This is needed in order to allow the ldb_*ldap module
@@ -449,15 +478,12 @@ if (defined($ENV{SMBD_MAXTIME}) and $ENV{SMBD_MAXTIME} ne "") {
     $server_maxtime = $ENV{SMBD_MAXTIME};
 }
 
+$target = new Samba($bindir, $ldap, $srcdir, $server_maxtime);
 unless ($opt_list) {
        if ($opt_target eq "samba") {
                $testenv_default = "ad_dc";
-               require target::Samba;
-               $target = new Samba($bindir, $ldap, $srcdir, $server_maxtime);
        } elsif ($opt_target eq "samba3") {
                $testenv_default = "nt4_member";
-               require target::Samba3;
-               $target = new Samba3($bindir, $srcdir_abs, $server_maxtime);
        }
 }
 
@@ -488,12 +514,10 @@ foreach (@opt_include) {
        push (@includes, read_test_regexes($_));
 }
 
-my $interfaces = join(',', ("127.0.0.11/8",
-                           "127.0.0.12/8",
-                           "127.0.0.13/8",
-                           "127.0.0.14/8",
-                           "127.0.0.15/8",
-                           "127.0.0.16/8"));
+# We give the selftest client 6 different IPv4 addresses to use. Most tests
+# only use the first (.11) IP. Note that winsreplication.c is one test that
+# uses the other IPs (search for iface_list_count()).
+my $interfaces = Samba::get_interfaces_config("client", 6);
 
 my $clientdir = "$prefix_abs/client";
 
@@ -683,11 +707,6 @@ $ENV{SELFTEST_PREFIX} = "$prefix_abs";
 $ENV{SELFTEST_TMPDIR} = "$tmpdir_abs";
 $ENV{TMPDIR} = "$tmpdir_abs";
 $ENV{TEST_DATA_PREFIX} = "$tmpdir_abs";
-if ($opt_socket_wrapper) {
-       $ENV{SELFTEST_INTERFACES} = $interfaces;
-} else {
-       $ENV{SELFTEST_INTERFACES} = "";
-}
 if ($opt_quick) {
        $ENV{SELFTEST_QUICK} = "1";
 } else {
@@ -725,7 +744,7 @@ $individual_tests = {};
 
 foreach my $testsuite (@available) {
        my $name = $$testsuite[0];
-       my $skipreason = skip($name);
+       my $skipreason = skip(@$testsuite);
        if (defined($restricted)) {
                # Find the testsuite for this test
                my $match = undef;
@@ -773,7 +792,7 @@ my $suitestotal = $#todo + 1;
 
 unless ($opt_list) {
        Subunit::progress($suitestotal);
-       Subunit::report_time(time());
+       Subunit::report_time();
 }
 
 my $i = 0;
@@ -792,112 +811,6 @@ sub get_running_env($)
        return $running_envs{$envname};
 }
 
-my @exported_envvars = (
-       # domain stuff
-       "DOMAIN",
-       "REALM",
-
-       # stuff related to a trusted domain
-       "TRUST_SERVER",
-       "TRUST_SERVER_IP",
-       "TRUST_SERVER_IPV6",
-       "TRUST_NETBIOSNAME",
-       "TRUST_USERNAME",
-       "TRUST_PASSWORD",
-       "TRUST_DOMAIN",
-       "TRUST_REALM",
-
-       # domain controller stuff
-       "DC_SERVER",
-       "DC_SERVER_IP",
-       "DC_SERVER_IPV6",
-       "DC_NETBIOSNAME",
-       "DC_NETBIOSALIAS",
-
-       # domain member
-       "MEMBER_SERVER",
-       "MEMBER_SERVER_IP",
-       "MEMBER_SERVER_IPV6",
-       "MEMBER_NETBIOSNAME",
-       "MEMBER_NETBIOSALIAS",
-
-       # rpc proxy controller stuff
-       "RPC_PROXY_SERVER",
-       "RPC_PROXY_SERVER_IP",
-       "RPC_PROXY_SERVER_IPV6",
-       "RPC_PROXY_NETBIOSNAME",
-       "RPC_PROXY_NETBIOSALIAS",
-
-       # domain controller stuff for Vampired DC
-       "VAMPIRE_DC_SERVER",
-       "VAMPIRE_DC_SERVER_IP",
-       "VAMPIRE_DC_SERVER_IPV6",
-       "VAMPIRE_DC_NETBIOSNAME",
-       "VAMPIRE_DC_NETBIOSALIAS",
-
-       # domain controller stuff for RODC
-       "RODC_DC_SERVER",
-       "RODC_DC_SERVER_IP",
-       "RODC_DC_SERVER_IPV6",
-       "RODC_DC_NETBIOSNAME",
-
-       # domain controller stuff for FL 2000 Vampired DC
-       "VAMPIRE_2000_DC_SERVER",
-       "VAMPIRE_2000_DC_SERVER_IP",
-       "VAMPIRE_2000_DC_SERVER_IPV6",
-       "VAMPIRE_2000_DC_NETBIOSNAME",
-       "VAMPIRE_2000_DC_NETBIOSALIAS",
-
-       "PROMOTED_DC_SERVER",
-       "PROMOTED_DC_SERVER_IP",
-       "PROMOTED_DC_SERVER_IPV6",
-       "PROMOTED_DC_NETBIOSNAME",
-       "PROMOTED_DC_NETBIOSALIAS",
-
-       # server stuff
-       "SERVER",
-       "SERVER_IP",
-       "SERVER_IPV6",
-       "NETBIOSNAME",
-       "NETBIOSALIAS",
-
-       # user stuff
-       "USERNAME",
-       "USERID",
-       "PASSWORD",
-       "DC_USERNAME",
-       "DC_PASSWORD",
-
-       # UID/GID for rfc2307 mapping tests
-       "UID_RFC2307TEST",
-       "GID_RFC2307TEST",
-
-       # misc stuff
-       "KRB5_CONFIG",
-       "KRB5CCNAME",
-       "SELFTEST_WINBINDD_SOCKET_DIR",
-       "NMBD_SOCKET_DIR",
-       "LOCAL_PATH",
-       "DNS_FORWARDER1",
-       "DNS_FORWARDER2",
-       "RESOLV_CONF",
-       "UNACCEPTABLE_PASSWORD",
-       "LOCK_DIR",
-       "SMBD_TEST_LOG",
-
-       # nss_wrapper
-       "NSS_WRAPPER_PASSWD",
-       "NSS_WRAPPER_GROUP",
-       "NSS_WRAPPER_HOSTS",
-       "NSS_WRAPPER_HOSTNAME",
-       "NSS_WRAPPER_MODULE_SO_PATH",
-       "NSS_WRAPPER_MODULE_FN_PREFIX",
-
-       # resolv_wrapper
-       "RESOLV_WRAPPER_CONF",
-       "RESOLV_WRAPPER_HOSTS",
-);
-
 sub sighandler($)
 {
        my $signame = shift;
@@ -931,6 +844,13 @@ sub setup_env($$)
 
        $option = "client" if $option eq "";
 
+       # Initially clear out the environment for the provision, so previous envs'
+       # variables don't leak in. Provisioning steps must explicitly set their
+       # necessary variables when calling out to other executables
+       Samba::clear_exported_envvars();
+       delete $ENV{SOCKET_WRAPPER_DEFAULT_IFACE};
+       delete $ENV{SMB_CONF_PATH};
+
        $ENV{KRB5CCNAME} = "FILE:${selftest_krbt_ccache_path}.${envname}/ignore";
 
        if (defined(get_running_env($envname))) {
@@ -947,7 +867,11 @@ sub setup_env($$)
                        $testenv_vars->{target} = $target;
                }
                if (not defined($testenv_vars)) {
-                       warn("$opt_target can't start up known environment '$envname'");
+                       if ($opt_one) {
+                               die("$opt_target can't start up known environment '$envname'");
+                       } else {
+                               warn("$opt_target can't start up known environment '$envname'");
+                       }
                }
        }
 
@@ -966,13 +890,8 @@ sub setup_env($$)
                die("Unknown option[$option] for envname[$envname]");
        }
 
-       foreach (@exported_envvars) {
-               if (defined($testenv_vars->{$_})) {
-                       $ENV{$_} = $testenv_vars->{$_};
-               } else {
-                       delete $ENV{$_};
-               }
-       }
+       # export the environment variables for the testenv (SERVER, SERVER_IP, etc)
+       Samba::export_envvars($testenv_vars);
 
        my $krb5_ccache_path = "${selftest_krbt_ccache_path}.${envname}.${option}";
        unlink($krb5_ccache_path);
@@ -980,19 +899,6 @@ sub setup_env($$)
        return $testenv_vars;
 }
 
-sub exported_envvars_str($)
-{
-       my ($testenv_vars) = @_;
-       my $out = "";
-
-       foreach (@exported_envvars) {
-               next unless defined($testenv_vars->{$_});
-               $out .= $_."=".$testenv_vars->{$_}."\n";
-       }
-
-       return $out;
-}
-
 sub getlog_env($)
 {
        my ($envname) = @_;
@@ -1041,7 +947,7 @@ if ($opt_testenv) {
        $ENV{PIDDIR} = $testenv_vars->{PIDDIR};
        $ENV{ENVNAME} = $testenv_name;
 
-       my $envvarstr = exported_envvars_str($testenv_vars);
+       my $envvarstr = Samba::exported_envvars_str($testenv_vars);
 
        my @term_args = ("echo -e \"
 Welcome to the Samba4 Test environment '$testenv_name'
@@ -1100,43 +1006,17 @@ $envvarstr
                my $cmd = $$_[2];
                my $name = $$_[0];
                my $envname = $$_[1];
-               my ($env_basename, $env_localpart) = split(/:/, $envname);
-               my $envvars = "SKIP";
+               my $envvars = setup_env($envname, $prefix);
 
-               if (@opt_include_env) {
-                   foreach my $env (@opt_include_env) {
-                       if ($env_basename eq $env) {
-                           $envvars = setup_env($envname, $prefix);
-                       }
-                   }
-               } elsif (@opt_exclude_env) {
-                   my $excluded = 0;
-                   foreach my $env (@opt_exclude_env) {
-                       if ($env_basename eq $env) {
-                           $excluded = 1;
-                       }
-                   }
-                   if ($excluded == 0) {
-                       $envvars = setup_env($envname, $prefix);
-                   }
-               } else {
-                   $envvars = setup_env($envname, $prefix);
-               }
-               
                if (not defined($envvars)) {
                        Subunit::start_testsuite($name);
                        Subunit::end_testsuite($name, "error",
                                "unable to set up environment $envname - exiting");
                        next;
-               } elsif ($envvars eq "SKIP") {
-                       Subunit::start_testsuite($name);
-                       Subunit::end_testsuite($name, "skip",
-                               "environment $envname is disabled (via --exclude-env / --include-env command line options) in this test run - skipping");
-                       next;
                } elsif ($envvars eq "UNKNOWN") {
                        Subunit::start_testsuite($name);
-                       Subunit::end_testsuite($name, "skip",
-                               "environment $envname is unknown in this test backend - skipping");
+                       Subunit::end_testsuite($name, "error",
+                               "environment $envname is unknown - exiting");
                        next;
                }