platformNet.cpp 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050
  1. //-----------------------------------------------------------------------------
  2. // Copyright (c) 2012 GarageGames, LLC
  3. //
  4. // Permission is hereby granted, free of charge, to any person obtaining a copy
  5. // of this software and associated documentation files (the "Software"), to
  6. // deal in the Software without restriction, including without limitation the
  7. // rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
  8. // sell copies of the Software, and to permit persons to whom the Software is
  9. // furnished to do so, subject to the following conditions:
  10. //
  11. // The above copyright notice and this permission notice shall be included in
  12. // all copies or substantial portions of the Software.
  13. //
  14. // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  15. // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  16. // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  17. // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  18. // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  19. // FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
  20. // IN THE SOFTWARE.
  21. //-----------------------------------------------------------------------------
  22. #include "platform/platformNet.h"
  23. #include "core/strings/stringFunctions.h"
  24. #if defined (TORQUE_OS_WIN32)
  25. #define TORQUE_USE_WINSOCK
  26. #include <errno.h>
  27. #include <winsock.h>
  28. #define EINPROGRESS WSAEINPROGRESS
  29. #define ioctl ioctlsocket
  30. typedef int socklen_t;
  31. #elif defined ( TORQUE_OS_MAC )
  32. #include <unistd.h>
  33. #include <sys/types.h>
  34. #include <sys/socket.h>
  35. #include <sys/poll.h>
  36. #include <arpa/inet.h>
  37. #include <netdb.h>
  38. #include <netinet/in.h>
  39. #include <errno.h>
  40. #include <sys/ioctl.h>
  41. typedef sockaddr_in SOCKADDR_IN;
  42. typedef sockaddr * PSOCKADDR;
  43. typedef sockaddr SOCKADDR;
  44. typedef in_addr IN_ADDR;
  45. #define INVALID_SOCKET -1
  46. #define SOCKET_ERROR -1
  47. #define closesocket close
  48. #elif defined TORQUE_OS_LINUX
  49. #include <unistd.h>
  50. #include <sys/types.h>
  51. #include <sys/socket.h>
  52. #include <sys/poll.h>
  53. #include <arpa/inet.h>
  54. #include <netdb.h>
  55. #include <netinet/in.h>
  56. #include <errno.h>
  57. #include <sys/ioctl.h>
  58. typedef sockaddr_in SOCKADDR_IN;
  59. typedef sockaddr * PSOCKADDR;
  60. typedef sockaddr SOCKADDR;
  61. typedef in_addr IN_ADDR;
  62. #define INVALID_SOCKET -1
  63. #define SOCKET_ERROR -1
  64. #define closesocket close
  65. #elif defined( TORQUE_OS_XENON )
  66. #include <Xtl.h>
  67. #include <string>
  68. #define TORQUE_USE_WINSOCK
  69. #define EINPROGRESS WSAEINPROGRESS
  70. #define ioctl ioctlsocket
  71. typedef int socklen_t;
  72. DWORD _getLastErrorAndClear()
  73. {
  74. DWORD err = WSAGetLastError();
  75. WSASetLastError( 0 );
  76. return err;
  77. }
  78. #else
  79. #endif
  80. #if defined(TORQUE_USE_WINSOCK)
  81. static const char* strerror_wsa( int code )
  82. {
  83. switch( code )
  84. {
  85. #define E( name ) case name: return #name;
  86. E( WSANOTINITIALISED );
  87. E( WSAENETDOWN );
  88. E( WSAEADDRINUSE );
  89. E( WSAEINPROGRESS );
  90. E( WSAEALREADY );
  91. E( WSAEADDRNOTAVAIL );
  92. E( WSAEAFNOSUPPORT );
  93. E( WSAEFAULT );
  94. E( WSAEINVAL );
  95. E( WSAEISCONN );
  96. E( WSAENETUNREACH );
  97. E( WSAEHOSTUNREACH );
  98. E( WSAENOBUFS );
  99. E( WSAENOTSOCK );
  100. E( WSAETIMEDOUT );
  101. E( WSAEWOULDBLOCK );
  102. E( WSAEACCES );
  103. #undef E
  104. default:
  105. return "Unknown";
  106. }
  107. }
  108. #endif
  109. #include "core/util/tVector.h"
  110. #include "platform/platformNetAsync.h"
  111. #include "console/console.h"
  112. #include "core/util/journal/process.h"
  113. #include "core/util/journal/journal.h"
  114. static Net::Error getLastError();
  115. static S32 defaultPort = 28000;
  116. static S32 netPort = 0;
  117. static int udpSocket = InvalidSocket;
  118. ConnectionNotifyEvent Net::smConnectionNotify;
  119. ConnectionAcceptedEvent Net::smConnectionAccept;
  120. ConnectionReceiveEvent Net::smConnectionReceive;
  121. PacketReceiveEvent Net::smPacketReceive;
  122. // local enum for socket states for polled sockets
  123. enum SocketState
  124. {
  125. InvalidState,
  126. Connected,
  127. ConnectionPending,
  128. Listening,
  129. NameLookupRequired
  130. };
  131. // the Socket structure helps us keep track of the
  132. // above states
  133. struct Socket
  134. {
  135. Socket()
  136. {
  137. fd = InvalidSocket;
  138. state = InvalidState;
  139. remoteAddr[0] = 0;
  140. remotePort = -1;
  141. }
  142. NetSocket fd;
  143. S32 state;
  144. char remoteAddr[256];
  145. S32 remotePort;
  146. };
  147. // list of polled sockets
  148. static Vector<Socket*> gPolledSockets( __FILE__, __LINE__ );
  149. static Socket* addPolledSocket(NetSocket& fd, S32 state,
  150. char* remoteAddr = NULL, S32 port = -1)
  151. {
  152. Socket* sock = new Socket();
  153. sock->fd = fd;
  154. sock->state = state;
  155. if (remoteAddr)
  156. dStrcpy(sock->remoteAddr, remoteAddr);
  157. if (port != -1)
  158. sock->remotePort = port;
  159. gPolledSockets.push_back(sock);
  160. return sock;
  161. }
  162. enum {
  163. MaxConnections = 1024,
  164. };
  165. bool netSocketWaitForWritable(NetSocket fd, S32 timeoutMs)
  166. {
  167. fd_set writefds;
  168. timeval timeout;
  169. FD_ZERO(&writefds);
  170. FD_SET( fd, &writefds );
  171. timeout.tv_sec = timeoutMs / 1000;
  172. timeout.tv_usec = ( timeoutMs % 1000 ) * 1000;
  173. if( select(fd + 1, NULL, &writefds, NULL, &timeout) > 0 )
  174. return true;
  175. return false;
  176. }
  177. static S32 initCount = 0;
  178. bool Net::init()
  179. {
  180. #if defined(TORQUE_USE_WINSOCK)
  181. if(!initCount)
  182. {
  183. #ifdef TORQUE_OS_XENON
  184. // Configure startup parameters
  185. XNetStartupParams xnsp;
  186. memset( &xnsp, 0, sizeof( xnsp ) );
  187. xnsp.cfgSizeOfStruct = sizeof( XNetStartupParams );
  188. #ifndef TORQUE_DISABLE_PC_CONNECTIVITY
  189. xnsp.cfgFlags = XNET_STARTUP_BYPASS_SECURITY;
  190. Con::warnf("XNET_STARTUP_BYPASS_SECURITY enabled! This build can talk to PCs!");
  191. #endif
  192. AssertISV( !XNetStartup( &xnsp ), "Net::init - failed to init XNet" );
  193. #endif
  194. WSADATA stWSAData;
  195. AssertISV( !WSAStartup( 0x0101, &stWSAData ), "Net::init - failed to init WinSock!" );
  196. //logprintf("Winsock initialization %s", success ? "succeeded." : "failed!");
  197. }
  198. #endif
  199. initCount++;
  200. Process::notify(&Net::process, PROCESS_NET_ORDER);
  201. return(true);
  202. }
  203. void Net::shutdown()
  204. {
  205. Process::remove(&Net::process);
  206. while (gPolledSockets.size() > 0)
  207. closeConnectTo(gPolledSockets[0]->fd);
  208. closePort();
  209. initCount--;
  210. #if defined(TORQUE_USE_WINSOCK)
  211. if(!initCount)
  212. {
  213. WSACleanup();
  214. #ifdef TORQUE_OS_XENON
  215. XNetCleanup();
  216. #endif
  217. }
  218. #endif
  219. }
  220. Net::Error getLastError()
  221. {
  222. #if defined(TORQUE_USE_WINSOCK)
  223. S32 err = WSAGetLastError();
  224. switch(err)
  225. {
  226. case 0:
  227. return Net::NoError;
  228. case WSAEWOULDBLOCK:
  229. return Net::WouldBlock;
  230. default:
  231. return Net::UnknownError;
  232. }
  233. #else
  234. if (errno == EAGAIN)
  235. return Net::WouldBlock;
  236. if (errno == 0)
  237. return Net::NoError;
  238. return Net::UnknownError;
  239. #endif
  240. }
  241. static void netToIPSocketAddress(const NetAddress *address, struct sockaddr_in *sockAddr)
  242. {
  243. dMemset(sockAddr, 0, sizeof(struct sockaddr_in));
  244. sockAddr->sin_family = AF_INET;
  245. sockAddr->sin_port = htons(address->port);
  246. char tAddr[20];
  247. dSprintf(tAddr, 20, "%d.%d.%d.%d", address->netNum[0], address->netNum[1], address->netNum[2], address->netNum[3]);
  248. //fprintf(stdout,"netToIPSocketAddress(): %s\n",tAddr);fflush(NULL);
  249. sockAddr->sin_addr.s_addr = inet_addr(tAddr);
  250. }
  251. static void IPSocketToNetAddress(const struct sockaddr_in *sockAddr, NetAddress *address)
  252. {
  253. address->type = NetAddress::IPAddress;
  254. address->port = htons(sockAddr->sin_port);
  255. #ifndef TORQUE_OS_XENON
  256. char *tAddr;
  257. tAddr = inet_ntoa(sockAddr->sin_addr);
  258. //fprintf(stdout,"IPSocketToNetAddress(): %s\n",tAddr);fflush(NULL);
  259. U8 nets[4];
  260. nets[0] = atoi(strtok(tAddr, "."));
  261. nets[1] = atoi(strtok(NULL, "."));
  262. nets[2] = atoi(strtok(NULL, "."));
  263. nets[3] = atoi(strtok(NULL, "."));
  264. //fprintf(stdout,"0 = %d, 1 = %d, 2 = %d, 3 = %d\n", nets[0], nets[1], nets[2], nets[3]);
  265. address->netNum[0] = nets[0];
  266. address->netNum[1] = nets[1];
  267. address->netNum[2] = nets[2];
  268. address->netNum[3] = nets[3];
  269. #else
  270. address->netNum[0] = sockAddr->sin_addr.s_net;
  271. address->netNum[1] = sockAddr->sin_addr.s_host;
  272. address->netNum[2] = sockAddr->sin_addr.s_lh;
  273. address->netNum[3] = sockAddr->sin_addr.s_impno;
  274. #endif
  275. }
  276. NetSocket Net::openListenPort(U16 port)
  277. {
  278. if(Journal::IsPlaying())
  279. {
  280. U32 ret;
  281. Journal::Read(&ret);
  282. return NetSocket(ret);
  283. }
  284. NetSocket sock = openSocket();
  285. if (sock == InvalidSocket)
  286. {
  287. Con::errorf("Unable to open listen socket: %s", strerror(errno));
  288. return InvalidSocket;
  289. }
  290. if (bind(sock, port) != NoError)
  291. {
  292. Con::errorf("Unable to bind port %d: %s", port, strerror(errno));
  293. ::closesocket(sock);
  294. return InvalidSocket;
  295. }
  296. if (listen(sock, 4) != NoError)
  297. {
  298. Con::errorf("Unable to listen on port %d: %s", port, strerror(errno));
  299. ::closesocket(sock);
  300. return InvalidSocket;
  301. }
  302. setBlocking(sock, false);
  303. addPolledSocket(sock, Listening);
  304. if(Journal::IsRecording())
  305. Journal::Write(U32(sock));
  306. return sock;
  307. }
  308. NetSocket Net::openConnectTo(const char *addressString)
  309. {
  310. if(!dStrnicmp(addressString, "ipx:", 4))
  311. // ipx support deprecated
  312. return InvalidSocket;
  313. if(!dStrnicmp(addressString, "ip:", 3))
  314. addressString += 3; // eat off the ip:
  315. char remoteAddr[256];
  316. dStrcpy(remoteAddr, addressString);
  317. char *portString = dStrchr(remoteAddr, ':');
  318. U16 port;
  319. if(portString)
  320. {
  321. *portString++ = 0;
  322. port = htons(dAtoi(portString));
  323. }
  324. else
  325. port = htons(defaultPort);
  326. if(!dStricmp(remoteAddr, "broadcast"))
  327. return InvalidSocket;
  328. if(Journal::IsPlaying())
  329. {
  330. U32 ret;
  331. Journal::Read(&ret);
  332. return NetSocket(ret);
  333. }
  334. NetSocket sock = openSocket();
  335. setBlocking(sock, false);
  336. sockaddr_in ipAddr;
  337. dMemset(&ipAddr, 0, sizeof(ipAddr));
  338. ipAddr.sin_addr.s_addr = inet_addr(remoteAddr);
  339. if(ipAddr.sin_addr.s_addr != INADDR_NONE)
  340. {
  341. ipAddr.sin_port = port;
  342. ipAddr.sin_family = AF_INET;
  343. if(::connect(sock, (struct sockaddr *)&ipAddr, sizeof(ipAddr)) == -1)
  344. {
  345. S32 err = getLastError();
  346. if(err != Net::WouldBlock)
  347. {
  348. Con::errorf("Error connecting %s: %s",
  349. addressString, strerror(err));
  350. ::closesocket(sock);
  351. sock = InvalidSocket;
  352. }
  353. }
  354. if(sock != InvalidSocket)
  355. {
  356. // add this socket to our list of polled sockets
  357. addPolledSocket(sock, ConnectionPending);
  358. }
  359. }
  360. else
  361. {
  362. // need to do an asynchronous name lookup. first, add the socket
  363. // to the polled list
  364. addPolledSocket(sock, NameLookupRequired, remoteAddr, port);
  365. // queue the lookup
  366. gNetAsync.queueLookup(remoteAddr, sock);
  367. }
  368. if(Journal::IsRecording())
  369. Journal::Write(U32(sock));
  370. return sock;
  371. }
  372. void Net::closeConnectTo(NetSocket sock)
  373. {
  374. if(Journal::IsPlaying())
  375. return;
  376. // if this socket is in the list of polled sockets, remove it
  377. for (int i = 0; i < gPolledSockets.size(); ++i)
  378. {
  379. if (gPolledSockets[i]->fd == sock)
  380. {
  381. delete gPolledSockets[i];
  382. gPolledSockets.erase(i);
  383. break;
  384. }
  385. }
  386. closeSocket(sock);
  387. }
  388. Net::Error Net::sendtoSocket(NetSocket socket, const U8 *buffer, int bufferSize)
  389. {
  390. if(Journal::IsPlaying())
  391. {
  392. U32 e;
  393. Journal::Read(&e);
  394. return (Net::Error) e;
  395. }
  396. Net::Error e = send(socket, buffer, bufferSize);
  397. if(Journal::IsRecording())
  398. Journal::Write(U32(e));
  399. return e;
  400. }
  401. bool Net::openPort(S32 port, bool doBind)
  402. {
  403. if(udpSocket != InvalidSocket)
  404. ::closesocket(udpSocket);
  405. // we turn off VDP in non-release builds because VDP does not support broadcast packets
  406. // which are required for LAN queries (PC->Xbox connectivity). The wire protocol still
  407. // uses the VDP packet structure, though.
  408. int protocol = 0;
  409. bool useVDP = false;
  410. #ifdef TORQUE_DISABLE_PC_CONNECTIVITY
  411. // Xbox uses a VDP (voice/data protocol) socket for networking
  412. protocol = IPPROTO_VDP;
  413. useVDP = true;
  414. #endif
  415. udpSocket = socket(AF_INET, SOCK_DGRAM, protocol);
  416. if(udpSocket != InvalidSocket)
  417. {
  418. Net::Error error = NoError;
  419. if (doBind)
  420. {
  421. error = bind(udpSocket, port);
  422. }
  423. if(error == NoError)
  424. error = setBufferSize(udpSocket, 32768);
  425. if(error == NoError && !useVDP)
  426. error = setBroadcast(udpSocket, true);
  427. if(error == NoError)
  428. error = setBlocking(udpSocket, false);
  429. if(error == NoError)
  430. Con::printf("UDP initialized on port %d", port);
  431. else
  432. {
  433. ::closesocket(udpSocket);
  434. udpSocket = InvalidSocket;
  435. Con::printf("Unable to initialize UDP - error %d", error);
  436. }
  437. }
  438. netPort = port;
  439. return udpSocket != InvalidSocket;
  440. }
  441. NetSocket Net::getPort()
  442. {
  443. return udpSocket;
  444. }
  445. void Net::closePort()
  446. {
  447. if(udpSocket != InvalidSocket)
  448. ::closesocket(udpSocket);
  449. }
  450. Net::Error Net::sendto(const NetAddress *address, const U8 *buffer, S32 bufferSize)
  451. {
  452. if(Journal::IsPlaying())
  453. return NoError;
  454. if(address->type == NetAddress::IPAddress)
  455. {
  456. sockaddr_in ipAddr;
  457. netToIPSocketAddress(address, &ipAddr);
  458. if(::sendto(udpSocket, (const char*)buffer, bufferSize, 0,
  459. (sockaddr *) &ipAddr, sizeof(sockaddr_in)) == SOCKET_ERROR)
  460. return getLastError();
  461. else
  462. return NoError;
  463. }
  464. else
  465. {
  466. SOCKADDR_IN ipAddr;
  467. netToIPSocketAddress(address, &ipAddr);
  468. if(::sendto(udpSocket, (const char*)buffer, bufferSize, 0,
  469. (PSOCKADDR) &ipAddr, sizeof(SOCKADDR_IN)) == SOCKET_ERROR)
  470. return getLastError();
  471. else
  472. return NoError;
  473. }
  474. }
  475. void Net::process()
  476. {
  477. sockaddr sa;
  478. sa.sa_family = AF_UNSPEC;
  479. NetAddress srcAddress;
  480. RawData tmpBuffer;
  481. tmpBuffer.alloc(MaxPacketDataSize);
  482. for(;;)
  483. {
  484. socklen_t addrLen = sizeof(sa);
  485. S32 bytesRead = -1;
  486. if(udpSocket != InvalidSocket)
  487. bytesRead = recvfrom(udpSocket, (char *) tmpBuffer.data, MaxPacketDataSize, 0, &sa, &addrLen);
  488. if(bytesRead == -1)
  489. break;
  490. if(sa.sa_family == AF_INET)
  491. IPSocketToNetAddress((sockaddr_in *) &sa, &srcAddress);
  492. else
  493. continue;
  494. if(bytesRead <= 0)
  495. continue;
  496. if(srcAddress.type == NetAddress::IPAddress &&
  497. srcAddress.netNum[0] == 127 &&
  498. srcAddress.netNum[1] == 0 &&
  499. srcAddress.netNum[2] == 0 &&
  500. srcAddress.netNum[3] == 1 &&
  501. srcAddress.port == netPort)
  502. continue;
  503. tmpBuffer.size = bytesRead;
  504. Net::smPacketReceive.trigger(srcAddress, tmpBuffer);
  505. }
  506. // process the polled sockets. This blob of code performs functions
  507. // similar to WinsockProc in winNet.cc
  508. if (gPolledSockets.size() == 0)
  509. return;
  510. S32 optval;
  511. socklen_t optlen = sizeof(S32);
  512. S32 bytesRead;
  513. Net::Error err;
  514. bool removeSock = false;
  515. Socket *currentSock = NULL;
  516. sockaddr_in ipAddr;
  517. NetSocket incoming = InvalidSocket;
  518. char out_h_addr[1024];
  519. int out_h_length = 0;
  520. RawData readBuff;
  521. for (S32 i = 0; i < gPolledSockets.size();
  522. /* no increment, this is done at end of loop body */)
  523. {
  524. removeSock = false;
  525. currentSock = gPolledSockets[i];
  526. switch (currentSock->state)
  527. {
  528. case ::InvalidState:
  529. Con::errorf("Error, InvalidState socket in polled sockets list");
  530. break;
  531. case ::ConnectionPending:
  532. // see if it is now connected
  533. #ifdef TORQUE_OS_XENON
  534. // WSASetLastError has no return value, however part of the SO_ERROR behavior
  535. // is to clear the last error, so this needs to be done here.
  536. if( ( optval = _getLastErrorAndClear() ) == -1 )
  537. #else
  538. if (getsockopt(currentSock->fd, SOL_SOCKET, SO_ERROR,
  539. (char*)&optval, &optlen) == -1)
  540. #endif
  541. {
  542. Con::errorf("Error getting socket options: %s", strerror(errno));
  543. Net::smConnectionNotify.trigger(currentSock->fd, Net::ConnectFailed);
  544. removeSock = true;
  545. }
  546. else
  547. {
  548. if (optval == EINPROGRESS)
  549. // still connecting...
  550. break;
  551. if (optval == 0)
  552. {
  553. // poll for writable status to be sure we're connected.
  554. bool ready = netSocketWaitForWritable(currentSock->fd,0);
  555. if(!ready)
  556. break;
  557. currentSock->state = ::Connected;
  558. Net::smConnectionNotify.trigger(currentSock->fd, Net::Connected);
  559. }
  560. else
  561. {
  562. // some kind of error
  563. Con::errorf("Error connecting: %s", strerror(errno));
  564. Net::smConnectionNotify.trigger(currentSock->fd, Net::ConnectFailed);
  565. removeSock = true;
  566. }
  567. }
  568. break;
  569. case ::Connected:
  570. // try to get some data
  571. bytesRead = 0;
  572. readBuff.alloc(MaxPacketDataSize);
  573. err = Net::recv(currentSock->fd, (U8*)readBuff.data, MaxPacketDataSize, &bytesRead);
  574. if(err == Net::NoError)
  575. {
  576. if (bytesRead > 0)
  577. {
  578. // got some data, post it
  579. readBuff.size = bytesRead;
  580. Net::smConnectionReceive.trigger(currentSock->fd, readBuff);
  581. }
  582. else
  583. {
  584. // ack! this shouldn't happen
  585. if (bytesRead < 0)
  586. Con::errorf("Unexpected error on socket: %s", strerror(errno));
  587. // zero bytes read means EOF
  588. Net::smConnectionNotify.trigger(currentSock->fd, Net::Disconnected);
  589. removeSock = true;
  590. }
  591. }
  592. else if (err != Net::NoError && err != Net::WouldBlock)
  593. {
  594. Con::errorf("Error reading from socket: %s", strerror(errno));
  595. Net::smConnectionNotify.trigger(currentSock->fd, Net::Disconnected);
  596. removeSock = true;
  597. }
  598. break;
  599. case ::NameLookupRequired:
  600. // is the lookup complete?
  601. if (!gNetAsync.checkLookup(
  602. currentSock->fd, out_h_addr, &out_h_length,
  603. sizeof(out_h_addr)))
  604. break;
  605. U32 newState;
  606. if (out_h_length == -1)
  607. {
  608. Con::errorf("DNS lookup failed: %s", currentSock->remoteAddr);
  609. newState = Net::DNSFailed;
  610. removeSock = true;
  611. }
  612. else
  613. {
  614. // try to connect
  615. dMemcpy(&(ipAddr.sin_addr.s_addr), out_h_addr, out_h_length);
  616. ipAddr.sin_port = currentSock->remotePort;
  617. ipAddr.sin_family = AF_INET;
  618. if(::connect(currentSock->fd, (struct sockaddr *)&ipAddr,
  619. sizeof(ipAddr)) == -1)
  620. {
  621. int errorCode;
  622. #if defined(TORQUE_USE_WINSOCK)
  623. errorCode = WSAGetLastError();
  624. if( errorCode == WSAEINPROGRESS || errorCode == WSAEWOULDBLOCK )
  625. #else
  626. errorCode = errno;
  627. if (errno == EINPROGRESS)
  628. #endif
  629. {
  630. newState = Net::DNSResolved;
  631. currentSock->state = ::ConnectionPending;
  632. }
  633. else
  634. {
  635. const char* errorString;
  636. #if defined(TORQUE_USE_WINSOCK)
  637. errorString = strerror_wsa( errorCode );
  638. #else
  639. errorString = strerror( errorCode );
  640. #endif
  641. Con::errorf("Error connecting to %s: %s (%i)",
  642. currentSock->remoteAddr, errorString, errorCode);
  643. newState = Net::ConnectFailed;
  644. removeSock = true;
  645. }
  646. }
  647. else
  648. {
  649. newState = Net::Connected;
  650. currentSock->state = Net::Connected;
  651. }
  652. }
  653. Net::smConnectionNotify.trigger(currentSock->fd, newState);
  654. break;
  655. case ::Listening:
  656. NetAddress incomingAddy;
  657. incoming = Net::accept(currentSock->fd, &incomingAddy);
  658. if(incoming != InvalidSocket)
  659. {
  660. setBlocking(incoming, false);
  661. addPolledSocket(incoming, Connected);
  662. Net::smConnectionAccept.trigger(currentSock->fd, incoming, incomingAddy);
  663. }
  664. break;
  665. }
  666. // only increment index if we're not removing the connection, since
  667. // the removal will shift the indices down by one
  668. if (removeSock)
  669. closeConnectTo(currentSock->fd);
  670. else
  671. i++;
  672. }
  673. }
  674. NetSocket Net::openSocket()
  675. {
  676. int retSocket;
  677. retSocket = socket(AF_INET, SOCK_STREAM, 0);
  678. if(retSocket == InvalidSocket)
  679. return InvalidSocket;
  680. else
  681. return retSocket;
  682. }
  683. Net::Error Net::closeSocket(NetSocket socket)
  684. {
  685. if(socket != InvalidSocket)
  686. {
  687. if(!closesocket(socket))
  688. return NoError;
  689. else
  690. return getLastError();
  691. }
  692. else
  693. return NotASocket;
  694. }
  695. Net::Error Net::connect(NetSocket socket, const NetAddress *address)
  696. {
  697. if(address->type != NetAddress::IPAddress)
  698. return WrongProtocolType;
  699. sockaddr_in socketAddress;
  700. netToIPSocketAddress(address, &socketAddress);
  701. if(!::connect(socket, (sockaddr *) &socketAddress, sizeof(socketAddress)))
  702. return NoError;
  703. return getLastError();
  704. }
  705. Net::Error Net::listen(NetSocket socket, S32 backlog)
  706. {
  707. if(!::listen(socket, backlog))
  708. return NoError;
  709. return getLastError();
  710. }
  711. NetSocket Net::accept(NetSocket acceptSocket, NetAddress *remoteAddress)
  712. {
  713. sockaddr_in socketAddress;
  714. socklen_t addrLen = sizeof(socketAddress);
  715. int retVal = ::accept(acceptSocket, (sockaddr *) &socketAddress, &addrLen);
  716. if(retVal != InvalidSocket)
  717. {
  718. IPSocketToNetAddress(&socketAddress, remoteAddress);
  719. return retVal;
  720. }
  721. return InvalidSocket;
  722. }
  723. Net::Error Net::bind(NetSocket socket, U16 port)
  724. {
  725. S32 error;
  726. sockaddr_in socketAddress;
  727. dMemset((char *)&socketAddress, 0, sizeof(socketAddress));
  728. socketAddress.sin_family = AF_INET;
  729. // It's entirely possible that there are two NIC cards.
  730. // We let the user specify which one the server runs on.
  731. // thanks to [TPG]P1aGu3 for the name
  732. const char* serverIP = Con::getVariable( "pref::Net::BindAddress" );
  733. // serverIP is guaranteed to be non-0.
  734. AssertFatal( serverIP, "serverIP is NULL!" );
  735. if( serverIP[0] != '\0' ) {
  736. // we're not empty
  737. socketAddress.sin_addr.s_addr = inet_addr( serverIP );
  738. if( socketAddress.sin_addr.s_addr != INADDR_NONE ) {
  739. Con::printf( "Binding server port to %s", serverIP );
  740. } else {
  741. Con::warnf( ConsoleLogEntry::General,
  742. "inet_addr() failed for %s while binding!",
  743. serverIP );
  744. socketAddress.sin_addr.s_addr = INADDR_ANY;
  745. }
  746. } else {
  747. Con::printf( "Binding server port to default IP" );
  748. socketAddress.sin_addr.s_addr = INADDR_ANY;
  749. }
  750. socketAddress.sin_port = htons(port);
  751. error = ::bind(socket, (sockaddr *) &socketAddress, sizeof(socketAddress));
  752. if(!error)
  753. return NoError;
  754. return getLastError();
  755. }
  756. Net::Error Net::setBufferSize(NetSocket socket, S32 bufferSize)
  757. {
  758. S32 error;
  759. error = setsockopt(socket, SOL_SOCKET, SO_RCVBUF, (char *) &bufferSize, sizeof(bufferSize));
  760. if(!error)
  761. error = setsockopt(socket, SOL_SOCKET, SO_SNDBUF, (char *) &bufferSize, sizeof(bufferSize));
  762. if(!error)
  763. return NoError;
  764. return getLastError();
  765. }
  766. Net::Error Net::setBroadcast(NetSocket socket, bool broadcast)
  767. {
  768. S32 bc = broadcast;
  769. S32 error = setsockopt(socket, SOL_SOCKET, SO_BROADCAST, (char*)&bc, sizeof(bc));
  770. if(!error)
  771. return NoError;
  772. return getLastError();
  773. }
  774. Net::Error Net::setBlocking(NetSocket socket, bool blockingIO)
  775. {
  776. unsigned long notblock = !blockingIO;
  777. S32 error = ioctl(socket, FIONBIO, &notblock);
  778. if(!error)
  779. return NoError;
  780. return getLastError();
  781. }
  782. Net::Error Net::send(NetSocket socket, const U8 *buffer, S32 bufferSize)
  783. {
  784. errno = 0;
  785. S32 bytesWritten = ::send(socket, (const char*)buffer, bufferSize, 0);
  786. if(bytesWritten == -1)
  787. #if defined(TORQUE_USE_WINSOCK)
  788. Con::errorf("Could not write to socket. Error: %s",strerror_wsa( WSAGetLastError() ));
  789. #else
  790. Con::errorf("Could not write to socket. Error: %s",strerror(errno));
  791. #endif
  792. return getLastError();
  793. }
  794. Net::Error Net::recv(NetSocket socket, U8 *buffer, S32 bufferSize, S32 *bytesRead)
  795. {
  796. *bytesRead = ::recv(socket, (char*)buffer, bufferSize, 0);
  797. if(*bytesRead == -1)
  798. return getLastError();
  799. return NoError;
  800. }
  801. bool Net::compareAddresses(const NetAddress *a1, const NetAddress *a2)
  802. {
  803. if((a1->type != a2->type) ||
  804. (*((U32 *)a1->netNum) != *((U32 *)a2->netNum)) ||
  805. (a1->port != a2->port))
  806. return false;
  807. if(a1->type == NetAddress::IPAddress)
  808. return true;
  809. for(S32 i = 0; i < 6; i++)
  810. if(a1->nodeNum[i] != a2->nodeNum[i])
  811. return false;
  812. return true;
  813. }
  814. bool Net::stringToAddress(const char *addressString, NetAddress *address)
  815. {
  816. if(!dStrnicmp(addressString, "ipx:", 4))
  817. // ipx support deprecated
  818. return false;
  819. if(!dStrnicmp(addressString, "ip:", 3))
  820. addressString += 3; // eat off the ip:
  821. sockaddr_in ipAddr;
  822. char remoteAddr[256];
  823. if(strlen(addressString) > 255)
  824. return false;
  825. dStrcpy(remoteAddr, addressString);
  826. char *portString = dStrchr(remoteAddr, ':');
  827. if(portString)
  828. *portString++ = '\0';
  829. if(!dStricmp(remoteAddr, "broadcast"))
  830. ipAddr.sin_addr.s_addr = htonl(INADDR_BROADCAST);
  831. else
  832. {
  833. ipAddr.sin_addr.s_addr = inet_addr(remoteAddr);
  834. if (ipAddr.sin_addr.s_addr == INADDR_NONE) // error
  835. {
  836. // On the Xbox, 'gethostbyname' does not exist so...
  837. #ifndef TORQUE_OS_XENON
  838. struct hostent *hp;
  839. if((hp = gethostbyname(remoteAddr)) == 0)
  840. return false;
  841. else
  842. memcpy(&ipAddr.sin_addr.s_addr, hp->h_addr, sizeof(in_addr));
  843. #else
  844. // On the Xbox do XNetDnsLookup
  845. XNDNS *pxndns = NULL;
  846. HANDLE hEvent = CreateEvent(NULL, false, false, NULL);
  847. XNetDnsLookup(remoteAddr, hEvent, &pxndns);
  848. // Wait for event (passing NULL as a handle to XNetDnsLookup will NOT
  849. // cause it to behave synchronously, so do not remove the handle/wait
  850. while(pxndns->iStatus == WSAEINPROGRESS)
  851. WaitForSingleObject(hEvent, INFINITE);
  852. bool foundAddr = pxndns->iStatus == 0 && pxndns->cina > 0;
  853. if(foundAddr)
  854. {
  855. // Lets just grab the first address returned, for now
  856. memcpy(&ipAddr.sin_addr, pxndns->aina, sizeof(IN_ADDR));
  857. }
  858. XNetDnsRelease(pxndns);
  859. CloseHandle(hEvent);
  860. // If we didn't successfully resolve the DNS lookup, bail after the
  861. // handles are released
  862. if(!foundAddr)
  863. return false;
  864. #endif
  865. }
  866. }
  867. if(portString)
  868. ipAddr.sin_port = htons(dAtoi(portString));
  869. else
  870. ipAddr.sin_port = htons(defaultPort);
  871. ipAddr.sin_family = AF_INET;
  872. IPSocketToNetAddress(&ipAddr, address);
  873. return true;
  874. }
  875. void Net::addressToString(const NetAddress *address, char addressString[256])
  876. {
  877. if(address->type == NetAddress::IPAddress)
  878. {
  879. sockaddr_in ipAddr;
  880. netToIPSocketAddress(address, &ipAddr);
  881. if(ipAddr.sin_addr.s_addr == htonl(INADDR_BROADCAST))
  882. dSprintf(addressString, 256, "IP:Broadcast:%d", ntohs(ipAddr.sin_port));
  883. else
  884. {
  885. #ifndef TORQUE_OS_XENON
  886. dSprintf(addressString, 256, "IP:%s:%d", inet_ntoa(ipAddr.sin_addr),
  887. ntohs(ipAddr.sin_port));
  888. #else
  889. dSprintf(addressString, 256, "IP:%d.%d.%d.%d:%d", ipAddr.sin_addr.s_net,
  890. ipAddr.sin_addr.s_host, ipAddr.sin_addr.s_lh,
  891. ipAddr.sin_addr.s_impno, ntohs( ipAddr.sin_port ) );
  892. #endif
  893. }
  894. }
  895. else
  896. {
  897. *addressString = 0;
  898. return;
  899. }
  900. }