s4:drepl_out_helpers: don't look at the internals of 'struct rpc_request'
[ira/wip.git] / source4 / dsdb / repl / drepl_service.c
1 /* 
2    Unix SMB/CIFS mplementation.
3    DSDB replication service
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 "dsdb/samdb/samdb.h"
24 #include "auth/auth.h"
25 #include "smbd/service.h"
26 #include "lib/events/events.h"
27 #include "lib/messaging/irpc.h"
28 #include "dsdb/repl/drepl_service.h"
29 #include "lib/ldb/include/ldb_errors.h"
30 #include "../lib/util/dlinklist.h"
31 #include "librpc/gen_ndr/ndr_misc.h"
32 #include "librpc/gen_ndr/ndr_drsuapi.h"
33 #include "librpc/gen_ndr/ndr_drsblobs.h"
34 #include "param/param.h"
35
36 static WERROR dreplsrv_init_creds(struct dreplsrv_service *service)
37 {
38         service->system_session_info = system_session(service->task->lp_ctx);
39         if (service->system_session_info == NULL) {
40                 return WERR_NOMEM;
41         }
42
43         return WERR_OK;
44 }
45
46 static WERROR dreplsrv_connect_samdb(struct dreplsrv_service *service, struct loadparm_context *lp_ctx)
47 {
48         const struct GUID *ntds_guid;
49         struct drsuapi_DsBindInfo28 *bind_info28;
50
51         service->samdb = samdb_connect(service, service->task->event_ctx, lp_ctx, service->system_session_info);
52         if (!service->samdb) {
53                 return WERR_DS_UNAVAILABLE;
54         }
55
56         ntds_guid = samdb_ntds_objectGUID(service->samdb);
57         if (!ntds_guid) {
58                 return WERR_DS_UNAVAILABLE;
59         }
60
61         service->ntds_guid = *ntds_guid;
62
63         bind_info28                             = &service->bind_info28;
64         bind_info28->supported_extensions       |= DRSUAPI_SUPPORTED_EXTENSION_BASE;
65         bind_info28->supported_extensions       |= DRSUAPI_SUPPORTED_EXTENSION_ASYNC_REPLICATION;
66         bind_info28->supported_extensions       |= DRSUAPI_SUPPORTED_EXTENSION_REMOVEAPI;
67         bind_info28->supported_extensions       |= DRSUAPI_SUPPORTED_EXTENSION_MOVEREQ_V2;
68         bind_info28->supported_extensions       |= DRSUAPI_SUPPORTED_EXTENSION_GETCHG_COMPRESS;
69         bind_info28->supported_extensions       |= DRSUAPI_SUPPORTED_EXTENSION_DCINFO_V1;
70         bind_info28->supported_extensions       |= DRSUAPI_SUPPORTED_EXTENSION_RESTORE_USN_OPTIMIZATION;
71         bind_info28->supported_extensions       |= DRSUAPI_SUPPORTED_EXTENSION_KCC_EXECUTE;
72         bind_info28->supported_extensions       |= DRSUAPI_SUPPORTED_EXTENSION_ADDENTRY_V2;
73         bind_info28->supported_extensions       |= DRSUAPI_SUPPORTED_EXTENSION_LINKED_VALUE_REPLICATION;
74         bind_info28->supported_extensions       |= DRSUAPI_SUPPORTED_EXTENSION_DCINFO_V2;
75         bind_info28->supported_extensions       |= DRSUAPI_SUPPORTED_EXTENSION_INSTANCE_TYPE_NOT_REQ_ON_MOD;
76         bind_info28->supported_extensions       |= DRSUAPI_SUPPORTED_EXTENSION_CRYPTO_BIND;
77         bind_info28->supported_extensions       |= DRSUAPI_SUPPORTED_EXTENSION_GET_REPL_INFO;
78         bind_info28->supported_extensions       |= DRSUAPI_SUPPORTED_EXTENSION_STRONG_ENCRYPTION;
79         bind_info28->supported_extensions       |= DRSUAPI_SUPPORTED_EXTENSION_DCINFO_V01;
80         bind_info28->supported_extensions       |= DRSUAPI_SUPPORTED_EXTENSION_TRANSITIVE_MEMBERSHIP;
81         bind_info28->supported_extensions       |= DRSUAPI_SUPPORTED_EXTENSION_ADD_SID_HISTORY;
82         bind_info28->supported_extensions       |= DRSUAPI_SUPPORTED_EXTENSION_POST_BETA3;
83         bind_info28->supported_extensions       |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREQ_V5;
84         bind_info28->supported_extensions       |= DRSUAPI_SUPPORTED_EXTENSION_GET_MEMBERSHIPS2;
85         bind_info28->supported_extensions       |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREQ_V6;
86         bind_info28->supported_extensions       |= DRSUAPI_SUPPORTED_EXTENSION_NONDOMAIN_NCS;
87         bind_info28->supported_extensions       |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREQ_V8;
88         bind_info28->supported_extensions       |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREPLY_V5;
89         bind_info28->supported_extensions       |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREPLY_V6;
90         bind_info28->supported_extensions       |= DRSUAPI_SUPPORTED_EXTENSION_ADDENTRYREPLY_V3;
91         bind_info28->supported_extensions       |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREPLY_V7;
92         bind_info28->supported_extensions       |= DRSUAPI_SUPPORTED_EXTENSION_VERIFY_OBJECT;
93 #if 0 /* we don't support XPRESS compression yet */
94         bind_info28->supported_extensions       |= DRSUAPI_SUPPORTED_EXTENSION_XPRESS_COMPRESS;
95 #endif
96         /* TODO: fill in site_guid */
97         bind_info28->site_guid                  = GUID_zero();
98         /* TODO: find out how this is really triggered! */
99         bind_info28->pid                        = 0;
100         bind_info28->repl_epoch                 = 0;
101
102         return WERR_OK;
103 }
104
105 /*
106   DsReplicaSync messages from the DRSUAPI server are forwarded here
107  */
108 static NTSTATUS drepl_replica_sync(struct irpc_message *msg, 
109                                    struct drsuapi_DsReplicaSync *r)
110 {
111         struct dreplsrv_service *service = talloc_get_type(msg->private_data,
112                                                            struct dreplsrv_service);
113         struct GUID *guid = &r->in.req->req1.naming_context->guid;
114
115         r->out.result = dreplsrv_schedule_partition_pull_by_guid(service, msg, guid);
116         if (W_ERROR_IS_OK(r->out.result)) {
117                 DEBUG(3,("drepl_replica_sync: forcing sync of partition %s\n",
118                          GUID_string(msg, guid)));
119                 dreplsrv_run_pending_ops(service);
120         } else {
121                 DEBUG(3,("drepl_replica_sync: failed setup of sync of partition %s - %s\n",
122                          GUID_string(msg, guid), win_errstr(r->out.result)));
123         }
124         return NT_STATUS_OK;
125 }
126
127 /*
128   startup the dsdb replicator service task
129 */
130 static void dreplsrv_task_init(struct task_server *task)
131 {
132         WERROR status;
133         struct dreplsrv_service *service;
134         uint32_t periodic_startup_interval;
135
136         switch (lp_server_role(task->lp_ctx)) {
137         case ROLE_STANDALONE:
138                 task_server_terminate(task, "dreplsrv: no DSDB replication required in standalone configuration", 
139                                       false);
140                 return;
141         case ROLE_DOMAIN_MEMBER:
142                 task_server_terminate(task, "dreplsrv: no DSDB replication required in domain member configuration", 
143                                       false);
144                 return;
145         case ROLE_DOMAIN_CONTROLLER:
146                 /* Yes, we want DSDB replication */
147                 break;
148         }
149
150         task_server_set_title(task, "task[dreplsrv]");
151
152         service = talloc_zero(task, struct dreplsrv_service);
153         if (!service) {
154                 task_server_terminate(task, "dreplsrv_task_init: out of memory", true);
155                 return;
156         }
157         service->task           = task;
158         service->startup_time   = timeval_current();
159         task->private_data      = service;
160
161         status = dreplsrv_init_creds(service);
162         if (!W_ERROR_IS_OK(status)) {
163                 task_server_terminate(task, talloc_asprintf(task,
164                                       "dreplsrv: Failed to obtain server credentials: %s\n",
165                                                             win_errstr(status)), true);
166                 return;
167         }
168
169         status = dreplsrv_connect_samdb(service, task->lp_ctx);
170         if (!W_ERROR_IS_OK(status)) {
171                 task_server_terminate(task, talloc_asprintf(task,
172                                       "dreplsrv: Failed to connect to local samdb: %s\n",
173                                                             win_errstr(status)), true);
174                 return;
175         }
176
177         status = dreplsrv_load_partitions(service);
178         if (!W_ERROR_IS_OK(status)) {
179                 task_server_terminate(task, talloc_asprintf(task,
180                                       "dreplsrv: Failed to load partitions: %s\n",
181                                                             win_errstr(status)), true);
182                 return;
183         }
184
185         periodic_startup_interval       = lp_parm_int(task->lp_ctx, NULL, "dreplsrv", "periodic_startup_interval", 15); /* in seconds */
186         service->periodic.interval      = lp_parm_int(task->lp_ctx, NULL, "dreplsrv", "periodic_interval", 300); /* in seconds */
187
188         status = dreplsrv_periodic_schedule(service, periodic_startup_interval);
189         if (!W_ERROR_IS_OK(status)) {
190                 task_server_terminate(task, talloc_asprintf(task,
191                                       "dreplsrv: Failed to periodic schedule: %s\n",
192                                                             win_errstr(status)), true);
193                 return;
194         }
195
196         service->notify.interval = lp_parm_int(task->lp_ctx, NULL, "dreplsrv", 
197                                                "notify_interval", 5); /* in seconds */
198         status = dreplsrv_notify_schedule(service, service->notify.interval);
199         if (!W_ERROR_IS_OK(status)) {
200                 task_server_terminate(task, talloc_asprintf(task,
201                                       "dreplsrv: Failed to setup notify schedule: %s\n",
202                                                             win_errstr(status)), true);
203                 return;
204         }
205
206         irpc_add_name(task->msg_ctx, "dreplsrv");
207
208         IRPC_REGISTER(task->msg_ctx, drsuapi, DRSUAPI_DSREPLICASYNC, drepl_replica_sync, service);
209         messaging_register(task->msg_ctx, service, MSG_DREPL_ALLOCATE_RID, dreplsrv_allocate_rid);
210 }
211
212 /*
213   register ourselves as a available server
214 */
215 NTSTATUS server_service_drepl_init(void)
216 {
217         return register_server_service("drepl", dreplsrv_task_init);
218 }