forked from epics_driver_modules/motorBase
MarkRivers
287 lines
8.5 KiB
C++
Executable File
287 lines
8.5 KiB
C++
Executable File
/*///////////////////////////////////////////////////////////////////////////////
|
|
* Socket.cpp
|
|
*/
|
|
#include "Socket.h"
|
|
|
|
#define TIMEOUT 300
|
|
#define SMALL_BUFFER_SIZE 256
|
|
#define MAX_NB_SOCKETS 100
|
|
|
|
void Delay(double timedelay);
|
|
|
|
CAsyncSocket m_sConnectSocket[MAX_NB_SOCKETS];
|
|
BOOL UsedSocket[MAX_NB_SOCKETS] = { FALSE};
|
|
double TimeoutSocket[MAX_NB_SOCKETS];
|
|
int NbSockets = 0;
|
|
|
|
/***************************************************************************************/
|
|
int ConnectToServer(char *Ip_Address, int Ip_Port, double TimeOut)
|
|
{
|
|
int flag = 1;
|
|
int socketID = 0;
|
|
DWORD sockPendingFlag = 1;
|
|
|
|
if (!AfxSocketInit())
|
|
{
|
|
AfxMessageBox("Fatal Error: MFC Socket initialization failed");
|
|
return -1;
|
|
}
|
|
/* Select a socket number */
|
|
if (NbSockets < MAX_NB_SOCKETS)
|
|
{
|
|
while ((UsedSocket[socketID] == TRUE) && (socketID < MAX_NB_SOCKETS))
|
|
socketID++;
|
|
|
|
if (socketID == MAX_NB_SOCKETS)
|
|
return -1;
|
|
}
|
|
else
|
|
return -1;
|
|
UsedSocket[socketID] = TRUE;
|
|
NbSockets++;
|
|
|
|
/* Socket creation */
|
|
if ((m_sConnectSocket[socketID].Create() == 0)
|
|
|| (m_sConnectSocket[socketID].SetSockOpt(TCP_NODELAY,(char *)&flag,(int)sizeof( flag ),IPPROTO_TCP) == 0))
|
|
{
|
|
UsedSocket[socketID] = FALSE;
|
|
NbSockets--;
|
|
return -1;
|
|
}
|
|
|
|
/* Connect */
|
|
if (m_sConnectSocket[socketID].Connect(Ip_Address,Ip_Port) == 0)
|
|
{
|
|
int SocketError = m_sConnectSocket[socketID].GetLastError();
|
|
if (SocketError != WSAEWOULDBLOCK)
|
|
{
|
|
UsedSocket[socketID] = FALSE;
|
|
NbSockets--;
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
/* Set timeout */
|
|
if (TimeOut > 0)
|
|
{
|
|
if (TimeOut < 1e-3)
|
|
TimeoutSocket[socketID] = 1e-3;
|
|
else
|
|
TimeoutSocket[socketID] = TimeOut;
|
|
}
|
|
else
|
|
TimeoutSocket[socketID] = TIMEOUT;
|
|
|
|
/* Socket array */
|
|
struct fd_set *Sockets = new struct fd_set;
|
|
FD_ZERO(Sockets);
|
|
FD_SET(m_sConnectSocket[socketID], Sockets);
|
|
|
|
/* Time structure */
|
|
struct timeval *TimeOutStruct = new struct timeval;
|
|
TimeOutStruct->tv_sec = (long) TimeoutSocket[socketID];
|
|
TimeOutStruct->tv_usec = (long) ((TimeoutSocket[socketID] - (long)TimeoutSocket[socketID])* 1e9);
|
|
|
|
/* Checking connection is ok */
|
|
int SelectReturn = select(0, NULL, Sockets, NULL, TimeOutStruct);
|
|
if (SelectReturn == SOCKET_ERROR
|
|
|| SelectReturn == 0)
|
|
{
|
|
UsedSocket[socketID] = FALSE;
|
|
NbSockets--;
|
|
return -1;
|
|
}
|
|
|
|
/* Force no pending */
|
|
if (m_sConnectSocket[socketID].IOCtl(FIONBIO,&sockPendingFlag) == 0)
|
|
{
|
|
m_sConnectSocket[socketID].Close();
|
|
UsedSocket[socketID] = FALSE;
|
|
NbSockets--;
|
|
return -1;
|
|
}
|
|
|
|
/* Delay unless -1 return for the first API */
|
|
Sleep(10);
|
|
|
|
/* Return socket ID */
|
|
return socketID;
|
|
}
|
|
|
|
/***************************************************************************************/
|
|
void SetTCPTimeout(int SocketIndex, double TimeOut)
|
|
{
|
|
if ((SocketIndex >= 0) && (SocketIndex < MAX_NB_SOCKETS) && (UsedSocket[SocketIndex] == TRUE))
|
|
{
|
|
if (TimeOut > 0)
|
|
{
|
|
if (TimeOut < 1e-3)
|
|
TimeoutSocket[SocketIndex] = 1e-3;
|
|
else
|
|
TimeoutSocket[SocketIndex] = TimeOut;
|
|
}
|
|
}
|
|
}
|
|
|
|
/***************************************************************************************/
|
|
void SendAndReceive(int socketID, char sSendString[], char sReturnString[], int iReturnStringSize)
|
|
{
|
|
char sSocketBuffer[SMALL_BUFFER_SIZE + 1] = {'\0'};
|
|
int iReceiveByteNumber = 0;
|
|
int iErrorNo = 0;
|
|
fd_set readFds;
|
|
int iSelectStatus;
|
|
double dTimeout;
|
|
struct timeval cTimeout;
|
|
clock_t start, stop;
|
|
|
|
if ((socketID >= 0) && (socketID < MAX_NB_SOCKETS) && (UsedSocket[socketID] == TRUE))
|
|
{
|
|
/* Clear receive buffer */
|
|
do
|
|
{
|
|
iReceiveByteNumber = m_sConnectSocket[socketID].Receive(sSocketBuffer,SMALL_BUFFER_SIZE);
|
|
}
|
|
while (iReceiveByteNumber != SOCKET_ERROR);
|
|
sReturnString[0] = '\0';
|
|
|
|
/* Send String to controller and wait for response */
|
|
m_sConnectSocket[socketID].Send(sSendString,strlen(sSendString));
|
|
|
|
/* Get reply with timeout */
|
|
dTimeout = TimeoutSocket[socketID];
|
|
do
|
|
{
|
|
/* Get time */
|
|
start = clock();
|
|
|
|
/* Check reply */
|
|
iReceiveByteNumber = m_sConnectSocket[socketID].Receive(sSocketBuffer,SMALL_BUFFER_SIZE);
|
|
iErrorNo = GetLastError() & 0xffff;
|
|
|
|
/* Wait for reply */
|
|
if ((iReceiveByteNumber == SOCKET_ERROR) && (iErrorNo == WSAEWOULDBLOCK))
|
|
{
|
|
FD_ZERO(&readFds);
|
|
FD_SET(m_sConnectSocket[socketID].m_hSocket, &readFds);
|
|
cTimeout.tv_sec = (long)dTimeout;
|
|
cTimeout.tv_usec = (long)((dTimeout - (long)dTimeout) * 1e6);
|
|
iSelectStatus = select(FD_SETSIZE, (fd_set *)&readFds, (fd_set *) NULL, (fd_set *) NULL, &cTimeout);
|
|
if ((iSelectStatus > 0) && (FD_ISSET(m_sConnectSocket[socketID].m_hSocket, &readFds)))
|
|
iReceiveByteNumber = m_sConnectSocket[socketID].Receive(sSocketBuffer,SMALL_BUFFER_SIZE);
|
|
else
|
|
{
|
|
iErrorNo = GetLastError() & 0xffff;
|
|
sprintf(sSocketBuffer,"-2,%s,EndOfAPI",sSendString);
|
|
strncpyWithEOS(sReturnString, sSocketBuffer, strlen(sSocketBuffer), iReturnStringSize);
|
|
iReceiveByteNumber = SOCKET_ERROR;
|
|
}
|
|
}
|
|
|
|
/* Concatenation */
|
|
if ((iReceiveByteNumber >= 0) && (iReceiveByteNumber <= SMALL_BUFFER_SIZE))
|
|
{
|
|
sSocketBuffer[iReceiveByteNumber] = '\0';
|
|
strncat(sReturnString, sSocketBuffer, iReturnStringSize - strlen(sReturnString) - 1);
|
|
}
|
|
|
|
/* Calculate new timeout */
|
|
stop = clock();
|
|
dTimeout = dTimeout - (double)(stop - start) / CLOCKS_PER_SEC;
|
|
if (dTimeout < 1e-3)
|
|
dTimeout = 1e-3;
|
|
}
|
|
while ((iReceiveByteNumber != SOCKET_ERROR) && (strstr(sReturnString, "EndOfAPI") == NULL));
|
|
}
|
|
else
|
|
sReturnString[0] = '\0';
|
|
|
|
return;
|
|
}
|
|
|
|
/***************************************************************************************/
|
|
void CloseSocket(int socketID)
|
|
{
|
|
if ((socketID >= 0) && (socketID < MAX_NB_SOCKETS))
|
|
{
|
|
if (UsedSocket[socketID] == TRUE)
|
|
{
|
|
m_sConnectSocket[socketID].Close();
|
|
UsedSocket[socketID] = FALSE;
|
|
TimeoutSocket[socketID] = TIMEOUT;
|
|
NbSockets--;
|
|
}
|
|
}
|
|
}
|
|
|
|
/***************************************************************************************/
|
|
char * GetError(int socketID)
|
|
{
|
|
if ((socketID >= 0) && (socketID < MAX_NB_SOCKETS))
|
|
{
|
|
int error = m_sConnectSocket[socketID].GetLastError();
|
|
switch (error)
|
|
{
|
|
case WSANOTINITIALISED:
|
|
return (_T("A successful AfxSocketInit must occur before using this API."));
|
|
case WSAENETDOWN:
|
|
return(_T("The Windows Sockets implementation detected that the network subsystem failed."));
|
|
case WSAEADDRINUSE:
|
|
return(_T("The specified address is already in use."));
|
|
case WSAEINPROGRESS:
|
|
return(_T("A blocking Windows Sockets call is in progress."));
|
|
case WSAEADDRNOTAVAIL:
|
|
return(_T("The specified address is not available from the local machine."));
|
|
case WSAEAFNOSUPPORT:
|
|
return(_T("Addresses in the specified family cannot be used with this socket."));
|
|
case WSAECONNREFUSED:
|
|
return(_T("The attempt to connect was rejected."));
|
|
case WSAEDESTADDRREQ:
|
|
return(_T("A destination address is required."));
|
|
case WSAEFAULT:
|
|
return(_T("The nSockAddrLen argument is incorrect."));
|
|
case WSAEINVAL:
|
|
return(_T("Invalid host address."));
|
|
case WSAEISCONN:
|
|
return(_T("The socket is already connected."));
|
|
case WSAEMFILE:
|
|
return(_T("No more file descriptors are available."));
|
|
case WSAENETUNREACH:
|
|
return(_T("The network cannot be reached from this host at this time."));
|
|
case WSAENOBUFS:
|
|
return(_T("No buffer space is available. The socket cannot be connected."));
|
|
case WSAENOTSOCK:
|
|
return(_T("The descriptor is not a socket."));
|
|
case WSAETIMEDOUT:
|
|
return(_T("Attempt to connect timed out without establishing a connection."));
|
|
case WSAEWOULDBLOCK:
|
|
return(_T("The socket is marked as nonblocking and the connection cannot be completed immediately."));
|
|
case WSAEPROTONOSUPPORT:
|
|
return(_T("The specified port is not supported."));
|
|
case WSAEPROTOTYPE:
|
|
return(_T("The specified port is the wrong type for this socket."));
|
|
case WSAESOCKTNOSUPPORT:
|
|
return(_T("The specified socket type is not supported in this address family."));
|
|
}
|
|
return(_T(""));
|
|
}
|
|
else
|
|
return(_T(""));
|
|
}
|
|
|
|
/***************************************************************************************/
|
|
void strncpyWithEOS(char * szStringOut, const char * szStringIn, int nNumberOfCharToCopy, int nStringOutSize)
|
|
{
|
|
if (nNumberOfCharToCopy < nStringOutSize)
|
|
{
|
|
strncpy (szStringOut, szStringIn, nNumberOfCharToCopy);
|
|
szStringOut[nNumberOfCharToCopy] = '\0';
|
|
}
|
|
else
|
|
{
|
|
strncpy (szStringOut, szStringIn, nStringOutSize - 1);
|
|
szStringOut[nStringOutSize - 1] = '\0';
|
|
}
|
|
}
|
|
|