816203ea3c9b620bab6fb9aa6948d4ded866c9a2
[kai/samba.git] / source3 / namedbsubnet.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-1997
6    
7    This program is free software; you can redistribute it and/or modify
8    it under the terms of the GNU General Public License as published by
9    the Free Software Foundation; either version 2 of the License, or
10    (at your option) any later version.
11    
12    This program is distributed in the hope that it will be useful,
13    but WITHOUT ANY WARRANTY; without even the implied warranty of
14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15    GNU General Public License for more details.
16    
17    You should have received a copy of the GNU General Public License
18    along with this program; if not, write to the Free Software
19    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
20    
21    Revision History:
22
23    14 jan 96: lkcl@pires.co.uk
24    added multiple workgroup domain master support
25
26    04 jul 96: lkcl@pires.co.uk
27    created module namedbsubnet containing subnet database functions
28
29 */
30
31 #include "includes.h"
32 #include "smb.h"
33
34 extern int ClientNMB;
35 extern int ClientDGRAM;
36 extern int global_nmb_port;
37
38 extern int DEBUGLEVEL;
39
40 extern struct in_addr wins_ip;
41 extern struct in_addr ipzero;
42
43 extern pstring myname;
44 extern fstring myworkgroup;
45 extern char **my_netbios_names;
46
47 BOOL updatedlists = True;
48 int updatecount = 0;
49
50 /* local interfaces structure */
51 extern struct interface *local_interfaces;
52
53 /* this is our domain/workgroup/server database */
54 struct subnet_record *subnetlist = NULL;
55
56 /* WINS subnet - keep this separate so enumeration code doesn't
57    run onto it by mistake. */
58 struct subnet_record *wins_client_subnet = NULL;
59
60 extern uint16 nb_type; /* samba's NetBIOS name type */
61
62 /****************************************************************************
63   add a domain into the list
64   **************************************************************************/
65 static void add_subnet(struct subnet_record *d)
66 {
67   struct subnet_record *d2;
68
69   if (!subnetlist)
70   {
71     subnetlist = d;
72     d->prev = NULL;
73     d->next = NULL;
74     return;
75   }
76
77   for (d2 = subnetlist; d2->next; d2 = d2->next);
78
79   d2->next = d;
80   d->next = NULL;
81   d->prev = d2;
82 }
83
84
85 /****************************************************************************
86   find a subnet in the subnetlist that a given IP address could
87   match - not including WINS. Returns NULL if no match.
88   **************************************************************************/
89 struct subnet_record *find_subnet(struct in_addr ip)
90 {   
91   struct subnet_record *d = NULL;
92   
93   /* search through subnet list for broadcast/netmask that matches
94      the source ip address. */
95   
96   for (d = FIRST_SUBNET; d; d = NEXT_SUBNET_EXCLUDING_WINS(d))
97     {
98       if (same_net(ip, d->bcast_ip, d->mask_ip))
99         break;
100     }
101   
102   return d;
103 }
104
105 /****************************************************************************
106   find a subnet in the subnetlist - if the subnet is not found
107   then return the WINS client subnet.
108   **************************************************************************/
109 struct subnet_record *find_subnet_all(struct in_addr ip)
110 {
111   struct subnet_record *d = find_subnet(ip);
112   if(!d)
113     return wins_client_subnet;
114   return d;
115 }
116
117 /****************************************************************************
118   create a subnet entry
119   ****************************************************************************/
120 static struct subnet_record *make_subnet(struct in_addr myip, struct in_addr bcast_ip, 
121                                          struct in_addr mask_ip, BOOL add)
122 {
123   struct subnet_record *d = NULL;
124   int nmb_sock, dgram_sock;
125
126   /* Check if we are creating the WINS subnet - if so don't create
127      sockets, use the ClientNMB and ClientDGRAM sockets instead.
128    */
129
130   if(ip_equal(bcast_ip, wins_ip))
131   {
132     nmb_sock = -1;
133     dgram_sock = -1;
134   }
135   else
136   {
137     /*
138      * Attempt to open the sockets on port 137/138 for this interface
139      * and bind them.
140      * Fail the subnet creation if this fails.
141      */
142
143     if((nmb_sock = open_socket_in(SOCK_DGRAM, global_nmb_port,0, myip.s_addr)) == -1)
144     {
145       DEBUG(0,("make_subnet: Failed to open nmb socket on interface %s \
146 for port %d. Error was %s\n", inet_ntoa(myip), global_nmb_port, strerror(errno)));
147       return NULL;
148     }
149
150     if((dgram_sock = open_socket_in(SOCK_DGRAM,DGRAM_PORT,3, myip.s_addr)) == -1)
151     {
152       DEBUG(0,("make_subnet: Failed to open dgram socket on interface %s \
153 for port %d. Error was %s\n", inet_ntoa(myip), DGRAM_PORT, strerror(errno)));
154       return NULL;
155     }
156
157     /* Make sure we can broadcast from these sockets. */
158     set_socket_options(nmb_sock,"SO_BROADCAST");
159     set_socket_options(dgram_sock,"SO_BROADCAST");
160
161   }
162
163   d = (struct subnet_record *)malloc(sizeof(*d));
164   
165   if (!d) 
166   {
167     DEBUG(0,("make_subnet: malloc fail !\n"));
168     close(nmb_sock);
169     close(dgram_sock);
170     return(NULL);
171   }
172   
173   bzero((char *)d,sizeof(*d));
174   
175   DEBUG(4, ("making subnet %s ", inet_ntoa(bcast_ip)));
176   DEBUG(4, ("%s\n", inet_ntoa(mask_ip)));
177   
178   d->bcast_ip = bcast_ip;
179   d->mask_ip  = mask_ip;
180   d->myip = myip;
181   d->nmb_sock = nmb_sock;
182   d->dgram_sock = dgram_sock;
183   d->workgrouplist = NULL;
184   
185   if(add)
186     add_subnet(d);
187   
188   return d;
189 }
190
191 /****************************************************************************
192   add a domain entry. creates a workgroup, if necessary, and adds the domain
193   to the named a workgroup.
194   ****************************************************************************/
195 static struct subnet_record *add_subnet_entry(struct in_addr myip,
196                                        struct in_addr bcast_ip, 
197                                        struct in_addr mask_ip, char *name, 
198                                        BOOL create_subnets, BOOL add)
199 {
200   struct subnet_record *d = NULL;
201
202   if (zero_ip(bcast_ip)) 
203     bcast_ip = *iface_bcast(bcast_ip);
204   
205   /* Note that we should also add into the WINS subnet as add_subnet_entry
206     should be called to add NetBIOS names and server entries on all
207     interfaces, including the WINS interface
208    */
209
210   if(create_subnets == True)
211   {
212     /* Create new subnets. */
213     if((d = make_subnet(myip, bcast_ip, mask_ip, add)) == NULL)
214     {
215       DEBUG(0,("add_subnet_entry: Unable to create subnet %s\n",
216                inet_ntoa(bcast_ip) ));
217       return NULL;
218     }
219     return d;
220   }
221   if(ip_equal(bcast_ip, wins_ip))
222     return wins_client_subnet;
223   return find_subnet(bcast_ip);
224 }
225
226 /****************************************************************************
227  Add a workgroup into a subnet, and if it's our primary workgroup,
228  add the required names to it.
229 **************************************************************************/
230
231 void add_workgroup_to_subnet( struct subnet_record *d, char *group)
232 {
233   struct work_record *w = NULL;
234
235   DEBUG(5,("add_workgroup_to_subnet: Adding workgroup %s to subnet %s\n",
236             group, inet_ntoa(d->bcast_ip)));
237
238   /* This next statement creates the workgroup struct if it doesn't
239      already exist. 
240    */
241   if((w = find_workgroupstruct(d, group, True)) == NULL)
242   {
243     DEBUG(0,("add_workgroup_to_subnet: Unable to add workgroup %s to subnet %s\n",
244               group, inet_ntoa(d->bcast_ip) ));
245     return;
246   }
247
248   /* add WORKGROUP(00) entries into name database
249      or register with WINS server, if it's our workgroup.
250    */
251   if (strequal(myworkgroup, group))
252   {
253     int n;
254
255     add_my_name_entry(d,group,0x0 ,nb_type|NB_ACTIVE|NB_GROUP);
256
257     /* Register the WORKGROUP<0x1e> name. */
258     add_my_name_entry(d,group,0x1e,nb_type|NB_ACTIVE|NB_GROUP);
259
260     /* Add all our server names to the workgroup list. We remove any
261        browser or logon server flags from all but the primary name.
262      */
263     for( n = 0; my_netbios_names[n]; n++)
264     {    
265       char *name = my_netbios_names[n];
266       int stype = w->ServerType;
267
268       if(!strequal(myname, name))
269           stype &= ~(SV_TYPE_MASTER_BROWSER|SV_TYPE_POTENTIAL_BROWSER|
270                      SV_TYPE_DOMAIN_MASTER|SV_TYPE_DOMAIN_MEMBER);
271
272       add_server_entry(d,w,name,stype|SV_TYPE_LOCAL_LIST_ONLY,0,
273                 lp_serverstring(),True);
274       DEBUG(3,("add_workgroup_to_subnet: Added server name entry %s \
275 to subnet %s\n", name, inet_ntoa(d->bcast_ip)));
276     }
277   }
278 }
279
280 /****************************************************************************
281   create subnet / workgroup / server entries
282      
283   - add or create the subnet lists
284   - add or create the workgroup entries in each subnet entry
285   - register appropriate NetBIOS names for the workgroup entries
286      
287 **************************************************************************/
288 void add_my_subnets(char *group)
289 {    
290   static BOOL create_subnets = True;
291   struct subnet_record *d = NULL;
292   struct interface *i = NULL;
293
294   if (*group == '*') return;
295
296   /* Create subnets from all the local interfaces and thread them onto
297      the linked list. 
298    */
299   for (i = local_interfaces; i; i = i->next)
300   {
301     add_subnet_entry(i->ip, i->bcast,i->nmask,group, create_subnets, True);
302   }
303
304   /* If we are using WINS, then we must add the workgroup to the WINS
305      subnet. This is used as a place to keep collated server lists.
306    */
307
308   /* Create the WINS subnet if we are using WINS - but don't thread it
309      onto the linked subnet list. 
310    */    
311   if (lp_wins_support() || lp_wins_server())
312   {
313     struct in_addr wins_nmask = ipzero;
314     wins_client_subnet = add_subnet_entry(ipzero, wins_ip, wins_nmask, group, create_subnets, False);
315   }
316
317   /* Ensure we only create the subnets once. */
318   create_subnets = False;
319
320   /* Now we have created all the subnets - we can add the names
321      that make us a client member in the workgroup.
322    */
323   for (d = FIRST_SUBNET; d; d = NEXT_SUBNET_INCLUDING_WINS(d))
324     add_workgroup_to_subnet(d, group);
325 }
326
327 /*******************************************************************
328   write out browse.dat
329   ******************************************************************/
330 void write_browse_list(time_t t)
331 {
332   struct subnet_record *d;
333   pstring fname,fnamenew;
334   FILE *f;
335
336   static time_t lasttime = 0;
337
338   if (!lasttime) lasttime = t;
339   if (!updatedlists || t - lasttime < 5) return;
340   
341   lasttime = t;
342   updatedlists = False;
343   updatecount++;
344   
345   dump_names();
346   dump_workgroups();
347   
348   pstrcpy(fname,lp_lockdir());
349   trim_string(fname,NULL,"/");
350   strcat(fname,"/");
351   strcat(fname,SERVER_LIST);
352   pstrcpy(fnamenew,fname);
353   strcat(fnamenew,".");
354   
355   f = fopen(fnamenew,"w");
356   
357   if (!f)
358     {
359       DEBUG(4,("Can't open %s - %s\n",fnamenew,strerror(errno)));
360       return;
361     }
362   
363   for (d = FIRST_SUBNET; d ; d = NEXT_SUBNET_INCLUDING_WINS(d))
364     {
365       struct work_record *work;
366       for (work = d->workgrouplist; work ; work = work->next)
367         {
368           struct server_record *s;
369           for (s = work->serverlist; s ; s = s->next)
370             {
371               fstring tmp;
372               
373               /* don't list domains I don't have a master for */
374               if ((s->serv.type & SV_TYPE_DOMAIN_ENUM) && !s->serv.comment[0])
375                 {
376                   continue;
377                 }
378               
379               /* output server details, plus what workgroup/domain
380                  they're in. without the domain information, the
381                  combined list of all servers in all workgroups gets
382                  sent to anyone asking about any workgroup! */
383               
384               sprintf(tmp, "\"%s\"", s->serv.name);
385               fprintf(f, "%-25s ", tmp);
386               fprintf(f, "%08x ", s->serv.type);
387               sprintf(tmp, "\"%s\" ", s->serv.comment);
388               fprintf(f, "%-30s", tmp);
389               fprintf(f, "\"%s\"\n", work->work_group);
390             }
391         }
392     }
393   
394   fclose(f);
395   unlink(fname);
396   chmod(fnamenew,0644);
397   rename(fnamenew,fname);   
398   DEBUG(3,("Wrote browse list %s\n",fname));
399 }
400