primary.c revision 225830
150476Speter/*-
21802Sphk * Copyright (c) 2009 The FreeBSD Foundation
31802Sphk * Copyright (c) 2010-2011 Pawel Jakub Dawidek <pawel@dawidek.net>
418845Swollman * All rights reserved.
51802Sphk *
61802Sphk * This software was developed by Pawel Jakub Dawidek under sponsorship from
71802Sphk * the FreeBSD Foundation.
81802Sphk *
918845Swollman * Redistribution and use in source and binary forms, with or without
101802Sphk * modification, are permitted provided that the following conditions
111802Sphk * are met:
121802Sphk * 1. Redistributions of source code must retain the above copyright
131802Sphk *    notice, this list of conditions and the following disclaimer.
1418845Swollman * 2. Redistributions in binary form must reproduce the above copyright
151802Sphk *    notice, this list of conditions and the following disclaimer in the
161802Sphk *    documentation and/or other materials provided with the distribution.
171802Sphk *
181802Sphk * THIS SOFTWARE IS PROVIDED BY THE AUTHORS AND CONTRIBUTORS ``AS IS'' AND
1918845Swollman * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
201802Sphk * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
211802Sphk * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE LIABLE
22 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
23 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
24 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
25 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
26 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
27 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
28 * SUCH DAMAGE.
29 */
30
31#include <sys/cdefs.h>
32__FBSDID("$FreeBSD: head/sbin/hastd/primary.c 225830 2011-09-28 13:08:51Z pjd $");
33
34#include <sys/types.h>
35#include <sys/time.h>
36#include <sys/bio.h>
37#include <sys/disk.h>
38#include <sys/refcount.h>
39#include <sys/stat.h>
40
41#include <geom/gate/g_gate.h>
42
43#include <err.h>
44#include <errno.h>
45#include <fcntl.h>
46#include <libgeom.h>
47#include <pthread.h>
48#include <signal.h>
49#include <stdint.h>
50#include <stdio.h>
51#include <string.h>
52#include <sysexits.h>
53#include <unistd.h>
54
55#include <activemap.h>
56#include <nv.h>
57#include <rangelock.h>
58
59#include "control.h"
60#include "event.h"
61#include "hast.h"
62#include "hast_proto.h"
63#include "hastd.h"
64#include "hooks.h"
65#include "metadata.h"
66#include "proto.h"
67#include "pjdlog.h"
68#include "subr.h"
69#include "synch.h"
70
71/* The is only one remote component for now. */
72#define	ISREMOTE(no)	((no) == 1)
73
74struct hio {
75	/*
76	 * Number of components we are still waiting for.
77	 * When this field goes to 0, we can send the request back to the
78	 * kernel. Each component has to decrease this counter by one
79	 * even on failure.
80	 */
81	unsigned int		 hio_countdown;
82	/*
83	 * Each component has a place to store its own error.
84	 * Once the request is handled by all components we can decide if the
85	 * request overall is successful or not.
86	 */
87	int			*hio_errors;
88	/*
89	 * Structure used to communicate with GEOM Gate class.
90	 */
91	struct g_gate_ctl_io	 hio_ggio;
92	TAILQ_ENTRY(hio)	*hio_next;
93};
94#define	hio_free_next	hio_next[0]
95#define	hio_done_next	hio_next[0]
96
97/*
98 * Free list holds unused structures. When free list is empty, we have to wait
99 * until some in-progress requests are freed.
100 */
101static TAILQ_HEAD(, hio) hio_free_list;
102static pthread_mutex_t hio_free_list_lock;
103static pthread_cond_t hio_free_list_cond;
104/*
105 * There is one send list for every component. One requests is placed on all
106 * send lists - each component gets the same request, but each component is
107 * responsible for managing his own send list.
108 */
109static TAILQ_HEAD(, hio) *hio_send_list;
110static pthread_mutex_t *hio_send_list_lock;
111static pthread_cond_t *hio_send_list_cond;
112/*
113 * There is one recv list for every component, although local components don't
114 * use recv lists as local requests are done synchronously.
115 */
116static TAILQ_HEAD(, hio) *hio_recv_list;
117static pthread_mutex_t *hio_recv_list_lock;
118static pthread_cond_t *hio_recv_list_cond;
119/*
120 * Request is placed on done list by the slowest component (the one that
121 * decreased hio_countdown from 1 to 0).
122 */
123static TAILQ_HEAD(, hio) hio_done_list;
124static pthread_mutex_t hio_done_list_lock;
125static pthread_cond_t hio_done_list_cond;
126/*
127 * Structure below are for interaction with sync thread.
128 */
129static bool sync_inprogress;
130static pthread_mutex_t sync_lock;
131static pthread_cond_t sync_cond;
132/*
133 * The lock below allows to synchornize access to remote connections.
134 */
135static pthread_rwlock_t *hio_remote_lock;
136
137/*
138 * Lock to synchronize metadata updates. Also synchronize access to
139 * hr_primary_localcnt and hr_primary_remotecnt fields.
140 */
141static pthread_mutex_t metadata_lock;
142
143/*
144 * Maximum number of outstanding I/O requests.
145 */
146#define	HAST_HIO_MAX	256
147/*
148 * Number of components. At this point there are only two components: local
149 * and remote, but in the future it might be possible to use multiple local
150 * and remote components.
151 */
152#define	HAST_NCOMPONENTS	2
153
154#define	ISCONNECTED(res, no)	\
155	((res)->hr_remotein != NULL && (res)->hr_remoteout != NULL)
156
157#define	QUEUE_INSERT1(hio, name, ncomp)	do {				\
158	bool _wakeup;							\
159									\
160	mtx_lock(&hio_##name##_list_lock[(ncomp)]);			\
161	_wakeup = TAILQ_EMPTY(&hio_##name##_list[(ncomp)]);		\
162	TAILQ_INSERT_TAIL(&hio_##name##_list[(ncomp)], (hio),		\
163	    hio_next[(ncomp)]);						\
164	mtx_unlock(&hio_##name##_list_lock[ncomp]);			\
165	if (_wakeup)							\
166		cv_signal(&hio_##name##_list_cond[(ncomp)]);		\
167} while (0)
168#define	QUEUE_INSERT2(hio, name)	do {				\
169	bool _wakeup;							\
170									\
171	mtx_lock(&hio_##name##_list_lock);				\
172	_wakeup = TAILQ_EMPTY(&hio_##name##_list);			\
173	TAILQ_INSERT_TAIL(&hio_##name##_list, (hio), hio_##name##_next);\
174	mtx_unlock(&hio_##name##_list_lock);				\
175	if (_wakeup)							\
176		cv_signal(&hio_##name##_list_cond);			\
177} while (0)
178#define	QUEUE_TAKE1(hio, name, ncomp, timeout)	do {			\
179	bool _last;							\
180									\
181	mtx_lock(&hio_##name##_list_lock[(ncomp)]);			\
182	_last = false;							\
183	while (((hio) = TAILQ_FIRST(&hio_##name##_list[(ncomp)])) == NULL && !_last) { \
184		cv_timedwait(&hio_##name##_list_cond[(ncomp)],		\
185		    &hio_##name##_list_lock[(ncomp)], (timeout));	\
186		if ((timeout) != 0)					\
187			_last = true;					\
188	}								\
189	if (hio != NULL) {						\
190		TAILQ_REMOVE(&hio_##name##_list[(ncomp)], (hio),	\
191		    hio_next[(ncomp)]);					\
192	}								\
193	mtx_unlock(&hio_##name##_list_lock[(ncomp)]);			\
194} while (0)
195#define	QUEUE_TAKE2(hio, name)	do {					\
196	mtx_lock(&hio_##name##_list_lock);				\
197	while (((hio) = TAILQ_FIRST(&hio_##name##_list)) == NULL) {	\
198		cv_wait(&hio_##name##_list_cond,			\
199		    &hio_##name##_list_lock);				\
200	}								\
201	TAILQ_REMOVE(&hio_##name##_list, (hio), hio_##name##_next);	\
202	mtx_unlock(&hio_##name##_list_lock);				\
203} while (0)
204
205#define	SYNCREQ(hio)		do {					\
206	(hio)->hio_ggio.gctl_unit = -1;					\
207	(hio)->hio_ggio.gctl_seq = 1;					\
208} while (0)
209#define	ISSYNCREQ(hio)		((hio)->hio_ggio.gctl_unit == -1)
210#define	SYNCREQDONE(hio)	do { (hio)->hio_ggio.gctl_unit = -2; } while (0)
211#define	ISSYNCREQDONE(hio)	((hio)->hio_ggio.gctl_unit == -2)
212
213static struct hast_resource *gres;
214
215static pthread_mutex_t range_lock;
216static struct rangelocks *range_regular;
217static bool range_regular_wait;
218static pthread_cond_t range_regular_cond;
219static struct rangelocks *range_sync;
220static bool range_sync_wait;
221static pthread_cond_t range_sync_cond;
222static bool fullystarted;
223
224static void *ggate_recv_thread(void *arg);
225static void *local_send_thread(void *arg);
226static void *remote_send_thread(void *arg);
227static void *remote_recv_thread(void *arg);
228static void *ggate_send_thread(void *arg);
229static void *sync_thread(void *arg);
230static void *guard_thread(void *arg);
231
232static void
233cleanup(struct hast_resource *res)
234{
235	int rerrno;
236
237	/* Remember errno. */
238	rerrno = errno;
239
240	/* Destroy ggate provider if we created one. */
241	if (res->hr_ggateunit >= 0) {
242		struct g_gate_ctl_destroy ggiod;
243
244		bzero(&ggiod, sizeof(ggiod));
245		ggiod.gctl_version = G_GATE_VERSION;
246		ggiod.gctl_unit = res->hr_ggateunit;
247		ggiod.gctl_force = 1;
248		if (ioctl(res->hr_ggatefd, G_GATE_CMD_DESTROY, &ggiod) < 0) {
249			pjdlog_errno(LOG_WARNING,
250			    "Unable to destroy hast/%s device",
251			    res->hr_provname);
252		}
253		res->hr_ggateunit = -1;
254	}
255
256	/* Restore errno. */
257	errno = rerrno;
258}
259
260static __dead2 void
261primary_exit(int exitcode, const char *fmt, ...)
262{
263	va_list ap;
264
265	PJDLOG_ASSERT(exitcode != EX_OK);
266	va_start(ap, fmt);
267	pjdlogv_errno(LOG_ERR, fmt, ap);
268	va_end(ap);
269	cleanup(gres);
270	exit(exitcode);
271}
272
273static __dead2 void
274primary_exitx(int exitcode, const char *fmt, ...)
275{
276	va_list ap;
277
278	va_start(ap, fmt);
279	pjdlogv(exitcode == EX_OK ? LOG_INFO : LOG_ERR, fmt, ap);
280	va_end(ap);
281	cleanup(gres);
282	exit(exitcode);
283}
284
285static int
286hast_activemap_flush(struct hast_resource *res)
287{
288	const unsigned char *buf;
289	size_t size;
290
291	buf = activemap_bitmap(res->hr_amp, &size);
292	PJDLOG_ASSERT(buf != NULL);
293	PJDLOG_ASSERT((size % res->hr_local_sectorsize) == 0);
294	if (pwrite(res->hr_localfd, buf, size, METADATA_SIZE) !=
295	    (ssize_t)size) {
296		pjdlog_errno(LOG_ERR, "Unable to flush activemap to disk");
297		return (-1);
298	}
299	if (res->hr_metaflush == 1 && g_flush(res->hr_localfd) == -1) {
300		if (errno == EOPNOTSUPP) {
301			pjdlog_warning("The %s provider doesn't support flushing write cache. Disabling it.",
302			    res->hr_localpath);
303			res->hr_metaflush = 0;
304		} else {
305			pjdlog_errno(LOG_ERR,
306			    "Unable to flush disk cache on activemap update");
307			return (-1);
308		}
309	}
310	return (0);
311}
312
313static bool
314real_remote(const struct hast_resource *res)
315{
316
317	return (strcmp(res->hr_remoteaddr, "none") != 0);
318}
319
320static void
321init_environment(struct hast_resource *res __unused)
322{
323	struct hio *hio;
324	unsigned int ii, ncomps;
325
326	/*
327	 * In the future it might be per-resource value.
328	 */
329	ncomps = HAST_NCOMPONENTS;
330
331	/*
332	 * Allocate memory needed by lists.
333	 */
334	hio_send_list = malloc(sizeof(hio_send_list[0]) * ncomps);
335	if (hio_send_list == NULL) {
336		primary_exitx(EX_TEMPFAIL,
337		    "Unable to allocate %zu bytes of memory for send lists.",
338		    sizeof(hio_send_list[0]) * ncomps);
339	}
340	hio_send_list_lock = malloc(sizeof(hio_send_list_lock[0]) * ncomps);
341	if (hio_send_list_lock == NULL) {
342		primary_exitx(EX_TEMPFAIL,
343		    "Unable to allocate %zu bytes of memory for send list locks.",
344		    sizeof(hio_send_list_lock[0]) * ncomps);
345	}
346	hio_send_list_cond = malloc(sizeof(hio_send_list_cond[0]) * ncomps);
347	if (hio_send_list_cond == NULL) {
348		primary_exitx(EX_TEMPFAIL,
349		    "Unable to allocate %zu bytes of memory for send list condition variables.",
350		    sizeof(hio_send_list_cond[0]) * ncomps);
351	}
352	hio_recv_list = malloc(sizeof(hio_recv_list[0]) * ncomps);
353	if (hio_recv_list == NULL) {
354		primary_exitx(EX_TEMPFAIL,
355		    "Unable to allocate %zu bytes of memory for recv lists.",
356		    sizeof(hio_recv_list[0]) * ncomps);
357	}
358	hio_recv_list_lock = malloc(sizeof(hio_recv_list_lock[0]) * ncomps);
359	if (hio_recv_list_lock == NULL) {
360		primary_exitx(EX_TEMPFAIL,
361		    "Unable to allocate %zu bytes of memory for recv list locks.",
362		    sizeof(hio_recv_list_lock[0]) * ncomps);
363	}
364	hio_recv_list_cond = malloc(sizeof(hio_recv_list_cond[0]) * ncomps);
365	if (hio_recv_list_cond == NULL) {
366		primary_exitx(EX_TEMPFAIL,
367		    "Unable to allocate %zu bytes of memory for recv list condition variables.",
368		    sizeof(hio_recv_list_cond[0]) * ncomps);
369	}
370	hio_remote_lock = malloc(sizeof(hio_remote_lock[0]) * ncomps);
371	if (hio_remote_lock == NULL) {
372		primary_exitx(EX_TEMPFAIL,
373		    "Unable to allocate %zu bytes of memory for remote connections locks.",
374		    sizeof(hio_remote_lock[0]) * ncomps);
375	}
376
377	/*
378	 * Initialize lists, their locks and theirs condition variables.
379	 */
380	TAILQ_INIT(&hio_free_list);
381	mtx_init(&hio_free_list_lock);
382	cv_init(&hio_free_list_cond);
383	for (ii = 0; ii < HAST_NCOMPONENTS; ii++) {
384		TAILQ_INIT(&hio_send_list[ii]);
385		mtx_init(&hio_send_list_lock[ii]);
386		cv_init(&hio_send_list_cond[ii]);
387		TAILQ_INIT(&hio_recv_list[ii]);
388		mtx_init(&hio_recv_list_lock[ii]);
389		cv_init(&hio_recv_list_cond[ii]);
390		rw_init(&hio_remote_lock[ii]);
391	}
392	TAILQ_INIT(&hio_done_list);
393	mtx_init(&hio_done_list_lock);
394	cv_init(&hio_done_list_cond);
395	mtx_init(&metadata_lock);
396
397	/*
398	 * Allocate requests pool and initialize requests.
399	 */
400	for (ii = 0; ii < HAST_HIO_MAX; ii++) {
401		hio = malloc(sizeof(*hio));
402		if (hio == NULL) {
403			primary_exitx(EX_TEMPFAIL,
404			    "Unable to allocate %zu bytes of memory for hio request.",
405			    sizeof(*hio));
406		}
407		hio->hio_countdown = 0;
408		hio->hio_errors = malloc(sizeof(hio->hio_errors[0]) * ncomps);
409		if (hio->hio_errors == NULL) {
410			primary_exitx(EX_TEMPFAIL,
411			    "Unable allocate %zu bytes of memory for hio errors.",
412			    sizeof(hio->hio_errors[0]) * ncomps);
413		}
414		hio->hio_next = malloc(sizeof(hio->hio_next[0]) * ncomps);
415		if (hio->hio_next == NULL) {
416			primary_exitx(EX_TEMPFAIL,
417			    "Unable allocate %zu bytes of memory for hio_next field.",
418			    sizeof(hio->hio_next[0]) * ncomps);
419		}
420		hio->hio_ggio.gctl_version = G_GATE_VERSION;
421		hio->hio_ggio.gctl_data = malloc(MAXPHYS);
422		if (hio->hio_ggio.gctl_data == NULL) {
423			primary_exitx(EX_TEMPFAIL,
424			    "Unable to allocate %zu bytes of memory for gctl_data.",
425			    MAXPHYS);
426		}
427		hio->hio_ggio.gctl_length = MAXPHYS;
428		hio->hio_ggio.gctl_error = 0;
429		TAILQ_INSERT_HEAD(&hio_free_list, hio, hio_free_next);
430	}
431}
432
433static bool
434init_resuid(struct hast_resource *res)
435{
436
437	mtx_lock(&metadata_lock);
438	if (res->hr_resuid != 0) {
439		mtx_unlock(&metadata_lock);
440		return (false);
441	} else {
442		/* Initialize unique resource identifier. */
443		arc4random_buf(&res->hr_resuid, sizeof(res->hr_resuid));
444		mtx_unlock(&metadata_lock);
445		if (metadata_write(res) < 0)
446			exit(EX_NOINPUT);
447		return (true);
448	}
449}
450
451static void
452init_local(struct hast_resource *res)
453{
454	unsigned char *buf;
455	size_t mapsize;
456
457	if (metadata_read(res, true) < 0)
458		exit(EX_NOINPUT);
459	mtx_init(&res->hr_amp_lock);
460	if (activemap_init(&res->hr_amp, res->hr_datasize, res->hr_extentsize,
461	    res->hr_local_sectorsize, res->hr_keepdirty) < 0) {
462		primary_exit(EX_TEMPFAIL, "Unable to create activemap");
463	}
464	mtx_init(&range_lock);
465	cv_init(&range_regular_cond);
466	if (rangelock_init(&range_regular) < 0)
467		primary_exit(EX_TEMPFAIL, "Unable to create regular range lock");
468	cv_init(&range_sync_cond);
469	if (rangelock_init(&range_sync) < 0)
470		primary_exit(EX_TEMPFAIL, "Unable to create sync range lock");
471	mapsize = activemap_ondisk_size(res->hr_amp);
472	buf = calloc(1, mapsize);
473	if (buf == NULL) {
474		primary_exitx(EX_TEMPFAIL,
475		    "Unable to allocate buffer for activemap.");
476	}
477	if (pread(res->hr_localfd, buf, mapsize, METADATA_SIZE) !=
478	    (ssize_t)mapsize) {
479		primary_exit(EX_NOINPUT, "Unable to read activemap");
480	}
481	activemap_copyin(res->hr_amp, buf, mapsize);
482	free(buf);
483	if (res->hr_resuid != 0)
484		return;
485	/*
486	 * We're using provider for the first time. Initialize local and remote
487	 * counters. We don't initialize resuid here, as we want to do it just
488	 * in time. The reason for this is that we want to inform secondary
489	 * that there were no writes yet, so there is no need to synchronize
490	 * anything.
491	 */
492	res->hr_primary_localcnt = 0;
493	res->hr_primary_remotecnt = 0;
494	if (metadata_write(res) < 0)
495		exit(EX_NOINPUT);
496}
497
498static int
499primary_connect(struct hast_resource *res, struct proto_conn **connp)
500{
501	struct proto_conn *conn;
502	int16_t val;
503
504	val = 1;
505	if (proto_send(res->hr_conn, &val, sizeof(val)) < 0) {
506		primary_exit(EX_TEMPFAIL,
507		    "Unable to send connection request to parent");
508	}
509	if (proto_recv(res->hr_conn, &val, sizeof(val)) < 0) {
510		primary_exit(EX_TEMPFAIL,
511		    "Unable to receive reply to connection request from parent");
512	}
513	if (val != 0) {
514		errno = val;
515		pjdlog_errno(LOG_WARNING, "Unable to connect to %s",
516		    res->hr_remoteaddr);
517		return (-1);
518	}
519	if (proto_connection_recv(res->hr_conn, true, &conn) < 0) {
520		primary_exit(EX_TEMPFAIL,
521		    "Unable to receive connection from parent");
522	}
523	if (proto_connect_wait(conn, res->hr_timeout) < 0) {
524		pjdlog_errno(LOG_WARNING, "Unable to connect to %s",
525		    res->hr_remoteaddr);
526		proto_close(conn);
527		return (-1);
528	}
529	/* Error in setting timeout is not critical, but why should it fail? */
530	if (proto_timeout(conn, res->hr_timeout) < 0)
531		pjdlog_errno(LOG_WARNING, "Unable to set connection timeout");
532
533	*connp = conn;
534
535	return (0);
536}
537
538static int
539init_remote(struct hast_resource *res, struct proto_conn **inp,
540    struct proto_conn **outp)
541{
542	struct proto_conn *in, *out;
543	struct nv *nvout, *nvin;
544	const unsigned char *token;
545	unsigned char *map;
546	const char *errmsg;
547	int32_t extentsize;
548	int64_t datasize;
549	uint32_t mapsize;
550	size_t size;
551	int error;
552
553	PJDLOG_ASSERT((inp == NULL && outp == NULL) || (inp != NULL && outp != NULL));
554	PJDLOG_ASSERT(real_remote(res));
555
556	in = out = NULL;
557	errmsg = NULL;
558
559	if (primary_connect(res, &out) == -1)
560		return (ECONNREFUSED);
561
562	error = ECONNABORTED;
563
564	/*
565	 * First handshake step.
566	 * Setup outgoing connection with remote node.
567	 */
568	nvout = nv_alloc();
569	nv_add_string(nvout, res->hr_name, "resource");
570	if (nv_error(nvout) != 0) {
571		pjdlog_common(LOG_WARNING, 0, nv_error(nvout),
572		    "Unable to allocate header for connection with %s",
573		    res->hr_remoteaddr);
574		nv_free(nvout);
575		goto close;
576	}
577	if (hast_proto_send(res, out, nvout, NULL, 0) < 0) {
578		pjdlog_errno(LOG_WARNING,
579		    "Unable to send handshake header to %s",
580		    res->hr_remoteaddr);
581		nv_free(nvout);
582		goto close;
583	}
584	nv_free(nvout);
585	if (hast_proto_recv_hdr(out, &nvin) < 0) {
586		pjdlog_errno(LOG_WARNING,
587		    "Unable to receive handshake header from %s",
588		    res->hr_remoteaddr);
589		goto close;
590	}
591	errmsg = nv_get_string(nvin, "errmsg");
592	if (errmsg != NULL) {
593		pjdlog_warning("%s", errmsg);
594		if (nv_exists(nvin, "wait"))
595			error = EBUSY;
596		nv_free(nvin);
597		goto close;
598	}
599	token = nv_get_uint8_array(nvin, &size, "token");
600	if (token == NULL) {
601		pjdlog_warning("Handshake header from %s has no 'token' field.",
602		    res->hr_remoteaddr);
603		nv_free(nvin);
604		goto close;
605	}
606	if (size != sizeof(res->hr_token)) {
607		pjdlog_warning("Handshake header from %s contains 'token' of wrong size (got %zu, expected %zu).",
608		    res->hr_remoteaddr, size, sizeof(res->hr_token));
609		nv_free(nvin);
610		goto close;
611	}
612	bcopy(token, res->hr_token, sizeof(res->hr_token));
613	nv_free(nvin);
614
615	/*
616	 * Second handshake step.
617	 * Setup incoming connection with remote node.
618	 */
619	if (primary_connect(res, &in) == -1)
620		goto close;
621
622	nvout = nv_alloc();
623	nv_add_string(nvout, res->hr_name, "resource");
624	nv_add_uint8_array(nvout, res->hr_token, sizeof(res->hr_token),
625	    "token");
626	if (res->hr_resuid == 0) {
627		/*
628		 * The resuid field was not yet initialized.
629		 * Because we do synchronization inside init_resuid(), it is
630		 * possible that someone already initialized it, the function
631		 * will return false then, but if we successfully initialized
632		 * it, we will get true. True means that there were no writes
633		 * to this resource yet and we want to inform secondary that
634		 * synchronization is not needed by sending "virgin" argument.
635		 */
636		if (init_resuid(res))
637			nv_add_int8(nvout, 1, "virgin");
638	}
639	nv_add_uint64(nvout, res->hr_resuid, "resuid");
640	nv_add_uint64(nvout, res->hr_primary_localcnt, "localcnt");
641	nv_add_uint64(nvout, res->hr_primary_remotecnt, "remotecnt");
642	if (nv_error(nvout) != 0) {
643		pjdlog_common(LOG_WARNING, 0, nv_error(nvout),
644		    "Unable to allocate header for connection with %s",
645		    res->hr_remoteaddr);
646		nv_free(nvout);
647		goto close;
648	}
649	if (hast_proto_send(res, in, nvout, NULL, 0) < 0) {
650		pjdlog_errno(LOG_WARNING,
651		    "Unable to send handshake header to %s",
652		    res->hr_remoteaddr);
653		nv_free(nvout);
654		goto close;
655	}
656	nv_free(nvout);
657	if (hast_proto_recv_hdr(out, &nvin) < 0) {
658		pjdlog_errno(LOG_WARNING,
659		    "Unable to receive handshake header from %s",
660		    res->hr_remoteaddr);
661		goto close;
662	}
663	errmsg = nv_get_string(nvin, "errmsg");
664	if (errmsg != NULL) {
665		pjdlog_warning("%s", errmsg);
666		nv_free(nvin);
667		goto close;
668	}
669	datasize = nv_get_int64(nvin, "datasize");
670	if (datasize != res->hr_datasize) {
671		pjdlog_warning("Data size differs between nodes (local=%jd, remote=%jd).",
672		    (intmax_t)res->hr_datasize, (intmax_t)datasize);
673		nv_free(nvin);
674		goto close;
675	}
676	extentsize = nv_get_int32(nvin, "extentsize");
677	if (extentsize != res->hr_extentsize) {
678		pjdlog_warning("Extent size differs between nodes (local=%zd, remote=%zd).",
679		    (ssize_t)res->hr_extentsize, (ssize_t)extentsize);
680		nv_free(nvin);
681		goto close;
682	}
683	res->hr_secondary_localcnt = nv_get_uint64(nvin, "localcnt");
684	res->hr_secondary_remotecnt = nv_get_uint64(nvin, "remotecnt");
685	res->hr_syncsrc = nv_get_uint8(nvin, "syncsrc");
686	if (nv_exists(nvin, "virgin")) {
687		/*
688		 * Secondary was reinitialized, bump localcnt if it is 0 as
689		 * only we have the data.
690		 */
691		PJDLOG_ASSERT(res->hr_syncsrc == HAST_SYNCSRC_PRIMARY);
692		PJDLOG_ASSERT(res->hr_secondary_localcnt == 0);
693
694		if (res->hr_primary_localcnt == 0) {
695			PJDLOG_ASSERT(res->hr_secondary_remotecnt == 0);
696
697			mtx_lock(&metadata_lock);
698			res->hr_primary_localcnt++;
699			pjdlog_debug(1, "Increasing localcnt to %ju.",
700			    (uintmax_t)res->hr_primary_localcnt);
701			(void)metadata_write(res);
702			mtx_unlock(&metadata_lock);
703		}
704	}
705	map = NULL;
706	mapsize = nv_get_uint32(nvin, "mapsize");
707	if (mapsize > 0) {
708		map = malloc(mapsize);
709		if (map == NULL) {
710			pjdlog_error("Unable to allocate memory for remote activemap (mapsize=%ju).",
711			    (uintmax_t)mapsize);
712			nv_free(nvin);
713			goto close;
714		}
715		/*
716		 * Remote node have some dirty extents on its own, lets
717		 * download its activemap.
718		 */
719		if (hast_proto_recv_data(res, out, nvin, map,
720		    mapsize) < 0) {
721			pjdlog_errno(LOG_ERR,
722			    "Unable to receive remote activemap");
723			nv_free(nvin);
724			free(map);
725			goto close;
726		}
727		/*
728		 * Merge local and remote bitmaps.
729		 */
730		activemap_merge(res->hr_amp, map, mapsize);
731		free(map);
732		/*
733		 * Now that we merged bitmaps from both nodes, flush it to the
734		 * disk before we start to synchronize.
735		 */
736		(void)hast_activemap_flush(res);
737	}
738	nv_free(nvin);
739#ifdef notyet
740	/* Setup directions. */
741	if (proto_send(out, NULL, 0) == -1)
742		pjdlog_errno(LOG_WARNING, "Unable to set connection direction");
743	if (proto_recv(in, NULL, 0) == -1)
744		pjdlog_errno(LOG_WARNING, "Unable to set connection direction");
745#endif
746	pjdlog_info("Connected to %s.", res->hr_remoteaddr);
747	if (inp != NULL && outp != NULL) {
748		*inp = in;
749		*outp = out;
750	} else {
751		res->hr_remotein = in;
752		res->hr_remoteout = out;
753	}
754	event_send(res, EVENT_CONNECT);
755	return (0);
756close:
757	if (errmsg != NULL && strcmp(errmsg, "Split-brain condition!") == 0)
758		event_send(res, EVENT_SPLITBRAIN);
759	proto_close(out);
760	if (in != NULL)
761		proto_close(in);
762	return (error);
763}
764
765static void
766sync_start(void)
767{
768
769	mtx_lock(&sync_lock);
770	sync_inprogress = true;
771	mtx_unlock(&sync_lock);
772	cv_signal(&sync_cond);
773}
774
775static void
776sync_stop(void)
777{
778
779	mtx_lock(&sync_lock);
780	if (sync_inprogress)
781		sync_inprogress = false;
782	mtx_unlock(&sync_lock);
783}
784
785static void
786init_ggate(struct hast_resource *res)
787{
788	struct g_gate_ctl_create ggiocreate;
789	struct g_gate_ctl_cancel ggiocancel;
790
791	/*
792	 * We communicate with ggate via /dev/ggctl. Open it.
793	 */
794	res->hr_ggatefd = open("/dev/" G_GATE_CTL_NAME, O_RDWR);
795	if (res->hr_ggatefd < 0)
796		primary_exit(EX_OSFILE, "Unable to open /dev/" G_GATE_CTL_NAME);
797	/*
798	 * Create provider before trying to connect, as connection failure
799	 * is not critical, but may take some time.
800	 */
801	bzero(&ggiocreate, sizeof(ggiocreate));
802	ggiocreate.gctl_version = G_GATE_VERSION;
803	ggiocreate.gctl_mediasize = res->hr_datasize;
804	ggiocreate.gctl_sectorsize = res->hr_local_sectorsize;
805	ggiocreate.gctl_flags = 0;
806	ggiocreate.gctl_maxcount = 0;
807	ggiocreate.gctl_timeout = 0;
808	ggiocreate.gctl_unit = G_GATE_NAME_GIVEN;
809	snprintf(ggiocreate.gctl_name, sizeof(ggiocreate.gctl_name), "hast/%s",
810	    res->hr_provname);
811	if (ioctl(res->hr_ggatefd, G_GATE_CMD_CREATE, &ggiocreate) == 0) {
812		pjdlog_info("Device hast/%s created.", res->hr_provname);
813		res->hr_ggateunit = ggiocreate.gctl_unit;
814		return;
815	}
816	if (errno != EEXIST) {
817		primary_exit(EX_OSERR, "Unable to create hast/%s device",
818		    res->hr_provname);
819	}
820	pjdlog_debug(1,
821	    "Device hast/%s already exists, we will try to take it over.",
822	    res->hr_provname);
823	/*
824	 * If we received EEXIST, we assume that the process who created the
825	 * provider died and didn't clean up. In that case we will start from
826	 * where he left of.
827	 */
828	bzero(&ggiocancel, sizeof(ggiocancel));
829	ggiocancel.gctl_version = G_GATE_VERSION;
830	ggiocancel.gctl_unit = G_GATE_NAME_GIVEN;
831	snprintf(ggiocancel.gctl_name, sizeof(ggiocancel.gctl_name), "hast/%s",
832	    res->hr_provname);
833	if (ioctl(res->hr_ggatefd, G_GATE_CMD_CANCEL, &ggiocancel) == 0) {
834		pjdlog_info("Device hast/%s recovered.", res->hr_provname);
835		res->hr_ggateunit = ggiocancel.gctl_unit;
836		return;
837	}
838	primary_exit(EX_OSERR, "Unable to take over hast/%s device",
839	    res->hr_provname);
840}
841
842void
843hastd_primary(struct hast_resource *res)
844{
845	pthread_t td;
846	pid_t pid;
847	int error, mode, debuglevel;
848
849	/*
850	 * Create communication channel for sending control commands from
851	 * parent to child.
852	 */
853	if (proto_client(NULL, "socketpair://", &res->hr_ctrl) < 0) {
854		/* TODO: There's no need for this to be fatal error. */
855		KEEP_ERRNO((void)pidfile_remove(pfh));
856		pjdlog_exit(EX_OSERR,
857		    "Unable to create control sockets between parent and child");
858	}
859	/*
860	 * Create communication channel for sending events from child to parent.
861	 */
862	if (proto_client(NULL, "socketpair://", &res->hr_event) < 0) {
863		/* TODO: There's no need for this to be fatal error. */
864		KEEP_ERRNO((void)pidfile_remove(pfh));
865		pjdlog_exit(EX_OSERR,
866		    "Unable to create event sockets between child and parent");
867	}
868	/*
869	 * Create communication channel for sending connection requests from
870	 * child to parent.
871	 */
872	if (proto_client(NULL, "socketpair://", &res->hr_conn) < 0) {
873		/* TODO: There's no need for this to be fatal error. */
874		KEEP_ERRNO((void)pidfile_remove(pfh));
875		pjdlog_exit(EX_OSERR,
876		    "Unable to create connection sockets between child and parent");
877	}
878
879	pid = fork();
880	if (pid < 0) {
881		/* TODO: There's no need for this to be fatal error. */
882		KEEP_ERRNO((void)pidfile_remove(pfh));
883		pjdlog_exit(EX_TEMPFAIL, "Unable to fork");
884	}
885
886	if (pid > 0) {
887		/* This is parent. */
888		/* Declare that we are receiver. */
889		proto_recv(res->hr_event, NULL, 0);
890		proto_recv(res->hr_conn, NULL, 0);
891		/* Declare that we are sender. */
892		proto_send(res->hr_ctrl, NULL, 0);
893		res->hr_workerpid = pid;
894		return;
895	}
896
897	gres = res;
898	mode = pjdlog_mode_get();
899	debuglevel = pjdlog_debug_get();
900
901	/* Declare that we are sender. */
902	proto_send(res->hr_event, NULL, 0);
903	proto_send(res->hr_conn, NULL, 0);
904	/* Declare that we are receiver. */
905	proto_recv(res->hr_ctrl, NULL, 0);
906	descriptors_cleanup(res);
907
908	descriptors_assert(res, mode);
909
910	pjdlog_init(mode);
911	pjdlog_debug_set(debuglevel);
912	pjdlog_prefix_set("[%s] (%s) ", res->hr_name, role2str(res->hr_role));
913	setproctitle("%s (%s)", res->hr_name, role2str(res->hr_role));
914
915	init_local(res);
916	init_ggate(res);
917	init_environment(res);
918
919	if (drop_privs(res) != 0) {
920		cleanup(res);
921		exit(EX_CONFIG);
922	}
923	pjdlog_info("Privileges successfully dropped.");
924
925	/*
926	 * Create the guard thread first, so we can handle signals from the
927	 * very begining.
928	 */
929	error = pthread_create(&td, NULL, guard_thread, res);
930	PJDLOG_ASSERT(error == 0);
931	/*
932	 * Create the control thread before sending any event to the parent,
933	 * as we can deadlock when parent sends control request to worker,
934	 * but worker has no control thread started yet, so parent waits.
935	 * In the meantime worker sends an event to the parent, but parent
936	 * is unable to handle the event, because it waits for control
937	 * request response.
938	 */
939	error = pthread_create(&td, NULL, ctrl_thread, res);
940	PJDLOG_ASSERT(error == 0);
941	if (real_remote(res)) {
942		error = init_remote(res, NULL, NULL);
943		if (error == 0) {
944			sync_start();
945		} else if (error == EBUSY) {
946			time_t start = time(NULL);
947
948			pjdlog_warning("Waiting for remote node to become %s for %ds.",
949			    role2str(HAST_ROLE_SECONDARY),
950			    res->hr_timeout);
951			for (;;) {
952				sleep(1);
953				error = init_remote(res, NULL, NULL);
954				if (error != EBUSY)
955					break;
956				if (time(NULL) > start + res->hr_timeout)
957					break;
958			}
959			if (error == EBUSY) {
960				pjdlog_warning("Remote node is still %s, starting anyway.",
961				    role2str(HAST_ROLE_PRIMARY));
962			}
963		}
964	}
965	error = pthread_create(&td, NULL, ggate_recv_thread, res);
966	PJDLOG_ASSERT(error == 0);
967	error = pthread_create(&td, NULL, local_send_thread, res);
968	PJDLOG_ASSERT(error == 0);
969	error = pthread_create(&td, NULL, remote_send_thread, res);
970	PJDLOG_ASSERT(error == 0);
971	error = pthread_create(&td, NULL, remote_recv_thread, res);
972	PJDLOG_ASSERT(error == 0);
973	error = pthread_create(&td, NULL, ggate_send_thread, res);
974	PJDLOG_ASSERT(error == 0);
975	fullystarted = true;
976	(void)sync_thread(res);
977}
978
979static void
980reqlog(int loglevel, int debuglevel, struct g_gate_ctl_io *ggio, const char *fmt, ...)
981{
982	char msg[1024];
983	va_list ap;
984	int len;
985
986	va_start(ap, fmt);
987	len = vsnprintf(msg, sizeof(msg), fmt, ap);
988	va_end(ap);
989	if ((size_t)len < sizeof(msg)) {
990		switch (ggio->gctl_cmd) {
991		case BIO_READ:
992			(void)snprintf(msg + len, sizeof(msg) - len,
993			    "READ(%ju, %ju).", (uintmax_t)ggio->gctl_offset,
994			    (uintmax_t)ggio->gctl_length);
995			break;
996		case BIO_DELETE:
997			(void)snprintf(msg + len, sizeof(msg) - len,
998			    "DELETE(%ju, %ju).", (uintmax_t)ggio->gctl_offset,
999			    (uintmax_t)ggio->gctl_length);
1000			break;
1001		case BIO_FLUSH:
1002			(void)snprintf(msg + len, sizeof(msg) - len, "FLUSH.");
1003			break;
1004		case BIO_WRITE:
1005			(void)snprintf(msg + len, sizeof(msg) - len,
1006			    "WRITE(%ju, %ju).", (uintmax_t)ggio->gctl_offset,
1007			    (uintmax_t)ggio->gctl_length);
1008			break;
1009		default:
1010			(void)snprintf(msg + len, sizeof(msg) - len,
1011			    "UNKNOWN(%u).", (unsigned int)ggio->gctl_cmd);
1012			break;
1013		}
1014	}
1015	pjdlog_common(loglevel, debuglevel, -1, "%s", msg);
1016}
1017
1018static void
1019remote_close(struct hast_resource *res, int ncomp)
1020{
1021
1022	rw_wlock(&hio_remote_lock[ncomp]);
1023	/*
1024	 * A race is possible between dropping rlock and acquiring wlock -
1025	 * another thread can close connection in-between.
1026	 */
1027	if (!ISCONNECTED(res, ncomp)) {
1028		PJDLOG_ASSERT(res->hr_remotein == NULL);
1029		PJDLOG_ASSERT(res->hr_remoteout == NULL);
1030		rw_unlock(&hio_remote_lock[ncomp]);
1031		return;
1032	}
1033
1034	PJDLOG_ASSERT(res->hr_remotein != NULL);
1035	PJDLOG_ASSERT(res->hr_remoteout != NULL);
1036
1037	pjdlog_debug(2, "Closing incoming connection to %s.",
1038	    res->hr_remoteaddr);
1039	proto_close(res->hr_remotein);
1040	res->hr_remotein = NULL;
1041	pjdlog_debug(2, "Closing outgoing connection to %s.",
1042	    res->hr_remoteaddr);
1043	proto_close(res->hr_remoteout);
1044	res->hr_remoteout = NULL;
1045
1046	rw_unlock(&hio_remote_lock[ncomp]);
1047
1048	pjdlog_warning("Disconnected from %s.", res->hr_remoteaddr);
1049
1050	/*
1051	 * Stop synchronization if in-progress.
1052	 */
1053	sync_stop();
1054
1055	event_send(res, EVENT_DISCONNECT);
1056}
1057
1058/*
1059 * Thread receives ggate I/O requests from the kernel and passes them to
1060 * appropriate threads:
1061 * WRITE - always goes to both local_send and remote_send threads
1062 * READ (when the block is up-to-date on local component) -
1063 *	only local_send thread
1064 * READ (when the block isn't up-to-date on local component) -
1065 *	only remote_send thread
1066 * DELETE - always goes to both local_send and remote_send threads
1067 * FLUSH - always goes to both local_send and remote_send threads
1068 */
1069static void *
1070ggate_recv_thread(void *arg)
1071{
1072	struct hast_resource *res = arg;
1073	struct g_gate_ctl_io *ggio;
1074	struct hio *hio;
1075	unsigned int ii, ncomp, ncomps;
1076	int error;
1077
1078	ncomps = HAST_NCOMPONENTS;
1079
1080	for (;;) {
1081		pjdlog_debug(2, "ggate_recv: Taking free request.");
1082		QUEUE_TAKE2(hio, free);
1083		pjdlog_debug(2, "ggate_recv: (%p) Got free request.", hio);
1084		ggio = &hio->hio_ggio;
1085		ggio->gctl_unit = res->hr_ggateunit;
1086		ggio->gctl_length = MAXPHYS;
1087		ggio->gctl_error = 0;
1088		pjdlog_debug(2,
1089		    "ggate_recv: (%p) Waiting for request from the kernel.",
1090		    hio);
1091		if (ioctl(res->hr_ggatefd, G_GATE_CMD_START, ggio) < 0) {
1092			if (sigexit_received)
1093				pthread_exit(NULL);
1094			primary_exit(EX_OSERR, "G_GATE_CMD_START failed");
1095		}
1096		error = ggio->gctl_error;
1097		switch (error) {
1098		case 0:
1099			break;
1100		case ECANCELED:
1101			/* Exit gracefully. */
1102			if (!sigexit_received) {
1103				pjdlog_debug(2,
1104				    "ggate_recv: (%p) Received cancel from the kernel.",
1105				    hio);
1106				pjdlog_info("Received cancel from the kernel, exiting.");
1107			}
1108			pthread_exit(NULL);
1109		case ENOMEM:
1110			/*
1111			 * Buffer too small? Impossible, we allocate MAXPHYS
1112			 * bytes - request can't be bigger than that.
1113			 */
1114			/* FALLTHROUGH */
1115		case ENXIO:
1116		default:
1117			primary_exitx(EX_OSERR, "G_GATE_CMD_START failed: %s.",
1118			    strerror(error));
1119		}
1120		for (ii = 0; ii < ncomps; ii++)
1121			hio->hio_errors[ii] = EINVAL;
1122		reqlog(LOG_DEBUG, 2, ggio,
1123		    "ggate_recv: (%p) Request received from the kernel: ",
1124		    hio);
1125		/*
1126		 * Inform all components about new write request.
1127		 * For read request prefer local component unless the given
1128		 * range is out-of-date, then use remote component.
1129		 */
1130		switch (ggio->gctl_cmd) {
1131		case BIO_READ:
1132			res->hr_stat_read++;
1133			pjdlog_debug(2,
1134			    "ggate_recv: (%p) Moving request to the send queue.",
1135			    hio);
1136			refcount_init(&hio->hio_countdown, 1);
1137			mtx_lock(&metadata_lock);
1138			if (res->hr_syncsrc == HAST_SYNCSRC_UNDEF ||
1139			    res->hr_syncsrc == HAST_SYNCSRC_PRIMARY) {
1140				/*
1141				 * This range is up-to-date on local component,
1142				 * so handle request locally.
1143				 */
1144				 /* Local component is 0 for now. */
1145				ncomp = 0;
1146			} else /* if (res->hr_syncsrc ==
1147			    HAST_SYNCSRC_SECONDARY) */ {
1148				PJDLOG_ASSERT(res->hr_syncsrc ==
1149				    HAST_SYNCSRC_SECONDARY);
1150				/*
1151				 * This range is out-of-date on local component,
1152				 * so send request to the remote node.
1153				 */
1154				 /* Remote component is 1 for now. */
1155				ncomp = 1;
1156			}
1157			mtx_unlock(&metadata_lock);
1158			QUEUE_INSERT1(hio, send, ncomp);
1159			break;
1160		case BIO_WRITE:
1161			res->hr_stat_write++;
1162			if (res->hr_resuid == 0) {
1163				/*
1164				 * This is first write, initialize localcnt and
1165				 * resuid.
1166				 */
1167				res->hr_primary_localcnt = 1;
1168				(void)init_resuid(res);
1169			}
1170			for (;;) {
1171				mtx_lock(&range_lock);
1172				if (rangelock_islocked(range_sync,
1173				    ggio->gctl_offset, ggio->gctl_length)) {
1174					pjdlog_debug(2,
1175					    "regular: Range offset=%jd length=%zu locked.",
1176					    (intmax_t)ggio->gctl_offset,
1177					    (size_t)ggio->gctl_length);
1178					range_regular_wait = true;
1179					cv_wait(&range_regular_cond, &range_lock);
1180					range_regular_wait = false;
1181					mtx_unlock(&range_lock);
1182					continue;
1183				}
1184				if (rangelock_add(range_regular,
1185				    ggio->gctl_offset, ggio->gctl_length) < 0) {
1186					mtx_unlock(&range_lock);
1187					pjdlog_debug(2,
1188					    "regular: Range offset=%jd length=%zu is already locked, waiting.",
1189					    (intmax_t)ggio->gctl_offset,
1190					    (size_t)ggio->gctl_length);
1191					sleep(1);
1192					continue;
1193				}
1194				mtx_unlock(&range_lock);
1195				break;
1196			}
1197			mtx_lock(&res->hr_amp_lock);
1198			if (activemap_write_start(res->hr_amp,
1199			    ggio->gctl_offset, ggio->gctl_length)) {
1200				res->hr_stat_activemap_update++;
1201				(void)hast_activemap_flush(res);
1202			}
1203			mtx_unlock(&res->hr_amp_lock);
1204			/* FALLTHROUGH */
1205		case BIO_DELETE:
1206		case BIO_FLUSH:
1207			switch (ggio->gctl_cmd) {
1208			case BIO_DELETE:
1209				res->hr_stat_delete++;
1210				break;
1211			case BIO_FLUSH:
1212				res->hr_stat_flush++;
1213				break;
1214			}
1215			pjdlog_debug(2,
1216			    "ggate_recv: (%p) Moving request to the send queues.",
1217			    hio);
1218			refcount_init(&hio->hio_countdown, ncomps);
1219			for (ii = 0; ii < ncomps; ii++)
1220				QUEUE_INSERT1(hio, send, ii);
1221			break;
1222		}
1223	}
1224	/* NOTREACHED */
1225	return (NULL);
1226}
1227
1228/*
1229 * Thread reads from or writes to local component.
1230 * If local read fails, it redirects it to remote_send thread.
1231 */
1232static void *
1233local_send_thread(void *arg)
1234{
1235	struct hast_resource *res = arg;
1236	struct g_gate_ctl_io *ggio;
1237	struct hio *hio;
1238	unsigned int ncomp, rncomp;
1239	ssize_t ret;
1240
1241	/* Local component is 0 for now. */
1242	ncomp = 0;
1243	/* Remote component is 1 for now. */
1244	rncomp = 1;
1245
1246	for (;;) {
1247		pjdlog_debug(2, "local_send: Taking request.");
1248		QUEUE_TAKE1(hio, send, ncomp, 0);
1249		pjdlog_debug(2, "local_send: (%p) Got request.", hio);
1250		ggio = &hio->hio_ggio;
1251		switch (ggio->gctl_cmd) {
1252		case BIO_READ:
1253			ret = pread(res->hr_localfd, ggio->gctl_data,
1254			    ggio->gctl_length,
1255			    ggio->gctl_offset + res->hr_localoff);
1256			if (ret == ggio->gctl_length)
1257				hio->hio_errors[ncomp] = 0;
1258			else if (!ISSYNCREQ(hio)) {
1259				/*
1260				 * If READ failed, try to read from remote node.
1261				 */
1262				if (ret < 0) {
1263					reqlog(LOG_WARNING, 0, ggio,
1264					    "Local request failed (%s), trying remote node. ",
1265					    strerror(errno));
1266				} else if (ret != ggio->gctl_length) {
1267					reqlog(LOG_WARNING, 0, ggio,
1268					    "Local request failed (%zd != %jd), trying remote node. ",
1269					    ret, (intmax_t)ggio->gctl_length);
1270				}
1271				QUEUE_INSERT1(hio, send, rncomp);
1272				continue;
1273			}
1274			break;
1275		case BIO_WRITE:
1276			ret = pwrite(res->hr_localfd, ggio->gctl_data,
1277			    ggio->gctl_length,
1278			    ggio->gctl_offset + res->hr_localoff);
1279			if (ret < 0) {
1280				hio->hio_errors[ncomp] = errno;
1281				reqlog(LOG_WARNING, 0, ggio,
1282				    "Local request failed (%s): ",
1283				    strerror(errno));
1284			} else if (ret != ggio->gctl_length) {
1285				hio->hio_errors[ncomp] = EIO;
1286				reqlog(LOG_WARNING, 0, ggio,
1287				    "Local request failed (%zd != %jd): ",
1288				    ret, (intmax_t)ggio->gctl_length);
1289			} else {
1290				hio->hio_errors[ncomp] = 0;
1291			}
1292			break;
1293		case BIO_DELETE:
1294			ret = g_delete(res->hr_localfd,
1295			    ggio->gctl_offset + res->hr_localoff,
1296			    ggio->gctl_length);
1297			if (ret < 0) {
1298				hio->hio_errors[ncomp] = errno;
1299				reqlog(LOG_WARNING, 0, ggio,
1300				    "Local request failed (%s): ",
1301				    strerror(errno));
1302			} else {
1303				hio->hio_errors[ncomp] = 0;
1304			}
1305			break;
1306		case BIO_FLUSH:
1307			ret = g_flush(res->hr_localfd);
1308			if (ret < 0) {
1309				hio->hio_errors[ncomp] = errno;
1310				reqlog(LOG_WARNING, 0, ggio,
1311				    "Local request failed (%s): ",
1312				    strerror(errno));
1313			} else {
1314				hio->hio_errors[ncomp] = 0;
1315			}
1316			break;
1317		}
1318		if (refcount_release(&hio->hio_countdown)) {
1319			if (ISSYNCREQ(hio)) {
1320				mtx_lock(&sync_lock);
1321				SYNCREQDONE(hio);
1322				mtx_unlock(&sync_lock);
1323				cv_signal(&sync_cond);
1324			} else {
1325				pjdlog_debug(2,
1326				    "local_send: (%p) Moving request to the done queue.",
1327				    hio);
1328				QUEUE_INSERT2(hio, done);
1329			}
1330		}
1331	}
1332	/* NOTREACHED */
1333	return (NULL);
1334}
1335
1336static void
1337keepalive_send(struct hast_resource *res, unsigned int ncomp)
1338{
1339	struct nv *nv;
1340
1341	rw_rlock(&hio_remote_lock[ncomp]);
1342
1343	if (!ISCONNECTED(res, ncomp)) {
1344		rw_unlock(&hio_remote_lock[ncomp]);
1345		return;
1346	}
1347
1348	PJDLOG_ASSERT(res->hr_remotein != NULL);
1349	PJDLOG_ASSERT(res->hr_remoteout != NULL);
1350
1351	nv = nv_alloc();
1352	nv_add_uint8(nv, HIO_KEEPALIVE, "cmd");
1353	if (nv_error(nv) != 0) {
1354		rw_unlock(&hio_remote_lock[ncomp]);
1355		nv_free(nv);
1356		pjdlog_debug(1,
1357		    "keepalive_send: Unable to prepare header to send.");
1358		return;
1359	}
1360	if (hast_proto_send(res, res->hr_remoteout, nv, NULL, 0) < 0) {
1361		rw_unlock(&hio_remote_lock[ncomp]);
1362		pjdlog_common(LOG_DEBUG, 1, errno,
1363		    "keepalive_send: Unable to send request");
1364		nv_free(nv);
1365		remote_close(res, ncomp);
1366		return;
1367	}
1368
1369	rw_unlock(&hio_remote_lock[ncomp]);
1370	nv_free(nv);
1371	pjdlog_debug(2, "keepalive_send: Request sent.");
1372}
1373
1374/*
1375 * Thread sends request to secondary node.
1376 */
1377static void *
1378remote_send_thread(void *arg)
1379{
1380	struct hast_resource *res = arg;
1381	struct g_gate_ctl_io *ggio;
1382	time_t lastcheck, now;
1383	struct hio *hio;
1384	struct nv *nv;
1385	unsigned int ncomp;
1386	bool wakeup;
1387	uint64_t offset, length;
1388	uint8_t cmd;
1389	void *data;
1390
1391	/* Remote component is 1 for now. */
1392	ncomp = 1;
1393	lastcheck = time(NULL);
1394
1395	for (;;) {
1396		pjdlog_debug(2, "remote_send: Taking request.");
1397		QUEUE_TAKE1(hio, send, ncomp, HAST_KEEPALIVE);
1398		if (hio == NULL) {
1399			now = time(NULL);
1400			if (lastcheck + HAST_KEEPALIVE <= now) {
1401				keepalive_send(res, ncomp);
1402				lastcheck = now;
1403			}
1404			continue;
1405		}
1406		pjdlog_debug(2, "remote_send: (%p) Got request.", hio);
1407		ggio = &hio->hio_ggio;
1408		switch (ggio->gctl_cmd) {
1409		case BIO_READ:
1410			cmd = HIO_READ;
1411			data = NULL;
1412			offset = ggio->gctl_offset;
1413			length = ggio->gctl_length;
1414			break;
1415		case BIO_WRITE:
1416			cmd = HIO_WRITE;
1417			data = ggio->gctl_data;
1418			offset = ggio->gctl_offset;
1419			length = ggio->gctl_length;
1420			break;
1421		case BIO_DELETE:
1422			cmd = HIO_DELETE;
1423			data = NULL;
1424			offset = ggio->gctl_offset;
1425			length = ggio->gctl_length;
1426			break;
1427		case BIO_FLUSH:
1428			cmd = HIO_FLUSH;
1429			data = NULL;
1430			offset = 0;
1431			length = 0;
1432			break;
1433		default:
1434			PJDLOG_ABORT("invalid condition");
1435		}
1436		nv = nv_alloc();
1437		nv_add_uint8(nv, cmd, "cmd");
1438		nv_add_uint64(nv, (uint64_t)ggio->gctl_seq, "seq");
1439		nv_add_uint64(nv, offset, "offset");
1440		nv_add_uint64(nv, length, "length");
1441		if (nv_error(nv) != 0) {
1442			hio->hio_errors[ncomp] = nv_error(nv);
1443			pjdlog_debug(2,
1444			    "remote_send: (%p) Unable to prepare header to send.",
1445			    hio);
1446			reqlog(LOG_ERR, 0, ggio,
1447			    "Unable to prepare header to send (%s): ",
1448			    strerror(nv_error(nv)));
1449			/* Move failed request immediately to the done queue. */
1450			goto done_queue;
1451		}
1452		pjdlog_debug(2,
1453		    "remote_send: (%p) Moving request to the recv queue.",
1454		    hio);
1455		/*
1456		 * Protect connection from disappearing.
1457		 */
1458		rw_rlock(&hio_remote_lock[ncomp]);
1459		if (!ISCONNECTED(res, ncomp)) {
1460			rw_unlock(&hio_remote_lock[ncomp]);
1461			hio->hio_errors[ncomp] = ENOTCONN;
1462			goto done_queue;
1463		}
1464		/*
1465		 * Move the request to recv queue before sending it, because
1466		 * in different order we can get reply before we move request
1467		 * to recv queue.
1468		 */
1469		mtx_lock(&hio_recv_list_lock[ncomp]);
1470		wakeup = TAILQ_EMPTY(&hio_recv_list[ncomp]);
1471		TAILQ_INSERT_TAIL(&hio_recv_list[ncomp], hio, hio_next[ncomp]);
1472		mtx_unlock(&hio_recv_list_lock[ncomp]);
1473		if (hast_proto_send(res, res->hr_remoteout, nv, data,
1474		    data != NULL ? length : 0) < 0) {
1475			hio->hio_errors[ncomp] = errno;
1476			rw_unlock(&hio_remote_lock[ncomp]);
1477			pjdlog_debug(2,
1478			    "remote_send: (%p) Unable to send request.", hio);
1479			reqlog(LOG_ERR, 0, ggio,
1480			    "Unable to send request (%s): ",
1481			    strerror(hio->hio_errors[ncomp]));
1482			remote_close(res, ncomp);
1483			/*
1484			 * Take request back from the receive queue and move
1485			 * it immediately to the done queue.
1486			 */
1487			mtx_lock(&hio_recv_list_lock[ncomp]);
1488			TAILQ_REMOVE(&hio_recv_list[ncomp], hio, hio_next[ncomp]);
1489			mtx_unlock(&hio_recv_list_lock[ncomp]);
1490			goto done_queue;
1491		}
1492		rw_unlock(&hio_remote_lock[ncomp]);
1493		nv_free(nv);
1494		if (wakeup)
1495			cv_signal(&hio_recv_list_cond[ncomp]);
1496		continue;
1497done_queue:
1498		nv_free(nv);
1499		if (ISSYNCREQ(hio)) {
1500			if (!refcount_release(&hio->hio_countdown))
1501				continue;
1502			mtx_lock(&sync_lock);
1503			SYNCREQDONE(hio);
1504			mtx_unlock(&sync_lock);
1505			cv_signal(&sync_cond);
1506			continue;
1507		}
1508		if (ggio->gctl_cmd == BIO_WRITE) {
1509			mtx_lock(&res->hr_amp_lock);
1510			if (activemap_need_sync(res->hr_amp, ggio->gctl_offset,
1511			    ggio->gctl_length)) {
1512				(void)hast_activemap_flush(res);
1513			}
1514			mtx_unlock(&res->hr_amp_lock);
1515		}
1516		if (!refcount_release(&hio->hio_countdown))
1517			continue;
1518		pjdlog_debug(2,
1519		    "remote_send: (%p) Moving request to the done queue.",
1520		    hio);
1521		QUEUE_INSERT2(hio, done);
1522	}
1523	/* NOTREACHED */
1524	return (NULL);
1525}
1526
1527/*
1528 * Thread receives answer from secondary node and passes it to ggate_send
1529 * thread.
1530 */
1531static void *
1532remote_recv_thread(void *arg)
1533{
1534	struct hast_resource *res = arg;
1535	struct g_gate_ctl_io *ggio;
1536	struct hio *hio;
1537	struct nv *nv;
1538	unsigned int ncomp;
1539	uint64_t seq;
1540	int error;
1541
1542	/* Remote component is 1 for now. */
1543	ncomp = 1;
1544
1545	for (;;) {
1546		/* Wait until there is anything to receive. */
1547		mtx_lock(&hio_recv_list_lock[ncomp]);
1548		while (TAILQ_EMPTY(&hio_recv_list[ncomp])) {
1549			pjdlog_debug(2, "remote_recv: No requests, waiting.");
1550			cv_wait(&hio_recv_list_cond[ncomp],
1551			    &hio_recv_list_lock[ncomp]);
1552		}
1553		mtx_unlock(&hio_recv_list_lock[ncomp]);
1554		rw_rlock(&hio_remote_lock[ncomp]);
1555		if (!ISCONNECTED(res, ncomp)) {
1556			rw_unlock(&hio_remote_lock[ncomp]);
1557			/*
1558			 * Connection is dead, so move all pending requests to
1559			 * the done queue (one-by-one).
1560			 */
1561			mtx_lock(&hio_recv_list_lock[ncomp]);
1562			hio = TAILQ_FIRST(&hio_recv_list[ncomp]);
1563			PJDLOG_ASSERT(hio != NULL);
1564			TAILQ_REMOVE(&hio_recv_list[ncomp], hio,
1565			    hio_next[ncomp]);
1566			mtx_unlock(&hio_recv_list_lock[ncomp]);
1567			goto done_queue;
1568		}
1569		if (hast_proto_recv_hdr(res->hr_remotein, &nv) < 0) {
1570			pjdlog_errno(LOG_ERR,
1571			    "Unable to receive reply header");
1572			rw_unlock(&hio_remote_lock[ncomp]);
1573			remote_close(res, ncomp);
1574			continue;
1575		}
1576		rw_unlock(&hio_remote_lock[ncomp]);
1577		seq = nv_get_uint64(nv, "seq");
1578		if (seq == 0) {
1579			pjdlog_error("Header contains no 'seq' field.");
1580			nv_free(nv);
1581			continue;
1582		}
1583		mtx_lock(&hio_recv_list_lock[ncomp]);
1584		TAILQ_FOREACH(hio, &hio_recv_list[ncomp], hio_next[ncomp]) {
1585			if (hio->hio_ggio.gctl_seq == seq) {
1586				TAILQ_REMOVE(&hio_recv_list[ncomp], hio,
1587				    hio_next[ncomp]);
1588				break;
1589			}
1590		}
1591		mtx_unlock(&hio_recv_list_lock[ncomp]);
1592		if (hio == NULL) {
1593			pjdlog_error("Found no request matching received 'seq' field (%ju).",
1594			    (uintmax_t)seq);
1595			nv_free(nv);
1596			continue;
1597		}
1598		error = nv_get_int16(nv, "error");
1599		if (error != 0) {
1600			/* Request failed on remote side. */
1601			hio->hio_errors[ncomp] = error;
1602			reqlog(LOG_WARNING, 0, &hio->hio_ggio,
1603			    "Remote request failed (%s): ", strerror(error));
1604			nv_free(nv);
1605			goto done_queue;
1606		}
1607		ggio = &hio->hio_ggio;
1608		switch (ggio->gctl_cmd) {
1609		case BIO_READ:
1610			rw_rlock(&hio_remote_lock[ncomp]);
1611			if (!ISCONNECTED(res, ncomp)) {
1612				rw_unlock(&hio_remote_lock[ncomp]);
1613				nv_free(nv);
1614				goto done_queue;
1615			}
1616			if (hast_proto_recv_data(res, res->hr_remotein, nv,
1617			    ggio->gctl_data, ggio->gctl_length) < 0) {
1618				hio->hio_errors[ncomp] = errno;
1619				pjdlog_errno(LOG_ERR,
1620				    "Unable to receive reply data");
1621				rw_unlock(&hio_remote_lock[ncomp]);
1622				nv_free(nv);
1623				remote_close(res, ncomp);
1624				goto done_queue;
1625			}
1626			rw_unlock(&hio_remote_lock[ncomp]);
1627			break;
1628		case BIO_WRITE:
1629		case BIO_DELETE:
1630		case BIO_FLUSH:
1631			break;
1632		default:
1633			PJDLOG_ABORT("invalid condition");
1634		}
1635		hio->hio_errors[ncomp] = 0;
1636		nv_free(nv);
1637done_queue:
1638		if (refcount_release(&hio->hio_countdown)) {
1639			if (ISSYNCREQ(hio)) {
1640				mtx_lock(&sync_lock);
1641				SYNCREQDONE(hio);
1642				mtx_unlock(&sync_lock);
1643				cv_signal(&sync_cond);
1644			} else {
1645				pjdlog_debug(2,
1646				    "remote_recv: (%p) Moving request to the done queue.",
1647				    hio);
1648				QUEUE_INSERT2(hio, done);
1649			}
1650		}
1651	}
1652	/* NOTREACHED */
1653	return (NULL);
1654}
1655
1656/*
1657 * Thread sends answer to the kernel.
1658 */
1659static void *
1660ggate_send_thread(void *arg)
1661{
1662	struct hast_resource *res = arg;
1663	struct g_gate_ctl_io *ggio;
1664	struct hio *hio;
1665	unsigned int ii, ncomp, ncomps;
1666
1667	ncomps = HAST_NCOMPONENTS;
1668
1669	for (;;) {
1670		pjdlog_debug(2, "ggate_send: Taking request.");
1671		QUEUE_TAKE2(hio, done);
1672		pjdlog_debug(2, "ggate_send: (%p) Got request.", hio);
1673		ggio = &hio->hio_ggio;
1674		for (ii = 0; ii < ncomps; ii++) {
1675			if (hio->hio_errors[ii] == 0) {
1676				/*
1677				 * One successful request is enough to declare
1678				 * success.
1679				 */
1680				ggio->gctl_error = 0;
1681				break;
1682			}
1683		}
1684		if (ii == ncomps) {
1685			/*
1686			 * None of the requests were successful.
1687			 * Use the error from local component except the
1688			 * case when we did only remote request.
1689			 */
1690			if (ggio->gctl_cmd == BIO_READ &&
1691			    res->hr_syncsrc == HAST_SYNCSRC_SECONDARY)
1692				ggio->gctl_error = hio->hio_errors[1];
1693			else
1694				ggio->gctl_error = hio->hio_errors[0];
1695		}
1696		if (ggio->gctl_error == 0 && ggio->gctl_cmd == BIO_WRITE) {
1697			mtx_lock(&res->hr_amp_lock);
1698			if (activemap_write_complete(res->hr_amp,
1699			    ggio->gctl_offset, ggio->gctl_length)) {
1700				res->hr_stat_activemap_update++;
1701				(void)hast_activemap_flush(res);
1702			}
1703			mtx_unlock(&res->hr_amp_lock);
1704		}
1705		if (ggio->gctl_cmd == BIO_WRITE) {
1706			/*
1707			 * Unlock range we locked.
1708			 */
1709			mtx_lock(&range_lock);
1710			rangelock_del(range_regular, ggio->gctl_offset,
1711			    ggio->gctl_length);
1712			if (range_sync_wait)
1713				cv_signal(&range_sync_cond);
1714			mtx_unlock(&range_lock);
1715			/*
1716			 * Bump local count if this is first write after
1717			 * connection failure with remote node.
1718			 */
1719			ncomp = 1;
1720			rw_rlock(&hio_remote_lock[ncomp]);
1721			if (!ISCONNECTED(res, ncomp)) {
1722				mtx_lock(&metadata_lock);
1723				if (res->hr_primary_localcnt ==
1724				    res->hr_secondary_remotecnt) {
1725					res->hr_primary_localcnt++;
1726					pjdlog_debug(1,
1727					    "Increasing localcnt to %ju.",
1728					    (uintmax_t)res->hr_primary_localcnt);
1729					(void)metadata_write(res);
1730				}
1731				mtx_unlock(&metadata_lock);
1732			}
1733			rw_unlock(&hio_remote_lock[ncomp]);
1734		}
1735		if (ioctl(res->hr_ggatefd, G_GATE_CMD_DONE, ggio) < 0)
1736			primary_exit(EX_OSERR, "G_GATE_CMD_DONE failed");
1737		pjdlog_debug(2,
1738		    "ggate_send: (%p) Moving request to the free queue.", hio);
1739		QUEUE_INSERT2(hio, free);
1740	}
1741	/* NOTREACHED */
1742	return (NULL);
1743}
1744
1745/*
1746 * Thread synchronize local and remote components.
1747 */
1748static void *
1749sync_thread(void *arg __unused)
1750{
1751	struct hast_resource *res = arg;
1752	struct hio *hio;
1753	struct g_gate_ctl_io *ggio;
1754	struct timeval tstart, tend, tdiff;
1755	unsigned int ii, ncomp, ncomps;
1756	off_t offset, length, synced;
1757	bool dorewind;
1758	int syncext;
1759
1760	ncomps = HAST_NCOMPONENTS;
1761	dorewind = true;
1762	synced = 0;
1763	offset = -1;
1764
1765	for (;;) {
1766		mtx_lock(&sync_lock);
1767		if (offset >= 0 && !sync_inprogress) {
1768			gettimeofday(&tend, NULL);
1769			timersub(&tend, &tstart, &tdiff);
1770			pjdlog_info("Synchronization interrupted after %#.0T. "
1771			    "%NB synchronized so far.", &tdiff,
1772			    (intmax_t)synced);
1773			event_send(res, EVENT_SYNCINTR);
1774		}
1775		while (!sync_inprogress) {
1776			dorewind = true;
1777			synced = 0;
1778			cv_wait(&sync_cond, &sync_lock);
1779		}
1780		mtx_unlock(&sync_lock);
1781		/*
1782		 * Obtain offset at which we should synchronize.
1783		 * Rewind synchronization if needed.
1784		 */
1785		mtx_lock(&res->hr_amp_lock);
1786		if (dorewind)
1787			activemap_sync_rewind(res->hr_amp);
1788		offset = activemap_sync_offset(res->hr_amp, &length, &syncext);
1789		if (syncext != -1) {
1790			/*
1791			 * We synchronized entire syncext extent, we can mark
1792			 * it as clean now.
1793			 */
1794			if (activemap_extent_complete(res->hr_amp, syncext))
1795				(void)hast_activemap_flush(res);
1796		}
1797		mtx_unlock(&res->hr_amp_lock);
1798		if (dorewind) {
1799			dorewind = false;
1800			if (offset < 0)
1801				pjdlog_info("Nodes are in sync.");
1802			else {
1803				pjdlog_info("Synchronization started. %NB to go.",
1804				    (intmax_t)(res->hr_extentsize *
1805				    activemap_ndirty(res->hr_amp)));
1806				event_send(res, EVENT_SYNCSTART);
1807				gettimeofday(&tstart, NULL);
1808			}
1809		}
1810		if (offset < 0) {
1811			sync_stop();
1812			pjdlog_debug(1, "Nothing to synchronize.");
1813			/*
1814			 * Synchronization complete, make both localcnt and
1815			 * remotecnt equal.
1816			 */
1817			ncomp = 1;
1818			rw_rlock(&hio_remote_lock[ncomp]);
1819			if (ISCONNECTED(res, ncomp)) {
1820				if (synced > 0) {
1821					int64_t bps;
1822
1823					gettimeofday(&tend, NULL);
1824					timersub(&tend, &tstart, &tdiff);
1825					bps = (int64_t)((double)synced /
1826					    ((double)tdiff.tv_sec +
1827					    (double)tdiff.tv_usec / 1000000));
1828					pjdlog_info("Synchronization complete. "
1829					    "%NB synchronized in %#.0lT (%NB/sec).",
1830					    (intmax_t)synced, &tdiff,
1831					    (intmax_t)bps);
1832					event_send(res, EVENT_SYNCDONE);
1833				}
1834				mtx_lock(&metadata_lock);
1835				res->hr_syncsrc = HAST_SYNCSRC_UNDEF;
1836				res->hr_primary_localcnt =
1837				    res->hr_secondary_remotecnt;
1838				res->hr_primary_remotecnt =
1839				    res->hr_secondary_localcnt;
1840				pjdlog_debug(1,
1841				    "Setting localcnt to %ju and remotecnt to %ju.",
1842				    (uintmax_t)res->hr_primary_localcnt,
1843				    (uintmax_t)res->hr_primary_remotecnt);
1844				(void)metadata_write(res);
1845				mtx_unlock(&metadata_lock);
1846			}
1847			rw_unlock(&hio_remote_lock[ncomp]);
1848			continue;
1849		}
1850		pjdlog_debug(2, "sync: Taking free request.");
1851		QUEUE_TAKE2(hio, free);
1852		pjdlog_debug(2, "sync: (%p) Got free request.", hio);
1853		/*
1854		 * Lock the range we are going to synchronize. We don't want
1855		 * race where someone writes between our read and write.
1856		 */
1857		for (;;) {
1858			mtx_lock(&range_lock);
1859			if (rangelock_islocked(range_regular, offset, length)) {
1860				pjdlog_debug(2,
1861				    "sync: Range offset=%jd length=%jd locked.",
1862				    (intmax_t)offset, (intmax_t)length);
1863				range_sync_wait = true;
1864				cv_wait(&range_sync_cond, &range_lock);
1865				range_sync_wait = false;
1866				mtx_unlock(&range_lock);
1867				continue;
1868			}
1869			if (rangelock_add(range_sync, offset, length) < 0) {
1870				mtx_unlock(&range_lock);
1871				pjdlog_debug(2,
1872				    "sync: Range offset=%jd length=%jd is already locked, waiting.",
1873				    (intmax_t)offset, (intmax_t)length);
1874				sleep(1);
1875				continue;
1876			}
1877			mtx_unlock(&range_lock);
1878			break;
1879		}
1880		/*
1881		 * First read the data from synchronization source.
1882		 */
1883		SYNCREQ(hio);
1884		ggio = &hio->hio_ggio;
1885		ggio->gctl_cmd = BIO_READ;
1886		ggio->gctl_offset = offset;
1887		ggio->gctl_length = length;
1888		ggio->gctl_error = 0;
1889		for (ii = 0; ii < ncomps; ii++)
1890			hio->hio_errors[ii] = EINVAL;
1891		reqlog(LOG_DEBUG, 2, ggio, "sync: (%p) Sending sync request: ",
1892		    hio);
1893		pjdlog_debug(2, "sync: (%p) Moving request to the send queue.",
1894		    hio);
1895		mtx_lock(&metadata_lock);
1896		if (res->hr_syncsrc == HAST_SYNCSRC_PRIMARY) {
1897			/*
1898			 * This range is up-to-date on local component,
1899			 * so handle request locally.
1900			 */
1901			 /* Local component is 0 for now. */
1902			ncomp = 0;
1903		} else /* if (res->hr_syncsrc == HAST_SYNCSRC_SECONDARY) */ {
1904			PJDLOG_ASSERT(res->hr_syncsrc == HAST_SYNCSRC_SECONDARY);
1905			/*
1906			 * This range is out-of-date on local component,
1907			 * so send request to the remote node.
1908			 */
1909			 /* Remote component is 1 for now. */
1910			ncomp = 1;
1911		}
1912		mtx_unlock(&metadata_lock);
1913		refcount_init(&hio->hio_countdown, 1);
1914		QUEUE_INSERT1(hio, send, ncomp);
1915
1916		/*
1917		 * Let's wait for READ to finish.
1918		 */
1919		mtx_lock(&sync_lock);
1920		while (!ISSYNCREQDONE(hio))
1921			cv_wait(&sync_cond, &sync_lock);
1922		mtx_unlock(&sync_lock);
1923
1924		if (hio->hio_errors[ncomp] != 0) {
1925			pjdlog_error("Unable to read synchronization data: %s.",
1926			    strerror(hio->hio_errors[ncomp]));
1927			goto free_queue;
1928		}
1929
1930		/*
1931		 * We read the data from synchronization source, now write it
1932		 * to synchronization target.
1933		 */
1934		SYNCREQ(hio);
1935		ggio->gctl_cmd = BIO_WRITE;
1936		for (ii = 0; ii < ncomps; ii++)
1937			hio->hio_errors[ii] = EINVAL;
1938		reqlog(LOG_DEBUG, 2, ggio, "sync: (%p) Sending sync request: ",
1939		    hio);
1940		pjdlog_debug(2, "sync: (%p) Moving request to the send queue.",
1941		    hio);
1942		mtx_lock(&metadata_lock);
1943		if (res->hr_syncsrc == HAST_SYNCSRC_PRIMARY) {
1944			/*
1945			 * This range is up-to-date on local component,
1946			 * so we update remote component.
1947			 */
1948			 /* Remote component is 1 for now. */
1949			ncomp = 1;
1950		} else /* if (res->hr_syncsrc == HAST_SYNCSRC_SECONDARY) */ {
1951			PJDLOG_ASSERT(res->hr_syncsrc == HAST_SYNCSRC_SECONDARY);
1952			/*
1953			 * This range is out-of-date on local component,
1954			 * so we update it.
1955			 */
1956			 /* Local component is 0 for now. */
1957			ncomp = 0;
1958		}
1959		mtx_unlock(&metadata_lock);
1960
1961		pjdlog_debug(2, "sync: (%p) Moving request to the send queues.",
1962		    hio);
1963		refcount_init(&hio->hio_countdown, 1);
1964		QUEUE_INSERT1(hio, send, ncomp);
1965
1966		/*
1967		 * Let's wait for WRITE to finish.
1968		 */
1969		mtx_lock(&sync_lock);
1970		while (!ISSYNCREQDONE(hio))
1971			cv_wait(&sync_cond, &sync_lock);
1972		mtx_unlock(&sync_lock);
1973
1974		if (hio->hio_errors[ncomp] != 0) {
1975			pjdlog_error("Unable to write synchronization data: %s.",
1976			    strerror(hio->hio_errors[ncomp]));
1977			goto free_queue;
1978		}
1979
1980		synced += length;
1981free_queue:
1982		mtx_lock(&range_lock);
1983		rangelock_del(range_sync, offset, length);
1984		if (range_regular_wait)
1985			cv_signal(&range_regular_cond);
1986		mtx_unlock(&range_lock);
1987		pjdlog_debug(2, "sync: (%p) Moving request to the free queue.",
1988		    hio);
1989		QUEUE_INSERT2(hio, free);
1990	}
1991	/* NOTREACHED */
1992	return (NULL);
1993}
1994
1995void
1996primary_config_reload(struct hast_resource *res, struct nv *nv)
1997{
1998	unsigned int ii, ncomps;
1999	int modified, vint;
2000	const char *vstr;
2001
2002	pjdlog_info("Reloading configuration...");
2003
2004	PJDLOG_ASSERT(res->hr_role == HAST_ROLE_PRIMARY);
2005	PJDLOG_ASSERT(gres == res);
2006	nv_assert(nv, "remoteaddr");
2007	nv_assert(nv, "sourceaddr");
2008	nv_assert(nv, "replication");
2009	nv_assert(nv, "checksum");
2010	nv_assert(nv, "compression");
2011	nv_assert(nv, "timeout");
2012	nv_assert(nv, "exec");
2013	nv_assert(nv, "metaflush");
2014
2015	ncomps = HAST_NCOMPONENTS;
2016
2017#define MODIFIED_REMOTEADDR	0x01
2018#define MODIFIED_SOURCEADDR	0x02
2019#define MODIFIED_REPLICATION	0x04
2020#define MODIFIED_CHECKSUM	0x08
2021#define MODIFIED_COMPRESSION	0x10
2022#define MODIFIED_TIMEOUT	0x20
2023#define MODIFIED_EXEC		0x40
2024#define MODIFIED_METAFLUSH	0x80
2025	modified = 0;
2026
2027	vstr = nv_get_string(nv, "remoteaddr");
2028	if (strcmp(gres->hr_remoteaddr, vstr) != 0) {
2029		/*
2030		 * Don't copy res->hr_remoteaddr to gres just yet.
2031		 * We want remote_close() to log disconnect from the old
2032		 * addresses, not from the new ones.
2033		 */
2034		modified |= MODIFIED_REMOTEADDR;
2035	}
2036	vstr = nv_get_string(nv, "sourceaddr");
2037	if (strcmp(gres->hr_sourceaddr, vstr) != 0) {
2038		strlcpy(gres->hr_sourceaddr, vstr, sizeof(gres->hr_sourceaddr));
2039		modified |= MODIFIED_SOURCEADDR;
2040	}
2041	vint = nv_get_int32(nv, "replication");
2042	if (gres->hr_replication != vint) {
2043		gres->hr_replication = vint;
2044		modified |= MODIFIED_REPLICATION;
2045	}
2046	vint = nv_get_int32(nv, "checksum");
2047	if (gres->hr_checksum != vint) {
2048		gres->hr_checksum = vint;
2049		modified |= MODIFIED_CHECKSUM;
2050	}
2051	vint = nv_get_int32(nv, "compression");
2052	if (gres->hr_compression != vint) {
2053		gres->hr_compression = vint;
2054		modified |= MODIFIED_COMPRESSION;
2055	}
2056	vint = nv_get_int32(nv, "timeout");
2057	if (gres->hr_timeout != vint) {
2058		gres->hr_timeout = vint;
2059		modified |= MODIFIED_TIMEOUT;
2060	}
2061	vstr = nv_get_string(nv, "exec");
2062	if (strcmp(gres->hr_exec, vstr) != 0) {
2063		strlcpy(gres->hr_exec, vstr, sizeof(gres->hr_exec));
2064		modified |= MODIFIED_EXEC;
2065	}
2066	vint = nv_get_int32(nv, "metaflush");
2067	if (gres->hr_metaflush != vint) {
2068		gres->hr_metaflush = vint;
2069		modified |= MODIFIED_METAFLUSH;
2070	}
2071
2072	/*
2073	 * Change timeout for connected sockets.
2074	 * Don't bother if we need to reconnect.
2075	 */
2076	if ((modified & MODIFIED_TIMEOUT) != 0 &&
2077	    (modified & (MODIFIED_REMOTEADDR | MODIFIED_SOURCEADDR |
2078	    MODIFIED_REPLICATION)) == 0) {
2079		for (ii = 0; ii < ncomps; ii++) {
2080			if (!ISREMOTE(ii))
2081				continue;
2082			rw_rlock(&hio_remote_lock[ii]);
2083			if (!ISCONNECTED(gres, ii)) {
2084				rw_unlock(&hio_remote_lock[ii]);
2085				continue;
2086			}
2087			rw_unlock(&hio_remote_lock[ii]);
2088			if (proto_timeout(gres->hr_remotein,
2089			    gres->hr_timeout) < 0) {
2090				pjdlog_errno(LOG_WARNING,
2091				    "Unable to set connection timeout");
2092			}
2093			if (proto_timeout(gres->hr_remoteout,
2094			    gres->hr_timeout) < 0) {
2095				pjdlog_errno(LOG_WARNING,
2096				    "Unable to set connection timeout");
2097			}
2098		}
2099	}
2100	if ((modified & (MODIFIED_REMOTEADDR | MODIFIED_SOURCEADDR |
2101	    MODIFIED_REPLICATION)) != 0) {
2102		for (ii = 0; ii < ncomps; ii++) {
2103			if (!ISREMOTE(ii))
2104				continue;
2105			remote_close(gres, ii);
2106		}
2107		if (modified & MODIFIED_REMOTEADDR) {
2108			vstr = nv_get_string(nv, "remoteaddr");
2109			strlcpy(gres->hr_remoteaddr, vstr,
2110			    sizeof(gres->hr_remoteaddr));
2111		}
2112	}
2113#undef	MODIFIED_REMOTEADDR
2114#undef	MODIFIED_SOURCEADDR
2115#undef	MODIFIED_REPLICATION
2116#undef	MODIFIED_CHECKSUM
2117#undef	MODIFIED_COMPRESSION
2118#undef	MODIFIED_TIMEOUT
2119#undef	MODIFIED_EXEC
2120#undef	MODIFIED_METAFLUSH
2121
2122	pjdlog_info("Configuration reloaded successfully.");
2123}
2124
2125static void
2126guard_one(struct hast_resource *res, unsigned int ncomp)
2127{
2128	struct proto_conn *in, *out;
2129
2130	if (!ISREMOTE(ncomp))
2131		return;
2132
2133	rw_rlock(&hio_remote_lock[ncomp]);
2134
2135	if (!real_remote(res)) {
2136		rw_unlock(&hio_remote_lock[ncomp]);
2137		return;
2138	}
2139
2140	if (ISCONNECTED(res, ncomp)) {
2141		PJDLOG_ASSERT(res->hr_remotein != NULL);
2142		PJDLOG_ASSERT(res->hr_remoteout != NULL);
2143		rw_unlock(&hio_remote_lock[ncomp]);
2144		pjdlog_debug(2, "remote_guard: Connection to %s is ok.",
2145		    res->hr_remoteaddr);
2146		return;
2147	}
2148
2149	PJDLOG_ASSERT(res->hr_remotein == NULL);
2150	PJDLOG_ASSERT(res->hr_remoteout == NULL);
2151	/*
2152	 * Upgrade the lock. It doesn't have to be atomic as no other thread
2153	 * can change connection status from disconnected to connected.
2154	 */
2155	rw_unlock(&hio_remote_lock[ncomp]);
2156	pjdlog_debug(2, "remote_guard: Reconnecting to %s.",
2157	    res->hr_remoteaddr);
2158	in = out = NULL;
2159	if (init_remote(res, &in, &out) == 0) {
2160		rw_wlock(&hio_remote_lock[ncomp]);
2161		PJDLOG_ASSERT(res->hr_remotein == NULL);
2162		PJDLOG_ASSERT(res->hr_remoteout == NULL);
2163		PJDLOG_ASSERT(in != NULL && out != NULL);
2164		res->hr_remotein = in;
2165		res->hr_remoteout = out;
2166		rw_unlock(&hio_remote_lock[ncomp]);
2167		pjdlog_info("Successfully reconnected to %s.",
2168		    res->hr_remoteaddr);
2169		sync_start();
2170	} else {
2171		/* Both connections should be NULL. */
2172		PJDLOG_ASSERT(res->hr_remotein == NULL);
2173		PJDLOG_ASSERT(res->hr_remoteout == NULL);
2174		PJDLOG_ASSERT(in == NULL && out == NULL);
2175		pjdlog_debug(2, "remote_guard: Reconnect to %s failed.",
2176		    res->hr_remoteaddr);
2177	}
2178}
2179
2180/*
2181 * Thread guards remote connections and reconnects when needed, handles
2182 * signals, etc.
2183 */
2184static void *
2185guard_thread(void *arg)
2186{
2187	struct hast_resource *res = arg;
2188	unsigned int ii, ncomps;
2189	struct timespec timeout;
2190	time_t lastcheck, now;
2191	sigset_t mask;
2192	int signo;
2193
2194	ncomps = HAST_NCOMPONENTS;
2195	lastcheck = time(NULL);
2196
2197	PJDLOG_VERIFY(sigemptyset(&mask) == 0);
2198	PJDLOG_VERIFY(sigaddset(&mask, SIGINT) == 0);
2199	PJDLOG_VERIFY(sigaddset(&mask, SIGTERM) == 0);
2200
2201	timeout.tv_sec = HAST_KEEPALIVE;
2202	timeout.tv_nsec = 0;
2203	signo = -1;
2204
2205	for (;;) {
2206		switch (signo) {
2207		case SIGINT:
2208		case SIGTERM:
2209			sigexit_received = true;
2210			primary_exitx(EX_OK,
2211			    "Termination signal received, exiting.");
2212			break;
2213		default:
2214			break;
2215		}
2216
2217		/*
2218		 * Don't check connections until we fully started,
2219		 * as we may still be looping, waiting for remote node
2220		 * to switch from primary to secondary.
2221		 */
2222		if (fullystarted) {
2223			pjdlog_debug(2, "remote_guard: Checking connections.");
2224			now = time(NULL);
2225			if (lastcheck + HAST_KEEPALIVE <= now) {
2226				for (ii = 0; ii < ncomps; ii++)
2227					guard_one(res, ii);
2228				lastcheck = now;
2229			}
2230		}
2231		signo = sigtimedwait(&mask, NULL, &timeout);
2232	}
2233	/* NOTREACHED */
2234	return (NULL);
2235}
2236