hosts_gatherer.c
上传用户:tjescc
上传日期:2021-02-23
资源大小:419k
文件大小:5k
源码类别:

Telnet服务器

开发平台:

Unix_Linux

  1. /* HostsGatherer
  2.  *
  3.  * Copyright (C) 1999 Renaud Deraison
  4.  *
  5.  * This library is free software; you can redistribute it and/or
  6.  * modify it under the terms of the GNU Library General Public
  7.  * License as published by the Free Software Foundation; either
  8.  * version 2 of the License, or (at your option) any later version.
  9.  *
  10.  * This library is distributed in the hope that it will be useful,
  11.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13.  * Library General Public License for more details.
  14.  *
  15.  * You should have received a copy of the GNU Library General Public
  16.  * License along with this library; if not, write to the Free
  17.  * Software Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  18.  *
  19.  */
  20. #include <includes.h>
  21. #include "hosts_gatherer.h"
  22. #include "hg_add_hosts.h" 
  23. #include "hg_subnet.h"
  24. #include "hg_utils.h"
  25. #include "hg_filter.h"
  26. #include "hg_dns_axfr.h"
  27. int hg_test_syntax(char * hostname, int flags)
  28. {
  29.  struct hg_globals * globals = malloc(sizeof(struct hg_globals));
  30.  int err;
  31.  
  32.  hostname = strdup(hostname);
  33.  bzero(globals, sizeof(struct hg_globals));
  34.  globals->flags = flags;
  35.  globals->host_list = malloc(sizeof(struct hg_host));
  36.  bzero(globals->host_list, sizeof(struct hg_host));
  37.  
  38.  globals->tested = malloc(sizeof(struct hg_host));
  39.  bzero(globals->tested, sizeof(struct hg_host));
  40.  
  41.  globals->input = strdup(hostname);
  42.  globals->marker = globals->input;
  43.  
  44.  globals->distribute = 0;
  45.  
  46.  err = hg_add_comma_delimited_hosts(globals, 0);
  47.  free(hostname);
  48.  hg_cleanup(globals);
  49.  return err;
  50. }
  51. struct hg_globals * 
  52. hg_init(hostname, flags)
  53.  char * hostname;
  54.  int flags;
  55. {
  56.  struct hg_globals * globals = malloc(sizeof(struct hg_globals));
  57.  hostname = strdup(hostname);
  58.  bzero(globals, sizeof(struct hg_globals));
  59.  globals->flags = flags;
  60.  globals->host_list = malloc(sizeof(struct hg_host));
  61.  bzero(globals->host_list, sizeof(struct hg_host));
  62.  
  63.  globals->tested = malloc(sizeof(struct hg_host));
  64.  bzero(globals->tested, sizeof(struct hg_host));
  65.  
  66.  globals->input = strdup(hostname);
  67.  globals->marker = globals->input;
  68.  
  69.  globals->distribute = 0;
  70.  
  71.  hg_add_comma_delimited_hosts(globals, 256);
  72.  free(hostname);
  73.  return(globals);
  74. }
  75. int hg_next_host(globals, ip, hostname, sz)
  76.  struct hg_globals * globals;
  77.  struct in_addr *ip;
  78.  char * hostname;
  79.  int sz;
  80. {
  81.  struct hg_host * host;
  82.  
  83.  if(!globals) return -1;
  84. #ifdef DEBUG_HIGH
  85.  printf("Hosts list : n");
  86.  hg_dump_hosts(globals->host_list);
  87. #endif
  88.  host = globals->host_list;
  89.  
  90.  while(host->tested && host->next){
  91.         struct hg_host * next = host->next;
  92.         globals->host_list = next;
  93.         hg_host_cleanup(host);
  94.         host = next;
  95.         }
  96.      
  97.  if( globals->flags & HG_DISTRIBUTE )
  98.  {
  99.   struct hg_host * first = host;
  100.   unsigned int i;
  101.   
  102.   
  103.   i = 0;
  104. again:
  105.   host = first;
  106.   while (host != NULL && host->next != NULL )
  107.   {
  108.    if (  host->tested == 0 ){
  109.  if( globals->distribute == i ) break;
  110.  }
  111.    i ++;
  112.    host = host->next;
  113.   }
  114.   globals->distribute ++;
  115.     
  116.   if( host == NULL || host->next == NULL ) { 
  117.    if ( i == 0 ) return -1 ;
  118.    globals->distribute = 0 ; 
  119. i = 0; 
  120. goto again; 
  121. }
  122.  }
  123.  
  124.  
  125.  if( host != NULL && host->next == NULL )
  126.  {
  127.   if(globals->marker != NULL)
  128.    {
  129.    hg_add_comma_delimited_hosts(globals, 0);
  130. return hg_next_host(globals, ip, hostname, sz);
  131. }
  132.   else return -1;
  133.  }
  134.    
  135.   if((globals->flags & HG_DNS_AXFR) && hg_filter_domain(globals, host->domain) == 0 )
  136.         {
  137.       hg_dns_axfr_add_hosts(globals, host->domain);
  138. }
  139.   
  140.   if(!host->use_max || (host->addr.s_addr == host->max.s_addr))host->tested = 1;
  141.   host->alive = 1;
  142.   
  143.   if(ip)ip->s_addr = host->addr.s_addr;
  144.    
  145.    if(!host->use_max)
  146.    {
  147.    if((globals->flags & HG_REVLOOKUP))
  148.      {
  149.       if(!host->hostname ||
  150.         (inet_addr(host->hostname) != INADDR_NONE)) 
  151.   return hg_get_name_from_ip(host->addr, hostname, sz);
  152.          else
  153.   {
  154.           strncpy(hostname, host->hostname, sz - 1);
  155.      return 0;
  156.   }
  157.      }
  158.    else
  159.     {
  160.      if(host->hostname && (inet_addr(host->hostname) == INADDR_NONE))
  161.        strncpy(hostname, host->hostname, sz - 1);
  162.      else 
  163.        strncpy(hostname, inet_ntoa(host->addr), sz - 1);
  164.       return 0;
  165.      }
  166.    }
  167.    else
  168.    {
  169.     if(globals->flags & HG_REVLOOKUP)
  170.       hg_get_name_from_ip(host->addr, hostname, sz);
  171.     else
  172.       strncpy(hostname, inet_ntoa(host->addr), sz - 1);
  173.     
  174.     host->addr.s_addr = htonl(ntohl(host->addr.s_addr) + 1);
  175.     return 0;
  176.    }
  177. }
  178. void hg_cleanup(globals) 
  179. struct hg_globals * globals;
  180. {
  181.  struct hg_host * hosts = globals->host_list;
  182.  struct hg_host * tested = globals->tested;
  183.  free(globals->input);
  184.  free(globals);
  185.  
  186.  hg_hosts_cleanup(hosts);
  187.  hg_hosts_cleanup(tested);
  188. }
  189.