/*
 * 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 <gtest/gtest.h>
#include <gmock/gmock.h>

#include "znctest.h"
#include "znctestconfig.h"

using testing::HasSubstr;
using testing::Contains;
using testing::ContainsRegex;
using testing::SizeIs;
using testing::Lt;
using testing::Gt;
using testing::AllOf;
using testing::AnyOf;

namespace znc_inttest {
namespace {

TEST(Config, AlreadyExists) {
    QTemporaryDir dir;
    WriteConfig(dir.path());
    Process p(ZNC_BIN_DIR "/znc", QStringList() << "--debug"
                                                << "--datadir" << dir.path()
                                                << "--makeconf");
    p.ReadUntil("already exists");
    p.CanDie();
}

TEST_F(ZNCTest, Connect) {
    auto znc = Run();

    auto ircd = ConnectIRCd();
    ircd.ReadUntil("CAP LS");

    auto client = ConnectClient();
    client.Write("PASS :hunter2");
    client.Write("NICK nick");
    client.Write("USER user/test x x :x");
    client.ReadUntil("Welcome");
    client.Close();

    client = ConnectClient();
    client.Write("PASS :user:hunter2");
    client.Write("NICK nick");
    client.Write("USER u x x x");
    client.ReadUntil("Welcome");
    client.Close();

    client = ConnectClient();
    client.Write("NICK nick");
    client.Write("USER user x x x");
    client.ReadUntil("Configure your client to send a server password");
    client.Close();

    ircd.Write(":server 001 nick :Hello");
    ircd.ReadUntil("WHO");
}

TEST_F(ZNCTest, Channel) {
    auto znc = Run();
    auto ircd = ConnectIRCd();

    auto client = LoginClient();
    client.ReadUntil("Welcome");
    client.Write("JOIN #znc");
    client.Close();

    ircd.Write(":server 001 nick :Hello");
    ircd.ReadUntil("JOIN #znc");
    ircd.Write(":nick JOIN :#znc");
    ircd.Write(":server 353 nick #znc :nick");
    ircd.Write(":server 366 nick #znc :End of /NAMES list");
    ircd.Write(":server PING :1");
    ircd.ReadUntil("PONG 1");

    client = LoginClient();
    client.ReadUntil(":nick JOIN :#znc");
}

TEST_F(ZNCTest, HTTP) {
    auto znc = Run();
    auto ircd = ConnectIRCd();

    auto client = LoginClient();
    int port = PickPortNumber();
    client.Write(QStringLiteral("znc addport %1 all all").arg(port).toUtf8());
    client.ReadUntil(":Port added");

    auto reply = HttpGet(QNetworkRequest(
        QUrl(QStringLiteral("http://127.0.0.1:%1/").arg(port))));
    EXPECT_THAT(reply->rawHeader("Server").toStdString(), HasSubstr("ZNC"));
}

TEST_F(ZNCTest, FixCVE20149403) {
    auto znc = Run();
    auto ircd = ConnectIRCd();
    ircd.Write(":server 001 nick :Hello");
    ircd.Write(":server 005 nick CHANTYPES=# :supports");
    ircd.Write(":server PING :1");
    ircd.ReadUntil("PONG 1");

    auto client = LoginClient();
    int port = PickPortNumber();
    client.Write(QStringLiteral("znc addport %1 all all").arg(port).toUtf8());
    client.ReadUntil(":Port added");

    QNetworkRequest request;
    request.setRawHeader("Authorization",
                         "Basic " + QByteArray("user:hunter2").toBase64());
    request.setUrl(
        QUrl(QStringLiteral("http://127.0.0.1:%1/mods/global/webadmin/addchan")
                 .arg(port)));
    HttpPost(request, {
                          {"user", "user"},
                          {"network", "test"},
                          {"submitted", "1"},
                          {"name", "znc"},
                          {"enabled", "1"},
                      });
    EXPECT_THAT(HttpPost(request,
                         {
                             {"user", "user"},
                             {"network", "test"},
                             {"submitted", "1"},
                             {"name", "znc"},
                             {"enabled", "1"},
                         })
                    ->readAll()
                    .toStdString(),
                HasSubstr("Channel [#znc] already exists"));
}

TEST_F(ZNCTest, FixFixOfCVE20149403) {
    auto znc = Run();
    auto ircd = ConnectIRCd();
    ircd.Write(":server 001 nick :Hello");
    ircd.Write(":nick JOIN @#znc");
    ircd.ReadUntil("MODE @#znc");
    ircd.Write(":server 005 nick STATUSMSG=@ :supports");
    ircd.Write(":server PING :12345");
    ircd.ReadUntil("PONG 12345");

    auto client = LoginClient();
    int port = PickPortNumber();
    client.Write(QStringLiteral("znc addport %1 all all").arg(port).toUtf8());
    client.ReadUntil(":Port added");

    QNetworkRequest request;
    request.setRawHeader("Authorization",
                         "Basic " + QByteArray("user:hunter2").toBase64());
    request.setUrl(
        QUrl(QStringLiteral("http://127.0.0.1:%1/mods/global/webadmin/addchan")
                 .arg(port)
                 .toUtf8()));
    auto reply = HttpPost(request, {
                                       {"user", "user"},
                                       {"network", "test"},
                                       {"submitted", "1"},
                                       {"name", "@#znc"},
                                       {"enabled", "1"},
                                   });
    EXPECT_THAT(reply->readAll().toStdString(),
                HasSubstr("Could not add channel [@#znc]"));
}

TEST_F(ZNCTest, InvalidConfigInChan) {
    QFile conf(m_dir.path() + "/configs/znc.conf");
    ASSERT_TRUE(conf.open(QIODevice::Append | QIODevice::Text));
    QTextStream out(&conf);
    out << R"(
        <User foo>
            <Network bar>
                <Chan #baz>
                    Invalid = Line
                </Chan>
            </Network>
        </User>
    )";
    out.flush();
    auto znc = Run();
    znc->ShouldFinishItself(1);
}

TEST_F(ZNCTest, Encoding) {
    auto znc = Run();
    auto ircd = ConnectIRCd();
    auto client = LoginClient();
    ircd.Write(":server 001 nick :hello");
    // legacy
    ircd.Write(":n!u@h PRIVMSG nick :Hello\xE6world");
    client.ReadUntil("Hello\xE6world");
    client.Write("PRIVMSG *controlpanel :SetNetwork Encoding $me $net UTF-8");
    client.ReadUntil("Encoding = UTF-8");
    ircd.Write(":n!u@h PRIVMSG nick :Hello\xE6world");
    client.ReadUntil("Hello\xEF\xBF\xBDworld");
    client.Write(
        "PRIVMSG *controlpanel :SetNetwork Encoding $me $net ^CP-1251");
    client.ReadUntil("Encoding = ^CP-1251");
    ircd.Write(":n!u@h PRIVMSG nick :Hello\xE6world");
    client.ReadUntil("Hello\xD0\xB6world");
    ircd.Write(":n!u@h PRIVMSG nick :Hello\xD0\xB6world");
    client.ReadUntil("Hello\xD0\xB6world");
}

TEST_F(ZNCTest, BuildMod) {
    auto znc = Run();
    auto ircd = ConnectIRCd();
    auto client = LoginClient();
    QTemporaryDir srcd;
    QDir srcdir(srcd.path());
    QFile file(srcdir.filePath("testmod.cpp"));
    ASSERT_TRUE(file.open(QIODevice::WriteOnly | QIODevice::Text));
    QTextStream out(&file);
    out << R"(
        #include <znc/Modules.h>
        class TestModule : public CModule {
          public:
            MODCONSTRUCTOR(TestModule) {}
            void OnModCommand(const CString& sLine) override {
                PutModule("Lorem ipsum");
            }
        };
        MODULEDEFS(TestModule, "Test")
    )";
    file.close();
    QDir dir(m_dir.path());
    EXPECT_TRUE(dir.mkdir("modules"));
    EXPECT_TRUE(dir.cd("modules"));
    {
        Process p(ZNC_BIN_DIR "/znc-buildmod",
                  QStringList() << srcdir.filePath("file-not-found.cpp"),
                  [&](QProcess* proc) {
                      proc->setWorkingDirectory(dir.absolutePath());
                      proc->setProcessChannelMode(QProcess::ForwardedChannels);
                  });
        p.ShouldFinishItself(1);
        p.ShouldFinishInSec(300);
    }
    {
        Process p(ZNC_BIN_DIR "/znc-buildmod",
                  QStringList() << srcdir.filePath("testmod.cpp"),
                  [&](QProcess* proc) {
                      proc->setWorkingDirectory(dir.absolutePath());
                      proc->setProcessChannelMode(QProcess::ForwardedChannels);
                  });
        p.ShouldFinishItself();
        p.ShouldFinishInSec(300);
    }
    client.Write("znc loadmod testmod");
    client.Write("PRIVMSG *testmod :hi");
    client.ReadUntil("Lorem ipsum");
}

TEST_F(ZNCTest, AwayNotify) {
    auto znc = Run();
    auto ircd = ConnectIRCd();
    auto client = ConnectClient();
    client.Write("CAP LS");
    client.Write("PASS :hunter2");
    client.Write("NICK nick");
    client.Write("USER user/test x x :x");
    QByteArray cap_ls;
    client.ReadUntilAndGet(" LS :", cap_ls);
    ASSERT_THAT(cap_ls.toStdString(),
                AllOf(HasSubstr("cap-notify"), Not(HasSubstr("away-notify"))));
    client.Write("CAP REQ :cap-notify");
    client.ReadUntil("ACK :cap-notify");
    client.Write("CAP END");
    client.ReadUntil(" 001 ");
    ircd.ReadUntil("USER");
    ircd.Write("CAP user LS :away-notify");
    ircd.ReadUntil("CAP REQ :away-notify");
    ircd.Write("CAP user ACK :away-notify");
    ircd.ReadUntil("CAP END");
    ircd.Write(":server 001 user :welcome");
    client.ReadUntil("CAP user NEW :away-notify");
    client.Write("CAP REQ :away-notify");
    client.ReadUntil("ACK :away-notify");
    ircd.Write(":x!y@z AWAY :reason");
    client.ReadUntil(":x!y@z AWAY :reason");
    ircd.Close();
    client.ReadUntil("DEL :away-notify");
    // This test often fails on macos due to ZNC process not finishing.
    // No idea why. Let's try to shutdown it more explicitly...
    client.Write("znc shutdown");
}

TEST_F(ZNCTest, ExtendedJoin) {
    auto znc = Run();
    auto ircd = ConnectIRCd();
    auto client = LoginClient();
    ircd.Write(":server 001 user :welcome");
    client.ReadUntil(" 001 ");
    ircd.Write(":nick!user@host JOIN #channel account :Real Name");
    // Not sure why it is like this when server sends such format unexpectedly.
    client.ReadUntil("JOIN #channel account :Real Name");
    ircd.Write("CAP nick ACK extended-join");
    ircd.Write(":nick!user@host JOIN #channel2 account :Real Name");
    QByteArray line;
    client.ReadUntilAndGet("JOIN", line);
    EXPECT_EQ(line.toStdString(), "JOIN #channel2");
    client.Write("CAP REQ extended-join");
    client.ReadUntil("CAP user ACK :extended-join");
    ircd.Write(":nick!user@host JOIN #channel3 account :Real Name");
    client.ReadUntil("JOIN #channel3 account :Real Name");
}

TEST_F(ZNCTest, CAP302LSWaitFull) {
    auto znc = Run();
    auto ircd = ConnectIRCd();
    ircd.ReadUntil("CAP LS 302");
    ircd.Write("CAP user LS * :away-notify");
    ASSERT_THAT(ircd.ReadRemainder().toStdString(), Not(HasSubstr("away-notify")));
    ircd.Write("CAP user LS :blahblah");
    ircd.ReadUntil("CAP REQ :away-notify");
}

TEST_F(ZNCTest, CAP302NewDel) {
    auto znc = Run();
    auto ircd = ConnectIRCd();
    auto client = LoginClient();
    ircd.Write("CAP nick LS :blahblah");
    ircd.ReadUntil("CAP END");
    ircd.Write(":server 001 nick :Hello");
    client.Write("CAP REQ :away-notify");
    client.ReadUntil("NAK :away-notify");
    client.Write("CAP REQ :cap-notify");
    client.ReadUntil("ACK :cap-notify");
    ircd.Write("CAP nick NEW :away-notify");
    ircd.ReadUntil("CAP REQ :away-notify");
    ircd.Write("CAP nick ACK :away-notify");
    client.ReadUntil("CAP nick NEW :away-notify");
    ircd.Write("CAP nick DEL :away-notify");
    client.ReadUntil("CAP nick DEL :away-notify");
}

TEST_F(ZNCTest, JoinKey) {
    QFile conf(m_dir.path() + "/configs/znc.conf");
    ASSERT_TRUE(conf.open(QIODevice::Append | QIODevice::Text));
    QTextStream(&conf) << "ServerThrottle = 1\n";
    auto znc = Run();

    auto ircd = ConnectIRCd();
    auto client = LoginClient();
    ircd.Write(":server 001 nick :Hello");
    client.Write("JOIN #znc secret");
    ircd.ReadUntil("JOIN #znc secret");
    ircd.Write(":nick JOIN :#znc");
    client.ReadUntil("JOIN :#znc");
    ircd.Close();

    ircd = ConnectIRCd();
    ircd.Write(":server 001 nick :Hello");
    ircd.ReadUntil("JOIN #znc secret");
}

TEST_F(ZNCTest, StatusEchoMessage) {
    auto znc = Run();
    auto ircd = ConnectIRCd();
    auto client = LoginClient();
    client.Write("CAP REQ :echo-message");
    client.Write("PRIVMSG *status :blah");
    client.ReadUntil(":nick!user@irc.znc.in PRIVMSG *status :blah");
    client.ReadUntil(":*status!status@znc.in PRIVMSG nick :Unknown command");
    client.Write("znc delnetwork test");
    client.ReadUntil("Network deleted");
    auto client2 = LoginClient();
    client2.Write("PRIVMSG *status :blah2");
    client2.ReadUntil(":*status!status@znc.in PRIVMSG nick :Unknown command");
    auto client3 = LoginClient();
    client3.Write("PRIVMSG *status :blah3");
    client3.ReadUntil(":*status!status@znc.in PRIVMSG nick :Unknown command");
}

TEST_F(ZNCTest, MoveChannels) {
    auto znc = Run();
    auto ircd = ConnectIRCd();

    auto client = LoginClient();
    client.Write("JOIN #foo,#bar");
    client.Close();

    ircd.Write(":server 001 nick :Hello");
    QByteArray joins;
    ircd.ReadUntilAndGet("JOIN", joins);
    ASSERT_THAT(joins.toStdString(), AnyOf(HasSubstr("JOIN #foo,#bar"),
                                           HasSubstr("JOIN #bar,#foo")));
    ircd.Write(":nick JOIN :#foo");
    ircd.Write(":server 353 nick #foo :nick");
    ircd.Write(":server 366 nick #foo :End of /NAMES list");
    ircd.Write(":nick JOIN :#bar");
    ircd.Write(":server 353 nick #bar :nick");
    ircd.Write(":server 366 nick #bar :End of /NAMES list");

    client = LoginClient();
    client.ReadUntil(":nick JOIN :#foo");
    client.ReadUntil(":nick JOIN :#bar");
    client.Write("znc movechan #foo 2");
    client.ReadUntil("Moved channel #foo to index 2");
    client.Close();

    client = LoginClient();
    client.ReadUntil(":nick JOIN :#bar");
    client.ReadUntil(":nick JOIN :#foo");
    client.Write("znc swapchans #foo #bar");
    client.ReadUntil("Swapped channels #foo and #bar");
    client.Close();

    client = LoginClient();
    client.ReadUntil(":nick JOIN :#foo");
    client.ReadUntil(":nick JOIN :#bar");
}

TEST_F(ZNCTest, DenyOptions) {
    auto znc = Run();
    auto ircd = ConnectIRCd();

    auto client1 = LoginClient();
    client1.Write("PRIVMSG *controlpanel :CloneUser user user2");
    client1.ReadUntil("User user2 added!");
    client1.Write("PRIVMSG *controlpanel :Set Admin user2 false");
    client1.ReadUntil("Admin = false");
    client1.Write("PRIVMSG *controlpanel :Set MaxNetworks user2 5");
    client1.ReadUntil("MaxNetworks = 5");

    auto client2 = ConnectClient();
    client2.Write("PASS :hunter2");
    client2.Write("NICK nick2");
    client2.Write("USER user2/test x x :x");

    // DenySetNetwork
    // This is false by default so we should be able to add/delete networks/servers
    client2.Write("PRIVMSG *controlpanel :AddNetwork user2 test2");
    client2.ReadUntil("Network test2 added to user user2.");
    client2.Write("PRIVMSG *controlpanel :AddServer user2 test2 127.0.0.1");
    client2.ReadUntil("Added IRC Server 127.0.0.1 to network test2 for user user2.");
    client2.Write("PRIVMSG *controlpanel :DelServer user2 test2 127.0.0.1");
    client2.ReadUntil("Deleted IRC Server 127.0.0.1 from network test2 for user user2.");
    client2.Write("PRIVMSG *controlpanel :DelNetwork user2 test2");
    client2.ReadUntil("Network test2 deleted for user user2.");

    // Set it to true
    client1.Write("PRIVMSG *controlpanel :Set DenySetNetwork user2 true");
    client1.ReadUntil("DenySetNetwork = true");

    // Now we should be denied
    client2.Write("PRIVMSG *controlpanel :AddNetwork user2 test2");
    client2.ReadUntil("Access denied!");
    client2.Write("PRIVMSG *controlpanel :AddServer user2 test 127.0.0.2");
    client2.ReadUntil("Access denied!");
    client2.Write("PRIVMSG *controlpanel :DelServer user2 test 127.0.0.1");
    client2.ReadUntil("Access denied!");
    client2.Write("PRIVMSG *controlpanel :DelNetwork user2 test");
    client2.ReadUntil("Access denied!");

    // DenySetBindHost
    client2.Write("PRIVMSG *controlpanel :Set BindHost user2 127.0.0.1");
    client2.ReadUntil("BindHost = 127.0.0.1");
    client1.Write("PRIVMSG *controlpanel :Set DenySetBindHost user2 true");
    client1.ReadUntil("DenySetBindHost = true");
    client2.Write("PRIVMSG *controlpanel :Set BindHost user2 127.0.0.2");
    client2.ReadUntil("Access denied!");

    // DenySetIdent
    client2.Write("PRIVMSG *controlpanel :Set Ident user2 test");
    client2.ReadUntil("Ident = test");
    client1.Write("PRIVMSG *controlpanel :Set DenySetIdent user2 true");
    client1.ReadUntil("DenySetIdent = true");
    client2.Write("PRIVMSG *controlpanel :Set Ident user2 test2");
    client2.ReadUntil("Access denied!");

    // DenySetRealName
    client2.Write("PRIVMSG *controlpanel :Set RealName user2 test");
    client2.ReadUntil("RealName = test");
    client1.Write("PRIVMSG *controlpanel :Set DenySetRealName user2 true");
    client1.ReadUntil("DenySetRealName = true");
    client2.Write("PRIVMSG *controlpanel :Set RealName user2 test2");
    client2.ReadUntil("Access denied!");

    // DenySetQuitMsg
    client2.Write("PRIVMSG *controlpanel :Set QuitMsg user2 test");
    client2.ReadUntil("QuitMsg = test");
    client1.Write("PRIVMSG *controlpanel :Set DenySetQuitMsg user2 true");
    client1.ReadUntil("DenySetQuitMsg = true");
    client2.Write("PRIVMSG *controlpanel :Set QuitMsg user2 test2");
    client2.ReadUntil("Access denied!");

    // DenySetCTCPReplies
    client2.Write("PRIVMSG *controlpanel :AddCTCP user2 FOO BAR");
    client2.ReadUntil("CTCP requests FOO to user user2 will now get reply: BAR");
    client2.Write("PRIVMSG *controlpanel :DelCTCP user2 FOO");
    client2.ReadUntil("CTCP requests FOO to user user2 will now be sent to IRC clients");
    client1.Write("PRIVMSG *controlpanel :Set DenySetCTCPReplies user2 true");
    client1.ReadUntil("DenySetCTCPReplies = true");
    client2.Write("PRIVMSG *controlpanel :AddCTCP user2 FOO BAR");
    client2.ReadUntil("Access denied!");
    client2.Write("PRIVMSG *controlpanel :DelCTCP user2 FOO");
    client2.ReadUntil("Access denied!");
}

TEST_F(ZNCTest, CAP302MultiLS) {
    auto znc = Run();
    auto ircd = ConnectIRCd();
    auto client = LoginClient();
    InstallModule("testmod.cpp", R"(
        #include <znc/Client.h>
        #include <znc/Modules.h>
        class TestModule : public CModule {
          public:
            MODCONSTRUCTOR(TestModule) {}
            void OnClientCapLs(CClient* pClient, SCString& ssCaps) override {
                for (int i = 0; i < 100; ++i) {
                    ssCaps.insert("testcap-" + CString(i));
                }
            }
        };
        GLOBALMODULEDEFS(TestModule, "Test")
    )");
    client.Write("znc loadmod testmod");
    client.ReadUntil("Loaded module testmod");

    auto client2 = ConnectClient();
    client2.Write("CAP LS");
    client2.ReadUntil("LS :");
    auto rem = client2.ReadRemainder();
    ASSERT_GT(rem.indexOf("testcap-10"), 10);
    ASSERT_EQ(rem.indexOf("testcap-80"), -1);
    ASSERT_EQ(rem.indexOf("LS"), -1);

    client2 = ConnectClient();
    client2.Write("CAP LS 302");
    client2.ReadUntil("LS * :");
    rem = client2.ReadRemainder();
    int w = 0;
    ASSERT_GT(w = rem.indexOf("testcap-10"), 1);
    ASSERT_GT(w = rem.indexOf("testcap-22", w), 1);
    ASSERT_GT(w = rem.indexOf("LS * :", w), 1);
    ASSERT_GT(rem.indexOf("testcap-80", w), 1);
    ASSERT_GT(rem.indexOf("LS :", w), 1);
}

TEST_F(ZNCTest, CAP302LSValue) {
    auto znc = Run();
    auto ircd = ConnectIRCd();
    auto client = LoginClient();
    InstallModule("testmod.cpp", R"(
        #include <znc/Client.h>
        #include <znc/Modules.h>
        class TestModule : public CModule {
          public:
            MODCONSTRUCTOR(TestModule) {}
            void OnClientCapLs(CClient* pClient, SCString& ssCaps) override {
                if (pClient->HasCap302()) {
                    ssCaps.insert("testcap=blah");
                } else {
                    ssCaps.insert("testcap");
                }
            }
        };
        GLOBALMODULEDEFS(TestModule, "Test")
    )");
    client.Write("znc loadmod testmod");
    client.ReadUntil("Loaded module testmod");

    auto client2 = ConnectClient();
    client2.Write("CAP LS");
    client2.ReadUntil("testcap ");

    client2 = ConnectClient();
    client2.Write("CAP LS 302");
    client2.ReadUntil("testcap=");
}

class AllLanguages : public ZNCTest, public testing::WithParamInterface<int> {};

INSTANTIATE_TEST_CASE_P(LanguagesTests, AllLanguages, testing::Values(1, 2, 3));

TEST_P(AllLanguages, ServerDependentCapInModule) {
    auto znc = Run();
    auto ircd = ConnectIRCd();
    auto client = LoginClient();
    switch (GetParam()) {
        case 1:
            InstallModule("testmod.cpp", R"(
                #include <znc/Modules.h>
                class TestModule : public CModule {
                    class TestCap : public CCapability {
                        void OnServerChangedSupport(CIRCNetwork* pNetwork, bool bState) override {
                            GetModule()->PutModule("Server changed support: " + CString(bState));
                        }
                        void OnClientChangedSupport(CClient* pClient, bool bState) override {
                            GetModule()->PutModule("Client changed support: " + CString(bState));
                        }
                    };
                  public:
                    MODCONSTRUCTOR(TestModule) {
                        AddServerDependentCapability("testcap", std::make_unique<TestCap>());
                    }
                };
                MODULEDEFS(TestModule, "Test")
            )");
            break;
        case 2:
#ifndef WANT_PYTHON
            GTEST_SKIP() << "Modpython is disabled";
#endif
            znc->CanLeak();
            InstallModule("testmod.py", R"(
                import znc
                class testmod(znc.Module):
                    def OnLoad(self, args, ret):
                        def server_change(net, state):
                            self.PutModule('Server changed support: ' + ('true' if state else 'false'))
                        def client_change(client, state):
                            self.PutModule('Client changed support: ' + ('true' if state else 'false'))
                        self.AddServerDependentCapability('testcap', server_change, client_change)
                        return True
                )");
            client.Write("znc loadmod modpython");
            break;
        case 3:
#ifndef WANT_PERL
            GTEST_SKIP() << "Modperl is disabled";
#endif
            znc->CanLeak();
            InstallModule("testmod.pm", R"(
                package testmod;
                use base 'ZNC::Module';
                sub OnLoad {
                    my $self = shift;
                    $self->AddServerDependentCapability('testcap', sub {
                        my ($net, $state) = @_;
                        $self->PutModule('Server changed support: ' . ($state ? 'true' : 'false'));
                    }, sub {
                        my ($client, $state) = @_;
                        $self->PutModule('Client changed support: ' . ($state ? 'true' : 'false'));
                    });
                    return 1;
                }
                1;
            )");
            client.Write("znc loadmod modperl");
            break;
    }
    client.Write("znc loadmod testmod");
    client.ReadUntil("Loaded module testmod");
    client.Write("znc addnetwork net2");
    client.Close();

    client = ConnectClient();
    client.Write("CAP LS 302");
    client.Write("PASS :hunter2");
    client.Write("NICK nick");
    client.Write("USER user/test x x :x");
    client.Write("CAP END");
    client.ReadUntil("Welcome");

    ircd.Write("001 nick Welcome");
    ircd.Write("CAP nick NEW testcap=value");
    ircd.ReadUntil("CAP REQ :testcap");
    ircd.Write("CAP nick ACK :testcap");
    client.ReadUntil(":Server changed support: true");
    client.ReadUntil("CAP nick NEW :testcap=value");
    client.Write("CAP REQ testcap");
    client.ReadUntil(":Client changed support: true");
    client.ReadUntil("CAP nick ACK :testcap");

    client.Write("CAP LS");
    client.ReadUntil(" testcap=value ");

    ircd.Write("CAP nick DEL testcap");
    client.ReadUntil(":Server changed support: false");
    client.ReadUntil("CAP nick DEL :testcap");
    client.ReadUntil(":Client changed support: false");

    ircd.Close();
    // TODO combine multiple DELs to single line
    client.ReadUntil("CAP nick DEL :testcap");
    client.ReadUntil(":Client changed support: false");

    ircd = ConnectIRCd();
    ircd.ReadUntil("CAP LS 302");
    ircd.Write("CAP nick LS :testcap=new");
    ircd.ReadUntil("CAP REQ :testcap");
    ircd.Write("CAP nick ACK :testcap");
    client.ReadUntil(":Server changed support: true");
    ircd.ReadUntil("CAP END");
    // NEW waits until 001
    ASSERT_THAT(ircd.ReadRemainder().toStdString(), Not(HasSubstr("testcap")));
    ircd.Write("001 nick Welcome");
    // TODO combine multiple NEWs to single line
    client.ReadUntil("CAP nick NEW :testcap=new");
    client.ReadUntil("Welcome");

    // NEW with new value without DEL
    ircd.Write("CAP nick NEW testcap=another");
    client.ReadUntil("CAP nick NEW :testcap=another");

    client.Write("znc jumpnetwork net2");
    client.ReadUntil("CAP nick DEL :testcap");
    client.ReadUntil(":Client changed support: false");

    client.Write("znc jumpnetwork test");
    client.ReadUntil("CAP nick NEW :testcap=another");
}

TEST_F(ZNCTest, HashUpgrade) {
    QFile conf(m_dir.path() + "/configs/znc.conf");
    ASSERT_TRUE(conf.open(QIODevice::Append | QIODevice::Text));
    QTextStream out(&conf);
    out << R"(
        <User foo>
            <Pass pass>
                Method = MD5
                Salt = abc
                Hash = defdf93cef7fa7a8ee88e65d0e277b99
            </Pass>
        </User>
    )";
    out.flush();
    conf.close();
    auto znc = Run();
    auto ircd = ConnectIRCd();

    auto client = ConnectClient();
    client.Write("PASS :hunter2");
    client.Write("NICK nick");
    client.Write("USER foo x x :x");
    client.ReadUntil("Welcome");
    client.Close();

    client = LoginClient();
    client.Write("znc saveconfig");
    client.ReadUntil("Wrote config");

    ASSERT_TRUE(conf.open(QIODevice::ReadOnly | QIODevice::Text));
    QTextStream in(&conf);
    QString config = in.readAll();
    // It was upgraded to either Argon2 or SHA256
    EXPECT_THAT(config.toStdString(), Not(ContainsRegex("Method.*MD5")));

    // Check that still can login after the upgrade
    client = ConnectClient();
    client.Write("PASS :hunter2");
    client.Write("NICK nick");
    client.Write("USER foo x x :x");
    client.ReadUntil("Welcome");
    client.Close();
}

TEST_F(ZNCTest, CapReqWithoutLs) {
    auto znc = Run();
    auto ircd = ConnectIRCd();

    auto client = ConnectClient();
    client.Write("CAP REQ nonono");
    client.Write("PASS :hunter2");
    client.Write("NICK nick");
    client.Write("USER foo x x :x");
    ASSERT_THAT(client.ReadRemainder().toStdString(), Not(HasSubstr("Welcome")));
}

TEST_F(ZNCTest, ChgHostEmulation) {
    auto znc = Run();
    auto ircd = ConnectIRCd();
    ircd.Write("CAP user LS :chghost");
    ircd.ReadUntil("CAP REQ :chghost");
    ircd.Write("CAP user ACK :chghost");

    auto client1 = LoginClient();
    auto client2 = LoginClient();
    client2.Write("CAP REQ :chghost");
    client2.ReadUntil("ACK");

    ircd.Write(":user!oldident@oldhost JOIN #chan");

    ircd.Write(":user!oldident@oldhost CHGHOST newident newhost");
    client1.ReadUntil(":user!oldident@oldhost QUIT :Changing hostname");
    client1.ReadUntil(":user!newident@newhost JOIN #chan");
    ASSERT_THAT(client1.ReadRemainder().toStdString(), Not(HasSubstr("MODE")));
    client2.ReadUntil(":user!oldident@oldhost CHGHOST newident newhost");
    client2.Close();

    ircd.Write(":server MODE #chan +v user");
    client1.ReadUntil("MODE");
    ircd.Write(":user!newident@newhost CHGHOST ident-2 host-2");
    client1.ReadUntil(":irc.znc.in MODE #chan +v user");

    ircd.Write(":server MODE #chan +o user");
    client1.ReadUntil("MODE");
    ircd.Write(":user!ident-2@host-2 CHGHOST ident-3 host-3");
    client1.ReadUntil(":irc.znc.in MODE #chan +ov user user");

    // Only attached channel should receive emulation
    ircd.Write(":user!ident-3@host-3 JOIN #chan2");
    client1.ReadUntil("JOIN #chan2");
    client1.Write("DETACH #chan2");
    client1.ReadUntil("Detached 1 channel");
    ircd.Write(":user!ident-3@host-3 CHGHOST ident host");
    ASSERT_THAT(client1.ReadRemainder().toStdString(),
                Not(HasSubstr("#chan2")));
}

TEST_F(ZNCTest, ChgHostOnce) {
    auto znc = Run();
    auto ircd = ConnectIRCd();
    ircd.Write("CAP user LS :chghost");
    ircd.ReadUntil("CAP REQ :chghost");
    ircd.Write("CAP user ACK :chghost");

    auto client = LoginClient();
    client.Write("CAP REQ :chghost");
    client.ReadUntil("ACK");

    ircd.Write(":user!oldident@oldhost JOIN #chan");
    ircd.Write(":user!oldident@oldhost JOIN #chan2");
    ircd.Write(":user!oldident@oldhost CHGHOST newident newhost");
    client.ReadUntil("CHGHOST");
    ASSERT_THAT(client.ReadRemainder().toStdString(),
                Not(HasSubstr("CHGHOST")));
}

TEST_F(ZNCTest, ChgHostOnlyNicksAlreadyOnChannels) {
    auto znc = Run();
    auto ircd = ConnectIRCd();
    ircd.Write("CAP user LS :chghost");
    ircd.ReadUntil("CAP REQ :chghost");
    ircd.Write("CAP user ACK :chghost");

    auto client = LoginClient();
    ircd.Write(":user!ident@host JOIN #chan1");
    ircd.Write(":user!ident@host JOIN #chan2");
    ircd.Write(":another!ident@host JOIN #chan1");
    client.ReadUntil("another");

    ircd.Write(":another!ident@host CHGHOST i2 h2");
    ASSERT_THAT(client.ReadRemainder().toStdString(),
                AllOf(HasSubstr("JOIN #chan1"),
                    Not(HasSubstr("JOIN #chan2"))));
}

TEST_F(ZNCTest, SaslAuthPlainSimple) {
    auto znc = Run();
    auto ircd = ConnectIRCd();
    auto client = ConnectClient();
    client.Write("NICK foo");
    client.Write("CAP LS");
    client.ReadUntil(" sasl ");
    client.Write("CAP REQ :sasl");
    client.ReadUntil(":irc.znc.in CAP foo ACK :sasl");
    client.Write("USER bar");
    client.Write("AUTHENTICATE PLAIN");
    client.ReadUntil("AUTHENTICATE +");
    client.Write("AUTHENTICATE " + QByteArrayLiteral("\0user\0hunter2").toBase64());
    client.ReadUntil(":irc.znc.in 903 foo :SASL authentication successful");
}

TEST_F(ZNCTest, SaslAuthPlainCopyInZ) {
    auto znc = Run();
    auto ircd = ConnectIRCd();
    auto client = ConnectClient();
    client.Write("NICK foo");
    client.Write("CAP LS");
    client.ReadUntil(" sasl ");
    client.Write("CAP REQ :sasl");
    client.ReadUntil(":irc.znc.in CAP foo ACK :sasl");
    client.Write("USER bar");
    client.Write("AUTHENTICATE PLAIN");
    client.ReadUntil("AUTHENTICATE +");
    client.Write("AUTHENTICATE " + QByteArrayLiteral("user@phone\0user@phone\0hunter2").toBase64());
    client.ReadUntil(":irc.znc.in 903 foo :SASL authentication successful");
    client.Write("CAP END");
    client.Write("znc listclients");
    client.ReadUntil("phone");
}

TEST_F(ZNCTest, SaslAuthPlainPartialInZ) {
    auto znc = Run();
    auto ircd = ConnectIRCd();
    auto client = ConnectClient();
    client.Write("NICK foo");
    client.Write("CAP LS");
    client.ReadUntil(" sasl ");
    client.Write("CAP REQ :sasl");
    client.ReadUntil(":irc.znc.in CAP foo ACK :sasl");
    client.Write("USER bar");
    client.Write("AUTHENTICATE PLAIN");
    client.ReadUntil("AUTHENTICATE +");
    client.Write("AUTHENTICATE " + QByteArrayLiteral("user@phone\0user\0hunter2").toBase64());
    client.ReadUntil(":irc.znc.in 903 foo :SASL authentication successful");
    client.Write("CAP END");
    client.Write("znc listclients");
    client.ReadUntil("phone");
}

TEST_F(ZNCTest, SaslAuthPlainDifferentZ) {
    auto znc = Run();
    auto ircd = ConnectIRCd();
    auto client = ConnectClient();
    client.Write("NICK foo");
    client.Write("CAP LS");
    client.ReadUntil(" sasl ");
    client.Write("CAP REQ :sasl");
    client.ReadUntil(":irc.znc.in CAP foo ACK :sasl");
    client.Write("USER bar");
    client.Write("AUTHENTICATE PLAIN");
    client.ReadUntil("AUTHENTICATE +");
    client.Write("AUTHENTICATE " + QByteArrayLiteral("user@phone\0user@tablet\0hunter2").toBase64());
    client.ReadUntil(":irc.znc.in 904 foo :No support for custom AuthzId");
}

TEST_F(ZNCTest, SaslAuthPlainWrongPassword) {
    auto znc = Run();
    auto ircd = ConnectIRCd();
    auto client = ConnectClient();
    client.Write("NICK foo");
    client.Write("CAP LS");
    client.ReadUntil(" sasl ");
    client.Write("CAP REQ :sasl");
    client.ReadUntil(":irc.znc.in CAP foo ACK :sasl");
    client.Write("USER bar");
    client.Write("AUTHENTICATE PLAIN");
    client.ReadUntil("AUTHENTICATE +");
    client.Write("AUTHENTICATE " + QByteArrayLiteral("\0user\0hunter3").toBase64());
    client.ReadUntil(":irc.znc.in 904 foo :Invalid Password");

    // Try again on the same connection
    client.Write("AUTHENTICATE PLAIN");
    client.ReadUntil(":irc.znc.in 904 foo :SASL authentication failed");
}

TEST_F(ZNCTest, SaslAuthPlainWrongUser) {
    auto znc = Run();
    auto ircd = ConnectIRCd();
    auto client = ConnectClient();
    client.Write("NICK foo");
    client.Write("CAP LS");
    client.ReadUntil(" sasl ");
    client.Write("CAP REQ :sasl");
    client.ReadUntil(":irc.znc.in CAP foo ACK :sasl");
    client.Write("USER bar");
    client.Write("AUTHENTICATE PLAIN");
    client.ReadUntil("AUTHENTICATE +");
    client.Write("AUTHENTICATE " + QByteArrayLiteral("\0anotheruser\0hunter2").toBase64());
    client.ReadUntil(":irc.znc.in 904 foo :Invalid Password");
}

TEST_F(ZNCTest, SaslAuthUserAfterCapEnd) {
    // kvirc sends this
    auto znc = Run();
    auto ircd = ConnectIRCd();
    auto client = ConnectClient();
    client.Write("CAP LS");
    client.Write("PING :::1");
    client.Write("CAP REQ :sasl");
    client.Write("AUTHENTICATE PLAIN");
    client.Write("AUTHENTICATE " +
                 QByteArrayLiteral("\0user\0hunter2").toBase64());
    client.Write("CAP END");
    client.ReadUntil("903 unknown-nick :SASL authentication successful");
    client.Write("NICK nick");
    client.Write("USER user 0 1 :2");
    client.ReadUntil("001");
}

TEST_F(ZNCTest, SaslAuthAbort) {
    auto znc = Run();
    auto ircd = ConnectIRCd();
    auto client = ConnectClient();
    client.Write("NICK foo");
    client.Write("CAP LS");
    client.ReadUntil(" sasl ");
    client.Write("CAP REQ :sasl");
    client.ReadUntil(":irc.znc.in CAP foo ACK :sasl");
    client.Write("USER bar");
    client.Write("AUTHENTICATE PLAIN");
    client.ReadUntil("AUTHENTICATE +");
    client.Write("AUTHENTICATE *");
    client.ReadUntil(":irc.znc.in 906 foo :SASL authentication aborted");
}

TEST_F(ZNCTest, SpacedServerPassword) {
    auto znc = Run();
    auto ircd = ConnectIRCd();
    auto client = LoginClient();
    client.Write(("znc delserver unix:" + m_dir.path() + "/inttest.ircd").toUtf8());
    client.Write(("znc addserver unix:" + m_dir.path() + "/inttest.ircd a b").toUtf8());
    client.Write("znc jump");
    auto ircd2 = ConnectIRCd();
    ircd2.ReadUntil("PASS :a b");
    client.Write(("znc delserver unix:" + m_dir.path() + "/inttest.ircd").toUtf8());
    client.Write(("znc addserver unix:" + m_dir.path() + "/inttest.ircd a").toUtf8());
    client.Write("znc jump");
    auto ircd3 = ConnectIRCd();
    // No :
    ircd3.ReadUntil("PASS a");
}

TEST_F(ZNCTest, TagMsg) {
    auto znc = Run();
    auto ircd = ConnectIRCd();
    auto client = LoginClient();
    ircd.Write("001 nick Welcome");

    client.Write("@foo=bar PRIVMSG #foo hi");
    ircd.ReadUntil("\nPRIVMSG");

    client.Write("TAGMSG #foo");
    ASSERT_THAT(ircd.ReadRemainder().toStdString(), Not(HasSubstr("TAGMSG")));

    ircd.Write("@foo=bar :friend PRIVMSG #foo hi");
    client.ReadUntil("\n:friend PRIVMSG");

    ircd.Write("TAGMSG #foo");
    ASSERT_THAT(client.ReadRemainder().toStdString(), Not(HasSubstr("TAGMSG")));

    client.Write("CAP REQ :message-tags");
    client.ReadUntil("ACK");

    ircd.Write("@foo TAGMSG #foo");
    client.ReadUntil("@foo TAGMSG #foo");

    ircd.Write("CAP * ACK message-tags");
    // barrier to make client wait before sending TAGMSG
    ircd.Write(":friend PRIVMSG #foo hi");
    client.ReadUntil("friend");

    client.Write("@foo=bar TAGMSG #foo");
    ircd.ReadUntil("@foo=bar TAGMSG #foo");

    client.Write("CAP REQ echo-message");
    client.Write("@baz TAGMSG #foo");
    client.ReadUntil("@baz :nick TAGMSG #foo");

    // Check buffers
    client.Write("znc addnetwork other");
    client.Write("znc jumpnetwork other");
    client.ReadUntil(":Switched to other");

    ircd.Write(":nick JOIN #bar");
    ircd.Write("@bar TAGMSG #bar");
    ASSERT_THAT(client.ReadRemainder().toStdString(), Not(HasSubstr("TAGMSG")));
    client.Write("znc jumpnetwork test");
    client.ReadUntil("@bar TAGMSG #bar");
}

TEST_F(ZNCTest, StatusAction) {
    auto znc = Run();
    auto ircd = ConnectIRCd();
    auto client = LoginClient();
    ircd.Write("001 nick Welcome");

    client.Write("PRIVMSG *status :\1ACTION waves\1");
    client.Write("PRIVMSG *status :\1VERSION\1");
    ASSERT_THAT(ircd.ReadRemainder().toStdString(), Not(HasSubstr("PRIVMSG")));
}

TEST_F(ZNCTest, InviteNotify) {
    auto znc = Run();
    auto ircd = ConnectIRCd();
    auto client = LoginClient();
    client.Write("CAP REQ invite-notify");
    client.ReadUntil("ACK");
    auto client2 = LoginClient();

    InstallModule("testmod.cpp", R"(
        #include <znc/Modules.h>
        class TestModule : public CModule {
          public:
            MODCONSTRUCTOR(TestModule) {}
            EModRet OnInviteMessage(CInviteMessage& Msg) override {
                PutIRC("__MSG__ " + Msg.ToString());
                return CONTINUE;
            }
            EModRet OnInvite(const CNick& Nick, const CString& sChan) override {
                PutIRC("__INV__ " + Nick.GetNickMask() + " " + sChan);
                return CONTINUE;
            }
        };
        GLOBALMODULEDEFS(TestModule, "Test")
    )");
    client.Write("znc loadmod testmod");
    client.ReadUntil("Loaded module testmod");

    ircd.Write("001 nick Welcome");

    ircd.Write(":source!id@ho INVITE nick #chan");
    client.ReadUntil(":source!id@ho INVITE nick #chan");
    client2.ReadUntil(":source!id@ho INVITE nick #chan");
    ircd.ReadUntil("__MSG__ :source!id@ho INVITE nick #chan");
    ircd.ReadUntil("__INV__ source!id@ho #chan");

    ircd.Write(":source!id@ho INVITE someone #chan");
    client.ReadUntil(":source!id@ho INVITE someone #chan");
    ASSERT_THAT(client2.ReadRemainder().toStdString(), Not(HasSubstr("someone")));
    ircd.ReadUntil("__MSG__ :source!id@ho INVITE someone #chan");
    ASSERT_THAT(ircd.ReadRemainder().toStdString(), Not(HasSubstr("__INV__")));
}

TEST_F(ZNCTest, DisableCap) {
    {
        QFile conf(m_dir.path() + "/configs/znc.conf");
        ASSERT_TRUE(conf.open(QIODevice::Append | QIODevice::Text));
        QTextStream out(&conf);
        out << R"(
            DisableClientCap = sasl
            DisableClientCap = away-notify
            DisableServerCap = chghost
        )";
    }
    auto znc = Run();
    auto ircd = ConnectIRCd();

    ircd.Write("CAP user LS :chghost");
    ASSERT_THAT(ircd.ReadRemainder().toStdString(), Not(HasSubstr("chghost")));

    auto client = ConnectClient();
    client.Write("NICK foo");
    client.Write("CAP LS 302");
    ASSERT_THAT(client.ReadRemainder().toStdString(), Not(HasSubstr("sasl")));
    client.Write("CAP REQ sasl");
    client.ReadUntil("CAP foo NAK :sasl");

    client.Write("PASS :hunter2");
    client.Write("USER user/test x x :x");
    client.Write("CAP END");
    client.ReadUntil("001");

    // Server-dependent
    ircd.Write("001 nick Welcome");
    ircd.Write("CAP nick NEW away-notify");
    ircd.ReadUntil("CAP REQ :away-notify");
    ircd.Write("CAP nick ACK away-notify");
    ASSERT_THAT(client.ReadRemainder().toStdString(),
                Not(AllOf(HasSubstr("NEW"), HasSubstr("away-notify"))));
}

TEST_F(ZNCTest, ManyCapsInReq) {
    constexpr int prefix = std::string_view("CAP NAK :").length();
    auto znc = Run();
    auto ircd = ConnectIRCd();
    ircd.Write(
        "CAP user LS :away-notify invite-notify extended-join "
        "userhost-in-names multi-prefix cap-notify "
        "sasl=ANONYMOUS,EXTERNAL,PLAIN setname tls chghost account-notify "
        "message-tags batch server-time account-tag echo-message "
        "labeled-response");
    QByteArray caps, caps2, caps3;
    ircd.ReadUntilAndGet("CAP REQ :", caps);
    caps.remove(0, prefix);
    EXPECT_THAT(caps.toStdString(), SizeIs(AllOf(Gt(10), Lt(100))));
    ircd.Write("CAP user NAK :" + caps);
    ircd.ReadUntilAndGet("CAP REQ :", caps2);
    caps2.remove(0, prefix);
    EXPECT_THAT(caps2.toStdString(), SizeIs(AllOf(Gt(10), Lt(100))));
    ircd.Write("CAP user ACK :" + caps2);
    // Now it should retry one of the first batch
    ircd.ReadUntilAndGet("CAP REQ :", caps3);
    caps3.remove(0, prefix);
    EXPECT_THAT(caps3.toStdString(), Not(HasSubstr(" ")));
    EXPECT_THAT(caps.split(' '), Contains(caps3));
}

TEST_F(ZNCTest, JoinWhileRegistration) {
    auto znc = Run();
    auto ircd = ConnectIRCd();
    auto client = LoginClient();
    // First JOIN just adds channel to the list, second one updates the key and
    // sends JOIN from CChan::JoinUser. Both should be delayed until
    // registration ends.
    client.Write("JOIN #foo");
    client.Write("JOIN #foo");
    EXPECT_THAT(ircd.ReadRemainder().toStdString(), Not(HasSubstr("JOIN")));
    ircd.Write(":server 001 nick :Hello");
    ircd.ReadUntil("JOIN #foo");
}

TEST_F(ZNCTest, Issue1960) {
    auto znc = Run();
    auto ircd1 = ConnectIRCd();
    auto client = LoginClient();
    ircd1.Write("CAP user ACK :message-tags");
    ircd1.Write(":server 001 nick :Hello");
    ircd1.Write(":server 005 nick blahblah");
    client.ReadUntil("blahblah");
    client.Write("znc addnetwork second");
    client.Write("znc jumpnetwork second");
    client.Write("znc addserver unix:" + m_dir.path().toUtf8() + "/inttest.ircd");
    auto ircd2 = ConnectIRCd();
    ircd2.Write("CAP user ACK :message-tags");
    ircd2.Write(":server 001 nick :Hello");
    client.ReadUntil("Connected");
    client.Write("@foo TAGMSG #bar");
    ircd2.ReadUntil("@foo TAGMSG #bar");
}

TEST_F(ZNCTest, DisconnectedTagmsgCrash) {
    auto znc = Run();
    auto ircd = ConnectIRCd();
    auto client = LoginClient();
    client.Write("znc disconnect");
    client.ReadUntil("Disconnected");
    client.Write("@foo TAGMSG #foo");
    client.Write("znc help");
    client.ReadUntil("AddServer");
}

}  // namespace
}  // namespace znc_inttest