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