Head

GCC Code Coverage Report

Directory:	./		Exec	Total	Coverage
File:	usr.sbin/ldapd/imsgev.c	Lines:	10	60	16.7 %
Date:	2017-11-13	Branches:	0	34	0.0 %


/*	$OpenBSD: imsgev.c,v 1.6 2017/03/01 00:53:39 gsoares Exp $ */

/*
 * Copyright (c) 2009 Eric Faurot <eric@openbsd.org>
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include <sys/queue.h>
#include <sys/socket.h>
#include <sys/uio.h>

#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include "imsgev.h"

void imsgev_add(struct imsgev *);
void imsgev_dispatch(int, short, void *);
void imsgev_disconnect(struct imsgev *, int);

void
imsgev_init(struct imsgev *iev, int fd, void *data,
    void (*callback)(struct imsgev *, int, struct imsg *),
    void (*needfd)(struct imsgev *))
{
	imsg_init(&iev->ibuf, fd);
	iev->terminate = 0;

	iev->data = data;
	iev->handler = imsgev_dispatch;
	iev->callback = callback;
	iev->needfd = needfd;

	iev->events = EV_READ;
	event_set(&iev->ev, iev->ibuf.fd, iev->events, iev->handler, iev);
	event_add(&iev->ev, NULL);
}

int
imsgev_compose(struct imsgev *iev, u_int16_t type, u_int32_t peerid,
    uint32_t pid, int fd, void *data, u_int16_t datalen)
{
	int	r;

	r = imsg_compose(&iev->ibuf, type, peerid, pid, fd, data, datalen);
	if (r != -1)
		imsgev_add(iev);

	return (r);
}

void
imsgev_close(struct imsgev *iev)
{
	iev->terminate = 1;
	imsgev_add(iev);
}

void
imsgev_clear(struct imsgev *iev)
{
	event_del(&iev->ev);
	msgbuf_clear(&iev->ibuf.w);
	close(iev->ibuf.fd);
}

void
imsgev_add(struct imsgev *iev)
{
	short	events = 0;

	if (!iev->terminate)
		events = EV_READ;
	if (iev->ibuf.w.queued || iev->terminate)
		events |= EV_WRITE;

	/* optimization: skip event_{del/set/add} if already set */
	if (events == iev->events)
		return;

	iev->events = events;
	event_del(&iev->ev);
	event_set(&iev->ev, iev->ibuf.fd, iev->events, iev->handler, iev);
	event_add(&iev->ev, NULL);
}

void
imsgev_dispatch(int fd, short ev, void *humppa)
{
	struct imsgev	*iev = humppa;
	struct imsgbuf	*ibuf = &iev->ibuf;
	struct imsg	 imsg;
	ssize_t		 n;

	iev->events = 0;

	if (ev & EV_READ) {
		if ((n = imsg_read(ibuf)) == -1) {
			/* if we don't have enough fds, free one up and retry */
			if (errno == EAGAIN) {
				iev->needfd(iev);
				n = imsg_read(ibuf);
			}

			if (n == -1) {
				imsgev_disconnect(iev, IMSGEV_EREAD);
				return;
			}
		}
		if (n == 0) {
			/*
			 * Connection is closed for reading, and we assume
			 * it is also closed for writing, so we error out
			 * if write data is pending.
			 */
			imsgev_disconnect(iev,
			    (iev->ibuf.w.queued) ? IMSGEV_EWRITE : IMSGEV_DONE);
			return;
		}
	}

	if (ev & EV_WRITE) {
		/*
		 * We wanted to write data out but the connection is either
		 * closed, or some error occured. Both case are not recoverable
		 * from the imsg perspective, so we treat it as a WRITE error.
		 */
		if ((n = msgbuf_write(&ibuf->w)) <= 0 && errno != EAGAIN) {
			imsgev_disconnect(iev, IMSGEV_EWRITE);
			return;
		}
	}

	while (iev->terminate == 0) {
		if ((n = imsg_get(ibuf, &imsg)) == -1) {
			imsgev_disconnect(iev, IMSGEV_EIMSG);
			return;
		}
		if (n == 0)
			break;
		iev->callback(iev, IMSGEV_IMSG, &imsg);
		imsg_free(&imsg);
	}

	if (iev->terminate && iev->ibuf.w.queued == 0) {
		imsgev_disconnect(iev, IMSGEV_DONE);
		return;
	}

	imsgev_add(iev);
}

void
imsgev_disconnect(struct imsgev *iev, int code)
{
	iev->callback(iev, code, NULL);
}


Generated by: GCOVR (Version 3.3)

