remtask.cpp

#include "remtask.h"
#include <netdb.h>

RemoteTask::RemoteTask(TaskPool& taskPool, taskhandle_t handle, taskid_t id, pid_t pid,
			    uint16_t cmdPort, uint16_t dataPort, ipaddr_t remoteAddr) :
    ExternalTask(taskPool, handle, id, pid, cmdPort, dataPort), receiving(false), remoteAddr(remoteAddr)
{
}

bool RemoteTask::rejectTask(taskhandle_t task)
{
    if (::rejectTask(task)) {
	syslog(LOG_ERR, "rejecting task %s from %s", task.str(), remoteAddr.str().c_str());
	sendErrorToClient(ACNET_REQREJ);
	return true;
    }

    return false;
}

void RemoteTask::handleReceiveRequests()
{
    receiving = true;

    Ack ack;

    if (!sendAckToClient(&ack, sizeof(ack)))
	taskPool().removeTask(this);
}

void RemoteTask::handleBlockRequests()
{
    receiving = false;

    while (!replies.empty())
	taskPool().rpyPool.endRpyId(*replies.begin(), ACNET_DISCONNECTED);

    Ack ack;
    if (!sendAckToClient(&ack, sizeof(ack)))
	taskPool().removeTask(this);
}

void RemoteTask::handleSend(SendCommand const *cmd, size_t const len)
{
    if (!rejectTask(cmd->task()))
	ExternalTask::handleSend(cmd, len);
}

void RemoteTask::handleSendRequest(SendRequestCommand const *cmd, size_t const len)
{
    if (!rejectTask(cmd->task()))
	ExternalTask::handleSendRequest(cmd, len);
}

void RemoteTask::handleSendRequestWithTimeout(SendRequestWithTimeoutCommand const* cmd, size_t const len)
{
    if (!rejectTask(cmd->task()))
	ExternalTask::handleSendRequestWithTimeout(cmd, len);
}

size_t RemoteTask::totalProp() const
{
    return ExternalTask::totalProp() + 1;
}

char const* RemoteTask::propName(size_t ii) const
{
    if (ii < ExternalTask::totalProp())
	return ExternalTask::propName(ii);

    if (ii == (ExternalTask::totalProp()))
	return "Remote Address";

    return 0;
}

std::string RemoteTask::propVal(size_t ii) const
{
    if (ii < ExternalTask::totalProp())
	return ExternalTask::propVal(ii);

    if (ii == (ExternalTask::totalProp())) {
	sockaddr_in addr;

	addr.sin_family = AF_INET;
	addr.sin_port = 0;
	addr.sin_addr.s_addr = htonl(remoteAddr.value());

	char host[1024];

	if (getnameinfo((sockaddr *) &addr, sizeof(addr), host, sizeof(host), 0, 0, 0) == 0)
	    return std::string(host);
	else
	    return remoteAddr.str();
    }

    return "";
}