Add the cert module which allows you to use a SSL certificate on outgoing IRC connections

2026-07-04 08:51:14 +02:00 · 2011-03-29 17:55:50 +01:00
parent 322c63cd05
commit 8ad2d907b5
2 changed files with 114 additions and 0 deletions
@@ -0,0 +1,87 @@
+/*
+ * Copyright (C) 2004-2011  See the AUTHORS file for details.
+ *
+ * This program is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License version 2 as published
+ * by the Free Software Foundation.
+ */
+
+#define REQUIRESSL
+
+#include "User.h"
+#include "Modules.h"
+#include "IRCSock.h"
+
+class CCertMod : public CModule {
+public:
+	void Delete(const CString &line) {
+		if (CFile::Delete(PemFile())) {
+			PutModule("Pem file deleted");
+		} else {
+			PutModule("The pem file doesn't exist or there was a error deleting the pem file.");
+		}
+	}
+
+	void Info(const CString &line) {
+		if (HasPemFile()) {
+			PutModule("You have a certificate in: " + PemFile());
+		} else {
+			PutModule("You do not have a certificate. Please use the web interface to add a certificate");
+			if (m_pUser->IsAdmin()) {
+				PutModule("Alternatively you can either place one at " + PemFile() + " or use the GENERATE command to generate a new certificate");
+			}
+		}
+	}
+
+	MODCONSTRUCTOR(CCertMod) {
+		AddHelpCommand();
+		AddCommand("delete", static_cast<CModCommand::ModCmdFunc>(&CCertMod::Delete),
+			"", "Delete the current certificate");
+		AddCommand("info", static_cast<CModCommand::ModCmdFunc>(&CCertMod::Info));
+	}
+
+	virtual ~CCertMod() {}
+
+	CString PemFile() const {
+		return GetSavePath() + "/user.pem";
+	}
+
+	bool HasPemFile() const {
+		return (CFile::Exists(PemFile()));
+	}
+
+	virtual EModRet OnIRCConnecting(CIRCSock *pIRCSock) {
+		if (HasPemFile()) {
+			pIRCSock->SetPemLocation(PemFile());
+		}
+
+		return CONTINUE;
+	}
+
+	virtual CString GetWebMenuTitle() { return "Certificate"; }
+
+	virtual bool OnWebRequest(CWebSock& WebSock, const CString& sPageName, CTemplate& Tmpl) {
+		if (sPageName == "index") {
+			Tmpl["Cert"] = CString(HasPemFile());
+			return true;
+		} else if (sPageName == "update") {
+			CFile fPemFile(PemFile());
+
+			if (fPemFile.Open(O_WRONLY | O_TRUNC | O_CREAT)) {
+				fPemFile.Write(WebSock.GetParam("cert"));
+				fPemFile.Close();
+			}
+
+			WebSock.Redirect("/mods/cert/");
+			return true;
+		} else if (sPageName == "delete") {
+			CFile::Delete(PemFile());
+			WebSock.Redirect("/mods/cert/");
+			return true;
+		}
+
+		return false;
+	}
+};
+
+MODULEDEFS(CCertMod, "Use a ssl certificate to connect to a server")