]>
Commit | Line | Data |
---|---|---|
f52207a4 GS |
1 | #include "git-compat-util.h" |
2 | #include "bloom.h" | |
3 | #include "test-tool.h" | |
ed591feb | 4 | #include "commit.h" |
f52207a4 | 5 | |
066b70ae | 6 | static struct bloom_filter_settings settings = DEFAULT_BLOOM_FILTER_SETTINGS; |
f1294eaf GS |
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); | |
9794633b | 19 | clear_bloom_key(&key); |
f1294eaf GS |
20 | } |
21 | ||
22 | static void print_bloom_filter(struct bloom_filter *filter) { | |
23 | int i; | |
24 | ||
25 | if (!filter) { | |
26 | printf("No filter.\n"); | |
27 | return; | |
28 | } | |
29 | printf("Filter_Length:%d\n", (int)filter->len); | |
30 | printf("Filter_Data:"); | |
24b7d1e7 | 31 | for (i = 0; i < filter->len; i++) { |
f1294eaf GS |
32 | printf("%02x|", filter->data[i]); |
33 | } | |
34 | printf("\n"); | |
35 | } | |
36 | ||
ed591feb GS |
37 | static void get_bloom_filter_for_commit(const struct object_id *commit_oid) |
38 | { | |
39 | struct commit *c; | |
40 | struct bloom_filter *filter; | |
41 | setup_git_directory(); | |
42 | c = lookup_commit(the_repository, commit_oid); | |
312cff52 | 43 | filter = get_or_compute_bloom_filter(the_repository, c, 1, |
9a7a9ed1 | 44 | &settings, |
312cff52 | 45 | NULL); |
ed591feb GS |
46 | print_bloom_filter(filter); |
47 | } | |
48 | ||
1b4c57fa | 49 | static const char *bloom_usage = "\n" |
54c337be | 50 | " test-tool bloom get_murmur3 <string>\n" |
1b4c57fa | 51 | " test-tool bloom generate_filter <string> [<string>...]\n" |
dba94e3a | 52 | " test-tool bloom get_filter_for_commit <commit-hex>\n"; |
1b4c57fa | 53 | |
f52207a4 GS |
54 | int cmd__bloom(int argc, const char **argv) |
55 | { | |
094a685c | 56 | setup_git_directory(); |
57 | ||
1b4c57fa JK |
58 | if (argc < 2) |
59 | usage(bloom_usage); | |
60 | ||
f52207a4 | 61 | if (!strcmp(argv[1], "get_murmur3")) { |
1b4c57fa JK |
62 | uint32_t hashed; |
63 | if (argc < 3) | |
64 | usage(bloom_usage); | |
65 | hashed = murmur3_seeded(0, argv[2], strlen(argv[2])); | |
f52207a4 GS |
66 | printf("Murmur3 Hash with seed=0:0x%08x\n", hashed); |
67 | } | |
68 | ||
24b7d1e7 | 69 | if (!strcmp(argv[1], "generate_filter")) { |
f1294eaf GS |
70 | struct bloom_filter filter; |
71 | int i = 2; | |
72 | filter.len = (settings.bits_per_entry + BITS_PER_WORD - 1) / BITS_PER_WORD; | |
486f4bd1 | 73 | CALLOC_ARRAY(filter.data, filter.len); |
f1294eaf | 74 | |
1b4c57fa JK |
75 | if (argc - 1 < i) |
76 | usage(bloom_usage); | |
f1294eaf GS |
77 | |
78 | while (argv[i]) { | |
79 | add_string_to_filter(argv[i], &filter); | |
80 | i++; | |
81 | } | |
82 | ||
83 | print_bloom_filter(&filter); | |
9794633b | 84 | free(filter.data); |
f1294eaf GS |
85 | } |
86 | ||
24b7d1e7 | 87 | if (!strcmp(argv[1], "get_filter_for_commit")) { |
ed591feb GS |
88 | struct object_id oid; |
89 | const char *end; | |
1b4c57fa JK |
90 | if (argc < 3) |
91 | usage(bloom_usage); | |
ed591feb GS |
92 | if (parse_oid_hex(argv[2], &oid, &end)) |
93 | die("cannot parse oid '%s'", argv[2]); | |
94 | init_bloom_filters(); | |
95 | get_bloom_filter_for_commit(&oid); | |
96 | } | |
97 | ||
f52207a4 | 98 | return 0; |
24b7d1e7 | 99 | } |