]>
Commit | Line | Data |
---|---|---|
c142442b KW |
1 | /* |
2 | * Block driver for the QCOW version 2 format | |
3 | * | |
4 | * Copyright (c) 2004-2006 Fabrice Bellard | |
5 | * | |
6 | * Permission is hereby granted, free of charge, to any person obtaining a copy | |
7 | * of this software and associated documentation files (the "Software"), to deal | |
8 | * in the Software without restriction, including without limitation the rights | |
9 | * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | |
10 | * copies of the Software, and to permit persons to whom the Software is | |
11 | * furnished to do so, subject to the following conditions: | |
12 | * | |
13 | * The above copyright notice and this permission notice shall be included in | |
14 | * all copies or substantial portions of the Software. | |
15 | * | |
16 | * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | |
17 | * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | |
18 | * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL | |
19 | * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | |
20 | * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | |
21 | * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN | |
22 | * THE SOFTWARE. | |
23 | */ | |
24 | ||
25 | #include "qemu-common.h" | |
737e150e | 26 | #include "block/block_int.h" |
c142442b KW |
27 | #include "block/qcow2.h" |
28 | ||
541dc0d4 | 29 | typedef struct QEMU_PACKED QCowSnapshotHeader { |
c142442b KW |
30 | /* header is 8 byte aligned */ |
31 | uint64_t l1_table_offset; | |
32 | ||
33 | uint32_t l1_size; | |
34 | uint16_t id_str_size; | |
35 | uint16_t name_size; | |
36 | ||
37 | uint32_t date_sec; | |
38 | uint32_t date_nsec; | |
39 | ||
40 | uint64_t vm_clock_nsec; | |
41 | ||
42 | uint32_t vm_state_size; | |
43 | uint32_t extra_data_size; /* for extension */ | |
44 | /* extra data follows */ | |
45 | /* id_str follows */ | |
46 | /* name follows */ | |
47 | } QCowSnapshotHeader; | |
48 | ||
c2c9a466 KW |
49 | typedef struct QEMU_PACKED QCowSnapshotExtraData { |
50 | uint64_t vm_state_size_large; | |
90b27759 | 51 | uint64_t disk_size; |
c2c9a466 KW |
52 | } QCowSnapshotExtraData; |
53 | ||
ed6ccf0f | 54 | void qcow2_free_snapshots(BlockDriverState *bs) |
c142442b KW |
55 | { |
56 | BDRVQcowState *s = bs->opaque; | |
57 | int i; | |
58 | ||
59 | for(i = 0; i < s->nb_snapshots; i++) { | |
7267c094 AL |
60 | g_free(s->snapshots[i].name); |
61 | g_free(s->snapshots[i].id_str); | |
c142442b | 62 | } |
7267c094 | 63 | g_free(s->snapshots); |
c142442b KW |
64 | s->snapshots = NULL; |
65 | s->nb_snapshots = 0; | |
66 | } | |
67 | ||
ed6ccf0f | 68 | int qcow2_read_snapshots(BlockDriverState *bs) |
c142442b KW |
69 | { |
70 | BDRVQcowState *s = bs->opaque; | |
71 | QCowSnapshotHeader h; | |
c2c9a466 | 72 | QCowSnapshotExtraData extra; |
c142442b KW |
73 | QCowSnapshot *sn; |
74 | int i, id_str_size, name_size; | |
75 | int64_t offset; | |
76 | uint32_t extra_data_size; | |
42deb29f | 77 | int ret; |
c142442b KW |
78 | |
79 | if (!s->nb_snapshots) { | |
80 | s->snapshots = NULL; | |
81 | s->snapshots_size = 0; | |
82 | return 0; | |
83 | } | |
84 | ||
85 | offset = s->snapshots_offset; | |
7267c094 | 86 | s->snapshots = g_malloc0(s->nb_snapshots * sizeof(QCowSnapshot)); |
42deb29f | 87 | |
c142442b | 88 | for(i = 0; i < s->nb_snapshots; i++) { |
42deb29f | 89 | /* Read statically sized part of the snapshot header */ |
c142442b | 90 | offset = align_offset(offset, 8); |
42deb29f KW |
91 | ret = bdrv_pread(bs->file, offset, &h, sizeof(h)); |
92 | if (ret < 0) { | |
c142442b | 93 | goto fail; |
42deb29f KW |
94 | } |
95 | ||
c142442b KW |
96 | offset += sizeof(h); |
97 | sn = s->snapshots + i; | |
98 | sn->l1_table_offset = be64_to_cpu(h.l1_table_offset); | |
99 | sn->l1_size = be32_to_cpu(h.l1_size); | |
100 | sn->vm_state_size = be32_to_cpu(h.vm_state_size); | |
101 | sn->date_sec = be32_to_cpu(h.date_sec); | |
102 | sn->date_nsec = be32_to_cpu(h.date_nsec); | |
103 | sn->vm_clock_nsec = be64_to_cpu(h.vm_clock_nsec); | |
104 | extra_data_size = be32_to_cpu(h.extra_data_size); | |
105 | ||
106 | id_str_size = be16_to_cpu(h.id_str_size); | |
107 | name_size = be16_to_cpu(h.name_size); | |
108 | ||
c2c9a466 KW |
109 | /* Read extra data */ |
110 | ret = bdrv_pread(bs->file, offset, &extra, | |
111 | MIN(sizeof(extra), extra_data_size)); | |
112 | if (ret < 0) { | |
113 | goto fail; | |
114 | } | |
c142442b KW |
115 | offset += extra_data_size; |
116 | ||
c2c9a466 KW |
117 | if (extra_data_size >= 8) { |
118 | sn->vm_state_size = be64_to_cpu(extra.vm_state_size_large); | |
119 | } | |
120 | ||
90b27759 KW |
121 | if (extra_data_size >= 16) { |
122 | sn->disk_size = be64_to_cpu(extra.disk_size); | |
123 | } else { | |
124 | sn->disk_size = bs->total_sectors * BDRV_SECTOR_SIZE; | |
125 | } | |
126 | ||
42deb29f | 127 | /* Read snapshot ID */ |
7267c094 | 128 | sn->id_str = g_malloc(id_str_size + 1); |
42deb29f KW |
129 | ret = bdrv_pread(bs->file, offset, sn->id_str, id_str_size); |
130 | if (ret < 0) { | |
c142442b | 131 | goto fail; |
42deb29f | 132 | } |
c142442b KW |
133 | offset += id_str_size; |
134 | sn->id_str[id_str_size] = '\0'; | |
135 | ||
42deb29f | 136 | /* Read snapshot name */ |
7267c094 | 137 | sn->name = g_malloc(name_size + 1); |
42deb29f KW |
138 | ret = bdrv_pread(bs->file, offset, sn->name, name_size); |
139 | if (ret < 0) { | |
c142442b | 140 | goto fail; |
42deb29f | 141 | } |
c142442b KW |
142 | offset += name_size; |
143 | sn->name[name_size] = '\0'; | |
144 | } | |
42deb29f | 145 | |
c142442b KW |
146 | s->snapshots_size = offset - s->snapshots_offset; |
147 | return 0; | |
42deb29f KW |
148 | |
149 | fail: | |
ed6ccf0f | 150 | qcow2_free_snapshots(bs); |
42deb29f | 151 | return ret; |
c142442b KW |
152 | } |
153 | ||
154 | /* add at the end of the file a new list of snapshots */ | |
7c80ab3f | 155 | static int qcow2_write_snapshots(BlockDriverState *bs) |
c142442b KW |
156 | { |
157 | BDRVQcowState *s = bs->opaque; | |
158 | QCowSnapshot *sn; | |
159 | QCowSnapshotHeader h; | |
c2c9a466 | 160 | QCowSnapshotExtraData extra; |
c142442b | 161 | int i, name_size, id_str_size, snapshots_size; |
d69969c4 KW |
162 | struct { |
163 | uint32_t nb_snapshots; | |
164 | uint64_t snapshots_offset; | |
165 | } QEMU_PACKED header_data; | |
c142442b | 166 | int64_t offset, snapshots_offset; |
07fd8779 | 167 | int ret; |
c142442b KW |
168 | |
169 | /* compute the size of the snapshots */ | |
170 | offset = 0; | |
171 | for(i = 0; i < s->nb_snapshots; i++) { | |
172 | sn = s->snapshots + i; | |
173 | offset = align_offset(offset, 8); | |
174 | offset += sizeof(h); | |
c2c9a466 | 175 | offset += sizeof(extra); |
c142442b KW |
176 | offset += strlen(sn->id_str); |
177 | offset += strlen(sn->name); | |
178 | } | |
179 | snapshots_size = offset; | |
180 | ||
07fd8779 | 181 | /* Allocate space for the new snapshot list */ |
ed6ccf0f | 182 | snapshots_offset = qcow2_alloc_clusters(bs, snapshots_size); |
c142442b | 183 | offset = snapshots_offset; |
5d757b56 KW |
184 | if (offset < 0) { |
185 | return offset; | |
186 | } | |
f6977f15 SH |
187 | ret = bdrv_flush(bs); |
188 | if (ret < 0) { | |
189 | return ret; | |
190 | } | |
c142442b | 191 | |
07fd8779 | 192 | /* Write all snapshots to the new list */ |
c142442b KW |
193 | for(i = 0; i < s->nb_snapshots; i++) { |
194 | sn = s->snapshots + i; | |
195 | memset(&h, 0, sizeof(h)); | |
196 | h.l1_table_offset = cpu_to_be64(sn->l1_table_offset); | |
197 | h.l1_size = cpu_to_be32(sn->l1_size); | |
c2c9a466 KW |
198 | /* If it doesn't fit in 32 bit, older implementations should treat it |
199 | * as a disk-only snapshot rather than truncate the VM state */ | |
200 | if (sn->vm_state_size <= 0xffffffff) { | |
201 | h.vm_state_size = cpu_to_be32(sn->vm_state_size); | |
202 | } | |
c142442b KW |
203 | h.date_sec = cpu_to_be32(sn->date_sec); |
204 | h.date_nsec = cpu_to_be32(sn->date_nsec); | |
205 | h.vm_clock_nsec = cpu_to_be64(sn->vm_clock_nsec); | |
c2c9a466 KW |
206 | h.extra_data_size = cpu_to_be32(sizeof(extra)); |
207 | ||
208 | memset(&extra, 0, sizeof(extra)); | |
209 | extra.vm_state_size_large = cpu_to_be64(sn->vm_state_size); | |
90b27759 | 210 | extra.disk_size = cpu_to_be64(sn->disk_size); |
c142442b KW |
211 | |
212 | id_str_size = strlen(sn->id_str); | |
213 | name_size = strlen(sn->name); | |
214 | h.id_str_size = cpu_to_be16(id_str_size); | |
215 | h.name_size = cpu_to_be16(name_size); | |
216 | offset = align_offset(offset, 8); | |
07fd8779 KW |
217 | |
218 | ret = bdrv_pwrite(bs->file, offset, &h, sizeof(h)); | |
219 | if (ret < 0) { | |
c142442b | 220 | goto fail; |
07fd8779 | 221 | } |
c142442b | 222 | offset += sizeof(h); |
07fd8779 | 223 | |
c2c9a466 KW |
224 | ret = bdrv_pwrite(bs->file, offset, &extra, sizeof(extra)); |
225 | if (ret < 0) { | |
226 | goto fail; | |
227 | } | |
228 | offset += sizeof(extra); | |
229 | ||
07fd8779 KW |
230 | ret = bdrv_pwrite(bs->file, offset, sn->id_str, id_str_size); |
231 | if (ret < 0) { | |
c142442b | 232 | goto fail; |
07fd8779 | 233 | } |
c142442b | 234 | offset += id_str_size; |
07fd8779 KW |
235 | |
236 | ret = bdrv_pwrite(bs->file, offset, sn->name, name_size); | |
237 | if (ret < 0) { | |
c142442b | 238 | goto fail; |
07fd8779 | 239 | } |
c142442b KW |
240 | offset += name_size; |
241 | } | |
242 | ||
07fd8779 KW |
243 | /* |
244 | * Update the header to point to the new snapshot table. This requires the | |
245 | * new table and its refcounts to be stable on disk. | |
07fd8779 KW |
246 | */ |
247 | ret = bdrv_flush(bs); | |
248 | if (ret < 0) { | |
249 | goto fail; | |
250 | } | |
251 | ||
d69969c4 KW |
252 | QEMU_BUILD_BUG_ON(offsetof(QCowHeader, snapshots_offset) != |
253 | offsetof(QCowHeader, nb_snapshots) + sizeof(header_data.nb_snapshots)); | |
254 | ||
255 | header_data.nb_snapshots = cpu_to_be32(s->nb_snapshots); | |
256 | header_data.snapshots_offset = cpu_to_be64(snapshots_offset); | |
07fd8779 | 257 | |
07fd8779 | 258 | ret = bdrv_pwrite_sync(bs->file, offsetof(QCowHeader, nb_snapshots), |
d69969c4 | 259 | &header_data, sizeof(header_data)); |
07fd8779 | 260 | if (ret < 0) { |
c142442b | 261 | goto fail; |
07fd8779 | 262 | } |
c142442b KW |
263 | |
264 | /* free the old snapshot table */ | |
ed6ccf0f | 265 | qcow2_free_clusters(bs, s->snapshots_offset, s->snapshots_size); |
c142442b KW |
266 | s->snapshots_offset = snapshots_offset; |
267 | s->snapshots_size = snapshots_size; | |
268 | return 0; | |
07fd8779 KW |
269 | |
270 | fail: | |
271 | return ret; | |
c142442b KW |
272 | } |
273 | ||
274 | static void find_new_snapshot_id(BlockDriverState *bs, | |
275 | char *id_str, int id_str_size) | |
276 | { | |
277 | BDRVQcowState *s = bs->opaque; | |
278 | QCowSnapshot *sn; | |
279 | int i, id, id_max = 0; | |
280 | ||
281 | for(i = 0; i < s->nb_snapshots; i++) { | |
282 | sn = s->snapshots + i; | |
283 | id = strtoul(sn->id_str, NULL, 10); | |
284 | if (id > id_max) | |
285 | id_max = id; | |
286 | } | |
287 | snprintf(id_str, id_str_size, "%d", id_max + 1); | |
288 | } | |
289 | ||
290 | static int find_snapshot_by_id(BlockDriverState *bs, const char *id_str) | |
291 | { | |
292 | BDRVQcowState *s = bs->opaque; | |
293 | int i; | |
294 | ||
295 | for(i = 0; i < s->nb_snapshots; i++) { | |
296 | if (!strcmp(s->snapshots[i].id_str, id_str)) | |
297 | return i; | |
298 | } | |
299 | return -1; | |
300 | } | |
301 | ||
302 | static int find_snapshot_by_id_or_name(BlockDriverState *bs, const char *name) | |
303 | { | |
304 | BDRVQcowState *s = bs->opaque; | |
305 | int i, ret; | |
306 | ||
307 | ret = find_snapshot_by_id(bs, name); | |
308 | if (ret >= 0) | |
309 | return ret; | |
310 | for(i = 0; i < s->nb_snapshots; i++) { | |
311 | if (!strcmp(s->snapshots[i].name, name)) | |
312 | return i; | |
313 | } | |
314 | return -1; | |
315 | } | |
316 | ||
317 | /* if no id is provided, a new one is constructed */ | |
ed6ccf0f | 318 | int qcow2_snapshot_create(BlockDriverState *bs, QEMUSnapshotInfo *sn_info) |
c142442b KW |
319 | { |
320 | BDRVQcowState *s = bs->opaque; | |
d1ea98d5 KW |
321 | QCowSnapshot *new_snapshot_list = NULL; |
322 | QCowSnapshot *old_snapshot_list = NULL; | |
323 | QCowSnapshot sn1, *sn = &sn1; | |
c142442b KW |
324 | int i, ret; |
325 | uint64_t *l1_table = NULL; | |
5d757b56 | 326 | int64_t l1_table_offset; |
c142442b KW |
327 | |
328 | memset(sn, 0, sizeof(*sn)); | |
329 | ||
03343166 | 330 | /* Generate an ID if it wasn't passed */ |
c142442b | 331 | if (sn_info->id_str[0] == '\0') { |
c142442b KW |
332 | find_new_snapshot_id(bs, sn_info->id_str, sizeof(sn_info->id_str)); |
333 | } | |
334 | ||
03343166 KW |
335 | /* Check that the ID is unique */ |
336 | if (find_snapshot_by_id(bs, sn_info->id_str) >= 0) { | |
647cc472 | 337 | return -EEXIST; |
03343166 | 338 | } |
c142442b | 339 | |
03343166 | 340 | /* Populate sn with passed data */ |
7267c094 | 341 | sn->id_str = g_strdup(sn_info->id_str); |
7267c094 | 342 | sn->name = g_strdup(sn_info->name); |
03343166 | 343 | |
90b27759 | 344 | sn->disk_size = bs->total_sectors * BDRV_SECTOR_SIZE; |
c142442b KW |
345 | sn->vm_state_size = sn_info->vm_state_size; |
346 | sn->date_sec = sn_info->date_sec; | |
347 | sn->date_nsec = sn_info->date_nsec; | |
348 | sn->vm_clock_nsec = sn_info->vm_clock_nsec; | |
349 | ||
03343166 | 350 | /* Allocate the L1 table of the snapshot and copy the current one there. */ |
5d757b56 KW |
351 | l1_table_offset = qcow2_alloc_clusters(bs, s->l1_size * sizeof(uint64_t)); |
352 | if (l1_table_offset < 0) { | |
d1ea98d5 | 353 | ret = l1_table_offset; |
5d757b56 KW |
354 | goto fail; |
355 | } | |
356 | ||
357 | sn->l1_table_offset = l1_table_offset; | |
c142442b KW |
358 | sn->l1_size = s->l1_size; |
359 | ||
03343166 | 360 | l1_table = g_malloc(s->l1_size * sizeof(uint64_t)); |
c142442b KW |
361 | for(i = 0; i < s->l1_size; i++) { |
362 | l1_table[i] = cpu_to_be64(s->l1_table[i]); | |
363 | } | |
d1ea98d5 KW |
364 | |
365 | ret = bdrv_pwrite(bs->file, sn->l1_table_offset, l1_table, | |
366 | s->l1_size * sizeof(uint64_t)); | |
367 | if (ret < 0) { | |
c142442b | 368 | goto fail; |
d1ea98d5 KW |
369 | } |
370 | ||
7267c094 | 371 | g_free(l1_table); |
c142442b KW |
372 | l1_table = NULL; |
373 | ||
d1ea98d5 KW |
374 | /* |
375 | * Increase the refcounts of all clusters and make sure everything is | |
376 | * stable on disk before updating the snapshot table to contain a pointer | |
377 | * to the new L1 table. | |
378 | */ | |
379 | ret = qcow2_update_snapshot_refcount(bs, s->l1_table_offset, s->l1_size, 1); | |
380 | if (ret < 0) { | |
381 | goto fail; | |
382 | } | |
383 | ||
d1ea98d5 KW |
384 | /* Append the new snapshot to the snapshot list */ |
385 | new_snapshot_list = g_malloc((s->nb_snapshots + 1) * sizeof(QCowSnapshot)); | |
c142442b | 386 | if (s->snapshots) { |
d1ea98d5 KW |
387 | memcpy(new_snapshot_list, s->snapshots, |
388 | s->nb_snapshots * sizeof(QCowSnapshot)); | |
389 | old_snapshot_list = s->snapshots; | |
c142442b | 390 | } |
d1ea98d5 | 391 | s->snapshots = new_snapshot_list; |
c142442b KW |
392 | s->snapshots[s->nb_snapshots++] = *sn; |
393 | ||
d1ea98d5 KW |
394 | ret = qcow2_write_snapshots(bs); |
395 | if (ret < 0) { | |
396 | g_free(s->snapshots); | |
397 | s->snapshots = old_snapshot_list; | |
c142442b | 398 | goto fail; |
d1ea98d5 KW |
399 | } |
400 | ||
401 | g_free(old_snapshot_list); | |
402 | ||
c142442b | 403 | #ifdef DEBUG_ALLOC |
6cbc3031 PH |
404 | { |
405 | BdrvCheckResult result = {0}; | |
b35278f7 | 406 | qcow2_check_refcounts(bs, &result, 0); |
6cbc3031 | 407 | } |
c142442b KW |
408 | #endif |
409 | return 0; | |
03343166 KW |
410 | |
411 | fail: | |
412 | g_free(sn->id_str); | |
7267c094 AL |
413 | g_free(sn->name); |
414 | g_free(l1_table); | |
d1ea98d5 KW |
415 | |
416 | return ret; | |
c142442b KW |
417 | } |
418 | ||
419 | /* copy the snapshot 'snapshot_name' into the current disk image */ | |
ed6ccf0f | 420 | int qcow2_snapshot_goto(BlockDriverState *bs, const char *snapshot_id) |
c142442b KW |
421 | { |
422 | BDRVQcowState *s = bs->opaque; | |
423 | QCowSnapshot *sn; | |
35d7ace7 KW |
424 | int i, snapshot_index; |
425 | int cur_l1_bytes, sn_l1_bytes; | |
589f284b | 426 | int ret; |
43a0cac4 | 427 | uint64_t *sn_l1_table = NULL; |
c142442b | 428 | |
589f284b | 429 | /* Search the snapshot */ |
c142442b | 430 | snapshot_index = find_snapshot_by_id_or_name(bs, snapshot_id); |
589f284b | 431 | if (snapshot_index < 0) { |
c142442b | 432 | return -ENOENT; |
589f284b | 433 | } |
c142442b KW |
434 | sn = &s->snapshots[snapshot_index]; |
435 | ||
90b27759 KW |
436 | if (sn->disk_size != bs->total_sectors * BDRV_SECTOR_SIZE) { |
437 | error_report("qcow2: Loading snapshots with different disk " | |
438 | "size is not implemented"); | |
439 | ret = -ENOTSUP; | |
440 | goto fail; | |
441 | } | |
442 | ||
589f284b KW |
443 | /* |
444 | * Make sure that the current L1 table is big enough to contain the whole | |
445 | * L1 table of the snapshot. If the snapshot L1 table is smaller, the | |
446 | * current one must be padded with zeros. | |
447 | */ | |
448 | ret = qcow2_grow_l1_table(bs, sn->l1_size, true); | |
449 | if (ret < 0) { | |
c142442b | 450 | goto fail; |
589f284b | 451 | } |
c142442b | 452 | |
35d7ace7 KW |
453 | cur_l1_bytes = s->l1_size * sizeof(uint64_t); |
454 | sn_l1_bytes = sn->l1_size * sizeof(uint64_t); | |
455 | ||
589f284b KW |
456 | /* |
457 | * Copy the snapshot L1 table to the current L1 table. | |
458 | * | |
459 | * Before overwriting the old current L1 table on disk, make sure to | |
460 | * increase all refcounts for the clusters referenced by the new one. | |
43a0cac4 KW |
461 | * Decrease the refcount referenced by the old one only when the L1 |
462 | * table is overwritten. | |
589f284b | 463 | */ |
43a0cac4 KW |
464 | sn_l1_table = g_malloc0(cur_l1_bytes); |
465 | ||
466 | ret = bdrv_pread(bs->file, sn->l1_table_offset, sn_l1_table, sn_l1_bytes); | |
467 | if (ret < 0) { | |
468 | goto fail; | |
469 | } | |
470 | ||
471 | ret = qcow2_update_snapshot_refcount(bs, sn->l1_table_offset, | |
472 | sn->l1_size, 1); | |
589f284b | 473 | if (ret < 0) { |
c142442b | 474 | goto fail; |
589f284b KW |
475 | } |
476 | ||
43a0cac4 | 477 | ret = bdrv_pwrite_sync(bs->file, s->l1_table_offset, sn_l1_table, |
589f284b KW |
478 | cur_l1_bytes); |
479 | if (ret < 0) { | |
c142442b | 480 | goto fail; |
589f284b KW |
481 | } |
482 | ||
43a0cac4 KW |
483 | /* |
484 | * Decrease refcount of clusters of current L1 table. | |
485 | * | |
486 | * At this point, the in-memory s->l1_table points to the old L1 table, | |
487 | * whereas on disk we already have the new one. | |
488 | * | |
489 | * qcow2_update_snapshot_refcount special cases the current L1 table to use | |
490 | * the in-memory data instead of really using the offset to load a new one, | |
491 | * which is why this works. | |
492 | */ | |
493 | ret = qcow2_update_snapshot_refcount(bs, s->l1_table_offset, | |
494 | s->l1_size, -1); | |
495 | ||
496 | /* | |
497 | * Now update the in-memory L1 table to be in sync with the on-disk one. We | |
498 | * need to do this even if updating refcounts failed. | |
499 | */ | |
c142442b | 500 | for(i = 0;i < s->l1_size; i++) { |
43a0cac4 | 501 | s->l1_table[i] = be64_to_cpu(sn_l1_table[i]); |
c142442b KW |
502 | } |
503 | ||
43a0cac4 KW |
504 | if (ret < 0) { |
505 | goto fail; | |
506 | } | |
507 | ||
508 | g_free(sn_l1_table); | |
509 | sn_l1_table = NULL; | |
510 | ||
511 | /* | |
512 | * Update QCOW_OFLAG_COPIED in the active L1 table (it may have changed | |
513 | * when we decreased the refcount of the old snapshot. | |
514 | */ | |
515 | ret = qcow2_update_snapshot_refcount(bs, s->l1_table_offset, s->l1_size, 0); | |
589f284b | 516 | if (ret < 0) { |
c142442b | 517 | goto fail; |
589f284b | 518 | } |
c142442b KW |
519 | |
520 | #ifdef DEBUG_ALLOC | |
6cbc3031 PH |
521 | { |
522 | BdrvCheckResult result = {0}; | |
b35278f7 | 523 | qcow2_check_refcounts(bs, &result, 0); |
6cbc3031 | 524 | } |
c142442b KW |
525 | #endif |
526 | return 0; | |
589f284b KW |
527 | |
528 | fail: | |
43a0cac4 | 529 | g_free(sn_l1_table); |
589f284b | 530 | return ret; |
c142442b KW |
531 | } |
532 | ||
ed6ccf0f | 533 | int qcow2_snapshot_delete(BlockDriverState *bs, const char *snapshot_id) |
c142442b KW |
534 | { |
535 | BDRVQcowState *s = bs->opaque; | |
9a476780 | 536 | QCowSnapshot sn; |
c142442b KW |
537 | int snapshot_index, ret; |
538 | ||
9a476780 | 539 | /* Search the snapshot */ |
c142442b | 540 | snapshot_index = find_snapshot_by_id_or_name(bs, snapshot_id); |
9a476780 | 541 | if (snapshot_index < 0) { |
c142442b | 542 | return -ENOENT; |
9a476780 KW |
543 | } |
544 | sn = s->snapshots[snapshot_index]; | |
c142442b | 545 | |
9a476780 KW |
546 | /* Remove it from the snapshot list */ |
547 | memmove(s->snapshots + snapshot_index, | |
548 | s->snapshots + snapshot_index + 1, | |
549 | (s->nb_snapshots - snapshot_index - 1) * sizeof(sn)); | |
550 | s->nb_snapshots--; | |
551 | ret = qcow2_write_snapshots(bs); | |
552 | if (ret < 0) { | |
c142442b | 553 | return ret; |
9a476780 KW |
554 | } |
555 | ||
556 | /* | |
557 | * The snapshot is now unused, clean up. If we fail after this point, we | |
558 | * won't recover but just leak clusters. | |
559 | */ | |
560 | g_free(sn.id_str); | |
561 | g_free(sn.name); | |
562 | ||
563 | /* | |
564 | * Now decrease the refcounts of clusters referenced by the snapshot and | |
565 | * free the L1 table. | |
566 | */ | |
567 | ret = qcow2_update_snapshot_refcount(bs, sn.l1_table_offset, | |
568 | sn.l1_size, -1); | |
569 | if (ret < 0) { | |
c142442b | 570 | return ret; |
9a476780 KW |
571 | } |
572 | qcow2_free_clusters(bs, sn.l1_table_offset, sn.l1_size * sizeof(uint64_t)); | |
c142442b | 573 | |
9a476780 KW |
574 | /* must update the copied flag on the current cluster offsets */ |
575 | ret = qcow2_update_snapshot_refcount(bs, s->l1_table_offset, s->l1_size, 0); | |
c142442b | 576 | if (ret < 0) { |
c142442b KW |
577 | return ret; |
578 | } | |
9a476780 | 579 | |
c142442b | 580 | #ifdef DEBUG_ALLOC |
6cbc3031 PH |
581 | { |
582 | BdrvCheckResult result = {0}; | |
b35278f7 | 583 | qcow2_check_refcounts(bs, &result, 0); |
6cbc3031 | 584 | } |
c142442b KW |
585 | #endif |
586 | return 0; | |
587 | } | |
588 | ||
ed6ccf0f | 589 | int qcow2_snapshot_list(BlockDriverState *bs, QEMUSnapshotInfo **psn_tab) |
c142442b KW |
590 | { |
591 | BDRVQcowState *s = bs->opaque; | |
592 | QEMUSnapshotInfo *sn_tab, *sn_info; | |
593 | QCowSnapshot *sn; | |
594 | int i; | |
595 | ||
596 | if (!s->nb_snapshots) { | |
597 | *psn_tab = NULL; | |
598 | return s->nb_snapshots; | |
599 | } | |
600 | ||
7267c094 | 601 | sn_tab = g_malloc0(s->nb_snapshots * sizeof(QEMUSnapshotInfo)); |
c142442b KW |
602 | for(i = 0; i < s->nb_snapshots; i++) { |
603 | sn_info = sn_tab + i; | |
604 | sn = s->snapshots + i; | |
605 | pstrcpy(sn_info->id_str, sizeof(sn_info->id_str), | |
606 | sn->id_str); | |
607 | pstrcpy(sn_info->name, sizeof(sn_info->name), | |
608 | sn->name); | |
609 | sn_info->vm_state_size = sn->vm_state_size; | |
610 | sn_info->date_sec = sn->date_sec; | |
611 | sn_info->date_nsec = sn->date_nsec; | |
612 | sn_info->vm_clock_nsec = sn->vm_clock_nsec; | |
613 | } | |
614 | *psn_tab = sn_tab; | |
615 | return s->nb_snapshots; | |
616 | } | |
617 | ||
51ef6727 | 618 | int qcow2_snapshot_load_tmp(BlockDriverState *bs, const char *snapshot_name) |
619 | { | |
e3f652b3 | 620 | int i, snapshot_index; |
51ef6727 | 621 | BDRVQcowState *s = bs->opaque; |
622 | QCowSnapshot *sn; | |
e3f652b3 KW |
623 | uint64_t *new_l1_table; |
624 | int new_l1_bytes; | |
625 | int ret; | |
51ef6727 | 626 | |
e3f652b3 KW |
627 | assert(bs->read_only); |
628 | ||
629 | /* Search the snapshot */ | |
51ef6727 | 630 | snapshot_index = find_snapshot_by_id_or_name(bs, snapshot_name); |
631 | if (snapshot_index < 0) { | |
632 | return -ENOENT; | |
633 | } | |
51ef6727 | 634 | sn = &s->snapshots[snapshot_index]; |
51ef6727 | 635 | |
e3f652b3 KW |
636 | /* Allocate and read in the snapshot's L1 table */ |
637 | new_l1_bytes = s->l1_size * sizeof(uint64_t); | |
638 | new_l1_table = g_malloc0(align_offset(new_l1_bytes, 512)); | |
51ef6727 | 639 | |
e3f652b3 KW |
640 | ret = bdrv_pread(bs->file, sn->l1_table_offset, new_l1_table, new_l1_bytes); |
641 | if (ret < 0) { | |
642 | g_free(new_l1_table); | |
643 | return ret; | |
51ef6727 | 644 | } |
645 | ||
e3f652b3 KW |
646 | /* Switch the L1 table */ |
647 | g_free(s->l1_table); | |
648 | ||
649 | s->l1_size = sn->l1_size; | |
650 | s->l1_table_offset = sn->l1_table_offset; | |
651 | s->l1_table = new_l1_table; | |
652 | ||
51ef6727 | 653 | for(i = 0;i < s->l1_size; i++) { |
654 | be64_to_cpus(&s->l1_table[i]); | |
655 | } | |
e3f652b3 | 656 | |
51ef6727 | 657 | return 0; |
658 | } |