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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
# tio.py - I/O functions
#
# Copyright (C) 2010-2019 Arthur de Jong
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 2.1 of the License, or (at your option) any later version.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this library; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
# 02110-1301 USA
import os
import socket
import struct
import sys
# definition for reading and writing INT32 values
_int32 = struct.Struct('!i')
# FIXME: use something from constants.py to determine the correct size
_struct_timeval = struct.Struct('ll')
class TIOStreamError(Exception):
pass
class TIOStream(object):
"""File-like object for reading and writing nslcd-protocol entities."""
def __init__(self, conn):
conn.setblocking(1)
conn.setsockopt(socket.SOL_SOCKET, socket.SO_RCVTIMEO, _struct_timeval.pack(0, 500000))
conn.setsockopt(socket.SOL_SOCKET, socket.SO_SNDTIMEO, _struct_timeval.pack(60, 0))
self.readfd = os.fdopen(conn.fileno(), 'rb', 1024)
self.writefd = os.fdopen(conn.fileno(), 'wb', 1024 * 1024)
def read(self, size):
return self.readfd.read(size)
def read_int32(self):
return _int32.unpack(self.read(_int32.size))[0]
def read_bytes(self, maxsize=None):
num = self.read_int32()
if maxsize and num >= maxsize:
raise TIOStreamError()
return self.read(num)
def read_string(self):
value = self.read_bytes()
if sys.version_info[0] >= 3:
value = value.decode('utf-8')
return value
def read_address(self):
"""Read an address (usually IPv4 or IPv6) from the stream.
This returns the address as a string representation.
"""
af = self.read_int32()
return socket.inet_ntop(af, self.read_bytes(maxsize=64))
def write(self, value):
self.writefd.write(value)
def write_int32(self, value):
self.write(_int32.pack(value))
def write_bytes(self, value):
self.write_int32(len(value))
if value:
self.write(value)
def write_string(self, value):
if sys.version_info[0] >= 3:
value = value.encode('utf-8')
self.write_bytes(value)
def write_stringlist(self, value):
lst = tuple(value)
self.write_int32(len(lst))
for string in lst:
self.write_string(string)
@staticmethod
def _to_address(value):
# try IPv4 first
try:
return socket.AF_INET, socket.inet_pton(socket.AF_INET, value)
except socket.error:
pass # try the next one
# fall back to IPv6
return socket.AF_INET6, socket.inet_pton(socket.AF_INET6, value)
def write_address(self, value):
"""Write an address (usually IPv4 or IPv6) to the stream."""
# first try to make it into an IPv6 address
af, address = TIOStream._to_address(value)
self.write_int32(af)
self.write_bytes(address)
def close(self):
try:
self.writefd.close()
except IOError:
pass
try:
self.readfd.close()
except IOError:
pass
def __del__(self):
self.close()
|