wraith-ircbot/src/auth.cc

/*
 * auth.c -- handles:
 *   auth system functions
 *   auth system hooks
 */


#include "common.h"
#include "auth.h"
#include "misc.h"
#include "main.h"
#include "settings.h"
#include "types.h"
#include "userrec.h"
#include "set.h"
#include "core_binds.h"
#include "egg_timer.h"
#include "users.h"
#include "crypt.h"
#include <sys/stat.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <unistd.h>
#include <fcntl.h>
#include "chan.h"
#include "tandem.h"
#include "src/mod/server.mod/server.h"
#include <bdlib/src/String.h>
#include <bdlib/src/HashTable.h>
#include <pwd.h>
#include <errno.h>

#include <net/if.h>
#include <sys/ioctl.h>
#include <sys/socket.h>
#include <signal.h>


#include "stat.h"

bd::HashTable<bd::String, Auth*> Auth::ht_host(10);
bd::HashTable<RfcString, Auth*> Auth::ht_nick(10);

Auth::Auth(const RfcString& _nick, const char *_host, struct userrec *u)
{
  Status(AUTHING);
  nick = _nick;
  strlcpy(host, _host, UHOSTLEN);
  user = u;

  ht_host[host] = this;
  ht_nick[_nick] = this;

  sdprintf(STR("New auth created! (%s!%s) [%s]"), nick.c_str(), host,
      u ? u->handle : "*");
  authtime = atime = now;
  idx = -1;
}

Auth::~Auth()
{
  sdprintf(STR("Removing auth: (%s!%s) [%s]"), nick.c_str(), host,
      user ? user->handle : "*");
  ht_host.remove(host);
  ht_nick.remove(nick);
}

void Auth::MakeHash() noexcept
{
 make_rand_str(rand, 50);
 makehash(user, rand, hash, 50);
}

void Auth::Done() noexcept
{
  hash[0] = 0;
  rand[0] = 0;
  Status(AUTHED);
}

void Auth::NewNick(const RfcString& newnick) noexcept
{
  if (ht_nick.contains(nick)) {
    ht_nick.remove(nick);
  }
  nick = newnick;
  ht_nick[newnick] = this;
}

Auth *Auth::Find(const char *_host) noexcept
{

  if (ht_host.contains(_host)) {
    Auth *auth = ht_host[_host];
    sdprintf(STR("Found auth: (%s!%s) [%s]"), auth->nick.c_str(), auth->host,
        auth->user ? auth->user->handle : "*");
    return auth;
  }
  return NULL;
}

static void auth_clear_users_block(const bd::String& key, Auth* auth)
{
  if (auth->user) {
    sdprintf(STR("Clearing USER for auth: (%s!%s) [%s]"), auth->nick.c_str(),
        auth->host, auth->user ? auth->user->handle : "*");
    auth->user = NULL;
  }
}

void Auth::NullUsers(const RfcString& nick) noexcept
{
  if (!ht_nick.contains(nick))
    return;
  auto auth = ht_nick[nick];
  auth_clear_users_block(nick, auth);
}

void Auth::NullUsers(void) noexcept
{
  for (auto& kv : ht_host) {
    auth_clear_users_block(kv.first, kv.second);
  }
}

static void auth_fill_users_block(const bd::String& key, Auth* auth)
{
  char from[NICKLEN + UHOSTLEN];

  sdprintf(STR("Filling USER for auth: (%s!%s) [%s]"), auth->nick.c_str(), auth->host,
      auth->user ? auth->user->handle : "*");
  simple_snprintf(from, sizeof(from), "%s!%s", auth->nick.c_str(), auth->host);
  auth->user = get_user_by_host(from);
}

void Auth::FillUsers(void) noexcept
{
  for (auto& kv : ht_host) {
    auth_fill_users_block(kv.first, kv.second);
  }
}


void Auth::ExpireAuths() noexcept
{
  if (!ischanhub() || ht_host.size() == 0)
    return;
  std::vector<bd::String> expired_hosts;
  expired_hosts.reserve(ht_host.size());
  std::vector<const Auth*> delete_auths;
  delete_auths.reserve(ht_host.size());

  for (const auto& kv : ht_host) {
    const bd::String& host = kv.first;
    const auto& auth = kv.second;
    if (auth->Authed() && ((now - auth->atime) >= (1 * 60))) {
      putlog(LOG_DEBUG, "*", STR("Auth (%s!%s) [%s] expired."),
          auth->nick.c_str(), auth->host, auth->user ? auth->user->handle : "*");
      expired_hosts.push_back(host);
      ht_nick.remove(auth->nick);
      delete_auths.push_back(auth);
    }
  }
  for (const auto& host : expired_hosts) {
    ht_host.remove(host);
  }
  for (const auto& auth : delete_auths) {
    delete auth;
  }
}

