ICode9
精准搜索请尝试： 精确搜索
首页
编程语言>
数据库
系统相关
互联网
其他分享
Python
Java
JavaScript
android
PHP
首页 > 系统相关> 文章详细

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

2022-01-09 11:31:05  阅读：145  来源： 互联网

标签：-- printf 文件传输 char int file Linux size md5


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

运行结果：
在这里插入图片描述
代码：

/*
	文件传输
客户端向服务器传送文件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

标签：--,printf,文件传输,char,int,file,Linux,size,md5
来源： https://blog.csdn.net/weixin_45640148/article/details/122391285

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

关于我们 | 联系我们 | 留言反馈

专注分享技术，共同学习，共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有