Skip to content

Instantly share code, notes, and snippets.

@orklann
Forked from ishworgurung/kevent_srv.py
Created April 27, 2017 05:04
Show Gist options
  • Save orklann/aae8e153a92836faa2fa6ac42b06ff9b to your computer and use it in GitHub Desktop.
Save orklann/aae8e153a92836faa2fa6ac42b06ff9b to your computer and use it in GitHub Desktop.
So, how can I use kqueue/kevent on BSD/Mac using Python?
#!/usr/bin/env python
"""
Example on using Kqueue/Kevent on BSD/Mac
using Python.
The TCP server essentially echoes back the
message it receives on the client socket.
"""
__author__ = "Ishwor Gurung <ishwor@develworx.com>"
__license__ = "3 clause BSD"
from socket import socket, \
AF_INET, SOCK_STREAM,\
SOL_SOCKET, SO_REUSEADDR,\
SHUT_WR
import sys
import select
BUFSIZE = 512
def handle_connection(cl_socket):
"""
Handle each client socket. Receive data on it and send
the data back to the client. If there are no more data
available on the read side, shutdown and close the
socket.
"""
while True:
m = cl_socket.recv(BUFSIZE)
if m and len(m)>0:
cl_socket.send(m)
else:
cl_socket.shutdown(SHUT_WR)
cl_socket.close()
break
def main():
s = socket(AF_INET, SOCK_STREAM)
s.setsockopt(SOL_SOCKET, SO_REUSEADDR, 1)
s.bind(("127.0.0.1",10000))
s.listen(10)
kq = select.kqueue()
# Initialise the master fd(s.fileno()) from server socket
kevent = select.kevent(s.fileno(),
filter=select.KQ_FILTER_READ, # we are interested in reads
flags=select.KQ_EV_ADD | select.KQ_EV_ENABLE)
while True:
revents = kq.control([kevent], 1, None)
for event in revents:
# If the kernel notifies us saying there is a read event available
# on the master fd(s.fileno()), we accept() the
# connection so that we can recv()/send() on the the accept()ed
# socket
if (event.filter == select.KQ_FILTER_READ):
cl,_ = s.accept()
handle_connection(cl)
if __name__=="__main__":
sys.exit(main())
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment