回射程序,客户端和服务器都使用 recvn 定长数据接收【基于流式套接字的网络程序设计】

设计思路

预先给定了接收数据的总长度,增加对数据长度的判断。

客户端 recvn 定长数据接收

// 回射程序 send recvn 客户端 
#define NETWORK1_2_1_A_COMM_H
#define NETWORK1_2_1_A_COMM_H
#include <stdio.h>
#include <stdlib.h>
#include <winsock2.h>
#include <inaddr.h>
#include <time.h>
#include <string.h>

#pragma comment(lib, "ws2_32.lib")
#define MAXLINE 4096     // 接收缓冲区长度
#define LISTENQ 1024     // 监听队列长度
#define SEVER_PORT 13131 // 端口
#define FIXEDLEN 5 //recvn定长接收长度 
char recv_data[MAXLINE];
//初始化动态连接库 
int startUp()
{
    WORD w_version = MAKEWORD(2, 2);
    WSADATA wsa_data;
    int res = WSAStartup(w_version, &wsa_data);
    if (res != 0)
    {
    	printf("WSAStartup Error:%d\n",WSAGetLastError());
        return -1;
    }
    if (LOBYTE(wsa_data.wVersion) != 2 || HIBYTE(wsa_data.wVersion) != 2)
    {
    	printf("版本号错误:%d\n",WSAGetLastError()); 
        WSACleanup();
        return -1;
    }

    return 0; //WSAStartup()成功
}
/*释放动态连接库*/
int cleanUp()
{
    int res = WSACleanup();
    if (res == SOCKET_ERROR)
    {
    	printf("WSACleanup error:%d\n",WSAGetLastError());
        return -1;
    }
    return 0;
}
/*断开连接并释放资源*/
int closeConn(SOCKET sock_conn)
{
    int res = closesocket(sock_conn); //关闭连接
    if (res == SOCKET_ERROR)
    {
    	printf("close socket error:%d",WSAGetLastError());
        return -1;
    }
    res = cleanUp();
    return res;
}
/*客户端套接字初始化*/
SOCKET tcpClientInit(char *server_ip, u_short port)
{
    int res;
    SOCKET sock_conn;

    // 设置服务器地址
    sockaddr_in server_addr;
    memset(&server_addr, 0, sizeof(server_addr));
    server_addr.sin_family = AF_INET;
    server_addr.sin_port = htons(port);
    server_addr.sin_addr.S_un.S_addr = inet_addr(server_ip);

    //创建流式套接字
    sock_conn = socket(AF_INET, SOCK_STREAM, 0);
    if (sock_conn == INVALID_SOCKET)
    {
    	printf("socket error:%d\n",WSAGetLastError());
        cleanUp();
        return -1;
    }

    // 请求连接服务器
    res = connect(sock_conn, (sockaddr *)&server_addr, sizeof(server_addr));
    if (res == SOCKET_ERROR)
    {
    	printf("connect error:%d\n",WSAGetLastError());
        closeConn(sock_conn);
        return -1;
    }

    return sock_conn; //连接成功
}
int getline(char s[],int lim)
{
    int c,i;
    i=0;
    while((c=getchar())!=EOF&&c!='\n'&&i<lim-1)
        s[i++]=c;
    s[i]='\0';
    return i;
}
/*定长数据接收函数*/
int recvn(int s,char * recvbuf,unsigned int fixedlen)
{
	int iResult;//单次操作recv的返回值 
	int cnt;
	cnt=fixedlen;
	while(cnt>0)
	{
		iResult=recv(s,recvbuf,cnt,0);
		if(iResult<0)
		{
			printf("数据接收出现错误:%d\n",WSAGetLastError());
			return -1;
		}
	if(iResult==0)
	{
		puts("关闭连接");
		return fixedlen-cnt;
	}
	recvbuf+=iResult;
	cnt-=iResult;
	}
	return fixedlen;
}
/*tcp发送接收*/
int tcpEchoClient(SOCKET sock_conn)
{
    int res = -1;
    int len=0; 
    char send_data[MAXLINE];
	//char recv_data[MAXLINE];全局变量 
    memset(send_data, 0, MAXLINE);
    memset(recv_data, 0, MAXLINE);
    while (getline(send_data, MAXLINE))
    {
        if (*send_data == 'Q' || *send_data == 'q')
        {
        	puts("结束发送");

            // 关闭发送
            res = shutdown(sock_conn, SD_SEND);
            if (res == SOCKET_ERROR)
            {
            	printf("shutdow error:%d\n",WSAGetLastError());
            }
            return 0;//正常返回 
        }

        // 发送数据
        res = send(sock_conn, send_data, (int)strlen(send_data), 0);
        if (res == SOCKET_ERROR)
        {
        	printf("send error:%d\n",WSAGetLastError());
            return -1;
        }
		printf("发送出去的数据是:%s\n",send_data);

        // 接收数据 
        len=recvn(sock_conn,recv_data,FIXEDLEN);
      	if(len==-1)
		{
			puts("出错!");
			res=-1;
			break;
		}
		if(len<FIXEDLEN)
		{
			puts("对方提前关闭了连接");
			res=-1;
			break;
		}
		if(len==FIXEDLEN)
		{
			printf("接收到的数据:%s\n",recv_data); 
			// 回射接收的数据
		}
        // 清空缓存
        memset(recv_data, 0, MAXLINE);
        memset(send_data, 0, MAXLINE);
    }
    return res;
}
int main()
{
    int res = -1;
    SOCKET sock_conn;
    char server_ip[] = "10.236.10.200";

    // 启动
    res = startUp();
    if (res == -1) return -1;
    //连接
    sock_conn = tcpClientInit(server_ip, SEVER_PORT);
    if (sock_conn == -1) return -1;
	puts("Client Startup seccess!");
    res = tcpEchoClient(sock_conn);  //处理连接
    closeConn(sock_conn);            //关闭连接

    return res;
}

