Blame

Date:
Message:
Merge branch 'portable' Trying to maintain two branches was impossible at this stage for me alone, so I'll recognize my stupidity and merge the portable branch into the main one. The repository will build using autotools.
Actions:
History | Blob | Raw File
  1 fb1a36c0 2022-01-09 op /*

  2 fb1a36c0 2022-01-09 op  * Copyright (c) 2003, 2004 Henning Brauer <henning@openbsd.org>

  3 fb1a36c0 2022-01-09 op  *

  4 fb1a36c0 2022-01-09 op  * Permission to use, copy, modify, and distribute this software for any

  5 fb1a36c0 2022-01-09 op  * purpose with or without fee is hereby granted, provided that the above

  6 fb1a36c0 2022-01-09 op  * copyright notice and this permission notice appear in all copies.

  7 fb1a36c0 2022-01-09 op  *

  8 fb1a36c0 2022-01-09 op  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES

  9 fb1a36c0 2022-01-09 op  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF

 10 fb1a36c0 2022-01-09 op  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR

 11 fb1a36c0 2022-01-09 op  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES

 12 fb1a36c0 2022-01-09 op  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN

 13 fb1a36c0 2022-01-09 op  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF

 14 fb1a36c0 2022-01-09 op  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.

 15 fb1a36c0 2022-01-09 op  */

 16 fb1a36c0 2022-01-09 op 

 17 bbcba3ed 2022-01-10 op #include "compat.h"

 18 bbcba3ed 2022-01-10 op 

 19 fb1a36c0 2022-01-09 op #include <sys/types.h>

 20 fb1a36c0 2022-01-09 op #include <sys/stat.h>

 21 fb1a36c0 2022-01-09 op #include <sys/socket.h>

 22 fb1a36c0 2022-01-09 op #include <sys/uio.h>

 23 fb1a36c0 2022-01-09 op #include <sys/un.h>

 24 fb1a36c0 2022-01-09 op 

 25 fb1a36c0 2022-01-09 op #include <netinet/in.h>

 26 fb1a36c0 2022-01-09 op #include <net/if.h>

 27 fb1a36c0 2022-01-09 op 

 28 fb1a36c0 2022-01-09 op #include <errno.h>

 29 fb1a36c0 2022-01-09 op #include <stdint.h>

 30 fb1a36c0 2022-01-09 op #include <stdlib.h>

 31 fb1a36c0 2022-01-09 op #include <string.h>

 32 fb1a36c0 2022-01-09 op #include <unistd.h>

 33 fb1a36c0 2022-01-09 op 

 34 fb1a36c0 2022-01-09 op #include "control.h"

 35 fb1a36c0 2022-01-09 op #include "kamid.h"

 36 fb1a36c0 2022-01-09 op #include "listener.h"

 37 fb1a36c0 2022-01-09 op #include "log.h"

 38 fb1a36c0 2022-01-09 op #include "utils.h"

 39 fb1a36c0 2022-01-09 op 

 40 fb1a36c0 2022-01-09 op #define	CONTROL_BACKLOG	5

 41 fb1a36c0 2022-01-09 op 

 42 fb1a36c0 2022-01-09 op struct {

 43 fb1a36c0 2022-01-09 op 	struct event	ev;

 44 fb1a36c0 2022-01-09 op 	struct event	evt;

 45 fb1a36c0 2022-01-09 op 	int		fd;

 46 fb1a36c0 2022-01-09 op } control_state = {.fd = -1};

 47 fb1a36c0 2022-01-09 op 

 48 fb1a36c0 2022-01-09 op struct ctl_conn {

 49 fb1a36c0 2022-01-09 op 	TAILQ_ENTRY(ctl_conn)	entry;

 50 fb1a36c0 2022-01-09 op 	struct imsgev		iev;

 51 fb1a36c0 2022-01-09 op };

 52 fb1a36c0 2022-01-09 op 

 53 fb1a36c0 2022-01-09 op TAILQ_HEAD(ctl_conns, ctl_conn)	ctl_conns = TAILQ_HEAD_INITIALIZER(ctl_conns);

 54 fb1a36c0 2022-01-09 op 

 55 fb1a36c0 2022-01-09 op struct ctl_conn	*control_connbyfd(int);

 56 fb1a36c0 2022-01-09 op struct ctl_conn	*control_connbypid(pid_t);

 57 fb1a36c0 2022-01-09 op void		 control_close(int);

 58 fb1a36c0 2022-01-09 op 

 59 fb1a36c0 2022-01-09 op int

 60 fb1a36c0 2022-01-09 op control_init(const char *path)

 61 fb1a36c0 2022-01-09 op {

 62 fb1a36c0 2022-01-09 op 	struct sockaddr_un	 sun;

 63 fb1a36c0 2022-01-09 op 	int			 fd;

 64 fb1a36c0 2022-01-09 op 	mode_t			 old_umask;

 65 fb1a36c0 2022-01-09 op 

 66 fb1a36c0 2022-01-09 op 	if ((fd = socket(AF_UNIX, SOCK_STREAM | SOCK_CLOEXEC | SOCK_NONBLOCK,

 67 fb1a36c0 2022-01-09 op 	    0)) == -1) {

 68 fb1a36c0 2022-01-09 op 		log_warn("%s: socket", __func__);

 69 fb1a36c0 2022-01-09 op 		return (-1);

 70 fb1a36c0 2022-01-09 op 	}

 71 fb1a36c0 2022-01-09 op 

 72 fb1a36c0 2022-01-09 op 	memset(&sun, 0, sizeof(sun));

 73 fb1a36c0 2022-01-09 op 	sun.sun_family = AF_UNIX;

 74 fb1a36c0 2022-01-09 op 	strlcpy(sun.sun_path, path, sizeof(sun.sun_path));

 75 fb1a36c0 2022-01-09 op 

 76 fb1a36c0 2022-01-09 op 	if (unlink(path) == -1)

 77 fb1a36c0 2022-01-09 op 		if (errno != ENOENT) {

 78 fb1a36c0 2022-01-09 op 			log_warn("%s: unlink %s", __func__, path);

 79 fb1a36c0 2022-01-09 op 			close(fd);

 80 fb1a36c0 2022-01-09 op 			return (-1);

 81 fb1a36c0 2022-01-09 op 		}

 82 fb1a36c0 2022-01-09 op 

 83 fb1a36c0 2022-01-09 op 	old_umask = umask(S_IXUSR|S_IXGRP|S_IWOTH|S_IROTH|S_IXOTH);

 84 fb1a36c0 2022-01-09 op 	if (bind(fd, (struct sockaddr *)&sun, sizeof(sun)) == -1) {

 85 fb1a36c0 2022-01-09 op 		log_warn("%s: bind: %s", __func__, path);

 86 fb1a36c0 2022-01-09 op 		close(fd);

 87 fb1a36c0 2022-01-09 op 		umask(old_umask);

 88 fb1a36c0 2022-01-09 op 		return (-1);

 89 fb1a36c0 2022-01-09 op 	}

 90 fb1a36c0 2022-01-09 op 	umask(old_umask);

 91 fb1a36c0 2022-01-09 op 

 92 fb1a36c0 2022-01-09 op 	if (chmod(path, S_IRUSR|S_IWUSR|S_IRGRP|S_IWGRP) == -1) {

 93 fb1a36c0 2022-01-09 op 		log_warn("%s: chmod", __func__);

 94 fb1a36c0 2022-01-09 op 		close(fd);

 95 fb1a36c0 2022-01-09 op 		(void)unlink(path);

 96 fb1a36c0 2022-01-09 op 		return (-1);

 97 fb1a36c0 2022-01-09 op 	}

 98 fb1a36c0 2022-01-09 op 

 99 fb1a36c0 2022-01-09 op 	return (fd);

100 fb1a36c0 2022-01-09 op }

