day5-网络编程

作业：
1. 上交select服务器客户端代码，以及poll代码（今晚上交）

//select服务端
#include <stdio.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <sys/types.h>
#include <string.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <sys/select.h>
#include <sys/time.h>
#define PORT 6666
#define IP "192.168.10.121"

#define ERR_WSG(wsg)                   \
	do                                 \
	{                                  \
		printf("line:%d\n", __LINE__); \
		perror(wsg);                   \
	} while (0)

int main(int argc, const char *argv[])
{
	//socket创建套接字
	int sfd = socket(AF_INET, SOCK_STREAM, 0);
	if (sfd < 0)
	{
		ERR_WSG("socket");
		return -1;
	}

	//允许端口快速被复用
	int reuse = 1;
	if (setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)
	{
		ERR_WSG("setsockopt");
		return -1;
	}

	//bind绑定IP和端口-->网络字节序
	struct sockaddr_in sin;
	sin.sin_family = AF_INET;
	sin.sin_port = htons(PORT);			 //端口字节序
	sin.sin_addr.s_addr = inet_addr(IP); //ip字节序

	if (bind(sfd, (struct sockaddr *)&sin, sizeof(sin)) < 0) //第二个参数需注意强转
	{
		ERR_WSG("bind");
		return -1;
	}
	printf("bind success\n");

	//listen设置被动监听
	if (listen(sfd, 128) < 0)
	{
		ERR_WSG("listen");
		return -1;
	}
	printf("listen success\n");

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

	//由于fd_set是一个结构体，且结构体中是long类型的数组
	//集合可能会随机存[0,1023]文件描述符的整型数据
	//所以需要将集合中的数据初始化城无意义的整数
	FD_ZERO(&readfds);//清空集合
	FD_ZERO(&tempfds);//清空集合

	//将需要监测的文件描述符放入读集合中
	FD_SET(0, &readfds);
	FD_SET(sfd, &readfds);

	int maxfd = sfd; //记录最大的文件描述符
	
	int s_res = 0;

	struct sockaddr_in cin; //存储客户端信息
	socklen_t len = sizeof(struct sockaddr_in);

	struct sockaddr_in savecin[1024]; //0 1 2 sfd无对应客户端

	int newfd = -1;

	ssize_t res = 0;
	char buf[128]="";

	while(1)
	{
		tempfds = readfds;

		s_res = select(maxfd+1, &tempfds, NULL, NULL, NULL);
		if(s_res < 0)
		{
			ERR_WSG("select");
			return -1;
		}
		else if(0 == s_res)
		{
			printf("超时\n");
			break;
		}

		//能运行到当前位置，则代表集合中有文件描述符准备就绪
		//需要判断哪个文件描述符准备就绪，就走对应的处理函数即可

		//当集合中的文件描述符准备就绪后，集合中会只剩下产生时间的文件描述符
		//若0准备就绪，则集合中只剩下0
		//若sfd准备就绪，则集合中只剩下sfd
		//若0和sfd均准备就绪，则集合中0和sfd均存在
		//只需要判断集合中剩下哪个文件描述符，走对应函数即可

		for(int i=0; i<=maxfd; i++)
		{
			if(!FD_ISSET(i, &tempfds)) //若i所代表的文件描述符不在集合中，
				continue;              //结束本次循环，遍历下一个循环
		
			//能运行到当前位置，则说明i所代表的文件描述符，在tempfd集合中
			//0～maxfd：键盘输入事件  客户端连接事件  客户端交互事件

			if(0 == i) //键盘输入事件
			{
				printf("触发键盘输入事件\n");

				fgets(buf, sizeof(buf),stdin);
				buf[strlen(buf) -1] = 0;
				printf("buf = %s\n", buf);
			}

			else if(i == sfd) //客户端连接事件
			{
				printf("触发客户端连接事件\n");
				//从已完成连接的队伍里面，获取一个客户端信息，生成一个新的文件描述符
				//该文件描述符才是与客户端通信的文件描述符
				newfd = accept(sfd,(struct sockaddr*)&cin, &len);//2,3参数可填NULL
				if(newfd < 0)
				{
					ERR_WSG("accept");
					return -1;
				}
				printf("[%s:%d] 客户端连接成功 newfd = %d __%d__\n",\
						inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), newfd, __LINE__);
			
				savecin[newfd] = cin;
				
				//将新生成的newfd添加到readfds中，让select函数检测
				FD_SET(newfd, &readfds);

				//更新maxfd
				maxfd = maxfd < newfd ? newfd : maxfd;

			}
			else //客户端交互事件
			{
				printf("触发客户端交互事件__%d__\n",__LINE__);

				bzero(buf, sizeof(buf));

				//接收数据
				res =  recv(i, buf, sizeof(buf), 0); //0为默认方式，阻塞
				if(res < 0)
				{
					ERR_WSG("recv");
					return -1;
				}
				else if(0 == res)
				{
					printf("[%s :%d] 客户端下线 newfd = %d __%d__\n",\
							inet_ntoa(savecin[i].sin_addr), ntohs(savecin[i].sin_port), i, __LINE__);

					close(i);//关闭文件描述符
					//将 i 所代表的文件描述符从集合中删除 
					FD_CLR(i, &readfds);

					//更新maxfd
			 /*  	for(; maxfd >=0; maxfd--)
					{
						if(FD_ISSET(maxfd, &readfds) != 0)
							break;
					}
			*/		
					while(!FD_ISSET(maxfd, & readfds) && maxfd-- >= 0);

					continue;
				}

				printf("[%s:%d] newfd = %d : %s, __%d__\n",\
						inet_ntoa(savecin[i].sin_addr), ntohs(savecin[i].sin_port), i, buf, __LINE__);

				//发送数据
				strcat(buf,"*_*");
				if(send(i, buf, sizeof(buf), 0) < 0)
				{
					ERR_WSG("send");
					return -1;
				}
				printf("发送成功\n");
			}
		}
	}
	close(sfd);

	return 0;
}

