tahoma2d/toonz/sources/toonzfarm/tfarm/tserverproxy.cpp

#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);
}