Function loadFile

plugins/experimental/rate_limit/iprep_simu.cc:146–198

////////////////////////////////////////////////////////////////////////// Load a configuration file, and populate the two structures with the list of IPs (and their status) as well as the full sequence of requests. Returns a tuple with the number of good requests and bad requests, respectively.

// Returns a tuple with the number of good requests and bad requests, respectively.
//
std::tuple<uint32_t, uint32_t>
loadFile(const std::string &fname, IpMap &all_ips, IpList &ips)
{
  std::ifstream infile(fname);

  float       timestamp; // The timestamp from the request(relative)
  std::string ip;        // The IP
  bool        status;    // Bad (false) or Good (true) request?

  uint32_t good_ips      = 0;
  uint32_t bad_ips       = 0;
  uint32_t good_requests = 0;
  uint32_t bad_requests  = 0;

  // Load in the entire file, and fill the request vector as well as the IP lookup table (state)
  while (infile >> timestamp >> ip >> status) {
    auto ip_hash = IpReputation::SieveLru::hasher(ip, ip.find(':') != std::string::npos ? AF_INET6 : AF_INET);
    auto it      = all_ips.find(ip_hash);

    if (!status) {
      ++good_requests;
    } else {
      ++bad_requests;
    }

    if (all_ips.end() != it) {
      auto &[key, data]       = *it;
      auto &[count, d_status] = data;

      ++count;
      ips.push_back(it);
    } else {
      all_ips[ip_hash] = {0, status};
      ips.push_back(all_ips.find(ip_hash));
      if (!status) {
        ++good_ips;
      } else {
        ++bad_ips;
      }
    }
  }

  std::cout << std::setprecision(3);
  std::cout << "Total number of requests: " << ips.size() << "\n";
  std::cout << "\tGood requests: " << good_requests << " (" << 100.0 * good_requests / ips.size() << "%)\n";
  std::cout << "\tBad requests: " << bad_requests << " (" << 100.0 * bad_requests / ips.size() << "%)\n";
  std::cout << "Unique IPs in set: " << all_ips.size() << "\n";
  std::cout << "\tGood IPs: " << good_ips << " (" << 100.0 * good_ips / all_ips.size() << "%)\n";
  std::cout << "\tBad IPs: " << bad_ips << " (" << 100.0 * bad_ips / all_ips.size() << "%)\n";
  std::cout << "\n";

  return {good_requests, bad_requests};
}

int
main(int argc, char *argv[])