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