window/linux下获取文件MD5

MD5消息摘要算法(英语: MD5 Message-Digest Algorithm), 主要用于确保信息传输过程的一致性校验。

 

首先介绍两个工具：

window: WinMD5Free

Linux: md5sum

这两个工具的做用是验证本身写的程序获取MD5是否正确。

 

先上一段错误代码：

 1 string getFileMd5(const string& file)
 2 {
 3     MD5 md5;
 4     ifstream f(file.c_str(), ios_base::binrary);
 5     char buffer[64 * 1024];
 6     while (!f.eof())
 7     {
 8         f.read(buffer, sizeof(buffer));
 9         size_t length = strlen(buffer);
10         md5.update(buffer, length);
11     }
12     return md5.md5();
13 }

 

逻辑上，看似没有问题的一段代码。

实际验证的状况：

window下获取的MD5与第三方获取的值一致；

linux下获取的MD5与第三方获取的值不一致。

也就是说计算MD5时的输入不一致，致使的结果不一致。

 

问题分析：

上面的代码最有可能出问题的地方在用strlen获取buffer的大小。strlen以‘\0’标志肯定函数读取终止。从文件中读取的二进制数据中间是可能存在‘\0’的状况，打印每次strlen返回的结果，能够验证这个状况。

 

问题的缘由已经找到，那么如何解决这个问题呢？

若是你熟悉C++ IO操做，io库已经考虑过这个问题。获取每次读取的大小，使用fstream.gcount()返回正确读取大小。

 

正确的代码

 1 string getFileMd5(const string& file)
 2 {
 3     MD5 md5;
 4     ifstream f(file.c_str(), ios_base::binrary);
 5     char buffer[64 * 1024];
 6     while (!f.eof())
 7     {
 8         f.read(buffer, sizeof(buffer));
 9         md5.update(buffer, f.gcount());
10     }
11     return md5.md5();
12 }

 

工做中遇到这样的问题，经历过屡次尝试才发现这个问题，在此处记录下。