Hamlib/tests/testsecurity.c

127 wiersze
3.4 KiB
C

/* Borrowed for Hamlib from:
* String Crypt Test (Linux)
* Copyright (C) 2012, 2015
*
* Author: Paul E. Jones <paulej@packetizer.com>
*
* This software is licensed as "freeware." Permission to distribute
* this software in source and binary forms is hereby granted without a
* fee. THIS SOFTWARE IS PROVIDED 'AS IS' AND WITHOUT ANY EXPRESSED OR
* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
* THE AUTHOR SHALL NOT BE HELD LIABLE FOR ANY DAMAGES RESULTING FROM
* THE USE OF THIS SOFTWARE, EITHER DIRECTLY OR INDIRECTLY, INCLUDING,
* BUT NOT LIMITED TO, LOSS OF DATA OR DATA BEING RENDERED INACCURATE.
*
*/
#include <stdio.h>
#include <string.h>
#ifdef _WIN32
#include <windows.h>
//#include <Wincrypt.h>
#else
#include <unistd.h>
#endif
#include "AESStringCrypt.h"
#include "password.h"
#include "../src/misc.h"
#if defined(_WIN32)
// gmtime_r can be defined by mingw
#ifndef gmtime_r
static struct tm *gmtime_r(const time_t *t, struct tm *r)
{
// gmtime is threadsafe in windows because it uses TLS
struct tm *theTm = gmtime(t);
if (theTm)
{
*r = *theTm;
return r;
}
else
{
return 0;
}
}
#endif // gmtime_r
#endif // _WIN32
// using tv_usec with a sleep gives a fairly good random number
static int my_rand(int max)
{
time_t t;
struct timeval tv;
struct tm result;
t = time(NULL);
gmtime_r(&t, &result);
gettimeofday(&tv, NULL);
hl_usleep(100);
int val = tv.tv_usec % max;
return val;
}
void rig_make_key(char key[33])
{
char *all =
"abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ123467890!@#$%^&*()_=~<>/?";
int max = strlen(all);
int i;
for (i = 0; i < 32; ++i)
{
key[i] = all[my_rand(max)];
}
key[32] = 0;
}
int main()
{
char key1[33];
char key2[33];
char plaintext[33];
unsigned char ciphertext[1024];
int ciphertext_length;
int plaintext_length;
memset(ciphertext, 0, sizeof(ciphertext));
rig_make_key(key1);
rig_make_key(key2);
printf("key1=%s\n", key1);
printf("key2=%s\n", key2);
ciphertext_length = AESStringCrypt((unsigned char *) key1,
strlen(key1),
(unsigned char *) key2,
strlen(key2),
(unsigned char *) ciphertext);
for (int i = 0; i < ciphertext_length; ++i) { printf("%02x", ciphertext[i]); }
printf("\n");
if (ciphertext_length == AESSTRINGCRYPT_ERROR)
{
printf("Error encrypting the string\n");
}
printf("Ciphertext length: %d\n", ciphertext_length);
memset(plaintext, 0, sizeof(plaintext));
printf("Decrypting...\n");
plaintext_length = AESStringDecrypt((unsigned char *) key1,
strlen(key1),
(unsigned char *) ciphertext,
ciphertext_length,
(unsigned char *) plaintext);
if (plaintext_length == AESSTRINGCRYPT_ERROR)
{
printf("Error decrypting the string\n");
}
printf("Decrypted plaintext length: %d, %s\n", plaintext_length, plaintext);
}