nicolargo/glances
1
1
Fork 0
mirror of https://github.com/nicolargo/glances.git synced 2024-12-22 16:51:35 +03:00
Code Issues Projects Releases Wiki Activity

Prevent data truncation when receiving data from hddtemp server

Browse Source
This commit is contained in:
MendelGusmao 2017-10-01 20:24:41 -03:00
parent b263e315ab
commit 4bc25e5314
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
glances/plugins/glances_hddtemp.py
View File

@ -140,7 +140,12 @@ class GlancesGrabHDDTemp(object):
try:
sck = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sck.connect((self.host, self.port))
data = sck.recv(4096)
data = b''
while True:
received = sck.recv(4096)
if not received:
break
data += received
except socket.error as e:
logger.debug("Cannot connect to an HDDtemp server ({}:{} => {})".format(self.host, self.port, e))
logger.debug("Disable the HDDtemp module. Use the --disable-hddtemp to hide the previous message.")
@ -149,6 +154,8 @@ class GlancesGrabHDDTemp(object):
data = ""
finally:
sck.close()
if data != "":
logger.debug("Received data from the HDDtemp server: {}".format(data))
return data