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", |
Manuel Pégourié-Gonnard | cf38367 | 2014-02-01 10:22:21 +0100 | [diff] [blame] | 14 | "HMAC_DRBG", "MD2", "MD4", "MD5", "RIPEMD160", |
| 15 | "SHA1", "SHA256", "SHA512", "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"); |
Manuel Pégourié-Gonnard | cf38367 | 2014-02-01 10:22:21 +0100 | [diff] [blame] | 51 | $module_name = "HMAC_DRBG" if ($module_name eq "HMAC"); |
Paul Bakker | 9d78140 | 2011-05-09 16:17:09 +0000 | [diff] [blame] | 52 | |
| 53 | my $define_name = $module_name; |
Paul Bakker | 36713e8 | 2013-09-17 13:25:29 +0200 | [diff] [blame] | 54 | $define_name = "X509_USE,X509_CREATE" if ($define_name eq "X509"); |
Paul Bakker | dceecd8 | 2011-11-15 16:38:34 +0000 | [diff] [blame] | 55 | $define_name = "ASN1_PARSE" if ($define_name eq "ASN1"); |
Paul Bakker | e471cd1 | 2011-05-18 13:27:35 +0000 | [diff] [blame] | 56 | $define_name = "SSL_TLS" if ($define_name eq "SSL"); |
Paul Bakker | cff6842 | 2013-09-15 20:43:33 +0200 | [diff] [blame] | 57 | $define_name = "PEM_PARSE,PEM_WRITE" if ($define_name eq "PEM"); |
Paul Bakker | 9d78140 | 2011-05-09 16:17:09 +0000 | [diff] [blame] | 58 | |
| 59 | my $include_name = $module_name; |
| 60 | $include_name =~ tr/A-Z/a-z/; |
| 61 | $include_name = "" if ($include_name eq "asn1"); |
| 62 | |
| 63 | my $found_ll = grep $_ eq $module_name, @low_level_modules; |
| 64 | my $found_hl = grep $_ eq $module_name, @high_level_modules; |
| 65 | if (!$found_ll && !$found_hl) |
| 66 | { |
| 67 | printf("Error: Do not know how to handle: $module_name\n"); |
| 68 | exit 1; |
| 69 | } |
| 70 | |
| 71 | my $code_check; |
| 72 | my $old_define; |
| 73 | my $white_space; |
Paul Bakker | cff6842 | 2013-09-15 20:43:33 +0200 | [diff] [blame] | 74 | my $first; |
Paul Bakker | 9d78140 | 2011-05-09 16:17:09 +0000 | [diff] [blame] | 75 | |
| 76 | if ($found_ll) |
| 77 | { |
| 78 | $code_check = \$ll_code_check; |
| 79 | $old_define = \$ll_old_define; |
| 80 | $white_space = ' '; |
| 81 | } |
| 82 | else |
| 83 | { |
| 84 | $code_check = \$hl_code_check; |
| 85 | $old_define = \$hl_old_define; |
| 86 | $white_space = ' '; |
| 87 | } |
| 88 | |
| 89 | if ($define_name ne ${$old_define}) |
| 90 | { |
| 91 | if (${$old_define} ne "") |
| 92 | { |
Paul Bakker | cff6842 | 2013-09-15 20:43:33 +0200 | [diff] [blame] | 93 | ${$code_check} .= "#endif /* "; |
| 94 | $first = 0; |
| 95 | foreach my $dep (split(/,/, ${$old_define})) |
| 96 | { |
| 97 | ${$code_check} .= " || " if ($first++); |
| 98 | ${$code_check} .= "POLARSSL_${dep}_C"; |
| 99 | } |
| 100 | ${$code_check} .= " */\n\n"; |
Paul Bakker | 9d78140 | 2011-05-09 16:17:09 +0000 | [diff] [blame] | 101 | } |
| 102 | |
Paul Bakker | cff6842 | 2013-09-15 20:43:33 +0200 | [diff] [blame] | 103 | ${$code_check} .= "#if "; |
| 104 | $headers .= "#if " if ($include_name ne ""); |
| 105 | $first = 0; |
| 106 | foreach my $dep (split(/,/, ${define_name})) |
| 107 | { |
| 108 | ${$code_check} .= " || " if ($first); |
| 109 | $headers .= " || " if ($first++); |
| 110 | |
| 111 | ${$code_check} .= "defined(POLARSSL_${dep}_C)"; |
| 112 | $headers .= "defined(POLARSSL_${dep}_C)" if |
| 113 | ($include_name ne ""); |
| 114 | } |
| 115 | ${$code_check} .= "\n"; |
| 116 | $headers .= "\n#include \"polarssl/${include_name}.h\"\n". |
Paul Bakker | 9d78140 | 2011-05-09 16:17:09 +0000 | [diff] [blame] | 117 | "#endif\n\n" if ($include_name ne ""); |
| 118 | ${$old_define} = $define_name; |
| 119 | } |
| 120 | |
Paul Bakker | 3aac1da | 2012-05-08 13:12:27 +0000 | [diff] [blame] | 121 | if ($error_name eq "POLARSSL_ERR_SSL_FATAL_ALERT_MESSAGE") |
| 122 | { |
| 123 | ${$code_check} .= "${white_space}if( use_ret == -($error_name) )\n". |
| 124 | "${white_space}\{\n". |
| 125 | "${white_space} snprintf( buf, buflen, \"$module_name - $description\" );\n". |
| 126 | "${white_space} return;\n". |
| 127 | "${white_space}}\n" |
| 128 | } |
| 129 | else |
| 130 | { |
| 131 | ${$code_check} .= "${white_space}if( use_ret == -($error_name) )\n". |
| 132 | "${white_space} snprintf( buf, buflen, \"$module_name - $description\" );\n" |
| 133 | } |
Paul Bakker | 9d78140 | 2011-05-09 16:17:09 +0000 | [diff] [blame] | 134 | }; |
| 135 | |
| 136 | if ($ll_old_define ne "") |
| 137 | { |
Manuel Pégourié-Gonnard | fe671f4 | 2014-04-11 18:06:44 +0200 | [diff] [blame] | 138 | $ll_code_check .= "#endif /* POLARSSL_${ll_old_define}_C */\n"; |
Paul Bakker | 9d78140 | 2011-05-09 16:17:09 +0000 | [diff] [blame] | 139 | } |
| 140 | if ($hl_old_define ne "") |
| 141 | { |
Manuel Pégourié-Gonnard | fe671f4 | 2014-04-11 18:06:44 +0200 | [diff] [blame] | 142 | $hl_code_check .= "#endif /* POLARSSL_${hl_old_define}_C */\n"; |
Paul Bakker | 9d78140 | 2011-05-09 16:17:09 +0000 | [diff] [blame] | 143 | } |
| 144 | |
| 145 | $error_format =~ s/HEADER_INCLUDED\n/$headers/g; |
| 146 | $error_format =~ s/LOW_LEVEL_CODE_CHECKS\n/$ll_code_check/g; |
| 147 | $error_format =~ s/HIGH_LEVEL_CODE_CHECKS\n/$hl_code_check/g; |
| 148 | |
| 149 | open(ERROR_FILE, ">$error_file") or die "Opening destination file '$error_file': $!"; |
| 150 | print ERROR_FILE $error_format; |
| 151 | close(ERROR_FILE); |