Code

Further cleanup IO startup and initialization; fix io_run_append
[tig.git] / tig.c
diff --git a/tig.c b/tig.c
index eaff28cdc04f98bdbffbd4f8e89b8da8e054feb0..77e1d8183372669bc1fed7fda0cbb97ada649417 100644 (file)
--- a/tig.c
+++ b/tig.c
@@ -140,13 +140,6 @@ static struct ref_list *get_ref_list(const char *id);
 static void foreach_ref(bool (*visitor)(void *data, const struct ref *ref), void *data);
 static int load_refs(void);
 
-enum format_flags {
-       FORMAT_ALL,             /* Perform replacement in all arguments. */
-       FORMAT_NONE             /* No replacement should be performed. */
-};
-
-static bool format_argv(const char *dst[], const char *src[], enum format_flags flags);
-
 enum input_status {
        INPUT_OK,
        INPUT_SKIP,
@@ -685,6 +678,18 @@ argv_free(const char *argv[])
 
        for (argc = 0; argv[argc]; argc++)
                free((void *) argv[argc]);
+       argv[0] = NULL;
+}
+
+static bool
+argv_copy(const char *dst[], const char *src[])
+{
+       int argc;
+
+       for (argc = 0; src[argc]; argc++)
+               if (!(dst[argc] = strdup(src[argc])))
+                       return FALSE;
+       return TRUE;
 }
 
 
