Paul Bakker | 9d78140 | 2011-05-09 16:17:09 +0000 | [diff] [blame] | 1 | #!/usr/bin/perl |
| 2 | # |
| 3 | |
| 4 | use strict; |
| 5 | |
| 6 | my $include_dir = shift or die "Missing include directory"; |
| 7 | my $data_dir = shift or die "Missing data directory"; |
| 8 | my $error_file = shift or die "Missing destination file"; |
| 9 | my $error_format_file = $data_dir.'/error.fmt'; |
| 10 | |
Paul Bakker | 83f00bb | 2012-07-04 11:08:50 +0000 | [diff] [blame] | 11 | my @low_level_modules = ( "AES", "ASN1", "BLOWFISH", "CAMELLIA", "BIGNUM", |
Paul Bakker | 28144de | 2013-06-24 19:28:55 +0200 | [diff] [blame] | 12 | "BASE64", "XTEA", "PBKDF2", "OID", |
Paul Bakker | 69e095c | 2011-12-10 21:55:01 +0000 | [diff] [blame] | 13 | "PADLOCK", "DES", "NET", "CTR_DRBG", "ENTROPY", |
Paul Bakker | 2466d93 | 2013-09-28 14:40:38 +0200 | [diff] [blame] | 14 | "MD2", "MD4", "MD5", "SHA1", "SHA256", "SHA512", |
| 15 | "GCM", "THREADING" ); |
Paul Bakker | f1f21fe | 2013-06-24 19:17:19 +0200 | [diff] [blame] | 16 | my @high_level_modules = ( "PEM", "X509", "DHM", "RSA", "ECP", "MD", "CIPHER", "SSL", |
Paul Bakker | 5187656 | 2013-09-17 14:36:05 +0200 | [diff] [blame] | 17 | "PK", "PKCS12", "PKCS5" ); |
Paul Bakker | 9d78140 | 2011-05-09 16:17:09 +0000 | [diff] [blame] | 18 | |
| 19 | my $line_separator = $/; |
| 20 | undef $/; |
| 21 | |
| 22 | open(FORMAT_FILE, "$error_format_file") or die "Opening error format file '$error_format_file': $!"; |
| 23 | my $error_format = <FORMAT_FILE>; |
| 24 | close(FORMAT_FILE); |
| 25 | |
| 26 | $/ = $line_separator; |
| 27 | |
| 28 | open(GREP, "/bin/grep \"define POLARSSL_ERR_\" $include_dir/* |") || die("Failure when calling grep: $!"); |
| 29 | |
| 30 | my $ll_old_define = ""; |
| 31 | my $hl_old_define = ""; |
| 32 | |
| 33 | my $ll_code_check = ""; |
| 34 | my $hl_code_check = ""; |
| 35 | |
| 36 | my $headers = ""; |
| 37 | |
| 38 | while (my $line = <GREP>) |
| 39 | { |
Paul Bakker | 36713e8 | 2013-09-17 13:25:29 +0200 | [diff] [blame] | 40 | next if ($line =~ /compat-1.2.h/); |
Paul Bakker | 9d78140 | 2011-05-09 16:17:09 +0000 | [diff] [blame] | 41 | my ($error_name, $error_code) = $line =~ /(POLARSSL_ERR_\w+)\s+\-(0x\w+)/; |
| 42 | my ($description) = $line =~ /\/\*\*< (.*?)\.? \*\//; |
| 43 | $description =~ s/\\/\\\\/g; |
| 44 | $description = "DESCRIPTION MISSING" if ($description eq ""); |
| 45 | |
| 46 | my ($module_name) = $error_name =~ /^POLARSSL_ERR_([^_]+)/; |
| 47 | |
| 48 | # Fix faulty ones |
| 49 | $module_name = "BIGNUM" if ($module_name eq "MPI"); |
Paul Bakker | 880ac7e | 2011-11-27 14:50:49 +0000 | [diff] [blame] | 50 | $module_name = "CTR_DRBG" if ($module_name eq "CTR"); |
Paul Bakker | 9d78140 | 2011-05-09 16:17:09 +0000 | [diff] [blame] | 51 | |
| 52 | my $define_name = $module_name; |
Paul Bakker | 36713e8 | 2013-09-17 13:25:29 +0200 | [diff] [blame] | 53 | $define_name = "X509_USE,X509_CREATE" if ($define_name eq "X509"); |
Paul Bakker | dceecd8 | 2011-11-15 16:38:34 +0000 | [diff] [blame] | 54 | $define_name = "ASN1_PARSE" if ($define_name eq "ASN1"); |
Paul Bakker | e471cd1 | 2011-05-18 13:27:35 +0000 | [diff] [blame] | 55 | $define_name = "SSL_TLS" if ($define_name eq "SSL"); |
Paul Bakker | cff6842 | 2013-09-15 20:43:33 +0200 | [diff] [blame] | 56 | $define_name = "PEM_PARSE,PEM_WRITE" if ($define_name eq "PEM"); |
Paul Bakker | 9d78140 | 2011-05-09 16:17:09 +0000 | [diff] [blame] | 57 | |
| 58 | my $include_name = $module_name; |
| 59 | $include_name =~ tr/A-Z/a-z/; |
| 60 | $include_name = "" if ($include_name eq "asn1"); |
| 61 | |
| 62 | my $found_ll = grep $_ eq $module_name, @low_level_modules; |
| 63 | my $found_hl = grep $_ eq $module_name, @high_level_modules; |
| 64 | if (!$found_ll && !$found_hl) |
| 65 | { |
| 66 | printf("Error: Do not know how to handle: $module_name\n"); |
| 67 | exit 1; |
| 68 | } |
| 69 | |
| 70 | my $code_check; |
| 71 | my $old_define; |
| 72 | my $white_space; |
Paul Bakker | cff6842 | 2013-09-15 20:43:33 +0200 | [diff] [blame] | 73 | my $first; |
Paul Bakker | 9d78140 | 2011-05-09 16:17:09 +0000 | [diff] [blame] | 74 | |
| 75 | if ($found_ll) |
| 76 | { |
| 77 | $code_check = \$ll_code_check; |
| 78 | $old_define = \$ll_old_define; |
| 79 | $white_space = ' '; |
| 80 | } |
| 81 | else |
| 82 | { |
| 83 | $code_check = \$hl_code_check; |
| 84 | $old_define = \$hl_old_define; |
| 85 | $white_space = ' '; |
| 86 | } |
| 87 | |
| 88 | if ($define_name ne ${$old_define}) |
| 89 | { |
| 90 | if (${$old_define} ne "") |
| 91 | { |
Paul Bakker | cff6842 | 2013-09-15 20:43:33 +0200 | [diff] [blame] | 92 | ${$code_check} .= "#endif /* "; |
| 93 | $first = 0; |
| 94 | foreach my $dep (split(/,/, ${$old_define})) |
| 95 | { |
| 96 | ${$code_check} .= " || " if ($first++); |
| 97 | ${$code_check} .= "POLARSSL_${dep}_C"; |
| 98 | } |
| 99 | ${$code_check} .= " */\n\n"; |
Paul Bakker | 9d78140 | 2011-05-09 16:17:09 +0000 | [diff] [blame] | 100 | } |
| 101 | |
Paul Bakker | cff6842 | 2013-09-15 20:43:33 +0200 | [diff] [blame] | 102 | ${$code_check} .= "#if "; |
| 103 | $headers .= "#if " if ($include_name ne ""); |
| 104 | $first = 0; |
| 105 | foreach my $dep (split(/,/, ${define_name})) |
| 106 | { |
| 107 | ${$code_check} .= " || " if ($first); |
| 108 | $headers .= " || " if ($first++); |
| 109 | |
| 110 | ${$code_check} .= "defined(POLARSSL_${dep}_C)"; |
| 111 | $headers .= "defined(POLARSSL_${dep}_C)" if |
| 112 | ($include_name ne ""); |
| 113 | } |
| 114 | ${$code_check} .= "\n"; |
| 115 | $headers .= "\n#include \"polarssl/${include_name}.h\"\n". |
Paul Bakker | 9d78140 | 2011-05-09 16:17:09 +0000 | [diff] [blame] | 116 | "#endif\n\n" if ($include_name ne ""); |
| 117 | ${$old_define} = $define_name; |
| 118 | } |
| 119 | |
Paul Bakker | 3aac1da | 2012-05-08 13:12:27 +0000 | [diff] [blame] | 120 | if ($error_name eq "POLARSSL_ERR_SSL_FATAL_ALERT_MESSAGE") |
| 121 | { |
| 122 | ${$code_check} .= "${white_space}if( use_ret == -($error_name) )\n". |
| 123 | "${white_space}\{\n". |
| 124 | "${white_space} snprintf( buf, buflen, \"$module_name - $description\" );\n". |
| 125 | "${white_space} return;\n". |
| 126 | "${white_space}}\n" |
| 127 | } |
| 128 | else |
| 129 | { |
| 130 | ${$code_check} .= "${white_space}if( use_ret == -($error_name) )\n". |
| 131 | "${white_space} snprintf( buf, buflen, \"$module_name - $description\" );\n" |
| 132 | } |
Paul Bakker | 9d78140 | 2011-05-09 16:17:09 +0000 | [diff] [blame] | 133 | }; |
| 134 | |
| 135 | if ($ll_old_define ne "") |
| 136 | { |
| 137 | $ll_code_check .= "#endif /* POLARSSL_${ll_old_define}_C */\n\n"; |
| 138 | } |
| 139 | if ($hl_old_define ne "") |
| 140 | { |
| 141 | $hl_code_check .= "#endif /* POLARSSL_${hl_old_define}_C */\n\n"; |
| 142 | } |
| 143 | |
| 144 | $error_format =~ s/HEADER_INCLUDED\n/$headers/g; |
| 145 | $error_format =~ s/LOW_LEVEL_CODE_CHECKS\n/$ll_code_check/g; |
| 146 | $error_format =~ s/HIGH_LEVEL_CODE_CHECKS\n/$hl_code_check/g; |
| 147 | |
| 148 | open(ERROR_FILE, ">$error_file") or die "Opening destination file '$error_file': $!"; |
| 149 | print ERROR_FILE $error_format; |
| 150 | close(ERROR_FILE); |