Add files via upload
This commit is contained in:
Yutaka Sawada
861
source/sfv_md5/main.c
Normal file
861
source/sfv_md5/main.c
Normal file
@@ -0,0 +1,861 @@
|
||||
// main.c
|
||||
// Copyright : 2022-02-25 Yutaka Sawada
|
||||
// License : The MIT license
|
||||
|
||||
#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 <imagehlp.h>
|
||||
|
||||
#include "crc.h"
|
||||
#include "common.h"
|
||||
#include "create.h"
|
||||
#include "verify.h"
|
||||
#include "ini.h"
|
||||
#include "version.h"
|
||||
|
||||
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
|
||||
|
||||
void print_help(void)
|
||||
{
|
||||
printf(
|
||||
"Usage\n"
|
||||
"c(reate) [fo,t, d,u] <checksum file> [input files]\n"
|
||||
"v(erify) [vs,vd,d,u] <checksum file>\n"
|
||||
"\nOption\n"
|
||||
" /fo : Search file only for wildcard\n"
|
||||
" /t<n> : Save time stamp\n"
|
||||
" /vs<n>: Skip verification by recent result\n"
|
||||
" /vd\"*\": Set directory of recent result\n"
|
||||
" /d\"*\" : Set directory of input files\n"
|
||||
" /u : Console output is encoded with UTF-8\n"
|
||||
);
|
||||
}
|
||||
|
||||
// CRC-32 チェックサムを使って自分自身の破損を検出する
|
||||
long test_checksum(wchar_t *file_path) // 作業用
|
||||
{
|
||||
unsigned long rv, crc, chk, chk2;
|
||||
unsigned char *pAddr;
|
||||
HANDLE hFile, hMap;
|
||||
|
||||
init_crc_table(); // CRC 計算用のテーブルを作成する
|
||||
|
||||
// 実行ファイルのパスを取得する
|
||||
rv = GetModuleFileName(NULL, file_path, MAX_LEN);
|
||||
if ((rv == 0) || (rv >= MAX_LEN))
|
||||
return 1;
|
||||
|
||||
// 実行ファイルの PE checksum と CRC-32 を検証する
|
||||
hFile = CreateFile(file_path, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL);
|
||||
if (hFile == INVALID_HANDLE_VALUE){
|
||||
print_win32_err();
|
||||
return 1;
|
||||
}
|
||||
rv = GetFileSize(hFile, &chk2);
|
||||
if (rv == INVALID_FILE_SIZE)
|
||||
return 1;
|
||||
hMap = CreateFileMapping(hFile, NULL, PAGE_READONLY, chk2, rv, NULL);
|
||||
if (hMap == NULL){
|
||||
CloseHandle(hFile);
|
||||
return 1;
|
||||
}
|
||||
pAddr = MapViewOfFile(hMap, FILE_MAP_READ, 0, 0, rv);
|
||||
if (pAddr == NULL){
|
||||
CloseHandle(hMap);
|
||||
CloseHandle(hFile);
|
||||
return 1;
|
||||
}
|
||||
if (CheckSumMappedFile(pAddr, rv, &chk2, &chk) == NULL){ // PE checksum
|
||||
UnmapViewOfFile(pAddr);
|
||||
CloseHandle(hMap);
|
||||
CloseHandle(hFile);
|
||||
return 1;
|
||||
}
|
||||
crc = crc_update(0xFFFFFFFF, pAddr, rv) ^ 0xFFFFFFFF;
|
||||
UnmapViewOfFile(pAddr);
|
||||
CloseHandle(hMap);
|
||||
CloseHandle(hFile);
|
||||
|
||||
if (chk != chk2)
|
||||
return 2;
|
||||
if (crc != 0x00000000)
|
||||
return 3;
|
||||
return 0;
|
||||
}
|
||||
|
||||
// ファイルを検索してファイル・リストに追加する
|
||||
static wchar_t * search_files(
|
||||
wchar_t *list_buf, // ファイル・リスト
|
||||
wchar_t *search_path, // 検索するファイルのフル・パス
|
||||
long dir_len, // ディレクトリ部分の長さ
|
||||
long file_only, // ファイルのみにするかどうか
|
||||
long single_file, // -1 = *や?で検索指定、0～ = 単独指定
|
||||
long *list_max, // ファイル・リストの確保サイズ
|
||||
long *list_len, // ファイル・リストの文字数
|
||||
__int64 *total_size) // 合計ファイル・サイズ
|
||||
{
|
||||
wchar_t *tmp_p;
|
||||
long len, l_max, l_off, dir_len2;
|
||||
HANDLE hFind;
|
||||
WIN32_FIND_DATA FindData;
|
||||
|
||||
if (list_buf == NULL){
|
||||
l_off = 0;
|
||||
l_max = ALLOC_LEN;
|
||||
list_buf = (wchar_t *)malloc(l_max);
|
||||
if (list_buf == NULL){
|
||||
printf("malloc, %d\n", l_max);
|
||||
return NULL;
|
||||
}
|
||||
} else {
|
||||
l_max = *list_max;
|
||||
l_off = *list_len;
|
||||
}
|
||||
|
||||
// 検索する
|
||||
hFind = FindFirstFile(search_path, &FindData);
|
||||
if (hFind == INVALID_HANDLE_VALUE)
|
||||
return list_buf; // 見つからなかったらそのまま
|
||||
do {
|
||||
if ((single_file < 0) && (FindData.dwFileAttributes & FILE_ATTRIBUTE_HIDDEN))
|
||||
continue; // 検索中は隠し属性が付いてるファイルを無視する
|
||||
|
||||
len = wcslen(FindData.cFileName);
|
||||
if (dir_len + len >= MAX_LEN - ADD_LEN - 2){ // 末尾に「\*」を付けて再検索するので
|
||||
FindClose(hFind);
|
||||
free(list_buf);
|
||||
printf("filename is too long\n");
|
||||
return NULL;
|
||||
}
|
||||
|
||||
// 現在のディレクトリ部分に見つかったファイル名を連結する
|
||||
wcscpy(search_path + dir_len, FindData.cFileName);
|
||||
|
||||
// フォルダなら
|
||||
if (FindData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY){
|
||||
if ((file_only == 0) && wcscmp(FindData.cFileName, L".") && wcscmp(FindData.cFileName, L"..")){
|
||||
// フォルダの末尾は「\」にする
|
||||
wcscat(search_path, L"\\");
|
||||
// そのフォルダの中身を更に検索する
|
||||
dir_len2 = wcslen(search_path);
|
||||
search_path[dir_len2 ] = '*'; // 末尾に「*」を追加する
|
||||
search_path[dir_len2 + 1] = 0;
|
||||
list_buf = search_files(list_buf, search_path, dir_len2, file_only, single_file, &l_max, &l_off, total_size);
|
||||
if (list_buf == NULL){
|
||||
FindClose(hFind);
|
||||
printf("cannot search inner folder\n");
|
||||
return NULL;
|
||||
}
|
||||
}
|
||||
} else { // ファイルなら
|
||||
if (!search_file_path(list_buf, l_off, search_path + base_len)){ // ファイル名が重複しないようにする
|
||||
if ((l_off + dir_len- base_len + len) * 2 >= l_max){ // 領域が足りなくなるなら拡張する
|
||||
l_max += ALLOC_LEN;
|
||||
tmp_p = (wchar_t *)realloc(list_buf, l_max);
|
||||
if (tmp_p == NULL){
|
||||
FindClose(hFind);
|
||||
free(list_buf);
|
||||
printf("realloc, %d\n", l_max);
|
||||
return NULL;
|
||||
} else {
|
||||
list_buf = tmp_p;
|
||||
}
|
||||
}
|
||||
|
||||
// リストにコピーする
|
||||
wcscpy(list_buf + l_off, search_path + base_len);
|
||||
l_off += dir_len - base_len + len + 1;
|
||||
file_num++;
|
||||
(*total_size) += ((__int64)FindData.nFileSizeHigh << 32) | (__int64)FindData.nFileSizeLow;
|
||||
}
|
||||
}
|
||||
} while (FindNextFile(hFind, &FindData)); // 次のファイルを検索する
|
||||
FindClose(hFind);
|
||||
|
||||
*list_max = l_max;
|
||||
*list_len = l_off;
|
||||
return list_buf;
|
||||
}
|
||||
|
||||
// ファイルの詳細情報を記録する
|
||||
long save_detail(
|
||||
wchar_t *uni_buf,
|
||||
wchar_t *file_name) // 検査対象のファイル名
|
||||
{
|
||||
__int64 file_size;
|
||||
HANDLE hFile;
|
||||
FILETIME ftWrite;
|
||||
SYSTEMTIME stUTC, stLocal;
|
||||
|
||||
// 調べるファイルを開く
|
||||
wcscpy(uni_buf, base_dir);
|
||||
wcscpy(uni_buf + base_len, file_name);
|
||||
hFile = CreateFile(uni_buf, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL);
|
||||
if (hFile == INVALID_HANDLE_VALUE){
|
||||
print_win32_err();
|
||||
return 1;
|
||||
}
|
||||
// ファイル・サイズ
|
||||
if (!GetFileSizeEx(hFile, (PLARGE_INTEGER)&file_size)){
|
||||
print_win32_err();
|
||||
CloseHandle(hFile);
|
||||
return 1;
|
||||
}
|
||||
// 更新日時
|
||||
if (!GetFileTime(hFile, NULL, NULL, &ftWrite)){
|
||||
print_win32_err();
|
||||
CloseHandle(hFile);
|
||||
return 1;
|
||||
}
|
||||
FileTimeToSystemTime(&ftWrite, &stUTC);
|
||||
SystemTimeToTzSpecificLocalTime(NULL, &stUTC, &stLocal);
|
||||
CloseHandle(hFile);
|
||||
|
||||
// サイズと日時を書き込む
|
||||
wsprintf(uni_buf, L";%13I64d %02d:%02d.%02d %4d-%02d-%02d ", file_size,
|
||||
stLocal.wHour, stLocal.wMinute, stLocal.wSecond,
|
||||
stLocal.wYear, stLocal.wMonth, stLocal.wDay);
|
||||
add_text(uni_buf);
|
||||
|
||||
// ファイル名を書き込む
|
||||
wcscpy(uni_buf, file_name); // 変換前にコピーする
|
||||
unix_directory(uni_buf);
|
||||
add_text(uni_buf);
|
||||
add_text(L"\r\n");
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
// チェックサム・ファイルを書き込む
|
||||
static long write_checksum(wchar_t *uni_buf,
|
||||
wchar_t *list_buf, long list_len, __int64 total_size, long switch_t)
|
||||
{
|
||||
char *ascii_buf;
|
||||
wchar_t *ads_p;
|
||||
long err = 0, rv, len, format;
|
||||
__int64 prog_now = 0;
|
||||
HANDLE hFile;
|
||||
FILETIME ftWrite;
|
||||
|
||||
len = wcslen(checksum_file);
|
||||
if (_wcsicmp(checksum_file + (len - 4), L".sfv") == 0){ // SFVファイル
|
||||
format = 1;
|
||||
} else if (_wcsicmp(checksum_file + (len - 4), L".md5") == 0){ // MD5ファイル
|
||||
format = 2;
|
||||
// } else {
|
||||
// printf("unknown format\n");
|
||||
// return 1;
|
||||
}
|
||||
|
||||
// チェックサム・ファイルのテキストを UTF-16 の文字列で作成する
|
||||
text_len = 0;
|
||||
text_max = ALLOC_LEN;
|
||||
text_buf = (wchar_t *)malloc(text_max * 2);
|
||||
if (text_buf == NULL){
|
||||
printf("malloc, %d\n", text_max * 2);
|
||||
return 1;
|
||||
}
|
||||
|
||||
// コメントを書き込む
|
||||
wsprintf(uni_buf, L"; Generated by SFV/MD5 checker v%hs", PRODUCT_VERSION);
|
||||
add_text(uni_buf);
|
||||
if (switch_t >= 1){ // 作成した日時を書き込む
|
||||
SYSTEMTIME stLocal;
|
||||
GetLocalTime(&stLocal);
|
||||
wsprintf(uni_buf, L" on %4d-%02d-%02d at %02d:%02d.%02d",
|
||||
stLocal.wYear, stLocal.wMonth, stLocal.wDay,
|
||||
stLocal.wHour, stLocal.wMinute, stLocal.wSecond);
|
||||
add_text(uni_buf);
|
||||
if (switch_t >= 2){ // ファイルの詳細を書き込むなら
|
||||
add_text(L"\r\n;");
|
||||
if (file_num > 1){
|
||||
add_text(L"\r\n");
|
||||
add_text(L"; Size (Bytes) Time Date Filename\n");
|
||||
add_text(L"; ------------ -------- ---------- ------------");
|
||||
}
|
||||
}
|
||||
}
|
||||
add_text(L"\r\n");
|
||||
|
||||
if (switch_t >= 2){ // ファイルのサイズと更新日時を書き込む
|
||||
len = 0;
|
||||
while (len < list_len){
|
||||
if (save_detail(uni_buf, list_buf + len) != 0){
|
||||
printf("cannot save detail\n");
|
||||
free(text_buf);
|
||||
return 1;
|
||||
}
|
||||
len += wcslen(list_buf + len) + 1;
|
||||
}
|
||||
add_text(L";\r\n");
|
||||
}
|
||||
|
||||
// 各ファイルのチェックサムを計算する
|
||||
printf("\n");
|
||||
print_progress_text(0, "Computing file hash");
|
||||
len = 0;
|
||||
while (len < list_len){
|
||||
if (format == 1){
|
||||
rv = create_sfv(uni_buf, list_buf + len, &prog_now, total_size);
|
||||
} else if (format == 2){
|
||||
rv = create_md5(uni_buf, list_buf + len, &prog_now, total_size);
|
||||
} else {
|
||||
rv = 1;
|
||||
}
|
||||
if (rv != 0){
|
||||
free(text_buf);
|
||||
return rv; // エラー、キャンセルなど
|
||||
}
|
||||
len += wcslen(list_buf + len) + 1;
|
||||
}
|
||||
print_progress_done(); // 改行して行の先頭に戻しておく
|
||||
|
||||
// 既存ファイルの alternate data stream に書き込む場合は更新日時をそのままにする
|
||||
ads_p = wcschr(offset_file_name(checksum_file), ':');
|
||||
if (ads_p != NULL){
|
||||
*ads_p = 0;
|
||||
hFile = CreateFile(checksum_file, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, NULL);
|
||||
*ads_p = ':';
|
||||
if (hFile == INVALID_HANDLE_VALUE){
|
||||
ads_p = NULL;
|
||||
} else {
|
||||
if (!GetFileTime(hFile, NULL, NULL, &ftWrite)){
|
||||
//print_win32_err();
|
||||
//printf("cannot read time stamp\n");
|
||||
ads_p = NULL;
|
||||
}
|
||||
CloseHandle(hFile);
|
||||
}
|
||||
}
|
||||
|
||||
// UTF-16 から UTF-8 に変換する
|
||||
len = text_len * 3;
|
||||
ascii_buf = (char *)malloc(len);
|
||||
if (ascii_buf == NULL){
|
||||
printf("malloc, %d\n", len);
|
||||
free(text_buf);
|
||||
return 1;
|
||||
}
|
||||
if (utf16_to_cp(text_buf, ascii_buf, len, CP_UTF8) != 0){
|
||||
printf("cannot encode text\n");
|
||||
free(ascii_buf);
|
||||
free(text_buf);
|
||||
return 1;
|
||||
}
|
||||
len = strlen(ascii_buf);
|
||||
|
||||
// チェックサム・ファイルを開いて、エンコードしたテキストを書き込む
|
||||
hFile = CreateFile(checksum_file, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
|
||||
if (hFile == INVALID_HANDLE_VALUE){
|
||||
print_win32_err();
|
||||
printf("cannot create checksum file\n");
|
||||
free(ascii_buf);
|
||||
free(text_buf);
|
||||
return 1;
|
||||
}
|
||||
if (!WriteFile(hFile, ascii_buf, len, &rv, NULL)){
|
||||
print_win32_err();
|
||||
printf("cannot write checksum file\n");
|
||||
CloseHandle(hFile);
|
||||
free(ascii_buf);
|
||||
free(text_buf);
|
||||
return 1;
|
||||
}
|
||||
// main stream の更新日時を元に戻す
|
||||
if (ads_p != NULL)
|
||||
SetFileTime(hFile, NULL, NULL, &ftWrite);
|
||||
|
||||
CloseHandle(hFile); // チェックサム・ファイルを閉じる
|
||||
free(ascii_buf);
|
||||
free(text_buf);
|
||||
|
||||
printf("\nCreated successfully\n");
|
||||
return 0;
|
||||
}
|
||||
|
||||
// チェックサム・ファイルを読み込む
|
||||
static long read_checksum(char *ascii_buf, wchar_t *file_path)
|
||||
{
|
||||
unsigned char *data_buf;
|
||||
unsigned long err = 0, rv, file_size;
|
||||
HANDLE hFile;
|
||||
|
||||
// 読み込むファイルを開く
|
||||
hFile = CreateFile(checksum_file, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL);
|
||||
if (hFile == INVALID_HANDLE_VALUE){
|
||||
print_win32_err();
|
||||
printf("cannot open checksum file\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
// 2MB を越えるチェックサム・ファイルには対応しない
|
||||
file_size = GetFileSize(hFile, NULL);
|
||||
if ((file_size == INVALID_FILE_SIZE) || (file_size > (2 << 21))){
|
||||
CloseHandle(hFile);
|
||||
printf("checksum file is too large\n");
|
||||
return 1;
|
||||
}
|
||||
printf("Checksum Size\t: %d\n\n", file_size);
|
||||
fflush(stdout);
|
||||
|
||||
// ハッシュ値を記録するには最低でも 1(ファイル名)+1(スペース)+8(CRC-32) = 10バイト必要
|
||||
if (file_size < 10){
|
||||
CloseHandle(hFile);
|
||||
printf("Status : Damaged\n");
|
||||
printf("valid file is not found\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
// ファイル全体を読み込む
|
||||
data_buf = (unsigned char *)malloc(file_size + 2);
|
||||
if (data_buf == NULL){
|
||||
CloseHandle(hFile);
|
||||
printf("malloc, %d\n", file_size);
|
||||
return 1;
|
||||
}
|
||||
text_max = file_size + 1; // 全て ASCII 文字と仮定して、末尾に null 文字を付けた時の文字数
|
||||
text_buf = (wchar_t *)malloc(text_max * 2);
|
||||
if (text_buf == NULL){
|
||||
free(data_buf);
|
||||
CloseHandle(hFile);
|
||||
printf("malloc, %d\n", text_max * 2);
|
||||
return 1;
|
||||
}
|
||||
if (!ReadFile(hFile, data_buf, file_size, &rv, NULL) || (file_size != rv)){
|
||||
free(data_buf);
|
||||
free(text_buf);
|
||||
CloseHandle(hFile);
|
||||
printf("ReadFile, %d\n", file_size);
|
||||
return 1;
|
||||
}
|
||||
CloseHandle(hFile);
|
||||
data_buf[file_size ] = 0; // 末尾に null 文字を付けておく
|
||||
data_buf[file_size + 1] = 0;
|
||||
|
||||
// UTF-16 の文字列に変換する
|
||||
rv = utf16_to_utf16(data_buf, file_size + 2, text_buf);
|
||||
if (rv != 0){ // UTF-16 ではなかった
|
||||
if (check_utf8(data_buf) == 0){ // UTF-8 として変換する
|
||||
rv = 0; // BOMが付いてるかどうか
|
||||
if ((data_buf[0] == 0xEF) && (data_buf[1] == 0xBB) && (data_buf[2] == 0xBF))
|
||||
rv = 3;
|
||||
rv = cp_to_utf16(data_buf + rv, text_buf, file_size + 1, CP_UTF8);
|
||||
} else {
|
||||
rv = cp_to_utf16(data_buf, text_buf, file_size + 1, CP_ACP);
|
||||
if (rv != 0)
|
||||
rv = cp_to_utf16(data_buf, text_buf, file_size + 1, 1252); // Latin-1 CP1252 で変換を試みる
|
||||
}
|
||||
}
|
||||
free(data_buf);
|
||||
if (rv != 0){
|
||||
free(text_buf);
|
||||
printf("cannot decode text\n");
|
||||
return 1;
|
||||
}
|
||||
text_len = wcslen(text_buf); // 文字数には末尾の null 文字を含まないことに注意
|
||||
|
||||
// チェックサムを検証する
|
||||
rv = wcslen(checksum_file);
|
||||
if (_wcsicmp(checksum_file + (rv - 4), L".sfv") == 0){ // SFVファイル
|
||||
err = verify_sfv(ascii_buf, file_path);
|
||||
} else if (_wcsicmp(checksum_file + (rv - 4), L".md5") == 0){ // MD5ファイル
|
||||
err = verify_md5(ascii_buf, file_path);
|
||||
// } else {
|
||||
// err = 1;
|
||||
// printf("unknown format\n");
|
||||
}
|
||||
free(text_buf);
|
||||
close_ini_file();
|
||||
|
||||
// 検査結果を表示する
|
||||
printf("\n");
|
||||
if (err == 0){
|
||||
printf("All Files Complete\n");
|
||||
} else {
|
||||
if (err & 16)
|
||||
printf("Checksum File Incomplete\n");
|
||||
if (err & 0xFFFFFF00){
|
||||
printf("%d Files Missing or Damaged\n", err >> 8);
|
||||
//err &= 0xFF;
|
||||
}
|
||||
err = ((err & 0x10) << 4) | (err & 0xEF); // 16を 256 に変更する
|
||||
}
|
||||
|
||||
return err;
|
||||
}
|
||||
|
||||
// チェックサム・ファイルとソース・ファイルの名前に問題が無いか確かめる
|
||||
static long check_filename(
|
||||
wchar_t *list_buf,
|
||||
long list_len)
|
||||
{
|
||||
wchar_t *tmp_p;
|
||||
long num, list_off = 0;
|
||||
|
||||
// SFV ファイル作成時に、ソース・ファイルの先頭が「;」だとコメントと区別できない
|
||||
if (_wcsicmp(checksum_file + (wcslen(checksum_file) - 4), L".sfv") == 0){ // SFVファイル
|
||||
for (num = 0; num < file_num; num++){
|
||||
tmp_p = list_buf + list_off;
|
||||
while (list_buf[list_off] != 0)
|
||||
list_off++;
|
||||
list_off++;
|
||||
|
||||
// ファイル名の先頭を調べる
|
||||
if (tmp_p[0] == ';'){
|
||||
printf_cp("filename is invalid, %s\n", tmp_p);
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 基準ディレクトリ外にチェックサム・ファイルが存在するなら問題なし
|
||||
if (_wcsnicmp(checksum_file, base_dir, base_len) != 0)
|
||||
return 0;
|
||||
|
||||
// チェックサム・ファイルとソース・ファイルが同じ名前にならないようにする
|
||||
for (num = 0; num < file_num; num++){
|
||||
tmp_p = list_buf + list_off;
|
||||
while (list_buf[list_off] != 0)
|
||||
list_off++;
|
||||
list_off++;
|
||||
|
||||
// ファイル名との一致を調べる
|
||||
if (_wcsicmp(checksum_file + base_len, tmp_p) == 0){
|
||||
printf_cp("filename is invalid, %s\n", tmp_p);
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
wmain(long argc, wchar_t *argv[])
|
||||
{
|
||||
char ascii_buf[MAX_LEN * 3];
|
||||
wchar_t file_path[MAX_LEN], *tmp_p;
|
||||
long i, j, switch_set = 0;
|
||||
/*
|
||||
t = switch_set & 0x00000003
|
||||
fo= switch_set & 0x00000020
|
||||
*/
|
||||
printf("SFV/MD5 checker version " FILE_VERSION " by Yutaka Sawada\n\n");
|
||||
if (argc < 3){
|
||||
printf("Self-Test: ");
|
||||
i = test_checksum(file_path);
|
||||
if (i == 0){
|
||||
printf("Success");
|
||||
} else if (i == 2){
|
||||
printf("PE checksum is different");
|
||||
} else if (i == 3){
|
||||
printf("CRC-32 is different");
|
||||
} else {
|
||||
printf("Error\0thedummytext");
|
||||
}
|
||||
printf("\n\n");
|
||||
print_help();
|
||||
return 0;
|
||||
}
|
||||
|
||||
// 初期化
|
||||
checksum_file[0] = 0;
|
||||
base_dir[0] = 0;
|
||||
ini_path[0] = 0;
|
||||
file_num = 0;
|
||||
cp_output = GetConsoleOutputCP();
|
||||
|
||||
// コマンド
|
||||
switch (argv[1][0]){
|
||||
case 'c': // create
|
||||
case 'v': // verify
|
||||
break;
|
||||
default:
|
||||
print_help();
|
||||
return 0;
|
||||
}
|
||||
|
||||
// オプションとチェックサム・ファイルの指定
|
||||
for (i = 2; i < argc; i++){
|
||||
tmp_p = argv[i];
|
||||
// オプション
|
||||
if (((tmp_p[0] == '/') || (tmp_p[0] == '-')) && (tmp_p[0] == argv[2][0])){
|
||||
tmp_p++; // 先頭の識別文字をとばす
|
||||
// オプション
|
||||
if (wcscmp(tmp_p, L"u") == 0){
|
||||
cp_output = CP_UTF8;
|
||||
} else if (wcscmp(tmp_p, L"fo") == 0){
|
||||
switch_set |= 0x20;
|
||||
|
||||
// オプション (数値)
|
||||
} else if (wcsncmp(tmp_p, L"vs", 2) == 0){
|
||||
recent_data = 0;
|
||||
j = 2;
|
||||
while ((j < 2 + 2) && (tmp_p[j] >= '0') && (tmp_p[j] <= '9')){
|
||||
recent_data = (recent_data * 10) + (tmp_p[j] - '0');
|
||||
j++;
|
||||
}
|
||||
if ((recent_data == 8) || (recent_data > 15))
|
||||
recent_data = 0;
|
||||
} else if (wcsncmp(tmp_p, L"t", 1) == 0){
|
||||
j = -1;
|
||||
if ((tmp_p[1] >= '0') && (tmp_p[1] <= '2')) // 0～2 の範囲
|
||||
j = tmp_p[1] - '0';
|
||||
if (j != -1)
|
||||
switch_set |= j;
|
||||
|
||||
// オプション (文字列)
|
||||
} else if (wcsncmp(tmp_p, L"vd", 2) == 0){
|
||||
tmp_p += 2;
|
||||
j = copy_path_prefix(ini_path, MAX_LEN - INI_NAME_LEN - 1, tmp_p, NULL);
|
||||
if (j == 0){
|
||||
printf("save-directory is invalid\n");
|
||||
return 1;
|
||||
}
|
||||
if (ini_path[j - 1] != '\\'){ // 末尾が「\」でなければ付けておく
|
||||
ini_path[j ] = '\\';
|
||||
ini_path[j + 1] = 0;
|
||||
}
|
||||
//printf("Save Directory : %S\n", ini_path);
|
||||
j = GetFileAttributes(ini_path);
|
||||
if ((j == INVALID_FILE_ATTRIBUTES) || !(j & FILE_ATTRIBUTE_DIRECTORY)){
|
||||
printf("save-directory is invalid\n");
|
||||
return 1;
|
||||
}
|
||||
} else if (wcsncmp(tmp_p, L"d", 1) == 0){
|
||||
tmp_p++;
|
||||
j = copy_path_prefix(base_dir, MAX_LEN - 2, tmp_p, NULL); // 末尾に追加される分の余裕を見ておく
|
||||
if (j == 0){
|
||||
printf("base-directory is invalid\n");
|
||||
return 1;
|
||||
}
|
||||
if (base_dir[j - 1] != '\\'){ // 末尾が「\」でなければ付けておく
|
||||
base_dir[j ] = '\\';
|
||||
base_dir[j + 1] = 0;
|
||||
}
|
||||
j = GetFileAttributes(base_dir);
|
||||
if ((j == INVALID_FILE_ATTRIBUTES) || !(j & FILE_ATTRIBUTE_DIRECTORY)){
|
||||
printf("base-directory is invalid\n");
|
||||
return 1;
|
||||
}
|
||||
} else { // 未対応のオプション
|
||||
utf16_to_cp(tmp_p - 1, ascii_buf, MAX_LEN * 3, cp_output);
|
||||
printf("invalid option, %s\n", ascii_buf);
|
||||
return 1;
|
||||
}
|
||||
continue;
|
||||
}
|
||||
// オプション以外ならリカバリ・ファイル
|
||||
j = copy_path_prefix(checksum_file, MAX_LEN, tmp_p, NULL);
|
||||
if (j == 0){
|
||||
printf("checksum filename is invalid\n");
|
||||
return 1;
|
||||
}
|
||||
// 拡張子は SFV か MD5 のみ
|
||||
if (_wcsicmp(checksum_file + (j - 4), L".sfv") == 0){
|
||||
init_crc_table(); // CRC 計算用のテーブルを作る
|
||||
} else if (_wcsicmp(checksum_file + (j - 4), L".md5") == 0){
|
||||
if (recent_data != 0)
|
||||
init_crc_table();
|
||||
} else { // 拡張子が違うなら
|
||||
wcscpy(checksum_file + j, L".sfv"); // 標準で SFV 形式にする
|
||||
if (argv[1][0] == 'c'){ // 作成なら
|
||||
init_crc_table();
|
||||
} else { // 検査なら
|
||||
if (GetFileAttributes(checksum_file) == INVALID_FILE_ATTRIBUTES){
|
||||
wcscpy(checksum_file + j, L".md5"); // SFV で駄目なら MD5 にする
|
||||
if (GetFileAttributes(checksum_file) == INVALID_FILE_ATTRIBUTES){
|
||||
printf("file format is unknown\n");
|
||||
return 1;
|
||||
} else if (recent_data != 0){
|
||||
init_crc_table();
|
||||
}
|
||||
} else {
|
||||
init_crc_table();
|
||||
}
|
||||
}
|
||||
}
|
||||
if (argv[1][0] == 'c'){ // 作成なら
|
||||
// 指定されたファイル名が適切か調べる
|
||||
if (sanitize_filename(offset_file_name(checksum_file)) != 0){
|
||||
printf("checksum filename is invalid\n");
|
||||
return 1;
|
||||
}
|
||||
} else { // 検査なら
|
||||
j = GetFileAttributes(checksum_file);
|
||||
if ((j == INVALID_FILE_ATTRIBUTES) || (j & FILE_ATTRIBUTE_DIRECTORY)){
|
||||
wchar_t search_path[MAX_LEN];
|
||||
long name_len, dir_len, path_len, find_flag = 0;
|
||||
HANDLE hFind;
|
||||
WIN32_FIND_DATA FindData;
|
||||
if (wcspbrk(offset_file_name(checksum_file), L"*?") == NULL){
|
||||
// 「*」や「?」で検索しない場合、ファイルが見つからなければエラーにする
|
||||
printf("valid file is not found\n");
|
||||
return 1;
|
||||
}
|
||||
// 指定された拡張子で検索する
|
||||
path_len = wcslen(checksum_file);
|
||||
hFind = FindFirstFile(checksum_file, &FindData);
|
||||
if (hFind != INVALID_HANDLE_VALUE){
|
||||
get_base_dir(checksum_file, search_path);
|
||||
dir_len = wcslen(search_path);
|
||||
do {
|
||||
//printf("file name = %S\n", FindData.cFileName);
|
||||
name_len = wcslen(FindData.cFileName);
|
||||
if ((dir_len + name_len < MAX_LEN) && // ファイル名が長すぎない
|
||||
(_wcsicmp(FindData.cFileName + (name_len - 4), checksum_file + (path_len - 4)) == 0)){ // 拡張子が同じ
|
||||
find_flag = 1;
|
||||
break; // 見つけたファイル名で問題なし
|
||||
}
|
||||
} while (FindNextFile(hFind, &FindData)); // 次のファイルを検索する
|
||||
FindClose(hFind);
|
||||
}
|
||||
if (find_flag == 0){
|
||||
printf("valid file is not found\n");
|
||||
return 1;
|
||||
}
|
||||
wcscpy(checksum_file + dir_len, FindData.cFileName);
|
||||
}
|
||||
}
|
||||
break;
|
||||
}
|
||||
if (checksum_file[0] == 0){ // チェックサム・ファイルが指定されてないなら
|
||||
printf("checksum file is not specified\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
// input file の位置が指定されて無くて、
|
||||
// 最初のソース・ファイル指定に絶対パスや相対パスが含まれてるなら、それを使う
|
||||
if (argv[1][0] == 'c'){
|
||||
if ((base_dir[0] == 0) && (i + 1 < argc)){
|
||||
tmp_p = argv[i + 1];
|
||||
if (is_full_path(tmp_p) != 0){ // 絶対パスなら
|
||||
if (copy_path_prefix(file_path, MAX_LEN - 2, tmp_p, NULL) == 0){
|
||||
printf("base-directory is invalid\n");
|
||||
return 1;
|
||||
}
|
||||
get_base_dir(file_path, base_dir); // 最初のソース・ファイルの位置にする
|
||||
if (len_without_prefix(base_dir) == 0) // サブ・ディレクトリが無ければ
|
||||
base_dir[0] = 0;
|
||||
}
|
||||
}
|
||||
}
|
||||
if (base_dir[0] == 0) // input file の位置が指定されてないなら
|
||||
get_base_dir(checksum_file, base_dir); // リカバリ・ファイルの位置にする
|
||||
base_len = wcslen(base_dir);
|
||||
|
||||
// 動作環境の表示
|
||||
// 「\\?\」は常に付加されてるので表示する際には無視する
|
||||
printf_cp("Base Directory\t: \"%s\"\n", base_dir);
|
||||
printf_cp("Checksum File\t: \"%s\"\n", checksum_file);
|
||||
|
||||
if (argv[1][0] == 'c'){
|
||||
wchar_t *list_buf;
|
||||
long dir_len, list_len, list_max;
|
||||
__int64 total_size = 0; // 合計ファイル・サイズ
|
||||
// チェックサム・ファイル作成ならソース・ファイルのリストがいる
|
||||
i++;
|
||||
if (i >= argc){
|
||||
printf("input file is not specified\n");
|
||||
return 1;
|
||||
}
|
||||
// 入力ファイルの指定
|
||||
file_num = 0;
|
||||
list_len = 0;
|
||||
list_max = 0;
|
||||
list_buf = NULL;
|
||||
for (; i < argc; i++){
|
||||
// ファイルが基準ディレクトリ以下に存在することを確認する
|
||||
tmp_p = argv[i];
|
||||
j = copy_path_prefix(file_path, MAX_LEN - 2, tmp_p, base_dir); // 絶対パスにしてから比較する
|
||||
if (j == 0){
|
||||
free(list_buf);
|
||||
printf_cp("filename is invalid, %s\n", tmp_p);
|
||||
return 1;
|
||||
}
|
||||
if ((j <= base_len) || (_wcsnicmp(base_dir, file_path, base_len) != 0)){ // 基準ディレクトリ外なら
|
||||
free(list_buf);
|
||||
printf_cp("out of base-directory, %s\n", tmp_p);
|
||||
return 1;
|
||||
}
|
||||
//printf("%d = %S\n", i, argv[i]);
|
||||
//printf_cp("search = %s\n", file_path);
|
||||
// 「*」や「?」で検索しない場合、ファイルが見つからなければエラーにする
|
||||
j = -1;
|
||||
if (wcspbrk(file_path + base_len, L"*?") == NULL)
|
||||
j = file_num;
|
||||
// ファイルを検索する
|
||||
dir_len = wcslen(file_path) - 2; // ファイル名末尾の「\」を無視して、ディレクトリ部分の長さを求める
|
||||
while (file_path[dir_len] != '\\')
|
||||
dir_len--;
|
||||
dir_len++;
|
||||
list_buf = search_files(list_buf, file_path, dir_len, switch_set & 0x20, j, &list_max, &list_len, &total_size);
|
||||
if (list_buf == NULL)
|
||||
return 1;
|
||||
if ((j != -1) && (j == file_num)){ // ファイルが見つかったか確かめる
|
||||
free(list_buf);
|
||||
printf_cp("input file is not found, %s\n", file_path + base_len);
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
if (list_len == 0){
|
||||
if (list_buf)
|
||||
free(list_buf);
|
||||
printf("input file is not found\n");
|
||||
return 1;
|
||||
}
|
||||
if (check_filename(list_buf, list_len)){
|
||||
free(list_buf);
|
||||
return 1;
|
||||
}
|
||||
/*{
|
||||
FILE *fp;
|
||||
fp = fopen("list_buf.txt", "wb");
|
||||
fwrite(list_buf, 2, list_len, fp);
|
||||
fclose(fp);
|
||||
}*/
|
||||
|
||||
// ファイル・リストの内容を並び替える (リストの末尾は null 1個にすること)
|
||||
sort_list(list_buf, list_len);
|
||||
|
||||
printf("\nInput File count\t: %d\n", file_num);
|
||||
printf("Input File total size\t: %I64d\n", total_size);
|
||||
i = write_checksum(file_path, list_buf, list_len, total_size, switch_set & 3);
|
||||
|
||||
} else {
|
||||
// 検査結果ファイルの位置が指定されてないなら
|
||||
if ((recent_data != 0) && (ini_path[0] == 0)){
|
||||
// 実行ファイルのディレクトリにする
|
||||
j = GetModuleFileName(NULL, ini_path, MAX_LEN);
|
||||
if ((j == 0) || (j >= MAX_LEN)){
|
||||
printf("GetModuleFileName\n");
|
||||
return 1;
|
||||
}
|
||||
while (j > 0){
|
||||
if (ini_path[j - 1] == '\\'){
|
||||
ini_path[j] = 0;
|
||||
break;
|
||||
}
|
||||
j--;
|
||||
}
|
||||
if (j >= MAX_LEN - INI_NAME_LEN - 1){
|
||||
printf("save-directory is invalid\n");
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
|
||||
i = read_checksum(ascii_buf, file_path);
|
||||
}
|
||||
|
||||
//printf("ExitCode: %d\n", i);
|
||||
return i;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user