From dce289a5d459092e54bf1dac1cf15d64dd3407c0 Mon Sep 17 00:00:00 2001 From: J-P Nurmi Date: Tue, 4 Aug 2015 16:08:19 +0200 Subject: [PATCH] CUser: add API for removing and clearing allowed hosts --- include/znc/User.h | 2 ++ src/User.cpp | 6 ++++++ 2 files changed, 8 insertions(+) diff --git a/include/znc/User.h b/include/znc/User.h index de7b7a5b..84018e45 100644 --- a/include/znc/User.h +++ b/include/znc/User.h @@ -63,6 +63,8 @@ public: CConfig ToConfig() const; bool CheckPass(const CString& sPass) const; bool AddAllowedHost(const CString& sHostMask); + bool RemAllowedHost(const CString& sHostMask); + void ClearAllowedHosts(); bool IsHostAllowed(const CString& sHostMask) const; bool IsValid(CString& sErrMsg, bool bSkipPass = false) const; static bool IsValidUserName(const CString& sUserName); diff --git a/src/User.cpp b/src/User.cpp index b1e2b9d9..f3b5b91a 100644 --- a/src/User.cpp +++ b/src/User.cpp @@ -791,6 +791,12 @@ bool CUser::AddAllowedHost(const CString& sHostMask) { m_ssAllowedHosts.insert(sHostMask); return true; } +bool CUser::RemAllowedHost(const CString& sHostMask) { + return m_ssAllowedHosts.erase(sHostMask) > 0; +} +void CUser::ClearAllowedHosts() { + m_ssAllowedHosts.clear(); +} bool CUser::IsHostAllowed(const CString& sHostMask) const { if (m_ssAllowedHosts.empty()) {