//this file is part of eMule
//Copyright (C)2002-2007 Merkur ( strEmail.Format("%s@%s", "devteam", "emule-project.net") / http://www.emule-project.net )
//
//This program is free software; you can redistribute it and/or
//modify it under the terms of the GNU General Public License
//as published by the Free Software Foundation; either
//version 2 of the License, or (at your option) any later version.
//
//This program is distributed in the hope that it will be useful,
//but WITHOUT ANY WARRANTY; without even the implied warranty of
//MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.	See the
//GNU General Public License for more details.
//
//You should have received a copy of the GNU General Public License
//along with this program; if not, write to the Free Software
//Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
#include "stdafx.h"
#include "emule.h"
#include "ArchivePreviewDlg.h"
#include "KnownFile.h"
#include "partfile.h"
#include "preferences.h"
#include "UserMsgs.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif


// from free unRAR, Alexander L. Roshal
class EncodeFileName
{
  private:
	byte *EncName;
	byte Flags;
	int FlagBits;
	int FlagsPos;
	int DestSize;
  public:
	EncodeFileName();
	int Encode(char *Name,wchar_t *NameW,byte *EncName);
	void Decode(char *Name,byte *EncName,int EncSize,wchar_t *NameW,int MaxDecSize);
};

EncodeFileName::EncodeFileName()
{
  Flags=0;
  FlagBits=0;
  FlagsPos=0;
  DestSize=0;
}

void EncodeFileName::Decode(char *Name,byte *EncName,int EncSize,wchar_t *NameW,
							int MaxDecSize)
{
  int EncPos=0,DecPos=0;
  byte HighByte=EncName[EncPos++];
  while (EncPos<EncSize && DecPos<MaxDecSize)
  {
	if (FlagBits==0)
	{
	  Flags=EncName[EncPos++];
	  FlagBits=8;
	}
	switch(Flags>>6)
	{
	  case 0:
		NameW[DecPos++]=EncName[EncPos++];
		break;
	  case 1:
		NameW[DecPos++]=EncName[EncPos++]+(HighByte<<8);
		break;
	  case 2:
		NameW[DecPos++]=EncName[EncPos]+(EncName[EncPos+1]<<8);
		EncPos+=2;
		break;
	  case 3:
		{
		  int Length=EncName[EncPos++];
		  if (Length & 0x80)
		  {
			byte Correction=EncName[EncPos++];
			for (Length=(Length&0x7f)+2;Length>0 && DecPos<MaxDecSize;Length--,DecPos++)
			  NameW[DecPos]=((Name[DecPos]+Correction)&0xff)+(HighByte<<8);
		  }
		  else
			for (Length+=2;Length>0 && DecPos<MaxDecSize;Length--,DecPos++)
			  NameW[DecPos]=Name[DecPos];
		}
		break;
	}
	Flags<<=2;
	FlagBits-=2;
  }
  NameW[DecPos<MaxDecSize ? DecPos:MaxDecSize-1]=0;
}
/////////////////////////



//////////////////////////////////////////////////////////////////////////////
// COLUMN_INIT -- List View Columns

enum EArchiveCols
{
	ARCHPREV_COL_NAME = 0,
	ARCHPREV_COL_SIZE,
	ARCHPREV_COL_CRC,
	ARCHPREV_COL_ATTR,
	ARCHPREV_COL_TIME,
	ARCHPREV_COL_CMT
};

static LCX_COLUMN_INIT _aColumns[] =
{
	{ ARCHPREV_COL_NAME, _T("Name"),  IDS_DL_FILENAME,  LVCFMT_LEFT,  -1, 0, ASCENDING, NONE, _T("LONG FILENAME.DAT") },
	{ ARCHPREV_COL_SIZE, _T("Size"),  IDS_DL_SIZE,      LVCFMT_RIGHT, -1, 1, ASCENDING, NONE, _T("9999 MByte") },
	{ ARCHPREV_COL_CRC,  _T("CRC"),   (UINT)-1,         LVCFMT_LEFT,  -1, 2, ASCENDING, NONE, _T("1234abcd")  },
	{ ARCHPREV_COL_ATTR, _T("Attr"),  IDS_ATTRIBUTES,   LVCFMT_LEFT,  -1, 3, ASCENDING, NONE, _T("mmm") },
	{ ARCHPREV_COL_TIME, _T("Time"),  IDS_LASTMODIFIED, LVCFMT_LEFT,  -1, 4, ASCENDING, NONE, _T("12.12.1990 00:00:00") },
	{ ARCHPREV_COL_CMT,  _T("Cmt"),   IDS_COMMENT,      LVCFMT_LEFT,  -1, 5, ASCENDING, NONE, _T("Long long long long comment") }
};

#define	PREF_INI_SECTION	_T("ArchivePreviewDlg")

IMPLEMENT_DYNAMIC(CArchivePreviewDlg, CResizablePage)

