diff --git a/Implementierung/.gitignore b/Implementierung/.gitignore index d6dce6b..37e27d3 100644 --- a/Implementierung/.gitignore +++ b/Implementierung/.gitignore @@ -1,3 +1,3 @@ build md2 -testfile* \ No newline at end of file +t \ No newline at end of file diff --git a/Implementierung/Makefile b/Implementierung/Makefile index 228eb39..66d8453 100644 --- a/Implementierung/Makefile +++ b/Implementierung/Makefile @@ -1,10 +1,12 @@ -SRC = src/main.c src/helper.c src/io.c src/md2.c src/md2_impls/md2_common.c src/md2_impls/md2_0.c src/md2_impls/md2_2.c src/md2_impls/md2_1.c +SRC = src/main.c src/helper.c src/io.c src/md2.c src/md2_impls/md2_common.c src/md2_impls/md2_0.c src/md2_impls/md2_1.c src/md2_impls/md2_2.c src/md2_impls/md2_3.c OBJ = ${subst src,build,${SRC:.c=.o}} CC = gcc -CFLAGS = -Ilib -ggdb -std=c11 -g -Wall -Wextra -no-pie -O3 -LDFLAGS = +CFLAGS = -Ilib -ggdb -std=c11 -g -Wall -Wextra -no-pie -O3 +LDFLAGS = -pthread +TESTFILES = t/10000 t/1 t/10 t/100 t/1000 #t/2000 t/5000 t/10000 +TESTFILES_SIZES = ${subst t/,,${TESTFILES}} all: md2 @@ -18,9 +20,7 @@ help: @echo - all: build everything @echo - clean: clean distfiles @echo - help: show this help - -build: - mkdir build + @echo - benchmarks: run benchmarks (only works on linux!) build/%.o: src/%.c @mkdir -p build/md2_impls @@ -29,4 +29,33 @@ build/%.o: src/%.c md2: ${OBJ} ${CC} -o $@ $(OBJ) ${LDFLAGS} -.PHONY: all clean help \ No newline at end of file +t/%: + @echo + @echo "=== Generating ${subst t/,,$@}MB of random data... ===" + dd if=/dev/random of=$@ bs=1M count=${subst t/,,$@} status=progress + @echo "=== done ===" + @echo + +benchmarks.csv: md2 ${TESTFILES} + @rm -f $@ + @for i in 0 2 3; do \ + echo ;\ + echo "=== Testing implementation $$i ===";\ + for t in $(TESTFILES_SIZES); do \ + echo -n "- with $${t}MB ... "; \ + if ! rr=$$(./md2 t/$${t} -B1 -V$${i}); then \ + echo; \ + echo "ERROR!"; \ + exit 1; \ + fi; \ + r=$$(echo $$rr | xargs | sed -e 's/.*took \(.*\) seconds.*/\1/'); \ + echo "$${r}s"; \ + echo "$${i};$${t};$${r}" >> $@; \ + done; \ + echo "=== done ===";\ + echo;\ + done + +benchmarks: benchmarks.csv + +.PHONY: all clean help benchmarks \ No newline at end of file diff --git a/Implementierung/lib/io.h b/Implementierung/lib/io.h index 6887f25..532b58c 100644 --- a/Implementierung/lib/io.h +++ b/Implementierung/lib/io.h @@ -9,6 +9,7 @@ #include #include #include +#include /** * @brief Open a file and load its stats diff --git a/Implementierung/lib/md2_impls/md2_2.h b/Implementierung/lib/md2_impls/md2_2.h index 4e64689..d7b370d 100644 --- a/Implementierung/lib/md2_impls/md2_2.h +++ b/Implementierung/lib/md2_impls/md2_2.h @@ -18,6 +18,5 @@ * @param out */ void md2_hash_2(size_t len, const uint8_t buf[len], uint8_t out[16]); -void md2_checksum_2(size_t len, uint8_t* buf); #endif // MD2_2_H \ No newline at end of file diff --git a/Implementierung/lib/md2_impls/md2_3.h b/Implementierung/lib/md2_impls/md2_3.h new file mode 100644 index 0000000..e51c783 --- /dev/null +++ b/Implementierung/lib/md2_impls/md2_3.h @@ -0,0 +1,22 @@ +#ifndef MD2_3_H +#define MD2_3_H + +#include +#include +#include +#include +#include +#include +#include +#include + +/** + * @brief Diese Implementierung benutzt Threads zum Berechnen des Hashs + * + * @param len + * @param buf + * @param out + */ +void md2_hash_3(size_t len, const uint8_t buf[len], uint8_t out[16]); + +#endif // MD2_3_H \ No newline at end of file diff --git a/Implementierung/lib/md2_impls/md2_common.h b/Implementierung/lib/md2_impls/md2_common.h index f2b72c2..c66e756 100644 --- a/Implementierung/lib/md2_impls/md2_common.h +++ b/Implementierung/lib/md2_impls/md2_common.h @@ -1,6 +1,7 @@ #ifndef MD2_COMMON_H #define MD2_COMMON_H +#include #include #include #include diff --git a/Implementierung/src/io.c b/Implementierung/src/io.c index 3dbc7ff..857a996 100644 --- a/Implementierung/src/io.c +++ b/Implementierung/src/io.c @@ -33,9 +33,17 @@ uint8_t* read_file(const char* path, size_t* size) { return NULL; } + struct sysinfo info; + int r = sysinfo(&info); + if (r != 0 || info.freeram < (unsigned long)statOfFile.st_size * 2) { + errno = ENOMEM; + return NULL; + } + uint8_t* data = malloc(statOfFile.st_size); if (data == NULL) { - return; + fclose(f); + return NULL; } size_t bytesRead = diff --git a/Implementierung/src/main.c b/Implementierung/src/main.c index f4e5fc8..4441094 100644 --- a/Implementierung/src/main.c +++ b/Implementierung/src/main.c @@ -5,7 +5,6 @@ #include "../lib/io.h" #include "../lib/md2.h" -// Returns true when val is approx. equal to exp. static bool runTest(struct configuration* c, const char* message, const char* expectedHash) { uint8_t out[16]; @@ -140,8 +139,10 @@ int main(int argc, char** argv) { char hash[32]; if (!calculate_hash(c, hash)) { if (errno != 0) { - fprintf(stderr, "\n\033[0;31mAn error occured: %s\033[0m\n", + fprintf(stderr, "\n\033[1;31mAn error occured: %s\033[0m\n", strerror(errno)); + if (errno == ENOMEM) + fprintf(stderr, "\033[1;36mPlease try to use -V2!\033[0m\n"); } return EXIT_FAILURE; } diff --git a/Implementierung/src/md2.c b/Implementierung/src/md2.c index de9b908..f06ce10 100644 --- a/Implementierung/src/md2.c +++ b/Implementierung/src/md2.c @@ -4,6 +4,7 @@ #include "../lib/md2_impls/md2_0.h" #include "../lib/md2_impls/md2_1.h" #include "../lib/md2_impls/md2_2.h" +#include "../lib/md2_impls/md2_3.h" md2_hash_func md2_hash; md2_checksum_func md2_checksum; @@ -31,7 +32,12 @@ bool md2_choose_implementation(int i) { case 2: md2_hash = md2_hash_2; - md2_checksum = md2_checksum_2; + md2_checksum = NULL; + return true; + + case 3: + md2_hash = md2_hash_3; + md2_checksum = NULL; return true; default: diff --git a/Implementierung/src/md2_impls/md2_2.c b/Implementierung/src/md2_impls/md2_2.c index 90fd232..b165d05 100644 --- a/Implementierung/src/md2_impls/md2_2.c +++ b/Implementierung/src/md2_impls/md2_2.c @@ -38,9 +38,6 @@ void apply_padding(size_t len, uint8_t buf[16]) { } } -// unused! -void md2_checksum_2(size_t, uint8_t*) {} - void md2_hash_2(size_t len, const uint8_t buf[len], uint8_t out[16]) { FILE* file; struct stat file_stat; @@ -71,8 +68,8 @@ void md2_hash_2(size_t len, const uint8_t buf[len], uint8_t out[16]) { while (bytes_left_to_read != 0) { bytes_left_to_process = bytes_left_to_read >= 16 ? 16 : bytes_left_to_read; - fread(data, 1, bytes_left_to_process, file); - if (ferror(file) || feof(file)) { + size_t size = fread(data, 1, bytes_left_to_process, file); + if (size == 0 || ferror(file) || feof(file)) { if (errno == 0) errno = EIO; return; } diff --git a/Implementierung/src/md2_impls/md2_3.c b/Implementierung/src/md2_impls/md2_3.c new file mode 100644 index 0000000..63aebad --- /dev/null +++ b/Implementierung/src/md2_impls/md2_3.c @@ -0,0 +1,114 @@ +#include "../../lib/md2_impls/md2_3.h" + +#include "../../lib/md2_impls/md2_common.h" + +struct thread_args { + size_t len; + const uint8_t* buf; +}; + +void process_nothread_hash(size_t len, const uint8_t buf[len], + uint8_t messageDigestBuf[48]) { + for (size_t i = 0; i < (len + 16) / 16 - 1; i++) { + for (int j = 0; j < 16; j++) { + messageDigestBuf[16 + j] = buf[i * 16 + j]; + messageDigestBuf[32 + j] = + (messageDigestBuf[16 + j] ^ messageDigestBuf[j]); + } + + u_int8_t t = 0; + + for (int j = 0; j < 18; j++) { + for (int k = 0; k < 48; k++) { + t = messageDigestBuf[k] = messageDigestBuf[k] ^ MD2_PI_SUBST[t]; + } + t = (t + j) % 256; + } + } +} + +void* process_hash(void* threadArgs) { + struct thread_args* args = (struct thread_args*)threadArgs; + + uint8_t* messageDigestBuf = calloc(48, sizeof(uint8_t)); + if (messageDigestBuf == NULL) { + return NULL; + } + + process_nothread_hash(args->len, args->buf, messageDigestBuf); + pthread_exit(messageDigestBuf); +} + +void* process_checksum(void* threasdArgs) { + struct thread_args* args = (struct thread_args*)threasdArgs; + + uint8_t* checksum = calloc(16, sizeof(uint8_t)); + if (checksum == NULL) { + return NULL; + } + + uint8_t l = 0; + + for (size_t i = 0; i < args->len / 16; i++) { + for (int j = 0; j < 16; j++) { + u_int8_t c = args->buf[i * 16 + j]; + // reference is wrong. It says: Set C[j] to S[c xor L]. But it should be: + l = checksum[j] ^= MD2_PI_SUBST[c ^ l]; + } + } + pthread_exit(checksum); +} + +void md2_hash_3(size_t len, const uint8_t buf[len], uint8_t out[16]) { + int paddingNeeded = 16 - (len % 16); + uint8_t originalPadding = paddingNeeded; + len += paddingNeeded; + + uint8_t* newBuf = calloc(len + 16, sizeof(uint8_t)); + if (newBuf == NULL) { + return; + } + + memcpy(newBuf, buf, len - paddingNeeded); + + while (paddingNeeded > 0) { + newBuf[len - paddingNeeded] = originalPadding; + paddingNeeded--; + } + + pthread_t thread_1, thread_2; + + struct thread_args thread_args = {len, newBuf}; + + if (pthread_create(&thread_1, NULL, process_hash, (void*)&thread_args) != 0) { + printf("Error creating thread 1\n"); + if (errno == 0) errno = EAGAIN; + return; + } + if (pthread_create(&thread_2, NULL, process_checksum, (void*)&thread_args) != + 0) { + printf("Error creating thread 2\n"); + if (errno == 0) errno = EAGAIN; + return; + } + + u_int8_t* messageDigestBuf; + u_int8_t* checksum; + + if (pthread_join(thread_1, (void**)&messageDigestBuf) != 0) { + printf("Error joining thread 1\n"); + if (errno == 0) errno = EINVAL; + return; + } + if (pthread_join(thread_2, (void**)&checksum) != 0) { + printf("Error joining thread 2\n"); + if (errno == 0) errno = EINVAL; + return; + } + + process_nothread_hash(16, checksum, messageDigestBuf); + memcpy(out, messageDigestBuf, 16); + + free(messageDigestBuf); + free(checksum); +}