source4/dsdb/repl/drepl_periodic.c

   1 /*
   2    Unix SMB/CIFS mplementation.
   3    DSDB replication service periodic handling
   4
   5    Copyright (C) Stefan Metzmacher 2007
   6
   7    This program is free software; you can redistribute it and/or modify
   8    it under the terms of the GNU General Public License as published by
   9    the Free Software Foundation; either version 3 of the License, or
  10    (at your option) any later version.
  11
  12    This program is distributed in the hope that it will be useful,
  13    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15    GNU General Public License for more details.
  16
  17    You should have received a copy of the GNU General Public License
  18    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19
  20 */
  21
  22 #include "includes.h"
  23 #include "lib/events/events.h"
  24 #include "dsdb/samdb/samdb.h"
  25 #include "auth/auth.h"
  26 #include "smbd/service.h"
  27 #include "dsdb/repl/drepl_service.h"
  28 #include <ldb_errors.h>
  29 #include "../lib/util/dlinklist.h"
  30 #include "librpc/gen_ndr/ndr_misc.h"
  31 #include "librpc/gen_ndr/ndr_drsuapi.h"
  32 #include "librpc/gen_ndr/ndr_drsblobs.h"
  33
  34 static void dreplsrv_periodic_run(struct dreplsrv_service *service);
  35
  36 static void dreplsrv_periodic_handler_te(struct tevent_context *ev, struct tevent_timer *te,
  37                                          struct timeval t, void *ptr)
  38 {
  39         struct dreplsrv_service *service = talloc_get_type(ptr, struct dreplsrv_service);
  40         WERROR status;
  41
  42         service->periodic.te = NULL;
  43
  44         dreplsrv_periodic_run(service);
  45
  46         status = dreplsrv_periodic_schedule(service, service->periodic.interval);
  47         if (!W_ERROR_IS_OK(status)) {
  48                 task_server_terminate(service->task, win_errstr(status), false);
  49                 return;
  50         }
  51 }
  52
  53 WERROR dreplsrv_periodic_schedule(struct dreplsrv_service *service, uint32_t next_interval)
  54 {
  55         TALLOC_CTX *tmp_mem;
  56         struct tevent_timer *new_te;
  57         struct timeval next_time;
  58
  59         /* prevent looping */
  60         if (next_interval == 0) next_interval = 1;
  61
  62         next_time = timeval_current_ofs(next_interval, 50);
  63
  64         if (service->periodic.te) {
  65                 /*
  66                  * if the timestamp of the new event is higher,
  67                  * as current next we don't need to reschedule
  68                  */
  69                 if (timeval_compare(&next_time, &service->periodic.next_event) > 0) {
  70                         return WERR_OK;
  71                 }
  72         }
  73
  74         /* reset the next scheduled timestamp */
  75         service->periodic.next_event = next_time;
  76
  77         new_te = tevent_add_timer(service->task->event_ctx, service,
  78                                  service->periodic.next_event,
  79                                  dreplsrv_periodic_handler_te, service);
  80         W_ERROR_HAVE_NO_MEMORY(new_te);
  81
  82         tmp_mem = talloc_new(service);
  83         DEBUG(4,("dreplsrv_periodic_schedule(%u) %sscheduled for: %s\n",
  84                 next_interval,
  85                 (service->periodic.te?"re":""),
  86                 nt_time_string(tmp_mem, timeval_to_nttime(&next_time))));
  87         talloc_free(tmp_mem);
  88
  89         talloc_free(service->periodic.te);
  90         service->periodic.te = new_te;
  91
  92         return WERR_OK;
  93 }
  94
  95 static void dreplsrv_periodic_run(struct dreplsrv_service *service)
  96 {
  97         TALLOC_CTX *mem_ctx;
  98
  99         DEBUG(4,("dreplsrv_periodic_run(): schedule pull replication\n"));
 100
 101         /*
 102          * KCC or some administrative tool
 103          * might have changed Topology graph
 104          * i.e. repsFrom/repsTo
 105          */
 106         dreplsrv_refresh_partitions(service);
 107
 108         mem_ctx = talloc_new(service);
 109         dreplsrv_schedule_pull_replication(service, mem_ctx);
 110         talloc_free(mem_ctx);
 111
 112         DEBUG(4,("dreplsrv_periodic_run(): run pending_ops memory=%u\n",
 113                  (unsigned)talloc_total_blocks(service)));
 114
 115         dreplsrv_ridalloc_check_rid_pool(service);
 116
 117         dreplsrv_run_pending_ops(service);
 118 }
 119
 120 /*
 121   run the next pending op, either a notify or a pull
 122  */
 123 void dreplsrv_run_pending_ops(struct dreplsrv_service *s)
 124 {
 125         if (!s->ops.notifies && !s->ops.pending) {
 126                 return;
 127         }
 128         if (!s->ops.notifies ||
 129             (s->ops.pending &&
 130              s->ops.notifies->schedule_time > s->ops.pending->schedule_time)) {
 131                 dreplsrv_run_pull_ops(s);
 132         } else {
 133                 dreplsrv_notify_run_ops(s);
 134         }
 135 }
 136
 137 static void dreplsrv_pending_run(struct dreplsrv_service *service);
 138
 139 static void dreplsrv_pending_handler_te(struct tevent_context *ev, struct tevent_timer *te,
 140                                         struct timeval t, void *ptr)
 141 {
 142         struct dreplsrv_service *service = talloc_get_type(ptr, struct dreplsrv_service);
 143
 144         service->pending.te = NULL;
 145
 146         dreplsrv_pending_run(service);
 147 }
 148
 149 WERROR dreplsrv_pendingops_schedule(struct dreplsrv_service *service, uint32_t next_interval)
 150 {
 151         TALLOC_CTX *tmp_mem;
 152         struct tevent_timer *new_te;
 153         struct timeval next_time;
 154
 155         /* prevent looping */
 156         if (next_interval == 0) {
 157                 next_interval = 1;
 158         }
 159
 160         next_time = timeval_current_ofs(next_interval, 50);
 161
 162         if (service->pending.te) {
 163                 /*
 164                  * if the timestamp of the new event is higher,
 165                  * as current next we don't need to reschedule
 166                  */
 167                 if (timeval_compare(&next_time, &service->pending.next_event) > 0) {
 168                         return WERR_OK;
 169                 }
 170         }
 171
 172         /* reset the next scheduled timestamp */
 173         service->pending.next_event = next_time;
 174
 175         new_te = tevent_add_timer(service->task->event_ctx, service,
 176                                   service->pending.next_event,
 177                                  dreplsrv_pending_handler_te, service);
 178         W_ERROR_HAVE_NO_MEMORY(new_te);
 179
 180         tmp_mem = talloc_new(service);
 181         DEBUG(4,("dreplsrv_pending_schedule(%u) %sscheduled for: %s\n",
 182                 next_interval,
 183                 (service->pending.te?"re":""),
 184                 nt_time_string(tmp_mem, timeval_to_nttime(&next_time))));
 185         talloc_free(tmp_mem);
 186
 187         talloc_free(service->pending.te);
 188         service->pending.te = new_te;
 189
 190         return WERR_OK;
 191 }
 192
 193 static void dreplsrv_pending_run(struct dreplsrv_service *service)
 194 {
 195         dreplsrv_run_pending_ops(service);
 196 }