void Auth::DeleteAll() noexcept
{
  if (!ischanhub())
    return;
  putlog(LOG_DEBUG, "*", STR("Removing %zd auth entries."), ht_host.size());
  std::vector<const Auth*> delete_auths;
  delete_auths.reserve(ht_host.size());
  for (const auto& kv : ht_host) {
    const auto& auth = kv.second;
    putlog(LOG_DEBUG, "*", STR("Removing (%s!%s) [%s], from auth list."),
        auth->nick.c_str(), auth->host, auth->user ? auth->user->handle : "*");
    delete_auths.push_back(auth);
  }
  ht_host.clear();
  ht_nick.clear();
  for (const auto& auth : delete_auths) {
    delete auth;
  }
}

void Auth::InitTimer() noexcept
{
  timer_create_secs(60, STR("Auth::ExpireAuths"), (Function) Auth::ExpireAuths);
}

bool Auth::GetIdx(const char *chname) noexcept
{
sdprintf(STR("GETIDX: auth: %s, idx: %d"), nick.c_str(), idx);
  if (idx != -1) {
    if (!valid_idx(idx))
      idx = -1;
    else if (!dcc[idx].irc || dcc[idx].simul == -1)
      idx = -1;
    else if (user && strcmp(dcc[idx].nick, user->handle))
      idx = -1;
    else {
      sdprintf(STR("FIRST FOUND: %d"), idx);
      strlcpy(dcc[idx].simulbot, nick.c_str(), sizeof(dcc[idx].simulbot));
      strlcpy(dcc[idx].u.chat->con_chan, chname ? chname : "*", sizeof(dcc[idx].u.chat->con_chan));
      return 1;
    }
  }

  int i = 0;

  for (i = 0; i < dcc_total; i++) {
    if (dcc[i].type && dcc[i].irc &&
       (((chname && chname[0]) && !strcmp(dcc[i].simulbot, chname) &&
         user && !strcmp(dcc[i].nick, user->handle)) ||
       (!(chname && chname[0]) && !strcmp(dcc[i].simulbot, nick.c_str())))) {
      putlog(LOG_DEBUG, "*", STR("Simul found old idx for %s/%s: (%s!%s)"), nick.c_str(), chname, nick.c_str(), host);
      dcc[i].simultime = now;
      idx = i;
      strlcpy(dcc[idx].simulbot, nick.c_str(), sizeof(dcc[idx].simulbot));
      strlcpy(dcc[idx].u.chat->con_chan, chname ? chname : "*", sizeof(dcc[idx].u.chat->con_chan));

      return 1;
    }
  }

  idx = new_dcc(&DCC_CHAT, sizeof(struct chat_info));

  if (idx != -1) {
    char from[NICKLEN + UHOSTLEN];

    dcc[idx].sock = -1;
    dcc[idx].timeval = now;
    dcc[idx].irc = 1;
    dcc[idx].simultime = now;
    dcc[idx].simul = 0;		/* not -1, so it's cleaned up later */
    dcc[idx].status = STAT_COLOR;
    dcc[idx].u.chat->con_flags = 0;
    strlcpy(dcc[idx].simulbot, nick.c_str(), sizeof(dcc[idx].simulbot));
    strlcpy(dcc[idx].u.chat->con_chan, chname ? chname : "*", sizeof(dcc[idx].u.chat->con_chan));
    dcc[idx].u.chat->strip_flags = STRIP_ALL;
    strlcpy(dcc[idx].nick, user ? user->handle : "*", sizeof(dcc[idx].nick));
    strlcpy(dcc[idx].host, host, sizeof(dcc[idx].host));
    dcc[idx].addr = 0L;
    simple_snprintf(from, sizeof(from), "%s!%s", nick.c_str(), host);
    dcc[idx].user = user ? user : get_user_by_host(from);
    return 1;
  }

  return 0;
}

void Auth::TellAuthed(int idx) noexcept
{
  for (const auto& kv : ht_host) {
    const Auth* auth = kv.second;
    dprintf(idx, "(%s!%s) [%s] authtime: %li, atime: %li, Status: %d\n",
        auth->nick.c_str(),
        auth->host, auth->user ? auth->user->handle : "*",
        (long)auth->authtime, (long)auth->atime, auth->Status());
  }
}

void makehash(struct userrec *u, const char *randstring, char *out, size_t out_size)
{
  char hash[256] = "", *secpass = NULL;

  if (u && get_user(&USERENTRY_SECPASS, u)) {
    secpass = strdup((char *) get_user(&USERENTRY_SECPASS, u));
    secpass[strlen(secpass)] = 0;
  }
  simple_snprintf(hash, sizeof hash, "%s%s%s", randstring, (secpass && secpass[0]) ? secpass : "", auth_key);
  if (secpass)
    free(secpass);

  strlcpy(out, MD5(hash), out_size);
  OPENSSL_cleanse(hash, sizeof(hash));
}

int check_auth_dcc(Auth *auth, const char *cmd, const char *par)
{
  return real_check_bind_dcc(cmd, auth->idx, par, auth);
}
/* vim: set sts=2 sw=2 ts=8 et: */