netpoll_kqueue.go 2.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102
  1. // Copyright 2013 The Go Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style
  3. // license that can be found in the LICENSE file.
  4. // +build darwin dragonfly freebsd netbsd openbsd
  5. package runtime
  6. // Integrated network poller (kqueue-based implementation).
  7. import "unsafe"
  8. func kqueue() int32
  9. //go:noescape
  10. func kevent(kq int32, ch *keventt, nch int32, ev *keventt, nev int32, ts *timespec) int32
  11. func closeonexec(fd int32)
  12. var (
  13. kq int32 = -1
  14. netpolllasterr int32
  15. )
  16. func netpollinit() {
  17. kq = kqueue()
  18. if kq < 0 {
  19. println("netpollinit: kqueue failed with", -kq)
  20. gothrow("netpollinit: kqueue failed")
  21. }
  22. closeonexec(kq)
  23. }
  24. func netpollopen(fd uintptr, pd *pollDesc) int32 {
  25. // Arm both EVFILT_READ and EVFILT_WRITE in edge-triggered mode (EV_CLEAR)
  26. // for the whole fd lifetime. The notifications are automatically unregistered
  27. // when fd is closed.
  28. var ev [2]keventt
  29. *(*uintptr)(unsafe.Pointer(&ev[0].ident)) = fd
  30. ev[0].filter = _EVFILT_READ
  31. ev[0].flags = _EV_ADD | _EV_CLEAR
  32. ev[0].fflags = 0
  33. ev[0].data = 0
  34. ev[0].udata = (*byte)(unsafe.Pointer(pd))
  35. ev[1] = ev[0]
  36. ev[1].filter = _EVFILT_WRITE
  37. n := kevent(kq, &ev[0], 2, nil, 0, nil)
  38. if n < 0 {
  39. return -n
  40. }
  41. return 0
  42. }
  43. func netpollclose(fd uintptr) int32 {
  44. // Don't need to unregister because calling close()
  45. // on fd will remove any kevents that reference the descriptor.
  46. return 0
  47. }
  48. func netpollarm(pd *pollDesc, mode int) {
  49. gothrow("unused")
  50. }
  51. // Polls for ready network connections.
  52. // Returns list of goroutines that become runnable.
  53. func netpoll(block bool) (gp *g) {
  54. if kq == -1 {
  55. return
  56. }
  57. var tp *timespec
  58. var ts timespec
  59. if !block {
  60. tp = &ts
  61. }
  62. var events [64]keventt
  63. retry:
  64. n := kevent(kq, nil, 0, &events[0], int32(len(events)), tp)
  65. if n < 0 {
  66. if n != -_EINTR && n != netpolllasterr {
  67. netpolllasterr = n
  68. println("runtime: kevent on fd", kq, "failed with", -n)
  69. }
  70. goto retry
  71. }
  72. for i := 0; i < int(n); i++ {
  73. ev := &events[i]
  74. var mode int32
  75. if ev.filter == _EVFILT_READ {
  76. mode += 'r'
  77. }
  78. if ev.filter == _EVFILT_WRITE {
  79. mode += 'w'
  80. }
  81. if mode != 0 {
  82. netpollready((**g)(noescape(unsafe.Pointer(&gp))), (*pollDesc)(unsafe.Pointer(ev.udata)), mode)
  83. }
  84. }
  85. if block && gp == nil {
  86. goto retry
  87. }
  88. return gp
  89. }