/********************************************************************
*
*  Author            : Andrea Allievi (andreaa)
*  Date              : 06/23/2017
*  Filename			 : Log.h
*  Description       : My generic Log class used to write data into a .LOG
*					   file very easily from C++
*					   The LOGTITLE symbol is defined in "stdafx.h" precompiled header
*********************************************************************/

#pragma once
#include <windows.h>

class CLog {
public:
	// Default Constructor
	CLog();

	// Specialized Constructor
	CLog(LPTSTR logFile, LPTSTR logTitle = NULL);

	// Destructor
	~CLog();

	// Copy constructor
	CLog(CLog &log);

	// Set log Title
	void SetLogTitle(LPTSTR strTitle);

	// Create or open a log file
	bool Open(LPTSTR logFile, bool overwrite = false);

	// Get log file name
	const LPTSTR GetLogFileName() { 
		return this->g_strLogFile; }

	const bool IsOpened() {
		return (g_hLogFile && g_hLogFile != INVALID_HANDLE_VALUE);
	}

	// Close this log
	void Close(bool WriteEnd = true);

	// Write a log string (4 parameters max)
	void Write(LPTSTR dbgStr, LPVOID arg1, LPVOID arg2 = NULL, LPVOID arg3 = NULL, LPVOID arg4 = NULL);

	// Write an ANSI log string (4 parameters max)
	void Write(LPSTR dbgStr, LPVOID arg1, LPVOID arg2 = NULL, LPVOID arg3 = NULL, LPVOID arg4 = NULL);

	// Write an ANSI log string (NO parameters)
	void Write(LPSTR dbgStr);

	// Write a Unicode log string (NO parameters)
	void Write(LPWSTR dbgStr);

	// Write a log line (4 parameters max)
	void WriteLine(LPTSTR dbgStr, LPVOID arg1, LPVOID arg2 = NULL, LPVOID arg3 = NULL, LPVOID arg4 = NULL);

	// Write a log line (NO parameters)
	void WriteLine(LPTSTR dbgStr);

	// Flush file to perform actual disk write
	void Flush();

	// Get the current log file handle
	HANDLE GetCurrentLogHandle()			// Modification made on 08/10/2016 to correctly deal with the AntiRansom driver
	{ return g_hLogFile; }

private:
	// Write current time to log
	void WriteCurTime();

private:
	LPTSTR g_strLogFile;			// This instance log file string
	HANDLE g_hLogFile;				// Log file handle of this CLog
	LPTSTR g_strLogTitle;			// Log title (see LOGTITLE definition)
	bool g_bIsAutoDeleteLog;		// Set if this log has to delete itself if there are no writing
	bool g_bAtLeastOneWrite;		// Set if user called at least one Write function
	bool g_bImCopy;					// True if this instance is a copy of another ones
};

#pragma region Version Information Class
class CVersionInfo {
public:
	CVersionInfo();
	CVersionInfo(LPTSTR modName);
	CVersionInfo(HMODULE hModule);
	~CVersionInfo();

	VS_FIXEDFILEINFO GetFixedVersion();
	LPTSTR GetFileVersionString();
	LPTSTR GetProductName();
	LPTSTR GetCompanyName();

private:
	// Helper functions that receive versione information of a specific module (NULL = this executable)
	bool GetModuleVersionInfo(HMODULE hMod = NULL);
	bool GetModuleVersionInfo(LPTSTR modName);

	// Helper function that Get a Language Code Page Version Value
	LPTSTR VerQueryLangCpValue(LPTSTR value); 

	// For lang and code page data
	struct LANGANDCODEPAGE {
		WORD wLanguage;
		WORD wCodePage;
	} * pLangTable;
	UINT iLangTableLen;

	// Lang string 
	LPTSTR g_LangStr;

	// FileVersion String
	LPTSTR g_fVerStr;

	// This module buffer
	LPBYTE g_bVerInfoBuff;

};
#pragma endregion

// Static function to write data to an uninitialized log
void WriteToLog(LPTSTR dbgStr, LPVOID arg1 = NULL, LPVOID arg2 = NULL, LPVOID arg3 = NULL, LPVOID arg4 = NULL);