| 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); |