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

#pragma once

#include <threads.h>

#include <hypervisor/virtio.h>
#include <zircon/compiler.h>
#include <zircon/types.h>
#include <virtio/balloon.h>

#define VIRTIO_BALLOON_Q_INFLATEQ 0
#define VIRTIO_BALLOON_Q_DEFLATEQ 1
#define VIRTIO_BALLOON_Q_STATSQ 2
#define VIRTIO_BALLOON_Q_COUNT 3

/* Per Virtio 1.0 Section 5.5.6, This value is historical, and independent of
 * the guest page size.
 */
#define VIRTIO_BALLOON_PAGE_SIZE 4096

__BEGIN_CDECLS

/* Virtio memory balloon device. */
typedef struct balloon {
    mtx_t mutex;
    // Handle to the guest phsycial memory VMO for memory management.
    zx_handle_t vmo;
    // With on-demand deflation we won't commit memory up-front for balloon
    // deflate requests.
    bool deflate_on_demand;

    struct {
        // The index in the available ring of the stats descriptor.
        uint16_t desc_index;
        // Indicates if desc_index valid.
        bool has_buffer;
        // Holds exclusive access to the stats queue. At most one stats request
        // can be active at a time (by design). Specifically we need to hold
        // exclusive access of the queue from the time a buffer is returned to
        // the queue, initiating a stats request, until any logic processing
        // the result has finished.
        //
        // Also guards access to other members of this structure.
        mtx_t mutex;
    } stats;

    virtio_device_t virtio_device;
    virtio_queue_t queues[VIRTIO_BALLOON_Q_COUNT];
    virtio_balloon_config_t config;
} balloon_t;

void balloon_init(balloon_t* balloon, uintptr_t guest_physmem_addr, size_t guest_physmem_size,
                  zx_handle_t guest_physmem_vmo);

/* Callback for balloon_request_stats. */
typedef void (*balloon_stats_fn_t)(const virtio_balloon_stat_t* stats, size_t len, void* ctx);

/* Request balloon memory statistics from the guest.
 *
 * The callback will be executed syncronously with this thread once stats have
 * been received from the guest. Pointers to stats must not be held after the
 * callback returns.
 */
zx_status_t balloon_request_stats(balloon_t* balloon, balloon_stats_fn_t handler, void* ctx);

/* Update the 'num_pages' configuration field in the balloon. */
zx_status_t balloon_update_num_pages(balloon_t* balloon, uint32_t num_pages);

__END_CDECLS