1 /*- 2 * SPDX-License-Identifier: BSD-2-Clause-FreeBSD 3 * 4 * Copyright (c) 2016 Flavius Anton 5 * Copyright (c) 2016 Mihai Tiganus 6 * Copyright (c) 2016-2019 Mihai Carabas 7 * Copyright (c) 2017-2019 Darius Mihai 8 * Copyright (c) 2017-2019 Elena Mihailescu 9 * Copyright (c) 2018-2019 Sergiu Weisz 10 * All rights reserved. 11 * The bhyve-snapshot feature was developed under sponsorships 12 * from Matthew Grooms. 13 * 14 * Redistribution and use in source and binary forms, with or without 15 * modification, are permitted provided that the following conditions 16 * are met: 17 * 1. Redistributions of source code must retain the above copyright 18 * notice, this list of conditions and the following disclaimer. 19 * 2. Redistributions in binary form must reproduce the above copyright 20 * notice, this list of conditions and the following disclaimer in the 21 * documentation and/or other materials provided with the distribution. 22 * 23 * THIS SOFTWARE IS PROVIDED BY NETAPP, INC ``AS IS'' AND 24 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 25 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 26 * ARE DISCLAIMED. IN NO EVENT SHALL NETAPP, INC OR CONTRIBUTORS BE LIABLE 27 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 28 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 29 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 30 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 31 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 32 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 33 * SUCH DAMAGE. 34 */ 35 36 #include <sys/cdefs.h> 37 __FBSDID("$FreeBSD$"); 38 39 #include <sys/types.h> 40 #include <sys/systm.h> 41 42 #include <machine/vmm_snapshot.h> 43 44 void 45 vm_snapshot_buf_err(const char *bufname, const enum vm_snapshot_op op) 46 { 47 const char *opstr; 48 49 if (op == VM_SNAPSHOT_SAVE) 50 opstr = "save"; 51 else if (op == VM_SNAPSHOT_RESTORE) 52 opstr = "restore"; 53 else 54 opstr = "unknown"; 55 56 printf("%s: snapshot-%s failed for %s\r\n", __func__, opstr, bufname); 57 } 58 59 int 60 vm_snapshot_buf(void *data, size_t data_size, struct vm_snapshot_meta *meta) 61 { 62 struct vm_snapshot_buffer *buffer; 63 int op; 64 65 buffer = &meta->buffer; 66 op = meta->op; 67 68 if (buffer->buf_rem < data_size) { 69 printf("%s: buffer too small\r\n", __func__); 70 return (E2BIG); 71 } 72 73 if (op == VM_SNAPSHOT_SAVE) 74 copyout(data, buffer->buf, data_size); 75 else if (op == VM_SNAPSHOT_RESTORE) 76 copyin(buffer->buf, data, data_size); 77 else 78 return (EINVAL); 79 80 buffer->buf += data_size; 81 buffer->buf_rem -= data_size; 82 83 return (0); 84 } 85 86 size_t 87 vm_get_snapshot_size(struct vm_snapshot_meta *meta) 88 { 89 size_t length; 90 struct vm_snapshot_buffer *buffer; 91 92 buffer = &meta->buffer; 93 94 if (buffer->buf_size < buffer->buf_rem) { 95 printf("%s: Invalid buffer: size = %zu, rem = %zu\r\n", 96 __func__, buffer->buf_size, buffer->buf_rem); 97 length = 0; 98 } else { 99 length = buffer->buf_size - buffer->buf_rem; 100 } 101 102 return (length); 103 } 104 105 int 106 vm_snapshot_buf_cmp(void *data, size_t data_size, struct vm_snapshot_meta *meta) 107 { 108 struct vm_snapshot_buffer *buffer; 109 int op; 110 int ret; 111 112 buffer = &meta->buffer; 113 op = meta->op; 114 115 if (buffer->buf_rem < data_size) { 116 printf("%s: buffer too small\r\n", __func__); 117 ret = E2BIG; 118 goto done; 119 } 120 121 if (op == VM_SNAPSHOT_SAVE) { 122 ret = 0; 123 copyout(data, buffer->buf, data_size); 124 } else if (op == VM_SNAPSHOT_RESTORE) { 125 ret = memcmp(data, buffer->buf, data_size); 126 } else { 127 ret = EINVAL; 128 goto done; 129 } 130 131 buffer->buf += data_size; 132 buffer->buf_rem -= data_size; 133 134 done: 135 return (ret); 136 } 137