primary.c revision 214692
1204076Spjd/*-
2204076Spjd * Copyright (c) 2009 The FreeBSD Foundation
3210886Spjd * Copyright (c) 2010 Pawel Jakub Dawidek <pjd@FreeBSD.org>
4204076Spjd * All rights reserved.
5204076Spjd *
6204076Spjd * This software was developed by Pawel Jakub Dawidek under sponsorship from
7204076Spjd * the FreeBSD Foundation.
8204076Spjd *
9204076Spjd * Redistribution and use in source and binary forms, with or without
10204076Spjd * modification, are permitted provided that the following conditions
11204076Spjd * are met:
12204076Spjd * 1. Redistributions of source code must retain the above copyright
13204076Spjd *    notice, this list of conditions and the following disclaimer.
14204076Spjd * 2. Redistributions in binary form must reproduce the above copyright
15204076Spjd *    notice, this list of conditions and the following disclaimer in the
16204076Spjd *    documentation and/or other materials provided with the distribution.
17204076Spjd *
18204076Spjd * THIS SOFTWARE IS PROVIDED BY THE AUTHORS AND CONTRIBUTORS ``AS IS'' AND
19204076Spjd * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
20204076Spjd * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
21204076Spjd * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE LIABLE
22204076Spjd * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
23204076Spjd * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
24204076Spjd * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
25204076Spjd * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
26204076Spjd * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
27204076Spjd * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
28204076Spjd * SUCH DAMAGE.
29204076Spjd */
30204076Spjd
31204076Spjd#include <sys/cdefs.h>
32204076Spjd__FBSDID("$FreeBSD: head/sbin/hastd/primary.c 214692 2010-11-02 22:13:08Z pjd $");
33204076Spjd
34204076Spjd#include <sys/types.h>
35204076Spjd#include <sys/time.h>
36204076Spjd#include <sys/bio.h>
37204076Spjd#include <sys/disk.h>
38204076Spjd#include <sys/refcount.h>
39204076Spjd#include <sys/stat.h>
40204076Spjd
41204076Spjd#include <geom/gate/g_gate.h>
42204076Spjd
43204076Spjd#include <assert.h>
44204076Spjd#include <err.h>
45204076Spjd#include <errno.h>
46204076Spjd#include <fcntl.h>
47204076Spjd#include <libgeom.h>
48204076Spjd#include <pthread.h>
49211982Spjd#include <signal.h>
50204076Spjd#include <stdint.h>
51204076Spjd#include <stdio.h>
52204076Spjd#include <string.h>
53204076Spjd#include <sysexits.h>
54204076Spjd#include <unistd.h>
55204076Spjd
56204076Spjd#include <activemap.h>
57204076Spjd#include <nv.h>
58204076Spjd#include <rangelock.h>
59204076Spjd
60204076Spjd#include "control.h"
61212038Spjd#include "event.h"
62204076Spjd#include "hast.h"
63204076Spjd#include "hast_proto.h"
64204076Spjd#include "hastd.h"
65211886Spjd#include "hooks.h"
66204076Spjd#include "metadata.h"
67204076Spjd#include "proto.h"
68204076Spjd#include "pjdlog.h"
69204076Spjd#include "subr.h"
70204076Spjd#include "synch.h"
71204076Spjd
72210886Spjd/* The is only one remote component for now. */
73210886Spjd#define	ISREMOTE(no)	((no) == 1)
74210886Spjd
75204076Spjdstruct hio {
76204076Spjd	/*
77204076Spjd	 * Number of components we are still waiting for.
78204076Spjd	 * When this field goes to 0, we can send the request back to the
79204076Spjd	 * kernel. Each component has to decrease this counter by one
80204076Spjd	 * even on failure.
81204076Spjd	 */
82204076Spjd	unsigned int		 hio_countdown;
83204076Spjd	/*
84204076Spjd	 * Each component has a place to store its own error.
85204076Spjd	 * Once the request is handled by all components we can decide if the
86204076Spjd	 * request overall is successful or not.
87204076Spjd	 */
88204076Spjd	int			*hio_errors;
89204076Spjd	/*
90204076Spjd	 * Structure used to comunicate with GEOM Gate class.
91204076Spjd	 */
92204076Spjd	struct g_gate_ctl_io	 hio_ggio;
93204076Spjd	TAILQ_ENTRY(hio)	*hio_next;
94204076Spjd};
95204076Spjd#define	hio_free_next	hio_next[0]
96204076Spjd#define	hio_done_next	hio_next[0]
97204076Spjd
98204076Spjd/*
99204076Spjd * Free list holds unused structures. When free list is empty, we have to wait
100204076Spjd * until some in-progress requests are freed.
101204076Spjd */
102204076Spjdstatic TAILQ_HEAD(, hio) hio_free_list;
103204076Spjdstatic pthread_mutex_t hio_free_list_lock;
104204076Spjdstatic pthread_cond_t hio_free_list_cond;
105204076Spjd/*
106204076Spjd * There is one send list for every component. One requests is placed on all
107204076Spjd * send lists - each component gets the same request, but each component is
108204076Spjd * responsible for managing his own send list.
109204076Spjd */
110204076Spjdstatic TAILQ_HEAD(, hio) *hio_send_list;
111204076Spjdstatic pthread_mutex_t *hio_send_list_lock;
112204076Spjdstatic pthread_cond_t *hio_send_list_cond;
113204076Spjd/*
114204076Spjd * There is one recv list for every component, although local components don't
115204076Spjd * use recv lists as local requests are done synchronously.
116204076Spjd */
117204076Spjdstatic TAILQ_HEAD(, hio) *hio_recv_list;
118204076Spjdstatic pthread_mutex_t *hio_recv_list_lock;
119204076Spjdstatic pthread_cond_t *hio_recv_list_cond;
120204076Spjd/*
121204076Spjd * Request is placed on done list by the slowest component (the one that
122204076Spjd * decreased hio_countdown from 1 to 0).
123204076Spjd */
124204076Spjdstatic TAILQ_HEAD(, hio) hio_done_list;
125204076Spjdstatic pthread_mutex_t hio_done_list_lock;
126204076Spjdstatic pthread_cond_t hio_done_list_cond;
127204076Spjd/*
128204076Spjd * Structure below are for interaction with sync thread.
129204076Spjd */
130204076Spjdstatic bool sync_inprogress;
131204076Spjdstatic pthread_mutex_t sync_lock;
132204076Spjdstatic pthread_cond_t sync_cond;
133204076Spjd/*
134204076Spjd * The lock below allows to synchornize access to remote connections.
135204076Spjd */
136204076Spjdstatic pthread_rwlock_t *hio_remote_lock;
137204076Spjd
138204076Spjd/*
139204076Spjd * Lock to synchronize metadata updates. Also synchronize access to
140204076Spjd * hr_primary_localcnt and hr_primary_remotecnt fields.
141204076Spjd */
142204076Spjdstatic pthread_mutex_t metadata_lock;
143204076Spjd
144204076Spjd/*
145204076Spjd * Maximum number of outstanding I/O requests.
146204076Spjd */
147204076Spjd#define	HAST_HIO_MAX	256
148204076Spjd/*
149204076Spjd * Number of components. At this point there are only two components: local
150204076Spjd * and remote, but in the future it might be possible to use multiple local
151204076Spjd * and remote components.
152204076Spjd */
153204076Spjd#define	HAST_NCOMPONENTS	2
154204076Spjd/*
155211982Spjd * Number of seconds to sleep between reconnect retries or keepalive packets.
156204076Spjd */
157211982Spjd#define	RETRY_SLEEP		10
158204076Spjd
159204076Spjd#define	ISCONNECTED(res, no)	\
160204076Spjd	((res)->hr_remotein != NULL && (res)->hr_remoteout != NULL)
161204076Spjd
162204076Spjd#define	QUEUE_INSERT1(hio, name, ncomp)	do {				\
163204076Spjd	bool _wakeup;							\
164204076Spjd									\
165204076Spjd	mtx_lock(&hio_##name##_list_lock[(ncomp)]);			\
166204076Spjd	_wakeup = TAILQ_EMPTY(&hio_##name##_list[(ncomp)]);		\
167204076Spjd	TAILQ_INSERT_TAIL(&hio_##name##_list[(ncomp)], (hio),		\
168204076Spjd	    hio_next[(ncomp)]);						\
169204076Spjd	mtx_unlock(&hio_##name##_list_lock[ncomp]);			\
170204076Spjd	if (_wakeup)							\
171204076Spjd		cv_signal(&hio_##name##_list_cond[(ncomp)]);		\
172204076Spjd} while (0)
173204076Spjd#define	QUEUE_INSERT2(hio, name)	do {				\
174204076Spjd	bool _wakeup;							\
175204076Spjd									\
176204076Spjd	mtx_lock(&hio_##name##_list_lock);				\
177204076Spjd	_wakeup = TAILQ_EMPTY(&hio_##name##_list);			\
178204076Spjd	TAILQ_INSERT_TAIL(&hio_##name##_list, (hio), hio_##name##_next);\
179204076Spjd	mtx_unlock(&hio_##name##_list_lock);				\
180204076Spjd	if (_wakeup)							\
181204076Spjd		cv_signal(&hio_##name##_list_cond);			\
182204076Spjd} while (0)
183214692Spjd#define	QUEUE_TAKE1(hio, name, ncomp, timeout)	do {			\
184214692Spjd	bool _last;							\
185214692Spjd									\
186204076Spjd	mtx_lock(&hio_##name##_list_lock[(ncomp)]);			\
187214692Spjd	_last = false;							\
188214692Spjd	while (((hio) = TAILQ_FIRST(&hio_##name##_list[(ncomp)])) == NULL && !_last) { \
189214692Spjd		cv_timedwait(&hio_##name##_list_cond[(ncomp)],		\
190214692Spjd		    &hio_##name##_list_lock[(ncomp)], (timeout));	\
191214692Spjd		if ((timeout) != 0) 					\
192214692Spjd			_last = true;					\
193204076Spjd	}								\
194214692Spjd	if (hio != NULL) {						\
195214692Spjd		TAILQ_REMOVE(&hio_##name##_list[(ncomp)], (hio),	\
196214692Spjd		    hio_next[(ncomp)]);					\
197214692Spjd	}								\
198204076Spjd	mtx_unlock(&hio_##name##_list_lock[(ncomp)]);			\
199204076Spjd} while (0)
200204076Spjd#define	QUEUE_TAKE2(hio, name)	do {					\
201204076Spjd	mtx_lock(&hio_##name##_list_lock);				\
202204076Spjd	while (((hio) = TAILQ_FIRST(&hio_##name##_list)) == NULL) {	\
203204076Spjd		cv_wait(&hio_##name##_list_cond,			\
204204076Spjd		    &hio_##name##_list_lock);				\
205204076Spjd	}								\
206204076Spjd	TAILQ_REMOVE(&hio_##name##_list, (hio), hio_##name##_next);	\
207204076Spjd	mtx_unlock(&hio_##name##_list_lock);				\
208204076Spjd} while (0)
209204076Spjd
210209183Spjd#define	SYNCREQ(hio)		do {					\
211209183Spjd	(hio)->hio_ggio.gctl_unit = -1;					\
212209183Spjd	(hio)->hio_ggio.gctl_seq = 1;					\
213209183Spjd} while (0)
214204076Spjd#define	ISSYNCREQ(hio)		((hio)->hio_ggio.gctl_unit == -1)
215204076Spjd#define	SYNCREQDONE(hio)	do { (hio)->hio_ggio.gctl_unit = -2; } while (0)
216204076Spjd#define	ISSYNCREQDONE(hio)	((hio)->hio_ggio.gctl_unit == -2)
217204076Spjd
218204076Spjdstatic struct hast_resource *gres;
219204076Spjd
220204076Spjdstatic pthread_mutex_t range_lock;
221204076Spjdstatic struct rangelocks *range_regular;
222204076Spjdstatic bool range_regular_wait;
223204076Spjdstatic pthread_cond_t range_regular_cond;
224204076Spjdstatic struct rangelocks *range_sync;
225204076Spjdstatic bool range_sync_wait;
226204076Spjdstatic pthread_cond_t range_sync_cond;
227204076Spjd
228204076Spjdstatic void *ggate_recv_thread(void *arg);
229204076Spjdstatic void *local_send_thread(void *arg);
230204076Spjdstatic void *remote_send_thread(void *arg);
231204076Spjdstatic void *remote_recv_thread(void *arg);
232204076Spjdstatic void *ggate_send_thread(void *arg);
233204076Spjdstatic void *sync_thread(void *arg);
234204076Spjdstatic void *guard_thread(void *arg);
235204076Spjd
236211982Spjdstatic void
237204076Spjdcleanup(struct hast_resource *res)
238204076Spjd{
239204076Spjd	int rerrno;
240204076Spjd
241204076Spjd	/* Remember errno. */
242204076Spjd	rerrno = errno;
243204076Spjd
244204076Spjd	/* Destroy ggate provider if we created one. */
245204076Spjd	if (res->hr_ggateunit >= 0) {
246204076Spjd		struct g_gate_ctl_destroy ggiod;
247204076Spjd
248213533Spjd		bzero(&ggiod, sizeof(ggiod));
249204076Spjd		ggiod.gctl_version = G_GATE_VERSION;
250204076Spjd		ggiod.gctl_unit = res->hr_ggateunit;
251204076Spjd		ggiod.gctl_force = 1;
252204076Spjd		if (ioctl(res->hr_ggatefd, G_GATE_CMD_DESTROY, &ggiod) < 0) {
253213531Spjd			pjdlog_errno(LOG_WARNING,
254213531Spjd			    "Unable to destroy hast/%s device",
255204076Spjd			    res->hr_provname);
256204076Spjd		}
257204076Spjd		res->hr_ggateunit = -1;
258204076Spjd	}
259204076Spjd
260204076Spjd	/* Restore errno. */
261204076Spjd	errno = rerrno;
262204076Spjd}
263204076Spjd
264212899Spjdstatic __dead2 void
265204076Spjdprimary_exit(int exitcode, const char *fmt, ...)
266204076Spjd{
267204076Spjd	va_list ap;
268204076Spjd
269204076Spjd	assert(exitcode != EX_OK);
270204076Spjd	va_start(ap, fmt);
271204076Spjd	pjdlogv_errno(LOG_ERR, fmt, ap);
272204076Spjd	va_end(ap);
273204076Spjd	cleanup(gres);
274204076Spjd	exit(exitcode);
275204076Spjd}
276204076Spjd
277212899Spjdstatic __dead2 void
278204076Spjdprimary_exitx(int exitcode, const char *fmt, ...)
279204076Spjd{
280204076Spjd	va_list ap;
281204076Spjd
282204076Spjd	va_start(ap, fmt);
283204076Spjd	pjdlogv(exitcode == EX_OK ? LOG_INFO : LOG_ERR, fmt, ap);
284204076Spjd	va_end(ap);
285204076Spjd	cleanup(gres);
286204076Spjd	exit(exitcode);
287204076Spjd}
288204076Spjd
289204076Spjdstatic int
290204076Spjdhast_activemap_flush(struct hast_resource *res)
291204076Spjd{
292204076Spjd	const unsigned char *buf;
293204076Spjd	size_t size;
294204076Spjd
295204076Spjd	buf = activemap_bitmap(res->hr_amp, &size);
296204076Spjd	assert(buf != NULL);
297204076Spjd	assert((size % res->hr_local_sectorsize) == 0);
298204076Spjd	if (pwrite(res->hr_localfd, buf, size, METADATA_SIZE) !=
299204076Spjd	    (ssize_t)size) {
300204076Spjd		KEEP_ERRNO(pjdlog_errno(LOG_ERR,
301204076Spjd		    "Unable to flush activemap to disk"));
302204076Spjd		return (-1);
303204076Spjd	}
304204076Spjd	return (0);
305204076Spjd}
306204076Spjd
307210881Spjdstatic bool
308210881Spjdreal_remote(const struct hast_resource *res)
309210881Spjd{
310210881Spjd
311210881Spjd	return (strcmp(res->hr_remoteaddr, "none") != 0);
312210881Spjd}
313210881Spjd
314204076Spjdstatic void
315204076Spjdinit_environment(struct hast_resource *res __unused)
316204076Spjd{
317204076Spjd	struct hio *hio;
318204076Spjd	unsigned int ii, ncomps;
319204076Spjd
320204076Spjd	/*
321204076Spjd	 * In the future it might be per-resource value.
322204076Spjd	 */
323204076Spjd	ncomps = HAST_NCOMPONENTS;
324204076Spjd
325204076Spjd	/*
326204076Spjd	 * Allocate memory needed by lists.
327204076Spjd	 */
328204076Spjd	hio_send_list = malloc(sizeof(hio_send_list[0]) * ncomps);
329204076Spjd	if (hio_send_list == NULL) {
330204076Spjd		primary_exitx(EX_TEMPFAIL,
331204076Spjd		    "Unable to allocate %zu bytes of memory for send lists.",
332204076Spjd		    sizeof(hio_send_list[0]) * ncomps);
333204076Spjd	}
334204076Spjd	hio_send_list_lock = malloc(sizeof(hio_send_list_lock[0]) * ncomps);
335204076Spjd	if (hio_send_list_lock == NULL) {
336204076Spjd		primary_exitx(EX_TEMPFAIL,
337204076Spjd		    "Unable to allocate %zu bytes of memory for send list locks.",
338204076Spjd		    sizeof(hio_send_list_lock[0]) * ncomps);
339204076Spjd	}
340204076Spjd	hio_send_list_cond = malloc(sizeof(hio_send_list_cond[0]) * ncomps);
341204076Spjd	if (hio_send_list_cond == NULL) {
342204076Spjd		primary_exitx(EX_TEMPFAIL,
343204076Spjd		    "Unable to allocate %zu bytes of memory for send list condition variables.",
344204076Spjd		    sizeof(hio_send_list_cond[0]) * ncomps);
345204076Spjd	}
346204076Spjd	hio_recv_list = malloc(sizeof(hio_recv_list[0]) * ncomps);
347204076Spjd	if (hio_recv_list == NULL) {
348204076Spjd		primary_exitx(EX_TEMPFAIL,
349204076Spjd		    "Unable to allocate %zu bytes of memory for recv lists.",
350204076Spjd		    sizeof(hio_recv_list[0]) * ncomps);
351204076Spjd	}
352204076Spjd	hio_recv_list_lock = malloc(sizeof(hio_recv_list_lock[0]) * ncomps);
353204076Spjd	if (hio_recv_list_lock == NULL) {
354204076Spjd		primary_exitx(EX_TEMPFAIL,
355204076Spjd		    "Unable to allocate %zu bytes of memory for recv list locks.",
356204076Spjd		    sizeof(hio_recv_list_lock[0]) * ncomps);
357204076Spjd	}
358204076Spjd	hio_recv_list_cond = malloc(sizeof(hio_recv_list_cond[0]) * ncomps);
359204076Spjd	if (hio_recv_list_cond == NULL) {
360204076Spjd		primary_exitx(EX_TEMPFAIL,
361204076Spjd		    "Unable to allocate %zu bytes of memory for recv list condition variables.",
362204076Spjd		    sizeof(hio_recv_list_cond[0]) * ncomps);
363204076Spjd	}
364204076Spjd	hio_remote_lock = malloc(sizeof(hio_remote_lock[0]) * ncomps);
365204076Spjd	if (hio_remote_lock == NULL) {
366204076Spjd		primary_exitx(EX_TEMPFAIL,
367204076Spjd		    "Unable to allocate %zu bytes of memory for remote connections locks.",
368204076Spjd		    sizeof(hio_remote_lock[0]) * ncomps);
369204076Spjd	}
370204076Spjd
371204076Spjd	/*
372204076Spjd	 * Initialize lists, their locks and theirs condition variables.
373204076Spjd	 */
374204076Spjd	TAILQ_INIT(&hio_free_list);
375204076Spjd	mtx_init(&hio_free_list_lock);
376204076Spjd	cv_init(&hio_free_list_cond);
377204076Spjd	for (ii = 0; ii < HAST_NCOMPONENTS; ii++) {
378204076Spjd		TAILQ_INIT(&hio_send_list[ii]);
379204076Spjd		mtx_init(&hio_send_list_lock[ii]);
380204076Spjd		cv_init(&hio_send_list_cond[ii]);
381204076Spjd		TAILQ_INIT(&hio_recv_list[ii]);
382204076Spjd		mtx_init(&hio_recv_list_lock[ii]);
383204076Spjd		cv_init(&hio_recv_list_cond[ii]);
384204076Spjd		rw_init(&hio_remote_lock[ii]);
385204076Spjd	}
386204076Spjd	TAILQ_INIT(&hio_done_list);
387204076Spjd	mtx_init(&hio_done_list_lock);
388204076Spjd	cv_init(&hio_done_list_cond);
389204076Spjd	mtx_init(&metadata_lock);
390204076Spjd
391204076Spjd	/*
392204076Spjd	 * Allocate requests pool and initialize requests.
393204076Spjd	 */
394204076Spjd	for (ii = 0; ii < HAST_HIO_MAX; ii++) {
395204076Spjd		hio = malloc(sizeof(*hio));
396204076Spjd		if (hio == NULL) {
397204076Spjd			primary_exitx(EX_TEMPFAIL,
398204076Spjd			    "Unable to allocate %zu bytes of memory for hio request.",
399204076Spjd			    sizeof(*hio));
400204076Spjd		}
401204076Spjd		hio->hio_countdown = 0;
402204076Spjd		hio->hio_errors = malloc(sizeof(hio->hio_errors[0]) * ncomps);
403204076Spjd		if (hio->hio_errors == NULL) {
404204076Spjd			primary_exitx(EX_TEMPFAIL,
405204076Spjd			    "Unable allocate %zu bytes of memory for hio errors.",
406204076Spjd			    sizeof(hio->hio_errors[0]) * ncomps);
407204076Spjd		}
408204076Spjd		hio->hio_next = malloc(sizeof(hio->hio_next[0]) * ncomps);
409204076Spjd		if (hio->hio_next == NULL) {
410204076Spjd			primary_exitx(EX_TEMPFAIL,
411204076Spjd			    "Unable allocate %zu bytes of memory for hio_next field.",
412204076Spjd			    sizeof(hio->hio_next[0]) * ncomps);
413204076Spjd		}
414204076Spjd		hio->hio_ggio.gctl_version = G_GATE_VERSION;
415204076Spjd		hio->hio_ggio.gctl_data = malloc(MAXPHYS);
416204076Spjd		if (hio->hio_ggio.gctl_data == NULL) {
417204076Spjd			primary_exitx(EX_TEMPFAIL,
418204076Spjd			    "Unable to allocate %zu bytes of memory for gctl_data.",
419204076Spjd			    MAXPHYS);
420204076Spjd		}
421204076Spjd		hio->hio_ggio.gctl_length = MAXPHYS;
422204076Spjd		hio->hio_ggio.gctl_error = 0;
423204076Spjd		TAILQ_INSERT_HEAD(&hio_free_list, hio, hio_free_next);
424204076Spjd	}
425204076Spjd}
426204076Spjd
427214284Spjdstatic bool
428214284Spjdinit_resuid(struct hast_resource *res)
429214284Spjd{
430214284Spjd
431214284Spjd	mtx_lock(&metadata_lock);
432214284Spjd	if (res->hr_resuid != 0) {
433214284Spjd		mtx_unlock(&metadata_lock);
434214284Spjd		return (false);
435214284Spjd	} else {
436214284Spjd		/* Initialize unique resource identifier. */
437214284Spjd		arc4random_buf(&res->hr_resuid, sizeof(res->hr_resuid));
438214284Spjd		mtx_unlock(&metadata_lock);
439214284Spjd		if (metadata_write(res) < 0)
440214284Spjd			exit(EX_NOINPUT);
441214284Spjd		return (true);
442214284Spjd	}
443214284Spjd}
444214284Spjd
445204076Spjdstatic void
446204076Spjdinit_local(struct hast_resource *res)
447204076Spjd{
448204076Spjd	unsigned char *buf;
449204076Spjd	size_t mapsize;
450204076Spjd
451204076Spjd	if (metadata_read(res, true) < 0)
452204076Spjd		exit(EX_NOINPUT);
453204076Spjd	mtx_init(&res->hr_amp_lock);
454204076Spjd	if (activemap_init(&res->hr_amp, res->hr_datasize, res->hr_extentsize,
455204076Spjd	    res->hr_local_sectorsize, res->hr_keepdirty) < 0) {
456204076Spjd		primary_exit(EX_TEMPFAIL, "Unable to create activemap");
457204076Spjd	}
458204076Spjd	mtx_init(&range_lock);
459204076Spjd	cv_init(&range_regular_cond);
460204076Spjd	if (rangelock_init(&range_regular) < 0)
461204076Spjd		primary_exit(EX_TEMPFAIL, "Unable to create regular range lock");
462204076Spjd	cv_init(&range_sync_cond);
463204076Spjd	if (rangelock_init(&range_sync) < 0)
464204076Spjd		primary_exit(EX_TEMPFAIL, "Unable to create sync range lock");
465204076Spjd	mapsize = activemap_ondisk_size(res->hr_amp);
466204076Spjd	buf = calloc(1, mapsize);
467204076Spjd	if (buf == NULL) {
468204076Spjd		primary_exitx(EX_TEMPFAIL,
469204076Spjd		    "Unable to allocate buffer for activemap.");
470204076Spjd	}
471204076Spjd	if (pread(res->hr_localfd, buf, mapsize, METADATA_SIZE) !=
472204076Spjd	    (ssize_t)mapsize) {
473204076Spjd		primary_exit(EX_NOINPUT, "Unable to read activemap");
474204076Spjd	}
475204076Spjd	activemap_copyin(res->hr_amp, buf, mapsize);
476209181Spjd	free(buf);
477204076Spjd	if (res->hr_resuid != 0)
478204076Spjd		return;
479204076Spjd	/*
480214284Spjd	 * We're using provider for the first time. Initialize local and remote
481214284Spjd	 * counters. We don't initialize resuid here, as we want to do it just
482214284Spjd	 * in time. The reason for this is that we want to inform secondary
483214284Spjd	 * that there were no writes yet, so there is no need to synchronize
484214284Spjd	 * anything.
485204076Spjd	 */
486204076Spjd	res->hr_primary_localcnt = 1;
487204076Spjd	res->hr_primary_remotecnt = 0;
488204076Spjd	if (metadata_write(res) < 0)
489204076Spjd		exit(EX_NOINPUT);
490204076Spjd}
491204076Spjd
492205738Spjdstatic bool
493205738Spjdinit_remote(struct hast_resource *res, struct proto_conn **inp,
494205738Spjd    struct proto_conn **outp)
495204076Spjd{
496205738Spjd	struct proto_conn *in, *out;
497204076Spjd	struct nv *nvout, *nvin;
498204076Spjd	const unsigned char *token;
499204076Spjd	unsigned char *map;
500204076Spjd	const char *errmsg;
501204076Spjd	int32_t extentsize;
502204076Spjd	int64_t datasize;
503204076Spjd	uint32_t mapsize;
504204076Spjd	size_t size;
505204076Spjd
506205738Spjd	assert((inp == NULL && outp == NULL) || (inp != NULL && outp != NULL));
507210881Spjd	assert(real_remote(res));
508205738Spjd
509205738Spjd	in = out = NULL;
510211983Spjd	errmsg = NULL;
511205738Spjd
512204076Spjd	/* Prepare outgoing connection with remote node. */
513205738Spjd	if (proto_client(res->hr_remoteaddr, &out) < 0) {
514207347Spjd		primary_exit(EX_TEMPFAIL, "Unable to create connection to %s",
515204076Spjd		    res->hr_remoteaddr);
516204076Spjd	}
517204076Spjd	/* Try to connect, but accept failure. */
518205738Spjd	if (proto_connect(out) < 0) {
519204076Spjd		pjdlog_errno(LOG_WARNING, "Unable to connect to %s",
520204076Spjd		    res->hr_remoteaddr);
521204076Spjd		goto close;
522204076Spjd	}
523207371Spjd	/* Error in setting timeout is not critical, but why should it fail? */
524207371Spjd	if (proto_timeout(out, res->hr_timeout) < 0)
525207371Spjd		pjdlog_errno(LOG_WARNING, "Unable to set connection timeout");
526204076Spjd	/*
527204076Spjd	 * First handshake step.
528204076Spjd	 * Setup outgoing connection with remote node.
529204076Spjd	 */
530204076Spjd	nvout = nv_alloc();
531204076Spjd	nv_add_string(nvout, res->hr_name, "resource");
532204076Spjd	if (nv_error(nvout) != 0) {
533204076Spjd		pjdlog_common(LOG_WARNING, 0, nv_error(nvout),
534204076Spjd		    "Unable to allocate header for connection with %s",
535204076Spjd		    res->hr_remoteaddr);
536204076Spjd		nv_free(nvout);
537204076Spjd		goto close;
538204076Spjd	}
539205738Spjd	if (hast_proto_send(res, out, nvout, NULL, 0) < 0) {
540204076Spjd		pjdlog_errno(LOG_WARNING,
541204076Spjd		    "Unable to send handshake header to %s",
542204076Spjd		    res->hr_remoteaddr);
543204076Spjd		nv_free(nvout);
544204076Spjd		goto close;
545204076Spjd	}
546204076Spjd	nv_free(nvout);
547205738Spjd	if (hast_proto_recv_hdr(out, &nvin) < 0) {
548204076Spjd		pjdlog_errno(LOG_WARNING,
549204076Spjd		    "Unable to receive handshake header from %s",
550204076Spjd		    res->hr_remoteaddr);
551204076Spjd		goto close;
552204076Spjd	}
553204076Spjd	errmsg = nv_get_string(nvin, "errmsg");
554204076Spjd	if (errmsg != NULL) {
555204076Spjd		pjdlog_warning("%s", errmsg);
556204076Spjd		nv_free(nvin);
557204076Spjd		goto close;
558204076Spjd	}
559204076Spjd	token = nv_get_uint8_array(nvin, &size, "token");
560204076Spjd	if (token == NULL) {
561204076Spjd		pjdlog_warning("Handshake header from %s has no 'token' field.",
562204076Spjd		    res->hr_remoteaddr);
563204076Spjd		nv_free(nvin);
564204076Spjd		goto close;
565204076Spjd	}
566204076Spjd	if (size != sizeof(res->hr_token)) {
567204076Spjd		pjdlog_warning("Handshake header from %s contains 'token' of wrong size (got %zu, expected %zu).",
568204076Spjd		    res->hr_remoteaddr, size, sizeof(res->hr_token));
569204076Spjd		nv_free(nvin);
570204076Spjd		goto close;
571204076Spjd	}
572204076Spjd	bcopy(token, res->hr_token, sizeof(res->hr_token));
573204076Spjd	nv_free(nvin);
574204076Spjd
575204076Spjd	/*
576204076Spjd	 * Second handshake step.
577204076Spjd	 * Setup incoming connection with remote node.
578204076Spjd	 */
579205738Spjd	if (proto_client(res->hr_remoteaddr, &in) < 0) {
580204076Spjd		pjdlog_errno(LOG_WARNING, "Unable to create connection to %s",
581204076Spjd		    res->hr_remoteaddr);
582204076Spjd	}
583204076Spjd	/* Try to connect, but accept failure. */
584205738Spjd	if (proto_connect(in) < 0) {
585204076Spjd		pjdlog_errno(LOG_WARNING, "Unable to connect to %s",
586204076Spjd		    res->hr_remoteaddr);
587204076Spjd		goto close;
588204076Spjd	}
589207371Spjd	/* Error in setting timeout is not critical, but why should it fail? */
590207371Spjd	if (proto_timeout(in, res->hr_timeout) < 0)
591207371Spjd		pjdlog_errno(LOG_WARNING, "Unable to set connection timeout");
592204076Spjd	nvout = nv_alloc();
593204076Spjd	nv_add_string(nvout, res->hr_name, "resource");
594204076Spjd	nv_add_uint8_array(nvout, res->hr_token, sizeof(res->hr_token),
595204076Spjd	    "token");
596214284Spjd	if (res->hr_resuid == 0) {
597214284Spjd		/*
598214284Spjd		 * The resuid field was not yet initialized.
599214284Spjd		 * Because we do synchronization inside init_resuid(), it is
600214284Spjd		 * possible that someone already initialized it, the function
601214284Spjd		 * will return false then, but if we successfully initialized
602214284Spjd		 * it, we will get true. True means that there were no writes
603214284Spjd		 * to this resource yet and we want to inform secondary that
604214284Spjd		 * synchronization is not needed by sending "virgin" argument.
605214284Spjd		 */
606214284Spjd		if (init_resuid(res))
607214284Spjd			nv_add_int8(nvout, 1, "virgin");
608214284Spjd	}
609204076Spjd	nv_add_uint64(nvout, res->hr_resuid, "resuid");
610204076Spjd	nv_add_uint64(nvout, res->hr_primary_localcnt, "localcnt");
611204076Spjd	nv_add_uint64(nvout, res->hr_primary_remotecnt, "remotecnt");
612204076Spjd	if (nv_error(nvout) != 0) {
613204076Spjd		pjdlog_common(LOG_WARNING, 0, nv_error(nvout),
614204076Spjd		    "Unable to allocate header for connection with %s",
615204076Spjd		    res->hr_remoteaddr);
616204076Spjd		nv_free(nvout);
617204076Spjd		goto close;
618204076Spjd	}
619205738Spjd	if (hast_proto_send(res, in, nvout, NULL, 0) < 0) {
620204076Spjd		pjdlog_errno(LOG_WARNING,
621204076Spjd		    "Unable to send handshake header to %s",
622204076Spjd		    res->hr_remoteaddr);
623204076Spjd		nv_free(nvout);
624204076Spjd		goto close;
625204076Spjd	}
626204076Spjd	nv_free(nvout);
627205738Spjd	if (hast_proto_recv_hdr(out, &nvin) < 0) {
628204076Spjd		pjdlog_errno(LOG_WARNING,
629204076Spjd		    "Unable to receive handshake header from %s",
630204076Spjd		    res->hr_remoteaddr);
631204076Spjd		goto close;
632204076Spjd	}
633204076Spjd	errmsg = nv_get_string(nvin, "errmsg");
634204076Spjd	if (errmsg != NULL) {
635204076Spjd		pjdlog_warning("%s", errmsg);
636204076Spjd		nv_free(nvin);
637204076Spjd		goto close;
638204076Spjd	}
639204076Spjd	datasize = nv_get_int64(nvin, "datasize");
640204076Spjd	if (datasize != res->hr_datasize) {
641204076Spjd		pjdlog_warning("Data size differs between nodes (local=%jd, remote=%jd).",
642204076Spjd		    (intmax_t)res->hr_datasize, (intmax_t)datasize);
643204076Spjd		nv_free(nvin);
644204076Spjd		goto close;
645204076Spjd	}
646204076Spjd	extentsize = nv_get_int32(nvin, "extentsize");
647204076Spjd	if (extentsize != res->hr_extentsize) {
648204076Spjd		pjdlog_warning("Extent size differs between nodes (local=%zd, remote=%zd).",
649204076Spjd		    (ssize_t)res->hr_extentsize, (ssize_t)extentsize);
650204076Spjd		nv_free(nvin);
651204076Spjd		goto close;
652204076Spjd	}
653204076Spjd	res->hr_secondary_localcnt = nv_get_uint64(nvin, "localcnt");
654204076Spjd	res->hr_secondary_remotecnt = nv_get_uint64(nvin, "remotecnt");
655204076Spjd	res->hr_syncsrc = nv_get_uint8(nvin, "syncsrc");
656204076Spjd	map = NULL;
657204076Spjd	mapsize = nv_get_uint32(nvin, "mapsize");
658204076Spjd	if (mapsize > 0) {
659204076Spjd		map = malloc(mapsize);
660204076Spjd		if (map == NULL) {
661204076Spjd			pjdlog_error("Unable to allocate memory for remote activemap (mapsize=%ju).",
662204076Spjd			    (uintmax_t)mapsize);
663204076Spjd			nv_free(nvin);
664204076Spjd			goto close;
665204076Spjd		}
666204076Spjd		/*
667204076Spjd		 * Remote node have some dirty extents on its own, lets
668204076Spjd		 * download its activemap.
669204076Spjd		 */
670205738Spjd		if (hast_proto_recv_data(res, out, nvin, map,
671204076Spjd		    mapsize) < 0) {
672204076Spjd			pjdlog_errno(LOG_ERR,
673204076Spjd			    "Unable to receive remote activemap");
674204076Spjd			nv_free(nvin);
675204076Spjd			free(map);
676204076Spjd			goto close;
677204076Spjd		}
678204076Spjd		/*
679204076Spjd		 * Merge local and remote bitmaps.
680204076Spjd		 */
681204076Spjd		activemap_merge(res->hr_amp, map, mapsize);
682204076Spjd		free(map);
683204076Spjd		/*
684204076Spjd		 * Now that we merged bitmaps from both nodes, flush it to the
685204076Spjd		 * disk before we start to synchronize.
686204076Spjd		 */
687204076Spjd		(void)hast_activemap_flush(res);
688204076Spjd	}
689214274Spjd	nv_free(nvin);
690204076Spjd	pjdlog_info("Connected to %s.", res->hr_remoteaddr);
691205738Spjd	if (inp != NULL && outp != NULL) {
692205738Spjd		*inp = in;
693205738Spjd		*outp = out;
694205738Spjd	} else {
695205738Spjd		res->hr_remotein = in;
696205738Spjd		res->hr_remoteout = out;
697205738Spjd	}
698212038Spjd	event_send(res, EVENT_CONNECT);
699205738Spjd	return (true);
700205738Spjdclose:
701211983Spjd	if (errmsg != NULL && strcmp(errmsg, "Split-brain condition!") == 0)
702212038Spjd		event_send(res, EVENT_SPLITBRAIN);
703205738Spjd	proto_close(out);
704205738Spjd	if (in != NULL)
705205738Spjd		proto_close(in);
706205738Spjd	return (false);
707205738Spjd}
708205738Spjd
709205738Spjdstatic void
710205738Spjdsync_start(void)
711205738Spjd{
712205738Spjd
713204076Spjd	mtx_lock(&sync_lock);
714204076Spjd	sync_inprogress = true;
715204076Spjd	mtx_unlock(&sync_lock);
716204076Spjd	cv_signal(&sync_cond);
717204076Spjd}
718204076Spjd
719204076Spjdstatic void
720211878Spjdsync_stop(void)
721211878Spjd{
722211878Spjd
723211878Spjd	mtx_lock(&sync_lock);
724211878Spjd	if (sync_inprogress)
725211878Spjd		sync_inprogress = false;
726211878Spjd	mtx_unlock(&sync_lock);
727211878Spjd}
728211878Spjd
729211878Spjdstatic void
730204076Spjdinit_ggate(struct hast_resource *res)
731204076Spjd{
732204076Spjd	struct g_gate_ctl_create ggiocreate;
733204076Spjd	struct g_gate_ctl_cancel ggiocancel;
734204076Spjd
735204076Spjd	/*
736204076Spjd	 * We communicate with ggate via /dev/ggctl. Open it.
737204076Spjd	 */
738204076Spjd	res->hr_ggatefd = open("/dev/" G_GATE_CTL_NAME, O_RDWR);
739204076Spjd	if (res->hr_ggatefd < 0)
740204076Spjd		primary_exit(EX_OSFILE, "Unable to open /dev/" G_GATE_CTL_NAME);
741204076Spjd	/*
742204076Spjd	 * Create provider before trying to connect, as connection failure
743204076Spjd	 * is not critical, but may take some time.
744204076Spjd	 */
745213533Spjd	bzero(&ggiocreate, sizeof(ggiocreate));
746204076Spjd	ggiocreate.gctl_version = G_GATE_VERSION;
747204076Spjd	ggiocreate.gctl_mediasize = res->hr_datasize;
748204076Spjd	ggiocreate.gctl_sectorsize = res->hr_local_sectorsize;
749204076Spjd	ggiocreate.gctl_flags = 0;
750206669Spjd	ggiocreate.gctl_maxcount = G_GATE_MAX_QUEUE_SIZE;
751204076Spjd	ggiocreate.gctl_timeout = 0;
752204076Spjd	ggiocreate.gctl_unit = G_GATE_NAME_GIVEN;
753204076Spjd	snprintf(ggiocreate.gctl_name, sizeof(ggiocreate.gctl_name), "hast/%s",
754204076Spjd	    res->hr_provname);
755204076Spjd	if (ioctl(res->hr_ggatefd, G_GATE_CMD_CREATE, &ggiocreate) == 0) {
756204076Spjd		pjdlog_info("Device hast/%s created.", res->hr_provname);
757204076Spjd		res->hr_ggateunit = ggiocreate.gctl_unit;
758204076Spjd		return;
759204076Spjd	}
760204076Spjd	if (errno != EEXIST) {
761204076Spjd		primary_exit(EX_OSERR, "Unable to create hast/%s device",
762204076Spjd		    res->hr_provname);
763204076Spjd	}
764204076Spjd	pjdlog_debug(1,
765204076Spjd	    "Device hast/%s already exists, we will try to take it over.",
766204076Spjd	    res->hr_provname);
767204076Spjd	/*
768204076Spjd	 * If we received EEXIST, we assume that the process who created the
769204076Spjd	 * provider died and didn't clean up. In that case we will start from
770204076Spjd	 * where he left of.
771204076Spjd	 */
772213533Spjd	bzero(&ggiocancel, sizeof(ggiocancel));
773204076Spjd	ggiocancel.gctl_version = G_GATE_VERSION;
774204076Spjd	ggiocancel.gctl_unit = G_GATE_NAME_GIVEN;
775204076Spjd	snprintf(ggiocancel.gctl_name, sizeof(ggiocancel.gctl_name), "hast/%s",
776204076Spjd	    res->hr_provname);
777204076Spjd	if (ioctl(res->hr_ggatefd, G_GATE_CMD_CANCEL, &ggiocancel) == 0) {
778204076Spjd		pjdlog_info("Device hast/%s recovered.", res->hr_provname);
779204076Spjd		res->hr_ggateunit = ggiocancel.gctl_unit;
780204076Spjd		return;
781204076Spjd	}
782204076Spjd	primary_exit(EX_OSERR, "Unable to take over hast/%s device",
783204076Spjd	    res->hr_provname);
784204076Spjd}
785204076Spjd
786204076Spjdvoid
787204076Spjdhastd_primary(struct hast_resource *res)
788204076Spjd{
789204076Spjd	pthread_t td;
790204076Spjd	pid_t pid;
791204076Spjd	int error;
792204076Spjd
793204076Spjd	/*
794204076Spjd	 * Create communication channel between parent and child.
795204076Spjd	 */
796204076Spjd	if (proto_client("socketpair://", &res->hr_ctrl) < 0) {
797204076Spjd		KEEP_ERRNO((void)pidfile_remove(pfh));
798212034Spjd		pjdlog_exit(EX_OSERR,
799204076Spjd		    "Unable to create control sockets between parent and child");
800204076Spjd	}
801212038Spjd	/*
802212038Spjd	 * Create communication channel between child and parent.
803212038Spjd	 */
804212038Spjd	if (proto_client("socketpair://", &res->hr_event) < 0) {
805212038Spjd		KEEP_ERRNO((void)pidfile_remove(pfh));
806212038Spjd		pjdlog_exit(EX_OSERR,
807212038Spjd		    "Unable to create event sockets between child and parent");
808212038Spjd	}
809204076Spjd
810204076Spjd	pid = fork();
811204076Spjd	if (pid < 0) {
812204076Spjd		KEEP_ERRNO((void)pidfile_remove(pfh));
813212034Spjd		pjdlog_exit(EX_TEMPFAIL, "Unable to fork");
814204076Spjd	}
815204076Spjd
816204076Spjd	if (pid > 0) {
817204076Spjd		/* This is parent. */
818212038Spjd		/* Declare that we are receiver. */
819212038Spjd		proto_recv(res->hr_event, NULL, 0);
820204076Spjd		res->hr_workerpid = pid;
821204076Spjd		return;
822204076Spjd	}
823211977Spjd
824211984Spjd	gres = res;
825211984Spjd
826204076Spjd	(void)pidfile_close(pfh);
827211977Spjd	hook_fini();
828204076Spjd
829204076Spjd	setproctitle("%s (primary)", res->hr_name);
830204076Spjd
831212038Spjd	/* Declare that we are sender. */
832212038Spjd	proto_send(res->hr_event, NULL, 0);
833212038Spjd
834204076Spjd	init_local(res);
835213007Spjd	init_ggate(res);
836213007Spjd	init_environment(res);
837213007Spjd	/*
838213530Spjd	 * Create the guard thread first, so we can handle signals from the
839213530Spjd	 * very begining.
840213530Spjd	 */
841213530Spjd	error = pthread_create(&td, NULL, guard_thread, res);
842213530Spjd	assert(error == 0);
843213530Spjd	/*
844213007Spjd	 * Create the control thread before sending any event to the parent,
845213007Spjd	 * as we can deadlock when parent sends control request to worker,
846213007Spjd	 * but worker has no control thread started yet, so parent waits.
847213007Spjd	 * In the meantime worker sends an event to the parent, but parent
848213007Spjd	 * is unable to handle the event, because it waits for control
849213007Spjd	 * request response.
850213007Spjd	 */
851213007Spjd	error = pthread_create(&td, NULL, ctrl_thread, res);
852213007Spjd	assert(error == 0);
853210881Spjd	if (real_remote(res) && init_remote(res, NULL, NULL))
854205738Spjd		sync_start();
855204076Spjd	error = pthread_create(&td, NULL, ggate_recv_thread, res);
856204076Spjd	assert(error == 0);
857204076Spjd	error = pthread_create(&td, NULL, local_send_thread, res);
858204076Spjd	assert(error == 0);
859204076Spjd	error = pthread_create(&td, NULL, remote_send_thread, res);
860204076Spjd	assert(error == 0);
861204076Spjd	error = pthread_create(&td, NULL, remote_recv_thread, res);
862204076Spjd	assert(error == 0);
863204076Spjd	error = pthread_create(&td, NULL, ggate_send_thread, res);
864204076Spjd	assert(error == 0);
865213530Spjd	(void)sync_thread(res);
866204076Spjd}
867204076Spjd
868204076Spjdstatic void
869204076Spjdreqlog(int loglevel, int debuglevel, struct g_gate_ctl_io *ggio, const char *fmt, ...)
870204076Spjd{
871204076Spjd	char msg[1024];
872204076Spjd	va_list ap;
873204076Spjd	int len;
874204076Spjd
875204076Spjd	va_start(ap, fmt);
876204076Spjd	len = vsnprintf(msg, sizeof(msg), fmt, ap);
877204076Spjd	va_end(ap);
878204076Spjd	if ((size_t)len < sizeof(msg)) {
879204076Spjd		switch (ggio->gctl_cmd) {
880204076Spjd		case BIO_READ:
881204076Spjd			(void)snprintf(msg + len, sizeof(msg) - len,
882204076Spjd			    "READ(%ju, %ju).", (uintmax_t)ggio->gctl_offset,
883204076Spjd			    (uintmax_t)ggio->gctl_length);
884204076Spjd			break;
885204076Spjd		case BIO_DELETE:
886204076Spjd			(void)snprintf(msg + len, sizeof(msg) - len,
887204076Spjd			    "DELETE(%ju, %ju).", (uintmax_t)ggio->gctl_offset,
888204076Spjd			    (uintmax_t)ggio->gctl_length);
889204076Spjd			break;
890204076Spjd		case BIO_FLUSH:
891204076Spjd			(void)snprintf(msg + len, sizeof(msg) - len, "FLUSH.");
892204076Spjd			break;
893204076Spjd		case BIO_WRITE:
894204076Spjd			(void)snprintf(msg + len, sizeof(msg) - len,
895204076Spjd			    "WRITE(%ju, %ju).", (uintmax_t)ggio->gctl_offset,
896204076Spjd			    (uintmax_t)ggio->gctl_length);
897204076Spjd			break;
898204076Spjd		default:
899204076Spjd			(void)snprintf(msg + len, sizeof(msg) - len,
900204076Spjd			    "UNKNOWN(%u).", (unsigned int)ggio->gctl_cmd);
901204076Spjd			break;
902204076Spjd		}
903204076Spjd	}
904204076Spjd	pjdlog_common(loglevel, debuglevel, -1, "%s", msg);
905204076Spjd}
906204076Spjd
907204076Spjdstatic void
908204076Spjdremote_close(struct hast_resource *res, int ncomp)
909204076Spjd{
910204076Spjd
911204076Spjd	rw_wlock(&hio_remote_lock[ncomp]);
912204076Spjd	/*
913204076Spjd	 * A race is possible between dropping rlock and acquiring wlock -
914204076Spjd	 * another thread can close connection in-between.
915204076Spjd	 */
916204076Spjd	if (!ISCONNECTED(res, ncomp)) {
917204076Spjd		assert(res->hr_remotein == NULL);
918204076Spjd		assert(res->hr_remoteout == NULL);
919204076Spjd		rw_unlock(&hio_remote_lock[ncomp]);
920204076Spjd		return;
921204076Spjd	}
922204076Spjd
923204076Spjd	assert(res->hr_remotein != NULL);
924204076Spjd	assert(res->hr_remoteout != NULL);
925204076Spjd
926211881Spjd	pjdlog_debug(2, "Closing incoming connection to %s.",
927204076Spjd	    res->hr_remoteaddr);
928204076Spjd	proto_close(res->hr_remotein);
929204076Spjd	res->hr_remotein = NULL;
930211881Spjd	pjdlog_debug(2, "Closing outgoing connection to %s.",
931204076Spjd	    res->hr_remoteaddr);
932204076Spjd	proto_close(res->hr_remoteout);
933204076Spjd	res->hr_remoteout = NULL;
934204076Spjd
935204076Spjd	rw_unlock(&hio_remote_lock[ncomp]);
936204076Spjd
937211881Spjd	pjdlog_warning("Disconnected from %s.", res->hr_remoteaddr);
938211881Spjd
939204076Spjd	/*
940204076Spjd	 * Stop synchronization if in-progress.
941204076Spjd	 */
942211878Spjd	sync_stop();
943211984Spjd
944212038Spjd	event_send(res, EVENT_DISCONNECT);
945204076Spjd}
946204076Spjd
947204076Spjd/*
948204076Spjd * Thread receives ggate I/O requests from the kernel and passes them to
949204076Spjd * appropriate threads:
950204076Spjd * WRITE - always goes to both local_send and remote_send threads
951204076Spjd * READ (when the block is up-to-date on local component) -
952204076Spjd *	only local_send thread
953204076Spjd * READ (when the block isn't up-to-date on local component) -
954204076Spjd *	only remote_send thread
955204076Spjd * DELETE - always goes to both local_send and remote_send threads
956204076Spjd * FLUSH - always goes to both local_send and remote_send threads
957204076Spjd */
958204076Spjdstatic void *
959204076Spjdggate_recv_thread(void *arg)
960204076Spjd{
961204076Spjd	struct hast_resource *res = arg;
962204076Spjd	struct g_gate_ctl_io *ggio;
963204076Spjd	struct hio *hio;
964204076Spjd	unsigned int ii, ncomp, ncomps;
965204076Spjd	int error;
966204076Spjd
967204076Spjd	ncomps = HAST_NCOMPONENTS;
968204076Spjd
969204076Spjd	for (;;) {
970204076Spjd		pjdlog_debug(2, "ggate_recv: Taking free request.");
971204076Spjd		QUEUE_TAKE2(hio, free);
972204076Spjd		pjdlog_debug(2, "ggate_recv: (%p) Got free request.", hio);
973204076Spjd		ggio = &hio->hio_ggio;
974204076Spjd		ggio->gctl_unit = res->hr_ggateunit;
975204076Spjd		ggio->gctl_length = MAXPHYS;
976204076Spjd		ggio->gctl_error = 0;
977204076Spjd		pjdlog_debug(2,
978204076Spjd		    "ggate_recv: (%p) Waiting for request from the kernel.",
979204076Spjd		    hio);
980204076Spjd		if (ioctl(res->hr_ggatefd, G_GATE_CMD_START, ggio) < 0) {
981204076Spjd			if (sigexit_received)
982204076Spjd				pthread_exit(NULL);
983204076Spjd			primary_exit(EX_OSERR, "G_GATE_CMD_START failed");
984204076Spjd		}
985204076Spjd		error = ggio->gctl_error;
986204076Spjd		switch (error) {
987204076Spjd		case 0:
988204076Spjd			break;
989204076Spjd		case ECANCELED:
990204076Spjd			/* Exit gracefully. */
991204076Spjd			if (!sigexit_received) {
992204076Spjd				pjdlog_debug(2,
993204076Spjd				    "ggate_recv: (%p) Received cancel from the kernel.",
994204076Spjd				    hio);
995204076Spjd				pjdlog_info("Received cancel from the kernel, exiting.");
996204076Spjd			}
997204076Spjd			pthread_exit(NULL);
998204076Spjd		case ENOMEM:
999204076Spjd			/*
1000204076Spjd			 * Buffer too small? Impossible, we allocate MAXPHYS
1001204076Spjd			 * bytes - request can't be bigger than that.
1002204076Spjd			 */
1003204076Spjd			/* FALLTHROUGH */
1004204076Spjd		case ENXIO:
1005204076Spjd		default:
1006204076Spjd			primary_exitx(EX_OSERR, "G_GATE_CMD_START failed: %s.",
1007204076Spjd			    strerror(error));
1008204076Spjd		}
1009204076Spjd		for (ii = 0; ii < ncomps; ii++)
1010204076Spjd			hio->hio_errors[ii] = EINVAL;
1011204076Spjd		reqlog(LOG_DEBUG, 2, ggio,
1012204076Spjd		    "ggate_recv: (%p) Request received from the kernel: ",
1013204076Spjd		    hio);
1014204076Spjd		/*
1015204076Spjd		 * Inform all components about new write request.
1016204076Spjd		 * For read request prefer local component unless the given
1017204076Spjd		 * range is out-of-date, then use remote component.
1018204076Spjd		 */
1019204076Spjd		switch (ggio->gctl_cmd) {
1020204076Spjd		case BIO_READ:
1021204076Spjd			pjdlog_debug(2,
1022204076Spjd			    "ggate_recv: (%p) Moving request to the send queue.",
1023204076Spjd			    hio);
1024204076Spjd			refcount_init(&hio->hio_countdown, 1);
1025204076Spjd			mtx_lock(&metadata_lock);
1026204076Spjd			if (res->hr_syncsrc == HAST_SYNCSRC_UNDEF ||
1027204076Spjd			    res->hr_syncsrc == HAST_SYNCSRC_PRIMARY) {
1028204076Spjd				/*
1029204076Spjd				 * This range is up-to-date on local component,
1030204076Spjd				 * so handle request locally.
1031204076Spjd				 */
1032204076Spjd				 /* Local component is 0 for now. */
1033204076Spjd				ncomp = 0;
1034204076Spjd			} else /* if (res->hr_syncsrc ==
1035204076Spjd			    HAST_SYNCSRC_SECONDARY) */ {
1036204076Spjd				assert(res->hr_syncsrc ==
1037204076Spjd				    HAST_SYNCSRC_SECONDARY);
1038204076Spjd				/*
1039204076Spjd				 * This range is out-of-date on local component,
1040204076Spjd				 * so send request to the remote node.
1041204076Spjd				 */
1042204076Spjd				 /* Remote component is 1 for now. */
1043204076Spjd				ncomp = 1;
1044204076Spjd			}
1045204076Spjd			mtx_unlock(&metadata_lock);
1046204076Spjd			QUEUE_INSERT1(hio, send, ncomp);
1047204076Spjd			break;
1048204076Spjd		case BIO_WRITE:
1049214284Spjd			if (res->hr_resuid == 0) {
1050214284Spjd				/* This is first write, initialize resuid. */
1051214284Spjd				(void)init_resuid(res);
1052214284Spjd			}
1053204076Spjd			for (;;) {
1054204076Spjd				mtx_lock(&range_lock);
1055204076Spjd				if (rangelock_islocked(range_sync,
1056204076Spjd				    ggio->gctl_offset, ggio->gctl_length)) {
1057204076Spjd					pjdlog_debug(2,
1058204076Spjd					    "regular: Range offset=%jd length=%zu locked.",
1059204076Spjd					    (intmax_t)ggio->gctl_offset,
1060204076Spjd					    (size_t)ggio->gctl_length);
1061204076Spjd					range_regular_wait = true;
1062204076Spjd					cv_wait(&range_regular_cond, &range_lock);
1063204076Spjd					range_regular_wait = false;
1064204076Spjd					mtx_unlock(&range_lock);
1065204076Spjd					continue;
1066204076Spjd				}
1067204076Spjd				if (rangelock_add(range_regular,
1068204076Spjd				    ggio->gctl_offset, ggio->gctl_length) < 0) {
1069204076Spjd					mtx_unlock(&range_lock);
1070204076Spjd					pjdlog_debug(2,
1071204076Spjd					    "regular: Range offset=%jd length=%zu is already locked, waiting.",
1072204076Spjd					    (intmax_t)ggio->gctl_offset,
1073204076Spjd					    (size_t)ggio->gctl_length);
1074204076Spjd					sleep(1);
1075204076Spjd					continue;
1076204076Spjd				}
1077204076Spjd				mtx_unlock(&range_lock);
1078204076Spjd				break;
1079204076Spjd			}
1080204076Spjd			mtx_lock(&res->hr_amp_lock);
1081204076Spjd			if (activemap_write_start(res->hr_amp,
1082204076Spjd			    ggio->gctl_offset, ggio->gctl_length)) {
1083204076Spjd				(void)hast_activemap_flush(res);
1084204076Spjd			}
1085204076Spjd			mtx_unlock(&res->hr_amp_lock);
1086204076Spjd			/* FALLTHROUGH */
1087204076Spjd		case BIO_DELETE:
1088204076Spjd		case BIO_FLUSH:
1089204076Spjd			pjdlog_debug(2,
1090204076Spjd			    "ggate_recv: (%p) Moving request to the send queues.",
1091204076Spjd			    hio);
1092204076Spjd			refcount_init(&hio->hio_countdown, ncomps);
1093204076Spjd			for (ii = 0; ii < ncomps; ii++)
1094204076Spjd				QUEUE_INSERT1(hio, send, ii);
1095204076Spjd			break;
1096204076Spjd		}
1097204076Spjd	}
1098204076Spjd	/* NOTREACHED */
1099204076Spjd	return (NULL);
1100204076Spjd}
1101204076Spjd
1102204076Spjd/*
1103204076Spjd * Thread reads from or writes to local component.
1104204076Spjd * If local read fails, it redirects it to remote_send thread.
1105204076Spjd */
1106204076Spjdstatic void *
1107204076Spjdlocal_send_thread(void *arg)
1108204076Spjd{
1109204076Spjd	struct hast_resource *res = arg;
1110204076Spjd	struct g_gate_ctl_io *ggio;
1111204076Spjd	struct hio *hio;
1112204076Spjd	unsigned int ncomp, rncomp;
1113204076Spjd	ssize_t ret;
1114204076Spjd
1115204076Spjd	/* Local component is 0 for now. */
1116204076Spjd	ncomp = 0;
1117204076Spjd	/* Remote component is 1 for now. */
1118204076Spjd	rncomp = 1;
1119204076Spjd
1120204076Spjd	for (;;) {
1121204076Spjd		pjdlog_debug(2, "local_send: Taking request.");
1122214692Spjd		QUEUE_TAKE1(hio, send, ncomp, 0);
1123204076Spjd		pjdlog_debug(2, "local_send: (%p) Got request.", hio);
1124204076Spjd		ggio = &hio->hio_ggio;
1125204076Spjd		switch (ggio->gctl_cmd) {
1126204076Spjd		case BIO_READ:
1127204076Spjd			ret = pread(res->hr_localfd, ggio->gctl_data,
1128204076Spjd			    ggio->gctl_length,
1129204076Spjd			    ggio->gctl_offset + res->hr_localoff);
1130204076Spjd			if (ret == ggio->gctl_length)
1131204076Spjd				hio->hio_errors[ncomp] = 0;
1132204076Spjd			else {
1133204076Spjd				/*
1134204076Spjd				 * If READ failed, try to read from remote node.
1135204076Spjd				 */
1136204076Spjd				QUEUE_INSERT1(hio, send, rncomp);
1137204076Spjd				continue;
1138204076Spjd			}
1139204076Spjd			break;
1140204076Spjd		case BIO_WRITE:
1141204076Spjd			ret = pwrite(res->hr_localfd, ggio->gctl_data,
1142204076Spjd			    ggio->gctl_length,
1143204076Spjd			    ggio->gctl_offset + res->hr_localoff);
1144204076Spjd			if (ret < 0)
1145204076Spjd				hio->hio_errors[ncomp] = errno;
1146204076Spjd			else if (ret != ggio->gctl_length)
1147204076Spjd				hio->hio_errors[ncomp] = EIO;
1148204076Spjd			else
1149204076Spjd				hio->hio_errors[ncomp] = 0;
1150204076Spjd			break;
1151204076Spjd		case BIO_DELETE:
1152204076Spjd			ret = g_delete(res->hr_localfd,
1153204076Spjd			    ggio->gctl_offset + res->hr_localoff,
1154204076Spjd			    ggio->gctl_length);
1155204076Spjd			if (ret < 0)
1156204076Spjd				hio->hio_errors[ncomp] = errno;
1157204076Spjd			else
1158204076Spjd				hio->hio_errors[ncomp] = 0;
1159204076Spjd			break;
1160204076Spjd		case BIO_FLUSH:
1161204076Spjd			ret = g_flush(res->hr_localfd);
1162204076Spjd			if (ret < 0)
1163204076Spjd				hio->hio_errors[ncomp] = errno;
1164204076Spjd			else
1165204076Spjd				hio->hio_errors[ncomp] = 0;
1166204076Spjd			break;
1167204076Spjd		}
1168204076Spjd		if (refcount_release(&hio->hio_countdown)) {
1169204076Spjd			if (ISSYNCREQ(hio)) {
1170204076Spjd				mtx_lock(&sync_lock);
1171204076Spjd				SYNCREQDONE(hio);
1172204076Spjd				mtx_unlock(&sync_lock);
1173204076Spjd				cv_signal(&sync_cond);
1174204076Spjd			} else {
1175204076Spjd				pjdlog_debug(2,
1176204076Spjd				    "local_send: (%p) Moving request to the done queue.",
1177204076Spjd				    hio);
1178204076Spjd				QUEUE_INSERT2(hio, done);
1179204076Spjd			}
1180204076Spjd		}
1181204076Spjd	}
1182204076Spjd	/* NOTREACHED */
1183204076Spjd	return (NULL);
1184204076Spjd}
1185204076Spjd
1186214692Spjdstatic void
1187214692Spjdkeepalive_send(struct hast_resource *res, unsigned int ncomp)
1188214692Spjd{
1189214692Spjd	struct nv *nv;
1190214692Spjd
1191214692Spjd	if (!ISCONNECTED(res, ncomp))
1192214692Spjd		return;
1193214692Spjd
1194214692Spjd	assert(res->hr_remotein != NULL);
1195214692Spjd	assert(res->hr_remoteout != NULL);
1196214692Spjd
1197214692Spjd	nv = nv_alloc();
1198214692Spjd	nv_add_uint8(nv, HIO_KEEPALIVE, "cmd");
1199214692Spjd	if (nv_error(nv) != 0) {
1200214692Spjd		nv_free(nv);
1201214692Spjd		pjdlog_debug(1,
1202214692Spjd		    "keepalive_send: Unable to prepare header to send.");
1203214692Spjd		return;
1204214692Spjd	}
1205214692Spjd	if (hast_proto_send(res, res->hr_remoteout, nv, NULL, 0) < 0) {
1206214692Spjd		pjdlog_common(LOG_DEBUG, 1, errno,
1207214692Spjd		    "keepalive_send: Unable to send request");
1208214692Spjd		nv_free(nv);
1209214692Spjd		rw_unlock(&hio_remote_lock[ncomp]);
1210214692Spjd		remote_close(res, ncomp);
1211214692Spjd		rw_rlock(&hio_remote_lock[ncomp]);
1212214692Spjd		return;
1213214692Spjd	}
1214214692Spjd	nv_free(nv);
1215214692Spjd	pjdlog_debug(2, "keepalive_send: Request sent.");
1216214692Spjd}
1217214692Spjd
1218204076Spjd/*
1219204076Spjd * Thread sends request to secondary node.
1220204076Spjd */
1221204076Spjdstatic void *
1222204076Spjdremote_send_thread(void *arg)
1223204076Spjd{
1224204076Spjd	struct hast_resource *res = arg;
1225204076Spjd	struct g_gate_ctl_io *ggio;
1226214692Spjd	time_t lastcheck, now;
1227204076Spjd	struct hio *hio;
1228204076Spjd	struct nv *nv;
1229204076Spjd	unsigned int ncomp;
1230204076Spjd	bool wakeup;
1231204076Spjd	uint64_t offset, length;
1232204076Spjd	uint8_t cmd;
1233204076Spjd	void *data;
1234204076Spjd
1235204076Spjd	/* Remote component is 1 for now. */
1236204076Spjd	ncomp = 1;
1237214692Spjd	lastcheck = time(NULL);
1238204076Spjd
1239204076Spjd	for (;;) {
1240204076Spjd		pjdlog_debug(2, "remote_send: Taking request.");
1241214692Spjd		QUEUE_TAKE1(hio, send, ncomp, RETRY_SLEEP);
1242214692Spjd		if (hio == NULL) {
1243214692Spjd			now = time(NULL);
1244214692Spjd			if (lastcheck + RETRY_SLEEP <= now) {
1245214692Spjd				keepalive_send(res, ncomp);
1246214692Spjd				lastcheck = now;
1247214692Spjd			}
1248214692Spjd			continue;
1249214692Spjd		}
1250204076Spjd		pjdlog_debug(2, "remote_send: (%p) Got request.", hio);
1251204076Spjd		ggio = &hio->hio_ggio;
1252204076Spjd		switch (ggio->gctl_cmd) {
1253204076Spjd		case BIO_READ:
1254204076Spjd			cmd = HIO_READ;
1255204076Spjd			data = NULL;
1256204076Spjd			offset = ggio->gctl_offset;
1257204076Spjd			length = ggio->gctl_length;
1258204076Spjd			break;
1259204076Spjd		case BIO_WRITE:
1260204076Spjd			cmd = HIO_WRITE;
1261204076Spjd			data = ggio->gctl_data;
1262204076Spjd			offset = ggio->gctl_offset;
1263204076Spjd			length = ggio->gctl_length;
1264204076Spjd			break;
1265204076Spjd		case BIO_DELETE:
1266204076Spjd			cmd = HIO_DELETE;
1267204076Spjd			data = NULL;
1268204076Spjd			offset = ggio->gctl_offset;
1269204076Spjd			length = ggio->gctl_length;
1270204076Spjd			break;
1271204076Spjd		case BIO_FLUSH:
1272204076Spjd			cmd = HIO_FLUSH;
1273204076Spjd			data = NULL;
1274204076Spjd			offset = 0;
1275204076Spjd			length = 0;
1276204076Spjd			break;
1277204076Spjd		default:
1278204076Spjd			assert(!"invalid condition");
1279204076Spjd			abort();
1280204076Spjd		}
1281204076Spjd		nv = nv_alloc();
1282204076Spjd		nv_add_uint8(nv, cmd, "cmd");
1283204076Spjd		nv_add_uint64(nv, (uint64_t)ggio->gctl_seq, "seq");
1284204076Spjd		nv_add_uint64(nv, offset, "offset");
1285204076Spjd		nv_add_uint64(nv, length, "length");
1286204076Spjd		if (nv_error(nv) != 0) {
1287204076Spjd			hio->hio_errors[ncomp] = nv_error(nv);
1288204076Spjd			pjdlog_debug(2,
1289204076Spjd			    "remote_send: (%p) Unable to prepare header to send.",
1290204076Spjd			    hio);
1291204076Spjd			reqlog(LOG_ERR, 0, ggio,
1292204076Spjd			    "Unable to prepare header to send (%s): ",
1293204076Spjd			    strerror(nv_error(nv)));
1294204076Spjd			/* Move failed request immediately to the done queue. */
1295204076Spjd			goto done_queue;
1296204076Spjd		}
1297204076Spjd		pjdlog_debug(2,
1298204076Spjd		    "remote_send: (%p) Moving request to the recv queue.",
1299204076Spjd		    hio);
1300204076Spjd		/*
1301204076Spjd		 * Protect connection from disappearing.
1302204076Spjd		 */
1303204076Spjd		rw_rlock(&hio_remote_lock[ncomp]);
1304204076Spjd		if (!ISCONNECTED(res, ncomp)) {
1305204076Spjd			rw_unlock(&hio_remote_lock[ncomp]);
1306204076Spjd			hio->hio_errors[ncomp] = ENOTCONN;
1307204076Spjd			goto done_queue;
1308204076Spjd		}
1309204076Spjd		/*
1310204076Spjd		 * Move the request to recv queue before sending it, because
1311204076Spjd		 * in different order we can get reply before we move request
1312204076Spjd		 * to recv queue.
1313204076Spjd		 */
1314204076Spjd		mtx_lock(&hio_recv_list_lock[ncomp]);
1315204076Spjd		wakeup = TAILQ_EMPTY(&hio_recv_list[ncomp]);
1316204076Spjd		TAILQ_INSERT_TAIL(&hio_recv_list[ncomp], hio, hio_next[ncomp]);
1317204076Spjd		mtx_unlock(&hio_recv_list_lock[ncomp]);
1318204076Spjd		if (hast_proto_send(res, res->hr_remoteout, nv, data,
1319204076Spjd		    data != NULL ? length : 0) < 0) {
1320204076Spjd			hio->hio_errors[ncomp] = errno;
1321204076Spjd			rw_unlock(&hio_remote_lock[ncomp]);
1322204076Spjd			pjdlog_debug(2,
1323204076Spjd			    "remote_send: (%p) Unable to send request.", hio);
1324204076Spjd			reqlog(LOG_ERR, 0, ggio,
1325204076Spjd			    "Unable to send request (%s): ",
1326204076Spjd			    strerror(hio->hio_errors[ncomp]));
1327211979Spjd			remote_close(res, ncomp);
1328204076Spjd			/*
1329204076Spjd			 * Take request back from the receive queue and move
1330204076Spjd			 * it immediately to the done queue.
1331204076Spjd			 */
1332204076Spjd			mtx_lock(&hio_recv_list_lock[ncomp]);
1333204076Spjd			TAILQ_REMOVE(&hio_recv_list[ncomp], hio, hio_next[ncomp]);
1334204076Spjd			mtx_unlock(&hio_recv_list_lock[ncomp]);
1335204076Spjd			goto done_queue;
1336204076Spjd		}
1337204076Spjd		rw_unlock(&hio_remote_lock[ncomp]);
1338204076Spjd		nv_free(nv);
1339204076Spjd		if (wakeup)
1340204076Spjd			cv_signal(&hio_recv_list_cond[ncomp]);
1341204076Spjd		continue;
1342204076Spjddone_queue:
1343204076Spjd		nv_free(nv);
1344204076Spjd		if (ISSYNCREQ(hio)) {
1345204076Spjd			if (!refcount_release(&hio->hio_countdown))
1346204076Spjd				continue;
1347204076Spjd			mtx_lock(&sync_lock);
1348204076Spjd			SYNCREQDONE(hio);
1349204076Spjd			mtx_unlock(&sync_lock);
1350204076Spjd			cv_signal(&sync_cond);
1351204076Spjd			continue;
1352204076Spjd		}
1353204076Spjd		if (ggio->gctl_cmd == BIO_WRITE) {
1354204076Spjd			mtx_lock(&res->hr_amp_lock);
1355204076Spjd			if (activemap_need_sync(res->hr_amp, ggio->gctl_offset,
1356204076Spjd			    ggio->gctl_length)) {
1357204076Spjd				(void)hast_activemap_flush(res);
1358204076Spjd			}
1359204076Spjd			mtx_unlock(&res->hr_amp_lock);
1360204076Spjd		}
1361204076Spjd		if (!refcount_release(&hio->hio_countdown))
1362204076Spjd			continue;
1363204076Spjd		pjdlog_debug(2,
1364204076Spjd		    "remote_send: (%p) Moving request to the done queue.",
1365204076Spjd		    hio);
1366204076Spjd		QUEUE_INSERT2(hio, done);
1367204076Spjd	}
1368204076Spjd	/* NOTREACHED */
1369204076Spjd	return (NULL);
1370204076Spjd}
1371204076Spjd
1372204076Spjd/*
1373204076Spjd * Thread receives answer from secondary node and passes it to ggate_send
1374204076Spjd * thread.
1375204076Spjd */
1376204076Spjdstatic void *
1377204076Spjdremote_recv_thread(void *arg)
1378204076Spjd{
1379204076Spjd	struct hast_resource *res = arg;
1380204076Spjd	struct g_gate_ctl_io *ggio;
1381204076Spjd	struct hio *hio;
1382204076Spjd	struct nv *nv;
1383204076Spjd	unsigned int ncomp;
1384204076Spjd	uint64_t seq;
1385204076Spjd	int error;
1386204076Spjd
1387204076Spjd	/* Remote component is 1 for now. */
1388204076Spjd	ncomp = 1;
1389204076Spjd
1390204076Spjd	for (;;) {
1391204076Spjd		/* Wait until there is anything to receive. */
1392204076Spjd		mtx_lock(&hio_recv_list_lock[ncomp]);
1393204076Spjd		while (TAILQ_EMPTY(&hio_recv_list[ncomp])) {
1394204076Spjd			pjdlog_debug(2, "remote_recv: No requests, waiting.");
1395204076Spjd			cv_wait(&hio_recv_list_cond[ncomp],
1396204076Spjd			    &hio_recv_list_lock[ncomp]);
1397204076Spjd		}
1398204076Spjd		mtx_unlock(&hio_recv_list_lock[ncomp]);
1399204076Spjd		rw_rlock(&hio_remote_lock[ncomp]);
1400204076Spjd		if (!ISCONNECTED(res, ncomp)) {
1401204076Spjd			rw_unlock(&hio_remote_lock[ncomp]);
1402204076Spjd			/*
1403204076Spjd			 * Connection is dead, so move all pending requests to
1404204076Spjd			 * the done queue (one-by-one).
1405204076Spjd			 */
1406204076Spjd			mtx_lock(&hio_recv_list_lock[ncomp]);
1407204076Spjd			hio = TAILQ_FIRST(&hio_recv_list[ncomp]);
1408204076Spjd			assert(hio != NULL);
1409204076Spjd			TAILQ_REMOVE(&hio_recv_list[ncomp], hio,
1410204076Spjd			    hio_next[ncomp]);
1411204076Spjd			mtx_unlock(&hio_recv_list_lock[ncomp]);
1412204076Spjd			goto done_queue;
1413204076Spjd		}
1414204076Spjd		if (hast_proto_recv_hdr(res->hr_remotein, &nv) < 0) {
1415204076Spjd			pjdlog_errno(LOG_ERR,
1416204076Spjd			    "Unable to receive reply header");
1417204076Spjd			rw_unlock(&hio_remote_lock[ncomp]);
1418204076Spjd			remote_close(res, ncomp);
1419204076Spjd			continue;
1420204076Spjd		}
1421204076Spjd		rw_unlock(&hio_remote_lock[ncomp]);
1422204076Spjd		seq = nv_get_uint64(nv, "seq");
1423204076Spjd		if (seq == 0) {
1424204076Spjd			pjdlog_error("Header contains no 'seq' field.");
1425204076Spjd			nv_free(nv);
1426204076Spjd			continue;
1427204076Spjd		}
1428204076Spjd		mtx_lock(&hio_recv_list_lock[ncomp]);
1429204076Spjd		TAILQ_FOREACH(hio, &hio_recv_list[ncomp], hio_next[ncomp]) {
1430204076Spjd			if (hio->hio_ggio.gctl_seq == seq) {
1431204076Spjd				TAILQ_REMOVE(&hio_recv_list[ncomp], hio,
1432204076Spjd				    hio_next[ncomp]);
1433204076Spjd				break;
1434204076Spjd			}
1435204076Spjd		}
1436204076Spjd		mtx_unlock(&hio_recv_list_lock[ncomp]);
1437204076Spjd		if (hio == NULL) {
1438204076Spjd			pjdlog_error("Found no request matching received 'seq' field (%ju).",
1439204076Spjd			    (uintmax_t)seq);
1440204076Spjd			nv_free(nv);
1441204076Spjd			continue;
1442204076Spjd		}
1443204076Spjd		error = nv_get_int16(nv, "error");
1444204076Spjd		if (error != 0) {
1445204076Spjd			/* Request failed on remote side. */
1446204076Spjd			hio->hio_errors[ncomp] = 0;
1447204076Spjd			nv_free(nv);
1448204076Spjd			goto done_queue;
1449204076Spjd		}
1450204076Spjd		ggio = &hio->hio_ggio;
1451204076Spjd		switch (ggio->gctl_cmd) {
1452204076Spjd		case BIO_READ:
1453204076Spjd			rw_rlock(&hio_remote_lock[ncomp]);
1454204076Spjd			if (!ISCONNECTED(res, ncomp)) {
1455204076Spjd				rw_unlock(&hio_remote_lock[ncomp]);
1456204076Spjd				nv_free(nv);
1457204076Spjd				goto done_queue;
1458204076Spjd			}
1459204076Spjd			if (hast_proto_recv_data(res, res->hr_remotein, nv,
1460204076Spjd			    ggio->gctl_data, ggio->gctl_length) < 0) {
1461204076Spjd				hio->hio_errors[ncomp] = errno;
1462204076Spjd				pjdlog_errno(LOG_ERR,
1463204076Spjd				    "Unable to receive reply data");
1464204076Spjd				rw_unlock(&hio_remote_lock[ncomp]);
1465204076Spjd				nv_free(nv);
1466204076Spjd				remote_close(res, ncomp);
1467204076Spjd				goto done_queue;
1468204076Spjd			}
1469204076Spjd			rw_unlock(&hio_remote_lock[ncomp]);
1470204076Spjd			break;
1471204076Spjd		case BIO_WRITE:
1472204076Spjd		case BIO_DELETE:
1473204076Spjd		case BIO_FLUSH:
1474204076Spjd			break;
1475204076Spjd		default:
1476204076Spjd			assert(!"invalid condition");
1477204076Spjd			abort();
1478204076Spjd		}
1479204076Spjd		hio->hio_errors[ncomp] = 0;
1480204076Spjd		nv_free(nv);
1481204076Spjddone_queue:
1482204076Spjd		if (refcount_release(&hio->hio_countdown)) {
1483204076Spjd			if (ISSYNCREQ(hio)) {
1484204076Spjd				mtx_lock(&sync_lock);
1485204076Spjd				SYNCREQDONE(hio);
1486204076Spjd				mtx_unlock(&sync_lock);
1487204076Spjd				cv_signal(&sync_cond);
1488204076Spjd			} else {
1489204076Spjd				pjdlog_debug(2,
1490204076Spjd				    "remote_recv: (%p) Moving request to the done queue.",
1491204076Spjd				    hio);
1492204076Spjd				QUEUE_INSERT2(hio, done);
1493204076Spjd			}
1494204076Spjd		}
1495204076Spjd	}
1496204076Spjd	/* NOTREACHED */
1497204076Spjd	return (NULL);
1498204076Spjd}
1499204076Spjd
1500204076Spjd/*
1501204076Spjd * Thread sends answer to the kernel.
1502204076Spjd */
1503204076Spjdstatic void *
1504204076Spjdggate_send_thread(void *arg)
1505204076Spjd{
1506204076Spjd	struct hast_resource *res = arg;
1507204076Spjd	struct g_gate_ctl_io *ggio;
1508204076Spjd	struct hio *hio;
1509204076Spjd	unsigned int ii, ncomp, ncomps;
1510204076Spjd
1511204076Spjd	ncomps = HAST_NCOMPONENTS;
1512204076Spjd
1513204076Spjd	for (;;) {
1514204076Spjd		pjdlog_debug(2, "ggate_send: Taking request.");
1515204076Spjd		QUEUE_TAKE2(hio, done);
1516204076Spjd		pjdlog_debug(2, "ggate_send: (%p) Got request.", hio);
1517204076Spjd		ggio = &hio->hio_ggio;
1518204076Spjd		for (ii = 0; ii < ncomps; ii++) {
1519204076Spjd			if (hio->hio_errors[ii] == 0) {
1520204076Spjd				/*
1521204076Spjd				 * One successful request is enough to declare
1522204076Spjd				 * success.
1523204076Spjd				 */
1524204076Spjd				ggio->gctl_error = 0;
1525204076Spjd				break;
1526204076Spjd			}
1527204076Spjd		}
1528204076Spjd		if (ii == ncomps) {
1529204076Spjd			/*
1530204076Spjd			 * None of the requests were successful.
1531204076Spjd			 * Use first error.
1532204076Spjd			 */
1533204076Spjd			ggio->gctl_error = hio->hio_errors[0];
1534204076Spjd		}
1535204076Spjd		if (ggio->gctl_error == 0 && ggio->gctl_cmd == BIO_WRITE) {
1536204076Spjd			mtx_lock(&res->hr_amp_lock);
1537204076Spjd			activemap_write_complete(res->hr_amp,
1538204076Spjd			    ggio->gctl_offset, ggio->gctl_length);
1539204076Spjd			mtx_unlock(&res->hr_amp_lock);
1540204076Spjd		}
1541204076Spjd		if (ggio->gctl_cmd == BIO_WRITE) {
1542204076Spjd			/*
1543204076Spjd			 * Unlock range we locked.
1544204076Spjd			 */
1545204076Spjd			mtx_lock(&range_lock);
1546204076Spjd			rangelock_del(range_regular, ggio->gctl_offset,
1547204076Spjd			    ggio->gctl_length);
1548204076Spjd			if (range_sync_wait)
1549204076Spjd				cv_signal(&range_sync_cond);
1550204076Spjd			mtx_unlock(&range_lock);
1551204076Spjd			/*
1552204076Spjd			 * Bump local count if this is first write after
1553204076Spjd			 * connection failure with remote node.
1554204076Spjd			 */
1555204076Spjd			ncomp = 1;
1556204076Spjd			rw_rlock(&hio_remote_lock[ncomp]);
1557204076Spjd			if (!ISCONNECTED(res, ncomp)) {
1558204076Spjd				mtx_lock(&metadata_lock);
1559204076Spjd				if (res->hr_primary_localcnt ==
1560204076Spjd				    res->hr_secondary_remotecnt) {
1561204076Spjd					res->hr_primary_localcnt++;
1562204076Spjd					pjdlog_debug(1,
1563204076Spjd					    "Increasing localcnt to %ju.",
1564204076Spjd					    (uintmax_t)res->hr_primary_localcnt);
1565204076Spjd					(void)metadata_write(res);
1566204076Spjd				}
1567204076Spjd				mtx_unlock(&metadata_lock);
1568204076Spjd			}
1569204076Spjd			rw_unlock(&hio_remote_lock[ncomp]);
1570204076Spjd		}
1571204076Spjd		if (ioctl(res->hr_ggatefd, G_GATE_CMD_DONE, ggio) < 0)
1572204076Spjd			primary_exit(EX_OSERR, "G_GATE_CMD_DONE failed");
1573204076Spjd		pjdlog_debug(2,
1574204076Spjd		    "ggate_send: (%p) Moving request to the free queue.", hio);
1575204076Spjd		QUEUE_INSERT2(hio, free);
1576204076Spjd	}
1577204076Spjd	/* NOTREACHED */
1578204076Spjd	return (NULL);
1579204076Spjd}
1580204076Spjd
1581204076Spjd/*
1582204076Spjd * Thread synchronize local and remote components.
1583204076Spjd */
1584204076Spjdstatic void *
1585204076Spjdsync_thread(void *arg __unused)
1586204076Spjd{
1587204076Spjd	struct hast_resource *res = arg;
1588204076Spjd	struct hio *hio;
1589204076Spjd	struct g_gate_ctl_io *ggio;
1590204076Spjd	unsigned int ii, ncomp, ncomps;
1591204076Spjd	off_t offset, length, synced;
1592204076Spjd	bool dorewind;
1593204076Spjd	int syncext;
1594204076Spjd
1595204076Spjd	ncomps = HAST_NCOMPONENTS;
1596204076Spjd	dorewind = true;
1597211897Spjd	synced = 0;
1598211897Spjd	offset = -1;
1599204076Spjd
1600204076Spjd	for (;;) {
1601204076Spjd		mtx_lock(&sync_lock);
1602211897Spjd		if (offset >= 0 && !sync_inprogress) {
1603211879Spjd			pjdlog_info("Synchronization interrupted. "
1604211879Spjd			    "%jd bytes synchronized so far.",
1605211879Spjd			    (intmax_t)synced);
1606212038Spjd			event_send(res, EVENT_SYNCINTR);
1607211879Spjd		}
1608204076Spjd		while (!sync_inprogress) {
1609204076Spjd			dorewind = true;
1610204076Spjd			synced = 0;
1611204076Spjd			cv_wait(&sync_cond, &sync_lock);
1612204076Spjd		}
1613204076Spjd		mtx_unlock(&sync_lock);
1614204076Spjd		/*
1615204076Spjd		 * Obtain offset at which we should synchronize.
1616204076Spjd		 * Rewind synchronization if needed.
1617204076Spjd		 */
1618204076Spjd		mtx_lock(&res->hr_amp_lock);
1619204076Spjd		if (dorewind)
1620204076Spjd			activemap_sync_rewind(res->hr_amp);
1621204076Spjd		offset = activemap_sync_offset(res->hr_amp, &length, &syncext);
1622204076Spjd		if (syncext != -1) {
1623204076Spjd			/*
1624204076Spjd			 * We synchronized entire syncext extent, we can mark
1625204076Spjd			 * it as clean now.
1626204076Spjd			 */
1627204076Spjd			if (activemap_extent_complete(res->hr_amp, syncext))
1628204076Spjd				(void)hast_activemap_flush(res);
1629204076Spjd		}
1630204076Spjd		mtx_unlock(&res->hr_amp_lock);
1631204076Spjd		if (dorewind) {
1632204076Spjd			dorewind = false;
1633204076Spjd			if (offset < 0)
1634204076Spjd				pjdlog_info("Nodes are in sync.");
1635204076Spjd			else {
1636204076Spjd				pjdlog_info("Synchronization started. %ju bytes to go.",
1637204076Spjd				    (uintmax_t)(res->hr_extentsize *
1638204076Spjd				    activemap_ndirty(res->hr_amp)));
1639212038Spjd				event_send(res, EVENT_SYNCSTART);
1640204076Spjd			}
1641204076Spjd		}
1642204076Spjd		if (offset < 0) {
1643211878Spjd			sync_stop();
1644204076Spjd			pjdlog_debug(1, "Nothing to synchronize.");
1645204076Spjd			/*
1646204076Spjd			 * Synchronization complete, make both localcnt and
1647204076Spjd			 * remotecnt equal.
1648204076Spjd			 */
1649204076Spjd			ncomp = 1;
1650204076Spjd			rw_rlock(&hio_remote_lock[ncomp]);
1651204076Spjd			if (ISCONNECTED(res, ncomp)) {
1652204076Spjd				if (synced > 0) {
1653204076Spjd					pjdlog_info("Synchronization complete. "
1654204076Spjd					    "%jd bytes synchronized.",
1655204076Spjd					    (intmax_t)synced);
1656212038Spjd					event_send(res, EVENT_SYNCDONE);
1657204076Spjd				}
1658204076Spjd				mtx_lock(&metadata_lock);
1659204076Spjd				res->hr_syncsrc = HAST_SYNCSRC_UNDEF;
1660204076Spjd				res->hr_primary_localcnt =
1661204076Spjd				    res->hr_secondary_localcnt;
1662204076Spjd				res->hr_primary_remotecnt =
1663204076Spjd				    res->hr_secondary_remotecnt;
1664204076Spjd				pjdlog_debug(1,
1665204076Spjd				    "Setting localcnt to %ju and remotecnt to %ju.",
1666204076Spjd				    (uintmax_t)res->hr_primary_localcnt,
1667204076Spjd				    (uintmax_t)res->hr_secondary_localcnt);
1668204076Spjd				(void)metadata_write(res);
1669204076Spjd				mtx_unlock(&metadata_lock);
1670204076Spjd			}
1671204076Spjd			rw_unlock(&hio_remote_lock[ncomp]);
1672204076Spjd			continue;
1673204076Spjd		}
1674204076Spjd		pjdlog_debug(2, "sync: Taking free request.");
1675204076Spjd		QUEUE_TAKE2(hio, free);
1676204076Spjd		pjdlog_debug(2, "sync: (%p) Got free request.", hio);
1677204076Spjd		/*
1678204076Spjd		 * Lock the range we are going to synchronize. We don't want
1679204076Spjd		 * race where someone writes between our read and write.
1680204076Spjd		 */
1681204076Spjd		for (;;) {
1682204076Spjd			mtx_lock(&range_lock);
1683204076Spjd			if (rangelock_islocked(range_regular, offset, length)) {
1684204076Spjd				pjdlog_debug(2,
1685204076Spjd				    "sync: Range offset=%jd length=%jd locked.",
1686204076Spjd				    (intmax_t)offset, (intmax_t)length);
1687204076Spjd				range_sync_wait = true;
1688204076Spjd				cv_wait(&range_sync_cond, &range_lock);
1689204076Spjd				range_sync_wait = false;
1690204076Spjd				mtx_unlock(&range_lock);
1691204076Spjd				continue;
1692204076Spjd			}
1693204076Spjd			if (rangelock_add(range_sync, offset, length) < 0) {
1694204076Spjd				mtx_unlock(&range_lock);
1695204076Spjd				pjdlog_debug(2,
1696204076Spjd				    "sync: Range offset=%jd length=%jd is already locked, waiting.",
1697204076Spjd				    (intmax_t)offset, (intmax_t)length);
1698204076Spjd				sleep(1);
1699204076Spjd				continue;
1700204076Spjd			}
1701204076Spjd			mtx_unlock(&range_lock);
1702204076Spjd			break;
1703204076Spjd		}
1704204076Spjd		/*
1705204076Spjd		 * First read the data from synchronization source.
1706204076Spjd		 */
1707204076Spjd		SYNCREQ(hio);
1708204076Spjd		ggio = &hio->hio_ggio;
1709204076Spjd		ggio->gctl_cmd = BIO_READ;
1710204076Spjd		ggio->gctl_offset = offset;
1711204076Spjd		ggio->gctl_length = length;
1712204076Spjd		ggio->gctl_error = 0;
1713204076Spjd		for (ii = 0; ii < ncomps; ii++)
1714204076Spjd			hio->hio_errors[ii] = EINVAL;
1715204076Spjd		reqlog(LOG_DEBUG, 2, ggio, "sync: (%p) Sending sync request: ",
1716204076Spjd		    hio);
1717204076Spjd		pjdlog_debug(2, "sync: (%p) Moving request to the send queue.",
1718204076Spjd		    hio);
1719204076Spjd		mtx_lock(&metadata_lock);
1720204076Spjd		if (res->hr_syncsrc == HAST_SYNCSRC_PRIMARY) {
1721204076Spjd			/*
1722204076Spjd			 * This range is up-to-date on local component,
1723204076Spjd			 * so handle request locally.
1724204076Spjd			 */
1725204076Spjd			 /* Local component is 0 for now. */
1726204076Spjd			ncomp = 0;
1727204076Spjd		} else /* if (res->hr_syncsrc == HAST_SYNCSRC_SECONDARY) */ {
1728204076Spjd			assert(res->hr_syncsrc == HAST_SYNCSRC_SECONDARY);
1729204076Spjd			/*
1730204076Spjd			 * This range is out-of-date on local component,
1731204076Spjd			 * so send request to the remote node.
1732204076Spjd			 */
1733204076Spjd			 /* Remote component is 1 for now. */
1734204076Spjd			ncomp = 1;
1735204076Spjd		}
1736204076Spjd		mtx_unlock(&metadata_lock);
1737204076Spjd		refcount_init(&hio->hio_countdown, 1);
1738204076Spjd		QUEUE_INSERT1(hio, send, ncomp);
1739204076Spjd
1740204076Spjd		/*
1741204076Spjd		 * Let's wait for READ to finish.
1742204076Spjd		 */
1743204076Spjd		mtx_lock(&sync_lock);
1744204076Spjd		while (!ISSYNCREQDONE(hio))
1745204076Spjd			cv_wait(&sync_cond, &sync_lock);
1746204076Spjd		mtx_unlock(&sync_lock);
1747204076Spjd
1748204076Spjd		if (hio->hio_errors[ncomp] != 0) {
1749204076Spjd			pjdlog_error("Unable to read synchronization data: %s.",
1750204076Spjd			    strerror(hio->hio_errors[ncomp]));
1751204076Spjd			goto free_queue;
1752204076Spjd		}
1753204076Spjd
1754204076Spjd		/*
1755204076Spjd		 * We read the data from synchronization source, now write it
1756204076Spjd		 * to synchronization target.
1757204076Spjd		 */
1758204076Spjd		SYNCREQ(hio);
1759204076Spjd		ggio->gctl_cmd = BIO_WRITE;
1760204076Spjd		for (ii = 0; ii < ncomps; ii++)
1761204076Spjd			hio->hio_errors[ii] = EINVAL;
1762204076Spjd		reqlog(LOG_DEBUG, 2, ggio, "sync: (%p) Sending sync request: ",
1763204076Spjd		    hio);
1764204076Spjd		pjdlog_debug(2, "sync: (%p) Moving request to the send queue.",
1765204076Spjd		    hio);
1766204076Spjd		mtx_lock(&metadata_lock);
1767204076Spjd		if (res->hr_syncsrc == HAST_SYNCSRC_PRIMARY) {
1768204076Spjd			/*
1769204076Spjd			 * This range is up-to-date on local component,
1770204076Spjd			 * so we update remote component.
1771204076Spjd			 */
1772204076Spjd			 /* Remote component is 1 for now. */
1773204076Spjd			ncomp = 1;
1774204076Spjd		} else /* if (res->hr_syncsrc == HAST_SYNCSRC_SECONDARY) */ {
1775204076Spjd			assert(res->hr_syncsrc == HAST_SYNCSRC_SECONDARY);
1776204076Spjd			/*
1777204076Spjd			 * This range is out-of-date on local component,
1778204076Spjd			 * so we update it.
1779204076Spjd			 */
1780204076Spjd			 /* Local component is 0 for now. */
1781204076Spjd			ncomp = 0;
1782204076Spjd		}
1783204076Spjd		mtx_unlock(&metadata_lock);
1784204076Spjd
1785204076Spjd		pjdlog_debug(2, "sync: (%p) Moving request to the send queues.",
1786204076Spjd		    hio);
1787204076Spjd		refcount_init(&hio->hio_countdown, 1);
1788204076Spjd		QUEUE_INSERT1(hio, send, ncomp);
1789204076Spjd
1790204076Spjd		/*
1791204076Spjd		 * Let's wait for WRITE to finish.
1792204076Spjd		 */
1793204076Spjd		mtx_lock(&sync_lock);
1794204076Spjd		while (!ISSYNCREQDONE(hio))
1795204076Spjd			cv_wait(&sync_cond, &sync_lock);
1796204076Spjd		mtx_unlock(&sync_lock);
1797204076Spjd
1798204076Spjd		if (hio->hio_errors[ncomp] != 0) {
1799204076Spjd			pjdlog_error("Unable to write synchronization data: %s.",
1800204076Spjd			    strerror(hio->hio_errors[ncomp]));
1801204076Spjd			goto free_queue;
1802204076Spjd		}
1803211880Spjd
1804211880Spjd		synced += length;
1805204076Spjdfree_queue:
1806204076Spjd		mtx_lock(&range_lock);
1807204076Spjd		rangelock_del(range_sync, offset, length);
1808204076Spjd		if (range_regular_wait)
1809204076Spjd			cv_signal(&range_regular_cond);
1810204076Spjd		mtx_unlock(&range_lock);
1811204076Spjd		pjdlog_debug(2, "sync: (%p) Moving request to the free queue.",
1812204076Spjd		    hio);
1813204076Spjd		QUEUE_INSERT2(hio, free);
1814204076Spjd	}
1815204076Spjd	/* NOTREACHED */
1816204076Spjd	return (NULL);
1817204076Spjd}
1818204076Spjd
1819204076Spjdstatic void
1820210886Spjdconfig_reload(void)
1821210886Spjd{
1822210886Spjd	struct hastd_config *newcfg;
1823210886Spjd	struct hast_resource *res;
1824210886Spjd	unsigned int ii, ncomps;
1825210886Spjd	int modified;
1826210886Spjd
1827210886Spjd	pjdlog_info("Reloading configuration...");
1828210886Spjd
1829210886Spjd	ncomps = HAST_NCOMPONENTS;
1830210886Spjd
1831210886Spjd	newcfg = yy_config_parse(cfgpath, false);
1832210886Spjd	if (newcfg == NULL)
1833210886Spjd		goto failed;
1834210886Spjd
1835210886Spjd	TAILQ_FOREACH(res, &newcfg->hc_resources, hr_next) {
1836210886Spjd		if (strcmp(res->hr_name, gres->hr_name) == 0)
1837210886Spjd			break;
1838210886Spjd	}
1839210886Spjd	/*
1840210886Spjd	 * If resource was removed from the configuration file, resource
1841210886Spjd	 * name, provider name or path to local component was modified we
1842210886Spjd	 * shouldn't be here. This means that someone modified configuration
1843210886Spjd	 * file and send SIGHUP to us instead of main hastd process.
1844210886Spjd	 * Log advice and ignore the signal.
1845210886Spjd	 */
1846210886Spjd	if (res == NULL || strcmp(gres->hr_name, res->hr_name) != 0 ||
1847210886Spjd	    strcmp(gres->hr_provname, res->hr_provname) != 0 ||
1848210886Spjd	    strcmp(gres->hr_localpath, res->hr_localpath) != 0) {
1849210886Spjd		pjdlog_warning("To reload configuration send SIGHUP to the main hastd process (pid %u).",
1850210886Spjd		    (unsigned int)getppid());
1851210886Spjd		goto failed;
1852210886Spjd	}
1853210886Spjd
1854210886Spjd#define MODIFIED_REMOTEADDR	0x1
1855210886Spjd#define MODIFIED_REPLICATION	0x2
1856210886Spjd#define MODIFIED_TIMEOUT	0x4
1857211886Spjd#define MODIFIED_EXEC		0x8
1858210886Spjd	modified = 0;
1859210886Spjd	if (strcmp(gres->hr_remoteaddr, res->hr_remoteaddr) != 0) {
1860210886Spjd		/*
1861210886Spjd		 * Don't copy res->hr_remoteaddr to gres just yet.
1862210886Spjd		 * We want remote_close() to log disconnect from the old
1863210886Spjd		 * addresses, not from the new ones.
1864210886Spjd		 */
1865210886Spjd		modified |= MODIFIED_REMOTEADDR;
1866210886Spjd	}
1867210886Spjd	if (gres->hr_replication != res->hr_replication) {
1868210886Spjd		gres->hr_replication = res->hr_replication;
1869210886Spjd		modified |= MODIFIED_REPLICATION;
1870210886Spjd	}
1871210886Spjd	if (gres->hr_timeout != res->hr_timeout) {
1872210886Spjd		gres->hr_timeout = res->hr_timeout;
1873210886Spjd		modified |= MODIFIED_TIMEOUT;
1874210886Spjd	}
1875211886Spjd	if (strcmp(gres->hr_exec, res->hr_exec) != 0) {
1876211886Spjd		strlcpy(gres->hr_exec, res->hr_exec, sizeof(gres->hr_exec));
1877211886Spjd		modified |= MODIFIED_EXEC;
1878211886Spjd	}
1879210886Spjd	/*
1880210886Spjd	 * If only timeout was modified we only need to change it without
1881210886Spjd	 * reconnecting.
1882210886Spjd	 */
1883210886Spjd	if (modified == MODIFIED_TIMEOUT) {
1884210886Spjd		for (ii = 0; ii < ncomps; ii++) {
1885210886Spjd			if (!ISREMOTE(ii))
1886210886Spjd				continue;
1887210886Spjd			rw_rlock(&hio_remote_lock[ii]);
1888210886Spjd			if (!ISCONNECTED(gres, ii)) {
1889210886Spjd				rw_unlock(&hio_remote_lock[ii]);
1890210886Spjd				continue;
1891210886Spjd			}
1892210886Spjd			rw_unlock(&hio_remote_lock[ii]);
1893210886Spjd			if (proto_timeout(gres->hr_remotein,
1894210886Spjd			    gres->hr_timeout) < 0) {
1895210886Spjd				pjdlog_errno(LOG_WARNING,
1896210886Spjd				    "Unable to set connection timeout");
1897210886Spjd			}
1898210886Spjd			if (proto_timeout(gres->hr_remoteout,
1899210886Spjd			    gres->hr_timeout) < 0) {
1900210886Spjd				pjdlog_errno(LOG_WARNING,
1901210886Spjd				    "Unable to set connection timeout");
1902210886Spjd			}
1903210886Spjd		}
1904211886Spjd	} else if ((modified &
1905211886Spjd	    (MODIFIED_REMOTEADDR | MODIFIED_REPLICATION)) != 0) {
1906210886Spjd		for (ii = 0; ii < ncomps; ii++) {
1907210886Spjd			if (!ISREMOTE(ii))
1908210886Spjd				continue;
1909210886Spjd			remote_close(gres, ii);
1910210886Spjd		}
1911210886Spjd		if (modified & MODIFIED_REMOTEADDR) {
1912210886Spjd			strlcpy(gres->hr_remoteaddr, res->hr_remoteaddr,
1913210886Spjd			    sizeof(gres->hr_remoteaddr));
1914210886Spjd		}
1915210886Spjd	}
1916210886Spjd#undef	MODIFIED_REMOTEADDR
1917210886Spjd#undef	MODIFIED_REPLICATION
1918210886Spjd#undef	MODIFIED_TIMEOUT
1919211886Spjd#undef	MODIFIED_EXEC
1920210886Spjd
1921210886Spjd	pjdlog_info("Configuration reloaded successfully.");
1922210886Spjd	return;
1923210886Spjdfailed:
1924210886Spjd	if (newcfg != NULL) {
1925210886Spjd		if (newcfg->hc_controlconn != NULL)
1926210886Spjd			proto_close(newcfg->hc_controlconn);
1927210886Spjd		if (newcfg->hc_listenconn != NULL)
1928210886Spjd			proto_close(newcfg->hc_listenconn);
1929210886Spjd		yy_config_free(newcfg);
1930210886Spjd	}
1931210886Spjd	pjdlog_warning("Configuration not reloaded.");
1932210886Spjd}
1933210886Spjd
1934211882Spjdstatic void
1935211981Spjdguard_one(struct hast_resource *res, unsigned int ncomp)
1936211981Spjd{
1937211981Spjd	struct proto_conn *in, *out;
1938211981Spjd
1939211981Spjd	if (!ISREMOTE(ncomp))
1940211981Spjd		return;
1941211981Spjd
1942211981Spjd	rw_rlock(&hio_remote_lock[ncomp]);
1943211981Spjd
1944211981Spjd	if (!real_remote(res)) {
1945211981Spjd		rw_unlock(&hio_remote_lock[ncomp]);
1946211981Spjd		return;
1947211981Spjd	}
1948211981Spjd
1949211981Spjd	if (ISCONNECTED(res, ncomp)) {
1950211981Spjd		assert(res->hr_remotein != NULL);
1951211981Spjd		assert(res->hr_remoteout != NULL);
1952211981Spjd		rw_unlock(&hio_remote_lock[ncomp]);
1953211981Spjd		pjdlog_debug(2, "remote_guard: Connection to %s is ok.",
1954211981Spjd		    res->hr_remoteaddr);
1955211981Spjd		return;
1956211981Spjd	}
1957211981Spjd
1958211981Spjd	assert(res->hr_remotein == NULL);
1959211981Spjd	assert(res->hr_remoteout == NULL);
1960211981Spjd	/*
1961211981Spjd	 * Upgrade the lock. It doesn't have to be atomic as no other thread
1962211981Spjd	 * can change connection status from disconnected to connected.
1963211981Spjd	 */
1964211981Spjd	rw_unlock(&hio_remote_lock[ncomp]);
1965211981Spjd	pjdlog_debug(2, "remote_guard: Reconnecting to %s.",
1966211981Spjd	    res->hr_remoteaddr);
1967211981Spjd	in = out = NULL;
1968211981Spjd	if (init_remote(res, &in, &out)) {
1969211981Spjd		rw_wlock(&hio_remote_lock[ncomp]);
1970211981Spjd		assert(res->hr_remotein == NULL);
1971211981Spjd		assert(res->hr_remoteout == NULL);
1972211981Spjd		assert(in != NULL && out != NULL);
1973211981Spjd		res->hr_remotein = in;
1974211981Spjd		res->hr_remoteout = out;
1975211981Spjd		rw_unlock(&hio_remote_lock[ncomp]);
1976211981Spjd		pjdlog_info("Successfully reconnected to %s.",
1977211981Spjd		    res->hr_remoteaddr);
1978211981Spjd		sync_start();
1979211981Spjd	} else {
1980211981Spjd		/* Both connections should be NULL. */
1981211981Spjd		assert(res->hr_remotein == NULL);
1982211981Spjd		assert(res->hr_remoteout == NULL);
1983211981Spjd		assert(in == NULL && out == NULL);
1984211981Spjd		pjdlog_debug(2, "remote_guard: Reconnect to %s failed.",
1985211981Spjd		    res->hr_remoteaddr);
1986211981Spjd	}
1987211981Spjd}
1988211981Spjd
1989204076Spjd/*
1990204076Spjd * Thread guards remote connections and reconnects when needed, handles
1991204076Spjd * signals, etc.
1992204076Spjd */
1993204076Spjdstatic void *
1994204076Spjdguard_thread(void *arg)
1995204076Spjd{
1996204076Spjd	struct hast_resource *res = arg;
1997204076Spjd	unsigned int ii, ncomps;
1998211982Spjd	struct timespec timeout;
1999211981Spjd	time_t lastcheck, now;
2000211982Spjd	sigset_t mask;
2001211982Spjd	int signo;
2002204076Spjd
2003204076Spjd	ncomps = HAST_NCOMPONENTS;
2004211981Spjd	lastcheck = time(NULL);
2005204076Spjd
2006211982Spjd	PJDLOG_VERIFY(sigemptyset(&mask) == 0);
2007211982Spjd	PJDLOG_VERIFY(sigaddset(&mask, SIGHUP) == 0);
2008211982Spjd	PJDLOG_VERIFY(sigaddset(&mask, SIGINT) == 0);
2009211982Spjd	PJDLOG_VERIFY(sigaddset(&mask, SIGTERM) == 0);
2010211982Spjd
2011211982Spjd	timeout.tv_nsec = 0;
2012211982Spjd	signo = -1;
2013211982Spjd
2014204076Spjd	for (;;) {
2015211982Spjd		switch (signo) {
2016211982Spjd		case SIGHUP:
2017211982Spjd			config_reload();
2018211982Spjd			break;
2019211982Spjd		case SIGINT:
2020211982Spjd		case SIGTERM:
2021211982Spjd			sigexit_received = true;
2022204076Spjd			primary_exitx(EX_OK,
2023204076Spjd			    "Termination signal received, exiting.");
2024211982Spjd			break;
2025211982Spjd		default:
2026211982Spjd			break;
2027204076Spjd		}
2028211882Spjd
2029204076Spjd		pjdlog_debug(2, "remote_guard: Checking connections.");
2030211981Spjd		now = time(NULL);
2031211982Spjd		if (lastcheck + RETRY_SLEEP <= now) {
2032211982Spjd			for (ii = 0; ii < ncomps; ii++)
2033211981Spjd				guard_one(res, ii);
2034211981Spjd			lastcheck = now;
2035204076Spjd		}
2036211982Spjd		timeout.tv_sec = RETRY_SLEEP;
2037211982Spjd		signo = sigtimedwait(&mask, NULL, &timeout);
2038204076Spjd	}
2039204076Spjd	/* NOTREACHED */
2040204076Spjd	return (NULL);
2041204076Spjd}
2042