244 lines
6.3 KiB
C++
244 lines
6.3 KiB
C++
|
#include "rar.hpp"
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
#if defined(RARDLL) && defined(_MSC_VER) && !defined(_M_X64)
|
||
|
|
// Disable the run time stack check for unrar.dll, so we can manipulate
|
||
|
|
// with ChangeVolProc call type below. Run time check would intercept
|
||
|
|
// a wrong ESP before we restore it.
|
||
|
|
#pragma runtime_checks( "s", off )
|
||
|
|
#endif
|
||
|
|
|
||
|
|
bool MergeArchive(Archive &Arc,ComprDataIO *DataIO,bool ShowFileName,char Command)
|
||
|
|
{
|
||
|
|
RAROptions *Cmd=Arc.GetRAROptions();
|
||
|
|
|
||
|
|
int HeaderType=Arc.GetHeaderType();
|
||
|
|
FileHeader *hd=HeaderType==NEWSUB_HEAD ? &Arc.SubHead:&Arc.NewLhd;
|
||
|
|
bool SplitHeader=(HeaderType==FILE_HEAD || HeaderType==NEWSUB_HEAD) &&
|
||
|
|
(hd->Flags & LHD_SPLIT_AFTER)!=0;
|
||
|
|
|
||
|
|
if (DataIO!=NULL && SplitHeader && hd->UnpVer>=20 &&
|
||
|
|
hd->FileCRC!=0xffffffff && DataIO->PackedCRC!=~hd->FileCRC)
|
||
|
|
{
|
||
|
|
Log(Arc.FileName,St(MDataBadCRC),hd->FileName,Arc.FileName);
|
||
|
|
}
|
||
|
|
|
||
|
|
int64 PosBeforeClose=Arc.Tell();
|
||
|
|
|
||
|
|
if (DataIO!=NULL)
|
||
|
|
DataIO->ProcessedArcSize+=Arc.FileLength();
|
||
|
|
|
||
|
|
Arc.Close();
|
||
|
|
|
||
|
|
char NextName[NM];
|
||
|
|
wchar NextNameW[NM];
|
||
|
|
strcpy(NextName,Arc.FileName);
|
||
|
|
strcpyw(NextNameW,Arc.FileNameW);
|
||
|
|
NextVolumeName(NextName,NextNameW,ASIZE(NextName),(Arc.NewMhd.Flags & MHD_NEWNUMBERING)==0 || Arc.OldFormat);
|
||
|
|
|
||
|
|
#if !defined(SFX_MODULE) && !defined(RARDLL)
|
||
|
|
bool RecoveryDone=false;
|
||
|
|
#endif
|
||
|
|
bool FailedOpen=false,OldSchemeTested=false;
|
||
|
|
|
||
|
|
while (!Arc.Open(NextName,NextNameW))
|
||
|
|
{
|
||
|
|
// We need to open a new volume which size was not calculated
|
||
|
|
// in total size before, so we cannot calculate the total progress
|
||
|
|
// anymore. Let's reset the total size to zero and stop
|
||
|
|
// the total progress.
|
||
|
|
if (DataIO!=NULL)
|
||
|
|
DataIO->TotalArcSize=0;
|
||
|
|
|
||
|
|
if (!OldSchemeTested)
|
||
|
|
{
|
||
|
|
// Checking for new style volumes renamed by user to old style
|
||
|
|
// name format. Some users did it for unknown reason.
|
||
|
|
char AltNextName[NM];
|
||
|
|
wchar AltNextNameW[NM];
|
||
|
|
strcpy(AltNextName,Arc.FileName);
|
||
|
|
strcpyw(AltNextNameW,Arc.FileNameW);
|
||
|
|
NextVolumeName(AltNextName,AltNextNameW,ASIZE(AltNextName),true);
|
||
|
|
OldSchemeTested=true;
|
||
|
|
if (Arc.Open(AltNextName,AltNextNameW))
|
||
|
|
{
|
||
|
|
strcpy(NextName,AltNextName);
|
||
|
|
strcpyw(NextNameW,AltNextNameW);
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
#ifdef RARDLL
|
||
|
|
if (Cmd->Callback==NULL && Cmd->ChangeVolProc==NULL ||
|
||
|
|
Cmd->Callback!=NULL && Cmd->Callback(UCM_CHANGEVOLUME,Cmd->UserData,(LPARAM)NextName,RAR_VOL_ASK)==-1)
|
||
|
|
{
|
||
|
|
Cmd->DllError=ERAR_EOPEN;
|
||
|
|
FailedOpen=true;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
if (Cmd->ChangeVolProc!=NULL)
|
||
|
|
{
|
||
|
|
// Here we preserve ESP value. It is necessary for those developers,
|
||
|
|
// who still define ChangeVolProc callback as "C" type function,
|
||
|
|
// even though in year 2001 we announced in unrar.dll whatsnew.txt
|
||
|
|
// that it will be PASCAL type (for compatibility with Visual Basic).
|
||
|
|
#if defined(_MSC_VER)
|
||
|
|
#ifndef _M_X64
|
||
|
|
__asm mov ebx,esp
|
||
|
|
#endif
|
||
|
|
#elif defined(_WIN_32) && defined(__BORLANDC__)
|
||
|
|
_EBX=_ESP;
|
||
|
|
#endif
|
||
|
|
int RetCode=Cmd->ChangeVolProc(NextName,RAR_VOL_ASK);
|
||
|
|
|
||
|
|
// Restore ESP after ChangeVolProc with wrongly defined calling
|
||
|
|
// convention broken it.
|
||
|
|
#if defined(_MSC_VER)
|
||
|
|
#ifndef _M_X64
|
||
|
|
__asm mov esp,ebx
|
||
|
|
#endif
|
||
|
|
#elif defined(_WIN_32) && defined(__BORLANDC__)
|
||
|
|
_ESP=_EBX;
|
||
|
|
#endif
|
||
|
|
if (RetCode==0)
|
||
|
|
{
|
||
|
|
Cmd->DllError=ERAR_EOPEN;
|
||
|
|
FailedOpen=true;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
#else // RARDLL
|
||
|
|
|
||
|
|
#if !defined(SFX_MODULE) && !defined(_WIN_CE)
|
||
|
|
if (!RecoveryDone)
|
||
|
|
{
|
||
|
|
RecVolumes RecVol;
|
||
|
|
RecVol.Restore(Cmd,Arc.FileName,Arc.FileNameW,true);
|
||
|
|
RecoveryDone=true;
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#ifndef GUI
|
||
|
|
if (!Cmd->VolumePause && !IsRemovable(NextName))
|
||
|
|
{
|
||
|
|
FailedOpen=true;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
#ifndef SILENT
|
||
|
|
if (Cmd->AllYes || !AskNextVol(NextName))
|
||
|
|
#endif
|
||
|
|
{
|
||
|
|
FailedOpen=true;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif // RARDLL
|
||
|
|
*NextNameW=0;
|
||
|
|
}
|
||
|
|
if (FailedOpen)
|
||
|
|
{
|
||
|
|
#if !defined(SILENT) && !defined(_WIN_CE)
|
||
|
|
Log(Arc.FileName,St(MAbsNextVol),NextName);
|
||
|
|
#endif
|
||
|
|
Arc.Open(Arc.FileName,Arc.FileNameW);
|
||
|
|
Arc.Seek(PosBeforeClose,SEEK_SET);
|
||
|
|
return(false);
|
||
|
|
}
|
||
|
|
Arc.CheckArc(true);
|
||
|
|
#ifdef RARDLL
|
||
|
|
if (Cmd->Callback!=NULL &&
|
||
|
|
Cmd->Callback(UCM_CHANGEVOLUME,Cmd->UserData,(LPARAM)NextName,RAR_VOL_NOTIFY)==-1)
|
||
|
|
return(false);
|
||
|
|
if (Cmd->ChangeVolProc!=NULL)
|
||
|
|
{
|
||
|
|
#if defined(_WIN_32) && !defined(_MSC_VER) && !defined(__MINGW32__)
|
||
|
|
_EBX=_ESP;
|
||
|
|
#endif
|
||
|
|
int RetCode=Cmd->ChangeVolProc(NextName,RAR_VOL_NOTIFY);
|
||
|
|
#if defined(_WIN_32) && !defined(_MSC_VER) && !defined(__MINGW32__)
|
||
|
|
_ESP=_EBX;
|
||
|
|
#endif
|
||
|
|
if (RetCode==0)
|
||
|
|
return(false);
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
|
||
|
|
if (Command=='T' || Command=='X' || Command=='E')
|
||
|
|
mprintf(St(Command=='T' ? MTestVol:MExtrVol),Arc.FileName);
|
||
|
|
if (SplitHeader)
|
||
|
|
Arc.SearchBlock(HeaderType);
|
||
|
|
else
|
||
|
|
Arc.ReadHeader();
|
||
|
|
if (Arc.GetHeaderType()==FILE_HEAD)
|
||
|
|
{
|
||
|
|
Arc.ConvertAttributes();
|
||
|
|
Arc.Seek(Arc.NextBlockPos-Arc.NewLhd.FullPackSize,SEEK_SET);
|
||
|
|
}
|
||
|
|
#ifndef GUI
|
||
|
|
if (ShowFileName)
|
||
|
|
{
|
||
|
|
char OutName[NM];
|
||
|
|
IntToExt(Arc.NewLhd.FileName,OutName);
|
||
|
|
#ifdef UNICODE_SUPPORTED
|
||
|
|
bool WideName=(Arc.NewLhd.Flags & LHD_UNICODE) && UnicodeEnabled();
|
||
|
|
if (WideName)
|
||
|
|
{
|
||
|
|
wchar NameW[NM];
|
||
|
|
ConvertPath(Arc.NewLhd.FileNameW,NameW);
|
||
|
|
char Name[NM];
|
||
|
|
if (WideToChar(NameW,Name) && IsNameUsable(Name))
|
||
|
|
strcpy(OutName,Name);
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
mprintf(St(MExtrPoints),OutName);
|
||
|
|
if (!Cmd->DisablePercentage)
|
||
|
|
mprintf(" ");
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
if (DataIO!=NULL)
|
||
|
|
{
|
||
|
|
if (HeaderType==ENDARC_HEAD)
|
||
|
|
DataIO->UnpVolume=false;
|
||
|
|
else
|
||
|
|
{
|
||
|
|
DataIO->UnpVolume=(hd->Flags & LHD_SPLIT_AFTER)!=0;
|
||
|
|
DataIO->SetPackedSizeToRead(hd->FullPackSize);
|
||
|
|
}
|
||
|
|
#ifdef SFX_MODULE
|
||
|
|
DataIO->UnpArcSize=Arc.FileLength();
|
||
|
|
#endif
|
||
|
|
|
||
|
|
// Reset the size of packed data read from current volume. It is used
|
||
|
|
// to display the total progress and preceding volumes are already
|
||
|
|
// compensated with ProcessedArcSize, so we need to reset this variable.
|
||
|
|
DataIO->CurUnpRead=0;
|
||
|
|
|
||
|
|
DataIO->PackedCRC=0xffffffff;
|
||
|
|
// DataIO->SetFiles(&Arc,NULL);
|
||
|
|
}
|
||
|
|
return(true);
|
||
|
|
}
|
||
|
|
|
||
|
|
#if defined(RARDLL) && defined(_MSC_VER) && !defined(_M_X64)
|
||
|
|
// Restore the run time stack check for unrar.dll.
|
||
|
|
#pragma runtime_checks( "s", restore )
|
||
|
|
#endif
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
#ifndef SILENT
|
||
|
|
bool AskNextVol(char *ArcName)
|
||
|
|
{
|
||
|
|
eprintf(St(MAskNextVol),ArcName);
|
||
|
|
if (Ask(St(MContinueQuit))==2)
|
||
|
|
return(false);
|
||
|
|
return(true);
|
||
|
|
}
|
||
|
|
#endif
|