sockopt_asmreq_windows.go 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. // Copyright 2012 The Go Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style
  3. // license that can be found in the LICENSE file.
  4. package ipv4
  5. import (
  6. "net"
  7. "os"
  8. "syscall"
  9. "unsafe"
  10. )
  11. func setsockoptIPMreq(fd syscall.Handle, name int, ifi *net.Interface, grp net.IP) error {
  12. mreq := sysIPMreq{Multiaddr: [4]byte{grp[0], grp[1], grp[2], grp[3]}}
  13. if err := setIPMreqInterface(&mreq, ifi); err != nil {
  14. return err
  15. }
  16. return os.NewSyscallError("setsockopt", syscall.Setsockopt(fd, ianaProtocolIP, int32(name), (*byte)(unsafe.Pointer(&mreq)), int32(sysSizeofIPMreq)))
  17. }
  18. func getsockoptInterface(fd syscall.Handle, name int) (*net.Interface, error) {
  19. var b [4]byte
  20. l := int32(4)
  21. if err := syscall.Getsockopt(fd, ianaProtocolIP, int32(name), (*byte)(unsafe.Pointer(&b[0])), &l); err != nil {
  22. return nil, os.NewSyscallError("getsockopt", err)
  23. }
  24. ifi, err := netIP4ToInterface(net.IPv4(b[0], b[1], b[2], b[3]))
  25. if err != nil {
  26. return nil, err
  27. }
  28. return ifi, nil
  29. }
  30. func setsockoptInterface(fd syscall.Handle, name int, ifi *net.Interface) error {
  31. ip, err := netInterfaceToIP4(ifi)
  32. if err != nil {
  33. return err
  34. }
  35. var b [4]byte
  36. copy(b[:], ip)
  37. return os.NewSyscallError("setsockopt", syscall.Setsockopt(fd, ianaProtocolIP, int32(name), (*byte)(unsafe.Pointer(&b[0])), 4))
  38. }