storage_backend.c 68.4 KB
Newer Older
1
/*
2
 * storage_backend.c: internal storage driver backend contract
3
 *
4
 * Copyright (C) 2007-2016 Red Hat, Inc.
5 6 7 8 9 10 11 12 13 14 15 16 17
 * Copyright (C) 2007-2008 Daniel P. Berrange
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
18
 * License along with this library.  If not, see
O
Osier Yang 已提交
19
 * <http://www.gnu.org/licenses/>.
20 21 22 23 24 25 26
 *
 * Author: Daniel P. Berrange <berrange@redhat.com>
 */

#include <config.h>

#include <string.h>
27
#include <stdio.h>
28
#include <regex.h>
29
#include <sys/types.h>
E
Eric Blake 已提交
30
#include <sys/wait.h>
31
#include <unistd.h>
32 33
#include <fcntl.h>
#include <sys/stat.h>
34
#include <sys/param.h>
35
#include <dirent.h>
36
#include "dirname.h"
37 38 39
#ifdef __linux__
# include <sys/ioctl.h>
# include <linux/fs.h>
C
Chunyan Liu 已提交
40 41 42
# ifndef FS_NOCOW_FL
#  define FS_NOCOW_FL                     0x00800000 /* Do not cow file */
# endif
43
#endif
44

45
#if WITH_SELINUX
46
# include <selinux/selinux.h>
47
#endif
48

49 50 51 52
#if HAVE_LINUX_BTRFS_H
# include <linux/btrfs.h>
#endif

53
#include "datatypes.h"
54
#include "virerror.h"
55
#include "viralloc.h"
56
#include "internal.h"
57
#include "secret_conf.h"
58
#include "viruuid.h"
59
#include "virstoragefile.h"
D
Daniel P. Berrange 已提交
60
#include "storage_backend.h"
61
#include "virlog.h"
E
Eric Blake 已提交
62
#include "virfile.h"
63
#include "stat-time.h"
64
#include "virstring.h"
65
#include "virxml.h"
66
#include "fdstream.h"
67

68
#if WITH_STORAGE_LVM
69
# include "storage_backend_logical.h"
70 71
#endif
#if WITH_STORAGE_ISCSI
72
# include "storage_backend_iscsi.h"
73
#endif
74
#if WITH_STORAGE_SCSI
75
# include "storage_backend_scsi.h"
76
#endif
D
Dave Allan 已提交
77
#if WITH_STORAGE_MPATH
78
# include "storage_backend_mpath.h"
D
Dave Allan 已提交
79
#endif
80
#if WITH_STORAGE_DISK
81
# include "storage_backend_disk.h"
82 83
#endif
#if WITH_STORAGE_DIR
84
# include "storage_backend_fs.h"
85
#endif
86 87 88
#if WITH_STORAGE_RBD
# include "storage_backend_rbd.h"
#endif
89 90 91
#if WITH_STORAGE_SHEEPDOG
# include "storage_backend_sheepdog.h"
#endif
92 93 94
#if WITH_STORAGE_GLUSTER
# include "storage_backend_gluster.h"
#endif
R
Roman Bogorodskiy 已提交
95 96 97
#if WITH_STORAGE_ZFS
# include "storage_backend_zfs.h"
#endif
98

99 100
#define VIR_FROM_THIS VIR_FROM_STORAGE

101 102
VIR_LOG_INIT("storage.storage_backend");

103 104 105 106 107 108 109 110 111 112 113 114 115 116
static virStorageBackendPtr backends[] = {
#if WITH_STORAGE_DIR
    &virStorageBackendDirectory,
#endif
#if WITH_STORAGE_FS
    &virStorageBackendFileSystem,
    &virStorageBackendNetFileSystem,
#endif
#if WITH_STORAGE_LVM
    &virStorageBackendLogical,
#endif
#if WITH_STORAGE_ISCSI
    &virStorageBackendISCSI,
#endif
117 118 119
#if WITH_STORAGE_SCSI
    &virStorageBackendSCSI,
#endif
D
Dave Allan 已提交
120 121 122
#if WITH_STORAGE_MPATH
    &virStorageBackendMpath,
#endif
123 124
#if WITH_STORAGE_DISK
    &virStorageBackendDisk,
125 126 127
#endif
#if WITH_STORAGE_RBD
    &virStorageBackendRBD,
128 129 130
#endif
#if WITH_STORAGE_SHEEPDOG
    &virStorageBackendSheepdog,
131 132 133
#endif
#if WITH_STORAGE_GLUSTER
    &virStorageBackendGluster,
R
Roman Bogorodskiy 已提交
134 135 136
#endif
#if WITH_STORAGE_ZFS
    &virStorageBackendZFS,
137 138 139
#endif
    NULL
};
140

141 142

static virStorageFileBackendPtr fileBackends[] = {
143 144 145
#if WITH_STORAGE_FS
    &virStorageFileBackendFile,
    &virStorageFileBackendBlock,
146 147 148
#endif
#if WITH_STORAGE_GLUSTER
    &virStorageFileBackendGluster,
149
#endif
150 151 152 153
    NULL
};


154 155 156 157 158 159
enum {
    TOOL_QEMU_IMG,
    TOOL_KVM_IMG,
    TOOL_QCOW_CREATE,
};

160 161 162
#define READ_BLOCK_SIZE_DEFAULT  (1024 * 1024)
#define WRITE_BLOCK_SIZE_DEFAULT (4 * 1024)

163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182
/*
 * Perform the O(1) btrfs clone operation, if possible.
 * Upon success, return 0.  Otherwise, return -1 and set errno.
 */
#if HAVE_LINUX_BTRFS_H
static inline int
btrfsCloneFile(int dest_fd, int src_fd)
{
    return ioctl(dest_fd, BTRFS_IOC_CLONE, src_fd);
}
#else
static inline int
btrfsCloneFile(int dest_fd ATTRIBUTE_UNUSED,
               int src_fd ATTRIBUTE_UNUSED)
{
    errno = ENOTSUP;
    return -1;
}
#endif

183
static int ATTRIBUTE_NONNULL(2)
184
virStorageBackendCopyToFD(virStorageVolDefPtr vol,
185 186
                          virStorageVolDefPtr inputvol,
                          int fd,
187
                          unsigned long long *total,
188 189
                          bool want_sparse,
                          bool reflink_copy)
190 191 192
{
    int inputfd = -1;
    int amtread = -1;
193
    int ret = 0;
194
    size_t rbytes = READ_BLOCK_SIZE_DEFAULT;
L
Li Zhang 已提交
195
    int wbytes = 0;
196
    int interval;
197
    char *zerobuf = NULL;
198
    char *buf = NULL;
199
    struct stat st;
200

201
    if ((inputfd = open(inputvol->target.path, O_RDONLY)) < 0) {
202
        ret = -errno;
203
        virReportSystemError(errno,
204 205 206
                             _("could not open input path '%s'"),
                             inputvol->target.path);
        goto cleanup;
207 208
    }

209
#ifdef __linux__
210
    if (ioctl(fd, BLKBSZGET, &wbytes) < 0)
211 212 213 214 215 216 217 218 219 220 221
        wbytes = 0;
#endif
    if ((wbytes == 0) && fstat(fd, &st) == 0)
        wbytes = st.st_blksize;
    if (wbytes < WRITE_BLOCK_SIZE_DEFAULT)
        wbytes = WRITE_BLOCK_SIZE_DEFAULT;

    if (VIR_ALLOC_N(zerobuf, wbytes) < 0) {
        ret = -errno;
        goto cleanup;
    }
222

223
    if (VIR_ALLOC_N(buf, rbytes) < 0) {
224
        ret = -errno;
225 226 227
        goto cleanup;
    }

228 229 230 231 232 233 234 235 236 237 238 239 240
    if (reflink_copy) {
        if (btrfsCloneFile(fd, inputfd) < 0) {
            ret = -errno;
            virReportSystemError(errno,
                                 _("failed to clone files from '%s'"),
                                 inputvol->target.path);
            goto cleanup;
        } else {
            VIR_DEBUG("btrfs clone finished.");
            goto cleanup;
        }
    }

241 242 243
    while (amtread != 0) {
        int amtleft;

244 245
        if (*total < rbytes)
            rbytes = *total;
246

247
        if ((amtread = saferead(inputfd, buf, rbytes)) < 0) {
248
            ret = -errno;
249
            virReportSystemError(errno,
250 251 252 253
                                 _("failed reading from file '%s'"),
                                 inputvol->target.path);
            goto cleanup;
        }
254
        *total -= amtread;
255 256 257 258 259

        /* Loop over amt read in 512 byte increments, looking for sparse
         * blocks */
        amtleft = amtread;
        do {
260
            interval = ((wbytes > amtleft) ? amtleft : wbytes);
261 262
            int offset = amtread - amtleft;

263
            if (want_sparse && memcmp(buf+offset, zerobuf, interval) == 0) {
264
                if (lseek(fd, interval, SEEK_CUR) < 0) {
265
                    ret = -errno;
266
                    virReportSystemError(errno,
267 268 269 270 271
                                         _("cannot extend file '%s'"),
                                         vol->target.path);
                    goto cleanup;
                }
            } else if (safewrite(fd, buf+offset, interval) < 0) {
272
                ret = -errno;
273
                virReportSystemError(errno,
274 275 276 277 278
                                     _("failed writing to file '%s'"),
                                     vol->target.path);
                goto cleanup;

            }
279
        } while ((amtleft -= interval) > 0);
280 281
    }

282 283 284 285 286 287 288 289
    if (fdatasync(fd) < 0) {
        ret = -errno;
        virReportSystemError(errno, _("cannot sync data to file '%s'"),
                             vol->target.path);
        goto cleanup;
    }


290
    if (VIR_CLOSE(inputfd) < 0) {
291
        ret = -errno;
292
        virReportSystemError(errno,
293 294 295 296 297 298
                             _("cannot close file '%s'"),
                             inputvol->target.path);
        goto cleanup;
    }
    inputfd = -1;

299
 cleanup:
300
    VIR_FORCE_CLOSE(inputfd);
301

302
    VIR_FREE(zerobuf);
303 304
    VIR_FREE(buf);

305 306 307
    return ret;
}

308
static int
309
virStorageBackendCreateBlockFrom(virConnectPtr conn ATTRIBUTE_UNUSED,
310
                                 virStoragePoolObjPtr pool ATTRIBUTE_UNUSED,
311 312
                                 virStorageVolDefPtr vol,
                                 virStorageVolDefPtr inputvol,
E
Eric Blake 已提交
313
                                 unsigned int flags)
