// Copyright 2017 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/async/cpp/wait.h>

namespace async {

WaitBase::WaitBase(zx_handle_t object, zx_signals_t trigger, async_wait_handler_t* handler)
    : wait_{{ASYNC_STATE_INIT}, handler, object, trigger} {}

WaitBase::~WaitBase() {
    if (async_) {
        // Failure to cancel here may result in a dangling pointer...
        zx_status_t status = async_cancel_wait(async_, &wait_);
        ZX_ASSERT_MSG(status == ZX_OK, "status=%d", status);
    }
}

zx_status_t WaitBase::Begin(async_t* async) {
    if (async_)
        return ZX_ERR_ALREADY_EXISTS;

    async_ = async;
    zx_status_t status = async_begin_wait(async, &wait_);
    if (status != ZX_OK) {
        async_ = nullptr;
    }
    return status;
}

zx_status_t WaitBase::Cancel() {
    if (!async_)
        return ZX_ERR_NOT_FOUND;

    async_t* async = async_;
    async_ = nullptr;
    return async_cancel_wait(async, &wait_);
}

Wait::Wait(zx_handle_t object, zx_signals_t trigger, Handler handler)
    : WaitBase(object, trigger, &Wait::CallHandler), handler_(fbl::move(handler)) {}

Wait::~Wait() = default;

void Wait::CallHandler(async_t* async, async_wait_t* wait,
                       zx_status_t status, const zx_packet_signal_t* signal) {
    auto self = Dispatch<Wait>(wait);
    self->handler_(async, self, status, signal);
}

} // namespace async