aes.c File Reference

#include "aes.h"
#include "../common/insecure_memzero.h"
#include <string.h>
#include <assert.h>

Include dependency graph for aes.c:

Go to the source code of this file.

Functions
uint8_t	aes_set_encrypt_key (uint8_t *key, uint16_t key_len, aes_context *ctx)
uint8_t	aes_set_decrypt_key (uint8_t *key, uint16_t key_len, aes_context *ctx)
uint8_t	aes_encrypt (uint8_t *in, uint8_t *out, const aes_context *ctx)
uint8_t	aes_decrypt (uint8_t *in, uint8_t *out, const aes_context *ctx)
uint8_t	aes_cbc_encrypt (uint8_t *in, uint8_t *out, uint16_t len, uint8_t *iv, aes_context *ctx)
uint8_t	aes_cbc_decrypt (uint8_t *in, uint8_t *out, uint16_t len, uint8_t *iv, aes_context *ctx)
void	aes_add_padding (uint8_t *in, uint16_t *len)
void	aes_remove_padding (uint8_t *in, uint16_t *len)
void	aes_destroy (aes_context *ctx)

Function Documentation

aes_add_padding()

void aes_add_padding	(	uint8_t *	in,
		uint16_t *	len )

Definition at line 313 of file aes.c.

                                                 {
 
  in[(*len)++] = 0x80;
  while ((*len) % AES_BLOCK_SIZE != 0) {
    in[(*len)++] = 0x00;
  }
}

Here is the caller graph for this function:

aes_cbc_decrypt()

uint8_t aes_cbc_decrypt	(	uint8_t *	in,
		uint8_t *	out,
		uint16_t	len,
		uint8_t *	iv,
		aes_context *	ctx )

Definition at line 292 of file aes.c.

                                          {
#ifdef _WIN32_BCRYPT
  NTSTATUS status = STATUS_SUCCESS;
  ULONG cbResult = 0;
 
  if (!BCRYPT_SUCCESS(status = BCryptDecrypt(ctx->hKeyCBC, in, len, NULL, iv,
                                             AES_BLOCK_SIZE, out, len,
                                             &cbResult, 0))) {
    return -1;
  }
 
#else
 
  AES_cbc_encrypt(in, out, len, &ctx->key, iv, AES_DECRYPT);
 
#endif
 
  return 0;
}

aes_cbc_encrypt()

uint8_t aes_cbc_encrypt	(	uint8_t *	in,
		uint8_t *	out,
		uint16_t	len,
		uint8_t *	iv,
		aes_context *	ctx )

Definition at line 271 of file aes.c.

                                          {
#ifdef _WIN32_BCRYPT
  NTSTATUS status = STATUS_SUCCESS;
  ULONG cbResult = 0;
 
  if (!BCRYPT_SUCCESS(status = BCryptEncrypt(ctx->hKeyCBC, in, len, NULL, iv,
                                             AES_BLOCK_SIZE, out, len,
                                             &cbResult, 0))) {
    return -1;
  }
 
#else
 
  AES_cbc_encrypt(in, out, len, &ctx->key, iv, AES_ENCRYPT);
 
#endif
 
  return 0;
}

aes_decrypt()

uint8_t aes_decrypt	(	uint8_t *	in,
		uint8_t *	out,
		const aes_context *	ctx )

Definition at line 249 of file aes.c.

                                                                       {
#ifdef _WIN32_BCRYPT
  NTSTATUS status = STATUS_SUCCESS;
  ULONG cbResult = 0;
 
  if (!BCRYPT_SUCCESS(status = BCryptDecrypt(ctx->hKeyECB, in, AES_BLOCK_SIZE,
                                             NULL, NULL, 0, out, AES_BLOCK_SIZE,
                                             &cbResult, 0))) {
    return -1;
  }
 
  assert(cbResult == AES_BLOCK_SIZE);
 
#else
 
  AES_ecb_encrypt(in, out, &ctx->key, AES_DECRYPT);
 
#endif
 
  return 0;
}

aes_destroy()

void aes_destroy

(

aes_context *

ctx

)

