Juan Quintela | 84a899d | 2017-04-24 18:53:30 +0200 | [diff] [blame] | 1 | /* |
| 2 | * Global State configuration |
| 3 | * |
| 4 | * Copyright (c) 2014-2017 Red Hat Inc |
| 5 | * |
| 6 | * Authors: |
| 7 | * Juan Quintela <quintela@redhat.com> |
| 8 | * |
| 9 | * This work is licensed under the terms of the GNU GPL, version 2 or later. |
| 10 | * See the COPYING file in the top-level directory. |
| 11 | */ |
| 12 | |
| 13 | #include "qemu/osdep.h" |
| 14 | #include "qemu/cutils.h" |
| 15 | #include "qemu/error-report.h" |
Markus Armbruster | 54d3123 | 2019-08-12 07:23:59 +0200 | [diff] [blame] | 16 | #include "sysemu/runstate.h" |
Juan Quintela | 84a899d | 2017-04-24 18:53:30 +0200 | [diff] [blame] | 17 | #include "qapi/error.h" |
Peter Xu | 5272298 | 2017-06-27 12:10:14 +0800 | [diff] [blame] | 18 | #include "migration.h" |
Juan Quintela | 84a899d | 2017-04-24 18:53:30 +0200 | [diff] [blame] | 19 | #include "migration/global_state.h" |
| 20 | #include "migration/vmstate.h" |
Juan Quintela | 84a899d | 2017-04-24 18:53:30 +0200 | [diff] [blame] | 21 | #include "trace.h" |
| 22 | |
| 23 | typedef struct { |
Juan Quintela | 84a899d | 2017-04-24 18:53:30 +0200 | [diff] [blame] | 24 | uint32_t size; |
| 25 | uint8_t runstate[100]; |
| 26 | RunState state; |
| 27 | bool received; |
| 28 | } GlobalState; |
| 29 | |
| 30 | static GlobalState global_state; |
| 31 | |
| 32 | int global_state_store(void) |
| 33 | { |
| 34 | if (!runstate_store((char *)global_state.runstate, |
| 35 | sizeof(global_state.runstate))) { |
| 36 | error_report("runstate name too big: %s", global_state.runstate); |
| 37 | trace_migrate_state_too_big(); |
| 38 | return -EINVAL; |
| 39 | } |
| 40 | return 0; |
| 41 | } |
| 42 | |
| 43 | void global_state_store_running(void) |
| 44 | { |
Markus Armbruster | 977c736 | 2017-08-24 10:46:08 +0200 | [diff] [blame] | 45 | const char *state = RunState_str(RUN_STATE_RUNNING); |
Marc-André Lureau | 0a5526a | 2019-01-03 09:56:37 +0100 | [diff] [blame] | 46 | assert(strlen(state) < sizeof(global_state.runstate)); |
Thomas Huth | 043c2c1 | 2020-09-25 16:40:16 +0100 | [diff] [blame] | 47 | strpadcpy((char *)global_state.runstate, sizeof(global_state.runstate), |
| 48 | state, '\0'); |
Juan Quintela | 84a899d | 2017-04-24 18:53:30 +0200 | [diff] [blame] | 49 | } |
| 50 | |
| 51 | bool global_state_received(void) |
| 52 | { |
| 53 | return global_state.received; |
| 54 | } |
| 55 | |
| 56 | RunState global_state_get_runstate(void) |
| 57 | { |
| 58 | return global_state.state; |
| 59 | } |
| 60 | |
Juan Quintela | 84a899d | 2017-04-24 18:53:30 +0200 | [diff] [blame] | 61 | static bool global_state_needed(void *opaque) |
| 62 | { |
| 63 | GlobalState *s = opaque; |
| 64 | char *runstate = (char *)s->runstate; |
| 65 | |
| 66 | /* If it is not optional, it is mandatory */ |
| 67 | |
Peter Xu | 5272298 | 2017-06-27 12:10:14 +0800 | [diff] [blame] | 68 | if (migrate_get_current()->store_global_state) { |
Juan Quintela | 84a899d | 2017-04-24 18:53:30 +0200 | [diff] [blame] | 69 | return true; |
| 70 | } |
| 71 | |
| 72 | /* If state is running or paused, it is not needed */ |
| 73 | |
| 74 | if (strcmp(runstate, "running") == 0 || |
| 75 | strcmp(runstate, "paused") == 0) { |
| 76 | return false; |
| 77 | } |
| 78 | |
| 79 | /* for any other state it is needed */ |
| 80 | return true; |
| 81 | } |
| 82 | |
| 83 | static int global_state_post_load(void *opaque, int version_id) |
| 84 | { |
| 85 | GlobalState *s = opaque; |
| 86 | Error *local_err = NULL; |
| 87 | int r; |
| 88 | char *runstate = (char *)s->runstate; |
| 89 | |
| 90 | s->received = true; |
| 91 | trace_migrate_global_state_post_load(runstate); |
| 92 | |
Philippe Mathieu-Daudé | a346af9 | 2019-01-03 09:56:38 +0100 | [diff] [blame] | 93 | if (strnlen((char *)s->runstate, |
| 94 | sizeof(s->runstate)) == sizeof(s->runstate)) { |
| 95 | /* |
| 96 | * This condition should never happen during migration, because |
| 97 | * all runstate names are shorter than 100 bytes (the size of |
| 98 | * s->runstate). However, a malicious stream could overflow |
| 99 | * the qapi_enum_parse() call, so we force the last character |
| 100 | * to a NUL byte. |
| 101 | */ |
| 102 | s->runstate[sizeof(s->runstate) - 1] = '\0'; |
| 103 | } |
Marc-André Lureau | f7abe0e | 2017-08-24 10:46:10 +0200 | [diff] [blame] | 104 | r = qapi_enum_parse(&RunState_lookup, runstate, -1, &local_err); |
Juan Quintela | 84a899d | 2017-04-24 18:53:30 +0200 | [diff] [blame] | 105 | |
| 106 | if (r == -1) { |
| 107 | if (local_err) { |
| 108 | error_report_err(local_err); |
| 109 | } |
| 110 | return -EINVAL; |
| 111 | } |
| 112 | s->state = r; |
| 113 | |
| 114 | return 0; |
| 115 | } |
| 116 | |
Dr. David Alan Gilbert | 44b1ff3 | 2017-09-25 12:29:12 +0100 | [diff] [blame] | 117 | static int global_state_pre_save(void *opaque) |
Juan Quintela | 84a899d | 2017-04-24 18:53:30 +0200 | [diff] [blame] | 118 | { |
| 119 | GlobalState *s = opaque; |
| 120 | |
| 121 | trace_migrate_global_state_pre_save((char *)s->runstate); |
Philippe Mathieu-Daudé | a346af9 | 2019-01-03 09:56:38 +0100 | [diff] [blame] | 122 | s->size = strnlen((char *)s->runstate, sizeof(s->runstate)) + 1; |
| 123 | assert(s->size <= sizeof(s->runstate)); |
Dr. David Alan Gilbert | 44b1ff3 | 2017-09-25 12:29:12 +0100 | [diff] [blame] | 124 | |
| 125 | return 0; |
Juan Quintela | 84a899d | 2017-04-24 18:53:30 +0200 | [diff] [blame] | 126 | } |
| 127 | |
| 128 | static const VMStateDescription vmstate_globalstate = { |
| 129 | .name = "globalstate", |
| 130 | .version_id = 1, |
| 131 | .minimum_version_id = 1, |
| 132 | .post_load = global_state_post_load, |
| 133 | .pre_save = global_state_pre_save, |
| 134 | .needed = global_state_needed, |
| 135 | .fields = (VMStateField[]) { |
| 136 | VMSTATE_UINT32(size, GlobalState), |
| 137 | VMSTATE_BUFFER(runstate, GlobalState), |
| 138 | VMSTATE_END_OF_LIST() |
| 139 | }, |
| 140 | }; |
| 141 | |
| 142 | void register_global_state(void) |
| 143 | { |
| 144 | /* We would use it independently that we receive it */ |
| 145 | strcpy((char *)&global_state.runstate, ""); |
| 146 | global_state.received = false; |
| 147 | vmstate_register(NULL, 0, &vmstate_globalstate, &global_state); |
| 148 | } |