#pragma once

#include "nyx/snapshot/devices/state_reallocation.h"
#include <stdbool.h>
#include <stdint.h>

typedef struct nyx_device_state_s {
    state_reallocation_t *qemu_state;

    uint64_t tsc_value;
    uint64_t tsc_value_incremental;

    bool incremental_mode;

    void    *state_buf; /* QEMU's serialized state */
    uint32_t state_buf_size;

} nyx_device_state_t;


nyx_device_state_t *nyx_device_state_init(void);
nyx_device_state_t *nyx_device_state_init_from_snapshot(const char *snapshot_folder,
                                                        bool        pre_snapshot);

void nyx_device_state_restore(nyx_device_state_t *self);
void nyx_device_state_post_restore(nyx_device_state_t *self);

void nyx_device_state_switch_incremental(nyx_device_state_t *self);
void nyx_device_state_disable_incremental(nyx_device_state_t *self);

void nyx_device_state_save_tsc(nyx_device_state_t *self);
void nyx_device_state_save_tsc_incremental(nyx_device_state_t *self);

void nyx_device_state_serialize(nyx_device_state_t *self, const char *snapshot_folder);