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

Contents of /trunk/vz-clone.pl

Parent Directory Parent Directory | Revision Log Revision Log


Revision 31 - (show annotations)
Thu Oct 11 19:28:27 2007 UTC (16 years, 5 months ago) by dpavlin
File MIME type: text/plain
File size: 4656 byte(s)
delete files on cloned VE which no longer exist on master to create perfect copy
1 #!/usr/bin/perl -w
2
3 use strict;
4 use Shell qw/rsync vzlist lvcreate mount umount lvremove which lvdisplay mke2fs vzctl/;
5
6 my $vz = '/vz';
7 my $conf = '/etc/vz/conf';
8
9 my $snap_size = '100M';
10
11 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";
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";
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 my $vz_lv;
61
62 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 warn "found LV $vz_lv for $vz\n";
69 last;
70 }
71 }
72
73 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 sub copy_files {
85 my ( $from, $to ) = @_;
86 return if $action ne 'create';
87 warn "rsync $from -> $to\n";
88 rsync('-raSHAD', '--delete', "$from/", "$to/" );
89 }
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 my $clone = "vz-clone-$CVEID";
97
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 if ( $use_snapshot ) {
120
121 print "Creating $snap_size snapshot $snap from $vz_lv\n";
122 lvcreate( '--size', $snap_size, '--snapshot', '--name', $snap, $vz_lv );
123
124 do_mount( "$vz_lv_path/$snap", "/tmp/$snap" );
125
126 }
127
128 my $vg_name = $1 if ( $vz_lv_path =~ m!/([^/]+)/*$! );
129
130 if ( lvdisplay( "$vz_lv_path/$clone" ) ) {
131 warn "using existing $vz_lv_path/$clone\n";
132 } 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";
135 lvcreate( '--size', $clone_size, '--name', $clone, $vg_name );
136 mke2fs( '-m', 0, '-j', "$vz_lv_path/$clone" );
137 }
138
139 do_mount( "$vz_lv_path/$clone", "/tmp/$clone" );
140
141 test_mkdir( "/tmp/$clone/private" );
142 test_mkdir( "/tmp/$clone/root" );
143
144 mount_bind( "/tmp/$clone/private", "$vz/private/$CVEID" );
145 mount_bind( "/tmp/$clone/root", "$vz/root/$CVEID" );
146
147 if ( $use_snapshot ) {
148
149 copy_files( "/tmp/$snap/private/$VEID", "/tmp/$clone/private" );
150
151 print "Cleanup\n";
152
153 #umount( "$vz_lv_path/$clone" );
154 umount( "$vz_lv_path/$snap" );
155
156 lvremove( '-f', "$vz_lv_path/$snap" );
157
158 } else {
159 copy_files( "$vz/private/$VEID", "/tmp/$clone/private" );
160
161 }
162
163 } else {
164
165 copy_files( "$vz/private/$VEID", "$vz/private/$CVEID" );
166 }
167
168 exit if $action eq 'mount';
169
170 my $note = "# modified by $0\n";
171
172 sub fix_ip {
173 my $ip = shift;
174 $ip =~ s/['"]//g;
175 $ip =~ s/^\d+\./10./;
176 warn "$CVEID new IP number: $ip\n";
177 return $note . qq{IP_ADDRESS="$ip"};
178 }
179
180 sub fix_hostname {
181 my $hostname = shift;
182 $hostname =~ s/['"]//g;
183 $hostname = "clone-$hostname";
184 warn "$CVEID new hostname: $hostname\n";
185 return $note . qq{HOSTNAME="$hostname"};
186 }
187
188 open(my $o, '<', "$conf/$VEID.conf") || die "can't open $conf/$VEID.conf: $!";
189 open(my $n, '>', "$conf/$CVEID.conf") || die "can't open $conf/$CVEID.conf: $!";
190 while(<$o>) {
191 s!^HOSTNAME=(.*)$!fix_hostname($1)!ie;
192 s!^IP_ADDRESS=(.*)$!fix_ip($1)!ie;
193 s!^(ONBOOT=).*$!# modified by $0\n$1"no"!i;
194 print $n $_;
195 }
196
197 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"

Properties

Name Value
svn:executable *

  ViewVC Help
Powered by ViewVC 1.1.26