1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495 |
- // Copyright 2013 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 (
- "syscall"
- "time"
- )
- type pollDesc struct {
- fd *netFD
- closing bool
- }
- func (pd *pollDesc) Init(fd *netFD) error { pd.fd = fd; return nil }
- func (pd *pollDesc) Close() {}
- func (pd *pollDesc) Lock() {}
- func (pd *pollDesc) Unlock() {}
- func (pd *pollDesc) Wakeup() {}
- func (pd *pollDesc) Evict() bool {
- pd.closing = true
- if pd.fd != nil {
- syscall.StopIO(pd.fd.sysfd)
- }
- return false
- }
- func (pd *pollDesc) Prepare(mode int) error {
- if pd.closing {
- return errClosing
- }
- return nil
- }
- func (pd *pollDesc) PrepareRead() error { return pd.Prepare('r') }
- func (pd *pollDesc) PrepareWrite() error { return pd.Prepare('w') }
- func (pd *pollDesc) Wait(mode int) error {
- if pd.closing {
- return errClosing
- }
- return errTimeout
- }
- func (pd *pollDesc) WaitRead() error { return pd.Wait('r') }
- func (pd *pollDesc) WaitWrite() error { return pd.Wait('w') }
- func (pd *pollDesc) WaitCanceled(mode int) {}
- func (pd *pollDesc) WaitCanceledRead() {}
- func (pd *pollDesc) WaitCanceledWrite() {}
- func (fd *netFD) setDeadline(t time.Time) error {
- return setDeadlineImpl(fd, t, 'r'+'w')
- }
- func (fd *netFD) setReadDeadline(t time.Time) error {
- return setDeadlineImpl(fd, t, 'r')
- }
- func (fd *netFD) setWriteDeadline(t time.Time) error {
- return setDeadlineImpl(fd, t, 'w')
- }
- func setDeadlineImpl(fd *netFD, t time.Time, mode int) error {
- d := t.UnixNano()
- if t.IsZero() {
- d = 0
- }
- if err := fd.incref(); err != nil {
- return err
- }
- switch mode {
- case 'r':
- syscall.SetReadDeadline(fd.sysfd, d)
- case 'w':
- syscall.SetWriteDeadline(fd.sysfd, d)
- case 'r' + 'w':
- syscall.SetReadDeadline(fd.sysfd, d)
- syscall.SetWriteDeadline(fd.sysfd, d)
- }
- fd.decref()
- return nil
- }
|