Linux网络编程之文件传输2--带有MD5验证文件的完整性

2022-10-22 11:23:28

注意：这是在Linux中的，MD5的计算使用了oepnssl库，要注意你的Linux有没有，编译是要加库-lcrypto,

运行结果：
Linux网络编程之文件传输2--带有MD5验证文件的完整性
代码：

/*
	文件传输
客户端向服务器传送文件2.0

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

#define PORT 7777
#define IP "127.0.0.1"
typedef struct file_imformation{
	char file_name[120];
	unsigned long int file_length;
	unsigned char md5[16];
}FILE_IMFORMATION;
int file_transfer(int fd,char* path,unsigned long int file_size);
int file_imformation_transfer(int server_fd);
int get_file_md5(char *file_url,unsigned char md5 []);
int compare_str(char arr1[],char arr2[],int len);
int main(int argc,char* argv[]){
	
	int fd;
	struct sockaddr_in daddr;
	int err;
	//建立套接字
	fd = socket(AF_INET,SOCK_STREAM,0);
	if(fd < 0){
		printf("socket create fail!\n");
	}
	//初始化服务器地址
	memset(&daddr,0,sizeof(struct sockaddr));
	daddr.sin_family = AF_INET;
	daddr.sin_port = htons(PORT);
	daddr.sin_addr.s_addr = inet_addr(IP);
	//建立连接
	err = connect(fd,(struct sockaddr*)&daddr,sizeof(struct sockaddr));
	if(err < 0){
		printf("connect fail!\n");
	}
	file_imformation_transfer(fd);
	close(fd);
return 0;
}
//传输文件内容
int file_transfer(int fd,char* path,unsigned long int file_size){
	int size = 0;
	unsigned long sum = 0;
	int n = 0;
	char buffer[1024];
	memset(buffer,0,sizeof(buffer));
	int filefd = open(path,O_RDWR);
	if(filefd == -1){
		perror("file open fail");
	}
	while(1){
		size = read(filefd,buffer,sizeof(buffer));
		if(size == -1){
			printf("文件传输错误\n");
			close(filefd);
			return -1;
		}else if(size > 0){
			size = write(fd,buffer,size);
			sum += size;
			unsigned long int p = (sum * 100)/file_size;
			printf("\r文件已传输%ld\%%",p);
		}else{
			//当读取为0时，表示文件读取完毕
			printf("传输成功 %ld byte！\n",sum);
			close(filefd);
			return 0;
		}
	}
}
//在传输文件内容之前，先传送文件信息，比如文件名、文件大小、文件MD5值
int file_imformation_transfer(int server_fd){
	FILE_IMFORMATION file_im;
	struct stat sta;
	char path[200];
	
	memset(path,0,sizeof(path));
	memset(&file_im,0,sizeof(FILE_IMFORMATION));
	memset(&sta,0,sizeof(struct stat));
	printf("请输入文件路径和文件名：\n");
	scanf("%s",path);
	/*	
		获取文件名
		根据文件路径path确定文件名
		加入输入path为"/home/my/text.txt"那么i定位到最后一个反斜杠，然后将文件名复制到file_im.file_name中
	*/
	int i = strlen(path) - 1;			
	while(path[i] != '/' && i >= 0){
		i--;
	}
	i++;
	for(int j = 0;i < strlen(path)+1;j++){
		file_im.file_name[j] = path[i];
		i++;
	}
	printf("文件名为：%s\n",file_im.file_name);
	/*
		文件大小的获取
	*/
	if(stat(path,&sta) == -1){
		perror("获取文件信息错误");
		return -1;
	}
	file_im.file_length = sta.st_size;
	printf("文件大小为%ld\n",file_im.file_length);
	//获取文件MD5值
	get_file_md5(path,file_im.md5);
	printf("md5值为");
	for(int i = 0; i < 16;i++){
		printf("%02X",file_im.md5[i]);
	}
	printf("\n");
	//向服务器发送信息
	write(server_fd,&file_im,sizeof(FILE_IMFORMATION));
	//开始文件传输
	file_transfer(server_fd,path,sta.st_size);
	return 0;
}
//获取文件MD5值函数，file_url代表文件路径加文件名，根据其读取文件。char md5[]用于存放md5值
int get_file_md5(char *file_url,unsigned char md5 []){
	MD5_CTX ctx;
	char buffer[1024];
	int len=0;
	FILE * fp=NULL;
	memset(buffer,0,sizeof(buffer));
	//打开文件
	fp=fopen(file_url,"rb");
	if(fp==NULL)
	{
		printf("Can't open file\n");
		return 0;
	}
	//初始化
	MD5_Init(&ctx);
	//读取文件计算MD5值
	while((len=fread(buffer,1,1024,fp))>0)
	{
	MD5_Update(&ctx,buffer,len);
	memset(buffer,0,sizeof(buffer));
	}
	//获取MD5值
	MD5_Final(md5,&ctx);
	return 0;
}
//比较md5,两个字符数组中字符是否相等
int compare_str(char arr1[],char arr2[],int len){
	int i = len;
	for(;i < len;i++){
		if(arr1[i] == arr2[i]){
			continue;
		}else{
			return 0;
		}
		return 1;
	}
}


/*
	文件传输
服务器接收客户端传输文件2.0
*/
#include<stdio.h>
#include<unistd.h>
#include<fcntl.h>
#include<sys/stat.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<arpa/inet.h>
#include<stdlib.h>
#include<string.h>
#include<openssl/md5.h>

