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 | c70b982 | 2013-04-07 22:00:46 +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 | 7890405 | 2012-04-18 14:11:33 +0000 | [diff] [blame] | 14 | "MD2", "MD4", "MD5", "SHA1", "SHA2", "SHA4", "GCM" ); |
Paul Bakker | cf4365f | 2013-01-16 17:00:43 +0100 | [diff] [blame] | 15 | my @high_level_modules = ( "PEM", "X509", "DHM", "RSA", "ECP", "MD", "CIPHER", "SSL" ); |
Paul Bakker | 9d78140 | 2011-05-09 16:17:09 +0000 | [diff] [blame] | 16 | |
| 17 | my $line_separator = $/; |
| 18 | undef $/; |
| 19 | |
| 20 | open(FORMAT_FILE, "$error_format_file") or die "Opening error format file '$error_format_file': $!"; |
| 21 | my $error_format = <FORMAT_FILE>; |
| 22 | close(FORMAT_FILE); |
| 23 | |
| 24 | $/ = $line_separator; |
| 25 | |
| 26 | open(GREP, "/bin/grep \"define POLARSSL_ERR_\" $include_dir/* |") || die("Failure when calling grep: $!"); |
| 27 | |
| 28 | my $ll_old_define = ""; |
| 29 | my $hl_old_define = ""; |
| 30 | |
| 31 | my $ll_code_check = ""; |
| 32 | my $hl_code_check = ""; |
| 33 | |
| 34 | my $headers = ""; |
| 35 | |
| 36 | while (my $line = <GREP>) |
| 37 | { |
| 38 | my ($error_name, $error_code) = $line =~ /(POLARSSL_ERR_\w+)\s+\-(0x\w+)/; |
| 39 | my ($description) = $line =~ /\/\*\*< (.*?)\.? \*\//; |
| 40 | $description =~ s/\\/\\\\/g; |
| 41 | $description = "DESCRIPTION MISSING" if ($description eq ""); |
| 42 | |
| 43 | my ($module_name) = $error_name =~ /^POLARSSL_ERR_([^_]+)/; |
| 44 | |
| 45 | # Fix faulty ones |
| 46 | $module_name = "BIGNUM" if ($module_name eq "MPI"); |
Paul Bakker | 880ac7e | 2011-11-27 14:50:49 +0000 | [diff] [blame] | 47 | $module_name = "CTR_DRBG" if ($module_name eq "CTR"); |
Paul Bakker | 9d78140 | 2011-05-09 16:17:09 +0000 | [diff] [blame] | 48 | |
| 49 | my $define_name = $module_name; |
| 50 | $define_name = "X509_PARSE" if ($define_name eq "X509"); |
Paul Bakker | dceecd8 | 2011-11-15 16:38:34 +0000 | [diff] [blame] | 51 | $define_name = "ASN1_PARSE" if ($define_name eq "ASN1"); |
Paul Bakker | e471cd1 | 2011-05-18 13:27:35 +0000 | [diff] [blame] | 52 | $define_name = "SSL_TLS" if ($define_name eq "SSL"); |
Paul Bakker | 9d78140 | 2011-05-09 16:17:09 +0000 | [diff] [blame] | 53 | |
| 54 | my $include_name = $module_name; |
| 55 | $include_name =~ tr/A-Z/a-z/; |
| 56 | $include_name = "" if ($include_name eq "asn1"); |
| 57 | |
| 58 | my $found_ll = grep $_ eq $module_name, @low_level_modules; |
| 59 | my $found_hl = grep $_ eq $module_name, @high_level_modules; |
| 60 | if (!$found_ll && !$found_hl) |
| 61 | { |
| 62 | printf("Error: Do not know how to handle: $module_name\n"); |
| 63 | exit 1; |
| 64 | } |
| 65 | |
| 66 | my $code_check; |
| 67 | my $old_define; |
| 68 | my $white_space; |
| 69 | |
| 70 | if ($found_ll) |
| 71 | { |
| 72 | $code_check = \$ll_code_check; |
| 73 | $old_define = \$ll_old_define; |
| 74 | $white_space = ' '; |
| 75 | } |
| 76 | else |
| 77 | { |
| 78 | $code_check = \$hl_code_check; |
| 79 | $old_define = \$hl_old_define; |
| 80 | $white_space = ' '; |
| 81 | } |
| 82 | |
| 83 | if ($define_name ne ${$old_define}) |
| 84 | { |
| 85 | if (${$old_define} ne "") |
| 86 | { |
| 87 | ${$code_check} .= "#endif /* POLARSSL_${$old_define}_C */\n\n"; |
| 88 | } |
| 89 | |
| 90 | ${$code_check} .= "#if defined(POLARSSL_${define_name}_C)\n"; |
| 91 | $headers .= "#if defined(POLARSSL_${define_name}_C)\n". |
| 92 | "#include \"polarssl/${include_name}.h\"\n". |
| 93 | "#endif\n\n" if ($include_name ne ""); |
| 94 | ${$old_define} = $define_name; |
| 95 | } |
| 96 | |
Paul Bakker | 3aac1da | 2012-05-08 13:12:27 +0000 | [diff] [blame] | 97 | if ($error_name eq "POLARSSL_ERR_SSL_FATAL_ALERT_MESSAGE") |
| 98 | { |
| 99 | ${$code_check} .= "${white_space}if( use_ret == -($error_name) )\n". |
| 100 | "${white_space}\{\n". |
| 101 | "${white_space} snprintf( buf, buflen, \"$module_name - $description\" );\n". |
| 102 | "${white_space} return;\n". |
| 103 | "${white_space}}\n" |
| 104 | } |
| 105 | else |
| 106 | { |
| 107 | ${$code_check} .= "${white_space}if( use_ret == -($error_name) )\n". |
| 108 | "${white_space} snprintf( buf, buflen, \"$module_name - $description\" );\n" |
| 109 | } |
Paul Bakker | 9d78140 | 2011-05-09 16:17:09 +0000 | [diff] [blame] | 110 | }; |
| 111 | |
| 112 | if ($ll_old_define ne "") |
| 113 | { |
| 114 | $ll_code_check .= "#endif /* POLARSSL_${ll_old_define}_C */\n\n"; |
| 115 | } |
| 116 | if ($hl_old_define ne "") |
| 117 | { |
| 118 | $hl_code_check .= "#endif /* POLARSSL_${hl_old_define}_C */\n\n"; |
| 119 | } |
| 120 | |
| 121 | $error_format =~ s/HEADER_INCLUDED\n/$headers/g; |
| 122 | $error_format =~ s/LOW_LEVEL_CODE_CHECKS\n/$ll_code_check/g; |
| 123 | $error_format =~ s/HIGH_LEVEL_CODE_CHECKS\n/$hl_code_check/g; |
| 124 | |
| 125 | open(ERROR_FILE, ">$error_file") or die "Opening destination file '$error_file': $!"; |
| 126 | print ERROR_FILE $error_format; |
| 127 | close(ERROR_FILE); |