git.rot13.org Git - pxelator/blob - lib/PXElator/client.pm

   1 package client;
   2
   3 use warnings;
   4 use strict;
   5 use autodie;
   6
   7 use File::Slurp;
   8 use Net::Ping;
   9
  10 use server;
  11 use format;
  12 use ip;
  13
  14 our $debug = $server::debug;
  15
  16 sub mkbasedir {
  17         my $path = shift;
  18         $path =~ s{(^.*)/[^/]+$}{$1};
  19         mkdir $path unless -d $path;
  20         return $path;
  21 }
  22
  23 sub mac_path { $server::conf . '/mac/' . $_[0] }
  24 sub  ip_path { $server::conf . '/ip/'  . join('/', @_) }
  25 sub conf_value {
  26         my $path = shift;
  27         my $value;
  28         if ( -l $path ) {
  29                 $value = readlink $path;
  30                 $value =~ s{.*/([^/]+)$}{$1};
  31         } elsif ( -f $path ) {
  32                 $value = read_file $path;
  33         } else {
  34                 warn "W: $path not file or symlink\n";
  35         }
  36         return $value;
  37 }
  38
  39 sub conf {
  40         my $ip  = shift;
  41         my $name = shift;
  42         my ( $default, $value );
  43         if ( $#_ == 0 ) {
  44                 $value = shift;
  45         } elsif ( $#_ == 1 && $_[0] eq 'default' ) {
  46                 $default = $_[1]
  47         }
  48
  49         my $path = ip_path $ip;
  50         mkdir $path unless -d $path;
  51         $path .= '/' . $name;
  52
  53         if ( defined $value ) {
  54                 mkbasedir  $path;
  55                 write_file $path, $value;
  56                 warn "update $path = $value";
  57         } elsif ( ! -e $path && defined $default ) {
  58                 mkbasedir  $path;
  59                 write_file $path, $default;
  60                 warn "default $path = $default";
  61                 $value = $default;
  62         } elsif ( -f $path ) {
  63                 $value = read_file $path;
  64         } else {
  65                 warn "# $name missing $path\n" if $debug;
  66         }
  67         return $value;
  68 }
  69
  70 sub all_conf {
  71         my $ip = shift;
  72         my $path = ip_path $ip || return;
  73         my $conf;
  74         foreach my $file ( glob("$path/*") ) {
  75                 my $name = $file;
  76                 $name =~ s{^.+/([^/]+)$}{$1};
  77                 $conf->{ $name } = read_file $file;
  78         }
  79         return $conf;
  80 }
  81 sub next_ip($) {
  82         my $mac = shift;
  83         $mac = format::mac($mac);
  84
  85         my $p = Net::Ping->new;
  86
  87         my $prefix = $server::ip;
  88         $prefix =~ s{\.\d+$}{.};
  89         my $addr = $server::ip_from || die;
  90         my $ip = $prefix . $addr;
  91
  92         while ( -e ip_path($ip) || $p->ping( $ip, 0.7 ) ) {
  93                 $ip = $prefix . $addr++;
  94                 die "all addresses allocated!" if $addr == $server::ip_to;
  95                 warn "skip $ip\n";
  96         }
  97
  98         warn "next_ip $ip\n";
  99
 100         save_ip_mac( $ip, $mac );
 101
 102         return $ip;
 103 }
 104
 105 sub save_ip_mac {
 106         my ($ip,$mac) = @_;
 107         $mac = format::mac($mac);
 108         return if $mac eq '00:00:00:00:00:00';
 109
 110         mkdir ip_path($ip) unless -e ip_path($ip);
 111
 112         my $mac_path = mac_path($mac);
 113         unlink $mac_path if -l $mac_path;       # XXX audit?
 114         symlink ip_path($ip), $mac_path;
 115         write_file( ip_path($ip,'mac'), $mac );
 116 }
 117
 118 sub ip_from_mac($) {
 119         my $mac = shift;
 120         $mac = format::mac($mac);
 121
 122         my $mac_path = mac_path $mac;
 123         return unless -e $mac_path;
 124
 125         my $ip;
 126
 127         if ( -f $mac_path ) {
 128                 $ip = read_file $mac_path;
 129                 unlink $mac_path;
 130                 symlink ip_path($ip), $mac_path;
 131                 warn "I: upgrade to mac symlink $mac_path\n";
 132         } elsif ( -l $mac_path ) {
 133                 $ip = conf_value $mac_path;
 134         } else {
 135                 die "$mac_path not file or symlink";
 136         }
 137
 138         return $ip;
 139 }
 140
 141 sub mac_from_ip($) {
 142         my $ip = shift;
 143         conf_value ip_path($ip, 'mac');
 144 }
 145
 146 sub change_ip($$) {
 147         my ($old, $new) = @_;
 148         return if $old eq $new;
 149         my $mac = mac_from_ip($old) || die "no mac for $old";
 150         rename ip_path($old), ip_path($new);
 151         unlink mac_path($mac);
 152         symlink ip_path($new), mac_path($mac);
 153         return $new;
 154 }
 155
 156 sub all_ips {
 157         sort { ip::to_int($a) cmp ip::to_int($b) }
 158         map {
 159                 my $ip = $_;
 160                 $ip =~ s{^.+/ip/}{};
 161                 $ip;
 162         } glob("$server::conf/ip/*")
 163 }
 164
 165 sub remove {
 166         my $ip = shift;
 167         unlink $_ foreach grep { -e $_ } ( glob "$server::conf/ip/$ip/*" );
 168         if ( my $mac = mac_from_ip $ip ) {
 169                 unlink "$server::conf/mac/$mac";
 170         }
 171         rmdir "$server::conf/ip/$ip";
 172 }
 173
 174 1;