You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
76 lines
1.6 KiB
76 lines
1.6 KiB
5 years ago
|
#ifndef HEADER_AES_H
|
||
|
|
#define HEADER_AES_H
|
||
|
|
|
||
|
|
|
||
|
|
#include <stdint.h>
|
||
|
|
|
||
|
|
/**************************************************************************
|
||
|
|
* AES declarations
|
||
|
|
**************************************************************************/
|
||
|
|
|
||
|
|
#define AES_MAXROUNDS 14
|
||
|
|
#define AES_BLOCKSIZE 16
|
||
|
|
#define AES_IV_SIZE 16
|
||
|
|
|
||
|
|
#ifndef htonl
|
||
|
|
#define htonl(a) \
|
||
|
|
((((a) >> 24) & 0x000000ff) | \
|
||
|
|
(((a) >> 8) & 0x0000ff00) | \
|
||
|
|
(((a) << 8) & 0x00ff0000) | \
|
||
|
|
(((a) << 24) & 0xff000000))
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#ifndef ntohl
|
||
|
|
#define ntohl(a) htonl((a))
|
||
|
|
#endif
|
||
|
|
|
||
|
|
|
||
|
|
typedef struct aes_key_st
|
||
|
|
{
|
||
|
|
uint16_t rounds;
|
||
|
|
uint16_t key_size;
|
||
|
|
uint32_t ks[(AES_MAXROUNDS+1)*8];
|
||
|
|
uint8_t iv[AES_IV_SIZE];
|
||
|
|
} AES_CTX;
|
||
|
|
|
||
|
|
typedef enum
|
||
|
|
{
|
||
|
|
AES_MODE_128,
|
||
|
|
AES_MODE_256
|
||
|
|
} AES_MODE;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Set up AES with the key/iv and cipher size.
|
||
|
|
*/
|
||
|
|
void AES_set_key(AES_CTX *ctx, const uint8_t *key,
|
||
|
|
const uint8_t *iv, AES_MODE mode);
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Encrypt a byte sequence (with a block size 16) using the AES cipher.
|
||
|
|
*/
|
||
|
|
void AES_cbc_encrypt(AES_CTX *ctx, const uint8_t *msg,
|
||
|
|
uint8_t *out, int length);
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Decrypt a byte sequence (with a block size 16) using the AES cipher.
|
||
|
|
*/
|
||
|
|
void AES_cbc_decrypt(AES_CTX *ks, const uint8_t *in, uint8_t *out, int length);
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Change a key for decryption.
|
||
|
|
*/
|
||
|
|
void AES_convert_key(AES_CTX *ctx);
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Decrypt a single block (16 bytes) of data
|
||
|
|
*/
|
||
|
|
void AES_decrypt(const AES_CTX *ctx, uint32_t *data);
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Encrypt a single block (16 bytes) of data
|
||
|
|
*/
|
||
|
|
void AES_encrypt(const AES_CTX *ctx, uint32_t *data);
|
||
|
|
|
||
|
|
|
||
|
|
#endif /* !HEADER_AES_H */
|