builtin/mv: refactor to use struct strvec

Memory allocation patterns in git-mv(1) are extremely hard to follow:
We copy around string pointers into manually-managed arrays, some of
which alias each other, but only sometimes, while we also drop some of
those strings at other times without ever daring to free them.

While this may be my own subjective feeling, it seems like others have
given up as the code has multiple calls to `UNLEAK()`. These are not
sufficient though, and git-mv(1) is still leaking all over the place
even with them.

Refactor the code to instead track strings in `struct strvec`. While
this has the effect of effectively duplicating some of the strings
without an actual need, it is way easier to reason about and fixes all
of the aliasing of memory that has been going on. It allows us to get
rid of the `UNLEAK()` calls and also fixes leaks that those calls did
not paper over.

Mark tests which are now leak-free accordingly.

Signed-off-by: Patrick Steinhardt <ps@pks.im>
Signed-off-by: Junio C Hamano <gitster@pobox.com>
This commit is contained in:
Patrick Steinhardt
2024-05-27 13:47:18 +02:00
committed by Junio C Hamano
parent 9fcd9e4e72
commit 52a7dab439
8 changed files with 68 additions and 67 deletions

View File

@ -20,6 +20,7 @@
#include "read-cache-ll.h" #include "read-cache-ll.h"
#include "repository.h" #include "repository.h"
#include "setup.h" #include "setup.h"
#include "strvec.h"
#include "submodule.h" #include "submodule.h"
#include "entry.h" #include "entry.h"
@ -38,42 +39,32 @@ enum update_mode {
#define DUP_BASENAME 1 #define DUP_BASENAME 1
#define KEEP_TRAILING_SLASH 2 #define KEEP_TRAILING_SLASH 2
static const char **internal_prefix_pathspec(const char *prefix, static void internal_prefix_pathspec(struct strvec *out,
const char **pathspec, const char *prefix,
int count, unsigned flags) const char **pathspec,
int count, unsigned flags)
{ {
int i;
const char **result;
int prefixlen = prefix ? strlen(prefix) : 0; int prefixlen = prefix ? strlen(prefix) : 0;
ALLOC_ARRAY(result, count + 1);
/* Create an intermediate copy of the pathspec based on the flags */ /* Create an intermediate copy of the pathspec based on the flags */
for (i = 0; i < count; i++) { for (int i = 0; i < count; i++) {
int length = strlen(pathspec[i]); size_t length = strlen(pathspec[i]);
int to_copy = length; size_t to_copy = length;
char *it; const char *maybe_basename;
char *trimmed, *prefixed_path;
while (!(flags & KEEP_TRAILING_SLASH) && while (!(flags & KEEP_TRAILING_SLASH) &&
to_copy > 0 && is_dir_sep(pathspec[i][to_copy - 1])) to_copy > 0 && is_dir_sep(pathspec[i][to_copy - 1]))
to_copy--; to_copy--;
it = xmemdupz(pathspec[i], to_copy); trimmed = xmemdupz(pathspec[i], to_copy);
if (flags & DUP_BASENAME) { maybe_basename = (flags & DUP_BASENAME) ? basename(trimmed) : trimmed;
result[i] = xstrdup(basename(it)); prefixed_path = prefix_path(prefix, prefixlen, maybe_basename);
free(it); strvec_push(out, prefixed_path);
} else {
result[i] = it;
}
}
result[count] = NULL;
/* Prefix the pathspec and free the old intermediate strings */ free(prefixed_path);
for (i = 0; i < count; i++) { free(trimmed);
const char *match = prefix_path(prefix, prefixlen, result[i]);
free((char *) result[i]);
result[i] = match;
} }
return result;
} }
static char *add_slash(const char *path) static char *add_slash(const char *path)
@ -176,7 +167,10 @@ int cmd_mv(int argc, const char **argv, const char *prefix)
OPT_BOOL(0, "sparse", &ignore_sparse, N_("allow updating entries outside of the sparse-checkout cone")), OPT_BOOL(0, "sparse", &ignore_sparse, N_("allow updating entries outside of the sparse-checkout cone")),
OPT_END(), OPT_END(),
}; };
const char **source, **destination, **dest_path, **submodule_gitfile; struct strvec sources = STRVEC_INIT;
struct strvec dest_paths = STRVEC_INIT;
struct strvec destinations = STRVEC_INIT;
const char **submodule_gitfile;
char *dst_w_slash = NULL; char *dst_w_slash = NULL;
const char **src_dir = NULL; const char **src_dir = NULL;
int src_dir_nr = 0, src_dir_alloc = 0; int src_dir_nr = 0, src_dir_alloc = 0;
@ -201,7 +195,7 @@ int cmd_mv(int argc, const char **argv, const char *prefix)
if (repo_read_index(the_repository) < 0) if (repo_read_index(the_repository) < 0)
die(_("index file corrupt")); die(_("index file corrupt"));
source = internal_prefix_pathspec(prefix, argv, argc, 0); internal_prefix_pathspec(&sources, prefix, argv, argc, 0);
CALLOC_ARRAY(modes, argc); CALLOC_ARRAY(modes, argc);
/* /*
@ -212,41 +206,39 @@ int cmd_mv(int argc, const char **argv, const char *prefix)
flags = KEEP_TRAILING_SLASH; flags = KEEP_TRAILING_SLASH;
if (argc == 1 && is_directory(argv[0]) && !is_directory(argv[1])) if (argc == 1 && is_directory(argv[0]) && !is_directory(argv[1]))
flags = 0; flags = 0;
dest_path = internal_prefix_pathspec(prefix, argv + argc, 1, flags); internal_prefix_pathspec(&dest_paths, prefix, argv + argc, 1, flags);
dst_w_slash = add_slash(dest_path[0]); dst_w_slash = add_slash(dest_paths.v[0]);
submodule_gitfile = xcalloc(argc, sizeof(char *)); submodule_gitfile = xcalloc(argc, sizeof(char *));
if (dest_path[0][0] == '\0') if (dest_paths.v[0][0] == '\0')
/* special case: "." was normalized to "" */ /* special case: "." was normalized to "" */
destination = internal_prefix_pathspec(dest_path[0], argv, argc, DUP_BASENAME); internal_prefix_pathspec(&destinations, dest_paths.v[0], argv, argc, DUP_BASENAME);
else if (!lstat(dest_path[0], &st) && else if (!lstat(dest_paths.v[0], &st) && S_ISDIR(st.st_mode)) {
S_ISDIR(st.st_mode)) { internal_prefix_pathspec(&destinations, dst_w_slash, argv, argc, DUP_BASENAME);
destination = internal_prefix_pathspec(dst_w_slash, argv, argc, DUP_BASENAME); } else if (!path_in_sparse_checkout(dst_w_slash, the_repository->index) &&
empty_dir_has_sparse_contents(dst_w_slash)) {
internal_prefix_pathspec(&destinations, dst_w_slash, argv, argc, DUP_BASENAME);
dst_mode = SKIP_WORKTREE_DIR;
} else if (argc != 1) {
die(_("destination '%s' is not a directory"), dest_paths.v[0]);
} else { } else {
if (!path_in_sparse_checkout(dst_w_slash, the_repository->index) && strvec_pushv(&destinations, dest_paths.v);
empty_dir_has_sparse_contents(dst_w_slash)) {
destination = internal_prefix_pathspec(dst_w_slash, argv, argc, DUP_BASENAME); /*
dst_mode = SKIP_WORKTREE_DIR; * <destination> is a file outside of sparse-checkout
} else if (argc != 1) { * cone. Insist on cone mode here for backward
die(_("destination '%s' is not a directory"), dest_path[0]); * compatibility. We don't want dst_mode to be assigned
} else { * for a file when the repo is using no-cone mode (which
destination = dest_path; * is deprecated at this point) sparse-checkout. As
/* * SPARSE here is only considering cone-mode situation.
* <destination> is a file outside of sparse-checkout */
* cone. Insist on cone mode here for backward if (!path_in_cone_mode_sparse_checkout(destinations.v[0], the_repository->index))
* compatibility. We don't want dst_mode to be assigned dst_mode = SPARSE;
* for a file when the repo is using no-cone mode (which
* is deprecated at this point) sparse-checkout. As
* SPARSE here is only considering cone-mode situation.
*/
if (!path_in_cone_mode_sparse_checkout(destination[0], the_repository->index))
dst_mode = SPARSE;
}
} }
/* Checking */ /* Checking */
for (i = 0; i < argc; i++) { for (i = 0; i < argc; i++) {
const char *src = source[i], *dst = destination[i]; const char *src = sources.v[i], *dst = destinations.v[i];
int length; int length;
const char *bad = NULL; const char *bad = NULL;
int skip_sparse = 0; int skip_sparse = 0;
@ -330,8 +322,6 @@ dir_check:
src_dir[src_dir_nr++] = src; src_dir[src_dir_nr++] = src;
n = argc + last - first; n = argc + last - first;
REALLOC_ARRAY(source, n);
REALLOC_ARRAY(destination, n);
REALLOC_ARRAY(modes, n); REALLOC_ARRAY(modes, n);
REALLOC_ARRAY(submodule_gitfile, n); REALLOC_ARRAY(submodule_gitfile, n);
@ -341,12 +331,16 @@ dir_check:
for (j = 0; j < last - first; j++) { for (j = 0; j < last - first; j++) {
const struct cache_entry *ce = the_repository->index->cache[first + j]; const struct cache_entry *ce = the_repository->index->cache[first + j];
const char *path = ce->name; const char *path = ce->name;
source[argc + j] = path; char *prefixed_path = prefix_path(dst_with_slash, dst_with_slash_len, path + length + 1);
destination[argc + j] =
prefix_path(dst_with_slash, dst_with_slash_len, path + length + 1); strvec_push(&sources, path);
strvec_push(&destinations, prefixed_path);
memset(modes + argc + j, 0, sizeof(enum update_mode)); memset(modes + argc + j, 0, sizeof(enum update_mode));
modes[argc + j] |= ce_skip_worktree(ce) ? SPARSE : INDEX; modes[argc + j] |= ce_skip_worktree(ce) ? SPARSE : INDEX;
submodule_gitfile[argc + j] = NULL; submodule_gitfile[argc + j] = NULL;
free(prefixed_path);
} }
free(dst_with_slash); free(dst_with_slash);
@ -430,8 +424,8 @@ act_on_entry:
remove_entry: remove_entry:
if (--argc > 0) { if (--argc > 0) {
int n = argc - i; int n = argc - i;
MOVE_ARRAY(source + i, source + i + 1, n); strvec_remove(&sources, i);
MOVE_ARRAY(destination + i, destination + i + 1, n); strvec_remove(&destinations, i);
MOVE_ARRAY(modes + i, modes + i + 1, n); MOVE_ARRAY(modes + i, modes + i + 1, n);
MOVE_ARRAY(submodule_gitfile + i, MOVE_ARRAY(submodule_gitfile + i,
submodule_gitfile + i + 1, n); submodule_gitfile + i + 1, n);
@ -448,7 +442,7 @@ remove_entry:
} }
for (i = 0; i < argc; i++) { for (i = 0; i < argc; i++) {
const char *src = source[i], *dst = destination[i]; const char *src = sources.v[i], *dst = destinations.v[i];
enum update_mode mode = modes[i]; enum update_mode mode = modes[i];
int pos; int pos;
int sparse_and_dirty = 0; int sparse_and_dirty = 0;
@ -576,8 +570,9 @@ out:
string_list_clear(&src_for_dst, 0); string_list_clear(&src_for_dst, 0);
string_list_clear(&dirty_paths, 0); string_list_clear(&dirty_paths, 0);
string_list_clear(&only_match_skip_worktree, 0); string_list_clear(&only_match_skip_worktree, 0);
UNLEAK(source); strvec_clear(&sources);
UNLEAK(dest_path); strvec_clear(&dest_paths);
strvec_clear(&destinations);
free(submodule_gitfile); free(submodule_gitfile);
free(modes); free(modes);
return ret; return ret;

View File

@ -3,9 +3,9 @@
# Copyright (c) 2005 Junio C Hamano # Copyright (c) 2005 Junio C Hamano
# #
test_description='Test rename detection in diff engine. test_description='Test rename detection in diff engine.'
' TEST_PASSES_SANITIZE_LEAK=true
. ./test-lib.sh . ./test-lib.sh
. "$TEST_DIRECTORY"/lib-diff.sh . "$TEST_DIRECTORY"/lib-diff.sh

View File

@ -5,6 +5,7 @@
test_description='Move a binary file' test_description='Move a binary file'
TEST_PASSES_SANITIZE_LEAK=true
. ./test-lib.sh . ./test-lib.sh

View File

@ -5,6 +5,7 @@
test_description='git apply -p handling.' test_description='git apply -p handling.'
TEST_PASSES_SANITIZE_LEAK=true
. ./test-lib.sh . ./test-lib.sh
test_expect_success setup ' test_expect_success setup '

View File

@ -7,6 +7,7 @@ test_description='Test merge with directory/file conflicts'
GIT_TEST_DEFAULT_INITIAL_BRANCH_NAME=main GIT_TEST_DEFAULT_INITIAL_BRANCH_NAME=main
export GIT_TEST_DEFAULT_INITIAL_BRANCH_NAME export GIT_TEST_DEFAULT_INITIAL_BRANCH_NAME
TEST_PASSES_SANITIZE_LEAK=true
. ./test-lib.sh . ./test-lib.sh
test_expect_success 'prepare repository' ' test_expect_success 'prepare repository' '

View File

@ -4,6 +4,7 @@ test_description='merging with large rename matrix'
GIT_TEST_DEFAULT_INITIAL_BRANCH_NAME=main GIT_TEST_DEFAULT_INITIAL_BRANCH_NAME=main
export GIT_TEST_DEFAULT_INITIAL_BRANCH_NAME export GIT_TEST_DEFAULT_INITIAL_BRANCH_NAME
TEST_PASSES_SANITIZE_LEAK=true
. ./test-lib.sh . ./test-lib.sh
count() { count() {

View File

@ -22,6 +22,7 @@ test_description="merge cases"
# underscore notation is to differentiate different # underscore notation is to differentiate different
# files that might be renamed into each other's paths.) # files that might be renamed into each other's paths.)
TEST_PASSES_SANITIZE_LEAK=true
. ./test-lib.sh . ./test-lib.sh
. "$TEST_DIRECTORY"/lib-merge.sh . "$TEST_DIRECTORY"/lib-merge.sh

View File

@ -2,6 +2,7 @@
test_description="remember regular & dir renames in sequence of merges" test_description="remember regular & dir renames in sequence of merges"
TEST_PASSES_SANITIZE_LEAK=true
. ./test-lib.sh . ./test-lib.sh
# #