// Copyright 2018 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <lib/logger/logger.h>

#include <fuchsia/logger/c/fidl.h>
#include <lib/async-loop/cpp/loop.h>
#include <lib/syslog/global.h>
#include <unittest/unittest.h>

#include <fcntl.h>

#include <utility>

__BEGIN_CDECLS

// This does not come from header file as this function should only be used in
// tests and is not for general use.
void fx_log_reset_global(void);

__END_CDECLS
namespace {

bool ends_with(const char* str, const char* suffix) {
    size_t str_len = strlen(str);
    size_t suffix_len = strlen(suffix);
    if (str_len < suffix_len) {
        return false;
    }
    str += str_len - suffix_len;
    return strcmp(str, suffix) == 0;
}

class Fixture {
public:
    Fixture()
        : fds_valid_(false), loop_(&kAsyncLoopConfigNoAttachToThread), error_status_(0) {}
    ~Fixture() {
        fx_log_reset_global();
        if (fds_valid_) {
            close(pipefd_[0]);
            close(pipefd_[1]);
        }
    }

    zx_status_t error_status() const {
        return error_status_;
    }

    bool CreateLogger() {
        ASSERT_NE(pipe2(pipefd_, O_NONBLOCK), -1, "");
        zx::channel local, remote;
        ASSERT_EQ(ZX_OK, zx::channel::create(0, &local, &remote));
        logger_ = fbl::make_unique<logger::LoggerImpl>(std::move(remote), pipefd_[0]);
        ASSERT_EQ(ZX_OK, logger_->Begin(loop_.dispatcher()));
        logger_handle_.reset(local.release());
        logger_->set_error_handler([this](zx_status_t status) {
            error_status_ = status;
        });
        return true;
    }

    void ResetLoggerHandle() {
        logger_handle_.reset();
    }

    void ResetSocket() {
        socket_.reset();
    }

    bool ConnectToLogger() {
        ASSERT_TRUE(logger_handle_);
        zx::socket local, remote;
        ASSERT_EQ(ZX_OK, zx::socket::create(ZX_SOCKET_DATAGRAM, &local, &remote));
        fuchsia_logger_LogSinkConnectRequest req;
        memset(&req, 0, sizeof(req));
        req.hdr.ordinal = fuchsia_logger_LogSinkConnectOrdinal;
        req.socket = FIDL_HANDLE_PRESENT;
        zx_handle_t handles[1] = {remote.release()};
        ASSERT_EQ(ZX_OK, logger_handle_.write(0, &req, sizeof(req), handles, 1));
        loop_.RunUntilIdle();
        socket_.reset(local.release());
        return true;
    }

    bool InitSyslog(const char** tags, size_t ntags) {
        ASSERT_TRUE(socket_);
        fx_logger_config_t config = {.min_severity = FX_LOG_INFO,
                                     .console_fd = -1,
                                     .log_service_channel = socket_.release(),
                                     .tags = tags,
                                     .num_tags = ntags};

        ASSERT_EQ(ZX_OK, fx_log_init_with_config(&config));
        return true;
    }

    bool FullSetup() {
        ASSERT_TRUE(CreateLogger());
        ASSERT_TRUE(ConnectToLogger());
        ASSERT_TRUE(InitSyslog(nullptr, 0));
        return true;
    }

    void RunLoop() {
        loop_.RunUntilIdle();
        fsync(pipefd_[0]);
    }

