/[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 26 by dpavlin, Thu Oct 4 23:23:20 2007 UTC revision 69 by dpavlin, Tue Mar 3 20:58:48 2009 UTC
# Line 6  use Shell qw/rsync vzlist lvcreate mount Line 6  use Shell qw/rsync vzlist lvcreate mount
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 = '10G';
10    
11  my ( $action, $VEID ) = @ARGV;  my ( $action, $VEID ) = @ARGV;
12    
13  my $valid_actions = 'create|umount|destroy';  my $use_snapshot = 1;
14    
15    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  if ( $action eq 'destroy' ) {  my $CVEID = "10$VEID";
20    
21    sub vzctl_stop {
22            my $veid = shift;
23          my $vzlist = vzlist();          my $vzlist = vzlist();
24          if ( $vzlist =~ m/$VEID/ ) {          if ( $vzlist =~ m/$veid/ ) {
25                  print "stop $VEID\n";                  print "stop $veid\n";
26                  vzctl('stop', $VEID);                  vzctl('stop', $veid);
27          }          }
28  }  }
29    
30  if ( $action =~ m/(umount|destroy)/ ) {  if ( $action eq 'create' ) {
31            vzctl_stop( $CVEID );
32    } elsif ( $action eq 'destroy' ) {
33            vzctl_stop( $VEID );
34    }
35    
36    
37    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)/$VEID$! && $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-$VEID$! ) {                  } 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  #die "unknown action: $action" unless $action eq 'create';          vzctl('destroy', $VEID) if $action eq 'destroy';
58  exit unless $action eq 'create';  
59            exit if $action =~ m{(umount|destroy)};
60    }
61    
 my $CVEID = "10$VEID";  
