source3/wrepld/process.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    process incoming packets - main loop
   4    Copyright (C) Jean François Micouleau      1998-2002.
   5
   6    This program is free software; you can redistribute it and/or modify
   7    it under the terms of the GNU General Public License as published by
   8    the Free Software Foundation; either version 2 of the License, or
   9    (at your option) any later version.
  10
  11    This program is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14    GNU General Public License for more details.
  15
  16    You should have received a copy of the GNU General Public License
  17    along with this program; if not, write to the Free Software
  18    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  19 */
  20
  21 #include "includes.h"
  22 #include "wins_repl.h"
  23
  24 extern fd_set *listen_set;
  25 extern int listen_number;
  26 extern int *sock_array;
  27
  28 WINS_OWNER global_wins_table[64][64];
  29 int partner_count;
  30
  31 TALLOC_CTX *mem_ctx;
  32
  33 #define WINS_LIST "wins.tdb"
  34 #define INFO_VERSION    "INFO/version"
  35 #define INFO_COUNT      "INFO/num_entries"
  36 #define INFO_ID_HIGH    "INFO/id_high"
  37 #define INFO_ID_LOW     "INFO/id_low"
  38 #define ENTRY_PREFIX    "ENTRY/"
  39
  40
  41 /*******************************************************************
  42 fill the header of a reply.
  43 ********************************************************************/
  44 static void fill_header(GENERIC_PACKET *g, int opcode, int ctx, int mess)
  45 {
  46         if (g==NULL)
  47                 return;
  48
  49         g->header.opcode=opcode;
  50         g->header.assoc_ctx=ctx;
  51         g->header.mess_type=mess;
  52 }
  53
  54 /*******************************************************************
  55 dump the global table, that's a debug code.
  56 ********************************************************************/
  57 static void dump_global_table(void)
  58 {
  59         int i,j;
  60
  61         for (i=0;i<partner_count;i++) {
  62                 DEBUG(10,("\n%d ", i));
  63                 for (j=0; global_wins_table[i][j].address.s_addr!=0; j++)
  64                         DEBUG(10,("%s:%d \t", inet_ntoa(global_wins_table[i][j].address),
  65                                 (int)global_wins_table[i][j].max_version));
  66         }
  67         DEBUG(10,("\n"));
  68 }
  69
  70 /*******************************************************************
  71 start association
  72 ********************************************************************/
  73 static void start_assoc_process(GENERIC_PACKET *q, GENERIC_PACKET *r)
  74 {
  75         /*
  76          * add this request to our current wins partners list
  77          * this list is used to know with who we are in contact
  78          *
  79          */
  80         r->sa_rp.assoc_ctx=time(NULL);
  81         fill_header(r, OPCODE_NON_NBT, q->sa_rq.assoc_ctx, MESSAGE_TYPE_START_ASSOC_REPLY);
  82
  83         /* reply we are a NT4 server */
  84
  85         /* w2K is min=2, maj=5 */
  86
  87         r->sa_rp.min_ver=1;
  88         r->sa_rp.maj_ver=1;
  89
  90         add_partner(r->sa_rp.assoc_ctx, q->sa_rq.assoc_ctx, False, False);
  91 }
  92
  93 /*******************************************************************
  94 start association reply
  95 ********************************************************************/
  96 static void start_assoc_reply(GENERIC_PACKET *q, GENERIC_PACKET *r)
  97 {
  98         int i;
  99
 100         /* check if we have already registered this client */
 101         if (!check_partner(q->header.assoc_ctx)) {
 102                 DEBUG(0,("start_assoc_reply: unknown client\n"));
 103                 stop_packet(q, r, STOP_REASON_USER_REASON);
 104                 return;
 105         }
 106
 107         if (!update_server_partner(q->header.assoc_ctx, q->sa_rp.assoc_ctx)) {
 108                 DEBUG(0,("start_assoc_reply: can't update server ctx\n"));
 109                 stop_packet(q, r, STOP_REASON_USER_REASON);
 110                 return;
 111         }
 112
 113         /* if pull, request map table */
 114         if (check_pull_partner(q->header.assoc_ctx)) {
 115                 fill_header(r, OPCODE_NON_NBT, get_server_assoc(q->header.assoc_ctx), MESSAGE_TYPE_REPLICATE);
 116
 117                 r->rep.msg_type=MESSAGE_REP_ADD_VERSION_REQUEST;
 118                 DEBUG(5,("start_assoc_reply: requesting map table\n"));
 119
 120                 return;
 121         }
 122
 123         /* if push, send our table */
 124         if (check_push_partner(q->header.assoc_ctx)) {
 125                 fill_header(r, OPCODE_NON_NBT, get_server_assoc(q->header.assoc_ctx), MESSAGE_TYPE_REPLICATE);
 126                 r->rep.msg_type=MESSAGE_REP_UPDATE_NOTIFY_REQUEST;
 127                 r->rep.un_rq.partner_count=partner_count;
 128
 129                 r->rep.un_rq.wins_owner=(WINS_OWNER *)talloc(mem_ctx, partner_count*sizeof(WINS_OWNER));
 130                 if (r->rep.un_rq.wins_owner==NULL) {
 131                         DEBUG(0,("start_assoc_reply: can't alloc memory\n"));
 132                         stop_packet(q, r, STOP_REASON_USER_REASON);
 133                         return;
 134                 }
 135
 136                 for (i=0; i<partner_count; i++)
 137                         r->rep.un_rq.wins_owner[i]=global_wins_table[0][i];
 138
 139                 DEBUG(5,("start_assoc_reply: sending update table\n"));
 140                 return;
 141         }
 142
 143         /* neither push/pull, stop */
 144         /* we should not come here */
 145         DEBUG(0,("we have a partner which is neither push nor pull !\n"));
 146         stop_packet(q, r, STOP_REASON_USER_REASON);
 147 }
 148
 149 /****************************************************************************
 150 initialise and fill the in-memory partner table.
 151 ****************************************************************************/
 152 int init_wins_partner_table(void)
 153 {
 154         int i=1,j=0,k;
 155         char **partner = lp_list_make(lp_wins_partners());
 156
 157         if (partner==NULL) {
 158                 DEBUG(0,("wrepld: no partner list in smb.conf, exiting\n"));
 159                 exit_server("normal exit");
 160                 return(0);
 161         }
 162
 163         DEBUG(4, ("init_wins_partner_table: partners: %s\n", lp_wins_partners()));
 164
 165         global_wins_table[0][0].address=*iface_n_ip(0);
 166         global_wins_table[0][0].max_version=0;
 167         global_wins_table[0][0].min_version=0;
 168         global_wins_table[0][0].type=0;
 169
 170         while (partner[j]!=NULL) {
 171                 DEBUG(3,("init_wins_partner_table, adding partner: %s\n", partner[j]));
 172
 173                 global_wins_table[0][i].address=*interpret_addr2(partner[j]);
 174                 global_wins_table[0][i].max_version=0;
 175                 global_wins_table[0][i].min_version=0;
 176                 global_wins_table[0][i].type=0;
 177                 global_wins_table[0][i].last_pull=0;
 178                 global_wins_table[0][i].last_push=0;
 179
 180                 i++;
 181                 j++;
 182         }
 183
 184         for (k=1; k<i;k++)
 185                 for (j=0; j<i; j++)
 186                         global_wins_table[k][j]=global_wins_table[0][j];
 187
 188         lp_list_free (&partner);
 189
 190         return i;
 191 }
 192
 193 /****************************************************************************
 194 read the last ID from the wins tdb file.
 195 ****************************************************************************/
 196 static void get_our_last_id(WINS_OWNER *wins_owner)
 197 {
 198         TDB_CONTEXT *tdb;
 199
 200         tdb = tdb_open_log(lock_path(WINS_LIST), 0, TDB_DEFAULT, O_RDONLY, 0600);
 201         if (!tdb) {
 202                 DEBUG(2,("get_our_last_id: Can't open wins database file %s. Error was %s\n", WINS_LIST, strerror(errno) ));
 203                 return;
 204         }
 205
 206         wins_owner->max_version=((SMB_BIG_UINT)tdb_fetch_int32(tdb, INFO_ID_HIGH))<<32 |
 207                                  (SMB_BIG_UINT)tdb_fetch_int32(tdb, INFO_ID_LOW);
 208
 209         tdb_close(tdb);
 210 }
 211
 212 /****************************************************************************
 213 send the list of wins server we know.
 214 ****************************************************************************/
 215 static void send_version_number_map_table(GENERIC_PACKET *q, GENERIC_PACKET *r)
 216 {
 217         int i;
 218         int s_ctx=get_server_assoc(q->header.assoc_ctx);
 219
 220         if (s_ctx==0) {
 221                 DEBUG(5, ("send_version_number_map_table: request for a partner not in our table\n"));
 222                 stop_packet(q, r, STOP_REASON_USER_REASON);
 223                 return;
 224         }
 225
 226         /*
 227          * return an array of wins servers, we are partner with.
 228          * each entry contains the IP address and the version info
 229          * version: ID of the last entry we've got
 230          */
 231
 232         /* the first wins server must be self */
 233
 234         /*
 235          * get our last ID from the wins database
 236          * it can have been updated since last read
 237          * as nmbd got registration/release.
 238          */
 239         get_our_last_id(&global_wins_table[0][0]);
 240
 241         r->rep.avmt_rep.wins_owner=(WINS_OWNER *)talloc(mem_ctx, partner_count*sizeof(WINS_OWNER));
 242         if (r->rep.avmt_rep.wins_owner==NULL) {
 243                 stop_packet(q, r, STOP_REASON_USER_REASON);
 244                 return;
 245         }
 246
 247         DEBUG(5,("send_version_number_map_table: partner_count: %d\n", partner_count));
 248
 249         for (i=0; i<partner_count; i++) {
 250                 DEBUG(5,("send_version_number_map_table, partner: %d -> %s, \n", i, inet_ntoa(global_wins_table[0][i].address)));
 251                 r->rep.avmt_rep.wins_owner[i]=global_wins_table[0][i];
 252         }
 253
 254         r->rep.msg_type=1;
 255         r->rep.avmt_rep.partner_count=partner_count;
 256         r->rep.avmt_rep.initiating_wins_server.s_addr=0; /* blatant lie, NT4/w2K do the same ! */
 257         fill_header(r, OPCODE_NON_NBT, s_ctx, MESSAGE_TYPE_REPLICATE);
 258 }
 259
 260 /****************************************************************************
 261 for a given partner, ask it to send entries we don't have.
 262 ****************************************************************************/
 263 static BOOL check_partners_and_send_entries(GENERIC_PACKET *q, GENERIC_PACKET *r, int partner)
 264 {
 265         int server;
 266         int other;
 267         SMB_BIG_UINT temp;
 268         SMB_BIG_UINT current;
 269
 270
 271         /*
 272          * we check if our partner has more records than us.
 273          * we need to check more than our direct partners as
 274          * we can have this case:
 275          * us: A, partners: B,C, indirect partner: D
 276          * A<->B, A<->C, B<->D, C<->D
 277          *
 278          * So if we're talking to B, we need to check if between
 279          * B and C, which one have more records about D.
 280          * and also check if we don't already have the records.
 281          */
 282
 283
 284          /* check all servers even indirect */
 285          for (server=1; global_wins_table[0][server].address.s_addr!=0; server++) {
 286                 current = global_wins_table[partner][server].max_version;
 287
 288                 temp=0;
 289
 290                 for (other=1; other<partner_count; other++) {
 291                         /* skip the partner itself */
 292                         if (other==partner)
 293                                 continue;
 294
 295                         if (global_wins_table[other][server].max_version > temp)
 296                                 temp=global_wins_table[other][server].max_version;
 297                 }
 298
 299                 if (current >= temp && current > global_wins_table[0][server].max_version) {
 300                         /*
 301                          * it has more records than every body else and more than us,
 302                          * ask it the difference between what we have and what it has
 303                          */
 304                         fill_header(r, OPCODE_NON_NBT, get_server_assoc(q->header.assoc_ctx), MESSAGE_TYPE_REPLICATE);
 305
 306                         r->rep.msg_type=MESSAGE_REP_SEND_ENTRIES_REQUEST;
 307                         r->rep.se_rq.wins_owner.address=global_wins_table[partner][server].address;
 308
 309                         r->rep.se_rq.wins_owner.max_version=global_wins_table[partner][server].max_version;
 310                         r->rep.se_rq.wins_owner.min_version=global_wins_table[0][server].max_version;
 311                         r->rep.se_rq.wins_owner.type=0;
 312
 313                         write_server_assoc_table(q->header.assoc_ctx, global_wins_table[0][partner].address, global_wins_table[partner][server].address);
 314
 315                         /*
 316                          * and we update our version for this server
 317                          * as we can't use the IDs returned in the send_entries function
 318                          * the max ID can be larger than the largest ID returned
 319                          */
 320
 321                         global_wins_table[0][server].max_version=global_wins_table[partner][server].max_version;
 322
 323                         return True;
 324                 }
 325         }
 326         return False;
 327 }
 328
 329 /****************************************************************************
 330 receive the list of wins server we know.
 331 ****************************************************************************/
 332 static void receive_version_number_map_table(GENERIC_PACKET *q, GENERIC_PACKET *r)
 333 {
 334         fstring peer;
 335         struct in_addr addr;
 336         int i,j,k,l;
 337         int s_ctx=get_server_assoc(q->header.assoc_ctx);
 338
 339         if (s_ctx==0) {
 340                 DEBUG(5, ("receive_version_number_map_table: request for a partner not in our table\n"));
 341                 stop_packet(q, r, STOP_REASON_USER_REASON);
 342                 return;
 343         }
 344
 345         fstrcpy(peer,get_socket_addr(q->fd));
 346         addr=*interpret_addr2(peer);
 347
 348         get_our_last_id(&global_wins_table[0][0]);
 349
 350         DEBUG(5,("receive_version_number_map_table: received a map of %d server from: %s\n",
 351                   q->rep.avmt_rep.partner_count ,inet_ntoa(q->rep.avmt_rep.initiating_wins_server)));
 352         DEBUG(5,("real peer is: %s\n", peer));
 353
 354         for (i=0; global_wins_table[0][i].address.s_addr!=addr.s_addr && i<partner_count;i++)
 355                 ;
 356
 357         if (i==partner_count) {
 358                 DEBUG(5,("receive_version_number_map_table: unknown partner: %s\n", peer));
 359                 stop_packet(q, r, STOP_REASON_USER_REASON);
 360                 return;
 361         }
 362
 363         for (j=0; j<q->rep.avmt_rep.partner_count;j++) {
 364                 /*
 365                  * search if we already have this entry or if it's a new one
 366                  * it can be a new one in case of propagation
 367                  */
 368                 for (k=0; global_wins_table[0][k].address.s_addr!=0 &&
 369                           global_wins_table[0][k].address.s_addr!=q->rep.avmt_rep.wins_owner[j].address.s_addr; k++);
 370
 371                 global_wins_table[i][k].address.s_addr=q->rep.avmt_rep.wins_owner[j].address.s_addr;
 372                 global_wins_table[i][k].max_version=q->rep.avmt_rep.wins_owner[j].max_version;
 373                 global_wins_table[i][k].min_version=q->rep.avmt_rep.wins_owner[j].min_version;
 374                 global_wins_table[i][k].type=q->rep.avmt_rep.wins_owner[j].type;
 375
 376                 /*
 377                  * in case it's a new one, rewrite the address for all the partner
 378                  * to reserve the slot.
 379                  */
 380
 381                 for(l=0; l<partner_count; l++)
 382                         global_wins_table[l][k].address.s_addr=q->rep.avmt_rep.wins_owner[j].address.s_addr;
 383         }
 384
 385         dump_global_table();
 386
 387         /*
 388          * if this server have newer records than what we have
 389          * for several wins servers, we need to ask it.
 390          * Alas a send entry request is only on one server.
 391          * So in the send entry reply, we'll ask for the next server if required.
 392          */
 393
 394         if (check_partners_and_send_entries(q, r, i))
 395                 return;
 396
 397         /* it doesn't have more entries than us */
 398         stop_packet(q, r, STOP_REASON_USER_REASON);
 399 }
 400
 401 /****************************************************************************
 402 add an entry to the wins list we'll send.
 403 ****************************************************************************/
 404 static BOOL add_record_to_winsname(WINS_NAME **wins_name, int *max_names, char *name, int type, int wins_flags, int id, struct in_addr *ip_list, int num_ips)
 405 {
 406         WINS_NAME *temp_list;
 407         int i;
 408         int current=*max_names;
 409
 410         temp_list=talloc_realloc(mem_ctx, *wins_name, (current+1)*sizeof(WINS_NAME));
 411         if (temp_list==NULL)
 412                 return False;
 413
 414         temp_list[current].name_len=0x11;
 415
 416         safe_strcpy(temp_list[current].name, name, 15);
 417
 418         temp_list[current].type=type;
 419         temp_list[current].empty=0;
 420
 421         temp_list[current].name_flag=wins_flags;
 422
 423         if ( (wins_flags&0x03) == 1 || (wins_flags&0x03)==2)
 424                 temp_list[current].group_flag=0x01000000;
 425         else
 426                 temp_list[current].group_flag=0x00000000;
 427
 428         temp_list[current].id=id;
 429
 430         temp_list[current].owner.s_addr=ip_list[0].s_addr;
 431
 432         if (temp_list[current].name_flag & 2) {
 433                 temp_list[current].num_ip=num_ips;
 434                 temp_list[current].others=(struct in_addr *)talloc(mem_ctx, sizeof(struct in_addr)*num_ips);
 435                 if (temp_list[current].others==NULL)
 436                         return False;
 437
 438                 for (i=0; i<num_ips; i++)
 439                         temp_list[current].others[i].s_addr=ip_list[i].s_addr;
 440
 441         } else
 442                 temp_list[current].num_ip=1;
 443
 444         temp_list[current].foo=0xffffffff;
 445
 446         *wins_name=temp_list;
 447
 448         return True;
 449 }
 450
 451 /****************************************************************************
 452 send the list of name we have.
 453 ****************************************************************************/
 454 static void send_entry_request(GENERIC_PACKET *q, GENERIC_PACKET *r)
 455 {
 456         int max_names=0;
 457         int i;
 458         time_t time_now = time(NULL);
 459         WINS_OWNER *wins_owner;
 460         TDB_CONTEXT *tdb;
 461         TDB_DATA kbuf, dbuf, newkey;
 462         int s_ctx=get_server_assoc(q->header.assoc_ctx);
 463         int num_interfaces = iface_count();
 464
 465         if (s_ctx==0) {
 466                 DEBUG(1, ("send_entry_request: request for a partner not in our table\n"));
 467                 stop_packet(q, r, STOP_REASON_USER_REASON);
 468                 return;
 469         }
 470
 471
 472         wins_owner=&q->rep.se_rq.wins_owner;
 473         r->rep.se_rp.wins_name=NULL;
 474
 475         DEBUG(3,("send_entry_request: we have been asked to send the list of wins records\n"));
 476         DEBUGADD(3,("owned by: %s and between min: %d and max: %d\n", inet_ntoa(wins_owner->address),
 477                     (int)wins_owner->min_version, (int)wins_owner->max_version));
 478
 479         /*
 480          * if we are asked to send records owned by us
 481          * we overwrite the wins ip with 0.0.0.0
 482          * to make it easy in case of multihomed
 483          */
 484
 485         for (i=0; i<num_interfaces; i++)
 486                 if (ip_equal(wins_owner->address, *iface_n_ip(i))) {
 487                         wins_owner->address=*interpret_addr2("0.0.0.0");
 488                         break;
 489                 }
 490
 491
 492         tdb = tdb_open_log(lock_path(WINS_LIST), 0, TDB_DEFAULT, O_RDONLY, 0600);
 493         if (!tdb) {
 494                 DEBUG(2,("send_entry_request: Can't open wins database file %s. Error was %s\n", WINS_LIST, strerror(errno) ));
 495                 return;
 496         }
 497
 498         for (kbuf = tdb_firstkey(tdb);
 499              kbuf.dptr;
 500              newkey = tdb_nextkey(tdb, kbuf), safe_free(kbuf.dptr), kbuf=newkey) {
 501                 pstring name_type, name, ip_str;
 502                 char *p;
 503                 int type = 0;
 504                 int nb_flags;
 505                 int ttl;
 506                 unsigned int num_ips;
 507                 int low, high;
 508                 SMB_BIG_UINT version;
 509                 struct in_addr wins_ip;
 510                 struct in_addr *ip_list;
 511                 int wins_flags;
 512                 int len;
 513
 514                 if (strncmp(kbuf.dptr, ENTRY_PREFIX, strlen(ENTRY_PREFIX)) != 0)
 515                         continue;
 516
 517
 518                 dbuf = tdb_fetch(tdb, kbuf);
 519                 if (!dbuf.dptr)
 520                         continue;
 521
 522                 fstrcpy(name_type, kbuf.dptr+strlen(ENTRY_PREFIX));
 523                 pstrcpy(name, name_type);
 524
 525                 if((p = strchr(name,'#')) != NULL) {
 526                         *p = 0;
 527                         sscanf(p+1,"%x",&type);
 528                 }
 529
 530                 len = tdb_unpack(dbuf.dptr, dbuf.dsize, "dddfddd",
 531                                 &nb_flags,
 532                                 &high,
 533                                 &low,
 534                                 ip_str,
 535                                 &ttl,
 536                                 &num_ips,
 537                                 &wins_flags);
 538
 539                 wins_ip=*interpret_addr2(ip_str);
 540
 541                 /* Allocate the space for the ip_list. */
 542                 if((ip_list = (struct in_addr *)talloc(mem_ctx,  num_ips * sizeof(struct in_addr))) == NULL) {
 543                         DEBUG(0,("initialise_wins: talloc fail !\n"));
 544                         return;
 545                 }
 546
 547                 for (i = 0; i < num_ips; i++) {
 548                         len += tdb_unpack(dbuf.dptr+len, dbuf.dsize-len, "f", ip_str);
 549                         ip_list[i] = *interpret_addr2(ip_str);
 550                 }
 551
 552                 /* add all entries that have 60 seconds or more to live */
 553                 if ((ttl - 60) > time_now || ttl == PERMANENT_TTL) {
 554                         if(ttl != PERMANENT_TTL)
 555                                 ttl -= time_now;
 556
 557                         DEBUG( 4, ("send_entry_request: add name: %s#%02x ttl = %d first IP %s flags = %2x\n",
 558                             name, type, ttl, inet_ntoa(ip_list[0]), nb_flags));
 559
 560                         /* add the record to the list to send */
 561                         version=((SMB_BIG_UINT)high)<<32 | low;
 562
 563                         if (wins_owner->min_version<=version && wins_owner->max_version>=version &&
 564                             wins_owner->address.s_addr==wins_ip.s_addr) {
 565                                 if(!add_record_to_winsname(&r->rep.se_rp.wins_name, &max_names, name, type, wins_flags, version, ip_list, num_ips))
 566                                         return;
 567                                 max_names++;
 568                         }
 569
 570                 } else {
 571                         DEBUG(4, ("send_entry_request: not adding name (ttl problem) %s#%02x ttl = %d first IP %s flags = %2x\n",
 572                                   name, type, ttl, inet_ntoa(ip_list[0]), nb_flags));
 573                 }
 574         }
 575
 576         tdb_close(tdb);
 577
 578         DEBUG(4,("send_entry_request, sending %d records\n", max_names));
 579         fill_header(r, OPCODE_NON_NBT, s_ctx, MESSAGE_TYPE_REPLICATE);
 580         r->rep.msg_type=MESSAGE_REP_SEND_ENTRIES_REPLY; /* reply */
 581         r->rep.se_rp.max_names=max_names;
 582 }
 583
 584
 585 /****************************************************************************
 586 .
 587 ****************************************************************************/
 588 static void update_notify_request(GENERIC_PACKET *q, GENERIC_PACKET *r)
 589 {
 590         int i,j,k,l;
 591         UPDATE_NOTIFY_REQUEST *u;
 592         int s_ctx=get_server_assoc(q->header.assoc_ctx);
 593
 594         if (s_ctx==0) {
 595                 DEBUG(4, ("update_notify_request: request for a partner not in our table\n"));
 596                 stop_packet(q, r, STOP_REASON_USER_REASON);
 597                 return;
 598         }
 599
 600         u=&q->rep.un_rq;
 601
 602         /* check if we already have the range of records */
 603
 604         DEBUG(5,("update_notify_request: wins server: %s offered this list of %d records:\n",
 605                 inet_ntoa(u->initiating_wins_server), u->partner_count));
 606
 607         get_our_last_id(&global_wins_table[0][0]);
 608
 609         for (i=0; i<partner_count; i++) {
 610                 if (global_wins_table[0][i].address.s_addr==u->initiating_wins_server.s_addr) {
 611                         DEBUG(5,("update_notify_request: found initiator at index %d\n", i));
 612                         break;
 613                 }
 614         }
 615
 616         /*
 617          * some explanation is required, before someone say it's crap.
 618          *
 619          * let's take an example, we have 2 wins partners, we already now
 620          * that our max id is 10, partner 1 ID is 20 and partner 2 ID is 30
 621          * the array looks like:
 622          *
 623          *      0       1       2
 624          * 0    10      20      30
 625          * 1
 626          * 2
 627          *
 628          * we receive an update from partner 2 saying he has: 1:15, 2:40, 3:50
 629          * we must enlarge the array to add partner 3, it will look like:
 630          *
 631          *      0       1       2       3
 632          * 0    10      20      30
 633          * 1
 634          * 2            15      40      50
 635          *
 636          * now we know, we should pull from partner 2, the records 30->40 of 2 and 0->50 of 3.
 637          * once the pull will be over, our table will look like:
 638          *
 639          *      0       1       2       3
 640          * 0    10      20      40      50
 641          * 1
 642          * 2            15      40      50
 643          *
 644          *
 645          */
 646
 647         for (j=0; j<u->partner_count;j++) {
 648                 /*
 649                  * search if we already have this entry or if it's a new one
 650                  * it can be a new one in case of propagation
 651                  */
 652
 653                 for (k=0; global_wins_table[0][k].address.s_addr!=0 &&
 654                           global_wins_table[0][k].address.s_addr!=u->wins_owner[j].address.s_addr; k++);
 655
 656                 global_wins_table[i][k].address.s_addr=u->wins_owner[j].address.s_addr;
 657                 global_wins_table[i][k].max_version=u->wins_owner[j].max_version;
 658                 global_wins_table[i][k].min_version=u->wins_owner[j].min_version;
 659                 global_wins_table[i][k].type=u->wins_owner[j].type;
 660
 661                 /*
 662                  * in case it's a new one, rewrite the address for all the partner
 663                  * to reserve the slot.
 664                  */
 665
 666                 for(l=0; l<partner_count; l++)
 667                         global_wins_table[l][k].address.s_addr=u->wins_owner[j].address.s_addr;
 668         }
 669
 670         dump_global_table();
 671
 672         stop_packet(q, r, STOP_REASON_USER_REASON);
 673 }
 674
 675 /****************************************************************************
 676 .
 677 ****************************************************************************/
 678 static void send_entry_reply(GENERIC_PACKET *q, GENERIC_PACKET *r)
 679 {
 680         int i,j,k;
 681         struct in_addr partner, server;
 682         pid_t pid;
 683         int s_ctx=get_server_assoc(q->header.assoc_ctx);
 684         WINS_RECORD record;
 685
 686         if (s_ctx==0) {
 687                 DEBUG(1, ("send_entry_reply: request for a partner not in our table\n"));
 688                 stop_packet(q, r, STOP_REASON_USER_REASON);
 689                 return;
 690         }
 691
 692         DEBUG(5,("send_entry_reply:got %d new records\n", q->rep.se_rp.max_names));
 693
 694         /* we got records from a wins partner but that can be from another wins server */
 695         /* hopefully we track that */
 696
 697         /* and the only doc available from MS is wrong ! */
 698
 699         get_server_assoc_table(q->header.assoc_ctx, &partner, &server);
 700
 701         for (j=0; global_wins_table[0][j].address.s_addr!=0; j++) {
 702                 if (global_wins_table[0][j].address.s_addr==server.s_addr) {
 703                         DEBUG(5,("send_entry_reply: found server at index %d\n", j));
 704                         break;
 705                 }
 706         }
 707
 708         pid = pidfile_pid("nmbd");
 709         if (pid == 0) {
 710                 DEBUG(0,("send_entry_reply: Can't find pid for nmbd\n"));
 711                 return;
 712         }
 713
 714         for (k=0; k<q->rep.se_rp.max_names; k++) {
 715                 DEBUG(5,("send_entry_reply: %s<%02x> %d\n", q->rep.se_rp.wins_name[k].name, q->rep.se_rp.wins_name[k].type,
 716                          (int)q->rep.se_rp.wins_name[k].id));
 717
 718                 safe_strcpy(record.name, q->rep.se_rp.wins_name[k].name, 16);
 719                 record.type=q->rep.se_rp.wins_name[k].type;
 720                 record.id=q->rep.se_rp.wins_name[k].id;
 721                 record.wins_flags=q->rep.se_rp.wins_name[k].name_flag&0x00ff;
 722                 record.num_ips=q->rep.se_rp.wins_name[k].num_ip;
 723
 724                 record.wins_ip.s_addr=server.s_addr;
 725
 726                 if (record.num_ips==1)
 727                         record.ip[0]=q->rep.se_rp.wins_name[k].owner;
 728                 else
 729                         for (i=0; i<record.num_ips; i++)
 730                                 record.ip[i]=q->rep.se_rp.wins_name[k].others[i];
 731
 732                 record.nb_flags=0;
 733
 734                 if (record.wins_flags&WINS_NGROUP || record.wins_flags&WINS_SGROUP)
 735                         record.nb_flags|=NB_GROUP;
 736
 737                 if (record.wins_flags&WINS_ACTIVE)
 738                         record.nb_flags|=NB_ACTIVE;
 739
 740                 record.nb_flags|=record.wins_flags&WINS_HNODE;
 741
 742                 message_send_pid(pid, MSG_WINS_NEW_ENTRY, &record, sizeof(record), False);
 743
 744         }
 745
 746         dump_global_table();
 747
 748         /*
 749          * we got some entries,
 750          * ask the partner to send us the map table again
 751          * to get the other servers entries.
 752          *
 753          * we're getting the map table 1 time more than really
 754          * required. We could remove that call, but that
 755          * would complexify the code. I prefer this trade-of.
 756          */
 757         fill_header(r, OPCODE_NON_NBT, s_ctx, MESSAGE_TYPE_REPLICATE);
 758
 759         r->rep.msg_type=MESSAGE_REP_ADD_VERSION_REQUEST;
 760 }
 761
 762 /****************************************************************************
 763 decode the replication message and reply.
 764 ****************************************************************************/
 765 static void replicate(GENERIC_PACKET *q, GENERIC_PACKET *r)
 766 {
 767         switch (q->rep.msg_type) {
 768                 case 0:
 769                         /* add version number map table request */
 770                         send_version_number_map_table(q, r);
 771                         break;
 772                 case 1:
 773                         receive_version_number_map_table(q, r);
 774                         break;
 775                 case 2:
 776                         /* send entry request */
 777                         send_entry_request(q, r);
 778                         break;
 779                 case 3:
 780                         /* send entry reply */
 781                         send_entry_reply(q, r);
 782                         break;
 783                 case 4:
 784                         /* update notification request */
 785                         update_notify_request(q, r);
 786                         break;
 787         }
 788 }
 789
 790 /****************************************************************************
 791 do a switch on the message type, and return the response size
 792 ****************************************************************************/
 793 static BOOL switch_message(GENERIC_PACKET *q, GENERIC_PACKET *r)
 794 {
 795         switch (q->header.mess_type) {
 796                 case 0:
 797                         /* Start association type */
 798                         start_assoc_process(q, r);
 799                         return True;
 800                         break;
 801                 case 1:
 802                         /* start association reply */
 803                         start_assoc_reply(q, r);
 804                         return True;
 805                         break;
 806                 case 2:
 807                         /* stop association message */
 808                         return False;
 809                         break;
 810                 case 3:
 811                         /* replication message */
 812                         replicate(q, r);
 813                         return True;
 814                         break;
 815         }
 816
 817         return False;
 818 }
 819
 820
 821 /****************************************************************************
 822   construct a reply to the incoming packet
 823 ****************************************************************************/
 824 void construct_reply(struct wins_packet_struct *p)
 825 {
 826         GENERIC_PACKET r;
 827         struct BUFFER buffer;
 828
 829         buffer.buffer=NULL;
 830         buffer.offset=0;
 831         buffer.length=0;
 832
 833         DEBUG(5,("dump: received packet\n"));
 834         dump_generic_packet(p->packet);
 835
 836         /* Verify if the request we got is from a listed partner */
 837         if (!check_partner(p->packet->header.assoc_ctx)) {
 838                 fstring peer;
 839                 struct in_addr addr;
 840                 int i;
 841                 fstrcpy(peer,get_socket_addr(p->fd));
 842                 addr=*interpret_addr2(peer);
 843
 844                 for (i=1; i<partner_count; i++)
 845                         if (ip_equal(addr, global_wins_table[0][i].address))
 846                                 break;
 847
 848                 if (i==partner_count) {
 849                         DEBUG(1,("construct_reply: got a request from a non peer machine: %s\n", peer));
 850                         stop_packet(p->packet, &r, STOP_REASON_AUTH_FAILED);
 851                         p->stop_packet=True;
 852                         encode_generic_packet(&buffer, &r);
 853                         if (!send_smb(p->fd, buffer.buffer))
 854                                 exit_server("process_smb: send_smb failed.");
 855                         return;
 856                 }
 857         }
 858
 859         if (switch_message(p->packet, &r)) {
 860                 encode_generic_packet(&buffer, &r);
 861                 DEBUG(5,("dump: sending packet\n"));
 862                 dump_generic_packet(&r);
 863
 864                 if(buffer.offset > 0) {
 865                         if (!send_smb(p->fd, buffer.buffer))
 866                                 exit_server("process_smb: send_smb failed.");
 867                 }
 868         }
 869
 870         /* if we got a stop assoc or if we send a stop assoc, close the fd after */
 871         if (p->packet->header.mess_type==MESSAGE_TYPE_STOP_ASSOC ||
 872             r.header.mess_type==MESSAGE_TYPE_STOP_ASSOC) {
 873                 remove_partner(p->packet->header.assoc_ctx);
 874                 p->stop_packet=True;
 875         }
 876 }
 877
 878 /****************************************************************************
 879   contact periodically our wins partner to do a pull replication
 880 ****************************************************************************/
 881 void run_pull_replication(time_t t)
 882 {
 883         /* we pull every 30 minutes to query about new records*/
 884         int i, s;
 885         struct BUFFER buffer;
 886         GENERIC_PACKET p;
 887
 888         buffer.buffer=NULL;
 889         buffer.offset=0;
 890         buffer.length=0;
 891
 892         for (i=1; i<partner_count; i++) {
 893                 if (global_wins_table[0][i].last_pull < t) {
 894                         global_wins_table[0][i].last_pull=t+30*60; /* next in 30 minutes */
 895
 896                         /* contact the wins server */
 897                         p.header.mess_type=MESSAGE_TYPE_START_ASSOC_REQUEST;
 898                         p.header.opcode=OPCODE_NON_NBT;
 899                         p.header.assoc_ctx=0;
 900                         p.sa_rq.assoc_ctx=(int)t;
 901                         p.sa_rq.min_ver=1;
 902                         p.sa_rq.maj_ver=1;
 903
 904                         DEBUG(3,("run_pull_replication: contacting wins server %s.\n", inet_ntoa(global_wins_table[0][i].address)));
 905                         encode_generic_packet(&buffer, &p);
 906                         dump_generic_packet(&p);
 907
 908                         /* send the packet to the server and add the descriptor to receive answers */
 909                         s=open_socket_out(SOCK_STREAM, &global_wins_table[0][i].address, 42, LONG_CONNECT_TIMEOUT);
 910                         if (s==-1) {
 911                                 DEBUG(0,("run_pull_replication: can't contact wins server %s.\n", inet_ntoa(global_wins_table[0][i].address)));
 912                                 return;
 913                         }
 914
 915                         if(buffer.offset > 0) {
 916                                 if (!send_smb(s, buffer.buffer))
 917                                         exit_server("run_pull_replication: send_smb failed.");
 918                         }
 919
 920                         add_fd_to_sock_array(s);
 921                         FD_SET(s, listen_set);
 922
 923                         /* add ourself as a client */
 924                         add_partner((int)t, 0, True, False);
 925                 }
 926         }
 927 }
 928
 929 /****************************************************************************
 930   contact periodically our wins partner to do a push replication
 931 ****************************************************************************/
 932 void run_push_replication(time_t t)
 933 {
 934         /* we push every 30 minutes or 25 new entries */
 935         int i, s;
 936         struct BUFFER buffer;
 937         GENERIC_PACKET p;
 938
 939         buffer.buffer=NULL;
 940         buffer.offset=0;
 941         buffer.length=0;
 942
 943         for (i=1; i<partner_count; i++) {
 944                 if (global_wins_table[0][i].last_pull < t) {
 945                         global_wins_table[0][i].last_pull=t+30*60; /* next in 30 minutes */
 946
 947                         /* contact the wins server */
 948                         p.header.mess_type=MESSAGE_TYPE_START_ASSOC_REQUEST;
 949                         p.header.opcode=OPCODE_NON_NBT;
 950                         p.header.assoc_ctx=0;
 951                         p.sa_rq.assoc_ctx=(int)t;
 952                         p.sa_rq.min_ver=1;
 953                         p.sa_rq.maj_ver=1;
 954
 955                         DEBUG(3,("run_push_replication: contacting wins server %s.\n", inet_ntoa(global_wins_table[0][i].address)));
 956                         encode_generic_packet(&buffer, &p);
 957                         dump_generic_packet(&p);
 958
 959                         /* send the packet to the server and add the descriptor to receive answers */
 960                         s=open_socket_out(SOCK_STREAM, &global_wins_table[0][i].address, 42, LONG_CONNECT_TIMEOUT);
 961                         if (s==-1) {
 962                                 DEBUG(0,("run_push_replication: can't contact wins server %s.\n", inet_ntoa(global_wins_table[0][i].address)));
 963                                 return;
 964                         }
 965
 966                         if(buffer.offset > 0) {
 967                                 if (!send_smb(s, buffer.buffer))
 968                                         exit_server("run_push_replication: send_smb failed.");
 969                         }
 970
 971                         add_fd_to_sock_array(s);
 972                         FD_SET(s, listen_set);
 973
 974                         /* add ourself as a client */
 975                         add_partner((int)t, 0, False, True);
 976                 }
 977         }
 978 }
 979