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

Adding cache for sas return (#224)

Browse Source
This commit is contained in:
Anshuman Goel
2020-10-29 07:44:25 -07:00
committed by GitHub
parent f4b874e19e
commit 99b69d3e56
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/cli/onefuzz/api.py
View File

@ -10,11 +10,12 @@ import re
import subprocess # nosec
import uuid
from shutil import which
from typing import Callable, Dict, List, Optional, Tuple, Type, TypeVar
from typing import Callable, Dict, List, Optional, Tuple, Type, TypeVar, cast
from uuid import UUID
import pkg_resources
import semver
from memoization import cached
from onefuzztypes import enums, models, primitives, requests, responses
from pydantic import BaseModel
from six.moves import input # workaround for static analysis
@ -33,6 +34,8 @@ DEFAULT = {
# This was generated randomly and should be preserved moving forwards
ONEFUZZ_GUID_NAMESPACE = uuid.UUID("27f25e3f-6544-4b69-b309-9b096c5a9cbc")
ONE_HOUR_IN_SECONDS = 3600
DEFAULT_LINUX_IMAGE = "Canonical:UbuntuServer:18.04-LTS:latest"
DEFAULT_WINDOWS_IMAGE = "MicrosoftWindowsDesktop:Windows-10:rs5-pro:latest"
@ -146,6 +149,7 @@ class Files(Endpoint):
endpoint = "files"
@cached(ttl=ONE_HOUR_IN_SECONDS)
def _get_client(self, container: str) -> ContainerWrapper:
sas = self.onefuzz.containers.get(container).sas_url
return ContainerWrapper(sas)
@ -166,7 +170,8 @@ class Files(Endpoint):
""" get a file from a container """
self.logger.debug("getting file from container: %s:%s", container, filename)
client = self._get_client(container)
return client.download_blob(filename)
downloaded = cast(bytes, client.download_blob(filename))
return downloaded
def upload_file(
self, container: str, file_path: str, blob_name: Optional[str] = None