网络编程 8月15日

于 2023-08-16 09:07:50 发布
阅读量29 收藏
点赞数
文章标签： c语言 linux 学习 网络
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/yuyanshang/article/details/132311432
版权

    1.将select的TCP服务器重新搭建
    2.搭建select的TCP客户端
    3.搭建poll客户端
    4.上传的代码完成

1.代码

#include <stdio.h>
#include <head.h>

#define PORT 6644			//1024-49151
#define IP "192.168.255.87"	//ifconfig查看本机IP

int main(int argc, const char *argv[])
{
	//创建流式套接字
	int sfd = socket(AF_INET, SOCK_STREAM, 0);
	if (sfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("sfd = %d\n", sfd);

	//设置允许端口被快速复用
	int reuse = 1;
    if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)  
    {
        ERR_MSG("setsockopt");
        return -1;
    }
    printf("允许端口快速重用成功\n");

	//填充地址信息结构体，真实的地址信息结构体根据地址族制定
	//AF_INET: man 7 ip
	struct sockaddr_in sin;
	sin.sin_family 		= AF_INET;			//必须填AF_INET
	sin.sin_port   		= htons(PORT);		//端口号：1024-49151
	sin.sin_addr.s_addr = inet_addr(IP);	//本机IP ifconfig查看

	//绑定服务器的IP和端口 ---> 必须绑定
	if (bind(sfd, (struct sockaddr *)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success\n");

	//将套接字设置为被动监听状态
	if (listen(sfd, 128) < 0)
	{
		ERR_MSG("listen");
		return -1;
	}
	printf("listen success\n");

	//创建一个读集合
	fd_set readfds, tempfds;

	//fd_set本质上是一个结构体，成员是一个long类型的数组
	//集合目的是为了存储让内核监测的文件描述符，
	//若不初始化，则可能随机到有效的但不需要的文件描述符
	//所以需要将集合初始化成无意义的文件描述符
	FD_ZERO(&readfds);
	FD_ZERO(&tempfds);

	//将需要的文件描述符添加到集合中
	FD_SET(0, &readfds);
	FD_SET(sfd, &readfds);

	int s_res = -1;
	int newfd = -1;
	char buf[128] = "";
	ssize_t res = 0;
	struct sockaddr_in cin;	//存储客户端的地址信息
	socklen_t addrlen = sizeof(cin);
	int maxfd = sfd;
	struct sockaddr_in saveCin[1024];

	printf("输入任意字符后按回车触发键盘输入事件\n");

	//从已完成连接的队列中获取一个客户端信息，生产一个新的文件描述符
	//该文件描述符才是与客户端通信的文件描述符
	while (1)
	{
		tempfds = readfds;	//将文件描述符拷贝一份

		s_res = select(maxfd + 1, &tempfds, NULL, NULL, NULL);
		if (s_res < 0)
		{
			ERR_MSG("select");
			return -1;
		}
		else if (0 == s_res)
		{
			printf("time out...\n");
			break;
		}
		
		/*
		能运行到当前位置，则代表集合中有文件描述符准备就绪
		此时集合中会剩下触发事件的文件描述符
		例如：
		当0号文件描述符触发事件，则集合中只剩下0
		当sfd文件描述符触发事件，则集合中只剩下sfd
		当0号和sfd文件描述符同时触发事件，则集合中剩下0和sfd

		判断是哪个文件描述符准备就绪，走对应的处理函数即可
		判断集合中剩下哪个文件描述符，就代表哪个文件描述符准备就绪
		*/

		for (int i = 0; i <= maxfd; i++)
		{
			if (FD_ISSET(i, &tempfds) == 0) //若i所代表的文件描述符不在集合中
			{
				continue;	//则继续遍历下一个i
			}
			
			//能运行到当前位置，说明i所代表的文件描述符在集合中
			//判断i所代表的文件描述符触发的是什么事件
			if (0 == i)
			{
				while(getchar() != 10);
				//while(getchar() != 10);
				printf("触发键盘输入事件\n");
				int sndfd = -1;
				printf("请输入文件描述符>>> "); 
				scanf("%d", &sndfd);	//返回值：成功获取到的数据个数
				while(getchar() != 10);
				printf("请输入要发送的信息>>> "); 
				fgets(buf, sizeof(buf), stdin);	//若不将文件描述符对应的空间的数据读取出来，则文件描述符一直处于就绪状态，不阻塞
				buf[strlen(buf) - 1] = 0;
				// if (res != 2)
				// {
				// 	printf("请输入正确的格式: fd string\n");
				// 	continue;
				// }

				//能运行到当前位置，则代表输入的格式正确
				if (sndfd < 3 || !FD_ISSET(sndfd, &readfds))
				{
					printf("sndfd = %d 文件描述符错误\n", sndfd);
					continue;
				}
				
				if (send(sndfd, buf, sizeof(buf), 0) < 0)
				{
					ERR_MSG("send");
					return -1;
				}
				printf("send success\n");
				

				// printf(": %s\n", buf);
			}
			else if (sfd == i)
			{
				printf("触发客户端连接事件\n");
				newfd = accept(sfd, (struct sockaddr *)&cin, &addrlen);
				if (newfd < 0)
				{
					ERR_MSG("accept");
					return -1;
				}
				
				printf("[%s : %d] newfd = %d 客户端连接成功\n", inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), newfd);

				saveCin[newfd] = cin;

				//将newfd添加到select中
				FD_SET(newfd, &readfds);
				//更新maxfd
				maxfd = maxfd > newfd ? maxfd : newfd;
			}
			else //判断newfd在不在集合中，若在集合中则走recv函数
			{
				
				printf("触发客户端交互事件\n");
				bzero(buf, sizeof(buf));
				//接收数据
				res = recv(i, buf, sizeof(buf), 0);
				if (res < 0)
				{
					ERR_MSG("recv");
					return -1;
				}
				else if (0 == res)
				{
					printf("[%s : %d] newfd = %d 客户端下线\n", inet_ntoa(saveCin[i].sin_addr), ntohs(saveCin[i].sin_port), i);
					close(i);	//关闭下线客户端的文件描述符
					FD_CLR(i, &readfds);	//将下线客户端的文件描述符从集合中删除
					while(FD_ISSET(maxfd, &readfds) == 0 && maxfd-- >= 0);	//更新maxfd，从大到小判断，判断文件描述符是否在集合中，存在，该文件描述符为最大，不存在，则继续往小判断
					continue;
				}
				printf("[%s : %d] newfd = %d : %s\n", inet_ntoa(saveCin[i].sin_addr), ntohs(saveCin[i].sin_port), i, buf);

				//发送数据
				strcat(buf, "*_*");
				if (send(newfd, buf, sizeof(buf), 0) < 0)
				{
					ERR_MSG("send");
					return -1;
				}
				printf("send success\n");
			}
		}

	}

	//关闭所有文件描述符
	//close(newfd);
	close(sfd);

	return 0;
}

结果

 2.代码

#include <stdio.h>
#include <head.h>

#define PORT 6644			//1024-49151
#define IP "192.168.255.87"	//ifconfig查看本机IP

int main(int argc, const char *argv[])
{
	//创建流式套接字
	int cfd = socket(AF_INET, SOCK_STREAM, 0);
	if (cfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("cfd = %d\n", cfd);

	//设置允许端口被快速复用
	// int reuse = 1;
    // if(setsockopt(cfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)  
    // {
    //     ERR_MSG("setsockopt");
    //     return -1;
    // }
    // printf("允许端口快速重用成功\n");

	//填充服务器的地址信息结构体，真实的地址信息结构体根据地址族制定
	//AF_INET: man 7 ip
	//要连接哪个服务器，就填对应服务器的IP和端口
	struct sockaddr_in sin;
	sin.sin_family 		= AF_INET;			//必须填AF_INET
	sin.sin_port   		= htons(PORT);		//端口号：1024-49151
	sin.sin_addr.s_addr = inet_addr(IP);	//本机IP ifconfig查看

	//绑定服务器的IP和端口 ---> 非必须绑定
	//若不绑定，则操作系统会给客户端绑定上客户端所在的主机IP，以及随机端口
	// if (bind(sfd, (struct sockaddr *)&sin, sizeof(sin)) < 0)
	// {
	// 	ERR_MSG("bind");
	// 	return -1;
	// }
	// printf("bind success\n");

	//连接服务器
	if (connect(cfd, (struct sockaddr *)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("connect");
		return -1;
	}
	printf("connect success\n");

	fd_set readfds, tempfds;

	FD_ZERO(&readfds);
	FD_ZERO(&tempfds);

	FD_SET(0, &readfds);
	FD_SET(cfd, &readfds);

	int s_res = -1;


	char buf[128] = "";
	ssize_t res = 0;
	printf("输入任意字符后按回车触发键盘输入事件\n");

	while (1)
	{
		tempfds = readfds;

		s_res = select(cfd + 1, &tempfds, NULL, NULL, NULL);
		if (s_res < 0)
		{
			ERR_MSG("select");
			return -1;
		}
		else if (0 == s_res)
		{
			printf("time out...\n");
			break;
		}

		if (FD_ISSET(0, &tempfds))
		{
			while(getchar() != 10);
			printf("触发键盘输入和客户端发送事件\n");
			bzero(buf, sizeof(buf));
			printf("请输入>>> ");
			//fflush(stdin);
			fgets(buf, sizeof(buf), stdin);
			buf[strlen(buf) - 1] = 0;

			//发送数据
			//strcat(buf, "*_*");
			if (send(cfd, buf, sizeof(buf), 0) < 0)
			{
				ERR_MSG("send");
				return -1;
			}
			printf("send success\n");
		}
		
		
		if (FD_ISSET(cfd, &tempfds))
		{
			printf("触发客户端接收数据事件\n");
			bzero(buf, sizeof(buf));
			//接收数据
			res = recv(cfd, buf, sizeof(buf), 0);
			if (res < 0)
			{
				ERR_MSG("recv");
				return -1;
			}
			else if (0 == res)
			{
				printf("[%s : %d] cfd = %d 服务器掉线\n", IP, PORT, cfd);
				break;
			}
			printf("[%s : %d] cfd = %d : %s\n", IP, PORT, cfd, buf);
		}
	}

	//关闭所有文件描述符
	close(cfd);

	return 0;
}

结果

3.代码

#include <stdio.h>
#include <head.h>

#define PORT 6644			//1024-49151
#define IP "192.168.255.87"	//ifconfig查看本机IP

int main(int argc, const char *argv[])
{
	//创建流式套接字
	int cfd = socket(AF_INET, SOCK_STREAM, 0);
	if (cfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("cfd = %d\n", cfd);

	//设置允许端口被快速复用
	// int reuse = 1;
    // if(setsockopt(cfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)  
    // {
    //     ERR_MSG("setsockopt");
    //     return -1;
    // }
    // printf("允许端口快速重用成功\n");

	//填充服务器的地址信息结构体，真实的地址信息结构体根据地址族制定
	//AF_INET: man 7 ip
	//要连接哪个服务器，就填对应服务器的IP和端口
	struct sockaddr_in sin;
	sin.sin_family 		= AF_INET;			//必须填AF_INET
	sin.sin_port   		= htons(PORT);		//端口号：1024-49151
	sin.sin_addr.s_addr = inet_addr(IP);	//本机IP ifconfig查看

	//绑定服务器的IP和端口 ---> 非必须绑定
	//若不绑定，则操作系统会给客户端绑定上客户端所在的主机IP，以及随机端口
	// if (bind(sfd, (struct sockaddr *)&sin, sizeof(sin)) < 0)
	// {
	// 	ERR_MSG("bind");
	// 	return -1;
	// }
	// printf("bind success\n");

	//连接服务器
	if (connect(cfd, (struct sockaddr *)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("connect");
		return -1;
	}
	printf("connect success\n");

	//定义集合
	struct pollfd fds[2];

	//将需要监测的文件描述符添加到集合中
	fds[0].fd = 0;			//指定监测0号文件描述符
	fds[0].events = POLLIN;	//监测读事件

	fds[1].fd = cfd;		//指定监测0号文件描述符
	fds[1].events = POLLIN;	//监测读事件


	char buf[128] = "";
	ssize_t res = 0;
	int p_res = -1;

	printf("输入任意字符后按回车触发键盘输入事件\n");

	while (1)
	{
		

		p_res = poll(fds, 2, -1);
		if (p_res < 0)
		{
			ERR_MSG("poll");
			return -1;
		}
		else if (0 == p_res)
		{
			printf("time out...\n");
			break;
		}

		if ((fds[0].revents & POLLIN) != 0)
		{
			while(getchar() != 10);
			printf("触发键盘输入和客户端发送事件\n");
			bzero(buf, sizeof(buf));
			printf("请输入>>> ");
			//fflush(stdin);
			fgets(buf, sizeof(buf), stdin);
			buf[strlen(buf) - 1] = 0;

			//发送数据
			//strcat(buf, "*_*");
			if (send(cfd, buf, sizeof(buf), 0) < 0)
			{
				ERR_MSG("send");
				return -1;
			}
			printf("send success\n");
		}
		
		
		if ((fds[1].revents & POLLIN) != 0)
		{
			printf("触发客户端接收数据事件\n");
			bzero(buf, sizeof(buf));
			//接收数据
			res = recv(cfd, buf, sizeof(buf), 0);
			if (res < 0)
			{
				ERR_MSG("recv");
				return -1;
			}
			else if (0 == res)
			{
				printf("[%s : %d] cfd = %d 服务器掉线\n", IP, PORT, cfd);
				break;
			}
			printf("[%s : %d] cfd = %d : %s\n", IP, PORT, cfd, buf);
		}
	}

	//关闭所有文件描述符
	close(cfd);

	return 0;
}

 结果

4.代码

int do_upload(int cfd, struct sockaddr_in sin)
{
    char buf[N] = "";
    char filename[N - 10] = "";
    printf("请输入要上传的文件>>> ");
    fgets(filename, sizeof(filename), stdin);
    filename[strlen(filename) - 1] = 0;
    // scanf("%s", filename);
    // while(getchar() != 10);

    //判断该文件是否存在
    int fd = open(filename, O_RDONLY);
    if (fd < 0)
    {
        if (errno == ENONET)
        {
            printf(">>>文件不存在，请重新输入<<<\n");
            return -2;
        }
        else
        {
            ERR_MSG("open");
            return -1;
        }
    }
    
    //发送上传请求
    //上传协议
    int size = sprintf(buf, "%c%c%s%c%s%c", 0, 2, filename, 0, MODE, 0);

    if (sendto(cfd, buf, size, 0, (struct sockaddr *)&sin, sizeof(sin)) < 0)
    {
        ERR_MSG("sendto");
        return -1;
    }
    
    //循环接收发送数据包
    socklen_t addrlen = sizeof(sin);
    ssize_t res = 0;
    unsigned short num = 0;
    ssize_t rcv_len;

    while (1)
    {
        bzero(buf, N);

        rcv_len = recvfrom(cfd, buf, N, 0, (struct sockaddr *)&sin, &addrlen);
        if (rcv_len < 0)
        {
            ERR_MSG("recvfrom");
            return -1;
        }
        
        //操作码的范围是1-5，因为是网络字节序
        //所以有效操作码存储在高位，即buf[1]的位置
        if (4 == buf[1])
        {
            //判断当前数据包的编号是否等于应答包的编号
            //防止数据包在传送过程中丢包或者重复收包
            if (htons(num) == *(unsigned short *)(buf + 2))
            {
                //修改操作码为数据包
                buf[1] = 3;

                //填充块编号
                num++;
                *(unsigned short *)(buf + 2) = htons(num);

                //读取数据
                res = read(fd, buf + 4, N - 4);
                if (res < 0)
                {
                    ERR_MSG("read");
                    close(fd);
                    return -1;
                }
                else if (0 == res)
                {
                    printf("---%s 文件上传完毕---\n", filename);
                    break;
                }
                
                //发送数据包
                //发送的数据包的大小为读取到的字节数(res)+操作码(2byte)+块编号(2byte)
                if (sendto(cfd, buf, res + 4, 0, (struct sockaddr *)&sin, addrlen) < 0)
                {
                    ERR_MSG("sendto");
                    return -1;
                }
            }
            else
            {
                printf("---文件上传失败，请检查网络环境---\n");
                break;
            }
        }
        else if (5 == buf[1])
        {
            printf("---ERROR: %d %s---\n", ntohs(*(short *)(buf + 2)), buf + 4);
            break;
        }
        
    } 
    close(fd);
    return 0;

}

 结果

 

yuyanshang
关注
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值