select 상한선을 돌파하는 방법 (윈도우즈 아래)

2999 단어
귀에 익은 방법:
1. 정의된 매크로 수정
#undef FD_SETSIZE
#define FD_SETSIZE 1024
2. boost asio에서 추출한 코드
class win_fd_set_adapter : noncopyable
{
public:
  enum { default_fd_set_size = 1024 };

  win_fd_set_adapter()
    : capacity_(default_fd_set_size),
      max_descriptor_(invalid_socket)
  {
    fd_set_ = static_cast<win_fd_set*>(::operator new(
          sizeof(win_fd_set) - sizeof(SOCKET)
          + sizeof(SOCKET) * (capacity_)));
    fd_set_->fd_count = 0;
  }

  ~win_fd_set_adapter()
  {
    ::operator delete(fd_set_);
  }

  void reset()
  {
    fd_set_->fd_count = 0;
    max_descriptor_ = invalid_socket;
  }

  bool set(socket_type descriptor)
  {
    for (u_int i = 0; i < fd_set_->fd_count; ++i)
      if (fd_set_->fd_array[i] == descriptor)
        return true;

    if (fd_set_->fd_count == capacity_)
    {
      u_int new_capacity = capacity_ + capacity_ / 2;
      win_fd_set* new_fd_set = static_cast<win_fd_set*>(::operator new(
            sizeof(win_fd_set) - sizeof(SOCKET)
            + sizeof(SOCKET) * (new_capacity)));

      new_fd_set->fd_count = fd_set_->fd_count;
      for (u_int i = 0; i < fd_set_->fd_count; ++i)
        new_fd_set->fd_array[i] = fd_set_->fd_array[i];

      ::operator delete(fd_set_);
      fd_set_ = new_fd_set;
      capacity_ = new_capacity;
    }

    fd_set_->fd_array[fd_set_->fd_count++] = descriptor;
    return true;
  }

  bool is_set(socket_type descriptor) const
  {
    return !!__WSAFDIsSet(descriptor,
        const_cast<fd_set*>(reinterpret_cast<const fd_set*>(fd_set_)));
  }

  operator fd_set*()
  {
    return reinterpret_cast<fd_set*>(fd_set_);
  }

  socket_type max_descriptor() const
  {
    return max_descriptor_;
  }

private:

  // This structure is defined to be compatible with the Windows API fd_set
  // structure, but without being dependent on the value of FD_SETSIZE. We use
  // the "struct hack" to allow the number of descriptors to be varied at
  // runtime.
  struct win_fd_set
  {
    u_int fd_count;
    SOCKET fd_array[1];
  };

  win_fd_set* fd_set_;
  u_int capacity_;
  socket_type max_descriptor_;
};

fd 클래스를 사용하여 대체set 컬렉션, FDSET는 FDISSET는 isset 대체, 안에 있는 몇 개의 데이터 형식은 스스로 typedef를 사용하고 부족한 헤더 파일은 스스로 추가합니다.noncopyable는boost에서 정의한 복사 금지 클래스입니다(복사 구조 함수,부수 작업은private 형식,구조 함수와 분석 함수는protected 형식으로 정의됨)
class noncopyable
  {
   protected:
      noncopyable() {}
      ~noncopyable() {}
   private:  // emphasize the following members are private
      noncopyable( const noncopyable& );
      const noncopyable& operator=( const noncopyable& );
  };

그리고 윈도우즈 아래의 FDSETSIZE는 최대 폴링 개수를 가리키며, linux의 FD 는SETSIZE는 최대 파일 설명자를 가리키며, 폴링의 설명자가 이 값보다 크면 문제가 발생할 수 있습니다. man:
Executing FD_CLR() or FD_SET() with a value of fd that is negative or is equal to or larger than FD_SETSIZE will result in undefined behavior.

좋은 웹페이지 즐겨찾기