2009-11-01 12:15:34 -07:00
|
|
|
// 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.
|
|
|
|
|
build: add build comments to core packages
The go/build package already recognizes
system-specific file names like
mycode_darwin.go
mycode_darwin_386.go
mycode_386.s
However, it is also common to write files that
apply to multiple architectures, so a recent CL added
to go/build the ability to process comments
listing a set of conditions for building. For example:
// +build darwin freebsd openbsd/386
says that this file should be compiled only on
OS X, FreeBSD, or 32-bit x86 OpenBSD systems.
These conventions are not yet documented
(hence this long CL description).
This CL adds build comments to the multi-system
files in the core library, a step toward making it
possible to use go/build to build them.
With this change go/build can handle crypto/rand,
exec, net, path/filepath, os/user, and time.
os and syscall need additional adjustments.
R=golang-dev, r, gri, r, gustavo
CC=golang-dev
https://golang.org/cl/5011046
2011-09-15 14:48:57 -06:00
|
|
|
// +build darwin freebsd linux openbsd windows
|
|
|
|
|
2009-11-01 12:15:34 -07:00
|
|
|
// Sockets
|
|
|
|
|
|
|
|
package net
|
|
|
|
|
|
|
|
import (
|
2011-05-25 11:15:26 -06:00
|
|
|
"io"
|
2009-12-15 16:35:38 -07:00
|
|
|
"os"
|
|
|
|
"reflect"
|
|
|
|
"syscall"
|
2009-11-01 12:15:34 -07:00
|
|
|
)
|
|
|
|
|
|
|
|
// Boolean to int.
|
|
|
|
func boolint(b bool) int {
|
|
|
|
if b {
|
2009-11-09 13:07:39 -07:00
|
|
|
return 1
|
2009-11-01 12:15:34 -07:00
|
|
|
}
|
2009-12-15 16:35:38 -07:00
|
|
|
return 0
|
2009-11-01 12:15:34 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
// Generic socket creation.
|
2011-11-01 20:05:34 -06:00
|
|
|
func socket(net string, f, p, t int, la, ra syscall.Sockaddr, toAddr func(syscall.Sockaddr) Addr) (fd *netFD, err error) {
|
2009-11-01 12:15:34 -07:00
|
|
|
// See ../syscall/exec.go for description of ForkLock.
|
2009-12-15 16:35:38 -07:00
|
|
|
syscall.ForkLock.RLock()
|
|
|
|
s, e := syscall.Socket(f, p, t)
|
2011-11-13 20:44:52 -07:00
|
|
|
if err != nil {
|
2009-12-15 16:35:38 -07:00
|
|
|
syscall.ForkLock.RUnlock()
|
2011-11-13 20:44:52 -07:00
|
|
|
return nil, err
|
2009-11-01 12:15:34 -07:00
|
|
|
}
|
2009-12-15 16:35:38 -07:00
|
|
|
syscall.CloseOnExec(s)
|
|
|
|
syscall.ForkLock.RUnlock()
|
2009-11-01 12:15:34 -07:00
|
|
|
|
2011-05-02 08:50:12 -06:00
|
|
|
setKernelSpecificSockopt(s, f)
|
2010-09-24 14:52:06 -06:00
|
|
|
|
2011-03-29 12:23:42 -06:00
|
|
|
if la != nil {
|
|
|
|
e = syscall.Bind(s, la)
|
2011-11-13 20:44:52 -07:00
|
|
|
if e != nil {
|
2011-03-29 12:23:42 -06:00
|
|
|
closesocket(s)
|
2011-11-13 20:44:52 -07:00
|
|
|
return nil, e
|
2011-03-29 12:23:42 -06:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2011-03-28 21:40:01 -06:00
|
|
|
if fd, err = newFD(s, f, p, net); err != nil {
|
|
|
|
closesocket(s)
|
|
|
|
return nil, err
|
2009-11-01 12:15:34 -07:00
|
|
|
}
|
|
|
|
|
2011-03-29 12:23:42 -06:00
|
|
|
if ra != nil {
|
|
|
|
if err = fd.connect(ra); err != nil {
|
2011-10-11 10:53:16 -06:00
|
|
|
closesocket(s)
|
2011-07-01 08:18:07 -06:00
|
|
|
fd.Close()
|
2011-03-29 12:23:42 -06:00
|
|
|
return nil, err
|
|
|
|
}
|
2009-11-01 12:15:34 -07:00
|
|
|
}
|
|
|
|
|
2011-03-29 12:23:42 -06:00
|
|
|
sa, _ := syscall.Getsockname(s)
|
2009-12-15 16:35:38 -07:00
|
|
|
laddr := toAddr(sa)
|
2011-03-29 12:23:42 -06:00
|
|
|
sa, _ = syscall.Getpeername(s)
|
2009-12-15 16:35:38 -07:00
|
|
|
raddr := toAddr(sa)
|
2009-11-02 19:37:30 -07:00
|
|
|
|
2011-03-28 21:40:01 -06:00
|
|
|
fd.setAddr(laddr, raddr)
|
2009-12-15 16:35:38 -07:00
|
|
|
return fd, nil
|
2009-11-01 12:15:34 -07:00
|
|
|
}
|
|
|
|
|
2011-11-01 20:05:34 -06:00
|
|
|
func setsockoptInt(fd *netFD, level, opt int, value int) error {
|
2011-07-01 08:18:07 -06:00
|
|
|
return os.NewSyscallError("setsockopt", syscall.SetsockoptInt(fd.sysfd, level, opt, value))
|
2009-11-01 12:15:34 -07:00
|
|
|
}
|
|
|
|
|
2011-11-01 20:05:34 -06:00
|
|
|
func setsockoptNsec(fd *netFD, level, opt int, nsec int64) error {
|
2009-12-15 16:35:38 -07:00
|
|
|
var tv = syscall.NsecToTimeval(nsec)
|
2011-07-01 08:18:07 -06:00
|
|
|
return os.NewSyscallError("setsockopt", syscall.SetsockoptTimeval(fd.sysfd, level, opt, &tv))
|
2009-11-01 12:15:34 -07:00
|
|
|
}
|
|
|
|
|
2011-11-01 20:05:34 -06:00
|
|
|
func setReadBuffer(fd *netFD, bytes int) error {
|
2009-12-15 16:35:38 -07:00
|
|
|
fd.incref()
|
|
|
|
defer fd.decref()
|
2011-07-01 08:18:07 -06:00
|
|
|
return setsockoptInt(fd, syscall.SOL_SOCKET, syscall.SO_RCVBUF, bytes)
|
2009-11-01 12:15:34 -07:00
|
|
|
}
|
|
|
|
|
2011-11-01 20:05:34 -06:00
|
|
|
func setWriteBuffer(fd *netFD, bytes int) error {
|
2009-12-15 16:35:38 -07:00
|
|
|
fd.incref()
|
|
|
|
defer fd.decref()
|
2011-07-01 08:18:07 -06:00
|
|
|
return setsockoptInt(fd, syscall.SOL_SOCKET, syscall.SO_SNDBUF, bytes)
|
2009-11-01 12:15:34 -07:00
|
|
|
}
|
|
|
|
|
2011-11-01 20:05:34 -06:00
|
|
|
func setReadTimeout(fd *netFD, nsec int64) error {
|
2009-12-15 16:35:38 -07:00
|
|
|
fd.rdeadline_delta = nsec
|
|
|
|
return nil
|
2009-11-01 12:15:34 -07:00
|
|
|
}
|
|
|
|
|
2011-11-01 20:05:34 -06:00
|
|
|
func setWriteTimeout(fd *netFD, nsec int64) error {
|
2009-12-15 16:35:38 -07:00
|
|
|
fd.wdeadline_delta = nsec
|
|
|
|
return nil
|
2009-11-01 12:15:34 -07:00
|
|
|
}
|
|
|
|
|
2011-11-01 20:05:34 -06:00
|
|
|
func setTimeout(fd *netFD, nsec int64) error {
|
2009-11-02 19:37:30 -07:00
|
|
|
if e := setReadTimeout(fd, nsec); e != nil {
|
2009-11-09 13:07:39 -07:00
|
|
|
return e
|
2009-11-01 12:15:34 -07:00
|
|
|
}
|
2009-12-15 16:35:38 -07:00
|
|
|
return setWriteTimeout(fd, nsec)
|
2009-11-01 12:15:34 -07:00
|
|
|
}
|
|
|
|
|
2011-11-01 20:05:34 -06:00
|
|
|
func setReuseAddr(fd *netFD, reuse bool) error {
|
2009-12-15 16:35:38 -07:00
|
|
|
fd.incref()
|
|
|
|
defer fd.decref()
|
2011-07-01 08:18:07 -06:00
|
|
|
return setsockoptInt(fd, syscall.SOL_SOCKET, syscall.SO_REUSEADDR, boolint(reuse))
|
2009-11-01 12:15:34 -07:00
|
|
|
}
|
|
|
|
|
2011-11-01 20:05:34 -06:00
|
|
|
func bindToDevice(fd *netFD, dev string) error {
|
2009-11-01 12:15:34 -07:00
|
|
|
// TODO(rsc): call setsockopt with null-terminated string pointer
|
2009-11-09 13:07:39 -07:00
|
|
|
return os.EINVAL
|
2009-11-01 12:15:34 -07:00
|
|
|
}
|
|
|
|
|
2011-11-01 20:05:34 -06:00
|
|
|
func setDontRoute(fd *netFD, dontroute bool) error {
|
2009-12-15 16:35:38 -07:00
|
|
|
fd.incref()
|
|
|
|
defer fd.decref()
|
2011-07-01 08:18:07 -06:00
|
|
|
return setsockoptInt(fd, syscall.SOL_SOCKET, syscall.SO_DONTROUTE, boolint(dontroute))
|
2009-11-01 12:15:34 -07:00
|
|
|
}
|
|
|
|
|
2011-11-01 20:05:34 -06:00
|
|
|
func setKeepAlive(fd *netFD, keepalive bool) error {
|
2009-12-15 16:35:38 -07:00
|
|
|
fd.incref()
|
|
|
|
defer fd.decref()
|
2011-07-01 08:18:07 -06:00
|
|
|
return setsockoptInt(fd, syscall.SOL_SOCKET, syscall.SO_KEEPALIVE, boolint(keepalive))
|
2009-11-01 12:15:34 -07:00
|
|
|
}
|
|
|
|
|
2011-11-01 20:05:34 -06:00
|
|
|
func setNoDelay(fd *netFD, noDelay bool) error {
|
2010-07-26 17:19:39 -06:00
|
|
|
fd.incref()
|
|
|
|
defer fd.decref()
|
2011-07-01 08:18:07 -06:00
|
|
|
return setsockoptInt(fd, syscall.IPPROTO_TCP, syscall.TCP_NODELAY, boolint(noDelay))
|
2010-07-26 17:19:39 -06:00
|
|
|
}
|
|
|
|
|
2011-11-01 20:05:34 -06:00
|
|
|
func setLinger(fd *netFD, sec int) error {
|
2009-12-15 16:35:38 -07:00
|
|
|
var l syscall.Linger
|
2009-11-01 12:15:34 -07:00
|
|
|
if sec >= 0 {
|
2009-12-15 16:35:38 -07:00
|
|
|
l.Onoff = 1
|
|
|
|
l.Linger = int32(sec)
|
2009-11-01 12:15:34 -07:00
|
|
|
} else {
|
2009-12-15 16:35:38 -07:00
|
|
|
l.Onoff = 0
|
|
|
|
l.Linger = 0
|
2009-11-01 12:15:34 -07:00
|
|
|
}
|
2009-12-15 16:35:38 -07:00
|
|
|
fd.incref()
|
|
|
|
defer fd.decref()
|
|
|
|
e := syscall.SetsockoptLinger(fd.sysfd, syscall.SOL_SOCKET, syscall.SO_LINGER, &l)
|
|
|
|
return os.NewSyscallError("setsockopt", e)
|
2009-11-01 12:15:34 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
type UnknownSocketError struct {
|
2009-12-15 16:35:38 -07:00
|
|
|
sa syscall.Sockaddr
|
2009-11-01 12:15:34 -07:00
|
|
|
}
|
2009-11-05 00:16:46 -07:00
|
|
|
|
2011-11-01 20:05:34 -06:00
|
|
|
func (e *UnknownSocketError) Error() string {
|
2011-04-25 11:39:36 -06:00
|
|
|
return "unknown socket address type " + reflect.TypeOf(e.sa).String()
|
2009-11-01 12:15:34 -07:00
|
|
|
}
|
2011-05-25 11:15:26 -06:00
|
|
|
|
|
|
|
type writerOnly struct {
|
|
|
|
io.Writer
|
|
|
|
}
|
|
|
|
|
|
|
|
// Fallback implementation of io.ReaderFrom's ReadFrom, when sendfile isn't
|
|
|
|
// applicable.
|
2011-11-01 20:05:34 -06:00
|
|
|
func genericReadFrom(w io.Writer, r io.Reader) (n int64, err error) {
|
2011-05-25 11:15:26 -06:00
|
|
|
// Use wrapper to hide existing r.ReadFrom from io.Copy.
|
|
|
|
return io.Copy(writerOnly{w}, r)
|
|
|
|
}
|