22081-10-21 tffp客户端的上传，广播代码

最新推荐文章于 2024-09-10 19:16:08 发布

狸暗

最新推荐文章于 2024-09-10 19:16:08 发布

阅读量244

点赞数

分类专栏：嵌入式培训文章标签： linux 运维服务器 ubuntu

本文链接：https://blog.csdn.net/wyl2333/article/details/127469178

版权

嵌入式培训专栏收录该内容

63 篇文章 0 订阅

订阅专栏

本文展示了两个C程序，分别实现了TFTP客户端的文件上传和下载功能，以及UDP广播的发送与接收。TFTP客户端通过UDP协议与服务器进行交互，用户可以选择上传或下载文件，并能正确处理数据包的传输。同时，文章还提供了UDP广播的发送和接收示例，展示了如何在局域网内广播数据并接收响应。

摘要由CSDN通过智能技术生成

1、tftp客户端的上传和下载

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/stat.h>
#include <unistd.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <fcntl.h>
int do_download(int sfd,struct sockaddr_in sin);
int do_upload(int sfd,struct sockaddr_in sin);

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

int main(int argc, const char *argv[])
{
	//创建报式套接字
	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;//必须填AF_INET
	sin.sin_port = htons(69);//服务器端口号69
	sin.sin_addr.s_addr = inet_addr("192.168.1.105");
	//将点分十进制ip转化为网络字节序


	char c=0;
	//菜单
	while(1)
	{
		printf("------------------------------------\n");
		printf("--------------1.上传----------------\n");
		printf("--------------2.下载----------------\n");
		printf("--------------3.退出----------------\n");
		printf("请输入>>>");
		c=getchar();
		while(getchar()!=10);
		switch(c)
		{
		case '1':
			do_upload(sfd,sin);
			break;
		case '2':
			do_download(sfd,sin);
			break;
		case '3':
			goto END;
			break;
		default:
			printf("输入错误\n");
		}
	}
	//关闭文件描述符
END:	
	close(sfd);
	return 0;
}


