libgo/go/net/sockoptip.go

   1 // Copyright 2011 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
   5 // +build darwin freebsd linux netbsd openbsd windows
   6
   7 // IP-level socket options
   8
   9 package net
  10
  11 import (
  12         "os"
  13         "syscall"
  14 )
  15
  16 func ipv4TOS(fd *netFD) (int, error) {
  17         if err := fd.incref(false); err != nil {
  18                 return 0, err
  19         }
  20         defer fd.decref()
  21         v, err := syscall.GetsockoptInt(fd.sysfd, syscall.IPPROTO_IP, syscall.IP_TOS)
  22         if err != nil {
  23                 return 0, os.NewSyscallError("getsockopt", err)
  24         }
  25         return v, nil
  26 }
  27
  28 func setIPv4TOS(fd *netFD, v int) error {
  29         if err := fd.incref(false); err != nil {
  30                 return err
  31         }
  32         defer fd.decref()
  33         err := syscall.SetsockoptInt(fd.sysfd, syscall.IPPROTO_IP, syscall.IP_TOS, v)
  34         if err != nil {
  35                 return os.NewSyscallError("setsockopt", err)
  36         }
  37         return nil
  38 }
  39
  40 func ipv4TTL(fd *netFD) (int, error) {
  41         if err := fd.incref(false); err != nil {
  42                 return 0, err
  43         }
  44         defer fd.decref()
  45         v, err := syscall.GetsockoptInt(fd.sysfd, syscall.IPPROTO_IP, syscall.IP_TTL)
  46         if err != nil {
  47                 return 0, os.NewSyscallError("getsockopt", err)
  48         }
  49         return v, nil
  50 }
  51
  52 func setIPv4TTL(fd *netFD, v int) error {
  53         if err := fd.incref(false); err != nil {
  54                 return err
  55         }
  56         defer fd.decref()
  57         err := syscall.SetsockoptInt(fd.sysfd, syscall.IPPROTO_IP, syscall.IP_TTL, v)
  58         if err != nil {
  59                 return os.NewSyscallError("setsockopt", err)
  60         }
  61         return nil
  62 }
  63
  64 func joinIPv4Group(fd *netFD, ifi *Interface, ip IP) error {
  65         mreq := &syscall.IPMreq{Multiaddr: [4]byte{ip[0], ip[1], ip[2], ip[3]}}
  66         if err := setIPv4MreqToInterface(mreq, ifi); err != nil {
  67                 return err
  68         }
  69         if err := fd.incref(false); err != nil {
  70                 return err
  71         }
  72         defer fd.decref()
  73         return os.NewSyscallError("setsockopt", syscall.SetsockoptIPMreq(fd.sysfd, syscall.IPPROTO_IP, syscall.IP_ADD_MEMBERSHIP, mreq))
  74 }
  75
  76 func leaveIPv4Group(fd *netFD, ifi *Interface, ip IP) error {
  77         mreq := &syscall.IPMreq{Multiaddr: [4]byte{ip[0], ip[1], ip[2], ip[3]}}
  78         if err := setIPv4MreqToInterface(mreq, ifi); err != nil {
  79                 return err
  80         }
  81         if err := fd.incref(false); err != nil {
  82                 return err
  83         }
  84         defer fd.decref()
  85         return os.NewSyscallError("setsockopt", syscall.SetsockoptIPMreq(fd.sysfd, syscall.IPPROTO_IP, syscall.IP_DROP_MEMBERSHIP, mreq))
  86 }
  87
  88 func ipv6HopLimit(fd *netFD) (int, error) {
  89         if err := fd.incref(false); err != nil {
  90                 return 0, err
  91         }
  92         defer fd.decref()
  93         v, err := syscall.GetsockoptInt(fd.sysfd, syscall.IPPROTO_IPV6, syscall.IPV6_UNICAST_HOPS)
  94         if err != nil {
  95                 return 0, os.NewSyscallError("getsockopt", err)
  96         }
  97         return v, nil
  98 }
  99
 100 func setIPv6HopLimit(fd *netFD, v int) error {
 101         if err := fd.incref(false); err != nil {
 102                 return err
 103         }
 104         defer fd.decref()
 105         err := syscall.SetsockoptInt(fd.sysfd, syscall.IPPROTO_IPV6, syscall.IPV6_UNICAST_HOPS, v)
 106         if err != nil {
 107                 return os.NewSyscallError("setsockopt", err)
 108         }
 109         return nil
 110 }
 111
 112 func ipv6MulticastInterface(fd *netFD) (*Interface, error) {
 113         if err := fd.incref(false); err != nil {
 114                 return nil, err
 115         }
 116         defer fd.decref()
 117         v, err := syscall.GetsockoptInt(fd.sysfd, syscall.IPPROTO_IPV6, syscall.IPV6_MULTICAST_IF)
 118         if err != nil {
 119                 return nil, os.NewSyscallError("getsockopt", err)
 120         }
 121         if v == 0 {
 122                 return nil, nil
 123         }
 124         ifi, err := InterfaceByIndex(v)
 125         if err != nil {
 126                 return nil, err
 127         }
 128         return ifi, nil
 129 }
 130
 131 func setIPv6MulticastInterface(fd *netFD, ifi *Interface) error {
 132         var v int
 133         if ifi != nil {
 134                 v = ifi.Index
 135         }
 136         if err := fd.incref(false); err != nil {
 137                 return err
 138         }
 139         defer fd.decref()
 140         err := syscall.SetsockoptInt(fd.sysfd, syscall.IPPROTO_IPV6, syscall.IPV6_MULTICAST_IF, v)
 141         if err != nil {
 142                 return os.NewSyscallError("setsockopt", err)
 143         }
 144         return nil
 145 }
 146
 147 func ipv6MulticastHopLimit(fd *netFD) (int, error) {
 148         if err := fd.incref(false); err != nil {
 149                 return 0, err
 150         }
 151         defer fd.decref()
 152         v, err := syscall.GetsockoptInt(fd.sysfd, syscall.IPPROTO_IPV6, syscall.IPV6_MULTICAST_HOPS)
 153         if err != nil {
 154                 return 0, os.NewSyscallError("getsockopt", err)
 155         }
 156         return v, nil
 157 }
 158
 159 func setIPv6MulticastHopLimit(fd *netFD, v int) error {
 160         if err := fd.incref(false); err != nil {
 161                 return err
 162         }
 163         defer fd.decref()
 164         err := syscall.SetsockoptInt(fd.sysfd, syscall.IPPROTO_IPV6, syscall.IPV6_MULTICAST_HOPS, v)
 165         if err != nil {
 166                 return os.NewSyscallError("setsockopt", err)
 167         }
 168         return nil
 169 }
 170
 171 func ipv6MulticastLoopback(fd *netFD) (bool, error) {
 172         if err := fd.incref(false); err != nil {
 173                 return false, err
 174         }
 175         defer fd.decref()
 176         v, err := syscall.GetsockoptInt(fd.sysfd, syscall.IPPROTO_IPV6, syscall.IPV6_MULTICAST_LOOP)
 177         if err != nil {
 178                 return false, os.NewSyscallError("getsockopt", err)
 179         }
 180         return v == 1, nil
 181 }
 182
 183 func setIPv6MulticastLoopback(fd *netFD, v bool) error {
 184         if err := fd.incref(false); err != nil {
 185                 return err
 186         }
 187         defer fd.decref()
 188         err := syscall.SetsockoptInt(fd.sysfd, syscall.IPPROTO_IPV6, syscall.IPV6_MULTICAST_LOOP, boolint(v))
 189         if err != nil {
 190                 return os.NewSyscallError("setsockopt", err)
 191         }
 192         return nil
 193 }
 194
 195 func joinIPv6Group(fd *netFD, ifi *Interface, ip IP) error {
 196         mreq := &syscall.IPv6Mreq{}
 197         copy(mreq.Multiaddr[:], ip)
 198         if ifi != nil {
 199                 mreq.Interface = uint32(ifi.Index)
 200         }
 201         if err := fd.incref(false); err != nil {
 202                 return err
 203         }
 204         defer fd.decref()
 205         return os.NewSyscallError("setsockopt", syscall.SetsockoptIPv6Mreq(fd.sysfd, syscall.IPPROTO_IPV6, syscall.IPV6_JOIN_GROUP, mreq))
 206 }
 207
 208 func leaveIPv6Group(fd *netFD, ifi *Interface, ip IP) error {
 209         mreq := &syscall.IPv6Mreq{}
 210         copy(mreq.Multiaddr[:], ip)
 211         if ifi != nil {
 212                 mreq.Interface = uint32(ifi.Index)
 213         }
 214         if err := fd.incref(false); err != nil {
 215                 return err
 216         }
 217         defer fd.decref()
 218         return os.NewSyscallError("setsockopt", syscall.SetsockoptIPv6Mreq(fd.sysfd, syscall.IPPROTO_IPV6, syscall.IPV6_LEAVE_GROUP, mreq))
 219 }