【网络编程day3作业】

1.编写基于udp的文件传输的下载模块

 

#include <stdio.h>
#include <sys/types.h>         
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h> 
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>
#include <sys/stat.h>
#include <fcntl.h>
 
#define ERR_MSG(msg) do{\
	printf("line:%d\n",__LINE__);\
	perror(msg);\
}while(0);
 
#define SER_PORT 69
#define SER_IP "192.168.10.114"
 
int main(int argc, const char *argv[])
{
	//创建流式套接字
	int cfd = socket(AF_INET,SOCK_DGRAM,0);
	if(cfd<0)
	{
		ERR_MSG("socket");
		return -1;
	}
 
	printf("socket create sucess cfd=%d __%d__\n",cfd,__LINE__);
 
	//非必须绑定
	//如果不绑定，则程序启动后，操作系统会默认绑定本机SER_IP和随机端口
 
	//填充客户端的地址信息结构体，给bind函数使用
 
 
 
 
	//该结构体在 man 7 ip（表示真实的地址信息结构体）
	struct sockaddr_in sin;
	sin.sin_family 		= AF_INET; 				//必须填充AF_INET
	sin.sin_port 		= htons(SER_PORT); 			//端口号的网络字节序1024~49151
	sin.sin_addr.s_addr = inet_addr(SER_IP); 		//ifconfig查看本机IP
 
	struct sockaddr_in rcvaddr;
	socklen_t len =sizeof(rcvaddr);
	
 
 
 
 
	//组协议
	char buf[516]="";
	char buf1[20]="";
    printf("请输入你要下载的文件名：");
	scanf("%s",buf1);
	short *ptr1 = (short *)buf;
	*ptr1 = htons(1); 						//操作码，下载
 
	char *ptr2 = buf+2;
	strcpy(ptr2,buf1); 					//要下载的文件名
 
	char *ptr3 = ptr2+strlen(ptr2);
	*ptr3 = 0;
 
	char *ptr4 = ptr3+1;
	strcpy(ptr4,"octet"); 					//模式
 
	int size = 2+strlen(ptr2)+1+strlen(ptr4)+1;
 
 
 
	//打开文件
	int fd = open("./copy.txt",O_WRONLY|O_CREAT|O_TRUNC,0777);
	if(fd<-1)
	{
		ERR_MSG("open");
		return -1;
	}
 
 
 
	//发送下载请求包给服务器
	if(sendto(cfd,buf,size,0,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("sendto");
		return -1;
	}
	printf("发送成功\n");
 
 
 
 
 
	short *q1 = (short *)buf;
 
	short *q2 = q1+1;
 
	char *q3 = (char*)(q2+1);
 
 
 
 
	int i=1;
 
	ssize_t res = 0;
	while(1)
	{
		//清空数据包
		bzero(buf,sizeof(buf));
 
 
		//接收数据
		res = recvfrom(cfd,buf,sizeof(buf),0,(struct sockaddr *)&rcvaddr,&len);
		if(res < 0)
		{
			ERR_MSG("recvfrom");
			return -1;
		}
 
 
		//判断是否ERROR
		if(*q1==htons(5))
		{
			printf("ERROR:[%d]  %s\n",ntohs(*q2),q3);
		}
 
 
		//将数据写入文件
		if(write(fd,q3,res-4)<0)
		{
			ERR_MSG("write");
			return -1;
		}	
 
	
		//ACK应答	
		bzero(q3,512);
		*q1 = htons(4);
		if(sendto(cfd,buf,4,0,(struct sockaddr *)&rcvaddr,len)<0)
		{
			ERR_MSG("sendto");
			return -1;
		}
		printf("[%d]发送成功\n",i);
		i++;	
 
 
		if(res<516)
		{
			printf("文件读取完毕\n");
			break;
		}
 
		;
 
	}
	close(cfd);
 
 
		return 0;
}

结果：

socket create sucess cfd=3 __30__
请输入你要下载的文件名：5.png
发送成功
[1]发送成功
[2]发送成功
[3]发送成功
[4]发送成功
[5]发送成功
[6]发送成功
[7]发送成功

7]发送成功
[148]发送成功
[149]发送成功
[150]发送成功
[151]发送成功
[152]发送成功
[153]发送成功
[154]发送成功
[155]发送成功
[156]发送成功
文件读取完毕