125 lines
2.8 KiB
C++
125 lines
2.8 KiB
C++
|
|
|
|
#include "tserverproxy.h"
|
|
#include <iostream.h>
|
|
#include <Winsock.h>
|
|
#include <fcntl.h>
|
|
#include <errno.h>
|
|
#include <stdio.h>
|
|
#include <memory.h>
|
|
#include <time.h>
|
|
#include <assert.h>
|
|
|
|
using namespace std;
|
|
|
|
//=============================================================================
|
|
|
|
enum {
|
|
OK,
|
|
|
|
WSASTARTUP_FAILED,
|
|
HOST_UNKNOWN,
|
|
SOCKET_CREATION_FAILED,
|
|
CONNECTION_FAILED,
|
|
SEND_FAILED,
|
|
RECEIVE_FAILED
|
|
};
|
|
|
|
//=============================================================================
|
|
|
|
int askServer(string hostName, int portId, string question, string &answer)
|
|
{
|
|
struct hostent *he = gethostbyname(hostName.c_str());
|
|
if (!he)
|
|
return HOST_UNKNOWN;
|
|
|
|
int socket_id = socket(AF_INET, SOCK_STREAM, 0);
|
|
|
|
struct sockaddr_in addr;
|
|
memset((char *)&addr, 0, sizeof addr);
|
|
addr.sin_family = he->h_addrtype;
|
|
addr.sin_port = htons(portId);
|
|
memcpy((char *)&(addr.sin_addr), he->h_addr, he->h_length);
|
|
|
|
int rcConnect = connect(socket_id, (struct sockaddr *)&(addr), sizeof addr);
|
|
if (rcConnect == SOCKET_ERROR)
|
|
return CONNECTION_FAILED;
|
|
|
|
string message = question;
|
|
|
|
int rcSend = send(socket_id, message.c_str(), message.size(), 0);
|
|
if (rcSend == SOCKET_ERROR)
|
|
return SEND_FAILED;
|
|
|
|
char retBuffer[1024];
|
|
memset(&retBuffer, 0, 1024);
|
|
int rcRecv = recv(socket_id, retBuffer, sizeof(retBuffer) - 1, 0);
|
|
if (rcRecv == SOCKET_ERROR)
|
|
return RECEIVE_FAILED;
|
|
|
|
retBuffer[rcRecv] = '\0';
|
|
|
|
closesocket(socket_id);
|
|
|
|
answer = retBuffer;
|
|
return OK;
|
|
}
|
|
|
|
//=============================================================================
|
|
|
|
TServerProxy::TServerProxy(const string hostName, int portId)
|
|
: m_hostName(hostName), m_portId(portId)
|
|
{
|
|
#ifdef WIN32
|
|
WSADATA wsaData;
|
|
WORD wVersionRequested = MAKEWORD(1, 1);
|
|
int irc = WSAStartup(wVersionRequested, &wsaData);
|
|
#endif
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
|
|
TServerProxy::~TServerProxy()
|
|
{
|
|
#ifdef WIN32
|
|
WSACleanup();
|
|
#endif
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
|
|
bool TServerProxy::testConnection() const
|
|
{
|
|
struct hostent *he = gethostbyname(m_hostName.c_str());
|
|
if (!he)
|
|
return false;
|
|
|
|
int socket_id = socket(AF_INET, SOCK_STREAM, 0);
|
|
|
|
struct sockaddr_in addr;
|
|
memset((char *)&addr, 0, sizeof addr);
|
|
addr.sin_family = he->h_addrtype;
|
|
addr.sin_port = htons(m_portId);
|
|
memcpy((char *)&(addr.sin_addr), he->h_addr, he->h_length);
|
|
|
|
bool ret = (SOCKET_ERROR != connect(socket_id, (struct sockaddr *)&(addr), sizeof addr));
|
|
closesocket(socket_id);
|
|
return ret;
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
|
|
int TServerProxy::exec(int argc, char *argv[], string &reply)
|
|
{
|
|
string cmd;
|
|
assert(argc > 0);
|
|
if (argc > 0) {
|
|
cmd = argv[0];
|
|
for (int i = 1; i < argc; ++i) {
|
|
cmd += ",";
|
|
cmd += argv[i];
|
|
}
|
|
}
|
|
|
|
return askServer(m_hostName, m_portId, cmd, reply);
|
|
}
|