git.sesse.net Git - nbtscanner/blob - id_list.h

   1 /*
   2  * nbtscanner -- a tool for scanning large networks for SMB servers.
   3  *
   4  * id_list.h: Prototypes and structures for keeping track of the packet IDs.
   5  * Copyright (C) 2000 Steinar H. Gunderson
   6  *
   7  * Large amounts of code adapted from Samba (http://www.samba.org/)
   8  * Copyright (C) Andrew Tridgell 1994-1998, and others.
   9  *
  10  * This program is free software; you can redistribute it and/or modify
  11  * it under the terms of the GNU General Public License as published by
  12  * the Free Software Foundation; either version 2 of the License, or
  13  * (at your option) any later version.
  14  *
  15  * This program is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  * GNU General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU General Public License
  21  * along with this program; if not, write to the Free Software
  22  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  23  */
  24
  25 #ifndef _ID_LIST_H
  26 #define _ID_LIST_H 1
  27
  28 #include <sys/time.h>
  29 #include <netinet/in.h>
  30
  31 /*
  32  * There's quite some data duplication here, in the interest of speed and code
  33  * simplicity.
  34  */
  35
  36 struct id_entry {
  37         unsigned short id;              /* ID number from NBT packet */
  38         int in_use;                     /* Are we awaiting a reply on this ID? */
  39
  40         struct in_addr sin;             /* IP address we've sent it to */
  41         struct timeval timestamp;       /* When we sent the packet */
  42         int retries;                    /* Number of retries tried */
  43
  44         /* we list on index entry instead of using pointers */
  45         int prev_index;
  46         int next_index;
  47 };
  48
  49 struct id_list_entry {
  50         struct id_list_entry *next;
  51         struct id_entry *id;
  52 };
  53
  54 /*
  55  * a list of all the available ID numbers (sin is unused, in_use = 0)
  56  *
  57  * We could have a used_id list as well, but it would require a doubly
  58  * linked list, and perhaps doesn't help all that much anyways
  59  */
  60 extern struct id_list_entry *head_free_id;
  61
  62 /* an array with a one-to-one mapping of the array index vs. the id field */
  63 extern struct id_entry ids[];
  64
  65 void id_list_init();
  66 void id_list_destroy();
  67 void id_free(struct id_entry *i);
  68 void id_mark_free(struct id_entry *i);
  69 struct id_entry *id_get_free_id();
  70 int id_cleanup(int timeout_ms);
  71 int get_num_free_ids();
  72
  73 #endif /* _ID_LIST_H */
  74