314 315 316 317
{
    int fd = -1;
    int ret = -1;
    unsigned long long remain;
318 319 320
    struct stat st;
    gid_t gid;
    uid_t uid;
321
    mode_t mode;
322
    bool reflink_copy = false;
323

324 325 326
    virCheckFlags(VIR_STORAGE_VOL_CREATE_PREALLOC_METADATA |
                  VIR_STORAGE_VOL_CREATE_REFLINK,
                  -1);
327 328 329 330 331 332 333

    if (flags & VIR_STORAGE_VOL_CREATE_PREALLOC_METADATA) {
        virReportError(VIR_ERR_CONFIG_UNSUPPORTED, "%s",
                       _("metadata preallocation is not supported for block "
                         "volumes"));
        goto cleanup;
    }
E
Eric Blake 已提交
334

335 336 337
    if (flags & VIR_STORAGE_VOL_CREATE_REFLINK)
        reflink_copy = true;

338
    if ((fd = open(vol->target.path, O_RDWR)) < 0) {
339
        virReportSystemError(errno,
340 341 342 343 344
                             _("cannot create path '%s'"),
                             vol->target.path);
        goto cleanup;
    }

345
    remain = vol->target.capacity;
346 347

    if (inputvol) {
348 349
        if (virStorageBackendCopyToFD(vol, inputvol, fd, &remain,
                                      false, reflink_copy) < 0)
350 351 352
            goto cleanup;
    }

353
    if (fstat(fd, &st) == -1) {
354
        virReportSystemError(errno, _("stat of '%s' failed"),
355 356 357
                             vol->target.path);
        goto cleanup;
    }
358 359 360 361
    uid = (vol->target.perms->uid != st.st_uid) ? vol->target.perms->uid
        : (uid_t) -1;
    gid = (vol->target.perms->gid != st.st_gid) ? vol->target.perms->gid
        : (gid_t) -1;
P
Philipp Hahn 已提交
362
    if (((uid != (uid_t) -1) || (gid != (gid_t) -1))
363
        && (fchown(fd, uid, gid) < 0)) {
364
        virReportSystemError(errno,
365
                             _("cannot chown '%s' to (%u, %u)"),
366 367
                             vol->target.path, (unsigned int) uid,
                             (unsigned int) gid);
368 369
        goto cleanup;
    }
370 371 372 373

    mode = (vol->target.perms->mode == (mode_t) -1 ?
            VIR_STORAGE_DEFAULT_VOL_PERM_MODE : vol->target.perms->mode);
    if (fchmod(fd, mode) < 0) {
374
        virReportSystemError(errno,
375
                             _("cannot set mode of '%s' to %04o"),
376
                             vol->target.path, mode);
377 378
        goto cleanup;
    }
379
    if (VIR_CLOSE(fd) < 0) {
380
        virReportSystemError(errno,
381 382 383 384 385 386 387
                             _("cannot close file '%s'"),
                             vol->target.path);
        goto cleanup;
    }
    fd = -1;

    ret = 0;
388
 cleanup:
389
    VIR_FORCE_CLOSE(fd);
390 391 392 393

    return ret;
}

E
Eric Blake 已提交
394 395
static int
createRawFile(int fd, virStorageVolDefPtr vol,
396 397
              virStorageVolDefPtr inputvol,
              bool reflink_copy)
E
Eric Blake 已提交
398
{
J
Ján Tomko 已提交
399
    bool need_alloc = true;
400
    int ret = 0;
401
    unsigned long long pos = 0;
402

403 404 405 406 407 408
    /* If the new allocation is lower than the capacity of the original file,
     * the cloned volume will be sparse */
    if (inputvol &&
        vol->target.allocation < inputvol->target.capacity)
        need_alloc = false;

409 410
    /* Seek to the final size, so the capacity is available upfront
     * for progress reporting */
411
    if (ftruncate(fd, vol->target.capacity) < 0) {
412
        ret = -errno;
413
        virReportSystemError(errno,
414
                             _("cannot extend file '%s'"),
E
Eric Blake 已提交
415
                             vol->target.path);
416 417 418
        goto cleanup;
    }

E
Eric Blake 已提交
419 420
/* Avoid issues with older kernel's <linux/fs.h> namespace pollution. */
#if HAVE_FALLOCATE - 0
421
    /* Try to preallocate all requested disk space, but fall back to
422 423
     * other methods if this fails with ENOSYS or EOPNOTSUPP. If allocation
     * is 0 (or less than 0), then fallocate will fail with EINVAL.
424 425 426 427
     * NOTE: do not use posix_fallocate; posix_fallocate falls back
     * to writing zeroes block by block in case fallocate isn't
     * available, and since we're going to copy data from another
     * file it doesn't make sense to write the file twice. */
428
    if (vol->target.allocation && need_alloc) {
429
        if (fallocate(fd, 0, 0, vol->target.allocation) == 0) {
430 431 432 433 434
            need_alloc = false;
        } else if (errno != ENOSYS && errno != EOPNOTSUPP) {
            ret = -errno;
            virReportSystemError(errno,
                                 _("cannot allocate %llu bytes in file '%s'"),
435
                                 vol->target.allocation, vol->target.path);
436 437
            goto cleanup;
        }
438 439 440
    }
#endif

E
Eric Blake 已提交
441
    if (inputvol) {
442
        unsigned long long remain = inputvol->target.capacity;
443 444 445
        /* allow zero blocks to be skipped if we've requested sparse
         * allocation (allocation < capacity) or we have already
         * been able to allocate the required space. */
446 447
        if ((ret = virStorageBackendCopyToFD(vol, inputvol, fd, &remain,
                                             !need_alloc, reflink_copy)) < 0)
448
            goto cleanup;
449

450 451 452 453
        /* If the new allocation is greater than the original capacity,
         * but fallocate failed, fill the rest with zeroes.
         */
        pos = inputvol->target.capacity - remain;
454 455
    }

456
    if (need_alloc && (vol->target.allocation - pos > 0)) {
457
        if (safezero(fd, pos, vol->target.allocation - pos) < 0) {
458 459 460 461
            ret = -errno;
            virReportSystemError(errno, _("cannot fill file '%s'"),
                                 vol->target.path);
            goto cleanup;
462
        }
463 464 465
    }

    if (fsync(fd) < 0) {
466
        ret = -errno;
467
        virReportSystemError(errno, _("cannot sync data to file '%s'"),
E
Eric Blake 已提交
468
                             vol->target.path);
469
        goto cleanup;
470 471
    }

472
 cleanup:
473 474 475 476 477 478 479 480
    return ret;
}

int
virStorageBackendCreateRaw(virConnectPtr conn ATTRIBUTE_UNUSED,
                           virStoragePoolObjPtr pool,
                           virStorageVolDefPtr vol,
                           virStorageVolDefPtr inputvol,
E
Eric Blake 已提交
481
                           unsigned int flags)
482 483
{
    int ret = -1;
E
Eric Blake 已提交
484 485
    int fd = -1;
    int operation_flags;
486
    bool reflink_copy = false;
J
John Ferlan 已提交
487
    mode_t open_mode = VIR_STORAGE_DEFAULT_VOL_PERM_MODE;
488
    bool created = false;
E
Eric Blake 已提交
489

490 491 492
    virCheckFlags(VIR_STORAGE_VOL_CREATE_PREALLOC_METADATA |
                  VIR_STORAGE_VOL_CREATE_REFLINK,
                  -1);
493 494 495 496 497 498 499

    if (flags & VIR_STORAGE_VOL_CREATE_PREALLOC_METADATA) {
        virReportError(VIR_ERR_CONFIG_UNSUPPORTED, "%s",
                       _("metadata preallocation is not supported for raw "
                         "volumes"));
        goto cleanup;
    }
500

501 502 503 504 505 506
    if (vol->target.backingStore) {
        virReportError(VIR_ERR_NO_SUPPORT, "%s",
                       _("backing storage not supported for raw volumes"));
        goto cleanup;
    }

507 508 509 510
    if (flags & VIR_STORAGE_VOL_CREATE_REFLINK)
        reflink_copy = true;


511
    if (vol->target.encryption != NULL) {
512 513
        virReportError(VIR_ERR_CONFIG_UNSUPPORTED, "%s",
                       _("storage pool does not support encrypted volumes"));
514 515
        goto cleanup;
    }
516

L
Laine Stump 已提交
517
    operation_flags = VIR_FILE_OPEN_FORCE_MODE | VIR_FILE_OPEN_FORCE_OWNER;
E
Eric Blake 已提交
518
    if (pool->def->type == VIR_STORAGE_POOL_NETFS)
L
Laine Stump 已提交
519
        operation_flags |= VIR_FILE_OPEN_FORK;
E
Eric Blake 已提交
520

J
John Ferlan 已提交
521 522 523
    if (vol->target.perms->mode != (mode_t) -1)
        open_mode = vol->target.perms->mode;

524 525
    if ((fd = virFileOpenAs(vol->target.path,
                            O_RDWR | O_CREAT | O_EXCL,
J
John Ferlan 已提交
526
                            open_mode,
527 528
                            vol->target.perms->uid,
                            vol->target.perms->gid,
529
                            operation_flags)) < 0) {
E
Eric Blake 已提交
530
        virReportSystemError(-fd,
531
                             _("Failed to create file '%s'"),
E
Eric Blake 已提交
532 533 534
                             vol->target.path);
        goto cleanup;
    }
535
    created = true;
E
Eric Blake 已提交
536

C
Chunyan Liu 已提交
537 538 539 540 541 542
    if (vol->target.nocow) {
#ifdef __linux__
        int attr;

        /* Set NOCOW flag. This is an optimisation for btrfs.
         * The FS_IOC_SETFLAGS ioctl return value will be ignored since any
543
         * failure of this operation should not block the volume creation.
C
Chunyan Liu 已提交
544
         */
545 546 547
        if (ioctl(fd, FS_IOC_GETFLAGS, &attr) < 0) {
            virReportSystemError(errno, "%s", _("Failed to get fs flags"));
        } else {
C
Chunyan Liu 已提交
548
            attr |= FS_NOCOW_FL;
549 550 551 552
            if (ioctl(fd, FS_IOC_SETFLAGS, &attr) < 0) {
                virReportSystemError(errno, "%s",
                                     _("Failed to set NOCOW flag"));
            }
C
Chunyan Liu 已提交
553 554 555 556
        }
#endif
    }

557
    if ((ret = createRawFile(fd, vol, inputvol, reflink_copy)) < 0)
558
        /* createRawFile already reported the exact error. */
E
Eric Blake 已提交
559
        ret = -1;
560

561
 cleanup:
562 563 564 565
    if (ret < 0 && created)
        ignore_value(virFileRemove(vol->target.path,
                                   vol->target.perms->uid,
                                   vol->target.perms->gid));
E
Eric Blake 已提交
566
    VIR_FORCE_CLOSE(fd);
567 568 569
    return ret;
}

