2008-09-26 15:11:26 -06: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.
|
|
|
|
|
2008-09-29 14:37:00 -06:00
|
|
|
// Waiting for FDs via kqueue/kevent.
|
2008-09-26 15:11:26 -06:00
|
|
|
|
|
|
|
package net
|
|
|
|
|
|
|
|
import (
|
2008-09-26 15:46:28 -06:00
|
|
|
"net";
|
2008-09-26 15:11:26 -06:00
|
|
|
"os";
|
|
|
|
"syscall";
|
|
|
|
)
|
|
|
|
|
2009-05-08 15:40:20 -06:00
|
|
|
var kqueuePhaseError = &Error{"kqueue phase error"}
|
|
|
|
|
2009-03-05 16:48:12 -07:00
|
|
|
type pollster struct {
|
2008-09-29 14:37:00 -06:00
|
|
|
kq int64;
|
2009-01-16 12:36:44 -07:00
|
|
|
eventbuf [10]syscall.Kevent_t;
|
|
|
|
events []syscall.Kevent_t;
|
2008-09-26 15:11:26 -06:00
|
|
|
}
|
|
|
|
|
2009-04-17 01:08:24 -06:00
|
|
|
func newpollster() (p *pollster, err os.Error) {
|
2009-03-05 16:48:12 -07:00
|
|
|
p = new(pollster);
|
2008-09-29 14:37:00 -06:00
|
|
|
var e int64;
|
2009-01-16 12:36:44 -07:00
|
|
|
if p.kq, e = syscall.Kqueue(); e != 0 {
|
2008-09-26 15:46:28 -06:00
|
|
|
return nil, os.ErrnoToError(e)
|
|
|
|
}
|
2008-12-18 23:37:22 -07:00
|
|
|
p.events = p.eventbuf[0:0];
|
2008-09-29 14:37:00 -06:00
|
|
|
return p, nil
|
2008-09-26 15:11:26 -06:00
|
|
|
}
|
|
|
|
|
2009-04-17 01:08:24 -06:00
|
|
|
func (p *pollster) AddFD(fd int64, mode int, repeat bool) os.Error {
|
2008-09-26 15:46:28 -06:00
|
|
|
var kmode int16;
|
|
|
|
if mode == 'r' {
|
|
|
|
kmode = syscall.EVFILT_READ
|
|
|
|
} else {
|
|
|
|
kmode = syscall.EVFILT_WRITE
|
|
|
|
}
|
2009-01-16 12:36:44 -07:00
|
|
|
var events [1]syscall.Kevent_t;
|
2008-09-26 15:46:28 -06:00
|
|
|
ev := &events[0];
|
2009-01-16 12:36:44 -07:00
|
|
|
ev.Ident = fd;
|
|
|
|
ev.Filter = kmode;
|
2008-09-26 15:46:28 -06:00
|
|
|
|
|
|
|
// EV_ADD - add event to kqueue list
|
2008-09-29 14:37:00 -06:00
|
|
|
// EV_RECEIPT - generate fake EV_ERROR as result of add,
|
|
|
|
// rather than waiting for real event
|
2008-09-26 15:46:28 -06:00
|
|
|
// EV_ONESHOT - delete the event the first time it triggers
|
2009-01-16 12:36:44 -07:00
|
|
|
ev.Flags = syscall.EV_ADD | syscall.EV_RECEIPT;
|
2008-09-26 15:46:28 -06:00
|
|
|
if !repeat {
|
2009-01-16 12:36:44 -07:00
|
|
|
ev.Flags |= syscall.EV_ONESHOT
|
2008-09-26 15:46:28 -06:00
|
|
|
}
|
|
|
|
|
2009-04-15 21:27:45 -06:00
|
|
|
n, e := syscall.Kevent(p.kq, &events, &events, nil);
|
2008-09-26 15:46:28 -06:00
|
|
|
if e != 0 {
|
|
|
|
return os.ErrnoToError(e)
|
|
|
|
}
|
2009-01-16 12:36:44 -07:00
|
|
|
if n != 1 || (ev.Flags & syscall.EV_ERROR) == 0 || ev.Ident != fd || ev.Filter != kmode {
|
2009-05-08 15:40:20 -06:00
|
|
|
return kqueuePhaseError
|
2008-09-26 15:46:28 -06:00
|
|
|
}
|
2009-01-16 12:36:44 -07:00
|
|
|
if ev.Data != 0 {
|
|
|
|
return os.ErrnoToError(ev.Data)
|
2008-09-26 15:46:28 -06:00
|
|
|
}
|
|
|
|
return nil
|
2008-09-26 15:11:26 -06:00
|
|
|
}
|
|
|
|
|
2009-03-06 18:51:31 -07:00
|
|
|
func (p *pollster) DelFD(fd int64, mode int) {
|
|
|
|
var kmode int16;
|
|
|
|
if mode == 'r' {
|
|
|
|
kmode = syscall.EVFILT_READ
|
|
|
|
} else {
|
|
|
|
kmode = syscall.EVFILT_WRITE
|
|
|
|
}
|
|
|
|
var events [1]syscall.Kevent_t;
|
|
|
|
ev := &events[0];
|
|
|
|
ev.Ident = fd;
|
|
|
|
ev.Filter = kmode;
|
|
|
|
|
|
|
|
// EV_DELETE - delete event from kqueue list
|
|
|
|
// EV_RECEIPT - generate fake EV_ERROR as result of add,
|
|
|
|
// rather than waiting for real event
|
|
|
|
ev.Flags = syscall.EV_DELETE | syscall.EV_RECEIPT;
|
2009-04-15 21:27:45 -06:00
|
|
|
syscall.Kevent(p.kq, &events, &events, nil);
|
2009-03-06 18:51:31 -07:00
|
|
|
}
|
|
|
|
|
2009-04-17 01:08:24 -06:00
|
|
|
func (p *pollster) WaitFD(nsec int64) (fd int64, mode int, err os.Error) {
|
2009-03-06 18:51:31 -07:00
|
|
|
var t *syscall.Timespec;
|
2008-09-29 14:37:00 -06:00
|
|
|
for len(p.events) == 0 {
|
2009-03-06 18:51:31 -07:00
|
|
|
if nsec > 0 {
|
|
|
|
if t == nil {
|
|
|
|
t = new(syscall.Timespec);
|
|
|
|
}
|
|
|
|
t.Sec = nsec / 1e9;
|
|
|
|
t.Nsec = uint64(nsec % 1e9);
|
|
|
|
}
|
2009-04-15 21:27:45 -06:00
|
|
|
nn, e := syscall.Kevent(p.kq, nil, &p.eventbuf, t);
|
2008-09-29 14:37:00 -06:00
|
|
|
if e != 0 {
|
2009-03-06 18:51:31 -07:00
|
|
|
if e == syscall.EINTR {
|
2008-09-29 14:37:00 -06:00
|
|
|
continue
|
|
|
|
}
|
|
|
|
return -1, 0, os.ErrnoToError(e)
|
|
|
|
}
|
2009-03-06 18:51:31 -07:00
|
|
|
if nn == 0 {
|
|
|
|
return -1, 0, nil;
|
|
|
|
}
|
2008-12-18 23:37:22 -07:00
|
|
|
p.events = p.eventbuf[0:nn]
|
2008-09-26 15:46:28 -06:00
|
|
|
}
|
2008-09-29 14:37:00 -06:00
|
|
|
ev := &p.events[0];
|
|
|
|
p.events = p.events[1:len(p.events)];
|
2009-01-16 12:36:44 -07:00
|
|
|
fd = ev.Ident;
|
|
|
|
if ev.Filter == syscall.EVFILT_READ {
|
2008-09-26 15:46:28 -06:00
|
|
|
mode = 'r'
|
|
|
|
} else {
|
|
|
|
mode = 'w'
|
|
|
|
}
|
2008-09-29 14:37:00 -06:00
|
|
|
return fd, mode, nil
|
2008-09-26 15:46:28 -06:00
|
|
|
}
|
|
|
|
|
2009-04-17 01:08:24 -06:00
|
|
|
func (p *pollster) Close() os.Error {
|
2009-01-16 12:36:44 -07:00
|
|
|
r, e := syscall.Close(p.kq);
|
2008-09-29 14:37:00 -06:00
|
|
|
return os.ErrnoToError(e)
|
2008-09-26 15:46:28 -06:00
|
|
|
}
|