vz-tools/trunk/vz-clone.pl

#!/usr/bin/perl -w

use strict;
use Shell qw/rsync vzlist lvcreate mount umount lvremove which lvdisplay mke2fs vzctl/;

my $vz   = '/vz';
my $conf = '/etc/vz/conf';

my $snap_size = '100M';

my ( $action, $VEID ) = @ARGV;

my $valid_actions = 'create|umount|destroy';

die "Usage: $0 ($valid_actions) VEID\n\n",vzlist unless $VEID && $action && $action =~ m/($valid_actions)/;

if ( $action eq 'destroy' ) {
        my $vzlist = vzlist();
        if ( $vzlist =~ m/$VEID/ ) {
                print "stop $VEID\n";
                vzctl('stop', $VEID);
        }
}

if ( $action =~ m/(umount|destroy)/ ) {
        my $clone;
        open(my $m, '-|', 'mount') || die "can't open mount: $!";
        while(<$m>) {
                my @v = split(/\s+/,$_);
                if ( $v[2] =~ m!/(private|root)/$VEID$! && $v[5] =~ m/bind/ ) {
                        umount( $v[2] );
                } elsif ( $v[2] =~ m!-clone-$VEID$! ) {
                        $clone = $v[2];
                }
        }
        umount( $clone ) if $clone;
}

#die "unknown action: $action" unless $action eq 'create';
exit unless $action eq 'create';

my $CVEID = "10$VEID";

my $orig_conf = "/etc/vz/conf/$VEID.conf";

die "$VEID config $orig_conf doesn't exist\n" unless -e $orig_conf;

print "Clone VE $VEID -> $CVEID\n";

my $vz_lv;

open(my $m, '<', '/etc/fstab') || die "can't open /etc/fstab: $!";
while(<$m>) {
        next if m/^#/;
        my @v = split(/\s+/,$_);
        if ( $v[1] =~ m/\Q$vz\E/ ) {
                $vz_lv = $v[0];
                warn "found LV $vz_lv for $vz\n";
                last;
        }
}

sub fs_quota {
        my $id = shift;
        open(my $q, '-|', "vzquota -b show $id") || die "can't exec vzquota show $id: $!";
        my $l = <$q>;
        $l =~ s/^\s+//;
        my ( $usage, $soft, $hard ) = split(/\s+/,$l);
        warn "quota for $id | $soft < $hard | usage: $usage\n";
        return ( $usage, $soft, $hard ) if wantarray;
        return $soft;
}

sub copy_files {
        my ( $from, $to ) = @_;
        warn "rsync $from -> $to\n";
        rsync('-raSHAD', "$from/", "$to/" );
}

