source4/script/tests/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 FindBin qw($RealBin);
  10 use POSIX;
  11
  12 sub new($$$$) {
  13         my ($classname, $bindir, $ldap, $setupdir) = @_;
  14         my $self = { vars => {}, ldap => $ldap, bindir => $bindir, setupdir => $setupdir };
  15         bless $self;
  16         return $self;
  17 }
  18
  19 sub slapd_start($$)
  20 {
  21     my $count = 0;
  22         my ($self, $env_vars) = @_;
  23
  24         my $conf = $env_vars->{SLAPD_CONF};
  25         my $uri = $env_vars->{LDAP_URI};
  26
  27         # running slapd in the background means it stays in the same process group, so it can be
  28         # killed by timelimit
  29         if (defined($ENV{FEDORA_DS_PREFIX})) {
  30                 system("$ENV{FEDORA_DS_PREFIX}/sbin/ns-slapd -D $env_vars->{FEDORA_DS_DIR} -d$env_vars->{FEDORA_DS_LOGLEVEL} -i $env_vars->{FEDORA_DS_PIDFILE}> $env_vars->{LDAPDIR}/logs 2>&1 &");
  31         } else {
  32                 my $oldpath = $ENV{PATH};
  33                 $ENV{PATH} = "/usr/local/sbin:/usr/sbin:/sbin:$ENV{PATH}";
  34                 system("slapd -d$env_vars->{OPENLDAP_LOGLEVEL} -f $conf -h $uri > $env_vars->{LDAPDIR}/logs 2>&1 &");
  35                 $ENV{PATH} = $oldpath;
  36         }
  37         while (system("$self->{bindir}/ldbsearch -H $uri -s base -b \"\" supportedLDAPVersion > /dev/null") != 0) {
  38                 $count++;
  39                 if ($count > 10) {
  40                     $self->slapd_stop($env_vars);
  41                     return 0;
  42                 }
  43                 sleep(1);
  44         }
  45         return 1;
  46 }
  47
  48 sub slapd_stop($$)
  49 {
  50         my ($self, $envvars) = @_;
  51         if (defined($ENV{FEDORA_DS_PREFIX})) {
  52                 system("$envvars->{LDAPDIR}/slapd-samba4/stop-slapd");
  53         } else {
  54                 open(IN, "<$envvars->{OPENLDAP_PIDFILE}") or
  55                         die("unable to open slapd pid file: $envvars->{OPENLDAP_PIDFILE}");
  56                 kill 9, <IN>;
  57                 close(IN);
  58         }
  59 }
  60
  61 sub check_or_start($$$)
  62 {
  63         my ($self, $env_vars, $max_time) = @_;
  64         return 0 if ( -p $env_vars->{SMBD_TEST_FIFO});
  65
  66         # Start slapd before smbd
  67         if ($self->{ldap}) {
  68                 $self->slapd_start($env_vars) or
  69                         die("couldn't start slapd");
  70
  71                 print "LDAP PROVISIONING...";
  72                 $self->provision_ldap($env_vars);
  73         }
  74
  75         SocketWrapper::set_default_iface(1);
  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         my $valgrind = "";
  82         if (defined($ENV{SMBD_VALGRIND})) {
  83                 $valgrind = $ENV{SMBD_VALGRIND};
  84         }
  85
  86         print "STARTING SMBD... ";
  87         my $pid = fork();
  88         if ($pid == 0) {
  89                 open STDIN, $env_vars->{SMBD_TEST_FIFO};
  90                 open STDOUT, ">$env_vars->{SMBD_TEST_LOG}";
  91                 open STDERR, '>&STDOUT';
  92                 my $optarg = "";
  93                 if (defined($max_time)) {
  94                         $optarg = "--maximum-runtime=$max_time ";
  95                 }
  96                 my $ret = system("$valgrind $self->{bindir}/smbd $optarg -s $env_vars->{CONFFILE} -M single -i --leak-report-full");
  97                 if ($? == -1) {
  98                         print "Unable to start smbd: $ret: $!\n";
  99                         exit 1;
 100                 }
 101                 unlink($env_vars->{SMBD_TEST_FIFO});
 102                 my $exit = $? >> 8;
 103                 if ( $ret == 0 ) {
 104                         print "smbd exits with status $exit\n";
 105                 } elsif ( $ret & 127 ) {
 106                         print "smbd got signal ".($ret & 127)." and exits with $exit!\n";
 107                 } else {
 108                         $ret = $? >> 8;
 109                         print "smbd failed with status $exit!\n";
 110                 }
 111                 exit $exit;
 112         }
 113         print "DONE\n";
 114
 115         open(DATA, ">$env_vars->{SMBD_TEST_FIFO}");
 116
 117         return $pid;
 118 }
 119
 120 sub wait_for_start($$)
 121 {
 122         my ($self, $testenv_vars) = @_;
 123         # give time for nbt server to register its names
 124         print "delaying for nbt name registration\n";
 125
 126         # This will return quickly when things are up, but be slow if we
 127         # need to wait for (eg) SSL init
 128         system("bin/nmblookup $testenv_vars->{CONFIGURATION} $testenv_vars->{SERVER}");
 129         system("bin/nmblookup $testenv_vars->{CONFIGURATION} -U $testenv_vars->{SERVER} $testenv_vars->{SERVER}");
 130         system("bin/nmblookup $testenv_vars->{CONFIGURATION} $testenv_vars->{SERVER}");
 131         system("bin/nmblookup $testenv_vars->{CONFIGURATION} -U $testenv_vars->{SERVER} $testenv_vars->{NETBIOSNAME}");
 132         system("bin/nmblookup $testenv_vars->{CONFIGURATION} $testenv_vars->{NETBIOSNAME}");
 133         system("bin/nmblookup $testenv_vars->{CONFIGURATION} -U $testenv_vars->{SERVER} $testenv_vars->{NETBIOSNAME}");
 134         system("bin/nmblookup $testenv_vars->{CONFIGURATION} $testenv_vars->{NETBIOSNAME}");
 135         system("bin/nmblookup $testenv_vars->{CONFIGURATION} -U $testenv_vars->{SERVER} $testenv_vars->{NETBIOSNAME}");
 136 }
 137
 138 sub provision_member($$$)
 139 {
 140         my ($self, $prefix, $dcvars) = @_;
 141         my %ret = ();
 142         print "PROVISIONING...";
 143         open(IN, "$RealBin/mktestmember.sh $prefix $dcvars->{DOMAIN} $dcvars->{USERNAME} $dcvars->{PASSWORD}|") or die("Unable to setup");
 144         while (<IN>) {
 145                 die ("Error parsing `$_'") unless (/^([A-Z0-9a-z_]+)=(.*)$/);
 146                 $ret{$1} = $2;
 147         }
 148         close(IN);
 149
 150         $ret{SMBD_TEST_FIFO} = "$prefix/smbd_test.fifo";
 151         $ret{SMBD_TEST_LOG} = "$prefix/smbd_test.log";
 152         return \%ret;
 153 }
 154
 155 sub provision_dc($$)
 156 {
 157         my ($self, $prefix) = @_;
 158         my %ret = ();
 159         print "PROVISIONING...";
 160         open(IN, "$RealBin/mktestdc.sh $prefix|") or die("Unable to setup");
 161         while (<IN>) {
 162                 die ("Error parsing `$_'") unless (/^([A-Z0-9a-z_]+)=(.*)$/);
 163                 $ret{$1} = $2;
 164         }
 165         close(IN);
 166
 167         $ret{SMBD_TEST_FIFO} = "$prefix/smbd_test.fifo";
 168         $ret{SMBD_TEST_LOG} = "$prefix/smbd_test.log";
 169         return \%ret;
 170 }
 171
 172 sub provision_ldap($$)
 173 {
 174         my ($self, $envvars) = @_;
 175     system("$self->{bindir}/smbscript $self->{setupdir}/provision $envvars->{PROVISION_OPTIONS} \"$envvars->{PROVISION_ACI}\" --ldap-backend=$envvars->{LDAP_URI}") and
 176                 die("LDAP PROVISIONING failed: $self->{bindir}/smbscript $self->{setupdir}/provision $envvars->{PROVISION_OPTIONS} \"$envvars->{PROVISION_ACI}\" --ldap-backend=$envvars->{LDAP_URI}");
 177 }
 178
 179 sub teardown_env($$)
 180 {
 181         my ($self, $envvars) = @_;
 182
 183         close(DATA);
 184
 185         sleep(2);
 186
 187         my $failed = $? >> 8;
 188
 189         if (-f "$envvars->{PIDDIR}/smbd.pid" ) {
 190                 open(IN, "<$envvars->{PIDDIR}/smbd.pid") or die("unable to open smbd pid file");
 191                 kill 9, <IN>;
 192                 close(IN);
 193         }
 194
 195         $self->slapd_stop($envvars) if ($self->{ldap});
 196
 197         return $failed;
 198 }
 199
 200 sub setup_env($$$)
 201 {
 202         my ($self, $envname, $path) = @_;
 203
 204         if ($envname eq "dc") {
 205                 return $self->setup_dc("$path/dc");
 206         } elsif ($envname eq "member") {
 207                 if (not defined($self->{vars}->{dc})) {
 208                         $self->setup_dc("$path/dc");
 209                 }
 210                 return $self->setup_member("$path/member", $self->{vars}->{dc});
 211         } else {
 212                 die("Samba4 can't provide environment '$envname'");
 213         }
 214 }
 215
 216 sub setup_member($$$$)
 217 {
 218         my ($self, $path, $dc_vars) = @_;
 219
 220         my $env = $self->provision_member($path, $dc_vars);
 221
 222         $self->check_or_start($env, ($ENV{SMBD_MAX_TIME} or 5400));
 223
 224         $self->wait_for_start($env);
 225
 226         return $env;
 227 }
 228
 229 sub setup_dc($$)
 230 {
 231         my ($self, $path) = @_;
 232
 233         my $env = $self->provision_dc($path);
 234
 235         $self->check_or_start($env,
 236                 ($ENV{SMBD_MAX_TIME} or 5400));
 237
 238         $self->wait_for_start($env);
 239
 240         $self->{vars}->{dc} = $env;
 241
 242         return $env;
 243 }
 244
 245 sub stop($)
 246 {
 247         my ($self) = @_;
 248 }
 249
 250 1;