101 fb1a36c0 2022-01-09 op 

102 fb1a36c0 2022-01-09 op int

103 fb1a36c0 2022-01-09 op control_listen(int fd)

104 fb1a36c0 2022-01-09 op {

105 fb1a36c0 2022-01-09 op 	if (control_state.fd != -1)

106 fb1a36c0 2022-01-09 op 		fatalx("%s: received unexpected controlsock", __func__);

107 fb1a36c0 2022-01-09 op 

108 fb1a36c0 2022-01-09 op 	control_state.fd = fd;

109 fb1a36c0 2022-01-09 op 	if (listen(control_state.fd, CONTROL_BACKLOG) == -1) {

110 fb1a36c0 2022-01-09 op 		log_warn("%s: listen", __func__);

111 fb1a36c0 2022-01-09 op 		return (-1);

112 fb1a36c0 2022-01-09 op 	}

113 fb1a36c0 2022-01-09 op 

114 fb1a36c0 2022-01-09 op 	event_set(&control_state.ev, control_state.fd, EV_READ,

115 fb1a36c0 2022-01-09 op 	    control_accept, NULL);

116 fb1a36c0 2022-01-09 op 	event_add(&control_state.ev, NULL);

117 fb1a36c0 2022-01-09 op 	evtimer_set(&control_state.evt, control_accept, NULL);

118 fb1a36c0 2022-01-09 op 

119 fb1a36c0 2022-01-09 op 	return (0);

120 fb1a36c0 2022-01-09 op }

