def process(self, state):
"""Connect to a server"""
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.settimeout(5)
sock.connect((self.hostname, self.port))
# allow for later buffering of writes to the socket
sock = BufferedSocket(sock)
defragmenter = Defragmenter()
defragmenter.addStaticSize(ContentType.alert, 2)
defragmenter.addStaticSize(ContentType.change_cipher_spec, 1)
defragmenter.addDynamicSize(ContentType.handshake, 1, 3)
state.msg_sock = MessageSocket(sock, defragmenter)
state.msg_sock.version = self.version
def process(self, state):
"""Connect to a server."""
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.settimeout(5)
sock.connect((self.hostname, self.port))
# disable Nagle - we handle buffering and flushing ourselves
sock.setsockopt(socket.IPPROTO_TCP, socket.TCP_NODELAY, 1)
# allow for later buffering of writes to the socket
sock = BufferedSocket(sock)
defragmenter = Defragmenter()
defragmenter.add_static_size(ContentType.alert, 2)
defragmenter.add_static_size(ContentType.change_cipher_spec, 1)
defragmenter.add_dynamic_size(ContentType.handshake, 1, 3)
state.msg_sock = MessageSocket(sock, defragmenter)
state.msg_sock.version = self.version
def setUp(self):
self.raw_sock = mock.MagicMock()
self.sock = BufferedSocket(self.raw_sock)
