Major Update:
- Implemented global var DebugLevel - Update LogCore to check DebugLevel - Added many log messages and standadised all log messages - Further improved validation checks on all methods - Updated SelectCore, only remove SelectHandle from list during Test() - Close Handles in SelectableCore destructor Bug fixes: - Non-blocking Client Socket Connection now working correctly - Remove FD from Select lists at the correct time
This commit is contained in:
119
SelectCore.cpp
119
SelectCore.cpp
@@ -6,12 +6,20 @@
|
||||
*/
|
||||
|
||||
// redA Libraries
|
||||
#include "SelectableCore.h"
|
||||
#include "TimingCore.h"
|
||||
#include "LogCore.h"
|
||||
|
||||
// Standard C/C++ Libraries
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <errno.h>
|
||||
#include "SelectableCore.h"
|
||||
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
extern char ProcessName[];
|
||||
extern char LogStr[]; // Access to global temporary log messages string
|
||||
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
@@ -30,12 +38,27 @@ CSelect::CSelect( long SelectTimeout )
|
||||
|
||||
// Set Timeout
|
||||
SetInterval( &Timeout, SelectTimeout );
|
||||
|
||||
// Show status
|
||||
LogMessage( ProcessName, dlMedium, "Select - Created" );
|
||||
}
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
// Destroy Select
|
||||
CSelect::~CSelect()
|
||||
{
|
||||
TSelectHandle * NextHandle;
|
||||
|
||||
// Destroy handles
|
||||
while (FirstHandle)
|
||||
{
|
||||
NextHandle = FirstHandle->Next;
|
||||
free( FirstHandle );
|
||||
FirstHandle = NextHandle;
|
||||
}
|
||||
|
||||
// Show status
|
||||
LogMessage( ProcessName, dlMedium, "Select - Destroyed" );
|
||||
return;
|
||||
}
|
||||
//---------------------------------------------------------------------------
|
||||
@@ -75,15 +98,23 @@ void CSelect::Add( int FD, bool Read, bool Write, CSelectableCore * Function )
|
||||
}
|
||||
|
||||
// Add Read select
|
||||
if (Read) {
|
||||
if (Read && !(*Handle)->Read) {
|
||||
(*Handle)->Read = true;
|
||||
FD_SET( FD, &ReadTestFDS );
|
||||
|
||||
// Log event
|
||||
sprintf( LogStr, "FD [%d] - Add Read", FD );
|
||||
LogMessage( "Select", dlMedium, LogStr );
|
||||
}
|
||||
|
||||
// Add Write Select
|
||||
if (Write) {
|
||||
if (Write && !(*Handle)->Write) {
|
||||
(*Handle)->Write = true;
|
||||
FD_SET( FD, &WriteTestFDS );
|
||||
|
||||
// Log event
|
||||
sprintf( LogStr, "FD [%d] - Add Write", FD );
|
||||
LogMessage( "Select", dlMedium, LogStr );
|
||||
}
|
||||
|
||||
// Check Maximum File Handle
|
||||
@@ -95,8 +126,6 @@ void CSelect::Add( int FD, bool Read, bool Write, CSelectableCore * Function )
|
||||
void CSelect::Remove( int FD, bool Read, bool Write )
|
||||
{
|
||||
TSelectHandle ** Handle = NULL;
|
||||
TSelectHandle * NextHandle = NULL;
|
||||
int TestFD = 0;
|
||||
|
||||
// Check if Handle already exists
|
||||
Handle = &FirstHandle;
|
||||
@@ -108,42 +137,34 @@ void CSelect::Remove( int FD, bool Read, bool Write )
|
||||
return;
|
||||
|
||||
// Remove from set for select read check
|
||||
if (Read) {
|
||||
if (Read && (*Handle)->Read) {
|
||||
(*Handle)->Read = false;
|
||||
FD_CLR( FD, &ReadTestFDS);
|
||||
|
||||
// Log event
|
||||
sprintf( LogStr, "FD [%d] - Remove Read", FD );
|
||||
LogMessage( "Select", dlMedium, LogStr );
|
||||
}
|
||||
|
||||
// Remove from set for select write check
|
||||
if (Write) {
|
||||
if (Write && (*Handle)->Write) {
|
||||
(*Handle)->Write = false;
|
||||
FD_CLR( FD, &WriteTestFDS);
|
||||
}
|
||||
|
||||
// Check if to remove from list
|
||||
if (!(*Handle)->Read && !(*Handle)->Write)
|
||||
{
|
||||
// Remove from list
|
||||
NextHandle = (*Handle)->Next;
|
||||
free( *Handle );
|
||||
*Handle = NextHandle;
|
||||
|
||||
// Update Maximum Test FD
|
||||
if (FD == MaxFD-1) {
|
||||
for (TestFD = MaxFD-1; TestFD >= 0; TestFD--) {
|
||||
if (FD_ISSET( TestFD, &ReadTestFDS ) || FD_ISSET( TestFD, &WriteTestFDS )) {
|
||||
break;
|
||||
}
|
||||
}
|
||||
MaxFD = TestFD+1;
|
||||
}
|
||||
// Log event
|
||||
sprintf( LogStr, "FD [%d] - Remove Write", FD );
|
||||
LogMessage( "Select", dlMedium, LogStr );
|
||||
}
|
||||
// Handle will be removed in Test() if both Read & Write flags are false
|
||||
}
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
bool CSelect::Test()
|
||||
{
|
||||
TSelectHandle * Handle = NULL;
|
||||
int Events = 0;
|
||||
TSelectHandle * Handle = NULL;
|
||||
TSelectHandle ** HandlePtr = NULL;
|
||||
int TestFD = 0;
|
||||
int Events = 0;
|
||||
|
||||
// Set Test sets
|
||||
ReadFDS = ReadTestFDS;
|
||||
@@ -153,7 +174,7 @@ bool CSelect::Test()
|
||||
Events = select( MaxFD, &ReadFDS, &WriteFDS, (fd_set*)NULL, &Timeout );
|
||||
if (Events < 0)
|
||||
{
|
||||
printf( "Select operation failed (%s)\n", strerror(errno) );
|
||||
LogMessage( "Select", dlMedium, "Select operation failed" );
|
||||
return false;
|
||||
}
|
||||
|
||||
@@ -161,16 +182,42 @@ bool CSelect::Test()
|
||||
Handle = FirstHandle;
|
||||
while (Handle)
|
||||
{
|
||||
// Check read Event
|
||||
if (FD_ISSET( Handle->FD, &ReadFDS ) && Handle->Function) {
|
||||
Handle->Function->Read( Handle->FD );
|
||||
}
|
||||
// Check if to remove from list
|
||||
if (!Handle->Read && !Handle->Write)
|
||||
{
|
||||
// Update Maximum Test FD
|
||||
if (Handle->FD == MaxFD-1) {
|
||||
for (TestFD = MaxFD-1; TestFD >= 0; TestFD--) {
|
||||
if (FD_ISSET( TestFD, &ReadTestFDS ) || FD_ISSET( TestFD, &WriteTestFDS )) {
|
||||
break;
|
||||
}
|
||||
}
|
||||
MaxFD = TestFD+1;
|
||||
}
|
||||
|
||||
// Check Write Event
|
||||
if (FD_ISSET( Handle->FD, &WriteFDS ) && Handle->Function) {
|
||||
Handle->Function->Write( Handle->FD );
|
||||
}
|
||||
// Remove from list
|
||||
HandlePtr = &FirstHandle;
|
||||
while (*HandlePtr && (*HandlePtr != Handle))
|
||||
HandlePtr = &((*HandlePtr)->Next);
|
||||
*HandlePtr = (*HandlePtr)->Next;
|
||||
|
||||
// Destroy and go to next
|
||||
free( Handle );
|
||||
Handle = *HandlePtr;
|
||||
continue;
|
||||
}
|
||||
else
|
||||
{
|
||||
// Check read Event
|
||||
if (FD_ISSET( Handle->FD, &ReadFDS ) && Handle->Function) {
|
||||
Handle->Function->Read( Handle->FD );
|
||||
}
|
||||
|
||||
// Check Write Event
|
||||
if (FD_ISSET( Handle->FD, &WriteFDS ) && Handle->Function) {
|
||||
Handle->Function->Write( Handle->FD );
|
||||
}
|
||||
}
|
||||
// Next
|
||||
Handle = Handle->Next;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user