ksplice.pm.in

   1 package ksplice;
   2 use Cwd 'abs_path';
   3 use File::Temp qw/:mktemp/;
   4 use strict;
   5 use warnings;
   6 require Exporter;
   7 our @ISA = qw(Exporter);
   8 our @EXPORT = qw($datadir $libexecdir $verbose $version_str init_tmpdir runcd runstr runval runval_raw runsuc unpack_update check_in_proc);
   9
  10 our ($datadir, $libexecdir) = qw(KSPLICE_DATA_DIR KSPLICE_LIBEXEC_DIR);
  11 our $version_str = "Ksplice version PACKAGE_VERSION\n";
  12 our $verbose = 0;
  13
  14 sub init_tmpdir {
  15         my $argdir = $ENV{TMPDIR};
  16         $argdir = "/tmp" if(!defined $argdir || ! -d $argdir);
  17         die if(! -d $argdir);
  18
  19         my $tmpdir = mkdtemp("$argdir/ksplice-tmp-XXXXXX");
  20         die if(!defined $tmpdir || ! -d $tmpdir);
  21         return $tmpdir;
  22 }
  23
  24 sub runcd {
  25         my ($dir) = @_;
  26         print "chdir($dir)\n" if($verbose);
  27         if(!chdir($dir)) {
  28                 die "Failed during: chdir($dir)\n";
  29         }
  30 }
  31
  32 sub runval {
  33         my (@cmd) = @_;
  34         if(runval_raw(@cmd) != 0) {
  35                 die "Failed during: @cmd\n";
  36         }
  37 }
  38
  39 sub runval_raw {
  40         my (@cmd) = @_;
  41         my ($out, $err);
  42         print "@cmd\n" if($verbose);
  43         if(!$verbose) {
  44                 open $out, ">&STDOUT" or die "Can't dup STDOUT: $!";
  45                 open $err, ">&STDERR" or die "Can't dup STDERR: $!";
  46                 open STDOUT, '>', "/dev/null" or die "Can't hide STDOUT: $!";
  47                 open STDERR, '>', "/dev/null" or die "Can't hide STDERR: $!";
  48         }
  49         my $val = system(@cmd);
  50         if(!$verbose) {
  51                 open STDOUT, ">&", $out or die "Can't restore STDOUT: $!";
  52                 open STDERR, ">&", $err or die "Can't restore STDERR: $!";
  53         }
  54         return $val;
  55 }
  56
  57 sub runstr {
  58         my ($cmd) = @_;
  59         print "$cmd\n" if($verbose);
  60         return scalar(`$cmd`);
  61 }
  62
  63 sub runsuc {
  64         my ($cmd, $args) = @_;
  65         print "$libexecdir/ksplice-$cmd $args\n" if($verbose);
  66         my $output = `$libexecdir/ksplice-$cmd $args`;
  67         if($output =~ /ksplice: success/) {
  68                 $output = $`;
  69         }
  70         else {
  71                 print "Output: $output\n";
  72                 die "Failed during: $libexecdir/ksplice-$cmd $args";
  73         }
  74         return $output;
  75 }
  76
  77 sub unpack_update {
  78         my ($file) = @_;
  79         runval("tar zxf $file");
  80         my $ksplice = `find */ -maxdepth 0`;
  81         chomp($ksplice);
  82         chop($ksplice); # remove the trailing slash
  83         return $ksplice;
  84 }
  85
  86 sub check_in_proc {
  87         my ($kid) = @_;
  88
  89         if(! -e "/proc/1") {
  90                 print "/proc not mounted?\n";
  91                 exit(1);
  92         }
  93         if(! -e "/proc/ksplice_$kid") {
  94                 print "Ksplice id $kid is not present in the kernel\n";
  95                 exit(1);
  96         }
  97 }
  98
  99 1;