123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161 |
- // Copyright (c) 2012-2014 Jeremy Latt
- // Copyright (c) 2016 Daniel Oaks <daniel@danieloaks.net>
- // released under the MIT license
-
- package utils
-
- import "net"
- import "reflect"
- import "testing"
-
- func assertEqual(supplied, expected interface{}, t *testing.T) {
- if !reflect.DeepEqual(supplied, expected) {
- t.Errorf("expected %v but got %v", expected, supplied)
- }
- }
-
- // hostnames from https://github.com/DanielOaks/irc-parser-tests
- var (
- goodHostnames = []string{
- "irc.example.com",
- "i.coolguy.net",
- "irc-srv.net.uk",
- "iRC.CooLguY.NeT",
- "gsf.ds342.co.uk",
- "324.net.uk",
- "xn--bcher-kva.ch",
- "pentos",
- "pentos.",
- "www.google.com.",
- }
-
- badHostnames = []string{
- "-lol-.net.uk",
- "-lol.net.uk",
- "_irc._sctp.lol.net.uk",
- "irc.l%l.net.uk",
- "irc..net.uk",
- ".",
- "",
- }
- )
-
- func TestIsHostname(t *testing.T) {
- for _, name := range goodHostnames {
- if !IsHostname(name) {
- t.Error(
- "Expected to pass, but could not validate hostname",
- name,
- )
- }
- }
-
- for _, name := range badHostnames {
- if IsHostname(name) {
- t.Error(
- "Expected to fail, but successfully validated hostname",
- name,
- )
- }
- }
- }
-
- func TestIsServerName(t *testing.T) {
- if IsServerName("pentos") {
- t.Error("irc server names must contain a period")
- }
- if !IsServerName("darwin.network") {
- t.Error("failed to validate a perfectly good server name")
- }
- }
-
- func TestNormalizeToNet(t *testing.T) {
- a := net.ParseIP("8.8.8.8")
- b := net.ParseIP("8.8.4.4")
- if a == nil || b == nil {
- panic("something has gone very wrong")
- }
-
- aNetwork := NormalizeIPToNet(a)
- bNetwork := NormalizeIPToNet(b)
-
- assertEqual(aNetwork.Contains(a), true, t)
- assertEqual(bNetwork.Contains(b), true, t)
- assertEqual(aNetwork.Contains(b), false, t)
- assertEqual(bNetwork.Contains(a), false, t)
-
- c := net.ParseIP("2001:4860:4860::8888")
- d := net.ParseIP("2001:db8::1")
- if c == nil || d == nil {
- panic("something has gone very wrong")
- }
-
- cNetwork := NormalizeIPToNet(c)
- dNetwork := NormalizeIPToNet(d)
-
- assertEqual(cNetwork.Contains(c), true, t)
- assertEqual(dNetwork.Contains(d), true, t)
- assertEqual(dNetwork.Contains(c), false, t)
- assertEqual(dNetwork.Contains(a), false, t)
- assertEqual(cNetwork.Contains(b), false, t)
- assertEqual(aNetwork.Contains(c), false, t)
- assertEqual(bNetwork.Contains(c), false, t)
-
- assertEqual(NetToNormalizedString(aNetwork), "8.8.8.8", t)
- assertEqual(NetToNormalizedString(bNetwork), "8.8.4.4", t)
- assertEqual(NetToNormalizedString(cNetwork), "2001:4860:4860::8888", t)
- assertEqual(NetToNormalizedString(dNetwork), "2001:db8::1", t)
- }
-
- func TestNormalizedNetToString(t *testing.T) {
- _, network, err := net.ParseCIDR("8.8.0.0/16")
- if err != nil {
- panic(err)
- }
- assertEqual(NetToNormalizedString(*network), "8.8.0.0/16", t)
-
- normalized := NormalizeNet(*network)
- assertEqual(normalized.Contains(net.ParseIP("8.8.4.4")), true, t)
- assertEqual(normalized.Contains(net.ParseIP("1.1.1.1")), false, t)
- assertEqual(NetToNormalizedString(normalized), "8.8.0.0/16", t)
-
- _, network, err = net.ParseCIDR("8.8.4.4/32")
- if err != nil {
- panic(err)
- }
- assertEqual(NetToNormalizedString(*network), "8.8.4.4", t)
-
- normalized = NormalizeNet(*network)
- assertEqual(normalized.Contains(net.ParseIP("8.8.4.4")), true, t)
- assertEqual(normalized.Contains(net.ParseIP("8.8.8.8")), false, t)
- assertEqual(NetToNormalizedString(normalized), "8.8.4.4", t)
- }
-
- func TestNormalizedNet(t *testing.T) {
- _, network, err := net.ParseCIDR("::ffff:8.8.4.4/128")
- assertEqual(err, nil, t)
- assertEqual(NetToNormalizedString(*network), "8.8.4.4", t)
-
- normalizedNet := NormalizeIPToNet(net.ParseIP("8.8.4.4"))
- assertEqual(NetToNormalizedString(normalizedNet), "8.8.4.4", t)
-
- _, network, err = net.ParseCIDR("::ffff:8.8.0.0/112")
- assertEqual(err, nil, t)
- assertEqual(NetToNormalizedString(*network), "8.8.0.0/16", t)
- _, v4Network, err := net.ParseCIDR("8.8.0.0/16")
- assertEqual(err, nil, t)
- normalizedNet = NormalizeNet(*v4Network)
- assertEqual(NetToNormalizedString(normalizedNet), "8.8.0.0/16", t)
- }
-
- func TestNormalizedNetFromString(t *testing.T) {
- network, err := NormalizedNetFromString("8.8.4.4/16")
- assertEqual(err, nil, t)
- assertEqual(NetToNormalizedString(network), "8.8.0.0/16", t)
- assertEqual(network.Contains(net.ParseIP("8.8.8.8")), true, t)
-
- network, err = NormalizedNetFromString("2001:0db8::1")
- assertEqual(err, nil, t)
- assertEqual(NetToNormalizedString(network), "2001:db8::1", t)
- assertEqual(network.Contains(net.ParseIP("2001:0db8::1")), true, t)
- }
|