tftp下载与上传

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

#define SERVER_IP "192.168.8.161" //服务器 IP
#define SERVER_PORT 69             //服务器端口号

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

int main(int argc, char *argv[])
{
	int cfd,fd;
	char c;
	ssize_t res = 0, res1 = 0;
	char str[100] ;
	short int data[258];
	short int  ack[2];
	short int num = 1;
	int flag;
	struct sockaddr_in server_addr;
	socklen_t server_addr_size;

	//准备协议信息
	short int* p1 = (short int*)str;
	printf("请输入操作:   1. 下载   2.上传\n");
	c = getchar();
	if('1' == c)
		*p1 = htons(1);
	else if('2' == c)
		*p1 = htons(2);
	else
	{
		printf("输入错误\n");
		return -1;
	}
	while(getchar() != '\n');
	char* p2 = (char*)(p1+1);
	printf("请输入文件名:");
	scanf("%s",p2);
	char* p3 = p2 + strlen(p2);	
	*p3 = 0;
	char* p4 = p3 + 1;
	strcpy(p4,"octet");
	char* p5 = p4 + strlen(p4);
	*p5 = 0;

	//连接tftp服务器
	//1.创建数据报式套接字
	cfd = socket(AF_INET, SOCK_DGRAM, 0);
	if (cfd == -1)
	{
		ERROR("socket");
		return -1;
	}
	memset(&server_addr, 0, sizeof(struct sockaddr_in));
	//填充服务器地址信息结构体
	server_addr.sin_family = AF_INET;
	server_addr.sin_port = htons(SERVER_PORT);   //端口的网络字节序
	server_addr.sin_addr.s_addr = inet_addr(SERVER_IP);  //IP的网络字节序
	server_addr_size = sizeof(struct sockaddr_in);
	//向服务器发送读写请求
	if(sendto(cfd, str, sizeof(str), 0, (struct sockaddr*)&server_addr, server_addr_size) < 0)
	{
		ERROR("sendto");
		return -1;
	}
	//下载文件
	if('1' == c)
	{
		while(1)
		{
			bzero(data,sizeof(data));
			//接收信息
			res = recvfrom(cfd, data, sizeof(data), 0, (struct sockaddr*)&server_addr, &server_addr_size);
			if(res < 0)
			{
				ERROR("recv");
				return -1;
			}
			if( htons(3) == data[0])
			{
				if(flag)
				{
					if((fd = open(p2,O_RDWR|O_CREAT|O_TRUNC,0777)) < 0)
					{
						ERROR("open");
						return -1;
					}
					flag = 0;
				}
				if(htons(num) == data[1])
				{
					num++;
					if(write(fd, data+2,res-4) < 0)
					{
						ERROR("write");
						return -1;
					}

					if(res < 516)
					{
						printf("下载完毕\n");
						break;
					}

					ack[0] = htons(4);
					ack[1] = data[1];

					//发送信息
					if(sendto(cfd, ack, sizeof(ack), 0, (struct sockaddr*)&server_addr, server_addr_size) < 0)
					{
						ERROR("sendto");
						return -1;
					}	
				}
			}
			else if(htons(5) == data[0])
			{
				printf("%s\n",(char*)(data+2));
				break;
			}
		}
	}
	else if('2' == c)//上传文件
	{
		if((fd = open(p2,O_RDONLY)) < 0)
		{
			ERROR("open");
			return -1;
		}	
		num = 0;

		while(1)
		{
			bzero(data,sizeof(data));
			bzero(ack,sizeof(ack));
			//接收信息
			res = recvfrom(cfd, data, sizeof(data), 0, (struct sockaddr*)&server_addr, &server_addr_size);
			if(res < 0)
			{
				ERROR("recv");
				return -1;
			}
			if(htons(4) == data[0])
			{
				if(htons(num) != data[1])
				{
					lseek(fd, -res1,SEEK_CUR);
					printf("__%d__\n",__LINE__);
					num--;
				}
				if((res1 = read(fd, data+2,sizeof(data)-4)) < 0)
				{
					ERROR("write");
					return -1;
				}

				data[0] = htons(3);
				data[1] = htons(++num);

				//发送信息
				if((res = sendto(cfd, data, res1+4, 0, (struct sockaddr*)&server_addr, server_addr_size)) < 0)
				{
					ERROR("sendto");
					return -1;
				}
				if(res1 < 512)
				{
					printf("上传成功\n");
					break;
				}
			}
			else if(htons(5) == data[0])
			{
				printf("%s\n",(char*)(data+2));
				break;
			}

		}

	}
	return 0;
}