109 lines
2.5 KiB
C
109 lines
2.5 KiB
C
// 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.
|
|
|
|
// +build darwin
|
|
|
|
#include "runtime.h"
|
|
#include "defs_GOOS_GOARCH.h"
|
|
|
|
// Integrated network poller (kqueue-based implementation).
|
|
|
|
int32 runtime_kqueue(void);
|
|
int32 runtime_kevent(int32, Kevent*, int32, Kevent*, int32, Timespec*);
|
|
void runtime_closeonexec(int32);
|
|
|
|
static int32 kq = -1;
|
|
|
|
void
|
|
runtime_netpollinit(void)
|
|
{
|
|
kq = runtime_kqueue();
|
|
if(kq < 0) {
|
|
runtime_printf("netpollinit: kqueue failed with %d\n", -kq);
|
|
runtime_throw("netpollinit: kqueue failed");
|
|
}
|
|
runtime_closeonexec(kq);
|
|
}
|
|
|
|
int32
|
|
runtime_netpollopen(int32 fd, PollDesc *pd)
|
|
{
|
|
Kevent ev[2];
|
|
int32 n;
|
|
|
|
// Arm both EVFILT_READ and EVFILT_WRITE in edge-triggered mode (EV_CLEAR)
|
|
// for the whole fd lifetime. The notifications are automatically unregistered
|
|
// when fd is closed.
|
|
ev[0].ident = fd;
|
|
ev[0].filter = EVFILT_READ;
|
|
ev[0].flags = EV_ADD|EV_RECEIPT|EV_CLEAR;
|
|
ev[0].fflags = 0;
|
|
ev[0].data = 0;
|
|
ev[0].udata = (byte*)pd;
|
|
ev[1] = ev[0];
|
|
ev[1].filter = EVFILT_WRITE;
|
|
n = runtime_kevent(kq, ev, 2, ev, 2, nil);
|
|
if(n < 0)
|
|
return -n;
|
|
if(n != 2 ||
|
|
(ev[0].flags&EV_ERROR) == 0 || ev[0].ident != fd || ev[0].filter != EVFILT_READ ||
|
|
(ev[1].flags&EV_ERROR) == 0 || ev[1].ident != fd || ev[1].filter != EVFILT_WRITE)
|
|
return EFAULT; // just to mark out from other errors
|
|
if(ev[0].data != 0)
|
|
return ev[0].data;
|
|
if(ev[1].data != 0)
|
|
return ev[1].data;
|
|
return 0;
|
|
}
|
|
|
|
int32
|
|
runtime_netpollclose(int32 fd)
|
|
{
|
|
// Don't need to unregister because calling close()
|
|
// on fd will remove any kevents that reference the descriptor.
|
|
USED(fd);
|
|
return 0;
|
|
}
|
|
|
|
// Polls for ready network connections.
|
|
// Returns list of goroutines that become runnable.
|
|
G*
|
|
runtime_netpoll(bool block)
|
|
{
|
|
static int32 lasterr;
|
|
Kevent events[64], *ev;
|
|
Timespec ts, *tp;
|
|
int32 n, i;
|
|
G *gp;
|
|
|
|
if(kq == -1)
|
|
return nil;
|
|
tp = nil;
|
|
if(!block) {
|
|
ts.tv_sec = 0;
|
|
ts.tv_nsec = 0;
|
|
tp = &ts;
|
|
}
|
|
gp = nil;
|
|
retry:
|
|
n = runtime_kevent(kq, nil, 0, events, nelem(events), tp);
|
|
if(n < 0) {
|
|
if(n != -EINTR && n != lasterr) {
|
|
lasterr = n;
|
|
runtime_printf("runtime: kevent on fd %d failed with %d\n", kq, -n);
|
|
}
|
|
goto retry;
|
|
}
|
|
for(i = 0; i < n; i++) {
|
|
ev = &events[i];
|
|
if(ev->filter == EVFILT_READ)
|
|
runtime_netpollready(&gp, (PollDesc*)ev->udata, 'r');
|
|
if(ev->filter == EVFILT_WRITE)
|
|
runtime_netpollready(&gp, (PollDesc*)ev->udata, 'w');
|
|
}
|
|
if(block && gp == nil)
|
|
goto retry;
|
|
return gp;
|
|
}
|