Blinker

1) /* Blinker
2)    Copyright 2011-2014 Stefan Schuermans <stefan@blinkenarea.org>
3)    Copyleft GNU public license - http://www.gnu.org/copyleft/gpl.html
4)    a blinkenarea.org project */
5) 
6) #include <winsock2.h>
7) #include <set>
8) 
9) #include "Io.h"
10) #include "Time.h"
11) 
12) namespace Blinker {
13) 
14) /**
15)  * @brief wait for I/O events
16)  * @param[in] read I/O objects to check for readability
17)  * @param[out] read I/O objects that are readable
18)  * @param[in] write I/O objects to check for writability
19)  * @param[out] write I/O objects that are writable
20)  * @param[in] timeout maximum time to wait
21)  */
22) void Io::wait(Set &read, Set &write, const Time &timeout)
23) {

24)   static const Time maxTimeout(1); // maximum timeout for I/O wait

25) 

26)   // get read set and write set, also count entries
27)   unsigned int rds = 0;

28)   fd_set fd_rd;
29)   FD_ZERO(&fd_rd);
30)   for (Set::const_iterator it = read.begin(); it != read.end(); ++it) {
31)     if ((*it)->m_socket != INVALID_SOCKET) {
32)       FD_SET((*it)->m_socket, &fd_rd);

33)       ++rds;

34)     }
35)   }

36)   unsigned int wrs = 0;

37)   fd_set fd_wr;
38)   FD_ZERO(&fd_wr);
39)   for (Set::const_iterator it = write.begin(); it != write.end(); ++it) {
40)     if ((*it)->m_socket != INVALID_SOCKET) {
41)       FD_SET((*it)->m_socket, &fd_wr);

42)       ++wrs;

43)     }
44)   }
45) 

46)   /* special case for no sockets to check,
47)      because select does not work for allsets empty on Windows */
48)   if (rds == 0 && wrs == 0) {
49)     if (timeout <= Time::zero) {
50)       return;
51)     }
52)     Sleep(timeout.toMs());
53)   }
54) 

55)   // get timeout
56)   struct timeval to;

57)   if (timeout < Time::zero) { // don't use negative timeout

58)     Time::zero.toTimeval(to);

59)   } else if (timeout > maxTimeout) { // stay responsive -> no more than max
60)     maxTimeout.toTimeval(to);

61)   } else {

62)     timeout.toTimeval(to);

63)   }

64) 
65)   // wait for I/O event on sockets
66)   fd_set fd_err;
67)   FD_ZERO(&fd_err);

68)   int res = select(0 /* ignored on Windows */, &fd_rd, &fd_wr, &fd_err, &to);