r3936: Ensure LARGE_READX response must fit within reply buffer.
[ira/wip.git] / source3 / nmbd / nmbd_serverlistdb.c
index 2fee239b2dc73f77256f35f3782ee4846f8746fb..e6fad8319d91a35395b7b90a6d39cfba812fd634 100644 (file)
@@ -22,7 +22,6 @@
 */
 
 #include "includes.h"
-#include "smb.h"
 
 extern int ClientNMB;
 
@@ -34,28 +33,26 @@ int updatecount = 0;
 
 void remove_all_servers(struct work_record *work)
 {
-  struct server_record *servrec;
-  struct server_record *nexts;
+       struct server_record *servrec;
+       struct server_record *nexts;
 
-  for (servrec = work->serverlist; servrec; servrec = nexts)
-  {
-    DEBUG(7,("remove_all_servers: Removing server %s\n",servrec->serv.name));
-    nexts = servrec->next;
+       for (servrec = work->serverlist; servrec; servrec = nexts) {
+               DEBUG(7,("remove_all_servers: Removing server %s\n",servrec->serv.name));
+               nexts = servrec->next;
 
-    if (servrec->prev)
-      servrec->prev->next = servrec->next;
-    if (servrec->next)
-      servrec->next->prev = servrec->prev;
+               if (servrec->prev)
+                       servrec->prev->next = servrec->next;
+               if (servrec->next)
+                       servrec->next->prev = servrec->prev;
 
-    if (work->serverlist == servrec)
-      work->serverlist = servrec->next;
+               if (work->serverlist == servrec)
+                       work->serverlist = servrec->next;
 
-    ZERO_STRUCTP(servrec);
-    SAFE_FREE(servrec);
+               ZERO_STRUCTP(servrec);
+               SAFE_FREE(servrec);
+       }
 
-  }
-
-  work->subnet->work_changed = True;
+       work->subnet->work_changed = True;
 }
 
 /***************************************************************************
@@ -65,23 +62,22 @@ void remove_all_servers(struct work_record *work)
 static void add_server_to_workgroup(struct work_record *work,
                              struct server_record *servrec)
 {
-  struct server_record *servrec2;
-
-  if (!work->serverlist)
-  {
-    work->serverlist = servrec;
-    servrec->prev = NULL;
-    servrec->next = NULL;
-    return;
-  }
-
-  for (servrec2 = work->serverlist; servrec2->next; servrec2 = servrec2->next)
-    ;
-
-  servrec2->next = servrec;
-  servrec->next = NULL;
-  servrec->prev = servrec2;
-  work->subnet->work_changed = True;
+       struct server_record *servrec2;
+
+       if (!work->serverlist) {
+               work->serverlist = servrec;
+               servrec->prev = NULL;
+               servrec->next = NULL;
+               return;
+       }
+
+       for (servrec2 = work->serverlist; servrec2->next; servrec2 = servrec2->next)
+               ;
+
+       servrec2->next = servrec;
+       servrec->next = NULL;
+       servrec->prev = servrec2;
+       work->subnet->work_changed = True;
 }
 
 /****************************************************************************
@@ -90,14 +86,13 @@ static void add_server_to_workgroup(struct work_record *work,
 
 struct server_record *find_server_in_workgroup(struct work_record *work, const char *name)
 {
-  struct server_record *ret;
+       struct server_record *ret;
   
-  for (ret = work->serverlist; ret; ret = ret->next)
-  {
-    if (strequal(ret->serv.name,name))
-      return ret;
-  }
-  return NULL;
+       for (ret = work->serverlist; ret; ret = ret->next) {
+               if (strequal(ret->serv.name,name))
+                       return ret;
+       }
+       return NULL;
 }
 
 
@@ -107,17 +102,17 @@ struct server_record *find_server_in_workgroup(struct work_record *work, const c
 
 void remove_server_from_workgroup(struct work_record *work, struct server_record *servrec)
 {
-  if (servrec->prev)
-    servrec->prev->next = servrec->next;
-  if (servrec->next)
-    servrec->next->prev = servrec->prev;
+       if (servrec->prev)
+               servrec->prev->next = servrec->next;
+       if (servrec->next)
+               servrec->next->prev = servrec->prev;
 
-  if (work->serverlist == servrec) 
-    work->serverlist = servrec->next; 
+       if (work->serverlist == servrec) 
+               work->serverlist = servrec->next; 
 
-  ZERO_STRUCTP(servrec);
-  SAFE_FREE(servrec);
-  work->subnet->work_changed = True;
+       ZERO_STRUCTP(servrec);
+       SAFE_FREE(servrec);
+       work->subnet->work_changed = True;
 }
 
 /****************************************************************************
@@ -128,47 +123,44 @@ struct server_record *create_server_on_workgroup(struct work_record *work,
                                                  const char *name,int servertype, 
                                                  int ttl, const char *comment)
 {
-  struct server_record *servrec;
+       struct server_record *servrec;
   
-  if (name[0] == '*')
-  {
-      DEBUG(7,("create_server_on_workgroup: not adding name starting with '*' (%s)\n",
-             name));
-      return (NULL);
-  }
+       if (name[0] == '*') {
+               DEBUG(7,("create_server_on_workgroup: not adding name starting with '*' (%s)\n",
+                       name));
+               return (NULL);
+       }
   
-  if((servrec = find_server_in_workgroup(work, name)) != NULL)
-  {
-    DEBUG(0,("create_server_on_workgroup: Server %s already exists on \
+       if((servrec = find_server_in_workgroup(work, name)) != NULL) {
+               DEBUG(0,("create_server_on_workgroup: Server %s already exists on \
 workgroup %s. This is a bug.\n", name, work->work_group));
-    return NULL;
-  }
+               return NULL;
+       }
   
-  if((servrec = (struct server_record *)malloc(sizeof(*servrec))) == NULL)
-  {
-    DEBUG(0,("create_server_entry_on_workgroup: malloc fail !\n"));
-    return NULL;
-  }
+       if((servrec = (struct server_record *)malloc(sizeof(*servrec))) == NULL) {
+               DEBUG(0,("create_server_entry_on_workgroup: malloc fail !\n"));
+               return NULL;
+       }
 
-  memset((char *)servrec,'\0',sizeof(*servrec));
+       memset((char *)servrec,'\0',sizeof(*servrec));
  
-  servrec->subnet = work->subnet;
+       servrec->subnet = work->subnet;
  
-  StrnCpy(servrec->serv.name,name,sizeof(servrec->serv.name)-1);
-  StrnCpy(servrec->serv.comment,comment,sizeof(servrec->serv.comment)-1);
-  strupper(servrec->serv.name);
-  servrec->serv.type  = servertype;
+       fstrcpy(servrec->serv.name,name);
+       fstrcpy(servrec->serv.comment,comment);
+       strupper_m(servrec->serv.name);
+       servrec->serv.type  = servertype;
 
-  update_server_ttl(servrec, ttl);
+       update_server_ttl(servrec, ttl);
   
-  add_server_to_workgroup(work, servrec);
+       add_server_to_workgroup(work, servrec);
       
-  DEBUG(3,("create_server_on_workgroup: Created server entry %s of type %x (%s) on \
+       DEBUG(3,("create_server_on_workgroup: Created server entry %s of type %x (%s) on \
 workgroup %s.\n", name,servertype,comment, work->work_group));
  
-  work->subnet->work_changed = True;
+       work->subnet->work_changed = True;
  
-  return(servrec);
+       return(servrec);
 }
 
 /*******************************************************************
@@ -177,15 +169,15 @@ workgroup %s.\n", name,servertype,comment, work->work_group));
 
 void update_server_ttl(struct server_record *servrec, int ttl)
 {
-  if(ttl > lp_max_ttl())
-    ttl = lp_max_ttl();
+       if(ttl > lp_max_ttl())
+               ttl = lp_max_ttl();
 
-  if(is_myname(servrec->serv.name))
-    servrec->death_time = PERMANENT_TTL;
-  else
-    servrec->death_time = (ttl != PERMANENT_TTL) ? time(NULL)+(ttl*3) : PERMANENT_TTL;
+       if(is_myname(servrec->serv.name))
+               servrec->death_time = PERMANENT_TTL;
+       else
+               servrec->death_time = (ttl != PERMANENT_TTL) ? time(NULL)+(ttl*3) : PERMANENT_TTL;
 
-  servrec->subnet->work_changed = True;
+       servrec->subnet->work_changed = True;
 }
 
 /*******************************************************************
@@ -196,20 +188,18 @@ void update_server_ttl(struct server_record *servrec, int ttl)
 
 void expire_servers(struct work_record *work, time_t t)
 {
-  struct server_record *servrec;
-  struct server_record *nexts;
+       struct server_record *servrec;
+       struct server_record *nexts;
   
-  for (servrec = work->serverlist; servrec; servrec = nexts)
-  {
-    nexts = servrec->next;
-
-    if ((servrec->death_time != PERMANENT_TTL) && ((t == -1) || (servrec->death_time < t)))
-    {
-      DEBUG(3,("expire_old_servers: Removing timed out server %s\n",servrec->serv.name));
-      remove_server_from_workgroup(work, servrec);
-      work->subnet->work_changed = True;
-    }
-  }
+       for (servrec = work->serverlist; servrec; servrec = nexts) {
+               nexts = servrec->next;
+
+               if ((servrec->death_time != PERMANENT_TTL) && ((t == -1) || (servrec->death_time < t))) {
+                       DEBUG(3,("expire_old_servers: Removing timed out server %s\n",servrec->serv.name));
+                       remove_server_from_workgroup(work, servrec);
+                       work->subnet->work_changed = True;
+               }
+       }
 }
 
 /*******************************************************************
@@ -222,33 +212,30 @@ static uint32 write_this_server_name( struct subnet_record *subrec,
                                       struct work_record *work,
                                       struct server_record *servrec)
 {
-  struct subnet_record *ssub;
-  struct work_record *iwork;
-
-  /* Go through all the subnets we have already seen. */
-  for (ssub = FIRST_SUBNET; ssub != subrec; ssub = NEXT_SUBNET_INCLUDING_UNICAST(ssub)) 
-  {
-    for(iwork = ssub->workgrouplist; iwork; iwork = iwork->next)
-    {
-      if(find_server_in_workgroup( iwork, servrec->serv.name) != NULL)
-      {
-        /*
-         * We have already written out this server record, don't
-         * do it again. This gives precedence to servers we have seen
-         * on the broadcast subnets over servers that may have been
-         * added via a sync on the unicast_subet.
-         *
-         * The correct way to do this is to have a serverlist file
-         * per subnet - this means changes to smbd as well. I may
-         * add this at a later date (JRA).
-         */
-
-        return 0;
-      }
-    }
-  }
-
-  return servrec->serv.type;
+       struct subnet_record *ssub;
+       struct work_record *iwork;
+
+       /* Go through all the subnets we have already seen. */
+       for (ssub = FIRST_SUBNET; ssub != subrec; ssub = NEXT_SUBNET_INCLUDING_UNICAST(ssub)) {
+               for(iwork = ssub->workgrouplist; iwork; iwork = iwork->next) {
+                       if(find_server_in_workgroup( iwork, servrec->serv.name) != NULL) {
+                               /*
+                                * We have already written out this server record, don't
+                                * do it again. This gives precedence to servers we have seen
+                                * on the broadcast subnets over servers that may have been
+                                * added via a sync on the unicast_subet.
+                                *
+                                * The correct way to do this is to have a serverlist file
+                                * per subnet - this means changes to smbd as well. I may
+                                * add this at a later date (JRA).
+                                */
+
+                               return 0;
+                       }
+               }
+       }
+
+       return servrec->serv.type;
 }
 
 /*******************************************************************
@@ -262,30 +249,29 @@ static uint32 write_this_server_name( struct subnet_record *subrec,
 static uint32 write_this_workgroup_name( struct subnet_record *subrec, 
                                          struct work_record *work)
 {
-  struct subnet_record *ssub;
+       struct subnet_record *ssub;
 
-  if(strequal(lp_workgroup(), work->work_group))
-    return 0;
+       if(strequal(lp_workgroup(), work->work_group))
+               return 0;
 
-  /* This is a workgroup we have seen on a broadcast subnet. All
-     these have the same type. */
+       /* This is a workgroup we have seen on a broadcast subnet. All
+               these have the same type. */
 
