beenull/simple-login
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

hard exit when memory is more than 300MB

Browse source
This commit is contained in:
Son NK 2020-06-08 13:53:27 +02:00
parent 7a3a6784cc
commit e988573cb4
1 changed files with 13 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
app/pgp_utils.py
View file

@ -1,3 +1,4 @@
import os
from io import BytesIO
import gnupg
@ -24,11 +25,22 @@ def load_public_key(public_key: str) -> str:
raise PGPException("Cannot load key") from e
def hard_exit():
pid = os.getpid()
LOG.warning("kill pid %s", pid)
os.kill(pid, 9)
def encrypt_file(data: BytesIO, fingerprint: str) -> str:
LOG.d("encrypt for %s", fingerprint)
mem_usage = memory_usage(-1, interval=1, timeout=1)
mem_usage = memory_usage(-1, interval=1, timeout=1)[0]
LOG.d("mem_usage %s", mem_usage)
# todo
if mem_usage > 300:
LOG.error("Force exit")
hard_exit()
r = gpg.encrypt_file(data, fingerprint, always_trust=True)
if not r.ok:
LOG.error("Try encrypt again %s", fingerprint)