2
0

vmstate-types.c 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894
  1. /*
  2. * VMStateInfo's for basic typse
  3. *
  4. * Copyright (c) 2009-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. #include "qemu/osdep.h"
  13. #include "qemu/cpu-float.h"
  14. #include "qemu-file.h"
  15. #include "migration.h"
  16. #include "migration/vmstate.h"
  17. #include "qemu/error-report.h"
  18. #include "qemu/queue.h"
  19. #include "trace.h"
  20. /* bool */
  21. static int get_bool(QEMUFile *f, void *pv, size_t size,
  22. const VMStateField *field)
  23. {
  24. bool *v = pv;
  25. *v = qemu_get_byte(f);
  26. return 0;
  27. }
  28. static int put_bool(QEMUFile *f, void *pv, size_t size,
  29. const VMStateField *field, JSONWriter *vmdesc)
  30. {
  31. bool *v = pv;
  32. qemu_put_byte(f, *v);
  33. return 0;
  34. }
  35. const VMStateInfo vmstate_info_bool = {
  36. .name = "bool",
  37. .get = get_bool,
  38. .put = put_bool,
  39. };
  40. /* 8 bit int */
  41. static int get_int8(QEMUFile *f, void *pv, size_t size,
  42. const VMStateField *field)
  43. {
  44. int8_t *v = pv;
  45. qemu_get_s8s(f, v);
  46. return 0;
  47. }
  48. static int put_int8(QEMUFile *f, void *pv, size_t size,
  49. const VMStateField *field, JSONWriter *vmdesc)
  50. {
  51. int8_t *v = pv;
  52. qemu_put_s8s(f, v);
  53. return 0;
  54. }
  55. const VMStateInfo vmstate_info_int8 = {
  56. .name = "int8",
  57. .get = get_int8,
  58. .put = put_int8,
  59. };
  60. /* 16 bit int */
  61. static int get_int16(QEMUFile *f, void *pv, size_t size,
  62. const VMStateField *field)
  63. {
  64. int16_t *v = pv;
  65. qemu_get_sbe16s(f, v);
  66. return 0;
  67. }
  68. static int put_int16(QEMUFile *f, void *pv, size_t size,
  69. const VMStateField *field, JSONWriter *vmdesc)
  70. {
  71. int16_t *v = pv;
  72. qemu_put_sbe16s(f, v);
  73. return 0;
  74. }
  75. const VMStateInfo vmstate_info_int16 = {
  76. .name = "int16",
  77. .get = get_int16,
  78. .put = put_int16,
  79. };
  80. /* 32 bit int */
  81. static int get_int32(QEMUFile *f, void *pv, size_t size,
  82. const VMStateField *field)
  83. {
  84. int32_t *v = pv;
  85. qemu_get_sbe32s(f, v);
  86. return 0;
  87. }
  88. static int put_int32(QEMUFile *f, void *pv, size_t size,
  89. const VMStateField *field, JSONWriter *vmdesc)
  90. {
  91. int32_t *v = pv;
  92. qemu_put_sbe32s(f, v);
  93. return 0;
  94. }
  95. const VMStateInfo vmstate_info_int32 = {
  96. .name = "int32",
  97. .get = get_int32,
  98. .put = put_int32,
  99. };
  100. /* 32 bit int. See that the received value is the same than the one
  101. in the field */
  102. static int get_int32_equal(QEMUFile *f, void *pv, size_t size,
  103. const VMStateField *field)
  104. {
  105. int32_t *v = pv;
  106. int32_t v2;
  107. qemu_get_sbe32s(f, &v2);
  108. if (*v == v2) {
  109. return 0;
  110. }
  111. error_report("%" PRIx32 " != %" PRIx32, *v, v2);
  112. if (field->err_hint) {
  113. error_printf("%s\n", field->err_hint);
  114. }
  115. return -EINVAL;
  116. }
  117. const VMStateInfo vmstate_info_int32_equal = {
  118. .name = "int32 equal",
  119. .get = get_int32_equal,
  120. .put = put_int32,
  121. };
  122. /* 32 bit int. Check that the received value is non-negative
  123. * and less than or equal to the one in the field.
  124. */
  125. static int get_int32_le(QEMUFile *f, void *pv, size_t size,
  126. const VMStateField *field)
  127. {
  128. int32_t *cur = pv;
  129. int32_t loaded;
  130. qemu_get_sbe32s(f, &loaded);
  131. if (loaded >= 0 && loaded <= *cur) {
  132. *cur = loaded;
  133. return 0;
  134. }
  135. error_report("Invalid value %" PRId32
  136. " expecting positive value <= %" PRId32,
  137. loaded, *cur);
  138. return -EINVAL;
  139. }
  140. const VMStateInfo vmstate_info_int32_le = {
  141. .name = "int32 le",
  142. .get = get_int32_le,
  143. .put = put_int32,
  144. };
  145. /* 64 bit int */
  146. static int get_int64(QEMUFile *f, void *pv, size_t size,
  147. const VMStateField *field)
  148. {
  149. int64_t *v = pv;
  150. qemu_get_sbe64s(f, v);
  151. return 0;
  152. }
  153. static int put_int64(QEMUFile *f, void *pv, size_t size,
  154. const VMStateField *field, JSONWriter *vmdesc)
  155. {
  156. int64_t *v = pv;
  157. qemu_put_sbe64s(f, v);
  158. return 0;
  159. }
  160. const VMStateInfo vmstate_info_int64 = {
  161. .name = "int64",
  162. .get = get_int64,
  163. .put = put_int64,
  164. };
  165. /* 8 bit unsigned int */
  166. static int get_uint8(QEMUFile *f, void *pv, size_t size,
  167. const VMStateField *field)
  168. {
  169. uint8_t *v = pv;
  170. qemu_get_8s(f, v);
  171. return 0;
  172. }
  173. static int put_uint8(QEMUFile *f, void *pv, size_t size,
  174. const VMStateField *field, JSONWriter *vmdesc)
  175. {
  176. uint8_t *v = pv;
  177. qemu_put_8s(f, v);
  178. return 0;
  179. }
  180. const VMStateInfo vmstate_info_uint8 = {
  181. .name = "uint8",
  182. .get = get_uint8,
  183. .put = put_uint8,
  184. };
  185. /* 16 bit unsigned int */
  186. static int get_uint16(QEMUFile *f, void *pv, size_t size,
  187. const VMStateField *field)
  188. {
  189. uint16_t *v = pv;
  190. qemu_get_be16s(f, v);
  191. return 0;
  192. }
  193. static int put_uint16(QEMUFile *f, void *pv, size_t size,
  194. const VMStateField *field, JSONWriter *vmdesc)
  195. {
  196. uint16_t *v = pv;
  197. qemu_put_be16s(f, v);
  198. return 0;
  199. }
  200. const VMStateInfo vmstate_info_uint16 = {
  201. .name = "uint16",
  202. .get = get_uint16,
  203. .put = put_uint16,
  204. };
  205. /* 32 bit unsigned int */
  206. static int get_uint32(QEMUFile *f, void *pv, size_t size,
  207. const VMStateField *field)
  208. {
  209. uint32_t *v = pv;
  210. qemu_get_be32s(f, v);
  211. return 0;
  212. }
  213. static int put_uint32(QEMUFile *f, void *pv, size_t size,
  214. const VMStateField *field, JSONWriter *vmdesc)
  215. {
  216. uint32_t *v = pv;
  217. qemu_put_be32s(f, v);
  218. return 0;
  219. }
  220. const VMStateInfo vmstate_info_uint32 = {
  221. .name = "uint32",
  222. .get = get_uint32,
  223. .put = put_uint32,
  224. };
  225. /* 32 bit uint. See that the received value is the same than the one
  226. in the field */
  227. static int get_uint32_equal(QEMUFile *f, void *pv, size_t size,
  228. const VMStateField *field)
  229. {
  230. uint32_t *v = pv;
  231. uint32_t v2;
  232. qemu_get_be32s(f, &v2);
  233. if (*v == v2) {
  234. return 0;
  235. }
  236. error_report("%" PRIx32 " != %" PRIx32, *v, v2);
  237. if (field->err_hint) {
  238. error_printf("%s\n", field->err_hint);
  239. }
  240. return -EINVAL;
  241. }
  242. const VMStateInfo vmstate_info_uint32_equal = {
  243. .name = "uint32 equal",
  244. .get = get_uint32_equal,
  245. .put = put_uint32,
  246. };
  247. /* 64 bit unsigned int */
  248. static int get_uint64(QEMUFile *f, void *pv, size_t size,
  249. const VMStateField *field)
  250. {
  251. uint64_t *v = pv;
  252. qemu_get_be64s(f, v);
  253. return 0;
  254. }
  255. static int put_uint64(QEMUFile *f, void *pv, size_t size,
  256. const VMStateField *field, JSONWriter *vmdesc)
  257. {
  258. uint64_t *v = pv;
  259. qemu_put_be64s(f, v);
  260. return 0;
  261. }
  262. const VMStateInfo vmstate_info_uint64 = {
  263. .name = "uint64",
  264. .get = get_uint64,
  265. .put = put_uint64,
  266. };
  267. static int get_nullptr(QEMUFile *f, void *pv, size_t size,
  268. const VMStateField *field)
  269. {
  270. if (qemu_get_byte(f) == VMS_NULLPTR_MARKER) {
  271. return 0;
  272. }
  273. error_report("vmstate: get_nullptr expected VMS_NULLPTR_MARKER");
  274. return -EINVAL;
  275. }
  276. static int put_nullptr(QEMUFile *f, void *pv, size_t size,
  277. const VMStateField *field, JSONWriter *vmdesc)
  278. {
  279. if (pv == NULL) {
  280. qemu_put_byte(f, VMS_NULLPTR_MARKER);
  281. return 0;
  282. }
  283. error_report("vmstate: put_nullptr must be called with pv == NULL");
  284. return -EINVAL;
  285. }
  286. const VMStateInfo vmstate_info_nullptr = {
  287. .name = "uint64",
  288. .get = get_nullptr,
  289. .put = put_nullptr,
  290. };
  291. /* 64 bit unsigned int. See that the received value is the same than the one
  292. in the field */
  293. static int get_uint64_equal(QEMUFile *f, void *pv, size_t size,
  294. const VMStateField *field)
  295. {
  296. uint64_t *v = pv;
  297. uint64_t v2;
  298. qemu_get_be64s(f, &v2);
  299. if (*v == v2) {
  300. return 0;
  301. }
  302. error_report("%" PRIx64 " != %" PRIx64, *v, v2);
  303. if (field->err_hint) {
  304. error_printf("%s\n", field->err_hint);
  305. }
  306. return -EINVAL;
  307. }
  308. const VMStateInfo vmstate_info_uint64_equal = {
  309. .name = "int64 equal",
  310. .get = get_uint64_equal,
  311. .put = put_uint64,
  312. };
  313. /* 8 bit int. See that the received value is the same than the one
  314. in the field */
  315. static int get_uint8_equal(QEMUFile *f, void *pv, size_t size,
  316. const VMStateField *field)
  317. {
  318. uint8_t *v = pv;
  319. uint8_t v2;
  320. qemu_get_8s(f, &v2);
  321. if (*v == v2) {
  322. return 0;
  323. }
  324. error_report("%x != %x", *v, v2);
  325. if (field->err_hint) {
  326. error_printf("%s\n", field->err_hint);
  327. }
  328. return -EINVAL;
  329. }
  330. const VMStateInfo vmstate_info_uint8_equal = {
  331. .name = "uint8 equal",
  332. .get = get_uint8_equal,
  333. .put = put_uint8,
  334. };
  335. /* 16 bit unsigned int int. See that the received value is the same than the one
  336. in the field */
  337. static int get_uint16_equal(QEMUFile *f, void *pv, size_t size,
  338. const VMStateField *field)
  339. {
  340. uint16_t *v = pv;
  341. uint16_t v2;
  342. qemu_get_be16s(f, &v2);
  343. if (*v == v2) {
  344. return 0;
  345. }
  346. error_report("%x != %x", *v, v2);
  347. if (field->err_hint) {
  348. error_printf("%s\n", field->err_hint);
  349. }
  350. return -EINVAL;
  351. }
  352. const VMStateInfo vmstate_info_uint16_equal = {
  353. .name = "uint16 equal",
  354. .get = get_uint16_equal,
  355. .put = put_uint16,
  356. };
  357. /* CPU_DoubleU type */
  358. static int get_cpudouble(QEMUFile *f, void *pv, size_t size,
  359. const VMStateField *field)
  360. {
  361. CPU_DoubleU *v = pv;
  362. qemu_get_be32s(f, &v->l.upper);
  363. qemu_get_be32s(f, &v->l.lower);
  364. return 0;
  365. }
  366. static int put_cpudouble(QEMUFile *f, void *pv, size_t size,
  367. const VMStateField *field, JSONWriter *vmdesc)
  368. {
  369. CPU_DoubleU *v = pv;
  370. qemu_put_be32s(f, &v->l.upper);
  371. qemu_put_be32s(f, &v->l.lower);
  372. return 0;
  373. }
  374. const VMStateInfo vmstate_info_cpudouble = {
  375. .name = "CPU_Double_U",
  376. .get = get_cpudouble,
  377. .put = put_cpudouble,
  378. };
  379. /* uint8_t buffers */
  380. static int get_buffer(QEMUFile *f, void *pv, size_t size,
  381. const VMStateField *field)
  382. {
  383. uint8_t *v = pv;
  384. qemu_get_buffer(f, v, size);
  385. return 0;
  386. }
  387. static int put_buffer(QEMUFile *f, void *pv, size_t size,
  388. const VMStateField *field, JSONWriter *vmdesc)
  389. {
  390. uint8_t *v = pv;
  391. qemu_put_buffer(f, v, size);
  392. return 0;
  393. }
  394. const VMStateInfo vmstate_info_buffer = {
  395. .name = "buffer",
  396. .get = get_buffer,
  397. .put = put_buffer,
  398. };
  399. /* unused buffers: space that was used for some fields that are
  400. not useful anymore */
  401. static int get_unused_buffer(QEMUFile *f, void *pv, size_t size,
  402. const VMStateField *field)
  403. {
  404. uint8_t buf[1024];
  405. int block_len;
  406. while (size > 0) {
  407. block_len = MIN(sizeof(buf), size);
  408. size -= block_len;
  409. qemu_get_buffer(f, buf, block_len);
  410. }
  411. return 0;
  412. }
  413. static int put_unused_buffer(QEMUFile *f, void *pv, size_t size,
  414. const VMStateField *field, JSONWriter *vmdesc)
  415. {
  416. static const uint8_t buf[1024];
  417. int block_len;
  418. while (size > 0) {
  419. block_len = MIN(sizeof(buf), size);
  420. size -= block_len;
  421. qemu_put_buffer(f, buf, block_len);
  422. }
  423. return 0;
  424. }
  425. const VMStateInfo vmstate_info_unused_buffer = {
  426. .name = "unused_buffer",
  427. .get = get_unused_buffer,
  428. .put = put_unused_buffer,
  429. };
  430. /* vmstate_info_tmp, see VMSTATE_WITH_TMP, the idea is that we allocate
  431. * a temporary buffer and the pre_load/pre_save methods in the child vmsd
  432. * copy stuff from the parent into the child and do calculations to fill
  433. * in fields that don't really exist in the parent but need to be in the
  434. * stream.
  435. */
  436. static int get_tmp(QEMUFile *f, void *pv, size_t size,
  437. const VMStateField *field)
  438. {
  439. int ret;
  440. const VMStateDescription *vmsd = field->vmsd;
  441. int version_id = field->version_id;
  442. void *tmp = g_malloc(size);
  443. /* Writes the parent field which is at the start of the tmp */
  444. *(void **)tmp = pv;
  445. ret = vmstate_load_state(f, vmsd, tmp, version_id);
  446. g_free(tmp);
  447. return ret;
  448. }
  449. static int put_tmp(QEMUFile *f, void *pv, size_t size,
  450. const VMStateField *field, JSONWriter *vmdesc)
  451. {
  452. const VMStateDescription *vmsd = field->vmsd;
  453. void *tmp = g_malloc(size);
  454. int ret;
  455. /* Writes the parent field which is at the start of the tmp */
  456. *(void **)tmp = pv;
  457. ret = vmstate_save_state(f, vmsd, tmp, vmdesc);
  458. g_free(tmp);
  459. return ret;
  460. }
  461. const VMStateInfo vmstate_info_tmp = {
  462. .name = "tmp",
  463. .get = get_tmp,
  464. .put = put_tmp,
  465. };
  466. /* bitmaps (as defined by bitmap.h). Note that size here is the size
  467. * of the bitmap in bits. The on-the-wire format of a bitmap is 64
  468. * bit words with the bits in big endian order. The in-memory format
  469. * is an array of 'unsigned long', which may be either 32 or 64 bits.
  470. */
  471. /* This is the number of 64 bit words sent over the wire */
  472. #define BITS_TO_U64S(nr) DIV_ROUND_UP(nr, 64)
  473. static int get_bitmap(QEMUFile *f, void *pv, size_t size,
  474. const VMStateField *field)
  475. {
  476. unsigned long *bmp = pv;
  477. int i, idx = 0;
  478. for (i = 0; i < BITS_TO_U64S(size); i++) {
  479. uint64_t w = qemu_get_be64(f);
  480. bmp[idx++] = w;
  481. if (sizeof(unsigned long) == 4 && idx < BITS_TO_LONGS(size)) {
  482. bmp[idx++] = w >> 32;
  483. }
  484. }
  485. return 0;
  486. }
  487. static int put_bitmap(QEMUFile *f, void *pv, size_t size,
  488. const VMStateField *field, JSONWriter *vmdesc)
  489. {
  490. unsigned long *bmp = pv;
  491. int i, idx = 0;
  492. for (i = 0; i < BITS_TO_U64S(size); i++) {
  493. uint64_t w = bmp[idx++];
  494. if (sizeof(unsigned long) == 4 && idx < BITS_TO_LONGS(size)) {
  495. w |= ((uint64_t)bmp[idx++]) << 32;
  496. }
  497. qemu_put_be64(f, w);
  498. }
  499. return 0;
  500. }
  501. const VMStateInfo vmstate_info_bitmap = {
  502. .name = "bitmap",
  503. .get = get_bitmap,
  504. .put = put_bitmap,
  505. };
  506. /* get for QTAILQ
  507. * meta data about the QTAILQ is encoded in a VMStateField structure
  508. */
  509. static int get_qtailq(QEMUFile *f, void *pv, size_t unused_size,
  510. const VMStateField *field)
  511. {
  512. int ret = 0;
  513. const VMStateDescription *vmsd = field->vmsd;
  514. /* size of a QTAILQ element */
  515. size_t size = field->size;
  516. /* offset of the QTAILQ entry in a QTAILQ element */
  517. size_t entry_offset = field->start;
  518. int version_id = field->version_id;
  519. void *elm;
  520. trace_get_qtailq(vmsd->name, version_id);
  521. if (version_id > vmsd->version_id) {
  522. error_report("%s %s", vmsd->name, "too new");
  523. trace_get_qtailq_end(vmsd->name, "too new", -EINVAL);
  524. return -EINVAL;
  525. }
  526. if (version_id < vmsd->minimum_version_id) {
  527. error_report("%s %s", vmsd->name, "too old");
  528. trace_get_qtailq_end(vmsd->name, "too old", -EINVAL);
  529. return -EINVAL;
  530. }
  531. while (qemu_get_byte(f)) {
  532. elm = g_malloc(size);
  533. ret = vmstate_load_state(f, vmsd, elm, version_id);
  534. if (ret) {
  535. return ret;
  536. }
  537. QTAILQ_RAW_INSERT_TAIL(pv, elm, entry_offset);
  538. }
  539. trace_get_qtailq_end(vmsd->name, "end", ret);
  540. return ret;
  541. }
  542. /* put for QTAILQ */
  543. static int put_qtailq(QEMUFile *f, void *pv, size_t unused_size,
  544. const VMStateField *field, JSONWriter *vmdesc)
  545. {
  546. const VMStateDescription *vmsd = field->vmsd;
  547. /* offset of the QTAILQ entry in a QTAILQ element*/
  548. size_t entry_offset = field->start;
  549. void *elm;
  550. int ret;
  551. trace_put_qtailq(vmsd->name, vmsd->version_id);
  552. QTAILQ_RAW_FOREACH(elm, pv, entry_offset) {
  553. qemu_put_byte(f, true);
  554. ret = vmstate_save_state(f, vmsd, elm, vmdesc);
  555. if (ret) {
  556. return ret;
  557. }
  558. }
  559. qemu_put_byte(f, false);
  560. trace_put_qtailq_end(vmsd->name, "end");
  561. return 0;
  562. }
  563. const VMStateInfo vmstate_info_qtailq = {
  564. .name = "qtailq",
  565. .get = get_qtailq,
  566. .put = put_qtailq,
  567. };
  568. struct put_gtree_data {
  569. QEMUFile *f;
  570. const VMStateDescription *key_vmsd;
  571. const VMStateDescription *val_vmsd;
  572. JSONWriter *vmdesc;
  573. int ret;
  574. };
  575. static gboolean put_gtree_elem(gpointer key, gpointer value, gpointer data)
  576. {
  577. struct put_gtree_data *capsule = (struct put_gtree_data *)data;
  578. QEMUFile *f = capsule->f;
  579. int ret;
  580. qemu_put_byte(f, true);
  581. /* put the key */
  582. if (!capsule->key_vmsd) {
  583. qemu_put_be64(f, (uint64_t)(uintptr_t)(key)); /* direct key */
  584. } else {
  585. ret = vmstate_save_state(f, capsule->key_vmsd, key, capsule->vmdesc);
  586. if (ret) {
  587. capsule->ret = ret;
  588. return true;
  589. }
  590. }
  591. /* put the data */
  592. ret = vmstate_save_state(f, capsule->val_vmsd, value, capsule->vmdesc);
  593. if (ret) {
  594. capsule->ret = ret;
  595. return true;
  596. }
  597. return false;
  598. }
  599. static int put_gtree(QEMUFile *f, void *pv, size_t unused_size,
  600. const VMStateField *field, JSONWriter *vmdesc)
  601. {
  602. bool direct_key = (!field->start);
  603. const VMStateDescription *key_vmsd = direct_key ? NULL : &field->vmsd[1];
  604. const VMStateDescription *val_vmsd = &field->vmsd[0];
  605. const char *key_vmsd_name = direct_key ? "direct" : key_vmsd->name;
  606. struct put_gtree_data capsule = {
  607. .f = f,
  608. .key_vmsd = key_vmsd,
  609. .val_vmsd = val_vmsd,
  610. .vmdesc = vmdesc,
  611. .ret = 0};
  612. GTree **pval = pv;
  613. GTree *tree = *pval;
  614. uint32_t nnodes = g_tree_nnodes(tree);
  615. int ret;
  616. trace_put_gtree(field->name, key_vmsd_name, val_vmsd->name, nnodes);
  617. qemu_put_be32(f, nnodes);
  618. g_tree_foreach(tree, put_gtree_elem, (gpointer)&capsule);
  619. qemu_put_byte(f, false);
  620. ret = capsule.ret;
  621. if (ret) {
  622. error_report("%s : failed to save gtree (%d)", field->name, ret);
  623. }
  624. trace_put_gtree_end(field->name, key_vmsd_name, val_vmsd->name, ret);
  625. return ret;
  626. }
  627. static int get_gtree(QEMUFile *f, void *pv, size_t unused_size,
  628. const VMStateField *field)
  629. {
  630. bool direct_key = (!field->start);
  631. const VMStateDescription *key_vmsd = direct_key ? NULL : &field->vmsd[1];
  632. const VMStateDescription *val_vmsd = &field->vmsd[0];
  633. const char *key_vmsd_name = direct_key ? "direct" : key_vmsd->name;
  634. int version_id = field->version_id;
  635. size_t key_size = field->start;
  636. size_t val_size = field->size;
  637. int nnodes, count = 0;
  638. GTree **pval = pv;
  639. GTree *tree = *pval;
  640. void *key, *val;
  641. int ret = 0;
  642. /* in case of direct key, the key vmsd can be {}, ie. check fields */
  643. if (!direct_key && version_id > key_vmsd->version_id) {
  644. error_report("%s %s", key_vmsd->name, "too new");
  645. return -EINVAL;
  646. }
  647. if (!direct_key && version_id < key_vmsd->minimum_version_id) {
  648. error_report("%s %s", key_vmsd->name, "too old");
  649. return -EINVAL;
  650. }
  651. if (version_id > val_vmsd->version_id) {
  652. error_report("%s %s", val_vmsd->name, "too new");
  653. return -EINVAL;
  654. }
  655. if (version_id < val_vmsd->minimum_version_id) {
  656. error_report("%s %s", val_vmsd->name, "too old");
  657. return -EINVAL;
  658. }
  659. nnodes = qemu_get_be32(f);
  660. trace_get_gtree(field->name, key_vmsd_name, val_vmsd->name, nnodes);
  661. while (qemu_get_byte(f)) {
  662. if ((++count) > nnodes) {
  663. ret = -EINVAL;
  664. break;
  665. }
  666. if (direct_key) {
  667. key = (void *)(uintptr_t)qemu_get_be64(f);
  668. } else {
  669. key = g_malloc0(key_size);
  670. ret = vmstate_load_state(f, key_vmsd, key, version_id);
  671. if (ret) {
  672. error_report("%s : failed to load %s (%d)",
  673. field->name, key_vmsd->name, ret);
  674. goto key_error;
  675. }
  676. }
  677. val = g_malloc0(val_size);
  678. ret = vmstate_load_state(f, val_vmsd, val, version_id);
  679. if (ret) {
  680. error_report("%s : failed to load %s (%d)",
  681. field->name, val_vmsd->name, ret);
  682. goto val_error;
  683. }
  684. g_tree_insert(tree, key, val);
  685. }
  686. if (count != nnodes) {
  687. error_report("%s inconsistent stream when loading the gtree",
  688. field->name);
  689. return -EINVAL;
  690. }
  691. trace_get_gtree_end(field->name, key_vmsd_name, val_vmsd->name, ret);
  692. return ret;
  693. val_error:
  694. g_free(val);
  695. key_error:
  696. if (!direct_key) {
  697. g_free(key);
  698. }
  699. trace_get_gtree_end(field->name, key_vmsd_name, val_vmsd->name, ret);
  700. return ret;
  701. }
  702. const VMStateInfo vmstate_info_gtree = {
  703. .name = "gtree",
  704. .get = get_gtree,
  705. .put = put_gtree,
  706. };
  707. static int put_qlist(QEMUFile *f, void *pv, size_t unused_size,
  708. const VMStateField *field, JSONWriter *vmdesc)
  709. {
  710. const VMStateDescription *vmsd = field->vmsd;
  711. /* offset of the QTAILQ entry in a QTAILQ element*/
  712. size_t entry_offset = field->start;
  713. void *elm;
  714. int ret;
  715. trace_put_qlist(field->name, vmsd->name, vmsd->version_id);
  716. QLIST_RAW_FOREACH(elm, pv, entry_offset) {
  717. qemu_put_byte(f, true);
  718. ret = vmstate_save_state(f, vmsd, elm, vmdesc);
  719. if (ret) {
  720. error_report("%s: failed to save %s (%d)", field->name,
  721. vmsd->name, ret);
  722. return ret;
  723. }
  724. }
  725. qemu_put_byte(f, false);
  726. trace_put_qlist_end(field->name, vmsd->name);
  727. return 0;
  728. }
  729. static int get_qlist(QEMUFile *f, void *pv, size_t unused_size,
  730. const VMStateField *field)
  731. {
  732. int ret = 0;
  733. const VMStateDescription *vmsd = field->vmsd;
  734. /* size of a QLIST element */
  735. size_t size = field->size;
  736. /* offset of the QLIST entry in a QLIST element */
  737. size_t entry_offset = field->start;
  738. int version_id = field->version_id;
  739. void *elm, *prev = NULL;
  740. trace_get_qlist(field->name, vmsd->name, vmsd->version_id);
  741. if (version_id > vmsd->version_id) {
  742. error_report("%s %s", vmsd->name, "too new");
  743. return -EINVAL;
  744. }
  745. if (version_id < vmsd->minimum_version_id) {
  746. error_report("%s %s", vmsd->name, "too old");
  747. return -EINVAL;
  748. }
  749. while (qemu_get_byte(f)) {
  750. elm = g_malloc(size);
  751. ret = vmstate_load_state(f, vmsd, elm, version_id);
  752. if (ret) {
  753. error_report("%s: failed to load %s (%d)", field->name,
  754. vmsd->name, ret);
  755. g_free(elm);
  756. return ret;
  757. }
  758. if (!prev) {
  759. QLIST_RAW_INSERT_HEAD(pv, elm, entry_offset);
  760. } else {
  761. QLIST_RAW_INSERT_AFTER(pv, prev, elm, entry_offset);
  762. }
  763. prev = elm;
  764. }
  765. trace_get_qlist_end(field->name, vmsd->name);
  766. return ret;
  767. }
  768. const VMStateInfo vmstate_info_qlist = {
  769. .name = "qlist",
  770. .get = get_qlist,
  771. .put = put_qlist,
  772. };