//select 客户端
#include <stdio.h>
#include <string.h>
#include <fcntl.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/select.h>

#define ERR_MSG(msg){\
	printf("line:%d\n",__LINE__);\
	perror(msg);\
}

#define PORT 6666
#define IP "192.168.10.121"


int main(int argc, const char *argv[])
{
	int sfd = socket(AF_INET, SOCK_STREAM, 0);
	if(sfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("socket success\n");

	int reuse = 1;
	if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)
	{
		ERR_MSG("setsockopt");
		return -1;
	}

	struct sockaddr_in sin;
	sin.sin_family = AF_INET;
	sin.sin_port = htons(PORT);
	sin.sin_addr.s_addr = inet_addr(IP);
	
	if(connect(sfd, (struct sockaddr*)&sin, sizeof(sin)) != 0)
	{
		ERR_MSG("connect");
		return -1;
	}
	printf("connect success\n");
	//创建监测的集合
	fd_set readfds;

	//清空集合
	FD_ZERO(&readfds);

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


	int s_res = 0;

	struct sockaddr_in cin;
	socklen_t len = sizeof(cin);


	ssize_t res = 0;
	char buf[128] = "";

	while(1)
	{
		s_res = select(2, &readfds, NULL, NULL, NULL);
		if(s_res < 0)
		{
			ERR_MSG("select");
			return -1;
		}
		else if(0 == s_res)
		{
			printf("超时\n");
			break;
		}

		for(int i=0; i<=1; i++)
		{

			if(!FD_ISSET(i, &readfds))
			{
				continue;
			}

			//此处，表示文件描述符存在
			if(0 == i)
			{
				printf("键盘输入事件\n");

				fgets(buf, sizeof(buf), stdin);
				buf[strlen(buf)-1] = 0;
				printf("%s\n",buf);
			}

			else if(i == sfd) //客户端交互
			{		
				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]  服务器断开__%d__\n",\
							inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), __LINE__);

				}
				printf("[%s:%d] newfd = %d :%s, __%d__",\
						inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), i, buf, __LINE__);
				
				//发送数据
				strcat(buf,"*_*");
				if(send(i, buf, sizeof(buf), 0) < 0)
				{
					ERR_MSG("send");
					return -1;
				}
				printf("发送成功\n");
			}

		}

	}

	if(close(sfd) < 0)
	{
		ERR_MSG("close");
		return -1;
	}

	return 0;
}

