// Copyright 2015 The Android Open Source Project
//
// 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 "android/crashreport/CrashReporter.h"

#include "android/base/memory/LazyInstance.h"
#include "android/base/system/System.h"
#include "android/utils/debug.h"
#include "client/mac/handler/exception_handler.h"

#include <mach/mach.h>

#include <memory>

#include <inttypes.h>
#include <stdint.h>

#define E(...) derror(__VA_ARGS__)
#define W(...) dwarning(__VA_ARGS__)
#define D(...) VERBOSE_PRINT(init, __VA_ARGS__)
#define I(...) printf(__VA_ARGS__)

namespace android {
namespace crashreport {

namespace {

class HostCrashReporter : public CrashReporter {
public:
    HostCrashReporter() : CrashReporter(), mHandler() {}

    virtual ~HostCrashReporter() {}

    bool attachCrashHandler(const CrashSystem::CrashPipe& crashpipe) override {
        if (mHandler) {
            return false;
        }

        mHandler.reset(new google_breakpad::ExceptionHandler(
                getDumpDir(), &HostCrashReporter::exceptionFilterCallback,
                nullptr,  // no minidump callback
                nullptr,  // no callback context
                true,     // install signal handlers
                crashpipe.mClient.c_str()));

        return mHandler != nullptr;
    }

    bool waitServicePipeReady(const std::string& pipename,
                              int timeout_ms) override {
        static_assert(kWaitIntervalMS > 0, "kWaitIntervalMS must be greater than 0");
        mach_port_t task_bootstrap_port = 0;
        mach_port_t port;
        task_get_bootstrap_port(mach_task_self(), &task_bootstrap_port);
        for (; timeout_ms > 0; timeout_ms -= kWaitIntervalMS) {
            if (bootstrap_look_up(task_bootstrap_port, pipename.c_str(),
                                  &port) == KERN_SUCCESS) {
                return true;
            }
            ::android::base::System::get()->sleepMs(kWaitIntervalMS);
        }
        return false;
    }

    void setupChildCrashProcess(int pid) override {}

    void writeDump() override { mHandler->WriteMinidump(); }

   static bool exceptionFilterCallback(void* context);

private:
    bool onCrashPlatformSpecific() override;

    std::unique_ptr<google_breakpad::ExceptionHandler> mHandler;
};

::android::base::LazyInstance<HostCrashReporter> sCrashReporter =
        LAZY_INSTANCE_INIT;

bool HostCrashReporter::exceptionFilterCallback(void*) {
    return CrashReporter::get()->onCrash();
}

static void attachMemoryInfo()
{
    rusage usage = {};
    getrusage(RUSAGE_SELF, &usage);

    mach_task_basic_info info = {};
    mach_msg_type_number_t infoCount = MACH_TASK_BASIC_INFO_COUNT;
    task_info(mach_task_self(), MACH_TASK_BASIC_INFO,
            reinterpret_cast<task_info_t>(&info), &infoCount);

    char buf[1024] = {};
    snprintf(buf, sizeof(buf) - 1,
            "==== Process memory usage ====\n"
            "max resident size = %" PRIu64 " kB\n"
            "virtual size = %" PRIu64 " kB\n"
            "resident size = %" PRIu64 " kB\n"
            "messages sent = %" PRIu64 "\n"
            "messages received = %" PRIu64 "\n"
            "voluntary context switches = %" PRIu64 "\n"
            "involuntary context switches = %" PRIu64 "\n",
             uint64_t(info.resident_size_max / 1024),
             uint64_t(info.virtual_size / 1024),
             uint64_t(info.resident_size / 1024),
             uint64_t(usage.ru_msgsnd),
             uint64_t(usage.ru_msgrcv),
             uint64_t(usage.ru_nvcsw),
             uint64_t(usage.ru_nivcsw));

    CrashReporter::get()->attachData(
                CrashReporter::kProcessMemoryInfoFileName, buf);
}

bool HostCrashReporter::onCrashPlatformSpecific() {
    attachMemoryInfo();
    return true;    // proceed with handling the crash
}

}  // namespace anonymous

CrashReporter* CrashReporter::get() {
    return sCrashReporter.ptr();
}

}  // namespace crashreport
}  // namespace android