    const char* read_buffer() {
        memset(buf_, 0, sizeof(buf_));
        read(pipefd_[1], &buf_, sizeof(buf_));
        return buf_;
    }

private:
    bool fds_valid_;
    char buf_[4096]; // should be enough for logs
    async::Loop loop_;
    zx_status_t error_status_;
    fbl::unique_ptr<logger::LoggerImpl> logger_;
    zx::channel logger_handle_;
    zx::socket socket_;
    int pipefd_[2];
};

bool TestLogSimple(void) {
    BEGIN_TEST;
    Fixture fixture;
    ASSERT_TRUE(fixture.FullSetup());
    FX_LOG(INFO, nullptr, "test_message");
    fixture.RunLoop();
    const char* out = fixture.read_buffer();
    ASSERT_TRUE(ends_with(out, "test_message\n"), out);
    END_TEST;
}

bool TestLogMultipleMsgs(void) {
    BEGIN_TEST;
    Fixture fixture;
    ASSERT_TRUE(fixture.FullSetup());
    FX_LOG(INFO, nullptr, "test_message");
    fixture.RunLoop();
    const char* out = fixture.read_buffer();
    ASSERT_TRUE(ends_with(out, "INFO: test_message\n"), out);
    FX_LOG(INFO, nullptr, "test_message2");
    fixture.RunLoop();
    out = fixture.read_buffer();
    ASSERT_TRUE(ends_with(out, "INFO: test_message2\n"), out);
    END_TEST;
}

bool TestLogWithTag(void) {
    BEGIN_TEST;
    Fixture fixture;
    ASSERT_TRUE(fixture.FullSetup());
    FX_LOG(INFO, "tag", "test_message");
    fixture.RunLoop();
    const char* out = fixture.read_buffer();
    ASSERT_TRUE(ends_with(out, "[tag] INFO: test_message\n"), out);
    END_TEST;
}

bool TestLogWithMultipleTags(void) {
    BEGIN_TEST;
    Fixture fixture;
    ASSERT_TRUE(fixture.CreateLogger());
    ASSERT_TRUE(fixture.ConnectToLogger());
    const char* gtags[] = {"gtag1", "gtag2"};
    ASSERT_TRUE(fixture.InitSyslog(gtags, 2));
    FX_LOG(INFO, "tag", "test_message");
    fixture.RunLoop();
    const char* out = fixture.read_buffer();
    ASSERT_TRUE(ends_with(out, "[gtag1, gtag2, tag] INFO: test_message\n"), out);
    END_TEST;
}

bool TestLogSeverity(void) {
    BEGIN_TEST;
    Fixture fixture;
    ASSERT_TRUE(fixture.FullSetup());
    FX_LOG(INFO, "", "test_message");
    fixture.RunLoop();
    const char* out = fixture.read_buffer();
    ASSERT_TRUE(ends_with(out, "[] INFO: test_message\n"), out);

    FX_LOG(WARNING, "", "test_message");
    fixture.RunLoop();
    out = fixture.read_buffer();
    ASSERT_TRUE(ends_with(out, "[] WARNING: test_message\n"), out);

    FX_LOG(ERROR, "", "test_message");
    fixture.RunLoop();
    out = fixture.read_buffer();
    ASSERT_TRUE(ends_with(out, "[] ERROR: test_message\n"), out);

    END_TEST;
}

bool TestLogWhenLoggerHandleDies(void) {
    BEGIN_TEST;
    Fixture fixture;
    ASSERT_TRUE(fixture.FullSetup());
    fixture.ResetLoggerHandle();
    fixture.RunLoop();
    FX_LOG(INFO, "tag", "test_message");
    fixture.RunLoop();
    const char* out = fixture.read_buffer();
    ASSERT_TRUE(ends_with(out, "[tag] INFO: test_message\n"), out);
    ASSERT_EQ(ZX_OK, fixture.error_status());
    END_TEST;
}

bool TestLoggerDiesWithSocket(void) {
    BEGIN_TEST;
    Fixture fixture;
    ASSERT_TRUE(fixture.CreateLogger());
    ASSERT_TRUE(fixture.ConnectToLogger());
    fixture.ResetSocket();
    fixture.RunLoop();
    ASSERT_EQ(ZX_ERR_PEER_CLOSED, fixture.error_status());
    END_TEST;
}

bool TestLoggerDiesWithChannelWhenNoConnectCalled(void) {
    BEGIN_TEST;
    Fixture fixture;
    ASSERT_TRUE(fixture.CreateLogger());
    fixture.RunLoop();
    ASSERT_EQ(ZX_OK, fixture.error_status());
    fixture.ResetLoggerHandle();
    fixture.RunLoop();
    ASSERT_EQ(ZX_ERR_PEER_CLOSED, fixture.error_status());
    END_TEST;
}

} // namespace

BEGIN_TEST_CASE(logger_tests)
RUN_TEST(TestLogSimple)
RUN_TEST(TestLogSeverity)
RUN_TEST(TestLogMultipleMsgs)
RUN_TEST(TestLogWithTag)
RUN_TEST(TestLogWithMultipleTags)
RUN_TEST(TestLogWhenLoggerHandleDies)
RUN_TEST(TestLoggerDiesWithSocket)
RUN_TEST(TestLoggerDiesWithChannelWhenNoConnectCalled)
END_TEST_CASE(logger_tests)

int main(int argc, char** argv) {
    return unittest_run_all_tests(argc, argv) ? 0 : -1;
}