maxmind-geoip2 1.1.0 → 1.3.0

data/test/data/pkg/writer/nestedstructures.go

@@ -0,0 +1,73 @@
+package writer
+
+import (
+	"fmt"
+	"net/netip"
+
+	"github.com/maxmind/mmdbwriter"
+	"github.com/maxmind/mmdbwriter/mmdbtype"
+	"go4.org/netipx"
+)
+
+// WriteDeeplyNestedStructuresTestDB writes an mmdb file with deeply nested record value types.
+func (w *Writer) WriteDeeplyNestedStructuresTestDB() error {
+	dbWriter, err := mmdbwriter.New(
+		mmdbwriter.Options{
+			DatabaseType: "MaxMind DB Nested Data Structures",
+			Description: map[string]string{
+				"en": "MaxMind DB Nested Data Structures Test database - contains deeply nested map/array structures",
+			},
+			DisableIPv4Aliasing:     false,
+			IncludeReservedNetworks: true,
+			IPVersion:               6,
+			Languages:               []string{"en"},
+			RecordSize:              24,
+		},
+	)
+	if err != nil {
+		return fmt.Errorf("creating mmdbwriter: %w", err)
+	}
+
+	addrs, err := parseIPSlice(ipSample)
+	if err != nil {
+		return fmt.Errorf("parsing ip addresses: %w", err)
+	}
+	if err := insertNestedStructure(dbWriter, addrs); err != nil {
+		return fmt.Errorf("inserting all types records: %w", err)
+	}
+
+	if err := w.write(dbWriter, "MaxMind-DB-test-nested.mmdb"); err != nil {
+		return fmt.Errorf("writing database: %w", err)
+	}
+	return nil
+}
+
+// insertNestedStructure inserts records with deeply nested structures.
+func insertNestedStructure(w *mmdbwriter.Tree, ipAddresses []netip.Prefix) error {
+	nestedStruct := mmdbtype.Map{
+		"map1": mmdbtype.Map{
+			"map2": mmdbtype.Map{
+				"array": mmdbtype.Slice{
+					mmdbtype.Map{
+						"map3": mmdbtype.Map{
+							"a": mmdbtype.Uint32(1),
+							"b": mmdbtype.Uint32(2),
+							"c": mmdbtype.Uint32(3),
+						},
+					},
+				},
+			},
+		},
+	}
+
+	for _, addr := range ipAddresses {
+		err := w.Insert(
+			netipx.PrefixIPNet(addr),
+			nestedStruct,
+		)
+		if err != nil {
+			return fmt.Errorf("inserting ip: %w", err)
+		}
+	}
+	return nil
+}

data/test/data/pkg/writer/writer.go

@@ -0,0 +1,61 @@
+// Package writer defines database writers responsible
+// for generating test mmdb files.
+package writer
+
+import (
+	"errors"
+	"fmt"
+	"io/fs"
+	"os"
+	"path/filepath"
+
+	"github.com/maxmind/mmdbwriter"
+)
+
+var ipSample = []string{
+	"::1.1.1.0/120",
+	"::2.2.0.0/112",
+	"::3.0.0.0/104",
+	"::4.5.6.7/128",
+	"abcd::/64",
+	"1000::1234:0000/112",
+}
+
+// Writer is responsible for writing test mmdb databases
+// based on the provided data sources.
+type Writer struct {
+	source string
+	target string
+}
+
+// New initializes a new test database writer struct.
+func New(source, target string) (*Writer, error) {
+	s := filepath.Clean(source)
+	if _, err := os.Stat(s); errors.Is(err, fs.ErrNotExist) {
+		return nil, fmt.Errorf("source directory does not exist: %w", err)
+	}
+
+	t := filepath.Clean(target)
+	//nolint:gosec // not security sensitive.
+	if err := os.MkdirAll(t, os.ModePerm); err != nil {
+		return nil, fmt.Errorf("creating target directory: %w", err)
+	}
+
+	return &Writer{
+		source: s,
+		target: t,
+	}, nil
+}
+
+func (w *Writer) write(dbWriter *mmdbwriter.Tree, fileName string) error {
+	outputFile, err := os.Create(filepath.Clean(filepath.Join(w.target, fileName)))
+	if err != nil {
+		return fmt.Errorf("creating mmdb file: %w", err)
+	}
+	defer outputFile.Close()
+
+	if _, err := dbWriter.WriteTo(outputFile); err != nil {
+		return fmt.Errorf("writing mmdb file: %w", err)
+	}
+	return nil
+}

data/test/data/source-data/GeoIP-Anonymous-Plus-Test.json

