/[pxelator]/lib/PXElator/client.pm
This is repository of my old source code which isn't updated any more. Go to git.rot13.org for current projects!
ViewVC logotype

Diff of /lib/PXElator/client.pm

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 244 by dpavlin, Mon Aug 17 13:27:18 2009 UTC revision 541 by dpavlin, Sat Oct 9 12:28:01 2010 UTC
# Line 5  use strict; Line 5  use strict;
5  use autodie;  use autodie;
6    
7  use File::Slurp;  use File::Slurp;
8  use Net::Ping;  use Data::Dump qw/dump/;
9    use File::Path;
10    
11  use server;  use server;
12  use format;  use format;
13    use ip;
14    use ping;
15    use kvm;
16    
17  our $debug = $server::debug;  our $debug = $server::debug;
18    
# Line 46  sub conf { Line 50  sub conf {
50          }          }
51    
52          my $path = ip_path $ip;          my $path = ip_path $ip;
53          mkdir $path unless -d $path;          mkdir $path unless -e $path;
54            warn "WARNING: $path not directory" unless -d $path;
55          $path .= '/' . $name;          $path .= '/' . $name;
56    
57          if ( defined $value ) {          if ( defined $value ) {
# Line 58  sub conf { Line 63  sub conf {
63                  write_file $path, $default;                  write_file $path, $default;
64                  warn "default $path = $default";                  warn "default $path = $default";
65                  $value = $default;                  $value = $default;
66            } elsif ( -l $path ) {
67                    $value = readlink $path;
68          } elsif ( -f $path ) {          } elsif ( -f $path ) {
69                  $value = read_file $path;                  $value = read_file $path;
70                    chomp $value;
71          } else {          } else {
72                  warn "# $name missing $path\n" if $debug;                  warn "# $name missing $path\n" if $debug;
73          }          }
74          return $value;          return $value;
75  }  }
76    
77    sub all_conf {
78            my $ip = shift;
79            my $path = ip_path $ip || return;
80            my $conf;
81            foreach my $file ( glob("$path/*"), glob("$path/*/*") ) {
82                    my $name = $file;
83                    $name =~ s{^$path/+}{} || die "can't remove $path from $name";
84                    next if -d $file;
85                    $conf->{ $name } =
86                            -l $file ? readlink  $file :
87                            -f $file ? read_file $file :
88                            '?';
89            }
90            return $conf;
91    }
92  sub next_ip($) {  sub next_ip($) {
93          my $mac = shift;          my $mac = shift;
94            $mac = format::mac($mac);
95    
96          my $p = Net::Ping->new;          if ( $server::new_clients > 0 ) {
97                    warn "# clients left: ", --$server::new_clients;
98            } else {
99                    warn "W: no new clients accepted";
100                    return '0.0.0.0';
101            }
102    
103          my $prefix = $server::ip;          my $prefix = $server::ip;
104          $prefix =~ s{\.\d+$}{.};          $prefix =~ s{\.\d+$}{.};
105          my $addr = $server::ip_from || die;          my $addr = $server::ip_from || die;
106          my $ip = $prefix . $addr;          my $ip = $prefix . $addr;
107    
108          while ( -e ip_path($ip) || $p->ping( $ip, 0.7 ) ) {          while ( -e ip_path($ip) || ping::host($ip) ) {
109                  $ip = $prefix . $addr++;                  $ip = $prefix . $addr++;
110                  die "all addresses allocated!" if $addr == $server::ip_to;                  die "all addresses allocated!" if $addr == $server::ip_to;
111                  warn "skip $ip\n";                  warn "skip $ip\n";
# Line 91  sub next_ip($) { Line 120  sub next_ip($) {
120    
121  sub save_ip_mac {  sub save_ip_mac {
122          my ($ip,$mac) = @_;          my ($ip,$mac) = @_;
123            $mac = format::mac($mac);
124            return if $mac eq '00:00:00:00:00:00' || $ip eq '0.0.0.0';
125    
126          mkdir ip_path($ip);          mkdir ip_path($ip) unless -e ip_path($ip);
127    
128          my $mac_path = mac_path($mac);          my $mac_path = mac_path($mac);
129          unlink $mac_path if -e $mac_path;       # XXX audit?          unlink $mac_path if -l $mac_path;       # XXX audit?
130          symlink ip_path($ip), $mac_path;          symlink ip_path($ip), $mac_path;
131          write_file ip_path($ip,'mac'), $mac;          write_file( ip_path($ip,'mac'), $mac );
132  }  }
133    
134  sub ip_from_mac($) {  sub ip_from_mac($) {
135          my $mac = shift;          my $mac = shift;
136            $mac = format::mac($mac);
         $mac = lc $mac;  
         $mac =~ s{:}{}g;  
137    
138          my $mac_path = mac_path $mac;          my $mac_path = mac_path $mac;
139          return unless -e $mac_path;          return unless -e $mac_path;
# Line 132  sub mac_from_ip($) { Line 161  sub mac_from_ip($) {
161    
162  sub change_ip($$) {  sub change_ip($$) {
163          my ($old, $new) = @_;          my ($old, $new) = @_;
164          my $mac = mac_from_ip($old);          return if $old eq $new;
165            my $mac = mac_from_ip($old) || die "no mac for $old";
166          rename ip_path($old), ip_path($new);          rename ip_path($old), ip_path($new);
167          unlink mac_path($mac);          unlink mac_path($mac);
168          symlink ip_path($new), mac_path($mac);          symlink ip_path($new), mac_path($mac);
169            return $new;
170    }
171    
172    sub all_ips {
173            sort { ip::to_int($a) cmp ip::to_int($b) }
174            map {
175                    my $ip = $_;
176                    $ip =~ s{^.+/ip/}{};
177                    autocreate_params( $ip );
178                    $ip;
179            } glob("$server::conf/ip/*")
180    }
181    
182    sub remove {
183            my $ip = shift;
184            if ( my $mac = mac_from_ip $ip ) {
185                    unlink "$server::conf/mac/$mac";
186            }
187            rmtree "$server::conf/ip/$ip";
188    }
189    
190    sub arp_mac_dev {
191            my $arp = {
192                    map {
193                            my @c = split(/\s+/,$_);
194                            if ( $#c == 5 ) {
195                                    client::save_ip_mac( $c[0], $c[3] );
196                                    ( uc $c[3] => $c[5] )
197                            } else {
198                            }
199                    } read_file('/proc/net/arp')
200            };
201    
202            warn "# arp ",dump( $arp );
203            return $arp;
204    }
205    
206    sub rebuild_mac_links {
207            warn "# rebuild mac links";
208            foreach my $ip ( all_ips ) {
209                    my $mac = ip_path $ip, 'mac';
210                    if ( -e $mac ) {
211                            $mac = read_file $mac;
212                            chomp $mac;
213                            save_ip_mac( $ip, $mac );
214                            warn "## $ip $mac\n";
215                    }
216            }
217    }
218    
219    sub autocreate_params {
220            my $ip = shift;
221            my $mac = mac_from_ip $ip;
222            if ( $mac =~ m{^AC:DE:48:00:00} && ! defined conf( $ip, 'kvm' ) ) {
223                    conf( $ip, 'kvm', default => kvm::nr_from_mac( $mac ) );
224                    warn "# create kvm for $ip";
225            }
226  }  }
227    
228  1;  1;
Legend:
Removed from v.244  
changed lines
  Added in v.541
  ViewVC Help
Powered by ViewVC 1.1.26