华清远见第五课程day3作业

TFTP下载

#include<myhead.h>

#define ERR_MSG(msg)do{\
	fprintf(stderr,"__%d__:",__LINE__);\
	perror(msg);\
}while(0)
#define PORT 69
#define IP "192.168.114.114"
int do_download(int cfd,struct sockaddr_in sin);
int main(int argc, const char *argv[])
{
	//创建报式套接字 socket
	int cfd = socket(AF_INET,SOCK_DGRAM,0);
	if(cfd <0){
		ERR_MSG("socket");
		return -1;
	}
	printf("socket create success sfd=%d\n",cfd);

	//填充服务器的地址信息结构体给sendto函数使用
	//真实的地址信息结构体根据地址族制定 AF_INET
	struct sockaddr_in sin;
	sin.sin_family =AF_INET;
	sin.sin_port = htons(PORT);
	sin.sin_addr.s_addr= inet_addr(IP);
	
	char a;
	while(1){
		system("clear");
		printf("1.下载\n");
		printf("2.上传\n");
		printf("3.退出\n");
		printf("请输入>>");
		scanf("%c",&a);
		while(getchar()!=10);

		switch(a){
		case '1':
			do_download(cfd,sin);
			break;
		case '2':
			//do_upload();
		case '3':
			goto END;
			break;
		default:
			printf("输入错误请重新输入");
			break;
		}
		printf("输入任意字符继续\n");
		while(getchar()!=10);
	}
END:
	//关闭套接字
	close(cfd);

	return 0;
}

int do_download(int cfd,struct sockaddr_in sin){	
	socklen_t sinlen=sizeof(sin);
	unsigned char buff[600] = {0};
    //返回的ACK
    unsigned char _ack[4];
    unsigned short code = 0; //操作码
    unsigned short num = 0;  //块编号 或者 错误码
    //数据的长度以512Byte传输
    char text[512] = {0}; //文件内容 或 错误信息
    //校验收到的块编号
    int N = 0;
    //返回的文件描述符
    int fd;
    int ret = 0;

    //输入文件名
    char filename[32] = {0};
    printf("下载文件名: ");
    scanf("%s", filename);

    //使用 sprintf 组包
    //返回值：成功格式化字符的个数           //-读-  //文件名 //0 //二进制模式//0
    ret = sprintf(buff, "%c%c%s%c%s%c", 0, 1, filename, 0, "octet", 0);

    //首次发送请求-----想要发送的数据的字节数-阻塞

    if (-1 == sendto(cfd, buff, ret, 0, (struct sockaddr *)&sin,sizeof(sin)))
        ERR_MSG("recvfrom error");

    //循环接收服务器发来的数据包
    while (1)
    {
        //接收--需要保存服务器的网络信息结构体  因为里面有临时端口
        if (-1 == (ret = recvfrom(cfd, buff, 600, 0, (struct sockaddr *)&sin, &sinlen)))
            ERR_MSG("recvfrom error");

        //解析数据包中的内容
        //将无符号2字节整型  网络-->主机
        //解析操作码
        code = ntohs(*(unsigned short *)buff);
        //解析块编号 或者 错误码
        num = ntohs(*(unsigned short *)(buff + 2));
        //解析文件内容 或 错误信息
        strncpy(text, buff + 4, 512);
        if (3 == code && num == N + 1)
        {
            //校验块编号+1
            N++;
            //要接收的数据包
            //如果是第一次接到数据包 要创建文件
            if (num == 1)
            {
                if (-1 == (fd = open(filename, O_WRONLY | O_CREAT | O_TRUNC, 0664)))
                    ERR_MSG("open error");
            }
            //将文件内容写入文件
            if (-1 == write(fd, text, ret - 4))
                ERR_MSG("write error");
            // 组装ACK
            *(unsigned short *)_ack = htons(4);
            *(unsigned short *)(_ack + 2) = htons(num);
           // 回复ACK包
            if (-1 == sendto(cfd, _ack, 4, 0, (struct sockaddr *)&sin,sinlen))
                ERR_MSG("recvfrom error");
            //文件接收完毕
            if (ret < 512)
                break;
        }
        else if (5 == code)
        {
            printf("接收出错[%s]\n", text); //错误信息
			return -1;
        }
    }

    printf("文件[%s]下载完成\n", filename);
    close(cfd);
    return 0;
}