Files
MultiPar/source/par2j/md5_crc.c
2024-08-04 21:41:24 +09:00

1993 lines
65 KiB
C
Raw Blame History

This file contains ambiguous Unicode characters
This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.
// md5_crc.c
// Copyright : 2024-07-24 Yutaka Sawada
// License : GPL
#ifndef _UNICODE
#define _UNICODE
#endif
#ifndef UNICODE
#define UNICODE
#endif
#ifndef _WIN32_WINNT
#define _WIN32_WINNT 0x0600 // Windows Vista or later
#endif
#include <stdio.h>
#include <windows.h>
#include "common2.h"
#include "crc.h"
#include "phmd5.h"
#include "md5_crc.h"
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
// バイト配列の MD5 ハッシュ値を求める
void data_md5(
unsigned char *data_in, // ハッシュ値を求めるバイト配列
unsigned int data_len, // 入力バイト数
unsigned char *hash) // ハッシュ値 16バイト
{
PHMD5 ctx;
Phmd5Begin(&ctx);
Phmd5Process(&ctx, data_in, data_len);
Phmd5End(&ctx);
memcpy(hash, ctx.hash, 16);
}
// ブロックの MD5 ハッシュ値を求める
void data_md5_block(
unsigned char *data_in, // ハッシュ値を求めるバイト配列
unsigned int data_len, // 入力バイト数
unsigned char *hash) // ハッシュ値 16バイト
{
PHMD5 ctx;
Phmd5Begin(&ctx);
Phmd5Process(&ctx, data_in, data_len);
if (data_len < block_size) // ブロック・サイズまで 0で埋めて計算する
Phmd5ProcessZero(&ctx, block_size - data_len);
Phmd5End(&ctx);
memcpy(hash, ctx.hash, 16);
}
// ファイルの MD5-16k ハッシュ値を求める
int file_md5_16(
wchar_t *file_path, // ハッシュ値を求めるファイル
unsigned char *hash) // ファイルの先頭 16KB 分のハッシュ値 (16バイト)
{
unsigned char buf[16384];
unsigned int len;
PHMD5 hash_ctx;
HANDLE hFile;
// ソース・ファイルを開く
hFile = CreateFile(file_path, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL);
if (hFile == INVALID_HANDLE_VALUE)
return 1;
// 先頭 16KB のハッシュ値を計算する
Phmd5Begin(&hash_ctx); // 初期化
if (!ReadFile(hFile, buf, 16384, &len, NULL)){
CloseHandle(hFile);
return 2;
}
Phmd5Process(&hash_ctx, buf, len); // MD5 計算
Phmd5End(&hash_ctx); // 最終処理
memcpy(hash, hash_ctx.hash, 16);
CloseHandle(hFile);
return 0;
}
// ファイルの指定部分の MD5 ハッシュ値を求める
int file_md5(
HANDLE hFileRead, // ハッシュ値を求めるファイルのハンドル
__int64 offset,
unsigned int data_len, // 入力バイト数
unsigned char *hash) // ハッシュ値 (16バイト)
{
unsigned char buf[IO_SIZE];
unsigned int rv, len;
PHMD5 hash_ctx;
// ファイルの開始位置を offsetバイト目にする
if (!SetFilePointerEx(hFileRead, *((PLARGE_INTEGER)&offset), NULL, FILE_BEGIN))
return 1;
Phmd5Begin(&hash_ctx); // 初期化
while (data_len > 0){
len = IO_SIZE;
if (data_len < IO_SIZE)
len = data_len;
if (!ReadFile(hFileRead, buf, len, &rv, NULL) || (len != rv))
return 2;
data_len -= len;
Phmd5Process(&hash_ctx, buf, len); // MD5 計算
}
Phmd5End(&hash_ctx); // 最終処理
memcpy(hash, hash_ctx.hash, 16);
return 0;
}
/*------------------ 以下は MD5 と CRC-32 の組合せ関数 ------------------*/
// 内容が全て 0 のデータの MD5 ハッシュ値と CRC-32 を求める
void data_md5_crc32_zero(
unsigned char *hash) // ハッシュ値 (16 + 4バイト, MD5 + CRC-32)
{
PHMD5 ctx;
Phmd5Begin(&ctx);
Phmd5ProcessZero(&ctx, block_size);
Phmd5End(&ctx);
memcpy(hash, ctx.hash, 16);
memcpy(hash + 16, &window_mask, 4); // CRC-32 は window_mask と同じ
}
// 2GB 未満のファイルの開始位置以降のハッシュ値を計算する
unsigned int file_crc_part(HANDLE hFile)
{
unsigned char buf[4096];
unsigned int len, crc = 0xFFFFFFFF;
// 末尾まで読み込む
do {
if (!ReadFile(hFile, buf, 4096, &len, NULL) || (len == 0))
break;
crc = crc_update(crc, buf, len);
} while (len > 0);
return crc ^ 0xFFFFFFFF;
}
// ファイルの offset バイト目からブロック・サイズ分の MD5 と CRC-32 を求める
int file_md5_crc32_block(
HANDLE hFileRead, // MD5 と CRC を求めるファイルのハンドル
__int64 offset,
unsigned int avail_size, // 入力バイト数
unsigned char *hash) // ハッシュ値 (16 + 4バイト, MD5 + CRC-32)
{
unsigned char buf[IO_SIZE];
unsigned int rv, len, left, crc;
PHMD5 hash_ctx;
// ファイルの開始位置を offsetバイト目にする
if (!SetFilePointerEx(hFileRead, *((PLARGE_INTEGER)&offset), NULL, FILE_BEGIN))
return 1;
crc = 0xFFFFFFFF; // 初期化
Phmd5Begin(&hash_ctx);
left = avail_size; // 実際に読み込むサイズ
while (left > 0){
len = IO_SIZE;
if (left < IO_SIZE)
len = left;
if (!ReadFile(hFileRead, buf, len, &rv, NULL)){
print_win32_err(); // エラー通知
return 2;
} else if (len != rv){
return 3;
}
left -= len;
crc = crc_update(crc, buf, len); // CRC-32 計算
Phmd5Process(&hash_ctx, buf, len); // MD5 計算
}
if (avail_size < block_size){ // ブロック・サイズまで 0で埋めて計算する
crc = crc_update_zero(crc, block_size - avail_size);
Phmd5ProcessZero(&hash_ctx, block_size - avail_size);
}
crc ^= 0xFFFFFFFF; // 最終処理
memcpy(hash + 16, &crc, 4);
Phmd5End(&hash_ctx);
memcpy(hash, hash_ctx.hash, 16);
return 0;
}
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
//#define TIMER // 実験用
#ifdef TIMER
#include <time.h>
static double time_sec, time_speed;
static clock_t time_start, time1_start;
static clock_t time_total = 0, time2_total = 0, time3_total = 0;
#endif
#define MAX_BUF_SIZE 2097152 // ヒープ領域を使う場合の最大サイズ
// ファイルのハッシュ値と各スライスのチェックサムを同時に計算する
int file_hash_crc(
wchar_t *file_name, // ハッシュ値を求めるファイル
__int64 file_left,
unsigned char *hash, // (16)
unsigned char *sum, // チェックサム (MD5 + CRC-32) の配列
unsigned int *time_last, // 前回に経過表示した時刻
__int64 *prog_now) // 経過表示での現在位置
{
unsigned char *buf;
__declspec( align(64) ) unsigned char buf1[IO_SIZE * 2];
wchar_t file_path[MAX_LEN];
unsigned int err = 0, len, off, crc, block_left = 0, read_size;
__int64 file_off;
PHMD5 hash_ctx, block_ctx;
HANDLE hFile;
OVERLAPPED ol;
#ifdef TIMER
time1_start = clock();
#endif
// ソース・ファイルを開く
wcscpy(file_path, base_dir);
wcscpy(file_path + base_len, file_name);
hFile = CreateFile(file_path, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_FLAG_SEQUENTIAL_SCAN | FILE_FLAG_OVERLAPPED, NULL);
if (hFile == INVALID_HANDLE_VALUE){
print_win32_err();
return 1;
}
// 非同期ファイル・アクセスの準備をする
memset(&ol, 0, sizeof(OVERLAPPED));
ol.hEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
if (ol.hEvent == NULL){
err = 1;
goto error_end;
}
file_off = IO_SIZE;
buf = buf1 + IO_SIZE;
// 最初の分を読み込む
read_size = IO_SIZE;
if (file_left < IO_SIZE)
read_size = (unsigned int)file_left;
#ifdef TIMER
time_start = clock();
#endif
off = ReadFile(hFile, buf1, read_size, NULL, &ol);
#ifdef TIMER
time2_total += clock() - time_start;
#endif
if ((off == 0) && (GetLastError() != ERROR_IO_PENDING)){
print_win32_err();
err = 1;
goto error_end;
}
Phmd5Begin(&hash_ctx); // ファイルの MD5 計算を開始する
while (file_left > 0){
len = read_size;
file_left -= read_size;
(*prog_now) += read_size;
// 前回の読み込みが終わるのを待つ
WaitForSingleObject(ol.hEvent, INFINITE);
// 次の分を読み込み開始しておく
if (file_left > 0){
read_size = IO_SIZE;
if (file_left < IO_SIZE)
read_size = (unsigned int)file_left;
ol.Offset = (unsigned int)file_off;
ol.OffsetHigh = (unsigned int)(file_off >> 32);
file_off += IO_SIZE;
#ifdef TIMER
time_start = clock();
#endif
off = ReadFile(hFile, buf, read_size, NULL, &ol);
#ifdef TIMER
time2_total += clock() - time_start;
#endif
if ((off == 0) && (GetLastError() != ERROR_IO_PENDING)){
print_win32_err();
err = 1;
goto error_end;
}
}
// バッファーを入れ替える
if (buf == buf1){
buf = buf1 + IO_SIZE;
} else {
buf = buf1;
}
#ifdef TIMER
time_start = clock();
#endif
off = 0; // チェックサム計算
if (block_left > 0){ // 前回足りなかった分を追加する
//printf("file_left = %I64d, block_left = %d\n", file_left, block_left);
if (block_left <= len){
crc = crc_update(crc, buf, block_left) ^ 0xFFFFFFFF; // CRC-32 計算
Phmd5Process2(&hash_ctx, &block_ctx, buf, block_left); // MD5 計算
Phmd5End(&block_ctx); // 最終処理
memcpy(sum, block_ctx.hash, 16);
memcpy(sum + 16, &crc, 4);
sum += 20;
off += block_left;
block_left = 0;
} else {
crc = crc_update(crc, buf, len); // CRC-32 計算
Phmd5Process2(&hash_ctx, &block_ctx, buf, len); // MD5 計算
off = len;
block_left -= len;
}
}
for (; off < len; off += block_size){
Phmd5Begin(&block_ctx);
if (off + block_size <= len){
crc = crc_update(0xFFFFFFFF, buf + off, block_size) ^ 0xFFFFFFFF; // CRC-32 計算
Phmd5Process2(&hash_ctx, &block_ctx, buf + off, block_size); // MD5 計算
Phmd5End(&block_ctx); // 最終処理
memcpy(sum, block_ctx.hash, 16);
memcpy(sum + 16, &crc, 4);
sum += 20;
} else { // スライスが途中までなら
crc = crc_update(0xFFFFFFFF, buf + off, len - off); // CRC-32 計算
Phmd5Process2(&hash_ctx, &block_ctx, buf + off, len - off); // MD5 計算
block_left = block_size - len + off;
}
}
#ifdef TIMER
time3_total += clock() - time_start;
#endif
// 経過表示
if (GetTickCount() - (*time_last) >= UPDATE_TIME){
if (print_progress((int)(((*prog_now) * 1000) / total_file_size))){
err = 2;
goto error_end;
}
(*time_last) = GetTickCount();
}
}
// 最終ブロックが半端なら
if (block_left > 0){ // 残りを 0 でパディングする (PAR2 仕様の欠点)
crc = crc_update_zero(crc, block_left) ^ 0xFFFFFFFF; // CRC-32 計算
Phmd5ProcessZero(&block_ctx, block_left);
Phmd5End(&block_ctx);
memcpy(sum, block_ctx.hash, 16);
memcpy(sum + 16, &crc, 4);
}
Phmd5End(&hash_ctx); // 最終処理
memcpy(hash, hash_ctx.hash, 16);
error_end:
CancelIo(hFile); // 非同期 IO を取り消す
CloseHandle(hFile);
if (ol.hEvent)
CloseHandle(ol.hEvent);
#ifdef TIMER
time_total += clock() - time1_start;
if (*prog_now == total_file_size){
printf("\nread %.3f sec\n", (double)time2_total / CLOCKS_PER_SEC);
printf("main %.3f sec\n", (double)time3_total / CLOCKS_PER_SEC);
time_sec = (double)time_total / CLOCKS_PER_SEC;
if (time_sec > 0){
time_speed = (double)total_file_size / (time_sec * 1048576);
} else {
time_speed = 0;
}
printf("total %.3f sec, %.0f MB/s\n", time_sec, time_speed);
}
#endif
return err;
}
/* USBが切断されても再接続を待つ実験
int file_hash_crc(
wchar_t *file_name, // ハッシュ値を求めるファイル
__int64 file_left,
unsigned char *hash, // ハッシュ値 (16バイト)
unsigned char *sum, // チェックサム (MD5 + CRC-32) の配列
unsigned int *time_last, // 前回に経過表示した時刻
__int64 *prog_now) // 経過表示での現在位置
{
unsigned char *buf, *buf2;
__declspec( align(64) ) unsigned char buf1[IO_SIZE * 2];
wchar_t file_path[MAX_LEN];
unsigned int err = 0, err_last = 0, err_count = 0;
unsigned int len, off, crc, block_left = 0, read_size;
__int64 file_off;
PHMD5 hash_ctx, block_ctx;
HANDLE hFile;
OVERLAPPED ol;
#ifdef TIMER
time1_start = clock();
#endif
// ソース・ファイルを開く
wcscpy(file_path, base_dir);
wcscpy(file_path + base_len, file_name);
//printf_cp("\nfile = %s\n", file_path);
error_retry_read:
hFile = CreateFile(file_path, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_FLAG_SEQUENTIAL_SCAN | FILE_FLAG_OVERLAPPED, NULL);
if (hFile == INVALID_HANDLE_VALUE){
err = GetLastError();
if ((err_count <= 30) && ((err == ERROR_PATH_NOT_FOUND) || (err == ERROR_NOT_READY))){ // 3 or 21
if (error_progress(err, err_last) == 2)
return 2;
err_last = err;
err_count++;
err = 0;
goto error_retry_read; // ファイルを開きなおす
} else {
print_win32_err();
return 1;
}
}
// 非同期ファイル・アクセスの準備をする
memset(&ol, 0, sizeof(OVERLAPPED));
ol.hEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
if (ol.hEvent == NULL){
err = 1;
goto error_end;
}
file_off = IO_SIZE;
buf = buf1 + IO_SIZE;
// 最初の分を読み込む
read_size = IO_SIZE;
if (file_left < IO_SIZE)
read_size = (unsigned int)file_left;
#ifdef TIMER
time_start = clock();
#endif
off = ReadFile(hFile, buf1, read_size, NULL, &ol);
#ifdef TIMER
time2_total += clock() - time_start;
#endif
if ((off == 0) && (GetLastError() != ERROR_IO_PENDING)){
print_win32_err();
err = 1;
goto error_end;
}
Phmd5Begin(&hash_ctx); // ファイルの MD5 計算を開始する
while (file_left > 0){
len = read_size;
file_left -= read_size;
(*prog_now) += read_size;
// 前回の読み込みが終わるのを待つ
// WaitForSingleObject(ol.hEvent, INFINITE);
//Sleep(200);
error_retry_wait:
// 前回の読み込みが終わるのを待つ、エラーも取得できる
if (GetOverlappedResult(hFile, &ol, &off, TRUE) == 0){
err = GetLastError();
if ((err_count <= 30) && (err == ERROR_NOT_READY)){ // 21, The device is not ready.
CloseHandle(hFile); // 一度ファイルを閉じる
file_off -= IO_SIZE; // 前回の読み取り位置に戻す
ol.Offset = (unsigned int)file_off;
ol.OffsetHigh = (unsigned int)(file_off >> 32);
error_retry_pause:
if (error_progress(err, err_last) == 2){
err = 2;
goto error_end;
}
err_last = err;
err_count++;
err = 0;
// ファイルを開きなおす
hFile = CreateFile(file_path, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_FLAG_SEQUENTIAL_SCAN | FILE_FLAG_OVERLAPPED, NULL);
if (hFile == INVALID_HANDLE_VALUE){
err = GetLastError();
if ((err_count <= 30) && ((err == ERROR_NOT_READY) || (err == ERROR_PATH_NOT_FOUND))) // 21 or 3
goto error_retry_pause;
print_win32_err();
err = 1;
goto error_end;
}
// 再度読み込む
if (buf == buf1){
buf2 = buf1 + IO_SIZE;
} else {
buf2 = buf1;
}
off = ReadFile(hFile, buf2, read_size, NULL, &ol);
if (off == 0){
err = GetLastError();
if (err != ERROR_IO_PENDING){
if ((err_count <= 30) && (err == ERROR_NOT_READY)){
CancelIo(hFile); // 非同期 IO を取り消す
CloseHandle(hFile); // 一度ファイルを閉じる
goto error_retry_pause;
}
print_win32_err();
err = 1;
goto error_end;
}
err = 0;
}
file_off += IO_SIZE; // 読み取り位置を次の位置にする
goto error_retry_wait;
} else { // その他のエラー
print_win32_err();
printf("GetOverlappedResult: file_off = %I64d, file_left = %I64d\n", file_off, file_left);
err = 1;
goto error_end;
}
}
err_last = 0;
err_count = 0;
// 次の分を読み込み開始しておく
if (file_left > 0){
read_size = IO_SIZE;
if (file_left < IO_SIZE)
read_size = (unsigned int)file_left;
ol.Offset = (unsigned int)file_off;
ol.OffsetHigh = (unsigned int)(file_off >> 32);
file_off += IO_SIZE;
#ifdef TIMER
time_start = clock();
#endif
off = ReadFile(hFile, buf, read_size, NULL, &ol);
#ifdef TIMER
time2_total += clock() - time_start;
#endif
if ((off == 0) && (GetLastError() != ERROR_IO_PENDING)){
print_win32_err();
printf("ReadFile: file_off = %I64d, file_left = %I64d\n", file_off, file_left);
err = 1;
goto error_end;
}
}
// バッファーを入れ替える
if (buf == buf1){
buf = buf1 + IO_SIZE;
} else {
buf = buf1;
}
#ifdef TIMER
time_start = clock();
#endif
off = 0; // チェックサム計算
if (block_left > 0){ // 前回足りなかった分を追加する
//printf("file_left = %I64d, block_left = %d\n", file_left, block_left);
if (block_left <= len){
crc = crc_update(crc, buf, block_left) ^ 0xFFFFFFFF; // CRC-32 計算
Phmd5Process2(&hash_ctx, &block_ctx, buf, block_left); // MD5 計算
Phmd5End(&block_ctx); // 最終処理
memcpy(sum, block_ctx.hash, 16);
memcpy(sum + 16, &crc, 4);
sum += 20;
off += block_left;
block_left = 0;
} else {
crc = crc_update(crc, buf, len); // CRC-32 計算
Phmd5Process2(&hash_ctx, &block_ctx, buf, len); // MD5 計算
off = len;
block_left -= len;
}
}
for (; off < len; off += block_size){
Phmd5Begin(&block_ctx);
if (off + block_size <= len){
crc = crc_update(0xFFFFFFFF, buf + off, block_size) ^ 0xFFFFFFFF; // CRC-32 計算
Phmd5Process2(&hash_ctx, &block_ctx, buf + off, block_size); // MD5 計算
Phmd5End(&block_ctx); // 最終処理
memcpy(sum, block_ctx.hash, 16);
memcpy(sum + 16, &crc, 4);
sum += 20;
} else { // スライスが途中までなら
crc = crc_update(0xFFFFFFFF, buf + off, len - off); // CRC-32 計算
Phmd5Process2(&hash_ctx, &block_ctx, buf + off, len - off); // MD5 計算
block_left = block_size - len + off;
}
}
#ifdef TIMER
time3_total += clock() - time_start;
#endif
// 経過表示
if (GetTickCount() - (*time_last) >= UPDATE_TIME){
if (print_progress((int)(((*prog_now) * 1000) / total_file_size))){
err = 2;
goto error_end;
}
(*time_last) = GetTickCount();
}
}
// 最終ブロックが半端なら
if (block_left > 0){ // 残りを 0 でパディングする (PAR2 仕様の欠点)
crc = crc_update_zero(crc, block_left) ^ 0xFFFFFFFF; // CRC-32 計算
Phmd5ProcessZero(&block_ctx, block_left);
Phmd5End(&block_ctx);
memcpy(sum, block_ctx.hash, 16);
memcpy(sum + 16, &crc, 4);
}
Phmd5End(&hash_ctx); // 最終処理
memcpy(hash, hash_ctx.hash, 16);
error_end:
CancelIo(hFile); // 非同期 IO を取り消す
CloseHandle(hFile);
if (ol.hEvent)
CloseHandle(ol.hEvent);
#ifdef TIMER
time_total += clock() - time1_start;
if (*prog_now == total_file_size){
printf("\nread %.3f sec\n", (double)time2_total / CLOCKS_PER_SEC);
printf("main %.3f sec\n", (double)time3_total / CLOCKS_PER_SEC);
time_sec = (double)time_total / CLOCKS_PER_SEC;
if (time_sec > 0){
time_speed = (double)total_file_size / (time_sec * 1048576);
} else {
time_speed = 0;
}
printf("total %.3f sec, %.0f MB/s\n", time_sec, time_speed);
}
#endif
return err;
}
*/
/*
// ヒープ領域を使うバージョン
// ファイルのハッシュ値と各スライスのチェックサムを同時に計算する
int file_hash_crc(
wchar_t *file_name, // ハッシュ値を求めるファイル
__int64 file_left,
unsigned char *hash, // ハッシュ値 (16バイト)
unsigned char *sum, // チェックサム (MD5 + CRC-32) の配列
unsigned int *time_last, // 前回に経過表示した時刻
__int64 *prog_now) // 経過表示での現在位置
{
unsigned char *buf, *buf1;
wchar_t file_path[MAX_LEN];
unsigned int err = 0, len, off, crc, block_left = 0, read_size, io_size;
__int64 file_off;
PHMD5 hash_ctx, block_ctx;
HANDLE hFile;
OVERLAPPED ol;
#ifdef TIMER
time1_start = clock();
#endif
// ソース・ファイルを開く
wcscpy(file_path, base_dir);
wcscpy(file_path + base_len, file_name);
hFile = CreateFile(file_path, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_FLAG_SEQUENTIAL_SCAN | FILE_FLAG_OVERLAPPED, NULL);
if (hFile == INVALID_HANDLE_VALUE){
print_win32_err();
return 1;
}
// バッファー・サイズが大きいのでヒープ領域を使う
for (io_size = IO_SIZE; io_size <= MAX_BUF_SIZE; io_size += IO_SIZE){ // IO_SIZE の倍数にする
if ((io_size + IO_SIZE > (cpu_cache & 0xFFFF0000)) || ((__int64)(io_size + IO_SIZE) * 4 > file_left))
break;
}
buf1 = _aligned_malloc(io_size * 2, 64);
if (buf1 == NULL){
printf("malloc, %d\n", io_size * 2);
CloseHandle(hFile);
return 1;
}
buf = buf1 + io_size;
// 非同期ファイル・アクセスの準備をする
memset(&ol, 0, sizeof(OVERLAPPED));
ol.hEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
if (ol.hEvent == NULL){
err = 1;
goto error_end;
}
file_off = io_size;
// 最初の分を読み込む
read_size = io_size;
if (file_left < io_size)
read_size = (unsigned int)file_left;
#ifdef TIMER
time_start = clock();
#endif
off = ReadFile(hFile, buf1, read_size, NULL, &ol);
#ifdef TIMER
time2_total += clock() - time_start;
#endif
if ((off == 0) && (GetLastError() != ERROR_IO_PENDING)){
print_win32_err();
err = 1;
goto error_end;
}
Phmd5Begin(&hash_ctx); // ファイルの MD5 計算を開始する
while (file_left > 0){
len = read_size;
file_left -= read_size;
(*prog_now) += read_size;
// 前回の読み込みが終わるのを待つ
WaitForSingleObject(ol.hEvent, INFINITE);
// 次の分を読み込み開始しておく
if (file_left > 0){
read_size = io_size;
if (file_left < io_size)
read_size = (unsigned int)file_left;
ol.Offset = (unsigned int)file_off;
ol.OffsetHigh = (unsigned int)(file_off >> 32);
file_off += io_size;
#ifdef TIMER
time_start = clock();
#endif
off = ReadFile(hFile, buf, read_size, NULL, &ol);
#ifdef TIMER
time2_total += clock() - time_start;
#endif
if ((off == 0) && (GetLastError() != ERROR_IO_PENDING)){
print_win32_err();
err = 1;
goto error_end;
}
}
// バッファーを入れ替える
if (buf == buf1){
buf = buf1 + io_size;
} else {
buf = buf1;
}
#ifdef TIMER
time_start = clock();
#endif
off = 0; // チェックサム計算
if (block_left > 0){ // 前回足りなかった分を追加する
//printf("file_left = %I64d, block_left = %d\n", file_left, block_left);
if (block_left <= len){
crc = crc_update(crc, buf, block_left) ^ 0xFFFFFFFF; // CRC-32 計算
Phmd5Process2(&hash_ctx, &block_ctx, buf, block_left); // MD5 計算
Phmd5End(&block_ctx); // 最終処理
memcpy(sum, block_ctx.hash, 16);
memcpy(sum + 16, &crc, 4);
sum += 20;
off += block_left;
block_left = 0;
} else {
crc = crc_update(crc, buf, len); // CRC-32 計算
Phmd5Process2(&hash_ctx, &block_ctx, buf, len); // MD5 計算
off = len;
block_left -= len;
}
}
for (; off < len; off += block_size){
Phmd5Begin(&block_ctx);
if (off + block_size <= len){
crc = crc_update(0xFFFFFFFF, buf + off, block_size) ^ 0xFFFFFFFF; // CRC-32 計算
Phmd5Process2(&hash_ctx, &block_ctx, buf + off, block_size); // MD5 計算
Phmd5End(&block_ctx); // 最終処理
memcpy(sum, block_ctx.hash, 16);
memcpy(sum + 16, &crc, 4);
sum += 20;
} else { // スライスが途中までなら
crc = crc_update(0xFFFFFFFF, buf + off, len - off); // CRC-32 計算
Phmd5Process2(&hash_ctx, &block_ctx, buf + off, len - off); // MD5 計算
block_left = block_size - len + off;
}
}
#ifdef TIMER
time3_total += clock() - time_start;
#endif
// 経過表示
if (GetTickCount() - (*time_last) >= UPDATE_TIME){
if (print_progress((int)(((*prog_now) * 1000) / total_file_size))){
err = 2;
goto error_end;
}
(*time_last) = GetTickCount();
}
}
// 最終ブロックが半端なら
if (block_left > 0){ // 残りを 0 でパディングする (PAR2 仕様の欠点)
crc = crc_update_zero(crc, block_left) ^ 0xFFFFFFFF; // CRC-32 計算
Phmd5ProcessZero(&block_ctx, block_left);
Phmd5End(&block_ctx);
memcpy(sum, block_ctx.hash, 16);
memcpy(sum + 16, &crc, 4);
}
Phmd5End(&hash_ctx); // 最終処理
memcpy(hash, hash_ctx.hash, 16);
error_end:
CancelIo(hFile); // 非同期 IO を取り消す
CloseHandle(hFile);
if (ol.hEvent)
CloseHandle(ol.hEvent);
if (buf1)
_aligned_free(buf1);
#ifdef TIMER
time_total += clock() - time1_start;
if (*prog_now == total_file_size){
printf("\nread %.3f sec\n", (double)time2_total / CLOCKS_PER_SEC);
printf("main %.3f sec\n", (double)time3_total / CLOCKS_PER_SEC);
time_sec = (double)time_total / CLOCKS_PER_SEC;
if (time_sec > 0){
time_speed = (double)total_file_size / (time_sec * 1048576);
} else {
time_speed = 0;
}
printf("total %.3f sec, %.0f MB/s\n", time_sec, time_speed);
}
#endif
return err;
}
*/
// 複数ファイルのハッシュ値を同時に計算する
DWORD WINAPI file_hash_crc2(LPVOID lpParameter)
{
unsigned char *buf, *buf1, *hash, *sum;
wchar_t file_path[MAX_LEN];
int prog_loop, prog_tick, prog_rv;
unsigned int err = 0, len, off, crc, block_left = 0, read_size, io_size;
unsigned int time_last;
__int64 file_left, file_off;
PHMD5 hash_ctx, block_ctx;
HANDLE hFile;
OVERLAPPED ol;
FILE_HASH_TH *file_th;
file_th = (FILE_HASH_TH *)lpParameter;
wcscpy(file_path, base_dir);
wcscpy(file_path + base_len, file_th->file_name);
file_left = file_th->file_size;
hash = file_th->hash; // ハッシュ値 (16バイト)
sum = file_th->sum; // チェックサム (MD5 + CRC-32) の配列
prog_loop = 0;
// ソース・ファイルを開く
hFile = CreateFile(file_path, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_FLAG_SEQUENTIAL_SCAN | FILE_FLAG_OVERLAPPED, NULL);
// アクセス・モードで違いが出るかも?
//hFile = CreateFile(file_path, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_FLAG_RANDOM_ACCESS | FILE_FLAG_OVERLAPPED, NULL);
if (hFile == INVALID_HANDLE_VALUE){
print_win32_err();
return 1;
}
// バッファー・サイズが大きいのでヒープ領域を使う
prog_tick = 1;
for (io_size = IO_SIZE; io_size <= MAX_BUF_SIZE; io_size += IO_SIZE){ // IO_SIZE の倍数にする
if ((io_size + IO_SIZE > (cpu_cache & 0xFFFF0000)) || ((__int64)(io_size + IO_SIZE) * 4 > file_left))
break;
prog_tick++;
}
//printf("\n io_size = %d, prog_tick = %d\n", io_size, prog_tick);
buf1 = _aligned_malloc(io_size * 2, 64);
if (buf1 == NULL){
printf("malloc, %d\n", io_size * 2);
err = 1;
goto error_end;
}
buf = buf1 + io_size;
// 非同期ファイル・アクセスの準備をする
memset(&ol, 0, sizeof(OVERLAPPED));
ol.hEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
if (ol.hEvent == NULL){
err = 1;
goto error_end;
}
file_off = io_size;
// 最初の分を読み込む
read_size = io_size;
if (file_left < io_size)
read_size = (unsigned int)file_left;
off = ReadFile(hFile, buf1, read_size, NULL, &ol);
if ((off == 0) && (GetLastError() != ERROR_IO_PENDING)){
print_win32_err();
err = 1;
goto error_end;
}
Phmd5Begin(&hash_ctx); // ファイルの MD5 計算を開始する
time_last = GetTickCount();
while (file_left > 0){
len = read_size;
file_left -= read_size;
// 前回の読み込みが終わるのを待つ
WaitForSingleObject(ol.hEvent, INFINITE);
// 次の分を読み込み開始しておく
if (file_left > 0){
read_size = io_size;
if (file_left < io_size)
read_size = (unsigned int)file_left;
ol.Offset = (unsigned int)file_off;
ol.OffsetHigh = (unsigned int)(file_off >> 32);
file_off += io_size;
off = ReadFile(hFile, buf, read_size, NULL, &ol);
if ((off == 0) && (GetLastError() != ERROR_IO_PENDING)){
print_win32_err();
err = 1;
goto error_end;
}
}
// バッファーを入れ替える
if (buf == buf1){
buf = buf1 + io_size;
} else {
buf = buf1;
}
off = 0; // チェックサム計算
if (block_left > 0){ // 前回足りなかった分を追加する
//printf("file_left = %I64d, block_left = %d\n", file_left, block_left);
if (block_left <= len){
crc = crc_update(crc, buf, block_left) ^ 0xFFFFFFFF; // CRC-32 計算
//Phmd5Process(&block_ctx, buf, block_left);
Phmd5Process2(&hash_ctx, &block_ctx, buf, block_left); // MD5 計算
Phmd5End(&block_ctx); // 最終処理
memcpy(sum, block_ctx.hash, 16);
memcpy(sum + 16, &crc, 4);
sum += 20;
off += block_left;
block_left = 0;
} else {
crc = crc_update(crc, buf, len); // CRC-32 計算
//Phmd5Process(&block_ctx, buf, len);
Phmd5Process2(&hash_ctx, &block_ctx, buf, len); // MD5 計算
off = len;
block_left -= len;
}
}
for (; off < len; off += block_size){
Phmd5Begin(&block_ctx);
if (off + block_size <= len){
crc = crc_update(0xFFFFFFFF, buf + off, block_size) ^ 0xFFFFFFFF; // CRC-32 計算
//Phmd5Process(&block_ctx, buf + off, block_size);
Phmd5Process2(&hash_ctx, &block_ctx, buf + off, block_size); // MD5 計算
Phmd5End(&block_ctx); // 最終処理
memcpy(sum, block_ctx.hash, 16);
memcpy(sum + 16, &crc, 4);
sum += 20;
} else { // スライスが途中までなら
crc = crc_update(0xFFFFFFFF, buf + off, len - off); // CRC-32 計算
//Phmd5Process(&block_ctx, buf + off, len - off);
Phmd5Process2(&hash_ctx, &block_ctx, buf + off, len - off); // MD5 計算
block_left = block_size - len + off;
}
}
// 経過表示のために進捗状況を更新する
if (GetTickCount() - time_last >= UPDATE_TIME / 2){
prog_rv = InterlockedExchange(&(file_th->loop), prog_loop);
if (prog_rv == -1){
err = 2;
goto error_end;
}
time_last = GetTickCount();
}
prog_loop += prog_tick;
}
// 最終ブロックが半端なら
if (block_left > 0){ // 残りを 0 でパディングする (PAR2 仕様の欠点)
crc = crc_update_zero(crc, block_left) ^ 0xFFFFFFFF; // CRC-32 計算
Phmd5ProcessZero(&block_ctx, block_left);
Phmd5End(&block_ctx);
memcpy(sum, block_ctx.hash, 16);
memcpy(sum + 16, &crc, 4);
sum += 20;
}
Phmd5End(&hash_ctx); // 最終処理
memcpy(hash, hash_ctx.hash, 16);
// サブ・スレッド側でパケット2個を完成させる
data_md5(hash - 48, (int)(file_th->sum - hash) - 32, hash - 64); // File Description packet の MD5
data_md5(file_th->sum - 48, 32 + (int)(sum - file_th->sum) + 16, file_th->sum - 64); // Input File Slice Checksum packet の MD5
// ファイルの MD5-16k はもう不要なので、ブロックの CRC-32 に変更する
len = (int)((file_th->file_size + (__int64)block_size - 1) / block_size); // ブロック数
memset(file_th->crc, 0, 16);
for (off = 0; off < len; off++){ // XOR して 16バイトに減らす
memcpy(&crc, file_th->sum + (20 * off + 16), 4);
(file_th->crc)[off & 3] ^= crc;
}
error_end:
CancelIo(hFile); // 非同期 IO を取り消す
CloseHandle(hFile);
if (ol.hEvent)
CloseHandle(ol.hEvent);
if (buf1)
_aligned_free(buf1);
return err;
}
// ファイルのハッシュ値が同じか調べる (全てのスライスのチェックサムも)
// 0=破損してるが何個目まで同じ, -1=MD5-16kが異なる, -2=キャンセル, -3=同じ(完全か追加)
int file_hash_check(
int num, // file_ctx におけるファイル番号
wchar_t *file_name, // 表示するファイル名
HANDLE hFile, // ファイルのハンドル
int prog_min, // (INT_MAX )
file_ctx_r *files, // 各ソース・ファイルの情報
source_ctx_r *s_blk) // 各ソース・ブロックの情報 (NULL なら比較しない)
{
unsigned char *buf;
__declspec( align(64) ) unsigned char buf1[IO_SIZE * 2];
int find_next, comp_num = 0;
unsigned int len, off, crc, block_left = 0, read_size;
unsigned int time_last;
__int64 file_size, file_left, file_off;
PHMD5 hash_ctx, block_ctx;
OVERLAPPED ol;
#ifdef TIMER
time1_start = clock();
#endif
prog_last = -1; // 検証中のファイル名を毎回表示する
time_last = GetTickCount();
file_size = files[num].size; // 本来のファイル・サイズ
find_next = files[num].b_off; // 先頭ブロックの番号
// 非同期ファイル・アクセスの準備をする
memset(&ol, 0, sizeof(OVERLAPPED));
ol.hEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
if (ol.hEvent == NULL)
return -1;
buf = buf1 + IO_SIZE;
// まずは先頭 16KB のハッシュ値を比較する
len = 16384;
if (file_size < 16384){
len = (unsigned int)file_size;
file_left = 0;
} else {
file_left = file_size - 16384; // 本来のファイル・サイズまでしか検査しない
}
#ifdef TIMER
time_start = clock();
#endif
off = ReadFile(hFile, buf, len, NULL, &ol);
#ifdef TIMER
time2_total += clock() - time_start;
#endif
if ((off == 0) && (GetLastError() != ERROR_IO_PENDING)){
print_win32_err();
comp_num = -1;
goto error_end;
}
WaitForSingleObject(ol.hEvent, INFINITE);
Phmd5Begin(&hash_ctx); // 初期化
Phmd5Process(&hash_ctx, buf, len); // MD5 計算
Phmd5End(&hash_ctx); // 最終処理
if (file_left == 0){ // ファイルは 16KB 以下
if (s_blk != NULL){ // 全てのスライスを比較する
for (off = 0; off < len; off += block_size){
Phmd5Begin(&block_ctx);
if (off + block_size <= len){
crc = crc_update(0, buf + off, block_size);
Phmd5Process(&block_ctx, buf + off, block_size);
} else { // 末尾の半端なスライスなら
crc = crc_update(0, buf + off, len - off);
Phmd5Process(&block_ctx, buf + off, len - off);
block_left = block_size - len + off; // ブロック・サイズまで 0で埋めて計算する
crc = crc_update_zero(crc, block_left);
Phmd5ProcessZero(&block_ctx, block_left);
}
Phmd5End(&block_ctx);
if ((crc != s_blk[find_next].crc) || (memcmp(block_ctx.hash, s_blk[find_next].hash, 16) != 0))
goto error_end; // スライスの CRC-32 か MD5 が異なる
comp_num++;
find_next++;
}
}
if (memcmp(hash_ctx.hash, files[num].hash + 16, 16) != 0){ // 16k MD5 が異なる
comp_num = -1;
goto error_end;
}
comp_num = -3; // 同じ
goto error_end;
}
// 16KB よりも大きくて 16k MD5 が異なる場合は、スライス単位の検査をしない
if (memcmp(hash_ctx.hash, files[num].hash + 16, 16) != 0){
comp_num = -1;
goto error_end;
}
// 16k MD5 が一致したら残りの部分も計算する
Phmd5Begin(&hash_ctx); // 初期化
Phmd5Process(&hash_ctx, buf, 16384); // MD5 計算
if (s_blk != NULL){
for (off = 0; off < 16384; off += block_size){
Phmd5Begin(&block_ctx);
if (off + block_size <= 16384){
crc = crc_update(0, buf + off, block_size);
Phmd5Process(&block_ctx, buf + off, block_size);
Phmd5End(&block_ctx);
if ((crc != s_blk[find_next].crc) || (memcmp(block_ctx.hash, s_blk[find_next].hash, 16) != 0))
goto error_end; // スライスの CRC-32 か MD5 が異なる
comp_num++;
find_next++;
} else { // スライスが途中までなら
crc = crc_update(0, buf + off, 16384 - off);
Phmd5Process(&block_ctx, buf + off, 16384 - off);
block_left = block_size - 16384 + off;
}
}
}
// 次の分を読み込む
ol.Offset = 16384;
file_off = 16384 + IO_SIZE;
read_size = IO_SIZE;
if (file_left < IO_SIZE)
read_size = (unsigned int)file_left;
#ifdef TIMER
time_start = clock();
#endif
off = ReadFile(hFile, buf1, read_size, NULL, &ol);
#ifdef TIMER
time2_total += clock() - time_start;
#endif
if ((off == 0) && (GetLastError() != ERROR_IO_PENDING)){
print_win32_err();
goto error_end; // 読み取りが失敗した所で終わる
}
while (file_left > 0){
len = read_size; // 読み込んだサイズ
file_left -= read_size;
// 前回の読み込みが終わるのを待つ
WaitForSingleObject(ol.hEvent, INFINITE);
// 次の分を読み込み開始しておく
if (file_left > 0){
read_size = IO_SIZE;
if (file_left < IO_SIZE)
read_size = (unsigned int)file_left;
ol.Offset = (unsigned int)file_off;
ol.OffsetHigh = (unsigned int)(file_off >> 32);
file_off += IO_SIZE;
#ifdef TIMER
time_start = clock();
#endif
off = ReadFile(hFile, buf, read_size, NULL, &ol);
#ifdef TIMER
time2_total += clock() - time_start;
#endif
if ((off == 0) && (GetLastError() != ERROR_IO_PENDING)){
print_win32_err();
goto error_end; // 読み取りが失敗した所で終わる
}
}
// バッファーを入れ替える
if (buf == buf1){
buf = buf1 + IO_SIZE;
} else {
buf = buf1;
}
#ifdef TIMER
time_start = clock();
#endif
if (s_blk != NULL){
off = 0;
if (block_left > 0){ // 前回足りなかった分を追加する
if (block_left <= len){
crc = crc_update(crc, buf, block_left);
Phmd5Process2(&hash_ctx, &block_ctx, buf, block_left);
Phmd5End(&block_ctx);
if ((crc != s_blk[find_next].crc) || (memcmp(block_ctx.hash, s_blk[find_next].hash, 16) != 0))
goto error_end; // スライスの CRC-32 か MD5 が異なる
comp_num++;
find_next++;
off = block_left;
block_left = 0;
} else {
crc = crc_update(crc, buf, len);
Phmd5Process2(&hash_ctx, &block_ctx, buf, len);
off = len;
block_left -= len;
}
}
for (; off < len; off += block_size){
Phmd5Begin(&block_ctx);
if (off + block_size <= len){
// s_blk[].crc は初期値と最終処理の 0xFFFFFFFF を取り除いてる
crc = crc_update(0, buf + off, block_size);
Phmd5Process2(&hash_ctx, &block_ctx, buf + off, block_size);
Phmd5End(&block_ctx);
if ((crc != s_blk[find_next].crc) || (memcmp(block_ctx.hash, s_blk[find_next].hash, 16) != 0))
goto error_end; // スライスの CRC-32 か MD5 が異なる
comp_num++;
find_next++;
} else { // スライスが途中までなら
crc = crc_update(0, buf + off, len - off);
Phmd5Process2(&hash_ctx, &block_ctx, buf + off, len - off);
block_left = block_size - len + off;
}
}
} else {
Phmd5Process(&hash_ctx, buf, len); // MD5 計算
}
#ifdef TIMER
time3_total += clock() - time_start;
#endif
// 経過表示
if (GetTickCount() - time_last >= UPDATE_TIME){
if (comp_num <= prog_min){ // 最低値以下なら検出スライス数の合計は表示しない
len = -1;
} else {
len = first_num + comp_num - prog_min;
}
if (print_progress_file((int)(((file_size - file_left) * 1000) / file_size), len, file_name)){
comp_num = -2;
goto error_end;
}
time_last = GetTickCount();
}
}
Phmd5End(&hash_ctx); // 最終処理
if ((s_blk != NULL) && (block_left > 0)){ // 末尾の半端なスライスがあるなら
crc = crc_update_zero(crc, block_left);
Phmd5ProcessZero(&block_ctx, block_left);
Phmd5End(&block_ctx);
if ((crc != s_blk[find_next].crc) || (memcmp(block_ctx.hash, s_blk[find_next].hash, 16) != 0))
goto error_end; // スライスの CRC-32 か MD5 が異なる
comp_num++;
}
if (memcmp(hash_ctx.hash, files[num].hash, 16) == 0) // ファイル全域の MD5 が同じ
comp_num = -3;
error_end:
CancelIo(hFile); // 非同期 IO を取り消す
if (ol.hEvent)
CloseHandle(ol.hEvent);
#ifdef TIMER
time_total += clock() - time1_start;
printf("\nread %.3f sec\n", (double)time2_total / CLOCKS_PER_SEC);
printf("main %.3f sec\n", (double)time3_total / CLOCKS_PER_SEC);
time_sec = (double)time_total / CLOCKS_PER_SEC;
if (time_sec > 0){
time_speed = (double)file_size / (time_sec * 1048576);
} else {
time_speed = 0;
}
printf("total %.3f sec, %.0f MB/s\n", time_sec, time_speed);
#endif
return comp_num;
}
// バックグラウンドでファイルのハッシュ値が同じか調べる (全てのスライスのチェックサムも)
// comp_num 0=破損してるが何個目まで同じ, -1=MD5-16kが異なる, -3=同じ(完全か追加)
DWORD WINAPI file_hash_background(LPVOID lpParameter)
{
unsigned char *buf, *buf1;
int num, find_next, comp_num = 0;
unsigned int len, off, crc, block_left = 0, read_size, io_size;
unsigned int time_last;
__int64 file_size, file_left, file_off;
file_ctx_r *files;
source_ctx_r *s_blk;
HANDLE hFile;
PHMD5 hash_ctx, block_ctx;
OVERLAPPED ol;
FILE_CHECK_TH *file_th;
file_th = (FILE_CHECK_TH *)lpParameter;
num = file_th->num; // file_ctx におけるファイル番号
hFile = file_th->hFile; // ファイルのハンドル
files = file_th->files; // 各ソース・ファイルの情報
s_blk = file_th->s_blk; // 各ソース・ブロックの情報 (NULL なら比較しない)
file_size = files[num].size; // 本来のファイル・サイズ
find_next = files[num].b_off; // 先頭ブロックの番号
// バッファー・サイズが大きいのでヒープ領域を使う
for (io_size = IO_SIZE; io_size <= MAX_BUF_SIZE; io_size += IO_SIZE){ // IO_SIZE の倍数にする
if ((io_size + IO_SIZE > (cpu_cache & 0xFFFF0000)) || ((__int64)(io_size + IO_SIZE) * 4 > file_size))
break;
}
//printf("\n io_size = %d\n", io_size);
buf1 = _aligned_malloc(io_size * 2, 64);
if (buf1 == NULL){
printf("malloc, %d\n", io_size * 2);
comp_num = -1;
goto error_end;
}
buf = buf1 + io_size;
// 非同期ファイル・アクセスの準備をする
memset(&ol, 0, sizeof(OVERLAPPED));
ol.hEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
if (ol.hEvent == NULL){
comp_num = -1;
goto error_end;
}
time_last = GetTickCount();
// まずは先頭 16KB のハッシュ値を比較する
len = 16384;
if (file_size < 16384){
len = (unsigned int)file_size;
file_left = 0;
} else {
file_left = file_size - 16384; // 本来のファイル・サイズまでしか検査しない
}
off = ReadFile(hFile, buf, len, NULL, &ol);
if ((off == 0) && (GetLastError() != ERROR_IO_PENDING)){
print_win32_err();
comp_num = -1;
goto error_end;
}
WaitForSingleObject(ol.hEvent, INFINITE);
Phmd5Begin(&hash_ctx); // 初期化
Phmd5Process(&hash_ctx, buf, len); // MD5 計算
Phmd5End(&hash_ctx); // 最終処理
if (file_left == 0){ // ファイルは 16KB 以下
if (s_blk != NULL){ // 全てのスライスを比較する
for (off = 0; off < len; off += block_size){
Phmd5Begin(&block_ctx);
if (off + block_size <= len){
crc = crc_update(0, buf + off, block_size);
Phmd5Process(&block_ctx, buf + off, block_size);
} else { // 末尾の半端なスライスなら
crc = crc_update(0, buf + off, len - off);
Phmd5Process(&block_ctx, buf + off, len - off);
block_left = block_size - len + off; // ブロック・サイズまで 0で埋めて計算する
crc = crc_update_zero(crc, block_left);
Phmd5ProcessZero(&block_ctx, block_left);
}
Phmd5End(&block_ctx);
if ((crc != s_blk[find_next].crc) || (memcmp(block_ctx.hash, s_blk[find_next].hash, 16) != 0))
goto error_end; // スライスの CRC-32 か MD5 が異なる
comp_num++;
find_next++;
}
}
if (memcmp(hash_ctx.hash, files[num].hash + 16, 16) != 0){ // 16k MD5 が異なる
comp_num = -1;
goto error_end;
}
comp_num = -3; // 同じ
goto error_end;
}
// 16KB よりも大きくて 16k MD5 が異なる場合は、スライス単位の検査をしない
if (memcmp(hash_ctx.hash, files[num].hash + 16, 16) != 0){
comp_num = -1;
goto error_end;
}
// 16k MD5 が一致したら残りの部分も計算する
Phmd5Begin(&hash_ctx); // 初期化
Phmd5Process(&hash_ctx, buf, 16384); // MD5 計算
if (s_blk != NULL){
for (off = 0; off < 16384; off += block_size){
Phmd5Begin(&block_ctx);
if (off + block_size <= 16384){
crc = crc_update(0, buf + off, block_size);
Phmd5Process(&block_ctx, buf + off, block_size);
Phmd5End(&block_ctx);
if ((crc != s_blk[find_next].crc) || (memcmp(block_ctx.hash, s_blk[find_next].hash, 16) != 0))
goto error_end; // スライスの CRC-32 か MD5 が異なる
comp_num++;
find_next++;
} else { // スライスが途中までなら
crc = crc_update(0, buf + off, 16384 - off);
Phmd5Process(&block_ctx, buf + off, 16384 - off);
block_left = block_size - 16384 + off;
}
}
}
// 次の分を読み込む
ol.Offset = 16384;
file_off = 16384 + io_size;
read_size = io_size;
if (file_left < io_size)
read_size = (unsigned int)file_left;
off = ReadFile(hFile, buf1, read_size, NULL, &ol);
if ((off == 0) && (GetLastError() != ERROR_IO_PENDING)){
print_win32_err();
goto error_end; // 読み取りが失敗した所で終わる
}
while (file_left > 0){
len = read_size; // 読み込んだサイズ
file_left -= read_size;
// 前回の読み込みが終わるのを待つ
WaitForSingleObject(ol.hEvent, INFINITE);
// 次の分を読み込み開始しておく
if (file_left > 0){
read_size = io_size;
if (file_left < io_size)
read_size = (unsigned int)file_left;
ol.Offset = (unsigned int)file_off;
ol.OffsetHigh = (unsigned int)(file_off >> 32);
file_off += io_size;
off = ReadFile(hFile, buf, read_size, NULL, &ol);
if ((off == 0) && (GetLastError() != ERROR_IO_PENDING)){
print_win32_err();
goto error_end; // 読み取りが失敗した所で終わる
}
}
// バッファーを入れ替える
if (buf == buf1){
buf = buf1 + io_size;
} else {
buf = buf1;
}
if (s_blk != NULL){
off = 0;
if (block_left > 0){ // 前回足りなかった分を追加する
if (block_left <= len){
crc = crc_update(crc, buf, block_left);
Phmd5Process2(&hash_ctx, &block_ctx, buf, block_left);
Phmd5End(&block_ctx);
if ((crc != s_blk[find_next].crc) || (memcmp(block_ctx.hash, s_blk[find_next].hash, 16) != 0))
goto error_end; // スライスの CRC-32 か MD5 が異なる
comp_num++;
find_next++;
off = block_left;
block_left = 0;
} else {
crc = crc_update(crc, buf, len);
Phmd5Process2(&hash_ctx, &block_ctx, buf, len);
off = len;
block_left -= len;
}
}
for (; off < len; off += block_size){
Phmd5Begin(&block_ctx);
if (off + block_size <= len){
// s_blk[].crc は初期値と最終処理の 0xFFFFFFFF を取り除いてる
crc = crc_update(0, buf + off, block_size);
Phmd5Process2(&hash_ctx, &block_ctx, buf + off, block_size);
Phmd5End(&block_ctx);
if ((crc != s_blk[find_next].crc) || (memcmp(block_ctx.hash, s_blk[find_next].hash, 16) != 0))
goto error_end; // スライスの CRC-32 か MD5 が異なる
comp_num++;
find_next++;
} else { // スライスが途中までなら
crc = crc_update(0, buf + off, len - off);
Phmd5Process2(&hash_ctx, &block_ctx, buf + off, len - off);
block_left = block_size - len + off;
}
}
} else {
Phmd5Process(&hash_ctx, buf, len); // MD5 計算
}
// 経過更新
if (GetTickCount() - time_last >= UPDATE_TIME){
if (file_th->flag < 0) // キャンセル指示が出た
goto error_end;
// 上位 12-bit にパーセント、下位 20-bit に検出ブロック数を記録する
file_th->rv = ((int)(((file_size - file_left) * 1000) / file_size) << 20) | comp_num;
time_last = GetTickCount();
}
}
Phmd5End(&hash_ctx); // 最終処理
if ((s_blk != NULL) && (block_left > 0)){ // 末尾の半端なスライスがあるなら
crc = crc_update_zero(crc, block_left);
Phmd5ProcessZero(&block_ctx, block_left);
Phmd5End(&block_ctx);
if ((crc != s_blk[find_next].crc) || (memcmp(block_ctx.hash, s_blk[find_next].hash, 16) != 0))
goto error_end; // スライスの CRC-32 か MD5 が異なる
comp_num++;
}
if (memcmp(hash_ctx.hash, files[num].hash, 16) == 0) // ファイル全域の MD5 が同じ
comp_num = -3;
error_end:
CancelIo(hFile); // 非同期 IO を取り消す
CloseHandle(hFile); // ファイルはサブ・スレッド側で閉じる
if (ol.hEvent)
CloseHandle(ol.hEvent);
if (buf1)
_aligned_free(buf1);
file_th->rv = comp_num;
return 0;
}
// キャッシュ無しでファイルのハッシュ値が同じか調べる
// 0=破損してるが何個目まで同じ, -1=MD5-16kが異なる, -2=キャンセル, -3=最後まで同じ(完全), -4=存在しない, -5=追加
int file_hash_direct(
int num, // file_ctx におけるファイル番号
wchar_t *file_path, // ハッシュ値を求めるファイル
wchar_t *file_name, // 表示するファイル名
file_ctx_r *files, // 各ソース・ファイルの情報
source_ctx_r *s_blk) // 各ソース・ブロックの情報 (NULL なら比較しない)
{
unsigned char *buf, *buf1 = NULL;
int find_next, comp_num = 0;
unsigned int len, off, crc, block_left = 0, read_size;
unsigned int time_last;
__int64 file_size, file_left, file_off;
PHMD5 hash_ctx, block_ctx;
HANDLE hFile;
OVERLAPPED ol;
#ifdef TIMER
time1_start = clock();
#endif
prog_last = -1; // 検証中のファイル名を毎回表示する
time_last = GetTickCount();
file_size = files[num].size; // 本来のファイル・サイズ
find_next = files[num].b_off; // 先頭ブロックの番号
// ソース・ファイルを開く
hFile = CreateFile(file_path, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_FLAG_NO_BUFFERING | FILE_FLAG_OVERLAPPED, NULL);
if (hFile == INVALID_HANDLE_VALUE){
off = GetLastError();
if (off == 2) // file not found (2)
return -4;
if ((off == 32) || // ERROR_SHARING_VIOLATION
(off == 33)){ // ERROR_LOCK_VIOLATION
Sleep(300); // 少し待ってから再挑戦する
hFile = CreateFile(file_path, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_FLAG_NO_BUFFERING | FILE_FLAG_OVERLAPPED, NULL);
} else {
//printf("cannot set FILE_FLAG_NO_BUFFERING %d\n", GetLastError());
// ディスク・キャッシュを無効にできない場合
hFile = CreateFile(file_path, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_FLAG_OVERLAPPED | FILE_FLAG_SEQUENTIAL_SCAN, NULL);
}
if (hFile == INVALID_HANDLE_VALUE)
return -4;
}
// 非同期ファイル・アクセスの準備をする
memset(&ol, 0, sizeof(OVERLAPPED));
ol.hEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
if (ol.hEvent == NULL){
comp_num = -1;
goto error_end;
}
// バッファーの境界をセクター・サイズの倍数にする
// IO_SIZE は 16KB 以上で 4KB の倍数にすること
buf1 = _aligned_malloc(IO_SIZE * 2, 4096);
if (buf1 == NULL){
printf("malloc, %d\n", IO_SIZE * 2);
comp_num = -1;
goto error_end;
}
buf = buf1 + IO_SIZE;
// まずは先頭 16KB のハッシュ値を比較する
if (file_size < 16384){
len = (unsigned int)file_size;
read_size = (len + 4095) & ~4095; // 4KB の倍数にする
file_left = 0;
} else {
len = 16384;
read_size = 16384;
file_left = file_size - 16384; // 本来のファイル・サイズまでしか検査しない
}
#ifdef TIMER
time_start = clock();
#endif
off = ReadFile(hFile, buf, read_size, NULL, &ol);
#ifdef TIMER
time2_total += clock() - time_start;
#endif
if ((off == 0) && (GetLastError() != ERROR_IO_PENDING)){
comp_num = -1;
goto error_end;
}
WaitForSingleObject(ol.hEvent, INFINITE);
Phmd5Begin(&hash_ctx); // 初期化
Phmd5Process(&hash_ctx, buf, len); // MD5 計算
Phmd5End(&hash_ctx); // 最終処理
if (file_left == 0){ // ファイルは 16KB 以下
if (s_blk != NULL){ // 全てのスライスを比較する
for (off = 0; off < len; off += block_size){
Phmd5Begin(&block_ctx);
if (off + block_size <= len){
crc = crc_update(0, buf + off, block_size);
Phmd5Process(&block_ctx, buf + off, block_size);
} else { // 末尾の半端なスライスなら
crc = crc_update(0, buf + off, len - off);
Phmd5Process(&block_ctx, buf + off, len - off);
block_left = block_size - len + off; // ブロック・サイズまで 0で埋めて計算する
crc = crc_update_zero(crc, block_left);
Phmd5ProcessZero(&block_ctx, block_left);
}
Phmd5End(&block_ctx);
if ((crc != s_blk[find_next].crc) || (memcmp(block_ctx.hash, s_blk[find_next].hash, 16) != 0))
goto error_end; // スライスの CRC-32 か MD5 が異なる
comp_num++;
find_next++;
}
}
if (memcmp(hash_ctx.hash, files[num].hash + 16, 16) != 0){ // 16k MD5 が異なる
comp_num = -1;
goto error_end;
}
comp_num = -3; // 同じ
// ファイルサイズが大きい場合は末尾に追加されてる
if (!GetFileSizeEx(hFile, (PLARGE_INTEGER)&file_left))
file_left = 0;
if (file_left != file_size)
comp_num = -5;
goto error_end;
}
// 16KB よりも大きくて 16k MD5 が異なる場合は、スライス単位の検査をしない
if (memcmp(hash_ctx.hash, files[num].hash + 16, 16) != 0){
comp_num = -1;
goto error_end;
}
// 16k MD5 が一致したら残りの部分も計算する
Phmd5Begin(&hash_ctx); // 初期化
Phmd5Process(&hash_ctx, buf, 16384); // MD5 計算
if (s_blk != NULL){
for (off = 0; off < 16384; off += block_size){
Phmd5Begin(&block_ctx);
if (off + block_size <= 16384){
crc = crc_update(0, buf + off, block_size);
Phmd5Process(&block_ctx, buf + off, block_size);
Phmd5End(&block_ctx);
if ((crc != s_blk[find_next].crc) || (memcmp(block_ctx.hash, s_blk[find_next].hash, 16) != 0))
goto error_end; // スライスの CRC-32 か MD5 が異なる
comp_num++;
find_next++;
} else { // スライスが途中までなら
crc = crc_update(0, buf + off, 16384 - off);
Phmd5Process(&block_ctx, buf + off, 16384 - off);
block_left = block_size - 16384 + off;
}
}
}
// 次の分を読み込む
ol.Offset = 16384;
file_off = 16384 + IO_SIZE;
read_size = IO_SIZE;
if (file_left < IO_SIZE){
read_size = (unsigned int)file_left;
read_size = (read_size + 4095) & ~4095; // 4KB の倍数にする
}
#ifdef TIMER
time_start = clock();
#endif
off = ReadFile(hFile, buf1, read_size, NULL, &ol);
#ifdef TIMER
time2_total += clock() - time_start;
#endif
if ((off == 0) && (GetLastError() != ERROR_IO_PENDING)){
print_win32_err();
goto error_end; // 読み取りが失敗した所で終わる
}
while (file_left > 0){
len = IO_SIZE; // 読み込んだサイズ
if (file_left < IO_SIZE)
len = (unsigned int)file_left;
file_left -= len;
// 前回の読み込みが終わるのを待つ
WaitForSingleObject(ol.hEvent, INFINITE);
// 次の分を読み込み開始しておく
if (file_left > 0){
read_size = IO_SIZE;
if (file_left < IO_SIZE){
read_size = (unsigned int)file_left;
read_size = (read_size + 4095) & ~4095; // 4KB の倍数にする
}
ol.Offset = (unsigned int)file_off;
ol.OffsetHigh = (unsigned int)(file_off >> 32);
file_off += IO_SIZE;
#ifdef TIMER
time_start = clock();
#endif
off = ReadFile(hFile, buf, read_size, NULL, &ol);
#ifdef TIMER
time2_total += clock() - time_start;
#endif
if ((off == 0) && (GetLastError() != ERROR_IO_PENDING)){
print_win32_err();
goto error_end; // 読み取りが失敗した所で終わる
}
}
// バッファーを入れ替える
if (buf == buf1){
buf = buf1 + IO_SIZE;
} else {
buf = buf1;
}
#ifdef TIMER
time_start = clock();
#endif
if (s_blk != NULL){
off = 0;
if (block_left > 0){ // 前回足りなかった分を追加する
if (block_left <= len){
crc = crc_update(crc, buf, block_left);
Phmd5Process2(&hash_ctx, &block_ctx, buf, block_left);
Phmd5End(&block_ctx);
if ((crc != s_blk[find_next].crc) || (memcmp(block_ctx.hash, s_blk[find_next].hash, 16) != 0))
goto error_end; // スライスの CRC-32 か MD5 が異なる
comp_num++;
find_next++;
off = block_left;
block_left = 0;
} else {
crc = crc_update(crc, buf, len);
Phmd5Process2(&hash_ctx, &block_ctx, buf, len);
off = len;
block_left -= len;
}
}
for (; off < len; off += block_size){
Phmd5Begin(&block_ctx);
if (off + block_size <= len){
crc = crc_update(0, buf + off, block_size);
Phmd5Process2(&hash_ctx, &block_ctx, buf + off, block_size);
Phmd5End(&block_ctx);
if ((crc != s_blk[find_next].crc) || (memcmp(block_ctx.hash, s_blk[find_next].hash, 16) != 0))
goto error_end; // スライスの CRC-32 か MD5 が異なる
comp_num++;
find_next++;
} else { // スライスが途中までなら
crc = crc_update(0, buf + off, len - off);
Phmd5Process2(&hash_ctx, &block_ctx, buf + off, len - off);
block_left = block_size - len + off;
}
}
} else {
Phmd5Process(&hash_ctx, buf, len); // MD5 計算
}
#ifdef TIMER
time3_total += clock() - time_start;
#endif
// 経過表示
if (GetTickCount() - time_last >= UPDATE_TIME){
if (print_progress_file((int)(((file_size - file_left) * 1000) / file_size), -1, file_name)){
comp_num = -2;
goto error_end;
}
time_last = GetTickCount();
}
}
if ((s_blk != NULL) && (block_left > 0)){ // 末尾の半端なスライスがあるなら
crc = crc_update_zero(crc, block_left);
Phmd5ProcessZero(&block_ctx, block_left);
Phmd5End(&block_ctx);
if ((crc != s_blk[find_next].crc) || (memcmp(block_ctx.hash, s_blk[find_next].hash, 16) != 0))
goto error_end; // スライスの CRC-32 か MD5 が異なる
comp_num++;
}
Phmd5End(&hash_ctx); // 最終処理
if (memcmp(hash_ctx.hash, files[num].hash, 16) == 0) // ファイル全域の MD5 が同じ
comp_num = -3;
// ファイルサイズが大きい場合は末尾に追加されてる
if (!GetFileSizeEx(hFile, (PLARGE_INTEGER)&file_left))
file_left = 0;
if (file_left != file_size)
comp_num = -5;
error_end:
CancelIo(hFile); // 非同期 IO を取り消す
CloseHandle(hFile);
if (ol.hEvent)
CloseHandle(ol.hEvent);
if (buf1)
_aligned_free(buf1);
#ifdef TIMER
time_total += clock() - time1_start;
printf("\nread %.3f sec\n", (double)time2_total / CLOCKS_PER_SEC);
printf("main %.3f sec\n", (double)time3_total / CLOCKS_PER_SEC);
time_sec = (double)time_total / CLOCKS_PER_SEC;
if (time_sec > 0){
time_speed = (double)file_size / (time_sec * 1048576);
} else {
time_speed = 0;
}
printf("total %.3f sec, %.0f MB/s\n", time_sec, time_speed);
#endif
return comp_num;
}
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
// CRC-32 を使って 1バイト内のバースト・エラーを訂正する
// -1=エラー訂正できず, 0=エラー無し, 1=エラー訂正できた
// 全て 0 のデータの判定は他で行うこと
int correct_error(
unsigned char *data_in, // ハッシュ値を求めたバイト配列
unsigned int data_len, // 入力バイト数
unsigned char *hash, // (16, MD5)
unsigned int crc, // 入力バイト数分の本来の CRC-32
unsigned int *error_off, // エラー開始位置
unsigned char *error_mag) // エラー内容、XORでエラー訂正を取り消せる
{
unsigned char data_hash[16];
unsigned int data_crc, pos;
// まずエラーが存在するかを調べる
data_crc = crc_update(0, data_in, data_len);
if (data_crc == crc){
// MD5 はブロック・サイズ分を計算する
data_md5_block(data_in, data_len, data_hash);
if (memcmp(data_hash, hash, 16) == 0)
return 0; // エラー無し
return -1; // CRC-32 でエラーを検出できない場合は訂正もできない
}
if (data_len > 0x100000)
return -1; // 1MB 以上ならエラー訂正しない
// CRC-32 で 1バイト訂正を試みると、失敗する確率はエラー位置ごとに 1 / 2^24 ぐらい?
// データ・サイズが 1MB (1^20) だと、訂正できなくても 1/16 の確率で MD5 による検算が必要
// エラー位置と程度を計算する
crc ^= data_crc; // CRC-32 の差にする
for (pos = 1; pos <= data_len; pos++){
// 1 バイトずつ遡りながら最初のエラー発生位置を探す
crc = reverse_table[(crc >> 24)] ^ (crc << 8);
// 一定範囲内のエラーを訂正する
if ((crc & 0xFFFFFF00) == 0){
// 試しにエラーを訂正してみる
*error_mag = (unsigned char)crc;
data_in[data_len - pos] ^= *error_mag; // エラーが発生した 1バイト目
// 正しく訂正できてるかを確認する
data_md5_block(data_in, data_len, data_hash);
if (memcmp(data_hash, hash, 16) == 0){
//printf("position = %d byte from end, magnitude = 0x%08X\n", pos - 1, crc);
*error_off = data_len - pos;
return 1; // エラー訂正できた
}
// 訂正失敗ならデータを元に戻しておく
//printf("MD5 found errors\n");
data_in[data_len - pos] ^= *error_mag;
}
}
return -1; // 訂正できず
}
// CRC-32 を使って 1バイト内のバースト・エラーを訂正する
// -1=エラー訂正できず, 0=エラー無し, 1=エラー訂正できた
// CRC-32 は既に計算済み、エラー無しも比較済み
int correct_error2(
unsigned char *data_in, // ハッシュ値を求めたバイト配列
unsigned int data_len, // 入力バイト数
unsigned int data_crc, // 入力バイト数分の CRC-32
unsigned char *hash, // (16, MD5)
unsigned int crc, // 入力バイト数分の本来の CRC-32
unsigned int *error_off, // エラー開始位置
unsigned char *error_mag) // エラー内容、XORでエラー訂正を取り消せる
{
unsigned char data_hash[16];
unsigned int pos;
if (data_len > 0x100000)
return -1; // 1MB 以上ならエラー訂正しない
// エラー位置と程度を計算する
crc ^= data_crc; // CRC-32 の差にする
for (pos = 1; pos <= data_len; pos++){
// 1 バイトずつ遡りながら最初のエラー発生位置を探す
crc = reverse_table[(crc >> 24)] ^ (crc << 8);
// 一定範囲内のエラーを訂正する
if ((crc & 0xFFFFFF00) == 0){
// 試しにエラーを訂正してみる
*error_mag = (unsigned char)crc;
data_in[data_len - pos] ^= *error_mag; // エラーが発生した 1バイト目
// 正しく訂正できてるかを確認する
data_md5_block(data_in, data_len, data_hash);
if (memcmp(data_hash, hash, 16) == 0){
//printf("position = %d byte from end, magnitude = 0x%08X\n", pos - 1, crc);
*error_off = data_len - pos;
return 1; // エラー訂正できた
}
// 訂正失敗ならデータを元に戻しておく
//printf("MD5 found errors\n");
data_in[data_len - pos] ^= *error_mag;
}
}
return -1; // 訂正できず
}
/*
// MD5 と CRC32 から 8バイトのデータを逆算するための関数
void data_reverse8(
unsigned char *data_out, // 不明な入力データ (8バイト)
unsigned char *hash, // ハッシュ値 (16バイト)
unsigned char *crc) // CRC (4バイト)
{
unsigned int i, rv, crc1, crc2, data[16], state[4], time;
// crc = 0xffffffff ^ data[03]; // 初期値と最初のデータを XOR する
// crc = table[crc & 0xff] ^ (crc >> 8);
// crc = table[crc & 0xff] ^ (crc >> 8);
// crc = table[crc & 0xff] ^ (crc >> 8);
// crc = table[crc & 0xff] ^ (crc >> 8);
// crc ^= data[47]; // 結果と次のデータを XOR する
// crc = table[crc & 0xff] ^ (crc >> 8);
// crc = table[crc & 0xff] ^ (crc >> 8);
// crc = table[crc & 0xff] ^ (crc >> 8);
// crc = table[crc & 0xff] ^ (crc >> 8);
// crc ^= 0xffffffff; 最終処理
time = GetTickCount();
memcpy(&crc2, crc, 4);
crc2 ^= 0xffffffff; // 最終処理を取り消す
rv = CrcReverse4(crc2); // 2回目のCRC計算を取り消した値 = crc ^ data[47]
// 1回目のCRC計算の結果がわからないので、入力データもわからない (XORした値だけわかってる)
// そこで、入力データを変化させながら1回目のCRCを逆算して、候補をMD5で確認する
for (i = 0; i < 16; i++)
data[i] = 0;
data[2] = 0x00000080; // データ終端の 1bit
data[14] = 0x00000040; // データ・サイズ 8byte = 64bit
do {
crc1 = rv ^ data[1];
data[0] = CrcReverse4(crc1) ^ 0xffffffff; // 逆算して初期化処理を取り消す
if ((data[1] & 0x00ffffff) == 0)
printf("%08x%08x\n", data[0], data[1]);
// このデータで合ってるかをMD5で調べる
state[0] = 0x67452301;
state[1] = 0xefcdab89;
state[2] = 0x98badcfe;
state[3] = 0x10325476;
MD5Transform(data, state);
if (memcmp(hash, state, 16) == 0)
break;
data[1]++;
} while (data[1] != 0);
memcpy(data_out, (unsigned char *)data, 8);
// Pentium 3 800MHz だと完全走査に約50分かかる、テキストなら20分ぐらい?
time = (GetTickCount() - time);
printf("\n%3u.%03u sec\n", time / 1000, time % 1000);
}
*/