diff options
Diffstat (limited to 'src/modules')
| -rw-r--r-- | src/modules/m_conn_waitpong.cpp | 15 |
1 files changed, 13 insertions, 2 deletions
diff --git a/src/modules/m_conn_waitpong.cpp b/src/modules/m_conn_waitpong.cpp index ddbea55d0..245182536 100644 --- a/src/modules/m_conn_waitpong.cpp +++ b/src/modules/m_conn_waitpong.cpp @@ -62,7 +62,7 @@ class ModuleWaitPong : public Module void Implements(char* List) { - List[I_OnUserRegister] = List[I_OnCheckReady] = List[I_OnPreCommand] = List[I_OnRehash] = 1; + List[I_OnUserRegister] = List[I_OnCheckReady] = List[I_OnPreCommand] = List[I_OnRehash] = List[I_OnUserDisconnect] = 1; } virtual void OnUserRegister(userrec* user) @@ -104,7 +104,18 @@ class ModuleWaitPong : public Module { return (!user->GetExt("waitpong_pingstr")); } - + + virtual void OnUserDisconnect(userrec* user) + { + std::string* pingrpl = (std::string*)user->GetExt("waitpong_pingstr"); + + if(pingrpl) + { + delete pingrpl; + user->Shrink("waitpong_pingstr"); + } + } + virtual ~ModuleWaitPong() { } |