// 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 <unistd.h>

#include <new>
#include <stdbool.h>
#include <string.h>

#include <ddk/device.h>
#include <ddk/protocol/block.h>
#include <fbl/algorithm.h>
#include <fbl/alloc_checker.h>
#include <fbl/auto_call.h>
#include <fbl/auto_lock.h>
#include <fbl/ref_ptr.h>
#include <lib/zx/fifo.h>
#include <zircon/compiler.h>
#include <zircon/device/block.h>
#include <zircon/syscalls.h>

#include "server.h"

TransactionGroup::TransactionGroup() :
    fifo_(ZX_HANDLE_INVALID), flags_(0), ctr_(0) {
    memset(&response_, 0, sizeof(response_));
}

TransactionGroup::~TransactionGroup() {}

void TransactionGroup::Initialize(zx_handle_t fifo, groupid_t group) {
    ZX_DEBUG_ASSERT(fifo_ == ZX_HANDLE_INVALID);
    fifo_ = fifo;
    response_.group = group;
}

zx_status_t TransactionGroup::Enqueue(bool do_respond, reqid_t reqid) {
    zx_status_t status = ZX_OK;
    fbl::AutoLock lock(&lock_);
    if (flags_ & kTxnFlagRespond) {
        // Shouldn't get more than one response for a txn.
        response_.status = ZX_ERR_IO;
        status = ZX_ERR_IO;
    } else if (response_.status != ZX_OK) {
        // This operation already failed; don't bother processing it.
        status = ZX_ERR_IO;
    }
    ctr_++;
    if (do_respond) {
        response_.reqid = reqid;
        flags_ |= kTxnFlagRespond;
    }
    return status;
}

void TransactionGroup::CtrAdd(uint32_t n) {
    fbl::AutoLock lock(&lock_);
    ctr_ += n;
}

void TransactionGroup::Complete(zx_status_t status) {
    fbl::AutoLock lock(&lock_);
    if ((status != ZX_OK) && (response_.status == ZX_OK)) {
        response_.status = status;
    }

    response_.count++;
    ZX_DEBUG_ASSERT(ctr_ != 0);
    ZX_DEBUG_ASSERT(response_.count <= ctr_);

    if ((flags_ & kTxnFlagRespond) && (response_.count == ctr_)) {
        status = zx_fifo_write(fifo_, sizeof(response_), &response_, 1, nullptr);
        if (status != ZX_OK) {
            fprintf(stderr, "Block Server I/O error: Could not write response\n");
        }
        response_.count = 0;
        response_.status = ZX_OK;
        response_.reqid = 0;
        ctr_ = 0;
        flags_ &= ~kTxnFlagRespond;
    }
}