source/nmbd/nmbd_synclists.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-1998
   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 2 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, write to the Free Software
  20    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  21
  22 */
  23
  24 /* this file handles asynchronous browse synchronisation requests. The
  25    requests are done by forking and putting the result in a file in the
  26    locks directory. We do it this way because we don't want nmbd to be
  27    blocked waiting for some server to respond on a TCP connection. This
  28    also allows us to have more than 1 sync going at once (tridge) */
  29
  30 #include "includes.h"
  31
  32 struct sync_record {
  33         struct sync_record *next, *prev;
  34         unstring workgroup;
  35         unstring server;
  36         pstring fname;
  37         struct in_addr ip;
  38         pid_t pid;
  39 };
  40
  41 /* a linked list of current sync connections */
  42 static struct sync_record *syncs;
  43
  44 static XFILE *fp;
  45
  46 /*******************************************************************
  47   This is the NetServerEnum callback.
  48   Note sname and comment are in UNIX codepage format.
  49   ******************************************************************/
  50
  51 static void callback(const char *sname, uint32 stype,
  52                      const char *comment, void *state)
  53 {
  54         x_fprintf(fp,"\"%s\" %08X \"%s\"\n", sname, stype, comment);
  55 }
  56
  57 /*******************************************************************
  58   Synchronise browse lists with another browse server.
  59   Log in on the remote server's SMB port to their IPC$ service,
  60   do a NetServerEnum and record the results in fname
  61 ******************************************************************/
  62
  63 static void sync_child(char *name, int nm_type,
  64                        char *workgroup,
  65                        struct in_addr ip, BOOL local, BOOL servers,
  66                        char *fname)
  67 {
  68         extern fstring local_machine;
  69         fstring unix_workgroup;
  70         static struct cli_state cli;
  71         uint32 local_type = local ? SV_TYPE_LOCAL_LIST_ONLY : 0;
  72         struct nmb_name called, calling;
  73
  74         /* W2K DMB's return empty browse lists on port 445. Use 139.
  75          * Patch from Andy Levine andyl@epicrealm.com.
  76          */
  77
  78         if (!cli_initialise(&cli) || !cli_set_port(&cli, 139) || !cli_connect(&cli, name, &ip)) {
  79                 return;
  80         }
  81
  82         make_nmb_name(&calling, local_machine, 0x0);
  83         make_nmb_name(&called , name, nm_type);
  84
  85         if (!cli_session_request(&cli, &calling, &called)) {
  86                 cli_shutdown(&cli);
  87                 return;
  88         }
  89
  90         if (!cli_negprot(&cli)) {
  91                 cli_shutdown(&cli);
  92                 return;
  93         }
  94
  95         if (!cli_session_setup(&cli, "", "", 1, "", 0, workgroup)) {
  96                 cli_shutdown(&cli);
  97                 return;
  98         }
  99
 100         if (!cli_send_tconX(&cli, "IPC$", "IPC", "", 1)) {
 101                 cli_shutdown(&cli);
 102                 return;
 103         }
 104
 105         /* All the cli_XX functions take UNIX character set. */
 106         fstrcpy(unix_workgroup, cli.server_domain?cli.server_domain:workgroup);
 107
 108         /* Fetch a workgroup list. */
 109         cli_NetServerEnum(&cli, unix_workgroup,
 110                           local_type|SV_TYPE_DOMAIN_ENUM,
 111                           callback, NULL);
 112
 113         /* Now fetch a server list. */
 114         if (servers) {
 115                 fstrcpy(unix_workgroup, workgroup);
 116                 cli_NetServerEnum(&cli, unix_workgroup,
 117                                   local?SV_TYPE_LOCAL_LIST_ONLY:SV_TYPE_ALL,
 118                                   callback, NULL);
 119         }
 120
 121         cli_shutdown(&cli);
 122 }
 123
 124 /*******************************************************************
 125   initialise a browse sync with another browse server.  Log in on the
 126   remote server's SMB port to their IPC$ service, do a NetServerEnum
 127   and record the results
 128 ******************************************************************/
 129
 130 void sync_browse_lists(struct work_record *work,
 131                        char *name, int nm_type,
 132                        struct in_addr ip, BOOL local, BOOL servers)
 133 {
 134         struct sync_record *s;
 135         static int counter;
 136
 137         START_PROFILE(sync_browse_lists);
 138         /* Check we're not trying to sync with ourselves. This can
 139            happen if we are a domain *and* a local master browser. */
 140         if (ismyip(ip)) {
 141 done:
 142                 END_PROFILE(sync_browse_lists);
 143                 return;
 144         }
 145
 146         s = SMB_MALLOC_P(struct sync_record);
 147         if (!s) goto done;
 148
 149         ZERO_STRUCTP(s);
 150
 151         unstrcpy(s->workgroup, work->work_group);
 152         unstrcpy(s->server, name);
 153         s->ip = ip;
 154
 155         slprintf(s->fname, sizeof(pstring)-1,
 156                  "%s/sync.%d", lp_lockdir(), counter++);
 157         all_string_sub(s->fname,"//", "/", 0);
 158
 159         DLIST_ADD(syncs, s);
 160
 161         /* the parent forks and returns, leaving the child to do the
 162            actual sync and call END_PROFILE*/
 163         CatchChild();
 164         if ((s->pid = sys_fork())) return;
 165
 166         BlockSignals( False, SIGTERM );
 167
 168         DEBUG(2,("Initiating browse sync for %s to %s(%s)\n",
 169                  work->work_group, name, inet_ntoa(ip)));
 170
 171         fp = x_fopen(s->fname,O_WRONLY|O_CREAT|O_TRUNC, 0644);
 172         if (!fp) {
 173                 END_PROFILE(sync_browse_lists);
 174                 _exit(1);
 175         }
 176
 177         sync_child(name, nm_type, work->work_group, ip, local, servers,
 178                    s->fname);
 179
 180         x_fclose(fp);
 181         END_PROFILE(sync_browse_lists);
 182         _exit(0);
 183 }
 184
 185 /**********************************************************************
 186  Handle one line from a completed sync file.
 187  **********************************************************************/
 188
 189 static void complete_one(struct sync_record *s,
 190                          char *sname, uint32 stype, char *comment)
 191 {
 192         struct work_record *work;
 193         struct server_record *servrec;
 194
 195         stype &= ~SV_TYPE_LOCAL_LIST_ONLY;
 196
 197         if (stype & SV_TYPE_DOMAIN_ENUM) {
 198                 /* See if we can find the workgroup on this subnet. */
 199                 if((work=find_workgroup_on_subnet(unicast_subnet, sname))) {
 200                         /* We already know about this workgroup -
 201                            update the ttl. */
 202                         update_workgroup_ttl(work,lp_max_ttl());
 203                 } else {
 204                         /* Create the workgroup on the subnet. */
 205                         work = create_workgroup_on_subnet(unicast_subnet,
 206                                                           sname, lp_max_ttl());
 207                         if (work) {
 208                                 /* remember who the master is */
 209                                 unstrcpy(work->local_master_browser_name, comment);
 210                         }
 211                 }
 212                 return;
 213         }
 214
 215         work = find_workgroup_on_subnet(unicast_subnet, s->workgroup);
 216         if (!work) {
 217                 DEBUG(3,("workgroup %s doesn't exist on unicast subnet?\n",
 218                          s->workgroup));
 219                 return;
 220         }
 221
 222         if ((servrec = find_server_in_workgroup( work, sname))) {
 223                 /* Check that this is not a locally known
 224                    server - if so ignore the entry. */
 225                 if(!(servrec->serv.type & SV_TYPE_LOCAL_LIST_ONLY)) {
 226                         /* We already know about this server - update
 227                            the ttl. */
 228                         update_server_ttl(servrec, lp_max_ttl());
 229                         /* Update the type. */
 230                         servrec->serv.type = stype;
 231                 }
 232                 return;
 233         }
 234
 235         /* Create the server in the workgroup. */
 236         create_server_on_workgroup(work, sname,stype, lp_max_ttl(), comment);
 237 }
 238
 239 /**********************************************************************
 240  Read the completed sync info.
 241 **********************************************************************/
 242
 243 static void complete_sync(struct sync_record *s)
 244 {
 245         XFILE *f;
 246         unstring server, type_str;
 247         unsigned type;
 248         pstring comment;
 249         pstring line;
 250         const char *ptr;
 251         int count=0;
 252
 253         f = x_fopen(s->fname,O_RDONLY, 0);
 254
 255         if (!f)
 256                 return;
 257
 258         while (!x_feof(f)) {
 259
 260                 if (!fgets_slash(line,sizeof(pstring),f))
 261                         continue;
 262
 263                 ptr = line;
 264
 265                 if (!next_token(&ptr,server,NULL,sizeof(server)) ||
 266                     !next_token(&ptr,type_str,NULL, sizeof(type_str)) ||
 267                     !next_token(&ptr,comment,NULL, sizeof(comment))) {
 268                         continue;
 269                 }
 270
 271                 sscanf(type_str, "%X", &type);
 272
 273                 complete_one(s, server, type, comment);
 274
 275                 count++;
 276         }
 277
 278         x_fclose(f);
 279
 280         unlink(s->fname);
 281
 282         DEBUG(2,("sync with %s(%s) for workgroup %s completed (%d records)\n",
 283                  s->server, inet_ntoa(s->ip), s->workgroup, count));
 284 }
 285
 286 /**********************************************************************
 287  Check for completion of any of the child processes.
 288 **********************************************************************/
 289
 290 void sync_check_completion(void)
 291 {
 292         struct sync_record *s, *next;
 293
 294         for (s=syncs;s;s=next) {
 295                 next = s->next;
 296                 if (!process_exists(s->pid)) {
 297                         /* it has completed - grab the info */
 298                         complete_sync(s);
 299                         DLIST_REMOVE(syncs, s);
 300                         ZERO_STRUCTP(s);
 301                         SAFE_FREE(s);
 302                 }
 303         }
 304 }