570 571 572 573 574 575 576 577 578
static int
virStorageGenerateSecretUUID(virConnectPtr conn,
                             unsigned char *uuid)
{
    unsigned attempt;

    for (attempt = 0; attempt < 65536; attempt++) {
        virSecretPtr tmp;
        if (virUUIDGenerate(uuid) < 0) {
579 580
            virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                           _("unable to generate uuid"));
581 582
            return -1;
        }
583
        tmp = conn->secretDriver->secretLookupByUUID(conn, uuid);
584 585 586
        if (tmp == NULL)
            return 0;

587
        virObjectUnref(tmp);
588 589
    }

590
    virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
N
Nehal J Wani 已提交
591
                   _("too many conflicts when generating a uuid"));
592 593 594 595

    return -1;
}

596 597 598 599 600 601 602 603 604
static int
virStorageGenerateQcowEncryption(virConnectPtr conn,
                                 virStorageVolDefPtr vol)
{
    virSecretDefPtr def = NULL;
    virBuffer buf = VIR_BUFFER_INITIALIZER;
    virStorageEncryptionPtr enc;
    virStorageEncryptionSecretPtr enc_secret = NULL;
    virSecretPtr secret = NULL;
605
    char *xml;
606 607 608
    unsigned char value[VIR_STORAGE_QCOW_PASSPHRASE_SIZE];
    int ret = -1;

609
    if (conn->secretDriver == NULL ||
610 611 612
        conn->secretDriver->secretLookupByUUID == NULL ||
        conn->secretDriver->secretDefineXML == NULL ||
        conn->secretDriver->secretSetValue == NULL) {
613 614
        virReportError(VIR_ERR_CONFIG_UNSUPPORTED, "%s",
                       _("secret storage not supported"));
615 616 617 618 619
        goto cleanup;
    }

    enc = vol->target.encryption;
    if (enc->nsecrets != 0) {
620 621
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("secrets already defined"));
622 623 624 625
        goto cleanup;
    }

    if (VIR_ALLOC(enc_secret) < 0 || VIR_REALLOC_N(enc->secrets, 1) < 0 ||
626
        VIR_ALLOC(def) < 0)
627 628
        goto cleanup;

629
    def->ephemeral = false;
630
    def->private = false;
631
    if (virStorageGenerateSecretUUID(conn, def->uuid) < 0)
632
        goto cleanup;
633

634
    def->usage_type = VIR_SECRET_USAGE_TYPE_VOLUME;
635
    if (VIR_STRDUP(def->usage.volume, vol->target.path) < 0)
636
        goto cleanup;
637
    xml = virSecretDefFormat(def);
638 639 640 641 642
    virSecretDefFree(def);
    def = NULL;
    if (xml == NULL)
        goto cleanup;

643
    secret = conn->secretDriver->secretDefineXML(conn, xml, 0);
644 645 646 647 648 649
    if (secret == NULL) {
        VIR_FREE(xml);
        goto cleanup;
    }
    VIR_FREE(xml);

650
    if (virStorageGenerateQcowPassphrase(value) < 0)
651 652
        goto cleanup;

653
    if (conn->secretDriver->secretSetValue(secret, value, sizeof(value), 0) < 0)
654 655 656
        goto cleanup;

    enc_secret->type = VIR_STORAGE_ENCRYPTION_SECRET_TYPE_PASSPHRASE;
657
    memcpy(enc_secret->uuid, secret->uuid, VIR_UUID_BUFLEN);
658 659 660 661 662 663 664
    enc->format = VIR_STORAGE_ENCRYPTION_FORMAT_QCOW;
    enc->secrets[0] = enc_secret; /* Space for secrets[0] allocated above */
    enc_secret = NULL;
    enc->nsecrets = 1;

    ret = 0;

665
 cleanup:
666
    if (secret != NULL) {
667
        if (ret != 0 &&
668 669
            conn->secretDriver->secretUndefine != NULL)
            conn->secretDriver->secretUndefine(secret);
670
        virObjectUnref(secret);
671
    }
672
    virBufferFreeAndReset(&buf);
673 674 675 676 677
    virSecretDefFree(def);
    VIR_FREE(enc_secret);
    return ret;
}

678 679 680 681 682
static int
virStorageBackendCreateExecCommand(virStoragePoolObjPtr pool,
                                   virStorageVolDefPtr vol,
                                   virCommandPtr cmd)
{
683 684 685
    struct stat st;
    gid_t gid;
    uid_t uid;
686 687 688
    mode_t mode = (vol->target.perms->mode == (mode_t) -1 ?
                   VIR_STORAGE_DEFAULT_VOL_PERM_MODE :
                   vol->target.perms->mode);
689
    bool filecreated = false;
690
    int ret = -1;
691 692

    if ((pool->def->type == VIR_STORAGE_POOL_NETFS)
693
        && (((geteuid() == 0)
694 695 696 697
             && (vol->target.perms->uid != (uid_t) -1)
             && (vol->target.perms->uid != 0))
            || ((vol->target.perms->gid != (gid_t) -1)
                && (vol->target.perms->gid != getegid())))) {
698

699 700
        virCommandSetUID(cmd, vol->target.perms->uid);
        virCommandSetGID(cmd, vol->target.perms->gid);
701
        virCommandSetUmask(cmd, S_IRWXUGO ^ mode);
702 703

        if (virCommandRun(cmd, NULL) == 0) {
704
            /* command was successfully run, check if the file was created */
705
            if (stat(vol->target.path, &st) >= 0) {
706
                filecreated = true;
707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726

                /* seems qemu-img disregards umask and open/creates using 0644.
                 * If that doesn't match what we expect, then let's try to
                 * re-open the file and attempt to force the mode change.
                 */
                if (mode != (st.st_mode & S_IRWXUGO)) {
                    int fd = -1;
                    int flags = VIR_FILE_OPEN_FORK | VIR_FILE_OPEN_FORCE_MODE;

                    if ((fd = virFileOpenAs(vol->target.path, O_RDWR, mode,
                                            vol->target.perms->uid,
                                            vol->target.perms->gid,
                                            flags)) >= 0) {
                        /* Success - means we're good */
                        VIR_FORCE_CLOSE(fd);
                        ret = 0;
                        goto cleanup;
                    }
                }
            }
727 728
        }
    }
729

730
    if (!filecreated) {
731 732 733 734 735
        /* don't change uid/gid/mode if we retry */
        virCommandSetUID(cmd, -1);
        virCommandSetGID(cmd, -1);
        virCommandSetUmask(cmd, 0);

736
        if (virCommandRun(cmd, NULL) < 0)
737
            goto cleanup;
738
        if (stat(vol->target.path, &st) < 0) {
739
            virReportSystemError(errno,
740
                                 _("failed to create %s"), vol->target.path);
741
            goto cleanup;
742
        }
743
        filecreated = true;
744 745
    }

746 747 748 749
    uid = (vol->target.perms->uid != st.st_uid) ? vol->target.perms->uid
        : (uid_t) -1;
    gid = (vol->target.perms->gid != st.st_gid) ? vol->target.perms->gid
        : (gid_t) -1;
P
Philipp Hahn 已提交
750
    if (((uid != (uid_t) -1) || (gid != (gid_t) -1))
751
        && (chown(vol->target.path, uid, gid) < 0)) {
752
        virReportSystemError(errno,
753
                             _("cannot chown %s to (%u, %u)"),
754 755
                             vol->target.path, (unsigned int) uid,
                             (unsigned int) gid);
756
        goto cleanup;
757
    }
758

759 760
    if (mode != (st.st_mode & S_IRWXUGO) &&
        chmod(vol->target.path, mode) < 0) {
761
        virReportSystemError(errno,
762
                             _("cannot set mode of '%s' to %04o"),
763
                             vol->target.path, mode);
764
        goto cleanup;
765
    }
766 767 768 769

    ret = 0;

 cleanup:
770 771 772
    if (ret < 0 && filecreated)
        virFileRemove(vol->target.path, vol->target.perms->uid,
                      vol->target.perms->gid);
773
    return ret;
774 775
}

776 777 778 779
enum {
    QEMU_IMG_BACKING_FORMAT_NONE = 0,
    QEMU_IMG_BACKING_FORMAT_FLAG,
    QEMU_IMG_BACKING_FORMAT_OPTIONS,
780
    QEMU_IMG_BACKING_FORMAT_OPTIONS_COMPAT,
781 782
};

783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801
static bool
virStorageBackendQemuImgSupportsCompat(const char *qemuimg)
{
    bool ret = false;
    char *output;
    virCommandPtr cmd = NULL;

    cmd = virCommandNewArgList(qemuimg, "create", "-o", "?", "-f", "qcow2",
                               "/dev/null", NULL);

    virCommandAddEnvString(cmd, "LC_ALL=C");
    virCommandSetOutputBuffer(cmd, &output);

    if (virCommandRun(cmd, NULL) < 0)
        goto cleanup;

    if (strstr(output, "\ncompat "))
        ret = true;

802
 cleanup:
803 804 805 806 807
    virCommandFree(cmd);
    VIR_FREE(output);
    return ret;
}

808 809
static int
virStorageBackendQEMUImgBackingFormat(const char *qemuimg)
810 811 812 813 814 815
{
    char *help = NULL;
    char *start;
    char *end;
    char *tmp;
    int ret = -1;
816
    int exitstatus;
817
    virCommandPtr cmd = virCommandNewArgList(qemuimg, "-h", NULL);
818

819 820 821
    virCommandAddEnvString(cmd, "LC_ALL=C");
    virCommandSetOutputBuffer(cmd, &help);
    virCommandClearCaps(cmd);
822

823 824 825
    /* qemuimg doesn't return zero exit status on -h,
     * therefore we need to provide pointer for storing
     * exit status, although we don't parse it any later */
826
    if (virCommandRun(cmd, &exitstatus) < 0)
827 828
        goto cleanup;

829 830
    if ((start = strstr(help, " create ")) == NULL ||
        (end = strstr(start, "\n")) == NULL) {
831 832 833
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("unable to parse qemu-img output '%s'"),
                       help);
834 835
        goto cleanup;
    }
836
    if (((tmp = strstr(start, "-F fmt")) && tmp < end) ||
837
        ((tmp = strstr(start, "-F backing_fmt")) && tmp < end)) {
838
        ret = QEMU_IMG_BACKING_FORMAT_FLAG;
839 840 841 842 843 844
    } else if ((tmp = strstr(start, "[-o options]")) && tmp < end) {
        if (virStorageBackendQemuImgSupportsCompat(qemuimg))
            ret = QEMU_IMG_BACKING_FORMAT_OPTIONS_COMPAT;
        else
            ret = QEMU_IMG_BACKING_FORMAT_OPTIONS;
    } else {
845
        ret = QEMU_IMG_BACKING_FORMAT_NONE;
846
    }
847

848
 cleanup:
849
    virCommandFree(cmd);
850 851 852 853
    VIR_FREE(help);
    return ret;
}