-  if(subrec != unicast_subnet)
-    return (SV_TYPE_DOMAIN_ENUM|SV_TYPE_NT|SV_TYPE_LOCAL_LIST_ONLY);
+       if(subrec != unicast_subnet)
+               return (SV_TYPE_DOMAIN_ENUM|SV_TYPE_NT|SV_TYPE_LOCAL_LIST_ONLY);
 
-  for(ssub = FIRST_SUBNET; ssub;  ssub = NEXT_SUBNET_EXCLUDING_UNICAST(ssub))
-  {
-    /* This is the unicast subnet so check if we've already written out
-       this subnet when we passed over the broadcast subnets. */
+       for(ssub = FIRST_SUBNET; ssub;  ssub = NEXT_SUBNET_EXCLUDING_UNICAST(ssub)) {
+               /* This is the unicast subnet so check if we've already written out
+                       this subnet when we passed over the broadcast subnets. */
 
-    if(find_workgroup_on_subnet( ssub, work->work_group) != NULL)
-      return 0;
-  }
+               if(find_workgroup_on_subnet( ssub, work->work_group) != NULL)
+                       return 0;
+       }
 
-  /* All workgroups on the unicast subnet (except our own, which we
-     have already written out) cannot be local. */
+       /* All workgroups on the unicast subnet (except our own, which we
+               have already written out) cannot be local. */
 