服务器端 recvn 定长数据接收

//回射程序,服务器定长接收 recvn,send 
#define NETWORK1_2_1_A_COMM_H
#define NETWORK1_2_1_A_COMM_H
#include <stdio.h>
#include <stdlib.h>
#include <winsock2.h>
#include <inaddr.h>
#include <time.h>
#include <string.h>

#pragma comment(lib, "ws2_32.lib")

#define MAXLINE 4096     // 接收缓冲区长度
#define LISTENQ 1024     // 监听队列长度
#define SEVER_PORT 13131 // 端口
#define FIXEDLEN 5//recvn定长接收长度 

char recv_data[MAXLINE + 1];
//初始化动态连接库 
int startUp()
{
    WORD w_version = MAKEWORD(2, 2);
    WSADATA wsa_data;
    int res = WSAStartup(w_version, &wsa_data);
    if (res != 0)
    {
    	printf("WSAStartup Error:%d\n",WSAGetLastError());
        return -1;
    }
    if (LOBYTE(wsa_data.wVersion) != 2 || HIBYTE(wsa_data.wVersion) != 2)
    {
    	printf("版本号错误:%d\n",WSAGetLastError()); 
        WSACleanup();
        return -1;
    }

    return 0; //WSAStartup()成功
}
/*释放动态连接库*/
int cleanUp()
{
    int res = WSACleanup();
    if (res == SOCKET_ERROR)
    {
    	printf("WSACleanup error:%d\n",WSAGetLastError());
        return -1;
    }
    return 0;
}
/*断开连接并释放资源*/
int closeConn(SOCKET sock_conn)
{
    int res = closesocket(sock_conn); //关闭连接
    if (res == SOCKET_ERROR)
    {
    	printf("close socket error:%d",WSAGetLastError());
        return -1;
    }
    res = cleanUp();
    return res;
}
/*Server Init*/
SOCKET tcpServerInit(u_short port)
{
    int res = -1;

    // 设置地址
    sockaddr_in server_addr;
    memset(&server_addr, 0, sizeof(server_addr));
    server_addr.sin_family = AF_INET;
    server_addr.sin_addr.s_addr = htonl(INADDR_ANY);
    server_addr.sin_port = htons(port);

    //创建流式套接字
    SOCKET sock_listen = socket(AF_INET, SOCK_STREAM, 0);
    if (sock_listen == INVALID_SOCKET)
    {
    	printf("Sock error:%d",WSAGetLastError());
        WSACleanup();
        return -1;
    }

    //绑定服务器地址
    res = bind(sock_listen, (struct sockaddr *)&server_addr, sizeof(server_addr));
    if (res == SOCKET_ERROR)
    {
    	printf("bind error:%d\n",WSAGetLastError());
        closeConn(sock_listen);
        return -1;
    }

    //设置服务器为监听状态 最长长度LISTENQ
    res = listen(sock_listen, LISTENQ);
    if (res == SOCKET_ERROR)
    {
    	printf("listen error:%d\n",WSAGetLastError());
        closeConn(sock_listen);
        return -1;
    }

    return sock_listen;
}
/*定长数据接收函数*/ 
int recvn(int s,char * recvbuf,unsigned int fixedlen)
{
	int iResult;//单次操作recv的返回值 
	int cnt;
	cnt=fixedlen;
	while(cnt>0)
	{
		iResult=recv(s,recvbuf,cnt,0);
		if(iResult<0)
		{
			printf("数据接收出现错误:%d\n",WSAGetLastError());
			return -1;
		}
	if(iResult==0)
	{
		puts("关闭连接");
		return fixedlen-cnt;
	}
	recvbuf+=iResult;
	cnt-=iResult;
	}
	return fixedlen;
}
/*回射服务器*/
int tcpEchoServer(SOCKET sock_conn)
{
    int res=-1;
	int len=0;
   // char recv_data[MAXLINE + 1];全局
	do{
	memset(recv_data, 0, MAXLINE);
	len=recvn(sock_conn,recv_data,FIXEDLEN);
	if(len==-1)
	{
		puts("出错!");
		res=-1;
		break;
	}
	if(len<FIXEDLEN)
	{
		puts("对方提前关闭了连接");
		res=-1;
		break;
	}
	if(len==FIXEDLEN)
	{
		printf("接收到的数据:%s\n",recv_data); 
		// 回射接收的数据
            res = send(sock_conn, recv_data, len, 0);
           if (res == SOCKET_ERROR)
            {
            	printf("send error:%d\n",WSAGetLastError());
                res = -1;
				break;
			}
            else
           {
           	printf("发送数据:%s\n",recv_data);
			}
	}
	}while(res>0);
//	printf("接收到的字节数%d\n",len);
	return res;
}

