设计思路
预先给定了接收数据的总长度,增加对数据长度的判断。
客户端 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版权协议,转载请附上原文出处链接和本声明。