1 files changed, 294 insertions, 0 deletions
diff --git a/Swiften/TLS/Schannel/SchannelUtil.h b/Swiften/TLS/Schannel/SchannelUtil.h
new file mode 100644
index 0000000..0a54f16
--- /dev/null
+++ b/Swiften/TLS/Schannel/SchannelUtil.h
@@ -0,0 +1,294 @@
+/*
+ * Copyright (c) 2011 Soren Dreijer
+ * Licensed under the simplified BSD license.
+ * See Documentation/Licenses/BSD-simplified.txt for more information.
+ */
+
+#pragma once
+
+#define SECURITY_WIN32
+#include <Windows.h>
+#include <Schannel.h>
+#include <security.h>
+#include <schnlsp.h>
+
+#include <boost/noncopyable.hpp>
+
+namespace Swift 
+{
+	//
+	// Convenience wrapper around the Schannel CredHandle struct.
+	//
+	class ScopedCredHandle
+	{
+	private:
+		struct HandleContext
+		{
+			HandleContext()
+			{
+				ZeroMemory(&m_h, sizeof(m_h));
+			}
+
+			HandleContext(const CredHandle& h)
+			{
+				memcpy(&m_h, &h, sizeof(m_h));
+			}
+
+			~HandleContext()
+			{
+				::FreeCredentialsHandle(&m_h);
+			}
+
+			CredHandle m_h;
+		};
+
+	public:
+		ScopedCredHandle()
+		: m_pHandle( new HandleContext )
+		{
+		}
+
+		explicit ScopedCredHandle(const CredHandle& h)
+		: m_pHandle( new HandleContext(h) )
+		{
+		}
+
+		// Copy constructor
+		explicit ScopedCredHandle(const ScopedCredHandle& rhs)
+		{
+			m_pHandle = rhs.m_pHandle;			
+		}
+
+		~ScopedCredHandle()
+		{
+			m_pHandle.reset();
+		}
+
+		PCredHandle Reset()
+		{
+			CloseHandle();
+			return &m_pHandle->m_h;
+		}
+
+		operator PCredHandle() const
+		{
+			return &m_pHandle->m_h;
+		}
+
+		ScopedCredHandle& operator=(const ScopedCredHandle& sh)
+		{
+			// Only update the internal handle if it's different
+			if (&m_pHandle->m_h != &sh.m_pHandle->m_h)
+			{
+				m_pHandle = sh.m_pHandle;
+			}
+
+			return *this;
+		}
+
+		void CloseHandle()
+		{
+			m_pHandle.reset( new HandleContext );
+		}
+
+	private:
+		boost::shared_ptr<HandleContext> m_pHandle;
+	};
+
+	//------------------------------------------------------------------------
+
+	//
+	// Convenience wrapper around the Schannel CtxtHandle struct.
+	//
+	class ScopedCtxtHandle
+	{
+	private:
+		struct HandleContext
+		{
+			HandleContext()
+			{
+				ZeroMemory(&m_h, sizeof(m_h));
+			}
+
+			~HandleContext()
+			{
+				::DeleteSecurityContext(&m_h);
+			}
+
+			CtxtHandle m_h;
+		};
+
+	public:
+		ScopedCtxtHandle()
+		: m_pHandle( new HandleContext )
+		{
+		}
+
+		explicit ScopedCtxtHandle(CredHandle h)
+		: m_pHandle( new HandleContext )
+		{
+		}
+
+		// Copy constructor
+		explicit ScopedCtxtHandle(const ScopedCtxtHandle& rhs)
+		{
+			m_pHandle = rhs.m_pHandle;			
+		}
+
+		~ScopedCtxtHandle()
+		{
+			m_pHandle.reset();
+		}
+
+		PCredHandle Reset()
+		{
+			CloseHandle();
+			return &m_pHandle->m_h;
+		}
+
+		operator PCredHandle() const
+		{
+			return &m_pHandle->m_h;
+		}
+
+		ScopedCtxtHandle& operator=(const ScopedCtxtHandle& sh)
+		{
+			// Only update the internal handle if it's different
+			if (&m_pHandle->m_h != &sh.m_pHandle->m_h)
+			{
+				m_pHandle = sh.m_pHandle;
+			}
+
+			return *this;
+		}
+
+		void CloseHandle()
+		{
+			m_pHandle.reset( new HandleContext );
+		}
+
+	private:
+		boost::shared_ptr<HandleContext> m_pHandle;
+	};
+
+	//------------------------------------------------------------------------
+
+	//
+	// Convenience wrapper around the Schannel ScopedSecBuffer struct.
+	//
+	class ScopedSecBuffer : boost::noncopyable
+	{
+	public:
+		ScopedSecBuffer(PSecBuffer pSecBuffer)
+		: m_pSecBuffer(pSecBuffer)
+		{
+		}
+		
+		~ScopedSecBuffer()
+		{
+			// Loop through all the output buffers and make sure we free them
+			if (m_pSecBuffer->pvBuffer)
+				FreeContextBuffer(m_pSecBuffer->pvBuffer);
+		}
+
+		PSecBuffer AsPtr()
+		{
+			return m_pSecBuffer;
+		}
+
+		PSecBuffer operator->()
+		{
+			return m_pSecBuffer;
+		}
+
+	private:
+		PSecBuffer m_pSecBuffer;
+	};
+
+	//------------------------------------------------------------------------
+
+	//
+	// Convenience wrapper around the Schannel PCCERT_CONTEXT.
+	//
+	class ScopedCertContext
+	{
+	private:
+		struct HandleContext
+		{
+			HandleContext()
+			: m_pCertCtxt(NULL)
+			{
+			}
+
+			HandleContext(PCCERT_CONTEXT pCert)
+			: m_pCertCtxt(pCert)
+			{
+			}
+
+			~HandleContext()
+			{
+				if (m_pCertCtxt)
+					CertFreeCertificateContext(m_pCertCtxt);
+			}
+
+			PCCERT_CONTEXT m_pCertCtxt;
+		};
+
+	public:
+		ScopedCertContext()
+		: m_pHandle( new HandleContext )
+		{
+		}
+
+		explicit ScopedCertContext(PCCERT_CONTEXT pCert)
+		: m_pHandle( new HandleContext(pCert) )
+		{
+		}
+
+		// Copy constructor
+		explicit ScopedCertContext(const ScopedCertContext& rhs)
+		{
+			m_pHandle = rhs.m_pHandle;			
+		}
+
+		~ScopedCertContext()
+		{
+			m_pHandle.reset();
+		}
+
+		PCCERT_CONTEXT* Reset()
+		{
+			FreeContext();
+			return &m_pHandle->m_pCertCtxt;
+		}
+
+		operator PCCERT_CONTEXT() const
+		{
+			return m_pHandle->m_pCertCtxt;
+		}
+
+		PCCERT_CONTEXT operator->() const
+		{
+			return m_pHandle->m_pCertCtxt;
+		}
+
+		ScopedCertContext& operator=(const ScopedCertContext& sh)
+		{
+			// Only update the internal handle if it's different
+			if (&m_pHandle->m_pCertCtxt != &sh.m_pHandle->m_pCertCtxt)
+			{
+				m_pHandle = sh.m_pHandle;
+			}
+
+			return *this;
+		}
+
+		void FreeContext()
+		{
+			m_pHandle.reset( new HandleContext );
+		}
+
+	private:
+		boost::shared_ptr<HandleContext> m_pHandle;		
+	};	 
+}