854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870
struct _virStorageBackendQemuImgInfo {
    int format;
    const char *path;
    unsigned long long size_arg;
    bool encryption;
    bool preallocate;
    const char *compat;
    virBitmapPtr features;
    bool nocow;

    const char *backingPath;
    int backingFormat;

    const char *inputPath;
    int inputFormat;
};

871 872
static int
virStorageBackendCreateQemuImgOpts(char **opts,
873
                                   struct _virStorageBackendQemuImgInfo info)
874 875
{
    virBuffer buf = VIR_BUFFER_INITIALIZER;
876

877 878 879 880
    if (info.backingPath)
        virBufferAsprintf(&buf, "backing_fmt=%s,",
                          virStorageFileFormatTypeToString(info.backingFormat));
    if (info.encryption)
881
        virBufferAddLit(&buf, "encryption=on,");
882
    if (info.preallocate)
883
        virBufferAddLit(&buf, "preallocation=metadata,");
884
    if (info.nocow)
C
Chunyan Liu 已提交
885
        virBufferAddLit(&buf, "nocow=on,");
886

887 888
    if (info.compat)
        virBufferAsprintf(&buf, "compat=%s,", info.compat);
J
Ján Tomko 已提交
889

890
    if (info.features && info.format == VIR_STORAGE_FILE_QCOW2) {
J
Ján Tomko 已提交
891 892 893 894 895 896 897 898
        if (virBitmapIsBitSet(info.features,
                              VIR_STORAGE_FILE_FEATURE_LAZY_REFCOUNTS)) {
            if (STREQ_NULLABLE(info.compat, "0.10")) {
                virReportError(VIR_ERR_CONFIG_UNSUPPORTED,
                               _("lazy_refcounts not supported with compat"
                                 " level %s"),
                               info.compat);
                goto error;
899
            }
J
Ján Tomko 已提交
900
            virBufferAddLit(&buf, "lazy_refcounts,");
901 902 903
        }
    }

904 905
    virBufferTrim(&buf, ",", -1);

906 907
    if (virBufferCheckError(&buf) < 0)
        goto error;
908 909 910

    *opts = virBufferContentAndReset(&buf);
    return 0;
911

912
 error:
913 914
    virBufferFreeAndReset(&buf);
    return -1;
915 916
}

917 918 919
/* Create a qemu-img virCommand from the supplied binary path,
 * volume definitions and imgformat
 */
920
virCommandPtr
921 922 923 924 925 926 927
virStorageBackendCreateQemuImgCmdFromVol(virConnectPtr conn,
                                         virStoragePoolObjPtr pool,
                                         virStorageVolDefPtr vol,
                                         virStorageVolDefPtr inputvol,
                                         unsigned int flags,
                                         const char *create_tool,
                                         int imgformat)
928
{
929
    virCommandPtr cmd = NULL;
930 931 932
    const char *type;
    const char *backingType = NULL;
    const char *inputType = NULL;
933
    char *opts = NULL;
934 935 936 937 938 939 940 941 942
    struct _virStorageBackendQemuImgInfo info = {
        .format = vol->target.format,
        .path = vol->target.path,
        .encryption = vol->target.encryption != NULL,
        .preallocate = !!(flags & VIR_STORAGE_VOL_CREATE_PREALLOC_METADATA),
        .compat = vol->target.compat,
        .features = vol->target.features,
        .nocow = vol->target.nocow,
    };
943

944 945
    virCheckFlags(VIR_STORAGE_VOL_CREATE_PREALLOC_METADATA, NULL);

946
    /* Treat output block devices as 'raw' format */
947 948
    if (vol->type == VIR_STORAGE_VOL_BLOCK)
        info.format = VIR_STORAGE_FILE_RAW;
949

950
    if (!(type = virStorageFileFormatTypeToString(info.format))) {
951 952
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("unknown storage vol type %d"),
953
                       info.format);
954
        return NULL;
955
    }
956

957
    if (info.preallocate && info.format != VIR_STORAGE_FILE_QCOW2) {
958 959
        virReportError(VIR_ERR_CONFIG_UNSUPPORTED, "%s",
                       _("metadata preallocation only available with qcow2"));
960
        return NULL;
961
    }
962
    if (info.compat && info.format != VIR_STORAGE_FILE_QCOW2) {
963 964 965 966
        virReportError(VIR_ERR_CONFIG_UNSUPPORTED, "%s",
                       _("compatibility option only available with qcow2"));
        return NULL;
    }
967
    if (info.features && info.format != VIR_STORAGE_FILE_QCOW2) {
968 969 970 971
        virReportError(VIR_ERR_CONFIG_UNSUPPORTED, "%s",
                       _("format features only available with qcow2"));
        return NULL;
    }
972

973
    if (inputvol) {
974
        if (!(info.inputPath = inputvol->target.path)) {
975 976 977 978 979
            virReportError(VIR_ERR_INVALID_ARG, "%s",
                           _("missing input volume target path"));
            return NULL;
        }

980 981 982 983
        info.inputFormat = inputvol->target.format;
        if (inputvol->type == VIR_STORAGE_VOL_BLOCK)
            info.inputFormat = VIR_STORAGE_FILE_RAW;
        if (!(inputType = virStorageFileFormatTypeToString(info.inputFormat))) {
984 985
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("unknown storage vol type %d"),
986
                           info.inputFormat);
987 988 989 990
            return NULL;
        }
    }

991
    if (vol->target.backingStore) {
992 993
        int accessRetCode = -1;
        char *absolutePath = NULL;
994

995 996
        info.backingFormat = vol->target.backingStore->format;
        info.backingPath = vol->target.backingStore->path;
997

998
        if (info.preallocate) {
999 1000 1001
            virReportError(VIR_ERR_CONFIG_UNSUPPORTED, "%s",
                           _("metadata preallocation conflicts with backing"
                             " store"));
1002
            return NULL;
1003 1004
        }

1005 1006 1007 1008
        /* XXX: Not strictly required: qemu-img has an option a different
         * backing store, not really sure what use it serves though, and it
         * may cause issues with lvm. Untested essentially.
         */
1009
        if (inputvol && inputvol->target.backingStore &&
1010
            STRNEQ_NULLABLE(inputvol->target.backingStore->path, info.backingPath)) {
1011 1012
            virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                           _("a different backing store cannot be specified."));
1013
            return NULL;
1014 1015
        }

1016
        if (!(backingType = virStorageFileFormatTypeToString(info.backingFormat))) {
1017 1018
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("unknown storage vol backing store type %d"),
1019
                           info.backingFormat);
1020
            return NULL;
1021
        }
1022 1023 1024 1025

        /* Convert relative backing store paths to absolute paths for access
         * validation.
         */
1026
        if ('/' != *(info.backingPath) &&
1027
            virAsprintf(&absolutePath, "%s/%s", pool->def->target.path,
1028
                        info.backingPath) < 0)
1029
            return NULL;
1030
        accessRetCode = access(absolutePath ? absolutePath : info.backingPath, R_OK);
1031 1032
        VIR_FREE(absolutePath);
        if (accessRetCode != 0) {
1033
            virReportSystemError(errno,
1034
                                 _("inaccessible backing store volume %s"),
1035
                                 info.backingPath);
1036
            return NULL;
1037 1038 1039
        }
    }

1040
    if (info.encryption) {
1041 1042
        virStorageEncryptionPtr enc;

1043 1044
        if (info.format != VIR_STORAGE_FILE_QCOW &&
            info.format != VIR_STORAGE_FILE_QCOW2) {
1045 1046 1047
            virReportError(VIR_ERR_CONFIG_UNSUPPORTED,
                           _("qcow volume encryption unsupported with "
                             "volume format %s"), type);
1048
            return NULL;
1049
        }
1050 1051 1052
        enc = vol->target.encryption;
        if (enc->format != VIR_STORAGE_ENCRYPTION_FORMAT_QCOW &&
            enc->format != VIR_STORAGE_ENCRYPTION_FORMAT_DEFAULT) {
1053 1054 1055
            virReportError(VIR_ERR_CONFIG_UNSUPPORTED,
                           _("unsupported volume encryption format %d"),
                           vol->target.encryption->format);
1056
            return NULL;
1057
        }
1058
        if (enc->nsecrets > 1) {
1059 1060
            virReportError(VIR_ERR_XML_ERROR, "%s",
                           _("too many secrets for qcow encryption"));
1061
            return NULL;
1062
        }
1063 1064 1065
        if (enc->format == VIR_STORAGE_ENCRYPTION_FORMAT_DEFAULT ||
            enc->nsecrets == 0) {
            if (virStorageGenerateQcowEncryption(conn, vol) < 0)
1066
                return NULL;
1067
        }
1068 1069
    }

1070
    /* Size in KB */
1071
    info.size_arg = VIR_DIV_UP(vol->target.capacity, 1024);
1072

1073 1074
    cmd = virCommandNew(create_tool);

1075 1076 1077 1078 1079
    /* ignore the backing volume when we're converting a volume */
    if (info.inputPath) {
        info.backingPath = NULL;
        backingType = NULL;
    }
1080

1081
    if (info.inputPath)
1082 1083
        virCommandAddArgList(cmd, "convert", "-f", inputType, "-O", type, NULL);
    else
1084
        virCommandAddArgList(cmd, "create", "-f", type, NULL);
1085

1086 1087
    if (info.backingPath)
        virCommandAddArgList(cmd, "-b", info.backingPath, NULL);
1088

1089
    if (imgformat >= QEMU_IMG_BACKING_FORMAT_OPTIONS) {
1090
        if (info.format == VIR_STORAGE_FILE_QCOW2 && !info.compat &&
1091
            imgformat == QEMU_IMG_BACKING_FORMAT_OPTIONS_COMPAT)
1092 1093 1094
            info.compat = "0.10";

        if (virStorageBackendCreateQemuImgOpts(&opts, info) < 0) {
1095
            virCommandFree(cmd);
1096
            return NULL;
1097
        }
1098 1099 1100 1101
        if (opts)
            virCommandAddArgList(cmd, "-o", opts, NULL);
        VIR_FREE(opts);
    } else {
1102
        if (info.backingPath) {
1103 1104 1105 1106
            if (imgformat == QEMU_IMG_BACKING_FORMAT_FLAG)
                virCommandAddArgList(cmd, "-F", backingType, NULL);
            else
                VIR_DEBUG("Unable to set backing store format for %s with %s",
1107
                          info.path, create_tool);
1108
        }
1109
        if (info.encryption)
1110
            virCommandAddArg(cmd, "-e");
1111
    }
1112

1113 1114 1115
    if (info.inputPath)
        virCommandAddArg(cmd, info.inputPath);
    virCommandAddArg(cmd, info.path);
1116
    if (!info.inputPath && (info.size_arg || !info.backingPath))
1117
        virCommandAddArgFormat(cmd, "%lluK", info.size_arg);
