trunk/lib/VZ.pm

package VZ;
use Exporter 'import';
our @EXPORT = qw(
check_root

ip2hostname
hostname2ip

conf_veids
runscript

vzctl
vzlist

$vz_root
$vz_conf

dump
);

use warnings;
use strict;

use Socket;
use Data::Dump qw/dump/;

our $vz_root = '/vz';
our $vz_conf = '/etc/vz/conf';

$vz_root = '/var/lib/vz' if ! -e $vz_root && -e '/var/lib/vz';

=head1 NAME

VZ - helper functions for VZ tools

=cut

=head1 FUNCTIONS

=head2 check_root

Restart as C<root> user if needed using C<sudo>

=cut

sub check_root {
        if ( $> != 0 ) {
                warn "WARNING: restarting as root using sudo\n";
                exec 'sudo', $0, @ARGV;
        }
}

=head2 hostname2ip

  my $ip = hostname2ip('www.example.com');

=cut

sub hostname2ip {
        my $hostname = shift || return;

        my @addresses = gethostbyname($hostname)
                or die "Can't resolve $hostname: $!\n";
        @addresses = map { inet_ntoa($_) } @addresses[4 .. $#addresses];
        return shift @addresses;
}

=head2 ip2hostname

  my $hostname = ip2hostname('192.168.0.1');

=cut

sub ip2hostname {
        my $ip = shift || return;

        my $hostname = gethostbyaddr(inet_aton($ip), AF_INET)
                or die "Can't look up $ip: $!\n";
        return $hostname;
}

=head2 conf_veids

  my @VEIDs = conf_veids;

=cut

sub conf_veids {
        my @c;
        open( my $cfgs, 'find /etc/vz/conf -maxdepth 1 -name "*.conf" |' ) || die "can't run find: $!";
        while(<$cfgs>) {
                chomp;
                if ( m{^.+/(\d+)\.conf$} ) {
                        if ( -d "$vz_root/private/$1" ) {
                                push @c, $1;
                        } else {
                                warn "WARNING: have config for $1 but no private directory\n";
                        }
                } else {
                        warn "SKIPPED: $_\n";
                }
        }
        return @c;
}

=head2 runscript

  runscript( $veid, '/path/to/script' );

=cut

sub runscript {
        my ( $veid, $path ) = @_;

        if ( open(my $fh, $path) ) {
                while(<$fh>) {
                        chomp;
                        next if (m/^\s*$/);
                        if (/^#\s+(.+)$/) {
                                warn ">> $1\n";
                        } else {
                                vzctl('exec', $veid, $_);
                        }
                }
        } else {
                warn "can't open $path: $!";
        }
}
                

=head2 vzctl

  vzctl('set','--something',42);

=cut

sub vzctl {
        my @args = @_;
        my $a = join(' ', @args);
        # hide passwords from output
        $a =~ s/(--userpasswd\s+\w+:)\w+/$1********/;
        warn "## vzctl $a\n";
        system('vzctl',@args) == 0 or die "vzctl @args failed: $?"
}

=head2 vzlist

  my $output = vzlist();
  my $hash = vzlist( hash => 1 );

=cut

my @vzlist_fields;

sub vzlist {
        die "need hash as argument" unless $#_ % 2 == 1;
        my $args = {@_};


        my $output;
        $output = {} if defined $args->{hash};

        if ( ! @vzlist_fields ) {
                open(my $vzlist, 'vzlist -L |') || die "can't start vzlist -L: $!";
                while(<$vzlist>) {
                        push @vzlist_fields, (split(/\s+/, $_, 2 ))[0];
                }
        }

        open(my $vzlist, 'vzlist -a -H -o ' . join(',', @vzlist_fields) . ' |') || die "can't start vzlist: $!";
        while(<$vzlist>) {
                s/^\s+//;
                my @data = split(/\s+/, $_);
                
                if ( defined $args->{hash} ) {
                        my $hash;
                        $hash->{ $vzlist_fields[$_] } = $data[$_] foreach 1 .. $#data;
                        $output->{ $data[0] } = $hash;
                } else {
                        $output .= $_;
                }
        }

        return $output;
}

1;
1	dpavlin	38	package VZ;
2			use Exporter 'import';
3			our @EXPORT = qw(
4			check_root
5	dpavlin	42
6	dpavlin	40	ip2hostname
7	dpavlin	38	hostname2ip
8	dpavlin	42
9			conf_veids
10	dpavlin	44	runscript
11	dpavlin	42
12	dpavlin	40	vzctl
13	dpavlin	42	vzlist
14	dpavlin	40
15			$vz_root
16			$vz_conf
17	dpavlin	42
18			dump
19	dpavlin	38	);
20
21			use warnings;
22			use strict;
23
24			use Socket;
25	dpavlin	42	use Data::Dump qw/dump/;
26	dpavlin	38
27	dpavlin	40	our $vz_root = '/vz';
28			our $vz_conf = '/etc/vz/conf';
29
30	dpavlin	59	$vz_root = '/var/lib/vz' if ! -e $vz_root && -e '/var/lib/vz';
31
32	dpavlin	40	=head1 NAME
33
34			VZ - helper functions for VZ tools
35
36			=cut
37
38			=head1 FUNCTIONS
39
40			=head2 check_root
41
42			Restart as C<root> user if needed using C<sudo>
43
44			=cut
45
46	dpavlin	38	sub check_root {
47			if ( $> != 0 ) {
48			warn "WARNING: restarting as root using sudo\n";
49			exec 'sudo', $0, @ARGV;
50			}
51			}
52
53	dpavlin	40	=head2 hostname2ip
54
55			my $ip = hostname2ip('www.example.com');
56
57			=cut
58
59	dpavlin	38	sub hostname2ip {
60			my $hostname = shift \|\| return;
61
62	dpavlin	40	my @addresses = gethostbyname($hostname)
63			or die "Can't resolve $hostname: $!\n";
64	dpavlin	38	@addresses = map { inet_ntoa($_) } @addresses[4 .. $#addresses];
65			return shift @addresses;
66			}
67
68	dpavlin	40	=head2 ip2hostname
69
70			my $hostname = ip2hostname('192.168.0.1');
71
72			=cut
73
74			sub ip2hostname {
75			my $ip = shift \|\| return;
76
77	dpavlin	42	my $hostname = gethostbyaddr(inet_aton($ip), AF_INET)
78	dpavlin	40	or die "Can't look up $ip: $!\n";
79	dpavlin	42	return $hostname;
80	dpavlin	40	}
81
82	dpavlin	42	=head2 conf_veids
83	dpavlin	40
84	dpavlin	42	my @VEIDs = conf_veids;
85	dpavlin	40
86			=cut
87
88	dpavlin	42	sub conf_veids {
89	dpavlin	40	my @c;
90			open( my $cfgs, 'find /etc/vz/conf -maxdepth 1 -name "*.conf" \|' ) \|\| die "can't run find: $!";
91			while(<$cfgs>) {
92			chomp;
93	dpavlin	49	if ( m{^.+/(\d+)\.conf$} ) {
94	dpavlin	40	if ( -d "$vz_root/private/$1" ) {
95			push @c, $1;
96			} else {
97			warn "WARNING: have config for $1 but no private directory\n";
98			}
99			} else {
100			warn "SKIPPED: $_\n";
101			}
102			}
103			return @c;
104			}
105
106	dpavlin	44	=head2 runscript
107
108			runscript( $veid, '/path/to/script' );
109
110			=cut
111
112			sub runscript {
113			my ( $veid, $path ) = @_;
114
115	dpavlin	46	if ( open(my $fh, $path) ) {
116			while(<$fh>) {
117			chomp;
118			next if (m/^\s*$/);
119			if (/^#\s+(.+)$/) {
120			warn ">> $1\n";
121			} else {
122			vzctl('exec', $veid, $_);
123			}
124	dpavlin	44	}
125	dpavlin	46	} else {
126			warn "can't open $path: $!";
127	dpavlin	44	}
128			}
129
130
131	dpavlin	40	=head2 vzctl
132
133			vzctl('set','--something',42);
134
135			=cut
136
137			sub vzctl {
138			my @args = @_;
139	dpavlin	44	my $a = join(' ', @args);
140			# hide passwords from output
141			$a =~ s/(--userpasswd\s+\w+:)\w+/$1********/;
142			warn "## vzctl $a\n";
143	dpavlin	40	system('vzctl',@args) == 0 or die "vzctl @args failed: $?"
144			}
145
146	dpavlin	42	=head2 vzlist
147
148			my $output = vzlist();
149			my $hash = vzlist( hash => 1 );
150
151			=cut
152
153			my @vzlist_fields;
154
155			sub vzlist {
156			die "need hash as argument" unless $#_ % 2 == 1;
157			my $args = {@_};
158
159
160			my $output;
161	dpavlin	57	$output = {} if defined $args->{hash};
162	dpavlin	42
163			if ( ! @vzlist_fields ) {
164			open(my $vzlist, 'vzlist -L \|') \|\| die "can't start vzlist -L: $!";
165			while(<$vzlist>) {
166			push @vzlist_fields, (split(/\s+/, $_, 2 ))[0];
167			}
168			}
169
170			open(my $vzlist, 'vzlist -a -H -o ' . join(',', @vzlist_fields) . ' \|') \|\| die "can't start vzlist: $!";
171			while(<$vzlist>) {
172			s/^\s+//;
173			my @data = split(/\s+/, $_);
174
175			if ( defined $args->{hash} ) {
176			my $hash;
177			$hash->{ $vzlist_fields[$_] } = $data[$_] foreach 1 .. $#data;
178			$output->{ $data[0] } = $hash;
179			} else {
180			$output .= $_;
181			}
182			}
183
184			return $output;
185			}
186
187	dpavlin	38	1;