/*

 * Copyright (c) 1997-2009 Nokia Corporation and/or its subsidiary(-ies).

 * All rights reserved.

 * This component and the accompanying materials are made available

 * under the terms of "Eclipse Public License v1.0"

 * which accompanies this distribution, and is available

 * at the URL "http://www.eclipse.org/legal/epl-v10.html".

 *

 * Initial Contributors:

 * Nokia Corporation - initial contribution.

 *

 * Contributors:

 *

 * Description:

 * Test of AF_INET stream sockets

 * 

 *

 */

 #include <stdio.h>

 #include <errno.h>

 #include <stdlib.h>

 #include <string.h>

 #include <unistd.h>

 #include <sys/types.h>

 #include <sys/socket.h>

 #include <sys/ioctl.h>

 #include <libc/netinet/in.h>

 #include <libc/arpa/inet.h>

 #include <netdb.h>

 #include "CTEST.H"	/* includes C interface to EPOC32 threads, and SpawnPosixServer */

 test_Data;

 struct sockaddr_in testaddr;

 unsigned short int port = 2001;

 /**

 @SYMTestCaseID          SYSLIB-STDLIB-CT-1059

 @SYMTestCaseDesc	    Tests for AF_INET stream sockets 

 @SYMTestPriority 	    High

 @SYMTestActions  	    Tests for simple operations on stream sockets,open,bind,get socket name,listen operations and test for error code

 @SYMTestExpectedResults Test must not fail

 @SYMREQ                 REQ0000

 */		

 void testSimple()

 	{

 	int fd1, fd2;

 	size_t addrsize;

 	int err;

 	struct sockaddr_in addr1, addr2;

 	int optionbuf[20];

 	size_t optionsize;

 	test_Next("Create stream sockets");

 	fd1=socket(AF_INET, SOCK_STREAM, 0);

 	test_ok(fd1>=0);

 	fd2=socket(AF_INET, SOCK_STREAM, 0);

 	test_ok(fd2>=0);

 	test_Next("Some binding tests");

 	addr1.sin_family=2061;

 	addr1.sin_port=htons(65530);

 	addr1.sin_addr.s_addr=htonl(0x11223344);

 	err=bind(fd1, (struct sockaddr*)&addr1, sizeof(addr1));	/* wrong family, port out of range */

 	test_errno(err<0,ENOENT);

 	addr1=testaddr;

 	addr1.sin_port=htons(port);

 	err=bind(fd1, (struct sockaddr*)&addr1, sizeof(addr1));

 	test_ok(err==0);

 	addr1=testaddr;

 	addr1.sin_port=htons((unsigned short int)(port+1));

 	err=bind(fd1, (struct sockaddr*)&addr1, sizeof(addr1));	/* already bound */

 	test_errno(err!=0, EEXIST);

 	test_Next("Get associated addresses");

 	addrsize=sizeof(addr2);

 	err=getsockname(fd1,(struct sockaddr*)&addr2,&addrsize);

 	test_ok(err==0);

 	test(addr2.sin_family==AF_INET);

 	test(addr2.sin_port==htons(port));

 	test(addrsize<=sizeof(addr2));

 	addrsize=sizeof(addr2);

 	err=getpeername(fd1,(struct sockaddr*)&addr2,&addrsize);	/* not connected */

 	test_errno(err!=0,-2);	/* OMISSION - can't report proper reason for failure */

 	addrsize=sizeof(addr2);

 	err=getsockname(fd2,(struct sockaddr*)&addr2,&addrsize);	/* not bound */

 	test_errno(err!=0,-2);	/* OMISSION - can't report proper reason for failure */

 	test_Next("More binding");

 	addr1=testaddr;

 	addr1.sin_port=htons(port);

 	err=bind(fd2, (struct sockaddr*)&addr1, sizeof(addr1));	/* address in use */

 	test_errno(err!=0, EACCES);

 	addr1=testaddr;

 	addr1.sin_port=0;

 	err=bind(fd2, (struct sockaddr*)&addr1, sizeof(addr1));	/* unspecified port number */

 	test_ok(err==0);

 	addrsize=sizeof(addr2);

 	err=getsockname(fd2,(struct sockaddr*)&addr2,&addrsize);

 	test_ok(err==0);

 	test(addr2.sin_family==AF_INET);

 	test(addrsize<=sizeof(addr2));

 	test(addr2.sin_port!=htons(port));

 	test(addr2.sin_port!=0);

 	{

 		unsigned short port=ntohs(addr2.sin_port);

 		test(port>=IPPORT_RESERVED);

 		/* NB. TCPIP 030 has 9999 for IPPORT_USERRESERVED, but TCPIP 031 dropped it to 5000 */

 		test(port<=IPPORT_USERRESERVED);

 		printf("  allocated port %d\n",port);

 	}

 	err=listen(fd1,1);

 	test_ok(err==0);

 	test_Next("Socket options");

 	optionbuf[0]=3500000;	/* implausible size */

 	optionsize=sizeof(optionbuf[0]);

 	err=getsockopt(fd1,SOL_SOCKET,SO_SNDBUF,optionbuf,&optionsize);

 	test_ok(err==0);

 	test(optionbuf[0]!=3500000);

 	optionbuf[0]=7*1024;

 	optionsize=sizeof(optionbuf[0]);

 	err=setsockopt(fd1,SOL_SOCKET,SO_SNDBUF,optionbuf,optionsize);

 	test_ok(err==0);

 	optionbuf[0]=3500000;	/* implausible size */

 	optionsize=sizeof(optionbuf[0]);

 	err=getsockopt(fd1,SOL_SOCKET,SO_SNDBUF,optionbuf,&optionsize);

 	test_ok(err==0);

 	test(optionbuf[0]==7*1024);

 	optionbuf[0]=1;

 	optionsize=sizeof(optionbuf[0]);

 	err=getsockopt(fd1,SOL_SOCKET,1234,optionbuf,&optionsize);	/* invalid option */

 	test_errno(err<0,ENOSYS);

 	test(optionbuf[0]==1);

 	optionbuf[0]=13;

 	optionsize=sizeof(optionbuf[0]);

 	err=setsockopt(fd1,SOL_SOCKET,1234,optionbuf,optionsize);	/* invalid option */

 	test_errno(err<0,ENOSYS);

 	test(optionbuf[0]==13);

 	err=close(fd1);

 	test_ok(err==0);

 	err=close(fd2);

 	test_ok(err==0);

 	}

 /* Client and server take it in turns to send, starting with the client.

  * Each matches the message they receive with the string expected

  */

 char *message_sequence[] = {

 	"Hello from client",

 	"Hello from server",

 	"Test of send",

 	"Test of recv",

 	"Try sendto",

 	"Try recvfrom",

 	"Try sendto again",

 	"Try recvfrom again",

 	"Send to shutdown socket",

 	"Send to closed socket",

 	0

 	};

 /**

 @SYMTestCaseID          SYSLIB-STDLIB-CT-1060

 @SYMTestCaseDesc	    Tests for server socket

 @SYMTestPriority 	    High

 @SYMTestActions  	    Create a steam socket,bind and test for a transmission process.Check for error codes 

 @SYMTestExpectedResults Test must not fail

 @SYMREQ                 REQ0000

 */		

 void testServer()

 	{

 	int fd1,fd2,nbytes,i;

 	size_t addrsize;

 	int err;

 	struct sockaddr_in addr1, addr2;

 	char buf[80];

 	char **mp = message_sequence;

 	static int randbuf[10000];

 	 fd1=fd2=nbytes=i=0;

 	 /* Fill the buffer with a predetermined random number sequence */

 	randbuf[0]=4441302;

 	srand(randbuf[0]);

 	for (i=1; i<10000; i++)

 		randbuf[i]=rand();

 	test_Next("Create server socket");

 	fd1=socket(AF_INET, SOCK_STREAM, 0);

 	test_ok(fd1>=0);

 #if 0

 	/* causes ESOCK to panic the client */

 	addrsize=sizeof(addr2);

 	fd2=accept(fd1,&addr2,&addrsize);	/* can't accept on an unbound socket */

 	test_ok(fd2<0);

 #endif

 	IN_SET_LOOPBACK_ADDR(&addr1);

 	addr1.sin_port=htons(port);

 	err=bind(fd1, (struct sockaddr*)&addr1, sizeof(addr1));

 	test_ok(err==0);

 #if 0

 	/* causes ESOCK to panic the client */

 	addrsize=sizeof(addr2);

 	fd2=accept(fd1,&addr2,&addrsize);	/* can't accept before listening */

 	test_ok(fd2<0);

 #endif

 	err=listen(fd1,1);

 	test_ok(err==0);

 	addrsize=sizeof(addr2);

 	fd2=accept(fd1,(struct sockaddr*)&addr2,&addrsize);

 	test_ok(fd2>=0);

 	test(addr2.sin_family==AF_INET);

 	test(addr2.sin_port!=htons(port));

 	test(addr2.sin_port!=0);

 	test(addrsize<=sizeof(addr2));

 	test_Next("Server read/write");

 	/* read */

 	nbytes=strlen(*mp);

 	err=read(fd2, buf, nbytes+1);

 	test_ok(err==nbytes+1);

 	test(strcmp(buf,*mp)==0);

 	/* write */

 	mp++;

 	nbytes=strlen(*mp);

 	for (i=0; i<nbytes+1; i++)

 		{

 		err=write(fd2,(*mp)+i,1);

 		test_ok(err==1);

 		}

 	test_Next("Server send/recv");

 	/* recv */

 	mp++;

 	nbytes=strlen(*mp);

 	err=recv(fd2, buf, sizeof(buf) ,0);

 	test_ok(err==nbytes+1);

 	test(strcmp(buf,*mp)==0);

 	/* send */

 	mp++;

 	nbytes=strlen(*mp);

 	err=send(fd2, *mp, nbytes+1,0);

 	test_ok(err==nbytes+1);

 	/* recvfrom */

 	mp++;

 	nbytes=strlen(*mp);

 	addrsize=sizeof(addr2);

 	addr2.sin_port=0;

 	err=recvfrom(fd2, buf, nbytes+1,0,(struct sockaddr*)&addr2,&addrsize);

 /*

 	memset(&buf[0],0, sizeof(buf));

 	err=recvfrom(fd2, buf, nbytes+1,0,0,0);

 */

 	test_ok(err==nbytes+1);

 	test(strcmp(buf,*mp)==0);

 	test(addr2.sin_family==AF_INET);

 	test(addr2.sin_port!=htons(port));

 	test(addr2.sin_port!=0);

 	test(addrsize<=sizeof(addr2));

 	/* sendto */

 	mp++;

 	nbytes=strlen(*mp);

 	addrsize=sizeof(addr1);

 	err=sendto(fd2, *mp, nbytes+1,0,(struct sockaddr*)&addr1,addrsize);	/* not allowed on streams */

 	test_errno(err<0,ENOSYS);

 	err=send(fd2, *mp, nbytes+1,0);		/* to keep synchronisation */

 	test_ok(err==nbytes+1);

 	test_Next("Server revcfrom with null fromaddr");

 		/* recvfrom again*/

 	mp++;

 	nbytes=strlen(*mp);

 	memset(&buf[0],0xFF, sizeof(buf));

 	err=recvfrom(fd2, buf, nbytes+1,0,0,0);

 	test_ok(err==nbytes+1);

 	test(strcmp(buf,*mp)==0);

 	/* sendto again*/

 	mp++;

 	nbytes=strlen(*mp);

 	addrsize=sizeof(addr1);

 	err=send(fd2, *mp, nbytes+1,0);		/* to keep synchronisation */

 	test_ok(err==nbytes+1);

 	/*

 	test_Next("Server shutdown transmission");

 	err=shutdown(fd2,1);

 	test_ok(err==0);

 	sleep(4);	// so that the client's sleep(1) finishes before we awake 

 	mp++;

 	nbytes=strlen(*mp);

 	addrsize=sizeof(addr1);

 	err=send(fd2, *mp, nbytes+1,0);

 	test_errno(err<0,EPIPE);

 	nbytes=strlen(*mp);

 	err=recv(fd2, buf, nbytes+1,0);

 	test_ok(err==nbytes+1);

 	test(strcmp(buf,*mp)==0);

 	err=close(fd2);

 	test_ok(err==0);

 	// Large transmission 

 */

 	test_Next("Server large transfer"); 

 	addrsize=sizeof(addr2);

 	fd2=accept(fd1,(struct sockaddr*)&addr2,&addrsize);

 	test_ok(fd2>=0);

 	test(addr2.sin_family==AF_INET);

 	test(addr2.sin_port!=htons(port));

 	test(addr2.sin_port!=0);

 	test(addrsize<=sizeof(addr2));

 	nbytes=sizeof(randbuf);

 	err=write(fd2,(const char*)randbuf,nbytes);

 	test_ok(err==nbytes);

 	test_Next("Server close");

 //	err=shutdown(fd2,1);	

 //	test_ok(err==0);

 //	sleep(4);

 	err=close(fd2);

 	test_ok(err==0);

 	err=close(fd1);

 	test_ok(err==0);

 	}

 /**

 @SYMTestCaseID          SYSLIB-STDLIB-CT-1061

 @SYMTestCaseDesc	    Tests for client socket

 @SYMTestPriority 	    High

 @SYMTestActions  	    Create a steam socket,bind and test for a transmission process.Check for error codes 

 @SYMTestExpectedResults Test must not fail

 @SYMREQ                 REQ0000

 */		

 void testClient()

 	{

 	int fd1, nbytes, nbytes2, i, status;

 	size_t addrsize;

 	int err;

 	struct sockaddr_in addr1, addr2;

 	char buf[80];

 	char **mp = message_sequence;

 	static int randbuf[10001];

 	test_Next("Create client socket");

 	fd1=socket(AF_INET, SOCK_STREAM, 0);

 	test_ok(fd1>=0);

 	sleep(5);	/* give server a chance to run */

 	addr1=testaddr;

 	addr1.sin_port=htons(port);

 	addrsize=sizeof(addr1);

 	err=connect(fd1,(struct sockaddr*)&addr1,addrsize);

 	test_ok(err==0);

 	addrsize=sizeof(addr2);

 	err=getpeername(fd1,(struct sockaddr*)&addr2,&addrsize);

 	test_ok(err==0);

 	test(addr2.sin_family==AF_INET);

 	test(addr2.sin_port==htons(port));

 	test(addrsize<=sizeof(addr2));

 	addrsize=sizeof(addr2);

 	err=getsockname(fd1,(struct sockaddr*)&addr2,&addrsize);

 	test_ok(err==0);

 	test(addr2.sin_family==AF_INET);

 	test(addr2.sin_port!=htons(port));

 	test(addr2.sin_port!=0);

 	test(addrsize<=sizeof(addr2));

 	test_Next("Client read/write");

 	/* write */

 	nbytes=strlen(*mp);

 	err=write(fd1, *mp, nbytes+1);

 	test_ok(err==nbytes+1);

 	/* read */

 	mp++;

 	nbytes=strlen(*mp);

 	err=read(fd1, buf, nbytes+1);

 	test_ok(err==nbytes+1);

 	test(strcmp(buf,*mp)==0);

 	test_Next("Client send/recv");

 	/* send */

 	mp++;

 	nbytes=strlen(*mp);

 	err=send(fd1,*mp, nbytes+1,0);

 	test_ok(err==nbytes+1);

 	/* recv - get the first 2 bytes so that we know the buffer is full */

 	mp++;

 	nbytes=strlen(*mp);

 	err=recv(fd1,buf,2,0);

 	test_ok(err==2);

 	/* ioctl */

 	nbytes2=-1;

 	err=ioctl(fd1,E32IONREAD,&nbytes2);

 	test_ok(err==0);

 	test(nbytes2==(nbytes+1-2));

 	nbytes2=E32SELECT_READ|E32SELECT_WRITE|E32SELECT_EXCEPT;

 	err=ioctl(fd1,E32IOSELECT,&nbytes2);

 	test_ok(err==0);

 	test(nbytes2==(E32SELECT_READ|E32SELECT_WRITE));

 	nbytes2=E32SELECT_READ|E32SELECT_WRITE|E32SELECT_EXCEPT;

 	err=async_ioctl(fd1,E32IOSELECT,&nbytes2,&status);

 	test_ok(err==0);

 	err=async_ioctl_completion(fd1,E32IOSELECT,&nbytes2,&status);

 	test_ok(err==0);

 	if (nbytes2!=(E32SELECT_READ|E32SELECT_WRITE))

 		{

 		nbytes2=E32SELECT_READ|E32SELECT_WRITE|E32SELECT_EXCEPT;

 		err=ioctl(fd1,E32IOSELECT,&nbytes2);

 		test_ok(err==0);

 		test(nbytes2==(E32SELECT_READ|E32SELECT_WRITE));

 		}		

 	/* recv - get the rest of the data */

 	for (i=2; i<nbytes+1; i++)

 		{

 		err=recv(fd1,buf+i,1,0);

 		test_ok(err==1);

 		}

 	test(strcmp(buf,*mp)==0);

 	/* ioctl again - this time there is no data pending */

 	nbytes2=-1;

 	err=ioctl(fd1,E32IONREAD,&nbytes2);

 	test_ok(err==0);

 	test(nbytes2==0);

 	nbytes2=E32SELECT_READ|E32SELECT_WRITE|E32SELECT_EXCEPT;

 	err=ioctl(fd1,E32IOSELECT,&nbytes2);

 	test_ok(err==0);

 	test(nbytes2==E32SELECT_WRITE);

 	nbytes2=E32SELECT_READ|E32SELECT_WRITE|E32SELECT_EXCEPT;

 	err=async_ioctl(fd1,E32IOSELECT,&nbytes2,&status);

 	test_ok(err==0);

 	err=async_ioctl_completion(fd1,E32IOSELECT,&nbytes2,&status);

 	test_ok(err==0);

 	test(nbytes2==E32SELECT_WRITE);

 	/* sendto */

 	mp++;

 	nbytes=strlen(*mp);

 	addrsize=sizeof(addr1);

 	err=sendto(fd1, *mp, nbytes+1,0,(struct sockaddr*)&addr1,addrsize);

 	test_errno(err<0,ENOSYS);

 	err=send(fd1, *mp, nbytes+1,0);		/* to keep synchronisation */

 	test_ok(err==nbytes+1);

 	/* recvfrom */

 	mp++;

 	nbytes=strlen(*mp);

 	addrsize=sizeof(addr2);

 	addr2.sin_port=0;

 	err=recvfrom(fd1, buf, nbytes+1,0,(struct sockaddr*)&addr2,&addrsize);

 	test_ok(err==nbytes+1);

 	test(strcmp(buf,*mp)==0);

 	test(addr2.sin_family==AF_INET);

 	test(addr2.sin_port==htons(port));

 	test(addrsize<=sizeof(addr2));

 	/* another sendto */

 	test_Next("Client revcfrom with null fromaddr");

 	mp++;

 	nbytes=strlen(*mp);

 	addrsize=sizeof(addr1);

 	err=sendto(fd1, *mp, nbytes+1,0,(struct sockaddr*)&addr1,addrsize);

 	test_errno(err<0,ENOSYS);

 	err=send(fd1, *mp, nbytes+1,0);		/* to keep synchronisation */

 	test_ok(err==nbytes+1);

 	/* recvfrom */

 	mp++;

 	nbytes=strlen(*mp);

 	addrsize=sizeof(addr2);

 	memset(buf,0xFF,sizeof(buf));

 	err=recvfrom(fd1, buf, nbytes+1,0,0,0);	//do a recvfrom with a null address

 	test_ok(err==nbytes+1);

 	test(strcmp(buf,*mp)==0);

 	sleep(2);

 /*

 	test_Next("Test half-closed connection");

 	// read from half-closed socket 

 	mp++;

 	nbytes=strlen(*mp);

 	err=read(fd1, buf, nbytes+1);

 	test_errno(err<0, EPIPE);

 	nbytes=strlen(*mp);

 	err=write(fd1,*mp,nbytes+1);

 	test_ok(err==nbytes+1);

 	sleep(2);	

 	// read from a connection closed by the other end 

 	mp++;

 	nbytes=strlen(*mp);

 	err=read(fd1,buf,nbytes+1);

 	test_errno(err<0, EPIPE);

 */

 	err=close(fd1);

 	test_ok(err==0);

 	// Large transmission - and another connection to the server socket 

 	test_Next("Test large transmission");

 	fd1=socket(AF_INET, SOCK_STREAM, 0);

 	test_ok(fd1>=0);

 	sleep(2);	/* give server a chance to run */

 	addr1=testaddr;

 	addr1.sin_port=htons(port);

 	addrsize=sizeof(addr1);

 	//this connect fails.  Why is this?

 	err=connect(fd1,(struct sockaddr*)&addr1,addrsize);

 	test_ok(err==0);

 	addrsize=sizeof(addr2);

 	err=getpeername(fd1,(struct sockaddr*)&addr2,&addrsize);

 	test_ok(err==0);

 	test(addr2.sin_family==AF_INET);

 	test(addr2.sin_port==htons(port));

 	test(addrsize<=sizeof(addr2));

 	addrsize=sizeof(addr2);

 	err=getsockname(fd1,(struct sockaddr*)&addr2,&addrsize);

 	test_ok(err==0);

 	test(addr2.sin_family==AF_INET);

 	test(addr2.sin_port!=htons(port));

 	test(addr2.sin_port!=0);

 	test(addrsize<=sizeof(addr2));

 	nbytes=0;

 	{

 		int buf;

 		char * p;

 		do

 		{

 		buf=sizeof(randbuf)-nbytes;

 		if (buf>4095)

 				buf=4095;

 		p = (char*)randbuf+nbytes;

 //		err=read(fd1, p, buf); 

 		err=recv(fd1, p, buf, 0); 

 		if (err>0)

 			nbytes+=err;

 		}

 	while (nbytes<sizeof(randbuf) && err>0);

 	}

 	printf("\nnbytes=%d, err=%d, errno=%d\n", nbytes, err, errno);

 	test_ok(err==0);	/* want a graceful EOF */

 	test(nbytes==sizeof(randbuf)-sizeof(int));	/* expect a smaller test data set */

 	test(randbuf[0]==4441302);

 	srand(randbuf[0]);

 	for (i=1; i<10000; i++)

 		test(randbuf[i]==rand());

 	err=recv(fd1, (char*)randbuf, 1, 0); 

 	test_ok(err==0);

 	err=close(fd1);

 	test_ok(err==0);

 	}

 int close_console=0;

 void testConnection()

 	{

 	int err;

 	void* server;

 	char buf[100];

 	sprintf(buf, "TISTREAM server %d", close_console);

 	server=create_thread(testServer, buf);

 	test(server!=0);

 	start_thread(server);

 	sleep(2);	/* give the server a chance to get started */

 	testClient();

 	err=wait_for_thread(server);

 	test(err==0);

 	if (close_console)

 		{

 		test_Close();

 		close(0);

 		close(1);

 		close(2);

 		}

 	}

 int main(int argc, char *argv[])

 	{

 	void* client;

 	int err;

 	test_Title("AF_INET Streams");

 	err=CommInit(0);	/* ensure a workable comms environment */

 	test(err==0);

 	IN_SET_LOOPBACK_ADDR(&testaddr);

 	if (argc==1)

 		{

 		// Run the test(s) without a CPosixServer first 

 		testSimple();

 		port +=2;

 		testConnection();

 		port +=2;

 		}

 	test_Next("Test gethostname");

 	{

 		char * hname;

 		hname = (char*)malloc(200);

 		if (hname)

 			{

 			err = gethostname(hname, 200);

 			if (err)

 				printf("gethostname, err=%d, errno=%d\n", err, errno);

 			else

 				printf("host name is %s\n", hname);

 			free(hname);			}

 		else

 			printf("host name - malloc failure\n");

 	}

 	test_Next("Do it again using the CPosixServer (for them, not me)");

 	close_console=1;

 	start_posix_server();	/* calls SpawnPosixServer from C++ code */

 	client=create_thread(testSimple, "TISTREAM simple");

 	test(client!=0);

 	start_thread(client);

 	err=wait_for_thread(client);

 	test(err==0);

 	port +=2;

 	client=create_thread(testConnection, "TISTREAM client");

 	test(client!=0);

 	start_thread(client);

 	err=wait_for_thread(client);

 	test(err==0);

 	test_Close();

 	return 0;

 	}