//poll客户端
#include <stdio.h>
#include <string.h>
#include <fcntl.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/select.h>

#define ERR_MSG(msg){\
	printf("line:%d\n",__LINE__);\
	perror(msg);\
}

#define PORT 6666
#define IP "192.168.10.121"


int main(int argc, const char *argv[])
{
	int sfd = socket(AF_INET, SOCK_STREAM, 0);
	if(sfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("socket success\n");

	int reuse = 1;
	if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)
	{
		ERR_MSG("setsockopt");
		return -1;
	}

	struct sockaddr_in sin;
	sin.sin_family = AF_INET;
	sin.sin_port = htons(PORT);
	sin.sin_addr.s_addr = inet_addr(IP);
	
	if(connect(sfd, (struct sockaddr*)&sin, sizeof(sin)) != 0)
	{
		ERR_MSG("connect");
		return -1;
	}
	printf("connect success\n");
	//创建监测的集合
	fd_set readfds;

	//清空集合
	FD_ZERO(&readfds);

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


	int s_res = 0;

	struct sockaddr_in cin;
	socklen_t len = sizeof(cin);


	ssize_t res = 0;
	char buf[128] = "";

	while(1)
	{
		s_res = select(2, &readfds, NULL, NULL, NULL);
		if(s_res < 0)
		{
			ERR_MSG("select");
			return -1;
		}
		else if(0 == s_res)
		{
			printf("超时\n");
			break;
		}

		for(int i=0; i<=1; i++)
		{

			if(!FD_ISSET(i, &readfds))
			{
				continue;
			}

			//此处，表示文件描述符存在
			if(0 == i)
			{
				printf("键盘输入事件\n");

				fgets(buf, sizeof(buf), stdin);
				buf[strlen(buf)-1] = 0;
				printf("%s\n",buf);
			}

			else if(i == sfd) //客户端交互
			{		
				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]  服务器断开__%d__\n",\
							inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), __LINE__);

				}
				printf("[%s:%d] newfd = %d :%s, __%d__",\
						inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), i, buf, __LINE__);
				
				//发送数据
				strcat(buf,"*_*");
				if(send(i, buf, sizeof(buf), 0) < 0)
				{
					ERR_MSG("send");
					return -1;
				}
				printf("发送成功\n");
			}

		}

	}

	if(close(sfd) < 0)
	{
		ERR_MSG("close");
		return -1;
	}

	return 0;
}

2. 完成tftp的上传下载
 

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>
#include <fcntl.h>
#include <stdlib.h>
#include <errno.h>
#include <sys/stat.h>

#define ERR_MSG(msg) do {\
	printf("line:%d\n",__LINE__);\
	perror(msg);\
} while(0)

#define IP "192.168.10.104"
#define PORT 69
#define N 516


int do_download(int sfd, struct sockaddr_in sin);
int do_upload(int sfd, struct sockaddr_in sin);


int main(int argc, const char *argv[])
{
	if(argc < 2)
	{
		printf("请输入IP\n");
		return -1;
	}

	//创建报式套接字
	int sfd = socket(AF_INET, SOCK_DGRAM, 0);
	if(sfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}

	//填充服务器的地址信息结构体
	//给下面的sendto函数使用
	struct sockaddr_in sin;
	sin.sin_family      = AF_INET;
	sin.sin_port        = htons(PORT);
	sin.sin_addr.s_addr = inet_addr(argv[1]);

	char choose;
	while(1)
	{
		system("clear");
		printf("******************\n");
		printf("******1.下载******\n");
		printf("******2.上传******\n");
		printf("******3.退出******\n");
		printf("******************\n");

		scanf("%c", &choose);
		while(getchar()!= 10);

		switch(choose)
		{
		case '1':
			do_download(sfd, sin); //下载功能 todo
			break;
		case '2':
			do_upload; //上传功能，todo
			break;
		case '3':
			goto END;
			break;
		default:
			printf("输入错误\n");
		}
		printf("输入任意字符清屏>>:");
		while(getchar() != 10);
	}
END:
	close(sfd);
	return 0;
}

