| // 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. |
| |
| package net |
| |
| import ( |
| "internal/poll" |
| "io" |
| "os" |
| "syscall" |
| "time" |
| ) |
| |
| // Network file descriptor. |
| type netFD struct { |
| pfd poll.FD |
| |
| // immutable until Close |
| net string |
| n string |
| dir string |
| listen, ctl, data *os.File |
| laddr, raddr Addr |
| isStream bool |
| } |
| |
| var netdir = "/net" // default network |
| |
| func newFD(net, name string, listen, ctl, data *os.File, laddr, raddr Addr) (*netFD, error) { |
| ret := &netFD{ |
| net: net, |
| n: name, |
| dir: netdir + "/" + net + "/" + name, |
| listen: listen, |
| ctl: ctl, data: data, |
| laddr: laddr, |
| raddr: raddr, |
| } |
| ret.pfd.Destroy = ret.destroy |
| return ret, nil |
| } |
| |
| func (fd *netFD) init() error { |
| // stub for future fd.pd.Init(fd) |
| return nil |
| } |
| |
| func (fd *netFD) name() string { |
| var ls, rs string |
| if fd.laddr != nil { |
| ls = fd.laddr.String() |
| } |
| if fd.raddr != nil { |
| rs = fd.raddr.String() |
| } |
| return fd.net + ":" + ls + "->" + rs |
| } |
| |
| func (fd *netFD) ok() bool { return fd != nil && fd.ctl != nil } |
| |
| func (fd *netFD) destroy() { |
| if !fd.ok() { |
| return |
| } |
| err := fd.ctl.Close() |
| if fd.data != nil { |
| if err1 := fd.data.Close(); err1 != nil && err == nil { |
| err = err1 |
| } |
| } |
| if fd.listen != nil { |
| if err1 := fd.listen.Close(); err1 != nil && err == nil { |
| err = err1 |
| } |
| } |
| fd.ctl = nil |
| fd.data = nil |
| fd.listen = nil |
| } |
| |
| func (fd *netFD) Read(b []byte) (n int, err error) { |
| if !fd.ok() || fd.data == nil { |
| return 0, syscall.EINVAL |
| } |
| n, err = fd.pfd.Read(fd.data.Read, b) |
| if fd.net == "udp" && err == io.EOF { |
| n = 0 |
| err = nil |
| } |
| return |
| } |
| |
| func (fd *netFD) Write(b []byte) (n int, err error) { |
| if !fd.ok() || fd.data == nil { |
| return 0, syscall.EINVAL |
| } |
| return fd.pfd.Write(fd.data.Write, b) |
| } |
| |
| func (fd *netFD) closeRead() error { |
| if !fd.ok() { |
| return syscall.EINVAL |
| } |
| return syscall.EPLAN9 |
| } |
| |
| func (fd *netFD) closeWrite() error { |
| if !fd.ok() { |
| return syscall.EINVAL |
| } |
| return syscall.EPLAN9 |
| } |
| |
| func (fd *netFD) Close() error { |
| if err := fd.pfd.Close(); err != nil { |
| return err |
| } |
| if !fd.ok() { |
| return syscall.EINVAL |
| } |
| if fd.net == "tcp" { |
| // The following line is required to unblock Reads. |
| _, err := fd.ctl.WriteString("close") |
| if err != nil { |
| return err |
| } |
| } |
| err := fd.ctl.Close() |
| if fd.data != nil { |
| if err1 := fd.data.Close(); err1 != nil && err == nil { |
| err = err1 |
| } |
| } |
| if fd.listen != nil { |
| if err1 := fd.listen.Close(); err1 != nil && err == nil { |
| err = err1 |
| } |
| } |
| fd.ctl = nil |
| fd.data = nil |
| fd.listen = nil |
| return err |
| } |
| |
| // This method is only called via Conn. |
| func (fd *netFD) dup() (*os.File, error) { |
| if !fd.ok() || fd.data == nil { |
| return nil, syscall.EINVAL |
| } |
| return fd.file(fd.data, fd.dir+"/data") |
| } |
| |
| func (l *TCPListener) dup() (*os.File, error) { |
| if !l.fd.ok() { |
| return nil, syscall.EINVAL |
| } |
| return l.fd.file(l.fd.ctl, l.fd.dir+"/ctl") |
| } |
| |
| func (fd *netFD) file(f *os.File, s string) (*os.File, error) { |
| dfd, err := syscall.Dup(int(f.Fd()), -1) |
| if err != nil { |
| return nil, os.NewSyscallError("dup", err) |
| } |
| return os.NewFile(uintptr(dfd), s), nil |
| } |
| |
| func setReadBuffer(fd *netFD, bytes int) error { |
| return syscall.EPLAN9 |
| } |
| |
| func setWriteBuffer(fd *netFD, bytes int) error { |
| return syscall.EPLAN9 |
| } |
| |
| func (fd *netFD) SetDeadline(t time.Time) error { |
| return fd.pfd.SetDeadline(t) |
| } |
| |
| func (fd *netFD) SetReadDeadline(t time.Time) error { |
| return fd.pfd.SetReadDeadline(t) |
| } |
| |
| func (fd *netFD) SetWriteDeadline(t time.Time) error { |
| return fd.pfd.SetWriteDeadline(t) |
| } |