thhyl.cpp

// thhyl.cpp : Defines the class behaviors for the application.
//

#include "stdafx.h"
#include "cfgfile.h"
#include "thhyl.h"
#include "thhylDlg.h"
#include "RPYAnalyzer.h"
#include "global.h"
#include "indep/clipboard.h"
#include "indep/OSVer.h"

#include "filepath_utils.h"

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

#define delete_then_null(v) ((delete (v)), (v)=NULL)

inline static int MsgBox(LPCTSTR text, UINT options)
{
	return ::MessageBox( NULL, text, g_title, options );
}

/////////////////////////////////////////////////////////////////////////////
// CThhylApp

BEGIN_MESSAGE_MAP(CThhylApp, CWinApp)
	//{{AFX_MSG_MAP(CThhylApp)
		// NOTE - the ClassWizard will add and remove mapping macros here.
		//    DO NOT EDIT what you see in these blocks of generated code!
	//}}AFX_MSG
	ON_COMMAND(ID_HELP, CWinApp::OnHelp)
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CThhylApp construction

CThhylApp::CThhylApp()
{
	// TODO: add construction code here,
	// Place all significant initialization in InitInstance
	m_rpyfile = _T("");
	m_bOnTop = TRUE;
}

/////////////////////////////////////////////////////////////////////////////
// The one and only CThhylApp object

CThhylApp theApp;

/////////////////////////////////////////////////////////////////////////////
// CThhylApp initialization

//全局变量之初始化
static TCHAR s_exefullpath[MAX_PATH] = {0};
LPCTSTR g_exefullpath = (LPCTSTR)s_exefullpath;
////////////////

//分析命令行参数
void CThhylApp::ParseArguments()
{
	int argc;
	LPWSTR* argv=CommandLineToArgvW(GetCommandLineW(), &argc);
	ASSERT(argc>0 && argv!=NULL);

	//EXE 完整路径
	_tcsncpy(s_exefullpath, argv[0], MAX_PATH);
	s_exefullpath[MAX_PATH-1] = _T('\0');

	//config无视工作目录直接写在EXE所在目录下。
	CString cfgfilename = argv[0];
	filepath_utils::ChangeFileExtension(cfgfilename, _T("cfg"));
	SetConfigFilename(cfgfilename);
	LoadConfig();

	for (int i=1;i<argc;i++) {
		if (wcscmp(argv[i], _T("-n")) == 0)
			m_bOnTop = FALSE;
		else if (i == 1)
			m_rpyfile = argv[1];
	}

	LocalFree(argv); //释放argv所占内存
}

BOOL CThhylApp::InitInstance()
{
	AfxEnableControlContainer();

	// Standard initialization
	// If you are not using these features and wish to reduce the size
	//  of your final executable, you should remove from the following
	//  the specific initialization routines you do not need.

#ifdef _AFXDLL
	Enable3dControls();			// Call this when using MFC in a shared DLL
#else
	Enable3dControlsStatic();	// Call this when linking to MFC statically
#endif

	ParseArguments();
	OSVerInit();

	//检查是否按下ScrollLock
	if ( IsKeyDown(VK_SCROLL) && !m_rpyfile.IsEmpty()) { //按下，并且有录像文件名
		AnalyzeAndCopy(m_rpyfile); //分析录像，并将信息复制到剪贴板
	}
	else {
		CThhylDlg dlg;
		m_pMainWnd = &dlg;
		dlg.DoModal();
		
		SaveConfig();
	}

	// Since the dialog has been closed, return FALSE so that we exit the
	//  application, rather than start the application's message pump.
	return FALSE;

}



// return NULL if not found
static void* memfind(const void* buf, size_t buf_len,  const void* tofind, size_t tofind_len)
{
	const BYTE* const end_buf = (const BYTE*)buf + buf_len;
	for (const BYTE* curr_buf = (const BYTE*)buf; curr_buf < end_buf; ++curr_buf ) {
		if ( memcmp( curr_buf, tofind, tofind_len) == 0 ) {
			return (void*)curr_buf;
		}
	}

	return NULL;
}


