VxWorks网络编程TCP协议应用代码

在这在这里插入代码片里插入代码片

#include “vxWorks.h”
#include “sockLib.h”
#include “inetLib.h”
#include “stdioLib.h”
#include “strLib.h”
#include “hostLib.h”
#include “ioLib.h”
//#include “tcpExample.h”

#define SERVER_PORT_NUM 5001
#define SERVER_WORK_PRIORITY 100
#define SERVER_STACT_SIZE 10000
#define SERVER_MAX_CONNECTIONS 4
#define REQUEST_MSG_SIZE 1024
#define REPLY_MSG_SIZE 500
#define ERROR (-1)

int tcp_server(void)
{
struct sockaddr_in serverAddr;
struct sockaddr_in clientAddr;
char rcvBuf[REQUEST_MSG_SIZE];
int sockAddrSize;
int sFd;
int newFd;
int nByte;

//1 create socket
if((sFd = socket(AF_INET, SOCK_STREAM, 0)) == ERROR)
{
	printf("create socket error!\n");
	return ERROR;
}

//2 init some parameter
sockAddrSize=sizeof(struct sockaddr_in);
bzero((char *)&serverAddr, sockAddrSize);
serverAddr.sin_family = AF_INET;
serverAddr.sin_port = htons(SERVER_PORT_NUM);
serverAddr.sin_addr.s_addr=htonl(INADDR_ANY);
serverAddr.sin_len=(unsigned char)sockAddrSize;

//3 bind socket
if(bind(sFd,(struct sockaddr*)(&serverAddr),sockAddrSize)== ERROR)
{
	printf("bind error!\n");
	close(sFd);
	return ERROR;
}

//3 listen client
if(listen(sFd,SERVER_MAX_CONNECTIONS) == ERROR)
{
	printf("listen error!\n");
	close(sFd);
	return ERROR;
}	
    //下面可以使用死循环或线程创建accept，并创建相对应的接收线程
//4.accept server
newFd = accept(sFd,(struct sockaddr*)(&clientAddr),&sockAddrSize);
printf("server get connection from %s\n",inet_ntoa(clientAddr.sin_addr));
		
while(1)
{

	//5.receive
	nByte = recv(newFd, rcvBuf, REQUEST_MSG_SIZE, 0);//收不到数据阻塞
	rcvBuf[nByte] = '\0';
	printf("server received datas : %s\n", rcvBuf);
	
	if(rcvBuf[0]=='q')
		break;
}	
//6.close *Fd
close(newFd);
close(sFd);

return 0;

}

int tcp_client(char * serverName)
{
struct sockaddr_in serverAddr;
char sendBuf[REQUEST_MSG_SIZE];
int sockAddrSize;
int sFd;

//1 create socket
if((sFd = socket(AF_INET, SOCK_STREAM, 0)) == ERROR)
{
	printf("create socket error!\n");
	return ERROR;
}

//2 init some parameter
sockAddrSize=sizeof(struct sockaddr_in);
bzero((char *)&serverAddr, sockAddrSize);
serverAddr.sin_family = AF_INET;
serverAddr.sin_port = htons(SERVER_PORT_NUM);
serverAddr.sin_len=(unsigned char)sockAddrSize;

if((serverAddr.sin_addr.s_addr = inet_addr(serverName))== ERROR &&
   (serverAddr.sin_addr.s_addr = hostGetByName(serverName)) == ERROR)
{
	printf("unknown serverName\n");
	close(sFd);
	return ERROR;
}
//3 connnect server
if(connect(sFd,(struct sockaddr*)(&serverAddr),sockAddrSize) == ERROR)
{
	printf("connect error\n");
	return ERROR;
}

//4:send datas to server
printf("please input chars:\n");

//send(sFd,sendBuf, strlen(sendBuf), 0);
while(1){
	fgets(sendBuf,REQUEST_MSG_SIZE,stdin);	
	write(sFd,sendBuf, strlen(sendBuf));
	if(sendBuf[0] == 'q')
		break;
}
//5 close sFd
close(sFd);

return 0;

}