platformNet.cpp 54 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896
  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 "platform/threads/mutex.h"
  24. #include "core/strings/stringFunctions.h"
  25. #include "core/util/hashFunction.h"
  26. #include "console/consoleTypes.h"
  27. // jamesu - debug DNS
  28. //#define TORQUE_DEBUG_LOOKUPS
  29. #if defined (TORQUE_OS_WIN)
  30. #define TORQUE_USE_WINSOCK
  31. #include <errno.h>
  32. #include <ws2tcpip.h>
  33. #ifndef EINPROGRESS
  34. #define EINPROGRESS WSAEINPROGRESS
  35. #endif // EINPROGRESS
  36. #define ioctl ioctlsocket
  37. typedef S32 socklen_t;
  38. #elif defined ( TORQUE_OS_MAC )
  39. #include <unistd.h>
  40. #include <sys/types.h>
  41. #include <sys/socket.h>
  42. #include <sys/poll.h>
  43. #include <arpa/inet.h>
  44. #include <netdb.h>
  45. #include <netinet/in.h>
  46. #include <errno.h>
  47. #include <sys/ioctl.h>
  48. #include <net/if.h>
  49. typedef sockaddr_in SOCKADDR_IN;
  50. typedef sockaddr * PSOCKADDR;
  51. typedef sockaddr SOCKADDR;
  52. typedef in_addr IN_ADDR;
  53. typedef int SOCKET;
  54. #define INVALID_SOCKET -1
  55. #define SOCKET_ERROR -1
  56. #define closesocket close
  57. #elif defined( TORQUE_OS_LINUX )
  58. #include <unistd.h>
  59. #include <sys/types.h>
  60. #include <sys/socket.h>
  61. #include <sys/poll.h>
  62. #include <arpa/inet.h>
  63. #include <netdb.h>
  64. #include <netinet/in.h>
  65. #include <errno.h>
  66. #include <sys/ioctl.h>
  67. #include <net/if.h>
  68. typedef sockaddr_in SOCKADDR_IN;
  69. typedef sockaddr_in6 SOCKADDR_IN6;
  70. typedef sockaddr * PSOCKADDR;
  71. typedef sockaddr SOCKADDR;
  72. typedef in_addr IN_ADDR;
  73. typedef in6_addr IN6_ADDR;
  74. typedef int SOCKET;
  75. #define INVALID_SOCKET -1
  76. #define SOCKET_ERROR -1
  77. #define closesocket close
  78. #elif defined( TORQUE_OS_XENON )
  79. #include <Xtl.h>
  80. #include <string>
  81. #define TORQUE_USE_WINSOCK
  82. #define EINPROGRESS WSAEINPROGRESS
  83. #define ioctl ioctlsocket
  84. typedef S32 socklen_t;
  85. DWORD _getLastErrorAndClear()
  86. {
  87. DWORD err = WSAGetLastError();
  88. WSASetLastError( 0 );
  89. return err;
  90. }
  91. #else
  92. #endif
  93. #if defined(TORQUE_USE_WINSOCK)
  94. static const char* strerror_wsa( S32 code )
  95. {
  96. switch( code )
  97. {
  98. #define E( name ) case name: return #name;
  99. E( WSANOTINITIALISED );
  100. E( WSAENETDOWN );
  101. E( WSAEADDRINUSE );
  102. E( WSAEINPROGRESS );
  103. E( WSAEALREADY );
  104. E( WSAEADDRNOTAVAIL );
  105. E( WSAEAFNOSUPPORT );
  106. E( WSAEFAULT );
  107. E( WSAEINVAL );
  108. E( WSAEISCONN );
  109. E( WSAENETUNREACH );
  110. E( WSAEHOSTUNREACH );
  111. E( WSAENOBUFS );
  112. E( WSAENOTSOCK );
  113. E( WSAETIMEDOUT );
  114. E( WSAEWOULDBLOCK );
  115. E( WSAEACCES );
  116. #undef E
  117. default:
  118. return "Unknown";
  119. }
  120. }
  121. #endif
  122. #include "core/util/tVector.h"
  123. #include "platform/platformNetAsync.h"
  124. #include "console/console.h"
  125. #include "core/util/journal/process.h"
  126. #include "core/util/journal/journal.h"
  127. NetSocket NetSocket::INVALID = NetSocket::fromHandle(-1);
  128. template<class T> class ReservedSocketList
  129. {
  130. public:
  131. Vector<T> mSocketList;
  132. Mutex *mMutex;
  133. ReservedSocketList()
  134. {
  135. mMutex = new Mutex;
  136. }
  137. ~ReservedSocketList()
  138. {
  139. delete mMutex;
  140. }
  141. inline void modify() { Mutex::lockMutex(mMutex); }
  142. inline void endModify() { Mutex::unlockMutex(mMutex); }
  143. NetSocket reserve(SOCKET reserveId = -1, bool doLock = true);
  144. void remove(NetSocket socketToRemove, bool doLock = true);
  145. T activate(NetSocket socketToActivate, int family, bool useUDP, bool clearOnFail = false);
  146. T resolve(NetSocket socketToResolve);
  147. };
  148. const SOCKET InvalidSocketHandle = -1;
  149. static void IPSocketToNetAddress(const struct sockaddr_in *sockAddr, NetAddress *address);
  150. namespace PlatformNetState
  151. {
  152. static S32 initCount = 0;
  153. static const S32 defaultPort = 28000;
  154. static S32 netPort = 0;
  155. static NetSocket udpSocket = NetSocket::INVALID;
  156. static NetSocket udp6Socket = NetSocket::INVALID;
  157. static NetSocket multicast6Socket = NetSocket::INVALID;
  158. static ipv6_mreq multicast6Group;
  159. static ReservedSocketList<SOCKET> smReservedSocketList;
  160. static Net::Error getLastError()
  161. {
  162. #if defined(TORQUE_USE_WINSOCK)
  163. S32 err = WSAGetLastError();
  164. switch (err)
  165. {
  166. case 0:
  167. return Net::NoError;
  168. case WSAEWOULDBLOCK:
  169. return Net::WouldBlock;
  170. default:
  171. return Net::UnknownError;
  172. }
  173. #else
  174. if (errno == EAGAIN)
  175. return Net::WouldBlock;
  176. if (errno == 0)
  177. return Net::NoError;
  178. return Net::UnknownError;
  179. #endif
  180. }
  181. static S32 getDefaultGameProtocol()
  182. {
  183. // we turn off VDP in non-release builds because VDP does not support broadcast packets
  184. // which are required for LAN queries (PC->Xbox connectivity). The wire protocol still
  185. // uses the VDP packet structure, though.
  186. S32 protocol = IPPROTO_UDP;
  187. bool useVDP = false;
  188. #ifdef TORQUE_DISABLE_PC_CONNECTIVITY
  189. // Xbox uses a VDP (voice/data protocol) socket for networking
  190. protocol = IPPROTO_VDP;
  191. useVDP = true;
  192. #endif
  193. return protocol;
  194. }
  195. static struct addrinfo* pickAddressByProtocol(struct addrinfo* addr, int protocol)
  196. {
  197. for (addr; addr != NULL; addr = addr->ai_next)
  198. {
  199. if (addr->ai_family == protocol)
  200. return addr;
  201. }
  202. return NULL;
  203. }
  204. /// Extracts core address parts from an address string. Returns false if it's malformed.
  205. static bool extractAddressParts(const char *addressString, char outAddress[256], int &outPort, int &outFamily)
  206. {
  207. outPort = 0;
  208. outFamily = AF_UNSPEC;
  209. if (!dStrnicmp(addressString, "ipx:", 4))
  210. // ipx support deprecated
  211. return false;
  212. if (!dStrnicmp(addressString, "ip:", 3))
  213. {
  214. addressString += 3; // eat off the ip:
  215. outFamily = AF_INET;
  216. }
  217. else if (!dStrnicmp(addressString, "ip6:", 4))
  218. {
  219. addressString += 4; // eat off the ip6:
  220. outFamily = AF_INET6;
  221. }
  222. if (strlen(addressString) > 255)
  223. return false;
  224. char *portString = NULL;
  225. if (addressString[0] == '[')
  226. {
  227. // Must be ipv6 notation
  228. dStrcpy(outAddress, addressString+1);
  229. addressString = outAddress;
  230. portString = dStrchr(outAddress, ']');
  231. if (portString)
  232. {
  233. // Sort out the :port after the ]
  234. *portString++ = '\0';
  235. if (*portString != ':')
  236. {
  237. portString = NULL;
  238. }
  239. else
  240. {
  241. *portString++ = '\0';
  242. }
  243. }
  244. if (outFamily == AF_UNSPEC)
  245. {
  246. outFamily = AF_INET6;
  247. }
  248. }
  249. else
  250. {
  251. dStrcpy(outAddress, addressString);
  252. addressString = outAddress;
  253. // Check to see if we have multiple ":" which would indicate this is an ipv6 address
  254. char* scan = outAddress;
  255. int colonCount = 0;
  256. while (*scan != '\0' && colonCount < 2)
  257. {
  258. if (*scan++ == ':')
  259. colonCount++;
  260. }
  261. if (colonCount <= 1)
  262. {
  263. // either ipv4 or host
  264. portString = dStrchr(outAddress, ':');
  265. if (portString)
  266. {
  267. *portString++ = '\0';
  268. }
  269. }
  270. else if (outFamily == AF_UNSPEC)
  271. {
  272. // Must be ipv6
  273. outFamily = AF_INET6;
  274. }
  275. }
  276. if (portString)
  277. {
  278. outPort = dAtoi(portString);
  279. }
  280. return true;
  281. }
  282. };
  283. template<class T> NetSocket ReservedSocketList<T>::reserve(SOCKET reserveId, bool doLock)
  284. {
  285. MutexHandle handle;
  286. if (doLock)
  287. {
  288. handle.lock(mMutex, true);
  289. }
  290. S32 idx = mSocketList.find_next(-1);
  291. if (idx == -1)
  292. {
  293. mSocketList.push_back(reserveId);
  294. return NetSocket::fromHandle(mSocketList.size() - 1);
  295. }
  296. else
  297. {
  298. mSocketList[idx] = reserveId;
  299. }
  300. return NetSocket::fromHandle(idx);
  301. }
  302. template<class T> void ReservedSocketList<T>::remove(NetSocket socketToRemove, bool doLock)
  303. {
  304. MutexHandle handle;
  305. if (doLock)
  306. {
  307. handle.lock(mMutex, true);
  308. }
  309. if ((U32)socketToRemove.getHandle() >= (U32)mSocketList.size())
  310. return;
  311. mSocketList[socketToRemove.getHandle()] = -1;
  312. }
  313. template<class T> T ReservedSocketList<T>::activate(NetSocket socketToActivate, int family, bool useUDP, bool clearOnFail)
  314. {
  315. MutexHandle h;
  316. h.lock(mMutex, true);
  317. int typeID = useUDP ? SOCK_DGRAM : SOCK_STREAM;
  318. int protocol = useUDP ? PlatformNetState::getDefaultGameProtocol() : 0;
  319. if ((U32)socketToActivate.getHandle() >= (U32)mSocketList.size())
  320. return -1;
  321. T socketFd = mSocketList[socketToActivate.getHandle()];
  322. if (socketFd == -1)
  323. {
  324. socketFd = ::socket(family, typeID, protocol);
  325. if (socketFd == InvalidSocketHandle)
  326. {
  327. if (clearOnFail)
  328. {
  329. remove(socketToActivate, false);
  330. }
  331. return InvalidSocketHandle;
  332. }
  333. else
  334. {
  335. mSocketList[socketToActivate.getHandle()] = socketFd;
  336. return socketFd;
  337. }
  338. }
  339. return socketFd;
  340. }
  341. template<class T> T ReservedSocketList<T>::resolve(NetSocket socketToResolve)
  342. {
  343. MutexHandle h;
  344. h.lock(mMutex, true);
  345. if ((U32)socketToResolve.getHandle() >= (U32)mSocketList.size())
  346. return -1;
  347. return mSocketList[socketToResolve.getHandle()];
  348. }
  349. ConnectionNotifyEvent Net::smConnectionNotify;
  350. ConnectionAcceptedEvent Net::smConnectionAccept;
  351. ConnectionReceiveEvent Net::smConnectionReceive;
  352. PacketReceiveEvent Net::smPacketReceive;
  353. // Multicast stuff
  354. bool Net::smMulticastEnabled = true;
  355. //
  356. // Protocol Stuff
  357. bool Net::smIpv4Enabled = true;
  358. bool Net::smIpv6Enabled = false;
  359. //
  360. // the Socket structure helps us keep track of the
  361. // above states
  362. struct PolledSocket
  363. {
  364. // local enum for socket states for polled sockets
  365. enum SocketState
  366. {
  367. InvalidState,
  368. Connected,
  369. ConnectionPending,
  370. Listening,
  371. NameLookupRequired
  372. };
  373. PolledSocket()
  374. {
  375. fd = -1;
  376. handleFd = NetSocket::INVALID;
  377. state = InvalidState;
  378. remoteAddr[0] = 0;
  379. remotePort = -1;
  380. }
  381. SOCKET fd;
  382. NetSocket handleFd;
  383. S32 state;
  384. char remoteAddr[256];
  385. S32 remotePort;
  386. };
  387. // list of polled sockets
  388. static Vector<PolledSocket*> gPolledSockets( __FILE__, __LINE__ );
  389. static PolledSocket* addPolledSocket(NetSocket handleFd, SOCKET fd, S32 state,
  390. char* remoteAddr = NULL, S32 port = -1)
  391. {
  392. PolledSocket* sock = new PolledSocket();
  393. sock->fd = fd;
  394. sock->handleFd = handleFd;
  395. sock->state = state;
  396. if (remoteAddr)
  397. dStrcpy(sock->remoteAddr, remoteAddr);
  398. if (port != -1)
  399. sock->remotePort = port;
  400. gPolledSockets.push_back(sock);
  401. return sock;
  402. }
  403. bool netSocketWaitForWritable(NetSocket handleFd, S32 timeoutMs)
  404. {
  405. fd_set writefds;
  406. timeval timeout;
  407. SOCKET socketFd = PlatformNetState::smReservedSocketList.resolve(handleFd);
  408. FD_ZERO( &writefds );
  409. FD_SET( socketFd, &writefds );
  410. timeout.tv_sec = timeoutMs / 1000;
  411. timeout.tv_usec = ( timeoutMs % 1000 ) * 1000;
  412. if( select(socketFd + 1, NULL, &writefds, NULL, &timeout) > 0 )
  413. return true;
  414. return false;
  415. }
  416. bool Net::init()
  417. {
  418. #if defined(TORQUE_USE_WINSOCK)
  419. if(!PlatformNetState::initCount)
  420. {
  421. #ifdef TORQUE_OS_XENON
  422. // Configure startup parameters
  423. XNetStartupParams xnsp;
  424. memset( &xnsp, 0, sizeof( xnsp ) );
  425. xnsp.cfgSizeOfStruct = sizeof( XNetStartupParams );
  426. #ifndef TORQUE_DISABLE_PC_CONNECTIVITY
  427. xnsp.cfgFlags = XNET_STARTUP_BYPASS_SECURITY;
  428. Con::warnf("XNET_STARTUP_BYPASS_SECURITY enabled! This build can talk to PCs!");
  429. #endif
  430. AssertISV( !XNetStartup( &xnsp ), "Net::init - failed to init XNet" );
  431. #endif
  432. WSADATA stWSAData;
  433. AssertISV( !WSAStartup( 0x0101, &stWSAData ), "Net::init - failed to init WinSock!" );
  434. //logprintf("Winsock initialization %s", success ? "succeeded." : "failed!");
  435. }
  436. #endif
  437. PlatformNetState::initCount++;
  438. Process::notify(&Net::process, PROCESS_NET_ORDER);
  439. return(true);
  440. }
  441. void Net::shutdown()
  442. {
  443. Process::remove(&Net::process);
  444. while (gPolledSockets.size() > 0)
  445. closeConnectTo(gPolledSockets[0]->handleFd);
  446. closePort();
  447. PlatformNetState::initCount--;
  448. #if defined(TORQUE_USE_WINSOCK)
  449. if(!PlatformNetState::initCount)
  450. {
  451. WSACleanup();
  452. #ifdef TORQUE_OS_XENON
  453. XNetCleanup();
  454. #endif
  455. }
  456. #endif
  457. }
  458. // ipv4 version of name routines
  459. static void NetAddressToIPSocket(const NetAddress *address, struct sockaddr_in *sockAddr)
  460. {
  461. dMemset(sockAddr, 0, sizeof(struct sockaddr_in));
  462. sockAddr->sin_family = AF_INET;
  463. sockAddr->sin_port = htons(address->port);
  464. #if defined(TORQUE_OS_BSD)
  465. sockAddr->sin_len = sizeof(struct sockaddr_in);
  466. #endif
  467. if (address->type == NetAddress::IPBroadcastAddress)
  468. {
  469. sockAddr->sin_addr.s_addr = htonl(INADDR_BROADCAST);
  470. }
  471. else
  472. {
  473. dMemcpy(&sockAddr->sin_addr, &address->address.ipv4.netNum[0], 4);
  474. }
  475. }
  476. static void IPSocketToNetAddress(const struct sockaddr_in *sockAddr, NetAddress *address)
  477. {
  478. address->type = NetAddress::IPAddress;
  479. address->port = ntohs(sockAddr->sin_port);
  480. dMemcpy(&address->address.ipv4.netNum[0], &sockAddr->sin_addr, 4);
  481. }
  482. // ipv6 version of name routines
  483. static void NetAddressToIPSocket6(const NetAddress *address, struct sockaddr_in6 *sockAddr)
  484. {
  485. dMemset(sockAddr, 0, sizeof(struct sockaddr_in6));
  486. #ifdef SIN6_LEN
  487. sockAddr->sin6_len = sizeof(struct sockaddr_in6);
  488. #endif
  489. sockAddr->sin6_family = AF_INET6;
  490. sockAddr->sin6_port = ntohs(address->port);
  491. if (address->type == NetAddress::IPV6MulticastAddress)
  492. {
  493. sockAddr->sin6_addr = PlatformNetState::multicast6Group.ipv6mr_multiaddr;
  494. sockAddr->sin6_scope_id = PlatformNetState::multicast6Group.ipv6mr_interface;
  495. }
  496. else
  497. {
  498. sockAddr->sin6_flowinfo = address->address.ipv6.netFlow;
  499. sockAddr->sin6_scope_id = address->address.ipv6.netScope;
  500. dMemcpy(&sockAddr->sin6_addr, address->address.ipv6.netNum, sizeof(address->address.ipv6.netNum));
  501. }
  502. }
  503. static void IPSocket6ToNetAddress(const struct sockaddr_in6 *sockAddr, NetAddress *address)
  504. {
  505. address->type = NetAddress::IPV6Address;
  506. address->port = ntohs(sockAddr->sin6_port);
  507. dMemcpy(address->address.ipv6.netNum, &sockAddr->sin6_addr, sizeof(address->address.ipv6.netNum));
  508. address->address.ipv6.netFlow = sockAddr->sin6_flowinfo;
  509. address->address.ipv6.netScope = sockAddr->sin6_scope_id;
  510. }
  511. //
  512. NetSocket Net::openListenPort(U16 port, NetAddress::Type addressType)
  513. {
  514. if(Journal::IsPlaying())
  515. {
  516. U32 ret;
  517. Journal::Read(&ret);
  518. return NetSocket::fromHandle(ret);
  519. }
  520. Net::Error error = NoError;
  521. NetAddress address;
  522. if (!Net::getListenAddress(addressType, &address))
  523. error = Net::WrongProtocolType;
  524. NetSocket handleFd = NetSocket::INVALID;
  525. SOCKET sockId = InvalidSocketHandle;
  526. if (error == NoError)
  527. {
  528. handleFd = openSocket();
  529. sockId = PlatformNetState::smReservedSocketList.activate(handleFd, address.type == NetAddress::IPAddress ? AF_INET : AF_INET6, false, true);
  530. }
  531. if (error == NoError && (handleFd == NetSocket::INVALID || sockId == InvalidSocketHandle))
  532. {
  533. Con::errorf("Unable to open listen socket: %s", strerror(errno));
  534. error = NotASocket;
  535. handleFd = NetSocket::INVALID;
  536. }
  537. if (error == NoError)
  538. {
  539. address.port = port;
  540. error = bindAddress(address, handleFd, false);
  541. if (error != NoError)
  542. {
  543. Con::errorf("Unable to bind port %d: %s", port, strerror(errno));
  544. closeSocket(handleFd);
  545. handleFd = NetSocket::INVALID;
  546. }
  547. }
  548. if (error == NoError)
  549. {
  550. error = listen(handleFd, 4);
  551. if (error != NoError)
  552. {
  553. Con::errorf("Unable to listen on port %d: %s", port, strerror(errno));
  554. closeSocket(handleFd);
  555. handleFd = NetSocket::INVALID;
  556. }
  557. }
  558. if (error == NoError)
  559. {
  560. setBlocking(handleFd, false);
  561. addPolledSocket(handleFd, sockId, PolledSocket::Listening);
  562. }
  563. if(Journal::IsRecording())
  564. Journal::Write(U32(handleFd.getHandle()));
  565. return handleFd;
  566. }
  567. NetSocket Net::openConnectTo(const char *addressString)
  568. {
  569. if (Journal::IsPlaying())
  570. {
  571. U32 ret;
  572. Journal::Read(&ret);
  573. return NetSocket::fromHandle(ret);
  574. }
  575. NetAddress address;
  576. NetSocket handleFd = NetSocket::INVALID;
  577. Net::Error error = NoError;
  578. error = Net::stringToAddress(addressString, &address, false);
  579. if (error == NoError && address.type != NetAddress::IPAddress && address.type != NetAddress::IPV6Address)
  580. {
  581. error = Net::WrongProtocolType;
  582. }
  583. if (error != NoError || error == NeedHostLookup)
  584. {
  585. handleFd = openSocket();
  586. }
  587. // Attempt to connect or queue a lookup
  588. if (error == NoError && address.type == NetAddress::IPAddress)
  589. {
  590. sockaddr_in ipAddr;
  591. NetAddressToIPSocket(&address, &ipAddr);
  592. SOCKET socketFd = PlatformNetState::smReservedSocketList.activate(handleFd, AF_INET, false, true);
  593. if (socketFd != InvalidSocketHandle)
  594. {
  595. setBlocking(handleFd, false);
  596. if (::connect(socketFd, (struct sockaddr *)&ipAddr, sizeof(ipAddr)) == -1 &&
  597. errno != EINPROGRESS)
  598. {
  599. Con::errorf("Error connecting %s: %s",
  600. addressString, strerror(errno));
  601. closeSocket(handleFd);
  602. handleFd = NetSocket::INVALID;
  603. }
  604. }
  605. else
  606. {
  607. PlatformNetState::smReservedSocketList.remove(handleFd);
  608. handleFd = NetSocket::INVALID;
  609. }
  610. if (handleFd != NetSocket::INVALID)
  611. {
  612. // add this socket to our list of polled sockets
  613. addPolledSocket(handleFd, socketFd, PolledSocket::ConnectionPending);
  614. }
  615. }
  616. else if (error == NoError && address.type == NetAddress::IPV6Address)
  617. {
  618. sockaddr_in6 ipAddr6;
  619. NetAddressToIPSocket6(&address, &ipAddr6);
  620. SOCKET socketFd = PlatformNetState::smReservedSocketList.activate(handleFd, AF_INET6, false, true);
  621. if (::connect(socketFd, (struct sockaddr *)&ipAddr6, sizeof(ipAddr6)) == -1 &&
  622. errno != EINPROGRESS)
  623. {
  624. setBlocking(handleFd, false);
  625. Con::errorf("Error connecting %s: %s",
  626. addressString, strerror(errno));
  627. closeSocket(handleFd);
  628. handleFd = NetSocket::INVALID;
  629. }
  630. else
  631. {
  632. PlatformNetState::smReservedSocketList.remove(handleFd);
  633. handleFd = NetSocket::INVALID;
  634. }
  635. if (handleFd != NetSocket::INVALID)
  636. {
  637. // add this socket to our list of polled sockets
  638. addPolledSocket(handleFd, socketFd, PolledSocket::ConnectionPending);
  639. }
  640. }
  641. else if (error == Net::NeedHostLookup)
  642. {
  643. // need to do an asynchronous name lookup. first, add the socket
  644. // to the polled list
  645. char addressString[256];
  646. Net::addressToString(&address, addressString);
  647. addPolledSocket(handleFd, InvalidSocketHandle, PolledSocket::NameLookupRequired, addressString, address.port);
  648. // queue the lookup
  649. gNetAsync.queueLookup(addressString, handleFd);
  650. }
  651. else
  652. {
  653. handleFd = NetSocket::INVALID;
  654. }
  655. if (Journal::IsRecording())
  656. Journal::Write(U32(handleFd.getHandle()));
  657. return handleFd;
  658. }
  659. void Net::closeConnectTo(NetSocket handleFd)
  660. {
  661. if(Journal::IsPlaying())
  662. return;
  663. // if this socket is in the list of polled sockets, remove it
  664. for (S32 i = 0; i < gPolledSockets.size(); ++i)
  665. {
  666. if (gPolledSockets[i]->handleFd == handleFd)
  667. {
  668. delete gPolledSockets[i];
  669. gPolledSockets.erase(i);
  670. break;
  671. }
  672. }
  673. closeSocket(handleFd);
  674. }
  675. Net::Error Net::sendtoSocket(NetSocket handleFd, const U8 *buffer, S32 bufferSize)
  676. {
  677. if(Journal::IsPlaying())
  678. {
  679. U32 e;
  680. Journal::Read(&e);
  681. return (Net::Error) e;
  682. }
  683. Net::Error e = send(handleFd, buffer, bufferSize);
  684. if(Journal::IsRecording())
  685. Journal::Write(U32(e));
  686. return e;
  687. }
  688. bool Net::openPort(S32 port, bool doBind)
  689. {
  690. if (PlatformNetState::udpSocket != NetSocket::INVALID)
  691. {
  692. closeSocket(PlatformNetState::udpSocket);
  693. PlatformNetState::udpSocket = NetSocket::INVALID;
  694. }
  695. if (PlatformNetState::udp6Socket != NetSocket::INVALID)
  696. {
  697. closeSocket(PlatformNetState::udp6Socket);
  698. PlatformNetState::udp6Socket = NetSocket::INVALID;
  699. }
  700. // Frequently port "0" is used even though it makes no sense, so instead use the default port.
  701. if (port == 0)
  702. {
  703. port = PlatformNetState::defaultPort;
  704. }
  705. // Update prefs
  706. Net::smMulticastEnabled = Con::getBoolVariable("pref::Net::Multicast6Enabled", true);
  707. Net::smIpv4Enabled = Con::getBoolVariable("pref::Net::IPV4Enabled", true);
  708. Net::smIpv6Enabled = Con::getBoolVariable("pref::Net::IPV6Enabled", false);
  709. // we turn off VDP in non-release builds because VDP does not support broadcast packets
  710. // which are required for LAN queries (PC->Xbox connectivity). The wire protocol still
  711. // uses the VDP packet structure, though.
  712. S32 protocol = PlatformNetState::getDefaultGameProtocol();
  713. SOCKET socketFd = InvalidSocketHandle;
  714. NetAddress address;
  715. if (Net::smIpv4Enabled)
  716. {
  717. if (Net::getListenAddress(NetAddress::IPAddress, &address))
  718. {
  719. address.port = port;
  720. socketFd = ::socket(AF_INET, SOCK_DGRAM, protocol);
  721. if (socketFd != InvalidSocketHandle)
  722. {
  723. PlatformNetState::udpSocket = PlatformNetState::smReservedSocketList.reserve(socketFd);
  724. Net::Error error = NoError;
  725. if (doBind)
  726. {
  727. error = bindAddress(address, PlatformNetState::udpSocket, true);
  728. }
  729. if (error == NoError)
  730. error = setBufferSize(PlatformNetState::udpSocket, 32768 * 8);
  731. #ifndef TORQUE_DISABLE_PC_CONNECTIVITY
  732. if (error == NoError)
  733. error = setBroadcast(PlatformNetState::udpSocket, true);
  734. #endif
  735. if (error == NoError)
  736. error = setBlocking(PlatformNetState::udpSocket, false);
  737. if (error == NoError)
  738. {
  739. Con::printf("UDP initialized on ipv4 port %d", port);
  740. }
  741. else
  742. {
  743. closeSocket(PlatformNetState::udpSocket);
  744. PlatformNetState::udpSocket = NetSocket::INVALID;
  745. Con::printf("Unable to initialize UDP on ipv4 - error %d", error);
  746. }
  747. }
  748. }
  749. else
  750. {
  751. Con::errorf("Unable to initialize UDP on ipv4 - invalid address.");
  752. PlatformNetState::udpSocket = NetSocket::INVALID;
  753. }
  754. }
  755. if (Net::smIpv6Enabled)
  756. {
  757. if (Net::getListenAddress(NetAddress::IPV6Address, &address))
  758. {
  759. address.port = port;
  760. socketFd = ::socket(AF_INET6, SOCK_DGRAM, protocol);
  761. if (socketFd != InvalidSocketHandle)
  762. {
  763. PlatformNetState::udp6Socket = PlatformNetState::smReservedSocketList.reserve(socketFd);
  764. Net::Error error = NoError;
  765. int v = 1;
  766. setsockopt(socketFd, IPPROTO_IPV6, IPV6_V6ONLY, (const char *)&v, sizeof(v));
  767. PlatformNetState::getLastError();
  768. if (doBind)
  769. {
  770. error = bindAddress(address, PlatformNetState::udp6Socket, true);
  771. }
  772. if (error == NoError)
  773. error = setBufferSize(PlatformNetState::udp6Socket, 32768 * 8);
  774. if (error == NoError)
  775. error = setBlocking(PlatformNetState::udp6Socket, false);
  776. if (error == NoError)
  777. {
  778. Con::printf("UDP initialized on ipv6 port %d", port);
  779. }
  780. else
  781. {
  782. closeSocket(PlatformNetState::udp6Socket);
  783. PlatformNetState::udp6Socket = NetSocket::INVALID;
  784. Con::printf("Unable to initialize UDP on ipv6 - error %d", error);
  785. }
  786. if (Net::smMulticastEnabled && doBind)
  787. {
  788. Net::enableMulticast();
  789. }
  790. else
  791. {
  792. Net::disableMulticast();
  793. }
  794. }
  795. }
  796. }
  797. PlatformNetState::netPort = port;
  798. return PlatformNetState::udpSocket != NetSocket::INVALID || PlatformNetState::udp6Socket != NetSocket::INVALID;
  799. }
  800. NetSocket Net::getPort()
  801. {
  802. return PlatformNetState::udpSocket;
  803. }
  804. void Net::closePort()
  805. {
  806. if (PlatformNetState::udpSocket != NetSocket::INVALID)
  807. closeSocket(PlatformNetState::udpSocket);
  808. if (PlatformNetState::udp6Socket != NetSocket::INVALID)
  809. closeSocket(PlatformNetState::udp6Socket);
  810. }
  811. Net::Error Net::sendto(const NetAddress *address, const U8 *buffer, S32 bufferSize)
  812. {
  813. if(Journal::IsPlaying())
  814. return NoError;
  815. SOCKET socketFd;
  816. if(address->type == NetAddress::IPAddress || address->type == NetAddress::IPBroadcastAddress)
  817. {
  818. socketFd = PlatformNetState::smReservedSocketList.resolve(PlatformNetState::udpSocket);
  819. if (socketFd != InvalidSocketHandle)
  820. {
  821. sockaddr_in ipAddr;
  822. NetAddressToIPSocket(address, &ipAddr);
  823. if (::sendto(socketFd, (const char*)buffer, bufferSize, 0,
  824. (sockaddr *)&ipAddr, sizeof(sockaddr_in)) == SOCKET_ERROR)
  825. return PlatformNetState::getLastError();
  826. else
  827. return NoError;
  828. }
  829. else
  830. {
  831. return NotASocket;
  832. }
  833. }
  834. else if (address->type == NetAddress::IPV6Address || address->type == NetAddress::IPV6MulticastAddress)
  835. {
  836. socketFd = PlatformNetState::smReservedSocketList.resolve(address->type == NetAddress::IPV6MulticastAddress ? PlatformNetState::multicast6Socket : PlatformNetState::udp6Socket);
  837. if (socketFd != InvalidSocketHandle)
  838. {
  839. sockaddr_in6 ipAddr;
  840. NetAddressToIPSocket6(address, &ipAddr);
  841. if (::sendto(socketFd, (const char*)buffer, bufferSize, 0,
  842. (struct sockaddr *) &ipAddr, sizeof(sockaddr_in6)) == SOCKET_ERROR)
  843. return PlatformNetState::getLastError();
  844. else
  845. return NoError;
  846. }
  847. else
  848. {
  849. return NotASocket;
  850. }
  851. }
  852. return WrongProtocolType;
  853. }
  854. void Net::process()
  855. {
  856. // Process listening sockets
  857. processListenSocket(PlatformNetState::udpSocket);
  858. processListenSocket(PlatformNetState::udp6Socket);
  859. // process the polled sockets. This blob of code performs functions
  860. // similar to WinsockProc in winNet.cc
  861. if (gPolledSockets.size() == 0)
  862. return;
  863. S32 optval;
  864. socklen_t optlen = sizeof(S32);
  865. S32 bytesRead;
  866. Net::Error err;
  867. bool removeSock = false;
  868. PolledSocket *currentSock = NULL;
  869. NetSocket incomingHandleFd = NetSocket::INVALID;
  870. NetAddress out_h_addr;
  871. S32 out_h_length = 0;
  872. RawData readBuff;
  873. for (S32 i = 0; i < gPolledSockets.size();
  874. /* no increment, this is done at end of loop body */)
  875. {
  876. removeSock = false;
  877. currentSock = gPolledSockets[i];
  878. switch (currentSock->state)
  879. {
  880. case PolledSocket::InvalidState:
  881. Con::errorf("Error, InvalidState socket in polled sockets list");
  882. break;
  883. case PolledSocket::ConnectionPending:
  884. // see if it is now connected
  885. #ifdef TORQUE_OS_XENON
  886. // WSASetLastError has no return value, however part of the SO_ERROR behavior
  887. // is to clear the last error, so this needs to be done here.
  888. if( ( optval = _getLastErrorAndClear() ) == -1 )
  889. #else
  890. if (getsockopt(currentSock->fd, SOL_SOCKET, SO_ERROR,
  891. (char*)&optval, &optlen) == -1)
  892. #endif
  893. {
  894. Con::errorf("Error getting socket options: %s", strerror(errno));
  895. Net::smConnectionNotify.trigger(currentSock->handleFd, Net::ConnectFailed);
  896. removeSock = true;
  897. }
  898. else
  899. {
  900. if (optval == EINPROGRESS)
  901. // still connecting...
  902. break;
  903. if (optval == 0)
  904. {
  905. // poll for writable status to be sure we're connected.
  906. bool ready = netSocketWaitForWritable(currentSock->handleFd,0);
  907. if(!ready)
  908. break;
  909. currentSock->state = PolledSocket::Connected;
  910. Net::smConnectionNotify.trigger(currentSock->handleFd, Net::Connected);
  911. }
  912. else
  913. {
  914. // some kind of error
  915. Con::errorf("Error connecting: %s", strerror(errno));
  916. Net::smConnectionNotify.trigger(currentSock->handleFd, Net::ConnectFailed);
  917. removeSock = true;
  918. }
  919. }
  920. break;
  921. case PolledSocket::Connected:
  922. // try to get some data
  923. bytesRead = 0;
  924. readBuff.alloc(MaxPacketDataSize);
  925. err = Net::recv(currentSock->handleFd, (U8*)readBuff.data, MaxPacketDataSize, &bytesRead);
  926. if(err == Net::NoError)
  927. {
  928. if (bytesRead > 0)
  929. {
  930. // got some data, post it
  931. readBuff.size = bytesRead;
  932. Net::smConnectionReceive.trigger(currentSock->handleFd, readBuff);
  933. }
  934. else
  935. {
  936. // ack! this shouldn't happen
  937. if (bytesRead < 0)
  938. Con::errorf("Unexpected error on socket: %s", strerror(errno));
  939. // zero bytes read means EOF
  940. Net::smConnectionNotify.trigger(currentSock->handleFd, Net::Disconnected);
  941. removeSock = true;
  942. }
  943. }
  944. else if (err != Net::NoError && err != Net::WouldBlock)
  945. {
  946. Con::errorf("Error reading from socket: %s", strerror(errno));
  947. Net::smConnectionNotify.trigger(currentSock->handleFd, Net::Disconnected);
  948. removeSock = true;
  949. }
  950. break;
  951. case PolledSocket::NameLookupRequired:
  952. U32 newState;
  953. // is the lookup complete?
  954. if (!gNetAsync.checkLookup(
  955. currentSock->handleFd, &out_h_addr, &out_h_length,
  956. sizeof(out_h_addr)))
  957. break;
  958. if (out_h_length == -1)
  959. {
  960. Con::errorf("DNS lookup failed: %s", currentSock->remoteAddr);
  961. newState = Net::DNSFailed;
  962. removeSock = true;
  963. }
  964. else
  965. {
  966. // try to connect
  967. out_h_addr.port = currentSock->remotePort;
  968. const sockaddr *ai_addr = NULL;
  969. int ai_addrlen = 0;
  970. sockaddr_in socketAddress;
  971. sockaddr_in6 socketAddress6;
  972. if (out_h_addr.type == NetAddress::IPAddress)
  973. {
  974. ai_addr = (const sockaddr*)&socketAddress;
  975. ai_addrlen = sizeof(socketAddress);
  976. NetAddressToIPSocket(&out_h_addr, &socketAddress);
  977. currentSock->fd = PlatformNetState::smReservedSocketList.activate(currentSock->handleFd, AF_INET, false);
  978. setBlocking(currentSock->handleFd, false);
  979. #ifdef TORQUE_DEBUG_LOOKUPS
  980. char addrString[256];
  981. NetAddress addr;
  982. IPSocketToNetAddress(&socketAddress, &addr);
  983. Net::addressToString(&addr, addrString);
  984. Con::printf("DNS: lookup resolved to %s", addrString);
  985. #endif
  986. }
  987. else if (out_h_addr.type == NetAddress::IPV6Address)
  988. {
  989. ai_addr = (const sockaddr*)&socketAddress6;
  990. ai_addrlen = sizeof(socketAddress6);
  991. NetAddressToIPSocket6(&out_h_addr, &socketAddress6);
  992. currentSock->fd = PlatformNetState::smReservedSocketList.activate(currentSock->handleFd, AF_INET6, false);
  993. setBlocking(currentSock->handleFd, false);
  994. #ifdef TORQUE_DEBUG_LOOKUPS
  995. char addrString[256];
  996. NetAddress addr;
  997. IPSocket6ToNetAddress(&socketAddress6, &addr);
  998. Net::addressToString(&addr, addrString);
  999. Con::printf("DNS: lookup resolved to %s", addrString);
  1000. #endif
  1001. }
  1002. else
  1003. {
  1004. Con::errorf("Error connecting to %s: Invalid Protocol",
  1005. currentSock->remoteAddr);
  1006. newState = Net::ConnectFailed;
  1007. removeSock = true;
  1008. }
  1009. if (ai_addr)
  1010. {
  1011. if (::connect(currentSock->fd, ai_addr,
  1012. ai_addrlen) == -1)
  1013. {
  1014. if (errno == EINPROGRESS)
  1015. {
  1016. newState = Net::DNSResolved;
  1017. currentSock->state = PolledSocket::ConnectionPending;
  1018. }
  1019. else
  1020. {
  1021. Con::errorf("Error connecting to %s: %s",
  1022. currentSock->remoteAddr, strerror(errno));
  1023. newState = Net::ConnectFailed;
  1024. removeSock = true;
  1025. }
  1026. }
  1027. else
  1028. {
  1029. newState = Net::Connected;
  1030. currentSock->state = Connected;
  1031. }
  1032. }
  1033. }
  1034. Net::smConnectionNotify.trigger(currentSock->handleFd, newState);
  1035. break;
  1036. case PolledSocket::Listening:
  1037. NetAddress incomingAddy;
  1038. incomingHandleFd = Net::accept(currentSock->handleFd, &incomingAddy);
  1039. if(incomingHandleFd != NetSocket::INVALID)
  1040. {
  1041. setBlocking(incomingHandleFd, false);
  1042. addPolledSocket(incomingHandleFd, PlatformNetState::smReservedSocketList.resolve(incomingHandleFd), Connected);
  1043. Net::smConnectionAccept.trigger(currentSock->handleFd, incomingHandleFd, incomingAddy);
  1044. }
  1045. break;
  1046. }
  1047. // only increment index if we're not removing the connection, since
  1048. // the removal will shift the indices down by one
  1049. if (removeSock)
  1050. closeConnectTo(currentSock->handleFd);
  1051. else
  1052. i++;
  1053. }
  1054. }
  1055. void Net::processListenSocket(NetSocket socketHandle)
  1056. {
  1057. if (socketHandle == NetSocket::INVALID)
  1058. return;
  1059. sockaddr_storage sa;
  1060. sa.ss_family = AF_UNSPEC;
  1061. NetAddress srcAddress;
  1062. RawData tmpBuffer;
  1063. tmpBuffer.alloc(Net::MaxPacketDataSize);
  1064. SOCKET socketFd = PlatformNetState::smReservedSocketList.resolve(socketHandle);
  1065. for (;;)
  1066. {
  1067. socklen_t addrLen = sizeof(sa);
  1068. S32 bytesRead = -1;
  1069. if (socketHandle != NetSocket::INVALID)
  1070. bytesRead = ::recvfrom(socketFd, (char *)tmpBuffer.data, Net::MaxPacketDataSize, 0, (struct sockaddr*)&sa, &addrLen);
  1071. if (bytesRead == -1)
  1072. break;
  1073. if (sa.ss_family == AF_INET)
  1074. IPSocketToNetAddress((sockaddr_in *)&sa, &srcAddress);
  1075. else if (sa.ss_family == AF_INET6)
  1076. IPSocket6ToNetAddress((sockaddr_in6 *)&sa, &srcAddress);
  1077. else
  1078. continue;
  1079. if (bytesRead <= 0)
  1080. continue;
  1081. if (srcAddress.type == NetAddress::IPAddress &&
  1082. srcAddress.address.ipv4.netNum[0] == 127 &&
  1083. srcAddress.address.ipv4.netNum[1] == 0 &&
  1084. srcAddress.address.ipv4.netNum[2] == 0 &&
  1085. srcAddress.address.ipv4.netNum[3] == 1 &&
  1086. srcAddress.port == PlatformNetState::netPort)
  1087. continue;
  1088. tmpBuffer.size = bytesRead;
  1089. Net::smPacketReceive.trigger(srcAddress, tmpBuffer);
  1090. }
  1091. }
  1092. NetSocket Net::openSocket()
  1093. {
  1094. return PlatformNetState::smReservedSocketList.reserve();
  1095. }
  1096. Net::Error Net::closeSocket(NetSocket handleFd)
  1097. {
  1098. if(handleFd != NetSocket::INVALID)
  1099. {
  1100. SOCKET socketFd = PlatformNetState::smReservedSocketList.resolve(handleFd);
  1101. PlatformNetState::smReservedSocketList.remove(handleFd);
  1102. if(!::closesocket(socketFd))
  1103. return NoError;
  1104. else
  1105. return PlatformNetState::getLastError();
  1106. }
  1107. else
  1108. return NotASocket;
  1109. }
  1110. Net::Error Net::connect(NetSocket handleFd, const NetAddress *address)
  1111. {
  1112. if(!(address->type == NetAddress::IPAddress || address->type == NetAddress::IPV6Address))
  1113. return WrongProtocolType;
  1114. SOCKET socketFd = PlatformNetState::smReservedSocketList.resolve(handleFd);
  1115. if (address->type == NetAddress::IPAddress)
  1116. {
  1117. sockaddr_in socketAddress;
  1118. NetAddressToIPSocket(address, &socketAddress);
  1119. if (socketFd == InvalidSocketHandle)
  1120. {
  1121. socketFd = PlatformNetState::smReservedSocketList.activate(handleFd, AF_INET, false);
  1122. }
  1123. if (!::connect(socketFd, (struct sockaddr *) &socketAddress, sizeof(socketAddress)))
  1124. return NoError;
  1125. }
  1126. else if (address->type == NetAddress::IPV6Address)
  1127. {
  1128. sockaddr_in6 socketAddress;
  1129. NetAddressToIPSocket6(address, &socketAddress);
  1130. if (socketFd == InvalidSocketHandle)
  1131. {
  1132. socketFd = PlatformNetState::smReservedSocketList.activate(handleFd, AF_INET6, false);
  1133. }
  1134. if (!::connect(socketFd, (struct sockaddr *) &socketAddress, sizeof(socketAddress)))
  1135. return NoError;
  1136. }
  1137. return PlatformNetState::getLastError();
  1138. }
  1139. Net::Error Net::listen(NetSocket handleFd, S32 backlog)
  1140. {
  1141. SOCKET socketFd = PlatformNetState::smReservedSocketList.resolve(handleFd);
  1142. if (socketFd == InvalidSocketHandle)
  1143. return NotASocket;
  1144. if(!::listen(socketFd, backlog))
  1145. return NoError;
  1146. return PlatformNetState::getLastError();
  1147. }
  1148. NetSocket Net::accept(NetSocket handleFd, NetAddress *remoteAddress)
  1149. {
  1150. sockaddr_storage addr;
  1151. socklen_t addrLen = sizeof(addr);
  1152. SOCKET socketFd = PlatformNetState::smReservedSocketList.resolve(handleFd);
  1153. if (socketFd == InvalidSocketHandle)
  1154. return NetSocket::INVALID;
  1155. SOCKET acceptedSocketFd = ::accept(socketFd, (sockaddr *)&addr, &addrLen);
  1156. if (acceptedSocketFd != InvalidSocketHandle)
  1157. {
  1158. if (addr.ss_family == AF_INET)
  1159. {
  1160. // ipv4
  1161. IPSocketToNetAddress(((struct sockaddr_in*)&addr), remoteAddress);
  1162. }
  1163. else if (addr.ss_family == AF_INET6)
  1164. {
  1165. // ipv6
  1166. IPSocket6ToNetAddress(((struct sockaddr_in6*)&addr), remoteAddress);
  1167. }
  1168. NetSocket newHandleFd = PlatformNetState::smReservedSocketList.reserve(acceptedSocketFd);
  1169. return newHandleFd;
  1170. }
  1171. return NetSocket::INVALID;
  1172. }
  1173. Net::Error Net::bindAddress(const NetAddress &address, NetSocket handleFd, bool useUDP)
  1174. {
  1175. int error = 0;
  1176. sockaddr_storage socketAddress;
  1177. dMemset(&socketAddress, '\0', sizeof(socketAddress));
  1178. SOCKET socketFd = PlatformNetState::smReservedSocketList.resolve(handleFd);
  1179. if (socketFd == InvalidSocketHandle)
  1180. {
  1181. if (handleFd.getHandle() == -1)
  1182. return NotASocket;
  1183. }
  1184. if (address.type == NetAddress::IPAddress)
  1185. {
  1186. socketFd = PlatformNetState::smReservedSocketList.activate(handleFd, AF_INET, useUDP);
  1187. NetAddressToIPSocket(&address, (struct sockaddr_in*)&socketAddress);
  1188. error = ::bind(socketFd, (struct sockaddr*)&socketAddress, sizeof(sockaddr_in));
  1189. }
  1190. else if (address.type == NetAddress::IPV6Address)
  1191. {
  1192. socketFd = PlatformNetState::smReservedSocketList.activate(handleFd, AF_INET6, useUDP);
  1193. NetAddressToIPSocket6(&address, (struct sockaddr_in6*)&socketAddress);
  1194. error = ::bind(socketFd, (struct sockaddr*)&socketAddress, sizeof(sockaddr_in6));
  1195. }
  1196. if (!error)
  1197. return NoError;
  1198. return PlatformNetState::getLastError();
  1199. }
  1200. Net::Error Net::setBufferSize(NetSocket handleFd, S32 bufferSize)
  1201. {
  1202. S32 error;
  1203. SOCKET socketFd = PlatformNetState::smReservedSocketList.resolve(handleFd);
  1204. if (socketFd == InvalidSocketHandle)
  1205. return NotASocket;
  1206. error = ::setsockopt(socketFd, SOL_SOCKET, SO_RCVBUF, (char *) &bufferSize, sizeof(bufferSize));
  1207. if(!error)
  1208. error = ::setsockopt(socketFd, SOL_SOCKET, SO_SNDBUF, (char *) &bufferSize, sizeof(bufferSize));
  1209. if(!error)
  1210. return NoError;
  1211. return PlatformNetState::getLastError();
  1212. }
  1213. Net::Error Net::setBroadcast(NetSocket handleFd, bool broadcast)
  1214. {
  1215. S32 bc = broadcast;
  1216. SOCKET socketFd = PlatformNetState::smReservedSocketList.resolve(handleFd);
  1217. if (socketFd == InvalidSocketHandle)
  1218. return NotASocket;
  1219. S32 error = ::setsockopt(socketFd, SOL_SOCKET, SO_BROADCAST, (char*)&bc, sizeof(bc));
  1220. if(!error)
  1221. return NoError;
  1222. return PlatformNetState::getLastError();
  1223. }
  1224. Net::Error Net::setBlocking(NetSocket handleFd, bool blockingIO)
  1225. {
  1226. SOCKET socketFd = PlatformNetState::smReservedSocketList.resolve(handleFd);
  1227. if (socketFd == InvalidSocketHandle)
  1228. return NotASocket;
  1229. unsigned long notblock = !blockingIO;
  1230. S32 error = ioctl(socketFd, FIONBIO, &notblock);
  1231. if(!error)
  1232. return NoError;
  1233. return PlatformNetState::getLastError();
  1234. }
  1235. bool Net::getListenAddress(const NetAddress::Type type, NetAddress *address, bool forceDefaults)
  1236. {
  1237. if (type == NetAddress::IPAddress)
  1238. {
  1239. const char* serverIP = forceDefaults ? NULL : Con::getVariable("pref::Net::BindAddress");
  1240. if (!serverIP || serverIP[0] == '\0')
  1241. {
  1242. address->type = type;
  1243. address->port = PlatformNetState::defaultPort;
  1244. *((U32*)address->address.ipv4.netNum) = INADDR_ANY;
  1245. return true;
  1246. }
  1247. else
  1248. {
  1249. return Net::stringToAddress(serverIP, address, false);
  1250. }
  1251. }
  1252. else if (type == NetAddress::IPBroadcastAddress)
  1253. {
  1254. address->type = type;
  1255. address->port = PlatformNetState::defaultPort;
  1256. *((U32*)address->address.ipv4.netNum) = INADDR_BROADCAST;
  1257. return true;
  1258. }
  1259. else if (type == NetAddress::IPV6Address)
  1260. {
  1261. const char* serverIP6 = forceDefaults ? NULL : Con::getVariable("pref::Net::BindAddress6");
  1262. if (!serverIP6 || serverIP6[0] == '\0')
  1263. {
  1264. sockaddr_in6 addr;
  1265. dMemset(&addr, '\0', sizeof(addr));
  1266. addr.sin6_port = htons(PlatformNetState::defaultPort);
  1267. addr.sin6_addr = in6addr_any;
  1268. IPSocket6ToNetAddress(&addr, address);
  1269. return true;
  1270. }
  1271. else
  1272. {
  1273. return Net::stringToAddress(serverIP6, address, false);
  1274. }
  1275. }
  1276. else if (type == NetAddress::IPV6MulticastAddress)
  1277. {
  1278. const char* multicastAddressValue = forceDefaults ? NULL : Con::getVariable("pref::Net::Multicast6Address");
  1279. if (!multicastAddressValue || multicastAddressValue[0] == '\0')
  1280. {
  1281. multicastAddressValue = TORQUE_NET_DEFAULT_MULTICAST_ADDRESS;
  1282. }
  1283. return Net::stringToAddress(multicastAddressValue, address, false);
  1284. }
  1285. else
  1286. {
  1287. return false;
  1288. }
  1289. }
  1290. void Net::getIdealListenAddress(NetAddress *address)
  1291. {
  1292. dMemset(address, '\0', sizeof(NetAddress));
  1293. if (Net::smIpv6Enabled)
  1294. {
  1295. if (Net::getListenAddress(NetAddress::IPV6Address, address) == NeedHostLookup)
  1296. {
  1297. Net::getListenAddress(NetAddress::IPV6Address, address, true);
  1298. }
  1299. }
  1300. else
  1301. {
  1302. if (Net::getListenAddress(NetAddress::IPAddress, address) == NeedHostLookup)
  1303. {
  1304. Net::getListenAddress(NetAddress::IPAddress, address, true);
  1305. }
  1306. }
  1307. }
  1308. Net::Error Net::send(NetSocket handleFd, const U8 *buffer, S32 bufferSize)
  1309. {
  1310. SOCKET socketFd = PlatformNetState::smReservedSocketList.resolve(handleFd);
  1311. if (socketFd == InvalidSocketHandle)
  1312. return NotASocket;
  1313. errno = 0;
  1314. S32 bytesWritten = ::send(socketFd, (const char*)buffer, bufferSize, 0);
  1315. if(bytesWritten == -1)
  1316. #if defined(TORQUE_USE_WINSOCK)
  1317. Con::errorf("Could not write to socket. Error: %s",strerror_wsa( WSAGetLastError() ));
  1318. #else
  1319. Con::errorf("Could not write to socket. Error: %s",strerror(errno));
  1320. #endif
  1321. return PlatformNetState::getLastError();
  1322. }
  1323. Net::Error Net::recv(NetSocket handleFd, U8 *buffer, S32 bufferSize, S32 *bytesRead)
  1324. {
  1325. SOCKET socketFd = PlatformNetState::smReservedSocketList.resolve(handleFd);
  1326. if (socketFd == InvalidSocketHandle)
  1327. return NotASocket;
  1328. *bytesRead = ::recv(socketFd, (char*)buffer, bufferSize, 0);
  1329. if(*bytesRead == -1)
  1330. return PlatformNetState::getLastError();
  1331. return NoError;
  1332. }
  1333. bool Net::compareAddresses(const NetAddress *a1, const NetAddress *a2)
  1334. {
  1335. return a1->isSameAddressAndPort(*a2);
  1336. }
  1337. Net::Error Net::stringToAddress(const char *addressString, NetAddress *address, bool hostLookup, int requiredFamily)
  1338. {
  1339. char addr[256];
  1340. int port = 0;
  1341. int actualFamily = AF_UNSPEC;
  1342. if (!PlatformNetState::extractAddressParts(addressString, addr, port, actualFamily))
  1343. {
  1344. return WrongProtocolType;
  1345. }
  1346. // Make sure family matches (in cast we have IP: stuff in address)
  1347. if (requiredFamily != AF_UNSPEC && actualFamily != AF_UNSPEC && (actualFamily != requiredFamily))
  1348. {
  1349. return WrongProtocolType;
  1350. }
  1351. if (actualFamily == AF_UNSPEC)
  1352. {
  1353. actualFamily = requiredFamily;
  1354. }
  1355. addressString = addr;
  1356. dMemset(address, '\0', sizeof(NetAddress));
  1357. if (!dStricmp(addressString, "broadcast"))
  1358. {
  1359. address->type = NetAddress::IPBroadcastAddress;
  1360. if (!(actualFamily == AF_UNSPEC || actualFamily == AF_INET))
  1361. return WrongProtocolType;
  1362. if (port != 0)
  1363. address->port = port;
  1364. else
  1365. address->port = PlatformNetState::defaultPort;
  1366. }
  1367. else if (!dStricmp(addressString, "multicast"))
  1368. {
  1369. address->type = NetAddress::IPV6MulticastAddress;
  1370. if (!(actualFamily == AF_UNSPEC || actualFamily == AF_INET6))
  1371. return WrongProtocolType;
  1372. if (port != 0)
  1373. address->port = port;
  1374. else
  1375. address->port = PlatformNetState::defaultPort;
  1376. }
  1377. else
  1378. {
  1379. sockaddr_in ipAddr;
  1380. sockaddr_in6 ipAddr6;
  1381. dMemset(&ipAddr, 0, sizeof(ipAddr));
  1382. dMemset(&ipAddr6, 0, sizeof(ipAddr6));
  1383. bool hasInterface = dStrchr(addressString, '%') != NULL; // if we have an interface, best use getaddrinfo to parse
  1384. // Check if we've got a simple ipv4 / ipv6
  1385. if (inet_pton(AF_INET, addressString, &ipAddr.sin_addr) == 1)
  1386. {
  1387. if (!(actualFamily == AF_UNSPEC || actualFamily == AF_INET))
  1388. return WrongProtocolType;
  1389. IPSocketToNetAddress(((struct sockaddr_in*)&ipAddr), address);
  1390. if (port != 0)
  1391. address->port = port;
  1392. else
  1393. address->port = PlatformNetState::defaultPort;
  1394. return NoError;
  1395. }
  1396. else if (!hasInterface && inet_pton(AF_INET6, addressString, &ipAddr6.sin6_addr) == 1)
  1397. {
  1398. if (!(actualFamily == AF_UNSPEC || actualFamily == AF_INET6))
  1399. return WrongProtocolType;
  1400. IPSocket6ToNetAddress(((struct sockaddr_in6*)&ipAddr6), address);
  1401. if (port != 0)
  1402. address->port = port;
  1403. else
  1404. address->port = PlatformNetState::defaultPort;
  1405. return NoError;
  1406. }
  1407. else
  1408. {
  1409. if (!hostLookup && !hasInterface)
  1410. return NeedHostLookup;
  1411. int ret = 0;
  1412. struct addrinfo hint, *res = NULL;
  1413. dMemset(&hint, 0, sizeof(hint));
  1414. hint.ai_family = actualFamily;
  1415. hint.ai_flags = hostLookup ? 0 : AI_NUMERICHOST;
  1416. if (ret = getaddrinfo(addressString, NULL, &hint, &res) == 0)
  1417. {
  1418. if (actualFamily != AF_UNSPEC)
  1419. {
  1420. // Prefer desired protocol
  1421. res = PlatformNetState::pickAddressByProtocol(res, actualFamily);
  1422. }
  1423. if (res && res->ai_family == AF_INET)
  1424. {
  1425. // ipv4
  1426. IPSocketToNetAddress(((struct sockaddr_in*)res->ai_addr), address);
  1427. }
  1428. else if (res && res->ai_family == AF_INET6)
  1429. {
  1430. // ipv6
  1431. IPSocket6ToNetAddress(((struct sockaddr_in6*)res->ai_addr), address);
  1432. }
  1433. else
  1434. {
  1435. // unknown
  1436. return UnknownError;
  1437. }
  1438. if (port != 0)
  1439. address->port = port;
  1440. else
  1441. address->port = PlatformNetState::defaultPort;
  1442. }
  1443. }
  1444. }
  1445. return NoError;
  1446. }
  1447. void Net::addressToString(const NetAddress *address, char addressString[256])
  1448. {
  1449. if(address->type == NetAddress::IPAddress || address->type == NetAddress::IPBroadcastAddress)
  1450. {
  1451. sockaddr_in ipAddr;
  1452. NetAddressToIPSocket(address, &ipAddr);
  1453. if (ipAddr.sin_addr.s_addr == htonl(INADDR_BROADCAST) || address->type == NetAddress::IPBroadcastAddress)
  1454. {
  1455. if (ipAddr.sin_port == 0)
  1456. dSprintf(addressString, 256, "IP:Broadcast");
  1457. else
  1458. dSprintf(addressString, 256, "IP:Broadcast:%d", ntohs(ipAddr.sin_port));
  1459. }
  1460. else
  1461. {
  1462. char buffer[256];
  1463. buffer[0] = '\0';
  1464. sockaddr_in ipAddr;
  1465. NetAddressToIPSocket(address, &ipAddr);
  1466. inet_ntop(AF_INET, &(ipAddr.sin_addr), buffer, sizeof(buffer));
  1467. if (ipAddr.sin_port == 0)
  1468. dSprintf(addressString, 256, "IP:%s", buffer);
  1469. else
  1470. dSprintf(addressString, 256, "IP:%s:%i", buffer, ntohs(ipAddr.sin_port));
  1471. }
  1472. }
  1473. else if (address->type == NetAddress::IPV6Address)
  1474. {
  1475. char buffer[256];
  1476. buffer[0] = '\0';
  1477. sockaddr_in6 ipAddr;
  1478. NetAddressToIPSocket6(address, &ipAddr);
  1479. inet_ntop(AF_INET6, &(ipAddr.sin6_addr), buffer, sizeof(buffer));
  1480. if (ipAddr.sin6_port == 0)
  1481. dSprintf(addressString, 256, "IP6:%s", buffer);
  1482. else
  1483. dSprintf(addressString, 256, "IP6:[%s]:%i", buffer, ntohs(ipAddr.sin6_port));
  1484. }
  1485. else if (address->type == NetAddress::IPV6MulticastAddress)
  1486. {
  1487. if (address->port == 0)
  1488. dSprintf(addressString, 256, "IP6:Multicast");
  1489. else
  1490. dSprintf(addressString, 256, "IP6:Multicast:%d", address->port);
  1491. }
  1492. else
  1493. {
  1494. *addressString = 0;
  1495. return;
  1496. }
  1497. }
  1498. void Net::enableMulticast()
  1499. {
  1500. SOCKET socketFd;
  1501. if (Net::smIpv6Enabled)
  1502. {
  1503. socketFd = PlatformNetState::smReservedSocketList.resolve(PlatformNetState::udp6Socket);
  1504. if (socketFd != InvalidSocketHandle)
  1505. {
  1506. PlatformNetState::multicast6Socket = PlatformNetState::udp6Socket;
  1507. Net::Error error = NoError;
  1508. if (error == NoError)
  1509. {
  1510. unsigned long multicastTTL = 1;
  1511. if (setsockopt(socketFd, IPPROTO_IPV6, IPV6_MULTICAST_HOPS,
  1512. (char*)&multicastTTL, sizeof(multicastTTL)) < 0)
  1513. {
  1514. error = PlatformNetState::getLastError();
  1515. }
  1516. }
  1517. // Find multicast to bind to...
  1518. NetAddress multicastAddress;
  1519. sockaddr_in6 multicastSocketAddress;
  1520. const char *multicastAddressValue = Con::getVariable("pref::Net::Multicast6Address");
  1521. if (!multicastAddressValue || multicastAddressValue[0] == '\0')
  1522. {
  1523. multicastAddressValue = TORQUE_NET_DEFAULT_MULTICAST_ADDRESS;
  1524. }
  1525. error = Net::stringToAddress(multicastAddressValue, &multicastAddress, false);
  1526. if (error == NoError)
  1527. {
  1528. dMemset(&PlatformNetState::multicast6Group, '\0', sizeof(&PlatformNetState::multicast6Group));
  1529. NetAddressToIPSocket6(&multicastAddress, &multicastSocketAddress);
  1530. dMemcpy(&PlatformNetState::multicast6Group.ipv6mr_multiaddr, &multicastSocketAddress.sin6_addr, sizeof(PlatformNetState::multicast6Group.ipv6mr_multiaddr));
  1531. }
  1532. // Setup group
  1533. if (error == NoError)
  1534. {
  1535. const char *multicastInterface = Con::getVariable("pref::Net::Multicast6Interface");
  1536. if (multicastInterface && multicastInterface[0] != '\0')
  1537. {
  1538. #ifdef TORQUE_USE_WINSOCK
  1539. PlatformNetState::multicast6Group.ipv6mr_interface = dAtoi(multicastInterface);
  1540. #else
  1541. PlatformNetState::multicast6Group.ipv6mr_interface = if_nametoindex(multicastInterface);
  1542. #endif
  1543. }
  1544. else
  1545. {
  1546. PlatformNetState::multicast6Group.ipv6mr_interface = 0; // 0 == accept from any interface
  1547. }
  1548. if (PlatformNetState::multicast6Group.ipv6mr_interface && error == NoError)
  1549. {
  1550. if (setsockopt(socketFd, IPPROTO_IPV6, IPV6_MULTICAST_IF, (char *)&PlatformNetState::multicast6Group.ipv6mr_interface, sizeof(PlatformNetState::multicast6Group.ipv6mr_interface)) < 0)
  1551. {
  1552. error = PlatformNetState::getLastError();
  1553. }
  1554. }
  1555. if (error == NoError && setsockopt(socketFd, IPPROTO_IPV6, IPV6_JOIN_GROUP, (char*)&PlatformNetState::multicast6Group, sizeof(PlatformNetState::multicast6Group)) < 0)
  1556. {
  1557. error = PlatformNetState::getLastError();
  1558. }
  1559. }
  1560. if (error == NoError)
  1561. {
  1562. Con::printf("Multicast initialized on port %d", PlatformNetState::defaultPort);
  1563. }
  1564. else
  1565. {
  1566. PlatformNetState::multicast6Socket = NetSocket::INVALID;
  1567. Con::printf("Unable to multicast UDP - error %d", error);
  1568. }
  1569. }
  1570. }
  1571. }
  1572. void Net::disableMulticast()
  1573. {
  1574. if (PlatformNetState::multicast6Socket != NetSocket::INVALID)
  1575. {
  1576. PlatformNetState::multicast6Socket = NetSocket::INVALID;
  1577. }
  1578. }
  1579. bool Net::isMulticastEnabled()
  1580. {
  1581. return PlatformNetState::multicast6Socket != NetSocket::INVALID;
  1582. }
  1583. U32 NetAddress::getHash() const
  1584. {
  1585. U32 value = 0;
  1586. switch (type)
  1587. {
  1588. case NetAddress::IPAddress:
  1589. value = Torque::hash((const U8*)&address.ipv4.netNum, sizeof(address.ipv4.netNum), 0);
  1590. break;
  1591. case NetAddress::IPV6Address:
  1592. value = Torque::hash((const U8*)address.ipv6.netNum, sizeof(address.ipv6.netNum), 0);
  1593. break;
  1594. default:
  1595. value = 0;
  1596. break;
  1597. }
  1598. return value;
  1599. }
  1600. bool Net::isAddressTypeAvailable(NetAddress::Type addressType)
  1601. {
  1602. switch (addressType)
  1603. {
  1604. case NetAddress::IPAddress:
  1605. return PlatformNetState::udpSocket != NetSocket::INVALID;
  1606. case NetAddress::IPV6Address:
  1607. return PlatformNetState::udp6Socket != NetSocket::INVALID;
  1608. case NetAddress::IPBroadcastAddress:
  1609. return PlatformNetState::udpSocket != NetSocket::INVALID;
  1610. case NetAddress::IPV6MulticastAddress:
  1611. return PlatformNetState::multicast6Socket != NetSocket::INVALID;
  1612. default:
  1613. return false;
  1614. }
  1615. }