6c2ff065d1b54ede33a884458667229d7b9336a1
[openssl.git] / tools / c_rehash.in
1 #!/usr/local/bin/perl
2
3 # Perl c_rehash script, scan all files in a directory
4 # and add symbolic links to their hash values.
5
6 my $dir;
7 my $prefix;
8
9 my $openssl = $ENV{OPENSSL} || "openssl";
10 my $pwd;
11 my $x509hash = "-subject_hash";
12 my $crlhash = "-hash";
13 my $verbose = 0;
14 my $symlink_exists=eval {symlink("",""); 1};
15 my $removelinks = 1;
16
17 ##  Parse flags.
18 while ( $ARGV[0] =~ /^-/ ) {
19     my $flag = shift @ARGV;
20     last if ( $flag eq '--');
21     if ( $flag eq '-old') {
22             $x509hash = "-subject_hash_old";
23             $crlhash = "-hash_old";
24     } elsif ( $flag eq '-h') {
25             help();
26     } elsif ( $flag eq '-n' ) {
27             $removelinks = 0;
28     } elsif ( $flag eq '-v' ) {
29             $verbose++;
30     }
31     else {
32             print STDERR "Usage error; try -help.\n";
33             exit 1;
34     }
35 }
36
37 sub help {
38         print "Usage: c_rehash [-old] [-h] [-v] [dirs...]\n";
39         print "   -old use old-style digest\n";
40         print "   -h print this help text\n";
41         print "   -v print files removed and linked\n";
42         exit 0;
43 }
44
45 eval "require Cwd";
46 if (defined(&Cwd::getcwd)) {
47         $pwd=Cwd::getcwd();
48 } else {
49         $pwd=`pwd`;
50         chomp($pwd);
51 }
52
53 # DOS/Win32 or Unix delimiter?  Prefix our installdir, then search.
54 my $path_delim = ($pwd =~ /^[a-z]\:/i) ? ';' : ':';
55 $ENV{PATH} = "$prefix/bin" . ($ENV{PATH} ? $path_delim . $ENV{PATH} : "");
56
57 if (! -x $openssl) {
58         my $found = 0;
59         foreach (split /$path_delim/, $ENV{PATH}) {
60                 if (-x "$_/$openssl") {
61                         $found = 1;
62                         $openssl = "$_/$openssl";
63                         last;
64                 }       
65         }
66         if ($found == 0) {
67                 print STDERR "c_rehash: rehashing skipped ('openssl' program not available)\n";
68                 exit 0;
69         }
70 }
71
72 if (@ARGV) {
73         @dirlist = @ARGV;
74 } elsif ($ENV{SSL_CERT_DIR}) {
75         @dirlist = split /$path_delim/, $ENV{SSL_CERT_DIR};
76 } else {
77         $dirlist[0] = "$dir/certs";
78 }
79
80 if (-d $dirlist[0]) {
81         chdir $dirlist[0];
82         $openssl="$pwd/$openssl" if (!-x $openssl);
83         chdir $pwd;
84 }
85
86 foreach (@dirlist) {
87         if (-d $_ ) {
88             if ( -w $_) {
89                 hash_dir($_);
90             } else {
91                 print "Skipping $_, can't write\n";
92             }
93         }
94 }
95
96 sub hash_dir {
97         my %hashlist;
98         print "Doing $_[0]\n";
99         chdir $_[0];
100         opendir(DIR, ".");
101         my @flist = readdir(DIR);
102         closedir DIR;
103         if ( $removelinks ) {
104                 # Delete any existing symbolic links
105                 foreach (grep {/^[\da-f]+\.r{0,1}\d+$/} @flist) {
106                         if (-l $_) {
107                                 print "unlink $_" if $verbose;
108                                 unlink $_ || warn "Can't unlink $_, $!\n";
109                         }
110                 }
111         }
112         FILE: foreach $fname (grep {/\.(pem)|(crt)|(cer)|(crl)$/} @flist) {
113                 # Check to see if certificates and/or CRLs present.
114                 my ($cert, $crl) = check_file($fname);
115                 if (!$cert && !$crl) {
116                         print STDERR "WARNING: $fname does not contain a certificate or CRL: skipping\n";
117                         next;
118                 }
119                 link_hash_cert($fname) if ($cert);
120                 link_hash_crl($fname) if ($crl);
121         }
122 }
123
124 sub check_file {
125         my ($is_cert, $is_crl) = (0,0);
126         my $fname = $_[0];
127         open IN, $fname;
128         while(<IN>) {
129                 if (/^-----BEGIN (.*)-----/) {
130                         my $hdr = $1;
131                         if ($hdr =~ /^(X509 |TRUSTED |)CERTIFICATE$/) {
132                                 $is_cert = 1;
133                                 last if ($is_crl);
134                         } elsif ($hdr eq "X509 CRL") {
135                                 $is_crl = 1;
136                                 last if ($is_cert);
137                         }
138                 }
139         }
140         close IN;
141         return ($is_cert, $is_crl);
142 }
143
144
145 # Link a certificate to its subject name hash value, each hash is of
146 # the form <hash>.<n> where n is an integer. If the hash value already exists
147 # then we need to up the value of n, unless its a duplicate in which
148 # case we skip the link. We check for duplicates by comparing the
149 # certificate fingerprints
150
151 sub link_hash_cert {
152                 my $fname = $_[0];
153                 $fname =~ s/'/'\\''/g;
154                 my ($hash, $fprint) = `"$openssl" x509 $x509hash -fingerprint -noout -in "$fname"`;
155                 chomp $hash;
156                 chomp $fprint;
157                 $fprint =~ s/^.*=//;
158                 $fprint =~ tr/://d;
159                 my $suffix = 0;
160                 # Search for an unused hash filename
161                 while(exists $hashlist{"$hash.$suffix"}) {
162                         # Hash matches: if fingerprint matches its a duplicate cert
163                         if ($hashlist{"$hash.$suffix"} eq $fprint) {
164                                 print STDERR "WARNING: Skipping duplicate certificate $fname\n";
165                                 return;
166                         }
167                         $suffix++;
168                 }
169                 $hash .= ".$suffix";
170                 if ($symlink_exists) {
171                         print "link $fname -> $hash\n" if $verbose;
172                         symlink $fname, $hash || warn "Can't symlink, $!";
173                 } else {
174                         print "copy $fname -> $hash\n" if $verbose;
175                         if (open($in, "<", $fname)) {
176                             if (open($out,">", $hash)) {
177                                 print $out $_ while (<$in>);
178                                 close $out;
179                             } else {
180                                 warn "can't open $hash for write, $!";
181                             }
182                             close $in;
183                         } else {
184                             warn "can't open $fname for read, $!";
185                         }
186                 }
187                 $hashlist{$hash} = $fprint;
188 }
189
190 # Same as above except for a CRL. CRL links are of the form <hash>.r<n>
191
192 sub link_hash_crl {
193                 my $fname = $_[0];
194                 $fname =~ s/'/'\\''/g;
195                 my ($hash, $fprint) = `"$openssl" crl $crlhash -fingerprint -noout -in '$fname'`;
196                 chomp $hash;
197                 chomp $fprint;
198                 $fprint =~ s/^.*=//;
199                 $fprint =~ tr/://d;
200                 my $suffix = 0;
201                 # Search for an unused hash filename
202                 while(exists $hashlist{"$hash.r$suffix"}) {
203                         # Hash matches: if fingerprint matches its a duplicate cert
204                         if ($hashlist{"$hash.r$suffix"} eq $fprint) {
205                                 print STDERR "WARNING: Skipping duplicate CRL $fname\n";
206                                 return;
207                         }
208                         $suffix++;
209                 }
210                 $hash .= ".r$suffix";
211                 if ($symlink_exists) {
212                         print "link $fname -> $hash\n" if $verbose;
213                         symlink $fname, $hash || warn "Can't symlink, $!";
214                 } else {
215                         print "cp $fname -> $hash\n" if $verbose;
216                         system ("cp", $fname, $hash);
217                         warn "Can't copy, $!" if ($? >> 8) != 0;
218                 }
219                 $hashlist{$hash} = $fprint;
220 }