lib/PXElator/client.pm

package client;

use warnings;
use strict;
use autodie;

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

use server;
use format;

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 next_ip($) {
        my $mac = shift;

        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) = @_;

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

        my $mac_path = mac_path($mac);
        unlink $mac_path if -e $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 = lc $mac;
        $mac =~ s{:}{}g;

        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) = @_;
        my $mac = mac_from_ip($old);
        rename ip_path($old), ip_path($new);
        unlink mac_path($mac);
        symlink ip_path($new), mac_path($mac);
}

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	156
13	dpavlin	220	our $debug = $server::debug;
14
15	dpavlin	219	sub mkbasedir {
16			my $path = shift;
17			$path =~ s{(^.*)/[^/]+$}{$1};
18			mkdir $path unless -d $path;
19			return $path;
20			}
21
22	dpavlin	200	sub mac_path { $server::conf . '/mac/' . $_[0] }
23			sub ip_path { $server::conf . '/ip/' . join('/', @_) }
24	dpavlin	219	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	dpavlin	244	warn "W: $path not file or symlink\n";
34	dpavlin	219	}
35			return $value;
36			}
37	dpavlin	200
38	dpavlin	156	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	dpavlin	200	my $path = ip_path $ip;
49	dpavlin	156	mkdir $path unless -d $path;
50			$path .= '/' . $name;
51
52			if ( defined $value ) {
53	dpavlin	219	mkbasedir $path;
54	dpavlin	156	write_file $path, $value;
55			warn "update $path = $value";
56			} elsif ( ! -e $path && defined $default ) {
57	dpavlin	219	mkbasedir $path;
58	dpavlin	156	write_file $path, $default;
59			warn "default $path = $default";
60			$value = $default;
61	dpavlin	219	} elsif ( -f $path ) {
62			$value = read_file $path;
63			} else {
64	dpavlin	220	warn "# $name missing $path\n" if $debug;
65	dpavlin	156	}
66			return $value;
67			}
68
69	dpavlin	200	sub next_ip($) {
70			my $mac = shift;
71	dpavlin	168
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	dpavlin	200	while ( -e ip_path($ip) \|\| $p->ping( $ip, 0.7 ) ) {
80	dpavlin	168	$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	dpavlin	200
87	dpavlin	244	save_ip_mac( $ip, $mac );
88
89			return $ip;
90			}
91
92			sub save_ip_mac {
93			my ($ip,$mac) = @_;
94
95	dpavlin	246	mkdir ip_path($ip) unless -e $ip;
96	dpavlin	200
97	dpavlin	217	my $mac_path = mac_path($mac);
98			unlink $mac_path if -e $mac_path; # XXX audit?
99			symlink ip_path($ip), $mac_path;
100	dpavlin	200	write_file ip_path($ip,'mac'), $mac;
101	dpavlin	168	}
102
103	dpavlin	200	sub ip_from_mac($) {
104	dpavlin	194	my $mac = shift;
105
106			$mac = lc $mac;
107			$mac =~ s{:}{}g;
108
109	dpavlin	200	my $mac_path = mac_path $mac;
110	dpavlin	194	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	dpavlin	200	symlink ip_path($ip), $mac_path;
118	dpavlin	194	warn "I: upgrade to mac symlink $mac_path\n";
119			} elsif ( -l $mac_path ) {
120	dpavlin	219	$ip = conf_value $mac_path;
121	dpavlin	194	} else {
122			die "$mac_path not file or symlink";
123			}
124
125			return $ip;
126			}
127
128	dpavlin	200	sub mac_from_ip($) {
129			my $ip = shift;
130	dpavlin	219	conf_value ip_path($ip, 'mac');
131	dpavlin	200	}
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	dpavlin	156	1;