121 fb1a36c0 2022-01-09 op 

122 fb1a36c0 2022-01-09 op void

123 fb1a36c0 2022-01-09 op control_accept(int listenfd, short event, void *bula)

124 fb1a36c0 2022-01-09 op {

125 fb1a36c0 2022-01-09 op 	int			 connfd;

126 fb1a36c0 2022-01-09 op 	socklen_t		 len;

127 fb1a36c0 2022-01-09 op 	struct sockaddr_un	 sun;

128 fb1a36c0 2022-01-09 op 	struct ctl_conn		*c;

129 fb1a36c0 2022-01-09 op 

130 fb1a36c0 2022-01-09 op 	event_add(&control_state.ev, NULL);

131 fb1a36c0 2022-01-09 op 	if ((event & EV_TIMEOUT))

132 fb1a36c0 2022-01-09 op 		return;

133 fb1a36c0 2022-01-09 op 

134 fb1a36c0 2022-01-09 op 	len = sizeof(sun);

135 fb1a36c0 2022-01-09 op 	if ((connfd = accept4(listenfd, (struct sockaddr *)&sun, &len,

136 fb1a36c0 2022-01-09 op 	    SOCK_CLOEXEC | SOCK_NONBLOCK)) == -1) {

137 fb1a36c0 2022-01-09 op 		/*

138 fb1a36c0 2022-01-09 op 		 * Pause accept if we are out of file descriptors, or

139 fb1a36c0 2022-01-09 op 		 * libevent will haunt us here too.

140 fb1a36c0 2022-01-09 op 		 */

141 fb1a36c0 2022-01-09 op 		if (errno == ENFILE || errno == EMFILE) {

142 fb1a36c0 2022-01-09 op 			struct timeval evtpause = { 1, 0 };

143 fb1a36c0 2022-01-09 op 

144 fb1a36c0 2022-01-09 op 			event_del(&control_state.ev);

145 fb1a36c0 2022-01-09 op 			evtimer_add(&control_state.evt, &evtpause);

146 fb1a36c0 2022-01-09 op 		} else if (errno != EWOULDBLOCK && errno != EINTR &&

147 fb1a36c0 2022-01-09 op 		    errno != ECONNABORTED)

148 fb1a36c0 2022-01-09 op 			log_warn("%s: accept4", __func__);

149 fb1a36c0 2022-01-09 op 		return;

150 fb1a36c0 2022-01-09 op 	}

151 fb1a36c0 2022-01-09 op 

152 fb1a36c0 2022-01-09 op 	if ((c = calloc(1, sizeof(struct ctl_conn))) == NULL) {

153 fb1a36c0 2022-01-09 op 		log_warn("%s: calloc", __func__);

154 fb1a36c0 2022-01-09 op 		close(connfd);

155 fb1a36c0 2022-01-09 op 		return;

156 fb1a36c0 2022-01-09 op 	}

157 fb1a36c0 2022-01-09 op 

158 fb1a36c0 2022-01-09 op 	imsg_init(&c->iev.ibuf, connfd);

159 fb1a36c0 2022-01-09 op 	c->iev.handler = control_dispatch_imsg;

160 fb1a36c0 2022-01-09 op 	c->iev.events = EV_READ;

161 fb1a36c0 2022-01-09 op 	event_set(&c->iev.ev, c->iev.ibuf.fd, c->iev.events,

162 fb1a36c0 2022-01-09 op 	    c->iev.handler, &c->iev);

163 fb1a36c0 2022-01-09 op 	event_add(&c->iev.ev, NULL);

164 fb1a36c0 2022-01-09 op 

165 fb1a36c0 2022-01-09 op 	TAILQ_INSERT_TAIL(&ctl_conns, c, entry);

166 fb1a36c0 2022-01-09 op }

167 fb1a36c0 2022-01-09 op 

168 fb1a36c0 2022-01-09 op struct ctl_conn *

169 fb1a36c0 2022-01-09 op control_connbyfd(int fd)

