Import most of znc-extra

This imports all modules from znc-extra except modtcl under modules/extra. git-svn-id: https://znc.svn.sourceforge.net/svnroot/znc/trunk@1606 726aef4b-f618-498e-8847-2d620e286838
2026-08-07 09:22:55 +02:00 · 2009-08-20 19:54:27 +00:00
parent a593f584c0
commit fc4c3b974b
15 changed files with 1597 additions and 0 deletions
@@ -0,0 +1,208 @@
+/*
+ * Copyright (C) 2004-2009  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 "Modules.h"
+#include "User.h"
+#include "znc.h"
+#include <queue>
+
+class CSocketSorter {
+public:
+	CSocketSorter(Csock* p) {
+		m_pSock = p;
+	}
+	bool operator<(const CSocketSorter& other) const {
+		// The 'biggest' item is displayed first.
+		// return false: this is first
+		// return true: other is first
+
+		// Listeners go to the top
+		if (m_pSock->GetType() != other.m_pSock->GetType()) {
+			if (m_pSock->GetType() == Csock::LISTENER)
+				return false;
+			if (other.m_pSock->GetType() == Csock::LISTENER)
+				return true;
+		}
+		const CString& sMyName = m_pSock->GetSockName();
+		const CString& sMyName2 = sMyName.Token(1, true, "::");
+		bool bMyEmpty = sMyName2.empty();
+		const CString& sHisName = other.GetSock()->GetSockName();
+		const CString& sHisName2 = sHisName.Token(1, true, "::");
+		bool bHisEmpty = sHisName2.empty();
+
+		// Then sort by first token after "::"
+		if (bMyEmpty && !bHisEmpty)
+			return false;
+		if (bHisEmpty && !bMyEmpty)
+			return true;
+
+		if (!bMyEmpty && !bHisEmpty) {
+			int c = sMyName2.StrCmp(sHisName2);
+			if (c < 0)
+				return false;
+			if (c > 0)
+				return true;
+		}
+		// and finally sort by the whole socket name
+		return sMyName.StrCmp(sHisName) > 0;
+	}
+	Csock* GetSock() const { return m_pSock; }
+private:
+	Csock* m_pSock;
+};
+
+class CListSockets : public CModule {
+public:
+	MODCONSTRUCTOR(CListSockets) {}
+
+	virtual bool OnLoad(const CString& sArgs, CString& sMessage)
+	{
+#ifndef MOD_LISTSOCKETS_ALLOW_EVERYONE
+		if (!m_pUser->IsAdmin()) {
+			sMessage = "You must be admin to use this module";
+			return false;
+		}
+#endif
+
+		return true;
+	}
+
+	virtual void OnModCommand(const CString& sLine) {
+		CString sCommand = sLine.Token(0);
+		CString sArg = sLine.Token(1, true);
+
+		if (sCommand.Equals("LIST")) {
+			bool bShowHosts = true;
+			if (sArg.Equals("-n")) {
+				bShowHosts = false;
+			}
+			ShowSocks(bShowHosts);
+		} else {
+			PutModule("Use 'list' to view a list of active sockets");
+			PutModule("Use 'list -n' if you want IP addresses to be displayed");
+		}
+	}
+
+	void ShowSocks(bool bShowHosts) {
+		CSockManager& m = CZNC::Get().GetManager();
+		if (!m.size()) {
+			PutStatus("You have no open sockets.");
+			return;
+		}
+
+		std::priority_queue<CSocketSorter> socks;
+
+		for (unsigned int a = 0; a < m.size(); a++) {
+			socks.push(m[a]);
+		}
+
+		CTable Table;
+		Table.AddColumn("Name");
+		Table.AddColumn("Created");
+		Table.AddColumn("State");
+#ifdef HAVE_LIBSSL
+		Table.AddColumn("SSL");
+#endif
+		Table.AddColumn("Local");
+		Table.AddColumn("Remote");
+
+		while (!socks.empty()) {
+			Csock* pSocket = socks.top().GetSock();
+			socks.pop();
+
+			Table.AddRow();
+
+			switch (pSocket->GetType()) {
+			case Csock::LISTENER:
+				Table.SetCell("State", "Listen");
+				break;
+			case Csock::INBOUND:
+				Table.SetCell("State", "Inbound");
+				break;
+			case Csock::OUTBOUND:
+				if (pSocket->IsConnected())
+					Table.SetCell("State", "Outbound");
+				else
+					Table.SetCell("State", "Connecting");
+				break;
+			default:
+				Table.SetCell("State", "UNKNOWN");
+				break;
+			}
+
+			unsigned long long iStartTime = pSocket->GetStartTime();
+			time_t iTime = iStartTime / 1000;
+			Table.SetCell("Created", FormatTime("%Y-%m-%d %H:%M:%S", iTime));
+
+#ifdef HAVE_LIBSSL
+			if (pSocket->GetSSL()) {
+				Table.SetCell("SSL", "Yes");
+			} else {
+				Table.SetCell("SSL", "No");
+			}
+#endif
+
+
+			Table.SetCell("Name", pSocket->GetSockName());
+			CString sVHost;
+			if (bShowHosts) {
+				sVHost = pSocket->GetBindHost();
+			}
+			if (sVHost.empty()) {
+				sVHost = pSocket->GetLocalIP();
+			}
+			Table.SetCell("Local", sVHost + " " + CString(pSocket->GetLocalPort()));
+
+			CString sHost;
+			if (!bShowHosts) {
+				sHost = pSocket->GetRemoteIP();
+			}
+			// While connecting, there might be no ip available
+			if (sHost.empty()) {
+				sHost = pSocket->GetHostName();
+			}
+
+			u_short uPort;
+			// While connecting, GetRemotePort() would return 0
+			if (pSocket->GetType() == Csock::OUTBOUND) {
+				uPort = pSocket->GetPort();
+			} else {
+				uPort = pSocket->GetRemotePort();
+			}
+			if (uPort != 0) {
+				Table.SetCell("Remote", sHost + " " + CString(uPort));
+			} else {
+				Table.SetCell("Remote", sHost);
+			}
+		}
+
+		PutModule(Table);
+		return;
+	}
+
+	virtual ~CListSockets() {
+	}
+
+	CString FormatTime(const CString& sFormat, time_t tm = 0) const {
+		char szTimestamp[1024];
+
+		if (tm == 0) {
+			tm = time(NULL);
+		}
+
+		// offset is in hours
+		tm += (time_t)(m_pUser->GetTimezoneOffset() * 60 * 60);
+		strftime(szTimestamp, sizeof(szTimestamp) / sizeof(char),
+				sFormat.c_str(), localtime(&tm));
+
+		return szTimestamp;
+	}
+};
+
+MODULEDEFS(CListSockets, "List active sockets")
+