BEGIN_MESSAGE_MAP(CArchivePreviewDlg, CResizablePage)
	ON_BN_CLICKED(IDC_READARCH, OnBnClickedRead)
	ON_BN_CLICKED(IDC_RESTOREARCH, OnBnClickedCreateRestored)
	ON_BN_CLICKED(IDC_AP_EXPLAIN, OnBnExplain)
	ON_MESSAGE(UM_DATA_CHANGED, OnDataChanged)
	ON_MESSAGE(UM_ARCHIVESCANDONE, ShowScanResults)
	ON_WM_DESTROY()
	ON_NOTIFY(LVN_DELETEALLITEMS, IDC_FILELIST, OnLvnDeleteAllItemsArchiveEntries)
	ON_NOTIFY(NM_CUSTOMDRAW, IDC_FILELIST, OnNMCustomDrawArchiveEntries)
END_MESSAGE_MAP()

CArchivePreviewDlg::CArchivePreviewDlg()
   : CResizablePage(CArchivePreviewDlg::IDD, 0)
{
	m_paFiles = NULL;
	m_bDataChanged = false;
	m_strCaption = GetResString(IDS_CONTENT_INFO);
	m_psp.pszTitle = m_strCaption;
	m_psp.dwFlags |= PSP_USETITLE;
	m_activeTParams = NULL;
	m_ContentList.m_pParent = this;
	m_ContentList.SetRegistryKey(PREF_INI_SECTION);
	m_ContentList.SetRegistryPrefix(_T("ContentList_"));
	//m_ContentList.m_pfnFindItem = FindItem;
	//m_ContentList.m_lFindItemParam = (DWORD_PTR)this;
}

CArchivePreviewDlg::~CArchivePreviewDlg()
{
}

void CArchivePreviewDlg::DoDataExchange(CDataExchange* pDX)
{
	CResizablePage::DoDataExchange(pDX);
	DDX_Control(pDX, IDC_FILELIST, m_ContentList);
}

BOOL CArchivePreviewDlg::OnInitDialog()
{
	CResizablePage::OnInitDialog();
	InitWindowStyles(this);

	AddAnchor(IDC_READARCH, BOTTOM_LEFT);
	AddAnchor(IDC_RESTOREARCH, BOTTOM_LEFT);
	AddAnchor(IDC_FILELIST, TOP_LEFT, BOTTOM_RIGHT);
	AddAnchor(IDC_APV_FILEINFO, TOP_LEFT, TOP_RIGHT);
	AddAnchor(IDC_ARCP_ATTRIBS, TOP_CENTER);
	AddAnchor(IDC_INFO_ATTR, TOP_CENTER, TOP_RIGHT);
	AddAnchor(IDC_INFO_FILECOUNT, BOTTOM_RIGHT);
	AddAnchor(IDC_AP_EXPLAIN, BOTTOM_LEFT);
	AddAnchor(IDC_INFO_STATUS, TOP_LEFT, TOP_RIGHT);
	AddAnchor(IDC_ARCHPROGRESS, BOTTOM_LEFT, BOTTOM_RIGHT);

	// Win98: Explicitly set to Unicode to receive Unicode notifications.
	m_ContentList.SendMessage(CCM_SETUNICODEFORMAT, TRUE);
	// To support full sorting of the archive entries list we'd need a seperate list which
	// is holding the unified archive entries for all different supported archive formats so
	// that the ListView's sortproc can get valid 'lParam' values which are pointing to those
	// entries. This could be done, but for now we just let the default ListView sort
	// functionality sort the archive entries by filename (the content of the first column).
	ASSERT( m_ContentList.GetStyle() & LVS_SORTASCENDING );
	ASSERT( m_ContentList.GetStyle() & LVS_SHAREIMAGELISTS );
	m_ContentList.SendMessage(LVM_SETIMAGELIST, LVSIL_SMALL, (LPARAM)theApp.GetSystemImageList());
	m_ContentList.SetExtendedStyle(LVS_EX_FULLROWSELECT | LVS_EX_GRIDLINES | LVS_EX_INFOTIP);
	m_ContentList.EnableHdrCtrlSortBitmaps();
	m_ContentList.ReadColumnStats(_countof(_aColumns), _aColumns);
	m_ContentList.CreateColumns(_countof(_aColumns), _aColumns);
	m_ContentList.InitColumnOrders(_countof(_aColumns), _aColumns);
	m_ContentList.UpdateSortColumn(_countof(_aColumns), _aColumns);

	CResizablePage::UpdateData(FALSE);
	Localize();

	m_progressbar=(CProgressCtrl*)GetDlgItem(IDC_ARCHPROGRESS);
	m_progressbar->SetRange(0,1000);
	m_progressbar->SetPos(0);

	return TRUE;
}

void CArchivePreviewDlg::OnDestroy()
{
	m_ContentList.WriteColumnStats(_countof(_aColumns), _aColumns);

	// stop running scanner-thread if present
	if (m_activeTParams != NULL) {
		m_activeTParams->m_bIsValid = false;
		m_activeTParams = NULL;
	}

	// This property sheet's window may get destroyed and re-created several times although
	// the corresponding C++ class is kept -> explicitly reset ResizeableLib state
	RemoveAllAnchors();

	CResizablePage::OnDestroy();
}

