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 }

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

window/linux下获取文件MD5

猜你喜欢