mirror of
https://github.com/glest/glest-source.git
synced 2025-09-26 07:28:59 +02:00
151 lines
3.6 KiB
C++
151 lines
3.6 KiB
C++
// ==============================================================
|
||
// This file is part of Glest Shared Library (www.glest.org)
|
||
//
|
||
// Copyright (C) 2001-2008 Marti<74>o Figueroa
|
||
//
|
||
// You can redistribute this code 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
|
||
// ==============================================================
|
||
|
||
#ifndef _SHARED_PLATFORM_SOCKET_H_
|
||
#define _SHARED_PLATFORM_SOCKET_H_
|
||
|
||
#include <string>
|
||
#include <winsock.h>
|
||
#include <map>
|
||
#include <vector>
|
||
#include "thread.h"
|
||
|
||
using std::string;
|
||
|
||
const char* WSAGetLastErrorMessage(const char* pcMessagePrefix,int nErrorID = 0);
|
||
|
||
namespace Shared{ namespace Platform{
|
||
|
||
// =====================================================
|
||
// class IP
|
||
// =====================================================
|
||
|
||
class Ip{
|
||
private:
|
||
unsigned char bytes[4];
|
||
|
||
public:
|
||
Ip();
|
||
Ip(unsigned char byte0, unsigned char byte1, unsigned char byte2, unsigned char byte3);
|
||
Ip(const string& ipString);
|
||
|
||
unsigned char getByte(int byteIndex) {return bytes[byteIndex];}
|
||
string getString() const;
|
||
};
|
||
|
||
// =====================================================
|
||
// class Socket
|
||
// =====================================================
|
||
|
||
class Socket{
|
||
private:
|
||
class SocketManager{
|
||
public:
|
||
SocketManager();
|
||
~SocketManager();
|
||
};
|
||
|
||
protected:
|
||
static SocketManager socketManager;
|
||
SOCKET sock;
|
||
static int broadcast_portno;
|
||
|
||
public:
|
||
Socket(SOCKET sock);
|
||
Socket();
|
||
virtual ~Socket();
|
||
|
||
static int getBroadCastPort() { return broadcast_portno; }
|
||
static void setBroadCastPort(int value) { broadcast_portno = value; }
|
||
static std::vector<std::string> getLocalIPAddressList();
|
||
|
||
// Int lookup is socket fd while bool result is whether or not that socket was signalled for reading
|
||
static bool hasDataToRead(std::map<int,bool> &socketTriggeredList);
|
||
static bool hasDataToRead(int socket);
|
||
bool hasDataToRead();
|
||
void disconnectSocket();
|
||
|
||
int getSocketId() const { return sock; }
|
||
|
||
int getDataToRead();
|
||
int send(const void *data, int dataSize);
|
||
int receive(void *data, int dataSize);
|
||
int peek(void *data, int dataSize);
|
||
|
||
void setBlock(bool block);
|
||
bool isReadable();
|
||
bool isWritable(bool waitOnDelayedResponse);
|
||
bool isConnected();
|
||
|
||
static string getHostName();
|
||
static string getIp();
|
||
|
||
protected:
|
||
static void throwException(const string &str);
|
||
|
||
};
|
||
|
||
// =====================================================
|
||
// class ClientSocket
|
||
// =====================================================
|
||
|
||
class ClientSocket: public Socket{
|
||
public:
|
||
void connect(const Ip &ip, int port);
|
||
static std::vector<string> discoverServers();
|
||
};
|
||
|
||
class BroadCastSocketThread : public Thread
|
||
{
|
||
private:
|
||
Mutex mutexRunning;
|
||
Mutex mutexQuit;
|
||
|
||
bool quit;
|
||
bool running;
|
||
|
||
void setRunningStatus(bool value);
|
||
void setQuitStatus(bool value);
|
||
|
||
public:
|
||
BroadCastSocketThread();
|
||
virtual void execute();
|
||
void signalQuit();
|
||
bool getQuitStatus();
|
||
bool getRunningStatus();
|
||
};
|
||
|
||
// =====================================================
|
||
// class ServerSocket
|
||
// =====================================================
|
||
|
||
class ServerSocket: public Socket{
|
||
|
||
public:
|
||
ServerSocket();
|
||
virtual ~ServerSocket();
|
||
|
||
void bind(int port);
|
||
void listen(int connectionQueueSize= SOMAXCONN);
|
||
Socket *accept();
|
||
void stopBroadCastThread();
|
||
|
||
protected:
|
||
|
||
BroadCastSocketThread *broadCastThread;
|
||
void startBroadCastThread();
|
||
|
||
};
|
||
|
||
}}//end namespace
|
||
|
||
#endif
|