src/Server/main.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2008 by Sverre Rabbelier                                *
   3  *   sverre@rabbelier.nl                                                   *
   4  *                                                                         *
   5  *   This program is free software; you can redistribute it and/or modify  *
   6  *   it under the terms of the GNU General Public License as published by  *
   7  *   the Free Software Foundation; either version 3 of the License, or     *
   8  *   (at your option) any later version.                                   *
   9  *                                                                         *
  10  *   This program is distributed in the hope that it will be useful,       *
  11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  13  *   GNU General Public License for more details.                          *
  14  *                                                                         *
  15  *   You should have received a copy of the GNU General Public License     *
  16  *   along with this program; if not, write to the                         *
  17  *   Free Software Foundation, Inc.,                                       *
  18  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
  19  ***************************************************************************/
  20
  21 #include <iostream>
  22 #include <fstream>
  23
  24 #ifdef _WIN32
  25 #include <winsock2.h>
  26 #endif
  27
  28 #include "FieldImpls.h"
  29 #include "GameVersion.h"
  30 #include "SavableHeaders.h"
  31 #include "TableImpls.h"
  32 #include "DatabaseMgr.h"
  33 #include "Global.h"
  34 #include "ListenSocket.h"
  35 #include "SQLSocket.h"
  36 #include "UBHandler.h"
  37 #include "UBSocket.h"
  38 #include "SqliteMgr.h"
  39 #include "Managers.h"
  40 #include "StringUtilities.h"
  41
  42 void exitfunc()
  43 {
  44         printf("exiting...\n");
  45         std::cin.get();
  46         return;
  47 }
  48
  49 std::string add(SocketHandler* h,int port)
  50 {
  51         ListenSocket<UBSocket> *l = new ListenSocket<UBSocket>(*h);
  52
  53         std::string msg = "Attempting bind on port";
  54         msg.append(String::Get()->fromInt(port));
  55         msg.append("... ");
  56
  57         if (l -> Bind(port))
  58         {
  59                 msg.append("Not successful\n");
  60                 delete l;
  61         }
  62         else
  63         {
  64                 msg.append("OK\n");
  65                 l -> SetDeleteByHandler();
  66                 h->Add(l);
  67         }
  68
  69         return msg;
  70 }
  71
  72 std::string addSQL(SocketHandler* h, int port)
  73 {
  74         ListenSocket<SQLSocket> *l = new ListenSocket<SQLSocket>(*h);
  75
  76         std::string msg = "Attempting SQL bind on port ";
  77         msg.append(String::Get()->fromInt(port));
  78         msg.append("... ");
  79
  80
  81         if (l -> Bind(port))
  82         {
  83                 msg.append("Not successful\n");
  84                 delete l;
  85         }
  86         else
  87         {
  88                 msg.append("OK\n");
  89                 l -> SetDeleteByHandler();
  90                 h->Add(l);
  91         }
  92
  93         return msg;
  94 }
  95
  96 bool g_quit = false;
  97 extern bool g_shutdown;
  98 extern bool g_nocatch;
  99 extern bool g_printsql;
 100 extern bool g_printstatus;
 101 extern bool g_printbugs;
 102 extern bool g_printlogs;
 103
 104 static void logToFile(Global* global, const std::string& text)
 105 {
 106         std::string output = text;
 107         output.append("\n");
 108
 109         std::ofstream errorfile;
 110         errorfile.open("errorlog.txt", std::ios_base::app);
 111
 112         if(!errorfile.is_open())
 113         {
 114                 global->printexception("Could not open errorlog file.\n");
 115                 return;
 116         }
 117
 118         errorfile.write(output.c_str(), output.size());
 119         if(errorfile.bad())
 120         {
 121                 global->printexception("Could not write to errorlog file.\n");
 122                 return;
 123         }
 124
 125         return;
 126 }
 127
 128 static void parseArgs(int argc, char** argv)
 129 {
 130         for(int i = 1; i < argc; i++)
 131         {
 132                 if(!strncmp("-n", argv[i], 2) || !strncmp("--no-catch", argv[i], 10))
 133                 {
 134                         printf("[MODE  ] Not catching exceptions.\n");
 135                         g_nocatch = true;
 136                 }
 137
 138                 if(!strncmp("-s", argv[i], 2) || !strncmp("--sql", argv[i], 5))
 139                 {
 140                         printf("[MODE  ] Printing sql.\n");
 141                         g_printsql = true;
 142                 }
 143
 144                 if(!strncmp("-q", argv[i], 2) || !strncmp("--quiet", argv[i], 5))
 145                 {
 146                         printf("[MODE  ] Not printing status and bug msgs.\n");
 147                         g_printstatus = false;
 148                         g_printbugs = false;
 149                 }
 150
 151                 if(!strncmp("-b", argv[i], 2) || !strncmp("--bugs", argv[i], 5))
 152                 {
 153                         printf("[MODE  ] Printing bug msgs.\n");
 154                         g_printbugs = true;
 155                 }
 156
 157                 if(!strncmp("-l", argv[i], 2) || !strncmp("--log", argv[i], 5))
 158                 {
 159                         printf("[MODE  ] Printing log msgs.\n");
 160                         g_printlogs = true;
 161                 }
 162         }
 163 }
 164
 165 int main(int argc, char** argv)
 166 {
 167         if(argc > 1)
 168         {
 169                 parseArgs(argc, argv);
 170         }
 171
 172         Global* global = Global::Get();
 173
 174         global->printstatus("Opening database...\n");
 175         std::string dbname = game::vname;
 176         dbname.append(".db");
 177         DatabaseMgr::Initialize(dbname);
 178         global->printstatus("Database opened.\n");
 179
 180         global->printstatus("Initializing in-memory table definitions...\n");
 181         db::TableImpls::Get()->Initialize();
 182         global->printstatus("In-memory table definitions initialized.\n");
 183
 184         global->printstatus("Binding ports...\n");
 185         std::string portstatus;
 186
 187         portstatus = add(UBHandler::Get(), 4000);
 188         global->printstatus(portstatus);
 189
 190         portstatus = add(UBHandler::Get(), 4040);
 191         global->printstatus(portstatus);
 192
 193         portstatus = add(UBHandler::Get(), 5060);
 194         global->printstatus(portstatus);
 195
 196         portstatus = addSQL(UBHandler::Get(), 9090);
 197         global->printstatus(portstatus);
 198
 199         global->printstatus("Ports bound.\n");
 200         global->printstatus("Running!\n");
 201
 202         while (!g_quit)
 203         {
 204                 if(g_nocatch)
 205                 {
 206                         UBHandler::Get()->Select(0, 200000);
 207                         UBHandler::Get()->SwitchEditors();
 208                 }
 209                 else
 210                 {
 211                         try {
 212                                 UBHandler::Get()->Select(0, 200000);
 213                                 UBHandler::Get()->SwitchEditors();
 214                         } catch(std::exception& e) {
 215                                 global->printexception("While in main loop, trying to quit gracefully", e.what());
 216                                 logToFile(global, e.what());
 217                                 g_quit = true;
 218                         }
 219                 }
 220         }
 221
 222         try {
 223                 UBHandler::Get()->Shutdown();
 224                 UBHandler::Get()->Select();
 225         } catch(std::exception& e) {
 226                 global->printexception("While closing sockets, trying to quit gracefully.", e.what());
 227                 logToFile(global, e.what());
 228         }
 229
 230         try {
 231                 UBHandler::Free();
 232         } catch(std::exception& e) {
 233                 global->printexception("While freeing the UBHandler, trying to quit gracefully.", e.what());
 234                 logToFile(global, e.what());
 235         }
 236
 237         g_shutdown = true;
 238
 239         try {
 240                 mud::Managers::Free();
 241         } catch(std::exception& e) {
 242                 global->printexception("While freeing mud::Managers, trying to quit gracefully.", e.what());
 243                 logToFile(global, e.what());
 244         }
 245
 246         try {
 247                 SqliteMgr::Free();
 248         } catch(std::exception& e) {
 249                 global->printexception("While freeing the SqliteMgr, trying to quit gracefully.", e.what());
 250                 logToFile(global, e.what());
 251         }
 252
 253         try {
 254                 DatabaseMgr::Free();
 255         } catch(std::exception& e) {
 256                 global->printexception("While freeing the DatabaseMgr, trying to quit gracefully.", e.what());
 257                 logToFile(global, e.what());
 258         }
 259
 260         global->printstatus("End of program.\n");
 261         exitfunc();
 262         return 0;
 263 }