source4/wrepl_server/wrepl_server.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3
   4    WINS Replication server
   5
   6    Copyright (C) Stefan Metzmacher      2005
   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 #include "includes.h"
  24 #include "dlinklist.h"
  25 #include "lib/events/events.h"
  26 #include "lib/socket/socket.h"
  27 #include "smbd/service_task.h"
  28 #include "smbd/service_stream.h"
  29 #include "lib/messaging/irpc.h"
  30 #include "librpc/gen_ndr/ndr_winsrepl.h"
  31 #include "wrepl_server/wrepl_server.h"
  32 #include "nbt_server/wins/winsdb.h"
  33 #include "ldb/include/ldb.h"
  34 #include "ldb/include/ldb_errors.h"
  35 #include "auth/auth.h"
  36
  37 static struct ldb_context *wins_config_db_connect(TALLOC_CTX *mem_ctx)
  38 {
  39         return ldb_wrap_connect(mem_ctx, private_path(mem_ctx, lp_wins_config_url()),
  40                                 system_session(mem_ctx), NULL, 0, NULL);
  41 }
  42
  43 /*
  44   open winsdb
  45 */
  46 static NTSTATUS wreplsrv_open_winsdb(struct wreplsrv_service *service)
  47 {
  48         service->wins_db     = winsdb_connect(service);
  49         if (!service->wins_db) {
  50                 return NT_STATUS_INTERNAL_DB_ERROR;
  51         }
  52
  53         service->config.ldb = wins_config_db_connect(service);
  54         if (!service->config.ldb) {
  55                 return NT_STATUS_INTERNAL_DB_ERROR;
  56         }
  57
  58         /* the default renew interval is 6 days */
  59         service->config.renew_interval    = lp_parm_int(-1,"wreplsrv","renew_interval", 6*24*60*60);
  60
  61         /* the default tombstone (extinction) interval is 6 days */
  62         service->config.tombstone_interval= lp_parm_int(-1,"wreplsrv","tombstone_interval", 6*24*60*60);
  63
  64         /* the default tombstone (extinction) timeout is 1 day */
  65         service->config.tombstone_timeout = lp_parm_int(-1,"wreplsrv","tombstone_timeout", 1*24*60*60);
  66
  67         /* the default tombstone extra timeout is 3 days */
  68         service->config.tombstone_extra_timeout = lp_parm_int(-1,"wreplsrv","tombstone_extra_timeout", 3*24*60*60);
  69
  70         /* the default verify interval is 24 days */
  71         service->config.verify_interval   = lp_parm_int(-1,"wreplsrv","verify_interval", 24*24*60*60);
  72
  73         /* the default scavenging interval is 'renew_interval/2' */
  74         service->config.scavenging_interval=lp_parm_int(-1,"wreplsrv","scavenging_interval",
  75                                                         service->config.renew_interval/2);
  76
  77         /* the maximun interval to the next periodic processing event */
  78         service->config.periodic_interval = lp_parm_int(-1,"wreplsrv","periodic_interval", 60);
  79
  80         return NT_STATUS_OK;
  81 }
  82
  83 struct wreplsrv_partner *wreplsrv_find_partner(struct wreplsrv_service *service, const char *peer_addr)
  84 {
  85         struct wreplsrv_partner *cur;
  86
  87         for (cur = service->partners; cur; cur = cur->next) {
  88                 if (strcmp(cur->address, peer_addr) == 0) {
  89                         return cur;
  90                 }
  91         }
  92
  93         return NULL;
  94 }
  95
  96 /*
  97   load our replication partners
  98 */
  99 static NTSTATUS wreplsrv_load_partners(struct wreplsrv_service *service)
 100 {
 101         struct ldb_result *res = NULL;
 102         int ret;
 103         TALLOC_CTX *tmp_ctx = talloc_new(service);
 104         int i;
 105
 106         /* find the record in the WINS database */
 107         ret = ldb_search(service->config.ldb, ldb_dn_explode(tmp_ctx, "CN=PARTNERS"), LDB_SCOPE_SUBTREE,
 108                          "(objectClass=wreplPartner)", NULL, &res);
 109         if (ret != LDB_SUCCESS) goto failed;
 110         talloc_steal(tmp_ctx, res);
 111         if (res->count == 0) goto done;
 112
 113         for (i=0; i < res->count; i++) {
 114                 struct wreplsrv_partner *partner;
 115
 116                 partner = talloc_zero(service, struct wreplsrv_partner);
 117                 if (partner == NULL) goto failed;
 118
 119                 partner->service                = service;
 120                 partner->address                = ldb_msg_find_string(res->msgs[i], "address", NULL);
 121                 if (!partner->address) goto failed;
 122                 partner->name                   = ldb_msg_find_string(res->msgs[i], "name", partner->address);
 123                 partner->type                   = ldb_msg_find_uint(res->msgs[i], "type", WINSREPL_PARTNER_BOTH);
 124                 partner->pull.interval          = ldb_msg_find_uint(res->msgs[i], "pullInterval",
 125                                                                     WINSREPL_DEFAULT_PULL_INTERVAL);
 126                 partner->pull.retry_interval    = ldb_msg_find_uint(res->msgs[i], "pullRetryInterval",
 127                                                                     WINSREPL_DEFAULT_PULL_RETRY_INTERVAL);
 128                 partner->our_address            = ldb_msg_find_string(res->msgs[i], "ourAddress", NULL);
 129                 partner->push.change_count      = ldb_msg_find_uint(res->msgs[i], "pushChangeCount",
 130                                                                     WINSREPL_DEFAULT_PUSH_CHANGE_COUNT);
 131                 partner->push.use_inform        = ldb_msg_find_uint(res->msgs[i], "pushUseInform", False);
 132
 133                 talloc_steal(partner, partner->address);
 134                 talloc_steal(partner, partner->name);
 135                 talloc_steal(partner, partner->our_address);
 136
 137                 DLIST_ADD(service->partners, partner);
 138
 139                 DEBUG(3,("wreplsrv_load_partners: found partner: %s type: 0x%X\n",
 140                         partner->address, partner->type));
 141         }
 142 done:
 143         DEBUG(1,("wreplsrv_load_partners: %u partners found\n", res->count));
 144
 145         talloc_free(tmp_ctx);
 146         return NT_STATUS_OK;
 147 failed:
 148         talloc_free(tmp_ctx);
 149         return NT_STATUS_FOOBAR;
 150 }
 151
 152 BOOL wreplsrv_is_our_address(struct wreplsrv_service *service, const char *address)
 153 {
 154         const char *our_address;
 155
 156         if (lp_interfaces() && lp_bind_interfaces_only()) {
 157                 int num_interfaces = iface_count();
 158                 int i;
 159                 for(i = 0; i < num_interfaces; i++) {
 160                         our_address = iface_n_ip(i);
 161                         if (strcasecmp(our_address, address) == 0) {
 162                                 return True;
 163                         }
 164                 }
 165         } else {
 166                 our_address = lp_socket_address();
 167                 if (strcasecmp(our_address, address) == 0) {
 168                         return True;
 169                 }
 170         }
 171
 172         return False;
 173 }
 174
 175 uint64_t wreplsrv_local_max_version(struct wreplsrv_service *service)
 176 {
 177         int ret;
 178         struct ldb_context *ldb = service->wins_db;
 179         struct ldb_dn *dn;
 180         struct ldb_result *res = NULL;
 181         TALLOC_CTX *tmp_ctx = talloc_new(service);
 182         uint64_t maxVersion = 0;
 183
 184         dn = ldb_dn_explode(tmp_ctx, "CN=VERSION");
 185         if (!dn) goto failed;
 186
 187         /* find the record in the WINS database */
 188         ret = ldb_search(ldb, dn, LDB_SCOPE_BASE,
 189                          NULL, NULL, &res);
 190         if (ret != LDB_SUCCESS) goto failed;
 191         talloc_steal(tmp_ctx, res);
 192         if (res->count > 1) goto failed;
 193
 194         if (res->count == 1) {
 195                 maxVersion = ldb_msg_find_uint64(res->msgs[0], "maxVersion", 0);
 196         }
 197
 198 failed:
 199         talloc_free(tmp_ctx);
 200         return maxVersion;
 201 }
 202
 203 NTSTATUS wreplsrv_fill_wrepl_table(struct wreplsrv_service *service,
 204                                    TALLOC_CTX *mem_ctx,
 205                                    struct wrepl_table *table_out,
 206                                    const char *our_ip,
 207                                    const char *initiator,
 208                                    BOOL full_table)
 209 {
 210         struct wreplsrv_owner *cur;
 211         uint64_t local_max_version;
 212         uint32_t i = 0;
 213
 214         table_out->partner_count        = 0;
 215         table_out->partners             = NULL;
 216         table_out->initiator            = initiator;
 217
 218         local_max_version = wreplsrv_local_max_version(service);
 219         if (local_max_version > 0) {
 220                 table_out->partner_count++;
 221         }
 222
 223         for (cur = service->table; full_table && cur; cur = cur->next) {
 224                 table_out->partner_count++;
 225         }
 226
 227         table_out->partners = talloc_array(mem_ctx, struct wrepl_wins_owner, table_out->partner_count);
 228         NT_STATUS_HAVE_NO_MEMORY(table_out->partners);
 229
 230         if (local_max_version > 0) {
 231                 table_out->partners[i].address          = our_ip;
 232                 table_out->partners[i].min_version      = 0;
 233                 table_out->partners[i].max_version      = local_max_version;
 234                 table_out->partners[i].type             = 1;
 235                 i++;
 236         }
 237
 238         for (cur = service->table; full_table && cur; cur = cur->next) {
 239                 table_out->partners[i] = cur->owner;
 240                 i++;
 241         }
 242
 243         return NT_STATUS_OK;
 244 }
 245
 246 struct wreplsrv_owner *wreplsrv_find_owner(struct wreplsrv_owner *table, const char *wins_owner)
 247 {
 248         struct wreplsrv_owner *cur;
 249
 250         for (cur = table; cur; cur = cur->next) {
 251                 if (strcmp(cur->owner.address, wins_owner) == 0) {
 252                         return cur;
 253                 }
 254         }
 255
 256         return NULL;
 257 }
 258
 259 /*
 260  update the wins_owner_table max_version, if the given version is the highest version
 261  if no entry for the wins_owner exists yet, create one
 262 */
 263 NTSTATUS wreplsrv_add_table(struct wreplsrv_service *service,
 264                             TALLOC_CTX *mem_ctx, struct wreplsrv_owner **_table,
 265                             const char *wins_owner, uint64_t version)
 266 {
 267         struct wreplsrv_owner *table = *_table;
 268         struct wreplsrv_owner *cur;
 269
 270         if (strcmp(WINSDB_OWNER_LOCAL, wins_owner) == 0) {
 271                 return NT_STATUS_OK;
 272         }
 273
 274         cur = wreplsrv_find_owner(table, wins_owner);
 275
 276         /* if it doesn't exists yet, create one */
 277         if (!cur) {
 278                 cur = talloc_zero(mem_ctx, struct wreplsrv_owner);
 279                 NT_STATUS_HAVE_NO_MEMORY(cur);
 280
 281                 cur->owner.address      = talloc_strdup(cur, wins_owner);
 282                 NT_STATUS_HAVE_NO_MEMORY(cur->owner.address);
 283                 cur->owner.min_version  = 0;
 284                 cur->owner.max_version  = 0;
 285                 cur->owner.type         = 1; /* don't know why this is always 1 */
 286
 287                 cur->partner            = wreplsrv_find_partner(service, wins_owner);
 288
 289                 DLIST_ADD(table, cur);
 290                 *_table = table;
 291         }
 292
 293         /* the min_version is always 0 here, and won't be updated */
 294
 295         /* if the given version is higher the then current nax_version, update */
 296         if (cur->owner.max_version < version) {
 297                 cur->owner.max_version = version;
 298         }
 299
 300         return NT_STATUS_OK;
 301 }
 302
 303 /*
 304   load the partner table
 305 */
 306 static NTSTATUS wreplsrv_load_table(struct wreplsrv_service *service)
 307 {
 308         struct ldb_result *res = NULL;
 309         int ret;
 310         NTSTATUS status;
 311         TALLOC_CTX *tmp_ctx = talloc_new(service);
 312         int i;
 313         const char *wins_owner;
 314         uint64_t version;
 315         const char * const attrs[] = {
 316                 "winsOwner",
 317                 "versionID",
 318                 NULL
 319         };
 320
 321         /* find the record in the WINS database */
 322         ret = ldb_search(service->wins_db, NULL, LDB_SCOPE_SUBTREE,
 323                          "(objectClass=winsRecord)", attrs, &res);
 324         status = NT_STATUS_INTERNAL_DB_CORRUPTION;
 325         if (ret != LDB_SUCCESS) goto failed;
 326         talloc_steal(tmp_ctx, res);
 327         if (res->count == 0) goto done;
 328
 329         for (i=0; i < res->count; i++) {
 330                 wins_owner     = ldb_msg_find_string(res->msgs[i], "winsOwner", NULL);
 331                 version        = ldb_msg_find_uint64(res->msgs[i], "versionID", 0);
 332
 333                 if (wins_owner) {
 334                         status = wreplsrv_add_table(service,
 335                                                     service, &service->table,
 336                                                     wins_owner, version);
 337                         if (!NT_STATUS_IS_OK(status)) goto failed;
 338                 }
 339                 talloc_free(res->msgs[i]);
 340
 341                 /* TODO: what's abut the per address owners? */
 342         }
 343 done:
 344         talloc_free(tmp_ctx);
 345         return NT_STATUS_OK;
 346 failed:
 347         talloc_free(tmp_ctx);
 348         return status;
 349 }
 350
 351 /*
 352   setup our replication partners
 353 */
 354 static NTSTATUS wreplsrv_setup_partners(struct wreplsrv_service *service)
 355 {
 356         NTSTATUS status;
 357
 358         status = wreplsrv_load_partners(service);
 359         NT_STATUS_NOT_OK_RETURN(status);
 360
 361         status = wreplsrv_load_table(service);
 362         NT_STATUS_NOT_OK_RETURN(status);
 363
 364         return NT_STATUS_OK;
 365 }
 366
 367 /*
 368   startup the wrepl task
 369 */
 370 static void wreplsrv_task_init(struct task_server *task)
 371 {
 372         NTSTATUS status;
 373         struct wreplsrv_service *service;
 374
 375         service = talloc_zero(task, struct wreplsrv_service);
 376         if (!service) {
 377                 task_server_terminate(task, "wreplsrv_task_init: out of memory");
 378                 return;
 379         }
 380         service->task           = task;
 381         service->startup_time   = timeval_current();
 382         task->private           = service;
 383
 384         /*
 385          * setup up all partners, and open the winsdb
 386          */
 387         status = wreplsrv_open_winsdb(service);
 388         if (!NT_STATUS_IS_OK(status)) {
 389                 task_server_terminate(task, "wreplsrv_task_init: wreplsrv_open_winsdb() failed");
 390                 return;
 391         }
 392
 393         /*
 394          * setup timed events for each partner we want to pull from
 395          */
 396         status = wreplsrv_setup_partners(service);
 397         if (!NT_STATUS_IS_OK(status)) {
 398                 task_server_terminate(task, "wreplsrv_task_init: wreplsrv_setup_partners() failed");
 399                 return;
 400         }
 401
 402         /*
 403          * setup listen sockets, so we can anwser requests from our partners,
 404          * which pull from us
 405          */
 406         status = wreplsrv_setup_sockets(service);
 407         if (!NT_STATUS_IS_OK(status)) {
 408                 task_server_terminate(task, "wreplsrv_task_init: wreplsrv_setup_sockets() failed");
 409                 return;
 410         }
 411
 412         status = wreplsrv_setup_periodic(service);
 413         if (!NT_STATUS_IS_OK(status)) {
 414                 task_server_terminate(task, "wreplsrv_task_init: wreplsrv_setup_periodic() failed");
 415                 return;
 416         }
 417
 418         irpc_add_name(task->msg_ctx, "wrepl_server");
 419 }
 420
 421 /*
 422   initialise the WREPL server
 423  */
 424 static NTSTATUS wreplsrv_init(struct event_context *event_ctx, const struct model_ops *model_ops)
 425 {
 426         if (!lp_wins_support()) {
 427                 return NT_STATUS_OK;
 428         }
 429
 430         return task_server_startup(event_ctx, model_ops, wreplsrv_task_init);
 431 }
 432
 433 /*
 434   register ourselves as a available server
 435 */
 436 NTSTATUS server_service_wrepl_init(void)
 437 {
 438         return register_server_service("wrepl", wreplsrv_init);
 439 }