8.10 作业

实现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 <stdlib.h>
#include <fcntl.h>


//打印错误新的宏函数
#define ERR_MSG(msg)  do{\
	fprintf(stderr, " __%d__ ", __LINE__);\
	perror(msg);\
}while(0)


int main(int argc, const char *argv[])
{
	if(argc < 2)
	{
		fprintf(stderr, "请输入IP \n");
		return -1;
	}
/*	//将获取到的端口号字符串，转换成整形
	int port = atoi(argv[2]);
	if(port < 1024 || port > 49151)
	{
		fprintf(stderr, "port %d input error!! 1024~49151\n", port);
		return -1;
	}*/

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

	//绑定客户端自身的地址信息结构体 ---> 非必须绑定 
	
	//填充服务器的IP地址以及端口号 -->因为客户端要主动发送数据包给服务器
	struct sockaddr_in sin;
	sin.sin_family 		= AF_INET;
	sin.sin_port 		= htons(69);
	sin.sin_addr.s_addr = inet_addr(argv[1]);
//发送下载请求
    char name[20]="";
char buf[516]="";

	//请输入要请求的文件名
	printf("请输入要下载的文件名/n");
	scanf("%s",name);
    char *ptr = name;

	//存入操作码
	short int *pa =(short int *)name;
	*pa=htons(1);

	//存入文件名
	char *pm=ptr+2;
	strcpy(pm,name);

	char *a = pm+strlen(pm);
	*a = '\0';

	//存入模式
	char *pd = a+1;
	strcpy(pd,"octet");

	int size = 2+strlen(pm)+1+strlen(pd)+1;

    //
	int gf = open(argv[2],O_WRONLY|O_CREAT|O_TRUNC,0664);
		if(gf<0)
		{
			perror("open");
			return -1;
		}




	//向服务器发送请求包
	sendto(sfd,name,sizeof(name),0,(struct sockaddr*)&sfd,sizeof(sin));

//存储接受到的数据包来自哪里
	struct sockaddr_in bin;
	socklen_t addrlen = sizeof(bin);

	
    int num=1;
	char ack[4]="";
    ssize_t res; 
	while(1)
	{
		res=recvfrom(sfd,buf,516,0,(struct sockaddr *)&bin,&addrlen);
		if(res<0)
		{
			ERR_MSG("recvfrom");
			break;
		}

		//
		if(ntohs(*(short int*)buf) !=3)
		{
			continue;
		}

		//
		if(ntohs(*(short int*)(buf+2)) !=num)
		{
			continue;
		}

        write(gf,buf,res-4);
		if(res < 512)
		{
			printf("传输完成\n");
			break;
		}

		//
		*(short int*)ack=ntohs(4);
		*(short int*)(ack+2) = htons(num++);

		//向服务器发送ACK，表示想要接受下一个消息包
		if(sendto(sfd,buf,4,0,(struct sockaddr *)&bin,addrlen)<0)
		{
			ERR_MSG("sendto");
			return -1;
		}
	}

		close(gf);
		close(sfd);
		return 0;
	}