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

Add Compute Gallery images support (#1450)

Browse Source
This commit is contained in:
Peleus Uhley
2021-11-18 12:20:29 -08:00
committed by GitHub
parent dd03996961
commit 58091c2859
1 changed files with 19 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
src/api-service/__app__/onefuzzlib/azure/image.py
View File

@ -19,14 +19,27 @@ from .compute import get_compute_client
@cached(ttl=60) @cached(ttl=60)
def get_os(region: Region, image: str) -> Union[Error, OS]: def get_os(region: Region, image: str) -> Union[Error, OS]:
client = get_compute_client() client = get_compute_client()
# The dict returned here may not have any defined keys.
#
# See: https://github.com/Azure/msrestazure-for-python/blob/v0.6.3/msrestazure/tools.py#L134 # noqa: E501
parsed = parse_resource_id(image) parsed = parse_resource_id(image)
if "resource_group" in parsed: if "resource_group" in parsed:
try: if parsed["type"] == "galleries":
name = client.images.get( try:
parsed["resource_group"], parsed["name"] # See: https://docs.microsoft.com/en-us/rest/api/compute/gallery-images/get#galleryimage # noqa: E501
).storage_profile.os_disk.os_type.name name = client.gallery_images.get(
except (ResourceNotFoundError, CloudError) as err: parsed["resource_group"], parsed["name"], parsed["child_name_1"]
return Error(code=ErrorCode.INVALID_IMAGE, errors=[str(err)]) ).os_type.lower()
except (ResourceNotFoundError, CloudError) as err:
return Error(code=ErrorCode.INVALID_IMAGE, errors=[str(err)])
else:
try:
# See: https://docs.microsoft.com/en-us/rest/api/compute/images/get
name = client.images.get(
parsed["resource_group"], parsed["name"]
).storage_profile.os_disk.os_type.lower()
except (ResourceNotFoundError, CloudError) as err:
return Error(code=ErrorCode.INVALID_IMAGE, errors=[str(err)])
else: else:
publisher, offer, sku, version = image.split(":") publisher, offer, sku, version = image.split(":")
try: try: