- aes_encrypt_key(key, keylen, &((AesContext *)context)->enc);
- aes_decrypt_key(key, keylen, &((AesContext *)context)->dec);
+ if (encryption)
+ aes_encrypt_key(key, keylen, &((AesContext *)context)->u.enc);
+ else
+ aes_decrypt_key(key, keylen, &((AesContext *)context)->u.dec);