lib/PXElator/client.pm

package client;

use warnings;
use strict;
use autodie;

use File::Slurp;
use Net::Ping;

use server;
use format;
use ip;

our $debug = $server::debug;

sub mkbasedir {
        my $path = shift;
        $path =~ s{(^.*)/[^/]+$}{$1};
        mkdir $path unless -d $path;
        return $path;
}

sub mac_path { $server::conf . '/mac/' . $_[0] }
sub  ip_path { $server::conf . '/ip/'  . join('/', @_) }
sub conf_value {
        my $path = shift;
        my $value;
        if ( -l $path ) {
                $value = readlink $path;
                $value =~ s{.*/([^/]+)$}{$1};
        } elsif ( -f $path ) {
                $value = read_file $path;
        } else {
                warn "W: $path not file or symlink\n";
        }
        return $value;
}

sub conf {
        my $ip  = shift;
        my $name = shift;
        my ( $default, $value );
        if ( $#_ == 0 ) {
                $value = shift;
        } elsif ( $#_ == 1 && $_[0] eq 'default' ) {
                $default = $_[1]
        }

        my $path = ip_path $ip;
        mkdir $path unless -d $path;
        $path .= '/' . $name;

        if ( defined $value ) {
                mkbasedir  $path;
                write_file $path, $value;
                warn "update $path = $value";
        } elsif ( ! -e $path && defined $default ) {
                mkbasedir  $path;
                write_file $path, $default;
                warn "default $path = $default";
                $value = $default;
        } elsif ( -f $path ) {
                $value = read_file $path;
        } else {
                warn "# $name missing $path\n" if $debug;
        }
        return $value;
}

sub all_conf {
        my $ip = shift;
        my $path = ip_path $ip || return;
        my $conf;
        foreach my $file ( glob("$path/*") ) {
                my $name = $file;
                $name =~ s{^.+/([^/]+)$}{$1};
                $conf->{ $name } = read_file $file;
        }
        return $conf;
}
sub next_ip($) {
        my $mac = shift;
        $mac = format::mac($mac);

        my $p = Net::Ping->new;

        my $prefix = $server::ip;
        $prefix =~ s{\.\d+$}{.};
        my $addr = $server::ip_from || die;
        my $ip = $prefix . $addr;

        while ( -e ip_path($ip) || $p->ping( $ip, 0.7 ) ) {
                $ip = $prefix . $addr++;
                die "all addresses allocated!" if $addr == $server::ip_to;
                warn "skip $ip\n";
        }

        warn "next_ip $ip\n";

        save_ip_mac( $ip, $mac );

        return $ip;
}

sub save_ip_mac {
        my ($ip,$mac) = @_;
        $mac = format::mac($mac);
        return if $mac eq '00:00:00:00:00:00';

        mkdir ip_path($ip) unless -e ip_path($ip);

        my $mac_path = mac_path($mac);
        unlink $mac_path if -l $mac_path;       # XXX audit?
        symlink ip_path($ip), $mac_path;
        write_file( ip_path($ip,'mac'), $mac );
}

sub ip_from_mac($) {
        my $mac = shift;
        $mac = format::mac($mac);

        my $mac_path = mac_path $mac;
        return unless -e $mac_path;

        my $ip;

        if ( -f $mac_path ) {
                $ip = read_file $mac_path;
                unlink $mac_path;
                symlink ip_path($ip), $mac_path;
                warn "I: upgrade to mac symlink $mac_path\n";
        } elsif ( -l $mac_path ) {
                $ip = conf_value $mac_path;
        } else {
                die "$mac_path not file or symlink";
        }

        return $ip;
}

sub mac_from_ip($) {
        my $ip = shift;
        conf_value ip_path($ip, 'mac');
}

sub change_ip($$) {
        my ($old, $new) = @_;
        return if $old eq $new;
        my $mac = mac_from_ip($old) || die "no mac for $old";
        rename ip_path($old), ip_path($new);
        unlink mac_path($mac);
        symlink ip_path($new), mac_path($mac);
        return $new;
}

sub all_ips {
        sort { ip::to_int($a) cmp ip::to_int($b) }
        map {
                my $ip = $_;
                $ip =~ s{^.+/ip/}{};
                $ip;
        } glob("$server::conf/ip/*") 
}

sub remove {
        my $ip = shift;
        unlink $_ foreach grep { -e $_ } ( glob "$server::conf/ip/$ip/*" );
        if ( my $mac = mac_from_ip $ip ) {
                unlink "$server::conf/mac/$mac";
        }
        rmdir "$server::conf/ip/$ip";
}

