summaryrefslogtreecommitdiffstats
path: root/openaes/src/oaes.c
diff options
context:
space:
mode:
authorDees_Troy <dees_troy@teamw.in>2013-05-04 14:39:56 +0200
committerDees_Troy <dees_troy@yahoo.com>2013-06-22 23:16:55 +0200
commit83bd483f3db1df4b69c1612030e1a1be05b303ad (patch)
treed5c7b468b831244bfcd90aadda98b9d359d0d280 /openaes/src/oaes.c
parentFix building of updater binary (diff)
downloadandroid_bootable_recovery-83bd483f3db1df4b69c1612030e1a1be05b303ad.tar
android_bootable_recovery-83bd483f3db1df4b69c1612030e1a1be05b303ad.tar.gz
android_bootable_recovery-83bd483f3db1df4b69c1612030e1a1be05b303ad.tar.bz2
android_bootable_recovery-83bd483f3db1df4b69c1612030e1a1be05b303ad.tar.lz
android_bootable_recovery-83bd483f3db1df4b69c1612030e1a1be05b303ad.tar.xz
android_bootable_recovery-83bd483f3db1df4b69c1612030e1a1be05b303ad.tar.zst
android_bootable_recovery-83bd483f3db1df4b69c1612030e1a1be05b303ad.zip
Diffstat (limited to 'openaes/src/oaes.c')
-rw-r--r--openaes/src/oaes.c317
1 files changed, 317 insertions, 0 deletions
diff --git a/openaes/src/oaes.c b/openaes/src/oaes.c
new file mode 100644
index 000000000..9a5d26849
--- /dev/null
+++ b/openaes/src/oaes.c
@@ -0,0 +1,317 @@
+/*
+ * ---------------------------------------------------------------------------
+ * OpenAES License
+ * ---------------------------------------------------------------------------
+ * Copyright (c) 2012, Nabil S. Al Ramli, www.nalramli.com
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ * - Redistributions of source code must retain the above copyright notice,
+ * this list of conditions and the following disclaimer.
+ * - Redistributions in binary form must reproduce the above copyright
+ * notice, this list of conditions and the following disclaimer in the
+ * documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+ * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
+ * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
+ * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
+ * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+ * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
+ * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
+ * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
+ * POSSIBILITY OF SUCH DAMAGE.
+ * ---------------------------------------------------------------------------
+ */
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+#define OAES_DEBUG 1
+#include "../inc/oaes_lib.h"
+
+#if defined(_WIN32) && !defined(__SYMBIAN32__)
+#include <io.h>
+#else
+__inline static int setmode(int a, int b)
+{
+ return 0;
+}
+#endif
+
+#ifndef __max
+ #define __max(a,b) (((a) > (b)) ? (a) : (b))
+#endif // __max
+
+#ifndef __min
+ #define __min(a,b) (((a) < (b)) ? (a) : (b))
+#endif // __min
+
+#define OAES_BUF_LEN_ENC 4096 - 2 * OAES_BLOCK_SIZE
+#define OAES_BUF_LEN_DEC 4096
+
+static void usage( const char * exe_name )
+{
+ if( NULL == exe_name )
+ return;
+
+ fprintf( stderr,
+ "Usage:\n"
+ " %s <command> --key <key_data> [options]\n"
+ "\n"
+ " command:\n"
+ " enc: encrypt\n"
+ " dec: decrypt\n"
+ "\n"
+ " options:\n"
+ " --ecb: use ecb mode instead of cbc\n"
+ " --in <path_in>\n"
+ " --out <path_out>\n"
+ "\n",
+ exe_name
+ );
+}
+
+int main(int argc, char** argv)
+{
+ size_t _i = 0, _j = 0;
+ OAES_CTX * ctx = NULL;
+ uint8_t _buf_in[OAES_BUF_LEN_DEC];
+ uint8_t *_buf_out = NULL, _key_data[32] = "";
+ size_t _buf_in_len = 0, _buf_out_len = 0, _read_len = 0;
+ size_t _key_data_len = 0;
+ short _is_ecb = 0;
+ char *_file_in = NULL, *_file_out = NULL;
+ int _op = 0;
+ FILE *_f_in = stdin, *_f_out = stdout;
+
+ fprintf( stderr, "\n"
+ "*******************************************************************************\n"
+ "* OpenAES %-10s *\n"
+ "* Copyright (c) 2012, Nabil S. Al Ramli, www.nalramli.com *\n"
+ "*******************************************************************************\n\n",
+ OAES_VERSION );
+
+ // pad the key
+ for( _j = 0; _j < 32; _j++ )
+ _key_data[_j] = _j + 1;
+
+ if( argc < 2 )
+ {
+ usage( argv[0] );
+ return EXIT_FAILURE;
+ }
+
+ if( 0 == strcmp( argv[1], "enc" ) )
+ {
+ _op = 0;
+ _read_len = OAES_BUF_LEN_ENC;
+ }
+ else if( 0 == strcmp( argv[1], "dec" ) )
+ {
+ _op = 1;
+ _read_len = OAES_BUF_LEN_DEC;
+ }
+ else
+ {
+ fprintf(stderr, "Error: Unknown command '%s'.", argv[1]);
+ usage( argv[0] );
+ return EXIT_FAILURE;
+ }
+
+ for( _i = 2; _i < argc; _i++ )
+ {
+ int _found = 0;
+
+ if( 0 == strcmp( argv[_i], "--ecb" ) )
+ {
+ _found = 1;
+ _is_ecb = 1;
+ }
+
+ if( 0 == strcmp( argv[_i], "--key" ) )
+ {
+ _found = 1;
+ _i++; // key_data
+ if( _i >= argc )
+ {
+ fprintf(stderr, "Error: No value specified for '%s'.\n",
+ "--key");
+ usage( argv[0] );
+ return EXIT_FAILURE;
+ }
+ _key_data_len = strlen(argv[_i]);
+ if( 16 >= _key_data_len )
+ _key_data_len = 16;
+ else if( 24 >= _key_data_len )
+ _key_data_len = 24;
+ else
+ _key_data_len = 32;
+ memcpy(_key_data, argv[_i], __min(32, strlen(argv[_i])));
+ }
+
+ if( 0 == strcmp( argv[_i], "--in" ) )
+ {
+ _found = 1;
+ _i++; // path_in
+ if( _i >= argc )
+ {
+ fprintf(stderr, "Error: No value specified for '%s'.\n",
+ "--in");
+ usage( argv[0] );
+ return EXIT_FAILURE;
+ }
+ _file_in = argv[_i];
+ }
+
+ if( 0 == strcmp( argv[_i], "--out" ) )
+ {
+ _found = 1;
+ _i++; // path_out
+ if( _i >= argc )
+ {
+ fprintf(stderr, "Error: No value specified for '%s'.\n",
+ "--out");
+ usage( argv[0] );
+ return EXIT_FAILURE;
+ }
+ _file_out = argv[_i];
+ }
+
+ if( 0 == _found )
+ {
+ fprintf(stderr, "Error: Invalid option '%s'.\n", argv[_i]);
+ usage( argv[0] );
+ return EXIT_FAILURE;
+ }
+ }
+
+ if( 0 == _key_data_len )
+ {
+ fprintf(stderr, "Error: --key must be specified.\n");
+ return EXIT_FAILURE;
+ }
+
+ if( _file_in )
+ {
+ _f_in = fopen(_file_in, "rb");
+ if( NULL == _f_in )
+ {
+ fprintf(stderr,
+ "Error: Failed to open '-%s' for reading.\n", _file_in);
+ return EXIT_FAILURE;
+ }
+ }
+ else
+ {
+ if( setmode(fileno(stdin), 0x8000) < 0 )
+ fprintf(stderr,"Error: Failed in setmode().\n");
+ _f_in = stdin;
+ }
+
+ if( _file_out )
+ {
+ _f_out = fopen(_file_out, "wb");
+ if( NULL == _f_out )
+ {
+ fprintf(stderr,
+ "Error: Failed to open '-%s' for writing.\n", _file_out);
+ if( _file_in )
+ fclose(_f_in);
+ return EXIT_FAILURE;
+ }
+ }
+ else
+ {
+ if( setmode(fileno(stdout), 0x8000) < 0 )
+ fprintf(stderr, "Error: Failed in setmode().\n");
+ _f_out = stdout;
+ }
+
+ ctx = oaes_alloc();
+ if( NULL == ctx )
+ {
+ fprintf(stderr, "Error: Failed to initialize OAES.\n");
+ if( _file_in )
+ fclose(_f_in);
+ if( _file_out )
+ fclose(_f_out);
+ return EXIT_FAILURE;
+ }
+ if( _is_ecb )
+ if( OAES_RET_SUCCESS != oaes_set_option( ctx, OAES_OPTION_ECB, NULL ) )
+ fprintf(stderr, "Error: Failed to set OAES options.\n");
+
+ oaes_key_import_data( ctx, _key_data, _key_data_len );
+
+ while( _buf_in_len =
+ fread(_buf_in, sizeof(uint8_t), _read_len, _f_in) )
+ {
+ switch(_op)
+ {
+ case 0:
+ if( OAES_RET_SUCCESS != oaes_encrypt( ctx,
+ _buf_in, _buf_in_len, NULL, &_buf_out_len ) )
+ fprintf( stderr,
+ "Error: Failed to retrieve required buffer size for "
+ "encryption.\n" );
+ _buf_out = (uint8_t *) calloc( _buf_out_len, sizeof( char ) );
+ if( NULL == _buf_out )
+ {
+ fprintf(stderr, "Error: Failed to allocate memory.\n" );
+ if( _file_in )
+ fclose(_f_in);
+ if( _file_out )
+ fclose(_f_out);
+ return EXIT_FAILURE;
+ }
+ if( OAES_RET_SUCCESS != oaes_encrypt( ctx,
+ _buf_in, _buf_in_len, _buf_out, &_buf_out_len ) )
+ fprintf(stderr, "Error: Encryption failed.\n");
+ fwrite(_buf_out, sizeof(uint8_t), _buf_out_len, _f_out);
+ free(_buf_out);
+ break;
+ case 1:
+ if( OAES_RET_SUCCESS != oaes_decrypt( ctx,
+ _buf_in, _buf_in_len, NULL, &_buf_out_len ) )
+ fprintf( stderr,
+ "Error: Failed to retrieve required buffer size for "
+ "encryption.\n" );
+ _buf_out = (uint8_t *) calloc( _buf_out_len, sizeof( char ) );
+ if( NULL == _buf_out )
+ {
+ fprintf(stderr, "Error: Failed to allocate memory.\n" );
+ free( _buf_out );
+ if( _file_in )
+ fclose(_f_in);
+ if( _file_out )
+ fclose(_f_out);
+ return EXIT_FAILURE;
+ }
+ if( OAES_RET_SUCCESS != oaes_decrypt( ctx,
+ _buf_in, _buf_in_len, _buf_out, &_buf_out_len ) )
+ fprintf(stderr, "Error: Decryption failed.\n");
+ fwrite(_buf_out, sizeof(uint8_t), _buf_out_len, _f_out);
+ free(_buf_out);
+ break;
+ default:
+ break;
+ }
+ }
+
+
+ if( OAES_RET_SUCCESS != oaes_free( &ctx ) )
+ fprintf(stderr, "Error: Failed to uninitialize OAES.\n");
+
+ if( _file_in )
+ fclose(_f_in);
+ if( _file_out )
+ fclose(_f_out);
+
+ return (EXIT_SUCCESS);
+}