Paillat-dev/Botator
1
0
Fork 0
mirror of https://github.com/Paillat-dev/Botator.git synced 2026-01-02 01:06:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

🐛 fix(functionscalls.py): add timeout parameter to evaluate_math function to limit execution time and prevent potential abuse or infinite loops

Browse Source
This commit is contained in:
Paillat
2023-08-15 18:49:51 +02:00
parent d057fc7a78
commit 30ef28ce49
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/functionscalls.py
View File

@@ -316,13 +316,14 @@ async def send_ascii_art_image(
async def evaluate_math( async def evaluate_math(
message_in_channel_in_wich_to_send: discord.Message, arguments: dict message_in_channel_in_wich_to_send: discord.Message, arguments: dict, timeout=10
): ):
evaluable = arguments.get("string", "") evaluable = arguments.get("string", "")
if evaluable == "": if evaluable == "":
raise FuntionCallError("No string provided") raise FuntionCallError("No string provided")
loop = asyncio.get_event_loop()
try: try:
result = simple_eval(evaluable) result = await asyncio.wait_for(loop.run_in_executor(None, simple_eval, evaluable), timeout=timeout)
except Exception as e: except Exception as e:
result = f"Error: {e}" result = f"Error: {e}"
return f"Result to math eval of {evaluable}: ```\n{str(result)}```" return f"Result to math eval of {evaluable}: ```\n{str(result)}```"