event.c revision 225787 
1/*-
2 * Copyright (c) 2010 Pawel Jakub Dawidek <pjd@FreeBSD.org>
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 *    notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 *    notice, this list of conditions and the following disclaimer in the
12 *    documentation and/or other materials provided with the distribution.
13 *
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHORS AND CONTRIBUTORS ``AS IS'' AND
15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE LIABLE
18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24 * SUCH DAMAGE.
25 */
26
27#include <sys/cdefs.h>
28__FBSDID("$FreeBSD: head/sbin/hastd/event.c 225787 2011-09-27 08:50:37Z pjd $");
29
30#include <errno.h>
31
32#include "hast.h"
33#include "hast_proto.h"
34#include "hooks.h"
35#include "nv.h"
36#include "pjdlog.h"
37#include "proto.h"
38#include "subr.h"
39
40#include "event.h"
41
42void
43event_send(const struct hast_resource *res, int event)
44{
45	struct nv *nvin, *nvout;
46	int error;
47
48	PJDLOG_ASSERT(res != NULL);
49	PJDLOG_ASSERT(event >= EVENT_MIN && event <= EVENT_MAX);
50
51	nvin = nvout = NULL;
52
53	/*
54	 * Prepare and send event to parent process.
55	 */
56	nvout = nv_alloc();
57	nv_add_uint8(nvout, (uint8_t)event, "event");
58	error = nv_error(nvout);
59	if (error != 0) {
60		pjdlog_common(LOG_ERR, 0, error,
61		    "Unable to prepare event header");
62		goto done;
63	}
64	if (hast_proto_send(res, res->hr_event, nvout, NULL, 0) < 0) {
65		pjdlog_errno(LOG_ERR, "Unable to send event header");
66		goto done;
67	}
68	if (hast_proto_recv_hdr(res->hr_event, &nvin) < 0) {
69		pjdlog_errno(LOG_ERR, "Unable to receive event header");
70		goto done;
71	}
72	/*
73	 * Do nothing with the answer. We only wait for it to be sure not
74	 * to exit too quickly after sending an event and exiting immediately.
75	 */
76done:
77	if (nvin != NULL)
78		nv_free(nvin);
79	if (nvout != NULL)
80		nv_free(nvout);
81}
82
83int
84event_recv(const struct hast_resource *res)
85{
86	struct nv *nvin, *nvout;
87	const char *evstr;
88	uint8_t event;
89	int error;
90
91	PJDLOG_ASSERT(res != NULL);
92
93	nvin = nvout = NULL;
94
95	if (hast_proto_recv_hdr(res->hr_event, &nvin) < 0) {
96		/*
97		 * First error log as debug. This is because worker process
98		 * most likely exited.
99		 */
100		pjdlog_common(LOG_DEBUG, 1, errno,
101		    "Unable to receive event header");
102		goto fail;
103	}
104
105	event = nv_get_uint8(nvin, "event");
106	if (event == EVENT_NONE) {
107		pjdlog_error("Event header is missing 'event' field.");
108		goto fail;
109	}
110
111	switch (event) {
112	case EVENT_CONNECT:
113		evstr = "connect";
114		break;
115	case EVENT_DISCONNECT:
116		evstr = "disconnect";
117		break;
118	case EVENT_SYNCSTART:
119		evstr = "syncstart";
120		break;
121	case EVENT_SYNCDONE:
122		evstr = "syncdone";
123		break;
124	case EVENT_SYNCINTR:
125		evstr = "syncintr";
126		break;
127	case EVENT_SPLITBRAIN:
128		evstr = "split-brain";
129		break;
130	default:
131		pjdlog_error("Event header contain invalid event number (%hhu).",
132		    event);
133		goto fail;
134	}
135
136	pjdlog_prefix_set("[%s] (%s) ", res->hr_name, role2str(res->hr_role));
137	hook_exec(res->hr_exec, evstr, res->hr_name, NULL);
138	pjdlog_prefix_set("%s", "");
139
140	nvout = nv_alloc();
141	nv_add_int16(nvout, 0, "error");
142	error = nv_error(nvout);
143	if (error != 0) {
144		pjdlog_common(LOG_ERR, 0, error,
145		    "Unable to prepare event header");
146		goto fail;
147	}
148	if (hast_proto_send(res, res->hr_event, nvout, NULL, 0) < 0) {
149		pjdlog_errno(LOG_ERR, "Unable to send event header");
150		goto fail;
151	}
152	nv_free(nvin);
153	nv_free(nvout);
154	return (0);
155fail:
156	if (nvin != NULL)
157		nv_free(nvin);
158	if (nvout != NULL)
159		nv_free(nvout);
160	return (-1);
161}
162