1
2
3
4
5
6
7
8
9
10
11
12
13
14
15#include "qemu/osdep.h"
16#include "qemu/cutils.h"
17#include "trace.h"
18#include "block/block_int.h"
19#include "block/blockjob_int.h"
20#include "qapi/error.h"
21#include "qapi/qmp/qerror.h"
22#include "qemu/ratelimit.h"
23#include "sysemu/block-backend.h"
24
25enum {
26
27
28
29
30
31 COMMIT_BUFFER_SIZE = 512 * 1024,
32};
33
34typedef struct CommitBlockJob {
35 BlockJob common;
36 BlockDriverState *commit_top_bs;
37 BlockBackend *top;
38 BlockBackend *base;
39 BlockDriverState *base_bs;
40 BlockdevOnError on_error;
41 bool base_read_only;
42 bool chain_frozen;
43 char *backing_file_str;
44} CommitBlockJob;
45
46static int coroutine_fn commit_populate(BlockBackend *bs, BlockBackend *base,
47 int64_t offset, uint64_t bytes,
48 void *buf)
49{
50 int ret = 0;
51
52 assert(bytes < SIZE_MAX);
53
54 ret = blk_co_pread(bs, offset, bytes, buf, 0);
55 if (ret < 0) {
56 return ret;
57 }
58
59 ret = blk_co_pwrite(base, offset, bytes, buf, 0);
60 if (ret < 0) {
61 return ret;
62 }
63
64 return 0;
65}
66
67static int commit_prepare(Job *job)
68{
69 CommitBlockJob *s = container_of(job, CommitBlockJob, common.job);
70
71 bdrv_unfreeze_backing_chain(s->commit_top_bs, s->base_bs);
72 s->chain_frozen = false;
73
74
75
76 blk_unref(s->base);
77 s->base = NULL;
78
79
80
81 return bdrv_drop_intermediate(s->commit_top_bs, s->base_bs,
82 s->backing_file_str);
83}
84
85static void commit_abort(Job *job)
86{
87 CommitBlockJob *s = container_of(job, CommitBlockJob, common.job);
88 BlockDriverState *top_bs = blk_bs(s->top);
89
90 if (s->chain_frozen) {
91 bdrv_unfreeze_backing_chain(s->commit_top_bs, s->base_bs);
92 }
93
94
95 bdrv_ref(top_bs);
96 bdrv_ref(s->commit_top_bs);
97
98 if (s->base) {
99 blk_unref(s->base);
100 }
101
102
103
104 block_job_remove_all_bdrv(&s->common);
105
106
107
108
109
110
111
112
113 bdrv_replace_node(s->commit_top_bs, backing_bs(s->commit_top_bs),
114 &error_abort);
115
116 bdrv_unref(s->commit_top_bs);
117 bdrv_unref(top_bs);
118}
119
120static void commit_clean(Job *job)
121{
122 CommitBlockJob *s = container_of(job, CommitBlockJob, common.job);
123
124
125
126
127 if (s->base_read_only) {
128 bdrv_reopen_set_read_only(s->base_bs, true, NULL);
129 }
130
131 g_free(s->backing_file_str);
132 blk_unref(s->top);
133}
134
135static int coroutine_fn commit_run(Job *job, Error **errp)
136{
137 CommitBlockJob *s = container_of(job, CommitBlockJob, common.job);
138 int64_t offset;
139 uint64_t delay_ns = 0;
140 int ret = 0;
141 int64_t n = 0;
142 void *buf = NULL;
143 int bytes_written = 0;
144 int64_t len, base_len;
145
146 ret = len = blk_getlength(s->top);
147 if (len < 0) {
148 goto out;
149 }
150 job_progress_set_remaining(&s->common.job, len);
151
152 ret = base_len = blk_getlength(s->base);
153 if (base_len < 0) {
154 goto out;
155 }
156
157 if (base_len < len) {
158 ret = blk_truncate(s->base, len, PREALLOC_MODE_OFF, NULL);
159 if (ret) {
160 goto out;
161 }
162 }
163
164 buf = blk_blockalign(s->top, COMMIT_BUFFER_SIZE);
165
166 for (offset = 0; offset < len; offset += n) {
167 bool copy;
168
169
170
171
172 job_sleep_ns(&s->common.job, delay_ns);
173 if (job_is_cancelled(&s->common.job)) {
174 break;
175 }
176
177 ret = bdrv_is_allocated_above(blk_bs(s->top), blk_bs(s->base), false,
178 offset, COMMIT_BUFFER_SIZE, &n);
179 copy = (ret == 1);
180 trace_commit_one_iteration(s, offset, n, ret);
181 if (copy) {
182 ret = commit_populate(s->top, s->base, offset, n, buf);
183 bytes_written += n;
184 }
185 if (ret < 0) {
186 BlockErrorAction action =
187 block_job_error_action(&s->common, false, s->on_error, -ret);
188 if (action == BLOCK_ERROR_ACTION_REPORT) {
189 goto out;
190 } else {
191 n = 0;
192 continue;
193 }
194 }
195
196 job_progress_update(&s->common.job, n);
197
198 if (copy) {
199 delay_ns = block_job_ratelimit_get_delay(&s->common, n);
200 } else {
201 delay_ns = 0;
202 }
203 }
204
205 ret = 0;
206
207out:
208 qemu_vfree(buf);
209
210 return ret;
211}
212
213static const BlockJobDriver commit_job_driver = {
214 .job_driver = {
215 .instance_size = sizeof(CommitBlockJob),
216 .job_type = JOB_TYPE_COMMIT,
217 .free = block_job_free,
218 .user_resume = block_job_user_resume,
219 .drain = block_job_drain,
220 .run = commit_run,
221 .prepare = commit_prepare,
222 .abort = commit_abort,
223 .clean = commit_clean
224 },
225};
226
227static int coroutine_fn bdrv_commit_top_preadv(BlockDriverState *bs,
228 uint64_t offset, uint64_t bytes, QEMUIOVector *qiov, int flags)
229{
230 return bdrv_co_preadv(bs->backing, offset, bytes, qiov, flags);
231}
232
233static void bdrv_commit_top_refresh_filename(BlockDriverState *bs)
234{
235 pstrcpy(bs->exact_filename, sizeof(bs->exact_filename),
236 bs->backing->bs->filename);
237}
238
239static void bdrv_commit_top_child_perm(BlockDriverState *bs, BdrvChild *c,
240 const BdrvChildRole *role,
241 BlockReopenQueue *reopen_queue,
242 uint64_t perm, uint64_t shared,
243 uint64_t *nperm, uint64_t *nshared)
244{
245 *nperm = 0;
246 *nshared = BLK_PERM_ALL;
247}
248
249
250
251static BlockDriver bdrv_commit_top = {
252 .format_name = "commit_top",
253 .bdrv_co_preadv = bdrv_commit_top_preadv,
254 .bdrv_co_block_status = bdrv_co_block_status_from_backing,
255 .bdrv_refresh_filename = bdrv_commit_top_refresh_filename,
256 .bdrv_child_perm = bdrv_commit_top_child_perm,
257};
258
259void commit_start(const char *job_id, BlockDriverState *bs,
260 BlockDriverState *base, BlockDriverState *top,
261 int creation_flags, int64_t speed,
262 BlockdevOnError on_error, const char *backing_file_str,
263 const char *filter_node_name, Error **errp)
264{
265 CommitBlockJob *s;
266 BlockDriverState *iter;
267 BlockDriverState *commit_top_bs = NULL;
268 Error *local_err = NULL;
269 int ret;
270
271 assert(top != bs);
272 if (top == base) {
273 error_setg(errp, "Invalid files for merge: top and base are the same");
274 return;
275 }
276
277 s = block_job_create(job_id, &commit_job_driver, NULL, bs, 0, BLK_PERM_ALL,
278 speed, creation_flags, NULL, NULL, errp);
279 if (!s) {
280 return;
281 }
282
283
284 s->base_read_only = bdrv_is_read_only(base);
285 if (s->base_read_only) {
286 if (bdrv_reopen_set_read_only(base, false, errp) != 0) {
287 goto fail;
288 }
289 }
290
291
292
293 commit_top_bs = bdrv_new_open_driver(&bdrv_commit_top, filter_node_name, 0,
294 errp);
295 if (commit_top_bs == NULL) {
296 goto fail;
297 }
298 if (!filter_node_name) {
299 commit_top_bs->implicit = true;
300 }
301
302
303 commit_top_bs->never_freeze = true;
304
305 commit_top_bs->total_sectors = top->total_sectors;
306
307 bdrv_append(commit_top_bs, top, &local_err);
308 if (local_err) {
309 commit_top_bs = NULL;
310 error_propagate(errp, local_err);
311 goto fail;
312 }
313
314 s->commit_top_bs = commit_top_bs;
315
316
317
318 assert(bdrv_chain_contains(top, base));
319 for (iter = top; iter != base; iter = backing_bs(iter)) {
320
321
322
323
324 ret = block_job_add_bdrv(&s->common, "intermediate node", iter, 0,
325 BLK_PERM_WRITE_UNCHANGED | BLK_PERM_WRITE,
326 errp);
327 if (ret < 0) {
328 goto fail;
329 }
330 }
331
332 if (bdrv_freeze_backing_chain(commit_top_bs, base, errp) < 0) {
333 goto fail;
334 }
335 s->chain_frozen = true;
336
337 ret = block_job_add_bdrv(&s->common, "base", base, 0, BLK_PERM_ALL, errp);
338 if (ret < 0) {
339 goto fail;
340 }
341
342 s->base = blk_new(s->common.job.aio_context,
343 BLK_PERM_CONSISTENT_READ
344 | BLK_PERM_WRITE
345 | BLK_PERM_RESIZE,
346 BLK_PERM_CONSISTENT_READ
347 | BLK_PERM_GRAPH_MOD
348 | BLK_PERM_WRITE_UNCHANGED);
349 ret = blk_insert_bs(s->base, base, errp);
350 if (ret < 0) {
351 goto fail;
352 }
353 s->base_bs = base;
354
355
356 s->top = blk_new(s->common.job.aio_context, 0, BLK_PERM_ALL);
357 ret = blk_insert_bs(s->top, top, errp);
358 if (ret < 0) {
359 goto fail;
360 }
361
362 s->backing_file_str = g_strdup(backing_file_str);
363 s->on_error = on_error;
364
365 trace_commit_start(bs, base, top, s);
366 job_start(&s->common.job);
367 return;
368
369fail:
370 if (s->chain_frozen) {
371 bdrv_unfreeze_backing_chain(commit_top_bs, base);
372 }
373 if (s->base) {
374 blk_unref(s->base);
375 }
376 if (s->top) {
377 blk_unref(s->top);
378 }
379 if (s->base_read_only) {
380 bdrv_reopen_set_read_only(base, true, NULL);
381 }
382 job_early_fail(&s->common.job);
383
384
385 if (commit_top_bs) {
386 bdrv_replace_node(commit_top_bs, top, &error_abort);
387 }
388}
389
390
391#define COMMIT_BUF_SIZE (2048 * BDRV_SECTOR_SIZE)
392
393
394int bdrv_commit(BlockDriverState *bs)
395{
396 BlockBackend *src, *backing;
397 BlockDriverState *backing_file_bs = NULL;
398 BlockDriverState *commit_top_bs = NULL;
399 BlockDriver *drv = bs->drv;
400 AioContext *ctx;
401 int64_t offset, length, backing_length;
402 int ro;
403 int64_t n;
404 int ret = 0;
405 uint8_t *buf = NULL;
406 Error *local_err = NULL;
407
408 if (!drv)
409 return -ENOMEDIUM;
410
411 if (!bs->backing) {
412 return -ENOTSUP;
413 }
414
415 if (bdrv_op_is_blocked(bs, BLOCK_OP_TYPE_COMMIT_SOURCE, NULL) ||
416 bdrv_op_is_blocked(bs->backing->bs, BLOCK_OP_TYPE_COMMIT_TARGET, NULL)) {
417 return -EBUSY;
418 }
419
420 ro = bs->backing->bs->read_only;
421
422 if (ro) {
423 if (bdrv_reopen_set_read_only(bs->backing->bs, false, NULL)) {
424 return -EACCES;
425 }
426 }
427
428 ctx = bdrv_get_aio_context(bs);
429 src = blk_new(ctx, BLK_PERM_CONSISTENT_READ, BLK_PERM_ALL);
430 backing = blk_new(ctx, BLK_PERM_WRITE | BLK_PERM_RESIZE, BLK_PERM_ALL);
431
432 ret = blk_insert_bs(src, bs, &local_err);
433 if (ret < 0) {
434 error_report_err(local_err);
435 goto ro_cleanup;
436 }
437
438
439 backing_file_bs = backing_bs(bs);
440
441 commit_top_bs = bdrv_new_open_driver(&bdrv_commit_top, NULL, BDRV_O_RDWR,
442 &local_err);
443 if (commit_top_bs == NULL) {
444 error_report_err(local_err);
445 goto ro_cleanup;
446 }
447
448 bdrv_set_backing_hd(commit_top_bs, backing_file_bs, &error_abort);
449 bdrv_set_backing_hd(bs, commit_top_bs, &error_abort);
450
451 ret = blk_insert_bs(backing, backing_file_bs, &local_err);
452 if (ret < 0) {
453 error_report_err(local_err);
454 goto ro_cleanup;
455 }
456
457 length = blk_getlength(src);
458 if (length < 0) {
459 ret = length;
460 goto ro_cleanup;
461 }
462
463 backing_length = blk_getlength(backing);
464 if (backing_length < 0) {
465 ret = backing_length;
466 goto ro_cleanup;
467 }
468
469
470
471
472 if (length > backing_length) {
473 ret = blk_truncate(backing, length, PREALLOC_MODE_OFF, &local_err);
474 if (ret < 0) {
475 error_report_err(local_err);
476 goto ro_cleanup;
477 }
478 }
479
480
481
482 buf = blk_try_blockalign(src, COMMIT_BUF_SIZE);
483 if (buf == NULL) {
484 ret = -ENOMEM;
485 goto ro_cleanup;
486 }
487
488 for (offset = 0; offset < length; offset += n) {
489 ret = bdrv_is_allocated(bs, offset, COMMIT_BUF_SIZE, &n);
490 if (ret < 0) {
491 goto ro_cleanup;
492 }
493 if (ret) {
494 ret = blk_pread(src, offset, buf, n);
495 if (ret < 0) {
496 goto ro_cleanup;
497 }
498
499 ret = blk_pwrite(backing, offset, buf, n, 0);
500 if (ret < 0) {
501 goto ro_cleanup;
502 }
503 }
504 }
505
506 if (drv->bdrv_make_empty) {
507 ret = drv->bdrv_make_empty(bs);
508 if (ret < 0) {
509 goto ro_cleanup;
510 }
511 blk_flush(src);
512 }
513
514
515
516
517
518 blk_flush(backing);
519
520 ret = 0;
521ro_cleanup:
522 qemu_vfree(buf);
523
524 blk_unref(backing);
525 if (backing_file_bs) {
526 bdrv_set_backing_hd(bs, backing_file_bs, &error_abort);
527 }
528 bdrv_unref(commit_top_bs);
529 blk_unref(src);
530
531 if (ro) {
532
533 bdrv_reopen_set_read_only(bs->backing->bs, true, NULL);
534 }
535
536 return ret;
537}
538