62    
63  my $orig_conf = "/etc/vz/conf/$VEID.conf";  my $orig_conf = "/etc/vz/conf/$VEID.conf";
64    
# Line 73  sub fs_quota { Line 92  sub fs_quota {
92    
93  sub copy_files {  sub copy_files {
94          my ( $from, $to ) = @_;          my ( $from, $to ) = @_;
95            return if $action ne 'create';
96          warn "rsync $from -> $to\n";          warn "rsync $from -> $to\n";
97          rsync('-raSHAD', "$from/", "$to/" );          rsync('-raSHAD', '--delete', "$from/", "$to/" );
98  }  }
99    
100  if ( $vz_lv ) {  if ( $vz_lv ) {
101    
102          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{^(.+)/([^/]+)$} );
103    
104          my $snap = $vz_lv_name . '-snap';          my $snap = $vz_lv_name . '-snap';
105          my $clone = "vz-clone-$CVEID";          my $clone = "vz-clone-$CVEID";
# Line 105  if ( $vz_lv ) { Line 125  if ( $vz_lv ) {
125                  mount( '--bind', $from, $to );                  mount( '--bind', $from, $to );
126          }          }
127    
128          print "Creating $snap_size snapshot $snap from $vz_lv\n";          if ( $use_snapshot ) {
         lvcreate( '--size', $snap_size, '--snapshot', '--name', $snap, $vz_lv );  
129    
130          do_mount( "$vz_lv_path/$snap", "/tmp/$snap" );                  print "Creating $snap_size snapshot $snap from $vz_lv\n";
131                    lvcreate( '--size', $snap_size, '--snapshot', '--name', $snap, $vz_lv );
132    
133          my $clone_size = fs_quota( $VEID ) . 'k';                  do_mount( "$vz_lv_path/$snap", "/tmp/$snap" );
134          my $vg_name = $1 if ( $vz_lv_path =~ m!/([^/]+)/*$! );          
135                    mount_bind( "/tmp/$snap/private/$VEID", "$vz/private/$CVEID" );
136                    mount_bind( "/tmp/$snap/root/$VEID", "$vz/root/$CVEID" );
137    
         if ( lvdisplay( "$vz_lv_path/$clone" ) ) {  
                 warn "using existing $vz_lv_path/$clone\n";  
138          } else {          } 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" );  
139    
140          test_mkdir( "/tmp/$clone/private" );                  if ( lvdisplay( "$vz_lv_path/$clone" ) ) {
141          test_mkdir( "/tmp/$clone/root" );                          warn "using existing $vz_lv_path/$clone\n";
142                    } else {
143                            my $vg_name = $1 if ( $vz_lv_path =~ m{/([^/]+)/*$} );
144                            my $clone_size = fs_quota( $VEID ) . 'k';
145                            print "Creating LV $clone ($clone_size bytes) in VG $vg_name for $VEID clone filesystem\n";
146                            lvcreate( '--size', $clone_size, '--name', $clone, $vg_name );
147                            mke2fs( '-m', 0, '-j', "$vz_lv_path/$clone" );
148                    }
149    
150          mount_bind( "/tmp/$clone/private", "$vz/private/$CVEID" );                  do_mount( "$vz_lv_path/$clone", "/tmp/$clone" );
         mount_bind( "/tmp/$clone/root", "$vz/root/$CVEID" );  
151    
152          copy_files( "/tmp/$snap/private/$VEID", "$vz/private/$CVEID" );                  test_mkdir( "/tmp/$clone/private" );
153                    test_mkdir( "/tmp/$clone/root" );
154    
155          print "Cleanup\n";                  copy_files( "$vz/private/$VEID", "/tmp/$clone/private" );
156    
157          #umount( "$vz_lv_path/$clone" );                  mount_bind( "/tmp/$clone/private", "$vz/private/$CVEID" );
158          umount( "$vz_lv_path/$snap" );                  mount_bind( "/tmp/$clone/root", "$vz/root/$CVEID" );
159    
160          lvremove( '-f', "$vz_lv_path/$snap" );          }
161    
162  } else {  } else {
163    
164          copy_files( "$vz/private/$VEID", "$vz/private/$CVEID" );          copy_files( "$vz/private/$VEID", "$vz/private/$CVEID" );
165  }  }
166    
167    exit if $action eq 'mount';
168    
169  my $note = "# modified by $0\n";  my $note = "# modified by $0\n";
170    
171    my $ve_clone_ip;
172    
173  sub fix_ip {  sub fix_ip {
174          my $ip = shift;          my $ip = shift;
175          $ip =~ s/['"]//g;          $ip =~ s/['"]//g;
176          $ip =~ s/^\d+\./10./;  #       $ip =~ s/^\d+\./10./;
177            $ip =~ s/^\d+\.\d+\./192.168./;
178          warn "$CVEID new IP number: $ip\n";          warn "$CVEID new IP number: $ip\n";
179            $ve_clone_ip = $ip;
180          return $note . qq{IP_ADDRESS="$ip"};          return $note . qq{IP_ADDRESS="$ip"};
181  }  }
182    
# Line 164  sub fix_hostname { Line 191  sub fix_hostname {
191  open(my $o, '<', "$conf/$VEID.conf")  || die "can't open $conf/$VEID.conf: $!";  open(my $o, '<', "$conf/$VEID.conf")  || die "can't open $conf/$VEID.conf: $!";
192  open(my $n, '>', "$conf/$CVEID.conf") || die "can't open $conf/$CVEID.conf: $!";  open(my $n, '>', "$conf/$CVEID.conf") || die "can't open $conf/$CVEID.conf: $!";
193  while(<$o>) {  while(<$o>) {
194          s!^HOSTNAME=(.*)$!fix_hostname($1)!ie;          s{^HOSTNAME=(.*)$}{fix_hostname($1)}ie;
195          s!^IP_ADDRESS=(.*)$!fix_ip($1)!ie;          s{^IP_ADDRESS=(.*)$}{fix_ip($1)}ie;
196          s!^(ONBOOT=).*$!# modified by $0\n$1"no"!i;          s{^(ONBOOT=).*$}{# modified by $0\n$1"no"}i;
197          print $n $_;          print $n $_;
198  }  }
199    
200  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
201    
202    my $iptables="iptables -t nat -A POSTROUTING -s $ve_clone_ip -o eth0 -j MASQUERADE";
203    system( $iptables );
204    print "\nConfig file: $conf/$CVEID.conf\nAdded NAT for new VE with: $iptables\nStart clone of $VEID with: vzctl start $CVEID\n"
205    
Legend:
Removed from v.26  
changed lines
  Added in v.69
  ViewVC Help
Powered by ViewVC 1.1.26