1 files changed, 134 insertions, 0 deletions
diff --git a/NSCmpts/NightSecError.h b/NSCmpts/NightSecError.h
new file mode 100644
index 0000000..164c5fb
--- /dev/null
+++ b/NSCmpts/NightSecError.h
@@ -0,0 +1,134 @@
+// NightSecError.h: Definition of the CNightSecError class
+//
+//////////////////////////////////////////////////////////////////////
+
+#if !defined(AFX_NIGHTSECERROR_H__F6662FF3_94AB_11D3_BFC4_0020182B97FC__INCLUDED_)
+#define AFX_NIGHTSECERROR_H__F6662FF3_94AB_11D3_BFC4_0020182B97FC__INCLUDED_
+
+#if _MSC_VER > 1000
+#pragma once
+#endif // _MSC_VER > 1000
+
+#include "resource.h"       // main symbols
+#include "../common/Defines.h"
+#include "../common/interfaces.h"
+
+#include "ActionEngine.h"
+
+/////////////////////////////////////////////////////////////////////////////
+// CXXXXErrorImpl: A COM object that gets added to INightSecErrorLog on
+// the host 
+
+class CBaseErrorImpl :
+	public INightSecError
+{
+public:
+	CBaseErrorImpl()  
+		: m_lHelpContext(0) {};
+
+	// Set the error codes for this object
+	void SetError(const string& sDesc, HRESULT hRes, 
+				  LPCTSTR szHelpFile = _T(""), long lHelpContext = 0);
+	void SetError(const string& sDesc, HRESULT hRes, long lHelpContext)
+		{ SetError(sDesc, hRes, NS_HELP_FILE, lHelpContext); }
+
+// INightSecError
+public:
+	STDMETHOD(get_Err)(/*[out, retval]*/ HRESULT* phRet)
+		{ *phRet = m_hRes; return S_OK; }
+	STDMETHOD(get_Description)(/*[out, retval]*/ BSTR* pbsRet)
+		{ *pbsRet = m_bsDescription.Copy(); return S_OK; }
+	STDMETHOD(get_HelpFile)(/*[out, retval]*/ BSTR* pbsRet)
+		{ *pbsRet = m_bsHelpFile.Copy(); return S_OK; }
+	STDMETHOD(get_HelpContext)(/*[out, retval]*/ long* plRet)
+		{ *plRet = m_lHelpContext; return S_OK; }
+
+protected:
+	CComBSTR m_bsDescription;	// Description of Error
+	CComBSTR m_bsHelpFile;		// Helpfile for Error
+	long m_lHelpContext;		// Location in Help File
+	HRESULT m_hRes;				// Original HRESULT
+};
+
+////////////////////////////////////////////////////////////////
+// This is a special case CBaseErrorImpl that wraps a
+// CAction
+
+class CActionErrorImpl : 
+	public CComObjectRoot,
+	public CBaseErrorImpl,
+	public INightSecErrorFix
+{
+
+// Construction
+public:
+	CActionErrorImpl() 
+		{ m_pAction = NULL; };
+	~CActionErrorImpl()
+		{ m_pAction->release();	}
+
+	// Attach to this action (addrefs)
+	void Attach(CAction* pAction);
+
+// COM MAP
+BEGIN_COM_MAP(CActionErrorImpl)
+	COM_INTERFACE_ENTRY(INightSecError)
+	COM_INTERFACE_ENTRY(INightSecErrorFix)
+END_COM_MAP()
+
+DECLARE_NOT_AGGREGATABLE(CActionErrorImpl) 
+
+// INightSecErrorFix
+public:
+	STDMETHOD(Fix)();
+	STDMETHOD(get_Fixable)(/*[out, retval]*/ BOOL* pbRet);
+	STDMETHOD(Retry)();
+	STDMETHOD(get_Retryable)(/*[out, retval]*/ BOOL* pbRet);
+
+protected:
+	CAction* m_pAction;			// Action that caused the error
+};
+
+///////////////////////////////////////////////////////////
+// Special Case CBaseErrorIpml that wraps a 
+// ISecureShutdownWin Object
+
+class CNightSecErrorImpl : 
+	public CComObjectRoot,
+	public CBaseErrorImpl,
+	public INightSecErrorFix
+{
+
+// Construction
+public:
+	CNightSecErrorImpl() 
+		{ m_pShutdown = NULL; };
+	~CNightSecErrorImpl()
+		{ if(m_pShutdown) m_pShutdown->Release(); }
+
+	// Attach to object (addrefs)
+	void Attach(ISecureShutdownWin* pShutdown);
+
+// COM MAP
+BEGIN_COM_MAP(CActionErrorImpl)
+	COM_INTERFACE_ENTRY(INightSecError)
+	COM_INTERFACE_ENTRY(INightSecErrorFix)
+END_COM_MAP()
+
+DECLARE_NOT_AGGREGATABLE(CActionErrorImpl) 
+
+// INightSecErrorFix
+public:
+	STDMETHOD(Fix)()
+		{ return E_FAIL; };
+	STDMETHOD(get_Fixable)(/*[out, retval]*/ BOOL* pbRet)
+		{ *pbRet = FALSE; return S_OK; };
+	STDMETHOD(Retry)();
+	STDMETHOD(get_Retryable)(/*[out, retval]*/ BOOL* pbRet)
+		{ *pbRet = TRUE; return S_OK; };
+
+protected:
+	ISecureShutdownWin* m_pShutdown;	// Object that caused the error
+};
+
+#endif // !defined(AFX_NIGHTSECERROR_H__F6662FF3_94AB_11D3_BFC4_0020182B97FC__INCLUDED_)