luckyimnotanigger/Aegis
1
0
Fork 0
mirror of https://github.com/beemdevelopment/Aegis.git synced 2025-04-25 08:16:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge pull request #569 from alexbakker/slot-tests

Browse source 
Add tests for the Slot classes
This commit is contained in:
Michael Schättgen 2020-08-16 15:40:37 +02:00 • committed by GitHub
commit 1d00a2d8d6
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 92 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

92
app/src/test/java/com/beemdevelopment/aegis/SlotTest.java Normal file
View file

@ -0,0 +1,92 @@
package com.beemdevelopment.aegis;
import com.beemdevelopment.aegis.crypto.CryptoUtils;
import com.beemdevelopment.aegis.crypto.MasterKey;
import com.beemdevelopment.aegis.crypto.SCryptParameters;
import com.beemdevelopment.aegis.vault.slots.PasswordSlot;
import com.beemdevelopment.aegis.vault.slots.RawSlot;
import com.beemdevelopment.aegis.vault.slots.SlotException;
import com.beemdevelopment.aegis.vault.slots.SlotIntegrityException;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import java.security.InvalidAlgorithmParameterException;
import java.security.InvalidKeyException;
import java.security.NoSuchAlgorithmException;
import javax.crypto.Cipher;
import javax.crypto.NoSuchPaddingException;
import javax.crypto.SecretKey;
import javax.crypto.spec.SecretKeySpec;
import static org.junit.jupiter.api.Assertions.assertArrayEquals;
import static org.junit.jupiter.api.Assertions.assertThrows;
public class SlotTest {
private MasterKey _masterKey;
@BeforeEach
public void init() {
_masterKey = MasterKey.generate();
}
@Test
public void testRawSlotCrypto() throws
InvalidAlgorithmParameterException, NoSuchAlgorithmException,
InvalidKeyException, NoSuchPaddingException,
SlotException, SlotIntegrityException {
RawSlot slot = new RawSlot();
SecretKey rawKey = CryptoUtils.generateKey();
Cipher cipher = CryptoUtils.createEncryptCipher(rawKey);
slot.setKey(_masterKey, cipher);
cipher = slot.createDecryptCipher(rawKey);
MasterKey decryptedKey = slot.getKey(cipher);
assertArrayEquals(_masterKey.getBytes(), decryptedKey.getBytes());
}
@Test
public void testPasswordSlotCrypto() throws
InvalidAlgorithmParameterException, NoSuchAlgorithmException,
InvalidKeyException, NoSuchPaddingException,
SlotException, SlotIntegrityException {
final char[] password = "test".toCharArray();
final SCryptParameters scryptParams = new SCryptParameters(
CryptoUtils.CRYPTO_SCRYPT_N,
CryptoUtils.CRYPTO_SCRYPT_p,
CryptoUtils.CRYPTO_SCRYPT_r,
new byte[CryptoUtils.CRYPTO_AEAD_KEY_SIZE]
);
PasswordSlot slot = new PasswordSlot();
SecretKey passwordKey = slot.deriveKey(password, scryptParams);
Cipher cipher = CryptoUtils.createEncryptCipher(passwordKey);
slot.setKey(_masterKey, cipher);
cipher = slot.createDecryptCipher(passwordKey);
MasterKey decryptedKey = slot.getKey(cipher);
assertArrayEquals(_masterKey.getBytes(), decryptedKey.getBytes());
}
@Test
public void testSlotIntegrity() throws
InvalidAlgorithmParameterException, NoSuchAlgorithmException,
InvalidKeyException, NoSuchPaddingException,
SlotException {
RawSlot slot = new RawSlot();
SecretKey rawKey = CryptoUtils.generateKey();
Cipher cipher = CryptoUtils.createEncryptCipher(rawKey);
slot.setKey(_masterKey, cipher);
// garble the first byte of the key
byte[] rawKeyBytes = rawKey.getEncoded();
rawKeyBytes[0] = (byte) ~rawKeyBytes[0];
rawKey = new SecretKeySpec(rawKeyBytes, "AES");
Cipher decryptCipher = slot.createDecryptCipher(rawKey);
assertThrows(SlotIntegrityException.class, () -> slot.getKey(decryptCipher));
}
}