Create a identfile module, this replaces the old ident spoofing built into znc

2026-03-28 17:42:41 +01:00 · 2011-03-30 19:19:55 +01:00
parent 703f2cd5bb
commit 51ac51a252
1 changed files with 164 additions and 0 deletions
--- a/modules/identfile.cpp
+++ b/modules/identfile.cpp
@@ -0,0 +1,164 @@
+/*
+ * 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.
+ */
+
+#include "znc.h"
+#include "IRCSock.h"
+#include "User.h"
+
+class CIdentFileModule : public CGlobalModule {
+	CString m_sOrigISpoof;
+	CFile* m_pISpoofLockFile;
+	CIRCSock *m_pIRCSock;
+
+public:
+	GLOBALMODCONSTRUCTOR(CIdentFileModule) {
+		AddHelpCommand();
+		AddCommand("GetFile",   static_cast<CModCommand::ModCmdFunc>(&CIdentFileModule::GetFile));
+		AddCommand("SetFile",   static_cast<CModCommand::ModCmdFunc>(&CIdentFileModule::SetFile),
+			"<file>");
+		AddCommand("GetFormat", static_cast<CModCommand::ModCmdFunc>(&CIdentFileModule::GetFormat));
+		AddCommand("SetFormat", static_cast<CModCommand::ModCmdFunc>(&CIdentFileModule::SetFormat),
+			"<format>");
+	}
+
+	virtual ~CIdentFileModule() {
+		ReleaseISpoof();
+	}
+
+	void GetFile(const CString& sLine) {
+		PutModule("File is set to: " + GetNV("File"));
+	}
+
+	void SetFile(const CString& sLine) {
+		SetNV("File", sLine.Token(1, true));
+		PutModule("File has been set to: " + GetNV("File"));
+	}
+
+	void SetFormat(const CString& sLine) {
+		SetNV("Format", sLine.Token(1, true));
+		PutModule("Format has been set to: " + GetNV("Format"));
+		PutModule("Format would be expanded to: " + m_pUser->ExpandString(GetNV("Format")));
+	}
+
+	void GetFormat(const CString& sLine) {
+		PutModule("Format is set to: " + GetNV("Format"));
+		PutModule("Format would be expanded to: " + m_pUser->ExpandString(GetNV("Format")));
+	}
+
+	void OnModCommand(const CString& sCommand) {
+		if (m_pUser->IsAdmin()) {
+			HandleCommand(sCommand);
+		} else {
+			PutModule("Access denied");
+		}
+	}
+
+	bool WriteISpoof() {
+		if (m_pISpoofLockFile != NULL) {
+			return false;
+		}
+
+		m_pISpoofLockFile = new CFile;
+		if (!m_pISpoofLockFile->TryExLock(GetNV("File"), O_RDWR | O_CREAT)) {
+			delete m_pISpoofLockFile;
+			m_pISpoofLockFile = NULL;
+			return false;
+		}
+
+		char buf[1024];
+		memset((char*) buf, 0, 1024);
+		m_pISpoofLockFile->Read(buf ,1024);
+		m_sOrigISpoof = buf;
+
+		if (!m_pISpoofLockFile->Seek(0) || !m_pISpoofLockFile->Truncate()) {
+			delete m_pISpoofLockFile;
+			m_pISpoofLockFile = NULL;
+			return false;
+		}
+
+		CString sData = m_pUser->ExpandString(GetNV("Format"));
+
+		// If the format doesn't contain anything expandable, we'll
+		// assume this is an "old"-style format string.
+		if (sData == GetNV("Format")) {
+			sData.Replace("%", m_pUser->GetIdent());
+		}
+
+		DEBUG("Writing [" + sData + "] to ISpoofFile [" + m_pISpoofLockFile->GetLongName() + "]");
+
+		m_pISpoofLockFile->Write(sData + "\n");
+
+		return true;
+	}
+
+	void ReleaseISpoof() {
+		if (m_pISpoofLockFile != NULL) {
+			if (m_pISpoofLockFile->Seek(0) && m_pISpoofLockFile->Truncate()) {
+				m_pISpoofLockFile->Write(m_sOrigISpoof);
+			}
+
+			delete m_pISpoofLockFile;
+			m_pISpoofLockFile = NULL;
+		}
+	}
+
+	virtual bool OnLoad(const CString& sArgs, CString& sMessage) {
+		m_pISpoofLockFile = NULL;
+		m_pIRCSock = NULL;
+
+		if (GetNV("Format").empty()) {
+			SetNV("Format", "global { reply \"%ident%\" }");
+		}
+
+		if (GetNV("File").empty()) {
+			SetNV("File", "~/.oidentd.conf");
+		}
+
+		return true;
+	}
+
+	virtual EModRet OnIRCConnecting(CIRCSock *pIRCSock) {
+		if (m_pISpoofLockFile != NULL) {
+			DEBUG("Aborting connection, ISpoofLockFile exists");
+			PutModule("Aborting connection, ISpoofLockFile exists");
+			return HALTCORE;
+		}
+
+		if (!WriteISpoof()) {
+			DEBUG("ISpoof [" + GetNV("File") + "] could not be written");
+			PutModule("ISpoof [" + GetNV("File") + "] could not be written, retrying...");
+			return HALTCORE;
+		}
+
+		m_pIRCSock = pIRCSock;
+		return CONTINUE;
+	}
+
+	virtual void OnIRCConnected() {
+		if (m_pIRCSock == m_pUser->GetIRCSock()) {
+			m_pIRCSock = NULL;
+			ReleaseISpoof();
+		}
+	}
+
+	virtual void OnIRCConnectionError(CIRCSock *pIRCSock) {
+		if (m_pIRCSock == pIRCSock) {
+			m_pIRCSock = NULL;
+			ReleaseISpoof();
+		}
+	}
+
+	virtual void OnIRCDisconnected() {
+		if (m_pIRCSock == m_pUser->GetIRCSock()) {
+			m_pIRCSock = NULL;
+			ReleaseISpoof();
+		}
+	}
+};
+
+GLOBALMODULEDEFS(CIdentFileModule, "Write the ident of a user to a file when they are trying to connect")