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