mirror of
https://github.com/celisej567/mcpe.git
synced 2026-09-17 20:10:23 +03:00
* Initial commit.
:)
This commit is contained in:
iProgramInCpp
513
thirdparty/raknet/RakNetSocket2.cpp
vendored
Normal file
513
thirdparty/raknet/RakNetSocket2.cpp
vendored
Normal file
@@ -0,0 +1,513 @@
|
||||
/*
|
||||
* Copyright (c) 2014, Oculus VR, Inc.
|
||||
* All rights reserved.
|
||||
*
|
||||
* This source code is licensed under the BSD-style license found in the
|
||||
* LICENSE file in the root directory of this source tree. An additional grant
|
||||
* of patent rights can be found in the PATENTS file in the same directory.
|
||||
*
|
||||
*/
|
||||
|
||||
#include "RakNetSocket2.h"
|
||||
#include "RakMemoryOverride.h"
|
||||
#include "RakAssert.h"
|
||||
#include "RakSleep.h"
|
||||
#include "SocketDefines.h"
|
||||
#include "GetTime.h"
|
||||
#include <stdio.h>
|
||||
#include <string.h> // memcpy
|
||||
|
||||
using namespace RakNet;
|
||||
|
||||
#ifdef _WIN32
|
||||
#else
|
||||
#include <unistd.h>
|
||||
#include <fcntl.h>
|
||||
#include <arpa/inet.h>
|
||||
#include <errno.h> // error numbers
|
||||
#if !defined(ANDROID)
|
||||
#include <ifaddrs.h>
|
||||
#endif
|
||||
#include <netinet/in.h>
|
||||
#include <net/if.h>
|
||||
#include <sys/types.h>
|
||||
#include <sys/socket.h>
|
||||
#include <sys/ioctl.h>
|
||||
#endif
|
||||
|
||||
#ifdef TEST_NATIVE_CLIENT_ON_WINDOWS
|
||||
#else
|
||||
#define RAKNET_SOCKET_2_INLINE_FUNCTIONS
|
||||
#include "RakNetSocket2_360_720.cpp"
|
||||
#include "RakNetSocket2_PS3_PS4.cpp"
|
||||
#include "RakNetSocket2_PS4.cpp"
|
||||
#include "RakNetSocket2_Windows_Linux.cpp"
|
||||
#include "RakNetSocket2_Windows_Linux_360.cpp"
|
||||
#include "RakNetSocket2_Vita.cpp"
|
||||
#include "RakNetSocket2_NativeClient.cpp"
|
||||
#include "RakNetSocket2_Berkley.cpp"
|
||||
#include "RakNetSocket2_Berkley_NativeClient.cpp"
|
||||
#include "RakNetSocket2_WindowsStore8.cpp"
|
||||
#undef RAKNET_SOCKET_2_INLINE_FUNCTIONS
|
||||
|
||||
#endif
|
||||
|
||||
#ifndef INVALID_SOCKET
|
||||
#define INVALID_SOCKET -1
|
||||
#endif
|
||||
|
||||
void RakNetSocket2Allocator::DeallocRNS2(RakNetSocket2 *s) {RakNet::OP_DELETE(s,_FILE_AND_LINE_);}
|
||||
RakNetSocket2::RakNetSocket2() {eventHandler=0;}
|
||||
RakNetSocket2::~RakNetSocket2() {}
|
||||
void RakNetSocket2::SetRecvEventHandler(RNS2EventHandler *_eventHandler) {eventHandler=_eventHandler;}
|
||||
RNS2Type RakNetSocket2::GetSocketType(void) const {return socketType;}
|
||||
void RakNetSocket2::SetSocketType(RNS2Type t) {socketType=t;}
|
||||
bool RakNetSocket2::IsBerkleySocket(void) const {
|
||||
return socketType!=RNS2T_CHROME && socketType!=RNS2T_WINDOWS_STORE_8;
|
||||
}
|
||||
SystemAddress RakNetSocket2::GetBoundAddress(void) const {return boundAddress;}
|
||||
|
||||
RakNetSocket2* RakNetSocket2Allocator::AllocRNS2(void)
|
||||
{
|
||||
RakNetSocket2* s2;
|
||||
#if defined(WINDOWS_STORE_RT)
|
||||
s2 = RakNet::OP_NEW<RNS2_WindowsStore8>(_FILE_AND_LINE_);
|
||||
s2->SetSocketType(RNS2T_WINDOWS_STORE_8);
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
#elif defined(__native_client__)
|
||||
s2 = RakNet::OP_NEW<RNS2_NativeClient>(_FILE_AND_LINE_);
|
||||
s2->SetSocketType(RNS2T_CHROME);
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
#elif defined(_WIN32)
|
||||
s2 = RakNet::OP_NEW<RNS2_Windows>(_FILE_AND_LINE_);
|
||||
s2->SetSocketType(RNS2T_WINDOWS);
|
||||
#else
|
||||
s2 = RakNet::OP_NEW<RNS2_Linux>(_FILE_AND_LINE_);
|
||||
s2->SetSocketType(RNS2T_LINUX);
|
||||
#endif
|
||||
return s2;
|
||||
}
|
||||
void RakNetSocket2::GetMyIP( SystemAddress addresses[MAXIMUM_NUMBER_OF_INTERNAL_IDS] )
|
||||
{
|
||||
#if defined(WINDOWS_STORE_RT)
|
||||
RNS2_WindowsStore8::GetMyIP( addresses );
|
||||
|
||||
|
||||
|
||||
|
||||
#elif defined(__native_client__)
|
||||
RNS2_NativeClient::GetMyIP( addresses );
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
#elif defined(_WIN32)
|
||||
RNS2_Windows::GetMyIP( addresses );
|
||||
#else
|
||||
RNS2_Linux::GetMyIP( addresses );
|
||||
#endif
|
||||
}
|
||||
|
||||
unsigned int RakNetSocket2::GetUserConnectionSocketIndex(void) const {return userConnectionSocketIndex;}
|
||||
void RakNetSocket2::SetUserConnectionSocketIndex(unsigned int i) {userConnectionSocketIndex=i;}
|
||||
RNS2EventHandler * RakNetSocket2::GetEventHandler(void) const {return eventHandler;}
|
||||
|
||||
void RakNetSocket2::DomainNameToIP( const char *domainName, char ip[65] ) {
|
||||
#if defined(WINDOWS_STORE_RT)
|
||||
return RNS2_WindowsStore8::DomainNameToIP( domainName, ip );
|
||||
#elif defined(__native_client__)
|
||||
return DomainNameToIP_Berkley( domainName, ip );
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
#elif defined(_WIN32)
|
||||
return DomainNameToIP_Berkley( domainName, ip );
|
||||
#else
|
||||
return DomainNameToIP_Berkley( domainName, ip );
|
||||
#endif
|
||||
}
|
||||
|
||||
#if defined(WINDOWS_STORE_RT)
|
||||
#elif defined(__native_client__)
|
||||
RNS2_NativeClient::RNS2_NativeClient() {bindState = BS_UNBOUND; sendInProgress=false;}
|
||||
RNS2_NativeClient::~RNS2_NativeClient()
|
||||
{
|
||||
bufferedSendsMutex.Lock();
|
||||
while (bufferedSends.Size())
|
||||
RakNet::OP_DELETE(bufferedSends.Pop(), _FILE_AND_LINE_);
|
||||
bufferedSendsMutex.Unlock();
|
||||
}
|
||||
void RNS2_NativeClient::onSocketBound(void* pData, int32_t dataSize)
|
||||
{
|
||||
RAKNET_DEBUG_PRINTF("onSocketBound ==> %d\n", dataSize);
|
||||
RNS2_NativeClient *csc = (RNS2_NativeClient *)pData;
|
||||
|
||||
//any error codes will be given to us in the dataSize value
|
||||
if(dataSize < 0)
|
||||
{
|
||||
csc->bindState=BS_FAILED;
|
||||
fprintf(stderr,"onSocketBound exiting, dataSize = %d\n", dataSize);
|
||||
return;
|
||||
}
|
||||
|
||||
csc->bindState=BS_BOUND;
|
||||
|
||||
csc->ProcessBufferedSend();
|
||||
csc->IssueReceiveCall();
|
||||
}
|
||||
void RNS2_NativeClient::ProcessBufferedSend(void)
|
||||
{
|
||||
// Don't send until bound
|
||||
if (bindState!=BS_BOUND)
|
||||
return;
|
||||
// Fast non-threadsafe check
|
||||
if (bufferedSends.IsEmpty()==true)
|
||||
return;
|
||||
|
||||
sendInProgressMutex.Lock();
|
||||
if (sendInProgress==true) {sendInProgressMutex.Unlock(); return;}
|
||||
else {sendInProgress=true;}
|
||||
sendInProgressMutex.Unlock();
|
||||
|
||||
RNS2_SendParameters_NativeClient *sp;
|
||||
bufferedSendsMutex.Lock();
|
||||
if (bufferedSends.IsEmpty()==false)
|
||||
sp=bufferedSends.Pop();
|
||||
else
|
||||
sp=0;
|
||||
bufferedSendsMutex.Unlock();
|
||||
if (sp==0)
|
||||
{
|
||||
sendInProgressMutex.Lock();
|
||||
sendInProgress=false;
|
||||
sendInProgressMutex.Unlock();
|
||||
return; // Nothing to send after all
|
||||
}
|
||||
|
||||
SendImmediate(sp);
|
||||
// sp remains in memory until the callback completes
|
||||
// DeallocSP(sp);
|
||||
}
|
||||
void RNS2_NativeClient::DeallocSP(RNS2_SendParameters_NativeClient *sp)
|
||||
{
|
||||
rakFree_Ex(sp->data, _FILE_AND_LINE_);
|
||||
RakNet::OP_DELETE(sp, _FILE_AND_LINE_);
|
||||
}
|
||||
RNS2_SendParameters_NativeClient* RNS2_NativeClient::CloneSP(RNS2_SendParameters *sp, RNS2_NativeClient *socket2, const char *file, unsigned int line)
|
||||
{
|
||||
RNS2_SendParameters_NativeClient *spNew = RakNet::OP_NEW<RNS2_SendParameters_NativeClient>(file, line);
|
||||
spNew->data=(char*) rakMalloc(sp->length);
|
||||
memcpy(spNew->data,sp->data,sp->length);
|
||||
spNew->length = sp->length;
|
||||
spNew->socket2=socket2;
|
||||
spNew->systemAddress=sp->systemAddress;
|
||||
spNew->ttl=0; // Unused
|
||||
return spNew;
|
||||
}
|
||||
void RNS2_NativeClient::onSendTo(void* pData, int32_t dataSize)
|
||||
{
|
||||
if(dataSize <= 0)
|
||||
RAKNET_DEBUG_PRINTF("onSendTo: send failed with error %d\n", dataSize);
|
||||
|
||||
RNS2_SendParameters_NativeClient *sp = (RNS2_SendParameters_NativeClient*) pData;
|
||||
|
||||
// Caller will check sendInProgress to send again if needed
|
||||
sp->socket2->sendInProgressMutex.Lock();
|
||||
sp->socket2->sendInProgress=false;
|
||||
sp->socket2->sendInProgressMutex.Unlock();
|
||||
|
||||
DeallocSP(sp);
|
||||
|
||||
// if(dataSize == PP_ERROR_ABORTED)
|
||||
// return;
|
||||
}
|
||||
RNS2SendResult RNS2_NativeClient::Send( RNS2_SendParameters *sendParameters, const char *file, unsigned int line )
|
||||
{
|
||||
if (bindState==BS_FAILED)
|
||||
return -1;
|
||||
|
||||
// This is called from multiple threads. Always buffer the send, until native client is threadsafe
|
||||
BufferSend(sendParameters, file, line);
|
||||
return sendParameters->length;
|
||||
}
|
||||
void RNS2_NativeClient::BufferSend( RNS2_SendParameters *sendParameters, const char *file, unsigned int line )
|
||||
{
|
||||
if (bindState==BS_FAILED)
|
||||
return;
|
||||
|
||||
RNS2_SendParameters_NativeClient* sp = CloneSP(sendParameters, this, file, line);
|
||||
bufferedSendsMutex.Lock();
|
||||
bufferedSends.Push(sp, file, line);
|
||||
bufferedSendsMutex.Unlock();
|
||||
|
||||
// Do not check to send immediately, because this was probably invoked from a thread and native client is not threadsafe
|
||||
}
|
||||
void RNS2_NativeClient::GetMyIP( SystemAddress addresses[MAXIMUM_NUMBER_OF_INTERNAL_IDS] ) {addresses[0]=UNASSIGNED_SYSTEM_ADDRESS; RakAssert("GetMyIP Unsupported?" && 0);}
|
||||
const NativeClientBindParameters *RNS2_NativeClient::GetBindings(void) const {return &binding;}
|
||||
void RNS2_NativeClient::Update(void)
|
||||
{
|
||||
// Don't send until bound
|
||||
if (bindState==BS_BOUND)
|
||||
{
|
||||
do
|
||||
{
|
||||
ProcessBufferedSend();
|
||||
} while (sendInProgress==false && bufferedSends.Size()>1);
|
||||
}
|
||||
}
|
||||
|
||||
#else // defined(__native_client__)
|
||||
bool IRNS2_Berkley::IsPortInUse(unsigned short port, const char *hostAddress, unsigned short addressFamily, int type ) {
|
||||
RNS2_BerkleyBindParameters bbp;
|
||||
bbp.remotePortRakNetWasStartedOn_PS3_PS4_PSP2=0;
|
||||
bbp.port=port; bbp.hostAddress=(char*) hostAddress; bbp.addressFamily=addressFamily;
|
||||
bbp.type=type; bbp.protocol=0; bbp.nonBlockingSocket=false;
|
||||
bbp.setBroadcast=false; bbp.doNotFragment=false; bbp.protocol=0;
|
||||
bbp.setIPHdrIncl=false;
|
||||
SystemAddress boundAddress;
|
||||
RNS2_Berkley *rns2 = (RNS2_Berkley*) RakNetSocket2Allocator::AllocRNS2();
|
||||
RNS2BindResult bindResult = rns2->Bind(&bbp, _FILE_AND_LINE_);
|
||||
RakNetSocket2Allocator::DeallocRNS2(rns2);
|
||||
return bindResult==BR_FAILED_TO_BIND_SOCKET;
|
||||
}
|
||||
|
||||
#if defined(__APPLE__)
|
||||
void SocketReadCallback(CFSocketRef s, CFSocketCallBackType type, CFDataRef address, const void *data, void *info)
|
||||
// This C routine is called by CFSocket when there's data waiting on our
|
||||
// UDP socket. It just redirects the call to Objective-C code.
|
||||
{ }
|
||||
#endif
|
||||
|
||||
RNS2BindResult RNS2_Berkley::BindShared( RNS2_BerkleyBindParameters *bindParameters, const char *file, unsigned int line ) {
|
||||
RNS2BindResult br;
|
||||
#if RAKNET_SUPPORT_IPV6==1
|
||||
br=BindSharedIPV4And6(bindParameters, file, line);
|
||||
#else
|
||||
br=BindSharedIPV4(bindParameters, file, line);
|
||||
#endif
|
||||
|
||||
if (br!=BR_SUCCESS)
|
||||
return br;
|
||||
|
||||
unsigned long zero=0;
|
||||
RNS2_SendParameters bsp;
|
||||
bsp.data=(char*) &zero;
|
||||
bsp.length=4;
|
||||
bsp.systemAddress=boundAddress;
|
||||
bsp.ttl=0;
|
||||
RNS2SendResult sr = Send(&bsp, _FILE_AND_LINE_);
|
||||
if (sr<0)
|
||||
return BR_FAILED_SEND_TEST;
|
||||
|
||||
memcpy(&binding, bindParameters, sizeof(RNS2_BerkleyBindParameters));
|
||||
|
||||
/*
|
||||
#if defined(__APPLE__)
|
||||
const CFSocketContext context = { 0, this, NULL, NULL, NULL };
|
||||
_cfSocket = CFSocketCreateWithNative(NULL, rns2Socket, kCFSocketReadCallBack, SocketReadCallback, &context);
|
||||
#endif
|
||||
*/
|
||||
|
||||
return br;
|
||||
}
|
||||
|
||||
RAK_THREAD_DECLARATION(RNS2_Berkley::RecvFromLoop)
|
||||
{
|
||||
|
||||
|
||||
|
||||
RNS2_Berkley *b = ( RNS2_Berkley * ) arguments;
|
||||
|
||||
b->RecvFromLoopInt();
|
||||
return 0;
|
||||
}
|
||||
unsigned RNS2_Berkley::RecvFromLoopInt(void)
|
||||
{
|
||||
isRecvFromLoopThreadActive.Increment();
|
||||
|
||||
while ( endThreads == false )
|
||||
{
|
||||
RNS2RecvStruct *recvFromStruct;
|
||||
recvFromStruct=binding.eventHandler->AllocRNS2RecvStruct(_FILE_AND_LINE_);
|
||||
if (recvFromStruct != NULL)
|
||||
{
|
||||
recvFromStruct->socket=this;
|
||||
RecvFromBlocking(recvFromStruct);
|
||||
|
||||
if (recvFromStruct->bytesRead>0)
|
||||
{
|
||||
RakAssert(recvFromStruct->systemAddress.GetPort());
|
||||
binding.eventHandler->OnRNS2Recv(recvFromStruct);
|
||||
}
|
||||
else
|
||||
{
|
||||
RakSleep(0);
|
||||
binding.eventHandler->DeallocRNS2RecvStruct(recvFromStruct, _FILE_AND_LINE_);
|
||||
}
|
||||
}
|
||||
}
|
||||
isRecvFromLoopThreadActive.Decrement();
|
||||
|
||||
|
||||
|
||||
|
||||
return 0;
|
||||
|
||||
}
|
||||
RNS2_Berkley::RNS2_Berkley()
|
||||
{
|
||||
rns2Socket=(RNS2Socket)INVALID_SOCKET;
|
||||
}
|
||||
RNS2_Berkley::~RNS2_Berkley()
|
||||
{
|
||||
if (rns2Socket!=INVALID_SOCKET)
|
||||
{
|
||||
/*
|
||||
#if defined(__APPLE__)
|
||||
CFSocketInvalidate(_cfSocket);
|
||||
#endif
|
||||
*/
|
||||
|
||||
closesocket__(rns2Socket);
|
||||
}
|
||||
|
||||
}
|
||||
int RNS2_Berkley::CreateRecvPollingThread(int threadPriority)
|
||||
{
|
||||
endThreads=false;
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
int errorCode = RakNet::RakThread::Create(RecvFromLoop, this, threadPriority);
|
||||
|
||||
return errorCode;
|
||||
}
|
||||
void RNS2_Berkley::SignalStopRecvPollingThread(void)
|
||||
{
|
||||
endThreads=true;
|
||||
}
|
||||
void RNS2_Berkley::BlockOnStopRecvPollingThread(void)
|
||||
{
|
||||
endThreads=true;
|
||||
|
||||
// Get recvfrom to unblock
|
||||
RNS2_SendParameters bsp;
|
||||
unsigned long zero=0;
|
||||
bsp.data=(char*) &zero;
|
||||
bsp.length=4;
|
||||
bsp.systemAddress=boundAddress;
|
||||
bsp.ttl=0;
|
||||
Send(&bsp, _FILE_AND_LINE_);
|
||||
|
||||
RakNet::TimeMS timeout = RakNet::GetTimeMS()+1000;
|
||||
while ( isRecvFromLoopThreadActive.GetValue()>0 && RakNet::GetTimeMS()<timeout )
|
||||
{
|
||||
// Get recvfrom to unblock
|
||||
Send(&bsp, _FILE_AND_LINE_);
|
||||
RakSleep(30);
|
||||
}
|
||||
}
|
||||
const RNS2_BerkleyBindParameters *RNS2_Berkley::GetBindings(void) const {return &binding;}
|
||||
RNS2Socket RNS2_Berkley::GetSocket(void) const {return rns2Socket;}
|
||||
// See RakNetSocket2_Berkley.cpp for WriteSharedIPV4, BindSharedIPV4And6 and other implementations
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
#if defined(_WIN32)
|
||||
RNS2_Windows::RNS2_Windows() {slo=0;}
|
||||
RNS2_Windows::~RNS2_Windows() {}
|
||||
RNS2BindResult RNS2_Windows::Bind( RNS2_BerkleyBindParameters *bindParameters, const char *file, unsigned int line ) {
|
||||
RNS2BindResult bindResult = BindShared(bindParameters, file, line);
|
||||
if (bindResult == BR_FAILED_TO_BIND_SOCKET)
|
||||
{
|
||||
// Sometimes windows will fail if the socket is recreated too quickly
|
||||
RakSleep(100);
|
||||
bindResult = BindShared(bindParameters, file, line);
|
||||
}
|
||||
return bindResult;
|
||||
}
|
||||
RNS2SendResult RNS2_Windows::Send( RNS2_SendParameters *sendParameters, const char *file, unsigned int line ) {
|
||||
if (slo)
|
||||
{
|
||||
RNS2SendResult len;
|
||||
len = slo->RakNetSendTo(sendParameters->data, sendParameters->length,sendParameters->systemAddress);
|
||||
if (len>=0)
|
||||
return len;
|
||||
}
|
||||
return Send_Windows_Linux_360NoVDP(rns2Socket,sendParameters, file, line);
|
||||
}
|
||||
void RNS2_Windows::GetMyIP( SystemAddress addresses[MAXIMUM_NUMBER_OF_INTERNAL_IDS] ) {return GetMyIP_Windows_Linux(addresses);}
|
||||
void RNS2_Windows::SetSocketLayerOverride(SocketLayerOverride *_slo) {slo = _slo;}
|
||||
SocketLayerOverride* RNS2_Windows::GetSocketLayerOverride(void) {return slo;}
|
||||
#else
|
||||
RNS2BindResult RNS2_Linux::Bind( RNS2_BerkleyBindParameters *bindParameters, const char *file, unsigned int line ) {return BindShared(bindParameters, file, line);}
|
||||
RNS2SendResult RNS2_Linux::Send( RNS2_SendParameters *sendParameters, const char *file, unsigned int line ) {return Send_Windows_Linux_360NoVDP(rns2Socket,sendParameters, file, line);}
|
||||
void RNS2_Linux::GetMyIP( SystemAddress addresses[MAXIMUM_NUMBER_OF_INTERNAL_IDS] ) {return GetMyIP_Windows_Linux(addresses);}
|
||||
#endif // Linux
|
||||
|
||||
#endif // defined(__native_client__)
|
||||
Reference in New Issue
Block a user