--- trunk/vz-clone.pl 2007/10/04 13:19:26 22 +++ trunk/vz-clone.pl 2009/03/03 20:58:48 69 @@ -1,17 +1,65 @@ #!/usr/bin/perl -w use strict; -use Shell qw/rsync vzlist lvcreate mount umount lvremove/; +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 $snap_size = '10G'; -my $VEID = shift @ARGV || die "Usage: $0 VEID\n\n",vzlist; +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; + my $snap; + open(my $m, '-|', 'mount') || die "can't open mount: $!"; + while(<$m>) { + my @v = split(/\s+/,$_); + if ( $v[2] =~ m{/(private|root)/($VEID|$CVEID)$} && $v[5] =~ m{bind} ) { + umount( $v[2] ); + } elsif ( $v[2] =~ m{vz-clone-$CVEID$} ) { + $clone = $v[2]; + } elsif ( $v[2] =~ m{-snap} ) { + $snap = $v[0]; + umount( $snap ); + } + } + umount( $clone ) if $clone; + if ( $action eq 'destroy' && -e $snap ) { + lvremove( '-f', $snap ); + } + + vzctl('destroy', $VEID) if $action eq 'destroy'; + + exit if $action =~ m{(umount|destroy)}; +} + + my $orig_conf = "/etc/vz/conf/$VEID.conf"; die "$VEID config $orig_conf doesn't exist\n" unless -e $orig_conf; @@ -26,54 +74,109 @@ my @v = split(/\s+/,$_); if ( $v[1] =~ m/\Q$vz\E/ ) { $vz_lv = $v[0]; - warn "found lv $vz_lv for $vz\n"; + 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('-ra', "$from/", "$to/" ); + rsync('-raSHAD', '--delete', "$from/", "$to/" ); } if ( $vz_lv ) { - my ( $vz_lv_path, $vz_lv_name ) = ( $1, $2 ) if ( $vz_lv =~ m!^(.+)/([^/]+)$! ); + my ( $vz_lv_path, $vz_lv_name ) = ( $1, $2 ) if ( $vz_lv =~ m{^(.+)/([^/]+)$} ); my $snap = $vz_lv_name . '-snap'; - - print "Creating $snap_size snapshot $snap from $vz_lv\n"; + 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 ) { - lvcreate( '--size', $snap_size, '--snapshot', '--name', $snap, $vz_lv ); + print "Creating $snap_size snapshot $snap from $vz_lv\n"; + lvcreate( '--size', $snap_size, '--snapshot', '--name', $snap, $vz_lv ); - my $snap_path = "/tmp/$snap"; - mkdir $snap_path || die "can't create $snap_path: $!"; + 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" ); - print "Mounting $vz_lv_path/$snap to $snap_path\n"; + } else { - mount( "$vz_lv_path/$snap", $snap_path, '-o', 'noatime' ); + 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" ); - copy_files( "$snap_path/private/$VEID", "$vz/private/$CVEID" ); + test_mkdir( "/tmp/$clone/private" ); + test_mkdir( "/tmp/$clone/root" ); - print "Umount $snap_path and remove $vz_lv_path/$snap\n"; + copy_files( "$vz/private/$VEID", "/tmp/$clone/private" ); - umount( $snap_path ); + mount_bind( "/tmp/$clone/private", "$vz/private/$CVEID" ); + mount_bind( "/tmp/$clone/root", "$vz/root/$CVEID" ); - lvremove( '-f', "$vz_lv_path/$snap" ); + } } 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+\./10./; + $ip =~ s/^\d+\.\d+\./192.168./; warn "$CVEID new IP number: $ip\n"; + $ve_clone_ip = $ip; return $note . qq{IP_ADDRESS="$ip"}; } @@ -88,10 +191,15 @@ 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; + 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" +print $n qq{DISK_QUOTA="no"\n} if $use_snapshot; # don't wait for quota rebuild on startup + +my $iptables="iptables -t nat -A POSTROUTING -s $ve_clone_ip -o eth0 -j MASQUERADE"; +system( $iptables ); +print "\nConfig file: $conf/$CVEID.conf\nAdded NAT for new VE with: $iptables\nStart clone of $VEID with: vzctl start $CVEID\n" +