Kaydet (Commit) 322c0d18 authored tarafından Martin v. Löwis's avatar Martin v. Löwis

Only close sockets if they have been created. Reported by Blake Winton.

üst fb163784
...@@ -122,7 +122,8 @@ class FTP: ...@@ -122,7 +122,8 @@ class FTP:
self.sock = socket.socket(af, socktype, proto) self.sock = socket.socket(af, socktype, proto)
self.sock.connect(sa) self.sock.connect(sa)
except socket.error, msg: except socket.error, msg:
self.sock.close() if self.sock:
self.sock.close()
self.sock = None self.sock = None
continue continue
break break
...@@ -272,13 +273,15 @@ class FTP: ...@@ -272,13 +273,15 @@ class FTP:
def makeport(self): def makeport(self):
'''Create a new socket and send a PORT command for it.''' '''Create a new socket and send a PORT command for it.'''
msg = "getaddrinfo returns an empty list" msg = "getaddrinfo returns an empty list"
sock = None
for res in socket.getaddrinfo(None, 0, self.af, socket.SOCK_STREAM, 0, socket.AI_PASSIVE): for res in socket.getaddrinfo(None, 0, self.af, socket.SOCK_STREAM, 0, socket.AI_PASSIVE):
af, socktype, proto, canonname, sa = res af, socktype, proto, canonname, sa = res
try: try:
sock = socket.socket(af, socktype, proto) sock = socket.socket(af, socktype, proto)
sock.bind(sa) sock.bind(sa)
except socket.error, msg: except socket.error, msg:
sock.close() if sock:
sock.close()
sock = None sock = None
continue continue
break break
......
...@@ -368,7 +368,8 @@ class HTTPConnection: ...@@ -368,7 +368,8 @@ class HTTPConnection:
except socket.error, msg: except socket.error, msg:
if self.debuglevel > 0: if self.debuglevel > 0:
print 'connect fail:', (self.host, self.port) print 'connect fail:', (self.host, self.port)
self.sock.close() if self.sock:
self.sock.close()
self.sock = None self.sock = None
continue continue
break break
......
...@@ -76,13 +76,15 @@ class POP3: ...@@ -76,13 +76,15 @@ class POP3:
self.host = host self.host = host
self.port = port self.port = port
msg = "getaddrinfo returns an empty list" msg = "getaddrinfo returns an empty list"
self.sock = None
for res in socket.getaddrinfo(self.host, self.port, 0, socket.SOCK_STREAM): for res in socket.getaddrinfo(self.host, self.port, 0, socket.SOCK_STREAM):
af, socktype, proto, canonname, sa = res af, socktype, proto, canonname, sa = res
try: try:
self.sock = socket.socket(af, socktype, proto) self.sock = socket.socket(af, socktype, proto)
self.sock.connect(sa) self.sock.connect(sa)
except socket.error, msg: except socket.error, msg:
self.sock.close() if self.sock:
self.sock.close()
self.sock = None self.sock = None
continue continue
break break
......
...@@ -265,6 +265,7 @@ class SMTP: ...@@ -265,6 +265,7 @@ class SMTP:
if not port: port = SMTP_PORT if not port: port = SMTP_PORT
if self.debuglevel > 0: print 'connect:', (host, port) if self.debuglevel > 0: print 'connect:', (host, port)
msg = "getaddrinfo returns an empty list" msg = "getaddrinfo returns an empty list"
self.sock = None
for res in socket.getaddrinfo(host, port, 0, socket.SOCK_STREAM): for res in socket.getaddrinfo(host, port, 0, socket.SOCK_STREAM):
af, socktype, proto, canonname, sa = res af, socktype, proto, canonname, sa = res
try: try:
...@@ -273,7 +274,8 @@ class SMTP: ...@@ -273,7 +274,8 @@ class SMTP:
self.sock.connect(sa) self.sock.connect(sa)
except socket.error, msg: except socket.error, msg:
if self.debuglevel > 0: print 'connect fail:', (host, port) if self.debuglevel > 0: print 'connect fail:', (host, port)
self.sock.close() if self.sock:
self.sock.close()
self.sock = None self.sock = None
continue continue
break break
......
...@@ -210,7 +210,8 @@ class Telnet: ...@@ -210,7 +210,8 @@ class Telnet:
self.sock = socket.socket(af, socktype, proto) self.sock = socket.socket(af, socktype, proto)
self.sock.connect(sa) self.sock.connect(sa)
except socket.error, msg: except socket.error, msg:
self.sock.close() if self.sock:
self.sock.close()
self.sock = None self.sock = None
continue continue
break break
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment