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