High5! Code

Blob

Date:: Mon Jul 18 16:21:15 2022 UTC
Message:: Initial commit
Actions:: History | Blame | Raw File
1 #!/usr/bin/perl -w
2 # -*- Perl -*-
3 #
4 # This is in CVS!
5 # $Header$
6 
7 my($VERSION) = q$Revision$ =~ /:\s*(\d+(?:\.\d+)?)/;
8 
9 use strict;
10 use File::Copy;
11 
12 my $usage = "
13 Usage for rvi $VERSION:
14 
15   $0 [-r] [+] [-] filename...
16 
17 -r    Enter RCS description
18 +     auto-increment SOA serial number (default)
19 -     do NOT auto-increment SOA serial number.
20 ";
21 
22 @ARGV || die($usage);
23 
24 $ENV{EDITOR} = "vi" unless exists $ENV{EDITOR};
25 
26 my $autoinc = 1; # now default
27 my $setdescr = 0;
28 
29 my $euid = $>;
30 my $egid = (split(/\s+/,$)))[0];
31 my $user = getlogin() || $ENV{USER} || getpwuid($<);
32 
33 my $namedpath = "/usr/local/etc/named";
34 my $ndc;
35 if ($ENV{RVI_AUTO_NDC_RELOAD} or $ENV{RVI_AUTO_NDC_RECONFIG}) {
36     $ndc = which("ndc") or exit 1;
37 }
38 my $ci = which("ci") or exit 1;
39 my $co = which("co") or exit 1;
40 my $cmp = which("cmp") or exit 1;
41 my $rcsdiff = which("rcsdiff") or exit 1;
42 
43 while ( 1 ) {
44     $ARGV[0] eq "+" && do {
45 	$autoinc = 1;
46 	shift;
47 	next;
48     };
49     $ARGV[0] eq "-" && do {
50 	$autoinc = 0;
51 	shift;
52 	next;
53     };
54     $ARGV[0] eq "-r" && do {
55 	$setdescr = 1;
56 	shift;
57 	next;
58     };
59     last;
60 }
61 
62 my($file, %serial, %origfile, %checked, %fileinfo, %rcsinfo);
63 
64 my $wd = `pwd`;
65 chomp($wd);
66 
67 foreach $file ( @ARGV ) {
68     my $fullname = makefullname($wd, $file);
69     ($fileinfo{$file}, $rcsinfo{$file}) = getinfo($fullname);
70     if ($rcsinfo{$file}->{'exists'}) {
71 	unless (execute_ok($rcsdiff,"-c",$file)) {
72 	    warn "$file would change when checking out of RCS!\n";
73 	    exit cleanmess();
74 	}
75 	unless (execute_ok($co,"-l",$file)) {
76 	    exit cleanmess();
77 	}
78     }
79 
80     $checked{$file} = 1;
81 
82     # Begin magic incrementing of serial no.
83     if ( $fullname =~ m-^$namedpath-o && $fileinfo{$file}->{readable} ) {
84 	checkDNS($file, $fullname, $file, 2);
85     }
86 }
87 
88 execute_ok($ENV{EDITOR},@ARGV);
89 
90 FILE: foreach $file ( keys %serial ) {
91     unless (execute_ok($cmp,"-s",$origfile{$file},"$origfile{$file}.tmp")) {
92 	if ( ! exists $checked{$file} && $autoinc ) {
93 	    if (execute_ok($co,"-l",$file)) {
94 		$checked{$file} = 1;
95 		($fileinfo{$file}, $rcsinfo{$file})
96 		= getinfo(makefullname($wd,$file));
97 	    }
98 	    else {
99 		warn "You'll have to update the serial number yourself.\n";
100 		next FILE;
101 	    }
102 	}
103 	updateDNS($file);
104 	$fileinfo{$file}->{changed} = 1;
105     }
106     unlink("$file.tmp");
107 }
108 
109 my($reconfig,@zones);
110 foreach $file ( keys %checked ) {
111     my($success);
112 
113     if ( $setdescr ) {
114 	execute_ok($ci,"-u",$file);
115     }
116     else {
117 	open(CI, "|$ci -u $file") or die("Cannot fork $ci: $!\n");
118 	print CI "updated by $user on " . (localtime) . "\n";
119 	close(CI);
120 	if ($?) {
121 	    warn "WARNING: $ci -u $file returned error: ", $?/256, "\n";
122 	}
123     }
124     if ( $? ) {
125 	delete $checked{$file};
126     } else {
127 	if ($rcsinfo{$file}->{'exists'}
128 	and $rcsinfo{$file}->{uid} != $euid
129 	||  $rcsinfo{$file}->{gid} != $egid) {
130 	    chown $rcsinfo{$file}->{uid}, $rcsinfo{$file}->{gid},
131 		$file, $rcsinfo{$file}->{file};
132 	}
133     }
134     if ($ENV{RVI_AUTO_NDC_RECONFIG}) {
135 	if ($fileinfo{$file}->{changed} and $fileinfo{$file}->{src}) {
136 	    $reconfig++;
137 	}
138     }
139     if ($ENV{RVI_AUTO_NDC_RELOAD}) {
140 	if ($fileinfo{$file}->{changed} and $fileinfo{$file}->{zone}) {
141 	    push @zones, $fileinfo{$file}->{zone};
142 	}
143     }
144 }
145 
146 if ($ndc) {
147     execute_ok($ndc,"reconfig") if $reconfig;
148     foreach my $zone (@zones) {
149 	warn "Reloading $zone:\n";
150 	execute_ok($ndc,"reload",$zone);
151     }
152 }
153 
154 sub checkDNS {
155     my($file, $fullname, $origfile, $maxrecurse) = @_;
156     my($count, $dir, $which, $command, @files);
157 
158     open(FILE, $file) or die("Cannot open $file: $!\n");
159     $count = 0;
160     while ( <FILE> ) {
161 	if ( /^\@\s+IN\s+SOA\s+\S+\s+\S+\s+\(\s*$/i ) {
162 	    $_ = <FILE>;
163 	    last if !defined $_;
164 	    if ( /^\s*(\d+)/ ) {
165 		$serial{$file} = $1;
166 		$origfile{$file} = $origfile;
167 		copy($origfile,"$origfile.tmp");
168 		last;
169 	    }
170 	}
171 	last if ++$count > 10;
172     }
173     close(FILE);
174     if ( $maxrecurse > 0 && ! exists $serial{$file} ) {
175 	# not found, try if we're included somewhere...
176 	$fullname =~ s-^$namedpath/--o;
177 	($dir = $file) =~ s-[^/]*$--;
178 	$dir .= '*';
179 	@files = grep { !/~$/ && -f && -T } < $dir >;
180         $command = "grep -li \'\$INCLUDE *$fullname\' " . join(" ", @files);
181 	$which = `$command`;
182 	@files = split(' ', $which);
183 	if ( @files == 1 ) {
184 	    $fullname = makefullname($wd, $files[0]);
185 	    checkDNS($files[0], $fullname, $origfile, $maxrecurse - 1);
186 	}
187     }
188 }
189 
190 sub updateDNS {
191     my($file) = @_;
192     my($count);
193 
194     open(FILE, $file) or die("Cannot open $file: $!\n");
195     $count = 0;
196     while ( <FILE> ) {
197 	if ( /^\@\s+IN\s+SOA\s+\S+\s+\S+\s+\(\s*$/i ) {
198 	    $_ = <FILE>;
199 	    last if !defined $_;
200 	    close(FILE);
201 	    if ( /^\s*(\d+)/ ) {
202 		if ( $1 < $serial{$file} ) {
203 		    warn "You LOWERED the serial number for $file. Why?\n";
204 		}
205 		elsif ( $1 > $serial{$file} ) {
206 		    warn
207 		     "I see you updated the serial number yourself. Stoer!\n";
208 		}
209 		else {
210 		    newserialDNS($file);
211 		}
212 		return;
213 	    }
214 	}
215 	last if ++$count > 10;
216     }
217     warn "What did you do to the serial number?\n";
218     close(FILE);
219 }
220 
221 sub newserialDNS {
222     my($file) = @_;
223     my($serial, $new_serial);
224     my $hadSOA = 0;
225    
226     if ( $autoinc == 0 ) {
227 	warn "You should update the serial number in $file.\n";
228 	return;
229     }
230     use POSIX 'strftime';
231     warn "Automatically updating the serial number in $file for you.\n";
232     open(FILE, $file) or die("Cannot open $file: $!\n");
233     open(NEW, ">$file.magicinc") or die("Cannot write $file.magicinc: $!\n");
234     while ( <FILE> ) {
235 	print NEW;
236 	if ( ! $hadSOA && /^\@\s+IN\s+SOA\s+\S+\s+\S+\s+\(\s*$/i ) {
237 	    $hadSOA = 1;
238 	    $_ = <FILE>;
239 	    unless ( /^(\s*)\d+([\w\W]*)$/ ) {
240 		die("Aargh! internal error, stop");
241 	    }
242 	    $new_serial = strftime ("%Y%m%d", localtime()) . "01";
243 	    $serial = "" . $serial{$file};
244 	    if ($new_serial > $serial) {
245 	      $serial = $new_serial;
246 	    } else {
247 	      $serial++;
248 	    }
249 	    print NEW $1 . $serial . $2;
250 	}
251     }
252     close(FILE);
253     close(NEW) or die("Aargh! closing $file.magicinc: $!\n");
254     unlink($file) or die("Aargh! unlink $file: $!\n");
255     rename("$file.magicinc", $file) or die("Aargh! rename $file: $!\n");
256 }
257 
258 sub makefullname {
259     my($wd, $file) = @_;
260     my($fullname);
261 
262     if ( $file =~ m-^/- ) {
263 	$fullname = $file;
264     }
265     else {
266 	$fullname = "$wd/$file";
267     }
268     1 while $fullname =~ s-[^/]+/\.\./--;
269     $fullname;
270 }
271 
272 sub getinfo {
273     my($file) = @_;
274     my($fileinfo,$rcsinfo);
275 
276     my($path,$name);
277     unless ( ($path, $name) = $file =~ m-^(.*)/([^/]*)$- ) {
278 	$path = ".";
279 	$name = $file;
280     }
281     $fileinfo->{name} = $name;
282     if ($path =~ m-^$namedpath/(primary|secondary)-o) {
283         $fileinfo->{zone} = $name;
284     }
285     elsif ($path =~ m-^$namedpath/reverse-o) {
286     	($fileinfo->{zone}) = $name =~ /^(\d+\.\d+\.\d+)/;
287 	$fileinfo->{zone} .= ".in-addr.arpa";
288     }
289     elsif ($path =~ m-^$namedpath/src-o) {
290         $fileinfo->{src} = 1;
291     }
292 
293     $rcsinfo->{file} = "$path/RCS/$name,v";
294     if ( -f $rcsinfo->{file} ) {
295 	$rcsinfo->{'exists'} = 1;
296 	$rcsinfo->{uid} = (stat(_))[4];
297 	$rcsinfo->{gid} = (stat(_))[5];
298     }
299     elsif ( ! -d "$path/RCS" ) {
300 	warn "WARNING: There is no $path/RCS directory!\n";
301     	exit cleanmess();
302     }
303 
304     if ( -f $file ) {
305 	unless ($rcsinfo->{'exists'}) {
306 	    warn "WARNING: $file already exists but is not in RCS!\n";
307 	    warn "Manually check it in if you want it to be in RCS.\n";
308 	    exit cleanmess();
309 	}
310         $fileinfo->{mode} = (stat(_))[2];
311 	$fileinfo->{readable} = -r _;
312     }
313 
314     return $fileinfo, $rcsinfo;
315 }
316 
317 sub cleanmess {
318     if ( keys %checked ) {
319 	warn "Cleaning up the mess...\n";
320 	foreach my $file ( keys %checked ) {
321 	    if ($fileinfo{$file}->{mode}) {
322 		chmod $fileinfo{$file}->{mode} & 07577, $file;
323 	    }
324 	    if ($rcsinfo{$file}->{'exists'}) {
325 		execute_ok($co,"-u",$file);
326 		if ($rcsinfo{$file}->{uid} != $euid
327 		or  $rcsinfo{$file}->{gid} != $egid) {
328 		    chown $rcsinfo{$file}->{uid}, $rcsinfo{$file}->{gid},
329 			$file, $rcsinfo{$file}->{file};
330 		}
331 	    }
332 	}
333     }
334     return 1;
335 }
336 
337 sub execute_ok {
338     my(@command) = @_;
339     my $ret = system(@command) / 256;
340     unless ($command[0] eq $cmp) {
341 	warn "WARNING: @command returned $ret\n" if $ret;
342     }
343     return not $ret;
344 }
345 
346 sub which {
347     my($prog) = @_;
348     my($fullprog) = grep { -x } (
349     	"/usr/local/bin/$prog",
350     	"/usr/bin/$prog",
351     	"/usr/local/sbin/$prog",
352     	"/usr/sbin/$prog",
353     ) or warn "WARNING: cannot find $prog\n";
354     return $fullprog;
355 }
356