git.oblomov.eu Git - git/blob - 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 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_bloom_filter(the_repository, c, 1);
  43         print_bloom_filter(filter);
  44 }
  45
  46 int cmd__bloom(int argc, const char **argv)
  47 {
  48         if (!strcmp(argv[1], "get_murmur3")) {
  49                 uint32_t hashed = murmur3_seeded(0, argv[2], strlen(argv[2]));
  50                 printf("Murmur3 Hash with seed=0:0x%08x\n", hashed);
  51         }
  52
  53     if (!strcmp(argv[1], "generate_filter")) {
  54                 struct bloom_filter filter;
  55                 int i = 2;
  56                 filter.len =  (settings.bits_per_entry + BITS_PER_WORD - 1) / BITS_PER_WORD;
  57                 filter.data = xcalloc(filter.len, sizeof(unsigned char));
  58
  59                 if (!argv[2]){
  60                         die("at least one input string expected");
  61                 }
  62
  63                 while (argv[i]) {
  64                         add_string_to_filter(argv[i], &filter);
  65                         i++;
  66                 }
  67
  68                 print_bloom_filter(&filter);
  69         }
  70
  71     if (!strcmp(argv[1], "get_filter_for_commit")) {
  72                 struct object_id oid;
  73                 const char *end;
  74                 if (parse_oid_hex(argv[2], &oid, &end))
  75                         die("cannot parse oid '%s'", argv[2]);
  76                 init_bloom_filters();
  77                 get_bloom_filter_for_commit(&oid);
  78         }
  79
  80         return 0;
  81 }