ExternalVendorCode/onefuzz
1
0
Fork 0
mirror of https://github.com/microsoft/onefuzz.git synced 2025-06-21 13:51:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

reading command from the body instead of URI (#72)

Browse Source
This commit is contained in:
Cheick Keita
2020-10-01 16:34:31 -07:00
committed by GitHub
parent b0c82d2036
commit ecb4eec469
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/api-service/__app__/agent_commands/__init__.py
View File

@ -10,11 +10,11 @@ from onefuzztypes.responses import BoolResult, PendingNodeCommand
from ..onefuzzlib.agent_authorization import verify_token from ..onefuzzlib.agent_authorization import verify_token
from ..onefuzzlib.pools import NodeMessage from ..onefuzzlib.pools import NodeMessage
from ..onefuzzlib.request import not_ok, ok, parse_uri from ..onefuzzlib.request import not_ok, ok, parse_request
def get(req: func.HttpRequest) -> func.HttpResponse: def get(req: func.HttpRequest) -> func.HttpResponse:
request = parse_uri(NodeCommandGet, req) request = parse_request(NodeCommandGet, req)
if isinstance(request, Error): if isinstance(request, Error):
return not_ok(request, context="NodeCommandGet") return not_ok(request, context="NodeCommandGet")
@ -31,7 +31,7 @@ def get(req: func.HttpRequest) -> func.HttpResponse:
def delete(req: func.HttpRequest) -> func.HttpResponse: def delete(req: func.HttpRequest) -> func.HttpResponse:
request = parse_uri(NodeCommandDelete, req) request = parse_request(NodeCommandDelete, req)
if isinstance(request, Error): if isinstance(request, Error):
return not_ok(request, context="NodeCommandDelete") return not_ok(request, context="NodeCommandDelete")

4
src/api-service/__app__/agent_registration/__init__.py
View File

@ -20,8 +20,8 @@ from ..onefuzzlib.request import not_ok, ok, parse_uri
def create_registration_response(machine_id: UUID, pool: Pool) -> func.HttpResponse: def create_registration_response(machine_id: UUID, pool: Pool) -> func.HttpResponse:
base_address = "https://%s.azurewebsites.net" % get_instance_name() base_address = "https://%s.azurewebsites.net" % get_instance_name()
events_url = "%s/api/agents/events?machine_id=%s" % (base_address, machine_id) events_url = "%s/api/agents/events" % base_address
commands_url = "%s/api/agents/commands?machine_id=%s" % (base_address, machine_id) commands_url = "%s/api/agents/commands" % base_address
work_queue = get_queue_sas( work_queue = get_queue_sas(
pool.get_pool_queue(), pool.get_pool_queue(),
account_id=get_fuzz_storage(), account_id=get_fuzz_storage(),