1*4882a593Smuzhiyun // SPDX-License-Identifier: GPL-2.0-only
2*4882a593Smuzhiyun /*
3*4882a593Smuzhiyun * Copyright (C) 2014, Red Hat Inc, Arnaldo Carvalho de Melo <acme@redhat.com>
4*4882a593Smuzhiyun */
5*4882a593Smuzhiyun #include "array.h"
6*4882a593Smuzhiyun #include <errno.h>
7*4882a593Smuzhiyun #include <fcntl.h>
8*4882a593Smuzhiyun #include <poll.h>
9*4882a593Smuzhiyun #include <stdlib.h>
10*4882a593Smuzhiyun #include <unistd.h>
11*4882a593Smuzhiyun #include <string.h>
12*4882a593Smuzhiyun
fdarray__init(struct fdarray * fda,int nr_autogrow)13*4882a593Smuzhiyun void fdarray__init(struct fdarray *fda, int nr_autogrow)
14*4882a593Smuzhiyun {
15*4882a593Smuzhiyun fda->entries = NULL;
16*4882a593Smuzhiyun fda->priv = NULL;
17*4882a593Smuzhiyun fda->nr = fda->nr_alloc = 0;
18*4882a593Smuzhiyun fda->nr_autogrow = nr_autogrow;
19*4882a593Smuzhiyun }
20*4882a593Smuzhiyun
fdarray__grow(struct fdarray * fda,int nr)21*4882a593Smuzhiyun int fdarray__grow(struct fdarray *fda, int nr)
22*4882a593Smuzhiyun {
23*4882a593Smuzhiyun struct priv *priv;
24*4882a593Smuzhiyun int nr_alloc = fda->nr_alloc + nr;
25*4882a593Smuzhiyun size_t psize = sizeof(fda->priv[0]) * nr_alloc;
26*4882a593Smuzhiyun size_t size = sizeof(struct pollfd) * nr_alloc;
27*4882a593Smuzhiyun struct pollfd *entries = realloc(fda->entries, size);
28*4882a593Smuzhiyun
29*4882a593Smuzhiyun if (entries == NULL)
30*4882a593Smuzhiyun return -ENOMEM;
31*4882a593Smuzhiyun
32*4882a593Smuzhiyun priv = realloc(fda->priv, psize);
33*4882a593Smuzhiyun if (priv == NULL) {
34*4882a593Smuzhiyun free(entries);
35*4882a593Smuzhiyun return -ENOMEM;
36*4882a593Smuzhiyun }
37*4882a593Smuzhiyun
38*4882a593Smuzhiyun memset(&entries[fda->nr_alloc], 0, sizeof(struct pollfd) * nr);
39*4882a593Smuzhiyun memset(&priv[fda->nr_alloc], 0, sizeof(fda->priv[0]) * nr);
40*4882a593Smuzhiyun
41*4882a593Smuzhiyun fda->nr_alloc = nr_alloc;
42*4882a593Smuzhiyun fda->entries = entries;
43*4882a593Smuzhiyun fda->priv = priv;
44*4882a593Smuzhiyun return 0;
45*4882a593Smuzhiyun }
46*4882a593Smuzhiyun
fdarray__new(int nr_alloc,int nr_autogrow)47*4882a593Smuzhiyun struct fdarray *fdarray__new(int nr_alloc, int nr_autogrow)
48*4882a593Smuzhiyun {
49*4882a593Smuzhiyun struct fdarray *fda = calloc(1, sizeof(*fda));
50*4882a593Smuzhiyun
51*4882a593Smuzhiyun if (fda != NULL) {
52*4882a593Smuzhiyun if (fdarray__grow(fda, nr_alloc)) {
53*4882a593Smuzhiyun free(fda);
54*4882a593Smuzhiyun fda = NULL;
55*4882a593Smuzhiyun } else {
56*4882a593Smuzhiyun fda->nr_autogrow = nr_autogrow;
57*4882a593Smuzhiyun }
58*4882a593Smuzhiyun }
59*4882a593Smuzhiyun
60*4882a593Smuzhiyun return fda;
61*4882a593Smuzhiyun }
62*4882a593Smuzhiyun
fdarray__exit(struct fdarray * fda)63*4882a593Smuzhiyun void fdarray__exit(struct fdarray *fda)
64*4882a593Smuzhiyun {
65*4882a593Smuzhiyun free(fda->entries);
66*4882a593Smuzhiyun free(fda->priv);
67*4882a593Smuzhiyun fdarray__init(fda, 0);
68*4882a593Smuzhiyun }
69*4882a593Smuzhiyun
fdarray__delete(struct fdarray * fda)70*4882a593Smuzhiyun void fdarray__delete(struct fdarray *fda)
71*4882a593Smuzhiyun {
72*4882a593Smuzhiyun fdarray__exit(fda);
73*4882a593Smuzhiyun free(fda);
74*4882a593Smuzhiyun }
75*4882a593Smuzhiyun
fdarray__add(struct fdarray * fda,int fd,short revents,enum fdarray_flags flags)76*4882a593Smuzhiyun int fdarray__add(struct fdarray *fda, int fd, short revents, enum fdarray_flags flags)
77*4882a593Smuzhiyun {
78*4882a593Smuzhiyun int pos = fda->nr;
79*4882a593Smuzhiyun
80*4882a593Smuzhiyun if (fda->nr == fda->nr_alloc &&
81*4882a593Smuzhiyun fdarray__grow(fda, fda->nr_autogrow) < 0)
82*4882a593Smuzhiyun return -ENOMEM;
83*4882a593Smuzhiyun
84*4882a593Smuzhiyun fda->entries[fda->nr].fd = fd;
85*4882a593Smuzhiyun fda->entries[fda->nr].events = revents;
86*4882a593Smuzhiyun fda->priv[fda->nr].flags = flags;
87*4882a593Smuzhiyun fda->nr++;
88*4882a593Smuzhiyun return pos;
89*4882a593Smuzhiyun }
90*4882a593Smuzhiyun
fdarray__filter(struct fdarray * fda,short revents,void (* entry_destructor)(struct fdarray * fda,int fd,void * arg),void * arg)91*4882a593Smuzhiyun int fdarray__filter(struct fdarray *fda, short revents,
92*4882a593Smuzhiyun void (*entry_destructor)(struct fdarray *fda, int fd, void *arg),
93*4882a593Smuzhiyun void *arg)
94*4882a593Smuzhiyun {
95*4882a593Smuzhiyun int fd, nr = 0;
96*4882a593Smuzhiyun
97*4882a593Smuzhiyun if (fda->nr == 0)
98*4882a593Smuzhiyun return 0;
99*4882a593Smuzhiyun
100*4882a593Smuzhiyun for (fd = 0; fd < fda->nr; ++fd) {
101*4882a593Smuzhiyun if (!fda->entries[fd].events)
102*4882a593Smuzhiyun continue;
103*4882a593Smuzhiyun
104*4882a593Smuzhiyun if (fda->entries[fd].revents & revents) {
105*4882a593Smuzhiyun if (entry_destructor)
106*4882a593Smuzhiyun entry_destructor(fda, fd, arg);
107*4882a593Smuzhiyun
108*4882a593Smuzhiyun fda->entries[fd].revents = fda->entries[fd].events = 0;
109*4882a593Smuzhiyun continue;
110*4882a593Smuzhiyun }
111*4882a593Smuzhiyun
112*4882a593Smuzhiyun if (!(fda->priv[fd].flags & fdarray_flag__nonfilterable))
113*4882a593Smuzhiyun ++nr;
114*4882a593Smuzhiyun }
115*4882a593Smuzhiyun
116*4882a593Smuzhiyun return nr;
117*4882a593Smuzhiyun }
118*4882a593Smuzhiyun
fdarray__poll(struct fdarray * fda,int timeout)119*4882a593Smuzhiyun int fdarray__poll(struct fdarray *fda, int timeout)
120*4882a593Smuzhiyun {
121*4882a593Smuzhiyun return poll(fda->entries, fda->nr, timeout);
122*4882a593Smuzhiyun }
123*4882a593Smuzhiyun
fdarray__fprintf(struct fdarray * fda,FILE * fp)124*4882a593Smuzhiyun int fdarray__fprintf(struct fdarray *fda, FILE *fp)
125*4882a593Smuzhiyun {
126*4882a593Smuzhiyun int fd, printed = fprintf(fp, "%d [ ", fda->nr);
127*4882a593Smuzhiyun
128*4882a593Smuzhiyun for (fd = 0; fd < fda->nr; ++fd)
129*4882a593Smuzhiyun printed += fprintf(fp, "%s%d", fd ? ", " : "", fda->entries[fd].fd);
130*4882a593Smuzhiyun
131*4882a593Smuzhiyun return printed + fprintf(fp, " ]");
132*4882a593Smuzhiyun }
133