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



int udp_client_put(int sockfd,SI addr)
{
		printf("请输入要上传的文件\n");
		char str[256];
		scanf("%s",str);
		socklen_t len = sizeof(addr);
		char buf[256] = {};
		strcpy(buf,str);
		FILE* fd = fopen(str,"rb");
		
		if(NULL == fd)
		{
			strcpy(buf,"quit");
			sendto(sockfd,buf,strlen(buf)+1,0,(SA)&addr,len);
			printf("文件不存在\n");
			return ERROR;
		}
		//发送文件名
		sendto(sockfd,buf,strlen(buf)+1,0,(SA)&addr,len);
		
		//校验时间
		struct stat state = {};
		stat(buf,&state);
		sprintf(buf,"%lu",state.st_atime);
		sendto(sockfd,buf,strlen(buf)+1,0,(SA)&addr,len);
		//接收命令
		recvfrom(sockfd,buf,sizeof(buf),0,(SA)&addr,&len);
		if(!strcmp("exis",buf))
		{
			//接收从第几个字节开始
			recvfrom(sockfd,buf,sizeof(buf),0,(SA)&addr,&len);
			long ret = 0;
			sscanf(buf,"%ld",&ret);
			fseek(fd,ret,SEEK_CUR);
		}
		
		while(1)
		{
			char ch[1024] = {};
			int ret = fread(ch,1,10,fd);
			sendto(sockfd,ch,strlen(ch)+1,0,(SA)&addr,len);
			chmod(str,0644);
			if(10 > ret)
			{
				strcpy(ch,"close");
				sendto(sockfd,ch,strlen(ch)+1,0,(SA)&addr,len);
				
				uint8_t enc1[16] = {};
				uint8_t enc2[16] = {};
				getmd5_file(fd,enc1);
				recvfrom(sockfd,enc2,sizeof(enc2),0,(SA)&addr,&len);
				if(0 == md5cmp(enc1,enc2))
				{
					printf("传输完成\n");	
				}
				else
				{
					printf("传输失败文件有损坏\n");	
				}	
				fclose(fd);
				return SUCCESS;		
			}
		}		
	return SUCCESS;		
}


int udp_client_get(int sockfd,SI addr)
{
	printf("请输入要下载的文件\n");
	char str[256];
	char buf[256] = {};
	scanf("%s",str);
	socklen_t len = sizeof(addr);
	//发送名字
	sendto(sockfd,str,strlen(str)+1,0,(SA)&addr,len);
	recvfrom(sockfd,buf,sizeof(buf),0,(SA)&addr,&len);
	if(!strcmp(buf,"quit"))
	{
		printf("文件不存在\n");
		return ERROR;
	}

	//打开文件
	FILE* fd = fopen(str,"rb");
	//接收时间
	recvfrom(sockfd,buf,sizeof(buf),0,(SA)&addr,&len);
	long unsigned int time = 0;
	sscanf(buf,"%lu",&time);
	struct stat state = {}; 
	stat(str,&state);
	
	//如果文件存在 时间相同
	if(fd != NULL  && state.st_atime-time < 10)
	{
		printf("文件重传\n");
		strcpy(buf,"exis");
		sendto(sockfd,buf,strlen(buf)+1,0,(SA)&addr,len);
		//告知第几个字节开始
		fseek(fd,0,SEEK_END);
		long ret = ftell(fd);
		sprintf(buf,"%ld",ret);
		sendto(sockfd,buf,strlen(buf)+1,0,(SA)&addr,len);
		fd = fopen(str,"ab+");
	}
	else	//覆盖
	{
		printf("传输文件\n");
		strcpy(buf,"continue");
		sendto(sockfd,buf,strlen(buf)+1,0,(SA)&addr,len);
		fd = fopen(str,"wb+");
	}
	
	while(1)
	{
		char ch[1024] = {};
		recvfrom(sockfd,ch,sizeof(ch),0,(SA)&addr,&len);
		if(!strcmp(ch,"close"))
		{
			uint8_t encrypt[16] = {};
			getmd5_file(fd,encrypt);
			sendto(sockfd,encrypt,sizeof(encrypt),0,(SA)&addr,len);
			
			bzero(ch,sizeof(ch));
			recvfrom(sockfd,ch,sizeof(ch),0,(SA)&addr,&len);
			
			puts(ch);
			fclose(fd);
			return SUCCESS;
		}
		int ret = strlen(ch);
		fwrite(ch,1,ret,fd);
	}
	return SUCCESS;
}