1118

1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129
    return cmd;
}

static int
virStorageBackendCreateQemuImg(virConnectPtr conn,
                               virStoragePoolObjPtr pool,
                               virStorageVolDefPtr vol,
                               virStorageVolDefPtr inputvol,
                               unsigned int flags)
{
    int ret = -1;
1130
    char *create_tool;
1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150
    int imgformat;
    virCommandPtr cmd;

    virCheckFlags(VIR_STORAGE_VOL_CREATE_PREALLOC_METADATA, -1);

    /* KVM is usually ahead of qemu on features, so try that first */
    create_tool = virFindFileInPath("kvm-img");
    if (!create_tool)
        create_tool = virFindFileInPath("qemu-img");

    if (!create_tool) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       "%s", _("unable to find kvm-img or qemu-img"));
        return -1;
    }

    imgformat = virStorageBackendQEMUImgBackingFormat(create_tool);
    if (imgformat < 0)
        goto cleanup;

1151 1152
    cmd = virStorageBackendCreateQemuImgCmdFromVol(conn, pool, vol, inputvol,
                                                   flags, create_tool, imgformat);
1153 1154 1155
    if (!cmd)
        goto cleanup;

1156
    ret = virStorageBackendCreateExecCommand(pool, vol, cmd);
1157 1158

    virCommandFree(cmd);
1159
 cleanup:
1160
    VIR_FREE(create_tool);
1161
    return ret;
1162 1163 1164 1165 1166 1167 1168
}

/*
 * Xen removed the fully-functional qemu-img, and replaced it
 * with a partially functional qcow-create. Go figure ??!?
 */
static int
1169
virStorageBackendCreateQcowCreate(virConnectPtr conn ATTRIBUTE_UNUSED,
1170
                                  virStoragePoolObjPtr pool,
1171
                                  virStorageVolDefPtr vol,
1172
                                  virStorageVolDefPtr inputvol,
E
Eric Blake 已提交
1173
                                  unsigned int flags)
1174
{
1175
    int ret;
E
Eric Blake 已提交
1176
    char *size;
1177
    virCommandPtr cmd;
1178

1179 1180 1181 1182 1183 1184 1185 1186
    virCheckFlags(VIR_STORAGE_VOL_CREATE_PREALLOC_METADATA, -1);

    if (flags & VIR_STORAGE_VOL_CREATE_PREALLOC_METADATA) {
        virReportError(VIR_ERR_CONFIG_UNSUPPORTED, "%s",
                       _("metadata preallocation is not supported with "
                         "qcow-create"));
        return -1;
    }
E
Eric Blake 已提交
1187

1188
    if (inputvol) {
1189 1190
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("cannot copy from volume with qcow-create"));
1191 1192 1193
        return -1;
    }

1194
    if (vol->target.format != VIR_STORAGE_FILE_QCOW2) {
1195 1196 1197
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("unsupported storage vol type %d"),
                       vol->target.format);
1198 1199
        return -1;
    }
1200
    if (vol->target.backingStore != NULL) {
1201 1202 1203
        virReportError(VIR_ERR_CONFIG_UNSUPPORTED, "%s",
                       _("copy-on-write image not supported with "
                         "qcow-create"));
1204 1205
        return -1;
    }
1206
    if (vol->target.encryption != NULL) {
1207 1208 1209
        virReportError(VIR_ERR_CONFIG_UNSUPPORTED,
                       "%s", _("encrypted volumes not supported with "
                               "qcow-create"));
1210 1211
        return -1;
    }
1212 1213

    /* Size in MB - yes different units to qemu-img :-( */
1214
    if (virAsprintf(&size, "%llu",
1215
                    VIR_DIV_UP(vol->target.capacity, (1024 * 1024))) < 0)
E
Eric Blake 已提交
1216
        return -1;
1217

1218
    cmd = virCommandNewArgList("qcow-create", size, vol->target.path, NULL);
1219

1220 1221
    ret = virStorageBackendCreateExecCommand(pool, vol, cmd);
    virCommandFree(cmd);
E
Eric Blake 已提交
1222
    VIR_FREE(size);
1223

1224
    return ret;
1225 1226
}

1227
virStorageBackendBuildVolFrom
1228
virStorageBackendFSImageToolTypeToFunc(int tool_type)
1229 1230 1231 1232 1233 1234 1235 1236
{
    switch (tool_type) {
    case TOOL_KVM_IMG:
    case TOOL_QEMU_IMG:
        return virStorageBackendCreateQemuImg;
    case TOOL_QCOW_CREATE:
        return virStorageBackendCreateQcowCreate;
    default:
1237 1238 1239
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Unknown file create tool type '%d'."),
                       tool_type);
1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266
    }

    return NULL;
}

int
virStorageBackendFindFSImageTool(char **tool)
{
    int tool_type = -1;
    char *tmp = NULL;

    if ((tmp = virFindFileInPath("kvm-img")) != NULL) {
        tool_type = TOOL_KVM_IMG;
    } else if ((tmp = virFindFileInPath("qemu-img")) != NULL) {
        tool_type = TOOL_QEMU_IMG;
    } else if ((tmp = virFindFileInPath("qcow-create")) != NULL) {
        tool_type = TOOL_QCOW_CREATE;
    }

    if (tool)
        *tool = tmp;
    else
        VIR_FREE(tmp);

    return tool_type;
}

1267
virStorageBackendBuildVolFrom
1268
virStorageBackendGetBuildVolFromFunction(virStorageVolDefPtr vol,
1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279
                                         virStorageVolDefPtr inputvol)
{
    int tool_type;

    if (!inputvol)
        return NULL;

    /* If either volume is a non-raw file vol, we need to use an external
     * tool for converting
     */
    if ((vol->type == VIR_STORAGE_VOL_FILE &&
1280
         vol->target.format != VIR_STORAGE_FILE_RAW) ||
1281
        (inputvol->type == VIR_STORAGE_VOL_FILE &&
1282
         inputvol->target.format != VIR_STORAGE_FILE_RAW)) {
1283

D
Daniel P. Berrange 已提交
1284
        if ((tool_type = virStorageBackendFindFSImageTool(NULL)) < 0) {
1285 1286 1287
            virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                           _("creation of non-raw file images is "
                             "not supported without qemu-img."));
1288 1289 1290
            return NULL;
        }

1291
        return virStorageBackendFSImageToolTypeToFunc(tool_type);
1292 1293
    }

1294 1295 1296 1297
    if (vol->type == VIR_STORAGE_VOL_BLOCK)
        return virStorageBackendCreateBlockFrom;
    else
        return virStorageBackendCreateRaw;
1298
}
1299

1300

1301
virStorageBackendPtr
1302 1303
virStorageBackendForType(int type)
{
1304
    size_t i;
1305
    for (i = 0; backends[i]; i++)
1306 1307 1308
        if (backends[i]->type == type)
            return backends[i];

1309
    virReportError(VIR_ERR_INTERNAL_ERROR,
1310 1311
                   _("missing backend for pool type %d (%s)"),
                   type, NULLSTR(virStoragePoolTypeToString(type)));
1312 1313 1314 1315
    return NULL;
}


1316
virStorageFileBackendPtr
1317 1318 1319
virStorageFileBackendForTypeInternal(int type,
                                     int protocol,
                                     bool report)
1320 1321 1322 1323 1324
{
    size_t i;

    for (i = 0; fileBackends[i]; i++) {
        if (fileBackends[i]->type == type) {
E
Eric Blake 已提交
1325
            if (type == VIR_STORAGE_TYPE_NETWORK &&
1326 1327 1328 1329 1330 1331 1332
                fileBackends[i]->protocol != protocol)
                continue;

            return fileBackends[i];
        }
    }

1333 1334 1335
    if (!report)
        return NULL;

E
Eric Blake 已提交
1336
    if (type == VIR_STORAGE_TYPE_NETWORK) {
1337 1338 1339
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("missing storage backend for network files "
                         "using %s protocol"),
1340
                       virStorageNetProtocolTypeToString(protocol));
1341 1342 1343
    } else {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("missing storage backend for '%s' storage"),
E
Eric Blake 已提交
1344
                       virStorageTypeToString(type));
1345 1346 1347 1348 1349 1350
    }

    return NULL;
}


1351 1352 1353 1354 1355 1356 1357 1358
virStorageFileBackendPtr
virStorageFileBackendForType(int type,
                             int protocol)
{
    return virStorageFileBackendForTypeInternal(type, protocol, true);
}


1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387 1388 1389 1390 1391
struct diskType {
    int part_table_type;
    unsigned short offset;
    unsigned short length;
    unsigned long long magic;
};


static struct diskType const disk_types[] = {
    { VIR_STORAGE_POOL_DISK_LVM2, 0x218, 8, 0x31303020324D564CULL },
    { VIR_STORAGE_POOL_DISK_GPT,  0x200, 8, 0x5452415020494645ULL },
    { VIR_STORAGE_POOL_DISK_DVH,  0x0,   4, 0x41A9E50BULL },
    { VIR_STORAGE_POOL_DISK_MAC,  0x0,   2, 0x5245ULL },
    { VIR_STORAGE_POOL_DISK_BSD,  0x40,  4, 0x82564557ULL },
    { VIR_STORAGE_POOL_DISK_SUN,  0x1fc, 2, 0xBEDAULL },
    /*
     * NOTE: pc98 is funky; the actual signature is 0x55AA (just like dos), so
     * we can't use that.  At the moment I'm relying on the "dummy" IPL
     * bootloader data that comes from parted.  Luckily, the chances of running
     * into a pc98 machine running libvirt are approximately nil.
     */
    /*{ 0x1fe, 2, 0xAA55UL },*/
    { VIR_STORAGE_POOL_DISK_PC98, 0x0,   8, 0x314C5049000000CBULL },
    /*
     * NOTE: the order is important here; some other disk types (like GPT and
     * and PC98) also have 0x55AA at this offset.  For that reason, the DOS
     * one must be the last one.
     */
    { VIR_STORAGE_POOL_DISK_DOS,  0x1fe, 2, 0xAA55ULL },
    { -1,                         0x0,   0, 0x0ULL },
};


1392 1393 1394 1395
/*
 * virStorageBackendDetectBlockVolFormatFD
 * @target: target definition ptr of volume to update
 * @fd: fd of storage volume to update,
J
John Ferlan 已提交
1396 1397
 * @readflags: VolReadErrorMode flags to handle read error after open
 *             is successful, but read is not.
1398
 *
J
John Ferlan 已提交
1399 1400 1401
 * Returns 0 for success, -1 on a legitimate error condition, -2 if
 * the read error is desired to be ignored (along with appropriate
 * VIR_WARN of the issue).
1402
 */
1403
static int
1404
virStorageBackendDetectBlockVolFormatFD(virStorageSourcePtr target,
1405
                                        int fd,
J
John Ferlan 已提交
1406
                                        unsigned int readflags)
