addr.go 972 B

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748
  1. package ice
  2. import "net"
  3. func parseMulticastAnswerAddr(in net.Addr) (net.IP, bool) {
  4. switch addr := in.(type) {
  5. case *net.IPAddr:
  6. return addr.IP, true
  7. case *net.UDPAddr:
  8. return addr.IP, true
  9. case *net.TCPAddr:
  10. return addr.IP, true
  11. }
  12. return nil, false
  13. }
  14. func parseAddr(in net.Addr) (net.IP, int, NetworkType, bool) {
  15. switch addr := in.(type) {
  16. case *net.UDPAddr:
  17. return addr.IP, addr.Port, NetworkTypeUDP4, true
  18. case *net.TCPAddr:
  19. return addr.IP, addr.Port, NetworkTypeTCP4, true
  20. }
  21. return nil, 0, 0, false
  22. }
  23. func createAddr(network NetworkType, ip net.IP, port int) net.Addr {
  24. switch {
  25. case network.IsTCP():
  26. return &net.TCPAddr{IP: ip, Port: port}
  27. default:
  28. return &net.UDPAddr{IP: ip, Port: port}
  29. }
  30. }
  31. func addrEqual(a, b net.Addr) bool {
  32. aIP, aPort, aType, aOk := parseAddr(a)
  33. if !aOk {
  34. return false
  35. }
  36. bIP, bPort, bType, bOk := parseAddr(b)
  37. if !bOk {
  38. return false
  39. }
  40. return aType == bType && aIP.Equal(bIP) && aPort == bPort
  41. }