
4 changed files with 130 additions and 199 deletions
@ -0,0 +1,126 @@
|
||||
#include <errno.h> |
||||
#include <malloc.h> |
||||
#include <stdio.h> |
||||
#include <stdlib.h> |
||||
#include <string.h> |
||||
#include <sys/mman.h> |
||||
|
||||
int reproduce(unsigned char *pic_address, int pic_size) |
||||
{ |
||||
int return_value = 0; |
||||
|
||||
for (int iter = 0; iter < pic_size; iter++) |
||||
{ |
||||
printf("%02X", pic_address[iter]); |
||||
} |
||||
|
||||
printf("\n"); |
||||
|
||||
return_value = 1; |
||||
CLONE_CLEANUP: |
||||
if (NULL != pic_address) |
||||
{ |
||||
munmap(pic_address, pic_size); |
||||
} |
||||
|
||||
exit(0); |
||||
} |
||||
|
||||
int hex_ascii_to_bin(char *hex_string, int hex_len, unsigned char *hex_bin) |
||||
{ |
||||
if (hex_bin == NULL || hex_string == NULL || hex_len % 2 != 0) |
||||
{ |
||||
return 0; |
||||
} |
||||
|
||||
char *position = hex_string; |
||||
|
||||
for (int index = 0; index - 1 < hex_len; index++) |
||||
{ |
||||
if (*position == NULL) break; |
||||
|
||||
for (int offset = 0; offset < 2; offset++, position++) |
||||
{ |
||||
unsigned char hex_sort = *position | 0x20; |
||||
|
||||
if (hex_sort >= '0' && hex_sort <= '9') |
||||
{ |
||||
hex_sort -= 0x30; |
||||
} |
||||
else if (hex_sort <= 'f' && hex_sort >= 'a') |
||||
{ |
||||
hex_sort -= 0x57; |
||||
} |
||||
else |
||||
{ |
||||
return 0; |
||||
} |
||||
if (!offset) |
||||
{ |
||||
hex_bin[index] = (hex_sort << 4); |
||||
} |
||||
else |
||||
{ |
||||
hex_bin[index] |= hex_sort; |
||||
} |
||||
} |
||||
} |
||||
|
||||
return 1; |
||||
} |
||||
|
||||
int generation(char *parent_hex, int rand_offset, char rand_flip) |
||||
{ |
||||
int return_value = 0; |
||||
int parent_hex_len = strlen(parent_hex); |
||||
int parent_bin_len = parent_hex_len / 2; |
||||
|
||||
int prot = (PROT_READ | PROT_WRITE | PROT_EXEC); |
||||
int flags = (MAP_ANON | MAP_PRIVATE); |
||||
unsigned char *pic_buffer = mmap(NULL, parent_bin_len, prot, flags, -1, 0); |
||||
if (MAP_FAILED == pic_buffer) |
||||
{ |
||||
return_value = errno; |
||||
goto GEN_CLEANUP; |
||||
} |
||||
|
||||
hex_ascii_to_bin(parent_hex, parent_hex_len, pic_buffer); |
||||
|
||||
pic_buffer[rand_offset] = pic_buffer[rand_offset] ^ rand_flip; |
||||
|
||||
int (*reproduce_function)(unsigned char *, int) = reproduce; |
||||
void (*pic_function)(void *, int, void *) = pic_buffer; |
||||
|
||||
pic_function(pic_buffer, parent_bin_len, reproduce_function); |
||||
|
||||
return_value = 1; |
||||
GEN_CLEANUP: |
||||
if (NULL != pic_buffer) |
||||
{ |
||||
munmap(pic_buffer, parent_bin_len); |
||||
} |
||||
|
||||
return return_value; |
||||
} |
||||
|
||||
int main(int argc, const char **argv) |
||||
{ |
||||
if (3 > argc || argv[1] == NULL || argv[2] == NULL || argv[3] == NULL) |
||||
{ |
||||
exit(1); |
||||
} |
||||
|
||||
char *hex_string = argv[1]; |
||||
int rand_offset = atoi(argv[2]); |
||||
char rand_flip = atoi(argv[3]); |
||||
|
||||
int return_value = 0; |
||||
|
||||
return_value = generation(hex_string, rand_offset, rand_flip); |
||||
if (!return_value) |
||||
{ |
||||
exit(1); |
||||
} |
||||
|
||||
return 0; |
||||
} |
@ -1,193 +0,0 @@
|
||||
#include <errno.h> |
||||
#include <malloc.h> |
||||
#include <openssl/sha.h> |
||||
#include <stdio.h> |
||||
#include <stdlib.h> |
||||
#include <string.h> |
||||
#include <sys/mman.h> |
||||
#include <sys/stat.h> |
||||
#include <sys/types.h> |
||||
#include <sys/wait.h> |
||||
#include <time.h> |
||||
#include <unistd.h> |
||||
|
||||
#include "list.h" |
||||
|
||||
#pragma comment(lib, "openssl/sha.lib") |
||||
#define SHA_SUM_LENGTH (SHA_DIGEST_LENGTH + SHA_DIGEST_LENGTH + 1) |
||||
|
||||
int generation(char *seed_path); |
||||
int reproduce(void *pic_address, size_t pic_size); |
||||
|
||||
struct process_list |
||||
{ |
||||
struct list_head list; |
||||
pid_t process_id; |
||||
}; |
||||
|
||||
char seed_path[SHA_SUM_LENGTH]; |
||||
FILE *seed_handle = NULL; |
||||
void *pic_buffer = NULL; |
||||
|
||||
int main(int argc, const char **argv) |
||||
{ |
||||
struct process_list scraps; |
||||
struct process_list *iter = NULL; |
||||
pid_t process_id; |
||||
int status = 0; |
||||
int count = 0; |
||||
int return_value = 0; |
||||
|
||||
INIT_LIST_HEAD(&scraps.list); |
||||
|
||||
strncpy(seed_path, argv[1], SHA_SUM_LENGTH); |
||||
|
||||
while (1) |
||||
{ |
||||
process_id = fork(); |
||||
iter = (struct process_list *)malloc(sizeof(struct process_list)); |
||||
iter->process_id = process_id; |
||||
list_add_tail(&(iter->list), &(scraps.list)); |
||||
|
||||
if (process_id == 0) |
||||
{ |
||||
generation(seed_path); |
||||
break; |
||||
} |
||||
|
||||
if (process_id < 0) |
||||
{ |
||||
count = 0; |
||||
list_for_each_entry(iter, &scraps.list, list) |
||||
{ |
||||
count++; /* @todo clean up pids */ |
||||
return_value = waitpid(iter->process_id, &status, 0); |
||||
return_value = WEXITSTATUS(return_value); |
||||
} |
||||
printf("%d", count); |
||||
break; |
||||
} |
||||
} |
||||
|
||||
return count; |
||||
} |
||||
|
||||
int generation(char *seed_path) |
||||
{ |
||||
int return_value = 0; |
||||
long int mutation_value; |
||||
struct stat pic_statistics; |
||||
unsigned int mutation_offset = 0; |
||||
unsigned char pic_mutated = 0; |
||||
struct drand48_data drand_data; |
||||
|
||||
seed_handle = fopen(seed_path, "rb"); |
||||
if (NULL == seed_handle) |
||||
{ |
||||
return_value = errno; |
||||
goto GEN_CLEANUP; |
||||
} |
||||
|
||||
return_value = fstat(fileno(seed_handle), &pic_statistics); |
||||
if (-1 == return_value) |
||||
{ |
||||
return_value = errno; |
||||
goto GEN_CLEANUP; |
||||
} |
||||
|
||||
int prot = (PROT_READ | PROT_WRITE | PROT_EXEC); |
||||
int flags = (MAP_ANON | MAP_PRIVATE); |
||||
pic_buffer = mmap(NULL, pic_statistics.st_size, prot, flags, -1, 0); |
||||
if (MAP_FAILED == pic_buffer) |
||||
{ |
||||
return_value = errno; |
||||
goto GEN_CLEANUP; |
||||
} |
||||
|
||||
return_value = fread(pic_buffer, 1, pic_statistics.st_size, seed_handle); |
||||
if (return_value != pic_statistics.st_size) |
||||
{ |
||||
return_value = errno; |
||||
goto GEN_CLEANUP; |
||||
} |
||||
|
||||
if (NULL != seed_handle) |
||||
{ |
||||
fclose(seed_handle); |
||||
seed_handle = NULL; |
||||
} |
||||
|
||||
srand48_r(time(NULL), &drand_data); |
||||
lrand48_r(&drand_data, &mutation_value); |
||||
|
||||
mutation_offset = (mutation_value % (pic_statistics.st_size + 1)); |
||||
pic_mutated = |
||||
((unsigned char *)pic_buffer)[mutation_offset] & (mutation_value % 2); |
||||
((char *)pic_buffer)[mutation_offset] = pic_mutated; |
||||
|
||||
int (*reproduce_function)(void *, size_t) = reproduce; |
||||
void (*pic_function)(void *, size_t, void *) = pic_buffer; |
||||
|
||||
pic_function(pic_buffer, pic_statistics.st_size, reproduce_function); |
||||
|
||||
GEN_CLEANUP: |
||||
if (NULL != pic_buffer) |
||||
{ |
||||
munmap(pic_buffer, pic_statistics.st_size); |
||||
} |
||||
if (NULL != seed_handle) |
||||
{ |
||||
fclose(seed_handle); |
||||
seed_handle = NULL; |
||||
} |
||||
|
||||
return return_value; |
||||
} |
||||
|
||||
int reproduce(void *pic_address, size_t pic_size) |
||||
{ |
||||
int return_value = 0; |
||||
unsigned char digest[SHA_DIGEST_LENGTH]; |
||||
struct stat pic_statistics; |
||||
|
||||
memset(seed_path, 0, SHA_SUM_LENGTH); |
||||
SHA1((const unsigned char *)pic_address, pic_size, digest); |
||||
|
||||
for (int iter = 0; iter < SHA_DIGEST_LENGTH; iter++) |
||||
{ |
||||
sprintf(&seed_path[iter * 2], "%02x", digest[iter]); |
||||
} |
||||
|
||||
FILE *survived_store = fopen(seed_path, "w+"); |
||||
if (NULL == survived_store) |
||||
{ |
||||
return_value = errno; |
||||
goto CLONE_CLEANUP; |
||||
} |
||||
|
||||
return_value = fwrite(pic_address, 1, pic_size, survived_store); |
||||
if (return_value != pic_size) |
||||
{ |
||||
return_value = errno; |
||||
goto CLONE_CLEANUP; |
||||
} |
||||
|
||||
return_value = 1; |
||||
CLONE_CLEANUP: |
||||
if (survived_store) |
||||
{ |
||||
fclose(survived_store); |
||||
} |
||||
fstat(fileno(seed_handle), &pic_statistics); |
||||
if (NULL != pic_buffer) |
||||
{ |
||||
munmap(pic_buffer, pic_statistics.st_size); |
||||
} |
||||
if (NULL != seed_handle) |
||||
{ |
||||
fclose(seed_handle); |
||||
seed_handle = NULL; |
||||
} |
||||
|
||||
return generation(seed_path); |
||||
} |
Loading…
Reference in new issue