BOOL CArchivePreviewDlg::OnSetActive()
{
	if (!CResizablePage::OnSetActive())
		return FALSE;
	
	if (m_bDataChanged)
	{
		UpdateArchiveDisplay(thePrefs.m_bAutomaticArcPreviewStart);
		m_bDataChanged = false;
	}
	return TRUE;
}

LRESULT CArchivePreviewDlg::OnDataChanged(WPARAM, LPARAM)
{
	m_bDataChanged = true;
	return 1;
}

void CArchivePreviewDlg::Localize(void)
{
	SetDlgItemText(IDC_READARCH,	GetResString(IDS_SV_UPDATE) );
	SetDlgItemText(IDC_RESTOREARCH, GetResString(IDS_AP_CREATEPREVCOPY) );
	SetDlgItemText(IDC_ARCP_TYPE ,	GetResString(IDS_ARCHTYPE)+_T(":") );
	SetDlgItemText(IDC_ARCP_STATUS,	GetResString(IDS_STATUS)+_T(":") );
	SetDlgItemText(IDC_ARCP_ATTRIBS,GetResString(IDS_INFO)+_T(":")  );
}

void CArchivePreviewDlg::OnLvnDeleteAllItemsArchiveEntries(NMHDR *, LRESULT *pResult)
{
	// To suppress subsequent LVN_DELETEITEM notification messages, return TRUE.
	*pResult = TRUE;
}

void CArchivePreviewDlg::OnNMCustomDrawArchiveEntries(NMHDR *pNMHDR, LRESULT *plResult)
{
	LPNMLVCUSTOMDRAW pnmlvcd = (LPNMLVCUSTOMDRAW)pNMHDR;

	if (pnmlvcd->nmcd.dwDrawStage == CDDS_PREPAINT) {
		*plResult = CDRF_NOTIFYITEMDRAW;
		return;
	}

	if (pnmlvcd->nmcd.dwDrawStage == CDDS_ITEMPREPAINT)
	{
		if (pnmlvcd->nmcd.lItemlParam == 1) {
			// Show not available or incomplete entries in gray.
			pnmlvcd->clrText = RGB(128,128,128);
		}
	}

	*plResult = CDRF_DODEFAULT;
}

void CArchivePreviewDlg::OnBnExplain() {
	AfxMessageBox(GetResString(IDS_ARCHPREV_ATTRIBINFO), MB_OK | MB_ICONINFORMATION);
}

void CArchivePreviewDlg::OnBnClickedRead()
{
	UpdateArchiveDisplay(true);
}

void CArchivePreviewDlg::OnBnClickedCreateRestored() {
	if ( ((CKnownFile*)STATIC_DOWNCAST(CKnownFile, (*m_paFiles)[0])->IsPartFile()==false))
		return;

	CPartFile* file=STATIC_DOWNCAST(CPartFile, (*m_paFiles)[0]);

	if (!file->m_bRecoveringArchive && !file->m_bPreviewing)
			CArchiveRecovery::recover((CPartFile*)file, true, thePrefs.GetPreviewCopiedArchives());
}

// ###########################################################################

/*****************************************
				A C E
 *****************************************/