int main(int argc, char *argv[])
{
    int res = -1;
    char buff[MAXLINE]; //缓冲区
    SOCKET sock_conn;
    SOCKET sock_listen;

    // 启动
    res = startUp();
    if (res == -1) return -1;

    //监听
    sock_listen = tcpServerInit(SEVER_PORT);
    if (sock_listen == -1) return -1;
	puts("Server  StartUp seccess!");

    // 循环服务器
    while (true)
    {
        //接受客户端连接请求  建立连接
        sock_conn = accept(sock_listen, NULL, NULL);

        if (sock_conn != INVALID_SOCKET)
        {
        	puts("连接成功");
            res = tcpEchoServer(sock_conn);
            if (res == -1)
            {
            	puts("连接失败");
            }
        }
        else
        {
        	printf("Accept Error:%d\n",WSAGetLastError());
            closeConn(sock_listen);
            return -1;
        }

        if (closesocket(sock_conn) == SOCKET_ERROR)
        {
        	printf("close error:%d\n",WSAGetLastError());
        }
    }

    // 关闭连接
    closeConn(sock_listen);
    return 0;
} 

用户使用说明(输入 / 输出规定)

recvn 定长数据接收:
客户端必须输入固定长度的数据,(我设定的固定长度为 5 个字节),客户端会显示从服务器接收到的字节数。客户端输入 Q 或 q 断开连接。服务器端显示接收到的数据,以及回射给客户端的数据。

运行结果

recvn 回射程序,客户端及服务器端:
在这里插入图片描述


版权声明:本文为qq_41581354原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。