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
  13 our $debug = $server::debug;
  14
  15 sub mkbasedir {
  16         my $path = shift;
  17         $path =~ s{(^.*)/[^/]+$}{$1};
  18         mkdir $path unless -d $path;
  19         return $path;
  20 }
  21
  22 sub mac_path { $server::conf . '/mac/' . $_[0] }
  23 sub  ip_path { $server::conf . '/ip/'  . join('/', @_) }
  24 sub conf_value {
  25         my $path = shift;
  26         my $value;
  27         if ( -l $path ) {
  28                 $value = readlink $path;
  29                 $value =~ s{.*/([^/]+)$}{$1};
  30         } elsif ( -f $path ) {
  31                 $value = read_file $path;
  32         } else {
  33                 warn "W: $path not file or symlink\n";
  34         }
  35         return $value;
  36 }
  37
  38 sub conf {
  39         my $ip  = shift;
  40         my $name = shift;
  41         my ( $default, $value );
  42         if ( $#_ == 0 ) {
  43                 $value = shift;
  44         } elsif ( $#_ == 1 && $_[0] eq 'default' ) {
  45                 $default = $_[1]
  46         }
  47
  48         my $path = ip_path $ip;
  49         mkdir $path unless -d $path;
  50         $path .= '/' . $name;
  51
  52         if ( defined $value ) {
  53                 mkbasedir  $path;
  54                 write_file $path, $value;
  55                 warn "update $path = $value";
  56         } elsif ( ! -e $path && defined $default ) {
  57                 mkbasedir  $path;
  58                 write_file $path, $default;
  59                 warn "default $path = $default";
  60                 $value = $default;
  61         } elsif ( -f $path ) {
  62                 $value = read_file $path;
  63         } else {
  64                 warn "# $name missing $path\n" if $debug;
  65         }
  66         return $value;
  67 }
  68
  69 sub next_ip($) {
  70         my $mac = shift;
  71
  72         my $p = Net::Ping->new;
  73
  74         my $prefix = $server::ip;
  75         $prefix =~ s{\.\d+$}{.};
  76         my $addr = $server::ip_from || die;
  77         my $ip = $prefix . $addr;
  78
  79         while ( -e ip_path($ip) || $p->ping( $ip, 0.7 ) ) {
  80                 $ip = $prefix . $addr++;
  81                 die "all addresses allocated!" if $addr == $server::ip_to;
  82                 warn "skip $ip\n";
  83         }
  84
  85         warn "next_ip $ip\n";
  86
  87         save_ip_mac( $ip, $mac );
  88
  89         return $ip;
  90 }
  91
  92 sub save_ip_mac {
  93         my ($ip,$mac) = @_;
  94
  95         mkdir ip_path($ip) unless -e $ip;
  96
  97         my $mac_path = mac_path($mac);
  98         unlink $mac_path if -e $mac_path;       # XXX audit?
  99         symlink ip_path($ip), $mac_path;
 100         write_file ip_path($ip,'mac'), $mac;
 101 }
 102
 103 sub ip_from_mac($) {
 104         my $mac = shift;
 105
 106         $mac = lc $mac;
 107         $mac =~ s{:}{}g;
 108
 109         my $mac_path = mac_path $mac;
 110         return unless -e $mac_path;
 111
 112         my $ip;
 113
 114         if ( -f $mac_path ) {
 115                 $ip = read_file $mac_path;
 116                 unlink $mac_path;
 117                 symlink ip_path($ip), $mac_path;
 118                 warn "I: upgrade to mac symlink $mac_path\n";
 119         } elsif ( -l $mac_path ) {
 120                 $ip = conf_value $mac_path;
 121         } else {
 122                 die "$mac_path not file or symlink";
 123         }
 124
 125         return $ip;
 126 }
 127
 128 sub mac_from_ip($) {
 129         my $ip = shift;
 130         conf_value ip_path($ip, 'mac');
 131 }
 132
 133 sub change_ip($$) {
 134         my ($old, $new) = @_;
 135         my $mac = mac_from_ip($old);
 136         rename ip_path($old), ip_path($new);
 137         unlink mac_path($mac);
 138         symlink ip_path($new), mac_path($mac);
 139 }
 140
 141 1;