udp_plan9.go 1000 B

1234567891011121314151617181920212223242526272829303132333435
  1. package dns
  2. import (
  3. "net"
  4. )
  5. func setUDPSocketOptions(conn *net.UDPConn) error { return nil }
  6. // SessionUDP holds the remote address and the associated
  7. // out-of-band data.
  8. type SessionUDP struct {
  9. raddr *net.UDPAddr
  10. context []byte
  11. }
  12. // RemoteAddr returns the remote network address.
  13. func (s *SessionUDP) RemoteAddr() net.Addr { return s.raddr }
  14. // ReadFromSessionUDP acts just like net.UDPConn.ReadFrom(), but returns a session object instead of a
  15. // net.UDPAddr.
  16. func ReadFromSessionUDP(conn *net.UDPConn, b []byte) (int, *SessionUDP, error) {
  17. oob := make([]byte, 40)
  18. n, oobn, _, raddr, err := conn.ReadMsgUDP(b, oob)
  19. if err != nil {
  20. return n, nil, err
  21. }
  22. return n, &SessionUDP{raddr, oob[:oobn]}, err
  23. }
  24. // WriteToSessionUDP acts just like net.UDPConn.WritetTo(), but uses a *SessionUDP instead of a net.Addr.
  25. func WriteToSessionUDP(conn *net.UDPConn, b []byte, session *SessionUDP) (int, error) {
  26. n, _, err := conn.WriteMsgUDP(b, session.context, session.raddr)
  27. return n, err
  28. }