int CArchivePreviewDlg::ShowAceResults(int succ, archiveScannerThreadParams_s* tp) {

	if (succ==0) {
		SetDlgItemText(IDC_INFO_STATUS, GetResString(IDS_ARCPREV_INSUFFDATA));
		return -1;
	}

	// file content into list
	bool statusEncrypted=false;
	CString temp;
	UINT uArchiveFileEntries = 0;

	SetDlgItemText(IDC_INFO_STATUS, GetResString(IDS_ARCPARSED) + _T(" ") +
		(tp->ai->ACEdir->IsEmpty()?GetResString(IDS_ARCPREV_INSUFFDATA):
		(tp->file->IsPartFile()?GetResString(IDS_ARCPREV_LISTMAYBEINCOMPL):_T("")))		);

	if (!tp->ai->ACEdir->IsEmpty())
	{
		CTime lm;
		int iItem;
		int uSubId=1;
		ACE_BlockFile *block;

		m_ContentList.SetRedraw(FALSE);
		POSITION pos = tp->ai->ACEdir->GetHeadPosition();
		while (pos != NULL)
		{
			block = tp->ai->ACEdir->GetNext(pos);
			uSubId = 1;

			bool bCompleteEntry = !tp->file->IsPartFile() || STATIC_DOWNCAST(CPartFile, tp->file)->IsComplete(block->data_offset, block->data_offset + block->PACK_SIZE, true);
			bool bIsDirectory = ((block->FILE_ATTRIBS & FILE_ATTRIBUTE_DIRECTORY) != 0);
			if (!bIsDirectory)
				uArchiveFileEntries++;

			// file/folder name
			temp = CString(block->FNAME);
			int iSystemImage = bIsDirectory ? theApp.GetFileTypeSystemImageIdx(_T("\\"), 1) : theApp.GetFileTypeSystemImageIdx(temp, temp.GetLength());
			iItem = m_ContentList.InsertItem(LVIF_TEXT | LVIF_PARAM | (iSystemImage > 0 ? LVIF_IMAGE : 0),
											 INT_MAX, temp, 0, 0, iSystemImage, 
											 !bCompleteEntry ? 0x00000001 : 0x00000000);

			if (bIsDirectory)
				uSubId += 3;
			else {
				// size (uncompressed)
				UINT64 size=block->ORIG_SIZE;
				temp.Format(_T("%s"), CastItoXBytes(size));
				m_ContentList.SetItemText(iItem,uSubId++,temp);

				// crc
				temp.Format(_T("%08X"),block->CRC32);
				m_ContentList.SetItemText(iItem,uSubId++,temp);
			}

			// attribs
			temp.Empty();

			if (block->HEAD_FLAGS & 0x4000) {
				statusEncrypted=true;
				temp+=_T('P');
			}

			if (bIsDirectory) {
				if (!temp.IsEmpty()) temp+=_T(',');
				temp+=_T('D');
			}

			if (block->HEAD_FLAGS & 0x2000) {
				if (!temp.IsEmpty()) temp+=_T(',');
				temp+=_T('>');
			}
			if (block->HEAD_FLAGS & 0x1000) {
				if (!temp.IsEmpty()) temp+=_T(',');
				temp+=_T('<');
			}
			if (block->HEAD_FLAGS & 0x02) {
				// file comment - theoretically
				if (!temp.IsEmpty()) temp+=_T(',');
				temp+=_T('C');
			}

			if (!bCompleteEntry) {
				// packed data not yet downloaded
				if (!temp.IsEmpty()) temp+=_T(',');
				temp+=_T('M');
			}

			if (!bIsDirectory) {
				// compression level
				if (!temp.IsEmpty()) temp+=_T(", ");
				if ((BYTE)block->TECHINFO==0)
					temp.Append(_T("L0"));
				else
					temp.AppendFormat(_T("L%i"), (BYTE)(block->TECHINFO>>8) );
				m_ContentList.SetItemText(iItem,uSubId++,temp);
			}

			// date time
			UINT16 fdate,ftime;
			ftime=(UINT16)block->FTIME;
			fdate=(UINT16)(block->FTIME>>16);
			lm=CTime((WORD)(((fdate>>9) & 0x7f) + 1980),
					 (WORD)((fdate>>5) & 0xf),
					 (WORD)(fdate & 0x1f),
					 (WORD)((ftime >> 11) & 0x1f),
					 (WORD)((ftime >> 5) & 0x3f),
					 (WORD)((ftime & 0x1f) * 2));
			m_ContentList.SetItemText(iItem,uSubId++,lm.Format(thePrefs.GetDateTimeFormat4Log()));

			// cleanup
			delete block;
		}
		m_ContentList.SetRedraw(TRUE);
	}

	temp.Format(_T("%s: %u"), GetResString(IDS_FILES), uArchiveFileEntries);
	SetDlgItemText( IDC_INFO_FILECOUNT, temp);

	// general info / archive attribs
	CString status;
	if (statusEncrypted)
		status+= GetResString(IDS_PASSWPROT);

	if (tp->ai->ACEhdr) {
		if (tp->ai->ACEhdr->HEAD_FLAGS & 0x8000){
			if (!status.IsEmpty()) status+=_T(',');
			status+= _T("Solid");
		}
		if (tp->ai->ACEhdr->HEAD_FLAGS & 0x4000){
			if (!status.IsEmpty()) status+=_T(',');
			status+= _T("Locked");
		}
		if (tp->ai->ACEhdr->HEAD_FLAGS & 0x2000){
			if (!status.IsEmpty()) status+=_T(',');
			status+= _T("RecoveryRec");
		}
		if (tp->ai->ACEhdr->COMMENT_SIZE > 0){
			if (!status.IsEmpty()) status+=_T(',');
			status+= GetResString(IDS_COMMENT);
		}
	}

	SetDlgItemText(IDC_INFO_ATTR, status);
	//... any other info?

	delete tp->ai->ACEdir;
	tp->ai->ACEdir=NULL;
	if (tp->ai->ACEhdr) {
		delete tp->ai->ACEhdr;
		tp->ai->ACEhdr=NULL;
	}

	return 0;
}

/*****************************************
				R A R
 *****************************************/
