replay/replay-events.c

   1 /*
   2  * replay-events.c
   3  *
   4  * Copyright (c) 2010-2015 Institute for System Programming
   5  *                         of the Russian Academy of Sciences.
   6  *
   7  * This work is licensed under the terms of the GNU GPL, version 2 or later.
   8  * See the COPYING file in the top-level directory.
   9  *
  10  */
  11
  12 #include "qemu/osdep.h"
  13 #include "qemu-common.h"
  14 #include "qemu/error-report.h"
  15 #include "sysemu/replay.h"
  16 #include "replay-internal.h"
  17 #include "block/aio.h"
  18 #include "ui/input.h"
  19
  20 typedef struct Event {
  21     ReplayAsyncEventKind event_kind;
  22     void *opaque;
  23     void *opaque2;
  24     uint64_t id;
  25
  26     QTAILQ_ENTRY(Event) events;
  27 } Event;
  28
  29 static QTAILQ_HEAD(, Event) events_list = QTAILQ_HEAD_INITIALIZER(events_list);
  30 static unsigned int read_event_kind = -1;
  31 static uint64_t read_id = -1;
  32 static int read_checkpoint = -1;
  33
  34 static bool events_enabled;
  35
  36 /* Functions */
  37
  38 static void replay_run_event(Event *event)
  39 {
  40     switch (event->event_kind) {
  41     case REPLAY_ASYNC_EVENT_BH:
  42         aio_bh_call(event->opaque);
  43         break;
  44     case REPLAY_ASYNC_EVENT_INPUT:
  45         qemu_input_event_send_impl(NULL, (InputEvent *)event->opaque);
  46         qapi_free_InputEvent((InputEvent *)event->opaque);
  47         break;
  48     case REPLAY_ASYNC_EVENT_INPUT_SYNC:
  49         qemu_input_event_sync_impl();
  50         break;
  51     case REPLAY_ASYNC_EVENT_CHAR_READ:
  52         replay_event_char_read_run(event->opaque);
  53         break;
  54     case REPLAY_ASYNC_EVENT_BLOCK:
  55         aio_bh_call(event->opaque);
  56         break;
  57     case REPLAY_ASYNC_EVENT_NET:
  58         replay_event_net_run(event->opaque);
  59         break;
  60     default:
  61         error_report("Replay: invalid async event ID (%d) in the queue",
  62                     event->event_kind);
  63         exit(1);
  64         break;
  65     }
  66 }
  67
  68 void replay_enable_events(void)
  69 {
  70     if (replay_mode != REPLAY_MODE_NONE) {
  71         events_enabled = true;
  72     }
  73 }
  74
  75 bool replay_has_events(void)
  76 {
  77     return !QTAILQ_EMPTY(&events_list);
  78 }
  79
  80 void replay_flush_events(void)
  81 {
  82     g_assert(replay_mutex_locked());
  83
  84     while (!QTAILQ_EMPTY(&events_list)) {
  85         Event *event = QTAILQ_FIRST(&events_list);
  86         replay_run_event(event);
  87         QTAILQ_REMOVE(&events_list, event, events);
  88         g_free(event);
  89     }
  90 }
  91
  92 void replay_disable_events(void)
  93 {
  94     if (replay_mode != REPLAY_MODE_NONE) {
  95         events_enabled = false;
  96         /* Flush events queue before waiting of completion */
  97         replay_flush_events();
  98     }
  99 }
 100
 101 void replay_clear_events(void)
 102 {
 103     g_assert(replay_mutex_locked());
 104
 105     while (!QTAILQ_EMPTY(&events_list)) {
 106         Event *event = QTAILQ_FIRST(&events_list);
 107         QTAILQ_REMOVE(&events_list, event, events);
 108
 109         g_free(event);
 110     }
 111 }
 112
 113 /*! Adds specified async event to the queue */
 114 void replay_add_event(ReplayAsyncEventKind event_kind,
 115                       void *opaque,
 116                       void *opaque2, uint64_t id)
 117 {
 118     assert(event_kind < REPLAY_ASYNC_COUNT);
 119
 120     if (!replay_file || replay_mode == REPLAY_MODE_NONE
 121         || !events_enabled) {
 122         Event e;
 123         e.event_kind = event_kind;
 124         e.opaque = opaque;
 125         e.opaque2 = opaque2;
 126         e.id = id;
 127         replay_run_event(&e);
 128         return;
 129     }
 130
 131     Event *event = g_malloc0(sizeof(Event));
 132     event->event_kind = event_kind;
 133     event->opaque = opaque;
 134     event->opaque2 = opaque2;
 135     event->id = id;
 136
 137     g_assert(replay_mutex_locked());
 138     QTAILQ_INSERT_TAIL(&events_list, event, events);
 139 }
 140
 141 void replay_bh_schedule_event(QEMUBH *bh)
 142 {
 143     if (events_enabled) {
 144         uint64_t id = replay_get_current_step();
 145         replay_add_event(REPLAY_ASYNC_EVENT_BH, bh, NULL, id);
 146     } else {
 147         qemu_bh_schedule(bh);
 148     }
 149 }
 150
 151 void replay_add_input_event(struct InputEvent *event)
 152 {
 153     replay_add_event(REPLAY_ASYNC_EVENT_INPUT, event, NULL, 0);
 154 }
 155
 156 void replay_add_input_sync_event(void)
 157 {
 158     replay_add_event(REPLAY_ASYNC_EVENT_INPUT_SYNC, NULL, NULL, 0);
 159 }
 160
 161 void replay_block_event(QEMUBH *bh, uint64_t id)
 162 {
 163     if (events_enabled) {
 164         replay_add_event(REPLAY_ASYNC_EVENT_BLOCK, bh, NULL, id);
 165     } else {
 166         qemu_bh_schedule(bh);
 167     }
 168 }
 169
 170 static void replay_save_event(Event *event, int checkpoint)
 171 {
 172     if (replay_mode != REPLAY_MODE_PLAY) {
 173         /* put the event into the file */
 174         replay_put_event(EVENT_ASYNC);
 175         replay_put_byte(checkpoint);
 176         replay_put_byte(event->event_kind);
 177
 178         /* save event-specific data */
 179         switch (event->event_kind) {
 180         case REPLAY_ASYNC_EVENT_BH:
 181             replay_put_qword(event->id);
 182             break;
 183         case REPLAY_ASYNC_EVENT_INPUT:
 184             replay_save_input_event(event->opaque);
 185             break;
 186         case REPLAY_ASYNC_EVENT_INPUT_SYNC:
 187             break;
 188         case REPLAY_ASYNC_EVENT_CHAR_READ:
 189             replay_event_char_read_save(event->opaque);
 190             break;
 191         case REPLAY_ASYNC_EVENT_BLOCK:
 192             replay_put_qword(event->id);
 193             break;
 194         case REPLAY_ASYNC_EVENT_NET:
 195             replay_event_net_save(event->opaque);
 196             break;
 197         default:
 198             error_report("Unknown ID %" PRId64 " of replay event", event->id);
 199             exit(1);
 200         }
 201     }
 202 }
 203
 204 /* Called with replay mutex locked */
 205 void replay_save_events(int checkpoint)
 206 {
 207     g_assert(replay_mutex_locked());
 208     while (!QTAILQ_EMPTY(&events_list)) {
 209         Event *event = QTAILQ_FIRST(&events_list);
 210         replay_save_event(event, checkpoint);
 211         replay_run_event(event);
 212         QTAILQ_REMOVE(&events_list, event, events);
 213         g_free(event);
 214     }
 215 }
 216
 217 static Event *replay_read_event(int checkpoint)
 218 {
 219     Event *event;
 220     if (read_event_kind == -1) {
 221         read_checkpoint = replay_get_byte();
 222         read_event_kind = replay_get_byte();
 223         read_id = -1;
 224         replay_check_error();
 225     }
 226
 227     if (checkpoint != read_checkpoint) {
 228         return NULL;
 229     }
 230
 231     /* Events that has not to be in the queue */
 232     switch (read_event_kind) {
 233     case REPLAY_ASYNC_EVENT_BH:
 234         if (read_id == -1) {
 235             read_id = replay_get_qword();
 236         }
 237         break;
 238     case REPLAY_ASYNC_EVENT_INPUT:
 239         event = g_malloc0(sizeof(Event));
 240         event->event_kind = read_event_kind;
 241         event->opaque = replay_read_input_event();
 242         return event;
 243     case REPLAY_ASYNC_EVENT_INPUT_SYNC:
 244         event = g_malloc0(sizeof(Event));
 245         event->event_kind = read_event_kind;
 246         event->opaque = 0;
 247         return event;
 248     case REPLAY_ASYNC_EVENT_CHAR_READ:
 249         event = g_malloc0(sizeof(Event));
 250         event->event_kind = read_event_kind;
 251         event->opaque = replay_event_char_read_load();
 252         return event;
 253     case REPLAY_ASYNC_EVENT_BLOCK:
 254         if (read_id == -1) {
 255             read_id = replay_get_qword();
 256         }
 257         break;
 258     case REPLAY_ASYNC_EVENT_NET:
 259         event = g_malloc0(sizeof(Event));
 260         event->event_kind = read_event_kind;
 261         event->opaque = replay_event_net_load();
 262         return event;
 263     default:
 264         error_report("Unknown ID %d of replay event", read_event_kind);
 265         exit(1);
 266         break;
 267     }
 268
 269     QTAILQ_FOREACH(event, &events_list, events) {
 270         if (event->event_kind == read_event_kind
 271             && (read_id == -1 || read_id == event->id)) {
 272             break;
 273         }
 274     }
 275
 276     if (event) {
 277         QTAILQ_REMOVE(&events_list, event, events);
 278     } else {
 279         return NULL;
 280     }
 281
 282     /* Read event-specific data */
 283
 284     return event;
 285 }
 286
 287 /* Called with replay mutex locked */
 288 void replay_read_events(int checkpoint)
 289 {
 290     g_assert(replay_mutex_locked());
 291     while (replay_state.data_kind == EVENT_ASYNC) {
 292         Event *event = replay_read_event(checkpoint);
 293         if (!event) {
 294             break;
 295         }
 296         replay_finish_event();
 297         read_event_kind = -1;
 298         replay_run_event(event);
 299
 300         g_free(event);
 301     }
 302 }
 303
 304 void replay_init_events(void)
 305 {
 306     read_event_kind = -1;
 307 }
 308
 309 void replay_finish_events(void)
 310 {
 311     events_enabled = false;
 312     replay_clear_events();
 313 }
 314
 315 bool replay_events_enabled(void)
 316 {
 317     return events_enabled;
 318 }
 319
 320 uint64_t blkreplay_next_id(void)
 321 {
 322     if (replay_events_enabled()) {
 323         return replay_state.block_request_id++;
 324     }
 325     return 0;
 326 }