summaryrefslogtreecommitdiff
path: root/tools/testing/selftests/bpf/io_helpers.c
blob: 4ada0a74aa1fd10e9c85c786c5e353d2a5c9d6b9 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
// SPDX-License-Identifier: GPL-2.0
#include <sys/select.h>
#include <unistd.h>
#include <errno.h>

int read_with_timeout(int fd, char *buf, size_t count, long usec)
{
	const long M = 1000 * 1000;
	struct timeval tv = { usec / M, usec % M };
	fd_set fds;
	int err;

	FD_ZERO(&fds);
	FD_SET(fd, &fds);
	err = select(fd + 1, &fds, NULL, NULL, &tv);
	if (err < 0)
		return err;
	if (FD_ISSET(fd, &fds))
		return read(fd, buf, count);
	return -EAGAIN;
}