Esempio n. 1
0
/**
 * @brief	Helps to test NetModeUdp objects.
 *
 * Client ID used with DealWithData is 10, and operation ID is 5.
 *
 * @param [in,out]	obj				The object. 
 * @param [in,out]	packet			The packet. 
 * @param str						The string stored in @a packet.
 * @param dealWithDataClientID		Client ID to use with DealWithData().
 * @param expectedClientID			Expected result of destination.GetClientFrom().
 * @param operationID				Operation ID to use with DealWithData().
 *
 * @return	false if the test was failed and a problem was found while testing, true if not.
 */
bool NetModeUdp::_HelperTestClass(NetModeUdp & obj, Packet & packet, const char * str, size_t dealWithDataClientID, size_t expectedClientID, size_t operationID)
{
	Packet destination;
	WSABUF buffer;

	bool problem = false;

	// Counter: 500
	packet.SetCursor(0);
	packet.AddSizeT(500);
	packet.PtrIntoWSABUF(buffer);
	obj.DealWithData(buffer,packet.GetUsedSize(),NULL,dealWithDataClientID,1);
	
	if(obj.GetPacketFromStore(&destination,expectedClientID,operationID) != 1 ||
	   destination.GetClientFrom() != expectedClientID ||
	   destination.GetInstance() != 1||
	   destination.GetAge() != 500 ||
	   (destination.GetOperation() != operationID && destination.GetOperation() != 0))
	{
		cout << "DealWithData is bad (packet 1)\n";
		problem = true;
	}
	else
	{
		destination.Erase(0,destination.GetCursor());
		if(destination != str)
		{
			cout << "DealWithData is bad due to contents (packet 1)\n";
			problem = true;
		}
		else
		{
			cout << "DealWithData is good (packet 1)\n";
		}
	}

	// Counter: 499
	packet.SetCursor(0);
	packet.AddSizeT(499);
	packet.PtrIntoWSABUF(buffer);
	obj.DealWithData(buffer,packet.GetUsedSize(),NULL,dealWithDataClientID,1);
	
	if(obj.GetPacketFromStore(&destination,expectedClientID,operationID) != 0)
	{
		cout << "DealWithData is bad (packet 2)\n";
		problem = true;
	}
	else
	{
		cout << "DealWithData is good (packet 2)\n";
	}

	// Counter: 501
	packet.SetCursor(0);
	packet.AddSizeT(501);
	packet.PtrIntoWSABUF(buffer);
	obj.DealWithData(buffer,packet.GetUsedSize(),NULL,dealWithDataClientID,3);
	
	if(obj.GetPacketFromStore(&destination,expectedClientID,operationID) != 1 ||
	   destination.GetClientFrom() != expectedClientID ||
	   destination.GetInstance() != 3 ||
	   destination.GetAge() != 501 ||
	   (destination.GetOperation() != operationID && destination.GetOperation() != 0))
	{
		cout << "DealWithData is bad (packet 3)\n";
		problem = true;
	}
	else
	{
		destination.Erase(0,destination.GetCursor());
		if(destination != str)
		{
			cout << "DealWithData is bad due to contents (packet 3)\n";
			problem = true;
		}
		else
		{
			cout << "DealWithData is good (packet 3)\n";
		}
	}
	
	// Counter: very large (1000 from maximum) 
	packet.SetCursor(0);
	packet.AddSizeT(-1000);
	packet.PtrIntoWSABUF(buffer);
	obj.DealWithData(buffer,packet.GetUsedSize(),NULL,dealWithDataClientID,3);
	
	if(obj.GetPacketFromStore(&destination,expectedClientID,operationID) != 1 ||
	   destination.GetClientFrom() != expectedClientID ||
	   destination.GetInstance() != 3 ||
	   destination.GetAge() != -1000 ||
	   (destination.GetOperation() != operationID && destination.GetOperation() != 0))
	{
		cout << "DealWithData is bad (packet 4)\n";
		problem = true;
	}
	else
	{
		destination.Erase(0,destination.GetCursor());
		if(destination != str)
		{
			cout << "DealWithData is bad due to contents (packet 4)\n";
			problem = true;
		}
		else
		{
			cout << "DealWithData is good (packet 4)\n";
		}
	}
	
	// Counter: 1 (testing code that an entity can continue running
	// forever, when the counter reaches maximum it will loop round to 0).
	packet.SetCursor(0);
	packet.AddSizeT(1);
	packet.PtrIntoWSABUF(buffer);
	obj.DealWithData(buffer,packet.GetUsedSize(),NULL,dealWithDataClientID,3);
	
	if(obj.GetPacketFromStore(&destination,expectedClientID,operationID) != 1 ||
	   destination.GetClientFrom() != expectedClientID ||
	   destination.GetInstance() != 3 ||
	   destination.GetAge() != 1 ||
	   (destination.GetOperation() != operationID && destination.GetOperation() != 0))
	{
		cout << "DealWithData is bad (packet 5)\n";
		problem = true;
	}
	else
	{
		destination.Erase(0,destination.GetCursor());
		if(destination != str)
		{
			cout << "DealWithData is bad due to contents (packet 5)\n";
			problem = true;
		}
		else
		{
			cout << "DealWithData is good (packet 5)\n";
		}
	}

	return !problem;
}
Esempio n. 2
0
/**
 * @brief Tests class.
 *
 * @return true if no problems while testing were found, false if not.
 * Note that not all tests automatically check for problems so some tests
 * require manual verification.
 */
