ZNC-Extra no more.

Few the most useless modules are just removed, the others are moved to standard modules.
2026-05-07 22:04:46 +02:00 · 2012-02-21 19:34:36 +07:00
parent fc8e789b08
commit 7ac61474ed
25 changed files with 27 additions and 444 deletions
@@ -0,0 +1,131 @@
+/*
+ * Copyright (C) 2004-2012  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/FileUtils.h>
+#include <znc/User.h>
+#include <znc/znc.h>
+#include <znc/ExecSock.h>
+
+// Forward Declaration
+class CShellMod;
+
+class CShellSock : public CExecSock {
+public:
+	CShellSock(CShellMod* pShellMod, CClient* pClient, const CString& sExec) : CExecSock() {
+		EnableReadLine();
+		m_pParent = pShellMod;
+		m_pClient = pClient;
+
+		if (Execute(sExec) == -1) {
+			CString s = "Failed to execute: ";
+			s += strerror(errno);
+			ReadLine(s);
+			return;
+		}
+
+		// Get rid of that write fd, we aren't going to use it
+		// (And clients expecting input will fail this way).
+		close(GetWSock());
+		SetWSock(open("/dev/null", O_WRONLY));
+	}
+	// These next two function's bodies are at the bottom of the file since they reference CShellMod
+	virtual void ReadLine(const CString& sData);
+	virtual void Disconnected();
+
+	CShellMod* m_pParent;
+
+private:
+	CClient*   m_pClient;
+};
+
+class CShellMod : public CModule {
+public:
+	MODCONSTRUCTOR(CShellMod) {
+		m_sPath = CZNC::Get().GetHomePath();
+	}
+
+	virtual ~CShellMod() {
+		vector<Csock*> vSocks = m_pManager->FindSocksByName("SHELL");
+
+		for (unsigned int a = 0; a < vSocks.size(); a++) {
+			m_pManager->DelSockByAddr(vSocks[a]);
+		}
+	}
+
+	virtual bool OnLoad(const CString& sArgs, CString& sMessage)
+	{
+#ifndef MOD_SHELL_ALLOW_EVERYONE
+		if (!m_pUser->IsAdmin()) {
+			sMessage = "You must be admin to use the shell module";
+			return false;
+		}
+#endif
+
+		return true;
+	}
+
+	virtual void OnModCommand(const CString& sLine) {
+		CString sCommand = sLine.Token(0);
+		if (sCommand.Equals("cd")) {
+			CString sArg = sLine.Token(1, true);
+			CString sPath = CDir::ChangeDir(m_sPath, (sArg.empty() ? CString(CZNC::Get().GetHomePath()) : sArg), CZNC::Get().GetHomePath());
+			CFile Dir(sPath);
+
+			if (Dir.IsDir()) {
+				m_sPath = sPath;
+			} else if (Dir.Exists()) {
+				PutShell("cd: not a directory [" + sPath + "]");
+			} else {
+				PutShell("cd: no such directory [" + sPath + "]");
+			}
+
+			PutShell("znc$");
+		} else {
+			RunCommand(sLine);
+		}
+	}
+
+	void PutShell(const CString& sMsg) {
+		CString sPath = m_sPath.Replace_n(" ", "_");
+		CString sSource = ":" + GetModNick() + "!shell@" + sPath;
+		CString sLine = sSource + " PRIVMSG " + m_pClient->GetNick() + " :" + sMsg;
+		m_pClient->PutClient(sLine);
+	}
+
+	void RunCommand(const CString& sCommand) {
+		m_pManager->AddSock(new CShellSock(this, m_pClient, "cd " + m_sPath + " && " + sCommand), "SHELL");
+	}
+private:
+	CString m_sPath;
+};
+
+void CShellSock::ReadLine(const CString& sData) {
+	CString sLine = sData;
+
+	sLine.TrimRight("\r\n");
+	sLine.Replace("\t", "    ");
+
+	m_pParent->SetClient(m_pClient);
+	m_pParent->PutShell(sLine);
+	m_pParent->SetClient(NULL);
+}
+
+void CShellSock::Disconnected() {
+	// If there is some incomplete line in the buffer, read it
+	// (e.g. echo echo -n "hi" triggered this)
+	CString &sBuffer = GetInternalReadBuffer();
+	if (!sBuffer.empty())
+		ReadLine(sBuffer);
+
+	m_pParent->SetClient(m_pClient);
+	m_pParent->PutShell("znc$");
+	m_pParent->SetClient(NULL);
+}
+
+USERMODULEDEFS(CShellMod, "Gives shell access")
+