//上传功能
int do_upload(int sfd, struct sockaddr_in sin)
{
	char filename[20] = "";
	printf("请输入要上传的文件名>>:");
	fgets(filename, 20, stdin);
	filename[strlen(filename) - 1] = 0;
	
	//判断该文件是否存在
	int fd = open(filename, O_RDONLY);
	if(fd < 0)
	{
		if(errno == ENOENT)
		{
			printf(">>>文件不存在，请重新输入<<<\n");
			return -2;
		}
		else
		{
			ERR_MSG("open");
			return -1;
		}
	}

	//发送上传请求
	//上传协议
	char buf[N] = "";
	int size = sprintf(buf, "%c%c%s%c%s%c", 0, 2, filename, 0, "octet", 0);

	if(sendto(sfd, buf, size, 0, (struct sockaddr *)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("sendto");
		return -1;
	}

	//循环接收发送数据包
	int recv_len;
	unsigned short num = 0;
	socklen_t addrlen = sizeof(sin);
	while(1)
	{
		bzero(buf, N);
		recv_len = recvfrom(sfd, buf, sizeof(buf), 0, (struct sockaddr*)&sin, &addrlen);
		if(recv_len < 0)
		{
			ERR_MSG("recvfrom");
			return -1;
		}

		//操作码范围是1-5，因为是网络字节序
		//所以有效操作码存储在高位，即buf[1]的位置
		//printf("buf[1] = %d\n",buf[1]); //4 服务器返回应打包
		if(4 == buf[1])
		{
			//判断当前数据包的编号是否等于应答包的编号
			//防止数据包在传递过程丢包或重复收包
			if(num == ntohs(*(unsigned short *)(buf+2)))
			{
				//修改操作码为数据包
				buf[1] = 3;

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

				//读取数据
				//发送数据
				int res = read(fd, buf+4, N-4);
				if(res < 0)
				{
					ERR_MSG("read");
					return -1;
				}
				else if(0 == res)
				{
					printf("-----文件上传完毕-----");
					break;
				}

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



//下载功能
int do_download(int sfd, struct sockaddr_in sin)
{
	char filename[20] = "";
	printf("请输入要下载的文件名>>:");
	fgets(filename, 20, stdin);
	filename[strlen(filename) - 1] = 0;

	//发送下载请求
	char buf[N] = "";

	int size = sprintf(buf, "%c%c%s%c%s%c", 0, 1, filename,  0, "octet", 0);

	//发送下载请求包给服务器
	if(sendto(sfd, buf, size, 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("sendto");
		return -1;
	}

	//打开文件，保存下载的内容，写的方式打开，不存在则新建，存在则清空
	int fd = -1;
	int flag = 0;

	//循环接收发送应打包
	ssize_t recv_len;;
	socklen_t addrlen = sizeof(sin);
	unsigned short num = 1;
	while(1)
	{
		//接收数据
		bzero(buf, N);
		recv_len = recvfrom(sfd, buf, N, 0, (struct sockaddr*)&sin, &addrlen);
		if(recv_len < 0)
		{
			ERR_MSG("recvfrom");
			return -1;
		}

		//判断接收到的是数据包还是错误包
		if(3 == buf[1])
		{
			if(0 == flag) //防止文件重复打开
			{
				//存储从服务器下载下来的文件，接收数据成功后再创建数据
				//创建并打开filename文件
				fd = open(filename, O_WRONLY | O_CREAT | O_TRUNC, 0664);
				if(fd < 0)
				{
					ERR_MSG("open");
					return -1;
				}
				flag = 1;
			}
			//判断当前的块编号，是否与期望的块编号一直
			if(htons(num) == *(unsigned short *)(buf+2)) //防止数据包重复到达
			{
				//提取数据，写入到文件中，buf+4,

				if(write(fd, buf+4, recv_len-4) < 0)
				{
					printf("fd:%d recv_len = %ld\n", fd, recv_len);
					ERR_MSG("write");
					return -1;
				}

				//组ACK包
				buf[1] = 4;

				//发送数据 回复ACK
				if(sendto(sfd, buf, 4, 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
				{
					ERR_MSG("sendto");
				}

				if(recv_len < 512+2+2 )
				{
					printf("下载完成\n");
					break;
				}
				num++;
			}
		}
		else if(5 == buf[1])
		{
			printf("----ERROR:%s-----\n", buf+4);
			break;
		}
	}

	close(fd);
	return 0;
}