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