@@ -702,12 +707,9 @@ enum io_type {
 };
 
 struct io {
-       enum io_type type;      /* The requested type of pipe. */
-       const char *dir;        /* Directory from which to execute. */
-       pid_t pid;              /* PID of spawned process. */
        int pipe;               /* Pipe end for reading or writing. */
+       pid_t pid;              /* PID of spawned process. */
        int error;              /* Error status. */
-       const char *argv[SIZEOF_ARG];   /* Shell command arguments. */
        char *buf;              /* Read buffer. */
        size_t bufalloc;        /* Allocated buffer size. */
        size_t bufsize;         /* Buffer content size. */
@@ -716,30 +718,10 @@ struct io {
 };
 
 static void
-io_reset(struct io *io)
+io_init(struct io *io)
 {
+       memset(io, 0, sizeof(*io));
        io->pipe = -1;
-       io->pid = 0;
-       io->buf = io->bufpos = NULL;
-       io->bufalloc = io->bufsize = 0;
-       io->error = 0;
-       io->eof = 0;
-}
-
-static void
-io_init(struct io *io, const char *dir, enum io_type type)
-{
-       io_reset(io);
-       io->type = type;
-       io->dir = dir;
-}
-
-static bool
-io_format(struct io *io, const char *dir, enum io_type type,
-         const char *argv[], enum format_flags flags)
-{
-       io_init(io, dir, type);
-       return format_argv(io->argv, argv, flags);
 }
 
 static bool
@@ -749,7 +731,7 @@ io_open(struct io *io, const char *fmt, ...)
        bool fits;
        va_list args;
 
-       io_init(io, NULL, IO_FD);
+       io_init(io);
 
        va_start(args, fmt);
        fits = vsnprintf(name, sizeof(name), fmt, args) < sizeof(name);
@@ -779,7 +761,7 @@ io_done(struct io *io)
        if (io->pipe != -1)
                close(io->pipe);
        free(io->buf);
-       io_reset(io);
+       io_init(io);
 
        while (pid > 0) {
                int status;
@@ -802,35 +784,37 @@ io_done(struct io *io)
 }
 
 static bool
-io_start(struct io *io)
+io_run(struct io *io, enum io_type type, const char *dir, const char *argv[], ...)
 {
        int pipefds[2] = { -1, -1 };
+       va_list args;
 
-       if (io->type == IO_FD)
-               return TRUE;
+       io_init(io);
 
-       if ((io->type == IO_RD || io->type == IO_WR) && pipe(pipefds) < 0) {
+       if ((type == IO_RD || type == IO_WR) && pipe(pipefds) < 0) {
                io->error = errno;
                return FALSE;
-       } else if (io->type == IO_AP) {
-               pipefds[1] = io->pipe;
+       } else if (type == IO_AP) {
+               va_start(args, argv);
+               pipefds[1] = va_arg(args, int);
+               va_end(args);
        }
 
        if ((io->pid = fork())) {
                if (io->pid == -1)
                        io->error = errno;
-               if (pipefds[!(io->type == IO_WR)] != -1)
-                       close(pipefds[!(io->type == IO_WR)]);
+               if (pipefds[!(type == IO_WR)] != -1)
+                       close(pipefds[!(type == IO_WR)]);
                if (io->pid != -1) {
-                       io->pipe = pipefds[!!(io->type == IO_WR)];
+                       io->pipe = pipefds[!!(type == IO_WR)];
                        return TRUE;
                }
 
        } else {
-               if (io->type != IO_FG) {
+               if (type != IO_FG) {
                        int devnull = open("/dev/null", O_RDWR);
-                       int readfd  = io->type == IO_WR ? pipefds[0] : devnull;
-                       int writefd = (io->type == IO_RD || io->type == IO_AP)
+                       int readfd  = type == IO_WR ? pipefds[0] : devnull;
+                       int writefd = (type == IO_RD || type == IO_AP)
                                                        ? pipefds[1] : devnull;
 
                        dup2(readfd,  STDIN_FILENO);
@@ -844,71 +828,42 @@ io_start(struct io *io)
                                close(pipefds[1]);
                }
 
-               if (io->dir && *io->dir && chdir(io->dir) == -1)
+               if (dir && *dir && chdir(dir) == -1)
                        exit(errno);
 
-               execvp(io->argv[0], (char *const*) io->argv);
+               execvp(argv[0], (char *const*) argv);
                exit(errno);
        }
 
-       if (pipefds[!!(io->type == IO_WR)] != -1)
-               close(pipefds[!!(io->type == IO_WR)]);
+       if (pipefds[!!(type == IO_WR)] != -1)
+               close(pipefds[!!(type == IO_WR)]);
        return FALSE;
 }
 
 static bool
-io_run(struct io *io, const char **argv, const char *dir, enum io_type type)
+io_complete(enum io_type type, const char **argv, const char *dir, int fd)
 {
-       io_init(io, dir, type);
-       if (!format_argv(io->argv, argv, FORMAT_NONE))
-               return FALSE;
-       return io_start(io);
-}
+       struct io io;
 
-static int
-io_complete(struct io *io)
-{
-       return io_start(io) && io_done(io);
+       return io_run(&io, type, dir, argv, fd) && io_done(&io);
 }
 
-static int
+static bool
 io_run_bg(const char **argv)
 {
-       struct io io = {};
-
-       if (!io_format(&io, NULL, IO_BG, argv, FORMAT_NONE))
-               return FALSE;
-       return io_complete(&io);
+       return io_complete(IO_BG, argv, NULL, -1);
 }
 
 static bool
 io_run_fg(const char **argv, const char *dir)
 {
-       struct io io = {};
-
-       if (!io_format(&io, dir, IO_FG, argv, FORMAT_NONE))
-               return FALSE;
-       return io_complete(&io);
+       return io_complete(IO_FG, argv, dir, -1);
 }
 
 static bool
-io_run_append(const char **argv, enum format_flags flags, int fd)
+io_run_append(const char **argv, int fd)
 {
-       struct io io = {};
-
-       if (!io_format(&io, NULL, IO_AP, argv, flags)) {
-               close(fd);
-               return FALSE;
-       }
-
-       io.pipe = fd;
-       return io_complete(&io);
-}
-
-static bool
-io_run_rd(struct io *io, const char **argv, const char *dir, enum format_flags flags)
-{
-       return io_format(io, dir, IO_RD, argv, flags) && io_start(io);
+       return io_complete(IO_AP, argv, NULL, fd);
 }
 
 static bool
@@ -1043,10 +998,9 @@ io_read_buf(struct io *io, char buf[], size_t bufsize)
 static bool
 io_run_buf(const char **argv, char buf[], size_t bufsize)
 {
-       struct io io = {};
+       struct io io;
 
-       return io_run_rd(&io, argv, NULL, FORMAT_NONE)
-           && io_read_buf(&io, buf, bufsize);
+       return io_run(&io, IO_RD, NULL, argv) && io_read_buf(&io, buf, bufsize);
 }
 
 static int
@@ -1056,9 +1010,6 @@ io_load(struct io *io, const char *separators,
        char *name;
        int state = OK;
 
-       if (!io_start(io))
-               return ERR;
-
        while (state == OK && (name = io_get(io, '\n', TRUE))) {
                char *value;
                size_t namelen;
@@ -1091,10 +1042,11 @@ static int
 io_run_load(const char **argv, const char *separators,
            int (*read_property)(char *, size_t, char *, size_t))
 {
-       struct io io = {};
+       struct io io;
 
-       return io_format(&io, NULL, IO_RD, argv, FORMAT_NONE)
-               ? io_load(&io, separators, read_property) : ERR;
+       if (!io_run(&io, IO_RD, NULL, argv))
+               return ERR;
+       return io_load(&io, separators, read_property);
 }
 
 
@@ -1746,7 +1698,7 @@ add_run_request(enum keymap keymap, int key, int argc, const char **argv)
        req->key = key;
        req->argv[0] = NULL;
 
-       if (!format_argv(req->argv, argv, FORMAT_NONE))
+       if (!argv_copy(req->argv, argv))
                return REQ_NONE;
 
        return REQ_NONE + ++run_requests;
@@ -2136,7 +2088,7 @@ read_option(char *opt, size_t optlen, char *value, size_t valuelen)
 static void
 load_option_file(const char *path)
 {
-       struct io io = {};
+       struct io io;
 
        /* It's OK that the file doesn't exist. */
        if (!io_open(&io, "%s", path))
@@ -2261,6 +2213,8 @@ struct view {
        bool has_scrolled;      /* View was scrolled. */
 
        /* Loading */
+       const char *argv[SIZEOF_ARG];   /* Shell command arguments. */
+       const char *dir;        /* Directory from which to execute. */
        struct io io;
        struct io *pipe;
        time_t start_time;
@@ -3188,11 +3142,10 @@ format_arg(const char *name)
 }
 
 static bool
-format_argv(const char *dst_argv[], const char *src_argv[], enum format_flags flags)
+format_argv(const char *dst_argv[], const char *src_argv[], bool replace)
 {
        char buf[SIZEOF_STR];
        int argc;
-       bool noreplace = flags == FORMAT_NONE;
 
        argv_free(dst_argv);
 
@@ -3205,7 +3158,7 @@ format_argv(const char *dst_argv[], const char *src_argv[], enum format_flags fl
                        int len = next - arg;
                        const char *value;
 
-                       if (!next || noreplace) {
+                       if (!next || !replace) {
                                len = strlen(arg);
                                value = "";
 
@@ -3220,7 +3173,7 @@ format_argv(const char *dst_argv[], const char *src_argv[], enum format_flags fl
                        if (!string_format_from(buf, &bufpos, "%.*s%s", len, arg, value))
                                return FALSE;
 
-                       arg = next && !noreplace ? strchr(next, ')') + 1 : NULL;
+                       arg = next && replace ? strchr(next, ')') + 1 : NULL;
                }
 
                dst_argv[argc] = strdup(buf);
@@ -3279,12 +3232,28 @@ setup_update(struct view *view, const char *vid)
        view->start_time = time(NULL);
 }
 
+static bool
+prepare_io(struct view *view, const char *dir, const char *argv[], bool replace)
+{
+       view->dir = dir;
+       return format_argv(view->argv, argv, replace);
+}
+
 static bool
 prepare_update(struct view *view, const char *argv[], const char *dir)
 {
        if (view->pipe)
                end_update(view, TRUE);
-       return io_format(&view->io, dir, IO_RD, argv, FORMAT_NONE);
+       return prepare_io(view, dir, argv, FALSE);
+}
+
+static bool
+start_update(struct view *view, const char **argv, const char *dir)
+{
+       if (view->pipe)
+               io_done(view->pipe);
+       return prepare_io(view, dir, argv, FALSE) &&
+              io_run(&view->io, IO_RD, dir, view->argv);
 }
 
 static bool
@@ -3292,6 +3261,7 @@ prepare_update_file(struct view *view, const char *name)
 {
        if (view->pipe)
                end_update(view, TRUE);
+       argv_free(view->argv);
        return io_open(&view->io, "%s/%s", opt_cdup[0] ? opt_cdup : ".", name);
 }
 
@@ -3305,7 +3275,7 @@ begin_update(struct view *view, bool refresh)
                if (view->ops->prepare) {
                        if (!view->ops->prepare(view))
                                return FALSE;
-               } else if (!io_format(&view->io, NULL, IO_RD, view->ops->argv, FORMAT_ALL)) {
+               } else if (!prepare_io(view, NULL, view->ops->argv, TRUE)) {
                        return FALSE;
                }
 
@@ -3316,7 +3286,7 @@ begin_update(struct view *view, bool refresh)
                string_copy_rev(view->ref, view->id);
        }
 
-       if (!io_start(&view->io))
+       if (view->argv[0] && !io_run(&view->io, IO_RD, view->dir, view->argv))
                return FALSE;
 
        setup_update(view, view->id);
@@ -3602,7 +3572,7 @@ open_run_request(enum request request)
                return;
        }
 
-       if (format_argv(argv, req->argv, FORMAT_ALL))
+       if (format_argv(argv, req->argv, TRUE))
                open_external_viewer(argv, NULL);
        argv_free(argv);
 }
@@ -4519,7 +4489,6 @@ tree_read_date(struct view *view, char *text, bool *read_date)
                        "git", "log", "--no-color", "--pretty=raw",
                                "--cc", "--raw", view->id, "--", path, NULL
                };
-               struct io io = {};
 
                if (!view->lines) {
                        tree_entry(view, LINE_TREE_HEAD, opt_path, NULL, NULL);
@@ -4527,13 +4496,11 @@ tree_read_date(struct view *view, char *text, bool *read_date)
                        return TRUE;
                }
 
-               if (!io_run_rd(&io, log_file, opt_cdup, FORMAT_NONE)) {
+               if (!start_update(view, log_file, opt_cdup)) {
                        report("Failed to load tree data");
                        return TRUE;
                }
 
-               io_done(view->pipe);
-               view->io = io;
                *read_date = TRUE;
                return FALSE;
 
@@ -4670,7 +4637,7 @@ open_blob_editor(const char *id)
 
        if (fd == -1)
                report("Failed to create temporary file");
-       else if (!io_run_append(blob_argv, FORMAT_NONE, fd))
+       else if (!io_run_append(blob_argv, fd))
                report("Failed to save blob data to file");
        else
                open_editor(file);
@@ -4817,7 +4784,7 @@ tree_prepare(struct view *view)
                opt_path[0] = 0;
        }
 
-       return io_format(&view->io, opt_cdup, IO_RD, view->ops->argv, FORMAT_ALL);
+       return prepare_io(view, opt_cdup, view->ops->argv, TRUE);
 }
 
 static const char *tree_argv[SIZEOF_ARG] = {
@@ -4882,18 +4849,6 @@ static struct view_ops blob_ops = {
  *     reading output from git-blame.
  */
 
-static const char *blame_head_argv[] = {
-       "git", "blame", "--incremental", "--", "%(file)", NULL
-};
-
-static const char *blame_ref_argv[] = {
-       "git", "blame", "--incremental", "%(ref)", "--", "%(file)", NULL
-};
-
-static const char *blame_cat_file_argv[] = {
-       "git", "cat-file", "blob", "%(ref):%(file)", NULL
-};
-
 struct blame_commit {
        char id[SIZEOF_REV];            /* SHA1 ID. */
        char title[128];                /* First line of the commit message. */
@@ -4921,7 +4876,12 @@ blame_open(struct view *view)
        }
 
        if (*opt_ref || !io_open(&view->io, "%s%s", opt_cdup, opt_file)) {
-               if (!io_run_rd(&view->io, blame_cat_file_argv, opt_cdup, FORMAT_ALL))
+               const char *blame_cat_file_argv[] = {
+                       "git", "cat-file", "blob", path, NULL
+               };
+
+               if (!string_format(path, "%s:%s", opt_ref, opt_file) ||
+                   !start_update(view, blame_cat_file_argv, opt_cdup))
                        return FALSE;
        }
 
@@ -5011,19 +4971,19 @@ static bool
 blame_read_file(struct view *view, const char *line, bool *read_file)
 {
        if (!line) {
-               const char **argv = *opt_ref ? blame_ref_argv : blame_head_argv;
-               struct io io = {};
+               const char *blame_argv[] = {
+                       "git", "blame", "--incremental",
+                               *opt_ref ? opt_ref : "--incremental", "--", opt_file, NULL
+               };
 
                if (view->lines == 0 && !view->prev)
                        die("No blame exist for %s", view->vid);
 
-               if (view->lines == 0 || !io_run_rd(&io, argv, opt_cdup, FORMAT_ALL)) {
+               if (view->lines == 0 || !start_update(view, blame_argv, opt_cdup)) {
                        report("Failed to load blame data");
                        return TRUE;
                }
 
-               io_done(view->pipe);
-               view->io = io;
                *read_file = FALSE;
                return FALSE;
 
@@ -5154,12 +5114,12 @@ setup_blame_parent_line(struct view *view, struct blame *blame)
                "git", "diff-tree", "-U0", blame->commit->id,
                        "--", blame->commit->filename, NULL
        };
-       struct io io = {};
+       struct io io;
        int parent_lineno = -1;
        int blamed_lineno = -1;
        char *line;
 
-       if (!io_run(&io, diff_tree_argv, NULL, IO_RD))
+       if (!io_run(&io, IO_RD, NULL, diff_tree_argv))
                return;
 
        while ((line = io_get(&io, '\n', TRUE))) {
@@ -5451,7 +5411,7 @@ branch_open(struct view *view)
                        "--simplify-by-decoration", "--all", NULL
        };
 
-       if (!io_run_rd(&view->io, branch_log, NULL, FORMAT_NONE)) {
+       if (!start_update(view, branch_log, NULL)) {
                report("Failed to load branch data");
                return TRUE;
        }
@@ -5570,9 +5530,9 @@ status_run(struct view *view, const char *argv[], char status, enum line_type ty
 {
        struct status *unmerged = NULL;
        char *buf;
-       struct io io = {};
+       struct io io;
 
-       if (!io_run(&io, argv, opt_cdup, IO_RD))
+       if (!io_run(&io, IO_RD, opt_cdup, argv))
                return FALSE;
 
        add_line_data(view, NULL, type);
@@ -5720,7 +5680,7 @@ status_update_onbranch(void)
                        continue;
 
                if (!*opt_head) {
-                       struct io io = {};
+                       struct io io;
 
                        if (io_open(&io, "%s/rebase-merge/head-name", opt_git_dir) &&
                            io_read_buf(&io, buf, sizeof(buf))) {
@@ -5967,11 +5927,11 @@ status_update_prepare(struct io *io, enum line_type type)
 
        switch (type) {
        case LINE_STAT_STAGED:
-               return io_run(io, staged_argv, opt_cdup, IO_WR);
+               return io_run(io, IO_WR, opt_cdup, staged_argv);
 
        case LINE_STAT_UNSTAGED:
        case LINE_STAT_UNTRACKED:
-               return io_run(io, others_argv, opt_cdup, IO_WR);
+               return io_run(io, IO_WR, opt_cdup, others_argv);
 
        default:
                die("line type %d not handled in switch", type);
@@ -6010,7 +5970,7 @@ status_update_write(struct io *io, struct status *status, enum line_type type)
 static bool
 status_update_file(struct status *status, enum line_type type)
 {
-       struct io io = {};
+       struct io io;
        bool result;
 
        if (!status_update_prepare(&io, type))
@@ -6024,7 +5984,7 @@ static bool
 status_update_files(struct view *view, struct line *line)
 {
        char buf[sizeof(view->ref)];
-       struct io io = {};
+       struct io io;
        bool result = TRUE;
        struct line *pos = view->line + view->lines;
        int files = 0;
@@ -6297,7 +6257,7 @@ stage_apply_chunk(struct view *view, struct line *chunk, bool revert)
                "git", "apply", "--whitespace=nowarn", NULL
        };
        struct line *diff_hdr;
-       struct io io = {};
+       struct io io;
        int argc = 3;
 
        diff_hdr = stage_diff_find(view, chunk, LINE_DIFF_HEADER);
@@ -6310,7 +6270,7 @@ stage_apply_chunk(struct view *view, struct line *chunk, bool revert)
                apply_argv[argc++] = "-R";
        apply_argv[argc++] = "-";
        apply_argv[argc++] = NULL;
-       if (!io_run(&io, apply_argv, opt_cdup, IO_WR))
+       if (!io_run(&io, IO_WR, opt_cdup, apply_argv))
                return FALSE;
 
        if (!stage_diff_write(&io, diff_hdr, chunk) ||