/[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 24 by dpavlin, Thu Oct 4 18:54:41 2007 UTC revision 64 by dpavlin, Fri Feb 27 18:41:06 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/;  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|mount|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    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;
39            open(my $m, '-|', 'mount') || die "can't open mount: $!";
40            while(<$m>) {
41                    my @v = split(/\s+/,$_);
42                    if ( $v[2] =~ m{/(private|root)/$CVEID$} && $v[5] =~ m/bind/ ) {
43                            umount( $v[2] );
44                    } elsif ( $v[2] =~ m{vz-clone-$CVEID$} ) {
45                            $clone = $v[2];
46                    }
47            }
48            umount( $clone ) if $clone;
49    
50            exit if $action eq 'umount';
51    }
52    
53    
54  my $orig_conf = "/etc/vz/conf/$VEID.conf";  my $orig_conf = "/etc/vz/conf/$VEID.conf";
55    
56  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 44  sub fs_quota { Line 83  sub fs_quota {
83    
84  sub copy_files {  sub copy_files {
85          my ( $from, $to ) = @_;          my ( $from, $to ) = @_;
86            return if $action ne 'create';
87          warn "rsync $from -> $to\n";          warn "rsync $from -> $to\n";
88          rsync('-raSHAD', "$from/", "$to/" );          rsync('-raSHAD', '--delete', "$from/", "$to/" );
89  }  }
90    
91  if ( $vz_lv ) {  if ( $vz_lv ) {
92    
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_lv_name-clone-$CVEID";          my $clone = "vz-clone-$CVEID";
97    
98          sub do_mount {          sub do_mount {
99                  my ( $from, $to ) = @_;                  my ( $from, $to ) = @_;
# Line 76  if ( $vz_lv ) { Line 116  if ( $vz_lv ) {
116                  mount( '--bind', $from, $to );                  mount( '--bind', $from, $to );
117          }          }
118    
119          print "Creating $snap_size snapshot $snap from $vz_lv\n";          if ( $use_snapshot ) {
         lvcreate( '--size', $snap_size, '--snapshot', '--name', $snap, $vz_lv );  
120    
121          do_mount( "$vz_lv_path/$snap", "/tmp/$snap" );                  print "Creating $snap_size snapshot $snap from $vz_lv\n";
122                    lvcreate( '--size', $snap_size, '--snapshot', '--name', $snap, $vz_lv );
123    
124          my $clone_size = fs_quota( $VEID ) . 'k';                  do_mount( "$vz_lv_path/$snap", "/tmp/$snap" );
125          my $vg_name = $1 if ( $vz_lv_path =~ m!/([^/]+)/*$! );          
126            }
127    
128            my $vg_name = $1 if ( $vz_lv_path =~ m{/([^/]+)/*$} );
129    
130          if ( lvdisplay( "$vz_lv_path/$clone" ) ) {          if ( lvdisplay( "$vz_lv_path/$clone" ) ) {
131                  warn "using existing $vz_lv_path/$clone\n";                  warn "using existing $vz_lv_path/$clone\n";
132          } else {          } else {
133                    my $clone_size = fs_quota( $VEID ) . 'k';
134                  print "Creating LV $clone ($clone_size bytes) in VG $vg_name for $VEID clone filesystem\n";                  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 );                  lvcreate( '--size', $clone_size, '--name', $clone, $vg_name );
136                  mke2fs( '-m', 0, '-j', "$vz_lv_path/$clone" );                  mke2fs( '-m', 0, '-j', "$vz_lv_path/$clone" );
# Line 100  if ( $vz_lv ) { Line 144  if ( $vz_lv ) {
144          mount_bind( "/tmp/$clone/private", "$vz/private/$CVEID" );          mount_bind( "/tmp/$clone/private", "$vz/private/$CVEID" );
145          mount_bind( "/tmp/$clone/root", "$vz/root/$CVEID" );          mount_bind( "/tmp/$clone/root", "$vz/root/$CVEID" );
146    
147          copy_files( "/tmp/$snap/private/$VEID", "$vz/private/$CVEID" );          if ( $use_snapshot ) {
148    
149                    copy_files( "/tmp/$snap/private/$VEID", "/tmp/$clone/private" );
150    
151          print "Cleanup\n";                  print "Cleanup\n";
152    
153          #umount( "$vz_lv_path/$clone" );                  #umount( "$vz_lv_path/$clone" );
154          umount( "$vz_lv_path/$snap" );                  umount( "$vz_lv_path/$snap" );
155    
156          lvremove( '-f', "$vz_lv_path/$snap" );                  lvremove( '-f', "$vz_lv_path/$snap" );
157    
158            } else {
159                    copy_files( "$vz/private/$VEID", "/tmp/$clone/private" );
160    
161            }
162    
163  } else {  } else {
164    
165          copy_files( "$vz/private/$VEID", "$vz/private/$CVEID" );          copy_files( "$vz/private/$VEID", "$vz/private/$CVEID" );
166  }  }
167    
168    exit if $action eq 'mount';
169    
170  my $note = "# modified by $0\n";  my $note = "# modified by $0\n";
171    
172    my $ve_clone_ip;
173    
174  sub fix_ip {  sub fix_ip {
175          my $ip = shift;          my $ip = shift;
176          $ip =~ s/['"]//g;          $ip =~ s/['"]//g;
177          $ip =~ s/^\d+\./10./;  #       $ip =~ s/^\d+\./10./;
178            $ip =~ s/^\d+\.\d+\./192.168./;
179          warn "$CVEID new IP number: $ip\n";          warn "$CVEID new IP number: $ip\n";
180            $ve_clone_ip = $ip;
181          return $note . qq{IP_ADDRESS="$ip"};          return $note . qq{IP_ADDRESS="$ip"};
182  }  }
183    
# Line 135  sub fix_hostname { Line 192  sub fix_hostname {
192  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: $!";
193  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: $!";
194  while(<$o>) {  while(<$o>) {
195          s!^HOSTNAME=(.*)$!fix_hostname($1)!ie;          s{^HOSTNAME=(.*)$}{fix_hostname($1)}ie;
196          s!^IP_ADDRESS=(.*)$!fix_ip($1)!ie;          s{^IP_ADDRESS=(.*)$}{fix_ip($1)}ie;
197          s!^(ONBOOT=).*$!# modified by $0\n$1"no"!i;          s{^(ONBOOT=).*$}{# modified by $0\n$1"no"}i;
198          print $n $_;          print $n $_;
199  }  }
200    
201  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 "\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"
Legend:
Removed from v.24  
changed lines
  Added in v.64
  ViewVC Help
Powered by ViewVC 1.1.26