btrfs-progs: image: close all_devices at the end
[platform/upstream/btrfs-progs.git] / cmds-send.c
index 69e9bce..581b25e 100644 (file)
@@ -16,7 +16,6 @@
  * Boston, MA 021110-1307, USA.
  */
 
-#define _GNU_SOURCE
 
 #include "kerncompat.h"
 
 #include <sys/ioctl.h>
 #include <libgen.h>
 #include <mntent.h>
-
+#include <assert.h>
+#include <getopt.h>
 #include <uuid/uuid.h>
+#include <limits.h>
 
 #include "ctree.h"
 #include "ioctl.h"
 #include "commands.h"
 #include "list.h"
+#include "utils.h"
 
 #include "send.h"
 #include "send-utils.h"
@@ -56,39 +58,6 @@ struct btrfs_send {
        struct subvol_uuid_search sus;
 };
 
-int find_mount_root(const char *path, char **mount_root)
-{
-       FILE *mnttab;
-       int fd;
-       struct mntent *ent;
-       int len;
-       int longest_matchlen = 0;
-       char *longest_match = NULL;
-
-       fd = open(path, O_RDONLY | O_NOATIME);
-       if (fd < 0)
-               return -errno;
-       close(fd);
-
-       mnttab = fopen("/etc/mtab", "r");
-       while ((ent = getmntent(mnttab))) {
-               len = strlen(ent->mnt_dir);
-               if (strncmp(ent->mnt_dir, path, len) == 0) {
-                       /* match found */
-                       if (longest_matchlen < len) {
-                               free(longest_match);
-                               longest_matchlen = len;
-                               longest_match = strdup(ent->mnt_dir);
-                       }
-               }
-       }
-
-       *mount_root = realpath(longest_match, NULL);
-       free(longest_match);
-
-       return 0;
-}
-
 static int get_root_id(struct btrfs_send *s, const char *path, u64 *root_id)
 {
        struct subvol_info *si;
@@ -98,30 +67,33 @@ static int get_root_id(struct btrfs_send *s, const char *path, u64 *root_id)
        if (!si)
                return -ENOENT;
        *root_id = si->root_id;
+       free(si->path);
+       free(si);
        return 0;
 }
 
 static struct subvol_info *get_parent(struct btrfs_send *s, u64 root_id)
 {
+       struct subvol_info *si_tmp;
        struct subvol_info *si;
 
-       si = subvol_uuid_search(&s->sus, root_id, NULL, 0, NULL,
+       si_tmp = subvol_uuid_search(&s->sus, root_id, NULL, 0, NULL,
                        subvol_search_by_root_id);
-       if (!si)
+       if (!si_tmp)
                return NULL;
 
-       si = subvol_uuid_search(&s->sus, 0, si->parent_uuid, 0, NULL,
+       si = subvol_uuid_search(&s->sus, 0, si_tmp->parent_uuid, 0, NULL,
                        subvol_search_by_uuid);
-       if (!si)
-               return NULL;
+       free(si_tmp->path);
+       free(si_tmp);
        return si;
 }
 
 static int find_good_parent(struct btrfs_send *s, u64 root_id, u64 *found)
 {
        int ret;
-       struct subvol_info *parent;
-       struct subvol_info *parent2;
+       struct subvol_info *parent = NULL;
+       struct subvol_info *parent2 = NULL;
        struct subvol_info *best_parent = NULL;
        __s64 tmp;
        u64 best_diff = (u64)-1;
@@ -136,24 +108,46 @@ static int find_good_parent(struct btrfs_send *s, u64 root_id, u64 *found)
        for (i = 0; i < s->clone_sources_count; i++) {
                if (s->clone_sources[i] == parent->root_id) {
                        best_parent = parent;
+                       parent = NULL;
                        goto out_found;
                }
        }
 
        for (i = 0; i < s->clone_sources_count; i++) {
                parent2 = get_parent(s, s->clone_sources[i]);
-               if (parent2 != parent)
+               if (!parent2)
                        continue;
+               if (parent2->root_id != parent->root_id) {
+                       free(parent2->path);
+                       free(parent2);
+                       parent2 = NULL;
+                       continue;
+               }
 
+               free(parent2->path);
+               free(parent2);
                parent2 = subvol_uuid_search(&s->sus, s->clone_sources[i], NULL,
                                0, NULL, subvol_search_by_root_id);
 
+               if (!parent2) {
+                       ret = -ENOENT;
+                       goto out;
+               }
                tmp = parent2->ctransid - parent->ctransid;
                if (tmp < 0)
                        tmp *= -1;
                if (tmp < best_diff) {
+                       if (best_parent) {
+                               free(best_parent->path);
+                               free(best_parent);
+                       }
                        best_parent = parent2;
+                       parent2 = NULL;
                        best_diff = tmp;
+               } else {
+                       free(parent2->path);
+                       free(parent2);
+                       parent2 = NULL;
                }
        }
 
@@ -167,14 +161,32 @@ out_found:
        ret = 0;
 
 out:
+       if (parent) {
+               free(parent->path);
+               free(parent);
+       }
+       if (best_parent) {
+               free(best_parent->path);
+               free(best_parent);
+       }
        return ret;
 }
 
-static void add_clone_source(struct btrfs_send *s, u64 root_id)
+static int add_clone_source(struct btrfs_send *s, u64 root_id)
 {
+       void *tmp;
+
+       tmp = s->clone_sources;
        s->clone_sources = realloc(s->clone_sources,
                sizeof(*s->clone_sources) * (s->clone_sources_count + 1));
+
+       if (!s->clone_sources) {
+               free(tmp);
+               return -ENOMEM;
+       }
        s->clone_sources[s->clone_sources_count++] = root_id;
+
+       return 0;
 }
 
 static int write_buf(int fd, const void *buf, int size)
@@ -186,13 +198,13 @@ static int write_buf(int fd, const void *buf, int size)
                ret = write(fd, (char*)buf + pos, size - pos);
                if (ret < 0) {
                        ret = -errno;
-                       fprintf(stderr, "ERROR: failed to dump stream. %s",
+                       fprintf(stderr, "ERROR: failed to dump stream. %s\n",
                                        strerror(-ret));
                        goto out;
                }
                if (!ret) {
                        ret = -EIO;
-                       fprintf(stderr, "ERROR: failed to dump stream. %s",
+                       fprintf(stderr, "ERROR: failed to dump stream. %s\n",
                                        strerror(-ret));
                        goto out;
                }
@@ -236,36 +248,25 @@ out:
        return ERR_PTR(ret);
 }
 
-static int do_send(struct btrfs_send *send, u64 root_id, u64 parent_root_id)
+static int do_send(struct btrfs_send *send, u64 parent_root_id,
+                  int is_first_subvol, int is_last_subvol, char *subvol,
+                  u64 flags)
 {
        int ret;
        pthread_t t_read;
-       pthread_attr_t t_attr;
        struct btrfs_ioctl_send_args io_send;
-       struct subvol_info *si;
        void *t_err = NULL;
        int subvol_fd = -1;
-       int pipefd[2];
+       int pipefd[2] = {-1, -1};
 
-       si = subvol_uuid_search(&send->sus, root_id, NULL, 0, NULL,
-                       subvol_search_by_root_id);
-       if (!si) {
-               ret = -ENOENT;
-               fprintf(stderr, "ERROR: could not find subvol info for %llu",
-                               root_id);
-               goto out;
-       }
-
-       subvol_fd = openat(send->mnt_fd, si->path, O_RDONLY | O_NOATIME);
+       subvol_fd = openat(send->mnt_fd, subvol, O_RDONLY | O_NOATIME);
        if (subvol_fd < 0) {
                ret = -errno;
-               fprintf(stderr, "ERROR: open %s failed. %s\n", si->path,
+               fprintf(stderr, "ERROR: open %s failed. %s\n", subvol,
                                strerror(-ret));
                goto out;
        }
 
-       ret = pthread_attr_init(&t_attr);
-
        ret = pipe(pipefd);
        if (ret < 0) {
                ret = -errno;
@@ -278,7 +279,7 @@ static int do_send(struct btrfs_send *send, u64 root_id, u64 parent_root_id)
        send->send_fd = pipefd[0];
 
        if (!ret)
-               ret = pthread_create(&t_read, &t_attr, dump_thread,
+               ret = pthread_create(&t_read, NULL, dump_thread,
                                        send);
        if (ret) {
                ret = -ret;
@@ -287,14 +288,22 @@ static int do_send(struct btrfs_send *send, u64 root_id, u64 parent_root_id)
                goto out;
        }
 
+       io_send.flags = flags;
        io_send.clone_sources = (__u64*)send->clone_sources;
        io_send.clone_sources_count = send->clone_sources_count;
        io_send.parent_root = parent_root_id;
+       if (!is_first_subvol)
+               io_send.flags |= BTRFS_SEND_FLAG_OMIT_STREAM_HEADER;
+       if (!is_last_subvol)
+               io_send.flags |= BTRFS_SEND_FLAG_OMIT_END_CMD;
        ret = ioctl(subvol_fd, BTRFS_IOC_SEND, &io_send);
        if (ret) {
                ret = -errno;
                fprintf(stderr, "ERROR: send ioctl failed with %d: %s\n", ret,
                        strerror(-ret));
+               if (ret == -EINVAL && (!is_first_subvol || !is_last_subvol))
+                       fprintf(stderr,
+                               "Try upgrading your kernel or don't use -e.\n");
                goto out;
        }
        if (g_verbose > 0)
@@ -304,7 +313,7 @@ static int do_send(struct btrfs_send *send, u64 root_id, u64 parent_root_id)
                fprintf(stderr, "joining genl thread\n");
 
        close(pipefd[1]);
-       pipefd[1] = 0;
+       pipefd[1] = -1;
 
        ret = pthread_join(t_read, &t_err);
        if (ret) {
@@ -320,16 +329,14 @@ static int do_send(struct btrfs_send *send, u64 root_id, u64 parent_root_id)
                goto out;
        }
 
-       pthread_attr_destroy(&t_attr);
-
        ret = 0;
 
 out:
        if (subvol_fd != -1)
                close(subvol_fd);
-       if (pipefd[0])
+       if (pipefd[0] != -1)
                close(pipefd[0]);
-       if (pipefd[1])
+       if (pipefd[1] != -1)
                close(pipefd[1]);
        return ret;
 }
@@ -354,9 +361,17 @@ static int init_root_path(struct btrfs_send *s, const char *subvol)
 
        ret = find_mount_root(subvol, &s->root_path);
        if (ret < 0) {
+               fprintf(stderr,
+                       "ERROR: failed to determine mount point for %s: %s\n",
+                       subvol, strerror(-ret));
+               ret = -EINVAL;
+               goto out;
+       }
+       if (ret > 0) {
+               fprintf(stderr,
+                       "ERROR: %s doesn't belong to btrfs mount point\n",
+                       subvol);
                ret = -EINVAL;
-               fprintf(stderr, "ERROR: failed to determine mount point "
-                               "for %s\n", subvol);
                goto out;
        }
 
@@ -414,28 +429,42 @@ out:
        return ret;
 }
 
-int cmd_send_start(int argc, char **argv)
+int cmd_send(int argc, char **argv)
 {
        char *subvol = NULL;
-       int c;
        int ret;
-       char *outname = NULL;
+       char outname[PATH_MAX];
        struct btrfs_send send;
        u32 i;
        char *mount_root = NULL;
        char *snapshot_parent = NULL;
-       u64 root_id;
+       u64 root_id = 0;
        u64 parent_root_id = 0;
        int full_send = 1;
+       int new_end_cmd_semantic = 0;
+       u64 send_flags = 0;
 
        memset(&send, 0, sizeof(send));
        send.dump_fd = fileno(stdout);
+       outname[0] = 0;
+
+       while (1) {
+               enum { GETOPT_VAL_SEND_NO_DATA = 256 };
+               static const struct option long_options[] = {
+                       { "no-data", no_argument, NULL, GETOPT_VAL_SEND_NO_DATA }
+               };
+               int c = getopt_long(argc, argv, "vec:f:i:p:", long_options, NULL);
+
+               if (c < 0)
+                       break;
 
-       while ((c = getopt(argc, argv, "vc:f:i:p:")) != -1) {
                switch (c) {
                case 'v':
                        g_verbose++;
                        break;
+               case 'e':
+                       new_end_cmd_semantic = 1;
+                       break;
                case 'c':
                        subvol = realpath(optarg, NULL);
                        if (!subvol) {
@@ -456,17 +485,48 @@ int cmd_send_start(int argc, char **argv)
                                                "root_id for %s\n", subvol);
                                goto out;
                        }
-                       add_clone_source(&send, root_id);
+
+                       ret = is_subvol_ro(&send, subvol);
+                       if (ret < 0)
+                               goto out;
+                       if (!ret) {
+                               ret = -EINVAL;
+                               fprintf(stderr,
+                               "ERROR: cloned subvol %s is not read-only.\n",
+                                       subvol);
+                               goto out;
+                       }
+
+                       ret = add_clone_source(&send, root_id);
+                       if (ret < 0) {
+                               fprintf(stderr, "ERROR: not enough memory\n");
+                               goto out;
+                       }
+                       subvol_uuid_search_finit(&send.sus);
                        free(subvol);
+                       subvol = NULL;
+                       if (send.mnt_fd >= 0) {
+                               close(send.mnt_fd);
+                               send.mnt_fd = -1;
+                       }
+                       free(send.root_path);
+                       send.root_path = NULL;
                        full_send = 0;
                        break;
                case 'f':
-                       outname = optarg;
+                       if (arg_copy_path(outname, optarg, sizeof(outname))) {
+                               fprintf(stderr,
+                                   "ERROR: output file path too long (%zu)\n",
+                                   strlen(optarg));
+                               ret = 1;
+                               goto out;
+                       }
                        break;
                case 'p':
                        if (snapshot_parent) {
                                fprintf(stderr, "ERROR: you cannot have more than one parent (-p)\n");
-                               return 1;
+                               ret = 1;
+                               goto out;
                        }
                        snapshot_parent = realpath(optarg, NULL);
                        if (!snapshot_parent) {
@@ -475,25 +535,40 @@ int cmd_send_start(int argc, char **argv)
                                                "%s\n", optarg, strerror(-ret));
                                goto out;
                        }
+
+                       ret = is_subvol_ro(&send, snapshot_parent);
+                       if (ret < 0)
+                               goto out;
+                       if (!ret) {
+                               ret = -EINVAL;
+                               fprintf(stderr,
+                                       "ERROR: parent %s is not read-only.\n",
+                                       snapshot_parent);
+                               goto out;
+                       }
+
                        full_send = 0;
                        break;
                case 'i':
                        fprintf(stderr,
                                "ERROR: -i was removed, use -c instead\n");
-                       return 1;
+                       ret = 1;
+                       goto out;
+               case GETOPT_VAL_SEND_NO_DATA:
+                       send_flags |= BTRFS_SEND_FLAG_NO_FILE_DATA;
+                       break;
                case '?':
                default:
                        fprintf(stderr, "ERROR: send args invalid.\n");
-                       return 1;
+                       ret = 1;
+                       goto out;
                }
        }
 
-       if (optind == argc) {
-               fprintf(stderr, "ERROR: send needs path to snapshot\n");
-               return 1;
-       }
+       if (check_argc_min(argc - optind, 1))
+               usage(cmd_send_usage);
 
-       if (outname != NULL) {
+       if (outname[0]) {
                send.dump_fd = creat(outname, 0600);
                if (send.dump_fd == -1) {
                        ret = -errno;
@@ -507,7 +582,8 @@ int cmd_send_start(int argc, char **argv)
                fprintf(stderr, 
                        "ERROR: not dumping send stream into a terminal, "
                        "redirect it into a file\n");
-               return 1;
+               ret = 1;
+               goto out;
        }
 
        /* use first send subvol to determine mount_root */
@@ -534,10 +610,15 @@ int cmd_send_start(int argc, char **argv)
                        goto out;
                }
 
-               add_clone_source(&send, parent_root_id);
+               ret = add_clone_source(&send, parent_root_id);
+               if (ret < 0) {
+                       fprintf(stderr, "ERROR: not enough memory\n");
+                       goto out;
+               }
        }
 
        for (i = optind; i < argc; i++) {
+               free(subvol);
                subvol = realpath(argv[i], NULL);
                if (!subvol) {
                        ret = -errno;
@@ -552,6 +633,13 @@ int cmd_send_start(int argc, char **argv)
                                strerror(-ret));
                        goto out;
                }
+               if (ret > 0) {
+                       fprintf(stderr,
+                       "ERROR: %s doesn't belong to btrfs mount point\n",
+                               subvol);
+                       ret = -EINVAL;
+                       goto out;
+               }
                if (strcmp(send.root_path, mount_root) != 0) {
                        ret = -EINVAL;
                        fprintf(stderr, "ERROR: all subvols must be from the "
@@ -569,10 +657,16 @@ int cmd_send_start(int argc, char **argv)
                                        subvol);
                        goto out;
                }
-               free(subvol);
        }
 
+       if (send_flags & BTRFS_SEND_FLAG_NO_FILE_DATA)
+               printf("Mode NO_FILE_DATA enabled\n");
+
        for (i = optind; i < argc; i++) {
+               int is_first_subvol;
+               int is_last_subvol;
+
+               free(subvol);
                subvol = argv[i];
 
                fprintf(stderr, "At subvol %s\n", subvol);
@@ -585,14 +679,6 @@ int cmd_send_start(int argc, char **argv)
                        goto out;
                }
 
-               ret = get_root_id(&send, get_subvol_name(send.root_path, subvol),
-                               &root_id);
-               if (ret < 0) {
-                       fprintf(stderr, "ERROR: could not resolve root_id "
-                                       "for %s\n", subvol);
-                       goto out;
-               }
-
                if (!full_send && !parent_root_id) {
                        ret = find_good_parent(&send, root_id, &parent_root_id);
                        if (ret < 0) {
@@ -612,38 +698,51 @@ int cmd_send_start(int argc, char **argv)
                        goto out;
                }
 
-               ret = do_send(&send, root_id, parent_root_id);
+               if (new_end_cmd_semantic) {
+                       /* require new kernel */
+                       is_first_subvol = (i == optind);
+                       is_last_subvol = (i == argc - 1);
+               } else {
+                       /* be compatible to old and new kernel */
+                       is_first_subvol = 1;
+                       is_last_subvol = 1;
+               }
+               ret = do_send(&send, parent_root_id, is_first_subvol,
+                             is_last_subvol, subvol, send_flags);
                if (ret < 0)
                        goto out;
 
                /* done with this subvol, so add it to the clone sources */
-               add_clone_source(&send, root_id);
+               ret = add_clone_source(&send, root_id);
+               if (ret < 0) {
+                       fprintf(stderr, "ERROR: not enough memory\n");
+                       goto out;
+               }
 
                parent_root_id = 0;
                full_send = 0;
-               free(subvol);
        }
 
        ret = 0;
 
 out:
+       free(subvol);
+       free(snapshot_parent);
+       free(send.clone_sources);
        if (send.mnt_fd >= 0)
                close(send.mnt_fd);
-       return ret;
+       free(send.root_path);
+       subvol_uuid_search_finit(&send.sus);
+       return !!ret;
 }
 
-static const char * const send_cmd_group_usage[] = {
-       "btrfs send <command> <args>",
-       NULL
-};
-
 const char * const cmd_send_usage[] = {
-       "btrfs send [-v] [-p <parent>] [-c <clone-src>] <subvol>",
-       "Send the subvolume to stdout.",
-       "Sends the subvolume specified by <subvol> to stdout.",
+       "btrfs send [-ve] [-p <parent>] [-c <clone-src>] [-f <outfile>] <subvol> [<subvol>...]",
+       "Send the subvolume(s) to stdout.",
+       "Sends the subvolume(s) specified by <subvol> to stdout.",
        "By default, this will send the whole subvolume. To do an incremental",
        "send, use '-p <parent>'. If you want to allow btrfs to clone from",
-       "any additional local snapshots, use -c <clone-src> (multiple times",
+       "any additional local snapshots, use '-c <clone-src>' (multiple times",
        "where applicable). You must not specify clone sources unless you",
        "guarantee that these snapshots are exactly in the same state on both",
        "sides, the sender and the receiver. It is allowed to omit the",
@@ -653,6 +752,8 @@ const char * const cmd_send_usage[] = {
        "\n",
        "-v               Enable verbose debug output. Each occurrence of",
        "                 this option increases the verbose level more.",
+       "-e               If sending multiple subvols at once, use the new",
+       "                 format and omit the end-cmd between the subvols.",
        "-p <parent>      Send an incremental stream from <parent> to",
        "                 <subvol>.",
        "-c <clone-src>   Use this snapshot as a clone source for an ",
@@ -660,10 +761,9 @@ const char * const cmd_send_usage[] = {
        "-f <outfile>     Output is normally written to stdout. To write to",
        "                 a file, use this option. An alternative would be to",
        "                 use pipes.",
+       "--no-data        send in NO_FILE_DATA mode, Note: the output stream",
+       "                 does not contain any file data and thus cannot be used",
+       "                 to transfer changes. This mode is faster and useful to",
+       "                 show the differences in metadata.",
        NULL
 };
-
-int cmd_send(int argc, char **argv)
-{
-       return cmd_send_start(argc, argv);
-}