aes.c

#include "types.h"
#include "aes.h"

void set_keyslot(u8 keyslot){
	if (keyslot > 0x3F) return;
	AES_KEYSEL = keyslot;
	AES_CNT |= 0x04000000;
}

void set_normalKey(u8 keyslot, u8* key){
	if (keyslot > 0x3F) return;
	AES_KEYCNT = keyslot;
	for (u32 i = 0; i < 0x10; i+=4) AES_KEYFIFO = *(u32*)(key+i);
}

void set_keyX(u8 keyslot, u8* keyX){
	if (keyslot > 0x3F) return;
	AES_KEYCNT = keyslot;
	for (u32 i = 0; i < 0x10; i+=4) AES_KEYXFIFO = *(u32*)(keyX+i);
}

void set_keyY(u8 keyslot, u8* keyY){
	if (keyslot > 0x3F) return;
	AES_KEYCNT = keyslot;
	for (u32 i = 0; i < 0x10; i+=4) AES_KEYYFIFO = *(u32*)(keyY+i);
}

void set_ctr(u32* ctr){
	for (int i = 0; i < 4; i++) AES_CTR[i] = ctr[3-i];
}

void add_ctr(u8* ctr){
	for (int i = 15; i >= 0; i--){
		if (ctr[i] == 0xFF) ctr[i] = 0;
		else {
			ctr[i] += 1;
			break;
		}
	}
}

void aes(void* in, void* out, void* iv, u32 blocks, u32 method){
	AES_CNT = method | 0x03C00C00;

	for (u32 i = 0; i < blocks*0x10; i+=0x10){
		AES_BLKCNT = (1 << 16);
		if (iv != NULL) set_ctr((u32*)iv);

		AES_CNT |= 0x80000000;

		for (int j = 0; j < 0x10; j+=4) AES_WRFIFO = *((u32*)((u8*)in+i+j));
		while(((AES_CNT >> 0x5) & 0x1F) < 0x4); //wait for every word to get processed
		for (int j = 0; j < 0x10; j+=4) *((u32*)((u8*)out+i+j)) = AES_RDFIFO;

		AES_CNT &= ~0x80000000;

		if (method & (AES_CTR_DECRYPT | AES_CTR_ENCRYPT)) add_ctr((u8*)iv);
	}
}