X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/4271ab433cd55bbd2612292bcf39e4dc3d7274f1..0907aa4037b12b6b88ee24495d4577a064d4f8db:/rscriptd/listener.h diff --git a/rscriptd/listener.h b/rscriptd/listener.h deleted file mode 100644 index 7fb71ee3..00000000 --- a/rscriptd/listener.h +++ /dev/null @@ -1,144 +0,0 @@ -/* - * This program is free software; you can redistribute it and/or modify - * it under the terms of the GNU General Public License as published by - * the Free Software Foundation; either version 2 of the License, or - * (at your option) any later version. - * - * This program is distributed in the hope that it will be useful, - * but WITHOUT ANY WARRANTY; without even the implied warranty of - * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - * GNU General Public License for more details. - * - * You should have received a copy of the GNU General Public License - * along with this program; if not, write to the Free Software - * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA - */ - -/* - * Author : Boris Mikhailenko - * Author : Maxim Mamontov - */ - -#include - -#include -#include -#include -#include -#include - -#include "stg/blowfish.h" -#include "stg/rs_packets.h" -#include "stg/logger.h" - -struct UserData -{ - std::string params; - std::string login; - uint32_t ip; - uint32_t id; -}; - -struct PendingData : public UserData -{ - enum {CONNECT, ALIVE, DISCONNECT} type; -}; - -struct AliveData : public UserData -{ - explicit AliveData(const UserData & data) - : UserData(data), - lastAlive(time(NULL)) - {}; - bool operator<(const std::string & rvalue) const { return login < rvalue; }; - time_t lastAlive; -}; - -class IsNotTimedOut : public std::unary_function { - public: - explicit IsNotTimedOut(double to) : timeout(to), now(time(NULL)) {} - bool operator()(const AliveData & data) const - { - return difftime(now, data.lastAlive) < timeout; - } - private: - double timeout; - time_t now; -}; - -class LISTENER -{ -public: - LISTENER(); - ~LISTENER(){}; - - void SetPort(uint16_t p) { port = p; }; - void SetPassword(const std::string & p); - void SetUserTimeout(int t) { userTimeout = t; }; - void SetScriptOnConnect(const std::string & script) { scriptOnConnect = script; }; - void SetScriptOnDisconnect(const std::string & script) { scriptOnDisconnect = script; }; - - bool Start(); - bool Stop(); - bool IsRunning() const { return !receiverStopped && !processorStopped; }; - - const std::string & GetStrError() const { return errorStr; }; - const std::string & GetVersion() const { return version; }; - -private: - // Threading stuff - static void * Run(void * self); - static void * RunProcessor(void * self); - void Runner(); - void ProcessorRunner(); - // Networking stuff - bool PrepareNet(); - bool FinalizeNet(); - bool RecvPacket(); - // Parsing stuff - bool CheckHeader(const RS::PACKET_HEADER & header) const; - bool GetParams(char * buffer, UserData & data); - // Processing stuff - void ProcessPending(); - void ProcessTimeouts(); - bool Disconnect(const UserData & data) const; - bool Connect(const UserData & data) const; - - BLOWFISH_CTX ctxS; - STG::Logger& WriteServLog; - - mutable std::string errorStr; - std::string scriptOnConnect; - std::string scriptOnDisconnect; - std::string password; - uint16_t port; - - bool running; - bool receiverStopped; - bool processorStopped; - std::vector users; - std::list pending; - int userTimeout; - - pthread_t receiverThread; - pthread_t processorThread; - pthread_mutex_t mutex; - - int listenSocket; - - std::string version; - - friend class DisconnectUser; -}; - -class DisconnectUser : public std::unary_function { - public: - explicit DisconnectUser(LISTENER & l) : listener(l) {}; - void operator()(const UserData & data) - { - listener.Disconnect(data); - }; - private: - LISTENER & listener; -}; -//-----------------------------------------------------------------------------