网络编程：3

1. 将UDP的服务器客户端重新搭建

#include <myhead.h>

#define ERR_MSG(msg) do{\
	fprintf(stderr, "__%d__", __LINE__);\
	perror("msg");\
}while(0)

#define IP "192.168.10.139"
#define PORT 6666

int main(int argc, const char *argv[])
{
	//获取UDP套接字
	int sfd = socket(AF_INET, SOCK_DGRAM, 0);
	if (sfd < 0)
	{
		ERR_MSG("socket");
		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 (bind(sfd, (struct sockaddr *)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success\n");



	//接收客户端信息
	struct sockaddr_in cin;
	socklen_t cinler = sizeof(cin);
	char buf[128]="";
	while (1)
	{
		bzero(buf, sizeof(buf));
		//接收信息
		if (recvfrom(sfd, buf, sizeof(buf), 0, (struct sockaddr *)&cin, &cinler) < 0)
		{
			ERR_MSG("recvfrom");
			return -1;
		}
		printf("[%d : %s] : %s\n", ntohs(cin.sin_port), inet_ntoa(cin.sin_addr), buf);



		if (connect(sfd, (struct sockaddr *)&cin, sizeof(cin)) < 0)
		{
			ERR_MSG("connect");
			return -1;
		}
		printf("connect success\n");
		while(1)
		{
			bzero(buf, sizeof(buf));
			//接收信息
			if (recvfrom(sfd, buf, sizeof(buf), 0, (struct sockaddr *)&cin, &cinler) < 0)
			{
				ERR_MSG("recvfrom");
				return -1;
			}
			printf("[%d : %s] : %s\n", ntohs(cin.sin_port), inet_ntoa(cin.sin_addr), buf);



			if (strcmp(buf, "quit") == 0)
			{
				break;
			}



			//发送信息
			strcat(buf, "*_*");
			if (sendto(sfd, buf, sizeof(buf), 0, (struct sockaddr *)&cin, sizeof(cin)) < 0)
			{
				ERR_MSG("sendto");
				return -1;
			}
			printf("发送成功\n");
		}


		//清除内核对端信息
		cin.sin_family = AF_UNSPEC;
		if (connect(sfd, (struct sockaddr *)&cin, sizeof(cin)) < 0)
		{
			ERR_MSG("connect");
			return -1;
		}
		printf("清除成功\n");


	}

	close(sfd);

	return 0;
}

#include <myhead.h>

#define ERR_MSG(msg) do{\
	fprintf(stderr, "__%d__", __LINE__);\
	perror("msg");\
}while(0)

#define IP "192.168.10.139"
#define PORT 6666

int main(int argc, const char *argv[])
{
	//获取UDP套接字
	int sfd = socket(AF_INET, SOCK_DGRAM, 0);
	if (sfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}



	//绑定服务器信息
	struct sockaddr_in sin;
	sin.sin_family      = AF_INET;
	sin.sin_port        = htons(PORT);
	sin.sin_addr.s_addr = inet_addr(IP);



	//接收服务器信息
	struct sockaddr_in cin;
	socklen_t cinlen = sizeof(cin);
	char buf[128]="";
	while (1)
	{
		bzero(buf, sizeof(buf));
		//发送信息
		fgets(buf, sizeof(buf), stdin);
		buf[strlen(buf) - 1] = 0;
		if (sendto(sfd, buf, sizeof(buf), 0, (struct sockaddr *)&sin, sizeof(sin)) < 0)
		{
			ERR_MSG("sendto");
			return -1;
		}
		printf("发送成功\n");



		//接收信息
		if (recvfrom(sfd, buf, sizeof(buf), 0, (struct sockaddr *)&cin, &cinlen) < 0)
		{
			ERR_MSG("recvfrom");
			return -1;
		}
		printf("[%d : %s] : %s\n", ntohs(cin.sin_port), inet_ntoa(cin.sin_addr), buf);

	}

	close(sfd);

	return 0;
}

2. 编写tftp的下载代码

#include <myhead.h>

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

#define IP "192.168.10.138"
#define PORT 69

int do_download(int cfd,struct sockaddr_in sin);

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



   //填充服务器的地址信息结构体
	struct sockaddr_in sin;
	sin.sin_family      = AF_INET;        
	sin.sin_port        = htons(PORT);   
	sin.sin_addr.s_addr = inet_addr(IP);   
	
	
	
	char c = 0;
	while (1)
	{

		printf("请选择：\n\t 1>下载\n\t 2>上传\n\t 3>退出\n");

		c = getchar();
		while (getchar() != 10)

		printf("%c\n",c);
		switch (c)
		{
			case '1':
				do_download(cfd, sin);
				break;
			case '2':
							
				break;
			case '3':
				goto END;
			default:
				printf("输入错误，请重新输入:\n");
		}

	}



END:
	//关闭文件描述符
	close(cfd);

	return 0;
}
int do_download(int cfd,struct sockaddr_in sin)
{
   	char name[20] = "";
	printf("请输入文件名：");
	scanf("%s", name);
	while(getchar() != 10);



	//组数据包
	char buf[516] = "";

	unsigned short *ptr1 = (unsigned short *)buf;
	*ptr1 = htons(1);

	char *ptr2 = buf + 2;
	strcpy(ptr2, name);

	char *ptr3 = ptr2 + strlen(name);
	*ptr3 = 0;

	char *ptr4 = ptr3 + 1;
	strcpy(ptr4, "octet");

	int size = 2 + 2 + 1 + strlen(ptr2) + strlen(ptr4);



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



    //打开一个用于写入的文件
    int fd = open(ptr2, O_WRONLY|O_CREAT|O_APPEND, 0664);
    if(fd < 0)
	{
        ERR_MSG("fd err");
		return -1;
	}



    while(1)
	{
		bzero(buf, sizeof(buf));
		socklen_t addlen = sizeof(sin);
		int res = 0;
		if((res = recvfrom(cfd, buf, sizeof(buf), 0, (struct sockaddr *)&sin, &addlen)) < -1)
		{
			ERR_MSG("recvfrom");
			return -1;
		}



		//写入本地
		char *data = buf + 4;
		write(fd, data, 512);



		//组ACK包
		unsigned short net = *(ptr1 + 1);
		bzero(buf, sizeof(buf));
		*ptr1 = htons(4);
		*(ptr1 + 1) = net;



		//回复ACK
		if(sendto(cfd, buf, 4, 0, (struct sockaddr *)&sin, sizeof(sin)) < 0)
		{
			ERR_MSG("sendto");
			return -1;
		}



		//判断数据是否小于512，若小于512，则传输结束
		if(res<516)
			break;

	}

    close(fd);
    return -1;
}

3. 尝试编写tftp的上传代码

4. 将今天总结的重点进行整理归纳