network.h

Go to the documentation of this file.

/*
 *  The Mana Client
 *  Copyright (C) 2004-2009  The Mana World Development Team
 *  Copyright (C) 2009-2012  The Mana Developers
 *
 *  This file is part of The Mana Client.
 *
 *  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
 *  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, see <http://www.gnu.org/licenses/>.
 */
 
#pragma once
 
#include "utils/mutex.h"
 
#include "net/serverinfo.h"
 
#include "net/tmwa/messagehandler.h"
#include "net/tmwa/messageout.h"
 
#include <SDL_net.h>
#include <SDL_thread.h>
 
#include <map>
#include <string>
#include <unordered_map>
 
#define CLIENT_PROTOCOL_VERSION      1
// 10 -> 11: SMSG_MAP_MASK DONE
 
namespace TmwAthena {
 
struct PacketInfo;
 
class Network
{
    public:
        Network();
 
        ~Network();
 
        bool connect(const ServerInfo &server);
 
        void disconnect();
 
        const ServerInfo &getServer() const
        { return mServer; }
 
        void registerHandler(MessageHandler *handler);
 
        void unregisterHandler(MessageHandler *handler);
 
        void clearHandlers();
 
        const char *messageName(uint16_t id) const;
 
        int getState() const { return mState; }
 
        const std::string &getError() const { return mError; }
 
        bool isConnected() const { return mState == CONNECTED; }
 
        int getInSize() const { return mInSize; }
 
        void skip(int len);
 
        void dispatchMessages();
 
        void flush();
 
        // ERROR replaced by NET_ERROR because already defined in Windows
        enum {
            IDLE,
            CONNECTED,
            CONNECTING,
            DATA,
            NET_ERROR
        };
 
    private:
        friend int networkThread(void *data);
        friend class MessageOut;
 
        void setError(const std::string &error);
 
        uint16_t readWord(int pos);
 
        bool realConnect();
 
        void receive();
 
        TCPsocket mSocket = nullptr;
 
        ServerInfo mServer;
 
        char *mInBuffer, *mOutBuffer;
        unsigned int mInSize = 0;
        unsigned int mOutSize = 0;
 
        unsigned int mToSkip = 0;
 
        int mState = IDLE;
        std::string mError;
 
        SDL_Thread *mWorkerThread = nullptr;
        Mutex mMutex;
 
        std::unordered_map<uint16_t, const PacketInfo*> mPacketInfo;
        std::map<uint16_t, MessageHandler *> mMessageHandlers;
 
        static Network *mInstance;
};
 
} // namespace TmwAthena