• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• vanilla-master
  • .coveragerc
  • examples
    • ticktock.py
    • tcp-basic.py
    • tcp-echo.py
  • xx
    • react.py
    • pipe.py
    • router.py
    • notes.md
    • pipe2.py
    • stream.py
  • LICENSE
  • TODO.md
  • README.rst
  • setup.py
  • CHANGELOG.rst
  • .travis.yml
  • tests
    • integration
      • test_http.py
    • unit
      • test_poll.py
      • test_thread.py
      • test_process.py
      • test_udp.py
      • test_io.py
      • test_message.py
      • test_http.py
      • test_signal.py
      • test_tcp.py
      • test_core.py
  • requirements.txt
  • bench
    • heap.py
  • .gitignore
  • vanilla
    • signal.py
    • udp.py
    • http.py
    • tcp.py
    • poll.py
    • message.py
    • io.py
    • meta.py
    • __init__.py
    • exception.py
    • core.py
    • process.py
    • thread.py
  • docs
    • Makefile
    • api.rst
    • requirements.txt
    • index.rst
    • conf.py
    • _static
  • MANIFEST.in
  • tox.ini

import socket
import errno

import vanilla.exception
import vanilla.poll


class __plugin__(object):
    def __init__(self, hub):
        self.hub = hub

    def listen(self, port=0, host='127.0.0.1'):
        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        sock.bind((host, port))
        sock.listen(socket.SOMAXCONN)
        sock.setblocking(0)
        port = sock.getsockname()[1]
        server = self.hub.register(sock.fileno(), vanilla.poll.POLLIN)

        @server.pipe
        def server(upstream, downstream):
            for mask in upstream:
                while True:
                    try:
                        conn, host = sock.accept()
                        downstream.send(self.hub.io.socket(conn))
                    except (socket.error, OSError), e:
                        if e.errno == errno.EAGAIN:
                            break
                        raise
            self.hub.unregister(sock.fileno())
            sock.close()

        server.port = port
        return server

    def connect(self, port, host='127.0.0.1'):
        conn = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        # TODO: this shouldn't block on the connect
        conn.connect((host, port))
        return self.hub.io.socket(conn)