Inetd.cpp

Go to the documentation of this file.

/*

$Header$

*/

#include "Error.h"
#include "Value.h"
#include "Telnet.h"

#include "Inetd.h"

// Public methods.

unsigned int Inetd::instancesNumberLimit()
{
  return 1;
}

// Protected methods.

Inetd::Inetd(const ValueList &params) : Service<Inetd>(params)
{
  BIND(services);

  int res;

  if (FAILED(res = REGISTRY_GET(Inetd, services, _services))) {
    failure(ERROR_BACKTRACE(res));
    return;
  }

  unsigned int i, size = _services.size();

  if (size == 0) {
    failure(ERROR(SERVICE_MSG_INVALID_PARAMS, CSTRING(NAME())));
    return;
  }

  _core = CORE();

  TcpServer *server;
  for (i = 0; i < size; ++i) {
    if (_services[i].service.empty() || _services[i].port == 0) {
      failure(ERROR(SERVICE_MSG_INVALID_PARAMS, CSTRING(NAME())));
      return;
    }
    DEBUG("Starting to listen port %u for service '%s'.", _services[i].port, CSTRING(_services[i].service));

    if (!(server = new TcpServer("", String::number(_services[i].port)))) {
      failure(ERROR(MSG_OBJECT_CANNOT_CREATE, "TcpServer"));
      return;
    }
    if (FAILED(res = server->failureCode())) {
      failure(ERROR_BACKTRACE(res));
      return;
    }

    if (FAILED(res = CONNECT(server, accepted, this, Inetd, startService))) {
      failure(ERROR_BACKTRACE(res));
      return;
    }
    _servers[server] = _services[i];
    _serversByPort[_services[i].port] = server;
  }
}

Inetd::~Inetd()
{
  if (!_servers.empty()) {
    InetdServerList::iterator i = _servers.begin();
    InetdServerList::iterator e = _servers.end();

    while (i != e) {
      delete i->first;
      ++i;
    }
  }
}

// Private methods.

int Inetd::startService(TcpConnection *connection)
{
  InetdServersByPort::const_iterator fServer = _serversByPort.find(connection->localPort());
  if (fServer == _serversByPort.end())
    return ERROR(MSG_INETD_UNKNOWN_SERVER);

  InetdServerList::const_iterator f = _servers.find(fServer->second);
  if (f == _servers.end())
    return ERROR(MSG_INETD_UNKNOWN_SERVER);

  bool valid;
  int res;

  if (FAILED(res = authorize(f->second, connection, valid)))
    return ERROR_BACKTRACE(res);

  if (!valid) {
    delete connection;
    return ERROR(MSG_INETD_AUTHORIZATION_FAILED);
  }

  ValueList params;

  params.push_back(new Value<Object *>(connection));
  if (FAILED(res = _core->spawn(f->second.service, params)))
    return ERROR_BACKTRACE(res);

  INFO(MSG_INETD_SERVICE_INSTANCE_STARTED, CSTRING(f->second.service), CSTRING(f->second.login), CSTRING(connection->peerIp()));
  return OK;
}

int Inetd::authorize(const InetdItem &item, TcpConnection *connection, bool &valid)
{
  if (item.login.empty() && item.password.empty()) {
    valid = true;
    return OK;
  }

  String str;
  int res;

  valid = true;

  if (FAILED(res = connection->clear()))
    return ERROR_BACKTRACE(res);

  if (!item.login.empty()) {
    if (FAILED(res = connection->writeStr(MSG_INETD_LOGIN)))
      return ERROR_BACKTRACE(res);
    if (FAILED(res = connection->readLn(str)))
      return ERROR_BACKTRACE(res);
    if (str != item.login)
      valid = false;
  }
  if (!item.password.empty()) {
    int res1;

    if (FAILED(res = connection->writeStr(MSG_INETD_PASSWORD)))
      return ERROR_BACKTRACE(res);
    if (FAILED(res = Telnet::setEcho(connection, false)))
      return ERROR_BACKTRACE(res);
    res1 = connection->readLn(str);
    str.trim(true);
    DEBUG("Entered password: '%s'.", CSTRING(str));
    if (FAILED(res = connection->writeStr(String::eol())))
      return ERROR_BACKTRACE(res);
    if (FAILED(res = Telnet::setEcho(connection, true)))
      return ERROR_BACKTRACE(res);
    if (FAILED(res1))
      return ERROR_BACKTRACE(res1);
    if (!item.password.equals(str.md5(), true))
      valid = false;
  }
  if (!valid)
    if (FAILED(res = connection->writeStr(String(MSG_INETD_AUTHORIZATION_FAILED) + String::eol())))
      return ERROR_BACKTRACE(res);
  if (FAILED(res = connection->clear()))
    return ERROR_BACKTRACE(res);
  return OK;
}

---