t/helper/test-tool: implement sha1-unsafe helper

With the new "unsafe" SHA-1 build knob, it is convenient to have a
test-tool that can exercise Git's unsafe SHA-1 wrappers for testing,
similar to 't/helper/test-tool sha1'.

Implement that helper by altering the implementation of that test-tool
(in cmd_hash_impl(), which is generic and parameterized over different
hash functions) to conditionally run the unsafe variants of the chosen
hash function, and expose the new behavior via a new 'sha1-unsafe' test
helper.

Signed-off-by: Taylor Blau <me@ttaylorr.com>
Signed-off-by: Junio C Hamano <gitster@pobox.com>
This commit is contained in:
Taylor Blau
2025-01-23 12:34:19 -05:00
committed by Junio C Hamano
parent fbe8d3079d
commit d9213e4716
6 changed files with 45 additions and 23 deletions

View File

@ -1,7 +1,7 @@
#include "test-tool.h" #include "test-tool.h"
#include "hex.h" #include "hex.h"
int cmd_hash_impl(int ac, const char **av, int algo) int cmd_hash_impl(int ac, const char **av, int algo, int unsafe)
{ {
git_hash_ctx ctx; git_hash_ctx ctx;
unsigned char hash[GIT_MAX_HEXSZ]; unsigned char hash[GIT_MAX_HEXSZ];
@ -27,6 +27,9 @@ int cmd_hash_impl(int ac, const char **av, int algo)
die("OOPS"); die("OOPS");
} }
if (unsafe)
algop->unsafe_init_fn(&ctx);
else
algop->init_fn(&ctx); algop->init_fn(&ctx);
while (1) { while (1) {
@ -46,8 +49,14 @@ int cmd_hash_impl(int ac, const char **av, int algo)
} }
if (this_sz == 0) if (this_sz == 0)
break; break;
if (unsafe)
algop->unsafe_update_fn(&ctx, buffer, this_sz);
else
algop->update_fn(&ctx, buffer, this_sz); algop->update_fn(&ctx, buffer, this_sz);
} }
if (unsafe)
algop->unsafe_final_fn(hash, &ctx);
else
algop->final_fn(hash, &ctx); algop->final_fn(hash, &ctx);
if (binary) if (binary)

View File

@ -3,7 +3,7 @@
int cmd__sha1(int ac, const char **av) int cmd__sha1(int ac, const char **av)
{ {
return cmd_hash_impl(ac, av, GIT_HASH_SHA1); return cmd_hash_impl(ac, av, GIT_HASH_SHA1, 0);
} }
int cmd__sha1_is_sha1dc(int argc UNUSED, const char **argv UNUSED) int cmd__sha1_is_sha1dc(int argc UNUSED, const char **argv UNUSED)
@ -13,3 +13,8 @@ int cmd__sha1_is_sha1dc(int argc UNUSED, const char **argv UNUSED)
#endif #endif
return 1; return 1;
} }
int cmd__sha1_unsafe(int ac, const char **av)
{
return cmd_hash_impl(ac, av, GIT_HASH_SHA1, 1);
}

View File

@ -3,25 +3,31 @@
dd if=/dev/zero bs=1048576 count=100 2>/dev/null | dd if=/dev/zero bs=1048576 count=100 2>/dev/null |
/usr/bin/time t/helper/test-tool sha1 >/dev/null /usr/bin/time t/helper/test-tool sha1 >/dev/null
dd if=/dev/zero bs=1048576 count=100 2>/dev/null |
/usr/bin/time t/helper/test-tool sha1-unsafe >/dev/null
while read expect cnt pfx while read expect cnt pfx
do do
case "$expect" in '#'*) continue ;; esac case "$expect" in '#'*) continue ;; esac
for sha1 in sha1 sha1-unsafe
do
actual=$( actual=$(
{ {
test -z "$pfx" || echo "$pfx" test -z "$pfx" || echo "$pfx"
dd if=/dev/zero bs=1048576 count=$cnt 2>/dev/null | dd if=/dev/zero bs=1048576 count=$cnt 2>/dev/null |
perl -pe 'y/\000/g/' perl -pe 'y/\000/g/'
} | ./t/helper/test-tool sha1 $cnt } | ./t/helper/test-tool $sha1 $cnt
) )
if test "$expect" = "$actual" if test "$expect" = "$actual"
then then
echo "OK: $expect $cnt $pfx" echo "OK ($sha1): $expect $cnt $pfx"
else else
echo >&2 "OOPS: $cnt" echo >&2 "OOPS ($sha1): $cnt"
echo >&2 "expect: $expect" echo >&2 "expect ($sha1): $expect"
echo >&2 "actual: $actual" echo >&2 "actual ($sha1): $actual"
exit 1 exit 1
fi fi
done
done <<EOF done <<EOF
da39a3ee5e6b4b0d3255bfef95601890afd80709 0 da39a3ee5e6b4b0d3255bfef95601890afd80709 0
3f786850e387550fdab836ed7e6dc881de23001b 0 a 3f786850e387550fdab836ed7e6dc881de23001b 0 a

View File

@ -3,5 +3,5 @@
int cmd__sha256(int ac, const char **av) int cmd__sha256(int ac, const char **av)
{ {
return cmd_hash_impl(ac, av, GIT_HASH_SHA256); return cmd_hash_impl(ac, av, GIT_HASH_SHA256, 0);
} }

View File

@ -70,6 +70,7 @@ static struct test_cmd cmds[] = {
{ "serve-v2", cmd__serve_v2 }, { "serve-v2", cmd__serve_v2 },
{ "sha1", cmd__sha1 }, { "sha1", cmd__sha1 },
{ "sha1-is-sha1dc", cmd__sha1_is_sha1dc }, { "sha1-is-sha1dc", cmd__sha1_is_sha1dc },
{ "sha1-unsafe", cmd__sha1_unsafe },
{ "sha256", cmd__sha256 }, { "sha256", cmd__sha256 },
{ "sigchain", cmd__sigchain }, { "sigchain", cmd__sigchain },
{ "simple-ipc", cmd__simple_ipc }, { "simple-ipc", cmd__simple_ipc },

View File

@ -63,6 +63,7 @@ int cmd__scrap_cache_tree(int argc, const char **argv);
int cmd__serve_v2(int argc, const char **argv); int cmd__serve_v2(int argc, const char **argv);
int cmd__sha1(int argc, const char **argv); int cmd__sha1(int argc, const char **argv);
int cmd__sha1_is_sha1dc(int argc, const char **argv); int cmd__sha1_is_sha1dc(int argc, const char **argv);
int cmd__sha1_unsafe(int argc, const char **argv);
int cmd__sha256(int argc, const char **argv); int cmd__sha256(int argc, const char **argv);
int cmd__sigchain(int argc, const char **argv); int cmd__sigchain(int argc, const char **argv);
int cmd__simple_ipc(int argc, const char **argv); int cmd__simple_ipc(int argc, const char **argv);
@ -81,6 +82,6 @@ int cmd__windows_named_pipe(int argc, const char **argv);
#endif #endif
int cmd__write_cache(int argc, const char **argv); int cmd__write_cache(int argc, const char **argv);
int cmd_hash_impl(int ac, const char **av, int algo); int cmd_hash_impl(int ac, const char **av, int algo, int unsafe);
#endif #endif