//===========================================================================
//スレッド実行
//===========================================================================
void __fastcall TAttacheCaseFileDecrypt1::Execute()
{
int i, c, len;
float ProgressPercentNumF; //進捗パーセンテージ（浮動小数点）
// バッファ
char buffer[BUF_SIZE];
char source_buffer[BUF_SIZE];
char temp_buffer[BUF_SIZE];
char output_buffer[LARGE_BUF_SIZE];
char *headerbuffer;
//パスワード
bool fPasswordOk;
String FilePath, FileName;
// ファイルストリーム
TFileStream *fsIn;
TFileStream *fsOut;
bool fInputFileOpen = false;
bool fOutputFileOpen = false;
float free_space_required;
__int64 CurrentPos, TotalSize;
__int64 CurrentDriveFreeSpaceSize;
//処理する合計サイズ
AllTotalSize = 0;
int ret; //バッファ出力の返値
int FileIndex = 0;
//Blowfish
CBlowFish *bf;
char token[17] = {0, };
const String PrefixString = "Fn_";
const char charTokenString[17] = "_AttacheCaseData"; //復号の正否に使う
String AtcFileTokenString; //暗号化ファイルのトークン（文字列）
String AtcFileCreateDateString; //暗号化ファイルの生成日時（文字列）
//同名ファイル/フォルダーはすべて上書きして復号する
//（ユーザーがダイアログで「すべてはい」を選択したとき = true ）
fOverwirteYesToAll = false;
//「復号したファイルを関連付けされたソフトで開く」一時的な設定
fTempOpenFile = fOpenFile;
//フォルダーを一度開いたか（深いフォルダーすべてを開かないように）
fOpenFolderOnce = false;
// 出力するディレクトリ
OutDirPath = IncludeTrailingPathDelimiter(OutDirPath);
String TempRelativePath;
// 暗号化部分のヘッダサイズ
int EncryptHeaderSize = 0;
int DataVersion; // ver.2.00～は "5", ver.2.70～は "6"
int flush, status; // zlib
z_stream z; // zlibライブラリとやりとりするための構造体
bool fInputEnd = false; // 入力ストリームの終了
//ヘッダデータから必要情報を取り出すための
TMemoryStream *pms; // メモリストリーム
int idx;
TStringList *DataList;
TStringList *tsv;
TStringList *FileList = new TStringList(); // 0: ファイル名
__int64 *FileSizeList = 0; // 1: ファイルサイズ（フォルダは-1）
int *FileAttrList = 0; // 2: 属性
int *FileAgeList = 0; // 3: タイムスタンプ
int rest;
int buf_size;
//----------------------------------------------------------------------
// 平文のヘッダ内容チェック
try {
fsIn = new TFileStream(AtcFilePath, fmOpenRead | fmShareDenyNone);
}
catch(...) {
//'復号するファイルを開けません。他のアプリケーションが使用中の可能性があります。'
MsgText = LoadResourceString(&Msgdecrypt::_MSG_ERROR_FILE_OPEN);
MsgType = mtError;
MsgButtons = TMsgDlgButtons() << mbOK;
MsgDefaultButton = mbOK;
Synchronize(&PostConfirmMessageForm);
goto LabelError;
}
fInputFileOpen = true;
// 暗号部ヘッダサイズを取得
fsIn->Read(&EncryptHeaderSize, sizeof(int));
// トークンを取得
fsIn->Read(token, 16);
if (StrComp(token, charTokenString) != 0 ) {
//--------------------------------------------------------
//実は自己実行形式ファイル？（拡張子偽装されている場合も）
//--------------------------------------------------------
// サイズを再取得
fsIn->Seek(-(__int64)sizeof(__int64), TSeekOrigin::soEnd);
fsIn->Read(&AllTotalSize, sizeof(__int64));
// 位置を戻す
fsIn->Seek(-(AllTotalSize + sizeof(__int64)), TSeekOrigin::soEnd);
// もう一度、暗号部ヘッダサイズを取得
fsIn->Read(&EncryptHeaderSize, sizeof(int));
// もう一度、トークンを取得
fsIn->Read(token, 16);
// トークンを再チェック
if (StrComp(token, charTokenString) != 0 ) {
// '暗号化ファイルではありません。アタッシェケースによって暗号化されたファイルでないか、'+#13+
// 'ファイルが壊れている可能性があります。'
MsgText = LoadResourceString(&Msgdecrypt::_MSG_ERROR_FILE_NOT_ATC);
MsgType = mtError;
MsgButtons = TMsgDlgButtons() << mbOK;
MsgDefaultButton = mbOK;
Synchronize(&PostConfirmMessageForm);
goto LabelError;
}
}
else{
AllTotalSize = fsIn->Size;
}
//-----------------------------------
// データバージョンチェック
//-----------------------------------
DataVersion = -1;
fsIn->Read(&DataVersion, sizeof(int));
if (DataVersion <= 103) {
// Blowfishで暗号化されたファイル
}
else{
//'バージョンがちがいます。復号できません。'+#13+
//'ファイルはver.1のアタッシェケースで暗号化されていません。';
MsgText = LoadResourceString(&Msgdecrypt::_MSG_ERROR_NOT_BLOWFISH_ENCRYPTION);
MsgType = mtError;
MsgButtons = TMsgDlgButtons() << mbOK;
MsgDefaultButton = mbOK;
Synchronize(&PostConfirmMessageForm);
goto LabelError;
}
//-----------------------------------
// 復号の準備
//-----------------------------------
bf = new CBlowFish;
bf->Initialize( KeyString.c_str(), KeyString.Length() ); //初期化
//-----------------------------------
// 暗号部ヘッダの復号（ECBモード）
//-----------------------------------
pms = new TMemoryStream;
len = 0;
while (len < EncryptHeaderSize) {
// 読み出しバッファ
for (c = 0; c < BUF_SIZE; c++) {
source_buffer[c] = 0;
}
// 暗号化されたデータブロックの読み出し
len += fsIn->Read(source_buffer, BUF_SIZE);
// 復号処理
bf->Decode( source_buffer, buffer, BUF_SIZE);
pms->Write(buffer, BUF_SIZE);
}
pms->Seek((__int64)0, TSeekOrigin::soBeginning); //ポインタを先頭へ戻す
DataList = new TStringList;
DataList->LoadFromStream(pms, TEncoding::GetEncoding(932)); // shift-jis
//-----------------------------------
// 復号正否（復号できたか）
//-----------------------------------
if (DataList->Count == 0 || DataList->Strings[0].Pos("AttacheCase") == 0) {
fPasswordOk = false;
}
else{
fPasswordOk = true; //パスワード合致
}
if ( fPasswordOk == false ) {
//'パスワードがちがいます。復号できません。'+#13+
//'場合によってはファイルが壊れている可能性もあります。';
MsgText = LoadResourceString(&Msgdecrypt::_MSG_ERROR_PASSWORD_WRONG);
MsgType = mtError;
MsgButtons = TMsgDlgButtons() << mbOK;
MsgDefaultButton = mbOK;
Synchronize(&PostConfirmMessageForm);
delete DataList;
goto LabelTypeMiss;
}
//-----------------------------------
// 復号時のエンコーディング判定
//-----------------------------------
pms->Position = 0;
DataList->LoadFromStream(pms, TEncoding::GetEncoding(932)); // shift-jis
//===================================
// デバッグ
//ShowMessage(DataList->Text);
//===================================
delete pms;
//-----------------------------------
// 暗号化ファイルの生成日時
//-----------------------------------
//※特に今は使用していないが、将来的に
// 期限付きでファイルを復号できなくなる
// などの機能を追加しても良いかも。
//-----------------------------------
AtcFileCreateDateString = DataList->Strings[1];
//-----------------------------------
// ヘッダデータからファイルリストや
// ファイル情報などを各変数を動的確保
//-----------------------------------
FileSizeList = new __int64[DataList->Count]; // 1: ファイルサイズ（フォルダは-1）
FileAttrList = new int[DataList->Count]; // 2: 属性
FileAgeList = new int[DataList->Count]; // 3: タイムスタンプ
DataList->NameValueSeparator = ':';
tsv = new TStringList;
tsv->Delimiter = '\t';
tsv->StrictDelimiter = true;
char lpPath[MAX_PATH];
for (i = 0; i < DataList->Count; i++) {
idx = DataList->IndexOfName(PrefixString+IntToStr(i));
if (idx > 0) {
tsv->DelimitedText = DataList->ValueFromIndex[idx];
// ディレクトリ・トラバーサル対策（ver.2.8.5.0～）
bool fDirectoryTraversal = false;
AnsiString CanonicalizePath = AnsiString(OutDirPath + tsv->Strings[0]);
if (CanonicalizePath.Length() < MAX_PATH) {
// ファイルパスを正規化
if (PathCanonicalize(lpPath, CanonicalizePath.c_str()) == true) {
// 正規化したパスが保存先と一致するか
if (AnsiString(lpPath).Pos(OutDirPath) != 1 ){
fDirectoryTraversal = true;
}
}
else{
fDirectoryTraversal = true;
}
}
else{
fDirectoryTraversal = true;
}
if (fDirectoryTraversal == true) {
//'不正なファイルパスが含まれています。復号できません。';
MsgText = LoadResourceString(&Msgdecrypt::_MSG_ERROR_INVALID_FILE_PATH) +
"\n" + CanonicalizePath;
MsgType = mtError;
MsgButtons = TMsgDlgButtons() << mbOK;
MsgDefaultButton = mbOK;
Synchronize(&PostConfirmMessageForm);
delete DataList;
goto LabelTypeMiss;
}
FileList->Add(tsv->Strings[0]); // 0: ファイルパス
FileSizeList[i] = StrToIntDef(tsv->Strings[1], -1); // 1: ファイルサイズ（フォルダは-1）
FileAttrList[i] = StrToIntDef(tsv->Strings[2], -1); // 2: 属性
FileAgeList[i] = StrToIntDef(tsv->Strings[3], -1); // 3: タイムスタンプ
}
}
delete tsv;
delete DataList;
//-----------------------------------
//ディスクの空き容量チェック
//-----------------------------------
CurrentDriveFreeSpaceSize = GetDiskFreeSpaceNum(OutDirPath);
if (CurrentDriveFreeSpaceSize > -1) {
if ( AllTotalSize > CurrentDriveFreeSpaceSize ) {
//'復号する先のドライブの空き容量が足りません。'+#13+
//'復号するには、約%dMBの空き容量が必要です。復号処理を中止します。';
MsgText = LoadResourceString(&Msgdecrypt::_MSG_ERROR_DRIVE_NO_FREE_SPACE);
free_space_required = (float)AllTotalSize/1024/1024; // MB
MsgText = String().sprintf(MsgText.c_str(), free_space_required);
MsgType = mtError;
MsgButtons = TMsgDlgButtons() << mbOK;
MsgDefaultButton = mbOK;
Synchronize(&PostConfirmMessageForm);
goto LabelError;
}
}
else{
// OK!
//
}
//-----------------------------------
//複数フォルダ/ファイルを開こうとしているので確認
//-----------------------------------
//復号したファイルを関連付けされたソフトで開くか
if ( fTempOpenFile == true && FileList->Count > 4) {
//'5つ以上のファイルを復号後に開こうとしていますが、処理を続けますか？'+#13+
//'「いいえ」を選択すると、開かず復号します。';
MsgText = LoadResourceString(&Msgdecrypt::_MSG_CONFIRM_OPEN_DECRYPTED_FILES);
MsgType = mtConfirmation;
MsgButtons = TMsgDlgButtons() << mbYes << mbNo << mbCancel;
MsgDefaultButton = mbCancel;
Synchronize(&PostConfirmMessageForm);
if ( MsgReturnVal == mrCancel ) {
goto LabelStop;
}
else if ( MsgReturnVal == mrNo ) {
fTempOpenFile = true;
}
}
//-----------------------------------
// 復号開始
//-----------------------------------
//'復号しています...'
ProgressStatusText = LoadResourceString(&Msgdecrypt::_LABEL_STATUS_TITLE_DECRYPTING);
ProgressMsgText = ExtractFileName(AtcFilePath);
// ファイル（データ本体）サイズを取得する
AllTotalSize = fsIn->Size - fsIn->Position + 1;
// zlib 前準備
// 圧縮においてすべてのメモリ管理をライブラリに任せる
z.zalloc = Z_NULL;
z.zfree = Z_NULL;
z.opaque = Z_NULL;
if (inflateInit(&z) != Z_OK) {
// エラー
// zlibエラーは最後でまとめてメッセージ処理
goto LabelError;
}
// 通常は deflate() の第2引数は Z_NO_FLUSH にして呼び出す
flush = Z_NO_FLUSH;
// 入出力バッファ
for (i = 0; i < LARGE_BUF_SIZE; i++) {
output_buffer[i] = 0;
}
z.avail_in = 0; // 入力バッファ中のデータのバイト数
z.next_in = Z_NULL; // 入力バッファ中のデータのバイト数
z.next_out = output_buffer; // 出力バッファ残量
z.avail_out = LARGE_BUF_SIZE; // 出力ポインタ（展開するので大きめに）
status = Z_OK;
while (Terminated == false) {
//-----------------------------------
// 入力
//-----------------------------------
if (z.avail_in == 0) {
TotalSize = InputBuffer(bf, len, source_buffer, fsIn, fInputFileOpen, TotalSize);
z.avail_in = len;
z.next_in = source_buffer;
if ( len == 0 ) { //入力ストリーム終了
fInputEnd = true;
}
}
//-----------------------------------
// 展開
//-----------------------------------
status = inflate(&z, flush);
//-----------------------------------
// 処理ステータス
//-----------------------------------
if ( status == Z_OK ){
if ( z.avail_out == 0 ) {
ret = OutputBuffer(output_buffer, LARGE_BUF_SIZE,
fsOut, fOutputFileOpen,
FileList, FileIndex,
FileSizeList, FileAttrList, FileAgeList);
if ( ret == 0) {
z.next_out = output_buffer;
z.avail_out = LARGE_BUF_SIZE;
}
else if (ret == -1) {
goto LabelError;
}
else{
goto LabelStop;
}
}//end if (z.avail_out == 0);
}
//-----------------------------------
// バッファエラー
//-----------------------------------
else if ( status == Z_BUF_ERROR ) { //出力バッファがいっぱいの可能性
//出力バッファをクリアにして継続させる
len = LARGE_BUF_SIZE - z.avail_out;
ret = OutputBuffer(output_buffer, len,
fsOut, fOutputFileOpen,
FileList, FileIndex,
FileSizeList, FileAttrList, FileAgeList);
if (ret == 0) {
z.next_out = output_buffer;
z.avail_out = LARGE_BUF_SIZE;
}
else if ( ret == -1) {
goto LabelError;
}
else{
goto LabelStop;
}
}
//-----------------------------------
// 終了
//-----------------------------------
else if (status == Z_STREAM_END) {
break;
}
//-----------------------------------
// エラー
//-----------------------------------
else{
// #define Z_OK 0
// #define Z_STREAM_END 1
// #define Z_NEED_DICT 2
// #define Z_ERRNO (-1)
// #define Z_STREAM_ERROR (-2)
// #define Z_DATA_ERROR (-3)
// #define Z_MEM_ERROR (-4)
// #define Z_BUF_ERROR (-5)
// #define Z_VERSION_ERROR (-6)
goto LabelError;
}
//-----------------------------------
//進捗状況表示
ProgressPercentNumF = (float)TotalSize/AllTotalSize;
ProgressPercentNum = (int)(ProgressPercentNumF*100);
if (AllTotalSize < 104857600) { // 100MB未満
ProgressPercentNumText = IntToStr(ProgressPercentNum)+"%";
}
else{
ProgressPercentNumText = FloatToStrF(ProgressPercentNumF*100, ffNumber, 4, 1)+"%";
}
if ( fInputEnd == true) {
break;
}
}//while (!Terminated);
if (Terminated == true) {
//ユーザーキャンセルで抜けてきた
goto LabelStop;
}
//----------------------------------------------------------------------
// 万が一、出力バッファに余りがある場合
//----------------------------------------------------------------------
len = LARGE_BUF_SIZE - z.avail_out;
ret = OutputBuffer(output_buffer, len,
fsOut, fOutputFileOpen,
FileList, FileIndex,
FileSizeList, FileAttrList, FileAgeList);
if ( ret == 0 ) {
}
else if ( ret == -1 ){
goto LabelError;
}
else{
goto LabelStop;
}
if (inflateEnd(&z) != Z_OK) {
goto LabelError;
}
ProgressPercentNum = 100;
//'完了'
ProgressStatusText = LoadResourceString(&Msgdecrypt::_LABEL_STATUS_TITLE_COMPLETE);
ProgressMsgText = ExtractFileName(AtcFilePath);
if ( fInputFileOpen == true ) {
delete fsIn;
fInputFileOpen = false;
}
if ( fOutputFileOpen == true ) {
delete fsOut;
fOutputFileOpen = false;
}
delete FileList;
delete [] FileSizeList; // 1: ファイルサイズ（フォルダは-1）
delete [] FileAttrList; // 2: 属性
delete [] FileAgeList; // 3: 更新日
//復号成功
StatusNum = 1;
return;
//-----------------------------------
//パスワード入力ミスの後始末
//-----------------------------------
LabelTypeMiss:
ProgressPercentNum = 0;
//'エラー'
ProgressStatusText = LoadResourceString(&Msgdecrypt::_LABEL_STATUS_TITLE_ERROR);
//'復号に失敗しました。'
ProgressMsgText = LoadResourceString(&Msgdecrypt::_LABEL_STATUS_DETAIL_FAILED);
if ( fInputFileOpen == true ) {
delete fsIn;
fInputFileOpen = false;
}
if ( fOutputFileOpen == true ) {
delete fsOut;
fOutputFileOpen = false;
}
delete FileList;
delete [] FileSizeList;
delete [] FileAttrList;
delete [] FileAgeList;
StatusNum = -1;
return;
//-----------------------------------
//エラーの後始末
//-----------------------------------
LabelError:
ProgressPercentNum = 0;
if ( status < 0 ){
//'zlibライブラリからエラーを返されました。'
//'エラー番号：'
MsgText = LoadResourceString(&Msgdecrypt::_MSG_ERROR_ZLIB) + IntToStr(status) + "\n" + z.msg;
MsgType = mtError;
MsgButtons = TMsgDlgButtons() << mbOK;
MsgDefaultButton = mbOK;
Synchronize(&PostConfirmMessageForm);
}
//'エラー'
ProgressStatusText = LoadResourceString(&Msgdecrypt::_LABEL_STATUS_TITLE_ERROR);
//'復号に失敗しました。'
ProgressMsgText = LoadResourceString(&Msgdecrypt::_LABEL_STATUS_DETAIL_FAILED);
if ( fInputFileOpen == true ) {
delete fsIn;
fInputFileOpen = false;
}
if ( fOutputFileOpen == true ) {
delete fsOut;
fOutputFileOpen = false;
}
delete FileList;
delete [] FileSizeList;
delete [] FileAttrList;
delete [] FileAgeList;
StatusNum = -2;
return;
//-----------------------------------
//ユーザーキャンセルの後始末
//-----------------------------------
LabelStop:
ProgressPercentNum = 0;
//'キャンセル'
ProgressStatusText = LoadResourceString(&Msgdecrypt::_LABEL_STATUS_TITLE_USER_CANCEL);
//'復号が中止されました。'
ProgressMsgText = LoadResourceString(&Msgdecrypt::_LABEL_STATUS_DETAIL_STOPPED);
if ( fInputFileOpen == true ) {
delete fsIn;
fInputFileOpen = false;
}
if ( fOutputFileOpen == true ) {
delete fsOut;
fOutputFileOpen = false;
}
delete FileList;
delete [] FileSizeList;
delete [] FileAttrList;
delete [] FileAgeList;
StatusNum = 0;
return;
}
//===========================================================================
//スレッド実行
//===========================================================================
void __fastcall TAttacheCaseFileDecrypt2::Execute()
{
int i, c, len;
float ProgressPercentNumF; //進捗パーセンテージ（浮動小数点）
// バッファ
char source_buffer[BUF_SIZE];
char temp_buffer[BUF_SIZE];
char chain_buffer[BUF_SIZE];
char output_buffer[LARGE_BUF_SIZE];
char *headerbuffer;
//パスワード
bool fPasswordOk;
//const int KeyArrayNum = sizeof(key)/sizeof(key[0]);
String FilePath, FileName;
// ファイルストリーム
TFileStream *fsIn;
TFileStream *fsOut;
bool fInputFileOpen = false;
bool fOutputFileOpen = false;
float free_space_required;
__int64 CurrentPos, TotalSize;
__int64 CurrentDriveFreeSpaceSize;
//処理する合計サイズ
AllTotalSize = 0;
int ret; //バッファ出力の返値
int FileIndex = 0;
char token[17] = {0, };
const char charTokenString[17] = "_AttacheCaseData"; //復号の正否に使う
const char charBrokenToken[17] = "_Atc_Broken_Data"; //ファイルが破壊されていることを示すトークン
String AtcFileTokenString; //暗号化ファイルのトークン（文字列）
String AtcFileCreateDateString; //暗号化ファイルの生成日時（文字列）
//「復号したファイルを関連付けされたソフトで開く」一時的な設定
fTempOpenFile = fOpenFile;
//フォルダーを一度開いたか（深いフォルダーすべてを開かないように）
fOpenFolderOnce = false;
// 出力するディレクトリ
OutDirPath = IncludeTrailingPathDelimiter(OutDirPath);
String TempRelativePath;
// 平文ヘッダサイズ（データサブバージョン、破壊設定など）
int PlainHeaderSize = 0;
// 暗号化部分のヘッダサイズ
int EncryptHeaderSize = 0;
int DataVersion; // ver.2.00～は "5", ver.2.70～は "6"
int AlgorismType;
char reserved; // 0
//int TypeLimits; // ミスタイプ回数 0～10 （グローバル:public宣言とした）
//bool fDestroy; // 破壊するか否か 0 or 1 （グローバル:public宣言とした）
String PrefixString; //ファイルリストの接頭辞（Fn_*, U_*）
int flush, status; // zlib
z_stream z; // zlibライブラリとやりとりするための構造体
bool fInputEnd = false; // 入力ストリームの終了
//ヘッダデータから必要情報を取り出すための
TMemoryStream *pms; // メモリストリーム
int idx;
TStringList *DataList;
TStringList *tsv;
TStringList *FileList = new TStringList(); // 0: ファイル名
__int64 *FileSizeList = 0; // 1: ファイルサイズ（フォルダは-1）
int *FileAttrList = 0; // 2: 属性
int *FileDtChangeList = 0; // 3: 更新日
int *FileTmChangeList = 0; // 4: 更新時
int *FileDtCreateList = 0; // 5: 作成日
int *FileTmCreateList = 0; // 6: 作成時
int rest;
int buf_size;
//----------------------------------------------------------------------
// 平文のヘッダ内容チェック
try {
fsIn = new TFileStream(AtcFilePath, fmOpenRead | fmShareDenyNone);
}
catch(...) {
//'復号するファイルを開けません。他のアプリケーションが使用中の可能性があります。'
MsgText = LoadResourceString(&Msgdecrypt::_MSG_ERROR_FILE_OPEN);
MsgType = mtError;
MsgButtons = TMsgDlgButtons() << mbOK;
MsgDefaultButton = mbOK;
Synchronize(&PostConfirmMessageForm);
goto LabelError;
}
fInputFileOpen = true;
// 平文ヘッダサイズを取得
fsIn->Read(&PlainHeaderSize, sizeof(int));
// トークンを取得
fsIn->Read(token, 16);
if (StrComp(token, charTokenString) != 0 ) {
//--------------------------------------------------------
//実は自己実行形式ファイル？（拡張子偽装されている場合も）
//--------------------------------------------------------
// サイズを再取得
fsIn->Seek(-(__int64)sizeof(__int64), TSeekOrigin::soEnd);
fsIn->Read(&AllTotalSize, sizeof(__int64));
// 位置を戻す
fsIn->Seek(-(AllTotalSize + sizeof(__int64)), TSeekOrigin::soEnd);
// もう一度、平文ヘッダサイズを読み込む
fsIn->Read(&PlainHeaderSize, sizeof(int));
// もう一度、トークンを取得
fsIn->Read(token, 16);
// トークンを再チェック
if (StrComp(token, charTokenString) != 0 ) {
if ( StrComp(token, charBrokenToken) == 0 ) {
//'復号するファイルを開けません。この暗号化ファイルは破壊されています。'
MsgText = LoadResourceString(&Msgdecrypt::_MSG_ERROR_FILE_BROKEN);
MsgType = mtError;
MsgButtons = TMsgDlgButtons() << mbOK;
MsgDefaultButton = mbOK;
Synchronize(&PostConfirmMessageForm);
}
else {
// '暗号化ファイルではありません。アタッシェケースによって暗号化されたファイルでないか、'+#13+
// 'ファイルが壊れている可能性があります。'
MsgText = LoadResourceString(&Msgdecrypt::_MSG_ERROR_FILE_NOT_ATC);
MsgType = mtError;
MsgButtons = TMsgDlgButtons() << mbOK;
MsgDefaultButton = mbOK;
Synchronize(&PostConfirmMessageForm);
}
goto LabelError;
}
}
//-----------------------------------
// データバージョンチェック
//-----------------------------------
DataVersion = -1;
fsIn->Read(&DataVersion, sizeof(int));
if (DataVersion > ATC_DATA_FILE_VERSION && DataVersion < 200) {
//'データバージョンがちがいます。復号できません。'+#13+
//'ファイルは上位バージョンのアタッシェケースで暗号化されています。'+#13+
//'最新版を入手して復号を試みてください。';
MsgText = LoadResourceString(&Msgdecrypt::_MSG_ERROR_FILE_HIGHER_VERSION);
MsgType = mtError;
MsgButtons = TMsgDlgButtons() << mbOK;
MsgDefaultButton = mbOK;
Synchronize(&PostConfirmMessageForm);
goto LabelError;
}
else if (DataVersion <= 103) {
// Blowfishで暗号化されたファイル
// 正式版では正常に復号されるが、ここではエラーとする
//'データバージョンがちがいます。復号できません。'+#13+
//'ファイルはver.1のアタッシェケースで暗号化されています。'+#13+
//'古いバージョン（ver.2.75以前）か正式版を入手して復号を試みてください。';
MsgText = LoadResourceString(&Msgdecrypt::_MSG_ERROR_FILE_LOWER_VERSION);
MsgType = mtError;
MsgButtons = TMsgDlgButtons() << mbOK;
MsgDefaultButton = mbOK;
Synchronize(&PostConfirmMessageForm);
goto LabelError;
}
else {
// 104 ～
// Rijndaelで暗号化されている
// アルゴリズムタイプ
fsIn->Read(&AlgorismType, sizeof(int));
// ヘッダサイズを読み込む
fsIn->Read(&EncryptHeaderSize, sizeof(int));
// int型からポインタキャストでchar配列を取り出す
headerbuffer = (char*) & PlainHeaderSize;
// データサブバージョンチェック（ver.2.70～）
if (headerbuffer[0] >= 6) {
TypeLimits = (int)headerbuffer[2];
fDestroy = (bool)headerbuffer[3];
// 有効範囲（1～10）かチェック
if (TypeLimits < 1 || 10 < TypeLimits) {
TypeLimits = 3; // デフォルト
}
}
else { // headerbuffer[0] == 5 //旧バージョン
TypeLimits = 3;
fDestroy = false;
}
}
//-----------------------------------
// 復号の準備
//-----------------------------------
// テーブル生成
gentables();
// パスワードのセット
gkey(8, 8, key);
//-----------------------------------
// 暗号部ヘッダの復号
//-----------------------------------
pms = new TMemoryStream;
// IVの読み込み
fsIn->Read(chain_buffer, BUF_SIZE);
len = 0;
while (len < EncryptHeaderSize) {
// 読み出しバッファ
for (c = 0; c < BUF_SIZE; c++) {
source_buffer[c] = 0;
}
// 暗号化されたデータブロックの読み出し
len += fsIn->Read(source_buffer, BUF_SIZE);
for (c = 0; c < BUF_SIZE; c++) {
// あとのxorのためによけておく
temp_buffer[c] = source_buffer[c];
}
// 復号処理
rijndael_decrypt(source_buffer);
// xor
for (c = 0; c < BUF_SIZE; c++) {
source_buffer[c] ^= chain_buffer[c];
chain_buffer[c] = temp_buffer[c]; // CBC
}
pms->Write(source_buffer, BUF_SIZE);
}
pms->Seek((__int64)0, TSeekOrigin::soBeginning); //ポインタを先頭へ戻す
DataList = new TStringList;
DataList->LoadFromStream(pms, TEncoding::GetEncoding(932)); // shift-jis
//-----------------------------------
// 復号正否（復号できたか）
//-----------------------------------
//MsgText = DataList->Strings[0];
//MsgType = mtError;
//MsgButtons = TMsgDlgButtons() << mbOK;
//MsgDefaultButton = mbOK;
//Synchronize(&PostConfirmMessageForm);
if (DataList->Count == 0 || DataList->Strings[0].Pos("AttacheCase") == 0) {
fPasswordOk = false;
}
else{
fPasswordOk = true; //パスワード合致
}
if ( fPasswordOk == false ) {
//'パスワードがちがいます。復号できません。'+#13+
//'場合によってはファイルが壊れている可能性もあります。';
MsgText = LoadResourceString(&Msgdecrypt::_MSG_ERROR_PASSWORD_WRONG);
if ( fCompare == true ) {
//メッセージに'コンペアに失敗しました。'を追加
MsgText += "\n" + LoadResourceString(&Msgdecrypt::_MSG_ERROR_COMPARE_FILE);
}
MsgType = mtError;
MsgButtons = TMsgDlgButtons() << mbOK;
MsgDefaultButton = mbOK;
Synchronize(&PostConfirmMessageForm);
delete DataList;
goto LabelTypeMiss;
}
//-----------------------------------
// 復号時のエンコーディング判定
//-----------------------------------
pms->Position = 0;
//DataList->LoadFromStream(pms, TEncoding::UTF8);
DataList->LoadFromStream(pms, TEncoding::GetEncoding(65001));
PrefixString = "Fn_";
for (i = 0; i < DataList->Count; i++) {
if ( DataList->Strings[i].Pos("U_0") == 1){
PrefixString = "U_"; //新バージョン（ver.2.8.0～）で暗号化されているようだ
break;
}
}
// Unicodeではないので従来のShift-JISで再度読み直し
if (PrefixString == "Fn_") {
pms->Position = 0;
DataList->LoadFromStream(pms, TEncoding::GetEncoding(932));
}
//===================================
// デバッグ
//ShowMessage(DataList->Text);
//===================================
#ifdef _DEBUG
DataList->SaveToFile(OutDirPath + ExtractFileName(AtcFilePath) + ".txt");
#endif
delete pms;
//-----------------------------------
// 暗号化ファイルの生成日時
//-----------------------------------
//※特に今は使用していないが、将来的に
// 期限付きでファイルを復号できなくなる
// などの機能を追加しても良いかも。
//-----------------------------------
AtcFileCreateDateString = DataList->Strings[1];
//-----------------------------------
// ヘッダデータからファイルリストや
// ファイル情報などを各変数を動的確保
//-----------------------------------
FileSizeList = new __int64[DataList->Count]; // 1: ファイルサイズ（フォルダは-1）
FileAttrList = new int[DataList->Count]; // 2: 属性
FileDtChangeList = new int[DataList->Count]; // 3: 更新日
FileTmChangeList = new int[DataList->Count]; // 4: 更新時
FileDtCreateList = new int[DataList->Count]; // 5: 作成日
FileTmCreateList = new int[DataList->Count]; // 6: 作成時
DataList->NameValueSeparator = ':';
tsv = new TStringList;
tsv->Delimiter = '\t';
tsv->StrictDelimiter = true;
char lpPath[MAX_PATH];
for (i = 0; i < DataList->Count; i++) {
idx = DataList->IndexOfName(PrefixString+IntToStr(i));
if (idx > 0) {
tsv->DelimitedText = DataList->ValueFromIndex[idx];
// ディレクトリ・トラバーサル対策（ver.2.8.5.0～）
bool fDirectoryTraversal = false;
AnsiString CanonicalizePath = AnsiString(OutDirPath + tsv->Strings[0]);
if (CanonicalizePath.Length() < MAX_PATH) {
// ファイルパスを正規化
if (PathCanonicalize(lpPath, CanonicalizePath.c_str()) == true) {
// 正規化したパスが保存先と一致するか
if (AnsiString(lpPath).Pos(OutDirPath) != 1 ){
fDirectoryTraversal = true;
}
}
else{
fDirectoryTraversal = true;
}
}
else{
fDirectoryTraversal = true;
}
if (fDirectoryTraversal == true) {
//'不正なファイルパスが含まれています。復号できません。';
MsgText = LoadResourceString(&Msgdecrypt::_MSG_ERROR_INVALID_FILE_PATH) +
"\n" + CanonicalizePath;
MsgType = mtError;
MsgButtons = TMsgDlgButtons() << mbOK;
MsgDefaultButton = mbOK;
Synchronize(&PostConfirmMessageForm);
delete DataList;
goto LabelTypeMiss;
}
FileList->Add(tsv->Strings[0]); // 0: ファイルパス
FileSizeList[i] = StrToIntDef(tsv->Strings[1], -1); // 1: ファイルサイズ（フォルダは-1）
FileAttrList[i] = StrToIntDef(tsv->Strings[2], -1); // 2: 属性
FileDtChangeList[i] = StrToIntDef(tsv->Strings[3], -1);// 3: 更新日
FileTmChangeList[i] = StrToIntDef(tsv->Strings[4], -1);// 4: 更新時
FileDtCreateList[i] = StrToIntDef(tsv->Strings[5], -1);// 5: 作成日
FileTmCreateList[i] = StrToIntDef(tsv->Strings[6], -1);// 6: 作成時
}
}
delete tsv;
delete DataList;
//-----------------------------------
//ディスクの空き容量チェック
//-----------------------------------
CurrentDriveFreeSpaceSize = GetDiskFreeSpaceNum(OutDirPath);
if (CurrentDriveFreeSpaceSize > -1 && fCompare == false) {
if ( AllTotalSize > CurrentDriveFreeSpaceSize ) {
//'復号する先のドライブの空き容量が足りません。'+#13+
//'復号するには、約%dMBの空き容量が必要です。復号処理を中止します。';
MsgText = LoadResourceString(&Msgdecrypt::_MSG_ERROR_DRIVE_NO_FREE_SPACE);
free_space_required = (float)AllTotalSize/1024/1024; // MB
MsgText = String().sprintf(MsgText.c_str(), free_space_required);
MsgType = mtError;
MsgButtons = TMsgDlgButtons() << mbOK;
MsgDefaultButton = mbOK;
Synchronize(&PostConfirmMessageForm);
goto LabelError;
}
}
else{
// OK!
//
}
//-----------------------------------
//複数フォルダ/ファイルを開こうとしているので確認
//-----------------------------------
//復号したファイルを関連付けされたソフトで開くか
if ( fTempOpenFile == true && FileList->Count > 4 && fCompare == false) {
//'5つ以上のファイルを復号後に開こうとしていますが、処理を続けますか？'+#13+
//'「いいえ」を選択すると、開かず復号します。';
MsgText = LoadResourceString(&Msgdecrypt::_MSG_CONFIRM_OPEN_DECRYPTED_FILES);
MsgType = mtConfirmation;
MsgButtons = TMsgDlgButtons() << mbYes << mbNo << mbCancel;
MsgDefaultButton = mbCancel;
Synchronize(&PostConfirmMessageForm);
if ( MsgReturnVal == mrCancel ) {
goto LabelStop;
}
else if ( MsgReturnVal == mrNo ) {
fTempOpenFile = true;
}
}
//-----------------------------------
// 復号開始
//-----------------------------------
if (fCompare == false) {
//'復号しています...'
ProgressStatusText = LoadResourceString(&Msgdecrypt::_LABEL_STATUS_TITLE_DECRYPTING);
ProgressMsgText = ExtractFileName(AtcFilePath);
}
else{
//'コンペアしています...'
ProgressStatusText = LoadResourceString(&Msgdecrypt::_LABEL_STATUS_TITLE_COMPARING);
ProgressMsgText = ExtractFileName(AtcFilePath);
}
// ファイル（データ本体）サイズを取得する
AllTotalSize = fsIn->Size - fsIn->Position + 1;
// 初期化ベクトルの読み出し
TotalSize = fsIn->Read(chain_buffer, BUF_SIZE);
// zlib 前準備
// 圧縮においてすべてのメモリ管理をライブラリに任せる
z.zalloc = Z_NULL;
z.zfree = Z_NULL;
z.opaque = Z_NULL;
if (inflateInit(&z) != Z_OK) {
// エラー
// zlibエラーは最後でまとめてメッセージ処理
goto LabelError;
}
// 通常は deflate() の第2引数は Z_NO_FLUSH にして呼び出す
flush = Z_NO_FLUSH;
// 入出力バッファ
for (i = 0; i < LARGE_BUF_SIZE; i++) {
output_buffer[i] = 0;
}
z.avail_in = 0; // 入力バッファ中のデータのバイト数
z.next_in = Z_NULL; // 入力バッファ中のデータのバイト数
z.next_out = output_buffer; // 出力バッファ残量
z.avail_out = LARGE_BUF_SIZE; // 出力ポインタ（展開するので大きめに）
status = Z_OK;
while (Terminated == false) {
//-----------------------------------
// 入力
//-----------------------------------
if (z.avail_in == 0) {
TotalSize = InputBuffer(len, source_buffer, chain_buffer, fsIn, fInputFileOpen, TotalSize);
z.avail_in = len;
z.next_in = source_buffer;
if ( len == 0 ) { //入力ストリーム終了
fInputEnd = true;
}
}
//-----------------------------------
// 展開
//-----------------------------------
status = inflate(&z, flush);
//-----------------------------------
// 処理ステータス
//-----------------------------------
if ( status == Z_OK ){
if ( z.avail_out == 0 ) {
ret = OutputBuffer(output_buffer, LARGE_BUF_SIZE,
fsOut, fOutputFileOpen,
FileList, FileIndex,
FileSizeList, FileAttrList,
FileDtChangeList, FileTmChangeList,
FileDtCreateList, FileTmCreateList);
if ( ret == 0) {
z.next_out = output_buffer;
z.avail_out = LARGE_BUF_SIZE;
}
else if (ret == -1) {
goto LabelError;
}
else{
goto LabelStop;
}
}//end if (z.avail_out == 0);
}
//-----------------------------------
// バッファエラー
//-----------------------------------
else if ( status == Z_BUF_ERROR ) { //出力バッファがいっぱいの可能性
//出力バッファをクリアにして継続させる
len = LARGE_BUF_SIZE - z.avail_out;
ret = OutputBuffer(output_buffer, len,
fsOut, fOutputFileOpen,
FileList, FileIndex,
FileSizeList, FileAttrList,
FileDtChangeList, FileTmChangeList,
FileDtCreateList, FileTmCreateList);
if (ret == 0) {
z.next_out = output_buffer;
z.avail_out = LARGE_BUF_SIZE;
}
else if ( ret == -1) {
goto LabelError;
}
else{
goto LabelStop;
}
}
//-----------------------------------
// 終了
//-----------------------------------
else if (status == Z_STREAM_END) {
break;
}
//-----------------------------------
// エラー
//-----------------------------------
else{
// #define Z_OK 0
// #define Z_STREAM_END 1
// #define Z_NEED_DICT 2
// #define Z_ERRNO (-1)
// #define Z_STREAM_ERROR (-2)
// #define Z_DATA_ERROR (-3)
// #define Z_MEM_ERROR (-4)
// #define Z_BUF_ERROR (-5)
// #define Z_VERSION_ERROR (-6)
goto LabelError;
}
//-----------------------------------
//進捗状況表示
ProgressPercentNumF = (float)TotalSize/AllTotalSize;
ProgressPercentNum = (int)(ProgressPercentNumF*100);
if (AllTotalSize < 104857600) { // 100MB未満
ProgressPercentNumText = IntToStr(ProgressPercentNum)+"%";
}
else{
ProgressPercentNumText = FloatToStrF(ProgressPercentNumF*100, ffNumber, 4, 1)+"%";
}
if ( fInputEnd == true) {
break;
}
}//while (!Terminated);
if (Terminated == true) {
//ユーザーキャンセルで抜けてきた
goto LabelStop;
}
//----------------------------------------------------------------------
// 万が一、出力バッファに余りがある場合
//----------------------------------------------------------------------
len = LARGE_BUF_SIZE - z.avail_out;
ret = OutputBuffer(output_buffer, len,
fsOut, fOutputFileOpen,
FileList, FileIndex,
FileSizeList, FileAttrList,
FileDtChangeList, FileTmChangeList,
FileDtCreateList, FileTmCreateList);
if ( ret == 0 ) {
}
else if ( ret == -1 ){
goto LabelError;
}
else{
goto LabelStop;
}
if (inflateEnd(&z) != Z_OK) {
goto LabelError;
}
ProgressPercentNum = 100;
//'完了'
ProgressStatusText = LoadResourceString(&Msgdecrypt::_LABEL_STATUS_TITLE_COMPLETE);
ProgressMsgText = ExtractFileName(AtcFilePath);
if ( fInputFileOpen == true ) {
delete fsIn;
fInputFileOpen = false;
}
if ( fOutputFileOpen == true ) {
delete fsOut;
fOutputFileOpen = false;
}
delete FileList;
delete [] FileSizeList; // 1: ファイルサイズ（フォルダは-1）
delete [] FileAttrList; // 2: 属性
delete [] FileDtChangeList; // 3: 更新日
delete [] FileTmChangeList; // 4: 更新時
delete [] FileDtCreateList; // 5: 作成日
delete [] FileTmCreateList; // 6: 作成時
//復号成功
StatusNum = 1;
return;
//-----------------------------------
//パスワード入力ミスの後始末
//-----------------------------------
LabelTypeMiss:
ProgressPercentNum = 0;
//'エラー'
ProgressStatusText = LoadResourceString(&Msgdecrypt::_LABEL_STATUS_TITLE_ERROR);
//'復号に失敗しました。'
ProgressMsgText = LoadResourceString(&Msgdecrypt::_LABEL_STATUS_DETAIL_FAILED);
if ( fInputFileOpen == true ) {
delete fsIn;
fInputFileOpen = false;
}
if ( fOutputFileOpen == true ) {
delete fsOut;
fOutputFileOpen = false;
}
delete FileList;
delete [] FileSizeList;
delete [] FileAttrList;
delete [] FileDtChangeList;
delete [] FileTmChangeList;
delete [] FileDtCreateList;
delete [] FileTmCreateList;
StatusNum = -1;
return;
//-----------------------------------
//エラーの後始末
//-----------------------------------
LabelError:
ProgressPercentNum = 0;
if ( status < 0 ){
//'zlibライブラリからエラーを返されました。'
//'エラー番号：'
MsgText = LoadResourceString(&Msgdecrypt::_MSG_ERROR_ZLIB) + IntToStr(status) + "\n" + z.msg;
if ( fCompare == true ) {
//メッセージに'コンペアに失敗しました。'を追加
MsgText += "\n" + LoadResourceString(&Msgdecrypt::_MSG_ERROR_COMPARE_FILE);
}
MsgType = mtError;
MsgButtons = TMsgDlgButtons() << mbOK;
MsgDefaultButton = mbOK;
Synchronize(&PostConfirmMessageForm);
}
//'エラー'
ProgressStatusText = LoadResourceString(&Msgdecrypt::_LABEL_STATUS_TITLE_ERROR);
if (fCompare == true) {
//'コンペアで問題を見つけたようです。';
ProgressMsgText = LoadResourceString(&Msgdecrypt::_LABEL_STATUS_DETAIL_COMPARE_FAILED);
}
else{
//'復号に失敗しました。'
ProgressMsgText = LoadResourceString(&Msgdecrypt::_LABEL_STATUS_DETAIL_FAILED);
}
if ( fInputFileOpen == true ) {
delete fsIn;
fInputFileOpen = false;
}
if ( fOutputFileOpen == true ) {
delete fsOut;
fOutputFileOpen = false;
}
delete FileList;
delete [] FileSizeList;
delete [] FileAttrList;
delete [] FileDtChangeList;
delete [] FileTmChangeList;
delete [] FileDtCreateList;
delete [] FileTmCreateList;
StatusNum = -2;
return;
//-----------------------------------
//ユーザーキャンセルの後始末
//-----------------------------------
LabelStop:
ProgressPercentNum = 0;
//'キャンセル'
ProgressStatusText = LoadResourceString(&Msgdecrypt::_LABEL_STATUS_TITLE_USER_CANCEL);
//'復号が中止されました。'
ProgressMsgText = LoadResourceString(&Msgdecrypt::_LABEL_STATUS_DETAIL_STOPPED);
if ( fInputFileOpen == true ) {
delete fsIn;
fInputFileOpen = false;
}
if ( fOutputFileOpen == true ) {
delete fsOut;
fOutputFileOpen = false;
}
delete FileList;
delete [] FileSizeList;
delete [] FileAttrList;
delete [] FileDtChangeList;
delete [] FileTmChangeList;
delete [] FileDtCreateList;
delete [] FileTmCreateList;
StatusNum = 0;
return;
}
//===========================================================================
// スレッド実行
//===========================================================================
void __fastcall TAttacheCaseFileEncrypt::Execute()
{
int i, c;
int res;
float ProgressPercentNumF; //進捗パーセンテージ（浮動小数点）
z_stream z; // zlibライブラリとやりとりするための構造体
int flush, status; // zlib
//出力する暗号化ファイルのタイムスタンプを元ファイルに合わせる
HANDLE hFile;
//_WIN32_FIND_DATAW first_fd;
ZeroMemory(&first_fd, sizeof(_WIN32_FIND_DATAW));
int len, pos;
int FileIndex;
String FilePath;
int HeaderSize; //ヘッダデータサイズ
__int64 CurrentDriveFreeSpaceSize = -1; //保存するドライブの空き容量
//実行可能形式出力ファイルのデータサイズ
__int64 ExeAllSize = 0;
__int64 ExeSize = 0;
//全体のファイルサイズ
AllTotalSize = 0;
__int64 TotalSize = 0;
//バッファ
char source_buffer[BUF_SIZE];
char read_buffer[BUF_SIZE];
char out_buffer[BUF_SIZE];
char chain_buffer[BUF_SIZE]; // IVなどを格納するチェインバッファ
char margin_buffer[BUF_SIZE];
//ファイルストリーム
TFileStream *fsIn;
TFileStream *fsOut;
TFileStream *fsExe;
//オープン中か
bool fOpenIn;
bool fOpenOut;
//メモリストリーム
TMemoryStream *pms = new TMemoryStream;
// マージンバッファサイズ
int MarginBufSize = MARGIN_BUF_SIZE;
// PKCS #7 Pading num.
unsigned char paddingNum = 0;
//---------------------------------------
// 同名ファイルがあるのでダイアログ表示
//---------------------------------------
if ( fConfirmOverwirte == true && fOverwirteYesToAll == false ) {
if (FileExists(OutFilePath) == true) {
//同名ファイルの上書き確認メッセージダイアログ
MsgText = LoadResourceString(&Msgencrypt::_MSG_CONFIRM_OVER_WRITE_SAME_FILE)+"\n"+OutFilePath;
Synchronize(&PostConfirmOverwriteMessageForm);
if ( MsgReturnVal == mrYes ) {
//上書きOKなのでFilePathはそのまま
}
else if ( MsgReturnVal == mrNo ) {
//別名保存でFilePath文字列が書き換えられてきている
OutFilePath = MsgReturnPath;
}
else if ( MsgReturnVal == mrYesToAll ) {
//すべて上書き（YesToAll）
fOverwirteYesToAll = true;
}
else if ( MsgReturnVal == mrCancel ) {
//キャンセル
delete pms;
goto LabelStop;
}
}
}
//---------------------------------------
// ヘッダ情報の生成＆ファイル総サイズ取得
//---------------------------------------
//'暗号化するファイルリストの生成中...'
ProgressStatusText = LoadResourceString(&Msgencrypt::_LABEL_STATUS_TITLE_LISTING);
if ( CreateHeaderData( pms, InputFileList, FilePathList, AllTotalSize) == false ){
if (Terminated == true) {
//ユーザーキャンセルで抜けてきた
delete pms;
goto LabelStop;
}
else{
//'暗号化するファイルを開けません。他のアプリケーションが使用中の可能性があります。'
MsgText = LoadResourceString(&Msgencrypt::_MSG_ERROR_FILE_OPEN);
MsgType = mtError;
MsgButtons = TMsgDlgButtons() << mbOK;
MsgDefaultButton = mbOK;
Synchronize(&PostConfirmMessageForm);
delete pms;
goto LabelError;
}
}
//-----------------------------------
// ディスクの空き容量チェック
//-----------------------------------
CurrentDriveFreeSpaceSize = GetDiskFreeSpaceNum(OutFilePath);
if (CurrentDriveFreeSpaceSize > -1 ) {
if ( AllTotalSize > CurrentDriveFreeSpaceSize ) {
//"ディスクの空き容量が足りません！ 暗号化ファイルを保存できません。\n
//暗号化を中止します。;"
MsgText = LoadResourceString(&Msgencrypt::_MSG_ERROR_NO_DISK_FREE_SPACE);
MsgType = mtError;
MsgButtons = TMsgDlgButtons() << mbOK;
MsgDefaultButton = mbOK;
Synchronize(&PostConfirmMessageForm);
delete pms;
goto LabelError;
}
}
else{
// -1はネットワークドライブの可能性があるので無視
//(万が一、別のエラーの場合、実際書き込みに移行したときエラーが発生する)
}
//-----------------------------------
// 実行可能形式でかつ
// 合計バイト数が4GBを越えたときのエラー
//-----------------------------------
if ( fExeOutputOption == true && fOver4gbOk == false && AllTotalSize > SIZE_4GB ){
//実行形式ファイルのサイズが4GBを超えてしまう可能性があります！\n
//Win32アプリケーションとして実行できなくなるかもしれませんがよろしいですか？';
MsgText = LoadResourceString(&Msgencrypt::_MSG_ERROR_OVER_4GB_EXE);
MsgType = mtError;
MsgButtons = TMsgDlgButtons() << mbYes << mbNo;
MsgDefaultButton = mbNo;
Synchronize(&PostConfirmMessageForm);
if ( MsgReturnVal == mbNo) {
//キャンセル
delete pms;
goto LabelStop;
}
}
//-----------------------------------
// 暗号化ファイルの生成開始
//-----------------------------------
//'暗号化しています...'
ProgressStatusText = LoadResourceString(&Msgencrypt::_LABEL_STATUS_TITLE_ENCRYPTING);
ProgressMsgText = ExtractFileName(OutFilePath);
TotalSize = 0;
try{
fsOut = new TFileStream(OutFilePath, fmCreate);
fOpenOut = true;
}
catch(...){
//'保存する先のファイルが開けません。他のアプリケーションが使用中の可能性があります。'
MsgText = LoadResourceString(&Msgencrypt::_MSG_ERROR_OUT_FILE_OPEN) + "\n" + OutFilePath;
MsgType = mtError;
MsgButtons = TMsgDlgButtons() << mbOK;
MsgDefaultButton = mbOK;
Synchronize(&PostConfirmMessageForm);
delete pms;
goto LabelError;
}
//-----------------------------------
// 実行可能形式の出力
//-----------------------------------
if ( fExeOutputOption == true ){
//-----------------------------------
// 自分のお尻から実行データを抽出
//-----------------------------------
//自分自身の実行ファイルを開く
try{
fsExe = new TFileStream(Application->ExeName, fmOpenRead | fmShareDenyWrite);
}
catch(...){
//'実行可能形式出力に失敗しました。暗号化処理を中止します。'
MsgText = LoadResourceString(&Msgencrypt::_MSG_ERROR_EXEOUT_FAILED);
MsgType = mtError;
MsgButtons = TMsgDlgButtons() << mbOK;
MsgDefaultButton = mbOK;
Synchronize(&PostConfirmMessageForm);
delete pms;
goto LabelError;
}
//切り出すサイズを取得
fsExe->Seek(-(__int64)sizeof(__int64), TSeekOrigin::soEnd);
fsExe->Read(&ExeAllSize, sizeof(__int64));
//処理する合計サイズに実行形式分を加える
AllTotalSize += ExeAllSize;
//自己実行可能形式データの境界へ
fsExe->Seek(-(__int64)ExeAllSize-sizeof(__int64), TSeekOrigin::soEnd);
while(fsExe->Read(read_buffer, BUF_SIZE) != 0 ){
ExeSize+=BUF_SIZE;
//書き込む
if ( ExeSize < ExeAllSize ){
fsOut->Write(read_buffer, BUF_SIZE);
TotalSize += BUF_SIZE;
}
else{
fsOut->Write(read_buffer, ExeSize-ExeAllSize);
TotalSize += (ExeSize-ExeAllSize);
}
//進捗表示
ProgressPercentNumF = (float)TotalSize/AllTotalSize;
ProgressPercentNum = (int)(ProgressPercentNumF*100);
if (AllTotalSize < 104857600) { // 100MB未満
ProgressPercentNumText = IntToStr(ProgressPercentNum)+"%";
}
else{
ProgressPercentNumText = FloatToStrF(ProgressPercentNumF*100, ffNumber, 4, 1)+"%";
}
}
//自分自身を閉じる
delete fsExe;
}
//-----------------------------------
// ヘッダ情報の描き込み
//-----------------------------------
pms->SaveToStream(fsOut); //fsOutに追記
delete pms;
//-----------------------------------
// Rijndaelの初期化
//-----------------------------------
gentables();
gkey( 8, 8, key);
// 初期化ベクトルを生成して先頭に書き込む
fillrand(chain_buffer, BUF_SIZE);
if ( fsOut->Write(chain_buffer, BUF_SIZE) < BUF_SIZE ){
//''保存先に指定された暗号化ファイルに書き込めません。
MsgText = LoadResourceString(&Msgencrypt::_MSG_ERROR_OUT_FILE_WRITE) + "\n" + OutFilePath;
MsgType = mtError;
MsgButtons = TMsgDlgButtons() << mbOK;
MsgDefaultButton = mbOK;
Synchronize(&PostConfirmMessageForm);
goto LabelError;
}
//-----------------------------------
// zlib 初期化（圧縮においてすべてのメモリ管理をライブラリに任せる）
z.zalloc = Z_NULL;
z.zfree = Z_NULL;
z.opaque = Z_NULL;
//z.next_in = Z_NULL;
// 第2引数は圧縮の度合。0～9 の範囲の整数で，0 は無圧縮
// Z_DEFAULT_COMPRESSION (= 6) が標準
if (deflateInit(&z, CompressRateNum) != Z_OK){
//zlibエラー表示はラベル先で
goto LabelError;
}
//出力バッファの初期化
for(i = 0; i < BUF_SIZE; i++){
out_buffer[i] = 0;
}
// zlibに入出力バッファをセットする
z.avail_in = 0; // 入力バッファ中のデータのバイト数
z.next_out = out_buffer; // 出力バッファ残量
z.avail_out = BUF_SIZE; // 出力ポインタ
// 通常は deflate() の第2引数は Z_NO_FLUSH にして呼び出す
flush = Z_NO_FLUSH;
FileIndex = 0;
while(!Terminated) {
//-----------------------------------
//入力
//-----------------------------------
if ( z.avail_in == 0 && flush != Z_FINISH){
pos = 0;
for(i = 0; i < BUF_SIZE; i++){
source_buffer[i] = 0;
read_buffer[i] = 0;
}
while ( pos < BUF_SIZE ){
//オープン中のファイルがあればそこから読む
if ( fOpenIn == true ) {
if (pos < BUF_SIZE) {
len = fsIn->Read(read_buffer, BUF_SIZE - pos);
TotalSize+=len;
for (i = 0; i < len; i++) {
source_buffer[pos+i] = read_buffer[i];
}
if (len < BUF_SIZE - pos) {
fOpenIn = false; //ファイルを閉じる
delete fsIn;
}
}
pos += len;
}
//ファイルを開く
else{
if (FileIndex < FilePathList->Count) {
while(FileIndex < FilePathList->Count){
if (FilePathList->Strings[FileIndex] != "") {
try{
fsIn = new TFileStream(FilePathList->Strings[FileIndex], fmOpenRead | fmShareDenyWrite);
fOpenIn = true;
FileIndex++;
break;
}
catch(...){
//'暗号化するファイルを開けません。他のアプリケーションが使用中の可能性があります。'
MsgText = LoadResourceString(&Msgencrypt::_MSG_ERROR_FILE_OPEN);
MsgType = mtError;
MsgButtons = TMsgDlgButtons() << mbOK;
MsgDefaultButton = mbOK;
Synchronize(&PostConfirmMessageForm);
fOpenIn = false;
goto LabelError;
}
}
FileIndex++;
}
}
else{
//読み込むファイルがなくなったので、
//お尻にダミーのマージンデータを挿入する
//
//【補足】
// 本来はここにあるマージンデータ挿入処理は不要ですが、
// 昔に作った際に復号の際に圧縮データ境界のチェックを
// 怠っていたため、このように余分なデータを
// 入れておくという力業を使っています（すみません...）
fillrand(margin_buffer, BUF_SIZE);
for (i = pos; i < BUF_SIZE; i++) {
source_buffer[i] = margin_buffer[i];
}
pos = BUF_SIZE;
MarginBufSize -= BUF_SIZE;
}//end if (FileIndex < FilePathList->Count);
}//end if ( fOpenIn == true );
}//while ( pos < BUF_SIZE && 0 < MarginBufSize );
if (MarginBufSize < 1) {
flush = Z_FINISH; //入力バッファはこれが最後
}
z.next_in = source_buffer;
z.avail_in = pos;
}//end if ( z.avail_in == 0 );
//-----------------------------------
//圧縮
//-----------------------------------
if ( z.avail_out > 0 ){
status = deflate(&z, flush);
}
if (status == Z_STREAM_END){
break;
}
if (status != Z_OK ){
//#define Z_OK 0
//#define Z_STREAM_END 1
//#define Z_NEED_DICT 2
//#define Z_ERRNO (-1)
//#define Z_STREAM_ERROR (-2)
//#define Z_DATA_ERROR (-3)
//#define Z_MEM_ERROR (-4)
//#define Z_BUF_ERROR (-5)
//#define Z_VERSION_ERROR (-6)
goto LabelError;
}
//-----------------------------------
//出力
//-----------------------------------
if ( z.avail_out == 0 ){
// CBC - xor the file bytes with the IV bytes
for(i = 0; i < BUF_SIZE; i++){
out_buffer[i] ^= chain_buffer[i];
}
//encrypt!
rijndael_encrypt(out_buffer);
len = fsOut->Write(out_buffer, BUF_SIZE);
if (len < BUF_SIZE) {
//'保存先に指定された暗号化ファイルに書き込めません。
MsgText = LoadResourceString(&Msgencrypt::_MSG_ERROR_OUT_FILE_WRITE) + "\n" + OutFilePath;
MsgType = mtError;
MsgButtons = TMsgDlgButtons() << mbOK;
MsgDefaultButton = mbOK;
Synchronize(&PostConfirmMessageForm);
goto LabelError;
}
for(i = 0; i < BUF_SIZE; i++){
chain_buffer[i] = out_buffer[i];
out_buffer[i] = 0;
}
z.next_out = out_buffer; // 出力バッファ残量を元に戻す
z.avail_out = BUF_SIZE; // 出力ポインタを元に戻す
}
//-----------------------------------
//進捗状況表示
if (AllTotalSize == 0) {
ProgressPercentNum = 100;
ProgressPercentNumText = "100%";
}
else if (TotalSize == 0) {
ProgressPercentNum = 0;
ProgressPercentNumText = "0%";
}
else{
ProgressPercentNumF = (float)TotalSize/AllTotalSize;
ProgressPercentNum = (int)(ProgressPercentNumF*100);
if (AllTotalSize < 104857600) { // 100MB未満
ProgressPercentNumText = IntToStr(ProgressPercentNum)+"%";
}
else{
ProgressPercentNumText = FloatToStrF(ProgressPercentNumF*100, ffNumber, 4, 1)+"%";
}
}
//-----------------------------------
if ( fOpenIn == true ){
ProgressMsgText = ExtractFileName(fsIn->FileName);
}
else{
ProgressMsgText = ExtractFileName(OutFilePath);
}
}//while(!Terminated);
if (Terminated == true) {
//ユーザーキャンセルで抜けてきた
goto LabelStop;
}
//残りのバッファ
if (z.avail_out > 0) {
// PKCS #7 パディング
len = BUF_SIZE - z.avail_out;
paddingNum = (char)z.avail_out;
for(i = len; i < BUF_SIZE; i++){
out_buffer[i] = paddingNum;
}
// CBC - xor the file bytes with the IV bytes
for(i = 0; i < BUF_SIZE; i++){
out_buffer[i] ^= chain_buffer[i];
}
//encrypt!
rijndael_encrypt(out_buffer);
if ((len = fsOut->Write(out_buffer, BUF_SIZE)) != BUF_SIZE){
//'保存先に指定された暗号化ファイルに書き込めません。
MsgText = LoadResourceString(&Msgencrypt::_MSG_ERROR_OUT_FILE_WRITE) + "\n" + OutFilePath;
MsgType = mtError;
MsgButtons = TMsgDlgButtons() << mbOK;
MsgDefaultButton = mbOK;
Synchronize(&PostConfirmMessageForm);
goto LabelError;
}
}
if (deflateEnd(&z) != Z_OK){
//zlibエラー
goto LabelError;
}
//-----------------------------------
// 実行可能形式ファイルは
// 末尾へ暗号化データサイズを書き込む
//-----------------------------------
if ( fExeOutputOption == true ){
ExeSize = fsOut->Seek((__int64)0, TSeekOrigin::soEnd);
ExeSize = ExeSize-ExeAllSize;
fsOut->Write(&ExeSize, sizeof(__int64));
}
//-----------------------------------
// 完了
//-----------------------------------
ProgressPercentNum = 100;
//'完了'
ProgressStatusText = LoadResourceString(&Msgencrypt::_LABEL_STATUS_TITLE_COMPLETE);
ProgressMsgText = ExtractFileName(OutFilePath);
if (fOpenIn == true) {
delete fsIn;
}
if (fOpenOut == true) {
delete fsOut;
}
//出力する暗号化ファイルのタイムスタンプを元ファイルに合わせる
if ( fKeepTimeStamp == true && first_fd.cFileName[0] != NULL ) {
hFile = CreateFileW(FilePath.c_str(), GENERIC_WRITE, FILE_SHARE_READ, NULL,
OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
if (hFile != INVALID_HANDLE_VALUE) {
SetFileTime( hFile, &first_fd.ftCreationTime, &first_fd.ftLastAccessTime, &first_fd.ftLastWriteTime);
CloseHandle(hFile);
}
}
StatusNum = 1;
return;
//-----------------------------------
// エラーの後始末
//-----------------------------------
LabelError:
ProgressPercentNum = 0;
if ( status < 0 ){
//'zlibライブラリからエラーを返されました。'
//'エラー番号：'
MsgText = LoadResourceString(&Msgencrypt::_MSG_ERROR_ZLIB) + IntToStr(status) + "\n" + z.msg;
MsgType = mtError;
MsgButtons = TMsgDlgButtons() << mbOK;
MsgDefaultButton = mbOK;
Synchronize(&PostConfirmMessageForm);
}
//'エラー'
ProgressStatusText = LoadResourceString(&Msgencrypt::_LABEL_STATUS_TITLE_ERROR);
//'暗号化に失敗しました。'
ProgressMsgText = LoadResourceString(&Msgencrypt::_LABEL_STATUS_DETAIL_FAILED);
if (fOpenIn == true) {
delete fsIn;
}
if (fOpenOut == true) {
delete fsOut;
}
StatusNum = -1;
return;
//-----------------------------------
// ユーザーキャンセルの後始末
//-----------------------------------
LabelStop:
ProgressPercentNum = 0;
//'キャンセル'
ProgressStatusText = LoadResourceString(&Msgencrypt::_LABEL_STATUS_TITLE_USER_CANCEL);
//'暗号化が中止されました。'
ProgressMsgText = LoadResourceString(&Msgencrypt::_LABEL_STATUS_DETAIL_STOPPED);
if (fOpenIn == true) {
delete fsIn;
}
if (fOpenOut == true) {
delete fsOut;
}
StatusNum = -2;
return;
}
//---------------------------------------------------------------------------
//暗号化ファイルを破壊する
//---------------------------------------------------------------------------
bool __fastcall TForm1::DestroyAtcFile(void)
{
int i;
int fh;
String MsgText;
char buffer[BUF_SIZE];
for ( i = 0; i < BUF_SIZE; i++ ){
buffer[i]=NULL;
}
TFileStream *fsIn;
char token[16];
const char charTokenString[16] = "_AttacheCaseData"; //復号の正否に使う
const char charDestroyTokenString[16] = "_Atc_Broken_Data"; //破壊されているとき
String AtcFileTokenString; //暗号化ファイルのトークン（文字列）
String AtcFileCreateDateString; //暗号化ファイルの生成日時（文字列）
__int64 AllTotalSize;
int PlaneHeaderSize;
int HeaderBufSize;
try {
fsIn = new TFileStream(Application->ExeName, fmShareDenyNone);
}
catch(...) {
//'ファイルを開けません。他のアプリケーションで使用中の可能性があります。'
MsgText = LoadResourceString(&Msgexeout::_MSG_FILE_OPEN_ERROR);
ShowConfirmMassageForm(MsgText, mtError, TMsgDlgButtons()<<mbOK, mbOK);
return(false);
}
//総サイズ取得
AllTotalSize = fsIn->Seek((__int64)0, TSeekOrigin::soEnd);
fsIn->Seek((__int64)0, TSeekOrigin::soBeginning);
//-----------------------------------
//ヘッダ情報のチェック
//-----------------------------------
// サイズを取得
fsIn->Seek(-(__int64)sizeof(__int64), TSeekOrigin::soEnd);
fsIn->Read(&AllTotalSize, sizeof(__int64));
// 位置を戻す
fsIn->Seek(-(AllTotalSize + sizeof(__int64)), TSeekOrigin::soEnd);
// 平文ヘッダサイズを読み込む
fsIn->Read(&PlaneHeaderSize, sizeof(int));
// トークンを取得
fsIn->Read(token, 16);
// トークンを再チェック
if (memcmp(token, charTokenString, 16) != 0 ) {
//すでに壊れている？ サイレントに処理終了
delete fsIn;
return(true);
}
else{
fsIn->Seek((__int64)-16, TSeekOrigin::soCurrent);
//"_Atc_Broken_Data"を書き込む
fsIn->Write(charDestroyTokenString, 16);
}
//「データバージョン」「アルゴリズムの種類」分だけ進める
fsIn->Seek((__int64)(sizeof(int)*2), TSeekOrigin::soCurrent);
//暗号部ヘッダサイズを取得する
fsIn->Read(&HeaderBufSize, sizeof(int));
//暗号部ヘッダのIVを書き換えて破壊する
fsIn->Write(buffer, BUF_SIZE);
//「暗号部ヘッダサイズ」分だけ進める
fsIn->Seek((__int64)(HeaderBufSize-BUF_SIZE), TSeekOrigin::soCurrent);
// IV部分を書き換えて破壊する
fsIn->Write(buffer, BUF_SIZE);
delete fsIn;
return(true);
}