Modified source engine (2017) developed by valve and leaked in 2020. Not for commercial purporses
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

1607 lines
37 KiB

/*
* A simple socket-like package.
* This could undoubtedly be improved, since it does polling and busy-waiting.
* At least it uses asynch I/O and implements timeouts!
*
* Other funkiness includes the use of my own (possibly brain-damaged) error-handling infrastructure.
*
* -Roy Wood (roy@centricsystems.ca)
*
*/
/* ====================================================================
* Copyright (c) 1998-1999 The OpenSSL Project. All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
*
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
*
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in
* the documentation and/or other materials provided with the
* distribution.
*
* 3. All advertising materials mentioning features or use of this
* software must display the following acknowledgment:
* "This product includes software developed by the OpenSSL Project
* for use in the OpenSSL Toolkit. (http://www.openssl.org/)"
*
* 4. The names "OpenSSL Toolkit" and "OpenSSL Project" must not be used to
* endorse or promote products derived from this software without
* prior written permission. For written permission, please contact
* openssl-core@openssl.org.
*
* 5. Products derived from this software may not be called "OpenSSL"
* nor may "OpenSSL" appear in their names without prior written
* permission of the OpenSSL Project.
*
* 6. Redistributions of any form whatsoever must retain the following
* acknowledgment:
* "This product includes software developed by the OpenSSL Project
* for use in the OpenSSL Toolkit (http://www.openssl.org/)"
*
* THIS SOFTWARE IS PROVIDED BY THE OpenSSL PROJECT ``AS IS'' AND ANY
* EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE OpenSSL PROJECT OR
* ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
* STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
* OF THE POSSIBILITY OF SUCH DAMAGE.
* ====================================================================
*
* This product includes cryptographic software written by Eric Young
* (eay@cryptsoft.com). This product includes software written by Tim
* Hudson (tjh@cryptsoft.com).
*
*/
#include "MacSocket.h"
#include <Threads.h>
#include <OpenTransport.h>
#include <OpenTpTInternet.h>
#include <OpenTptClient.h>
#include "CPStringUtils.hpp"
#include "ErrorHandling.hpp"
// #define MACSOCKET_DEBUG 1
#ifdef MACSOCKET_DEBUG
#include <stdio.h>
#endif
extern int errno;
#define kMaxNumSockets 4
struct SocketStruct
{
Boolean mIsInUse;
Boolean mEndpointIsBound;
Boolean mLocalEndIsConnected;
Boolean mRemoteEndIsConnected;
Boolean mReceivedTOpenComplete;
Boolean mReceivedTBindComplete;
Boolean mReceivedTConnect;
Boolean mReceivedTListen;
Boolean mReceivedTPassCon;
Boolean mReceivedTDisconnect;
Boolean mReceivedTOrdRel;
Boolean mReceivedTDisconnectComplete;
long mTimeoutTicks;
long mOperationStartTicks;
MacSocket_IdleWaitCallback mIdleWaitCallback;
void *mUserRefPtr;
OTEventCode mExpectedCode;
OTResult mAsyncOperationResult;
EndpointRef mEndPointRef;
TBind *mBindRequestedAddrInfo;
TBind *mAssignedAddrInfo;
TCall *mRemoteAddrInfo;
Boolean mReadyToReadData;
Boolean mReadyToWriteData;
Ptr mReadBuffer;
Ptr mWriteBuffer;
int mLastError;
char mErrMessage[256];
};
typedef struct SocketStruct SocketStruct;
static SocketStruct sSockets[kMaxNumSockets];
static Boolean sSocketsSetup = false;
static OSErr MyBusyWait(SocketStruct *ioSocket,Boolean returnImmediatelyOnError,OTResult *outOTResult,Boolean *inAsyncOperationCompleteFlag);
static pascal void OTNonYieldingNotifier(void *contextPtr,OTEventCode code,OTResult result,void *cookie);
static Boolean SocketIndexIsValid(const int inSocketNum);
static void InitSocket(SocketStruct *ioSocket);
static void PrepareForAsyncOperation(SocketStruct *ioSocket,const OTEventCode inExpectedCode);
static Boolean TimeoutElapsed(const SocketStruct *inSocket);
static OSStatus NegotiateIPReuseAddrOption(EndpointRef inEndpoint,const Boolean inEnableReuseIP);
void MacSocket_GetSocketErrorInfo(const int inSocketNum,int *outSocketErrCode,char *outSocketErrString,const int inSocketErrStringMaxLength)
{
if (outSocketErrCode != nil)
{
*outSocketErrCode = -1;
}
if (outSocketErrString != nil)
{
CopyCStrToCStr("",outSocketErrString,inSocketErrStringMaxLength);
}
if (SocketIndexIsValid(inSocketNum))
{
SocketStruct *theSocketStruct = &(sSockets[inSocketNum]);
if (outSocketErrCode != nil)
{
*outSocketErrCode = theSocketStruct->mLastError;
}
if (outSocketErrString != nil)
{
CopyCStrToCStr(theSocketStruct->mErrMessage,outSocketErrString,inSocketErrStringMaxLength);
}
}
}
void MacSocket_SetUserRefPtr(const int inSocketNum,void *inNewRefPtr)
{
if (SocketIndexIsValid(inSocketNum))
{
SocketStruct *theSocketStruct = &(sSockets[inSocketNum]);
theSocketStruct->mUserRefPtr = inNewRefPtr;
}
}
void MacSocket_GetLocalIPAndPort(const int inSocketNum,char *outIPAndPort,const int inIPAndPortLength)
{
if (outIPAndPort != nil && SocketIndexIsValid(inSocketNum))
{
char tempString[256];
SocketStruct *theSocketStruct = &(sSockets[inSocketNum]);
CopyCStrToCStr("",tempString,sizeof(tempString));
if (theSocketStruct->mAssignedAddrInfo != nil)
{
InetAddress *theInetAddress = (InetAddress *) theSocketStruct->mAssignedAddrInfo->addr.buf;
InetHost theInetHost = theInetAddress->fHost;
if (theInetHost == 0)
{
InetInterfaceInfo theInetInterfaceInfo;
if (::OTInetGetInterfaceInfo(&theInetInterfaceInfo,kDefaultInetInterface) == noErr)
{
theInetHost = theInetInterfaceInfo.fAddress;
}
}
::OTInetHostToString(theInetHost,tempString);
ConcatCStrToCStr(":",tempString,sizeof(tempString));
ConcatLongIntToCStr(theInetAddress->fPort,tempString,sizeof(tempString));
}
CopyCStrToCStr(tempString,outIPAndPort,inIPAndPortLength);
}
}
void MacSocket_GetRemoteIPAndPort(const int inSocketNum,char *outIPAndPort,const int inIPAndPortLength)
{
if (outIPAndPort != nil && SocketIndexIsValid(inSocketNum))
{
char tempString[256];
SocketStruct *theSocketStruct = &(sSockets[inSocketNum]);
CopyCStrToCStr("",tempString,sizeof(tempString));
if (theSocketStruct->mRemoteAddrInfo != nil)
{
InetAddress *theInetAddress = (InetAddress *) theSocketStruct->mRemoteAddrInfo->addr.buf;
InetHost theInetHost = theInetAddress->fHost;
if (theInetHost == 0)
{
InetInterfaceInfo theInetInterfaceInfo;
if (::OTInetGetInterfaceInfo(&theInetInterfaceInfo,kDefaultInetInterface) == noErr)
{
theInetHost = theInetInterfaceInfo.fAddress;
}
}
::OTInetHostToString(theInetHost,tempString);
ConcatCStrToCStr(":",tempString,sizeof(tempString));
ConcatLongIntToCStr(theInetAddress->fPort,tempString,sizeof(tempString));
}
CopyCStrToCStr(tempString,outIPAndPort,inIPAndPortLength);
}
}
Boolean MacSocket_RemoteEndIsClosing(const int inSocketNum)
{
Boolean theResult = false;
if (SocketIndexIsValid(inSocketNum))
{
SocketStruct *theSocketStruct = &(sSockets[inSocketNum]);
theResult = theSocketStruct->mReceivedTOrdRel;
}
return(theResult);
}
Boolean MacSocket_ListenCompleted(const int inSocketNum)
{
Boolean theResult = false;
if (SocketIndexIsValid(inSocketNum))
{
SocketStruct *theSocketStruct = &(sSockets[inSocketNum]);
theResult = theSocketStruct->mReceivedTPassCon;
}
return(theResult);
}
Boolean MacSocket_RemoteEndIsOpen(const int inSocketNum)
{
if (SocketIndexIsValid(inSocketNum))
{
SocketStruct *theSocketStruct = &(sSockets[inSocketNum]);
return(theSocketStruct->mRemoteEndIsConnected);
}
else
{
return(false);
}
}
Boolean MacSocket_LocalEndIsOpen(const int inSocketNum)
{
if (SocketIndexIsValid(inSocketNum))
{
SocketStruct *theSocketStruct = &(sSockets[inSocketNum]);
return(theSocketStruct->mLocalEndIsConnected);
}
else
{
return(false);
}
}
static Boolean TimeoutElapsed(const SocketStruct *inSocket)
{
Boolean timeIsUp = false;
if (inSocket != nil && inSocket->mTimeoutTicks > 0 && ::TickCount() > inSocket->mOperationStartTicks + inSocket->mTimeoutTicks)
{
timeIsUp = true;
}
return(timeIsUp);
}
static Boolean SocketIndexIsValid(const int inSocketNum)
{
if (inSocketNum >= 0 && inSocketNum < kMaxNumSockets && sSockets[inSocketNum].mEndPointRef != kOTInvalidEndpointRef)
{
return(true);
}
else
{
return(false);
}
}
static void InitSocket(SocketStruct *ioSocket)
{
ioSocket->mIsInUse = false;
ioSocket->mEndpointIsBound = false;
ioSocket->mLocalEndIsConnected = false;
ioSocket->mRemoteEndIsConnected = false;
ioSocket->mReceivedTOpenComplete = false;
ioSocket->mReceivedTBindComplete = false;
ioSocket->mReceivedTConnect = false;
ioSocket->mReceivedTListen = false;
ioSocket->mReceivedTPassCon = false;
ioSocket->mReceivedTDisconnect = false;
ioSocket->mReceivedTOrdRel = false;
ioSocket->mReceivedTDisconnectComplete = false;
ioSocket->mTimeoutTicks = 30 * 60;
ioSocket->mOperationStartTicks = -1;
ioSocket->mIdleWaitCallback = nil;
ioSocket->mUserRefPtr = nil;
ioSocket->mExpectedCode = 0;
ioSocket->mAsyncOperationResult = noErr;
ioSocket->mEndPointRef = kOTInvalidEndpointRef;
ioSocket->mBindRequestedAddrInfo = nil;
ioSocket->mAssignedAddrInfo = nil;
ioSocket->mRemoteAddrInfo = nil;
ioSocket->mReadyToReadData = false;
ioSocket->mReadyToWriteData = true;
ioSocket->mReadBuffer = nil;
ioSocket->mWriteBuffer = nil;
ioSocket->mLastError = noErr;
CopyCStrToCStr("",ioSocket->mErrMessage,sizeof(ioSocket->mErrMessage));
}
static void PrepareForAsyncOperation(SocketStruct *ioSocket,const OTEventCode inExpectedCode)
{
ioSocket->mOperationStartTicks = ::TickCount();
ioSocket->mAsyncOperationResult = noErr;
ioSocket->mExpectedCode = inExpectedCode;
}
// The wait function....
static OSErr MyBusyWait(SocketStruct *ioSocket,Boolean returnImmediatelyOnError,OTResult *outOTResult,Boolean *inAsyncOperationCompleteFlag)
{
OSErr errCode = noErr;
OTResult theOTResult = noErr;
SetErrorMessageAndBailIfNil(ioSocket,"MyBusyWait: Bad parameter, ioSocket = nil");
SetErrorMessageAndBailIfNil(inAsyncOperationCompleteFlag,"MyBusyWait: Bad parameter, inAsyncOperationCompleteFlag = nil");
for (;;)
{
if (*inAsyncOperationCompleteFlag)
{
theOTResult = ioSocket->mAsyncOperationResult;
break;
}
if (ioSocket->mIdleWaitCallback != nil)
{
theOTResult = (*(ioSocket->mIdleWaitCallback))(ioSocket->mUserRefPtr);
if (theOTResult != noErr && returnImmediatelyOnError)
{
break;
}
}
if (TimeoutElapsed(ioSocket))
{
theOTResult = kMacSocket_TimeoutErr;
break;
}
}
EXITPOINT:
if (outOTResult != nil)
{
*outOTResult = theOTResult;
}
return(errCode);
}
// I used to do thread switching, but stopped. It could easily be rolled back in though....
static pascal void OTNonYieldingNotifier(void *contextPtr,OTEventCode code,OTResult result,void *cookie)
{
SocketStruct *theSocketStruct = (SocketStruct *) contextPtr;
if (theSocketStruct != nil)
{
if (theSocketStruct->mExpectedCode != 0 && code == theSocketStruct->mExpectedCode)
{
theSocketStruct->mAsyncOperationResult = result;
theSocketStruct->mExpectedCode = 0;
}
switch (code)
{
case T_OPENCOMPLETE:
{
theSocketStruct->mReceivedTOpenComplete = true;
theSocketStruct->mEndPointRef = (EndpointRef) cookie;
break;
}
case T_BINDCOMPLETE:
{
theSocketStruct->mReceivedTBindComplete = true;
break;
}
case T_CONNECT:
{
theSocketStruct->mReceivedTConnect = true;
theSocketStruct->mLocalEndIsConnected = true;
theSocketStruct->mRemoteEndIsConnected = true;
break;
}
case T_LISTEN:
{
theSocketStruct->mReceivedTListen = true;
break;
}
case T_PASSCON:
{
theSocketStruct->mReceivedTPassCon = true;
theSocketStruct->mLocalEndIsConnected = true;
theSocketStruct->mRemoteEndIsConnected = true;
break;
}
case T_DATA:
{
theSocketStruct->mReadyToReadData = true;
break;
}
case T_GODATA:
{
theSocketStruct->mReadyToWriteData = true;
break;
}
case T_DISCONNECT:
{
theSocketStruct->mReceivedTDisconnect = true;
theSocketStruct->mRemoteEndIsConnected = false;
theSocketStruct->mLocalEndIsConnected = false;
::OTRcvDisconnect(theSocketStruct->mEndPointRef,nil);
break;
}
case T_ORDREL:
{
theSocketStruct->mReceivedTOrdRel = true;
// We can still write data, so don't clear mRemoteEndIsConnected
::OTRcvOrderlyDisconnect(theSocketStruct->mEndPointRef);
break;
}
case T_DISCONNECTCOMPLETE:
{
theSocketStruct->mReceivedTDisconnectComplete = true;
theSocketStruct->mRemoteEndIsConnected = false;
theSocketStruct->mLocalEndIsConnected = false;
break;
}
}
}
/*
T_LISTEN OTListen
T_CONNECT OTRcvConnect
T_DATA OTRcv, OTRcvUData
T_DISCONNECT OTRcvDisconnect
T_ORDREL OTRcvOrderlyDisconnect
T_GODATA OTSnd, OTSndUData, OTLook
T_PASSCON none
T_EXDATA OTRcv
T_GOEXDATA OTSnd, OTLook
T_UDERR OTRcvUDErr
*/
}
// Initialize the main socket data structure
OSErr MacSocket_Startup(void)
{
if (!sSocketsSetup)
{
for (int i = 0;i < kMaxNumSockets;i++)
{
InitSocket(&(sSockets[i]));
}
::InitOpenTransport();
sSocketsSetup = true;
}
return(noErr);
}
// Cleanup before exiting
OSErr MacSocket_Shutdown(void)
{
if (sSocketsSetup)
{
for (int i = 0;i < kMaxNumSockets;i++)
{
SocketStruct *theSocketStruct = &(sSockets[i]);
if (theSocketStruct->mIsInUse)
{
if (theSocketStruct->mEndPointRef != kOTInvalidEndpointRef)
{
OTResult theOTResult;
// Since we're killing the endpoint, I don't bother to send the disconnect (sorry!)
/*
if (theSocketStruct->mLocalEndIsConnected)
{
// This is an abortive action, so we do a hard disconnect instead of an OTSndOrderlyDisconnect
theOTResult = ::OTSndDisconnect(theSocketStruct->mEndPointRef, nil);
// Now we have to watch for T_DISCONNECTCOMPLETE event
theSocketStruct->mLocalEndIsConnected = false;
}
*/
theOTResult = ::OTCloseProvider(theSocketStruct->mEndPointRef);
theSocketStruct->mEndPointRef = kOTInvalidEndpointRef;
}
if (theSocketStruct->mBindRequestedAddrInfo != nil)
{
::OTFree((void *) theSocketStruct->mBindRequestedAddrInfo,T_BIND);
theSocketStruct->mBindRequestedAddrInfo = nil;
}
if (theSocketStruct->mAssignedAddrInfo != nil)
{
::OTFree((void *) theSocketStruct->mAssignedAddrInfo,T_BIND);
theSocketStruct->mAssignedAddrInfo = nil;
}
if (theSocketStruct->mRemoteAddrInfo != nil)
{
::OTFree((void *) theSocketStruct->mRemoteAddrInfo,T_CALL);
theSocketStruct->mRemoteAddrInfo = nil;
}
}
}
::CloseOpenTransport();
sSocketsSetup = false;
}
return(noErr);
}
// Allocate a socket
OSErr MacSocket_socket(int *outSocketNum,const Boolean inDoThreadSwitching,const long inTimeoutTicks,MacSocket_IdleWaitCallback inIdleWaitCallback,void *inUserRefPtr)
{
// Gotta roll support back in for threads eventually.....
#pragma unused(inDoThreadSwitching)
OSErr errCode = noErr;
SetErrorMessageAndBailIfNil(outSocketNum,"MacSocket_socket: Bad parameter, outSocketNum == nil");
*outSocketNum = -1;
// Find an unused socket
for (int i = 0;i < kMaxNumSockets;i++)
{
if (sSockets[i].mIsInUse == false)
{
OTResult theOTResult;
SocketStruct *theSocketStruct = &(sSockets[i]);
InitSocket(theSocketStruct);
theSocketStruct->mIdleWaitCallback = inIdleWaitCallback;
theSocketStruct->mUserRefPtr = inUserRefPtr;
theSocketStruct->mTimeoutTicks = inTimeoutTicks;
// Set up OT endpoint
PrepareForAsyncOperation(theSocketStruct,T_OPENCOMPLETE);
theOTResult = ::OTAsyncOpenEndpoint(OTCreateConfiguration(kTCPName),0,nil,OTNonYieldingNotifier,(void *) theSocketStruct);
SetErrorMessageAndLongIntAndBailIfError(theOTResult,"MacSocket_socket: Can't create OT endpoint, OTAsyncOpenEndpoint() = ",theOTResult);
BailIfError(MyBusyWait(theSocketStruct,false,&theOTResult,&(theSocketStruct->mReceivedTOpenComplete)));
SetErrorMessageAndLongIntAndBailIfError(theOTResult,"MacSocket_socket: Can't create OT endpoint, OTAsyncOpenEndpoint() = ",theOTResult);
*outSocketNum = i;
errCode = noErr;
theSocketStruct->mIsInUse = true;
break;
}
else if (i == kMaxNumSockets - 1)
{
SetErrorMessageAndBail("MacSocket_socket: No sockets available");
}
}
EXITPOINT:
errno = errCode;
return(errCode);
}
OSErr MacSocket_listen(const int inSocketNum,const int inPortNum)
{
OSErr errCode = noErr;
SocketStruct *theSocketStruct = nil;
if (!SocketIndexIsValid(inSocketNum))
{
SetErrorMessageAndBail("MacSocket_listen: Invalid socket number specified");
}
theSocketStruct = &(sSockets[inSocketNum]);
OTResult theOTResult;
if (theSocketStruct->mBindRequestedAddrInfo == nil)
{
theSocketStruct->mBindRequestedAddrInfo = (TBind *) ::OTAlloc(theSocketStruct->mEndPointRef,T_BIND,T_ADDR,&theOTResult);
SetErrorMessageAndLongIntAndBailIfError(theOTResult,"MacSocket_listen: Can't allocate OT T_BIND structure, OTAlloc() = ",theOTResult);
SetErrorMessageAndBailIfNil(theSocketStruct->mBindRequestedAddrInfo,"MacSocket_listen: Can't allocate OT T_BIND structure, OTAlloc() returned nil");
}
if (theSocketStruct->mAssignedAddrInfo == nil)
{
theSocketStruct->mAssignedAddrInfo = (TBind *) ::OTAlloc(theSocketStruct->mEndPointRef,T_BIND,T_ADDR,&theOTResult);
SetErrorMessageAndLongIntAndBailIfError(theOTResult,"MacSocket_listen: Can't allocate OT T_BIND structure, OTAlloc() = ",theOTResult);
SetErrorMessageAndBailIfNil(theSocketStruct->mAssignedAddrInfo,"MacSocket_listen: Can't allocate OT T_BIND structure, OTAlloc() returned nil");
}
if (theSocketStruct->mRemoteAddrInfo == nil)
{
theSocketStruct->mRemoteAddrInfo = (TCall *) ::OTAlloc(theSocketStruct->mEndPointRef,T_CALL,T_ADDR,&theOTResult);
SetErrorMessageAndLongIntAndBailIfError(theOTResult,"MacSocket_listen: Can't allocate OT T_CALL structure, OTAlloc() = ",theOTResult);
SetErrorMessageAndBailIfNil(theSocketStruct->mRemoteAddrInfo,"MacSocket_listen: Can't allocate OT T_CALL structure, OTAlloc() returned nil");
}
if (!theSocketStruct->mEndpointIsBound)
{
InetInterfaceInfo theInetInterfaceInfo;
theOTResult = ::OTInetGetInterfaceInfo(&theInetInterfaceInfo,kDefaultInetInterface);
SetErrorMessageAndLongIntAndBailIfError(theOTResult,"MacSocket_listen: Can't determine OT interface info, OTInetGetInterfaceInfo() = ",theOTResult);
InetAddress *theInetAddress = (InetAddress *) theSocketStruct->mBindRequestedAddrInfo->addr.buf;
// theInetAddress->fAddressType = AF_INET;
// theInetAddress->fPort = inPortNum;
// theInetAddress->fHost = theInetInterfaceInfo.fAddress;
::OTInitInetAddress(theInetAddress,inPortNum,theInetInterfaceInfo.fAddress);
theSocketStruct->mBindRequestedAddrInfo->addr.len = sizeof(InetAddress);
theSocketStruct->mBindRequestedAddrInfo->qlen = 1;
theOTResult = ::OTSetSynchronous(theSocketStruct->mEndPointRef);
SetErrorMessageAndLongIntAndBailIfError(theOTResult,"MacSocket_listen: Can't set OT endpoint mode, OTSetSynchronous() = ",theOTResult);
theOTResult = NegotiateIPReuseAddrOption(theSocketStruct->mEndPointRef,true);
SetErrorMessageAndLongIntAndBailIfError(theOTResult,"MacSocket_listen: Can't set OT IP address reuse flag, NegotiateIPReuseAddrOption() = ",theOTResult);
theOTResult = ::OTSetAsynchronous(theSocketStruct->mEndPointRef);
SetErrorMessageAndLongIntAndBailIfError(theOTResult,"MacSocket_listen: Can't set OT endpoint mode, OTSetAsynchronous() = ",theOTResult);
PrepareForAsyncOperation(theSocketStruct,T_BINDCOMPLETE);
theOTResult = ::OTBind(theSocketStruct->mEndPointRef,theSocketStruct->mBindRequestedAddrInfo,theSocketStruct->mAssignedAddrInfo);
SetErrorMessageAndLongIntAndBailIfError(theOTResult,"MacSocket_listen: Can't bind OT endpoint, OTBind() = ",theOTResult);
BailIfError(MyBusyWait(theSocketStruct,false,&theOTResult,&(theSocketStruct->mReceivedTBindComplete)));
SetErrorMessageAndLongIntAndBailIfError(theOTResult,"MacSocket_listen: Can't bind OT endpoint, OTBind() = ",theOTResult);
theSocketStruct->mEndpointIsBound = true;
}
PrepareForAsyncOperation(theSocketStruct,T_LISTEN);
theOTResult = ::OTListen(theSocketStruct->mEndPointRef,theSocketStruct->mRemoteAddrInfo);
if (theOTResult == noErr)
{
PrepareForAsyncOperation(theSocketStruct,T_PASSCON);
theOTResult = ::OTAccept(theSocketStruct->mEndPointRef,theSocketStruct->mEndPointRef,theSocketStruct->mRemoteAddrInfo);
SetErrorMessageAndLongIntAndBailIfError(theOTResult,"MacSocket_listen: Can't begin OT accept, OTAccept() = ",theOTResult);
BailIfError(MyBusyWait(theSocketStruct,false,&theOTResult,&(theSocketStruct->mReceivedTPassCon)));
SetErrorMessageAndLongIntAndBailIfError(theOTResult,"MacSocket_listen: Can't accept OT connection, OTAccept() = ",theOTResult);
}
else if (theOTResult == kOTNoDataErr)
{
theOTResult = noErr;
}
else
{
SetErrorMessageAndLongIntAndBail("MacSocket_listen: Can't begin OT listen, OTListen() = ",theOTResult);
}
errCode = noErr;
EXITPOINT:
if (theSocketStruct != nil)
{
theSocketStruct->mLastError = noErr;
CopyCStrToCStr("",theSocketStruct->mErrMessage,sizeof(theSocketStruct->mErrMessage));
if (errCode != noErr)
{
theSocketStruct->mLastError = errCode;
CopyCStrToCStr(GetErrorMessage(),theSocketStruct->mErrMessage,sizeof(theSocketStruct->mErrMessage));
}
}
errno = errCode;
return(errCode);
}
OSErr MacSocket_connect(const int inSocketNum,char *inTargetAddressAndPort)
{
OSErr errCode = noErr;
SocketStruct *theSocketStruct = nil;
if (!SocketIndexIsValid(inSocketNum))
{
SetErrorMessageAndBail("MacSocket_connect: Invalid socket number specified");
}
theSocketStruct = &(sSockets[inSocketNum]);
if (theSocketStruct->mEndpointIsBound)
{
SetErrorMessageAndBail("MacSocket_connect: Socket previously bound");
}
OTResult theOTResult;
theSocketStruct->mBindRequestedAddrInfo = (TBind *) ::OTAlloc(theSocketStruct->mEndPointRef,T_BIND,T_ADDR,&theOTResult);
SetErrorMessageAndLongIntAndBailIfError(theOTResult,"MacSocket_connect: Can't allocate OT T_BIND structure, OTAlloc() = ",theOTResult);
SetErrorMessageAndBailIfNil(theSocketStruct->mBindRequestedAddrInfo,"MacSocket_connect: Can't allocate OT T_BIND structure, OTAlloc() returned nil");
theSocketStruct->mAssignedAddrInfo = (TBind *) ::OTAlloc(theSocketStruct->mEndPointRef,T_BIND,T_ADDR,&theOTResult);
SetErrorMessageAndLongIntAndBailIfError(theOTResult,"MacSocket_connect: Can't allocate OT T_BIND structure, OTAlloc() = ",theOTResult);
SetErrorMessageAndBailIfNil(theSocketStruct->mAssignedAddrInfo,"MacSocket_connect: Can't allocate OT T_BIND structure, OTAlloc() returned nil");
theSocketStruct->mRemoteAddrInfo = (TCall *) ::OTAlloc(theSocketStruct->mEndPointRef,T_CALL,T_ADDR,&theOTResult);
SetErrorMessageAndLongIntAndBailIfError(theOTResult,"MacSocket_connect: Can't allocate OT T_CALL structure, OTAlloc() = ",theOTResult);
SetErrorMessageAndBailIfNil(theSocketStruct->mRemoteAddrInfo,"MacSocket_connect: Can't allocate OT T_CALL structure, OTAlloc() returned nil");
PrepareForAsyncOperation(theSocketStruct,T_BINDCOMPLETE);
theOTResult = ::OTBind(theSocketStruct->mEndPointRef,nil,theSocketStruct->mAssignedAddrInfo);
SetErrorMessageAndLongIntAndBailIfError(theOTResult,"MacSocket_connect: Can't bind OT endpoint, OTBind() = ",theOTResult);
BailIfError(MyBusyWait(theSocketStruct,false,&theOTResult,&(theSocketStruct->mReceivedTBindComplete)));
SetErrorMessageAndLongIntAndBailIfError(theOTResult,"MacSocket_connect: Can't bind OT endpoint, OTBind() = ",theOTResult);
theSocketStruct->mEndpointIsBound = true;
TCall sndCall;
DNSAddress hostDNSAddress;
// Set up target address
sndCall.addr.buf = (UInt8 *) &hostDNSAddress;
sndCall.addr.len = ::OTInitDNSAddress(&hostDNSAddress,inTargetAddressAndPort);
sndCall.opt.buf = nil;
sndCall.opt.len = 0;
sndCall.udata.buf = nil;
sndCall.udata.len = 0;
sndCall.sequence = 0;
// Connect!
PrepareForAsyncOperation(theSocketStruct,T_CONNECT);
theOTResult = ::OTConnect(theSocketStruct->mEndPointRef,&sndCall,nil);
if (theOTResult == kOTNoDataErr)
{
theOTResult = noErr;
}
SetErrorMessageAndLongIntAndBailIfError(theOTResult,"MacSocket_connect: Can't connect OT endpoint, OTConnect() = ",theOTResult);
BailIfError(MyBusyWait(theSocketStruct,false,&theOTResult,&(theSocketStruct->mReceivedTConnect)));
if (theOTResult == kMacSocket_TimeoutErr)
{
SetErrorMessageAndBail("MacSocket_connect: Can't connect OT endpoint, OTConnect() = kMacSocket_TimeoutErr");
}
else
{
SetErrorMessageAndLongIntAndBailIfError(theOTResult,"MacSocket_connect: Can't connect OT endpoint, OTConnect() = ",theOTResult);
}
theOTResult = ::OTRcvConnect(theSocketStruct->mEndPointRef,nil);
SetErrorMessageAndLongIntAndBailIfError(theOTResult,"MacSocket_connect: Can't complete connect on OT endpoint, OTRcvConnect() = ",theOTResult);
errCode = noErr;
#ifdef MACSOCKET_DEBUG
printf("MacSocket_connect: connect completed\n");
#endif
EXITPOINT:
if (theSocketStruct != nil)
{
theSocketStruct->mLastError = noErr;
CopyCStrToCStr("",theSocketStruct->mErrMessage,sizeof(theSocketStruct->mErrMessage));
if (errCode != noErr)
{
theSocketStruct->mLastError = errCode;
CopyCStrToCStr(GetErrorMessage(),theSocketStruct->mErrMessage,sizeof(theSocketStruct->mErrMessage));
}
}
errno = errCode;
return(errCode);
}
// Close a connection
OSErr MacSocket_close(const int inSocketNum)
{
OSErr errCode = noErr;
SocketStruct *theSocketStruct = nil;
if (!SocketIndexIsValid(inSocketNum))
{
SetErrorMessageAndBail("MacSocket_close: Invalid socket number specified");
}
theSocketStruct = &(sSockets[inSocketNum]);
if (theSocketStruct->mEndPointRef != kOTInvalidEndpointRef)
{
OTResult theOTResult = noErr;
// Try to play nice
if (theSocketStruct->mReceivedTOrdRel)
{
// Already did an OTRcvOrderlyDisconnect() in the notifier
if (theSocketStruct->mLocalEndIsConnected)
{
theOTResult = ::OTSndOrderlyDisconnect(theSocketStruct->mEndPointRef);
theSocketStruct->mLocalEndIsConnected = false;
}
}
else if (theSocketStruct->mLocalEndIsConnected)
{
theOTResult = ::OTSndOrderlyDisconnect(theSocketStruct->mEndPointRef);
theSocketStruct->mLocalEndIsConnected = false;
// Wait for other end to hang up too!
// PrepareForAsyncOperation(theSocketStruct,T_ORDREL);
//
// errCode = MyBusyWait(theSocketStruct,false,&theOTResult,&(theSocketStruct->mReceivedTOrdRel));
}
if (theOTResult != noErr)
{
::OTCloseProvider(theSocketStruct->mEndPointRef);
}
else
{
theOTResult = ::OTCloseProvider(theSocketStruct->mEndPointRef);
}
theSocketStruct->mEndPointRef = kOTInvalidEndpointRef;
errCode = theOTResult;
}
theSocketStruct->mIsInUse = false;
EXITPOINT:
if (theSocketStruct != nil)
{
theSocketStruct->mLastError = noErr;
CopyCStrToCStr("",theSocketStruct->mErrMessage,sizeof(theSocketStruct->mErrMessage));
if (errCode != noErr)
{
theSocketStruct->mLastError = errCode;
CopyCStrToCStr(GetErrorMessage(),theSocketStruct->mErrMessage,sizeof(theSocketStruct->mErrMessage));
}
}
errno = errCode;
return(errCode);
}
// Receive some bytes
int MacSocket_recv(const int inSocketNum,void *outBuff,int outBuffLength,const Boolean inBlock)
{
OSErr errCode = noErr;
int totalBytesRead = 0;
SocketStruct *theSocketStruct = nil;
SetErrorMessageAndBailIfNil(outBuff,"MacSocket_recv: Bad parameter, outBuff = nil");
if (outBuffLength <= 0)
{
SetErrorMessageAndBail("MacSocket_recv: Bad parameter, outBuffLength <= 0");
}
if (!SocketIndexIsValid(inSocketNum))
{
SetErrorMessageAndBail("MacSocket_recv: Invalid socket number specified");
}
theSocketStruct = &(sSockets[inSocketNum]);
if (!theSocketStruct->mLocalEndIsConnected)
{
SetErrorMessageAndBail("MacSocket_recv: Socket not connected");
}
if (theSocketStruct->mReceivedTOrdRel)
{
totalBytesRead = 0;
goto EXITPOINT;
}
PrepareForAsyncOperation(theSocketStruct,0);
for (;;)
{
int bytesRead;
OTResult theOTResult;
theOTResult = ::OTRcv(theSocketStruct->mEndPointRef,(void *) ((unsigned long) outBuff + (unsigned long) totalBytesRead),outBuffLength - totalBytesRead,nil);
if (theOTResult >= 0)
{
bytesRead = theOTResult;
#ifdef MACSOCKET_DEBUG
printf("MacSocket_recv: read %d bytes in part\n",bytesRead);
#endif
}
else if (theOTResult == kOTNoDataErr)
{
bytesRead = 0;
}
else
{
SetErrorMessageAndLongIntAndBail("MacSocket_recv: Can't receive OT data, OTRcv() = ",theOTResult);
}
totalBytesRead += bytesRead;
if (totalBytesRead <= 0)
{
if (theSocketStruct->mReceivedTOrdRel)
{
break;
}
// This seems pretty stupid to me now. Maybe I'll delete this blocking garbage.
if (inBlock)
{
if (TimeoutElapsed(theSocketStruct))
{
SetErrorCodeAndMessageAndBail(kMacSocket_TimeoutErr,"MacSocket_recv: Receive operation timed-out");
}
if (theSocketStruct->mIdleWaitCallback != nil)
{
theOTResult = (*(theSocketStruct->mIdleWaitCallback))(theSocketStruct->mUserRefPtr);
SetErrorMessageAndBailIfError(theOTResult,"MacSocket_recv: User cancelled operation");
}
continue;
}
}
break;
}
errCode = noErr;
#ifdef MACSOCKET_DEBUG
printf("MacSocket_recv: read %d bytes in total\n",totalBytesRead);
#endif
EXITPOINT:
if (theSocketStruct != nil)
{
theSocketStruct->mLastError = noErr;
CopyCStrToCStr("",theSocketStruct->mErrMessage,sizeof(theSocketStruct->mErrMessage));
if (errCode != noErr)
{
theSocketStruct->mLastError = errCode;
CopyCStrToCStr(GetErrorMessage(),theSocketStruct->mErrMessage,sizeof(theSocketStruct->mErrMessage));
}
}
errno = errCode;
return(totalBytesRead);
}
// Send some bytes
int MacSocket_send(const int inSocketNum,const void *inBuff,int inBuffLength)
{
OSErr errCode = noErr;
int bytesSent = 0;
SocketStruct *theSocketStruct = nil;
SetErrorMessageAndBailIfNil(inBuff,"MacSocket_send: Bad parameter, inBuff = nil");
if (inBuffLength <= 0)
{
SetErrorMessageAndBail("MacSocket_send: Bad parameter, inBuffLength <= 0");
}
if (!SocketIndexIsValid(inSocketNum))
{
SetErrorMessageAndBail("MacSocket_send: Invalid socket number specified");
}
theSocketStruct = &(sSockets[inSocketNum]);
if (!theSocketStruct->mLocalEndIsConnected)
{
SetErrorMessageAndBail("MacSocket_send: Socket not connected");
}
OTResult theOTResult;
PrepareForAsyncOperation(theSocketStruct,0);
while (bytesSent < inBuffLength)
{
if (theSocketStruct->mIdleWaitCallback != nil)
{
theOTResult = (*(theSocketStruct->mIdleWaitCallback))(theSocketStruct->mUserRefPtr);
SetErrorMessageAndBailIfError(theOTResult,"MacSocket_send: User cancelled");
}
theOTResult = ::OTSnd(theSocketStruct->mEndPointRef,(void *) ((unsigned long) inBuff + bytesSent),inBuffLength - bytesSent,0);
if (theOTResult >= 0)
{
bytesSent += theOTResult;
theOTResult = noErr;
// Reset timer....
PrepareForAsyncOperation(theSocketStruct,0);
}
if (theOTResult == kOTFlowErr)
{
if (TimeoutElapsed(theSocketStruct))
{
SetErrorCodeAndMessageAndBail(kMacSocket_TimeoutErr,"MacSocket_send: Send timed-out")
}
theOTResult = noErr;
}
SetErrorMessageAndLongIntAndBailIfError(theOTResult,"MacSocket_send: Can't send OT data, OTSnd() = ",theOTResult);
}
errCode = noErr;
#ifdef MACSOCKET_DEBUG
printf("MacSocket_send: sent %d bytes\n",bytesSent);
#endif
EXITPOINT:
if (theSocketStruct != nil)
{
theSocketStruct->mLastError = noErr;
CopyCStrToCStr("",theSocketStruct->mErrMessage,sizeof(theSocketStruct->mErrMessage));
if (errCode != noErr)
{
theSocketStruct->mLastError = errCode;
CopyCStrToCStr(GetErrorMessage(),theSocketStruct->mErrMessage,sizeof(theSocketStruct->mErrMessage));
}
}
if (errCode != noErr)
{
::SysBeep(1);
}
errno = errCode;
return(bytesSent);
}
static OSStatus NegotiateIPReuseAddrOption(EndpointRef inEndpoint,const Boolean inEnableReuseIP)
{
OSStatus errCode;
UInt8 buf[kOTFourByteOptionSize];
TOption* theOTOption;
TOptMgmt theOTRequest;
TOptMgmt theOTResult;
if (!OTIsSynchronous(inEndpoint))
{
SetErrorMessageAndBail("NegotiateIPReuseAddrOption: Open Transport endpoint is not synchronous");
}
theOTRequest.opt.buf = buf;
theOTRequest.opt.len = sizeof(buf);
theOTRequest.flags = T_NEGOTIATE;
theOTResult.opt.buf = buf;
theOTResult.opt.maxlen = kOTFourByteOptionSize;
theOTOption = (TOption *) buf;
theOTOption->level = INET_IP;
theOTOption->name = IP_REUSEADDR;
theOTOption->len = kOTFourByteOptionSize;
theOTOption->status = 0;
*((UInt32 *) (theOTOption->value)) = inEnableReuseIP;
errCode = ::OTOptionManagement(inEndpoint,&theOTRequest,&theOTResult);
if (errCode == kOTNoError)
{
if (theOTOption->status != T_SUCCESS)
{
errCode = theOTOption->status;
}
else
{
errCode = kOTNoError;
}
}
EXITPOINT:
errno = errCode;
return(errCode);
}
// Some rough notes....
// OTAckSends(ep);
// OTAckSends(ep) // enable AckSend option
// ......
// buf = OTAllocMem( nbytes); // Allocate nbytes of memory from OT
// OTSnd(ep, buf, nbytes, 0); // send a packet
// ......
// NotifyProc( .... void* theParam) // Notifier Proc
// case T_MEMORYRELEASED: // process event
// OTFreeMem( theParam); // free up memory
// break;
/*
struct InetInterfaceInfo
{
InetHost fAddress;
InetHost fNetmask;
InetHost fBroadcastAddr;
InetHost fDefaultGatewayAddr;
InetHost fDNSAddr;
UInt16 fVersion;
UInt16 fHWAddrLen;
UInt8* fHWAddr;
UInt32 fIfMTU;
UInt8* fReservedPtrs[2];
InetDomainName fDomainName;
UInt32 fIPSecondaryCount;
UInt8 fReserved[252];
};
typedef struct InetInterfaceInfo InetInterfaceInfo;
((InetAddress *) addr.buf)->fHost
struct TBind
{
TNetbuf addr;
OTQLen qlen;
};
typedef struct TBind TBind;
struct TNetbuf
{
size_t maxlen;
size_t len;
UInt8* buf;
};
typedef struct TNetbuf TNetbuf;
struct InetAddress
{
OTAddressType fAddressType; // always AF_INET
InetPort fPort; // Port number
InetHost fHost; // Host address in net byte order
UInt8 fUnused[8]; // Traditional unused bytes
};
typedef struct InetAddress InetAddress;
*/
/*
static pascal void Notifier(void* context, OTEventCode event, OTResult result, void* cookie)
{
EPInfo* epi = (EPInfo*) context;
switch (event)
{
case T_LISTEN:
{
DoListenAccept();
return;
}
case T_ACCEPTCOMPLETE:
{
if (result != kOTNoError)
DBAlert1("Notifier: T_ACCEPTCOMPLETE - result %d",result);
return;
}
case T_PASSCON:
{
if (result != kOTNoError)
{
DBAlert1("Notifier: T_PASSCON result %d", result);
return;
}
OTAtomicAdd32(1, &gCntrConnections);
OTAtomicAdd32(1, &gCntrTotalConnections);
OTAtomicAdd32(1, &gCntrIntervalConnects);
if ( OTAtomicSetBit(&epi->stateFlags, kPassconBit) != 0 )
{
ReadData(epi);
}
return;
}
case T_DATA:
{
if ( OTAtomicSetBit(&epi->stateFlags, kPassconBit) != 0 )
{
ReadData(epi);
}
return;
}
case T_GODATA:
{
SendData(epi);
return;
}
case T_DISCONNECT:
{
DoRcvDisconnect(epi);
return;
}
case T_DISCONNECTCOMPLETE:
{
if (result != kOTNoError)
DBAlert1("Notifier: T_DISCONNECT_COMPLETE result %d",result);
return;
}
case T_MEMORYRELEASED:
{
OTAtomicAdd32(-1, &epi->outstandingSends);
return;
}
default:
{
DBAlert1("Notifier: unknown event <%x>", event);
return;
}
}
}
*/