基于UDP的tftp通信

最新推荐文章于 2024-09-14 14:53:27 发布

犩未

最新推荐文章于 2024-09-14 14:53:27 发布

阅读量197

点赞数

CC 4.0 BY-SA版权

分类专栏：网络编程文章标签： c#

本文链接：https://blog.youkuaiyun.com/ww1106/article/details/126923919

网络编程专栏收录该内容

7 篇文章

订阅专栏

本文介绍了一个基于UDP协议的客户端程序，该程序实现了文件的下载和上传功能。客户端通过发送特定格式的数据包与服务器进行交互，支持文件下载和上传操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >


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

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

#define PORT 69//服务器绑定的端口号
#define IP  "192.168.1.3"//服务器绑定的ＩＰ

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[])
{
	//创建报式套接字
	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(IP);

	char c = 0;
	while(1)
	{
		printf(" _______________________________\n");
		printf("|                               |\n");
		printf("|-*************菜单************-|\n");
		printf("|-***********１、下载**********-|\n");
		printf("|-***********２、上传**********-|\n");
		printf("|-***********３、退出**********-|\n");
		printf("|_______________________________|\n");
		printf("请输入(1,2,3)--->");

		c=getchar();

		while(getchar()!=10);
		switch(c)
		{
		case '1'://下载
			do_download(sfd,sin);
			break;
		case '2'://上传
			do_upload(sfd,sin);
			break;
		case '3'://退出
			goto END;
		default:
			printf("输入错误，请重新输入\n");
		}
	}
END:
	close(sfd);
	return 0;
}
//下载功能
int do_download(int sfd,struct sockaddr_in sin)
{
	char filename[20]="";
	printf("请输入要下载的文件名:");
	fgets(filename,20,stdin);
	filename[strlen(filename)-1]=0;
	printf("%s\n",filename);

	char buf[516] = "";//初始化字符串
	//组下载请求包
	char *ptr = buf;

	unsigned short* ptr1 = (unsigned short*)ptr;
	*ptr1 = htons(1);//下载请求的网络字节序

	char *ptr2 = ptr+2;
	strcpy(ptr2,filename);


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

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

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

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


	printf("发送下载请求包成功\n");
	//打开一个文件
	int fp = open("./copy.png",O_WRONLY|O_CREAT|O_TRUNC,0664);

	if(fp < 0)
	{
		ERR_MSG("open");
		return -1;
	}
	char data[516] ="";
	socklen_t addrlen=sizeof(sin);
	unsigned short num = 1;

	while(1)
	{
		bzero(data,sizeof(data));
	//	printf("111");
		//循环接收服务器返回的数据包
		ssize_t res = recvfrom(sfd,data,sizeof(data),0,(struct sockaddr*)&sin,&addrlen);
		if(res<0)
		{
			ERR_MSG("recvfrom");
			return -1;
		}
	//	printf("999");
		//分析是数据包还是错误包
		//如果是数据包，则将数据提取出来，数据在ｂｕｆ＋４的位置上
		unsigned short *p = (unsigned short *)data;
		if(htons(3)==*p)
		{
		//	printf("接收的包为数据包\n");
			if(htons(num) == *(unsigned short*)(data+2))
			{
			//	printf("00");
				char *q=(char*)p+4;
				if(write(fp,q,res-4) < 0)
				{
					ERR_MSG("write");
					return -1;
				}

				//ＡＣＫ

			//	short *b=(short*)data+1;
				char ACK[4]="";
				short *A=(short*)ACK;
				*A=htons(4);
				*(A+1)=htons(num);

				//	unsigned short *l=(unsigned short *)data;
				//	*(l+1)=htons(4);
				if(sendto(sfd,ACK,sizeof(ACK),0,(struct sockaddr*)&sin,sizeof(sin))<0)
				{
					ERR_MSG("sendto");
					return -1;
				}
				//printf("888");
				if(res<516)
				{
					printf("文件下载完毕\n");
					break;
				}
				num++;
			//	printf("%d",num);
			}
			else if(htons(5)==*p)
			{
				printf("接收的包为错误包,错误码及错误信息为：%s\n",data);
			}
			//printf("%d",num);
		//	num++;
		}

		//将提取出来的数据写入到文件中
		//组ＡＣＫ包，回复ＡＣＫ给服务器
		//直到数据内容的长度小鱼５１２个字节
		//如果是错误包，则打印错误信息，退出
	}

	return 0;
}

int do_upload(int sfd,struct sockaddr_in sin)
{
	//组上传请求包
	char filename[20]="";
	printf("请输入要上传的文件名:");
	fgets(filename,20,stdin);
	filename[strlen(filename)-1]=0;
	printf("%s\n",filename);
	char buf[516] = "";//初始化字符串
	//组上传请求包
	char *ptr = buf;

	unsigned short* ptr1 = (unsigned short*)ptr;
	*ptr1 = htons(2);//下载请求的网络字节序
	
	char *ptr2 = ptr+2;
	strcpy(ptr2,filename);


	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(filename,O_RDONLY);
	if(fd<0)
	{
		if(errno == ENOENT)
		{
			printf("文件不存在\n");
			return -1;
		}
		else
		{
			ERR_MSG("open");
			return -1;

		}
	}

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

	char data[516]="";
	unsigned short num = 0;
	ssize_t res;
	//发送上传请求包
	while(1)
	{

		bzero(data,sizeof(data));
		res = recvfrom(sfd,data,sizeof(data),0,(struct sockaddr*)&sin,&addrlen);
		if(res<0)
		{
			ERR_MSG("recvfrom");
			return -1;
		}
		unsigned short *p = (unsigned short *)data;
		if(htons(4)==*p)
		{
			if(htons(num)==*(unsigned short*)(data+2))
			{
				num++;
				*(unsigned short*)data=htons(3);
				*(unsigned short*)(data+2)=htons(num);
				char *q=data+4;
				int re;
				re= read(fd,q,512);
				if(re<0)
				{
					ERR_MSG("read");
					return -1;
				}
				else if(re==0)
				{
					printf("文件上传完毕");
					break;
				}

				if(sendto(sfd,data,re+4,0,(struct sockaddr*)&sin,sizeof(sin))<0)
				{
					ERR_MSG("sendto");
					return -1;
				}
			//	num++;
			}
			else
			{
				printf("上传失败\n");
				break;
			}
		}
		else if(htons(5)==data[1])
		{
			printf("上传失败，错误码是%s\n",data+4);
			break;
		}

	}
	return 0;
}