ExternalVendorCode/onefuzz
1
0
Fork 0
mirror of https://github.com/microsoft/onefuzz.git synced 2025-06-16 03:48:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

set max stdout/stderr size (#460)

Browse Source
This commit is contained in:
bmc-msft
2021-01-25 13:07:35 -05:00
committed by GitHub
parent 3ae70cea6b
commit 4bc90a7564
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/api-service/__app__/onefuzzlib/agent_events.py
View File

@ -29,6 +29,8 @@ from ..onefuzzlib.pools import Node, NodeTasks
from ..onefuzzlib.task_event import TaskEvent from ..onefuzzlib.task_event import TaskEvent
from ..onefuzzlib.tasks.main import Task from ..onefuzzlib.tasks.main import Task
MAX_OUTPUT_SIZE = 4096
def get_node(machine_id: UUID) -> Result[Node]: def get_node(machine_id: UUID) -> Result[Node]:
node = Node.get_by_machine_id(machine_id) node = Node.get_by_machine_id(machine_id)
@ -220,8 +222,8 @@ def on_worker_event_done(machine_id: UUID, event: WorkerDoneEvent) -> Result[Non
code=ErrorCode.TASK_FAILED, code=ErrorCode.TASK_FAILED,
errors=[ errors=[
"task failed. exit_status:%s" % event.exit_status, "task failed. exit_status:%s" % event.exit_status,
event.stdout[-4096:], event.stdout[-MAX_OUTPUT_SIZE:],
event.stderr[-4096:], event.stderr[-MAX_OUTPUT_SIZE:],
], ],
) )
) )
@ -233,6 +235,8 @@ def on_worker_event_done(machine_id: UUID, event: WorkerDoneEvent) -> Result[Non
node.debug_keep_node = True node.debug_keep_node = True
node.save() node.save()
event.stdout = event.stdout[-MAX_OUTPUT_SIZE:]
event.stderr = event.stderr[-MAX_OUTPUT_SIZE:]
task_event = TaskEvent( task_event = TaskEvent(
task_id=task.task_id, machine_id=machine_id, event_data=WorkerEvent(done=event) task_id=task.task_id, machine_id=machine_id, event_data=WorkerEvent(done=event)
) )