mirror of
https://github.com/znc/znc.git
synced 2026-03-28 17:42:41 +01:00
8cce595fbe
This was introduced in 232d2612fe which
tries to split a line by spaces and then tries splitted[1] and
splitted[2] which will cause a SIGABRT if the line isn't actually that
long.
This also uses the 4th argument to the 301 line to determine if this
message is because we sent a message to ourselfs. Previously the 3rd
option was used, which is always our own nick. This was resulting in any
301 line being halted.
114 lines
2.7 KiB
C++
114 lines
2.7 KiB
C++
/*
|
|
* 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/Nick.h>
|
|
#include <znc/IRCNetwork.h>
|
|
|
|
class CAntiIdle;
|
|
|
|
class CAntiIdleJob : public CTimer
|
|
{
|
|
public:
|
|
CAntiIdleJob(CModule* pModule, unsigned int uInterval, unsigned int uCycles, const CString& sLabel, const CString& sDescription)
|
|
: CTimer(pModule, uInterval, uCycles, sLabel, sDescription) {}
|
|
virtual ~CAntiIdleJob() {}
|
|
|
|
protected:
|
|
virtual void RunJob();
|
|
};
|
|
|
|
class CAntiIdle : public CModule
|
|
{
|
|
public:
|
|
MODCONSTRUCTOR(CAntiIdle) {
|
|
SetInterval(30);
|
|
}
|
|
|
|
virtual ~CAntiIdle() {}
|
|
|
|
virtual bool OnLoad(const CString& sArgs, CString& sErrorMsg)
|
|
{
|
|
if(!sArgs.Trim_n().empty())
|
|
SetInterval(sArgs.ToInt());
|
|
return true;
|
|
}
|
|
|
|
virtual void OnModCommand( const CString& sCommand )
|
|
{
|
|
CString sCmdName = sCommand.Token(0).AsLower();
|
|
if(sCmdName == "set")
|
|
{
|
|
CString sInterval = sCommand.Token(1, true);
|
|
SetInterval(sInterval.ToInt());
|
|
|
|
if(m_uiInterval == 0)
|
|
PutModule("AntiIdle is now turned off.");
|
|
else
|
|
PutModule("AntiIdle is now set to " + CString(m_uiInterval) + " seconds.");
|
|
} else if(sCmdName == "off") {
|
|
SetInterval(0);
|
|
PutModule("AntiIdle is now turned off");
|
|
} else if(sCmdName == "show") {
|
|
if(m_uiInterval == 0)
|
|
PutModule("AntiIdle is turned off.");
|
|
else
|
|
PutModule("AntiIdle is set to " + CString(m_uiInterval) + " seconds.");
|
|
} else {
|
|
PutModule("Commands: set, off, show");
|
|
}
|
|
}
|
|
|
|
virtual EModRet OnPrivMsg(CNick &Nick, CString &sMessage)
|
|
{
|
|
if(Nick.GetNick() == m_pNetwork->GetIRCNick().GetNick()
|
|
&& sMessage == "\xAE")
|
|
return HALT;
|
|
|
|
return CONTINUE;
|
|
}
|
|
|
|
virtual EModRet OnRaw(CString &sLine) {
|
|
/* If we send a message to ourselfs while we are away, this
|
|
* will result in the server sending a 301 which we shouldn't
|
|
* forward to the client */
|
|
|
|
if (sLine.Token(1).Equals("301") && sLine.Token(3).Equals(m_pNetwork->GetIRCNick().GetNick())) {
|
|
return HALT;
|
|
}
|
|
|
|
return CONTINUE;
|
|
}
|
|
|
|
private:
|
|
void SetInterval(int i)
|
|
{
|
|
if(i < 0)
|
|
return;
|
|
|
|
m_uiInterval = i;
|
|
|
|
RemTimer("AntiIdle");
|
|
|
|
if(m_uiInterval == 0) {
|
|
return;
|
|
}
|
|
|
|
AddTimer(new CAntiIdleJob(this, m_uiInterval, 0, "AntiIdle", "Periodically sends a msg to the user"));
|
|
}
|
|
|
|
unsigned int m_uiInterval;
|
|
};
|
|
|
|
//! This function sends a query with (r) back to the user
|
|
void CAntiIdleJob::RunJob() {
|
|
CString sNick = GetModule()->GetNetwork()->GetIRCNick().GetNick();
|
|
GetModule()->PutIRC("PRIVMSG " + sNick + " :\xAE");
|
|
}
|
|
|
|
NETWORKMODULEDEFS(CAntiIdle, "Hides your real idle time")
|