if ( $vz_lv ) {

        my ( $vz_lv_path, $vz_lv_name ) = ( $1, $2 ) if ( $vz_lv =~ m!^(.+)/([^/]+)$! );

        my $snap = $vz_lv_name . '-snap';
        my $clone = "$vz_lv_name-clone-$CVEID";

        sub do_mount {
                my ( $from, $to ) = @_;
                mkdir $to || die "can't create $to: $!";
                print "Mounting $from to $to\n";
                mount( $from, $to, '-o', 'noatime' );
        }

        sub test_mkdir {
                my $dir = shift;
                if ( ! -d $dir ) {
                        mkdir $dir || die "can't mkdir $dir: $!";
                }
        }

        sub mount_bind {
                my ( $from, $to ) = @_;
                die "$from doesn't exist!" unless -d $from;
                test_mkdir( $to );
                mount( '--bind', $from, $to );
        }

        print "Creating $snap_size snapshot $snap from $vz_lv\n";
        lvcreate( '--size', $snap_size, '--snapshot', '--name', $snap, $vz_lv );

        do_mount( "$vz_lv_path/$snap", "/tmp/$snap" );

        my $clone_size = fs_quota( $VEID ) . 'k';
        my $vg_name = $1 if ( $vz_lv_path =~ m!/([^/]+)/*$! );

        if ( lvdisplay( "$vz_lv_path/$clone" ) ) {
                warn "using existing $vz_lv_path/$clone\n";
        } else {
                print "Creating LV $clone ($clone_size bytes) in VG $vg_name for $VEID clone filesystem\n";
                lvcreate( '--size', $clone_size, '--name', $clone, $vg_name );
                mke2fs( '-m', 0, '-j', "$vz_lv_path/$clone" );
        }

        do_mount( "$vz_lv_path/$clone", "/tmp/$clone" );

        test_mkdir( "/tmp/$clone/private" );
        test_mkdir( "/tmp/$clone/root" );

        mount_bind( "/tmp/$clone/private", "$vz/private/$CVEID" );
        mount_bind( "/tmp/$clone/root", "$vz/root/$CVEID" );

        copy_files( "/tmp/$snap/private/$VEID", "$vz/private/$CVEID" );

        print "Cleanup\n";

        #umount( "$vz_lv_path/$clone" );
        umount( "$vz_lv_path/$snap" );

        lvremove( '-f', "$vz_lv_path/$snap" );

} else {

        copy_files( "$vz/private/$VEID", "$vz/private/$CVEID" );
}

my $note = "# modified by $0\n";

sub fix_ip {
        my $ip = shift;
        $ip =~ s/['"]//g;
        $ip =~ s/^\d+\./10./;
        warn "$CVEID new IP number: $ip\n";
        return $note . qq{IP_ADDRESS="$ip"};
}

sub fix_hostname {
        my $hostname = shift;
        $hostname =~ s/['"]//g;
        $hostname = "clone-$hostname";
        warn "$CVEID new hostname: $hostname\n";
        return $note . qq{HOSTNAME="$hostname"};
}

open(my $o, '<', "$conf/$VEID.conf")  || die "can't open $conf/$VEID.conf: $!";
open(my $n, '>', "$conf/$CVEID.conf") || die "can't open $conf/$CVEID.conf: $!";
while(<$o>) {
        s!^HOSTNAME=(.*)$!fix_hostname($1)!ie;
        s!^IP_ADDRESS=(.*)$!fix_ip($1)!ie;
        s!^(ONBOOT=).*$!# modified by $0\n$1"no"!i;
        print $n $_;
}

print "\nPlease review config file: $conf/$CVEID.conf\nAdd NAT for new VE with: iptables -t nat -A POSTROUTING -o eth0 -j MASQUERADE\nStart clone of $VEID with: vzctl start $CVEID\n"
1	dpavlin	20	#!/usr/bin/perl -w
2
3			use strict;
4	dpavlin	25	use Shell qw/rsync vzlist lvcreate mount umount lvremove which lvdisplay mke2fs vzctl/;
5	dpavlin	20
6			my $vz = '/vz';
7			my $conf = '/etc/vz/conf';
8
9	dpavlin	21	my $snap_size = '100M';
10
11	dpavlin	25	my ( $action, $VEID ) = @ARGV;
12	dpavlin	20
13	dpavlin	25	my $valid_actions = 'create\|umount\|destroy';
14
15			die "Usage: $0 ($valid_actions) VEID\n\n",vzlist unless $VEID && $action && $action =~ m/($valid_actions)/;
16
17			if ( $action eq 'destroy' ) {
18			my $vzlist = vzlist();
19			if ( $vzlist =~ m/$VEID/ ) {
20			print "stop $VEID\n";
21			vzctl('stop', $VEID);
22			}
23			}
24
25			if ( $action =~ m/(umount\|destroy)/ ) {
26			my $clone;
27			open(my $m, '-\|', 'mount') \|\| die "can't open mount: $!";
28			while(<$m>) {
29			my @v = split(/\s+/,$_);
30			if ( $v[2] =~ m!/(private\|root)/$VEID$! && $v[5] =~ m/bind/ ) {
31			umount( $v[2] );
32			} elsif ( $v[2] =~ m!-clone-$VEID$! ) {
33			$clone = $v[2];
34			}
35			}
36			umount( $clone ) if $clone;
37			}
38
39			#die "unknown action: $action" unless $action eq 'create';
40			exit unless $action eq 'create';
41
42	dpavlin	21	my $CVEID = "10$VEID";
43	dpavlin	20
44			my $orig_conf = "/etc/vz/conf/$VEID.conf";
45
46			die "$VEID config $orig_conf doesn't exist\n" unless -e $orig_conf;
47
48			print "Clone VE $VEID -> $CVEID\n";
49
50	dpavlin	21	my $vz_lv;
51	dpavlin	20
52	dpavlin	21	open(my $m, '<', '/etc/fstab') \|\| die "can't open /etc/fstab: $!";
53			while(<$m>) {
54			next if m/^#/;
55			my @v = split(/\s+/,$_);
56			if ( $v[1] =~ m/\Q$vz\E/ ) {
57			$vz_lv = $v[0];
58	dpavlin	23	warn "found LV $vz_lv for $vz\n";
59	dpavlin	21	last;
60			}
61			}
62
63	dpavlin	23	sub fs_quota {
64			my $id = shift;
65			open(my $q, '-\|', "vzquota -b show $id") \|\| die "can't exec vzquota show $id: $!";
66			my $l = <$q>;
67			$l =~ s/^\s+//;
68			my ( $usage, $soft, $hard ) = split(/\s+/,$l);
69			warn "quota for $id \| $soft < $hard \| usage: $usage\n";
70			return ( $usage, $soft, $hard ) if wantarray;
71			return $soft;
72			}
73
74	dpavlin	21	sub copy_files {
75			my ( $from, $to ) = @_;
76	dpavlin	24	warn "rsync $from -> $to\n";
77			rsync('-raSHAD', "$from/", "$to/" );
78	dpavlin	21	}
79
80			if ( $vz_lv ) {
81
82			my ( $vz_lv_path, $vz_lv_name ) = ( $1, $2 ) if ( $vz_lv =~ m!^(.+)/([^/]+)$! );
83
84			my $snap = $vz_lv_name . '-snap';
85	dpavlin	23	my $clone = "$vz_lv_name-clone-$CVEID";
86
87			sub do_mount {
88			my ( $from, $to ) = @_;
89			mkdir $to \|\| die "can't create $to: $!";
90			print "Mounting $from to $to\n";
91			mount( $from, $to, '-o', 'noatime' );
92			}
93
94			sub test_mkdir {
95			my $dir = shift;
96			if ( ! -d $dir ) {
97			mkdir $dir \|\| die "can't mkdir $dir: $!";
98			}
99			}
100
101			sub mount_bind {
102			my ( $from, $to ) = @_;
103			die "$from doesn't exist!" unless -d $from;
104			test_mkdir( $to );
105			mount( '--bind', $from, $to );
106			}
107
108	dpavlin	21	print "Creating $snap_size snapshot $snap from $vz_lv\n";
109			lvcreate( '--size', $snap_size, '--snapshot', '--name', $snap, $vz_lv );
110
111	dpavlin	23	do_mount( "$vz_lv_path/$snap", "/tmp/$snap" );
112	dpavlin	21
113	dpavlin	23	my $clone_size = fs_quota( $VEID ) . 'k';
114			my $vg_name = $1 if ( $vz_lv_path =~ m!/([^/]+)/*$! );
115	dpavlin	21
116	dpavlin	23	if ( lvdisplay( "$vz_lv_path/$clone" ) ) {
117			warn "using existing $vz_lv_path/$clone\n";
118			} else {
119			print "Creating LV $clone ($clone_size bytes) in VG $vg_name for $VEID clone filesystem\n";
120			lvcreate( '--size', $clone_size, '--name', $clone, $vg_name );
121			mke2fs( '-m', 0, '-j', "$vz_lv_path/$clone" );
122			}
123	dpavlin	21
124	dpavlin	23	do_mount( "$vz_lv_path/$clone", "/tmp/$clone" );
125	dpavlin	21
126	dpavlin	23	test_mkdir( "/tmp/$clone/private" );
127			test_mkdir( "/tmp/$clone/root" );
128	dpavlin	21
129	dpavlin	23	mount_bind( "/tmp/$clone/private", "$vz/private/$CVEID" );
130			mount_bind( "/tmp/$clone/root", "$vz/root/$CVEID" );
131	dpavlin	21
132	dpavlin	23	copy_files( "/tmp/$snap/private/$VEID", "$vz/private/$CVEID" );
133
134			print "Cleanup\n";
135
136	dpavlin	24	#umount( "$vz_lv_path/$clone" );
137	dpavlin	23	umount( "$vz_lv_path/$snap" );
138
139	dpavlin	21	lvremove( '-f', "$vz_lv_path/$snap" );
140
141			} else {
142
143			copy_files( "$vz/private/$VEID", "$vz/private/$CVEID" );
144			}
145
146			my $note = "# modified by $0\n";
147
148	dpavlin	20	sub fix_ip {
149			my $ip = shift;
150	dpavlin	21	$ip =~ s/['"]//g;
151	dpavlin	20	$ip =~ s/^\d+\./10./;
152			warn "$CVEID new IP number: $ip\n";
153	dpavlin	21	return $note . qq{IP_ADDRESS="$ip"};
154	dpavlin	20	}
155
156	dpavlin	21	sub fix_hostname {
157			my $hostname = shift;
158			$hostname =~ s/['"]//g;
159			$hostname = "clone-$hostname";
160			warn "$CVEID new hostname: $hostname\n";
161			return $note . qq{HOSTNAME="$hostname"};
162			}
163
164	dpavlin	20	open(my $o, '<', "$conf/$VEID.conf") \|\| die "can't open $conf/$VEID.conf: $!";
165			open(my $n, '>', "$conf/$CVEID.conf") \|\| die "can't open $conf/$CVEID.conf: $!";
166			while(<$o>) {
167	dpavlin	21	s!^HOSTNAME=(.*)$!fix_hostname($1)!ie;
168			s!^IP_ADDRESS=(.*)$!fix_ip($1)!ie;
169			s!^(ONBOOT=).*$!# modified by $0\n$1"no"!i;
170	dpavlin	20	print $n $_;
171			}
172
173	dpavlin	22	print "\nPlease review config file: $conf/$CVEID.conf\nAdd NAT for new VE with: iptables -t nat -A POSTROUTING -o eth0 -j MASQUERADE\nStart clone of $VEID with: vzctl start $CVEID\n"