@@ -0,0 +1,175 @@
+[
+  {
+    "1.2.0.0/32": {
+      "is_anonymous": true,
+      "is_anonymous_vpn": true
+    }
+  },
+  {
+    "1.2.0.1/32": {
+      "anonymizer_confidence": 30,
+      "is_anonymous": true,
+      "is_anonymous_vpn": true,
+      "network_last_seen": "2025-04-14",
+      "provider_name": "foo"
+    }
+  },
+  {
+    "1.2.0.2/32": {
+      "anonymizer_confidence": 99,
+      "is_anonymous": true,
+      "is_anonymous_vpn": true
+    }
+  },
+  {
+    "1.2.0.3/32": {
+      "is_anonymous": true,
+      "is_anonymous_vpn": true,
+      "network_last_seen": "2025-04-14"
+    }
+  },
+  {
+    "1.2.0.4/32": {
+      "is_anonymous": true,
+      "is_anonymous_vpn": true,
+      "provider_name": "foo"
+    }
+  },
+  {
+    "1.2.0.5/32": {
+      "is_anonymous": true,
+      "is_anonymous_vpn": true
+    }
+  },
+  {
+    "1.2.0.6/31": {
+      "is_anonymous": true,
+      "is_anonymous_vpn": true
+    }
+  },
+  {
+    "1.2.0.8/29": {
+      "is_anonymous": true,
+      "is_anonymous_vpn": true
+    }
+  },
+  {
+    "1.2.0.16/28": {
+      "is_anonymous": true,
+      "is_anonymous_vpn": true
+    }
+  },
+  {
+    "1.2.0.32/27": {
+      "is_anonymous": true,
+      "is_anonymous_vpn": true
+    }
+  },
+  {
+    "1.2.0.64/26": {
+      "is_anonymous": true,
+      "is_anonymous_vpn": true
+    }
+  },
+  {
+    "1.2.0.128/25": {
+      "is_anonymous": true,
+      "is_anonymous_vpn": true
+    }
+  },
+  {
+    "1.2.1.0/24": {
+      "is_anonymous": true,
+      "is_anonymous_vpn": true
+    }
+  },
+  {
+    "1.2.2.0/23": {
+      "is_anonymous": true,
+      "is_anonymous_vpn": true
+    }
+  },
+  {
+    "1.2.4.0/22": {
+      "is_anonymous": true,
+      "is_anonymous_vpn": true
+    }
+  },
+  {
+    "1.2.8.0/21": {
+      "is_anonymous": true,
+      "is_anonymous_vpn": true
+    }
+  },
+  {
+    "1.2.16.0/20": {
+      "is_anonymous": true,
+      "is_anonymous_vpn": true
+    }
+  },
+  {
+    "1.2.32.0/19": {
+      "is_anonymous": true,
+      "is_anonymous_vpn": true
+    }
+  },
+  {
+    "1.2.64.0/18": {
+      "is_anonymous": true,
+      "is_anonymous_vpn": true
+    }
+  },
+  {
+    "1.2.128.0/17": {
+      "is_anonymous": true,
+      "is_anonymous_vpn": true
+    }
+  },
+  {
+    "::1.124.213.1/128": {
+      "is_anonymous": true,
+      "is_anonymous_vpn": true,
+      "is_tor_exit_node": true
+    }
+  },
+  {
+    "::81.2.69.0/120": {
+      "is_anonymous": true,
+      "is_anonymous_vpn": true,
+      "is_hosting_provider": true,
+      "is_public_proxy": true,
+      "is_residential_proxy": true,
+      "is_tor_exit_node": true
+    }
+  },
+  {
+    "::71.160.223.0/120": {
+      "is_anonymous": true,
+      "is_hosting_provider": true
+    }
+  },
+  {
+    "::186.30.236.0/120": {
+      "is_anonymous": true,
+      "is_public_proxy": true
+    }
+  },
+  {
+    "::65.0.0.0/109": {
+      "is_anonymous": true,
+      "is_tor_exit_node": true
+    }
+  },
+  {
+    "abcd:1000::/112": {
+      "is_anonymous": true,
+      "is_public_proxy": true
+    }
+  },
+  {
+    "2001:480:3a::/64": {
+      "is_anonymous": true,
+      "is_public_proxy": true
+    }
+  }
+]

data/test/data/source-data/GeoIP2-Anonymous-IP-Test.json

@@ -45,5 +45,11 @@
          "is_anonymous" : true,
          "is_public_proxy" : true
       }
+   },
+   {
+      "2001:480:3a::/64" : {
+         "is_anonymous" : true,
+         "is_public_proxy" : true
+      }
    }
 ]