-  return (SV_TYPE_DOMAIN_ENUM|SV_TYPE_NT);
+       return (SV_TYPE_DOMAIN_ENUM|SV_TYPE_NT);
 }
 
 /*******************************************************************
@@ -307,143 +293,130 @@ void write_browse_list_entry(XFILE *fp, const char *name, uint32 rec_type,
 
 void write_browse_list(time_t t, BOOL force_write)
 {   
-  struct subnet_record *subrec;
-  struct work_record *work;
-  struct server_record *servrec;
-  pstring fname,fnamenew;
-  uint32 stype;
-  int i;
-  XFILE *fp;
-  BOOL list_changed = force_write;
-  static time_t lasttime = 0;
+       struct subnet_record *subrec;
+       struct work_record *work;
+       struct server_record *servrec;
+       pstring fname,fnamenew;
+       uint32 stype;
+       int i;
+       XFILE *fp;
+       BOOL list_changed = force_write;
+       static time_t lasttime = 0;
     
-  /* Always dump if we're being told to by a signal. */
-  if(force_write == False)
-  {
-    if (!lasttime)
-      lasttime = t;
-    if (t - lasttime < 5)
-      return;
-  }
-
-  lasttime = t;
-
-  dump_workgroups(force_write);
+       /* Always dump if we're being told to by a signal. */
+       if(force_write == False) {
+               if (!lasttime)
+                       lasttime = t;
+               if (t - lasttime < 5)
+                       return;
+       }
+
+       lasttime = t;
+
+       dump_workgroups(force_write);
  