int CArchivePreviewDlg::ShowRarResults(int succ, archiveScannerThreadParams_s* tp) {

	if (succ==0) {
		SetDlgItemText(IDC_INFO_STATUS, GetResString(IDS_ARCPREV_INSUFFDATA));
		return -1;
	}

	// file content into list
	bool statusEncrypted=false;
	CString temp;
	UINT uArchiveFileEntries = 0;

	if (tp->ai->rarFlags & 0x0080){
		SetDlgItemText(IDC_INFO_STATUS, GetResString(IDS_RARFULLENCR));
		statusEncrypted=true;
	}
	else
		SetDlgItemText(IDC_INFO_STATUS, GetResString(IDS_ARCPARSED) + _T(" ") +
			(tp->ai->RARdir->IsEmpty()?GetResString(IDS_ARCPREV_INSUFFDATA):
				(tp->file->IsPartFile()?GetResString(IDS_ARCPREV_LISTMAYBEINCOMPL):_T("")))		);

	if (!tp->ai->RARdir->IsEmpty())
	{
		char buf[MAX_PATH + MAX_PATH*2];
		CStringA strBuffA;
		CTime lm;
		int iItem;
		int uSubId = 1;
		RAR_BlockFile *block;

		m_ContentList.SetRedraw(FALSE);
		POSITION pos = tp->ai->RARdir->GetHeadPosition();
		while (pos != NULL)
		{
			block = tp->ai->RARdir->GetNext(pos);
			uSubId=1;

			bool bCompleteEntry = !tp->file->IsPartFile() || STATIC_DOWNCAST(CPartFile, tp->file)->IsComplete(block->offsetData, block->offsetData + block->dataLength, true);
			bool bIsDirectory = (block->HEAD_FLAGS & 0xE0) == 0xE0;
			if (!bIsDirectory)
				uArchiveFileEntries++;

			// file/folder name
			memcpy(buf, block->FILE_NAME, block->NAME_SIZE_var); // MORPH leuk_he rename to prevent conflict with upnp lib define (dumb...)
			buf[block->NAME_SIZE_var] = 0; // MORPH leuk_he rename to prevent conflict with upnp lib define (dumb...)
			// read unicode name from namebuffer and decode it
			if (block->HEAD_FLAGS & 0x0200) {
				unsigned int asciilen = strlen(buf) + 1;
				wchar_t nameuc[MAX_PATH];
				EncodeFileName enc;
				enc.Decode(buf, (byte*)buf + asciilen, block->NAME_SIZE_var - asciilen, nameuc, _countof(nameuc));  // MORPH leuk_he rename to prevent conflict with upnp lib define (dumb...)
				temp = nameuc;
			} else
				temp = buf;

			int iSystemImage = bIsDirectory ? theApp.GetFileTypeSystemImageIdx(_T("\\"), 1) : theApp.GetFileTypeSystemImageIdx(temp, temp.GetLength());
			iItem = m_ContentList.InsertItem(LVIF_TEXT | LVIF_PARAM | (iSystemImage > 0 ? LVIF_IMAGE : 0),
											 INT_MAX, temp, 0, 0, iSystemImage, 
											 !bCompleteEntry ? 0x00000001 : 0x00000000);

			if ( (block->HEAD_FLAGS & 0xE0) != 0xE0) {

				// size (uncompressed)
				UINT64 size=block->UNP_SIZE;
				if (block->ATTR & 0x100)
					size+=block->HIGH_UNP_SIZE*0x100000000;
				temp.Format(_T("%s"), CastItoXBytes(size));
				m_ContentList.SetItemText(iItem,uSubId++,temp);

				// crc
				temp.Format(_T("%08X"),block->FILE_CRC);
				m_ContentList.SetItemText(iItem,uSubId++,temp);

			} else
				uSubId+=2;

			// attribs
			temp.Empty();

			if (block->HEAD_FLAGS & 0x04)
			{
				statusEncrypted=true;
				temp+=_T('P');
			}

			//is directory ?
			if (bIsDirectory){
				if (!temp.IsEmpty()) temp+=_T(',');
				temp+=_T('D');
			}

			if (block->HEAD_FLAGS & 0x01) {
				if (!temp.IsEmpty()) temp+=_T(',');
				temp+=_T('<');
			}
			if (block->HEAD_FLAGS & 0x02) {
				if (!temp.IsEmpty()) temp+=_T(',');
				temp+=_T('>');
			}
			if (block->HEAD_FLAGS & 0x08) {
				// file comment - theoretically
				if (!temp.IsEmpty()) temp+=_T(',');
				temp+=_T('C');
			}

			if (!bCompleteEntry) {
				// packed data not yet downloaded
				if (!temp.IsEmpty()) temp+=_T(',');
				temp+=_T('M');
			}

			if (!temp.IsEmpty()) temp+=_T(", ");
			temp.AppendFormat(_T("L%i"),block->METHOD-0x30);

			m_ContentList.SetItemText(iItem,uSubId++,temp);

			// date time
			UINT16 fdate,ftime;
			ftime=(UINT16)block->FTIME;
			fdate=(UINT16)(block->FTIME>>16);
			lm=CTime((WORD)(((fdate>>9) & 0x7f) + 1980),
					 (WORD)((fdate>>5) & 0xf),
					 (WORD)(fdate & 0x1f),
					 (WORD)((ftime >> 11) & 0x1f),
					 (WORD)((ftime >> 5) & 0x3f),
					 (WORD)((ftime & 0x1f) * 2));
			m_ContentList.SetItemText(iItem,uSubId++,lm.Format(thePrefs.GetDateTimeFormat4Log()));

			// cleanup
			delete [] block->FILE_NAME;
			delete block;
		}
		m_ContentList.SetRedraw(TRUE);
	}

	temp.Format(_T("%s: %u"), GetResString(IDS_FILES), uArchiveFileEntries);
	SetDlgItemText(IDC_INFO_FILECOUNT, temp);

	// general info / archive attribs
	CString status;
	if (statusEncrypted)
		status+= GetResString(IDS_PASSWPROT);

	if (tp->ai->rarFlags & 0x0008){
		if (!status.IsEmpty()) status+=_T(',');
		status+= _T("Solid");
	}
	if (tp->ai->rarFlags & 0x0004){
		if (!status.IsEmpty()) status+=_T(',');
		status+= _T("Locked");
	}
	if (tp->ai->rarFlags & 0x0040){
		if (!status.IsEmpty()) status+=_T(',');
		status+= _T("RecoveryRec");
	}
	if (tp->ai->rarFlags & 0x0002){
		if (!status.IsEmpty()) status+=_T(',');
		status+= GetResString(IDS_COMMENT);
	}

	SetDlgItemText(IDC_INFO_ATTR, status);
	//... any other info?

	delete tp->ai->RARdir;
	tp->ai->RARdir=NULL;

	return 0;
}