170 fb1a36c0 2022-01-09 op {

171 fb1a36c0 2022-01-09 op 	struct ctl_conn	*c;

172 fb1a36c0 2022-01-09 op 

173 fb1a36c0 2022-01-09 op 	TAILQ_FOREACH(c, &ctl_conns, entry) {

174 fb1a36c0 2022-01-09 op 		if (c->iev.ibuf.fd == fd)

175 fb1a36c0 2022-01-09 op 			break;

176 fb1a36c0 2022-01-09 op 	}

177 fb1a36c0 2022-01-09 op 

178 fb1a36c0 2022-01-09 op 	return (c);

179 fb1a36c0 2022-01-09 op }

180 fb1a36c0 2022-01-09 op 

181 fb1a36c0 2022-01-09 op struct ctl_conn *

182 fb1a36c0 2022-01-09 op control_connbypid(pid_t pid)

183 fb1a36c0 2022-01-09 op {

184 fb1a36c0 2022-01-09 op 	struct ctl_conn	*c;

185 fb1a36c0 2022-01-09 op 

186 fb1a36c0 2022-01-09 op 	TAILQ_FOREACH(c, &ctl_conns, entry) {

187 fb1a36c0 2022-01-09 op 		if (c->iev.ibuf.pid == pid)

188 fb1a36c0 2022-01-09 op 			break;

189 fb1a36c0 2022-01-09 op 	}

190 fb1a36c0 2022-01-09 op 

191 fb1a36c0 2022-01-09 op 	return (c);

192 fb1a36c0 2022-01-09 op }

193 fb1a36c0 2022-01-09 op 

194 fb1a36c0 2022-01-09 op void

195 fb1a36c0 2022-01-09 op control_close(int fd)

196 fb1a36c0 2022-01-09 op {

197 fb1a36c0 2022-01-09 op 	struct ctl_conn	*c;

198 fb1a36c0 2022-01-09 op 

199 fb1a36c0 2022-01-09 op 	if ((c = control_connbyfd(fd)) == NULL) {

200 fb1a36c0 2022-01-09 op 		log_warnx("%s: fd %d: not found", __func__, fd);

201 fb1a36c0 2022-01-09 op 		return;

202 fb1a36c0 2022-01-09 op 	}

203 fb1a36c0 2022-01-09 op 

204 fb1a36c0 2022-01-09 op 	msgbuf_clear(&c->iev.ibuf.w);

205 fb1a36c0 2022-01-09 op 	TAILQ_REMOVE(&ctl_conns, c, entry);

206 fb1a36c0 2022-01-09 op 

207 fb1a36c0 2022-01-09 op 	event_del(&c->iev.ev);

208 fb1a36c0 2022-01-09 op 	close(c->iev.ibuf.fd);

209 fb1a36c0 2022-01-09 op 

210 fb1a36c0 2022-01-09 op 	/* Some file descriptors are available again. */

211 fb1a36c0 2022-01-09 op 	if (evtimer_pending(&control_state.evt, NULL)) {

212 fb1a36c0 2022-01-09 op 		evtimer_del(&control_state.evt);

213 fb1a36c0 2022-01-09 op 		event_add(&control_state.ev, NULL);

214 fb1a36c0 2022-01-09 op 	}

215 fb1a36c0 2022-01-09 op 

216 fb1a36c0 2022-01-09 op 	free(c);

217 fb1a36c0 2022-01-09 op }

218 fb1a36c0 2022-01-09 op 

219 fb1a36c0 2022-01-09 op void

220 fb1a36c0 2022-01-09 op control_dispatch_imsg(int fd, short event, void *bula)

