Annotation of embedaddon/dnsmasq/src/poll.c, revision 1.1.1.3

1.1.1.3 ! misho       1: /* dnsmasq is Copyright (c) 2000-2022 Simon Kelley
1.1       misho       2: 
                      3:    This program is free software; you can redistribute it and/or modify
                      4:    it under the terms of the GNU General Public License as published by
                      5:    the Free Software Foundation; version 2 dated June, 1991, or
                      6:    (at your option) version 3 dated 29 June, 2007.
                      7:  
                      8:    This program is distributed in the hope that it will be useful,
                      9:    but WITHOUT ANY WARRANTY; without even the implied warranty of
                     10:    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
                     11:    GNU General Public License for more details.
                     12:      
                     13:    You should have received a copy of the GNU General Public License
                     14:    along with this program.  If not, see <http://www.gnu.org/licenses/>.
                     15: */
                     16: 
                     17: #include "dnsmasq.h"
                     18: 
                     19: /* Wrapper for poll(). Allocates and extends array of struct pollfds,
                     20:    keeps them in fd order so that we can set and test conditions on
                     21:    fd using a simple but efficient binary chop. */
                     22: 
                     23: /* poll_reset()
                     24:    poll_listen(fd, event)
                     25:    .
                     26:    .
                     27:    poll_listen(fd, event);
                     28: 
                     29:    hits = do_poll(timeout);
                     30: 
                     31:    if (poll_check(fd, event)
                     32:     .
                     33:     .
                     34: 
                     35:    if (poll_check(fd, event)
                     36:     .
                     37:     .
                     38: 
                     39:     event is OR of POLLIN, POLLOUT, POLLERR, etc
                     40: */
                     41: 
                     42: static struct pollfd *pollfds = NULL;
                     43: static nfds_t nfds, arrsize = 0;
                     44: 
                     45: /* Binary search. Returns either the pollfd with fd, or
                     46:    if the fd doesn't match, or return equals nfds, the entry
                     47:    to the left of which a new record should be inserted. */
                     48: static nfds_t fd_search(int fd)
                     49: {
                     50:   nfds_t left, right, mid;
                     51:   
                     52:   if ((right = nfds) == 0)
                     53:     return 0;
                     54:   
                     55:   left = 0;
                     56:   
                     57:   while (1)
                     58:     {
                     59:       if (right == left + 1)
                     60:        return (pollfds[left].fd >= fd) ? left : right;
                     61:       
                     62:       mid = (left + right)/2;
                     63:       
                     64:       if (pollfds[mid].fd > fd)
                     65:        right = mid;
                     66:       else 
                     67:        left = mid;
                     68:     }
                     69: }
                     70: 
                     71: void poll_reset(void)
                     72: {
                     73:   nfds = 0;
                     74: }
                     75: 
                     76: int do_poll(int timeout)
                     77: {
                     78:   return poll(pollfds, nfds, timeout);
                     79: }
                     80: 
                     81: int poll_check(int fd, short event)
                     82: {
                     83:   nfds_t i = fd_search(fd);
                     84:   
                     85:   if (i < nfds && pollfds[i].fd == fd)
                     86:     return pollfds[i].revents & event;
                     87: 
                     88:   return 0;
                     89: }
                     90: 
                     91: void poll_listen(int fd, short event)
                     92: {
                     93:    nfds_t i = fd_search(fd);
                     94:   
                     95:    if (i < nfds && pollfds[i].fd == fd)
                     96:      pollfds[i].events |= event;
                     97:    else
                     98:      {
1.1.1.3 ! misho      99:        if (arrsize == nfds)
1.1       misho     100:         {
                    101:           /* Array too small, extend. */
                    102:           struct pollfd *new;
                    103: 
                    104:           arrsize = (arrsize == 0) ? 64 : arrsize * 2;
                    105: 
1.1.1.3 ! misho     106:           if (!(new = whine_realloc(pollfds, arrsize * sizeof(struct pollfd))))
1.1       misho     107:             return;
                    108: 
                    109:           pollfds = new;
                    110:         }
1.1.1.3 ! misho     111: 
        !           112:        memmove(&pollfds[i+1], &pollfds[i], (nfds - i) * sizeof(struct pollfd));
        !           113: 
1.1       misho     114:        pollfds[i].fd = fd;
                    115:        pollfds[i].events = event;
                    116:        nfds++;
                    117:      }
                    118: }

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>