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: stable/9/sbin/hastd/event.c 231017 2012-02-05 15:51:19Z trociny $");
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
42 void
event_send(const struct hast_resource * res,int event)43 event_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) == -1) {
65 pjdlog_errno(LOG_ERR, "Unable to send event header");
66 goto done;
67 }
68 if (hast_proto_recv_hdr(res->hr_event, &nvin) == -1) {
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 */
76 done:
77 if (nvin != NULL)
78 nv_free(nvin);
79 if (nvout != NULL)
80 nv_free(nvout);
81 }
82
83 int
event_recv(const struct hast_resource * res)84 event_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) == -1) {
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) == -1) {
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);
155 fail:
156 if (nvin != NULL)
157 nv_free(nvin);
158 if (nvout != NULL)
159 nv_free(nvout);
160 return (-1);
161 }
162