/[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

Annotation of /trunk/vz-clone.pl

Parent Directory Parent Directory | Revision Log Revision Log


Revision 62 - (hide annotations)
Fri Feb 27 18:37:34 2009 UTC (15 years, 1 month ago) by dpavlin
File MIME type: text/plain
File size: 4748 byte(s)
better s{}{} delimiters so we don't confuse vim syntax hightlight
1 dpavlin 20 #!/usr/bin/perl -w
2    
3     use strict;
4 dpavlin 25 use Shell qw/rsync vzlist lvcreate mount umount lvremove which lvdisplay mke2fs vzctl/;
5 dpavlin 20
6     my $vz = '/vz';
7     my $conf = '/etc/vz/conf';
8    
9 dpavlin 21 my $snap_size = '100M';
10    
11 dpavlin 25 my ( $action, $VEID ) = @ARGV;
12 dpavlin 20
13 dpavlin 27 my $use_snapshot = 0;
14    
15 dpavlin 29 my $valid_actions = 'create|mount|umount|destroy';
16 dpavlin 25
17     die "Usage: $0 ($valid_actions) VEID\n\n",vzlist unless $VEID && $action && $action =~ m/($valid_actions)/;
18    
19 dpavlin 27 my $CVEID = "10$VEID";
20    
21     sub vzctl_stop {
22     my $veid = shift;
23 dpavlin 25 my $vzlist = vzlist();
24 dpavlin 27 if ( $vzlist =~ m/$veid/ ) {
25     print "stop $veid\n";
26     vzctl('stop', $veid);
27 dpavlin 25 }
28     }
29    
30 dpavlin 27 if ( $action eq 'create' ) {
31     vzctl_stop( $CVEID );
32     } elsif ( $action eq 'destroy' ) {
33     vzctl_stop( $VEID );
34     }
35    
36    
37 dpavlin 30 if ( $action =~ m/(create|umount|destroy)/ ) {
38 dpavlin 25 my $clone;
39     open(my $m, '-|', 'mount') || die "can't open mount: $!";
40     while(<$m>) {
41     my @v = split(/\s+/,$_);
42 dpavlin 30 if ( $v[2] =~ m!/(private|root)/$CVEID$! && $v[5] =~ m/bind/ ) {
43 dpavlin 25 umount( $v[2] );
44 dpavlin 30 } elsif ( $v[2] =~ m!vz-clone-$CVEID$! ) {
45 dpavlin 25 $clone = $v[2];
46     }
47     }
48     umount( $clone ) if $clone;
49 dpavlin 30
50     exit if $action eq 'umount';
51 dpavlin 25 }
52    
53    
54 dpavlin 20 my $orig_conf = "/etc/vz/conf/$VEID.conf";
55    
56     die "$VEID config $orig_conf doesn't exist\n" unless -e $orig_conf;
57    
58     print "Clone VE $VEID -> $CVEID\n";
59    
60 dpavlin 21 my $vz_lv;
61 dpavlin 20
62 dpavlin 21 open(my $m, '<', '/etc/fstab') || die "can't open /etc/fstab: $!";
63     while(<$m>) {
64     next if m/^#/;
65     my @v = split(/\s+/,$_);
66     if ( $v[1] =~ m/\Q$vz\E/ ) {
67     $vz_lv = $v[0];
68 dpavlin 23 warn "found LV $vz_lv for $vz\n";
69 dpavlin 21 last;
70     }
71     }
72    
73 dpavlin 23 sub fs_quota {
74     my $id = shift;
75     open(my $q, '-|', "vzquota -b show $id") || die "can't exec vzquota show $id: $!";
76     my $l = <$q>;
77     $l =~ s/^\s+//;
78     my ( $usage, $soft, $hard ) = split(/\s+/,$l);
79     warn "quota for $id | $soft < $hard | usage: $usage\n";
80     return ( $usage, $soft, $hard ) if wantarray;
81     return $soft;
82     }
83    
84 dpavlin 21 sub copy_files {
85     my ( $from, $to ) = @_;
86 dpavlin 29 return if $action ne 'create';
87 dpavlin 24 warn "rsync $from -> $to\n";
88 dpavlin 31 rsync('-raSHAD', '--delete', "$from/", "$to/" );
89 dpavlin 21 }
90    
91     if ( $vz_lv ) {
92    
93     my ( $vz_lv_path, $vz_lv_name ) = ( $1, $2 ) if ( $vz_lv =~ m!^(.+)/([^/]+)$! );
94    
95     my $snap = $vz_lv_name . '-snap';
96 dpavlin 26 my $clone = "vz-clone-$CVEID";
97 dpavlin 23
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 dpavlin 27 if ( $use_snapshot ) {
120 dpavlin 21
121 dpavlin 27 print "Creating $snap_size snapshot $snap from $vz_lv\n";
122     lvcreate( '--size', $snap_size, '--snapshot', '--name', $snap, $vz_lv );
123 dpavlin 21
124 dpavlin 27 do_mount( "$vz_lv_path/$snap", "/tmp/$snap" );
125    
126     }
127    
128 dpavlin 23 my $vg_name = $1 if ( $vz_lv_path =~ m!/([^/]+)/*$! );
129 dpavlin 21
130 dpavlin 23 if ( lvdisplay( "$vz_lv_path/$clone" ) ) {
131     warn "using existing $vz_lv_path/$clone\n";
132     } else {
133 dpavlin 29 my $clone_size = fs_quota( $VEID ) . 'k';
134 dpavlin 23 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 dpavlin 21
139 dpavlin 23 do_mount( "$vz_lv_path/$clone", "/tmp/$clone" );
140 dpavlin 21
141 dpavlin 23 test_mkdir( "/tmp/$clone/private" );
142     test_mkdir( "/tmp/$clone/root" );
143 dpavlin 21
144 dpavlin 23 mount_bind( "/tmp/$clone/private", "$vz/private/$CVEID" );
145     mount_bind( "/tmp/$clone/root", "$vz/root/$CVEID" );
146 dpavlin 21
147 dpavlin 27 if ( $use_snapshot ) {
148 dpavlin 23
149 dpavlin 28 copy_files( "/tmp/$snap/private/$VEID", "/tmp/$clone/private" );
150 dpavlin 23
151 dpavlin 27 print "Cleanup\n";
152 dpavlin 23
153 dpavlin 27 #umount( "$vz_lv_path/$clone" );
154     umount( "$vz_lv_path/$snap" );
155 dpavlin 21
156 dpavlin 27 lvremove( '-f', "$vz_lv_path/$snap" );
157    
158     } else {
159 dpavlin 28 copy_files( "$vz/private/$VEID", "/tmp/$clone/private" );
160 dpavlin 27
161     }
162    
163 dpavlin 21 } else {
164    
165     copy_files( "$vz/private/$VEID", "$vz/private/$CVEID" );
166     }
167    
168 dpavlin 29 exit if $action eq 'mount';
169    
170 dpavlin 21 my $note = "# modified by $0\n";
171    
172 dpavlin 50 my $ve_clone_ip;
173    
174 dpavlin 20 sub fix_ip {
175     my $ip = shift;
176 dpavlin 21 $ip =~ s/['"]//g;
177 dpavlin 33 # $ip =~ s/^\d+\./10./;
178     $ip =~ s/^\d+\.\d+\./192.168./;
179 dpavlin 20 warn "$CVEID new IP number: $ip\n";
180 dpavlin 50 $ve_clone_ip = $ip;
181 dpavlin 21 return $note . qq{IP_ADDRESS="$ip"};
182 dpavlin 20 }
183    
184 dpavlin 21 sub fix_hostname {
185     my $hostname = shift;
186     $hostname =~ s/['"]//g;
187     $hostname = "clone-$hostname";
188     warn "$CVEID new hostname: $hostname\n";
189     return $note . qq{HOSTNAME="$hostname"};
190     }
191    
192 dpavlin 20 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: $!";
194     while(<$o>) {
195 dpavlin 62 s{^HOSTNAME=(.*)$}{fix_hostname($1)}ie;
196     s{^IP_ADDRESS=(.*)$}{fix_ip($1)}ie;
197     s{^(ONBOOT=).*$}{# modified by $0\n$1"no"}i;
198 dpavlin 20 print $n $_;
199     }
200    
201 dpavlin 50 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"

Properties

Name Value
svn:executable *

  ViewVC Help
Powered by ViewVC 1.1.26