iarv/znc
1
0
Fork 0
mirror of https://github.com/znc/znc.git synced 2026-03-28 17:42:41 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add a web interface to the certauth module

Browse Source
This commit is contained in:
Kyle Fuller
2011-05-25 13:24:14 +01:00
parent 1c0edff722
commit 456dfde05f
2 changed files with 94 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
modules/certauth.cpp
View File

@@ -84,6 +84,16 @@ public:
return SaveRegistry();
}
bool AddKey(CUser *pUser, CString sKey) {
pair<SCString::iterator, bool> pair = m_PubKeys[pUser->GetUserName()].insert(sKey);
if (pair.second) {
Save();
}
return pair.second;
}
virtual EModRet OnLoginAttempt(CSmartPtr<CAuthBase> Auth) {
CString sUser = Auth->GetUsername();
Csock *pSock = Auth->GetSocket();
@@ -139,10 +149,8 @@ public:
if (sPubKey.empty()) {
PutModule("You did not supply a public key or connect with one.");
} else {
pair<SCString::iterator, bool> res = m_PubKeys[m_pUser->GetUserName()].insert(sPubKey);
if (res.second) {
if (AddKey(m_pUser, sPubKey)) {
PutModule("'" + sPubKey + "' added.");
Save();
} else {
PutModule("The key '" + sPubKey + "' is already added.");
}
@@ -222,6 +230,46 @@ public:
}
}
virtual CString GetWebMenuTitle() { return "certauth"; }
virtual bool OnWebRequest(CWebSock& WebSock, const CString& sPageName, CTemplate& Tmpl) {
CUser *pUser = WebSock.GetSession()->GetUser();
if (sPageName == "index") {
MSCString::iterator it = m_PubKeys.find(pUser->GetUserName());
if (it != m_PubKeys.end()) {
SCString::iterator it2;
for (it2 = it->second.begin(); it2 != it->second.end(); it2++) {
CTemplate& row = Tmpl.AddRow("KeyLoop");
row["Key"] = *it2;
}
}
return true;
} else if (sPageName == "add") {
AddKey(pUser, WebSock.GetParam("key"));
WebSock.Redirect("/mods/certauth/");
return true;
} else if (sPageName == "delete") {
MSCString::iterator it = m_PubKeys.find(pUser->GetUserName());
if (it != m_PubKeys.end()) {
if (it->second.erase(WebSock.GetParam("key", false))) {
if (it->second.size() == 0) {
m_PubKeys.erase(it);
}
Save();
}
}
WebSock.Redirect("/mods/certauth/");
return true;
}
return false;
}
private:
// Maps user names to a list of allowed pubkeys
typedef map<CString, set<CString> > MSCString;