1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421 1422 1423 1424
{
    size_t i;
    off_t start;
    unsigned char buffer[1024];
    ssize_t bytes;

    /* make sure to set the target format "unknown" to begin with */
    target->format = VIR_STORAGE_POOL_DISK_UNKNOWN;

    start = lseek(fd, 0, SEEK_SET);
    if (start < 0) {
        virReportSystemError(errno,
                             _("cannot seek to beginning of file '%s'"),
                             target->path);
        return -1;
    }
    bytes = saferead(fd, buffer, sizeof(buffer));
    if (bytes < 0) {
J
John Ferlan 已提交
1425 1426 1427 1428 1429 1430 1431 1432 1433 1434
        if (readflags & VIR_STORAGE_VOL_READ_NOERROR) {
            VIR_WARN("ignoring failed saferead of file '%s'",
                     target->path);
            return -2;
        } else {
            virReportSystemError(errno,
                                 _("cannot read beginning of file '%s'"),
                                 target->path);
            return -1;
        }
1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446
    }

    for (i = 0; disk_types[i].part_table_type != -1; i++) {
        if (disk_types[i].offset + disk_types[i].length > bytes)
            continue;
        if (memcmp(buffer+disk_types[i].offset, &disk_types[i].magic,
            disk_types[i].length) == 0) {
            target->format = disk_types[i].part_table_type;
            break;
        }
    }

J
John Ferlan 已提交
1447 1448 1449 1450
    if (target->format == VIR_STORAGE_POOL_DISK_UNKNOWN)
        VIR_DEBUG("cannot determine the target format for '%s'",
                  target->path);

1451 1452 1453 1454
    return 0;
}


1455 1456 1457
/*
 * Allows caller to silently ignore files with improper mode
 *
1458 1459 1460
 * Returns -1 on error. If VIR_STORAGE_VOL_OPEN_NOERROR is passed, we
 * return -2 if file mode is unexpected or the volume is a dangling
 * symbolic link.
1461
 */
1462
int
1463 1464
virStorageBackendVolOpen(const char *path, struct stat *sb,
                         unsigned int flags)
1465
{
1466
    int fd, mode = 0;
1467
    char *base = last_component(path);
1468
    bool noerror = (flags & VIR_STORAGE_VOL_OPEN_NOERROR);
1469

E
Eric Blake 已提交
1470
    if (lstat(path, sb) < 0) {
1471 1472 1473 1474 1475 1476 1477 1478 1479
        if (errno == ENOENT) {
            if (noerror) {
                VIR_WARN("ignoring missing file '%s'", path);
                return -2;
            }
            virReportError(VIR_ERR_NO_STORAGE_VOL,
                           _("no storage vol with matching path '%s'"),
                           path);
            return -1;
1480
        }
1481 1482 1483 1484 1485 1486
        virReportSystemError(errno,
                             _("cannot stat file '%s'"),
                             path);
        return -1;
    }

E
Eric Blake 已提交
1487
    if (S_ISFIFO(sb->st_mode)) {
1488 1489 1490 1491 1492 1493 1494
        if (noerror) {
            VIR_WARN("ignoring FIFO '%s'", path);
            return -2;
        }
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Volume path '%s' is a FIFO"), path);
        return -1;
E
Eric Blake 已提交
1495
    } else if (S_ISSOCK(sb->st_mode)) {
1496 1497 1498 1499 1500 1501 1502
        if (noerror) {
            VIR_WARN("ignoring socket '%s'", path);
            return -2;
        }
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Volume path '%s' is a socket"), path);
        return -1;
1503 1504
    }

1505 1506 1507 1508
    /* O_NONBLOCK should only matter during open() for fifos and
     * sockets, which we already filtered; but using it prevents a
     * TOCTTOU race.  However, later on we will want to read() the
     * header from this fd, and virFileRead* routines require a
1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520
     * blocking fd, so fix it up after verifying we avoided a race.
     *
     * Use of virFileOpenAs allows this path to open a file using
     * the uid and gid as it was created in order to open. Since this
     * path is not using O_CREAT or O_TMPFILE, mode is meaningless.
     * Opening under user/group is especially important in an NFS
     * root-squash environment. If the target path isn't on shared
     * file system, the open will fail in the OPEN_FORK path.
     */
    if ((fd = virFileOpenAs(path, O_RDONLY|O_NONBLOCK|O_NOCTTY,
                            0, sb->st_uid, sb->st_gid,
                            VIR_FILE_OPEN_NOFORK|VIR_FILE_OPEN_FORK)) < 0) {
1521
        if ((errno == ENOENT || errno == ELOOP) &&
1522
            S_ISLNK(sb->st_mode) && noerror) {
1523 1524
            VIR_WARN("ignoring dangling symlink '%s'", path);
            return -2;
1525
        }
1526
        if (errno == ENOENT && noerror) {
1527 1528
            VIR_WARN("ignoring missing file '%s'", path);
            return -2;
1529
        }
1530 1531 1532 1533
        if (errno == ENXIO && noerror) {
            VIR_WARN("ignoring missing fifo '%s'", path);
            return -2;
        }
1534 1535 1536 1537
        if ((errno == EACCES || errno == EPERM) && noerror) {
            VIR_WARN("ignoring permission error for '%s'", path);
            return -2;
        }
1538

1539
        virReportSystemError(errno, _("cannot open volume '%s'"), path);
1540 1541 1542
        return -1;
    }

E
Eric Blake 已提交
1543
    if (fstat(fd, sb) < 0) {
1544
        virReportSystemError(errno, _("cannot stat file '%s'"), path);
1545 1546 1547 1548
        VIR_FORCE_CLOSE(fd);
        return -1;
    }

1549
    if (S_ISREG(sb->st_mode)) {
1550
        mode = VIR_STORAGE_VOL_OPEN_REG;
1551
    } else if (S_ISCHR(sb->st_mode)) {
1552
        mode = VIR_STORAGE_VOL_OPEN_CHAR;
1553
    } else if (S_ISBLK(sb->st_mode)) {
1554
        mode = VIR_STORAGE_VOL_OPEN_BLOCK;
1555
    } else if (S_ISDIR(sb->st_mode)) {
1556 1557 1558 1559 1560
        mode = VIR_STORAGE_VOL_OPEN_DIR;

        if (STREQ(base, ".") ||
            STREQ(base, "..")) {
            VIR_FORCE_CLOSE(fd);
1561 1562 1563 1564 1565 1566 1567
            if (noerror) {
                VIR_INFO("Skipping special dir '%s'", base);
                return -2;
            }
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("Cannot use volume path '%s'"), path);
            return -1;
1568
        }
1569 1570
    } else {
        VIR_FORCE_CLOSE(fd);
1571 1572 1573 1574 1575 1576 1577
        if (noerror) {
            VIR_WARN("ignoring unexpected type for file '%s'", path);
            return -2;
        }
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("unexpected type for file '%s'"), path);
        return -1;
1578 1579 1580
    }

    if (virSetBlocking(fd, true) < 0) {
1581
        VIR_FORCE_CLOSE(fd);
1582 1583
        virReportSystemError(errno, _("unable to set blocking mode for '%s'"),
                             path);
1584
        return -1;
1585
    }
1586 1587

    if (!(mode & flags)) {
1588
        VIR_FORCE_CLOSE(fd);
1589 1590 1591
        if (noerror) {
            VIR_INFO("Skipping volume '%s'", path);
            return -2;
1592 1593
        }

1594 1595 1596
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("unexpected storage mode for '%s'"), path);
        return -1;
1597 1598 1599 1600 1601
    }

    return fd;
}

1602 1603 1604 1605 1606
/*
 * virStorageBackendUpdateVolTargetInfo
 * @target: target definition ptr of volume to update
 * @withBlockVolFormat: true if caller determined a block file
 * @openflags: various VolOpenCheckMode flags to handle errors on open
J
John Ferlan 已提交
1607 1608
 * @readflags: VolReadErrorMode flags to handle read error after open
 *             is successful, but read is not.
1609 1610 1611
 *
 * Returns 0 for success, -1 on a legitimate error condition, and -2
 * if the openflags used VIR_STORAGE_VOL_OPEN_NOERROR and some sort of
J
John Ferlan 已提交
1612 1613
 * open error occurred. It is up to the caller to handle. A -2 may also
 * be returned if the caller passed a readflagsflag.
1614
 */
1615
int
1616
virStorageBackendUpdateVolTargetInfo(virStorageSourcePtr target,
1617
                                     bool withBlockVolFormat,
1618 1619
                                     unsigned int openflags,
                                     unsigned int readflags)
1620
{
1621
    int ret, fd = -1;
E
Eric Blake 已提交
1622
    struct stat sb;
1623 1624 1625
    virStorageSourcePtr meta = NULL;
    char *buf = NULL;
    ssize_t len = VIR_STORAGE_MAX_HEADER;
1626

1627
    if ((ret = virStorageBackendVolOpen(target->path, &sb, openflags)) < 0)
1628
        goto cleanup;
1629
    fd = ret;
1630

1631
    if ((ret = virStorageBackendUpdateVolTargetInfoFD(target, fd, &sb)) < 0)
1632 1633
        goto cleanup;

1634 1635 1636 1637 1638 1639 1640 1641 1642
    if (target->type == VIR_STORAGE_VOL_FILE &&
        target->format != VIR_STORAGE_FILE_NONE) {
        if (S_ISDIR(sb.st_mode)) {
            ret = 0;
            goto cleanup;
        }

        if (lseek(fd, 0, SEEK_SET) == (off_t)-1) {
            virReportSystemError(errno, _("cannot seek to start of '%s'"), target->path);
1643
            ret = -1;
1644 1645 1646 1647
            goto cleanup;
        }

        if ((len = virFileReadHeaderFD(fd, len, &buf)) < 0) {
J
John Ferlan 已提交
1648 1649 1650 1651 1652 1653 1654 1655 1656 1657
            if (readflags & VIR_STORAGE_VOL_READ_NOERROR) {
                VIR_WARN("ignoring failed header read for '%s'",
                         target->path);
                ret = -2;
            } else {
                virReportSystemError(errno,
                                     _("cannot read header '%s'"),
                                     target->path);
                ret = -1;
            }
1658 1659 1660 1661 1662
            goto cleanup;
        }

        if (!(meta = virStorageFileGetMetadataFromBuf(target->path, buf, len, target->format,
                                                      NULL))) {
1663
            ret = -1;
1664 1665 1666 1667 1668 1669 1670
            goto cleanup;
        }

        if (meta->capacity)
            target->capacity = meta->capacity;
    }

1671
    if (withBlockVolFormat) {
1672 1673
        if ((ret = virStorageBackendDetectBlockVolFormatFD(target, fd,
                                                           readflags)) < 0)
1674 1675 1676
            goto cleanup;
    }

1677
 cleanup:
1678
    virStorageSourceFree(meta);
1679
    VIR_FORCE_CLOSE(fd);
1680
    VIR_FREE(buf);
1681 1682 1683
    return ret;
}

