179 lines
4.2 KiB
Go
179 lines
4.2 KiB
Go
|
// Copyright 2009 The Go Authors. All rights reserved.
|
||
|
// Use of this source code is governed by a BSD-style
|
||
|
// license that can be found in the LICENSE file.
|
||
|
|
||
|
// Sockets
|
||
|
|
||
|
package net
|
||
|
|
||
|
import (
|
||
|
"os"
|
||
|
"reflect"
|
||
|
"syscall"
|
||
|
)
|
||
|
|
||
|
// Boolean to int.
|
||
|
func boolint(b bool) int {
|
||
|
if b {
|
||
|
return 1
|
||
|
}
|
||
|
return 0
|
||
|
}
|
||
|
|
||
|
// Generic socket creation.
|
||
|
func socket(net string, f, p, t int, la, ra syscall.Sockaddr, toAddr func(syscall.Sockaddr) Addr) (fd *netFD, err os.Error) {
|
||
|
// See ../syscall/exec.go for description of ForkLock.
|
||
|
syscall.ForkLock.RLock()
|
||
|
s, e := syscall.Socket(f, p, t)
|
||
|
if e != 0 {
|
||
|
syscall.ForkLock.RUnlock()
|
||
|
return nil, os.Errno(e)
|
||
|
}
|
||
|
syscall.CloseOnExec(s)
|
||
|
syscall.ForkLock.RUnlock()
|
||
|
|
||
|
// Allow reuse of recently-used addresses.
|
||
|
syscall.SetsockoptInt(s, syscall.SOL_SOCKET, syscall.SO_REUSEADDR, 1)
|
||
|
|
||
|
// Allow broadcast.
|
||
|
syscall.SetsockoptInt(s, syscall.SOL_SOCKET, syscall.SO_BROADCAST, 1)
|
||
|
|
||
|
if f == syscall.AF_INET6 {
|
||
|
// using ip, tcp, udp, etc.
|
||
|
// allow both protocols even if the OS default is otherwise.
|
||
|
syscall.SetsockoptInt(s, syscall.IPPROTO_IPV6, syscall.IPV6_V6ONLY, 0)
|
||
|
}
|
||
|
|
||
|
if la != nil {
|
||
|
e = syscall.Bind(s, la)
|
||
|
if e != 0 {
|
||
|
syscall.Close(s)
|
||
|
return nil, os.Errno(e)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if ra != nil {
|
||
|
e = syscall.Connect(s, ra)
|
||
|
if e != 0 {
|
||
|
syscall.Close(s)
|
||
|
return nil, os.Errno(e)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
sa, _ := syscall.Getsockname(s)
|
||
|
laddr := toAddr(sa)
|
||
|
sa, _ = syscall.Getpeername(s)
|
||
|
raddr := toAddr(sa)
|
||
|
|
||
|
fd, err = newFD(s, f, p, net, laddr, raddr)
|
||
|
if err != nil {
|
||
|
syscall.Close(s)
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
return fd, nil
|
||
|
}
|
||
|
|
||
|
func setsockoptInt(fd, level, opt int, value int) os.Error {
|
||
|
return os.NewSyscallError("setsockopt", syscall.SetsockoptInt(fd, level, opt, value))
|
||
|
}
|
||
|
|
||
|
func setsockoptNsec(fd, level, opt int, nsec int64) os.Error {
|
||
|
var tv = syscall.NsecToTimeval(nsec)
|
||
|
return os.NewSyscallError("setsockopt", syscall.SetsockoptTimeval(fd, level, opt, &tv))
|
||
|
}
|
||
|
|
||
|
func setReadBuffer(fd *netFD, bytes int) os.Error {
|
||
|
fd.incref()
|
||
|
defer fd.decref()
|
||
|
return setsockoptInt(fd.sysfd, syscall.SOL_SOCKET, syscall.SO_RCVBUF, bytes)
|
||
|
}
|
||
|
|
||
|
func setWriteBuffer(fd *netFD, bytes int) os.Error {
|
||
|
fd.incref()
|
||
|
defer fd.decref()
|
||
|
return setsockoptInt(fd.sysfd, syscall.SOL_SOCKET, syscall.SO_SNDBUF, bytes)
|
||
|
}
|
||
|
|
||
|
func setReadTimeout(fd *netFD, nsec int64) os.Error {
|
||
|
fd.rdeadline_delta = nsec
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func setWriteTimeout(fd *netFD, nsec int64) os.Error {
|
||
|
fd.wdeadline_delta = nsec
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func setTimeout(fd *netFD, nsec int64) os.Error {
|
||
|
if e := setReadTimeout(fd, nsec); e != nil {
|
||
|
return e
|
||
|
}
|
||
|
return setWriteTimeout(fd, nsec)
|
||
|
}
|
||
|
|
||
|
func setReuseAddr(fd *netFD, reuse bool) os.Error {
|
||
|
fd.incref()
|
||
|
defer fd.decref()
|
||
|
return setsockoptInt(fd.sysfd, syscall.SOL_SOCKET, syscall.SO_REUSEADDR, boolint(reuse))
|
||
|
}
|
||
|
|
||
|
func bindToDevice(fd *netFD, dev string) os.Error {
|
||
|
// TODO(rsc): call setsockopt with null-terminated string pointer
|
||
|
return os.EINVAL
|
||
|
}
|
||
|
|
||
|
func setDontRoute(fd *netFD, dontroute bool) os.Error {
|
||
|
fd.incref()
|
||
|
defer fd.decref()
|
||
|
return setsockoptInt(fd.sysfd, syscall.SOL_SOCKET, syscall.SO_DONTROUTE, boolint(dontroute))
|
||
|
}
|
||
|
|
||
|
func setKeepAlive(fd *netFD, keepalive bool) os.Error {
|
||
|
fd.incref()
|
||
|
defer fd.decref()
|
||
|
return setsockoptInt(fd.sysfd, syscall.SOL_SOCKET, syscall.SO_KEEPALIVE, boolint(keepalive))
|
||
|
}
|
||
|
|
||
|
func setNoDelay(fd *netFD, noDelay bool) os.Error {
|
||
|
fd.incref()
|
||
|
defer fd.decref()
|
||
|
return setsockoptInt(fd.sysfd, syscall.IPPROTO_TCP, syscall.TCP_NODELAY, boolint(noDelay))
|
||
|
}
|
||
|
|
||
|
func setLinger(fd *netFD, sec int) os.Error {
|
||
|
var l syscall.Linger
|
||
|
if sec >= 0 {
|
||
|
l.Onoff = 1
|
||
|
l.Linger = int32(sec)
|
||
|
} else {
|
||
|
l.Onoff = 0
|
||
|
l.Linger = 0
|
||
|
}
|
||
|
fd.incref()
|
||
|
defer fd.decref()
|
||
|
e := syscall.SetsockoptLinger(fd.sysfd, syscall.SOL_SOCKET, syscall.SO_LINGER, &l)
|
||
|
return os.NewSyscallError("setsockopt", e)
|
||
|
}
|
||
|
|
||
|
type UnknownSocketError struct {
|
||
|
sa syscall.Sockaddr
|
||
|
}
|
||
|
|
||
|
func (e *UnknownSocketError) String() string {
|
||
|
return "unknown socket address type " + reflect.Typeof(e.sa).String()
|
||
|
}
|
||
|
|
||
|
func sockaddrToString(sa syscall.Sockaddr) (name string, err os.Error) {
|
||
|
switch a := sa.(type) {
|
||
|
case *syscall.SockaddrInet4:
|
||
|
return joinHostPort(IP(a.Addr[0:]).String(), itoa(a.Port)), nil
|
||
|
case *syscall.SockaddrInet6:
|
||
|
return joinHostPort(IP(a.Addr[0:]).String(), itoa(a.Port)), nil
|
||
|
case *syscall.SockaddrUnix:
|
||
|
return a.Name, nil
|
||
|
}
|
||
|
|
||
|
return "", &UnknownSocketError{sa}
|
||
|
}
|