tftp服务器的下载与上传

程序代码

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

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

#define IP "192.168.8.86"     //填windowsIP
#define PORT 69

//声明
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;
	}


	//填充客户端自身的地址信息结构体
	struct sockaddr_in cin;
	cin.sin_family      = AF_INET;
	cin.sin_port        = htons(43210);
	cin.sin_addr.s_addr = inet_addr("192.168.8.86");


	//填充服务器自身的地址信息结构体，AF_INET: man 7  IP
	//供于下方的sendto使用，因为sendto必须要指明发给谁
	struct sockaddr_in sin;
	sin.sin_family 		= AF_INET;
	sin.sin_port 		= htons(PORT);
	sin.sin_addr.s_addr = inet_addr(IP);
	socklen_t addrlen = sizeof(sin);
	char choose = 0;
	while(1)
	{
		printf("*****************************\n");
		printf("************1.下载***********\n");
		printf("************2.上传***********\n");
		printf("************3.退出***********\n");
		printf("*****************************\n");
		printf("请输入选项>>>");
		choose = getchar();
		while(getchar()!=10);
		switch(choose)
		{
		case '1':
			//调用下载函数
			do_download(sfd,sin);
			break;
		case '2':
			//调用上传函数
			do_upload(sfd,sin);
			break;
		case '3':
			//退出
			goto END;
			break;
		}
	}
END:
	//关闭套接字
	close(sfd);
	return 0;
}


//下载函数
int do_download(int sfd,struct sockaddr_in sin)
{
	char buf[1000] = "";
	ssize_t res = 0;
	char filename[20] = "";
	printf("请输入要下载的文件名>>>");
	scanf("%s",filename);
	while(getchar()!=10);

	//组下载协议
	short* p1 = (short*)buf;
	*p1 = htons(1);
	char *p2 = buf + 2;
	strcpy(p2,filename);
	char *p3 = p2 + strlen(p2);
	*p3 = 0;
	char *p4 = p3 + 1;
	strcpy(p4,"octet");

	int size = 2+strlen(p2)+1+strlen(p4)+1;

	//发送下载协议
	if(sendto(sfd, buf,size, 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("sendto");
		return -1;
	}
	//printf("sendto success\n");
	socklen_t addrlen = sizeof(sin);
	unsigned short num = 0;
	int flag = 0;
	int fd = -1;
	short *q1 = NULL;
	while(1)
	{
		//接收服务器发送过来的数据包
		bzero(buf,sizeof(buf));
		res = recvfrom(sfd,buf,sizeof(buf),0,(struct sockaddr*)&sin,&addrlen);
		if(res < 0)
		{
			ERR_MSG("recvfrom");
			return -1;
		}
		//判断是否错误
		if(3 == buf[1])
		{
			if(0 == flag)
			{
				//以写的方式打开文件
				fd = open(filename,O_WRONLY|O_CREAT|O_TRUNC,0664);
				if(fd < 0)
				{
					ERR_MSG("open");
					return -1;
				}
			}
			flag = 1;
		}
		if(htons(num+1) == *(unsigned short*)(buf+2))
		{
			//将接收到的数据写入文件中
			if(write(fd,buf+4,res-4) < 0)
			{
				ERR_MSG("write");
				return -1;
			}

			//发送ACK包给服务器
			//组ACK包
			q1 = (short*)buf;
			*q1 = htons(4);
			//发送收到的数据包的前4个字节
			if(sendto(sfd, buf,4, 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
			{
				ERR_MSG("sendto");
				return -1;
			}
			//printf("sendto success\n");

			//判断数据包大小是否小于512
			if((res-4) < 512)
			{
				printf("下载完毕\n");
				break;
			}
			num++;
		}
	}
	close(fd);
	return 0;
}


//上传函数
int do_upload(int sfd,struct sockaddr_in sin)
{
	char buf[1000] = "";
	ssize_t res = 0;
	char filename[20] = "";
	printf("请输入要上传的文件名>>>");
	scanf("%s",filename);
	while(getchar()!=10);
    unsigned short num = 0;

	//判断文件是否存在
	int fd = open(filename,O_RDONLY);
	if(fd < 0)
	{
		ERR_MSG("open");
		return -1;
	}
	//发送上传请求到69号端口
	short* p1 = (short*)buf;
	*p1 = htons(2);
	char *p2 = buf + 2;
	strcpy(p2,filename);
	char *p3 = p2 + strlen(p2);
	*p3 = 0;
	char *p4 = p3 + 1;
	strcpy(p4,"octet");
	int size = 2+strlen(p2)+1+strlen(p4)+1;
	if(sendto(sfd, buf,size, 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("sendto");
		return -1;
	}
	//printf("sendto success\n");
	socklen_t addrlen = sizeof(sin);
	while(1)
	{
		//接收数据包
		bzero(buf,sizeof(buf));
		res = recvfrom(sfd,buf,sizeof(buf),0,(struct sockaddr*)&sin,&addrlen);
		if(res < 0)
		{
			ERR_MSG("recvfrom");
			return -1;
		}

	    //组数据包
		bzero(buf,sizeof(buf));
		short *p1 = (short*)buf;
		*p1 = htons(3);
		*(p1+1) = htons(num+1);
		ssize_t res = read(fd,buf+4,512);
		if(res < 0)
		{
			ERR_MSG("read");
			return -1;
		}
		//发送到服务器
		if(sendto(sfd, buf,res+4, 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
		{
			ERR_MSG("sendto");
			return -1;
		}
		//判断什么时候结束
		if(res < 512)
		{
			printf("上传完毕\n");
			break;
		}
		num++;
	}
	close(fd);
	return 0;
}

运行结果