--- trunk/vz-clone.pl	2007/10/04 13:19:26	22
+++ trunk/vz-clone.pl	2007/10/04 16:58:01	23
@@ -1,7 +1,7 @@
 #!/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 ssync lvdisplay mke2fs/;
 
 my $vz   = '/vz';
 my $conf = '/etc/vz/conf';
@@ -26,15 +26,31 @@
 	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 ) = @_;
-	warn "rsync $from -> $to\n";
-	rsync('-ra', "$from/", "$to/" );
+	if ( which('ssync') ) {
+		warn "ssync $from -> $to\n";
+		ssync(qw{--log-mode file --log-path /dev/null},'-f',$from,'-t',$to);
+	} else {
+		warn "rsync $from -> $to\n";
+		rsync('-ra', "$from/", "$to/" );
+	}
 }
 
 if ( $vz_lv ) {
@@ -42,23 +58,59 @@
 	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_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 );
 
-	my $snap_path = "/tmp/$snap";
-	mkdir $snap_path || die "can't create $snap_path: $!";
+	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" );
 
-	print "Mounting $vz_lv_path/$snap to $snap_path\n";
+	test_mkdir( "/tmp/$clone/private" );
+	test_mkdir( "/tmp/$clone/root" );
 
-	mount( "$vz_lv_path/$snap", $snap_path, '-o', 'noatime' );
+	mount_bind( "/tmp/$clone/private", "$vz/private/$CVEID" );
+	mount_bind( "/tmp/$clone/root", "$vz/root/$CVEID" );
 
-	copy_files( "$snap_path/private/$VEID", "$vz/private/$CVEID" );
+	copy_files( "/tmp/$snap/private/$VEID", "$vz/private/$CVEID" );
 
-	print "Umount $snap_path and remove $vz_lv_path/$snap\n";
+	print "Cleanup\n";
 
-	umount( $snap_path );
+	umount( "$vz_lv_path/$clone" );
+	umount( "$vz_lv_path/$snap" );
 
 	lvremove( '-f', "$vz_lv_path/$snap" );