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, "SERVER_ROLE=\"member server\" $RealBin/mksamba4server.sh $prefix|") 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         system("$self->{bindir}/net join $ret{CONFIGURATION} $dcvars->{DOMAIN} member -U$dcvars->{USERNAME}\%$dcvars->{PASSWORD}") or die("Join failed");
 151
 152         $ret{SMBD_TEST_FIFO} = "$prefix/smbd_test.fifo";
 153         $ret{SMBD_TEST_LOG} = "$prefix/smbd_test.log";
 154         print "$ret{DOMAIN}\n";
 155         return \%ret;
 156 }
 157
 158 sub provision_dc($$)
 159 {
 160         my ($self, $prefix) = @_;
 161         my %ret = ();
 162         print "PROVISIONING...";
 163         open(IN, "$RealBin/mksamba4server.sh $prefix|") or die("Unable to setup");
 164         while (<IN>) {
 165                 die ("Error parsing `$_'") unless (/^([A-Z0-9a-z_]+)=(.*)$/);
 166                 $ret{$1} = $2;
 167         }
 168         close(IN);
 169
 170         $ret{SMBD_TEST_FIFO} = "$prefix/smbd_test.fifo";
 171         $ret{SMBD_TEST_LOG} = "$prefix/smbd_test.log";
 172         return \%ret;
 173 }
 174
 175 sub provision_ldap($$)
 176 {
 177         my ($self, $envvars) = @_;
 178         system("$self->{bindir}/smbscript $self->{setupdir}/provision $envvars->{PROVISION_OPTIONS} \"$envvars->{PROVISION_ACI}\" --ldap-backend=$envvars->{LDAP_URI}") and
 179                 die("LDAP PROVISIONING failed: $self->{bindir}/smbscript $self->{setupdir}/provision $envvars->{PROVISION_OPTIONS} \"$envvars->{PROVISION_ACI}\" --ldap-backend=$envvars->{LDAP_URI}");
 180 }
 181
 182 sub teardown_env($$)
 183 {
 184         my ($self, $envvars) = @_;
 185
 186         close(DATA);
 187
 188         sleep(2);
 189
 190         my $failed = $? >> 8;
 191
 192         if (-f "$envvars->{PIDDIR}/smbd.pid" ) {
 193                 open(IN, "<$envvars->{PIDDIR}/smbd.pid") or die("unable to open smbd pid file");
 194                 kill 9, <IN>;
 195                 close(IN);
 196         }
 197
 198         $self->slapd_stop($envvars) if ($self->{ldap});
 199
 200         return $failed;
 201 }
 202
 203 sub setup_env($$$)
 204 {
 205         my ($self, $envname, $path) = @_;
 206
 207         if ($envname eq "dc") {
 208                 return $self->setup_dc("$path/dc");
 209         } elsif ($envname eq "member") {
 210                 if (not defined($self->{vars}->{dc})) {
 211                         $self->setup_dc("$path/dc");
 212                 }
 213                 return $self->setup_member("$path/member", $self->{vars}->{dc});
 214         } else {
 215                 die("Samba4 can't provide environment '$envname'");
 216         }
 217 }
 218
 219 sub setup_member($$$$)
 220 {
 221         my ($self, $path, $dc_vars) = @_;
 222
 223         my $env = $self->provision_member($path, $dc_vars);
 224
 225         $self->check_or_start($env, ($ENV{SMBD_MAX_TIME} or 5400));
 226
 227         $self->wait_for_start($env);
 228
 229         return $env;
 230 }
 231
 232 sub setup_dc($$)
 233 {
 234         my ($self, $path) = @_;
 235
 236         my $env = $self->provision_dc($path);
 237
 238         $self->check_or_start($env,
 239                 ($ENV{SMBD_MAX_TIME} or 5400));
 240
 241         $self->wait_for_start($env);
 242
 243         $self->{vars}->{dc} = $env;
 244
 245         return $env;
 246 }
 247
 248 sub stop($)
 249 {
 250         my ($self) = @_;
 251 }
 252
 253 1;