-  for (subrec = FIRST_SUBNET; subrec ; subrec = NEXT_SUBNET_INCLUDING_UNICAST(subrec))
-  {
-    if(subrec->work_changed)
-    {
-      list_changed = True;
-      break;
-    }
-  }
-
-  if(!list_changed)
-    return;
-
-  updatecount++;
+       for (subrec = FIRST_SUBNET; subrec ; subrec = NEXT_SUBNET_INCLUDING_UNICAST(subrec)) {
+               if(subrec->work_changed) {
+                       list_changed = True;
+                       break;
+               }
+       }
+
+       if(!list_changed)
+               return;
+
+       updatecount++;
     
-  pstrcpy(fname,lp_lockdir());
-  trim_string(fname,NULL,"/");
-  pstrcat(fname,"/");
-  pstrcat(fname,SERVER_LIST);
-  pstrcpy(fnamenew,fname);
-  pstrcat(fnamenew,".");
+       pstrcpy(fname,lp_lockdir());
+       trim_char(fname,'\0' ,'/');
+       pstrcat(fname,"/");
+       pstrcat(fname,SERVER_LIST);
+       pstrcpy(fnamenew,fname);
+       pstrcat(fnamenew,".");
  
-  fp = x_fopen(fnamenew,O_WRONLY|O_CREAT|O_TRUNC, 0644);
+       fp = x_fopen(fnamenew,O_WRONLY|O_CREAT|O_TRUNC, 0644);
  
