网络编程DAY3

TFTP服务器 下载&上传

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

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

#define IP "192.168.2.12"  //TFTP的主机IP
#define PORT 69           //TFTP端口号
int download_file(int cfd,struct sockaddr_in sin);
int upload_file(int cfd,struct sockaddr_in sin);

int main(int argc, const char *argv[])
{
	//创建报式套接字
	int cfd=socket(AF_INET,SOCK_DGRAM,0);
	if(cfd<0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("create socket success,cfd=%d __%d__\n",cfd, __LINE__);

	//填充服务器的地址信息结构体
	//真实的地址信息结构体根据地址族制定：AF_INET:man 7 IP

	struct sockaddr_in sin;
	sin.sin_family      =AF_INET;          //必须填AF_INET
	sin.sin_port        =htons(PORT);      //端口号 网络字节序
	sin.sin_addr.s_addr =inet_addr(IP);    //IP     网络字节序

	int mode;
	while(1)
	{
		putchar(10);
		printf("/***************************************/\n");
		printf("================TFTP文件客户端==========\n");
		printf("------------------功能菜单--------------\n");
		printf("\t\t1.下载文件\n");
		printf("\t\t2.上传文件\n");
		printf("\t\t3.显示当前文件夹列表\n");
		printf("\t\t4.清屏\n");
		printf("\t\t5.退出\n");
		printf("/***************************************/\n");
		printf("请输入您的选择：");
		scanf("%d",&mode);

		switch(mode)
		{
		case 1:
			download_file(cfd,sin);
			break;
		case 2:
			upload_file(cfd,sin);
			break;
		case 3:
			system("ls -lh");
			break;
		case 4:
			system("clear");
			break;
		case 5:
			goto END;
			break;
		default:
			printf("输入错误，请重新输入\n");
		}
	}
END:
	//关闭套接字文件描述符
	close(cfd);
	return 0;	
}

//下载文件
int download_file(int cfd,struct sockaddr_in sin)
{
	socklen_t addrlen =sizeof(sin);//地址信息结构体大小
	char buf[516]="";
	ssize_t res=0;
	int fd=-1;//定义文件描述符

	char file_name[20]="";
	printf("请输入下载的文件名：");
	scanf("%s",file_name);
	getchar();

	//下载(读)请求包
	int size=sprintf(buf,"%c%c%s%c%s%c",0,1,file_name,0,"octet",0);
	if( sendto(cfd,buf,size,0,(struct sockaddr *)&sin,addrlen) <0)
	{
		ERR_MSG("sendto");
		return -1;
	}
	printf("client  sendto upload  success __%d__\n",__LINE__);

	unsigned short num=0;//记录数据包的编号

	//循环接收数据包，发送ACK
	while(1)
	{
		bzero(buf,sizeof(buf));
		res=recvfrom(cfd,buf,sizeof(buf),0,(struct sockaddr *)&sin,&addrlen);
		if(res<0)
		{
			ERR_MSG("recvfrom");
			return -1;
		}
		if(buf[1]==3)
		{
			if(fd<0)
			{
				//打开文件写权限
				fd=open(file_name,O_WRONLY|O_CREAT|O_TRUNC,0664);
				if(fd<0)
				{
					ERR_MSG("open");
					return -1;
				}
			}

			//判断服务器发过来的数据包是否重复，验证块编号，重复则重新接收
			if( htons(num+1) == *( (unsigned short *)(buf+2) )  )
			{
				num++;
				write(fd,buf+4,res-4);//最后一次的数据长度小于512
				printf("块编号=%d\n",ntohs(*( (unsigned short *)(buf+2) )));

				//发送应答包
				buf[1]=4;
				if( sendto(cfd,buf,4,0,(struct sockaddr *)&sin,addrlen) < 0 )
				{
					ERR_MSG("sendto");
					return -1;
				}
				printf("sendto ACK  success __%d__\n",__LINE__);
				//若数据小于512，则下载完毕
				if(res-2-2<512)
				{
					printf("文件下载成功\n");
					break;
				}
			}
		}
		if(buf[1]==5)
		{
			//错误包
			printf("upload failed:%s\n",buf+4);
			break;
		}
	}
	//文件描述符
	close(fd);
}

//上传文件
int upload_file(int cfd,struct sockaddr_in sin)
{
	socklen_t addrlen =sizeof(sin);//地址信息结构体大小
	char buf[516]="";
	ssize_t res=0;
	int count;

	char file_name[20]="";
	printf("请输入上传的文件名：");
	scanf("%s",file_name);
	getchar();

	//上传请求包
	int size=sprintf(buf,"%c%c%s%c%s%c",0,2,file_name,0,"octet",0);
	if( sendto(cfd,buf,size,0,(struct sockaddr *)&sin,addrlen) <0)
	{
		ERR_MSG("sendto");
		return -1;
	}
	printf("client  sendto upload  success __%d__\n",__LINE__);

	//打开文件读取权限
	int fd=open(file_name,O_RDONLY);
	unsigned short num=0;//记录数据包的编号

	//循环发送数据包，接受ACK
	while(1)
	{
		bzero(buf,sizeof(buf));
		res=recvfrom(cfd,buf,sizeof(buf),0,(struct sockaddr *)&sin,&addrlen);
		if(res<0)
		{
			ERR_MSG("recvfrom");
			return -1;
		}
		if(buf[1]==4)
		{
			//构造数据包
			*((unsigned short *)buf)=htons(3);
			*((unsigned short *)(buf+2))=htons(num+1);
			res=read(fd,buf+4,512);

			//发送数据包
			count=sendto(cfd,buf,res+4,0,(struct sockaddr *)&sin,addrlen);
			if(count<0)
			{
				ERR_MSG("sendto");
				return -1;
			}
			printf("sendto  success __%d__\n",__LINE__);

			if(res<512)
			{
				printf("文件上传成功\n");
				break;
			}
			num++;
		}
		if(buf[1]==5)
		{
			//错误包
			printf("upload failed:%s\n",buf+4);
			break;
		}
	}

	//文件描述符
	close(fd);
}

运行效果