tcp_clie.c

基于vxworks操作系统的电话语音平台系统

#include "vxWorks.h"
#include "sockLib.h"
#include "inetLib.h"
#include "stdioLib.h"
#include "strLib.h"
#include "hostLib.h"
#include "ioLib.h"
#include "rtos\vxworks\tcpip\net_proc.h"

/**************************************************************************
*
* tcpClient - send requests to server over a TCP socket

*
* This routine connects over a TCP socket to a server, and sends a
* user-provided message to the server. Optionally, this routine
* waits for the server's reply message.
*
* This routine may be invoked as follows:
*       -> tcpClient "remoteSystem"
*       Message to send:
*       Hello out there
*       Would you like a reply (Y or N):
*       y
*       value = 0 = 0x0
*       -> MESSAGE FROM SERVER:
*       Server received your message
*
* RETURNS: OK, or ERROR if the message could not be sent to the server.

*/
STATUS tcpClient(char *serverName /* name or IP address of server */)
{
	struct request      myRequest;     /* request to send to server */
	struct sockaddr_in  serverAddr;    /* server's socket address */
	char                replyBuf[REPLY_MSG_SIZE]; /* buffer for reply */
	char                reply;         /* if TRUE, expect reply back */
	int                 sockAddrSize;  /* size of socket address structure */
	int                 sFd;           /* socket file descriptor */
	int                 mlen;          /* length of message */

	/* create client's socket */

	if((sFd = socket (AF_INET, SOCK_STREAM, 0)) == ERROR)
	{
		return (ERROR);
	}

/* bind not required - port number is dynamic */
/* build server socket address */

	sockAddrSize = sizeof (struct sockaddr_in);
	bzero ((char *) &serverAddr, sockAddrSize);
	serverAddr.sin_family = AF_INET;
	serverAddr.sin_port = htons (SERVER_PORT_NUM);

	if (((serverAddr.sin_addr.s_addr = inet_addr (serverName)) == ERROR) &&
		((serverAddr.sin_addr.s_addr = hostGetByName (serverName)) == ERROR))
	{
		close (sFd);
		return (ERROR);
	}

	/* connect to server */

	if(connect (sFd, (struct sockaddr *) &serverAddr, sockAddrSize) == ERROR)
	{
		close (sFd);
		return (ERROR);
	}

	/* build request, prompting user for message */


	mlen = read (STD_IN, myRequest.message, REQUEST_MSG_SIZE);
	myRequest.msgLen = mlen;
	myRequest.message[mlen - 1] = '\0';

	read (STD_IN, &reply, 1);
	switch (reply)
	{
		case 'y':
		case 'Y': myRequest.reply = TRUE;
			break;
		default: myRequest.reply = FALSE;
			break;
	}

	/* send request to server */

	if (write (sFd, (char *) &myRequest, sizeof (myRequest)) == ERROR)
	{
		close (sFd);
		return (ERROR);
	}

	if (myRequest.reply)        /* if expecting reply, read and display it */
	{
		if (read (sFd, replyBuf, REPLY_MSG_SIZE) < 0)
		{
			close (sFd);
			return (ERROR);
		}

	}

	close (sFd);
	return (OK);
}