221 fb1a36c0 2022-01-09 op {

222 fb1a36c0 2022-01-09 op 	struct ctl_conn	*c;

223 fb1a36c0 2022-01-09 op 	struct imsg	 imsg;

224 fb1a36c0 2022-01-09 op 	ssize_t		 n;

225 fb1a36c0 2022-01-09 op 	int		 verbose;

226 fb1a36c0 2022-01-09 op 

227 fb1a36c0 2022-01-09 op 	if ((c = control_connbyfd(fd)) == NULL) {

228 fb1a36c0 2022-01-09 op 		log_warnx("%s: fd %d: not found", __func__, fd);

229 fb1a36c0 2022-01-09 op 		return;

230 fb1a36c0 2022-01-09 op 	}

231 fb1a36c0 2022-01-09 op 

232 fb1a36c0 2022-01-09 op 	if (event & EV_READ) {

233 fb1a36c0 2022-01-09 op 		if (((n = imsg_read(&c->iev.ibuf)) == -1 && errno != EAGAIN) ||

234 fb1a36c0 2022-01-09 op 		    n == 0) {

235 fb1a36c0 2022-01-09 op 			control_close(fd);

236 fb1a36c0 2022-01-09 op 			return;

237 fb1a36c0 2022-01-09 op 		}

238 fb1a36c0 2022-01-09 op 	}

239 fb1a36c0 2022-01-09 op 	if (event & EV_WRITE) {

240 fb1a36c0 2022-01-09 op 		if (msgbuf_write(&c->iev.ibuf.w) <= 0 && errno != EAGAIN) {

241 fb1a36c0 2022-01-09 op 			control_close(fd);

242 fb1a36c0 2022-01-09 op 			return;

243 fb1a36c0 2022-01-09 op 		}

244 fb1a36c0 2022-01-09 op 	}

245 fb1a36c0 2022-01-09 op 

246 fb1a36c0 2022-01-09 op 	for (;;) {

247 fb1a36c0 2022-01-09 op 		if ((n = imsg_get(&c->iev.ibuf, &imsg)) == -1) {

248 fb1a36c0 2022-01-09 op 			control_close(fd);

249 fb1a36c0 2022-01-09 op 			return;

250 fb1a36c0 2022-01-09 op 		}

251 fb1a36c0 2022-01-09 op 		if (n == 0)

252 fb1a36c0 2022-01-09 op 			break;

253 fb1a36c0 2022-01-09 op 

254 fb1a36c0 2022-01-09 op 		switch (imsg.hdr.type) {

255 fb1a36c0 2022-01-09 op 		case IMSG_CTL_RELOAD:

256 fb1a36c0 2022-01-09 op 			listener_imsg_compose_main(imsg.hdr.type, 0, NULL, 0);

257 fb1a36c0 2022-01-09 op 			break;

258 fb1a36c0 2022-01-09 op 		case IMSG_CTL_LOG_VERBOSE:

259 fb1a36c0 2022-01-09 op 			if (IMSG_DATA_SIZE(imsg) != sizeof(verbose))

260 fb1a36c0 2022-01-09 op 				break;

261 fb1a36c0 2022-01-09 op 

262 fb1a36c0 2022-01-09 op 			/* Forward to all other processes. */

263 fb1a36c0 2022-01-09 op 			listener_imsg_compose_main(imsg.hdr.type, imsg.hdr.pid,

264 fb1a36c0 2022-01-09 op 			    imsg.data, IMSG_DATA_SIZE(imsg));

265 fb1a36c0 2022-01-09 op 

266 fb1a36c0 2022-01-09 op 			/* XXX: send to every client? */

267 fb1a36c0 2022-01-09 op 

268 fb1a36c0 2022-01-09 op 			memcpy(&verbose, imsg.data, sizeof(verbose));

269 fb1a36c0 2022-01-09 op 			log_setverbose(verbose);

270 fb1a36c0 2022-01-09 op 			break;

271 fb1a36c0 2022-01-09 op 		default:

272 fb1a36c0 2022-01-09 op 			log_debug("%s: error handling imsg %d", __func__,

273 fb1a36c0 2022-01-09 op 			    imsg.hdr.type);

274 fb1a36c0 2022-01-09 op 			break;

275 fb1a36c0 2022-01-09 op 		}

276 fb1a36c0 2022-01-09 op 		imsg_free(&imsg);

277 fb1a36c0 2022-01-09 op 	}

278 fb1a36c0 2022-01-09 op 

279 fb1a36c0 2022-01-09 op 	imsg_event_add(&c->iev);

280 fb1a36c0 2022-01-09 op }

281 fb1a36c0 2022-01-09 op 

282 fb1a36c0 2022-01-09 op int

283 fb1a36c0 2022-01-09 op control_imsg_relay(struct imsg *imsg)

284 fb1a36c0 2022-01-09 op {

285 fb1a36c0 2022-01-09 op 	struct ctl_conn	*c;

286 fb1a36c0 2022-01-09 op 

287 fb1a36c0 2022-01-09 op 	if ((c = control_connbypid(imsg->hdr.pid)) == NULL)

288 fb1a36c0 2022-01-09 op 		return (0);

289 fb1a36c0 2022-01-09 op 

290 fb1a36c0 2022-01-09 op 	return (imsg_compose_event(&c->iev, imsg->hdr.type, 0, imsg->hdr.pid,

291 fb1a36c0 2022-01-09 op 	    -1, imsg->data, IMSG_DATA_SIZE(*imsg)));

292 fb1a36c0 2022-01-09 op }