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
|
# Copyright (c) Twisted Matrix Laboratories.
# See LICENSE for details.
"""
Client-side of an example for sending file descriptors between processes over
UNIX sockets. This client connects to a server listening on a UNIX socket and
waits for one file descriptor to arrive over the connection. It displays the
name of the file and the first 80 bytes it contains, then exits.
To run this example, run this program with one argument: a path giving the UNIX
socket the server side of this example is already listening on. For example:
$ python recvfd.py /tmp/sendfd.sock
See sendfd.py for the server side of this example.
"""
if __name__ == "__main__":
import recvfd
raise SystemExit(recvfd.main())
import os
import sys
from zope.interface import implementer
from twisted.internet import reactor
from twisted.internet.defer import Deferred
from twisted.internet.endpoints import UNIXClientEndpoint
from twisted.internet.interfaces import IFileDescriptorReceiver
from twisted.internet.protocol import Factory
from twisted.protocols.basic import LineOnlyReceiver
from twisted.python.filepath import FilePath
from twisted.python.log import startLogging
@implementer(IFileDescriptorReceiver)
class ReceiveFDProtocol(LineOnlyReceiver):
descriptor = None
def __init__(self):
self.whenDisconnected = Deferred()
def fileDescriptorReceived(self, descriptor):
# Record the descriptor sent to us
self.descriptor = descriptor
def lineReceived(self, line):
if self.descriptor is None:
print(f"Received {line} without receiving descriptor!")
else:
# Use the previously received descriptor, along with the newly
# provided information about which file it is, to present some
# information to the user.
data = os.read(self.descriptor, 80)
print(f"Received {line} from the server.")
print(f"First 80 bytes are:\n{data}\n")
os.close(self.descriptor)
self.transport.loseConnection()
def connectionLost(self, reason):
self.whenDisconnected.callback(None)
def main():
address = FilePath(sys.argv[1])
startLogging(sys.stdout)
factory = Factory()
factory.protocol = ReceiveFDProtocol
factory.quiet = True
endpoint = UNIXClientEndpoint(reactor, address.path)
connected = endpoint.connect(factory)
def succeeded(client):
return client.whenDisconnected
def failed(reason):
print("Could not connect:", reason.getErrorMessage())
def disconnected(ignored):
reactor.stop()
connected.addCallbacks(succeeded, failed)
connected.addCallback(disconnected)
reactor.run()
|
