source3/nmbd/nmbd_subnetdb.c

   1 /*
   2    Unix SMB/Netbios implementation.
   3    Version 1.9.
   4    NBT netbios routines and daemon - version 2
   5    Copyright (C) Andrew Tridgell 1994-1998
   6    Copyright (C) Luke Kenneth Casson Leighton 1994-1998
   7    Copyright (C) Jeremy Allison 1994-1998
   8
   9    This program is free software; you can redistribute it and/or modify
  10    it under the terms of the GNU General Public License as published by
  11    the Free Software Foundation; either version 2 of the License, or
  12    (at your option) any later version.
  13
  14    This program is distributed in the hope that it will be useful,
  15    but WITHOUT ANY WARRANTY; without even the implied warranty of
  16    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17    GNU General Public License for more details.
  18
  19    You should have received a copy of the GNU General Public License
  20    along with this program; if not, write to the Free Software
  21    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  22
  23    Revision History:
  24
  25 */
  26
  27 #include "includes.h"
  28 #include "smb.h"
  29
  30 extern int ClientNMB;
  31 extern int ClientDGRAM;
  32 extern int global_nmb_port;
  33
  34 extern int DEBUGLEVEL;
  35
  36 extern fstring myworkgroup;
  37 extern char **my_netbios_names;
  38 extern struct in_addr ipzero;
  39
  40 /* This is the broadcast subnets database. */
  41 struct subnet_record *subnetlist = NULL;
  42
  43 /* Extra subnets - keep these separate so enumeration code doesn't
  44    run onto it by mistake. */
  45
  46 struct subnet_record *unicast_subnet = NULL;
  47 struct subnet_record *remote_broadcast_subnet = NULL;
  48 struct subnet_record *wins_server_subnet = NULL;
  49
  50 extern uint16 samba_nb_type; /* Samba's NetBIOS name type. */
  51
  52 /****************************************************************************
  53   Add a subnet into the list.
  54   **************************************************************************/
  55
  56 static void add_subnet(struct subnet_record *subrec)
  57 {
  58   struct subnet_record *subrec2;
  59
  60   if (!subnetlist)
  61   {
  62     subnetlist = subrec;
  63     subrec->prev = NULL;
  64     subrec->next = NULL;
  65     return;
  66   }
  67
  68   for (subrec2 = subnetlist; subrec2->next; subrec2 = subrec2->next)
  69     ;
  70
  71   subrec2->next = subrec;
  72   subrec->next = NULL;
  73   subrec->prev = subrec2;
  74 }
  75
  76 /* CRH!!! */
  77 /* ************************************************************************** ** * This will go away when we move to a "real" database back-end.
  78  * ************************************************************************** ** */
  79 int namelist_entry_compare( ubi_trItemPtr Item, ubi_trNodePtr Node )
  80   {
  81   struct name_record *NR = (struct name_record *)Node;
  82
  83   return( memcmp( Item, &(NR->name), sizeof(struct nmb_name) ) );
  84   } /* namelist_entry_compare */
  85 /* CRH!!! */
  86
  87 /****************************************************************************
  88   Create a subnet entry.
  89   ****************************************************************************/
  90
  91 static struct subnet_record *make_subnet(char *name, enum subnet_type type,
  92                                          struct in_addr myip, struct in_addr bcast_ip,
  93                                          struct in_addr mask_ip)
  94 {
  95   struct subnet_record *subrec = NULL;
  96   int nmb_sock, dgram_sock;
  97
  98   /* Check if we are creating a non broadcast subnet - if so don't create
  99      sockets.
 100    */
 101
 102   if(type != NORMAL_SUBNET)
 103   {
 104     nmb_sock = -1;
 105     dgram_sock = -1;
 106   }
 107   else
 108   {
 109     /*
 110      * Attempt to open the sockets on port 137/138 for this interface
 111      * and bind them.
 112      * Fail the subnet creation if this fails.
 113      */
 114
 115     if((nmb_sock = open_socket_in(SOCK_DGRAM, global_nmb_port,0, myip.s_addr)) == -1)
 116     {
 117       DEBUG(0,("make_subnet: Failed to open nmb socket on interface %s \
 118 for port %d. Error was %s\n", inet_ntoa(myip), global_nmb_port, strerror(errno)));
 119       return NULL;
 120     }
 121
 122     if((dgram_sock = open_socket_in(SOCK_DGRAM,DGRAM_PORT,3, myip.s_addr)) == -1)
 123     {
 124       DEBUG(0,("make_subnet: Failed to open dgram socket on interface %s \
 125 for port %d. Error was %s\n", inet_ntoa(myip), DGRAM_PORT, strerror(errno)));
 126       return NULL;
 127     }
 128
 129     /* Make sure we can broadcast from these sockets. */
 130     set_socket_options(nmb_sock,"SO_BROADCAST");
 131     set_socket_options(dgram_sock,"SO_BROADCAST");
 132
 133   }
 134
 135   subrec = (struct subnet_record *)malloc(sizeof(*subrec));
 136
 137   if (!subrec)
 138   {
 139     DEBUG(0,("make_subnet: malloc fail !\n"));
 140     close(nmb_sock);
 141     close(dgram_sock);
 142     return(NULL);
 143   }
 144
 145   bzero( (char *)subrec, sizeof(*subrec) );
 146   (void)ubi_trInitTree( subrec->namelist,
 147                         namelist_entry_compare,
 148                         ubi_trOVERWRITE );
 149
 150   if((subrec->subnet_name = strdup(name)) == NULL)
 151   {
 152     DEBUG(0,("make_subnet: malloc fail for subnet name !\n"));
 153     close(nmb_sock);
 154     close(dgram_sock);
 155     free((char *)subrec);
 156     return(NULL);
 157   }
 158
 159   DEBUG(2, ("making subnet name:%s ", name ));
 160   DEBUG(2, ("Broadcast address:%s ", inet_ntoa(bcast_ip)));
 161   DEBUG(2, ("Subnet mask:%s\n", inet_ntoa(mask_ip)));
 162
 163   subrec->namelist_changed = False;
 164   subrec->work_changed = False;
 165
 166   subrec->bcast_ip = bcast_ip;
 167   subrec->mask_ip  = mask_ip;
 168   subrec->myip = myip;
 169   subrec->type = type;
 170   subrec->nmb_sock = nmb_sock;
 171   subrec->dgram_sock = dgram_sock;
 172
 173   return subrec;
 174 }
 175
 176 /****************************************************************************
 177   Create subnet entries.
 178 **************************************************************************/
 179
 180 BOOL create_subnets(void)
 181 {
 182   int num_interfaces = iface_count();
 183   int i;
 184   struct in_addr unicast_ip;
 185
 186   if(num_interfaces == 0)
 187   {
 188     DEBUG(0,("create_subnets: No local interfaces !\n"));
 189     return False;
 190   }
 191
 192   /*
 193    * Create subnets from all the local interfaces and thread them onto
 194    * the linked list.
 195    */
 196
 197   for (i = 0 ; i < num_interfaces; i++)
 198   {
 199     struct subnet_record *subrec;
 200     struct interface *iface = get_interface(i);
 201
 202     if((subrec = make_subnet(inet_ntoa(iface->ip), NORMAL_SUBNET,
 203                  iface->ip, iface->bcast,iface->nmask)) == NULL)
 204       return False;
 205     add_subnet(subrec);
 206   }
 207
 208   /*
 209    * If we have been configured to use a WINS server, then try and
 210    * get the ip address of it here. If we are the WINS server then
 211    * set the unicast subnet address to be the first of our own real
 212    * addresses.
 213    */
 214
 215   if(*lp_wins_server())
 216   {
 217     struct in_addr real_wins_ip;
 218     real_wins_ip = *interpret_addr2(lp_wins_server());
 219
 220     if (!zero_ip(real_wins_ip))
 221     {
 222       unicast_ip = real_wins_ip;
 223     }
 224     else
 225     {
 226       /* The smb.conf's wins server parameter MUST be a host_name
 227          or an ip_address. */
 228       DEBUG(0,("invalid smb.conf parameter 'wins server'\n"));
 229       return False;
 230     }
 231   }
 232   else if(lp_we_are_a_wins_server())
 233   {
 234     /* Pick the first interface ip address as the WINS server ip. */
 235     unicast_ip = *iface_n_ip(0);
 236   }
 237   else
 238   {
 239     /* We should not be using a WINS server at all. Set the
 240       ip address of the subnet to be zero. */
 241     unicast_ip = ipzero;
 242   }
 243
 244   /*
 245    * Create the unicast and remote broadcast subnets.
 246    * Don't put these onto the linked list.
 247    * The ip address of the unicast subnet is set to be
 248    * the WINS server address, if it exists, or ipzero if not.
 249    */
 250
 251   unicast_subnet = make_subnet( "UNICAST_SUBNET", UNICAST_SUBNET,
 252                                  unicast_ip, unicast_ip, unicast_ip);
 253
 254   remote_broadcast_subnet = make_subnet( "REMOTE_BROADCAST_SUBNET",
 255                                          REMOTE_BROADCAST_SUBNET,
 256                                          ipzero, ipzero, ipzero);
 257
 258   if((unicast_subnet == NULL) || (remote_broadcast_subnet == NULL))
 259     return False;
 260
 261   /*
 262    * If we are WINS server, create the WINS_SERVER_SUBNET - don't put on
 263    * the linked list.
 264    */
 265
 266   if (lp_we_are_a_wins_server())
 267   {
 268     if((wins_server_subnet = make_subnet("WINS_SERVER_SUBNET",
 269                                        WINS_SERVER_SUBNET,
 270                                        ipzero, ipzero, ipzero)) == NULL)
 271       return False;
 272   }
 273
 274   return True;
 275 }
 276
 277 /*******************************************************************
 278 Function to tell us if we can use the unicast subnet.
 279 ******************************************************************/
 280
 281 BOOL we_are_a_wins_client(void)
 282 {
 283   static int cache_we_are_a_wins_client = -1;
 284
 285   if(cache_we_are_a_wins_client == -1)
 286     cache_we_are_a_wins_client = (ip_equal(ipzero, unicast_subnet->myip) ?
 287                                   False : True);
 288
 289   return cache_we_are_a_wins_client;
 290 }
 291
 292 /*******************************************************************
 293 Access function used by NEXT_SUBNET_INCLUDING_UNICAST
 294 ******************************************************************/
 295
 296 struct subnet_record *get_next_subnet_maybe_unicast(struct subnet_record *subrec)
 297 {
 298   if(subrec == unicast_subnet)
 299     return NULL;
 300   else if((subrec->next == NULL) && we_are_a_wins_client())
 301     return unicast_subnet;
 302   else
 303     return subrec->next;
 304 }
 305
 306 /*******************************************************************
 307  Access function used by retransmit_or_expire_response_records() in
 308  nmbd_packets.c. Patch from Andrey Alekseyev <fetch@muffin.arcadia.spb.ru>
 309  Needed when we need to enumerate all the broadcast, unicast and
 310  WINS subnets.
 311 ******************************************************************/
 312
 313 struct subnet_record *get_next_subnet_maybe_unicast_or_wins_server(struct subnet_record *subrec)
 314 {
 315   if(subrec == unicast_subnet)
 316   {
 317     if(wins_server_subnet)
 318       return wins_server_subnet;
 319     else
 320       return NULL;
 321   }
 322
 323   if(wins_server_subnet && subrec == wins_server_subnet)
 324     return NULL;
 325
 326   if((subrec->next == NULL) && we_are_a_wins_client())
 327     return unicast_subnet;
 328   else
 329     return subrec->next;
 330 }