udp_windows.go 760 B

1234567891011121314151617181920212223242526272829
  1. // +build windows
  2. package dns
  3. import "net"
  4. type SessionUDP struct {
  5. raddr *net.UDPAddr
  6. }
  7. func (s *SessionUDP) RemoteAddr() net.Addr { return s.raddr }
  8. // ReadFromSessionUDP acts just like net.UDPConn.ReadFrom(), but returns a session object instead of a
  9. // net.UDPAddr.
  10. func ReadFromSessionUDP(conn *net.UDPConn, b []byte) (int, *SessionUDP, error) {
  11. n, raddr, err := conn.ReadFrom(b)
  12. if err != nil {
  13. return n, nil, err
  14. }
  15. session := &SessionUDP{raddr.(*net.UDPAddr)}
  16. return n, session, err
  17. }
  18. // WriteToSessionUDP acts just like net.UDPConn.WritetTo(), but uses a *SessionUDP instead of a net.Addr.
  19. func WriteToSessionUDP(conn *net.UDPConn, b []byte, session *SessionUDP) (int, error) {
  20. n, err := conn.WriteTo(b, session.raddr)
  21. return n, err
  22. }