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.
|
|
|
|
|
2011-12-21 03:44:47 -07:00
|
|
|
// +build darwin freebsd linux netbsd openbsd windows
|
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
|
|
|
|
2009-11-01 12:15:34 -07:00
|
|
|
package net
|
|
|
|
|
|
|
|
import (
|
2013-08-06 15:15:50 -06:00
|
|
|
"os"
|
2009-12-15 16:35:38 -07:00
|
|
|
"syscall"
|
2012-11-08 09:35:16 -07:00
|
|
|
"time"
|
2009-11-01 12:15:34 -07:00
|
|
|
)
|
|
|
|
|
2013-07-28 01:15:07 -06:00
|
|
|
// A sockaddr represents a TCP, UDP, IP or Unix network endpoint
|
|
|
|
// address that can be converted into a syscall.Sockaddr.
|
2013-07-27 21:52:30 -06:00
|
|
|
type sockaddr interface {
|
|
|
|
Addr
|
2013-07-29 08:25:39 -06:00
|
|
|
|
|
|
|
// family returns the platform-dependent address family
|
|
|
|
// identifier.
|
2013-07-27 21:52:30 -06:00
|
|
|
family() int
|
2013-07-29 08:25:39 -06:00
|
|
|
|
|
|
|
// isWildcard reports whether the address is a wildcard
|
|
|
|
// address.
|
2013-07-27 21:52:30 -06:00
|
|
|
isWildcard() bool
|
2013-07-29 08:25:39 -06:00
|
|
|
|
|
|
|
// sockaddr returns the address converted into a syscall
|
|
|
|
// sockaddr type that implements syscall.Sockaddr
|
|
|
|
// interface. It returns a nil interface when the address is
|
|
|
|
// nil.
|
2013-07-27 21:52:30 -06:00
|
|
|
sockaddr(family int) (syscall.Sockaddr, error)
|
2013-07-29 08:25:39 -06:00
|
|
|
|
|
|
|
// toAddr returns the address represented in sockaddr
|
|
|
|
// interface. It returns a nil interface when the address is
|
|
|
|
// nil.
|
2013-07-27 21:52:30 -06:00
|
|
|
toAddr() sockaddr
|
|
|
|
}
|
|
|
|
|
2013-02-08 01:02:08 -07:00
|
|
|
// Generic POSIX socket creation.
|
2013-08-02 22:32:22 -06:00
|
|
|
func socket(net string, f, t, p int, ipv6only bool, laddr, raddr sockaddr, deadline time.Time, toAddr func(syscall.Sockaddr) Addr) (fd *netFD, err error) {
|
2013-01-28 09:54:15 -07:00
|
|
|
s, err := sysSocket(f, t, p)
|
2011-11-13 20:44:52 -07:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
2009-11-01 12:15:34 -07:00
|
|
|
}
|
|
|
|
|
2012-08-23 05:54:00 -06:00
|
|
|
if err = setDefaultSockopts(s, f, t, ipv6only); err != nil {
|
2012-01-31 09:53:26 -07:00
|
|
|
closesocket(s)
|
|
|
|
return nil, err
|
|
|
|
}
|
2010-09-24 14:52:06 -06:00
|
|
|
|
2013-08-06 15:15:50 -06:00
|
|
|
if fd, err = newFD(s, f, t, net); err != nil {
|
|
|
|
closesocket(s)
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
if err := fd.init(); err != nil {
|
|
|
|
fd.Close()
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2013-08-02 22:32:22 -06:00
|
|
|
// This function makes a network file descriptor for stream
|
|
|
|
// and datagram dialers, stream and datagram listeners.
|
|
|
|
//
|
|
|
|
// For dialers, they will require either named or unnamed
|
|
|
|
// sockets for their flights. We can assume that it's just a
|
|
|
|
// request from a dialer that wants a named socket when both
|
|
|
|
// laddr and raddr are not nil. A dialer will also require a
|
|
|
|
// connection setup initiated socket when raddr is not nil.
|
|
|
|
//
|
|
|
|
// For listeners and some dialers on datagram networks, they
|
|
|
|
// will only require named sockets. So we can assume that
|
|
|
|
// it's just for a listener or a datagram dialer when laddr is
|
|
|
|
// not nil but raddr is nil.
|
|
|
|
|
|
|
|
if laddr != nil && raddr == nil {
|
2013-08-06 15:15:50 -06:00
|
|
|
switch t {
|
|
|
|
case syscall.SOCK_STREAM, syscall.SOCK_SEQPACKET:
|
|
|
|
if err := fd.listenStream(laddr, toAddr); err != nil {
|
|
|
|
fd.Close()
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
return fd, nil
|
|
|
|
case syscall.SOCK_DGRAM:
|
|
|
|
if err := fd.listenDatagram(laddr, toAddr); err != nil {
|
|
|
|
fd.Close()
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
return fd, nil
|
2012-01-31 09:53:26 -07:00
|
|
|
}
|
2013-03-30 00:21:22 -06:00
|
|
|
}
|
2013-08-06 15:15:50 -06:00
|
|
|
if err := fd.dial(laddr, raddr, deadline, toAddr); err != nil {
|
2013-08-06 08:42:33 -06:00
|
|
|
fd.Close()
|
|
|
|
return nil, err
|
|
|
|
}
|
2013-08-06 15:15:50 -06:00
|
|
|
return fd, nil
|
|
|
|
}
|
2009-11-01 12:15:34 -07:00
|
|
|
|
2013-08-06 15:15:50 -06:00
|
|
|
func (fd *netFD) dial(laddr, raddr sockaddr, deadline time.Time, toAddr func(syscall.Sockaddr) Addr) error {
|
|
|
|
var err error
|
|
|
|
var lsa syscall.Sockaddr
|
|
|
|
if laddr != nil {
|
|
|
|
if lsa, err = laddr.sockaddr(fd.family); err != nil {
|
|
|
|
return err
|
|
|
|
} else if lsa != nil {
|
|
|
|
if err := syscall.Bind(fd.sysfd, lsa); err != nil {
|
|
|
|
return os.NewSyscallError("bind", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2013-08-02 22:32:22 -06:00
|
|
|
var rsa syscall.Sockaddr
|
|
|
|
if raddr != nil {
|
2013-08-06 15:15:50 -06:00
|
|
|
if rsa, err = raddr.sockaddr(fd.family); err != nil {
|
|
|
|
return err
|
|
|
|
} else if rsa != nil {
|
|
|
|
if !deadline.IsZero() {
|
|
|
|
setWriteDeadline(fd, deadline)
|
|
|
|
}
|
|
|
|
if err := fd.connect(lsa, rsa); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
fd.isConnected = true
|
|
|
|
if !deadline.IsZero() {
|
|
|
|
setWriteDeadline(fd, noDeadline)
|
|
|
|
}
|
2013-08-02 22:32:22 -06:00
|
|
|
}
|
|
|
|
}
|
2013-08-06 15:15:50 -06:00
|
|
|
lsa, _ = syscall.Getsockname(fd.sysfd)
|
|
|
|
if rsa, _ = syscall.Getpeername(fd.sysfd); rsa != nil {
|
|
|
|
fd.setAddr(toAddr(lsa), toAddr(rsa))
|
|
|
|
} else {
|
|
|
|
fd.setAddr(toAddr(lsa), raddr)
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
2013-08-02 22:32:22 -06:00
|
|
|
|
2013-08-06 15:15:50 -06:00
|
|
|
func (fd *netFD) listenStream(laddr sockaddr, toAddr func(syscall.Sockaddr) Addr) error {
|
|
|
|
if err := setDefaultListenerSockopts(fd.sysfd); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
if lsa, err := laddr.sockaddr(fd.family); err != nil {
|
|
|
|
return err
|
|
|
|
} else if lsa != nil {
|
|
|
|
if err := syscall.Bind(fd.sysfd, lsa); err != nil {
|
|
|
|
return os.NewSyscallError("bind", err)
|
2013-03-07 10:44:24 -07:00
|
|
|
}
|
2009-11-01 12:15:34 -07:00
|
|
|
}
|
2013-08-06 15:15:50 -06:00
|
|
|
lsa, _ := syscall.Getsockname(fd.sysfd)
|
|
|
|
fd.setAddr(toAddr(lsa), nil)
|
|
|
|
return nil
|
|
|
|
}
|
2009-11-01 12:15:34 -07:00
|
|
|
|
2013-08-06 15:15:50 -06:00
|
|
|
func (fd *netFD) listenDatagram(laddr sockaddr, toAddr func(syscall.Sockaddr) Addr) error {
|
|
|
|
switch addr := laddr.(type) {
|
|
|
|
case *UDPAddr:
|
|
|
|
// We provide a socket that listens to a wildcard
|
|
|
|
// address with reusable UDP port when the given laddr
|
|
|
|
// is an appropriate UDP multicast address prefix.
|
|
|
|
// This makes it possible for a single UDP listener to
|
|
|
|
// join multiple different group addresses, for
|
|
|
|
// multiple UDP listeners that listen on the same UDP
|
|
|
|
// port to join the same group address.
|
|
|
|
if addr.IP != nil && addr.IP.IsMulticast() {
|
|
|
|
if err := setDefaultMulticastSockopts(fd.sysfd); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
addr := *addr
|
|
|
|
switch fd.family {
|
|
|
|
case syscall.AF_INET:
|
|
|
|
addr.IP = IPv4zero
|
|
|
|
case syscall.AF_INET6:
|
|
|
|
addr.IP = IPv6unspecified
|
|
|
|
}
|
|
|
|
laddr = &addr
|
|
|
|
}
|
2013-03-23 07:32:19 -06:00
|
|
|
}
|
2013-08-06 15:15:50 -06:00
|
|
|
if lsa, err := laddr.sockaddr(fd.family); err != nil {
|
|
|
|
return err
|
|
|
|
} else if lsa != nil {
|
|
|
|
if err := syscall.Bind(fd.sysfd, lsa); err != nil {
|
|
|
|
return os.NewSyscallError("bind", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
lsa, _ := syscall.Getsockname(fd.sysfd)
|
|
|
|
fd.setAddr(toAddr(lsa), nil)
|
|
|
|
return nil
|
2009-11-01 12:15:34 -07:00
|
|
|
}
|