s4 smbd standard tests: limit forked processes
[amitay/samba.git] / selftest / target / Samba.pm
index 7bab462e0e6b4db740b0a6509b01db918abee00d..4820f987fd73c096ef24cd1251134fac4608a0ce 100644 (file)
@@ -22,33 +22,71 @@ sub new($$$$$) {
        return $self;
 }
 
+%Samba::ENV_DEPS = (%Samba3::ENV_DEPS, %Samba4::ENV_DEPS);
+our %ENV_DEPS;
+
+%Samba::ENV_TARGETS = (
+       (map { $_ => "Samba3" } keys %Samba3::ENV_DEPS),
+       (map { $_ => "Samba4" } keys %Samba4::ENV_DEPS),
+);
+our %ENV_TARGETS;
+
+%Samba::ENV_NEEDS_AD_DC = (
+       (map { $_ => 1 } keys %Samba4::ENV_DEPS)
+);
+our %ENV_NEEDS_AD_DC;
+foreach my $env (keys %Samba3::ENV_DEPS) {
+    $ENV_NEEDS_AD_DC{$env} = ($env =~ /^ad_/);
+}
+
 sub setup_env($$$)
 {
        my ($self, $envname, $path) = @_;
 
-       $ENV{ENVNAME} = $envname;
-
-       my $env = $self->{samba4}->setup_env($envname, $path);
-       if (defined($env) and $env ne "UNKNOWN") {
-           if (not defined($env->{target})) {
-               $env->{target} = $self->{samba4};
-           }
-       } elsif (defined($env) and $env eq "UNKNOWN") {
-               $env = $self->{samba3}->setup_env($envname, $path);
-               if (defined($env) and $env ne "UNKNOWN") {
-                   if (not defined($env->{target})) {
-                       $env->{target} = $self->{samba3};
-                   }
-               }
-       }
-       if (defined($env) and ($env eq "UNKNOWN")) {
+       my $targetname = $ENV_TARGETS{$envname};
+       if (not defined($targetname)) {
                warn("Samba can't provide environment '$envname'");
                return "UNKNOWN";
        }
-       if (not defined $env) {
+
+       my %targetlookup = (
+               "Samba3" => $self->{samba3},
+               "Samba4" => $self->{samba4}
+       );
+       my $target = $targetlookup{$targetname};
+
+       if (defined($target->{vars}->{$envname})) {
+               return $target->{vars}->{$envname};
+       }
+
+       my @dep_vars;
+       foreach(@{$ENV_DEPS{$envname}}) {
+               my $vars = $self->setup_env($_, $path);
+               if (defined($vars)) {
+                       push(@dep_vars, $vars);
+               } else {
+                       warn("Failed setting up $_ as a dependency of $envname");
+                       return undef;
+               }
+       }
+
+       $ENV{ENVNAME} = $envname;
+       # Avoid hitting system krb5.conf -
+       # An env that needs Kerberos will reset this to the real value.
+       $ENV{KRB5_CONFIG} = "$path/no_krb5.conf";
+
+       my $setup_name = $ENV_TARGETS{$envname}."::setup_".$envname;
+       my $setup_sub = \&$setup_name;
+       my $env = &$setup_sub($target, "$path/$envname", @dep_vars);
+
+       if (not defined($env)) {
                warn("failed to start up environment '$envname'");
                return undef;
        }
+
+       $target->{vars}->{$envname} = $env;
+       $target->{vars}->{$envname}->{target} = $target;
+
        return $env;
 }
 
@@ -56,9 +94,20 @@ sub bindir_path($$) {
        my ($object, $path) = @_;
 
        my $valpath = "$object->{bindir}/$path";
+       my $python_cmd = "";
+       my $result = $path;
+       if (defined $ENV{'PYTHON'}) {
+               $python_cmd = $ENV{'PYTHON'} . " ";
+       }
 
-       return $valpath if (-f $valpath or -d $valpath);
-       return $path;
+       if (-f $valpath or -d $valpath) {
+               $result = $valpath;
+       }
+       # make sure we prepend samba-tool with calling $PYTHON python version
+       if ($path eq "samba-tool") {
+               $result = $python_cmd . $result;
+       }
+       return $result;
 }
 
 sub nss_wrapper_winbind_so_path($) {
@@ -94,10 +143,14 @@ sub prepare_keyblobs($)
        my $dcdir = "$cadir/DCs/$dcdnsname";
        my $dccert = "$dcdir/DC-$dcdnsname-cert.pem";
        my $dckey_private = "$dcdir/DC-$dcdnsname-private-key.pem";
-       my $userprincipalname = "administrator\@$ctx->{dnsname}";
-       my $userdir = "$cadir/Users/$userprincipalname";
-       my $usercert = "$userdir/USER-$userprincipalname-cert.pem";
-       my $userkey_private = "$userdir/USER-$userprincipalname-private-key.pem";
+       my $adminprincipalname = "administrator\@$ctx->{dnsname}";
+       my $admindir = "$cadir/Users/$adminprincipalname";
+       my $admincert = "$admindir/USER-$adminprincipalname-cert.pem";
+       my $adminkey_private = "$admindir/USER-$adminprincipalname-private-key.pem";
+       my $pkinitprincipalname = "pkinit\@$ctx->{dnsname}";
+       my $pkinitdir = "$cadir/Users/$pkinitprincipalname";
+       my $pkinitcert = "$pkinitdir/USER-$pkinitprincipalname-cert.pem";
+       my $pkinitkey_private = "$pkinitdir/USER-$pkinitprincipalname-private-key.pem";
 
        my $tlsdir = "$ctx->{tlsdir}";
        my $pkinitdir = "$ctx->{prefix_abs}/pkinit";
@@ -107,8 +160,10 @@ sub prepare_keyblobs($)
        my $crlfile = "$tlsdir/crl.pem";
        my $certfile = "$tlsdir/cert.pem";
        my $keyfile = "$tlsdir/key.pem";
-       my $usercertfile = "$pkinitdir/USER-$userprincipalname-cert.pem";
-       my $userkeyfile = "$pkinitdir/USER-$userprincipalname-private-key.pem";
+       my $admincertfile = "$pkinitdir/USER-$adminprincipalname-cert.pem";
+       my $adminkeyfile = "$pkinitdir/USER-$adminprincipalname-private-key.pem";
+       my $pkinitcertfile = "$pkinitdir/USER-$pkinitprincipalname-cert.pem";
+       my $pkinitkeyfile = "$pkinitdir/USER-$pkinitprincipalname-private-key.pem";
 
        mkdir($tlsdir, 0700);
        mkdir($pkinitdir, 0700);
@@ -156,22 +211,18 @@ EOF
        copy_file_content(${cacrl_pem}, ${crlfile});
        copy_file_content(${dccert}, ${certfile});
        copy_file_content(${dckey_private}, ${keyfile});
-       if (-e ${userkey_private}) {
-               copy_file_content(${usercert}, ${usercertfile});
-               copy_file_content(${userkey_private}, ${userkeyfile});
+       if (-e ${adminkey_private}) {
+               copy_file_content(${admincert}, ${admincertfile});
+               copy_file_content(${adminkey_private}, ${adminkeyfile});
+       }
+       if (-e ${pkinitkey_private}) {
+               copy_file_content(${pkinitcert}, ${pkinitcertfile});
+               copy_file_content(${pkinitkey_private}, ${pkinitkeyfile});
        }
 
        # COMPAT stuff to be removed in a later commit
        my $kdccertfile = "$tlsdir/kdc.pem";
        copy_file_content(${dccert}, ${kdccertfile});
-       if (-e ${userkey_private}) {
-               my $adminkeyfile = "$tlsdir/adminkey.pem";
-               my $admincertfile = "$tlsdir/admincert.pem";
-               my $admincertupnfile = "$tlsdir/admincertupn.pem";
-               copy_file_content(${userkey_private}, ${adminkeyfile});
-               copy_file_content(${usercert}, ${admincertfile});
-               copy_file_content(${usercert}, ${admincertupnfile});
-       }
 
        umask $oldumask;
 }
@@ -199,9 +250,23 @@ sub mk_krb5_conf($$)
  ticket_lifetime = 24h
  forwardable = yes
  allow_weak_crypto = yes
+ # Set the grace clocskew to 5 seconds
+ # This is especially required by samba3.raw.session krb5 and
+ # reauth tests
+ clockskew = 5
+ # We are running on the same machine, do not correct
+ # system clock differences
+ kdc_timesync = 0
 
 ";
 
+       if (defined($ctx->{krb5_ccname})) {
+               print KRB5CONF "
+ default_ccache_name = $ctx->{krb5_ccname}
+";
+       }
+
+
         if (defined($ctx->{supported_enctypes})) {
                print KRB5CONF "
  default_etypes = $ctx->{supported_enctypes}
@@ -238,7 +303,7 @@ sub mk_realms_stanza($$$$)
 {
        my ($realm, $dnsname, $domain, $kdc_ipv4) = @_;
        my $lc_domain = lc($domain);
-       
+
        my $realms_stanza = "
  $realm = {
   kdc = $kdc_ipv4:88
@@ -265,6 +330,55 @@ sub mk_realms_stanza($$$$)
         return $realms_stanza;
 }
 
+sub mk_mitkdc_conf($$)
+{
+       # samba_kdb_dir is the path to mit_samba.so
+       my ($ctx, $samba_kdb_dir) = @_;
+
+       unless (open(KDCCONF, ">$ctx->{mitkdc_conf}")) {
+               warn("can't open $ctx->{mitkdc_conf}$?");
+               return undef;
+       }
+
+       print KDCCONF "
+# Generated kdc.conf for $ctx->{realm}
+
+[kdcdefaults]
+       kdc_ports = 88
+       kdc_tcp_ports = 88
+
+[realms]
+       $ctx->{realm} = {
+       }
+
+       $ctx->{dnsname} = {
+       }
+
+       $ctx->{domain} = {
+       }
+
+[dbmodules]
+       db_module_dir = $samba_kdb_dir
+
+       $ctx->{realm} = {
+               db_library = samba
+       }
+
+       $ctx->{dnsname} = {
+               db_library = samba
+       }
+
+       $ctx->{domain} = {
+               db_library = samba
+       }
+
+[logging]
+       kdc = FILE:$ctx->{logdir}/mit_kdc.log
+";
+
+       close(KDCCONF);
+}
+
 sub get_interface($)
 {
     my ($netbiosname) = @_;
@@ -281,6 +395,10 @@ sub get_interface($)
 
     # 11-16 used by selftest.pl for client interfaces
 
+    $interfaces{"addc_no_nss"} = 17;
+    $interfaces{"addc_no_ntlm"} = 18;
+    $interfaces{"idmapadmember"} = 19;
+    $interfaces{"idmapridmember"} = 20;
     $interfaces{"localdc"} = 21;
     $interfaces{"localvampiredc"} = 22;
     $interfaces{"s4member"} = 23;
@@ -298,10 +416,20 @@ sub get_interface($)
     $interfaces{"fileserver"} = 35;
     $interfaces{"fakednsforwarder1"} = 36;
     $interfaces{"fakednsforwarder2"} = 37;
+    $interfaces{"s4member_dflt"} = 38;
+    $interfaces{"vampire2000dc"} = 39;
+    $interfaces{"backupfromdc"} = 40;
+    $interfaces{"restoredc"} = 41;
+    $interfaces{"renamedc"} = 42;
+    $interfaces{"labdc"} = 43;
+    $interfaces{"offlinebackupdc"} = 44;
+    $interfaces{"customdc"} = 45;
+    $interfaces{"prockilldc"} = 46;
+    $interfaces{"proclimitdc"} = 47;
 
     # update lib/socket_wrapper/socket_wrapper.c
-    #  #define MAX_WRAPPED_INTERFACES 40
-    # if you wish to have more than 40 interfaces
+    #  #define MAX_WRAPPED_INTERFACES 64
+    # if you wish to have more than 64 interfaces
 
     if (not defined($interfaces{$netbiosname})) {
        die();
@@ -313,24 +441,31 @@ sub get_interface($)
 sub cleanup_child($$)
 {
     my ($pid, $name) = @_;
-    my $childpid = -1;
 
-    if (defined($pid)) {
-        $childpid = waitpid($pid, WNOHANG);
+    if (!defined($pid)) {
+        print STDERR "cleanup_child: pid not defined ... not calling waitpid\n";
+        return -1;
     }
 
+    my $childpid = waitpid($pid, WNOHANG);
+
     if ($childpid == 0) {
     } elsif ($childpid < 0) {
        printf STDERR "%s child process %d isn't here any more\n", $name, $pid;
        return $childpid;
-    }
-    elsif ($? & 127) {
+    } elsif ($? & 127) {
        printf STDERR "%s child process %d, died with signal %d, %s coredump\n",
-       $name, $childpid, ($? & 127),  ($? & 128) ? 'with' : 'without';
+               $name, $childpid, ($? & 127),  ($? & 128) ? 'with' : 'without';
     } else {
        printf STDERR "%s child process %d exited with value %d\n", $name, $childpid, $? >> 8;
     }
     return $childpid;
 }
 
+sub random_domain_sid()
+{
+       my $domain_sid = "S-1-5-21-". int(rand(4294967295)) . "-" . int(rand(4294967295)) . "-" . int(rand(4294967295));
+       return $domain_sid;
+}
+
 1;