/[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 64 by dpavlin, Fri Feb 27 18:41:06 2009 UTC revision 73 by dpavlin, Mon Mar 9 19:04:19 2009 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 which lvdisplay mke2fs vzctl/;  use Shell qw/rsync lvcreate mount umount lvremove which lvdisplay mke2fs/;
5    
6  my $vz   = '/vz';  use lib 'lib';
7  my $conf = '/etc/vz/conf';  use VZ;
8    
9  my $snap_size = '100M';  my $snap_size = '10G';
10    
11  my ( $action, $VEID ) = @ARGV;  my ( $action, $VEID ) = @ARGV;
12    
13  my $use_snapshot = 0;  my $use_snapshot = 1;
14    
15  my $valid_actions = 'create|mount|umount|destroy';  my $valid_actions = 'create|mount|umount|destroy';
16    
17  die "Usage: $0 ($valid_actions) VEID\n\n",vzlist unless $VEID && $action && $action =~ m/($valid_actions)/;  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    
# Line 34  if ( $action eq 'create' ) { Line 34  if ( $action eq 'create' ) {
34  }  }
35    
36    
37  if ( $action =~ m/(create|umount|destroy)/ ) {  if ( $action =~ m{(create|umount|destroy)} ) {
38          my $clone;          my $clone;
39            my $snap;
40          open(my $m, '-|', 'mount') || die "can't open mount: $!";          open(my $m, '-|', 'mount') || die "can't open mount: $!";
41          while(<$m>) {          while(<$m>) {
42                  my @v = split(/\s+/,$_);                  my @v = split(/\s+/,$_);
43                  if ( $v[2] =~ m{/(private|root)/$CVEID$} && $v[5] =~ m/bind/ ) {                  if ( $v[2] =~ m{/(private|root)/($VEID|$CVEID)$} && $v[5] =~ m{bind} ) {
44                          umount( $v[2] );                          umount( $v[2] );
45                  } elsif ( $v[2] =~ m{vz-clone-$CVEID$} ) {                  } elsif ( $v[2] =~ m{vz-clone-$CVEID$} ) {
46                          $clone = $v[2];                          $clone = $v[2];
47                    } elsif ( $v[2] =~ m{-snap} ) {
48                            $snap = $v[0];
49                            umount( $snap );
50                  }                  }
51          }          }
52          umount( $clone ) if $clone;          umount( $clone ) if $clone;
53            if ( $action eq 'destroy' && -e $snap ) {
54                    lvremove( '-f', $snap );
55            }
56    
57            vzctl('destroy', $VEID) if $action eq 'destroy';
58    
59          exit if $action eq 'umount';          exit if $action =~ m{(umount|destroy)};
60  }  }
61    
62    
# Line 59  print "Clone VE $VEID -> $CVEID\n"; Line 68  print "Clone VE $VEID -> $CVEID\n";
68    
69  my $vz_lv;  my $vz_lv;
70    
71  open(my $m, '<', '/etc/fstab') || die "can't open /etc/fstab: $!";  open(my $m, '-|', "df $vz_root") || die "can't open df $vz_root: $!";
72  while(<$m>) {  while(<$m>) {
73          next if m/^#/;          next if ! m{^/};
74          my @v = split(/\s+/,$_);          my @v = split(/\s+/,$_);
75          if ( $v[1] =~ m/\Q$vz\E/ ) {          $vz_lv = $v[0];
76                  $vz_lv = $v[0];          warn "found LV $vz_lv for $vz_root\n";
77                  warn "found LV $vz_lv for $vz\n";          last;
                 last;  
         }  
78  }  }
79    
80  sub fs_quota {  sub fs_quota {
# Line 123  if ( $vz_lv ) { Line 130  if ( $vz_lv ) {
130    
131                  do_mount( "$vz_lv_path/$snap", "/tmp/$snap" );                  do_mount( "$vz_lv_path/$snap", "/tmp/$snap" );
132                    
133          }                  mount_bind( "/tmp/$snap/private/$VEID", "$vz_root/private/$CVEID" );
134                    mount_bind( "/tmp/$snap/root/$VEID", "$vz_root/root/$CVEID" );
135    
         my $vg_name = $1 if ( $vz_lv_path =~ m{/([^/]+)/*$} );  
   
         if ( lvdisplay( "$vz_lv_path/$clone" ) ) {  
                 warn "using existing $vz_lv_path/$clone\n";  
136          } else {          } else {
                 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" );  
137    
138          mount_bind( "/tmp/$clone/private", "$vz/private/$CVEID" );                  if ( lvdisplay( "$vz_lv_path/$clone" ) ) {
139          mount_bind( "/tmp/$clone/root", "$vz/root/$CVEID" );                          warn "using existing $vz_lv_path/$clone\n";
140                    } else {
141          if ( $use_snapshot ) {                          my $vg_name = $1 if ( $vz_lv_path =~ m{/([^/]+)/*$} );
142                            my $clone_size = fs_quota( $VEID ) . 'k';
143                            print "Creating LV $clone ($clone_size bytes) in VG $vg_name for $VEID clone filesystem\n";
144                            lvcreate( '--size', $clone_size, '--name', $clone, $vg_name );
145                            mke2fs( '-m', 0, '-j', "$vz_lv_path/$clone" );
146                    }
147    
148                  copy_files( "/tmp/$snap/private/$VEID", "/tmp/$clone/private" );                  do_mount( "$vz_lv_path/$clone", "/tmp/$clone" );
149    
150                  print "Cleanup\n";                  test_mkdir( "/tmp/$clone/private" );
151                    test_mkdir( "/tmp/$clone/root" );
152    
153                  #umount( "$vz_lv_path/$clone" );                  copy_files( "$vz_root/private/$VEID", "/tmp/$clone/private" );
                 umount( "$vz_lv_path/$snap" );  
154    
155                  lvremove( '-f', "$vz_lv_path/$snap" );                  mount_bind( "/tmp/$clone/private", "$vz_root/private/$CVEID" );
156                    mount_bind( "/tmp/$clone/root", "$vz_root/root/$CVEID" );
         } else {  
                 copy_files( "$vz/private/$VEID", "/tmp/$clone/private" );  
157    
158          }          }
159    
160  } else {  } else {
161    
162          copy_files( "$vz/private/$VEID", "$vz/private/$CVEID" );          copy_files( "$vz_root/private/$VEID", "$vz_root/private/$CVEID" );
163  }  }
164    
165  exit if $action eq 'mount';  exit if $action eq 'mount';
# Line 189  sub fix_hostname { Line 186  sub fix_hostname {
186          return $note . qq{HOSTNAME="$hostname"};          return $note . qq{HOSTNAME="$hostname"};
187  }  }
188    
189  open(my $o, '<', "$conf/$VEID.conf")  || die "can't open $conf/$VEID.conf: $!";  open(my $o, '<', "$vz_conf/$VEID.conf")  || die "can't open $vz_conf/$VEID.conf: $!";
190  open(my $n, '>', "$conf/$CVEID.conf") || die "can't open $conf/$CVEID.conf: $!";  open(my $n, '>', "$vz_conf/$CVEID.conf") || die "can't open $vz_conf/$CVEID.conf: $!";
191  while(<$o>) {  while(<$o>) {
192          s{^HOSTNAME=(.*)$}{fix_hostname($1)}ie;          s{^HOSTNAME=(.*)$}{fix_hostname($1)}ie;
193          s{^IP_ADDRESS=(.*)$}{fix_ip($1)}ie;          s{^IP_ADDRESS=(.*)$}{fix_ip($1)}ie;
# Line 198  while(<$o>) { Line 195  while(<$o>) {
195          print $n $_;          print $n $_;
196  }  }
197    
198  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"  print $n qq{DISK_QUOTA="no"\n} if $use_snapshot; # don't wait for quota rebuild on startup
199    
200    my $iptables="iptables -t nat -A POSTROUTING -s $ve_clone_ip -o eth0 -j MASQUERADE";
201    system( $iptables );
202    print "\nConfig file: $vz_conf/$CVEID.conf\nAdded NAT for new VE with: $iptables\nStart clone of $VEID with: vzctl start $CVEID\n"
203    
Legend:
Removed from v.64  
changed lines
  Added in v.73
  ViewVC Help
Powered by ViewVC 1.1.26