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