/[vz-tools]/trunk/vz-clone.pl
This is repository of my old source code which isn't updated any more. Go to git.rot13.org for current projects!
ViewVC logotype

Diff of /trunk/vz-clone.pl

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 22 by dpavlin, Thu Oct 4 13:19:26 2007 UTC revision 28 by dpavlin, Fri Oct 5 00:28:43 2007 UTC
# Line 1  Line 1 
1  #!/usr/bin/perl -w  #!/usr/bin/perl -w
2    
3  use strict;  use strict;
4  use Shell qw/rsync vzlist lvcreate mount umount lvremove/;  use Shell qw/rsync vzlist lvcreate mount umount lvremove which lvdisplay mke2fs vzctl/;
5    
6  my $vz   = '/vz';  my $vz   = '/vz';
7  my $conf = '/etc/vz/conf';  my $conf = '/etc/vz/conf';
8    
9  my $snap_size = '100M';  my $snap_size = '100M';
10    
11  my $VEID = shift @ARGV || die "Usage: $0 VEID\n\n",vzlist;  my ( $action, $VEID ) = @ARGV;
12    
13    my $use_snapshot = 0;
14    
15    my $valid_actions = 'create|umount|destroy';
16    
17    die "Usage: $0 ($valid_actions) VEID\n\n",vzlist unless $VEID && $action && $action =~ m/($valid_actions)/;
18    
19  my $CVEID = "10$VEID";  my $CVEID = "10$VEID";
20    
21    sub vzctl_stop {
22            my $veid = shift;
23            my $vzlist = vzlist();
24            if ( $vzlist =~ m/$veid/ ) {
25                    print "stop $veid\n";
26                    vzctl('stop', $veid);
27            }
28    }
29    
30    
31    if ( $action eq 'create' ) {
32            vzctl_stop( $CVEID );
33    } elsif ( $action eq 'destroy' ) {
34            vzctl_stop( $VEID );
35    }
36    
37    
38    if ( 1 || $action =~ m/(umount|destroy)/ ) {
39            my $clone;
40            open(my $m, '-|', 'mount') || die "can't open mount: $!";
41            while(<$m>) {
42                    my @v = split(/\s+/,$_);
43                    if ( $v[2] =~ m!/(private|root)/$VEID$! && $v[5] =~ m/bind/ ) {
44                            umount( $v[2] );
45                    } elsif ( $v[2] =~ m!vz-clone-$VEID$! ) {
46                            $clone = $v[2];
47                    }
48            }
49            umount( $clone ) if $clone;
50    }
51    
52    #die "unknown action: $action" unless $action eq 'create';
53    exit unless $action eq 'create';
54    
55  my $orig_conf = "/etc/vz/conf/$VEID.conf";  my $orig_conf = "/etc/vz/conf/$VEID.conf";
56    
57  die "$VEID config $orig_conf doesn't exist\n" unless -e $orig_conf;  die "$VEID config $orig_conf doesn't exist\n" unless -e $orig_conf;
# Line 26  while(<$m>) { Line 66  while(<$m>) {
66          my @v = split(/\s+/,$_);          my @v = split(/\s+/,$_);
67          if ( $v[1] =~ m/\Q$vz\E/ ) {          if ( $v[1] =~ m/\Q$vz\E/ ) {
68                  $vz_lv = $v[0];                  $vz_lv = $v[0];
69                  warn "found lv $vz_lv for $vz\n";                  warn "found LV $vz_lv for $vz\n";
70                  last;                  last;
71          }          }
72  }  }
73    
74    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  sub copy_files {  sub copy_files {
86          my ( $from, $to ) = @_;          my ( $from, $to ) = @_;
87          warn "rsync $from -> $to\n";          warn "rsync $from -> $to\n";
88          rsync('-ra', "$from/", "$to/" );          rsync('-raSHAD', "$from/", "$to/" );
89  }  }
90    
91  if ( $vz_lv ) {  if ( $vz_lv ) {
# Line 42  if ( $vz_lv ) { Line 93  if ( $vz_lv ) {
93          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!^(.+)/([^/]+)$! );
94    
95          my $snap = $vz_lv_name . '-snap';          my $snap = $vz_lv_name . '-snap';
96            my $clone = "vz-clone-$CVEID";
97    
98            sub do_mount {
99                    my ( $from, $to ) = @_;
100                    mkdir $to || die "can't create $to: $!";
101                    print "Mounting $from to $to\n";
102                    mount( $from, $to, '-o', 'noatime' );
103            }
104    
105            sub test_mkdir {
106                    my $dir = shift;
107                    if ( ! -d $dir ) {
108                            mkdir $dir || die "can't mkdir $dir: $!";
109                    }
110            }
111    
112            sub mount_bind {
113                    my ( $from, $to ) = @_;
114                    die "$from doesn't exist!" unless -d $from;
115                    test_mkdir( $to );
116                    mount( '--bind', $from, $to );
117            }
118    
119            if ( $use_snapshot ) {
120    
121                    print "Creating $snap_size snapshot $snap from $vz_lv\n";
122                    lvcreate( '--size', $snap_size, '--snapshot', '--name', $snap, $vz_lv );
123    
124                    do_mount( "$vz_lv_path/$snap", "/tmp/$snap" );
125                    
126          print "Creating $snap_size snapshot $snap from $vz_lv\n";          }
127    
128            my $clone_size = fs_quota( $VEID ) . 'k';
129            my $vg_name = $1 if ( $vz_lv_path =~ m!/([^/]+)/*$! );
130    
131            if ( lvdisplay( "$vz_lv_path/$clone" ) ) {
132                    warn "using existing $vz_lv_path/$clone\n";
133            } else {
134                    print "Creating LV $clone ($clone_size bytes) in VG $vg_name for $VEID clone filesystem\n";
135                    lvcreate( '--size', $clone_size, '--name', $clone, $vg_name );
136                    mke2fs( '-m', 0, '-j', "$vz_lv_path/$clone" );
137            }
138    
139            do_mount( "$vz_lv_path/$clone", "/tmp/$clone" );
140    
141          lvcreate( '--size', $snap_size, '--snapshot', '--name', $snap, $vz_lv );          test_mkdir( "/tmp/$clone/private" );
142            test_mkdir( "/tmp/$clone/root" );
143    
144          my $snap_path = "/tmp/$snap";          mount_bind( "/tmp/$clone/private", "$vz/private/$CVEID" );
145          mkdir $snap_path || die "can't create $snap_path: $!";          mount_bind( "/tmp/$clone/root", "$vz/root/$CVEID" );
146    
147          print "Mounting $vz_lv_path/$snap to $snap_path\n";          if ( $use_snapshot ) {
148    
149          mount( "$vz_lv_path/$snap", $snap_path, '-o', 'noatime' );                  copy_files( "/tmp/$snap/private/$VEID", "/tmp/$clone/private" );
150    
151          copy_files( "$snap_path/private/$VEID", "$vz/private/$CVEID" );                  print "Cleanup\n";
152    
153          print "Umount $snap_path and remove $vz_lv_path/$snap\n";                  #umount( "$vz_lv_path/$clone" );
154                    umount( "$vz_lv_path/$snap" );
155    
156          umount( $snap_path );                  lvremove( '-f', "$vz_lv_path/$snap" );
157    
158          lvremove( '-f', "$vz_lv_path/$snap" );          } else {
159                    copy_files( "$vz/private/$VEID", "/tmp/$clone/private" );
160    
161            }
162    
163  } else {  } else {
164    
Legend:
Removed from v.22  
changed lines
  Added in v.28
  ViewVC Help
Powered by ViewVC 1.1.26