123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140 |
- // Copyright 2011 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 darwin dragonfly freebsd linux netbsd openbsd solaris
- package net
- import (
- "os"
- "syscall"
- )
- func newFileFD(f *os.File) (*netFD, error) {
- fd, err := dupCloseOnExec(int(f.Fd()))
- if err != nil {
- return nil, os.NewSyscallError("dup", err)
- }
- if err = syscall.SetNonblock(fd, true); err != nil {
- closesocket(fd)
- return nil, err
- }
- sotype, err := syscall.GetsockoptInt(fd, syscall.SOL_SOCKET, syscall.SO_TYPE)
- if err != nil {
- closesocket(fd)
- return nil, os.NewSyscallError("getsockopt", err)
- }
- family := syscall.AF_UNSPEC
- toAddr := sockaddrToTCP
- lsa, _ := syscall.Getsockname(fd)
- switch lsa.(type) {
- default:
- closesocket(fd)
- return nil, syscall.EINVAL
- case *syscall.SockaddrInet4:
- family = syscall.AF_INET
- if sotype == syscall.SOCK_DGRAM {
- toAddr = sockaddrToUDP
- } else if sotype == syscall.SOCK_RAW {
- toAddr = sockaddrToIP
- }
- case *syscall.SockaddrInet6:
- family = syscall.AF_INET6
- if sotype == syscall.SOCK_DGRAM {
- toAddr = sockaddrToUDP
- } else if sotype == syscall.SOCK_RAW {
- toAddr = sockaddrToIP
- }
- case *syscall.SockaddrUnix:
- family = syscall.AF_UNIX
- toAddr = sockaddrToUnix
- if sotype == syscall.SOCK_DGRAM {
- toAddr = sockaddrToUnixgram
- } else if sotype == syscall.SOCK_SEQPACKET {
- toAddr = sockaddrToUnixpacket
- }
- }
- laddr := toAddr(lsa)
- rsa, _ := syscall.Getpeername(fd)
- raddr := toAddr(rsa)
- netfd, err := newFD(fd, family, sotype, laddr.Network())
- if err != nil {
- closesocket(fd)
- return nil, err
- }
- if err := netfd.init(); err != nil {
- netfd.Close()
- return nil, err
- }
- netfd.setAddr(laddr, raddr)
- return netfd, nil
- }
- // FileConn returns a copy of the network connection corresponding to
- // the open file f. It is the caller's responsibility to close f when
- // finished. Closing c does not affect f, and closing f does not
- // affect c.
- func FileConn(f *os.File) (c Conn, err error) {
- fd, err := newFileFD(f)
- if err != nil {
- return nil, err
- }
- switch fd.laddr.(type) {
- case *TCPAddr:
- return newTCPConn(fd), nil
- case *UDPAddr:
- return newUDPConn(fd), nil
- case *IPAddr:
- return newIPConn(fd), nil
- case *UnixAddr:
- return newUnixConn(fd), nil
- }
- fd.Close()
- return nil, syscall.EINVAL
- }
- // FileListener returns a copy of the network listener corresponding
- // to the open file f. It is the caller's responsibility to close l
- // when finished. Closing l does not affect f, and closing f does not
- // affect l.
- func FileListener(f *os.File) (l Listener, err error) {
- fd, err := newFileFD(f)
- if err != nil {
- return nil, err
- }
- switch laddr := fd.laddr.(type) {
- case *TCPAddr:
- return &TCPListener{fd}, nil
- case *UnixAddr:
- return &UnixListener{fd, laddr.Name}, nil
- }
- fd.Close()
- return nil, syscall.EINVAL
- }
- // FilePacketConn returns a copy of the packet network connection
- // corresponding to the open file f. It is the caller's
- // responsibility to close f when finished. Closing c does not affect
- // f, and closing f does not affect c.
- func FilePacketConn(f *os.File) (c PacketConn, err error) {
- fd, err := newFileFD(f)
- if err != nil {
- return nil, err
- }
- switch fd.laddr.(type) {
- case *UDPAddr:
- return newUDPConn(fd), nil
- case *IPAddr:
- return newIPConn(fd), nil
- case *UnixAddr:
- return newUnixConn(fd), nil
- }
- fd.Close()
- return nil, syscall.EINVAL
- }
|