bool NetModeTcpPostfix::TestClass()
{
	cout << "Testing NetModeTcpPostfix class...\n";
	bool problem = false;

	Packet postfix("\r\n");
	NetModeTcpPostfix obj(1024,true,postfix,new MemoryRecyclePacket(10,12));

	WSABUF buf;
	buf.buf = "hello world";
	buf.len = static_cast<DWORD>(strlen(buf.buf));

	WSABUF buf2;
	buf2.buf = "\r\n";
	buf2.len = 2;

	WSABUF buf3;
	buf3.buf = "here we add multiple packets in one go\r\nThis is another packet!\r\nAnd another\r\n\r\n";
	buf3.len = static_cast<DWORD>(strlen(buf3.buf));

	obj.DealWithData(buf,buf.len,NULL,1,2);

	if(obj.GetPartialPacketUsedSize() != buf.len ||
	   obj.GetPacketAmount() != 0)
	{
		cout << "DealWithData is bad\n";
		problem = true;
	}
	else
	{
		cout << "DealWithData is good\n";
	}

	if(obj.GetMemorySize() == 0)
	{
		cout << "GetMemorySize is bad\n";
		problem = true;
	}
	else
	{
		cout << "GetMemorySize is good: " << obj.GetMemorySize() << "\n";
	}
	
	
	obj.DealWithData(buf2,buf2.len,NULL,1,2);

	if(obj.GetPartialPacketUsedSize() != 0 ||
	   obj.GetPacketAmount() != 1)
	{
		cout << "DealWithData is bad\n";
		problem = true;
	}
	else
	{
		cout << "DealWithData is good\n";
	}

	if(obj.GetMemorySize() == 0)
	{
		cout << "GetMemorySize is bad\n";
		problem = true;
	}
	else
	{
		cout << "GetMemorySize is good: " << obj.GetMemorySize() << "\n";
	}
	
	Packet retrieved;
	obj.GetPacketFromStore(&retrieved);

	if(retrieved.compareWSABUF(buf,buf.len) == false ||
	   retrieved.GetClientFrom() != 1 ||
	   retrieved.GetInstance() != 2)
	{
		cout << "GetPacketFromStore is bad\n";
		problem = true;
	}
	else
	{
		cout << "GetPacketFromStore is good\n";
	}

	if(obj.GetMemorySize() == 0)
	{
		cout << "GetMemorySize is bad\n";
		problem = true;
	}
	else
	{
		cout << "GetMemorySize is good: " << obj.GetMemorySize() << "\n";
	}
	
	obj.DealWithData(buf3,buf3.len,NULL,3,2);
	if(obj.GetPartialPacketUsedSize() != 0 ||
	   obj.GetPacketAmount() != 4)
	{
		cout << "DealWithData is bad\n";
		problem = true;
	}
	else
	{
		cout << "DealWithData is good\n";
	}

	if(obj.GetMemorySize() == 0)
	{
		cout << "GetMemorySize is bad\n";
		problem = true;
	}
	else
	{
		cout << "GetMemorySize is good: " << obj.GetMemorySize() << "\n";
	}

	obj.GetPacketFromStore(&retrieved);
	if(retrieved.GetUsedSize() != 38)
	{
		cout << "DealWithData is bad (packet 1)\n";
		problem = true;
	}
	else
	{
		cout << "DealWithData is good (packet 1)\n";
	}

	if(obj.GetMemorySize() == 0)
	{
		cout << "GetMemorySize is bad\n";
		problem = true;
	}
	else
	{
		cout << "GetMemorySize is good: " << obj.GetMemorySize() << "\n";
	}

	obj.GetPacketFromStore(&retrieved);
	if(retrieved.GetUsedSize() != 23)
	{
		cout << "DealWithData is bad (packet 2)\n";
		problem = true;
	}
	else
	{
		cout << "DealWithData is good (packet 2)\n";
	}

	if(obj.GetMemorySize() == 0)
	{
		cout << "GetMemorySize is bad\n";
		problem = true;
	}
	else
	{
		cout << "GetMemorySize is good: " << obj.GetMemorySize() << "\n";
	}

	obj.GetPacketFromStore(&retrieved);
	if(retrieved.GetUsedSize() != 11)
	{
		cout << "DealWithData is bad (packet 3)\n";
		problem = true;
	}
	else
	{
		cout << "DealWithData is good (packet 3)\n";
	}

	if(obj.GetMemorySize() == 0)
	{
		cout << "GetMemorySize is bad\n";
		problem = true;
	}
	else
	{
		cout << "GetMemorySize is good: " << obj.GetMemorySize() << "\n";
	}

	obj.GetPacketFromStore(&retrieved);
	if(retrieved.GetUsedSize() != 0)
	{
		cout << "DealWithData is bad (packet 4)\n";
		problem = true;
	}
	else
	{
		cout << "DealWithData is good (packet 4)\n";
	}

	if(obj.GetMemorySize() == 0)
	{
		cout << "GetMemorySize is bad\n";
		problem = true;
	}
	else
	{
		cout << "GetMemorySize is good: " << obj.GetMemorySize() << "\n";
	}

	// Adding of this packet succeeds.
	obj.SetMemoryLimit(obj.GetMemorySize() + (buf.len*2));
	obj.DealWithData(buf,buf.len,NULL,0,0);
	obj.DealWithData(buf,buf.len,NULL,0,0);
	obj.DealWithData(buf2,buf2.len,NULL,0,0);

	size_t originalMemorySize = obj.GetMemorySize();

	bool exception = false;
	try
	{
		// Adding of this packet fails due to memory limit.
		obj.DealWithData(buf,buf.len,NULL,0,0);
		obj.DealWithData(buf,buf.len,NULL,0,0);
		obj.DealWithData(buf2,buf2.len,NULL,0,0);
	}
	catch(ErrorReport & report)
	{
		exception = true;
	}

	if(exception && obj.GetMemorySize() == originalMemorySize)
	{
		cout << "DealWithData memory restriction success.\n";
	}
	else
	{
		cout << "DealWithData memory restriction failure " << obj.GetMemorySize() << "\n";
	}

	cout << "\n\n";
	return !problem;
}