myftp.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <signal.h>
#include <sys/time.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>	// "struct sockaddr_in"
#include <arpa/inet.h>	// "in_addr_t"
#include <dirent.h>
#include <malloc.h>
#include <pthread.h>
#include "myftp.h"
#include <ctype.h>
int verify_filename(char name_to_search[],char filename[]){
	int i=0,ul1=0,ul2=0;
	printf("name_to_search: %s, filename: %s\n", name_to_search,filename);
	char tmp[1024]="";
	for(i=0;i<strlen(filename);i++){
		// printf("char %c",filename[i] );
		if(filename[i]=='_'){
			if(ul1==0){
				ul1=1;
			}
			else{
				ul2=1;
				int j;
				for(j=i+1;j<strlen(filename);j++){
					tmp[j-i-1]=filename[j];
				}
				// strncpy(tmp,filename+i,strlen(filename)-i-1);
				// printf("strcmp(name_to_search,tmp) %d\n",strcmp(name_to_search,tmp) );
				return strcmp(name_to_search,tmp);
			}
		}
	}
	if(ul2==0){
		return 1;
	}
}
int ceiling(float x){
	if (x-(int)x>0){
			return (int)x+1;
	}
	else{
		return (int)x;
	}
}
void print_payload(unsigned char payload[PAYLEN],int len){
	int i;
	for(i=0;i<len;i++){
		printf("%d: %d\n",i,payload[i]);
	}
}
void construct_blockcode(char blockcode[1026],char filename[1024], int sid, int fd_code){
	char block_id[10];
	char tmp[2]="_";
	memset(block_id,0,sizeof(block_id));
	memset(blockcode,0,sizeof(blockcode));
	// memset(filename,0,sizeof(filename));
	snprintf(block_id,10,"%d",sid);

	strcat(blockcode,block_id);
	memset(block_id,0,sizeof(block_id));
	snprintf(block_id,10,"%d",fd_code);
	strcat(blockcode,tmp);
	strcat(blockcode,block_id);
	strcat(blockcode,tmp);
	strcat(blockcode,filename);

}
void tranp_file_data(int accept_fd, char filename[], char path[]){
	// printf("can i reach here\n");
	char filepath[100]="";
	strcat(filepath,path);
	strcat(filepath,filename);
		// printf("can i reach here\n");
	FILE * file=NULL;
	// printf("can i reach here\n");
	if((file=fopen(filepath,"rb"))<0){
		perror("fail to open the file\n");
	}
	// printf("can i reach here\n");
	unsigned int filelength=0;
	char c;
	c=fgetc(file);
	while(!feof(file)){
		filelength+=1;
		// printf("char is %X\n",c);
		c=fgetc(file);
	}
	struct message_s header;
	int len;
	char payload[PAYLEN];
	header.protocol[0]='m';
	header.protocol[1]='y';
	header.protocol[2]='f';
	header.protocol[3]='t';
	header.protocol[4]='p';
	header.type=0xFF;
	header.length=filelength+10;

	header.length = htonl(header.length);

	printf("file length is %d\n",filelength);
	fclose(file);

	if((file=fopen(filepath,"rb"))<0){
		perror("fail to open the file\n");
	}
	if((len=send(accept_fd,&header,sizeof(header),0))<0){
		perror("cannot send header to client\n");
	}
	 // char character[2]="";
	memset(payload,0,PAYLEN);
	int if_finish=0;
	unsigned int file_length=0;

	header.length = ntohl(header.length);

	int i;
	for(i=0;i<=(header.length-10-1)/(sizeof(payload)-1);i++){
		int j;
		for(j=0;j<sizeof(payload)-1;j++){
			// memset(character,0,sizeof(character));
			// character[0]=fgetc(file);
			payload[j]=fgetc(file);
			//printf("char: %c",payload[j]);
			if(!feof(file)){
				// printf("can i reach here,%d %d %s\n",j,strlen(payload), character);
				// printf("char is %X\n",character[0]);
				// strcat(payload,character);
				file_length+=1;
				// printf("can i reach there,\n");
			}
			else{
				if((len=send(accept_fd,payload,sizeof(payload)-1,0))<0){
					perror("can not send payload to client\n");
				}
				memset(payload,0,PAYLEN);
				if_finish=1;
				break;
			}
		}

		if(if_finish==0){
			while((len=send(accept_fd,payload,sizeof(payload)-1,0))<0){
				perror("can not send payload to client\n");

			}

		}

		memset(payload,0,PAYLEN);

	}
	printf("Sending finished\n");
	
	fclose(file);

}

void recv_file_data(int fd, char filename[], char path[]){
	int len;
	struct message_s header;
	char payload[PAYLEN];
	if((len=recv(fd,&header,sizeof(header),0))<0){
		perror("cannot recv the header from server\n");
	}

	header.length = ntohl(header.length);

	printf("length: %d\n",header.length);
	if(header.type!=0xFF){
		
		perror("type is wrong \n");

	}
	FILE * downfile=NULL;
	char filepath[100]="";
	strcat(filepath,path);
	strcat(filepath,filename);
	//printf("cat the filename\n");
	if((downfile=fopen(filepath,"wb"))==NULL){
		printf("open file: %s\n",filepath);
	}
	unsigned int file_length=0;
	int i;
	while(file_length<header.length-10){
		//printf("file_length: %d, true length: %d\n", file_length, header.length-10 );
		memset(payload,0, PAYLEN);
		//usleep(1);
		if((len=recv(fd, payload,sizeof(payload)-1,0))<0){
			printf("download file error!\n");
			// close(downfile);
		}
		if (file_length+len<header.length-10){
			if((fwrite(payload,len,1,downfile))<0){
				printf("wrong to write file!\n");
			}
			file_length+=len;
		}
		else{
			if((fwrite(payload,header.length-10-file_length,1,downfile))<0){
				printf("wrong to write file!\n");
			}
			file_length+=header.length-10-file_length;
		}
		// file_length+=len;
	}
	printf("Receiving finished\n");
	fflush(downfile);
	fclose(downfile);
}