-  if (!fp)
-  {
-    DEBUG(0,("write_browse_list: Can't open file %s. Error was %s\n",
-              fnamenew,strerror(errno)));
-    return;
-  } 
+       if (!fp) {
+               DEBUG(0,("write_browse_list: Can't open file %s. Error was %s\n",
+                       fnamenew,strerror(errno)));
+               return;
+       } 
   
-  /*
-   * Write out a record for our workgroup. Use the record from the first
-   * subnet.
-   */
-
-  if((work = find_workgroup_on_subnet(FIRST_SUBNET, lp_workgroup())) == NULL)
-  { 
-    DEBUG(0,("write_browse_list: Fatal error - cannot find my workgroup %s\n",
-             lp_workgroup()));
-    x_fclose(fp);
-    return;
-  }
-
-  write_browse_list_entry(fp, work->work_group,
-     SV_TYPE_DOMAIN_ENUM|SV_TYPE_NT|SV_TYPE_LOCAL_LIST_ONLY,
-     work->local_master_browser_name, work->work_group);
-
-  /* 
-   * We need to do something special for our own names.
-   * This is due to the fact that we may be a local master browser on
-   * one of our broadcast subnets, and a domain master on the unicast
-   * subnet. We iterate over the subnets and only write out the name
-   * once.
-   */
-
-  for (i=0; my_netbios_names(i); i++)
-  {
-    stype = 0;
-    for (subrec = FIRST_SUBNET; subrec ; subrec = NEXT_SUBNET_INCLUDING_UNICAST(subrec))
-    {
-      if((work = find_workgroup_on_subnet( subrec, lp_workgroup() )) == NULL)
-        continue;
-      if((servrec = find_server_in_workgroup( work, my_netbios_names(i))) == NULL)
-        continue;
-
-      stype |= servrec->serv.type;
-    }
-
-    /* Output server details, plus what workgroup they're in. */
-    write_browse_list_entry(fp, my_netbios_names(i), stype,
-        string_truncate(lp_serverstring(), MAX_SERVER_STRING_LENGTH), lp_workgroup());
-  }
+       /*
+        * Write out a record for our workgroup. Use the record from the first
+        * subnet.
+        */
+
+       if((work = find_workgroup_on_subnet(FIRST_SUBNET, lp_workgroup())) == NULL) { 
+               DEBUG(0,("write_browse_list: Fatal error - cannot find my workgroup %s\n",
+                       lp_workgroup()));
+               x_fclose(fp);
+               return;
+       }
+
+       write_browse_list_entry(fp, work->work_group,
+               SV_TYPE_DOMAIN_ENUM|SV_TYPE_NT|SV_TYPE_LOCAL_LIST_ONLY,
+               work->local_master_browser_name, work->work_group);
+
+       /* 
+        * We need to do something special for our own names.
+        * This is due to the fact that we may be a local master browser on
+        * one of our broadcast subnets, and a domain master on the unicast
+        * subnet. We iterate over the subnets and only write out the name
+        * once.
+        */
+
+       for (i=0; my_netbios_names(i); i++) {
+               stype = 0;
+               for (subrec = FIRST_SUBNET; subrec ; subrec = NEXT_SUBNET_INCLUDING_UNICAST(subrec)) {
+                       if((work = find_workgroup_on_subnet( subrec, lp_workgroup() )) == NULL)
+                               continue;
+                       if((servrec = find_server_in_workgroup( work, my_netbios_names(i))) == NULL)
+                               continue;
+
+                       stype |= servrec->serv.type;
+               }
+
+               /* Output server details, plus what workgroup they're in. */
+               write_browse_list_entry(fp, my_netbios_names(i), stype,
+                       string_truncate(lp_serverstring(), MAX_SERVER_STRING_LENGTH), lp_workgroup());
+       }
       