Line	Branch	Exec	Source
1			/* $OpenBSD: imsgev.c,v 1.6 2017/03/01 00:53:39 gsoares Exp $ */
2
3			/*
4			* Copyright (c) 2009 Eric Faurot <eric@openbsd.org>
5			*
6			* Permission to use, copy, modify, and distribute this software for any
7			* purpose with or without fee is hereby granted, provided that the above
8			* copyright notice and this permission notice appear in all copies.
9			*
10			* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11			* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12			* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13			* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14			* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15			* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16			* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17			*/
18
19			#include <sys/queue.h>
20			#include <sys/socket.h>
21			#include <sys/uio.h>
22
23			#include <errno.h>
24			#include <stdio.h>
25			#include <stdlib.h>
26			#include <string.h>
27			#include <unistd.h>
28
29			#include "imsgev.h"
30
31			void imsgev_add(struct imsgev *);
32			void imsgev_dispatch(int, short, void *);
33			void imsgev_disconnect(struct imsgev *, int);
34
35			void
36			imsgev_init(struct imsgev iev, int fd, void data,
37			void (callback)(struct imsgev , int, struct imsg *),
38			void (needfd)(struct imsgev ))
39			{
40		32	imsg_init(&iev->ibuf, fd);
41		16	iev->terminate = 0;
42
43		16	iev->data = data;
44		16	iev->handler = imsgev_dispatch;
45		16	iev->callback = callback;
46		16	iev->needfd = needfd;
47
48		16	iev->events = EV_READ;
49		16	event_set(&iev->ev, iev->ibuf.fd, iev->events, iev->handler, iev);
50		16	event_add(&iev->ev, NULL);
51		16	}
52
53			int
54			imsgev_compose(struct imsgev *iev, u_int16_t type, u_int32_t peerid,
55			uint32_t pid, int fd, void *data, u_int16_t datalen)
56			{
57			int r;
58
59			r = imsg_compose(&iev->ibuf, type, peerid, pid, fd, data, datalen);
60			if (r != -1)
61			imsgev_add(iev);
62
63			return (r);
64			}
65
66			void
67			imsgev_close(struct imsgev *iev)
68			{
69			iev->terminate = 1;
70			imsgev_add(iev);
71			}
72
73			void
74			imsgev_clear(struct imsgev *iev)
75			{
76			event_del(&iev->ev);
77			msgbuf_clear(&iev->ibuf.w);
78			close(iev->ibuf.fd);
79			}
80
81			void
82			imsgev_add(struct imsgev *iev)
83			{
84			short events = 0;
85
86			if (!iev->terminate)
87			events = EV_READ;
88			if (iev->ibuf.w.queued \|\| iev->terminate)
89			events \|= EV_WRITE;
90
91			/* optimization: skip event_{del/set/add} if already set */
92			if (events == iev->events)
93			return;
94
95			iev->events = events;
96			event_del(&iev->ev);
97			event_set(&iev->ev, iev->ibuf.fd, iev->events, iev->handler, iev);
98			event_add(&iev->ev, NULL);
99			}
100
101			void
102			imsgev_dispatch(int fd, short ev, void *humppa)
103			{
104			struct imsgev *iev = humppa;
105			struct imsgbuf *ibuf = &iev->ibuf;
106			struct imsg imsg;
107			ssize_t n;
108
109			iev->events = 0;
110
111			if (ev & EV_READ) {
112			if ((n = imsg_read(ibuf)) == -1) {
113			/* if we don't have enough fds, free one up and retry */
114			if (errno == EAGAIN) {
115			iev->needfd(iev);
116			n = imsg_read(ibuf);
117			}
118
119			if (n == -1) {
120			imsgev_disconnect(iev, IMSGEV_EREAD);
121			return;
122			}
123			}
124			if (n == 0) {
125			/*
126			* Connection is closed for reading, and we assume
127			* it is also closed for writing, so we error out
128			* if write data is pending.
129			*/
130			imsgev_disconnect(iev,
131			(iev->ibuf.w.queued) ? IMSGEV_EWRITE : IMSGEV_DONE);
132			return;
133			}
134			}
135
136			if (ev & EV_WRITE) {
137			/*
138			* We wanted to write data out but the connection is either
139			* closed, or some error occured. Both case are not recoverable
140			* from the imsg perspective, so we treat it as a WRITE error.
141			*/
142			if ((n = msgbuf_write(&ibuf->w)) <= 0 && errno != EAGAIN) {
143			imsgev_disconnect(iev, IMSGEV_EWRITE);
144			return;
145			}
146			}
147
148			while (iev->terminate == 0) {
149			if ((n = imsg_get(ibuf, &imsg)) == -1) {
150			imsgev_disconnect(iev, IMSGEV_EIMSG);
151			return;
152			}
153			if (n == 0)
154			break;
155			iev->callback(iev, IMSGEV_IMSG, &imsg);
156			imsg_free(&imsg);
157			}
158
159			if (iev->terminate && iev->ibuf.w.queued == 0) {
160			imsgev_disconnect(iev, IMSGEV_DONE);
161			return;
162			}
163
164			imsgev_add(iev);
165			}
166
167			void
168			imsgev_disconnect(struct imsgev *iev, int code)
169			{
170			iev->callback(iev, code, NULL);
171			}