Definition at line 330 of file aes.c.

                                   {
  if (!ctx) {
    return;
  }
 
#ifdef _WIN32_BCRYPT
 
  if (ctx->hKeyCBC) {
    BCryptDestroyKey(ctx->hKeyCBC);
  }
  if (ctx->pbKeyCBCObj) {
    free(ctx->pbKeyCBCObj);
  }
  if (ctx->hKeyECB) {
    BCryptDestroyKey(ctx->hKeyECB);
  }
  if (ctx->pbKeyECBObj) {
    free(ctx->pbKeyECBObj);
  }
  if (ctx->hAlgCBC) {
    BCryptCloseAlgorithmProvider(ctx->hAlgCBC, 0);
  }
  if (ctx->hAlgECB) {
    BCryptCloseAlgorithmProvider(ctx->hAlgECB, 0);
  }
 
#endif
 
  insecure_memzero(ctx, sizeof(aes_context));
}

Here is the caller graph for this function:

aes_encrypt()

uint8_t aes_encrypt	(	uint8_t *	in,
		uint8_t *	out,
		const aes_context *	ctx )

Definition at line 229 of file aes.c.

                                                                       {
#ifdef _WIN32_BCRYPT
  NTSTATUS status = STATUS_SUCCESS;
  ULONG cbResult = 0;
 
  if (!BCRYPT_SUCCESS(status = BCryptEncrypt(ctx->hKeyECB, in, AES_BLOCK_SIZE,
                                             NULL, NULL, 0, out, AES_BLOCK_SIZE,
                                             &cbResult, 0))) {
    return -2;
  }
 
#else
 
  AES_ecb_encrypt(in, out, &ctx->key, AES_ENCRYPT);
 
#endif
 
  return 0;
}

Here is the caller graph for this function:

aes_remove_padding()

void aes_remove_padding	(	uint8_t *	in,
		uint16_t *	len )

Definition at line 321 of file aes.c.

                                                    {
 
  while (in[(*len) - 1] == 0) {
    (*len)--;
  }
 
  (*len)--;
}

Here is the caller graph for this function:

aes_set_decrypt_key()

uint8_t aes_set_decrypt_key	(	uint8_t *	key,
		uint16_t	key_len,
		aes_context *	ctx )

Definition at line 201 of file aes.c.

                                                                              {
#ifdef _WIN32_BCRYPT
  NTSTATUS status = STATUS_SUCCESS;
 
  if (!BCRYPT_SUCCESS(status = init_ctx(ctx))) {
    return 1;
  }
 
  if (!BCRYPT_SUCCESS(status = import_key(ctx->hAlgCBC, &(ctx->hKeyCBC),
                                          &(ctx->pbKeyCBCObj), ctx->cbKeyObj,
                                          key, key_len))) {
    return 1;
  }
 
  if (!BCRYPT_SUCCESS(status = import_key(ctx->hAlgECB, &(ctx->hKeyECB),
                                          &(ctx->pbKeyECBObj), ctx->cbKeyObj,
                                          key, key_len))) {
    return 1;
  }
 
#else
  AES_set_decrypt_key(key, key_len * 8, &ctx->key);
  ctx->key_len = key_len;
 
#endif
  return 0;
}

aes_set_encrypt_key()

uint8_t aes_set_encrypt_key	(	uint8_t *	key,
		uint16_t	key_len,
		aes_context *	ctx )

Definition at line 172 of file aes.c.

                                                                              {
#ifdef _WIN32_BCRYPT
  NTSTATUS status = STATUS_SUCCESS;
 
  if (!BCRYPT_SUCCESS(status = init_ctx(ctx))) {
    return 1;
  }
 
  if (!BCRYPT_SUCCESS(status = import_key(ctx->hAlgCBC, &(ctx->hKeyCBC),
                                          &(ctx->pbKeyCBCObj), ctx->cbKeyObj,
                                          key, key_len))) {
    return 1;
  }
 
  if (!BCRYPT_SUCCESS(status = import_key(ctx->hAlgECB, &(ctx->hKeyECB),
                                          &(ctx->pbKeyECBObj), ctx->cbKeyObj,
                                          key, key_len))) {
    return 1;
  }
 
#else
  AES_set_encrypt_key(key, key_len * 8, &ctx->key);
  ctx->key_len = key_len;
 
#endif
 
  return 0;
}

Here is the caller graph for this function: