// Copyright 2016 The Fuchsia Authors
// Copyright (c) 2013 Google Inc.
// Copyright (c) 2015 Travis Geiselbrecht
//
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file or at
// https://opensource.org/licenses/MIT

#include <lib/watchdog.h>

#include <assert.h>
#include <err.h>
#include <inttypes.h>
#include <kernel/auto_lock.h>
#include <kernel/spinlock.h>
#include <kernel/thread.h>
#include <kernel/timer.h>
#include <platform.h>
#include <zircon/compiler.h>
#include <zircon/time.h>
#include <zircon/types.h>

static SpinLock lock;

__WEAK void watchdog_handler(watchdog_t* dog) {
    dprintf(INFO, "Watchdog \"%s\" (timeout %" PRId64 " mSec) just fired!!\n",
            dog->name, dog->timeout / (1000 * 1000));
    platform_halt(HALT_ACTION_HALT, HALT_REASON_SW_RESET);
}

static void watchdog_timer_callback(timer_t* timer, zx_time_t now, void* arg) {
    watchdog_handler((watchdog_t*)arg);

    /* We should never get here; watchdog handlers should always be fatal. */
    DEBUG_ASSERT(false);
}

zx_status_t watchdog_init(watchdog_t* dog, zx_duration_t timeout, const char* name) {
    DEBUG_ASSERT(NULL != dog);
    DEBUG_ASSERT(ZX_TIME_INFINITE != timeout);

    dog->magic = WATCHDOG_MAGIC;
    dog->name = name ? name : "unnamed watchdog";
    dog->enabled = false;
    dog->timeout = timeout;
    timer_init(&dog->expire_timer);

    return ZX_OK;
}

void watchdog_set_enabled(watchdog_t* dog, bool enabled) {
    AutoSpinLock guard(&lock);

    DEBUG_ASSERT((NULL != dog) && (WATCHDOG_MAGIC == dog->magic));

    if (dog->enabled == enabled) {
        return;
    }

    dog->enabled = enabled;
    zx_time_t deadline = zx_time_add_duration(current_time(), dog->timeout);
    if (enabled) {
        timer_set_oneshot(&dog->expire_timer, deadline, watchdog_timer_callback, dog);
    } else {
        timer_cancel(&dog->expire_timer);
    }
}

void watchdog_pet(watchdog_t* dog) {
    AutoSpinLock guard(&lock);

    DEBUG_ASSERT((NULL != dog) && (WATCHDOG_MAGIC == dog->magic));

    if (!dog->enabled) {
        return;
    }

    timer_cancel(&dog->expire_timer);
    zx_time_t deadline = zx_time_add_duration(current_time(), dog->timeout);
    timer_set_oneshot(&dog->expire_timer, deadline, watchdog_timer_callback, dog);
}

static timer_t hw_watchdog_timer;
static bool hw_watchdog_enabled;
static zx_duration_t hw_watchdog_pet_timeout;

static void hw_watchdog_timer_callback(timer_t* timer, zx_time_t now, void* arg) {
    timer_set_oneshot(timer,
                      zx_time_add_duration(now, hw_watchdog_pet_timeout),
                      hw_watchdog_timer_callback, NULL);
    platform_watchdog_pet();
}

zx_status_t watchdog_hw_init(zx_time_t timeout) {
    DEBUG_ASSERT(ZX_TIME_INFINITE != timeout);
    timer_init(&hw_watchdog_timer);
    return platform_watchdog_init(timeout, &hw_watchdog_pet_timeout);
}

void watchdog_hw_set_enabled(bool enabled) {
    AutoSpinLock guard(&lock);

    if (hw_watchdog_enabled == enabled) {
        return;
    }

    hw_watchdog_enabled = enabled;
    platform_watchdog_set_enabled(enabled);
    if (enabled) {
        timer_set_oneshot(&hw_watchdog_timer,
                          zx_time_add_duration(current_time(), hw_watchdog_pet_timeout),
                          hw_watchdog_timer_callback, NULL);
    } else {
        timer_cancel(&hw_watchdog_timer);
    }
}