dns/parse_test.go

package dns

import (
        "os"
        "time"
        "bufio"
        "strings"
	"testing"
	"crypto/rsa"
)

func TestSign(t *testing.T) {
	pub := "miek.nl. IN DNSKEY 256 3 5 AwEAAb+8lGNCxJgLS8rYVer6EnHVuIkQDghdjdtewDzU3G5R7PbMbKVRvH2Ma7pQyYceoaqWZQirSj72euPWfPxQnMy9ucCylA+FuH9cSjIcPf4PqJfdupHk9X6EBYjxrCLY4p1/yBwgyBIRJtZtAqM3ceAH2WovEJD6rTtOuHo5AluJ"

	priv := `Private-key-format: v1.3
Algorithm: 5 (RSASHA1)
Modulus: v7yUY0LEmAtLythV6voScdW4iRAOCF2N217APNTcblHs9sxspVG8fYxrulDJhx6hqpZlCKtKPvZ649Z8/FCczL25wLKUD4W4f1xKMhw9/g+ol926keT1foQFiPGsItjinX/IHCDIEhEm1m0Cozdx4AfZai8QkPqtO064ejkCW4k=
PublicExponent: AQAB
PrivateExponent: YPwEmwjk5HuiROKU4xzHQ6l1hG8Iiha4cKRG3P5W2b66/EN/GUh07ZSf0UiYB67o257jUDVEgwCuPJz776zfApcCB4oGV+YDyEu7Hp/rL8KcSN0la0k2r9scKwxTp4BTJT23zyBFXsV/1wRDK1A5NxsHPDMYi2SoK63Enm/1ptk=
Prime1: /wjOG+fD0ybNoSRn7nQ79udGeR1b0YhUA5mNjDx/x2fxtIXzygYk0Rhx9QFfDy6LOBvz92gbNQlzCLz3DJt5hw==
Prime2: wHZsJ8OGhkp5p3mrJFZXMDc2mbYusDVTA+t+iRPdS797Tj0pjvU2HN4vTnTj8KBQp6hmnY7dLp9Y1qserySGbw==
Exponent1: N0A7FsSRIg+IAN8YPQqlawoTtG1t1OkJ+nWrurPootScApX6iMvn8fyvw3p2k51rv84efnzpWAYiC8SUaQDNxQ==
Exponent2: SvuYRaGyvo0zemE3oS+WRm2scxR8eiA8WJGeOc+obwOKCcBgeZblXzfdHGcEC1KaOcetOwNW/vwMA46lpLzJNw==
Coefficient: 8+7ZN/JgByqv0NfULiFKTjtyegUcijRuyij7yNxYbCBneDvZGxJwKNi4YYXWx743pcAj4Oi4Oh86gcmxLs+hGw==
Created: 20110302104537
Publish: 20110302104537
Activate: 20110302104537`

	k := new(RR_DNSKEY)
        k.Read(strings.NewReader(pub))
	p, err := k.ReadPrivateKey(strings.NewReader(priv))
        if err != nil {
                t.Logf("%v\n", err)
                t.Fail()
        }
	switch priv := p.(type) {
	case *rsa.PrivateKey:
		if 65537 != priv.PublicKey.E {
			t.Log("Exponenent should be 65537")
			t.Fail()
		}
        default:
                t.Logf("We should have read an RSA key: %v", priv)
                t.Fail()
	}
	if k.KeyTag() != 37350 {
		t.Logf("%d %v\n", k.KeyTag(), k)
		t.Log("Keytag should be 37350")
		t.Fail()
	}

	soa := new(RR_SOA)
	soa.Hdr = RR_Header{"miek.nl.", TypeSOA, ClassINET, 14400, 0}
	soa.Ns = "open.nlnetlabs.nl."
	soa.Mbox = "miekg.atoom.net."
	soa.Serial = 1293945905
	soa.Refresh = 14400
	soa.Retry = 3600
	soa.Expire = 604800
	soa.Minttl = 86400

	sig := new(RR_RRSIG)
	sig.Hdr = RR_Header{"miek.nl.", TypeRRSIG, ClassINET, 14400, 0}
	sig.Expiration = 1296534305 // date -u '+%s' -d"2011-02-01 04:25:05"
	sig.Inception = 1293942305  // date -u '+%s' -d"2011-01-02 04:25:05"
	sig.KeyTag = k.KeyTag()
	sig.SignerName = k.Hdr.Name
	sig.Algorithm = k.Algorithm

	sig.Sign(p, []RR{soa})
        if sig.Signature != "D5zsobpQcmMmYsUMLxCVEtgAdCvTu8V/IEeP4EyLBjqPJmjt96bwM9kqihsccofA5LIJ7DN91qkCORjWSTwNhzCv7bMyr2o5vBZElrlpnRzlvsFIoAZCD9xg6ZY7ZyzUJmU6IcTwG4v3xEYajcpbJJiyaw/RqR90MuRdKPiBzSo=" {
                t.Log("Signature is not correct")
                t.Logf("%v\n", sig)
                t.Fail()
        }
}

func TestDotInName(t *testing.T) {
	buf := make([]byte, 20)
	packDomainName("aa\\.bb.nl.", buf, 0)
        // index 3 must be a real dot
        if buf[3] != '.' {
                t.Log("Dot should be a real dot")
                t.Fail()
        }

        if buf[6] != 2 {
                t.Log("This must have the value 2")
                t.Fail()
        }
        dom, _, _ := unpackDomainName(buf, 0)
        // printing it should yield the backspace again
        if dom != "aa\\.bb.nl." {
                t.Log("Dot should have been escaped: " + dom)
                t.Fail()
        }
}