1684 1685 1686 1687 1688
/*
 * virStorageBackendUpdateVolInfo
 * @vol: Pointer to a volume storage definition
 * @withBlockVolFormat: true if the caller determined a block file
 * @openflags: various VolOpenCheckMode flags to handle errors on open
J
John Ferlan 已提交
1689
 * @readflags: various VolReadErrorMode flags to handle errors on read
1690 1691 1692 1693 1694
 *
 * Returns 0 for success, -1 on a legitimate error condition, and -2
 * if the openflags used VIR_STORAGE_VOL_OPEN_NOERROR and some sort of
 * open error occurred. It is up to the caller to handle.
 */
1695
int
1696
virStorageBackendUpdateVolInfo(virStorageVolDefPtr vol,
1697
                               bool withBlockVolFormat,
1698 1699
                               unsigned int openflags,
                               unsigned int readflags)
1700 1701 1702
{
    int ret;

1703
    if ((ret = virStorageBackendUpdateVolTargetInfo(&vol->target,
1704
                                                    withBlockVolFormat,
1705
                                                    openflags, readflags)) < 0)
1706 1707
        return ret;

1708 1709
    if (vol->target.backingStore &&
        (ret = virStorageBackendUpdateVolTargetInfo(vol->target.backingStore,
1710
                                                    withBlockVolFormat,
1711
                                                    VIR_STORAGE_VOL_OPEN_DEFAULT |
1712 1713
                                                    VIR_STORAGE_VOL_OPEN_NOERROR,
                                                    readflags) < 0))
1714 1715 1716 1717 1718
        return ret;

    return 0;
}

1719 1720 1721
/*
 * virStorageBackendUpdateVolTargetInfoFD:
 * @target: target definition ptr of volume to update
1722 1723
 * @fd: fd of storage volume to update, via virStorageBackendOpenVol*, or -1
 * @sb: details about file (must match @fd, if that is provided)
1724
 *
1725
 * Returns 0 for success, -1 on a legitimate error condition.
1726
 */
1727
int
1728
virStorageBackendUpdateVolTargetInfoFD(virStorageSourcePtr target,
1729
                                       int fd,
1730
                                       struct stat *sb)
1731
{
1732
#if WITH_SELINUX
1733 1734 1735
    security_context_t filecon = NULL;
#endif

1736
    if (S_ISREG(sb->st_mode)) {
1737
#ifndef WIN32
1738 1739
        target->allocation = (unsigned long long)sb->st_blocks *
            (unsigned long long)DEV_BSIZE;
D
Daniel P. Berrange 已提交
1740
#else
1741
        target->allocation = sb->st_size;
D
Daniel P. Berrange 已提交
1742
#endif
1743 1744 1745
        /* Regular files may be sparse, so logical size (capacity) is not same
         * as actual allocation above
         */
1746
        target->capacity = sb->st_size;
1747 1748
    } else if (S_ISDIR(sb->st_mode)) {
        target->allocation = 0;
1749
        target->capacity = 0;
1750 1751 1752 1753 1754 1755 1756 1757 1758 1759 1760 1761 1762
    } else if (fd >= 0) {
        off_t end;
        /* XXX this is POSIX compliant, but doesn't work for CHAR files,
         * only BLOCK. There is a Linux specific ioctl() for getting
         * size of both CHAR / BLOCK devices we should check for in
         * configure
         */
        end = lseek(fd, 0, SEEK_END);
        if (end == (off_t)-1) {
            virReportSystemError(errno,
                                 _("cannot seek to end of file '%s'"),
                                 target->path);
            return -1;
1763
        }
1764
        target->allocation = end;
1765
        target->capacity = end;
1766 1767
    }

1768 1769 1770 1771 1772
    if (!target->perms && VIR_ALLOC(target->perms) < 0)
        return -1;
    target->perms->mode = sb->st_mode & S_IRWXUGO;
    target->perms->uid = sb->st_uid;
    target->perms->gid = sb->st_gid;
1773

1774
    if (!target->timestamps && VIR_ALLOC(target->timestamps) < 0)
1775
        return -1;
E
Eric Blake 已提交
1776 1777 1778 1779
    target->timestamps->atime = get_stat_atime(sb);
    target->timestamps->btime = get_stat_birthtime(sb);
    target->timestamps->ctime = get_stat_ctime(sb);
    target->timestamps->mtime = get_stat_mtime(sb);
1780

1781
    VIR_FREE(target->perms->label);
1782

1783
#if WITH_SELINUX
1784
    /* XXX: make this a security driver call */
1785 1786 1787 1788 1789 1790 1791 1792
    if (fd >= 0) {
        if (fgetfilecon_raw(fd, &filecon) == -1) {
            if (errno != ENODATA && errno != ENOTSUP) {
                virReportSystemError(errno,
                                     _("cannot get file context of '%s'"),
                                     target->path);
                return -1;
            }
1793
        } else {
1794
            if (VIR_STRDUP(target->perms->label, filecon) < 0) {
1795 1796 1797
                freecon(filecon);
                return -1;
            }
1798
            freecon(filecon);
1799
        }
1800 1801 1802 1803 1804 1805
    }
#endif

    return 0;
}

1806 1807 1808 1809 1810 1811
bool
virStorageBackendPoolPathIsStable(const char *path)
{
    if (path == NULL || STREQ(path, "/dev") || STREQ(path, "/dev/"))
        return false;

1812
    if (!STRPREFIX(path, "/dev/"))
1813 1814 1815 1816
        return false;

    return true;
}
D
Dave Allan 已提交
1817

1818 1819 1820 1821 1822 1823 1824 1825 1826 1827 1828 1829
/*
 * Given a volume path directly in /dev/XXX, iterate over the
 * entries in the directory pool->def->target.path and find the
 * first symlink pointing to the volume path.
 *
 * If, the target.path is /dev/, then return the original volume
 * path.
 *
 * If no symlink is found, then return the original volume path
 *
 * Typically target.path is one of the /dev/disk/by-XXX dirs
 * with stable paths.
1830
 *
1831
 * If 'loop' is true, we use a timeout loop to give dynamic paths
1832
 * a change to appear.
1833 1834
 */
char *
1835
virStorageBackendStablePath(virStoragePoolObjPtr pool,
1836
                            const char *devpath,
1837
                            bool loop)
1838 1839 1840
{
    DIR *dh;
    struct dirent *dent;
1841
    char *stablepath;
1842
    int opentries = 0;
1843
    int retry = 0;
E
Eric Blake 已提交
1844
    int direrr;
1845

1846
    /* Logical pools are under /dev but already have stable paths */
1847 1848
    if (pool->def->type == VIR_STORAGE_POOL_LOGICAL ||
        !virStorageBackendPoolPathIsStable(pool->def->target.path))
1849 1850
        goto ret_strdup;

1851 1852 1853
    /* We loop here because /dev/disk/by-{id,path} may not have existed
     * before we started this operation, so we have to give it some time to
     * get created.
1854
     */
1855
 reopen:
1856
    if ((dh = opendir(pool->def->target.path)) == NULL) {
1857
        opentries++;
1858
        if (loop && errno == ENOENT && opentries < 50) {
1859 1860 1861
            usleep(100 * 1000);
            goto reopen;
        }
1862
        virReportSystemError(errno,
1863 1864
                             _("cannot read dir '%s'"),
                             pool->def->target.path);
1865 1866 1867
        return NULL;
    }

1868 1869 1870
    /* The pool is pointing somewhere like /dev/disk/by-path
     * or /dev/disk/by-id, so we need to check all symlinks in
     * the target directory and figure out which one points
1871 1872
     * to this device node.
     *
1873
     * And it might need some time till the stable path shows
E
Eric Blake 已提交
1874 1875
     * up, so add timeout to retry here.  Ignore readdir failures,
     * since we have a fallback.
1876
     */
1877
 retry:
E
Eric Blake 已提交
1878
    while ((direrr = virDirRead(dh, &dent, NULL)) > 0) {
1879 1880 1881
        if (dent->d_name[0] == '.')
            continue;

1882 1883 1884
        if (virAsprintf(&stablepath, "%s/%s",
                        pool->def->target.path,
                        dent->d_name) == -1) {
1885 1886 1887 1888 1889 1890 1891 1892 1893
            closedir(dh);
            return NULL;
        }

        if (virFileLinkPointsTo(stablepath, devpath)) {
            closedir(dh);
            return stablepath;
        }

1894
        VIR_FREE(stablepath);
1895 1896
    }

E
Eric Blake 已提交
1897
    if (!direrr && loop && ++retry < 100) {
1898 1899 1900 1901
        usleep(100 * 1000);
        goto retry;
    }

1902 1903
    closedir(dh);

1904
 ret_strdup:
1905 1906 1907
    /* Couldn't find any matching stable link so give back
     * the original non-stable dev path
     */
1908

1909
    ignore_value(VIR_STRDUP(stablepath, devpath));
1910 1911

    return stablepath;
1912
}
1913

1914 1915 1916 1917 1918 1919 1920 1921 1922 1923 1924 1925 1926
int
virStorageBackendVolUploadLocal(virConnectPtr conn ATTRIBUTE_UNUSED,
                                virStoragePoolObjPtr pool ATTRIBUTE_UNUSED,
                                virStorageVolDefPtr vol,
                                virStreamPtr stream,
                                unsigned long long offset,
                                unsigned long long len,
                                unsigned int flags)
{
    virCheckFlags(0, -1);

    /* Not using O_CREAT because the file is required to already exist at
     * this point */
1927 1928
    return virFDStreamOpenBlockDevice(stream, vol->target.path,
                                      offset, len, O_WRONLY);
1929 1930 1931 1932 1933 1934 1935 1936 1937 1938 1939 1940 1941
}

int
virStorageBackendVolDownloadLocal(virConnectPtr conn ATTRIBUTE_UNUSED,
                                  virStoragePoolObjPtr pool ATTRIBUTE_UNUSED,
                                  virStorageVolDefPtr vol,
                                  virStreamPtr stream,
                                  unsigned long long offset,
                                  unsigned long long len,
                                  unsigned int flags)
{
    virCheckFlags(0, -1);

1942 1943
    return virFDStreamOpenBlockDevice(stream, vol->target.path,
                                      offset, len, O_RDONLY);
1944 1945
}

1946 1947 1948 1949 1950 1951 1952 1953 1954 1955 1956 1957 1958 1959 1960 1961 1962 1963 1964 1965 1966 1967 1968 1969 1970 1971 1972 1973 1974 1975 1976 1977 1978 1979 1980 1981 1982 1983 1984 1985 1986 1987 1988 1989

