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 = '1G';

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

my $use_snapshot = 1;

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

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

my $CVEID = "10$VEID";

sub vzctl_stop {
        my $veid = shift;
        my $vzlist = vzlist();
        if ( $vzlist =~ m/$veid/ ) {
                print "stop $veid\n";
                vzctl('stop', $veid);
        }
}

if ( $action eq 'create' ) {
        vzctl_stop( $CVEID );
} elsif ( $action eq 'destroy' ) {
        vzctl_stop( $VEID );
}


if ( $action =~ m/(create|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)/$CVEID$} && $v[5] =~ m/bind/ ) {
                        umount( $v[2] );
                } elsif ( $v[2] =~ m{vz-clone-$CVEID$} ) {
                        $clone = $v[2];
                }
        }
        umount( $clone ) if $clone;
        lvremove( '-f', $clone );

        exit if $action eq 'umount';
}


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 ) = @_;
        return if $action ne 'create';
        warn "rsync $from -> $to\n";
        rsync('-raSHAD', '--delete', "$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-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 );
        }

        if ( $use_snapshot ) {

                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" );
        
                mount_bind( "/tmp/$snap/private/$VEID", "$vz/private/$CVEID" );
                mount_bind( "/tmp/$snap/root/$VEID", "$vz/root/$CVEID" );

        } else {

                if ( lvdisplay( "$vz_lv_path/$clone" ) ) {
                        warn "using existing $vz_lv_path/$clone\n";
                } else {
                        my $vg_name = $1 if ( $vz_lv_path =~ m{/([^/]+)/*$} );
                        my $clone_size = fs_quota( $VEID ) . 'k';
                        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" );

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

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

        }

} else {

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

exit if $action eq 'mount';

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

my $ve_clone_ip;

sub fix_ip {
        my $ip = shift;
        $ip =~ s/['"]//g;
#       $ip =~ s/^\d+\./10./;
        $ip =~ s/^\d+\.\d+\./192.168./;
        warn "$CVEID new IP number: $ip\n";
        $ve_clone_ip = $ip;
        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 $n qq{DISK_QUOTA="no"\n} if $use_snapshot; # don't wait for quota rebuild on startup

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