First Commit

src/server/server.c

@@ -0,0 +1,104 @@
+#include "chunk.h"
+#include "config.h"
+#include "multiprocessing.h"
+#include "queue.h"
+#include "socket.h"
+#include "unistd.h"
+#include "utils.h"
+#include <stdio.h>
+#include <stdlib.h>
+#include <threads.h>
+
+FileReceive *receive_file_receive(int file_descriptor) {
+  char *path = (char *)receive_str(file_descriptor);
+  printf("%s\n", path);
+  DataFragment *file_data_fragment = receive_data(file_descriptor);
+  FileReceive *file = file_receive_create(path, file_data_fragment);
+  return file;
+}
+
+int receive_thread(void *pipeline_context) {
+  PipelineContextReceiver *context =
+      (PipelineContextReceiver *)pipeline_context;
+  mtx_lock(&context->mutex);
+  int file_descriptor = context->file_descriptor;
+  mtx_unlock(&context->mutex);
+
+  while (receive_status(file_descriptor) == NEXT) {
+    FileReceive *file = receive_file_receive(file_descriptor);
+    queue_enqueue_multithreaded(context->queue, file, &context->mutex,
+                                &context->condition_not_empty,
+                                &context->condition_not_full);
+  }
+  mtx_lock(&context->mutex);
+  context->receiver_done = true;
+  cnd_signal(&context->condition_not_empty);
+  mtx_unlock(&context->mutex);
+  return thrd_success;
+}
+
+int write_thread(void *pipeline_context) {
+  PipelineContextReceiver *context =
+      (PipelineContextReceiver *)pipeline_context;
+  mtx_lock(&context->mutex);
+  bool save_to_disk = context->config->save_to_disk;
+  char *root_directory = str_dup(context->config->receive_root_directory);
+  mtx_unlock(&context->mutex);
+
+  while (true) {
+    FileReceive *file = queue_dequeue_multithreaded(
+        context->queue, &context->mutex, &context->condition_not_empty,
+        &context->condition_not_full, &context->receiver_done);
+    if (file == NULL) {
+      free(root_directory);
+      return thrd_success;
+    }
+    if (save_to_disk)
+      to_disk(path_cat(root_directory, file->path), file->data_fragment->data,
+              file->data_fragment->size);
+  }
+}
+
+int receive_files(Config *config, int file_descriptor) {
+  Status status = receive_status(file_descriptor);
+  while (status == NEXT) {
+    FileReceive *file = receive_file_receive(file_descriptor);
+    if (config->save_to_disk)
+      to_disk(path_cat(config->receive_root_directory, file->path),
+              file->data_fragment->data, file->data_fragment->size);
+    file_receive_destroy(file);
+    status = receive_status(file_descriptor);
+  }
+  if (status != FINISHED) {
+    send_status(file_descriptor, ERROR);
+    return -1;
+  }
+  send_status(file_descriptor, OK);
+  return 0;
+}
+
+void handler(int file_descriptor) {
+  Config *config = config_receive(file_descriptor);
+  if (config->use_multithreading) {
+    PipelineContextReceiver *context = pipeline_context_receiver_create(
+        config, queue_create(100, file_receive_destroy), file_descriptor);
+    thrd_t receiver, writer;
+    if (thrd_create(&receiver, receive_thread, context) != thrd_success ||
+        thrd_create(&writer, write_thread, context) != thrd_success) {
+      perror("Error creating Threads!");
+      exit(EXIT_FAILURE);
+    }
+    thrd_join(receiver, NULL);
+    thrd_join(writer, NULL);
+    pipeline_context_receiver_destroy(context);
+  } else
+    receive_files(config, file_descriptor);
+  close(file_descriptor);
+}
+
+int main() {
+  Server *server = server_create(8080);
+  server_listen(server, handler);
+  server_delete(server);
+  return 0;
+}