152 lines
7.8 KiB
Python
152 lines
7.8 KiB
Python
import threading
|
|
import shared
|
|
import time
|
|
import sys
|
|
import os
|
|
import pickle
|
|
|
|
import tr#anslate
|
|
from helper_sql import *
|
|
from helper_threading import *
|
|
from debug import logger
|
|
|
|
"""
|
|
The singleCleaner class is a timer-driven thread that cleans data structures
|
|
to free memory, resends messages when a remote node doesn't respond, and
|
|
sends pong messages to keep connections alive if the network isn't busy.
|
|
It cleans these data structures in memory:
|
|
inventory (moves data to the on-disk sql database)
|
|
inventorySets (clears then reloads data out of sql database)
|
|
|
|
It cleans these tables on the disk:
|
|
inventory (clears expired objects)
|
|
pubkeys (clears pubkeys older than 4 weeks old which we have not used personally)
|
|
|
|
It resends messages when there has been no response:
|
|
resends getpubkey messages in 5 days (then 10 days, then 20 days, etc...)
|
|
resends msg messages in 5 days (then 10 days, then 20 days, etc...)
|
|
|
|
"""
|
|
|
|
|
|
class singleCleaner(threading.Thread, StoppableThread):
|
|
|
|
def __init__(self):
|
|
threading.Thread.__init__(self, name="singleCleaner")
|
|
self.initStop()
|
|
|
|
def run(self):
|
|
timeWeLastClearedInventoryAndPubkeysTables = 0
|
|
try:
|
|
shared.maximumLengthOfTimeToBotherResendingMessages = (float(shared.config.get('bitmessagesettings', 'stopresendingafterxdays')) * 24 * 60 * 60) + (float(shared.config.get('bitmessagesettings', 'stopresendingafterxmonths')) * (60 * 60 * 24 *365)/12)
|
|
except:
|
|
# Either the user hasn't set stopresendingafterxdays and stopresendingafterxmonths yet or the options are missing from the config file.
|
|
shared.maximumLengthOfTimeToBotherResendingMessages = float('inf')
|
|
|
|
while shared.shutdown == 0:
|
|
shared.UISignalQueue.put((
|
|
'updateStatusBar', 'Doing housekeeping (Flushing inventory in memory to disk...)'))
|
|
with shared.inventoryLock: # If you use both the inventoryLock and the sqlLock, always use the inventoryLock OUTSIDE of the sqlLock.
|
|
with SqlBulkExecute() as sql:
|
|
for hash, storedValue in shared.inventory.items():
|
|
objectType, streamNumber, payload, expiresTime, tag = storedValue
|
|
sql.execute(
|
|
'''INSERT INTO inventory VALUES (?,?,?,?,?,?)''',
|
|
hash,
|
|
objectType,
|
|
streamNumber,
|
|
payload,
|
|
expiresTime,
|
|
tag)
|
|
del shared.inventory[hash]
|
|
shared.UISignalQueue.put(('updateStatusBar', ''))
|
|
|
|
shared.broadcastToSendDataQueues((
|
|
0, 'pong', 'no data')) # commands the sendData threads to send out a pong message if they haven't sent anything else in the last five minutes. The socket timeout-time is 10 minutes.
|
|
# If we are running as a daemon then we are going to fill up the UI
|
|
# queue which will never be handled by a UI. We should clear it to
|
|
# save memory.
|
|
if shared.safeConfigGetBoolean('bitmessagesettings', 'daemon'):
|
|
shared.UISignalQueue.queue.clear()
|
|
if timeWeLastClearedInventoryAndPubkeysTables < int(time.time()) - 7380:
|
|
timeWeLastClearedInventoryAndPubkeysTables = int(time.time())
|
|
sqlExecute(
|
|
'''DELETE FROM inventory WHERE expirestime<? ''',
|
|
int(time.time()) - (60 * 60 * 3))
|
|
# pubkeys
|
|
sqlExecute(
|
|
'''DELETE FROM pubkeys WHERE time<? AND usedpersonally='no' ''',
|
|
int(time.time()) - shared.lengthOfTimeToHoldOnToAllPubkeys)
|
|
|
|
# Let us resend getpubkey objects if we have not yet heard a pubkey, and also msg objects if we have not yet heard an acknowledgement
|
|
queryreturn = sqlQuery(
|
|
'''select toaddress, ackdata, status FROM sent WHERE ((status='awaitingpubkey' OR status='msgsent') AND folder='sent' AND sleeptill<? AND senttime>?) ''',
|
|
int(time.time()),
|
|
int(time.time()) - shared.maximumLengthOfTimeToBotherResendingMessages)
|
|
for row in queryreturn:
|
|
if len(row) < 2:
|
|
logger.error('Something went wrong in the singleCleaner thread: a query did not return the requested fields. ' + repr(row))
|
|
self.stop.wait(3)
|
|
break
|
|
toAddress, ackData, status = row
|
|
if status == 'awaitingpubkey':
|
|
resendPubkeyRequest(toAddress)
|
|
elif status == 'msgsent':
|
|
resendMsg(ackData)
|
|
|
|
# Let's also clear and reload shared.inventorySets to keep it from
|
|
# taking up an unnecessary amount of memory.
|
|
for streamNumber in shared.inventorySets:
|
|
shared.inventorySets[streamNumber] = set()
|
|
queryData = sqlQuery('''SELECT hash FROM inventory WHERE streamnumber=?''', streamNumber)
|
|
for row in queryData:
|
|
shared.inventorySets[streamNumber].add(row[0])
|
|
with shared.inventoryLock:
|
|
for hash, storedValue in shared.inventory.items():
|
|
objectType, streamNumber, payload, expiresTime, tag = storedValue
|
|
if not streamNumber in shared.inventorySets:
|
|
shared.inventorySets[streamNumber] = set()
|
|
shared.inventorySets[streamNumber].add(hash)
|
|
|
|
# Let us write out the knowNodes to disk if there is anything new to write out.
|
|
if shared.needToWriteKnownNodesToDisk:
|
|
shared.knownNodesLock.acquire()
|
|
output = open(shared.appdata + 'knownnodes.dat', 'wb')
|
|
try:
|
|
pickle.dump(shared.knownNodes, output)
|
|
output.close()
|
|
except Exception as err:
|
|
if "Errno 28" in str(err):
|
|
logger.fatal('(while receiveDataThread shared.needToWriteKnownNodesToDisk) Alert: Your disk or data storage volume is full. ')
|
|
shared.UISignalQueue.put(('alert', (tr.translateText("MainWindow", "Disk full"), tr.translateText("MainWindow", 'Alert: Your disk or data storage volume is full. Bitmessage will now exit.'), True)))
|
|
if shared.daemon:
|
|
os._exit(0)
|
|
shared.knownNodesLock.release()
|
|
shared.needToWriteKnownNodesToDisk = False
|
|
self.stop.wait(300)
|
|
|
|
|
|
def resendPubkeyRequest(address):
|
|
logger.debug('It has been a long time and we haven\'t heard a response to our getpubkey request. Sending again.')
|
|
try:
|
|
del shared.neededPubkeys[
|
|
address] # We need to take this entry out of the shared.neededPubkeys structure because the shared.workerQueue checks to see whether the entry is already present and will not do the POW and send the message because it assumes that it has already done it recently.
|
|
except:
|
|
pass
|
|
|
|
shared.UISignalQueue.put((
|
|
'updateStatusBar', 'Doing work necessary to again attempt to request a public key...'))
|
|
sqlExecute(
|
|
'''UPDATE sent SET status='msgqueued' WHERE toaddress=?''',
|
|
address)
|
|
shared.workerQueue.put(('sendmessage', ''))
|
|
|
|
def resendMsg(ackdata):
|
|
logger.debug('It has been a long time and we haven\'t heard an acknowledgement to our msg. Sending again.')
|
|
sqlExecute(
|
|
'''UPDATE sent SET status='msgqueued' WHERE ackdata=?''',
|
|
ackdata)
|
|
shared.workerQueue.put(('sendmessage', ''))
|
|
shared.UISignalQueue.put((
|
|
'updateStatusBar', 'Doing work necessary to again attempt to deliver a message...'))
|