denozord/strongswan-ext
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

crypt-burn: Add a encryption buffer command line argument

Browse Source
This commit is contained in:
Martin Willi
2015-04-15 11:35:25 +02:00
parent 466d560a33
commit 3935d812b7
1 changed files with 14 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/crypt_burn.c
+14 -9
View File
@@ -16,7 +16,7 @@
#include <stdio.h> #include <stdio.h>
#include <library.h> #include <library.h>
static int burn_crypter(const proposal_token_t *token, u_int limit) static int burn_crypter(const proposal_token_t *token, u_int limit, u_int len)
{ {
chunk_t iv, key, data; chunk_t iv, key, data;
crypter_t *crypter; crypter_t *crypter;
@@ -34,7 +34,7 @@ static int burn_crypter(const proposal_token_t *token, u_int limit)
iv = chunk_alloc(crypter->get_iv_size(crypter)); iv = chunk_alloc(crypter->get_iv_size(crypter));
memset(iv.ptr, 0xFF, iv.len); memset(iv.ptr, 0xFF, iv.len);
data = chunk_alloc(round_up(1024, crypter->get_block_size(crypter))); data = chunk_alloc(round_up(len, crypter->get_block_size(crypter)));
memset(data.ptr, 0xDD, data.len); memset(data.ptr, 0xDD, data.len);
key = chunk_alloc(crypter->get_key_size(crypter)); key = chunk_alloc(crypter->get_key_size(crypter));
memset(key.ptr, 0xAA, key.len); memset(key.ptr, 0xAA, key.len);
@@ -68,7 +68,7 @@ static int burn_crypter(const proposal_token_t *token, u_int limit)
return ok; return ok;
} }
static bool burn_aead(const proposal_token_t *token, u_int limit) static bool burn_aead(const proposal_token_t *token, u_int limit, u_int len)
{ {
chunk_t iv, key, data, dataicv, assoc; chunk_t iv, key, data, dataicv, assoc;
aead_t *aead; aead_t *aead;
@@ -86,7 +86,7 @@ static bool burn_aead(const proposal_token_t *token, u_int limit)
iv = chunk_alloc(aead->get_iv_size(aead)); iv = chunk_alloc(aead->get_iv_size(aead));
memset(iv.ptr, 0xFF, iv.len); memset(iv.ptr, 0xFF, iv.len);
dataicv = chunk_alloc(round_up(1024, aead->get_block_size(aead)) + dataicv = chunk_alloc(round_up(len, aead->get_block_size(aead)) +
aead->get_icv_size(aead)); aead->get_icv_size(aead));
data = chunk_create(dataicv.ptr, dataicv.len - aead->get_icv_size(aead)); data = chunk_create(dataicv.ptr, dataicv.len - aead->get_icv_size(aead));
memset(data.ptr, 0xDD, data.len); memset(data.ptr, 0xDD, data.len);
@@ -127,7 +127,7 @@ static bool burn_aead(const proposal_token_t *token, u_int limit)
int main(int argc, char *argv[]) int main(int argc, char *argv[])
{ {
const proposal_token_t *token; const proposal_token_t *token;
int limit = 0; u_int limit = 0, len = 1024;
bool ok; bool ok;
library_init(NULL, "crypt_burn"); library_init(NULL, "crypt_burn");
@@ -138,12 +138,17 @@ int main(int argc, char *argv[])
if (argc < 2) if (argc < 2)
{ {
fprintf(stderr, "usage: %s <algorithm>!\n", argv[0]); fprintf(stderr, "usage: %s <algorithm> [buflen=%u] [rounds=%u]\n",
argv[0], len, limit);
return 1; return 1;
} }
if (argc > 2) if (argc > 2)
{ {
limit = atoi(argv[2]); len = atoi(argv[2]);
}
if (argc > 3)
{
limit = atoi(argv[3]);
} }
token = lib->proposal->get_token(lib->proposal, argv[1]); token = lib->proposal->get_token(lib->proposal, argv[1]);
@@ -158,11 +163,11 @@ int main(int argc, char *argv[])
case ENCRYPTION_ALGORITHM: case ENCRYPTION_ALGORITHM:
if (encryption_algorithm_is_aead(token->algorithm)) if (encryption_algorithm_is_aead(token->algorithm))
{ {
ok = burn_aead(token, limit); ok = burn_aead(token, limit, len);
} }
else else
{ {
ok = burn_crypter(token, limit); ok = burn_crypter(token, limit, len);
} }
break; break;
default: default: