blfchk.c

/* Copyright (c) 2015 Ryan Castellucci, All Rights Reserved */
#include <time.h>
#include <unistd.h>
#include <assert.h>
#include <stdlib.h>
#include <stdint.h>
#include <string.h>
#include <signal.h>
#include <stdio.h>
#include <fcntl.h>
#include <errno.h>

#include <arpa/inet.h> /* for ntohl/htonl */

#include <sys/time.h>
#include <sys/wait.h>
#include <sys/types.h>

#include "hex.h"
#include "bloom.h"
#include "mmapf.h"
#include "hash160.h"
#include "hsearchf.h"

int main(int argc, char **argv) {
  int ret;
  hash160_t hash;
  char *line = NULL;
  size_t line_sz = 0;
  unsigned char *bloom, *bloomfile, *hashfile;
  FILE *ifile = stdin, *ofile = stdout, *hfile = NULL;
  mmapf_ctx bloom_mmapf;

  if (argc < 2 || argc > 3) {
    fprintf(stderr, "Usage: %s BLOOM_FILTER_FILE HASH_FILE\n", argv[0]);
    return 1;
  }

  bloomfile = argv[1];

  if ((ret = mmapf(&bloom_mmapf, bloomfile, BLOOM_SIZE, MMAPF_RNDRD)) != MMAPF_OKAY) {
    fprintf(stderr, "failed to open bloom filter '%s': %s\n", bloomfile, mmapf_strerror(ret));
    return 1;
  } else if (bloom_mmapf.mem == NULL) {
    fprintf(stderr, "got NULL pointer trying to set up bloom filter\n");
    return 1;
  }

  bloom = bloom_mmapf.mem;

  if (argc == 3) {
    hashfile = argv[2];
    hfile = fopen(hashfile, "r");
  }

  while (getline(&line, &line_sz, ifile) > 0) {
    unhex(line, strlen(line), hash.uc, sizeof(hash.uc));

    unsigned int bit;
    bit = BH00(hash.uc); if (BLOOM_GET_BIT(bit) == 0) { continue; }
    bit = BH01(hash.uc); if (BLOOM_GET_BIT(bit) == 0) { continue; }
    bit = BH02(hash.uc); if (BLOOM_GET_BIT(bit) == 0) { continue; }
    bit = BH03(hash.uc); if (BLOOM_GET_BIT(bit) == 0) { continue; }
    bit = BH04(hash.uc); if (BLOOM_GET_BIT(bit) == 0) { continue; }
    bit = BH05(hash.uc); if (BLOOM_GET_BIT(bit) == 0) { continue; }
    bit = BH06(hash.uc); if (BLOOM_GET_BIT(bit) == 0) { continue; }
    bit = BH07(hash.uc); if (BLOOM_GET_BIT(bit) == 0) { continue; }
    bit = BH08(hash.uc); if (BLOOM_GET_BIT(bit) == 0) { continue; }
    bit = BH09(hash.uc); if (BLOOM_GET_BIT(bit) == 0) { continue; }
    bit = BH10(hash.uc); if (BLOOM_GET_BIT(bit) == 0) { continue; }
    bit = BH11(hash.uc); if (BLOOM_GET_BIT(bit) == 0) { continue; }
    bit = BH12(hash.uc); if (BLOOM_GET_BIT(bit) == 0) { continue; }
    bit = BH13(hash.uc); if (BLOOM_GET_BIT(bit) == 0) { continue; }
    bit = BH14(hash.uc); if (BLOOM_GET_BIT(bit) == 0) { continue; }
    bit = BH15(hash.uc); if (BLOOM_GET_BIT(bit) == 0) { continue; }
    bit = BH16(hash.uc); if (BLOOM_GET_BIT(bit) == 0) { continue; }
    bit = BH17(hash.uc); if (BLOOM_GET_BIT(bit) == 0) { continue; }
    bit = BH18(hash.uc); if (BLOOM_GET_BIT(bit) == 0) { continue; }
    bit = BH19(hash.uc); if (BLOOM_GET_BIT(bit) == 0) { continue; }

    if (hfile && !hsearchf(hfile, &hash)) {
      //fprintf(ofile, "%s (false positive)\n", hex(hash.uc, sizeof(hash.uc), buf, sizeof(buf)));
      continue;
    }
    //fprintf(ofile, "%s\n", hex(hash.uc, sizeof(hash.uc), buf, sizeof(buf)));
    fprintf(ofile, "%s", line);
  }

  return 0;
}

/*  vim: set ts=2 sw=2 et ai si: */