Blame src/PollVec.h

Packit 8f70b4
/*
Packit 8f70b4
 * lftp - file transfer program
Packit 8f70b4
 *
Packit 8f70b4
 * Copyright (c) 1996-2012 by Alexander V. Lukyanov (lav@yars.free.net)
Packit 8f70b4
 *
Packit 8f70b4
 * This program is free software; you can redistribute it and/or modify
Packit 8f70b4
 * it under the terms of the GNU General Public License as published by
Packit 8f70b4
 * the Free Software Foundation; either version 3 of the License, or
Packit 8f70b4
 * (at your option) any later version.
Packit 8f70b4
 *
Packit 8f70b4
 * This program is distributed in the hope that it will be useful,
Packit 8f70b4
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
Packit 8f70b4
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
Packit 8f70b4
 * GNU General Public License for more details.
Packit 8f70b4
 *
Packit 8f70b4
 * You should have received a copy of the GNU General Public License
Packit 8f70b4
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
Packit 8f70b4
 */
Packit 8f70b4
Packit 8f70b4
#ifndef POLLVEC_H
Packit 8f70b4
#define POLLVEC_H
Packit 8f70b4
Packit 8f70b4
#include <sys/types.h>
Packit 8f70b4
#include <sys/time.h>
Packit 8f70b4
#include <unistd.h>
Packit 8f70b4
#include <sys/select.h>
Packit 8f70b4
CDECL_BEGIN
Packit 8f70b4
#include <poll.h>
Packit 8f70b4
CDECL_END
Packit 8f70b4
Packit 8f70b4
class PollVec
Packit 8f70b4
{
Packit 8f70b4
   fd_set in;
Packit 8f70b4
   fd_set out;
Packit 8f70b4
   fd_set in_polled;
Packit 8f70b4
   fd_set out_polled;
Packit 8f70b4
   fd_set in_ready;
Packit 8f70b4
   fd_set out_ready;
Packit 8f70b4
   int nfds;
Packit 8f70b4
   struct timeval tv_timeout;
Packit 8f70b4
Packit 8f70b4
public:
Packit 8f70b4
   PollVec() {
Packit 8f70b4
      Empty();
Packit 8f70b4
      FD_ZERO(&in_polled);
Packit 8f70b4
      FD_ZERO(&out_polled);
Packit 8f70b4
      FD_ZERO(&in_ready);
Packit 8f70b4
      FD_ZERO(&out_ready);
Packit 8f70b4
   }
Packit 8f70b4
Packit 8f70b4
   void	 Empty()
Packit 8f70b4
      {
Packit 8f70b4
	 FD_ZERO(&in);
Packit 8f70b4
	 FD_ZERO(&out;;
Packit 8f70b4
	 nfds=0;
Packit 8f70b4
	 tv_timeout.tv_sec=-1;
Packit 8f70b4
	 tv_timeout.tv_usec=0;
Packit 8f70b4
      }
Packit 8f70b4
   void	 Block();
Packit 8f70b4
Packit 8f70b4
   enum {
Packit 8f70b4
      IN=1,
Packit 8f70b4
      OUT=4,
Packit 8f70b4
   };
Packit 8f70b4
Packit 8f70b4
   void SetTimeout(const timeval &t) { tv_timeout=t; }
Packit 8f70b4
   void SetTimeoutU(unsigned t) {
Packit 8f70b4
      tv_timeout.tv_sec=t/1000000;
Packit 8f70b4
      tv_timeout.tv_usec=t%1000000;
Packit 8f70b4
   }
Packit 8f70b4
   void AddTimeoutU(unsigned t);
Packit 8f70b4
   void AddFD(int fd,int events);
Packit 8f70b4
   bool FDReady(int fd,int events);
Packit 8f70b4
   void FDSetNotReady(int fd,int events);
Packit 8f70b4
   void NoWait() { tv_timeout.tv_sec=tv_timeout.tv_usec=0; }
Packit 8f70b4
   bool WillNotBlock() { return tv_timeout.tv_sec==0 && tv_timeout.tv_usec==0; }
Packit 8f70b4
};
Packit 8f70b4
Packit 8f70b4
#endif /* POLLVEC_H */