-  for (subrec = FIRST_SUBNET; subrec ; subrec = NEXT_SUBNET_INCLUDING_UNICAST(subrec)) 
-  { 
-    subrec->work_changed = False;
-
-    for (work = subrec->workgrouplist; work ; work = work->next)
-    {
-      /* Write out a workgroup record for a workgroup. */
-      uint32 wg_type = write_this_workgroup_name( subrec, work);
-
-      if(wg_type)
-      {
-        write_browse_list_entry(fp, work->work_group, wg_type,
-                                work->local_master_browser_name,
-                                work->work_group);
-      }
-
-      /* Now write out any server records a workgroup may have. */
-
-      for (servrec = work->serverlist; servrec ; servrec = servrec->next)
-      {
-        uint32 serv_type;
-
-        /* We have already written our names here. */
-        if(is_myname(servrec->serv.name))
-          continue; 
-
-        serv_type = write_this_server_name(subrec, work, servrec);
-
-        if(serv_type)
-        {
-          /* Output server details, plus what workgroup they're in. */
-          write_browse_list_entry(fp, servrec->serv.name, serv_type,
-                                 servrec->serv.comment, work->work_group);
-        }
-      }
-    }
-  } 
+       for (subrec = FIRST_SUBNET; subrec ; subrec = NEXT_SUBNET_INCLUDING_UNICAST(subrec)) { 
+               subrec->work_changed = False;
+
+               for (work = subrec->workgrouplist; work ; work = work->next) {
+                       /* Write out a workgroup record for a workgroup. */
+                       uint32 wg_type = write_this_workgroup_name( subrec, work);
+
+                       if(wg_type) {
+                               write_browse_list_entry(fp, work->work_group, wg_type,
+                                               work->local_master_browser_name,
+                                               work->work_group);
+                       }
+
+                       /* Now write out any server records a workgroup may have. */
+
+                       for (servrec = work->serverlist; servrec ; servrec = servrec->next) {
+                               uint32 serv_type;
+
+                               /* We have already written our names here. */
+                               if(is_myname(servrec->serv.name))
+                                       continue; 
+
+                               serv_type = write_this_server_name(subrec, work, servrec);
+                               if(serv_type) {
+                                       /* Output server details, plus what workgroup they're in. */
+                                       write_browse_list_entry(fp, servrec->serv.name, serv_type,
+                                               servrec->serv.comment, work->work_group);
+                               }
+                       }
+               }
+       } 
   
-  x_fclose(fp);
-  unlink(fname);
-  chmod(fnamenew,0644);
-  rename(fnamenew,fname);
-  DEBUG(3,("write_browse_list: Wrote browse list into file %s\n",fname));
+       x_fclose(fp);
+       unlink(fname);
+       chmod(fnamenew,0644);
+       rename(fnamenew,fname);
+       DEBUG(3,("write_browse_list: Wrote browse list into file %s\n",fname));
 }