247 lines
8.8 KiB
C
247 lines
8.8 KiB
C
#include <stdio.h>
|
||
#include <stdlib.h>
|
||
#include <string.h>
|
||
#include <threads.h>
|
||
#include <unistd.h>
|
||
|
||
#include "chunk.h"
|
||
#include "config.h"
|
||
#include "data.h"
|
||
#include "file.h"
|
||
#include "log.h"
|
||
#include "multiprocessing.h"
|
||
#include "queue.h"
|
||
#include "scanner.h"
|
||
#include "socket.h"
|
||
#include "utils.h"
|
||
#include <dirent.h>
|
||
|
||
int send_chunk(Client *client, Chunk *chunk, Config *config) {
|
||
if (config->use_chunk_serialization) {
|
||
send_status(client->file_descriptor, STATUS_CHUNK);
|
||
Data *data;
|
||
if (config->use_compression) {
|
||
data = chunk_compress(chunk, config->compression_level);
|
||
} else {
|
||
for (int i = 0; i < chunk->element_count; i++)
|
||
file_load_data(chunk->items[i]);
|
||
data = chunk_serialize(chunk);
|
||
}
|
||
send_data(client->file_descriptor, data->data, data->size);
|
||
data_destroy(data);
|
||
} else if (config->use_sendfile && !config->use_compression) {
|
||
for (int i = 0; i < chunk->element_count; i++) {
|
||
send_status(client->file_descriptor, STATUS_NEXT);
|
||
file_send_sendfile(chunk->items[i], client->file_descriptor);
|
||
}
|
||
} else {
|
||
for (int i = 0; i < chunk->element_count; i++) {
|
||
send_status(client->file_descriptor, STATUS_NEXT);
|
||
File *file = chunk->items[i];
|
||
if (config->use_compression) {
|
||
Data *compressed_data =
|
||
data_compress(file->data, config->compression_level);
|
||
data_destroy(file->data);
|
||
file->data = compressed_data;
|
||
}
|
||
file_send_single_calls(file, client->file_descriptor);
|
||
}
|
||
}
|
||
return 0;
|
||
}
|
||
|
||
int scan_directory_multithreaded(void *pipeline_context) {
|
||
PipelineContextSender *context = (PipelineContextSender *)pipeline_context;
|
||
mtx_lock(&context->mutex_scanner);
|
||
DirectoryScanner *scanner =
|
||
directory_scanner_create(context->config->send_directory);
|
||
mtx_unlock(&context->mutex_scanner);
|
||
|
||
Chunk *current_chunk;
|
||
while ((current_chunk = directory_scanner_next(scanner)) != NULL)
|
||
queue_enqueue_multithreaded(context->queue_scanner, current_chunk,
|
||
&context->mutex_scanner,
|
||
&context->condition_not_empty_scanner,
|
||
&context->condition_not_full_scanner);
|
||
mtx_lock(&context->mutex_scanner);
|
||
context->scanner_done = true;
|
||
cnd_signal(&context->condition_not_empty_scanner);
|
||
mtx_unlock(&context->mutex_scanner);
|
||
|
||
directory_scanner_destroy(scanner);
|
||
return thrd_success;
|
||
}
|
||
|
||
int load_files_multithreaded(void *pipeline_context) {
|
||
PipelineContextSender *context = (PipelineContextSender *)pipeline_context;
|
||
while (true) {
|
||
Chunk *chunk = queue_dequeue_multithreaded(
|
||
context->queue_scanner, &context->mutex_scanner,
|
||
&context->condition_not_empty_scanner,
|
||
&context->condition_not_full_scanner, &context->scanner_done);
|
||
if (chunk == NULL) {
|
||
mtx_lock(&context->mutex_loader);
|
||
context->loader_done = true;
|
||
cnd_signal(&context->condition_not_empty_loader);
|
||
mtx_unlock(&context->mutex_loader);
|
||
return thrd_success;
|
||
}
|
||
if (!context->config->use_sendfile) {
|
||
for (int i = 0; i < chunk->element_count; i++)
|
||
file_load_data(chunk->items[i]);
|
||
}
|
||
queue_enqueue_multithreaded(context->queue_loader, chunk,
|
||
&context->mutex_loader,
|
||
&context->condition_not_empty_loader,
|
||
&context->condition_not_full_loader);
|
||
}
|
||
}
|
||
|
||
int send_chunks_multithreaded(void *pipeline_context) {
|
||
PipelineContextSender *context = (PipelineContextSender *)pipeline_context;
|
||
Client *client = client_create();
|
||
const char *env_ip = getenv("FASTSYNC_SERVER_IP");
|
||
const char *ip = env_ip ? env_ip : "127.0.0.1";
|
||
const char *env_port = getenv("FASTSYNC_SERVER_PORT");
|
||
int port = env_port ? atoi(env_port) : 8080;
|
||
client_connect(client, (char *)ip, port);
|
||
config_send(client->file_descriptor, context->config);
|
||
|
||
while (true) {
|
||
Chunk *current_chunk = queue_dequeue_multithreaded(
|
||
context->queue_loader, &context->mutex_loader,
|
||
&context->condition_not_empty_loader,
|
||
&context->condition_not_full_loader, &context->loader_done);
|
||
if (current_chunk == NULL) {
|
||
send_status(client->file_descriptor, STATUS_FINISHED);
|
||
client_disconnect(client);
|
||
client_delete(client);
|
||
return thrd_success;
|
||
}
|
||
if (send_chunk(client, current_chunk, context->config) != 0) {
|
||
perror("Something unexpected happend while sending the chunk");
|
||
exit(EXIT_FAILURE);
|
||
}
|
||
chunk_destroy(current_chunk);
|
||
}
|
||
}
|
||
|
||
int send_files(Config *config) {
|
||
Client *client = client_create();
|
||
const char *env_ip = getenv("FASTSYNC_SERVER_IP");
|
||
const char *ip = env_ip ? env_ip : "127.0.0.1";
|
||
const char *env_port = getenv("FASTSYNC_SERVER_PORT");
|
||
int port = env_port ? atoi(env_port) : 8080;
|
||
client_connect(client, (char *)ip, port);
|
||
config_send(client->file_descriptor, config);
|
||
DirectoryScanner *scanner = directory_scanner_create(config->send_directory);
|
||
Chunk *current_chunk;
|
||
while ((current_chunk = directory_scanner_next(scanner)) != NULL) {
|
||
if (!config->use_sendfile) {
|
||
for (int i = 0; i < current_chunk->element_count; i++)
|
||
file_load_data(current_chunk->items[i]);
|
||
}
|
||
send_chunk(client, current_chunk, config);
|
||
chunk_destroy(current_chunk);
|
||
}
|
||
send_status(client->file_descriptor, STATUS_FINISHED);
|
||
if (receive_status(client->file_descriptor) != STATUS_OK)
|
||
return -1;
|
||
directory_scanner_destroy(scanner);
|
||
client_disconnect(client);
|
||
client_delete(client);
|
||
return 0;
|
||
}
|
||
|
||
int send_files_multithreaded(Config *config) {
|
||
PipelineContextSender *context =
|
||
pipeline_context_sender_create(config, queue_create(100, chunk_destroy),
|
||
queue_create(100, chunk_destroy));
|
||
|
||
thrd_t scanner, loader, sender;
|
||
if (thrd_create(&scanner, scan_directory_multithreaded, context) !=
|
||
thrd_success ||
|
||
thrd_create(&loader, load_files_multithreaded, context) != thrd_success ||
|
||
thrd_create(&sender, send_chunks_multithreaded, context) !=
|
||
thrd_success) {
|
||
perror("Error creating threads.\n");
|
||
return 1;
|
||
}
|
||
|
||
thrd_join(scanner, NULL);
|
||
thrd_join(loader, NULL);
|
||
thrd_join(sender, NULL);
|
||
|
||
pipeline_context_sender_destroy(context);
|
||
return 0;
|
||
}
|
||
|
||
void handle_arg(char *argument_given, char *argument_to_set, bool *result,
|
||
char *message) {
|
||
if (strcmp(argument_given, argument_to_set) == 0) {
|
||
*result = true;
|
||
log_message(LOG_LEVEL_INFO, message);
|
||
}
|
||
}
|
||
int main(int argc, char *argv[]) {
|
||
const char *env_source = getenv("FASTSYNC_SOURCE_DIR");
|
||
const char *env_dest = getenv("FASTSYNC_DEST_DIR");
|
||
const char *env_save = getenv("FASTSYNC_SAVE_TO_DISK");
|
||
|
||
char *source_dir =
|
||
env_source ? str_dup((char *)env_source)
|
||
: str_dup("/home/taptap/Nextcloud/Uni/moodle/B. Schnor： "
|
||
"Konzepte Paralleler Programmierung, SoSe 2026");
|
||
char *dest_dir =
|
||
env_dest ? str_dup((char *)env_dest) : str_dup("./data_copied");
|
||
bool save_to_disk = false;
|
||
if (env_save &&
|
||
(strcmp(env_save, "true") == 0 || strcmp(env_save, "1") == 0)) {
|
||
save_to_disk = true;
|
||
}
|
||
|
||
Config *config = config_create(str_dup("1.0.0"), source_dir, dest_dir,
|
||
save_to_disk, false, false, false, 5, 20, false);
|
||
for (int i = 1; i < argc; i++) {
|
||
if (strcmp(argv[i], "-c") == 0) {
|
||
config->use_compression = true;
|
||
log_message(LOG_LEVEL_INFO, "Enabled Compression");
|
||
|
||
if (i + 1 < argc) {
|
||
char *end_ptr;
|
||
int level = strtol(argv[i + 1], &end_ptr, 10);
|
||
if (*end_ptr == '\0') {
|
||
config->compression_level = level;
|
||
log_message(LOG_LEVEL_INFO, "Set Compression level to %d",
|
||
config->compression_level);
|
||
}
|
||
}
|
||
} else if (strcmp(argv[i], "--source-dir") == 0 && i + 1 < argc) {
|
||
free(config->send_directory);
|
||
config->send_directory = str_dup(argv[++i]);
|
||
} else if (strcmp(argv[i], "--dest-dir") == 0 && i + 1 < argc) {
|
||
free(config->receive_root_directory);
|
||
config->receive_root_directory = str_dup(argv[++i]);
|
||
} else if (strcmp(argv[i], "--save-to-disk") == 0) {
|
||
config->save_to_disk = true;
|
||
} else if (strcmp(argv[i], "-f") == 0 || strcmp(argv[i], "--sendfile") == 0) {
|
||
config->use_sendfile = true;
|
||
log_message(LOG_LEVEL_INFO, "Enabled sendfile");
|
||
} else {
|
||
handle_arg(argv[i], "-m", &config->use_multithreading,
|
||
"Enabled Multithreading");
|
||
handle_arg(argv[i], "-s", &config->use_chunk_serialization,
|
||
"Enabled Chunk Serialization");
|
||
}
|
||
}
|
||
|
||
if (config->use_sendfile && (config->use_chunk_serialization || config->use_compression)) {
|
||
fprintf(stderr, "Error: -f/--sendfile cannot be combined with -c (compression) or -s (chunk serialization)\n");
|
||
return 1;
|
||
}
|
||
|
||
if (config->use_multithreading)
|
||
return send_files_multithreaded(config);
|
||
return send_files(config);
|
||
}
|