int do_download(int sfd,struct sockaddr_in sin)
{
	//组下载请求，发送给服务器，服务器在69号端口等待请求
	char buf[520]="";
	//操作码
	unsigned short *pa=(unsigned short*)buf;
	*pa=htons(1);
	//文件名
	printf("请输入文件名>>");
	char name[32]="";
	scanf("%s",name);
	char *pb=buf+2;
	strcpy(pb,name);
	//0
	char *pc=pb+strlen(pb);
	//模式
	char *pd=pc+1;
	strcpy(pd,"octet");
	//0

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

	//客户端发送下载请求
	if(sendto(sfd,buf,size,0,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("sendto");
		return -1;
	}
	printf("下载请求成功\n");

	//打开文件
	int fd=open("./1.c",O_RDWR|O_CREAT|O_TRUNC,0777);
	if(fd<0)
	{
		ERR_MSG("open");
		return -1;
	}
	printf("文件打开成功\n");
	socklen_t addrlen=sizeof(sin);
	ssize_t res=0;
	unsigned short num=0;
	while(1)
	{
		//客户端接收来自服务器的数据包
		bzero(buf,sizeof(buf));
		res=recvfrom(sfd,buf,516,0,(struct sockaddr*)&sin,&addrlen);
		if(res<0)
		{
			ERR_MSG("recvfrom");
			return -1;
		}
	//		printf("%d %d\n",buf[0],buf[1]);
	//		printf("%d %d\n",*(unsigned short*)buf,ntohs(*(unsigned short*)buf));
		//	由于发送回来的操作码是网络字节序，因此只需要判断buf[1]是3还是5即可

		if(buf[1]==3)
		{
			//由于UDP传输 可能丢包或者重复，因此每次收到数据包时都要判断是不是我需要的
			if(num+1==ntohs(*(unsigned short*)(buf+2)))
			{
			/*	//将数据输出到终端
				printf("%s\n",buf+4);
				fflush(stdout);*/
				//将下载的数据写到文件中
					ssize_t reg=write(fd,buf+4,res-4);
					if(reg<0)
					{
					ERR_MSG("write");
					return -1;
					}

				//客户端回复ACK给服务器
				buf[1]=4;
				if(sendto(sfd,buf,4,0,(struct sockaddr*)&sin,sizeof(sin))<0)
				{
					ERR_MSG("sendto");
					return -1;
				}

				if(res-2-2<512)
				{
					printf("下载完成\n");
					break;
				}
				num++;
			}


		}
		else if(buf[1]==5)
		{
			//错误包
			printf("error=%s\n",buf+4);
			return -2;
		}
	}
		close(fd);
		return 0;
	}
//上传
int do_upload(int sfd,struct sockaddr_in sin)
{
	//组上传请求，发送给服务器，服务器在69号端口等待请求
	char buf[520]="";
	//操作码
	unsigned short *pa=(unsigned short*)buf;
	*pa=htons(2);
	//文件名
	printf("请输入文件名>>");
	char name[32]="";
	scanf("%s",name);
	char *pb=buf+2;
	strcpy(pb,name);
	//0
	char *pc=pb+strlen(pb);
	//模式
	char *pd=pc+1;
	strcpy(pd,"octet");
	//0

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

	socklen_t addrlen=sizeof(sin);
	ssize_t ret=0;
	unsigned short num=0;

	//客户端发送上传请求
	if(sendto(sfd,buf,size,0,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("sendto");
		return -1;
	}
	printf("上传请求发送成功\n");

	//打开文件
	int fd=open(name,O_RDONLY);
	if(fd<0)
	{
		ERR_MSG("open");
		return -1;
	}
	printf("open success\n");
	char ack[4];
	unsigned short*ack0=(unsigned short*)ack;
	unsigned short *ack1=ack0+1;
	int acklen=sizeof(ack);
	bzero(buf,sizeof(buf));
	while(1)
	{
		//接收服务器回复包
		if(recvfrom(sfd,ack,4,0,(struct sockaddr*)&sin,&acklen)<0)
		{
			ERR_MSG("recvfrom");
			return -1;
		}
		printf("recvfrom succ\n");
		if(*ack0==htons(4))
		{
				//组数据包
				//操作码
				unsigned short *p1=(unsigned short *)buf;
				*p1=htons(3);
				//块编号
				unsigned short *p2=(unsigned short*)(buf+2);
				//数据
				char *p3=buf+4;
				//读取文件数据
				ret=read(fd,p3,512);
				if(ret<0)
				{
					ERR_MSG("read");
					return -1;
				}
				if(sendto(sfd,buf,ret+4,0,(struct sockaddr*)&sin,sizeof(sin))<0)
				{
						ERR_MSG("sendto");
				}
				if(ret<512)
				{
					printf("上传成功\n");
					break;
				}
				bzero(buf,sizeof(buf));
		}
	}
	close(fd);
	return 0;

}

ubuntu@ubuntu:02_1020$ gcc 05_tftpcli.c 
ubuntu@ubuntu:02_1020$ ./a.out
------------------------------------
--------------1.上传----------------
--------------2.下载----------------
--------------3.退出----------------
请输入>>>2
请输入文件名>>5.png
下载请求成功
文件打开成功
下载完成
请输入>>>3
ubuntu@ubuntu:02_1020$ eog 1.c

ubuntu@ubuntu:02_1020$ gcc 05_tftpcli.c 
ubuntu@ubuntu:02_1020$ ./a.out
------------------------------------
--------------1.上传----------------
--------------2.下载----------------
--------------3.退出----------------
请输入>>>1
请输入文件名>>2.c
上传请求发送成功
open success
recvfrom succ
上传成功
请输入>>>3
ubuntu@ubuntu:02_1020$

2、广播

//广播的接收

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <netinet/in.h>
int main(int argc, const char *argv[])
{
	//创建报式套接字
	int sfd=socket(AF_INET,SOCK_DGRAM,0);
	if(sfd<0)
	{
		perror("socket");
		return -1;
	}

	//填充IP和端口
	struct sockaddr_in sin;
	sin.sin_family=AF_INET;//必填
	sin.sin_port=htons(2212);//端口1024-49151
	sin.sin_addr.s_addr=inet_addr("192.168.52.255");

	//绑定IP和端口
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		perror("bind");
		return -1;
	}
	//存储由谁发送数据包的IP和端口
	struct sockaddr_in cin;
	socklen_t addrlen=sizeof(cin);
	char buf[128]="";
	ssize_t ret=0;
	while(1)
	{
		bzero(buf,sizeof(buf));
		//接收数据
		ret=recvfrom(sfd,buf,sizeof(buf),0,(struct sockaddr*)&cin,&addrlen);
		if(ret<0)
		{
			perror("recvfrom");
			return -1;
		}
		printf("[%s:%d]%s\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),buf);

	
	}
	close(sfd);

	return 0;
}

//广播的发送

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <netinet/in.h>
int main(int argc, const char *argv[])
{
	//创建报式套接字
	int sfd=socket(AF_INET,SOCK_DGRAM,0);
	if(sfd<0)
	{
		perror("socket");
		return -1;
	}
	//设置允许广播
	int set_board=1;
	if(setsockopt(sfd,SOL_SOCKET,SO_BROADCAST,&set_board,sizeof(set_board))<0)
	{
		perror("setsockopt");
		return -1;
	}
	

	//填充接收端的IP和端口
	struct sockaddr_in sin;
	sin.sin_family=AF_INET;//必填
	sin.sin_port=htons(2212);//端口1024-49151
	sin.sin_addr.s_addr=inet_addr("192.168.52.255");

/*	//绑定IP和端口
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		perror("bind");
		return -1;
	}
	*/
	char buf[128]="";
	ssize_t ret=0;
	int i=0;
	while(1)
	{
		bzero(buf,sizeof(buf));
		printf("please input:");
		fgets(buf,sizeof(buf),stdin);
		buf[strlen(buf)-1]=0;
		//发送数据
		ret=sendto(sfd,buf,sizeof(buf),0,(struct sockaddr*)&sin,sizeof(sin));
		if(ret<0)
		{
			perror("recvfrom");
			return -1;
		}
		printf("send success\n");
	
	}
	close(sfd);

	return 0;
}