51b2519d837afca4b54c1ba9d9efcddef7d77496
[samba.git] / source3 / nmbd / nmbd_workgroupdb.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 */
24
25 #include "includes.h"
26 #include "smb.h"
27
28 extern int ClientNMB;
29
30 extern int DEBUGLEVEL;
31
32 extern pstring global_myname;
33 extern fstring global_myworkgroup;
34 extern char **my_netbios_names;
35 extern uint16 samba_nb_type;
36 extern struct in_addr ipzero;
37
38 int workgroup_count = 0; /* unique index key: one for each workgroup */
39
40 /****************************************************************************
41   Add a workgroup into the list.
42   **************************************************************************/
43
44 static void add_workgroup(struct subnet_record *subrec, struct work_record *work)
45 {
46   struct work_record *w2;
47
48   work->subnet = subrec;
49
50   if (!subrec->workgrouplist)
51   {
52     subrec->workgrouplist = work;
53     work->prev = NULL;
54     work->next = NULL;
55     return;
56   }
57   
58   for (w2 = subrec->workgrouplist; w2->next; w2 = w2->next)
59     ;
60   
61   w2->next = work;
62   work->next = NULL;
63   work->prev = w2;
64
65   subrec->work_changed = True;
66 }
67
68 /****************************************************************************
69   Create an empty workgroup.
70   **************************************************************************/
71
72 static struct work_record *create_workgroup(char *name, int ttl)
73 {
74   struct work_record *work;
75   struct subnet_record *subrec;
76   int t = -1;
77   
78   if((work = (struct work_record *)malloc(sizeof(*work))) == NULL)
79   {
80     DEBUG(0,("create_workgroup: malloc fail !\n"));
81     return NULL;
82   }
83   bzero((char *)work, sizeof(*work));
84  
85   StrnCpy(work->work_group,name,sizeof(work->work_group)-1);
86   work->serverlist = NULL;
87   
88   work->RunningElection = False;
89   work->ElectionCount = 0;
90   work->announce_interval = 0;
91   work->needelection = False;
92   work->needannounce = True;
93   work->lastannounce_time = time(NULL);
94   work->mst_state = lp_local_master() ? MST_POTENTIAL : MST_NONE;
95   work->dom_state = DOMAIN_NONE;
96   work->log_state = LOGON_NONE;
97   
98   work->death_time = (ttl != PERMANENT_TTL) ? time(NULL)+(ttl*3) : PERMANENT_TTL;
99
100   /* Make sure all token representations of workgroups are unique. */
101   
102   for (subrec = FIRST_SUBNET; subrec && (t == -1); 
103            subrec = NEXT_SUBNET_INCLUDING_UNICAST(subrec))
104   {
105     struct work_record *w;
106     for (w = subrec->workgrouplist; w && t == -1; w = w->next)
107     {
108       if (strequal(w->work_group, work->work_group))
109         t = w->token;
110     }
111   }
112   
113   if (t == -1)
114     work->token = ++workgroup_count;
115   else
116     work->token = t;
117   
118   /* No known local master browser as yet. */
119   *work->local_master_browser_name = '\0';
120
121   /* No known domain master browser as yet. */
122   *work->dmb_name.name = '\0';
123   putip((char *)&work->dmb_addr, &ipzero);
124
125   /* WfWg  uses 01040b01 */
126   /* Win95 uses 01041501 */
127   /* NTAS  uses ???????? */
128   work->ElectionCriterion  = (MAINTAIN_LIST)|(BROWSER_ELECTION_VERSION<<8); 
129   work->ElectionCriterion |= (lp_os_level() << 24);
130   if (lp_domain_master())
131     work->ElectionCriterion |= 0x80;
132   
133   return work;
134 }
135
136 /*******************************************************************
137   Remove a workgroup.
138   ******************************************************************/
139
140 static struct work_record *remove_workgroup_from_subnet(struct subnet_record *subrec, 
141                                      struct work_record *work)
142 {
143   struct work_record *ret_work = NULL;
144   
145   DEBUG(3,("remove_workgroup: Removing workgroup %s\n", work->work_group));
146   
147   ret_work = work->next;
148
149   remove_all_servers(work);
150   
151   if (!work->serverlist)
152   {
153     if (work->prev)
154       work->prev->next = work->next;
155     if (work->next)
156       work->next->prev = work->prev;
157   
158     if (subrec->workgrouplist == work)
159       subrec->workgrouplist = work->next; 
160   
161     free((char *)work);
162   }
163   
164   subrec->work_changed = True;
165
166   return ret_work;
167 }
168
169
170 /****************************************************************************
171   Find a workgroup in the workgroup list of a subnet.
172   **************************************************************************/
173
174 struct work_record *find_workgroup_on_subnet(struct subnet_record *subrec, 
175                                              fstring name)
176 {
177   struct work_record *ret;
178   
179   DEBUG(4, ("find_workgroup_on_subnet: workgroup search for %s on subnet %s: ",
180             name, subrec->subnet_name));
181   
182   for (ret = subrec->workgrouplist; ret; ret = ret->next)
183   {
184     if (!strcmp(ret->work_group,name))
185     {
186       DEBUGADD(4, ("found.\n"));
187       return(ret);
188     }
189   }
190   DEBUGADD(4, ("not found.\n"));
191   return NULL;
192 }
193
194 /****************************************************************************
195   Create a workgroup in the workgroup list of the subnet.
196   **************************************************************************/
197
198 struct work_record *create_workgroup_on_subnet(struct subnet_record *subrec,
199                                                fstring name, int ttl)
200 {
201   struct work_record *work = NULL;
202
203   DEBUG(4,("create_workgroup_on_subnet: creating group %s on subnet %s\n",
204            name, subrec->subnet_name));
205   
206   if ((work = create_workgroup(name, ttl)))
207   {
208     add_workgroup(subrec, work);
209
210     subrec->work_changed = True;
211
212     return(work);
213   }
214
215   return NULL;
216 }
217
218 /****************************************************************************
219   Update a workgroup ttl.
220   **************************************************************************/
221
222 void update_workgroup_ttl(struct work_record *work, int ttl)
223 {
224   if(work->death_time != PERMANENT_TTL)
225     work->death_time = time(NULL)+(ttl*3);
226   work->subnet->work_changed = True;
227 }
228
229 /****************************************************************************
230  Fail function called if we cannot register the WORKGROUP<0> and
231  WORKGROUP<1e> names on the net.
232 **************************************************************************/
233      
234 static void fail_register(struct subnet_record *subrec, struct response_record *rrec,
235                           struct nmb_name *nmbname)
236 {  
237   DEBUG(0,("fail_register: Failed to register name %s on subnet %s.\n",
238             namestr(nmbname), subrec->subnet_name));
239 }  
240
241 /****************************************************************************
242  If the workgroup is our primary workgroup, add the required names to it.
243 **************************************************************************/
244
245 void initiate_myworkgroup_startup(struct subnet_record *subrec, struct work_record *work)
246 {
247   int i;
248
249   if(!strequal(global_myworkgroup, work->work_group))
250     return;
251
252   /* If this is a broadcast subnet then start elections on it
253      if we are so configured. */
254
255   if ((subrec != unicast_subnet) && (subrec != remote_broadcast_subnet) &&
256       (subrec != wins_server_subnet) && lp_preferred_master() &&
257       lp_local_master())
258   {
259     DEBUG(3, ("initiate_myworkgroup_startup: preferred master startup for \
260 workgroup %s on subnet %s\n", work->work_group, subrec->subnet_name));
261     work->needelection = True;
262     work->ElectionCriterion |= (1<<3);
263   }  
264   
265   /* Register the WORKGROUP<0> and WORKGROUP<1e> names on the network. */
266
267   register_name(subrec,global_myworkgroup,0x0,samba_nb_type|NB_GROUP,
268                 NULL,
269                 fail_register,NULL);
270      
271   register_name(subrec,global_myworkgroup,0x1e,samba_nb_type|NB_GROUP,
272                 NULL,
273                 fail_register,NULL);
274      
275   for( i = 0; my_netbios_names[i]; i++)
276   {
277     char *name = my_netbios_names[i];
278     int stype = lp_default_server_announce() | (lp_local_master() ?
279                         SV_TYPE_POTENTIAL_BROWSER : 0 );
280    
281     if(!strequal(global_myname, name))
282         stype &= ~(SV_TYPE_MASTER_BROWSER|SV_TYPE_POTENTIAL_BROWSER|
283                    SV_TYPE_DOMAIN_MASTER|SV_TYPE_DOMAIN_MEMBER);
284    
285     create_server_on_workgroup(work,name,stype|SV_TYPE_LOCAL_LIST_ONLY,
286                 PERMANENT_TTL, lp_serverstring());
287     DEBUG(3,("initiate_myworkgroup_startup: Added server name entry %s \
288 on subnet %s\n", name, subrec->subnet_name));
289   }
290
291
292 /****************************************************************************
293   Dump a copy of the workgroup database into the log file.
294   **************************************************************************/
295
296 void dump_workgroups(BOOL force_write)
297 {
298   struct subnet_record *subrec;
299   int debuglevel = force_write ? 0 : 4;
300  
301   for (subrec = FIRST_SUBNET; subrec; subrec = NEXT_SUBNET_INCLUDING_UNICAST(subrec))
302   {
303     if (subrec->workgrouplist)
304     {
305       struct work_record *work;
306           
307       DEBUG(debuglevel,("dump_workgroups: dump workgroup on subnet %15s: ", subrec->subnet_name));
308       DEBUG(debuglevel,(" netmask=%15s:\n", inet_ntoa(subrec->mask_ip)));
309           
310       for (work = subrec->workgrouplist; work; work = work->next)
311       {
312         DEBUG(debuglevel,("\t%s(%d) current master browser = %s\n", work->work_group, 
313                   work->token, 
314                   *work->local_master_browser_name ? work->local_master_browser_name : "UNKNOWN" ));
315         if (work->serverlist)
316         {
317           struct server_record *servrec;                  
318           for (servrec = work->serverlist; servrec; servrec = servrec->next)
319           {
320             DEBUG(debuglevel,("\t\t%s %8x (%s)\n",
321                   servrec->serv.name, servrec->serv.type, servrec->serv.comment));
322           }
323         }
324       }
325     }
326   }
327 }
328
329 /****************************************************************************
330   Expire any dead servers on all workgroups. If the workgroup has expired
331   remove it.
332   **************************************************************************/
333
334 void expire_workgroups_and_servers(time_t t)
335 {
336   struct subnet_record *subrec;
337    
338   for (subrec = FIRST_SUBNET; subrec; subrec = NEXT_SUBNET_INCLUDING_UNICAST(subrec))
339   {
340     struct work_record *work;
341     struct work_record *nextwork;
342
343     for (work = subrec->workgrouplist; work; work = nextwork)
344     {
345       nextwork = work->next;
346       expire_servers(work, t);
347
348       if ((work->serverlist == NULL) && (work->death_time != PERMANENT_TTL) && 
349                      ((t == -1) || (work->death_time < t)))
350       {
351         DEBUG(3,("expire_workgroups_and_servers: Removing timed out workgroup %s\n",
352                   work->work_group));
353         remove_workgroup_from_subnet(subrec, work);
354       }
355     }
356   }
357 }