mirror of
git://xwords.git.sourceforge.net/gitroot/xwords/xwords
synced 2024-12-28 09:58:30 +01:00
102 lines
2.5 KiB
C
102 lines
2.5 KiB
C
/* -*-mode: C; fill-column: 78; c-basic-offset: 4; -*- */
|
|
|
|
#ifndef _CREF_H_
|
|
#define _CREF_H_
|
|
|
|
#include <map>
|
|
#include <vector>
|
|
#include <string>
|
|
#include <pthread.h>
|
|
#include "xwrelay_priv.h"
|
|
|
|
typedef unsigned short CookieID;
|
|
|
|
using namespace std;
|
|
|
|
class CookieMapIterator; /* forward */
|
|
|
|
class CookieRef {
|
|
|
|
public:
|
|
|
|
~CookieRef();
|
|
|
|
/* Within this cookie, remember that this hostID and socket go together.
|
|
If the hostID is HOST_ID_SERVER, it's the server. */
|
|
void Associate( int socket, HostID srcID );
|
|
short GetHeartbeat() { return 60; }
|
|
CookieID GetCookieID() { return m_connectionID; }
|
|
int SocketForHost( HostID dest );
|
|
void Remove( int socket );
|
|
int CountSockets() { return m_hostSockets.size(); }
|
|
string Name() { return m_name; }
|
|
|
|
void RecordSent( int nBytes, int socket ) {
|
|
/* This really needs a lock.... */
|
|
m_totalSent += nBytes;
|
|
}
|
|
|
|
/* for console */
|
|
void PrintCookieInfo( string& out );
|
|
void PrintSocketInfo( string& out, int socket );
|
|
|
|
static CookieMapIterator GetCookieIterator();
|
|
static CookieRef* AddNew( string s );
|
|
/* Nuke an existing */
|
|
static void Delete( CookieID id );
|
|
static void Delete( const char* name );
|
|
|
|
private:
|
|
CookieRef( string s );
|
|
|
|
map<HostID,int> m_hostSockets;
|
|
pthread_rwlock_t m_sockets_rwlock;
|
|
CookieID m_connectionID;
|
|
string m_name;
|
|
int m_totalSent;
|
|
|
|
static CookieID ms_nextConnectionID;
|
|
};
|
|
|
|
typedef map<CookieID,CookieRef*> CookieMap;
|
|
|
|
class CookieMapIterator {
|
|
public:
|
|
CookieMapIterator();
|
|
~CookieMapIterator() {}
|
|
CookieID Next();
|
|
private:
|
|
CookieMap::const_iterator _iter;
|
|
};
|
|
|
|
CookieRef* get_make_cookieRef( const char* cookie, CookieID connID );
|
|
CookieRef* get_cookieRef( unsigned short cookieID );
|
|
CookieID CookieIdForName( const char* name );
|
|
|
|
class SocketStuff;
|
|
typedef map< int, SocketStuff* > SocketMap;
|
|
|
|
class SocketsIterator {
|
|
public:
|
|
SocketsIterator( SocketMap::iterator iter );
|
|
int Next();
|
|
private:
|
|
SocketMap::iterator m_iter;
|
|
};
|
|
|
|
class SocketMgr {
|
|
public:
|
|
static void Associate( int socket, CookieRef* cref );
|
|
static pthread_mutex_t* GetWriteMutexForSocket( int socket );
|
|
static void RemoveSocketRefs( int socket );
|
|
static void PrintSocketInfo( int socket, string& out );
|
|
static SocketsIterator MakeSocketsIterator();
|
|
|
|
private:
|
|
static CookieRef* CookieRefForSocket( int socket );
|
|
static SocketMap ms_SocketStuff;
|
|
static pthread_mutex_t ms_SocketStuffMutex;
|
|
};
|
|
|
|
|
|
#endif
|