hosts.c

Go to the documentation of this file.

/* SPDX-FileCopyrightText: 2023 Greenbone AG
 * SPDX-FileCopyrightText: 2006 Software in the Public Interest, Inc.
 * SPDX-FileCopyrightText: 1998-2006 Tenable Network Security, Inc.
 *
 * SPDX-License-Identifier: GPL-2.0-only
 */

 
#include "hosts.h" /* for hosts_new() */
 
#include "../misc/network.h" /* for internal_recv */
#include "../misc/plugutils.h"
#include "utils.h" /* for data_left() */
 
#include <errno.h>               /* for errno() */
#include <glib.h>                /* for g_free() */
#include <gvm/base/networking.h> /* for gvm_resolve_list */
#include <stdio.h>               /* for snprintf() */
#include <string.h>              /* for strlen() */
#include <sys/wait.h>            /* for waitpid() */
#include <unistd.h>              /* for close() */
 
#undef G_LOG_DOMAIN
#define G_LOG_DOMAIN "sd   main"

struct host
{
  char *name;
  char *ip;
  pid_t pid;
  kb_t host_kb;
  kb_t results_kb;
  struct host *next;
  struct host *prev;
};

 
static struct host *hosts = NULL;
static int g_max_hosts = 15;
 
/*-------------------------------------------------------------------*/
extern int global_scan_stop;

void
host_set_time (kb_t kb, char *ip, char *type)
{
  char *timestr;
  char log_msg[1024];
  time_t t;
  int len;
 
  t = time (NULL);
  char ts[26];
  char *ts_ptr = ts;
  ctime_r (&t, ts_ptr);
  timestr = g_strdup (ts_ptr);
  len = strlen (timestr);
  if (timestr[len - 1] == '\n')
    timestr[len - 1] = '\0';
 
  snprintf (log_msg, sizeof (log_msg), "%s|||%s||||||||| |||%s", type, ip,
            timestr);
  g_free (timestr);
 
  kb_item_push_str_with_main_kb_check (kb, "internal/results", log_msg);
}
 
static void
host_rm (struct host *h)
{
  if (h->pid != 0)
    waitpid (h->pid, NULL, WNOHANG);
 
  if (h->next != NULL)
    h->next->prev = h->prev;
 
  if (h->prev != NULL)
    h->prev->next = h->next;
 
  if (h->host_kb)
    {
      kb_delete (h->host_kb);
      h->host_kb = NULL;
      kb_lnk_reset (h->results_kb);
    }
 
  g_free (h->name);
  g_free (h);
}
 
/*-----------------------------------------------------------------*/

static int
hosts_num (void)
{
  struct host *h = hosts;
  int num;
 
  for (num = 0; h != NULL; num++, h = h->next)
    ;
 
  return num;
}

static struct host *
hosts_get (char *name)
{
  struct host *h = hosts;
  while (h != NULL)
    {
      if (strcmp (h->name, name) == 0)
        return h;
      h = h->next;
    }
  return NULL;
}
 
int
hosts_init (int max_hosts)
{
  g_max_hosts = max_hosts;
  return 0;
}
 
int
hosts_new (char *name, kb_t kb, kb_t main_kb)
{
  struct host *h;
 
  while (hosts_num () >= g_max_hosts)
    {
      if (hosts_read () < 0)
        return -1;
    }
  if (global_scan_stop)
    return 0;
 
  h = g_malloc0 (sizeof (struct host));
  h->name = g_strdup (name);
  h->pid = 0;
  h->host_kb = kb;
  h->results_kb = main_kb;
  if (hosts != NULL)
    hosts->prev = h;
  h->next = hosts;
  h->prev = NULL;
  hosts = h;
  return 0;
}
 
int
hosts_set_pid (char *name, pid_t pid)
{
  struct host *h = hosts_get (name);
  if (h == NULL)
    {
      g_debug ("host_set_pid() failed!\n");
      return -1;
    }
 
  h->pid = pid;
  return 0;
}
 
/*-----------------------------------------------------------------*/
static int
hosts_stop_host (struct host *h)
{
  if (h == NULL)
    return -1;
 
  g_message ("Stopping host %s scan (pid: %d)", h->name, h->pid);
  kill (h->pid, SIGUSR1);
  return 0;
}
 
void
hosts_stop_all (void)
{
  struct host *host = hosts;
 
  global_scan_stop = 1;
  while (host)
    {
      hosts_stop_host (host);
      host = host->next;
    }
}
 
/*-----------------------------------------------------------------*/
 
static void
hosts_read_data (void)
{
  struct host *h = hosts;
  int ret = 1;
 
  while (ret > 0)
    {
      ret = waitpid (-1, NULL, WNOHANG);
      if (ret < 0)
        g_debug ("waitpid() failed. %s)", strerror (errno));
    }
 
  if (h == NULL)
    return;
 
  while (h)
    {
      if (h->pid != 0 && kill (h->pid, 0) < 0) /* Process is dead */
        {
          if (!h->prev)
            hosts = hosts->next;
          host_rm (h);
          h = hosts;
          if (!h)
            break;
        }
      h = h->next;
    }
}

int
hosts_read (void)
{
  if (hosts == NULL)
    return -1;
 
  hosts_read_data ();
  usleep (500000);
 
  return 0;
}

int
host_is_currently_scanned (gvm_host_t *host_to_check)
{
  struct host *h = hosts;
 
  GSList *list, *tmp;
  char *vhost = NULL;
 
  hosts_read ();
 
  if (h == NULL)
    return 0;
 
  vhost = gvm_host_reverse_lookup (host_to_check);
  if (!vhost)
    return 0;
 
  list = tmp = gvm_resolve_list (vhost);
  g_free (vhost);
  while (tmp)
    {
      h = hosts;
      char buffer[INET6_ADDRSTRLEN];
      addr6_to_str (tmp->data, buffer);
 
      while (h != NULL)
        {
          if (!strcasecmp (h->name, buffer))
            {
              g_slist_free_full (list, g_free);
              return 1;
            }
          h = h->next;
        }
      tmp = tmp->next;
    }
 
  g_slist_free_full (list, g_free);
  return 0;
}