226 lines
6.8 KiB
Perl
Executable File
226 lines
6.8 KiB
Perl
Executable File
#!/usr/bin/env perl
|
|
|
|
# Generate error.c
|
|
#
|
|
# Usage: ./generate_errors.pl or scripts/generate_errors.pl without arguments,
|
|
# or generate_errors.pl include_dir data_dir error_file include_crypto
|
|
# include_crypto can be either 0 (don't include) or 1 (include). On by default.
|
|
|
|
use strict;
|
|
|
|
my ($include_dir, $data_dir, $error_file, $include_crypto);
|
|
my $crypto_dir = "crypto";
|
|
|
|
if( @ARGV ) {
|
|
die "Invalid number of arguments" if scalar @ARGV != 4;
|
|
($include_dir, $data_dir, $error_file, $include_crypto) = @ARGV;
|
|
|
|
-d $include_dir or die "No such directory: $include_dir\n";
|
|
-d $data_dir or die "No such directory: $data_dir\n";
|
|
} else {
|
|
$include_dir = 'include/mbedtls';
|
|
$data_dir = 'scripts/data_files';
|
|
$error_file = 'library/error.c';
|
|
$include_crypto = 1;
|
|
|
|
unless( -d $include_dir && -d $data_dir ) {
|
|
chdir '..' or die;
|
|
-d $include_dir && -d $data_dir
|
|
or die "Without arguments, must be run from root or scripts\n"
|
|
}
|
|
}
|
|
|
|
if( $include_crypto ) {
|
|
-d $crypto_dir or die "Crypto submodule not present\n";
|
|
}
|
|
|
|
my $error_format_file = $data_dir.'/error.fmt';
|
|
|
|
my @low_level_modules = qw( AES ARC4 ARIA ASN1 BASE64 BIGNUM BLOWFISH
|
|
CAMELLIA CCM CHACHA20 CHACHAPOLY CMAC CTR_DRBG DES
|
|
ENTROPY GCM HKDF HMAC_DRBG MD2 MD4 MD5
|
|
NET OID PADLOCK PBKDF2 PLATFORM POLY1305 RIPEMD160
|
|
SHA1 SHA256 SHA512 THREADING XTEA );
|
|
my @high_level_modules = qw( CIPHER DHM ECP MD
|
|
PEM PK PKCS12 PKCS5
|
|
RSA SSL X509 );
|
|
|
|
my $line_separator = $/;
|
|
undef $/;
|
|
|
|
open(FORMAT_FILE, "$error_format_file") or die "Opening error format file '$error_format_file': $!";
|
|
my $error_format = <FORMAT_FILE>;
|
|
close(FORMAT_FILE);
|
|
|
|
$/ = $line_separator;
|
|
|
|
my @headers = ();
|
|
if ($include_crypto) {
|
|
@headers = <$crypto_dir/$include_dir/*.h>;
|
|
foreach my $header (<$include_dir/*.h>) {
|
|
my $basename = $header; $basename =~ s!.*/!!;
|
|
push @headers, $header unless -e "$crypto_dir/$include_dir/$basename";
|
|
}
|
|
} else {
|
|
@headers = <$include_dir/*.h>;
|
|
}
|
|
|
|
my @matches;
|
|
foreach my $file (@headers) {
|
|
open(FILE, "$file");
|
|
my @grep_res = grep(/^\s*#define\s+MBEDTLS_ERR_\w+\s+\-0x[0-9A-Fa-f]+/, <FILE>);
|
|
push(@matches, @grep_res);
|
|
close FILE;
|
|
}
|
|
|
|
my $ll_old_define = "";
|
|
my $hl_old_define = "";
|
|
|
|
my $ll_code_check = "";
|
|
my $hl_code_check = "";
|
|
|
|
my $headers = "";
|
|
|
|
my %error_codes_seen;
|
|
|
|
|
|
foreach my $line (@matches)
|
|
{
|
|
next if ($line =~ /compat-1.2.h/);
|
|
my ($error_name, $error_code) = $line =~ /(MBEDTLS_ERR_\w+)\s+\-(0x\w+)/;
|
|
my ($description) = $line =~ /\/\*\*< (.*?)\.? \*\//;
|
|
|
|
if( $error_codes_seen{$error_code}++ ) {
|
|
die "Duplicated error code: $error_code ($error_name)\n";
|
|
}
|
|
|
|
$description =~ s/\\/\\\\/g;
|
|
if ($description eq "") {
|
|
$description = "DESCRIPTION MISSING";
|
|
warn "Missing description for $error_name\n";
|
|
}
|
|
|
|
my ($module_name) = $error_name =~ /^MBEDTLS_ERR_([^_]+)/;
|
|
|
|
# Fix faulty ones
|
|
$module_name = "BIGNUM" if ($module_name eq "MPI");
|
|
$module_name = "CTR_DRBG" if ($module_name eq "CTR");
|
|
$module_name = "HMAC_DRBG" if ($module_name eq "HMAC");
|
|
|
|
my $define_name = $module_name;
|
|
$define_name = "X509_USE,X509_CREATE" if ($define_name eq "X509");
|
|
$define_name = "ASN1_PARSE" if ($define_name eq "ASN1");
|
|
$define_name = "SSL_TLS" if ($define_name eq "SSL");
|
|
$define_name = "PEM_PARSE,PEM_WRITE" if ($define_name eq "PEM");
|
|
|
|
my $include_name = $module_name;
|
|
$include_name =~ tr/A-Z/a-z/;
|
|
$include_name = "" if ($include_name eq "asn1");
|
|
|
|
# Fix faulty ones
|
|
$include_name = "net_sockets" if ($module_name eq "NET");
|
|
|
|
my $found_ll = grep $_ eq $module_name, @low_level_modules;
|
|
my $found_hl = grep $_ eq $module_name, @high_level_modules;
|
|
if (!$found_ll && !$found_hl)
|
|
{
|
|
printf("Error: Do not know how to handle: $module_name\n");
|
|
exit 1;
|
|
}
|
|
|
|
my $code_check;
|
|
my $old_define;
|
|
my $white_space;
|
|
my $first;
|
|
|
|
if ($found_ll)
|
|
{
|
|
$code_check = \$ll_code_check;
|
|
$old_define = \$ll_old_define;
|
|
$white_space = ' ';
|
|
}
|
|
else
|
|
{
|
|
$code_check = \$hl_code_check;
|
|
$old_define = \$hl_old_define;
|
|
$white_space = ' ';
|
|
}
|
|
|
|
if ($define_name ne ${$old_define})
|
|
{
|
|
if (${$old_define} ne "")
|
|
{
|
|
${$code_check} .= "#endif /* ";
|
|
$first = 0;
|
|
foreach my $dep (split(/,/, ${$old_define}))
|
|
{
|
|
${$code_check} .= " || " if ($first++);
|
|
${$code_check} .= "MBEDTLS_${dep}_C";
|
|
}
|
|
${$code_check} .= " */\n\n";
|
|
}
|
|
|
|
${$code_check} .= "#if ";
|
|
$headers .= "#if " if ($include_name ne "");
|
|
$first = 0;
|
|
foreach my $dep (split(/,/, ${define_name}))
|
|
{
|
|
${$code_check} .= " || " if ($first);
|
|
$headers .= " || " if ($first++);
|
|
|
|
${$code_check} .= "defined(MBEDTLS_${dep}_C)";
|
|
$headers .= "defined(MBEDTLS_${dep}_C)" if
|
|
($include_name ne "");
|
|
}
|
|
${$code_check} .= "\n";
|
|
$headers .= "\n#include \"mbedtls/${include_name}.h\"\n".
|
|
"#endif\n\n" if ($include_name ne "");
|
|
${$old_define} = $define_name;
|
|
}
|
|
|
|
if ($error_name eq "MBEDTLS_ERR_SSL_FATAL_ALERT_MESSAGE")
|
|
{
|
|
${$code_check} .= "${white_space}if( use_ret == -($error_name) )\n".
|
|
"${white_space}\{\n".
|
|
"${white_space} mbedtls_snprintf( buf, buflen, \"$module_name - $description\" );\n".
|
|
"${white_space} return;\n".
|
|
"${white_space}}\n"
|
|
}
|
|
else
|
|
{
|
|
${$code_check} .= "${white_space}if( use_ret == -($error_name) )\n".
|
|
"${white_space} mbedtls_snprintf( buf, buflen, \"$module_name - $description\" );\n"
|
|
}
|
|
};
|
|
|
|
if ($ll_old_define ne "")
|
|
{
|
|
$ll_code_check .= "#endif /* ";
|
|
my $first = 0;
|
|
foreach my $dep (split(/,/, $ll_old_define))
|
|
{
|
|
$ll_code_check .= " || " if ($first++);
|
|
$ll_code_check .= "MBEDTLS_${dep}_C";
|
|
}
|
|
$ll_code_check .= " */\n";
|
|
}
|
|
if ($hl_old_define ne "")
|
|
{
|
|
$hl_code_check .= "#endif /* ";
|
|
my $first = 0;
|
|
foreach my $dep (split(/,/, $hl_old_define))
|
|
{
|
|
$hl_code_check .= " || " if ($first++);
|
|
$hl_code_check .= "MBEDTLS_${dep}_C";
|
|
}
|
|
$hl_code_check .= " */\n";
|
|
}
|
|
|
|
$error_format =~ s/HEADER_INCLUDED\n/$headers/g;
|
|
$error_format =~ s/LOW_LEVEL_CODE_CHECKS\n/$ll_code_check/g;
|
|
$error_format =~ s/HIGH_LEVEL_CODE_CHECKS\n/$hl_code_check/g;
|
|
|
|
open(ERROR_FILE, ">$error_file") or die "Opening destination file '$error_file': $!";
|
|
print ERROR_FILE $error_format;
|
|
close(ERROR_FILE);
|