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

#include <block-client/cpp/client.h>
#include <fbl/unique_fd.h>
#include <lib/zx/fifo.h>
#include <zircon/status.h>

#include "pave-utils.h"
#include "pave-logging.h"

zx_status_t FlushClient(const block_client::Client& client) {
    block_fifo_request_t request;
    request.group = 0;
    request.vmoid = VMOID_INVALID;
    request.opcode = BLOCKIO_FLUSH;
    request.length = 0;
    request.vmo_offset = 0;
    request.dev_offset = 0;

    zx_status_t status = client.Transaction(&request, 1);
    if (status != ZX_OK) {
        ERROR("Error flushing: %s\n", zx_status_get_string(status));
        return status;
    }
    return ZX_OK;
}

zx_status_t FlushBlockDevice(const fbl::unique_fd& fd) {
    zx::fifo fifo;
    ssize_t result = ioctl_block_get_fifos(fd.get(), fifo.reset_and_get_address());
    if (result < 0) {
        ERROR("Couldn't attach fifo to partition\n");
        return static_cast<zx_status_t>(result);
    }

    block_client::Client client;
    zx_status_t status = block_client::Client::Create(fbl::move(fifo), &client);
    if (status != ZX_OK) {
        ERROR("Couldn't create block client\n");
        return status;
    }

    return FlushClient(client);
}