source/selftest/env/Samba4.pm

   1 #!/usr/bin/perl
   2 # Bootstrap Samba and run a number of tests against it.
   3 # Copyright (C) 2005-2007 Jelmer Vernooij <jelmer@samba.org>
   4 # Published under the GNU GPL, v3 or later.
   5
   6 package Samba4;
   7
   8 use strict;
   9 use Cwd qw(abs_path);
  10 use FindBin qw($RealBin);
  11 use POSIX;
  12
  13 sub new($$$$) {
  14         my ($classname, $bindir, $ldap, $setupdir) = @_;
  15         my $self = {
  16                 vars => {},
  17                 ldap => $ldap,
  18                 bindir => $bindir,
  19                 setupdir => $setupdir
  20         };
  21         bless $self;
  22         return $self;
  23 }
  24
  25 sub openldap_start($$$) {
  26         my ($slapd_conf, $uri, $logs) = @_;
  27         my $oldpath = $ENV{PATH};
  28         $ENV{PATH} = "/usr/local/sbin:/usr/sbin:/sbin:$ENV{PATH}";
  29         system("slapd -d0 -f $slapd_conf -h $uri > $logs 2>&1 &");
  30         $ENV{PATH} = $oldpath;
  31 }
  32
  33 sub slapd_start($$)
  34 {
  35         my $count = 0;
  36         my ($self, $env_vars) = @_;
  37
  38         my $uri = $env_vars->{LDAP_URI};
  39
  40         # running slapd in the background means it stays in the same process group, so it can be
  41         # killed by timelimit
  42         if ($self->{ldap} eq "fedora-ds") {
  43                 system("$ENV{FEDORA_DS_PREFIX}/sbin/ns-slapd -D $env_vars->{FEDORA_DS_DIR} -d0 -i $env_vars->{FEDORA_DS_PIDFILE}> $env_vars->{LDAPDIR}/logs 2>&1 &");
  44         } elsif ($self->{ldap} eq "openldap") {
  45                 openldap_start($env_vars->{SLAPD_CONF}, $uri, "$env_vars->{LDAPDIR}/logs");
  46         }
  47         while (system("$self->{bindir}/ldbsearch -H $uri -s base -b \"\" supportedLDAPVersion > /dev/null") != 0) {
  48                 $count++;
  49                 if ($count > 40) {
  50                     $self->slapd_stop($env_vars);
  51                     return 0;
  52                 }
  53                 sleep(1);
  54         }
  55         return 1;
  56 }
  57
  58 sub slapd_stop($$)
  59 {
  60         my ($self, $envvars) = @_;
  61         if ($self->{ldap} eq "fedora-ds") {
  62                 system("$envvars->{LDAPDIR}/slapd-samba4/stop-slapd");
  63         } elsif ($self->{ldap} eq "openldap") {
  64                 open(IN, "<$envvars->{OPENLDAP_PIDFILE}") or
  65                         die("unable to open slapd pid file: $envvars->{OPENLDAP_PIDFILE}");
  66                 kill 9, <IN>;
  67                 close(IN);
  68         }
  69         return 1;
  70 }
  71
  72 sub check_or_start($$$)
  73 {
  74         my ($self, $env_vars, $max_time) = @_;
  75         return 0 if ( -p $env_vars->{SMBD_TEST_FIFO});
  76
  77         unlink($env_vars->{SMBD_TEST_FIFO});
  78         POSIX::mkfifo($env_vars->{SMBD_TEST_FIFO}, 0700);
  79         unlink($env_vars->{SMBD_TEST_LOG});
  80
  81         print "STARTING SMBD... ";
  82         my $pid = fork();
  83         if ($pid == 0) {
  84                 open STDIN, $env_vars->{SMBD_TEST_FIFO};
  85                 open STDOUT, ">$env_vars->{SMBD_TEST_LOG}";
  86                 open STDERR, '>&STDOUT';
  87
  88                 SocketWrapper::set_default_iface($env_vars->{SOCKET_WRAPPER_DEFAULT_IFACE});
  89
  90                 my $valgrind = "";
  91                 if (defined($ENV{SMBD_VALGRIND})) {
  92                     $valgrind = $ENV{SMBD_VALGRIND};
  93                 }
  94
  95                 $ENV{KRB5_CONFIG} = $env_vars->{KRB5_CONFIG};
  96
  97                 # Start slapd before smbd, but with the fifo on stdin
  98                 if (defined($self->{ldap})) {
  99                     $self->slapd_start($env_vars) or
 100                         die("couldn't start slapd (2nd time)");
 101                 }
 102
 103                 my $optarg = "";
 104                 if (defined($max_time)) {
 105                         $optarg = "--maximum-runtime=$max_time ";
 106                 }
 107                 my $ret = system("$valgrind $self->{bindir}/smbd $optarg $env_vars->{CONFIGURATION} -M single -i --leak-report-full");
 108                 if ($? == -1) {
 109                         print "Unable to start smbd: $ret: $!\n";
 110                         exit 1;
 111                 }
 112                 unlink($env_vars->{SMBD_TEST_FIFO});
 113                 my $exit = $? >> 8;
 114                 if ( $ret == 0 ) {
 115                         print "smbd exits with status $exit\n";
 116                 } elsif ( $ret & 127 ) {
 117                         print "smbd got signal ".($ret & 127)." and exits with $exit!\n";
 118                 } else {
 119                         $ret = $? >> 8;
 120                         print "smbd failed with status $exit!\n";
 121                 }
 122                 exit $exit;
 123         }
 124         print "DONE\n";
 125
 126         open(DATA, ">$env_vars->{SMBD_TEST_FIFO}");
 127
 128         return $pid;
 129 }
 130
 131 sub wait_for_start($$)
 132 {
 133         my ($self, $testenv_vars) = @_;
 134         # give time for nbt server to register its names
 135         print "delaying for nbt name registration\n";
 136         sleep 2;
 137
 138         # This will return quickly when things are up, but be slow if we
 139         # need to wait for (eg) SSL init
 140         system("bin/nmblookup $testenv_vars->{CONFIGURATION} $testenv_vars->{SERVER}");
 141         system("bin/nmblookup $testenv_vars->{CONFIGURATION} -U $testenv_vars->{SERVER_IP} $testenv_vars->{SERVER}");
 142         system("bin/nmblookup $testenv_vars->{CONFIGURATION} $testenv_vars->{NETBIOSNAME}");
 143         system("bin/nmblookup $testenv_vars->{CONFIGURATION} -U $testenv_vars->{SERVER_IP} $testenv_vars->{NETBIOSNAME}");
 144         system("bin/nmblookup $testenv_vars->{CONFIGURATION} $testenv_vars->{NETBIOSALIAS}");
 145         system("bin/nmblookup $testenv_vars->{CONFIGURATION} -U $testenv_vars->{SERVER_IP} $testenv_vars->{NETBIOSALIAS}");
 146         system("bin/nmblookup $testenv_vars->{CONFIGURATION} $testenv_vars->{SERVER}");
 147         system("bin/nmblookup $testenv_vars->{CONFIGURATION} -U $testenv_vars->{SERVER_IP} $testenv_vars->{SERVER}");
 148         system("bin/nmblookup $testenv_vars->{CONFIGURATION} $testenv_vars->{NETBIOSNAME}");
 149         system("bin/nmblookup $testenv_vars->{CONFIGURATION} -U $testenv_vars->{SERVER_IP} $testenv_vars->{NETBIOSNAME}");
 150         system("bin/nmblookup $testenv_vars->{CONFIGURATION} $testenv_vars->{NETBIOSALIAS}");
 151         system("bin/nmblookup $testenv_vars->{CONFIGURATION} -U $testenv_vars->{SERVER_IP} $testenv_vars->{NETBIOSALIAS}");
 152
 153         print $self->getlog_env($testenv_vars);
 154 }
 155
 156 sub write_ldb_file($$$)
 157 {
 158         my ($self, $file, $ldif) = @_;
 159
 160         open(LDIF, "|$self->{bindir}/ldbadd -H $file >/dev/null");
 161         print LDIF $ldif;
 162         return close(LDIF);
 163 }
 164
 165 sub add_wins_config($$)
 166 {
 167         my ($self, $privatedir) = @_;
 168
 169         return $self->write_ldb_file("$privatedir/wins_config.ldb", "
 170 dn: name=TORTURE_6,CN=PARTNERS
 171 objectClass: wreplPartner
 172 name: TORTURE_6
 173 address: 127.0.0.6
 174 pullInterval: 0
 175 pushChangeCount: 0
 176 type: 0x3
 177 ");
 178 }
 179
 180 sub mk_fedora_ds($$$)
 181 {
 182         my ($self, $ldapdir, $configuration) = @_;
 183
 184         my $fedora_ds_inf = "$ldapdir/fedorads.inf";
 185         my $fedora_ds_extra_ldif = "$ldapdir/fedorads-partitions.ldif";
 186
 187         #Make the subdirectory be as fedora DS would expect
 188         my $fedora_ds_dir = "$ldapdir/slapd-samba4";
 189
 190         my $pidfile = "$fedora_ds_dir/logs/slapd-samba4.pid";
 191
 192 my $dir = getcwd();
 193 chdir "$ENV{FEDORA_DS_PREFIX}/bin" || die;
 194         if (system("perl $ENV{FEDORA_DS_PREFIX}/bin/ds_newinst.pl $fedora_ds_inf >&2") != 0) {
 195             chdir $dir;
 196             die("perl $ENV{FEDORA_DS_PREFIX}/bin/ds_newinst.pl $fedora_ds_inf FAILED: $?");
 197         }
 198         chdir $dir || die;
 199
 200         system("cat $fedora_ds_extra_ldif >> $fedora_ds_dir/dse.ldif");
 201
 202         system("$self->{bindir}/ad2oLschema $configuration -H $ldapdir/schema-tmp.ldb --option=convert:target=fedora-ds -I $self->{setupdir}/schema-map-fedora-ds-1.0 -O $fedora_ds_dir/schema/99_ad.ldif >&2") == 0 or die("schema conversion for Fedora DS failed");
 203
 204         return ($fedora_ds_dir, $pidfile);
 205 }
 206
 207 sub mk_openldap($$$)
 208 {
 209         my ($self, $ldapdir, $configuration) = @_;
 210
 211         my $slapd_conf = "$ldapdir/slapd.conf";
 212         my $pidfile = "$ldapdir/slapd.pid";
 213         my $modconf = "$ldapdir/modules.conf";
 214
 215         #This uses the backend provision we just did, to read out the schema
 216         system("$self->{bindir}/ad2oLschema $configuration --option=convert:target=openldap -H $ldapdir/schema-tmp.ldb -I $self->{setupdir}/schema-map-openldap-2.3 -O $ldapdir/backend-schema.schema >&2") == 0 or die("schema conversion for OpenLDAP failed");
 217
 218         my $oldpath = $ENV{PATH};
 219         $ENV{PATH} = "$ENV{OPENLDAP_PATH}/usr/local/sbin:/usr/sbin:/sbin:$ENV{PATH}";
 220
 221         unlink($modconf);
 222         open(CONF, ">$modconf"); close(CONF);
 223
 224         if (system("slaptest -u -f $slapd_conf >&2") != 0) {
 225                 open(CONF, ">$modconf");
 226                 # enable slapd modules
 227                 print CONF "
 228 moduleload      back_hdb
 229 moduleload      syncprov
 230 ";
 231                 close(CONF);
 232         }
 233
 234         if (system("slaptest -u -f $slapd_conf >&2") != 0) {
 235                 open(CONF, ">$modconf");
 236                 # enable slapd modules
 237                 print CONF "
 238 modulepath      /usr/lib/ldap
 239 moduleload      back_hdb
 240 moduleload      syncprov
 241 ";
 242                 close(CONF);
 243         }
 244
 245         if (system("slaptest -u -f $slapd_conf >&2") != 0) {
 246                 open(CONF, ">$modconf");
 247                 # enable slapd modules (Fedora layout)
 248                 print CONF "
 249 modulepath      /usr/lib/openldap
 250 moduleload      syncprov
 251 ";
 252                 close(CONF);
 253         }
 254
 255         if (system("slaptest -u -f $slapd_conf >&2") != 0) {
 256                 open(CONF, ">$modconf");
 257                 # enable slapd modules (Fedora x86_64 layout)
 258                 print CONF "
 259 modulepath      /usr/lib64/openldap
 260 moduleload      syncprov
 261 ";
 262                 close(CONF);
 263         }
 264
 265         system("slaptest -u -f $slapd_conf") == 0 or die("slaptest still fails after adding modules");
 266
 267
 268         $ENV{PATH} = $oldpath;
 269
 270         return ($slapd_conf, $pidfile);
 271 }
 272
 273 sub provision($$$$$$)
 274 {
 275         my ($self, $prefix, $server_role, $netbiosname, $netbiosalias, $swiface, $password) = @_;
 276
 277         my $smbd_loglevel = 1;
 278         my $username = "administrator";
 279         my $domain = "SAMBADOMAIN";
 280         my $realm = "SAMBA.EXAMPLE.COM";
 281         my $dnsname = "samba.example.com";
 282         my $basedn = "dc=samba,dc=example,dc=com";
 283         my $root = ($ENV{USER} or $ENV{LOGNAME} or `whoami`);
 284         my $srcdir="$RealBin/..";
 285         -d $prefix or mkdir($prefix, 0777) or die("Unable to create $prefix");
 286         my $prefix_abs = abs_path($prefix);
 287         my $tmpdir = "$prefix_abs/tmp";
 288         my $etcdir = "$prefix_abs/etc";
 289         my $piddir = "$prefix_abs/pid";
 290         my $conffile = "$etcdir/smb.conf";
 291         my $krb5_config = "$etcdir/krb5.conf";
 292         my $privatedir = "$prefix_abs/private";
 293         my $ncalrpcdir = "$prefix_abs/ncalrpc";
 294         my $lockdir = "$prefix_abs/lockdir";
 295         my $winbindd_socket_dir = "$prefix_abs/winbind_socket";
 296         my $winbindd_priv_pipe_dir = "$piddir/smbd.tmp/winbind_pipe";
 297
 298         my $configuration = "--configfile=$conffile";
 299         my $ldapdir = "$privatedir/ldap";
 300
 301         my $tlsdir = "$privatedir/tls";
 302
 303         my $ifaceipv4 = "127.0.0.$swiface";
 304         my $interfaces = "$ifaceipv4/8";
 305
 306         (system("rm -rf $prefix/*") == 0) or die("Unable to clean up");
 307         mkdir($_, 0777) foreach ($privatedir, $etcdir, $piddir, $ncalrpcdir, $lockdir,
 308                 $tmpdir);
 309
 310
 311         my $localbasedn = $basedn;
 312         $localbasedn = "DC=$netbiosname" if $server_role eq "member server";
 313
 314         open(CONFFILE, ">$conffile");
 315         print CONFFILE "
 316 [global]
 317         netbios name = $netbiosname
 318         netbios aliases = $netbiosalias
 319         workgroup = $domain
 320         realm = $realm
 321         private dir = $privatedir
 322         pid directory = $piddir
 323         ncalrpc dir = $ncalrpcdir
 324         lock dir = $lockdir
 325         setup directory = $self->{setupdir}
 326         js include = $srcdir/scripting/libjs
 327         winbindd socket directory = $winbindd_socket_dir
 328         winbind separator = /
 329         name resolve order = bcast
 330         interfaces = $interfaces
 331         tls dh params file = $tlsdir/dhparms.pem
 332         panic action = $srcdir/script/gdb_backtrace \%PID% \%PROG%
 333         wins support = yes
 334         server role = $server_role
 335         max xmit = 32K
 336         server max protocol = SMB2
 337         notify:inotify = false
 338         ldb:nosync = true
 339         system:anonymous = true
 340 #We don't want to pass our self-tests if the PAC code is wrong
 341         gensec:require_pac = true
 342         log level = $smbd_loglevel
 343
 344 [tmp]
 345         path = $tmpdir
 346         read only = no
 347         ntvfs handler = posix
 348         posix:sharedelay = 100000
 349         posix:eadb = $lockdir/eadb.tdb
 350
 351 [cifs]
 352         read only = no
 353         ntvfs handler = cifs
 354         cifs:server = $netbiosname
 355         cifs:share = tmp
 356 #There is no username specified here, instead the client is expected
 357 #to log in with kerberos, and smbd will used delegated credentials.
 358
 359 [simple]
 360         path = $tmpdir
 361         read only = no
 362         ntvfs handler = simple
 363
 364 [cifsposix]
 365         copy = simple
 366         ntvfs handler = cifsposix
 367 ";
 368         close(CONFFILE);
 369
 370         die ("Unable to create key blobs") if
 371                 (system("TLSDIR=$tlsdir $RealBin/mk-keyblobs.sh") != 0);
 372
 373         open(KRB5CONF, ">$krb5_config");
 374         print KRB5CONF "
 375 #Generated krb5.conf for $realm
 376
 377 [libdefaults]
 378  default_realm = $realm
 379  dns_lookup_realm = false
 380  dns_lookup_kdc = false
 381  ticket_lifetime = 24h
 382  forwardable = yes
 383
 384 [realms]
 385  $realm = {
 386   kdc = 127.0.0.1:88
 387   admin_server = 127.0.0.1:88
 388   default_domain = $dnsname
 389  }
 390  $dnsname = {
 391   kdc = 127.0.0.1:88
 392   admin_server = 127.0.0.1:88
 393   default_domain = $dnsname
 394  }
 395  $domain = {
 396   kdc = 127.0.0.1:88
 397   admin_server = 127.0.0.1:88
 398   default_domain = $dnsname
 399  }
 400
 401 [appdefaults]
 402         pkinit_anchors = FILE:$tlsdir/ca.pem
 403
 404 [kdc]
 405         enable-pkinit = true
 406         pkinit_identity = FILE:$tlsdir/kdc.pem,$tlsdir/key.pem
 407         pkinit_anchors = FILE:$tlsdir/ca.pem
 408
 409 [domain_realm]
 410  .$dnsname = $realm
 411 ";
 412         close(KRB5CONF);
 413
 414 #Ensure the config file is valid before we start
 415         if (system("$self->{bindir}/testparm $configuration -v --suppress-prompt >/dev/null 2>&1") != 0) {
 416                 system("$self->{bindir}/testparm -v --suppress-prompt $configuration >&2");
 417                 die("Failed to create a valid smb.conf configuration!");
 418         }
 419
 420         (system("($self->{bindir}/testparm $configuration -v --suppress-prompt --parameter-name=\"netbios name\" --section-name=global 2> /dev/null | grep -i \"^$netbiosname\" ) >/dev/null 2>&1") == 0) or die("Failed to create a valid smb.conf configuration!");
 421
 422 my @provision_options = ("$self->{bindir}/smbscript", "$self->{setupdir}/provision");
 423         push (@provision_options, split(' ', $configuration));
 424         push (@provision_options, "--host-name=$netbiosname");
 425         push (@provision_options, "--host-ip=$ifaceipv4");
 426         push (@provision_options, "--quiet");
 427         push (@provision_options, "--domain=$domain");
 428         push (@provision_options, "--realm=$realm");
 429         push (@provision_options, "--adminpass=$password");
 430         push (@provision_options, "--krbtgtpass=krbtgt$password");
 431         push (@provision_options, "--machinepass=machine$password");
 432         push (@provision_options, "--root=$root");
 433         push (@provision_options, "--simple-bind-dn=cn=Manager,$localbasedn");
 434         push (@provision_options, "--password=$password");
 435         push (@provision_options, "--root=$root");
 436         push (@provision_options, "--server-role=$server_role");
 437
 438         my $ldap_uri= "$ldapdir/ldapi";
 439         $ldap_uri =~ s|/|%2F|g;
 440         $ldap_uri = "ldapi://$ldap_uri";
 441
 442         my $ret = {
 443                 KRB5_CONFIG => $krb5_config,
 444                 PIDDIR => $piddir,
 445                 SERVER => $netbiosname,
 446                 SERVER_IP => $ifaceipv4,
 447                 NETBIOSNAME => $netbiosname,
 448                 NETBIOSALIAS => $netbiosalias,
 449                 LDAP_URI => $ldap_uri,
 450                 DOMAIN => $domain,
 451                 USERNAME => $username,
 452                 REALM => $realm,
 453                 PASSWORD => $password,
 454                 LDAPDIR => $ldapdir,
 455                 WINBINDD_SOCKET_DIR => $winbindd_socket_dir,
 456                 WINBINDD_PRIV_PIPE_DIR => $winbindd_priv_pipe_dir,
 457                 NCALRPCDIR => $ncalrpcdir,
 458                 LOCKDIR => $lockdir,
 459                 CONFIGURATION => $configuration,
 460                 SOCKET_WRAPPER_DEFAULT_IFACE => $swiface
 461         };
 462
 463         if (defined($self->{ldap})) {
 464
 465                 push (@provision_options, "--ldap-backend=$ldap_uri");
 466                 system("$self->{bindir}/smbscript $self->{setupdir}/provision-backend $configuration --ldap-manager-pass=$password --root=$root --realm=$realm --host-name=$netbiosname --ldap-backend-type=$self->{ldap}>&2") == 0 or die("backend provision failed");
 467
 468                 if ($self->{ldap} eq "openldap") {
 469                        ($ret->{SLAPD_CONF}, $ret->{OPENLDAP_PIDFILE}) = $self->mk_openldap($ldapdir, $configuration) or die("Unable to create openldap directories");
 470                 } elsif ($self->{ldap} eq "fedora-ds") {
 471                        ($ret->{FEDORA_DS_DIR}, $ret->{FEDORA_DS_PIDFILE}) = $self->mk_fedora_ds($ldapdir, $configuration) or die("Unable to create fedora ds directories");
 472                        push (@provision_options, "--ldap-module=nsuniqueid");
 473                        push (@provision_options, "--aci=aci:: KHRhcmdldGF0dHIgPSAiKiIpICh2ZXJzaW9uIDMuMDthY2wgImZ1bGwgYWNjZXNzIHRvIGFsbCBieSBhbGwiO2FsbG93IChhbGwpKHVzZXJkbiA9ICJsZGFwOi8vL2FueW9uZSIpOykK");
 474                  }
 475
 476                 $self->slapd_start($ret) or
 477                         die("couldn't start slapd");
 478         }
 479
 480         (system(@provision_options) == 0) or die("Unable to provision");
 481
 482         if (defined($self->{ldap})) {
 483                 $self->slapd_stop($ret) or
 484                         die("couldn't stop slapd");
 485         }
 486
 487         return $ret;
 488 }
 489
 490 sub provision_member($$$)
 491 {
 492         my ($self, $prefix, $dcvars) = @_;
 493         print "PROVISIONING MEMBER...";
 494
 495         my $ret = $self->provision($prefix,
 496                                    "member server",
 497                                    "localmember3",
 498                                    "localmember",
 499                                    3,
 500                                    "localmemberpass");
 501
 502         $ret or die("Unable to provision");
 503
 504         my $cmd = "";
 505         $cmd .= "SOCKET_WRAPPER_DEFAULT_IFACE=\"$ret->{SOCKET_WRAPPER_DEFAULT_IFACE}\" ";
 506         $cmd .= "KRB5_CONFIG=\"$ret->{KRB5_CONFIG}\" ";
 507         $cmd .= "$self->{bindir}/net join $ret->{CONFIGURATION} $dcvars->{DOMAIN} member";
 508         $cmd .= " -U$dcvars->{USERNAME}\%$dcvars->{PASSWORD}";
 509
 510         system($cmd) == 0 or die("Join failed\n$cmd");
 511
 512         $ret->{SMBD_TEST_FIFO} = "$prefix/smbd_test.fifo";
 513         $ret->{SMBD_TEST_LOG} = "$prefix/smbd_test.log";
 514         $ret->{SMBD_TEST_LOG_POS} = 0;
 515
 516         $ret->{DC_SERVER} = $dcvars->{SERVER};
 517         $ret->{DC_SERVER_IP} = $dcvars->{SERVER_IP};
 518         $ret->{DC_NETBIOSNAME} = $dcvars->{NETBIOSNAME};
 519         $ret->{DC_NETBIOSALIAS} = $dcvars->{NETBIOSALIAS};
 520         $ret->{DC_USERNAME} = $dcvars->{USERNAME};
 521         $ret->{DC_PASSWORD} = $dcvars->{PASSWORD};
 522
 523         return $ret;
 524 }
 525
 526 sub provision_dc($$)
 527 {
 528         my ($self, $prefix) = @_;
 529
 530         print "PROVISIONING DC...";
 531         my $ret = $self->provision($prefix,
 532                                    "domain controller",
 533                                    "localdc1",
 534                                    "localdc",
 535                                    1,
 536                                    "localdcpass");
 537
 538         $self->add_wins_config("$prefix/private") or
 539                 die("Unable to add wins configuration");
 540
 541         $ret->{SMBD_TEST_FIFO} = "$prefix/smbd_test.fifo";
 542         $ret->{SMBD_TEST_LOG} = "$prefix/smbd_test.log";
 543         $ret->{SMBD_TEST_LOG_POS} = 0;
 544         return $ret;
 545 }
 546
 547 sub teardown_env($$)
 548 {
 549         my ($self, $envvars) = @_;
 550         my $pid;
 551
 552         close(DATA);
 553
 554         if (-f "$envvars->{PIDDIR}/smbd.pid" ) {
 555                 open(IN, "<$envvars->{PIDDIR}/smbd.pid") or die("unable to open smbd pid file");
 556                 $pid = <IN>;
 557                 close(IN);
 558
 559                 # Give the process 20 seconds to exit.  gcov needs
 560                 # this time to write out the covarge data
 561                 my $count = 0;
 562                 until (kill(0, $pid) == 0) {
 563                     # if no process sucessfully signalled, then we are done
 564                     sleep(1);
 565                     $count++;
 566                     last if $count > 20;
 567                 }
 568
 569                 # If it is still around, kill it
 570                 if ($count > 20) {
 571                     print "smbd process $pid took more than $count seconds to exit, killing\n";
 572                     kill 9, $pid;
 573                 }
 574         }
 575
 576         my $failed = $? >> 8;
 577
 578         $self->slapd_stop($envvars) if ($self->{ldap});
 579
 580         print $self->getlog_env($envvars);
 581
 582         return $failed;
 583 }
 584
 585 sub getlog_env($$)
 586 {
 587         my ($self, $envvars) = @_;
 588         my $title = "SMBD LOG of: $envvars->{NETBIOSNAME}\n";
 589         my $out = $title;
 590
 591         open(LOG, "<$envvars->{SMBD_TEST_LOG}");
 592
 593         seek(LOG, $envvars->{SMBD_TEST_LOG_POS}, SEEK_SET);
 594         while (<LOG>) {
 595                 $out .= $_;
 596         }
 597         $envvars->{SMBD_TEST_LOG_POS} = tell(LOG);
 598         close(LOG);
 599
 600         return "" if $out eq $title;
 601
 602         return $out;
 603 }
 604
 605 sub check_env($$)
 606 {
 607         my ($self, $envvars) = @_;
 608
 609         return 1 if (-p $envvars->{SMBD_TEST_FIFO});
 610
 611         print $self->getlog_env($envvars);
 612
 613         return 0;
 614 }
 615
 616 sub setup_env($$$)
 617 {
 618         my ($self, $envname, $path) = @_;
 619
 620         if ($envname eq "dc") {
 621                 return $self->setup_dc("$path/dc");
 622         } elsif ($envname eq "member") {
 623                 if (not defined($self->{vars}->{dc})) {
 624                         $self->setup_dc("$path/dc");
 625                 }
 626                 return $self->setup_member("$path/member", $self->{vars}->{dc});
 627         } else {
 628                 die("Samba4 can't provide environment '$envname'");
 629         }
 630 }
 631
 632 sub setup_member($$$$)
 633 {
 634         my ($self, $path, $dc_vars) = @_;
 635
 636         my $env = $self->provision_member($path, $dc_vars);
 637
 638         $self->check_or_start($env, ($ENV{SMBD_MAXTIME} or 7500));
 639
 640         $self->wait_for_start($env);
 641
 642         return $env;
 643 }
 644
 645 sub setup_dc($$)
 646 {
 647         my ($self, $path) = @_;
 648
 649         my $env = $self->provision_dc($path);
 650
 651         $self->check_or_start($env,
 652                 ($ENV{SMBD_MAXTIME} or 7500));
 653
 654         $self->wait_for_start($env);
 655
 656         $self->{vars}->{dc} = $env;
 657
 658         return $env;
 659 }
 660
 661 sub stop($)
 662 {
 663         my ($self) = @_;
 664 }
 665
 666 1;