source/script/tests/SocketWrapper.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 SocketWrapper;
   7
   8 use Exporter;
   9 @ISA = qw(Exporter);
  10 @EXPORT_OK = qw(setup_dir setup_pcap set_default_iface);
  11
  12 use strict;
  13 use FindBin qw($RealBin);
  14
  15 sub setup_dir($)
  16 {
  17         my ($dir) = @_;
  18         if (defined($dir)) {
  19                 if ( -d $dir ) {
  20                         unlink <$dir/*>;
  21                 } else {
  22                         mkdir($dir);
  23                 }
  24         }
  25
  26         $ENV{SOCKET_WRAPPER_DIR} = $dir;
  27         return $dir;
  28 }
  29
  30 sub setup_pcap($)
  31 {
  32         my ($pcap_file) = @_;
  33
  34         $ENV{SOCKET_WRAPPER_PCAP_FILE} = $pcap_file;
  35 }
  36
  37 sub set_default_iface($)
  38 {
  39         my ($i) = @_;
  40         $ENV{SOCKET_WRAPPER_DEFAULT_IFACE} = $i;
  41 }
  42
  43 1;