header_rewrite: Various fixes for MaxMind support (#7746)

Including: * Demote scary errors and other log changes * ASN fieldname correction
apache · Apr 28, 2021 · 1b5139d · 1b5139d
1 parent 3a1c7f6
commit 1b5139d
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 10 deletions.
diff --git a/plugins/header_rewrite/conditions_geo_maxmind.cc b/plugins/header_rewrite/conditions_geo_maxmind.cc
@@ -36,14 +36,21 @@ void
 MMConditionGeo::initLibrary(const std::string &path)
 {
   if (path.empty()) {
-    TSError("[%s] Empty db path specified. Not initializing!", PLUGIN_NAME);
+    TSDebug(PLUGIN_NAME, "Empty MaxMind db path specified. Not initializing!");
     return;
   }
+
+  if (gMaxMindDB != nullptr) {
+    TSDebug(PLUGIN_NAME, "Maxmind library already initialized");
+    return;
+  }
+
   gMaxMindDB = new MMDB_s;
 
   int status = MMDB_open(path.c_str(), MMDB_MODE_MMAP, gMaxMindDB);
   if (MMDB_SUCCESS != status) {
     TSDebug(PLUGIN_NAME, "Cannot open %s - %s", path.c_str(), MMDB_strerror(status));
+    delete gMaxMindDB;
     return;
   }
   TSDebug(PLUGIN_NAME, "Loaded %s", path.c_str());
@@ -56,6 +63,7 @@ MMConditionGeo::get_geo_string(const sockaddr *addr) const
   int mmdb_error;
 
   if (gMaxMindDB == nullptr) {
+    TSDebug(PLUGIN_NAME, "MaxMind not initialized; using default value");
     return ret;
   }
 
@@ -120,6 +128,7 @@ MMConditionGeo::get_geo_int(const sockaddr *addr) const
   int mmdb_error;
 
   if (gMaxMindDB == nullptr) {
+    TSDebug(PLUGIN_NAME, "MaxMind not initialized; using default value");
     return ret;
   }
 
@@ -150,7 +159,7 @@ MMConditionGeo::get_geo_int(const sockaddr *addr) const
   const char *field_name;
   switch (_geo_qual) {
   case GEO_QUAL_ASN:
-    field_name = "autonomous_system";
+    field_name = "autonomous_system_number";
     break;
   default:
     TSDebug(PLUGIN_NAME, "Unsupported field %d", _geo_qual);

diff --git a/plugins/header_rewrite/header_rewrite.cc b/plugins/header_rewrite/header_rewrite.cc
@@ -40,6 +40,7 @@ static std::once_flag initGeoLibs;
 static void
 initGeoLib(const std::string &dbPath)
 {
+  TSDebug(PLUGIN_NAME, "Loading geo db %s", dbPath.c_str());
 #if TS_USE_HRW_GEOIP
   GeoIPConditionGeo::initLibrary(dbPath);
 #elif TS_USE_HRW_MAXMINDDB
@@ -334,13 +335,19 @@ TSPluginInit(int argc, const char *argv[])
     }
   }
 
+  if (!geoDBpath.empty() && geoDBpath.find("/") != 0) {
+    geoDBpath = std::string(TSConfigDirGet()) + '/' + geoDBpath;
+  }
+
+  TSDebug(PLUGIN_NAME, "Global geo db %s", geoDBpath.c_str());
+
+  std::call_once(initGeoLibs, [&geoDBpath]() { initGeoLib(geoDBpath); });
+
   // Parse the global config file(s). All rules are just appended
   // to the "global" Rules configuration.
   RulesConfig *conf = new RulesConfig;
   bool got_config   = false;
 
-  std::call_once(initGeoLibs, [&geoDBpath]() { initGeoLib(geoDBpath); });
-
   for (int i = optind; i < argc; ++i) {
     // Parse the config file(s). Note that multiple config files are
     // just appended to the configurations.
@@ -412,25 +419,29 @@ TSRemapNewInstance(int argc, char *argv[], void **ih, char * /* errbuf ATS_UNUSE
   --argc;
   ++argv;
 
-  std::string geoDBPath;
+  std::string geoDBpath;
   while (true) {
     int opt = getopt_long(argc, (char *const *)argv, "m:", longopt, NULL);
 
     switch (opt) {
     case 'm': {
-      geoDBPath = optarg;
+      geoDBpath = optarg;
     } break;
     }
     if (opt == -1) {
       break;
     }
   }
 
-  if (geoDBPath.find("/") != 0) {
-    geoDBPath = std::string(TSConfigDirGet()) + '/' + geoDBPath;
-  }
+  if (!geoDBpath.empty()) {
+    if (geoDBpath.find("/") != 0) {
+      geoDBpath = std::string(TSConfigDirGet()) + '/' + geoDBpath;
+    }
+
+    TSDebug(PLUGIN_NAME, "Remap geo db %s", geoDBpath.c_str());
 
-  std::call_once(initGeoLibs, [&geoDBPath]() { initGeoLib(geoDBPath); });
+    std::call_once(initGeoLibs, [&geoDBpath]() { initGeoLib(geoDBpath); });
+  }
 
   RulesConfig *conf = new RulesConfig;