// 对于国人制作的以下仿东方STG的REP文件，目前可以识别，但无法解析：
// 《东方幕华祭》、《东方百花宴》、《东方音乐绘》、《东方夏夜祭》
// 返回TRUE表示该 REP 疑似以上几个游戏之一，并在 strOutGameName 标明游戏名称
static BOOL IsUnanalyzableRPY(const BYTE* pData, DWORD dwFileSize, CString& strOutGameName)
{
	// 检查是不是 东方幕华祭……这貌似是固定的文件头？
	const char mhj_header[] = "\xF0\x67\x82\xFF\xCA\xE3\xC8\x5B\xFC\x91\xB5\x72\xCB\x84\xC2\xEF";
	BOOL result = FALSE;

	if ( memcmp(mhj_header, pData, sizeof(mhj_header) - 1) == 0 ) {
		strOutGameName = _T("東方幕華祭");
		result = TRUE;
	}
	else if ( *((WORD*)pData) == 0xe00e ) {
		const char start_of_info[] = "\x0e\xe0\xf0\x0f\x0d\x0a";
		size_t sizeof_info = sizeof(start_of_info) - 1;
		pData += 2;
		dwFileSize -= 2;
		const BYTE* pStartOfInfo = static_cast <const BYTE*>(memfind(pData, (size_t)dwFileSize, start_of_info, sizeof_info));
		if ( pStartOfInfo ) {
			strOutGameName = _T("东方百花宴/弹幕音乐绘/東方夏夜祭");
			result = TRUE;
		}
	}

	return result;
}

BYTE* ReadRPYFile(LPCTSTR szFileName, CString& strErrorInfo, HWND hWnd, CFileStatus* pFileStatus, DWORD& dwFileSize)
{
	CFile          cfRpyFile;
	CFileException ex;
	
	if ( !cfRpyFile.Open(szFileName, CFile::modeRead | CFile::shareDenyWrite | CFile::typeBinary, &ex) )
	{
		ex.GetErrorMessage(strErrorInfo.GetBuffer(512), 512);
		strErrorInfo.ReleaseBuffer();
		return NULL;
	}
	else {
		if ( !CheckRPYFileSize((HANDLE)cfRpyFile.m_hFile, hWnd)
			|| (pFileStatus != NULL && !cfRpyFile.GetStatus(*pFileStatus)) )
			return NULL;
	}

	cfRpyFile.SeekToBegin();
	
	dwFileSize = cfRpyFile.GetLength();
	BYTE *pBuf = new BYTE[dwFileSize+MINRPYSIZE];
	
	dwFileSize = cfRpyFile.Read(pBuf, dwFileSize+1);
	pBuf[dwFileSize] = '\0';
	cfRpyFile.Close();


	CString strGameName;
	if ( IsUnanalyzableRPY(pBuf, dwFileSize, strGameName) ) {
		strErrorInfo.Format(
			_T("这貌似是 %s 的录像哦……\r\n")
			_T("但是回映录解析不了喵……＝＿＝\r\n")
			, strGameName);
		return delete []pBuf, static_cast<BYTE*>(NULL);
	}

	return pBuf;
}


void CThhylApp::AnalyzeAndCopy(LPCTSTR szFileName)
{
	CString strInfo;
	DWORD dwRead;

	BYTE* pBuf = ReadRPYFile(szFileName, strInfo, NULL, NULL, dwRead);
	if (pBuf == NULL) { 
		if ( !strInfo.IsEmpty() ) {
			MsgBox(strInfo, MB_ICONSTOP);
		}
		return;
	}
	
	//Start
	CRPYAnalyzer* pRpyInfo = new CRPYAnalyzer(pBuf, dwRead, cfg.InfoCode);
	const int RPYAnalyzer_ret = pRpyInfo->Analyze();

	if ( RPYAnalyzer_ret != RPYINFO_UNKNOWNFORMAT ) {
		strInfo = pRpyInfo->GetFinalInfo();
		if ( RPYAnalyzer_ret == RPYINFO_OK ) {
			CString rpycomment(pRpyInfo->GetComment(cfg.CommentCode));
			if (!rpycomment.IsEmpty()) {
				strInfo += CString(_T("\r\n注释：\r\n")) + rpycomment;
			}
		}
	}
	delete_then_null(pRpyInfo);
	
	//将录像信息复制到剪贴板
	if (RPYAnalyzer_ret==RPYINFO_OK && !strInfo.IsEmpty()) {
		int ret = IDCANCEL;
		CString failedinfo;

		do {
			if ( ::SetClipTextW(strInfo) ) {
				MessageBeep(MB_ICONINFORMATION);
				break;
			}
			else {
				LPCTSTR ErrorMsg = GetErrorMessage( GetLastError() );
				failedinfo.Format(_T("复制数据到剪贴板失败！\n\n可能的原因：%s"), ErrorMsg);
				ret = MsgBox( failedinfo, MB_ICONSTOP | MB_RETRYCANCEL );
			}
		} while (ret==IDRETRY);
	}
	else {
		if (strInfo.IsEmpty())
			strInfo = LoadNotRpyString();

		CString msg;
		msg.Format(_T("%s\n\n%s"), szFileName, strInfo);
		MsgBox(msg, MB_ICONSTOP);
	}

	delete []pBuf;
}