cyphesis/server/CommMetaClient.cpp
Al Riddoch e3fdb2c7df 2003-08-23 Al Riddoch <alriddoch@zepler.org>
* server/server.cpp: Handle the case correctly if the metaserver
	  socket setup does not work.

	* server/CommMetaClient.h, server/CommMetaClient.cpp: Re-write
	  CommMetaClient to use skstreams udp_socket_stream.
2003-08-23 20:40:39 +00:00

127 lines
2.8 KiB
C++

// This file may be redistributed and modified only under the terms of
// the GNU General Public License (See COPYING for details).
// Copyright (C) 2000,2001 Alistair Riddoch
#include "CommMetaClient.h"
#include "protocol_instructions.h"
#include "common/debug.h"
#include "common/log.h"
#include "common/globals.h"
#include <cstring>
#include <iostream>
static const bool debug_flag = false;
static inline char *pack_uint32(uint32_t data, char *buffer, unsigned int *size)
{
uint32_t netorder;
netorder = htonl(data);
memcpy(buffer, &netorder, sizeof(uint32_t));
*size += sizeof(uint32_t);
return buffer+sizeof(uint32_t);
}
static inline char *unpack_uint32(uint32_t *dest, char *buffer)
{
uint32_t netorder;
memcpy(&netorder, buffer, sizeof(uint32_t));
*dest = ntohl(netorder);
return buffer+sizeof(uint32_t);
}
CommMetaClient::~CommMetaClient()
{
metaserverTerminate();
}
int CommMetaClient::getFd() const
{
return clientIos.getSocket();
}
bool CommMetaClient::eof()
{
return false;
}
bool CommMetaClient::isOpen() const
{
return true;
}
bool CommMetaClient::read()
{
metaserverReply();
return false;
}
bool CommMetaClient::setup(const std::string & mserver)
{
// Establish socket for communication with the metaserver
return clientIos.setTarget(mserver, metaserverPort);
}
static const int MAXLINE = 4096;
void CommMetaClient::metaserverKeepalive()
{
char mesg[MAXLINE];
unsigned int packet_size = 0;
pack_uint32(SKEEP_ALIVE, mesg, &packet_size);
clientIos.write(mesg, packet_size);
clientIos << std::flush;
}
void CommMetaClient::metaserverReply()
{
char mesg[MAXLINE];
char *mesg_ptr;
uint32_t handshake = 0, command = 0;
unsigned int packet_size;
clientIos.peek();
if (clientIos.readsome(mesg, MAXLINE) < (std::streamsize)sizeof(command)) {
log(WARNING, "WARNING: Reply from metaserver too short");
}
mesg_ptr = unpack_uint32(&command, mesg);
if(command == HANDSHAKE)
{
mesg_ptr = unpack_uint32(&handshake, mesg_ptr);
debug(std::cout << "MetaServer contacted successfully."
<< std::endl << std::flush;);
packet_size = 0;
mesg_ptr = pack_uint32(SERVERSHAKE, mesg, &packet_size);
mesg_ptr = pack_uint32(handshake, mesg_ptr, &packet_size);
clientIos.write(mesg, packet_size);
clientIos << std::flush;
}
}
void CommMetaClient::metaserverTerminate()
{
char mesg[MAXLINE];
unsigned int packet_size = 0;
pack_uint32(TERMINATE, mesg, &packet_size);
clientIos.write(mesg, packet_size);
clientIos << std::flush;
}
void CommMetaClient::idle(time_t t)
{
if (t > (lastTime + 5 * 60)) {
lastTime = t;
metaserverKeepalive();
}
}