LBRYCommunity/aioupnp
1
0
Fork 0
Code Issues 7 Pull requests 1 Projects Releases 1 Packages Wiki Activity

fix Host header in HTTP GET

Browse source
This commit is contained in:
Jack Robison 2018-10-04 18:35:17 -04:00
parent 0b0a82785f
commit 6a0d71e891
No known key found for this signature in database
GPG key ID: DF25C68FE0239BB2
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
txupnp/scpd.py
View file

@ -57,6 +57,7 @@ def parse_service_description(content: bytes):
class SCPDHTTPClientProtocol(Protocol): class SCPDHTTPClientProtocol(Protocol):
def connectionMade(self): def connectionMade(self):
self.response_buff = b"" self.response_buff = b""
log.debug("Sending HTTP:\n%s", self.factory.packet.decode())
self.factory.reactor.callLater(0, self.transport.write, self.factory.packet) self.factory.reactor.callLater(0, self.transport.write, self.factory.packet)
def dataReceived(self, data): def dataReceived(self, data):
@ -64,6 +65,7 @@ class SCPDHTTPClientProtocol(Protocol):
def connectionLost(self, reason): def connectionLost(self, reason):
if reason.trap(error.ConnectionDone): if reason.trap(error.ConnectionDone):
log.debug("Received HTTP:\n%s", self.response_buff.decode())
if XML_VERSION.encode() in self.response_buff: if XML_VERSION.encode() in self.response_buff:
parsed = CONTENT_PATTERN.findall(self.response_buff) parsed = CONTENT_PATTERN.findall(self.response_buff)
result = b'' if not parsed else parsed[0][0] result = b'' if not parsed else parsed[0][0]
@ -127,7 +129,7 @@ class SCPDHTTPClientFactory(ClientFactory):
'Accept-Encoding: gzip\r\n' 'Accept-Encoding: gzip\r\n'
'Host: %s\r\n' 'Host: %s\r\n'
'\r\n' '\r\n'
) % (control_url, address) ) % (control_url, address.split("http://")[1])
).encode() ).encode()
return cls(reactor, data) return cls(reactor, data)