1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
|
/*
* Copyright (C) 2004-2025 ZNC, see the NOTICE file for details.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include <znc/FileUtils.h>
#include <znc/User.h>
#include <znc/znc.h>
#include <znc/ExecSock.h>
using std::vector;
// 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) {
auto e = errno;
ReadLine(t_f("Failed to execute: {1}")(strerror(e)));
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
void ReadLine(const CString& sData) override;
void Disconnected() override;
bool IsOfClient(CClient* pClient) const { return pClient == m_pClient; }
bool IsOfModule(CShellMod* pParent) const { return pParent == m_pParent; }
CShellMod* m_pParent;
private:
CClient* m_pClient;
};
class CShellMod : public CModule {
public:
MODCONSTRUCTOR(CShellMod) { m_sPath = CZNC::Get().GetHomePath(); }
~CShellMod() override {
vector<Csock*> vSocks = GetManager()->FindSocksByName("SHELL");
for (unsigned int a = 0; a < vSocks.size(); a++) {
GetManager()->DelSockByAddr(vSocks[a]);
}
}
bool OnLoad(const CString& sArgs, CString& sMessage) override {
#ifndef MOD_SHELL_ALLOW_EVERYONE
if (!GetUser()->IsAdmin()) {
sMessage = t_s("You must be admin to use the shell module");
return false;
}
#endif
return true;
}
void OnModCommand(const CString& sLine) override {
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 " + GetClient()->GetNick() + " :" + sMsg;
GetClient()->PutClient(sLine);
}
void RunCommand(const CString& sCommand) {
GetManager()->AddSock(
new CShellSock(this, GetClient(),
"cd " + m_sPath + " && " + sCommand),
"SHELL");
}
void OnClientDisconnect() override {
std::vector<Csock*> vDeadCommands;
for (Csock* pSock : *GetManager()) {
if (CShellSock* pSSock = dynamic_cast<CShellSock*>(pSock)) {
if (pSSock->IsOfModule(this) &&
pSSock->IsOfClient(GetClient())) {
vDeadCommands.push_back(pSock);
}
}
}
for (Csock* pSock : vDeadCommands) {
GetManager()->DelSockByAddr(pSock);
}
}
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(nullptr);
}
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(nullptr);
}
template <>
void TModInfo<CShellMod>(CModInfo& Info) {
Info.SetWikiPage("shell");
}
#ifdef MOD_SHELL_ALLOW_EVERYONE
USERMODULEDEFS(CShellMod, t_s("Gives shell access"))
#else
USERMODULEDEFS(CShellMod,
t_s("Gives shell access. Only ZNC admins can use it."))
#endif
|
