third-party libs

third-party/BugTrap-1.3-57bd515/source/LogLink.h

@@ -0,0 +1,193 @@
+/*
+ * This is a part of the BugTrap package.
+ * Copyright (c) 2005-2009 IntelleSoft.
+ * All rights reserved.
+ *
+ * Description: Log file descriptor.
+ * Author: Maksim Pyatkovskiy.
+ *
+ * This source code is only intended as a supplement to the
+ * BugTrap package reference and related electronic documentation
+ * provided with the product. See these sources for detailed
+ * information regarding the BugTrap package.
+ */
+
+#pragma once
+
+#include "BugTrap.h"
+#include "BugTrapUtils.h"
+
+/// Log file descriptor.
+class CLogLink
+{
+public:
+	/// Initialize the object.
+	CLogLink(void);
+	/// Initialize the object.
+	CLogLink(PCTSTR pszLogFileName);
+	/// Destroy the object.
+	virtual ~CLogLink(void);
+	/// Get log type.
+	virtual BUGTRAP_LOGTYPE GetLogType(void) const;
+	/// Save log link entries.
+	virtual void SaveEntries(bool bCrash);
+	/// Get custom log file name.
+	PCTSTR GetLogFileName(void) const;
+	/// Set custom log file name.
+	void SetLogFileName(PCTSTR pszLogFileName);
+	/// Object comparison.
+	friend bool operator==(const CLogLink& rLogLink1, const CLogLink& rLogLink2);
+	/// Object comparison.
+	friend bool operator!=(const CLogLink& rLogLink1, const CLogLink& rLogLink2);
+
+protected:
+	/// Custom log file name.
+	TCHAR m_szLogFileName[MAX_PATH];
+};
+
+inline CLogLink::CLogLink(void)
+{
+	*m_szLogFileName = _T('\0');
+}
+
+/**
+ * @param pszLogFileName - pointer to custom log file name.
+ */
+inline CLogLink::CLogLink(PCTSTR pszLogFileName)
+{
+	GetCompleteLogFileName(m_szLogFileName, pszLogFileName, NULL);
+}
+
+inline CLogLink::~CLogLink(void)
+{
+}
+
+/**
+ * @return pointer to custom log file name.
+ */
+inline PCTSTR CLogLink::GetLogFileName(void) const
+{
+	return m_szLogFileName;
+}
+
+/**
+ * @param pszLogFileName - pointer to custom log file name.
+ */
+inline void CLogLink::SetLogFileName(PCTSTR pszLogFileName)
+{
+	GetCompleteLogFileName(m_szLogFileName, pszLogFileName, NULL);
+}
+
+/**
+ * @return type of the log.
+ */
+inline BUGTRAP_LOGTYPE CLogLink::GetLogType(void) const
+{
+	return BTLT_LOGFILE;
+}
+
+/**
+ * @param bCrash - true if crash has occurred.
+ */
+inline void CLogLink::SaveEntries(bool /*bCrash*/)
+{
+}
+
+/// Reg file descriptor.
+class CRegLink : public CLogLink
+{
+public:
+	/// Object constructor.
+	CRegLink(void);
+	/// Object constructor.
+	CRegLink(PCTSTR pszLogFileName);
+	/// Object constructor.
+	CRegLink(PCTSTR pszLogFileName, PCTSTR pszRegKey);
+	/// Get log type.
+	virtual BUGTRAP_LOGTYPE GetLogType(void) const;
+	/// Save log link entries.
+	virtual void SaveEntries(bool bCrash);
+	/// Get registry key path.
+	PCTSTR GetRegKey(void) const;
+	/// Set registry key path.
+	void SetRegKey(PCTSTR pszRegKey);
+
+protected:
+	/// Registry key path.
+	TCHAR m_szRegKey[MAX_PATH];
+};
+
+inline CRegLink::CRegLink(void)
+{
+	*m_szRegKey = _T('\0');
+}
+
+/**
+ * @param pszLogFileName - pointer to custom log file name.
+ */
+inline CRegLink::CRegLink(PCTSTR pszLogFileName) : CLogLink(pszLogFileName)
+{
+	*m_szRegKey = _T('\0');
+}
+
+/**
+ * @param pszLogFileName - pointer to custom log file name.
+ * @param pszRegKey - registry key path
+ */
+inline CRegLink::CRegLink(PCTSTR pszLogFileName, PCTSTR pszRegKey) : CLogLink(pszLogFileName)
+{
+	_tcscpy_s(m_szRegKey, countof(m_szRegKey), pszRegKey);
+}
+
+/**
+ * @return type of the log.
+ */
+inline BUGTRAP_LOGTYPE CRegLink::GetLogType(void) const
+{
+	return BTLT_REGEXPORT;
+}
+
+/**
+ * @param bCrash - true if crash has occurred.
+ */
+inline void CRegLink::SaveEntries(bool bCrash)
+{
+	if (bCrash)
+		BT_ExportRegistryKey(m_szLogFileName, m_szRegKey);
+}
+
+/**
+ * @param pszRegKey - registry key path
+ */
+inline void CRegLink::SetRegKey(PCTSTR pszRegKey)
+{
+	_tcscpy_s(m_szRegKey, countof(m_szRegKey), pszRegKey);
+}
+
+/**
+ * @return registry key path.
+ */
+inline PCTSTR CRegLink::GetRegKey(void) const
+{
+	return m_szRegKey;
+}
+
+/**
+ * @param rLogLink1 - 1st object.
+ * @param rLogLink2 - 2nd object.
+ * @return comparison result.
+ */
+inline bool operator==(const CLogLink& rLogLink1, const CLogLink& rLogLink2)
+{
+	return (_tcsicmp(rLogLink1.m_szLogFileName, rLogLink2.m_szLogFileName) == 0);
+}
+
+/**
+ * @param rLogLink1 - 1st object.
+ * @param rLogLink2 - 2nd object.
+ * @return comparison result.
+ */
+inline bool operator!=(const CLogLink& rLogLink1, const CLogLink& rLogLink2)
+{
+	return (_tcsicmp(rLogLink1.m_szLogFileName, rLogLink2.m_szLogFileName) != 0);
+}