Added ip spoofing protectection.

2010-06-22 15:31:11 -05:00
parent 80b7af4809
commit b13bbd7b40
11 changed files with 553 additions and 4 deletions
--- a/indra/llmessage/llnetcanary.cpp
+++ b/indra/llmessage/llnetcanary.cpp
@@ -0,0 +1,140 @@
+// <edit>
+#include "llnetcanary.h"
+#include "llerror.h"
+#ifdef _MSC_VER
+#include <winsock2.h>
+static WSADATA trapWSAData;
+#else
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <netdb.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <arpa/inet.h>
+#include <errno.h>
+#include <fcntl.h>
+
+extern int errno; //error number
+#define SOCKADDR_IN struct sockaddr_in
+#define SOCKADDR struct sockaddr
+#define SOCKET int
+#define INVALID_SOCKET -1
+#define SOCKET_ERROR -1
+#define SD_BOTH 2
+#define closesocket close
+#endif
+SOCKADDR_IN trapLclAddr;
+LLNetCanary::LLNetCanary(int requested_port)
+{
+	mGood = TRUE;
+	int nRet;
+	int hSocket;
+	int snd_size = 400000;
+	int rec_size = 400000;
+	int buff_size = 4;
+#ifdef _MSC_VER
+	if(WSAStartup(0x0202, &trapWSAData))
+	{
+		S32 err = WSAGetLastError();
+		WSACleanup();
+		llwarns << "WSAStartup() failure: " << err << llendl;
+		mGood = FALSE;
+		return;
+	}
+#endif
+	// Get a datagram socket
+    	hSocket = (int)socket(AF_INET, SOCK_DGRAM, 0);
+    	if (hSocket == INVALID_SOCKET)
+	{
+#ifdef _MSC_VER
+		S32 err = WSAGetLastError();
+		WSACleanup();
+#else
+		S32 err = errno;
+#endif
+		mGood = FALSE;
+		llwarns << "socket() failure: " << err << llendl;
+		return;
+	}
+
+	// Name the socket (assign the local port number to receive on)
+	trapLclAddr.sin_family      = AF_INET;
+	trapLclAddr.sin_addr.s_addr = htonl(INADDR_ANY);
+	trapLclAddr.sin_port        = htons(requested_port);
+	//llinfos << "bind() port: " << requested_port << llendl;
+	nRet = bind(hSocket, (struct sockaddr*) &trapLclAddr, sizeof(trapLclAddr));
+	if (nRet == SOCKET_ERROR)
+	{
+#ifdef _MSC_VER
+		S32 err = WSAGetLastError();
+		WSACleanup();
+#else
+		S32 err = errno;
+#endif
+		mGood = FALSE;
+		llwarns << "bind() failed: " << err << llendl;
+		return;
+	}
+
+	// Find out what address we got
+	SOCKADDR_IN socket_address;
+	S32 socket_address_size = sizeof(socket_address);
+	getsockname(hSocket, (SOCKADDR*) &socket_address, (socklen_t *)&socket_address_size);
+	mPort = ntohs(socket_address.sin_port);
+	//llinfos << "got port " << mPort << llendl;
+	
+	// Set socket to be non-blocking
+#ifdef _MSC_VER
+	unsigned long argp = 1;
+	nRet = ioctlsocket (hSocket, FIONBIO, &argp);
+#else
+	nRet = fcntl(hSocket, F_SETFL, O_NONBLOCK);
+#endif
+	if (nRet == SOCKET_ERROR) 
+	{
+#ifdef _MSC_VER
+		S32 err = WSAGetLastError();
+		WSACleanup();
+#else
+		S32 err = errno;
+#endif
+		mGood = FALSE;
+		llwarns << "Failed to set socket non-blocking, Err: " << err << llendl;
+		return;
+	}
+
+	// set a large receive buffer
+	nRet = setsockopt(hSocket, SOL_SOCKET, SO_RCVBUF, (char *)&rec_size, buff_size);
+	if (nRet)
+	{
+		llinfos << "Can't set receive buffer size!" << llendl;
+	}
+	// set a large send buffer
+	nRet = setsockopt(hSocket, SOL_SOCKET, SO_SNDBUF, (char *)&snd_size, buff_size);
+	if (nRet)
+	{
+		llinfos << "Can't set send buffer size!" << llendl;
+	}
+	//getsockopt(hSocket, SOL_SOCKET, SO_RCVBUF, (char *)&rec_size, &buff_size);
+	//getsockopt(hSocket, SOL_SOCKET, SO_SNDBUF, (char *)&snd_size, &buff_size);
+	//LL_DEBUGS("AppInit") << "startNet - receive buffer size : " << rec_size << LL_ENDL;
+	//LL_DEBUGS("AppInit") << "startNet - send buffer size    : " << snd_size << LL_ENDL;
+
+	mSocket = hSocket;
+}
+LLNetCanary::~LLNetCanary()
+{
+	if(mGood)
+	{
+		if(mSocket)
+		{
+			shutdown(mSocket, SD_BOTH);
+			closesocket(mSocket);
+		}
+#ifdef _MSC_VER
+		WSACleanup();
+#endif
+	}
+}
+// </edit>