From 50b998a5664d321e9417f78d54ff3ee8ce1535e8 Mon Sep 17 00:00:00 2001
From: prozacx <prozacx@726aef4b-f618-498e-8847-2d620e286838>
Date: Thu, 16 Mar 2006 01:20:12 +0000
Subject: [PATCH] Added support for auth modules

git-svn-id: https://znc.svn.sourceforge.net/svnroot/znc/trunk@692 726aef4b-f618-498e-8847-2d620e286838
---
 modules/webadmin.cpp | 63 ++++++++++++++++++++++++++++++++++++++++----
 1 file changed, 58 insertions(+), 5 deletions(-)

diff --git a/modules/webadmin.cpp b/modules/webadmin.cpp
index 0d8fce36..f9805cc6 100644
--- a/modules/webadmin.cpp
+++ b/modules/webadmin.cpp
@@ -5,10 +5,31 @@
 #include "Chan.h"
 #include "znc.h"
 #include "HTTPSock.h"
+#include "Client.h"
 #include "Server.h"
 #include "Template.h"
 
 class CWebAdminMod;
+class CWebAdminSock;
+
+
+class CWebAdminAuth : public CAuthBase {
+public:
+	CWebAdminAuth(CWebAdminSock* pWebAdminSock, const CString& sUsername, const CString& sPassword)
+		: CAuthBase(sUsername, sPassword) {
+			m_pWebAdminSock = pWebAdminSock;
+	}
+
+	virtual ~CWebAdminAuth() {}
+
+	void SetWebAdminSock(CWebAdminSock* pWebAdminSock) { m_pWebAdminSock = pWebAdminSock; }
+	void AcceptLogin(CUser& User);
+	void RefuseLogin(const CString& sReason);
+private:
+protected:
+	CWebAdminSock*	m_pWebAdminSock;
+};
+
 
 class CWebAdminSock : public CHTTPSock {
 public:
@@ -55,16 +76,22 @@ public:
 		return "";
 	}
 
+	// Setters
+	void SetSessionUser(CUser* p) { m_pSessionUser = p; m_bAdmin = p->IsAdmin(); }
+	void SetAdmin(bool b) { m_bAdmin = b; }
+	// !Setters
+
 	virtual Csock* GetSockObj(const CString& sHost, unsigned short uPort);
 	bool IsAdmin(bool bAllowUserAdmin = true) const { return m_bAdmin; }
 
 private:
 protected:
-	CWebAdminMod*	m_pModule;
-	CUser*			m_pUser;
-	CUser*			m_pSessionUser;
-	bool			m_bAdmin;
-	CTemplate		m_Template;
+	CWebAdminMod*			m_pModule;
+	CUser*					m_pUser;
+	CUser*					m_pSessionUser;
+	bool					m_bAdmin;
+	CTemplate				m_Template;
+	CSmartPtr<CAuthBase>	m_spAuth;
 };
 
 class CWebAdminMod : public CGlobalModule {
@@ -211,6 +238,13 @@ void CWebAdminSock::PrintPage(CString& sPageRet, const CString& sTmplName) {
 }
 
 bool CWebAdminSock::OnLogin(const CString& sUser, const CString& sPass) {
+	m_spAuth = new CWebAdminAuth(this, sUser, sPass);
+
+	if (CZNC::Get().GetModules().OnLoginAttempt(m_spAuth)) {
+		PauseRead();
+		return false;
+	}
+
 	CUser* pUser = CZNC::Get().FindUser(GetUser());
 
 	if (pUser) {
@@ -289,6 +323,11 @@ CWebAdminSock::CWebAdminSock(CWebAdminMod* pModule, const CString& sHostname, un
 
 CWebAdminSock::~CWebAdminSock() {
 	m_pModule->SockDestroyed(this);
+
+	if (!m_spAuth.IsNull()) {
+		CWebAdminAuth* pAuth = (CWebAdminAuth*) &(*m_spAuth);
+		pAuth->SetWebAdminSock(NULL);
+	}
 }
 
 bool CWebAdminSock::OnPageRequest(const CString& sURI, CString& sPageRet) {
@@ -1050,4 +1089,18 @@ CUser* CWebAdminSock::GetNewUser(CString& sPageRet, CUser* pUser) {
 	return pNewUser;
 }
 
+void CWebAdminAuth::AcceptLogin(CUser& User) {
+	if (m_pWebAdminSock) {
+		m_pWebAdminSock->SetSessionUser(&User);
+		m_pWebAdminSock->SetLoggedIn(true);
+		m_pWebAdminSock->UnPauseRead();
+	}
+}
+
+void CWebAdminAuth::RefuseLogin(const CString& sReason) {
+	if (m_pWebAdminSock) {
+		m_pWebAdminSock->UnPauseRead();
+	}
+}
+
 GLOBALMODULEDEFS(CWebAdminMod, "Dynamic configuration of users/settings through a web browser")