#define BACKLOG 3
#define PORT 7777
#define IP "127.0.0.1"
//默认接收文件路径
static char filepath[50] = "/home/ffj/fileTtest/";
typedef struct file_imformation{
	char file_name[120];
	unsigned long int file_length;
	unsigned char md5[16];
}FILE_IMFORMATION;
int file_access(int fd);
int file_imformation_accept(int fd,char buff[],unsigned long int *file_size,FILE_IMFORMATION *file_im);
int get_file_md5(char *file_url,unsigned char md5 []);
int compare_str(char arr1[],char arr2[],int len);
int main(int argc,char* argv[]){
	
	int fd;
	struct sockaddr_in daddr;
	char buff[1024];
	int err;
	
	//建立套接字
	fd = socket(AF_INET,SOCK_STREAM,0);
	if(fd < 0){
		printf("套接字建立失败!\n");
	}
	//初始化服务器地址
	memset(&daddr,0,sizeof(struct sockaddr));
	daddr.sin_family = AF_INET;
	daddr.sin_port = htons(PORT);
	daddr.sin_addr.s_addr = inet_addr(IP);
	//套接字绑定ip
	err = bind(fd,(struct sockaddr*)&daddr,sizeof(struct sockaddr));
	if(err < 0){
		printf("绑定失败\n");
	}
	//监听端口
	err = listen(fd,BACKLOG);
	if(err < 0){
		printf("监听错误！\n");
	}
	//接受连接	
	socklen_t len = sizeof(struct sockaddr);
	int acfd = accept(fd,(struct sockaddr*)&daddr,&len);
	file_access(acfd);
	close(acfd);		
	close(fd);
	return 0;
}
//接收文件内容
int file_access(int fd){
	unsigned long int size = -1;
	unsigned long int recv_size = 0;
	unsigned long int file_size;
	char buffer[1024];
	char file_url[200] = {0};
	FILE_IMFORMATION file_im;
	//创建并打开文件
	memset(file_url,0,sizeof(file_url));
	memset(buffer,0,sizeof(buffer));
	memset(&file_im,0,sizeof(FILE_IMFORMATION));
	//获取客户端发送信息
	file_imformation_accept(fd,file_url,&file_size,&file_im);
	//创建接收文件
	int filefd = open(file_url,O_RDWR|O_CREAT,0777);
	if(filefd == -1){
		perror("文件打开失败");
		return -1;
	}
	//从套接字中读取数据并写入文件
	while(size){
		size = read(fd,buffer,sizeof(buffer));
		if(size == -1){
			printf("文件传输错误\n");
			remove(file_url);
			close(filefd);
			return -1;
		}else if(size > 0){
			recv_size += size;
			write(filefd,buffer,size);
			memset(buffer,0,sizeof(buffer));
		}else{
				unsigned char file_md5 [16];
				memset(file_md5,0,sizeof(file_md5));
				get_file_md5(file_url,file_md5);
				if(recv_size != file_size){
					printf("错误：文件传输失败，文件大小不一致！\n");
					remove(file_url);
				}else if(compare_str(file_md5,file_im.md5,16) == 0){
					printf("错误：文件传输失败，文件MD5值不一致！\n");
					remove(file_url);
				}else{
					printf("接收文件成功！\n");
				}
				close(filefd);
				return 0;
			}
	
		}
	}
//在接受文件内容前，接受文件信息
int file_imformation_accept(int fd,char buff[],unsigned long int *file_size,FILE_IMFORMATION *file_im){
	int i = 0;
	while(i < strlen(filepath)){
		buff[i] = filepath[i];
		i++;
	}
	int len = read(fd,file_im,sizeof(FILE_IMFORMATION));
	if(len <= 0){
		printf("错误：接收文件信息为空");
	}
	printf("接收到文件信息中文件名称为%s\n",file_im->file_name);
	printf("接收到文件信息中文件大小为%ld\n",file_im->file_length);
	//获取文件大小
	*file_size = file_im->file_length;
	//获取文件名
	if(NULL == strcpy(&buff[i],file_im->file_name)){
		printf("错误：获取文件名失败\n");
	}
	printf("文件url是%s\n",buff);
	//获取文件md5值
	printf("接收到文件信息中md5值为");
	for(int i = 0; i < 16;i++){
		printf("%02X",file_im->md5[i]);
	}
	printf("\n");
	return 0;
}
int get_file_md5(char *file_url,unsigned char md5 []){
	MD5_CTX ctx;
	char buffer[1024];
	int len=0;
	FILE * fp=NULL;
	memset(buffer,0,sizeof(buffer));
	fp=fopen(file_url,"rb");
	if(fp==NULL)
	{
		printf("Can't open file\n");
		return 0;
	}
	MD5_Init(&ctx);
	while((len=fread(buffer,1,1024,fp))>0)
	{
	MD5_Update(&ctx,buffer,len);
	memset(buffer,0,sizeof(buffer));
	}
	MD5_Final(md5,&ctx);
	return 0;
}
//比较md5,两个字符数组中字符是否相等
int compare_str(char arr1[],char arr2[],int len){
	int i = len;
	for(;i < len;i++){
		if(arr1[i] == arr2[i]){
			continue;
		}else{
			return 0;
		}
		return 1;
	}
}

makefile

FLAGS=-lcrypto
server.out: fileTServer.c  client.out
	gcc -o $@ $< ${FLAGS}
client.out: fileT.c 
	gcc -o $@ $< ${FLAGS}
clean:
	rm *.out
上一篇：2021-2022-1 20211420《信息安全专业导论》XOR加密


下一篇：获取apk的MD5或SHA1签名信息