1;
1	dpavlin	156	package client;
2
3			use warnings;
4			use strict;
5			use autodie;
6
7			use File::Slurp;
8	dpavlin	168	use Net::Ping;
9	dpavlin	219
10			use server;
11	dpavlin	208	use format;
12	dpavlin	322	use ip;
13	dpavlin	156
14	dpavlin	220	our $debug = $server::debug;
15
16	dpavlin	219	sub mkbasedir {
17			my $path = shift;
18			$path =~ s{(^.*)/[^/]+$}{$1};
19			mkdir $path unless -d $path;
20			return $path;
21			}
22
23	dpavlin	200	sub mac_path { $server::conf . '/mac/' . $_[0] }
24			sub ip_path { $server::conf . '/ip/' . join('/', @_) }
25	dpavlin	219	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	dpavlin	244	warn "W: $path not file or symlink\n";
35	dpavlin	219	}
36			return $value;
37			}
38	dpavlin	200
39	dpavlin	156	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	dpavlin	200	my $path = ip_path $ip;
50	dpavlin	156	mkdir $path unless -d $path;
51			$path .= '/' . $name;
52
53			if ( defined $value ) {
54	dpavlin	219	mkbasedir $path;
55	dpavlin	156	write_file $path, $value;
56			warn "update $path = $value";
57			} elsif ( ! -e $path && defined $default ) {
58	dpavlin	219	mkbasedir $path;
59	dpavlin	156	write_file $path, $default;
60			warn "default $path = $default";
61			$value = $default;
62	dpavlin	219	} elsif ( -f $path ) {
63			$value = read_file $path;
64			} else {
65	dpavlin	220	warn "# $name missing $path\n" if $debug;
66	dpavlin	156	}
67			return $value;
68			}
69
70	dpavlin	305	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	dpavlin	200	sub next_ip($) {
82			my $mac = shift;
83	dpavlin	313	$mac = format::mac($mac);
84	dpavlin	168
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	dpavlin	200	while ( -e ip_path($ip) \|\| $p->ping( $ip, 0.7 ) ) {
93	dpavlin	168	$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	dpavlin	200
100	dpavlin	244	save_ip_mac( $ip, $mac );
101
102			return $ip;
103			}
104
105			sub save_ip_mac {
106			my ($ip,$mac) = @_;
107	dpavlin	313	$mac = format::mac($mac);
108	dpavlin	325	return if $mac eq '00:00:00:00:00:00';
109	dpavlin	244
110	dpavlin	254	mkdir ip_path($ip) unless -e ip_path($ip);
111	dpavlin	200
112	dpavlin	217	my $mac_path = mac_path($mac);
113	dpavlin	286	unlink $mac_path if -l $mac_path; # XXX audit?
114	dpavlin	217	symlink ip_path($ip), $mac_path;
115	dpavlin	323	write_file( ip_path($ip,'mac'), $mac );
116	dpavlin	168	}
117
118	dpavlin	200	sub ip_from_mac($) {
119	dpavlin	194	my $mac = shift;
120	dpavlin	313	$mac = format::mac($mac);
121	dpavlin	194
122	dpavlin	200	my $mac_path = mac_path $mac;
123	dpavlin	194	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	dpavlin	200	symlink ip_path($ip), $mac_path;
131	dpavlin	194	warn "I: upgrade to mac symlink $mac_path\n";
132			} elsif ( -l $mac_path ) {
133	dpavlin	219	$ip = conf_value $mac_path;
134	dpavlin	194	} else {
135			die "$mac_path not file or symlink";
136			}
137
138			return $ip;
139			}
140
141	dpavlin	200	sub mac_from_ip($) {
142			my $ip = shift;
143	dpavlin	219	conf_value ip_path($ip, 'mac');
144	dpavlin	200	}
145
146			sub change_ip($$) {
147			my ($old, $new) = @_;
148	dpavlin	323	return if $old eq $new;
149			my $mac = mac_from_ip($old) \|\| die "no mac for $old";
150	dpavlin	200	rename ip_path($old), ip_path($new);
151			unlink mac_path($mac);
152			symlink ip_path($new), mac_path($mac);
153	dpavlin	323	return $new;
154	dpavlin	200	}
155
156	dpavlin	322	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	dpavlin	323	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	dpavlin	156	1;