// Make this a decend test case. For now, good enough
// New style (Ragel) parsing
func TestParse(t *testing.T) {
        tests := map[string]string{
                "miek.nl. 3600 IN A 127.0.0.1": "miek.nl.\t3600\tIN\tA\t127.0.0.1",
                "miek.nl. 3600 IN MX 10 elektron.atoom.net.": "miek.nl.\t3600\tIN\tMX\t10 elektron.atoom.net.",
                "miek.nl. IN 3600 A 127.0.0.1": "miek.nl.\t3600\tIN\tA\t127.0.0.1",
                "miek.nl. A 127.0.0.1":         "miek.nl.\t0\tCLASS0\tA\t127.0.0.1",
                "miek.nl. IN AAAA ::1":         "miek.nl.\t0\tIN\tAAAA\t::1",
                "miek.nl. IN A 127.0.0.1":      "miek.nl.\t0\tIN\tA\t127.0.0.1",
                "miek.nl. IN DNSKEY 256 3 5 AwEAAb+8lGNCxJgLS8rYVer6EnHVuIkQDghdjdtewDzU3G5R7PbMbKVRvH2Ma7pQyYceoaqWZQirSj72euPWfPxQnMy9ucCylA+FuH9cSjIcPf4PqJfdupHk9X6EBYjxrCLY4p1/yBwgyBIRJtZtAqM3ceAH2WovEJD6rTtOuHo5AluJ":
                        "miek.nl.\t0\tIN\tDNSKEY\t256 3 5 AwEAAb+8lGNCxJgLS8rYVer6EnHVuIkQDghdjdtewDzU3G5R7PbMbKVRvH2Ma7pQyYceoaqWZQirSj72euPWfPxQnMy9ucCylA+FuH9cSjIcPf4PqJfdupHk9X6EBYjxrCLY4p1/yBwgyBIRJtZtAqM3ceAH2WovEJD6rTtOuHo5AluJ",
                "nlnetlabs.nl. 3175 IN DNSKEY 256 3 8 AwEAAdR7XR95OaAN9Rz7TbtPalQ9guQk7zfxTHYNKhsiwTZA9z+F16nD0VeBlk7dNik3ETpT2GLAwr9sntG898JwurCDe353wHPvjZtMCdiTVp3cRCrjuCEvoFpmZNN82H0gaH/4v8mkv/QBDAkDSncYjz/FqHKAeYy3cMcjY6RyVweh":
                        "nlnetlabs.nl.\t3175\tIN\tDNSKEY\t256 3 8 AwEAAdR7XR95OaAN9Rz7TbtPalQ9guQk7zfxTHYNKhsiwTZA9z+F16nD0VeBlk7dNik3ETpT2GLAwr9sntG898JwurCDe353wHPvjZtMCdiTVp3cRCrjuCEvoFpmZNN82H0gaH/4v8mkv/QBDAkDSncYjz/FqHKAeYy3cMcjY6RyVweh",
                }
        for test, result := range tests {
                p := NewParser(strings.NewReader(test))
                z, err := p.Zone()
                if err != nil || z == nil{
                        t.Logf("Error of nil r %v %s\n", err, test)
                        t.Fail()
                }
                r := z.Pop().(RR)
                if r.String() != result {
                        t.Logf("\"%s\" should be equal to\n\"%s\"\n", r, result)
                        t.Fail()
                }
        }
}

func TestSetString(t *testing.T) {
        a := NewRR("miek.nl. IN A 127.0.0.1")
        if a.String() != "miek.nl.\t0\tIN\tA\t127.0.0.1" {
                t.Log(a.String(), "!= miek.nl. IN A 127.0.0.1")
                t.Fail()
        }
        b := NewRR("miek.nl. IN AAAA ::1")
        if b.String() != "miek.nl.\t0\tIN\tAAAA\t::1" {
                t.Log(b.String(), "!= miek.nl. IN AAAA ::1")
                t.Fail()
        }
        c := NewRR("miek.nl. IN MX 10 miek.nl.")
        if c.String() != "miek.nl.\t0\tIN\tMX\t10 miek.nl." {
                t.Log(c.String(), "!= miek.nl. IN MX 10 miek.nl.")
                t.Fail()
        }
        d := NewRR("miek.nl. IN NS ns1.miek.nl")
        if d.String() != "miek.nl.\t0\tIN\tNS\tns1.miek.nl" {
                t.Log(d.String(), "!= miek.nl. IN NS ns1.miek.nl")
                t.Fail()
        }
}

func BenchmarkZoneParsing(b *testing.B) {
        file, err := os.Open("miek.nl")
        defer file.Close()
        if err != nil {
                return
        }
        p := NewParser(bufio.NewReader(file))
        // Don't care about errors (there shouldn't be any)
        p.Zone()
}

func TestZoneParsing(t *testing.T) {
        file, err := os.Open("miek.nl")
        defer file.Close()
        if err != nil {
                return
        }
        p := NewParser(bufio.NewReader(file))

        // Don't care about errors (there shouldn't be any)
        start := time.Nanoseconds()
        z, err := p.Zone()
        if err != nil {
                t.Logf("error %v\n", err.String())
                t.Fail()
        }
        delta := time.Nanoseconds() - start
        t.Logf("%d RRs parsed in %.2f s (%.2f RR/s)", z.Len()-1, float32(delta)/1e9, float32(z.Len()-1)/(float32(delta)/1e9))
}