Guy Resheff
92df7a383a
Unmodified upstream source from https://github.com/Raizo62/vwifi Commit: 4a9842e "CMakeLists.txt : VERSION = 7.0" Simulator of WiFi (802.11) interfaces to communicate between several Virtual Machines via mac80211_hwsim + TCP relay. License: Apache-2.0
100 lines
1.8 KiB
C++
100 lines
1.8 KiB
C++
#include <errno.h> // errno
|
|
#include <assert.h> // assert
|
|
#include <cstddef> // NULL
|
|
|
|
#include "cselect.h"
|
|
|
|
using namespace std;
|
|
|
|
CSelect::CSelect()
|
|
{
|
|
Init();
|
|
}
|
|
|
|
void CSelect::Init()
|
|
{
|
|
//clear the socket set
|
|
FD_ZERO(&Master);
|
|
|
|
//clear the socket set
|
|
FD_ZERO(&Dup);
|
|
|
|
MaxDescriptor=-1;
|
|
}
|
|
|
|
void CSelect::UpdateMaxDescriptor(TDescriptor descriptor)
|
|
{
|
|
//highest file descriptor number, need it for the select function
|
|
if ( descriptor > MaxDescriptor )
|
|
MaxDescriptor=descriptor;
|
|
}
|
|
|
|
bool CSelect::AddNode(TDescriptor descriptor)
|
|
{
|
|
//add new socket to array of sockets
|
|
ListNodes.push_back(descriptor);
|
|
|
|
//add child sockets to set
|
|
FD_SET( descriptor , &Master);
|
|
|
|
//highest file descriptor number, need it for the select function
|
|
UpdateMaxDescriptor(descriptor);
|
|
|
|
return true;
|
|
}
|
|
|
|
void CSelect::DelNode(TDescriptor descriptor)
|
|
{
|
|
MaxDescriptor=-1;
|
|
for (auto node = ListNodes.begin() ; node != ListNodes.end(); ++node)
|
|
{
|
|
if( *node == descriptor )
|
|
{
|
|
auto node_to_delete = node;
|
|
|
|
FD_CLR(descriptor , &Master);
|
|
|
|
for (++node ; node != ListNodes.end(); ++node)
|
|
UpdateMaxDescriptor(*node);
|
|
|
|
ListNodes.erase(node_to_delete);
|
|
|
|
return;
|
|
}
|
|
UpdateMaxDescriptor(*node);
|
|
}
|
|
}
|
|
|
|
TDescriptor CSelect::Wait()
|
|
{
|
|
return Wait(NULL);
|
|
}
|
|
|
|
TDescriptor CSelect::Wait(const sigset_t *sigmask)
|
|
{
|
|
/* back up master */
|
|
Dup = Master;
|
|
|
|
//wait for an activity on one of the sockets , timeout is NULL ,
|
|
//so wait indefinitely
|
|
int activity=pselect( MaxDescriptor + 1 , &Dup , NULL , NULL , NULL, sigmask);
|
|
|
|
if ((activity < 0) && (errno!=EINTR))
|
|
return SCHEDULER_ERROR;
|
|
|
|
return activity;
|
|
}
|
|
|
|
|
|
bool CSelect::DescriptorHasAction(TDescriptor descriptor)
|
|
{
|
|
return FD_ISSET( descriptor , &Dup);
|
|
}
|
|
|
|
bool CSelect::NodeHasAction(TIndex index)
|
|
{
|
|
assert( index < ListNodes.size() );
|
|
|
|
return FD_ISSET( ListNodes[index] , &Dup);
|
|
}
|