Server.h

/*
    Copyright (c) 2014 by Elvis Angelaccio

    Permission is hereby granted, free of charge, to any person obtaining a copy
    of this software and associated documentation files (the "Software"), to deal
    in the Software without restriction, including without limitation the rights
    to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
    copies of the Software, and to permit persons to whom the Software is
    furnished to do so, subject to the following conditions:

    The above copyright notice and this permission notice shall be included in
    all copies or substantial portions of the Software.

    THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
    IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
    FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
    AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
    LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
    OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
    THE SOFTWARE.
*/

#ifndef SERVER_H
#define SERVER_H

#include <csignal>

#include "SystemHandler.h"
#include "sync.h"
#include "../net/ServerHandler.h"
#include "../net/server_protocol.h"


namespace server
{   
    typedef enum {DEFAULT, RUNNING, STOPPED} ServerState;   
    extern sig_atomic_t currentServerState;             
    const std::string DEFAULT_DATA_DIR = "database";    
    const std::string DEFAULT_CONF_DIR = "conf";        
    const std::string VERSION = "1.0.0";

    struct NetworkThreadArgs;

    class Server
    {
                
    public:
        
        Server();
        ~Server();

        bool checkArguments(int argc, char **argv);
        
        void start();

    private:
        
        static const net::Port DEFAULT_PORT = 4444;          
        static const int POSIX_DSS_THREAD_TIMER = 30;       
        static const int WIN32_DSS_THREAD_TIMER = 30000;   
        std::string dataDirectory;      
        std::string confDirectory;      
        net::Port port;                 
        int connectionCount;            
        Mutex mutex;                    
        RWLock lock;                    
        /* System Handlers */
        net::ServerHandler *serverHandler;              
        SystemHandler systemHandler;                    
        /* System core functions */
        
        void setSignalHandler();
        
        void createDssThread();
        
        void createNetworkThread(int connectedSocket);
            
        
        void dssThread();

        void networkThread(int connectedSocket, net::SocketState& threadState);
        

        void doInsertCommand(int connectedSocket, net::SocketState& threadState, const net::InsertRequest& insertRequest);
        
        
        bool fillSdFields(const net::InsertRequest& insertRequest, entity::DataSet& dataSet);
        
        
        bool receiveFiles(int connectedSocket, net::SocketState& threadState, const net::InsertRequest& insertRequest, entity::DataSet& dataSet);
        
        bool processFiles(const net::InsertRequest& insertRequest, const entity::DataSet& dataSet);


        void doRemoveCommand(int connectedSocket, net::SocketState& threadState, const net::DataSetRequest& dataSetRequest);

        void doGetCommand(int connectedSocket, net::SocketState& threadState, const net::DataSetRequest& dataSetRequest);
        
        void doSearchCommand(int connectedSocket, net::SocketState& threadState, const net::SearchRequest& searchRequest);


        void doSpeclistCommand(int connectedSocket, net::SocketState& threadState);

        void shutdown();
        
        void printUsage(const char *program);
        
        void printInitialMessage();
        
        static Server *getContext();

        /* Platform-specific declarations for the threads starting functions and for the signal handling functions. */
#if (defined(_WIN32) || defined(__WIN32__) || defined(WIN32))
        
        static unsigned int __stdcall win32DssThread(void *args);       // __stdcall rqeuired by _beginthreadex() 
        static unsigned int __stdcall win32NetworkThread(void *args);   
        static BOOL WINAPI consoleHandler(DWORD signal);

        std::map<unsigned int, HANDLE>  networkThreadHandles;           
        std::map<unsigned int, NetworkThreadArgs*> networkThreadArgs;   
        HANDLE dssThreadHandle;                                         
        Event dssThreadEvent;                                           
#else
        static void *posixDssThread(void *args);
        static void *posixNetworkThread(void *args);
        static void signalHandler(int signal);
        
        std::map<pthread_t, NetworkThreadArgs*> networkThreads;
        pthread_t dssThreadId;
        
        CondVarBundle dssThreadCond;        
#endif

    };

    struct DssThreadArgs
    {
        Server *server;     
    };

    struct NetworkThreadArgs
    {
        int connectedSocket;            
        net::SocketState threadState;   
        Server *server;                 
    };

}




#endif