113 lines
2.6 KiB
C
113 lines
2.6 KiB
C
/*
|
|
|
|
Copyright (C) 2017 Sergej Schumilo
|
|
|
|
This file is part of QEMU-PT (HyperTrash / kAFL).
|
|
|
|
QEMU-PT is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 2 of the License, or
|
|
(at your option) any later version.
|
|
|
|
QEMU-PT is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with QEMU-PT. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
*/
|
|
|
|
#ifndef STATE_REALLOCATION
|
|
#define STATE_REALLOCATION
|
|
|
|
#include "qemu/osdep.h"
|
|
#include "monitor/monitor.h"
|
|
//#include "qemu-common.h"
|
|
#include "migration/migration.h"
|
|
#include "nyx/khash.h"
|
|
|
|
|
|
|
|
#define IO_BUF_SIZE 32768
|
|
|
|
struct QEMUFile_tmp {
|
|
void *ops;
|
|
void *hooks;
|
|
void *opaque;
|
|
|
|
int64_t bytes_xfer;
|
|
int64_t xfer_limit;
|
|
|
|
int64_t pos; /* start of buffer when writing, end of buffer
|
|
when reading */
|
|
volatile int buf_index;
|
|
int buf_size; /* 0 when writing */
|
|
uint8_t buf[IO_BUF_SIZE];
|
|
};
|
|
|
|
struct fast_savevm_opaque_t{
|
|
FILE* f;
|
|
uint8_t* buf;
|
|
size_t buflen;
|
|
uint64_t pos;
|
|
void* output_buffer;
|
|
uint32_t* output_buffer_size;
|
|
};
|
|
|
|
#define REALLOC_SIZE 0x8000
|
|
|
|
#define PRE_ALLOC_BLOCK_SIZE 0x8000000 /* 128 MB */
|
|
|
|
typedef struct state_reallocation_tmp_s{
|
|
void **copy;
|
|
uint32_t fast_state_size;
|
|
bool enabled;
|
|
} state_reallocation_tmp_t;
|
|
|
|
typedef struct state_reallocation_s{
|
|
void **ptr;
|
|
void **copy;
|
|
size_t *size;
|
|
|
|
uint32_t fast_state_size;
|
|
uint32_t fast_state_pos;
|
|
|
|
|
|
void **fptr;
|
|
void **opaque;
|
|
uint32_t *version;
|
|
|
|
uint32_t fast_state_fptr_size;
|
|
uint32_t fast_state_fptr_pos;
|
|
|
|
|
|
void **get_fptr;
|
|
void **get_opaque;
|
|
size_t *get_size;
|
|
void **get_data;
|
|
|
|
//QEMUFile** file;
|
|
|
|
uint32_t fast_state_get_fptr_size;
|
|
uint32_t fast_state_get_fptr_pos;
|
|
|
|
/* prevents heap fragmentation and additional 2GB mem usage */
|
|
void* pre_alloc_block;
|
|
uint32_t pre_alloc_block_offset;
|
|
|
|
state_reallocation_tmp_t tmp_snapshot;
|
|
|
|
} state_reallocation_t;
|
|
|
|
state_reallocation_t* state_reallocation_new(QEMUFile *f);
|
|
|
|
//void fdl_enumerate_global_states(QEMUFile *f);
|
|
void fdl_fast_reload(state_reallocation_t* self);
|
|
|
|
void fdl_fast_create_tmp(state_reallocation_t* self);
|
|
void fdl_fast_enable_tmp(state_reallocation_t* self);
|
|
void fdl_fast_disable_tmp(state_reallocation_t* self);
|
|
|
|
#endif |