Moving towards a server/clients structure.

1 files changed, 28 insertions, 420 deletions

diff --git a/src/main.c b/src/main.c
index bb4ae23..d2fac57 100644
--- a/src/main.c
+++ b/src/main.c
@@ -1,436 +1,44 @@
-#include <stdlib.h>
 #include <stdio.h>
-#include <string.h>
-#include <time.h>
-#include <signal.h>
 
-#include "../tool/strings.h"
+#include "cli/parameters.h"
 
-#include "../io/error.h"
-#include "../io/parameters.h"
-#include "../io/data_input.h"
-#include "../io/data_output.h"
-#include "../io/network.h"
+#include "server/server.h"
 
-#include "knowledge.h"
+#include "pervasive.h"
 
-#include "state_types.h"
-
-static int run = 1;
-
-static void request_termination (int const signo)
-{
-   if ((signo == SIGINT) || (signo == SIGTERM))
-   {
-      run = 0;
-   }
-}
-
-static int initialize
-(
-   struct ZoO_state s [const static 1],
-   int const argc,
-   const char * argv [const static argc]
-)
-{
-   ZoO_S_DEBUG(ZoO_DEBUG_PROGRAM_FLOW, "Zero of One is initializing...");
-
-   srand(time(NULL));
-
-   /* prevents s [restrict] */
-   if (ZoO_knowledge_initialize(&(s->knowledge)) < 0)
-   {
-      return -1;
-   }
-
-   if (ZoO_parameters_initialize(&(s->param), argc, argv) < 1)
-   {
-      ZoO_knowledge_finalize(&(s->knowledge));
-
-      return -1;
-   }
-
-   return 0;
-}
-
-static int load_data_file (struct ZoO_state s [const static 1])
-{
-   struct ZoO_data_input input;
-   char * result;
-
-   if (ZoO_data_input_open(&input, s->param.data_filename) < 0)
-   {
-      return -1;
-   }
-
-   while
-   (
-      ZoO_data_input_read_line
-      (
-         &input,
-         ZoO_knowledge_punctuation_chars_count,
-         ZoO_knowledge_punctuation_chars
-      ) == 0
-   )
-   {
-      (void) ZoO_knowledge_assimilate
-      (
-         &(s->knowledge),
-         &(input.string),
-         s->param.aliases_count,
-         s->param.aliases
-      );
-   }
-
-   ZoO_data_input_close(&input);
-
-   return 0;
-}
-
-static int finalize (struct ZoO_state s [const static 1])
-{
-   int error;
-
-   ZoO_S_DEBUG(ZoO_DEBUG_PROGRAM_FLOW, "Zero of One is finalizing...");
-
-   error = 0;
-
-   /* prevents s [restrict] */
-   ZoO_knowledge_finalize(&(s->knowledge));
-
-   return error;
-}
-
-static int network_connect (struct ZoO_state s [const static 1])
-{
-   return
-      ZoO_network_connect
-      (
-         &(s->network),
-         s->param.irc_server_addr,
-         s->param.irc_server_port,
-         s->param.irc_server_channel,
-         s->param.irc_username,
-         s->param.irc_realname,
-         s->param.aliases[0]
-      );
-}
-
-static int should_reply
-(
-   struct ZoO_parameters param [const restrict static 1],
-   struct ZoO_strings string [const restrict static 1],
-   int should_learn [const restrict static 1]
-)
-{
-   ZoO_index i, j;
-
-   for (i = 0; i < param->aliases_count; ++i)
-   {
-      if (ZoO_IS_PREFIX(param->aliases[i], string->words[0]))
-      {
-         *should_learn = 0;
-
-         return 1;
-      }
-
-      for (j = 1; j < string->words_count; ++j)
-      {
-         if (ZoO_IS_PREFIX(param->aliases[i], string->words[j]))
-         {
-            *should_learn = 1;
-
-            return 1;
-         }
-      }
-   }
-
-   *should_learn = 1;
-
-   return (param->reply_rate >= (rand() % 100));
-}
-
-static void handle_user_join
-(
-   struct ZoO_state s [const static 1],
-   struct ZoO_strings string [const restrict static 1],
-   ssize_t const msg_offset,
-   ssize_t const msg_size
-)
-{
-   ZoO_char * line;
-   ZoO_index loc;
-
-   if (s->param.reply_rate < (rand() % 100))
-   {
-      return;
-   }
-
-   if
-   (
-      ZoO_strings_parse
-      (
-         string,
-         (size_t) msg_size,
-         (s->network.in + msg_offset),
-         ZoO_knowledge_punctuation_chars_count,
-         ZoO_knowledge_punctuation_chars
-      ) < 0
-   )
-   {
-      ZoO_S_DEBUG(ZoO_DEBUG_PROGRAM_FLOW, "Could not dissect join username.");
-
-      return;
-   }
-
-   if
-   (
-      (
-      ZoO_knowledge_find
-         (
-            &(s->knowledge),
-            string->words[0],
-            &loc
-         ) < 0
-      )
-      || (s->knowledge.words[loc].backward_links_count <= 3)
-      || (s->knowledge.words[loc].forward_links_count <= 3)
-   )
-   {
-      if
-      (
-         ZoO_knowledge_extend
-         (
-            &(s->knowledge),
-            (struct ZoO_strings *) NULL,
-            0,
-            (const char **) NULL,
-            &line
-         ) == 0
-      )
-      {
-         if (line[0] == ' ')
-         {
-            strcpy((s->network.out), (line + 1));
-         }
-         else
-         {
-            strcpy((s->network.out), line);
-         }
-
-         free((void *) line);
-
-         ZoO_network_send(&(s->network));
-      }
-   }
-   else
-   {
-      if
-      (
-         ZoO_knowledge_extend
-         (
-            &(s->knowledge),
-            string,
-            0,
-            (const char **) NULL,
-            &line
-         ) == 0
-      )
-      {
-         if (line[0] == ' ')
-         {
-            strcpy((s->network.out), (line + 1));
-         }
-         else
-         {
-            strcpy((s->network.out), line);
-         }
-
-         free((void *) line);
-
-         ZoO_network_send(&(s->network));
-      }
-   }
-}
-
-static void handle_message
-(
-   struct ZoO_state s [const static 1],
-   struct ZoO_strings string [const restrict static 1],
-   ssize_t const msg_offset,
-   /* FIXME: somehow we end up using (msg_size + 1), meaning there's a mixup
-    *        between size and length.
-    */
-   ssize_t const msg_size
-)
-{
-   ZoO_char * line;
-   int reply, learn;
-
-   if
-   (
-      ZoO_strings_parse
-      (
-         string,
-         (size_t) msg_size,
-         (s->network.in + msg_offset),
-         ZoO_knowledge_punctuation_chars_count,
-         ZoO_knowledge_punctuation_chars
-      ) < 0
-   )
-   {
-      ZoO_S_DEBUG(ZoO_DEBUG_PROGRAM_FLOW, "Could not dissect msg.");
-
-      return;
-   }
-
-   if (string->words_count == 0)
-   {
-      return;
-   }
-
-   reply = should_reply(&(s->param), string, &learn);
-
-   if (learn)
-   {
-      /*
-       * It would be best to do that after replying, but by then we no longer
-       * have the string in 's->network.in'.
-       */
-      (void) ZoO_data_output_write_line
-      (
-         s->param.new_data_filename,
-         (s->network.in + msg_offset),
-         (size_t) (msg_size + 1)
-      );
-   }
-
-   if
-   (
-      reply
-      &&
-      (
-         ZoO_knowledge_extend
-         (
-            &(s->knowledge),
-            string,
-            s->param.aliases_count,
-            s->param.aliases,
-            &line
-         ) == 0
-      )
-   )
-   {
-      if (line[0] == ' ')
-      {
-         strcpy((s->network.out), (line + 1));
-      }
-      else
-      {
-         strcpy((s->network.out), line);
-      }
-
-      free((void *) line);
-
-      ZoO_network_send(&(s->network));
-   }
-
-   if (learn)
-   {
-      (void) ZoO_knowledge_assimilate
-      (
-         &(s->knowledge),
-         string,
-         s->param.aliases_count,
-         s->param.aliases
-      );
-   }
-}
-
-static int main_loop (struct ZoO_state s [const static 1])
+static void print_help ()
 {
-   struct ZoO_strings string;
-   ssize_t msg_offset, msg_size;
-   enum ZoO_msg_type msg_type;
-
-   msg_offset = 0;
-   msg_size = 0;
-
-   ZoO_strings_initialize(&string);
-
-   while (run)
-   {
-      if
-      (
-         ZoO_network_receive
-         (
-            &(s->network),
-            &msg_offset,
-            &msg_size,
-            &msg_type
-         ) == 0
-      )
-      {
-         switch (msg_type)
-         {
-            case ZoO_JOIN:
-               handle_user_join(s, &string, msg_offset, msg_size);
-               break;
-
-            case ZoO_PRIVMSG:
-               handle_message(s, &string, msg_offset, msg_size);
-               break;
-         }
-      }
-   }
-
-   ZoO_strings_finalize(&string);
-
-   ZoO_network_disconnect(&(s->network));
-
-   return 0;
+	printf
+	(
+		"Zero of One - server version %d - protocol version %d\n"
+		"\nUsages:\n"
+		"   SERVER:\tzero_of_one_server SESSION_NAME MARKOV_ORDER STORAGE_FILE\n"
+		"   CLEAN UP:\tzero_of_one_server -c SESSION_NAME\n"
+		"   SHOW HELP:\tAnything else\n"
+		"\nParameters:\n"
+		"   SESSION_NAME: valid POSIX message queue filename.\n"
+		"   MARKOV_ORDER: non-null positive integer.\n"
+		"   STORAGE_FILE: file in which the knowledge will be stored.",
+      ZoO_SERVER_VERSION,
+      ZoO_PROTOCOL_VERSION
+	);
 }
 
 int main (int const argc, const char * argv [const static argc])
 {
-   struct ZoO_state s;
-
-   if (initialize(&s, argc, argv) < 0)
-   {
-      return -1;
-   }
-
-   if (load_data_file(&s) < 0)
-   {
-      goto CRASH;
-   }
-
-   if (network_connect(&s) < 0)
-   {
-      goto CRASH;
-   }
-
-   if (main_loop(&s) < 0)
-   {
-      goto CRASH;
-   }
-
-   (void) finalize(&s);
-
-   ZoO_S_DEBUG(ZoO_DEBUG_PROGRAM_FLOW, "Zero of One terminated normally.");
-
-   return 0;
+   struct ZoO_parameters params;
 
-   CRASH:
+   switch (ZoO_parameters_initialize(&params, argc, argv))
    {
-      (void) finalize(&s);
+      case ZoO_CLEANS_UP:
+         return ZoO_server_cleanup_session(params.session);
 
-      ZoO_S_DEBUG
-      (
-         ZoO_DEBUG_PROGRAM_FLOW,
-         "Zero of One terminated by crashing."
-      );
+      case ZoO_RUNS:
+         return ZoO_server_main(params);
 
-      return -1;
+      default:
+      case ZoO_PRINTS_HELP:
+         print_help();
+         return 0;
    }
 }