]> git.ipfire.org Git - thirdparty/git.git/blob - t/helper/test-bloom.c
clone: allow "--bare" with "-o"
[thirdparty/git.git] / t / helper / test-bloom.c
1 #include "git-compat-util.h"
2 #include "bloom.h"
3 #include "test-tool.h"
4 #include "commit.h"
5
6 static struct bloom_filter_settings settings = DEFAULT_BLOOM_FILTER_SETTINGS;
7
8 static void add_string_to_filter(const char *data, struct bloom_filter *filter) {
9 struct bloom_key key;
10 int i;
11
12 fill_bloom_key(data, strlen(data), &key, &settings);
13 printf("Hashes:");
14 for (i = 0; i < settings.num_hashes; i++){
15 printf("0x%08x|", key.hashes[i]);
16 }
17 printf("\n");
18 add_key_to_filter(&key, filter, &settings);
19 }
20
21 static void print_bloom_filter(struct bloom_filter *filter) {
22 int i;
23
24 if (!filter) {
25 printf("No filter.\n");
26 return;
27 }
28 printf("Filter_Length:%d\n", (int)filter->len);
29 printf("Filter_Data:");
30 for (i = 0; i < filter->len; i++) {
31 printf("%02x|", filter->data[i]);
32 }
33 printf("\n");
34 }
35
36 static void get_bloom_filter_for_commit(const struct object_id *commit_oid)
37 {
38 struct commit *c;
39 struct bloom_filter *filter;
40 setup_git_directory();
41 c = lookup_commit(the_repository, commit_oid);
42 filter = get_or_compute_bloom_filter(the_repository, c, 1,
43 &settings,
44 NULL);
45 print_bloom_filter(filter);
46 }
47
48 static const char *bloom_usage = "\n"
49 " test-tool bloom get_murmur3 <string>\n"
50 " test-tool bloom generate_filter <string> [<string>...]\n"
51 " test-tool bloom get_filter_for_commit <commit-hex>\n";
52
53 int cmd__bloom(int argc, const char **argv)
54 {
55 setup_git_directory();
56
57 if (argc < 2)
58 usage(bloom_usage);
59
60 if (!strcmp(argv[1], "get_murmur3")) {
61 uint32_t hashed;
62 if (argc < 3)
63 usage(bloom_usage);
64 hashed = murmur3_seeded(0, argv[2], strlen(argv[2]));
65 printf("Murmur3 Hash with seed=0:0x%08x\n", hashed);
66 }
67
68 if (!strcmp(argv[1], "generate_filter")) {
69 struct bloom_filter filter;
70 int i = 2;
71 filter.len = (settings.bits_per_entry + BITS_PER_WORD - 1) / BITS_PER_WORD;
72 CALLOC_ARRAY(filter.data, filter.len);
73
74 if (argc - 1 < i)
75 usage(bloom_usage);
76
77 while (argv[i]) {
78 add_string_to_filter(argv[i], &filter);
79 i++;
80 }
81
82 print_bloom_filter(&filter);
83 }
84
85 if (!strcmp(argv[1], "get_filter_for_commit")) {
86 struct object_id oid;
87 const char *end;
88 if (argc < 3)
89 usage(bloom_usage);
90 if (parse_oid_hex(argv[2], &oid, &end))
91 die("cannot parse oid '%s'", argv[2]);
92 init_bloom_filters();
93 get_bloom_filter_for_commit(&oid);
94 }
95
96 return 0;
97 }