/* If the volume we're wiping is already a sparse file, we simply
 * truncate and extend it to its original size, filling it with
 * zeroes.  This behavior is guaranteed by POSIX:
 *
 * http://www.opengroup.org/onlinepubs/9699919799/functions/ftruncate.html
 *
 * If fildes refers to a regular file, the ftruncate() function shall
 * cause the size of the file to be truncated to length. If the size
 * of the file previously exceeded length, the extra data shall no
 * longer be available to reads on the file. If the file previously
 * was smaller than this size, ftruncate() shall increase the size of
 * the file. If the file size is increased, the extended area shall
 * appear as if it were zero-filled.
 */
static int
virStorageBackendVolZeroSparseFileLocal(virStorageVolDefPtr vol,
                                        off_t size,
                                        int fd)
{
    int ret = -1;

    ret = ftruncate(fd, 0);
    if (ret == -1) {
        virReportSystemError(errno,
                             _("Failed to truncate volume with "
                               "path '%s' to 0 bytes"),
                             vol->target.path);
        return ret;
    }

    ret = ftruncate(fd, size);
    if (ret == -1) {
        virReportSystemError(errno,
                             _("Failed to truncate volume with "
                               "path '%s' to %ju bytes"),
                             vol->target.path, (uintmax_t)size);
    }

    return ret;
}


static int
1990 1991 1992
virStorageBackendWipeLocal(virStorageVolDefPtr vol,
                           int fd,
                           unsigned long long wipe_len,
1993
                           size_t writebuf_length)
1994 1995
{
    int ret = -1, written = 0;
1996
    unsigned long long remaining = 0;
1997
    size_t write_size = 0;
1998
    char *writebuf = NULL;
1999

2000
    VIR_DEBUG("wiping start: 0 len: %llu", wipe_len);
2001

2002 2003 2004
    if (VIR_ALLOC_N(writebuf, writebuf_length) < 0)
        goto cleanup;

2005
    if (lseek(fd, 0, SEEK_SET) < 0) {
2006
        virReportSystemError(errno,
2007
                             _("Failed to seek to the start in volume "
2008
                               "with path '%s'"),
2009
                             vol->target.path);
2010 2011 2012
        goto cleanup;
    }

2013
    remaining = wipe_len;
2014 2015 2016 2017 2018 2019 2020 2021 2022 2023 2024 2025 2026 2027 2028 2029 2030 2031 2032 2033 2034 2035 2036
    while (remaining > 0) {

        write_size = (writebuf_length < remaining) ? writebuf_length : remaining;
        written = safewrite(fd, writebuf, write_size);
        if (written < 0) {
            virReportSystemError(errno,
                                 _("Failed to write %zu bytes to "
                                   "storage volume with path '%s'"),
                                 write_size, vol->target.path);

            goto cleanup;
        }

        remaining -= written;
    }

    if (fdatasync(fd) < 0) {
        virReportSystemError(errno,
                             _("cannot sync data to volume with path '%s'"),
                             vol->target.path);
        goto cleanup;
    }

2037 2038
    VIR_DEBUG("Wrote %llu bytes to volume with path '%s'",
              wipe_len, vol->target.path);
2039 2040 2041 2042

    ret = 0;

 cleanup:
2043
    VIR_FREE(writebuf);
2044 2045 2046 2047 2048 2049 2050 2051 2052 2053 2054 2055
    return ret;
}


int
virStorageBackendVolWipeLocal(virConnectPtr conn ATTRIBUTE_UNUSED,
                              virStoragePoolObjPtr pool ATTRIBUTE_UNUSED,
                              virStorageVolDefPtr vol,
                              unsigned int algorithm,
                              unsigned int flags)
{
    int ret = -1, fd = -1;
2056
    const char *alg_char = NULL;
2057 2058 2059 2060 2061 2062 2063 2064 2065 2066 2067 2068 2069 2070 2071 2072 2073 2074 2075 2076 2077 2078 2079
    struct stat st;
    virCommandPtr cmd = NULL;

    virCheckFlags(0, -1);

    VIR_DEBUG("Wiping volume with path '%s' and algorithm %u",
              vol->target.path, algorithm);

    fd = open(vol->target.path, O_RDWR);
    if (fd == -1) {
        virReportSystemError(errno,
                             _("Failed to open storage volume with path '%s'"),
                             vol->target.path);
        goto cleanup;
    }

    if (fstat(fd, &st) == -1) {
        virReportSystemError(errno,
                             _("Failed to stat storage volume with path '%s'"),
                             vol->target.path);
        goto cleanup;
    }

2080 2081 2082 2083 2084 2085 2086 2087 2088 2089 2090 2091 2092 2093 2094 2095 2096 2097 2098 2099 2100 2101 2102 2103 2104 2105 2106 2107
    switch ((virStorageVolWipeAlgorithm) algorithm) {
    case VIR_STORAGE_VOL_WIPE_ALG_ZERO:
        alg_char = "zero";
        break;
    case VIR_STORAGE_VOL_WIPE_ALG_NNSA:
        alg_char = "nnsa";
        break;
    case VIR_STORAGE_VOL_WIPE_ALG_DOD:
        alg_char = "dod";
        break;
    case VIR_STORAGE_VOL_WIPE_ALG_BSI:
        alg_char = "bsi";
        break;
    case VIR_STORAGE_VOL_WIPE_ALG_GUTMANN:
        alg_char = "gutmann";
        break;
    case VIR_STORAGE_VOL_WIPE_ALG_SCHNEIER:
        alg_char = "schneier";
        break;
    case VIR_STORAGE_VOL_WIPE_ALG_PFITZNER7:
        alg_char = "pfitzner7";
        break;
    case VIR_STORAGE_VOL_WIPE_ALG_PFITZNER33:
        alg_char = "pfitzner33";
        break;
    case VIR_STORAGE_VOL_WIPE_ALG_RANDOM:
        alg_char = "random";
        break;
2108 2109 2110 2111
    case VIR_STORAGE_VOL_WIPE_ALG_TRIM:
        virReportError(VIR_ERR_ARGUMENT_UNSUPPORTED, "%s",
                       _("'trim' algorithm not supported"));
        goto cleanup;
2112 2113 2114 2115 2116 2117 2118
    case VIR_STORAGE_VOL_WIPE_ALG_LAST:
        virReportError(VIR_ERR_INVALID_ARG,
                       _("unsupported algorithm %d"),
                       algorithm);
        goto cleanup;
    }

2119 2120 2121 2122 2123 2124 2125 2126 2127 2128 2129 2130 2131 2132
    if (algorithm != VIR_STORAGE_VOL_WIPE_ALG_ZERO) {
        cmd = virCommandNew(SCRUB);
        virCommandAddArgList(cmd, "-f", "-p", alg_char,
                             vol->target.path, NULL);

        if (virCommandRun(cmd, NULL) < 0)
            goto cleanup;

        ret = 0;
        goto cleanup;
    } else {
        if (S_ISREG(st.st_mode) && st.st_blocks < (st.st_size / DEV_BSIZE)) {
            ret = virStorageBackendVolZeroSparseFileLocal(vol, st.st_size, fd);
        } else {
2133 2134 2135
            ret = virStorageBackendWipeLocal(vol,
                                             fd,
                                             vol->target.allocation,
2136
                                             st.st_blksize);
2137 2138 2139 2140 2141 2142 2143 2144 2145 2146
        }
    }

 cleanup:
    virCommandFree(cmd);
    VIR_FORCE_CLOSE(fd);
    return ret;
}


2147 2148 2149 2150 2151 2152 2153 2154 2155 2156 2157 2158 2159 2160 2161 2162 2163 2164 2165 2166 2167 2168 2169 2170 2171 2172 2173 2174 2175 2176 2177 2178 2179 2180 2181 2182 2183 2184 2185 2186 2187 2188 2189 2190 2191 2192 2193 2194 2195 2196 2197 2198 2199 2200 2201 2202 2203 2204 2205 2206 2207 2208 2209 2210 2211 2212 2213 2214 2215 2216 2217 2218 2219 2220 2221 2222 2223 2224 2225 2226 2227 2228 2229 2230 2231 2232 2233 2234
#ifdef GLUSTER_CLI
int
virStorageBackendFindGlusterPoolSources(const char *host,
                                        int pooltype,
                                        virStoragePoolSourceListPtr list)
{
    char *outbuf = NULL;
    virCommandPtr cmd = NULL;
    xmlDocPtr doc = NULL;
    xmlXPathContextPtr ctxt = NULL;
    xmlNodePtr *nodes = NULL;
    virStoragePoolSource *src = NULL;
    size_t i;
    int nnodes;
    int rc;

    int ret = -1;

    cmd = virCommandNewArgList(GLUSTER_CLI,
                               "--xml",
                               "--log-file=/dev/null",
                               "volume", "info", "all", NULL);

    virCommandAddArgFormat(cmd, "--remote-host=%s", host);
    virCommandSetOutputBuffer(cmd, &outbuf);

    if (virCommandRun(cmd, &rc) < 0)
        goto cleanup;

    if (rc != 0) {
        VIR_INFO("failed to query host '%s' for gluster volumes: %s",
                 host, outbuf);
        ret = 0;
        goto cleanup;
    }

    if (!(doc = virXMLParseStringCtxt(outbuf, _("(gluster_cli_output)"),
                                      &ctxt)))
        goto cleanup;

    if ((nnodes = virXPathNodeSet("//volumes/volume", ctxt, &nodes)) <= 0) {
        VIR_INFO("no gluster volumes available on '%s'", host);
        ret = 0;
        goto cleanup;
    }

    for (i = 0; i < nnodes; i++) {
        ctxt->node = nodes[i];

        if (!(src = virStoragePoolSourceListNewSource(list)))
            goto cleanup;

        if (!(src->dir = virXPathString("string(//name)", ctxt))) {
            virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                           _("failed to extract gluster volume name"));
            goto cleanup;
        }

        if (VIR_ALLOC_N(src->hosts, 1) < 0)
            goto cleanup;
        src->nhost = 1;

        if (VIR_STRDUP(src->hosts[0].name, host) < 0)
            goto cleanup;

        src->format = pooltype;
    }

    ret = 0;

 cleanup:
    VIR_FREE(nodes);
    xmlXPathFreeContext(ctxt);
    xmlFreeDoc(doc);
    VIR_FREE(outbuf);
    virCommandFree(cmd);
    return ret;
}
#else /* #ifdef GLUSTER_CLI */
int
virStorageBackendFindGlusterPoolSources(const char *host ATTRIBUTE_UNUSED,
                                        int pooltype ATTRIBUTE_UNUSED,
                                        virStoragePoolSourceListPtr list ATTRIBUTE_UNUSED)
{
    VIR_INFO("gluster cli tool not installed");
    return 0;
}
#endif /* #ifdef GLUSTER_CLI */