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