secondary.c revision 220271
1/*-
2 * Copyright (c) 2009-2010 The FreeBSD Foundation
3 * Copyright (c) 2010 Pawel Jakub Dawidek <pjd@FreeBSD.org>
4 * All rights reserved.
5 *
6 * This software was developed by Pawel Jakub Dawidek under sponsorship from
7 * the FreeBSD Foundation.
8 *
9 * Redistribution and use in source and binary forms, with or without
10 * modification, are permitted provided that the following conditions
11 * are met:
12 * 1. Redistributions of source code must retain the above copyright
13 *    notice, this list of conditions and the following disclaimer.
14 * 2. Redistributions in binary form must reproduce the above copyright
15 *    notice, this list of conditions and the following disclaimer in the
16 *    documentation and/or other materials provided with the distribution.
17 *
18 * THIS SOFTWARE IS PROVIDED BY THE AUTHORS AND CONTRIBUTORS ``AS IS'' AND
19 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
20 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
21 * 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/secondary.c 220271 2011-04-02 09:25:13Z pjd $");
33
34#include <sys/param.h>
35#include <sys/time.h>
36#include <sys/bio.h>
37#include <sys/disk.h>
38#include <sys/stat.h>
39
40#include <err.h>
41#include <errno.h>
42#include <fcntl.h>
43#include <libgeom.h>
44#include <pthread.h>
45#include <signal.h>
46#include <stdint.h>
47#include <stdio.h>
48#include <string.h>
49#include <sysexits.h>
50#include <unistd.h>
51
52#include <activemap.h>
53#include <nv.h>
54#include <pjdlog.h>
55
56#include "control.h"
57#include "event.h"
58#include "hast.h"
59#include "hast_proto.h"
60#include "hastd.h"
61#include "hooks.h"
62#include "metadata.h"
63#include "proto.h"
64#include "subr.h"
65#include "synch.h"
66
67struct hio {
68	uint64_t	 hio_seq;
69	int		 hio_error;
70	struct nv	*hio_nv;
71	void		*hio_data;
72	uint8_t		 hio_cmd;
73	uint64_t	 hio_offset;
74	uint64_t	 hio_length;
75	TAILQ_ENTRY(hio) hio_next;
76};
77
78static struct hast_resource *gres;
79
80/*
81 * Free list holds unused structures. When free list is empty, we have to wait
82 * until some in-progress requests are freed.
83 */
84static TAILQ_HEAD(, hio) hio_free_list;
85static pthread_mutex_t hio_free_list_lock;
86static pthread_cond_t hio_free_list_cond;
87/*
88 * Disk thread (the one that do I/O requests) takes requests from this list.
89 */
90static TAILQ_HEAD(, hio) hio_disk_list;
91static pthread_mutex_t hio_disk_list_lock;
92static pthread_cond_t hio_disk_list_cond;
93/*
94 * There is one recv list for every component, although local components don't
95 * use recv lists as local requests are done synchronously.
96 */
97static TAILQ_HEAD(, hio) hio_send_list;
98static pthread_mutex_t hio_send_list_lock;
99static pthread_cond_t hio_send_list_cond;
100
101/*
102 * Maximum number of outstanding I/O requests.
103 */
104#define	HAST_HIO_MAX	256
105
106static void *recv_thread(void *arg);
107static void *disk_thread(void *arg);
108static void *send_thread(void *arg);
109
110#define	QUEUE_INSERT(name, hio)	do {					\
111	bool _wakeup;							\
112									\
113	mtx_lock(&hio_##name##_list_lock);				\
114	_wakeup = TAILQ_EMPTY(&hio_##name##_list);			\
115	TAILQ_INSERT_TAIL(&hio_##name##_list, (hio), hio_next);		\
116	mtx_unlock(&hio_##name##_list_lock);				\
117	if (_wakeup)							\
118		cv_signal(&hio_##name##_list_cond);			\
119} while (0)
120#define	QUEUE_TAKE(name, hio)	do {					\
121	mtx_lock(&hio_##name##_list_lock);				\
122	while (((hio) = TAILQ_FIRST(&hio_##name##_list)) == NULL) {	\
123		cv_wait(&hio_##name##_list_cond,			\
124		    &hio_##name##_list_lock);				\
125	}								\
126	TAILQ_REMOVE(&hio_##name##_list, (hio), hio_next);		\
127	mtx_unlock(&hio_##name##_list_lock);				\
128} while (0)
129
130static void
131init_environment(void)
132{
133	struct hio *hio;
134	unsigned int ii;
135
136	/*
137	 * Initialize lists, their locks and theirs condition variables.
138	 */
139	TAILQ_INIT(&hio_free_list);
140	mtx_init(&hio_free_list_lock);
141	cv_init(&hio_free_list_cond);
142	TAILQ_INIT(&hio_disk_list);
143	mtx_init(&hio_disk_list_lock);
144	cv_init(&hio_disk_list_cond);
145	TAILQ_INIT(&hio_send_list);
146	mtx_init(&hio_send_list_lock);
147	cv_init(&hio_send_list_cond);
148
149	/*
150	 * Allocate requests pool and initialize requests.
151	 */
152	for (ii = 0; ii < HAST_HIO_MAX; ii++) {
153		hio = malloc(sizeof(*hio));
154		if (hio == NULL) {
155			pjdlog_exitx(EX_TEMPFAIL,
156			    "Unable to allocate memory (%zu bytes) for hio request.",
157			    sizeof(*hio));
158		}
159		hio->hio_error = 0;
160		hio->hio_data = malloc(MAXPHYS);
161		if (hio->hio_data == NULL) {
162			pjdlog_exitx(EX_TEMPFAIL,
163			    "Unable to allocate memory (%zu bytes) for gctl_data.",
164			    (size_t)MAXPHYS);
165		}
166		TAILQ_INSERT_HEAD(&hio_free_list, hio, hio_next);
167	}
168}
169
170static void
171init_local(struct hast_resource *res)
172{
173
174	if (metadata_read(res, true) < 0)
175		exit(EX_NOINPUT);
176}
177
178static void
179init_remote(struct hast_resource *res, struct nv *nvin)
180{
181	uint64_t resuid;
182	struct nv *nvout;
183	unsigned char *map;
184	size_t mapsize;
185
186	/* Setup direction. */
187	if (proto_send(res->hr_remoteout, NULL, 0) == -1)
188		pjdlog_errno(LOG_WARNING, "Unable to set connection direction");
189
190	map = NULL;
191	mapsize = 0;
192	nvout = nv_alloc();
193	nv_add_int64(nvout, (int64_t)res->hr_datasize, "datasize");
194	nv_add_int32(nvout, (int32_t)res->hr_extentsize, "extentsize");
195	resuid = nv_get_uint64(nvin, "resuid");
196	res->hr_primary_localcnt = nv_get_uint64(nvin, "localcnt");
197	res->hr_primary_remotecnt = nv_get_uint64(nvin, "remotecnt");
198	nv_add_uint64(nvout, res->hr_secondary_localcnt, "localcnt");
199	nv_add_uint64(nvout, res->hr_secondary_remotecnt, "remotecnt");
200	mapsize = activemap_calc_ondisk_size(res->hr_local_mediasize -
201	    METADATA_SIZE, res->hr_extentsize, res->hr_local_sectorsize);
202	map = malloc(mapsize);
203	if (map == NULL) {
204		pjdlog_exitx(EX_TEMPFAIL,
205		    "Unable to allocate memory (%zu bytes) for activemap.",
206		    mapsize);
207	}
208	/*
209	 * When we work as primary and secondary is missing we will increase
210	 * localcnt in our metadata. When secondary is connected and synced
211	 * we make localcnt be equal to remotecnt, which means nodes are more
212	 * or less in sync.
213	 * Split-brain condition is when both nodes are not able to communicate
214	 * and are both configured as primary nodes. In turn, they can both
215	 * make incompatible changes to the data and we have to detect that.
216	 * Under split-brain condition we will increase our localcnt on first
217	 * write and remote node will increase its localcnt on first write.
218	 * When we connect we can see that primary's localcnt is greater than
219	 * our remotecnt (primary was modified while we weren't watching) and
220	 * our localcnt is greater than primary's remotecnt (we were modified
221	 * while primary wasn't watching).
222	 * There are many possible combinations which are all gathered below.
223	 * Don't pay too much attention to exact numbers, the more important
224	 * is to compare them. We compare secondary's local with primary's
225	 * remote and secondary's remote with primary's local.
226	 * Note that every case where primary's localcnt is smaller than
227	 * secondary's remotecnt and where secondary's localcnt is smaller than
228	 * primary's remotecnt should be impossible in practise. We will perform
229	 * full synchronization then. Those cases are marked with an asterisk.
230	 * Regular synchronization means that only extents marked as dirty are
231	 * synchronized (regular synchronization).
232	 *
233	 * SECONDARY METADATA PRIMARY METADATA
234	 * local=3 remote=3   local=2 remote=2*  ?! Full sync from secondary.
235	 * local=3 remote=3   local=2 remote=3*  ?! Full sync from primary.
236	 * local=3 remote=3   local=2 remote=4*  ?! Full sync from primary.
237	 * local=3 remote=3   local=3 remote=2   Primary is out-of-date,
238	 *                                       regular sync from secondary.
239	 * local=3 remote=3   local=3 remote=3   Regular sync just in case.
240	 * local=3 remote=3   local=3 remote=4*  ?! Full sync from primary.
241	 * local=3 remote=3   local=4 remote=2   Split-brain condition.
242	 * local=3 remote=3   local=4 remote=3   Secondary out-of-date,
243	 *                                       regular sync from primary.
244	 * local=3 remote=3   local=4 remote=4*  ?! Full sync from primary.
245	 */
246	if (res->hr_resuid == 0) {
247		/*
248		 * Provider is used for the first time. If primary node done no
249		 * writes yet as well (we will find "virgin" argument) then
250		 * there is no need to synchronize anything. If primary node
251		 * done any writes already we have to synchronize everything.
252		 */
253		PJDLOG_ASSERT(res->hr_secondary_localcnt == 0);
254		res->hr_resuid = resuid;
255		if (metadata_write(res) < 0)
256			exit(EX_NOINPUT);
257		if (nv_exists(nvin, "virgin")) {
258			free(map);
259			map = NULL;
260			mapsize = 0;
261		} else {
262			memset(map, 0xff, mapsize);
263		}
264		nv_add_uint8(nvout, HAST_SYNCSRC_PRIMARY, "syncsrc");
265	} else if (res->hr_resuid != resuid) {
266		char errmsg[256];
267
268		(void)snprintf(errmsg, sizeof(errmsg),
269		    "Resource unique ID mismatch (primary=%ju, secondary=%ju).",
270		    (uintmax_t)resuid, (uintmax_t)res->hr_resuid);
271		pjdlog_error("%s", errmsg);
272		nv_add_string(nvout, errmsg, "errmsg");
273		if (hast_proto_send(res, res->hr_remotein, nvout, NULL, 0) < 0) {
274			pjdlog_exit(EX_TEMPFAIL, "Unable to send response to %s",
275			    res->hr_remoteaddr);
276		}
277		nv_free(nvout);
278		exit(EX_CONFIG);
279	} else if (
280	    /* Is primary is out-of-date? */
281	    (res->hr_secondary_localcnt > res->hr_primary_remotecnt &&
282	     res->hr_secondary_remotecnt == res->hr_primary_localcnt) ||
283	    /* Nodes are more or less in sync? */
284	    (res->hr_secondary_localcnt == res->hr_primary_remotecnt &&
285	     res->hr_secondary_remotecnt == res->hr_primary_localcnt) ||
286	    /* Is secondary is out-of-date? */
287	    (res->hr_secondary_localcnt == res->hr_primary_remotecnt &&
288	     res->hr_secondary_remotecnt < res->hr_primary_localcnt)) {
289		/*
290		 * Nodes are more or less in sync or one of the nodes is
291		 * out-of-date.
292		 * It doesn't matter at this point which one, we just have to
293		 * send out local bitmap to the remote node.
294		 */
295		if (pread(res->hr_localfd, map, mapsize, METADATA_SIZE) !=
296		    (ssize_t)mapsize) {
297			pjdlog_exit(LOG_ERR, "Unable to read activemap");
298		}
299		if (res->hr_secondary_localcnt > res->hr_primary_remotecnt &&
300		     res->hr_secondary_remotecnt == res->hr_primary_localcnt) {
301			/* Primary is out-of-date, sync from secondary. */
302			nv_add_uint8(nvout, HAST_SYNCSRC_SECONDARY, "syncsrc");
303		} else {
304			/*
305			 * Secondary is out-of-date or counts match.
306			 * Sync from primary.
307			 */
308			nv_add_uint8(nvout, HAST_SYNCSRC_PRIMARY, "syncsrc");
309		}
310	} else if (res->hr_secondary_localcnt > res->hr_primary_remotecnt &&
311	     res->hr_primary_localcnt > res->hr_secondary_remotecnt) {
312		/*
313		 * Not good, we have split-brain condition.
314		 */
315		pjdlog_error("Split-brain detected, exiting.");
316		nv_add_string(nvout, "Split-brain condition!", "errmsg");
317		free(map);
318		map = NULL;
319		mapsize = 0;
320	} else /* if (res->hr_secondary_localcnt < res->hr_primary_remotecnt ||
321	    res->hr_primary_localcnt < res->hr_secondary_remotecnt) */ {
322		/*
323		 * This should never happen in practise, but we will perform
324		 * full synchronization.
325		 */
326		PJDLOG_ASSERT(res->hr_secondary_localcnt < res->hr_primary_remotecnt ||
327		    res->hr_primary_localcnt < res->hr_secondary_remotecnt);
328		mapsize = activemap_calc_ondisk_size(res->hr_local_mediasize -
329		    METADATA_SIZE, res->hr_extentsize,
330		    res->hr_local_sectorsize);
331		memset(map, 0xff, mapsize);
332		if (res->hr_secondary_localcnt > res->hr_primary_remotecnt) {
333			/* In this one of five cases sync from secondary. */
334			nv_add_uint8(nvout, HAST_SYNCSRC_SECONDARY, "syncsrc");
335		} else {
336			/* For the rest four cases sync from primary. */
337			nv_add_uint8(nvout, HAST_SYNCSRC_PRIMARY, "syncsrc");
338		}
339		pjdlog_warning("This should never happen, asking for full synchronization (primary(local=%ju, remote=%ju), secondary(local=%ju, remote=%ju)).",
340		    (uintmax_t)res->hr_primary_localcnt,
341		    (uintmax_t)res->hr_primary_remotecnt,
342		    (uintmax_t)res->hr_secondary_localcnt,
343		    (uintmax_t)res->hr_secondary_remotecnt);
344	}
345	nv_add_uint32(nvout, (uint32_t)mapsize, "mapsize");
346	if (hast_proto_send(res, res->hr_remotein, nvout, map, mapsize) < 0) {
347		pjdlog_exit(EX_TEMPFAIL, "Unable to send activemap to %s",
348		    res->hr_remoteaddr);
349	}
350	if (map != NULL)
351		free(map);
352	nv_free(nvout);
353	/* Setup direction. */
354	if (proto_recv(res->hr_remotein, NULL, 0) == -1)
355		pjdlog_errno(LOG_WARNING, "Unable to set connection direction");
356	if (res->hr_secondary_localcnt > res->hr_primary_remotecnt &&
357	     res->hr_primary_localcnt > res->hr_secondary_remotecnt) {
358		/* Exit on split-brain. */
359		event_send(res, EVENT_SPLITBRAIN);
360		exit(EX_CONFIG);
361	}
362}
363
364void
365hastd_secondary(struct hast_resource *res, struct nv *nvin)
366{
367	sigset_t mask;
368	pthread_t td;
369	pid_t pid;
370	int error, mode, debuglevel;
371
372	/*
373	 * Create communication channel between parent and child.
374	 */
375	if (proto_client(NULL, "socketpair://", &res->hr_ctrl) < 0) {
376		KEEP_ERRNO((void)pidfile_remove(pfh));
377		pjdlog_exit(EX_OSERR,
378		    "Unable to create control sockets between parent and child");
379	}
380	/*
381	 * Create communication channel between child and parent.
382	 */
383	if (proto_client(NULL, "socketpair://", &res->hr_event) < 0) {
384		KEEP_ERRNO((void)pidfile_remove(pfh));
385		pjdlog_exit(EX_OSERR,
386		    "Unable to create event sockets between child and parent");
387	}
388
389	pid = fork();
390	if (pid < 0) {
391		KEEP_ERRNO((void)pidfile_remove(pfh));
392		pjdlog_exit(EX_OSERR, "Unable to fork");
393	}
394
395	if (pid > 0) {
396		/* This is parent. */
397		proto_close(res->hr_remotein);
398		res->hr_remotein = NULL;
399		proto_close(res->hr_remoteout);
400		res->hr_remoteout = NULL;
401		/* Declare that we are receiver. */
402		proto_recv(res->hr_event, NULL, 0);
403		/* Declare that we are sender. */
404		proto_send(res->hr_ctrl, NULL, 0);
405		res->hr_workerpid = pid;
406		return;
407	}
408
409	gres = res;
410	mode = pjdlog_mode_get();
411	debuglevel = pjdlog_debug_get();
412
413	/* Declare that we are sender. */
414	proto_send(res->hr_event, NULL, 0);
415	/* Declare that we are receiver. */
416	proto_recv(res->hr_ctrl, NULL, 0);
417	descriptors_cleanup(res);
418
419	descriptors_assert(res, mode);
420
421	pjdlog_init(mode);
422	pjdlog_debug_set(debuglevel);
423	pjdlog_prefix_set("[%s] (%s) ", res->hr_name, role2str(res->hr_role));
424	setproctitle("%s (%s)", res->hr_name, role2str(res->hr_role));
425
426	PJDLOG_VERIFY(sigemptyset(&mask) == 0);
427	PJDLOG_VERIFY(sigprocmask(SIG_SETMASK, &mask, NULL) == 0);
428
429	/* Error in setting timeout is not critical, but why should it fail? */
430	if (proto_timeout(res->hr_remotein, 2 * HAST_KEEPALIVE) < 0)
431		pjdlog_errno(LOG_WARNING, "Unable to set connection timeout");
432	if (proto_timeout(res->hr_remoteout, res->hr_timeout) < 0)
433		pjdlog_errno(LOG_WARNING, "Unable to set connection timeout");
434
435	init_local(res);
436	init_environment();
437
438	if (drop_privs(true) != 0)
439		exit(EX_CONFIG);
440	pjdlog_info("Privileges successfully dropped.");
441
442	/*
443	 * Create the control thread before sending any event to the parent,
444	 * as we can deadlock when parent sends control request to worker,
445	 * but worker has no control thread started yet, so parent waits.
446	 * In the meantime worker sends an event to the parent, but parent
447	 * is unable to handle the event, because it waits for control
448	 * request response.
449	 */
450	error = pthread_create(&td, NULL, ctrl_thread, res);
451	PJDLOG_ASSERT(error == 0);
452
453	init_remote(res, nvin);
454	event_send(res, EVENT_CONNECT);
455
456	error = pthread_create(&td, NULL, recv_thread, res);
457	PJDLOG_ASSERT(error == 0);
458	error = pthread_create(&td, NULL, disk_thread, res);
459	PJDLOG_ASSERT(error == 0);
460	(void)send_thread(res);
461}
462
463static void
464reqlog(int loglevel, int debuglevel, int error, struct hio *hio, const char *fmt, ...)
465{
466	char msg[1024];
467	va_list ap;
468	int len;
469
470	va_start(ap, fmt);
471	len = vsnprintf(msg, sizeof(msg), fmt, ap);
472	va_end(ap);
473	if ((size_t)len < sizeof(msg)) {
474		switch (hio->hio_cmd) {
475		case HIO_READ:
476			(void)snprintf(msg + len, sizeof(msg) - len,
477			    "READ(%ju, %ju).", (uintmax_t)hio->hio_offset,
478			    (uintmax_t)hio->hio_length);
479			break;
480		case HIO_DELETE:
481			(void)snprintf(msg + len, sizeof(msg) - len,
482			    "DELETE(%ju, %ju).", (uintmax_t)hio->hio_offset,
483			    (uintmax_t)hio->hio_length);
484			break;
485		case HIO_FLUSH:
486			(void)snprintf(msg + len, sizeof(msg) - len, "FLUSH.");
487			break;
488		case HIO_WRITE:
489			(void)snprintf(msg + len, sizeof(msg) - len,
490			    "WRITE(%ju, %ju).", (uintmax_t)hio->hio_offset,
491			    (uintmax_t)hio->hio_length);
492			break;
493		case HIO_KEEPALIVE:
494			(void)snprintf(msg + len, sizeof(msg) - len, "KEEPALIVE.");
495			break;
496		default:
497			(void)snprintf(msg + len, sizeof(msg) - len,
498			    "UNKNOWN(%u).", (unsigned int)hio->hio_cmd);
499			break;
500		}
501	}
502	pjdlog_common(loglevel, debuglevel, error, "%s", msg);
503}
504
505static int
506requnpack(struct hast_resource *res, struct hio *hio)
507{
508
509	hio->hio_cmd = nv_get_uint8(hio->hio_nv, "cmd");
510	if (hio->hio_cmd == 0) {
511		pjdlog_error("Header contains no 'cmd' field.");
512		hio->hio_error = EINVAL;
513		goto end;
514	}
515	switch (hio->hio_cmd) {
516	case HIO_KEEPALIVE:
517		break;
518	case HIO_READ:
519	case HIO_WRITE:
520	case HIO_DELETE:
521		hio->hio_offset = nv_get_uint64(hio->hio_nv, "offset");
522		if (nv_error(hio->hio_nv) != 0) {
523			pjdlog_error("Header is missing 'offset' field.");
524			hio->hio_error = EINVAL;
525			goto end;
526		}
527		hio->hio_length = nv_get_uint64(hio->hio_nv, "length");
528		if (nv_error(hio->hio_nv) != 0) {
529			pjdlog_error("Header is missing 'length' field.");
530			hio->hio_error = EINVAL;
531			goto end;
532		}
533		if (hio->hio_length == 0) {
534			pjdlog_error("Data length is zero.");
535			hio->hio_error = EINVAL;
536			goto end;
537		}
538		if (hio->hio_length > MAXPHYS) {
539			pjdlog_error("Data length is too large (%ju > %ju).",
540			    (uintmax_t)hio->hio_length, (uintmax_t)MAXPHYS);
541			hio->hio_error = EINVAL;
542			goto end;
543		}
544		if ((hio->hio_offset % res->hr_local_sectorsize) != 0) {
545			pjdlog_error("Offset %ju is not multiple of sector size.",
546			    (uintmax_t)hio->hio_offset);
547			hio->hio_error = EINVAL;
548			goto end;
549		}
550		if ((hio->hio_length % res->hr_local_sectorsize) != 0) {
551			pjdlog_error("Length %ju is not multiple of sector size.",
552			    (uintmax_t)hio->hio_length);
553			hio->hio_error = EINVAL;
554			goto end;
555		}
556		if (hio->hio_offset + hio->hio_length >
557		    (uint64_t)res->hr_datasize) {
558			pjdlog_error("Data offset is too large (%ju > %ju).",
559			    (uintmax_t)(hio->hio_offset + hio->hio_length),
560			    (uintmax_t)res->hr_datasize);
561			hio->hio_error = EINVAL;
562			goto end;
563		}
564		break;
565	default:
566		pjdlog_error("Header contains invalid 'cmd' (%hhu).",
567		    hio->hio_cmd);
568		hio->hio_error = EINVAL;
569		goto end;
570	}
571	hio->hio_error = 0;
572end:
573	return (hio->hio_error);
574}
575
576static __dead2 void
577secondary_exit(int exitcode, const char *fmt, ...)
578{
579	va_list ap;
580
581	PJDLOG_ASSERT(exitcode != EX_OK);
582	va_start(ap, fmt);
583	pjdlogv_errno(LOG_ERR, fmt, ap);
584	va_end(ap);
585	event_send(gres, EVENT_DISCONNECT);
586	exit(exitcode);
587}
588
589/*
590 * Thread receives requests from the primary node.
591 */
592static void *
593recv_thread(void *arg)
594{
595	struct hast_resource *res = arg;
596	struct hio *hio;
597
598	for (;;) {
599		pjdlog_debug(2, "recv: Taking free request.");
600		QUEUE_TAKE(free, hio);
601		pjdlog_debug(2, "recv: (%p) Got request.", hio);
602		if (hast_proto_recv_hdr(res->hr_remotein, &hio->hio_nv) < 0) {
603			secondary_exit(EX_TEMPFAIL,
604			    "Unable to receive request header");
605		}
606		if (requnpack(res, hio) != 0) {
607			pjdlog_debug(2,
608			    "recv: (%p) Moving request to the send queue.",
609			    hio);
610			QUEUE_INSERT(send, hio);
611			continue;
612		}
613		reqlog(LOG_DEBUG, 2, -1, hio,
614		    "recv: (%p) Got request header: ", hio);
615		if (hio->hio_cmd == HIO_KEEPALIVE) {
616			pjdlog_debug(2,
617			    "recv: (%p) Moving request to the free queue.",
618			    hio);
619			nv_free(hio->hio_nv);
620			QUEUE_INSERT(free, hio);
621			continue;
622		} else if (hio->hio_cmd == HIO_WRITE) {
623			if (hast_proto_recv_data(res, res->hr_remotein,
624			    hio->hio_nv, hio->hio_data, MAXPHYS) < 0) {
625				secondary_exit(EX_TEMPFAIL,
626				    "Unable to receive request data");
627			}
628		}
629		pjdlog_debug(2, "recv: (%p) Moving request to the disk queue.",
630		    hio);
631		QUEUE_INSERT(disk, hio);
632	}
633	/* NOTREACHED */
634	return (NULL);
635}
636
637/*
638 * Thread reads from or writes to local component and also handles DELETE and
639 * FLUSH requests.
640 */
641static void *
642disk_thread(void *arg)
643{
644	struct hast_resource *res = arg;
645	struct hio *hio;
646	ssize_t ret;
647	bool clear_activemap;
648
649	clear_activemap = true;
650
651	for (;;) {
652		pjdlog_debug(2, "disk: Taking request.");
653		QUEUE_TAKE(disk, hio);
654		while (clear_activemap) {
655			unsigned char *map;
656			size_t mapsize;
657
658			/*
659			 * When first request is received, it means that primary
660			 * already received our activemap, merged it and stored
661			 * locally. We can now safely clear our activemap.
662			 */
663			mapsize =
664			    activemap_calc_ondisk_size(res->hr_local_mediasize -
665			    METADATA_SIZE, res->hr_extentsize,
666			    res->hr_local_sectorsize);
667			map = calloc(1, mapsize);
668			if (map == NULL) {
669				pjdlog_warning("Unable to allocate memory to clear local activemap.");
670				break;
671			}
672			if (pwrite(res->hr_localfd, map, mapsize,
673			    METADATA_SIZE) != (ssize_t)mapsize) {
674				pjdlog_errno(LOG_WARNING,
675				    "Unable to store cleared activemap");
676				free(map);
677				break;
678			}
679			free(map);
680			clear_activemap = false;
681			pjdlog_debug(1, "Local activemap cleared.");
682		}
683		reqlog(LOG_DEBUG, 2, -1, hio, "disk: (%p) Got request: ", hio);
684		/* Handle the actual request. */
685		switch (hio->hio_cmd) {
686		case HIO_READ:
687			ret = pread(res->hr_localfd, hio->hio_data,
688			    hio->hio_length,
689			    hio->hio_offset + res->hr_localoff);
690			if (ret < 0)
691				hio->hio_error = errno;
692			else if (ret != (int64_t)hio->hio_length)
693				hio->hio_error = EIO;
694			else
695				hio->hio_error = 0;
696			break;
697		case HIO_WRITE:
698			ret = pwrite(res->hr_localfd, hio->hio_data,
699			    hio->hio_length,
700			    hio->hio_offset + res->hr_localoff);
701			if (ret < 0)
702				hio->hio_error = errno;
703			else if (ret != (int64_t)hio->hio_length)
704				hio->hio_error = EIO;
705			else
706				hio->hio_error = 0;
707			break;
708		case HIO_DELETE:
709			ret = g_delete(res->hr_localfd,
710			    hio->hio_offset + res->hr_localoff,
711			    hio->hio_length);
712			if (ret < 0)
713				hio->hio_error = errno;
714			else
715				hio->hio_error = 0;
716			break;
717		case HIO_FLUSH:
718			ret = g_flush(res->hr_localfd);
719			if (ret < 0)
720				hio->hio_error = errno;
721			else
722				hio->hio_error = 0;
723			break;
724		}
725		if (hio->hio_error != 0) {
726			reqlog(LOG_ERR, 0, hio->hio_error, hio,
727			    "Request failed: ");
728		}
729		pjdlog_debug(2, "disk: (%p) Moving request to the send queue.",
730		    hio);
731		QUEUE_INSERT(send, hio);
732	}
733	/* NOTREACHED */
734	return (NULL);
735}
736
737/*
738 * Thread sends requests back to primary node.
739 */
740static void *
741send_thread(void *arg)
742{
743	struct hast_resource *res = arg;
744	struct nv *nvout;
745	struct hio *hio;
746	void *data;
747	size_t length;
748
749	for (;;) {
750		pjdlog_debug(2, "send: Taking request.");
751		QUEUE_TAKE(send, hio);
752		reqlog(LOG_DEBUG, 2, -1, hio, "send: (%p) Got request: ", hio);
753		nvout = nv_alloc();
754		/* Copy sequence number. */
755		nv_add_uint64(nvout, nv_get_uint64(hio->hio_nv, "seq"), "seq");
756		switch (hio->hio_cmd) {
757		case HIO_READ:
758			if (hio->hio_error == 0) {
759				data = hio->hio_data;
760				length = hio->hio_length;
761				break;
762			}
763			/*
764			 * We send no data in case of an error.
765			 */
766			/* FALLTHROUGH */
767		case HIO_DELETE:
768		case HIO_FLUSH:
769		case HIO_WRITE:
770			data = NULL;
771			length = 0;
772			break;
773		default:
774			abort();
775			break;
776		}
777		if (hio->hio_error != 0)
778			nv_add_int16(nvout, hio->hio_error, "error");
779		if (hast_proto_send(res, res->hr_remoteout, nvout, data,
780		    length) < 0) {
781			secondary_exit(EX_TEMPFAIL, "Unable to send reply.");
782		}
783		nv_free(nvout);
784		pjdlog_debug(2, "send: (%p) Moving request to the free queue.",
785		    hio);
786		nv_free(hio->hio_nv);
787		hio->hio_error = 0;
788		QUEUE_INSERT(free, hio);
789	}
790	/* NOTREACHED */
791	return (NULL);
792}
793