k-sone / critbitgo Goto Github PK

package main_test

import (
  "net"
  "testing"

  "github.com/k-sone/critbitgo"
)

func checkerr(err error) {
  if err != nil {
    panic(err)
  }  
}

func TestMatchIP(t *testing.T) {
  // loopback
  _, lo4, err := net.ParseCIDR("127.0.0.1/8")
  checkerr(err)

  _, lo6, err := net.ParseCIDR("::1/128")
  checkerr(err)

  // default routes
  _, dg4, err := net.ParseCIDR("0.0.0.0/0")
  checkerr(err)

  _, dg6, err := net.ParseCIDR("::/0")
  checkerr(err)

  // make new crit-bit trie
  trie := critbitgo.NewNet()

  // add some CIDRs, mixed IP version needed to reproduce the error
  for _, r := range []*net.IPNet{lo4, lo6, dg4, dg6} {
    err := trie.Add(r, nil)
    checkerr(err)
  }  

  // test IPv6 addr
  ip := net.ParseIP("fe80::1")

  // longest-prefix-match for test IPv6 addr, expect "::/0"
  cidr, _, err := trie.MatchIP(ip)
  checkerr(err)

  if cidr == nil {
    t.Fatalf("MatchIP(%q), expect %q, got %v", ip, dg6, cidr)
  }  
}

package main

import (
    "log"
    "github.com/k-sone/critbitgo"
)

func main() {
        log.SetFlags(0)
        trie := critbitgo.NewTrie()
        trie.Insert([]byte("/ctrl_ready"), 1)
        trie.Insert([]byte("/router_message"), 2)
        trie.Insert([]byte("/router_ready"), 3)
        key := []byte("/router_message")
        trie.Allprefixed(key, func(key []byte, value interface{}) bool {
                log.Printf("Allprefixed: %s %v", key, value)
                return true
        })
}