Firepup650/bitbot-3.11-fork
2
0
Fork 0
Code Issues Pull requests Activity Actions

add utils.deadline(func, seconds) to put hard limits on code exec time

Browse source
This commit is contained in:
jesopo 2019-08-30 17:17:03 +01:00
parent 4bfb4c3200
commit e97e3c8f30
1 changed files with 15 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
src/utils/__init__.py
View file

@ -1,4 +1,4 @@
import datetime, decimal, enum, io, ipaddress, re, threading, typing import datetime, decimal, enum, io, ipaddress, re, signal, threading, typing
from src.utils import cli, consts, irc, http, parse, security from src.utils import cli, consts, irc, http, parse, security
class Direction(enum.Enum): class Direction(enum.Enum):
@ -342,3 +342,17 @@ class OptionsSetting(Setting):
options = self._get_options() options = self._get_options()
options_str = ["'%s'" % option for option in options] options_str = ["'%s'" % option for option in options]
return "Options: %s" % ", ".join(options_str) return "Options: %s" % ", ".join(options_str)
class DeadlineExceededException(Exception):
pass
def _raise_deadline():
raise DeadlineExceededException()
def deadline(f: typing.Callable[[], typing.Any], seconds: int=10) -> typing.Any:
signal.signal(signal.SIGALRM, lambda _1, _2: _raise_deadline())
signal.alarm(seconds)
try:
return f()
finally:
signal.signal(signal.SIGALRM, signal.SIG_IGN)