completed set3, task 18

1 files changed, 83 insertions, 0 deletions

diff --git a/lib/lib3.c b/lib/lib3.c
new file mode 100644
index 0000000..87865ed
--- /dev/null
+++ b/lib/lib3.c
@@ -0,0 +1,83 @@
+#include "lib3.h"
+#include "lib2.h"
+#include "lib.h"
+
+
+#define NR_STIRNGS_CHALLENGE17 10
+
+char *challenge17_encrypt(int *length)
+{
+	int i, t;
+	char **string;
+	char **decoded;
+	string = malloc(sizeof(char *)*NR_STIRNGS_CHALLENGE17);
+	decoded = malloc(sizeof(char *)*NR_STIRNGS_CHALLENGE17);
+	char filename[] = "task17_0";	
+
+	for(i=0;i<10;i++) {
+		filename[strlen(filename)-1] = (char) (i+'0');
+		t = read_base64_file(filename, &string[i]);
+		decoded[i] = malloc(t);
+		length[i] = decode_base64(string[i], decoded[i]);
+		printf("read: %s\n", string[i]);
+	}
+	// choose one randomly
+	int random = rand() % NR_STIRNGS_CHALLENGE17;
+	int padding;
+	printf("plaintext: %s\n", string[random]);
+	*length = strlen(string[random]);
+	char *padded_string = __pkcs7_padding(string[random], *length, 16, &padding);
+	char *encrypted = malloc(strlen(padded_string));
+	*length += padding;
+	aes_cbc(padded_string, strlen(padded_string), encrypted, key, iv, 1);
+	return encrypted;	
+}
+
+int cbc_padding_oracle(char *encrypted, int length)
+{
+	char *decrypted = malloc(length);
+	char *unpadded= malloc(length);
+	
+	aes_cbc(encrypted, length, decrypted, key, iv, 0);
+
+	int valid = valid_pkcs7_padding(decrypted, length, unpadded, 16);
+	free(decrypted);
+	free(unpadded);
+	return valid;
+
+}
+
+int convert_to_little_endian(char *string)
+{
+
+}
+/**
+  * format is: 64 nonce concat with 64 bit counter
+  * calle has to make sure that nonce is at least 8 bytes
+  * its all little endian
+  */
+int aes_ctr(char *in, int length_in, char *out, char *string_key, char *nonce)
+{
+	long counter;
+	unsigned char tmp[16];
+	char keystream[16];
+	long nr_blocks = length_in / BLOCKSIZE;
+	int length_last_block = length_in % BLOCKSIZE;
+	memcpy(tmp, nonce, 16);
+
+	for(counter=0;counter<nr_blocks-1;counter++) {
+		// right now it only works for 256 block :P 
+		tmp[8] = counter % 256;
+		// encrypt nonce and counter and produce keystream
+		aes_ecb(tmp, 16, keystream, string_key, 16, 1);
+		// xor against keystream to encrpy/decrypt
+		xor_string(&in[counter*BLOCKSIZE], keystream, &out[counter*BLOCKSIZE], 16, 16); 
+	}
+
+	// do last block
+	tmp[8] = counter % 256;
+	aes_ecb(tmp, 16, keystream, string_key, 16, 1);
+	xor_string(&in[counter*BLOCKSIZE], keystream, &out[counter*BLOCKSIZE],
+			16, length_last_block);
+
+}