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';

=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;

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