s3:rpcclient: make use of dcerpc_binding_get_string_option("host")
[sfrench/samba-autobuild/.git] / source3 / nmbd / nmbd_become_lmb.c
1 /* 
2    Unix SMB/CIFS implementation.
3    NBT netbios routines and daemon - version 2
4    Copyright (C) Andrew Tridgell 1994-1998
5    Copyright (C) Luke Kenneth Casson Leighton 1994-1998
6    Copyright (C) Jeremy Allison 1994-2003
7    
8    This program is free software; you can redistribute it and/or modify
9    it under the terms of the GNU General Public License as published by
10    the Free Software Foundation; either version 3 of the License, or
11    (at your option) any later version.
12    
13    This program is distributed in the hope that it will be useful,
14    but WITHOUT ANY WARRANTY; without even the implied warranty of
15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16    GNU General Public License for more details.
17    
18    You should have received a copy of the GNU General Public License
19    along with this program.  If not, see <http://www.gnu.org/licenses/>.
20    
21 */
22
23 #include "includes.h"
24 #include "nmbd/nmbd.h"
25 #include "../librpc/gen_ndr/svcctl.h"
26
27 extern uint16 samba_nb_type; /* Samba's NetBIOS name type. */
28
29 /*******************************************************************
30  Utility function to add a name to the unicast subnet, or add in
31  our IP address if it already exists.
32 ******************************************************************/
33
34 void insert_permanent_name_into_unicast( struct subnet_record *subrec, 
35                                                 struct nmb_name *nmbname, uint16 nb_type )
36 {
37         unstring name;
38         struct name_record *namerec;
39
40         if((namerec = find_name_on_subnet(unicast_subnet, nmbname, FIND_SELF_NAME)) == NULL) {
41                 pull_ascii_nstring(name, sizeof(name), nmbname->name);
42                 /* The name needs to be created on the unicast subnet. */
43                 (void)add_name_to_subnet( unicast_subnet, name,
44                                 nmbname->name_type, nb_type,
45                                 PERMANENT_TTL, PERMANENT_NAME, 1, &subrec->myip);
46         } else {
47                 /* The name already exists on the unicast subnet. Add our local
48                 IP for the given broadcast subnet to the name. */
49                 add_ip_to_name_record( namerec, subrec->myip);
50         }
51 }
52
53 /*******************************************************************
54  Utility function to remove a name from the unicast subnet.
55 ******************************************************************/
56
57 static void remove_permanent_name_from_unicast( struct subnet_record *subrec,
58                                                 struct nmb_name *nmbname )
59 {
60         struct name_record *namerec;
61
62         if((namerec = find_name_on_subnet(unicast_subnet, nmbname, FIND_SELF_NAME)) != NULL) {
63                 /* Remove this broadcast subnet IP address from the name. */
64                 remove_ip_from_name_record( namerec, subrec->myip);
65                 if(namerec->data.num_ips == 0)
66                         remove_name_from_namelist( unicast_subnet, namerec);
67         }
68 }
69
70 /*******************************************************************
71  Utility function always called to set our workgroup and server
72  state back to potential browser, or none.
73 ******************************************************************/
74
75 static void reset_workgroup_state( struct subnet_record *subrec, const char *workgroup_name,
76                                    bool force_new_election )
77 {
78         struct work_record *work;
79         struct server_record *servrec;
80         struct nmb_name nmbname;
81
82         if((work = find_workgroup_on_subnet( subrec, workgroup_name)) == NULL) {
83                 DEBUG(0,("reset_workgroup_state: Error - cannot find workgroup %s on \
84 subnet %s.\n", workgroup_name, subrec->subnet_name ));
85                 return;
86         }
87
88         if((servrec = find_server_in_workgroup( work, lp_netbios_name())) == NULL) {
89                 DEBUG(0,("reset_workgroup_state: Error - cannot find server %s \
90 in workgroup %s on subnet %s\n",
91                         lp_netbios_name(), work->work_group, subrec->subnet_name));
92                 work->mst_state = lp_local_master() ? MST_POTENTIAL : MST_NONE;
93                 return;
94         }
95
96         /* Update our server status - remove any master flag and replace
97                 it with the potential browser flag. */
98         servrec->serv.type &= ~SV_TYPE_MASTER_BROWSER;
99         servrec->serv.type |= (lp_local_master() ? SV_TYPE_POTENTIAL_BROWSER : 0);
100
101         /* Tell the namelist writer to write out a change. */
102         subrec->work_changed = True;
103
104         /* Reset our election flags. */
105         work->ElectionCriterion &= ~0x4;
106
107         work->mst_state = lp_local_master() ? MST_POTENTIAL : MST_NONE;
108
109         /* Forget who the local master browser was for
110                 this workgroup. */
111
112         set_workgroup_local_master_browser_name( work, "");
113
114         /*
115          * Ensure the IP address of this subnet is not registered as one
116          * of the IP addresses of the WORKGROUP<1d> name on the unicast
117          * subnet. This undoes what we did below when we became a local
118          * master browser.
119          */
120
121         make_nmb_name(&nmbname, work->work_group, 0x1d);
122
123         remove_permanent_name_from_unicast( subrec, &nmbname);
124
125         if(force_new_election)
126                 work->needelection = True;
127 }
128
129 /*******************************************************************
130   Unbecome the local master browser name release success function.
131 ******************************************************************/
132
133 static void unbecome_local_master_success(struct subnet_record *subrec,
134                              struct userdata_struct *userdata,
135                              struct nmb_name *released_name,
136                              struct in_addr released_ip)
137
138         bool force_new_election = False;
139         unstring relname;
140
141         memcpy((char *)&force_new_election, userdata->data, sizeof(bool));
142
143         DEBUG(3,("unbecome_local_master_success: released name %s.\n",
144                 nmb_namestr(released_name)));
145
146         /* Now reset the workgroup and server state. */
147         pull_ascii_nstring(relname, sizeof(relname), released_name->name);
148         reset_workgroup_state( subrec, relname, force_new_election );
149
150         if( DEBUGLVL( 0 ) ) {
151                 dbgtext( "*****\n\n" );
152                 dbgtext( "Samba name server %s ", lp_netbios_name() );
153                 dbgtext( "has stopped being a local master browser " );
154                 dbgtext( "for workgroup %s ", relname );
155                 dbgtext( "on subnet %s\n\n*****\n", subrec->subnet_name );
156         }
157
158 }
159
160 /*******************************************************************
161   Unbecome the local master browser name release fail function.
162 ******************************************************************/
163
164 static void unbecome_local_master_fail(struct subnet_record *subrec, struct response_record *rrec,
165                        struct nmb_name *fail_name)
166 {
167         struct name_record *namerec;
168         struct userdata_struct *userdata = rrec->userdata;
169         bool force_new_election = False;
170         unstring failname;
171
172         memcpy((char *)&force_new_election, userdata->data, sizeof(bool));
173
174         DEBUG(0,("unbecome_local_master_fail: failed to release name %s. \
175 Removing from namelist anyway.\n", nmb_namestr(fail_name)));
176
177         /* Do it anyway. */
178         namerec = find_name_on_subnet(subrec, fail_name, FIND_SELF_NAME);
179         if(namerec)
180                 remove_name_from_namelist(subrec, namerec);
181
182         /* Now reset the workgroup and server state. */
183         pull_ascii_nstring(failname, sizeof(failname), fail_name->name);
184         reset_workgroup_state( subrec, failname, force_new_election );
185
186         if( DEBUGLVL( 0 ) ) {
187                 dbgtext( "*****\n\n" );
188                 dbgtext( "Samba name server %s ", lp_netbios_name() );
189                 dbgtext( "has stopped being a local master browser " );
190                 dbgtext( "for workgroup %s ", failname );
191                 dbgtext( "on subnet %s\n\n*****\n", subrec->subnet_name );
192         }
193 }
194
195 /*******************************************************************
196  Utility function to remove the WORKGROUP<1d> name.
197 ******************************************************************/
198
199 static void release_1d_name( struct subnet_record *subrec, const char *workgroup_name,
200                              bool force_new_election)
201 {
202         struct nmb_name nmbname;
203         struct name_record *namerec;
204
205         make_nmb_name(&nmbname, workgroup_name, 0x1d);
206         if((namerec = find_name_on_subnet( subrec, &nmbname, FIND_SELF_NAME))!=NULL) {
207                 struct userdata_struct *userdata;
208                 size_t size = sizeof(struct userdata_struct) + sizeof(bool);
209
210                 if((userdata = (struct userdata_struct *)SMB_MALLOC(size)) == NULL) {
211                         DEBUG(0,("release_1d_name: malloc fail.\n"));
212                         return;
213                 }
214
215                 userdata->copy_fn = NULL;
216                 userdata->free_fn = NULL;
217                 userdata->userdata_len = sizeof(bool);
218                 memcpy((char *)userdata->data, &force_new_election, sizeof(bool));
219
220                 release_name(subrec, namerec,
221                         unbecome_local_master_success,
222                         unbecome_local_master_fail,
223                         userdata);
224
225                 zero_free(userdata, size);
226         }
227 }
228
229 /*******************************************************************
230  Unbecome the local master browser MSBROWSE name release success function.
231 ******************************************************************/
232
233 static void release_msbrowse_name_success(struct subnet_record *subrec,
234                       struct userdata_struct *userdata,
235                       struct nmb_name *released_name,
236                       struct in_addr released_ip)
237 {
238         DEBUG(4,("release_msbrowse_name_success: Released name %s on subnet %s\n.",
239                 nmb_namestr(released_name), subrec->subnet_name ));
240
241         /* Remove the permanent MSBROWSE name added into the unicast subnet. */
242         remove_permanent_name_from_unicast( subrec, released_name);
243 }
244
245 /*******************************************************************
246  Unbecome the local master browser MSBROWSE name release fail function.
247 ******************************************************************/
248
249 static void release_msbrowse_name_fail( struct subnet_record *subrec, 
250                        struct response_record *rrec,
251                        struct nmb_name *fail_name)
252 {
253         struct name_record *namerec;
254
255         DEBUG(4,("release_msbrowse_name_fail: Failed to release name %s on subnet %s\n.",
256                 nmb_namestr(fail_name), subrec->subnet_name ));
257
258         /* Release the name anyway. */
259         namerec = find_name_on_subnet(subrec, fail_name, FIND_SELF_NAME);
260         if(namerec)
261                 remove_name_from_namelist(subrec, namerec);
262
263         /* Remove the permanent MSBROWSE name added into the unicast subnet. */
264         remove_permanent_name_from_unicast( subrec, fail_name);
265 }
266
267 /*******************************************************************
268   Unbecome the local master browser. If force_new_election is true, restart
269   the election process after we've unbecome the local master.
270 ******************************************************************/
271
272 void unbecome_local_master_browser(struct subnet_record *subrec, struct work_record *work,
273                                    bool force_new_election)
274 {
275         struct name_record *namerec;
276         struct nmb_name nmbname;
277
278   /* Sanity check. */
279
280         DEBUG(2,("unbecome_local_master_browser: unbecoming local master for workgroup %s \
281 on subnet %s\n",work->work_group, subrec->subnet_name));
282   
283         if(find_server_in_workgroup( work, lp_netbios_name()) == NULL) {
284                 DEBUG(0,("unbecome_local_master_browser: Error - cannot find server %s \
285 in workgroup %s on subnet %s\n",
286                         lp_netbios_name(), work->work_group, subrec->subnet_name));
287                         work->mst_state = lp_local_master() ? MST_POTENTIAL : MST_NONE;
288                 return;
289         }
290   
291         /* Set the state to unbecoming. */
292         work->mst_state = MST_UNBECOMING_MASTER;
293
294         /*
295          * Release the WORKGROUP<1d> name asap to allow another machine to
296          * claim it.
297          */
298
299         release_1d_name( subrec, work->work_group, force_new_election);
300
301         /* Deregister any browser names we may have. */
302         make_nmb_name(&nmbname, MSBROWSE, 0x1);
303         if((namerec = find_name_on_subnet( subrec, &nmbname, FIND_SELF_NAME))!=NULL) {
304                 release_name(subrec, namerec,
305                         release_msbrowse_name_success,
306                         release_msbrowse_name_fail,
307                         NULL);
308         }
309
310         /*
311          * Ensure we have sent and processed these release packets
312          * before returning - we don't want to process any election
313          * packets before dealing with the 1d release.
314          */
315
316         retransmit_or_expire_response_records(time(NULL));
317 }
318
319 /****************************************************************************
320   Success in registering the WORKGROUP<1d> name.
321   We are now *really* a local master browser.
322   ****************************************************************************/
323
324 static void become_local_master_stage2(struct subnet_record *subrec,
325                                         struct userdata_struct *userdata,
326                                         struct nmb_name *registered_name,
327                                         uint16 nb_flags,
328                                         int ttl, struct in_addr registered_ip)
329 {
330         int i = 0;
331         struct server_record *sl;
332         struct work_record *work;
333         struct server_record *servrec;
334         unstring regname;
335
336         pull_ascii_nstring(regname, sizeof(regname), registered_name->name);
337         work = find_workgroup_on_subnet( subrec, regname);
338
339         if(!work) {
340                 DEBUG(0,("become_local_master_stage2: Error - cannot find \
341 workgroup %s on subnet %s\n", regname, subrec->subnet_name));
342                 return;
343         }
344
345         if((servrec = find_server_in_workgroup( work, lp_netbios_name())) == NULL) {
346                 DEBUG(0,("become_local_master_stage2: Error - cannot find server %s \
347 in workgroup %s on subnet %s\n",
348                         lp_netbios_name(), regname, subrec->subnet_name));
349                         work->mst_state = lp_local_master() ? MST_POTENTIAL : MST_NONE;
350                 return;
351         }
352   
353         DEBUG(3,("become_local_master_stage2: registered as master browser for workgroup %s \
354 on subnet %s\n", work->work_group, subrec->subnet_name));
355
356         work->mst_state = MST_BROWSER; /* registering WORKGROUP(1d) succeeded */
357
358         /* update our server status */
359         servrec->serv.type |= SV_TYPE_MASTER_BROWSER;
360         servrec->serv.type &= ~SV_TYPE_POTENTIAL_BROWSER;
361
362         /* Tell the namelist writer to write out a change. */
363         subrec->work_changed = True;
364
365         /* Add this name to the workgroup as local master browser. */
366         set_workgroup_local_master_browser_name( work, lp_netbios_name());
367
368         /* Count the number of servers we have on our list. If it's
369                 less than 10 (just a heuristic) request the servers
370                 to announce themselves.
371         */
372         for( sl = work->serverlist; sl != NULL; sl = sl->next)
373                 i++;
374
375         if (i < 10) {
376                 /* Ask all servers on our local net to announce to us. */
377                 broadcast_announce_request(subrec, work);
378         }
379
380         /*
381          * Now we are a local master on a broadcast subnet, we need to add
382          * the WORKGROUP<1d> name to the unicast subnet so that we can answer
383          * unicast requests sent to this name. We can create this name directly on
384          * the unicast subnet as a WINS server always returns true when registering
385          * this name, and discards the registration. We use the number of IP
386          * addresses registered to this name as a reference count, as we
387          * remove this broadcast subnet IP address from it when we stop becoming a local
388          * master browser for this broadcast subnet.
389          */
390
391         insert_permanent_name_into_unicast( subrec, registered_name, nb_flags);
392
393         /* Reset the announce master browser timer so that we try and tell a domain
394                 master browser as soon as possible that we are a local master browser. */
395         reset_announce_timer();
396
397         if( DEBUGLVL( 0 ) ) {
398                 dbgtext( "*****\n\n" );
399                 dbgtext( "Samba name server %s ", lp_netbios_name() );
400                 dbgtext( "is now a local master browser " );
401                 dbgtext( "for workgroup %s ", work->work_group );
402                 dbgtext( "on subnet %s\n\n*****\n", subrec->subnet_name );
403         }
404 }
405
406 /****************************************************************************
407   Failed to register the WORKGROUP<1d> name.
408   ****************************************************************************/
409
410 static void become_local_master_fail2(struct subnet_record *subrec,
411                                       struct response_record *rrec,
412                                       struct nmb_name *fail_name)
413 {
414         unstring failname;
415         struct work_record *work;
416
417         DEBUG(0,("become_local_master_fail2: failed to register name %s on subnet %s. \
418 Failed to become a local master browser.\n", nmb_namestr(fail_name), subrec->subnet_name));
419
420         pull_ascii_nstring(failname, sizeof(failname), fail_name->name);
421         work = find_workgroup_on_subnet( subrec, failname);
422
423         if(!work) {
424                 DEBUG(0,("become_local_master_fail2: Error - cannot find \
425 workgroup %s on subnet %s\n", failname, subrec->subnet_name));
426                 return;
427         }
428
429         /* Roll back all the way by calling unbecome_local_master_browser(). */
430         unbecome_local_master_browser(subrec, work, False);
431 }
432
433 /****************************************************************************
434   Success in registering the MSBROWSE name.
435   ****************************************************************************/
436
437 static void become_local_master_stage1(struct subnet_record *subrec,
438                                         struct userdata_struct *userdata,
439                                         struct nmb_name *registered_name,
440                                         uint16 nb_flags,
441                                         int ttl, struct in_addr registered_ip)
442 {
443         char *work_name = userdata->data;
444         struct work_record *work = find_workgroup_on_subnet( subrec, work_name);
445
446         if(!work) {
447                 DEBUG(0,("become_local_master_stage1: Error - cannot find \
448                         %s on subnet %s\n", work_name, subrec->subnet_name));
449                 return;
450         }
451
452         DEBUG(3,("become_local_master_stage1: go to stage 2: register the %s<1d> name.\n",
453                 work->work_group));
454
455         work->mst_state = MST_MSB; /* Registering MSBROWSE was successful. */
456
457         /*
458          * We registered the MSBROWSE name on a broadcast subnet, now need to add
459          * the MSBROWSE name to the unicast subnet so that we can answer
460          * unicast requests sent to this name. We create this name directly on
461          * the unicast subnet.
462          */
463
464         insert_permanent_name_into_unicast( subrec, registered_name, nb_flags);
465
466         /* Attempt to register the WORKGROUP<1d> name. */
467         register_name(subrec, work->work_group,0x1d,samba_nb_type,
468                 become_local_master_stage2,
469                 become_local_master_fail2,
470                 NULL);
471 }
472
473 /****************************************************************************
474   Failed to register the MSBROWSE name.
475   ****************************************************************************/
476
477 static void become_local_master_fail1(struct subnet_record *subrec,
478                                       struct response_record *rrec,
479                                       struct nmb_name *fail_name)
480 {
481         char *work_name = rrec->userdata->data;
482         struct work_record *work = find_workgroup_on_subnet(subrec, work_name);
483
484         if(!work) {
485                 DEBUG(0,("become_local_master_fail1: Error - cannot find \
486 workgroup %s on subnet %s\n", work_name, subrec->subnet_name));
487                 return;
488         }
489
490         if(find_server_in_workgroup(work, lp_netbios_name()) == NULL) {
491                 DEBUG(0,("become_local_master_fail1: Error - cannot find server %s \
492 in workgroup %s on subnet %s\n",
493                         lp_netbios_name(), work->work_group, subrec->subnet_name));
494                 return;
495         }
496
497         reset_workgroup_state( subrec, work->work_group, False );
498
499         DEBUG(0,("become_local_master_fail1: Failed to become a local master browser for \
500 workgroup %s on subnet %s. Couldn't register name %s.\n",
501                 work->work_group, subrec->subnet_name, nmb_namestr(fail_name)));
502 }
503
504 /******************************************************************
505   Become the local master browser on a subnet.
506   This gets called if we win an election on this subnet.
507
508   Stage 1: mst_state was MST_POTENTIAL - go to MST_BACK register ^1^2__MSBROWSE__^2^1.
509   Stage 2: mst_state was MST_BACKUP  - go to MST_MSB  and register WORKGROUP<1d>.
510   Stage 3: mst_state was MST_MSB  - go to MST_BROWSER.
511 ******************************************************************/
512
513 void become_local_master_browser(struct subnet_record *subrec, struct work_record *work)
514 {
515         struct userdata_struct *userdata;
516         size_t size = sizeof(struct userdata_struct) + sizeof(fstring) + 1;
517
518         /* Sanity check. */
519         if (!lp_local_master()) { 
520                 DEBUG(0,("become_local_master_browser: Samba not configured as a local master browser.\n"));
521                 return;
522         }
523
524         if(!AM_POTENTIAL_MASTER_BROWSER(work)) {
525                 DEBUG(2,("become_local_master_browser: Awaiting potential browser state. Current state is %d\n",
526                         work->mst_state ));
527                 return;
528         }
529
530         if(find_server_in_workgroup( work, lp_netbios_name()) == NULL) {
531                 DEBUG(0,("become_local_master_browser: Error - cannot find server %s \
532 in workgroup %s on subnet %s\n",
533                         lp_netbios_name(), work->work_group, subrec->subnet_name));
534                 return;
535         }
536
537         DEBUG(2,("become_local_master_browser: Starting to become a master browser for workgroup \
538 %s on subnet %s\n", work->work_group, subrec->subnet_name));
539   
540         DEBUG(3,("become_local_master_browser: first stage - attempt to register ^1^2__MSBROWSE__^2^1\n"));
541         work->mst_state = MST_BACKUP; /* an election win was successful */
542
543         work->ElectionCriterion |= 0x5;
544
545         /* Tell the namelist writer to write out a change. */
546         subrec->work_changed = True;
547
548         /* Setup the userdata_struct. */
549         if((userdata = (struct userdata_struct *)SMB_MALLOC(size)) == NULL) {
550                 DEBUG(0,("become_local_master_browser: malloc fail.\n"));
551                 return;
552         }
553
554         userdata->copy_fn = NULL;
555         userdata->free_fn = NULL;
556         userdata->userdata_len = strlen(work->work_group)+1;
557         strlcpy(userdata->data, work->work_group, size - sizeof(*userdata));
558
559         /* Register the special browser group name. */
560         register_name(subrec, MSBROWSE, 0x01, samba_nb_type|NB_GROUP,
561                 become_local_master_stage1,
562                 become_local_master_fail1,
563                 userdata);
564
565         zero_free(userdata, size);
566 }
567
568 /***************************************************************
569  Utility function to set the local master browser name. Does
570  some sanity checking as old versions of Samba seem to sometimes
571  say that the master browser name for a workgroup is the same
572  as the workgroup name.
573 ****************************************************************/
574
575 void set_workgroup_local_master_browser_name( struct work_record *work, const char *newname)
576 {
577         DEBUG(5,("set_workgroup_local_master_browser_name: setting local master name to '%s' \
578 for workgroup %s.\n", newname, work->work_group ));
579
580 #if 0
581   /*
582    * Apparently some sites use the workgroup name as the local
583    * master browser name. Arrrrggghhhhh ! (JRA).
584    */
585   if(strequal( work->work_group, newname))
586   {
587     DEBUG(5, ("set_workgroup_local_master_browser_name: Refusing to set \
588 local_master_browser_name for workgroup %s to workgroup name.\n",
589          work->work_group ));
590     return;
591   }
592 #endif
593
594         unstrcpy(work->local_master_browser_name, newname);
595 }