/*****************************************
				Z I P
 *****************************************/
int CArchivePreviewDlg::ShowZipResults(int succ, archiveScannerThreadParams_s* tp) {

	if (succ==0) {
		SetDlgItemText(IDC_INFO_STATUS, GetResString(IDS_ARCPREV_INSUFFDATA));
		return 1;
	}

	// file content into list
	CString temp;
	UINT uArchiveFileEntries = 0;

	if (tp->ai->bZipCentralDir)
		SetDlgItemText(IDC_INFO_STATUS, GetResString(IDS_ARCPREV_DIRSUCCREAD) );
	else
		SetDlgItemText(IDC_INFO_STATUS, GetResString(IDS_ARCPREV_DIRNOTFOUND)	+ _T(" ") +
										GetResString(IDS_ARCPARSED)				+ _T(" ") +
										(tp->ai->centralDirectoryEntries->IsEmpty()?GetResString(IDS_ARCPREV_INSUFFDATA):
										GetResString(IDS_ARCPREV_LISTMAYBEINCOMPL)) );

	bool statusEncrypted=false;
	if (!tp->ai->centralDirectoryEntries->IsEmpty())
	{
		CStringA strBuffA;
		CTime lm;
		int iItem;
		int uSubId=1;
		ZIP_CentralDirectory *cdEntry;
		m_ContentList.SetRedraw(FALSE);
		POSITION pos = tp->ai->centralDirectoryEntries->GetHeadPosition();
		while (pos != NULL)
		{
			cdEntry = tp->ai->centralDirectoryEntries->GetNext(pos);
			uSubId = 1;

			// compressed data not available yet?
			bool bCompleteEntry = true;
			if (tp->file->IsPartFile()) {
				const CPartFile* pf = STATIC_DOWNCAST(CPartFile, tp->file);
				UINT64 dataoffset = cdEntry->relativeOffsetOfLocalHeader 
									+ sizeof(ZIP_Entry) - (3 * sizeof(BYTE*)) 
									+ cdEntry->lenComment 
									+ cdEntry->lenFilename 
									+ cdEntry->lenExtraField;
				bCompleteEntry = pf->IsComplete(dataoffset, dataoffset + cdEntry->lenCompressed, true);
			}

			// Is directory?
			bool bIsDirectory = (cdEntry->externalFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
								|| (cdEntry->crc32 == 0 && cdEntry->lenCompressed == 0 && cdEntry->lenUncompressed == 0 && cdEntry->lenFilename != 0);
			if (!bIsDirectory)
				uArchiveFileEntries++;

			// file/folder name
			strBuffA.SetString((char *)cdEntry->filename, cdEntry->lenFilename);
			temp = CString(strBuffA);
			int iSystemImage = bIsDirectory ? theApp.GetFileTypeSystemImageIdx(_T("\\"), 1) : theApp.GetFileTypeSystemImageIdx(temp, temp.GetLength());
			iItem = m_ContentList.InsertItem(LVIF_TEXT | LVIF_PARAM | (iSystemImage > 0 ? LVIF_IMAGE : 0),
											 INT_MAX, temp, 0, 0, iSystemImage,
											 !bCompleteEntry ? 0x00000001 : 0x00000000);

			// size (uncompressed)
			if (!bIsDirectory) {
				temp.Format(_T("%s"), CastItoXBytes(cdEntry->lenUncompressed));
				m_ContentList.SetItemText(iItem,uSubId++,temp);

				// crc
				temp.Format(_T("%08X"),cdEntry->crc32);
				m_ContentList.SetItemText(iItem,uSubId++,temp);
			} else
				uSubId+=2;

			// attribs
			temp.Empty();
			if (cdEntry->generalPurposeFlag & 0x01 || cdEntry->generalPurposeFlag & 0x40) {
				statusEncrypted=true;
				temp+=_T('P');
			}

			if (bIsDirectory) {
				if (!temp.IsEmpty()) temp+=_T(',');
				temp+=_T('D');
			}

			// compressed data not available yet?
			if (!bCompleteEntry) {
				if (!temp.IsEmpty()) temp+=_T(',');
				temp+=_T('M');
			}

			m_ContentList.SetItemText(iItem,uSubId++,temp);

			// date time
			lm=CTime((WORD)(((cdEntry->lastModFileDate>>9) & 0x7f) + 1980),
					 (WORD)((cdEntry->lastModFileDate>>5) & 0xf),
					 (WORD)(cdEntry->lastModFileDate & 0x1f),
					 (WORD)((cdEntry->lastModFileTime >> 11) & 0x1f),
					 (WORD)((cdEntry->lastModFileTime >> 5) & 0x3f),
					 (WORD)((cdEntry->lastModFileTime & 0x1f) * 2));
			m_ContentList.SetItemText(iItem,uSubId++,lm.Format(thePrefs.GetDateTimeFormat4Log()));

			// comment
			temp.Empty();
			if (cdEntry->lenComment) {
				strBuffA.SetString((char *)cdEntry->comment, cdEntry->lenComment);
				temp = CString(strBuffA);
			}
			m_ContentList.SetItemText(iItem, uSubId++, temp);


			// cleanup
			delete [] cdEntry->filename;
			if (cdEntry->lenExtraField > 0)
				delete [] cdEntry->extraField;
			if (cdEntry->lenComment > 0)
				delete [] cdEntry->comment;
			delete cdEntry;
		}
		m_ContentList.SetRedraw(TRUE);
	}

	temp.Format(_T("%s: %u"), GetResString(IDS_FILES), uArchiveFileEntries);
	SetDlgItemText(IDC_INFO_FILECOUNT, temp);

	// general info / archive attribs
	CString status;
	if (statusEncrypted)
		status+= GetResString(IDS_PASSWPROT);

	SetDlgItemText(IDC_INFO_ATTR, status);
	//... any other info?

	delete tp->ai->centralDirectoryEntries;
	tp->ai->centralDirectoryEntries=NULL;

	return 0;
}


// ##################################################################
static void FreeMemory(void* arg) {

	archiveScannerThreadParams_s* tp=(archiveScannerThreadParams_s*)arg;

	POSITION pos = tp->filled->GetHeadPosition();
	while (pos != NULL)
		delete tp->filled->GetNext(pos);
	tp->filled->RemoveAll();
	delete tp->filled;
	tp->filled=NULL;


	if (tp->ai->RARdir) {
		RAR_BlockFile *block;
		POSITION pos = tp->ai->RARdir->GetHeadPosition();
		while (pos != NULL)
		{
			block = tp->ai->RARdir->GetNext(pos);
			delete [] block->FILE_NAME;
			delete block;
		}
		delete tp->ai->RARdir;
		tp->ai->RARdir=NULL;
	}

	if (tp->ai->centralDirectoryEntries) {
		pos = tp->ai->centralDirectoryEntries->GetHeadPosition();
		while (pos != NULL)
		{
			ZIP_CentralDirectory *cdEntry = tp->ai->centralDirectoryEntries->GetNext(pos);
			// cleanup
			delete [] cdEntry->filename;
			if (cdEntry->lenExtraField > 0)
				delete [] cdEntry->extraField;
			if (cdEntry->lenComment > 0)
				delete [] cdEntry->comment;
			delete cdEntry;
		}
		delete tp->ai->centralDirectoryEntries;
		tp->ai->centralDirectoryEntries=NULL;
	}

	if (tp->ai->ACEhdr){
		delete tp->ai->ACEhdr;
		tp->ai->ACEhdr=NULL;
	}
	if (tp->ai->ACEdir){
		pos = tp->ai->ACEdir->GetHeadPosition();
		while (pos != NULL)
		{
			ACE_BlockFile *block= tp->ai->ACEdir->GetNext(pos);
			delete block;
		}
		delete tp->ai->ACEdir;
		tp->ai->ACEdir=NULL;
	}

	delete tp->ai;
	tp->ai=NULL;

	delete tp;
	tp=NULL;
}

void CArchivePreviewDlg::UpdateArchiveDisplay(bool doscan) {

	// invalidate previous scanning thread if still running
	if (m_activeTParams != NULL){
		m_activeTParams->m_bIsValid = false;
		m_activeTParams = NULL; // thread may still run but is not our active one anymore
	}
	m_progressbar->SetPos(0);

	m_ContentList.DeleteAllItems();
	m_ContentList.UpdateWindow();

	CKnownFile* file=STATIC_DOWNCAST(CKnownFile, (*m_paFiles)[0]);

	GetDlgItem(IDC_RESTOREARCH)->EnableWindow( file->IsPartFile()  && 
		(((CPartFile*)file)->IsArchive(true)) && 
		(((CPartFile*)file)->IsReadyForPreview() )	);

	// set infos
	SetDlgItemText(IDC_APV_FILEINFO, _T("") );
	SetDlgItemText(IDC_INFO_ATTR, _T(""));
	SetDlgItemText(IDC_INFO_STATUS, _T(""));
	SetDlgItemText(IDC_INFO_FILECOUNT, _T(""));

	EFileType type=GetFileTypeEx(file);
	switch(type) {
		case ARCHIVE_ZIP:
			SetDlgItemText(IDC_INFO_TYPE, _T("ZIP"));
			break;
		case ARCHIVE_RAR:
			SetDlgItemText(IDC_INFO_TYPE, _T("RAR"));
			break;
		case ARCHIVE_ACE:
			SetDlgItemText(IDC_INFO_TYPE, _T("ACE"));
			break;
		default:
			SetDlgItemText(IDC_INFO_TYPE, GetResString(IDS_ARCPREV_UNKNOWNFORMAT));
			return;
			break;
	}

	if (!doscan)
		return;

	// preparing archive processing

	archiveinfo_s* ai=new archiveinfo_s;

	// get filled area list
	CTypedPtrList<CPtrList, Gap_Struct*> *filled = new CTypedPtrList<CPtrList, Gap_Struct*>;
	if (file->IsPartFile()) {
		((CPartFile*)(file))->GetFilledList(filled);
		if (filled->GetCount()==0) {
			SetDlgItemText(IDC_INFO_STATUS, GetResString(IDS_ARCPREV_INSUFFDATA));
			delete filled;
			delete ai;
			return;
		}
	}
	else {
		Gap_Struct* gap = new Gap_Struct;
		gap->start=0;
		gap->end=file->GetFileSize();
		filled->AddTail(gap);
	}

	SetDlgItemText(IDC_INFO_STATUS, GetResString(IDS_ARCPREV_PLEASEWAIT) );

	switch(type) {
		case ARCHIVE_ZIP:
			ai->centralDirectoryEntries= new CTypedPtrList<CPtrList, ZIP_CentralDirectory*>;
			break;
		case ARCHIVE_RAR:
			ai->RARdir = new CTypedPtrList<CPtrList, RAR_BlockFile*>;
			break;
		case ARCHIVE_ACE:
			ai->ACEdir = new CTypedPtrList<CPtrList, ACE_BlockFile*>;
			break;
	}

	// prepare threadparams
	archiveScannerThreadParams_s *tp= new archiveScannerThreadParams_s;
	m_activeTParams=tp;
	tp->ai=ai;
	tp->file=file;
	tp->filled=filled;
	tp->type=type;
	tp->ownerHwnd=m_hWnd;
	tp->progressHwnd= GetDlgItem(IDC_ARCHPROGRESS)->m_hWnd;
	tp->m_bIsValid=true;

	// start scanning thread
	if (AfxBeginThread(RunArchiveScanner, (LPVOID)tp, THREAD_PRIORITY_BELOW_NORMAL) == NULL) {
		FreeMemory(tp);
		SetDlgItemText(IDC_INFO_STATUS, GetResString(IDS_ERROR) );
	}
}

UINT AFX_CDECL CArchivePreviewDlg::RunArchiveScanner(LPVOID pParam) {
	DbgSetThreadName("ArchiveScanner");
	//InitThreadLocale();

	archiveScannerThreadParams_s *tp=(archiveScannerThreadParams_s *)pParam;

	int ret=0;
	CFile inFile;

	if (!inFile.Open(tp->file->GetFilePath(), CFile::modeRead | CFile::shareDenyNone)){
		ret=-1;
	} else {
		switch (tp->type) {
			case ARCHIVE_ZIP:
				ret=CArchiveRecovery::recoverZip(&inFile, NULL, tp, tp->filled, (inFile.GetLength() == tp->file->GetFileSize()));
				break;
			case ARCHIVE_RAR:
				ret=CArchiveRecovery::recoverRar(&inFile, NULL, tp, tp->filled);
				break;
			case ARCHIVE_ACE:
				ret=CArchiveRecovery::recoverAce(&inFile, NULL, tp, tp->filled);
				break;
		}

		inFile.Close();
	}

	if (!tp->m_bIsValid || !IsWindow(tp->ownerHwnd) || !::SendMessage(tp->ownerHwnd, UM_ARCHIVESCANDONE, ret, (LPARAM)tp))
		FreeMemory(tp);
	return 0;
}

LRESULT CArchivePreviewDlg::ShowScanResults(WPARAM wParam, LPARAM lParam)
{
	CWaitCursor curHourglass;

	int ret = wParam;
	archiveScannerThreadParams_s* tp = (archiveScannerThreadParams_s*)lParam;
	ASSERT( tp->ownerHwnd == m_hWnd );

	// We may receive 'stopped' archive thread results here, just ignore them (but free the memory)
	if (tp->m_bIsValid)
	{
		m_progressbar->SetPos(0);
		if (ret == -1) {
			SetDlgItemText(IDC_INFO_STATUS, GetResString(IDS_IMP_ERR_IO));
		}
		else if (tp->m_bIsValid) {
			switch (tp->type) {
				case ARCHIVE_ZIP:
					ShowZipResults(ret, tp);
					break;
				case ARCHIVE_RAR:
					ShowRarResults(ret, tp);
					break;
				case ARCHIVE_ACE:
					ShowAceResults(ret, tp);
					break;
			}
		}
		ASSERT( tp == m_activeTParams );
		m_activeTParams = NULL;
	}

	FreeMemory(tp);
	return 1;
}
