conserver.c

嵌入式VxWorks开发的各种实验源代码以及相关教程

/* conServer.c - demo concurrent TCP server */

#include "vxWorks.h"
#include "sockLib.h"
#include "sys/socket.h"
#include "netinet/in.h"
#include "inetLib.h"
#include "stdio.h"
#include "stdlib.h"
#include "ioLib.h"
#include "taskLib.h"
#include "string.h"
#include "conServer.h"
#include "advTCPLab.h"

#define MAX_MSG_LEN 80
typedef int SOCK_FD;

/* forward declarations */
LOCAL void doRequest(SOCK_FD sock,
					struct sockaddr_in * pClientAddr);
LOCAL void error (char * str);

void vxServer
	(
	u_short port
	)
	{
	int clientAddrLength;
	SOCK_FD sockFd;
	SOCK_FD newSockFd;
	struct sockaddr_in * pClientAddr;
	struct sockaddr_in srvAddr;

	clientAddrLength = sizeof (struct sockaddr_in);
	if (port == 0)
		port = SRV_PORT;

	/* Create a socket */
	if ((sockFd = socket (PF_INET, SOCK_STREAM,
		 						IPPROTO_TCP)) < 0)
		error ("Socket failed");

	/*
	 * Bind to a well known address. INADDR_ANY says
	 * any network interface will do. htonX()
	 * routines put things in network byte order
	 */
	bzero ((char *)&srvAddr, sizeof(srvAddr));
	srvAddr.sin_family = AF_INET;
	srvAddr.sin_port = htons(port);
	srvAddr.sin_addr.s_addr = INADDR_ANY;
	if ( bind (sockFd, (struct sockaddr *) &srvAddr,
			sizeof(srvAddr)) < 0 )
		{
		close (sockFd);
		error ("Bind failed");
		}


	/* Queue up to 8 requests (max allowed by TCP) */
	if ( listen (sockFd, 5) < 0 )
		{
		close (sockFd);
		error ("Listen failed");
		}

	/* Service requests */
	FOREVER
		{
		pClientAddr =
				malloc (sizeof(struct sockaddr));

		newSockFd = accept (sockFd,
						(struct sockaddr *)pClientAddr,
						&clientAddrLength);
		if (newSockFd < 0)
	 		{
			close (sockFd);
	 		error ("Accept failed");
	 		}

		taskSpawn ("tSlaveServer", SLAVE_PRIORITY, 0,
						SLAVE_STACK_SIZE, 
						(FUNCPTR) doRequest,
						newSockFd,(int) pClientAddr,
						0,0,0,0,0,0,0,0);
		}

 	}


/************************************************** 
*
* doRequest - read the clients requests from the
* socket and honor them.
*
*/
void doRequest
	(
	SOCK_FD sock,
	struct sockaddr_in * pClientAddr
	)
	{
	char buf[MAX_MSG_LEN];
	int msgSize;
	char clientInet [INET_ADDR_LEN];
	u_short port = ntohs (pClientAddr->sin_port);

	inet_ntoa_b (pClientAddr->sin_addr, clientInet);
	free (pClientAddr);
 
 	FOREVER
		{
		msgSize = read (sock, buf, MAX_MSG_LEN);

		/* Read error ? */
		if (msgSize < 0)
			{
			close (sock);
			error ("Read failed");
			}

		/* Connection closed ? */
		else if (msgSize == 0)
			{
			close (sock);
			break;
			}

		/* Get client message and service it*/
		else
			{
			printf ("Task %s servicing client from"
					" inet %s, port %d\n\n",taskName(0),
					clientInet, port);
			}
		}
	}


void error
	(
	char * str
	)
	{
	perror (str);
	exit (1);
	}