yggdrasil-go/misc/tests/multicast.go

package main

import "fmt"
import "net"
import "time"

// TODO look into netmap + libpcap to bypass the kernel as much as possible

func basic_test() {

  // TODO need a way to look up who our link-local neighbors are for each iface!
  //addr, err := net.ResolveUDPAddr("udp", "[ff02::1%veth0]:9001")
  addr, err := net.ResolveUDPAddr("udp", "[ff02::1]:9001")
  if err != nil { panic(err) }
  sock, err := net.ListenMulticastUDP("udp", nil, addr)
  if err != nil { panic(err) }
  defer sock.Close()

  go func () {
    saddr, err := net.ResolveUDPAddr("udp", "[::]:0")
    if err != nil { panic(err) }
    send, err := net.ListenUDP("udp", saddr)
    if err != nil { panic(err) }
    defer send.Close()
    msg := make([]byte, 1280)
    for {
      //fmt.Println("Sending...")
      send.WriteTo(msg, addr)
    }
  }()

  numPackets := 1000
  start := time.Now()
  msg := make([]byte, 2000)
  for i := 0 ; i < numPackets ; i++ {
    //fmt.Println("Reading:", i)
    sock.ReadFromUDP(msg)
  }
  timed := time.Since(start)

  fmt.Printf("%f packets per second\n", float64(numPackets)/timed.Seconds())

}

func main () {

  basic_test()

}