/[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 20 by dpavlin, Thu Oct 4 11:30:05 2007 UTC revision 66 by dpavlin, Fri Feb 27 20:49: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/;  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 $VEID = shift @ARGV || die "Usage: $0 VEID\n\n",vzlist;  my $snap_size = '1G';
10    
11    my ( $action, $VEID ) = @ARGV;
12    
13    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)/;
18    
19    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            my $snap;
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|$CVEID)$} && $v[5] =~ m{bind} ) {
44                            umount( $v[2] );
45                    } elsif ( $v[2] =~ m{vz-clone-$CVEID$} ) {
46                            $clone = $v[2];
47                    } elsif ( $v[2] =~ m{-snap} ) {
48                            $snap = $v[0];
49                            umount( $snap );
50                    }
51            }
52            umount( $clone ) if $clone;
53            if ( $action eq 'destroy' && -e $snap ) {
54                    lvremove( '-f', $snap );
55                    exit;
56            }
57    
58            exit if $action eq 'umount';
59    }
60    
 my $CVEID = "99$VEID";  
61    
62  my $orig_conf = "/etc/vz/conf/$VEID.conf";  my $orig_conf = "/etc/vz/conf/$VEID.conf";
63    
# Line 16  die "$VEID config $orig_conf doesn't exi Line 65  die "$VEID config $orig_conf doesn't exi
65    
66  print "Clone VE $VEID -> $CVEID\n";  print "Clone VE $VEID -> $CVEID\n";
67    
68  rsync('-ra', "$vz/private/$VEID", "$vz/private/$CVEID" );  my $vz_lv;
69    
70    open(my $m, '<', '/etc/fstab') || die "can't open /etc/fstab: $!";
71    while(<$m>) {
72            next if m/^#/;
73            my @v = split(/\s+/,$_);
74            if ( $v[1] =~ m/\Q$vz\E/ ) {
75                    $vz_lv = $v[0];
76                    warn "found LV $vz_lv for $vz\n";
77                    last;
78            }
79    }
80    
81    sub fs_quota {
82            my $id = shift;
83            open(my $q, '-|', "vzquota -b show $id") || die "can't exec vzquota show $id: $!";
84            my $l = <$q>;
85            $l =~ s/^\s+//;
86            my ( $usage, $soft, $hard ) = split(/\s+/,$l);
87            warn "quota for $id | $soft < $hard | usage: $usage\n";
88            return ( $usage, $soft, $hard ) if wantarray;
89            return $soft;
90    }
91    
92    sub copy_files {
93            my ( $from, $to ) = @_;
94            return if $action ne 'create';
95            warn "rsync $from -> $to\n";
96            rsync('-raSHAD', '--delete', "$from/", "$to/" );
97    }
98    
99    if ( $vz_lv ) {
100    
101            my ( $vz_lv_path, $vz_lv_name ) = ( $1, $2 ) if ( $vz_lv =~ m{^(.+)/([^/]+)$} );
102    
103            my $snap = $vz_lv_name . '-snap';
104            my $clone = "vz-clone-$CVEID";
105    
106            sub do_mount {
107                    my ( $from, $to ) = @_;
108                    mkdir $to || die "can't create $to: $!";
109                    print "Mounting $from to $to\n";
110                    mount( $from, $to, '-o', 'noatime' );
111            }
112    
113            sub test_mkdir {
114                    my $dir = shift;
115                    if ( ! -d $dir ) {
116                            mkdir $dir || die "can't mkdir $dir: $!";
117                    }
118            }
119    
120            sub mount_bind {
121                    my ( $from, $to ) = @_;
122                    die "$from doesn't exist!" unless -d $from;
123                    test_mkdir( $to );
124                    mount( '--bind', $from, $to );
125            }
126    
127            if ( $use_snapshot ) {
128    
129                    print "Creating $snap_size snapshot $snap from $vz_lv\n";
130                    lvcreate( '--size', $snap_size, '--snapshot', '--name', $snap, $vz_lv );
131    
132                    do_mount( "$vz_lv_path/$snap", "/tmp/$snap" );
133            
134                    mount_bind( "/tmp/$snap/private/$VEID", "$vz/private/$CVEID" );
135                    mount_bind( "/tmp/$snap/root/$VEID", "$vz/root/$CVEID" );
136    
137            } else {
138    
139                    if ( lvdisplay( "$vz_lv_path/$clone" ) ) {
140                            warn "using existing $vz_lv_path/$clone\n";
141                    } else {
142                            my $vg_name = $1 if ( $vz_lv_path =~ m{/([^/]+)/*$} );
143                            my $clone_size = fs_quota( $VEID ) . 'k';
144                            print "Creating LV $clone ($clone_size bytes) in VG $vg_name for $VEID clone filesystem\n";
145                            lvcreate( '--size', $clone_size, '--name', $clone, $vg_name );
146                            mke2fs( '-m', 0, '-j', "$vz_lv_path/$clone" );
147                    }
148    
149                    do_mount( "$vz_lv_path/$clone", "/tmp/$clone" );
150    
151                    test_mkdir( "/tmp/$clone/private" );
152                    test_mkdir( "/tmp/$clone/root" );
153    
154                    copy_files( "$vz/private/$VEID", "/tmp/$clone/private" );
155    
156                    mount_bind( "/tmp/$clone/private", "$vz/private/$CVEID" );
157                    mount_bind( "/tmp/$clone/root", "$vz/root/$CVEID" );
158    
159            }
160    
161    } else {
162    
163            copy_files( "$vz/private/$VEID", "$vz/private/$CVEID" );
164    }
165    
166    exit if $action eq 'mount';
167    
168    my $note = "# modified by $0\n";
169    
170    my $ve_clone_ip;
171    
172  sub fix_ip {  sub fix_ip {
173          my $ip = shift;          my $ip = shift;
174          $ip =~ s/^\d+\./10./;          $ip =~ s/['"]//g;
175    #       $ip =~ s/^\d+\./10./;
176            $ip =~ s/^\d+\.\d+\./192.168./;
177          warn "$CVEID new IP number: $ip\n";          warn "$CVEID new IP number: $ip\n";
178          return qq{IP_ADDRESS=$ip};          $ve_clone_ip = $ip;
179            return $note . qq{IP_ADDRESS="$ip"};
180    }
181    
182    sub fix_hostname {
183            my $hostname = shift;
184            $hostname =~ s/['"]//g;
185            $hostname = "clone-$hostname";
186            warn "$CVEID new hostname: $hostname\n";
187            return $note . qq{HOSTNAME="$hostname"};
188  }  }
189    
190  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: $!";
191  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: $!";
192  while(<$o>) {  while(<$o>) {
193          s!^(HOSTNAME=)(.*)$!$1clone-$2!i;          s{^HOSTNAME=(.*)$}{fix_hostname($1)}ie;
194          s!^(IP_ADDRESS=.*)$!fix_up($1)!e;          s{^IP_ADDRESS=(.*)$}{fix_ip($1)}ie;
195          s!^(ONBOOT=).*$!# modified by $0\n$1="no"!gi;          s{^(ONBOOT=).*$}{# modified by $0\n$1"no"}i;
196          print $n $_;          print $n $_;
197  }  }
198    
199  print "Please edit config $conf/$CVEID.conf before vzctl start $CVEID\n"  print $n qq{DISK_QUOTA="no"\n} if $use_snapshot; # don't wait for quota rebuild on startup
200    
201    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"
202    
Legend:
Removed from v.20  
changed lines
  Added in v.66
  ViewVC Help
Powered by ViewVC 1.1.26