blob: 18e927c2609656bc7b9f07aa196a09757aad1cdf (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
import os
import socket
import telnetlib
from leap.eip import exceptions as eip_exceptions
class UDSTelnet(telnetlib.Telnet):
"""
a telnet-alike class, that can listen
on unix domain sockets
"""
def open(self, host, port=23, timeout=socket._GLOBAL_DEFAULT_TIMEOUT):
"""Connect to a host. If port is 'unix', it
will open a connection over unix docmain sockets.
The optional second argument is the port number, which
defaults to the standard telnet port (23).
Don't try to reopen an already connected instance.
"""
self.eof = 0
self.host = host
self.port = port
self.timeout = timeout
if self.port == "unix":
# unix sockets spoken
if not os.path.exists(self.host):
raise eip_exceptions.MissingSocketError
self.sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
try:
self.sock.connect(self.host)
except socket.error:
raise eip_exceptions.ConnectionRefusedError
else:
self.sock = socket.create_connection((host, port), timeout)
|
