//---------------------------------------------------------------------------
void __fastcall THttpTestForm::SendButtonClick(TObject *Sender)
{
TMemoryStream *DataIn;
TMemoryStream *DataOut;
AnsiString Buf;
BOOL bSuccess;
DisplayMemo->Clear();
DataIn = new TMemoryStream;
DataOut = new TMemoryStream;
bSuccess = TRUE;
try {
Buf = "ID=" + UserIDEdit->Text +
"&REMITE=" + EMailEdit->Text +
"&MENSAJE=" + Encode(MessageEdit->Text);
DataOut->Write(&Buf[1], Buf.Length());
DataOut->Seek(0, soFromBeginning);
HttpCli1->SendStream = DataOut;
HttpCli1->RcvdStream = DataIn;
HttpCli1->Proxy = ProxyEdit->Text;
HttpCli1->ProxyPort = "80";
HttpCli1->URL = "http://www.unired.net.pe/cgi-bin/a.out";
SendButton->Enabled = FALSE;
AbortButton->Enabled = TRUE;
try {
HttpCli1->Post();
}
__except (TRUE) {
bSuccess = FALSE;
}
if (bSuccess) {
DataIn->Seek(0, 0);
DisplayMemo->Lines->LoadFromStream(DataIn);
}
SendButton->Enabled = TRUE;
AbortButton->Enabled = FALSE;
}
__except (TRUE) {
bSuccess = FALSE;
}
if (!bSuccess)
DisplayMemo->Lines->Add("Failed : " + HttpCli1->ReasonPhrase);
delete DataOut;
delete DataIn;
}
//----------
//接口实现
//----------
TMemoryStream* TFComm::Read()
{
UINT8 buf[RECV_BUFF_MAX_LEN];
int rv;
TMemoryStream *p;
if (m_Recv->Count() > 0)
{
p = (TMemoryStream*)m_Recv->Pop();
p->Seek(0, soFromBeginning);
rv = p->Read(buf, RECV_BUFF_MAX_LEN);
LogMsg("Read" + IntToStr(rv) + ": " + StreamToText(buf, rv));
return p;
}
else return NULL;
}
//---------------------------------------------------------------------------
void __fastcall THttpTestForm::PostButtonClick(TObject *Sender)
{
TMemoryStream *DataOut;
TFileStream *DataIn;
AnsiString Buf;
int I;
DisplayMemo->Clear();
DocumentMemo->Clear();
SetButtonState(FALSE);
DataOut = new TMemoryStream;
Buf = DataEdit->Text;
DataOut->Write(&Buf[1], Buf.Length());
DataOut->Seek(0, soFromBeginning);
HttpCli1->SendStream = DataOut;
HttpCli1->Proxy = ProxyHostEdit->Text;
HttpCli1->ProxyPort = ProxyPortEdit->Text;
HttpCli1->RcvdStream = NULL;
HttpCli1->URL = URLEdit->Text;
try {
HttpCli1->Post();
} __except (TRUE) {
SetButtonState(TRUE);
delete DataOut;
DisplayMemo->Lines->Add("POST Failed !");
DisplayMemo->Lines->Add("StatusCode = " + IntToStr(HttpCli1->StatusCode));
DisplayMemo->Lines->Add("ReasonPhrase = " + HttpCli1->ReasonPhrase);
return;
}
delete DataOut;
DisplayMemo->Lines->Add("StatusCode = " + IntToStr(HttpCli1->StatusCode));
for (I = 0; I < HttpCli1->RcvdHeader->Count; I++)
DisplayMemo->Lines->Add("hdr>" + HttpCli1->RcvdHeader->Strings[I]);
DataIn = new TFileStream(HttpCli1->DocName, fmOpenRead);
DocumentMemo->Lines->LoadFromStream(DataIn);
delete DataIn;
SetButtonState(TRUE);
}
//===========================================================================
//スレッド実行
//===========================================================================
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;
}
//===========================================================================
// ヘッダ情報を生成する
//===========================================================================
bool __fastcall TAttacheCaseFileEncrypt::CreateHeaderData
(TMemoryStream *pms, TStringList *FileList, TStringList *FilePathList, __int64 &AllTotalFileSize)
{
int i, c;
int ret;
int Index = 0;
int HeaderSizeAddress = 0;
TSearchRec sr;
String OneLine;
String DirPath, FileName;
String MsgText;
//暗号部トークン
const AnsiString Passcode_AttacheCase = "Passcode:AttacheCase\n";
//暗号化ファイルの作成日
AnsiString LastDateTimeString = "LastDateTime:" + DateTimeToStr(Now()) + "\n";
//旧ヘッダーテキストすべて
AnsiString Fn_HeaderText;
//Unicode用ヘッダーテキストすべて
String U_HeaderText;
int EncryptHeaderSize = 0; //暗号部ヘッダサイズ
char buffer[BUF_SIZE];
char chain_buffer[BUF_SIZE];
TStringList *HeaderDataList;
TMemoryStream* tpms; //テンポラリメモリストリーム
//-----------------------------------
// ヘッダ情報（平文）
//-----------------------------------
const char charReservedValue[4] = { 0, 0, 0, 0 };
const char charDataSubVersion = ATC_DATA_SUB_VERSION;
const char charOptMissTypeLimitsNumOption = intOptMissTypeLimitsNumOption;
const char charOptBrokenFileOption = (fOptBrokenFileOption > 0 ? 1 : 0);
const char charTokenString[17] = "_AttacheCaseData";
const int DataFileVersion = ATC_DATA_FILE_VERSION;
const int AlgorismType = TYPE_ALGORISM_RIJNDAEL;
//データサブバージョン : 1byte
pms->Write(&charDataSubVersion, sizeof(char));
//予約データ(reserved） : 1byte
pms->Write(&charReservedValue, sizeof(char));
//ミスタイプ回数 : 1byte
pms->Write(&charOptMissTypeLimitsNumOption, sizeof(char));
//破壊するか否か : 1byte
pms->Write(&charOptBrokenFileOption, sizeof(char));
//トークン : 16byte
pms->Write(&charTokenString, 16);
//データファイルバージョン : 4byte
pms->Write(&DataFileVersion, sizeof(int));
//アルゴリズムタイプ : 4byte
pms->Write(&AlgorismType, sizeof(int));
//暗号化部分のヘッダデータサイズ（先に確保しておく）：4byte
HeaderSizeAddress = pms->Position;
pms->Write(&EncryptHeaderSize, sizeof(int));
//-----------------------------------
// ヘッダ情報（暗号化部分）
//-----------------------------------
//進捗状況表示
ProgressPercentNum = -1;
//'暗号化するファイルリストの生成中...'
ProgressStatusText = LoadResourceString(&Msgencrypt::_LABEL_STATUS_TITLE_LISTING);
//ヘッダデータリスト（文字列）
HeaderDataList = new TStringList;
//パスワード
HeaderDataList->Add(Passcode_AttacheCase);
//作成日
HeaderDataList->Add(LastDateTimeString);
for ( i = 0; i < FileList->Count; i++ ){
//ファイル
if (FileExists(FileList->Strings[i]) == true) {
DirPath = ExtractFileDir(FileList->Strings[i]);
FileName = ExtractFileName(FileList->Strings[i]);
ProgressMsgText = FileName; //処理中のファイル名
AllTotalFileSize +=
GetFileInfoList(Index, DirPath, FileName, FileList->Strings[i], FilePathList, HeaderDataList);
}
//ディレクトリ
else{
DirPath = ExtractFileDir(FileList->Strings[i]);
FileName = ExtractFileName(FileList->Strings[i]);
ProgressMsgText = FileName; //処理中のファイル名
//トップディレクトリ
GetFileInfoList(Index, DirPath, FileName, FileList->Strings[i], FilePathList, HeaderDataList);
//その配下
AllTotalFileSize +=
GetFileInfoList(Index, FileList->Strings[i], "", FileList->Strings[i], FilePathList, HeaderDataList);
}
//ユーザーキャンセル
if (Terminated == true) {
delete HeaderDataList;
return(false);
}
}// end for;
//進捗状況表示
ProgressPercentNum = -1;
//'ヘッダデータを書き込んでいます...'
ProgressStatusText = LoadResourceString(&Msgencrypt::_LABEL_STATUS_TITLE_ENCRYPTING_LIST);
ProgressMsgText = "";
//メモリストリームへ書き込み
tpms = new TMemoryStream;
//------------------------------------------------
// 暗号化時にヘッダデータの互換性維持
//---------------------------------------------------
HeaderDataList->SaveToStream(tpms, TEncoding::GetEncoding(932));
//新バージョン（ver.2.8.0～）用（UTF-8）に保存
for (i = 0; i < HeaderDataList->Count; i++) {
HeaderDataList->Strings[i] = StringReplace(HeaderDataList->Strings[i],"Fn_","U_",TReplaceFlags()<<rfIgnoreCase );
}
HeaderDataList->SaveToStream(tpms, TEncoding::UTF8);
delete HeaderDataList;
//-----------------------------------
//ヘッダ情報の暗号化
//-----------------------------------
//暗号化の準備
gentables();
//キー入力
gkey( 8, 8, key);
for (i = 0; i < BUF_SIZE; i++) {
buffer[i] = 0;
}
//初期化ベクトル（IV）を生成
fillrand(chain_buffer, BUF_SIZE);
pms->Write(chain_buffer, BUF_SIZE);
//先頭にポインタを戻す
tpms->Seek((__int64)0, TSeekOrigin::soBeginning);
EncryptHeaderSize = 0;
//CBCモードで書き込む
while (tpms->Read( buffer, BUF_SIZE ) != NULL){
EncryptHeaderSize += BUF_SIZE;
// xor
for ( i = 0; i < BUF_SIZE; i++ ){
buffer[i] ^= chain_buffer[i];
}
// rijndael
rijndael_encrypt(buffer);
pms->Write(buffer, BUF_SIZE);
//CBC＆バッファの初期化
for ( i = 0; i < BUF_SIZE; i++ ){
chain_buffer[i] = buffer[i];
buffer[i] = 0;
}
//ユーザーキャンセル
if (Terminated == true) {
delete tpms;
return(false);
}
}//loop;
delete tpms;
//暗号化部分のヘッダデータサイズ（確保しておいた場所へ改めて書き込む）
pms->Position = HeaderSizeAddress;
pms->Write(&EncryptHeaderSize, sizeof(int));
//先頭にポインタを戻す
pms->Seek((__int64)0, TSeekOrigin::soBeginning);
return(true);
}//end CreateHeaderData;
//===========================================================================
//スレッド実行
//===========================================================================
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 z3ResEx::Run( )
{
// Check the fileindex exists
if( TFileSize( msfName ) == 0 )
{
setMessage( "ERROR: Unable to open file (%s)", msfName );
}
else
{
TMemoryStream msf;
m_fileindexKey = nullptr;
m_fileindexKeyLength = 0;
//
// Brute-force the key (version 1)
//
unsigned int keyIndex( 0 );
// For all known keys
while( ( keyIndex < keyList1Count ) && ( msf.Size() == 0 ) )
{
// Try to read the fileindex
if( fsReadMSF( msf, keyList1[ keyIndex ].Data, KeyLength1, 0 ) )
{
m_fileindexKey = keyList1[ keyIndex ].Data;
m_fileindexKeyLength = KeyLength1;
m_fileindexVer = 0;
if( m_verboseMessages )
printf("Found key for %s!\n", keyList1[ keyIndex ].Desc );
}
++keyIndex;
}
// If key has not been found
if( m_fileindexKey == nullptr )
{
//
// Continue to brute-force the key (version 2)
//
keyIndex = 0;
// For all known keys
while( ( keyIndex < keyList2Count ) && ( msf.Size() == 0 ) )
{
// Try to read the fileindex
if( fsReadMSF( msf, keyList2[ keyIndex ].Data, KeyLength2, 1 ) )
{
m_fileindexKey = keyList2[ keyIndex ].Data;
m_fileindexKeyLength = KeyLength2;
m_fileindexVer = 1;
if( m_verboseMessages )
printf("Found key for %s!\n", keyList2[ keyIndex ].Desc );
}
++keyIndex;
}
}
// If a valid key has been found and fileindex loaded
if( !( ( m_fileindexKey == nullptr ) && ( msf.Size() == 0 ) ) )
{
// Attempt to parse it (to extract or list files)
msf.Seek( 0, bufo_start );
parseMsf( msf );
}
else
{
// No key found or incompatiable file (not checked)
setMessage( "ERROR: This file is using an updated key or unsupported method" );
}
msf.Close();
}
}
