blob: 019531843b4cf1c01471efc0503ef7a251272d56 [file] [log] [blame]
# -*- coding: utf-8 -*-
# Copyright 2019 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Payload API Service."""
from __future__ import print_function
from chromite.api import controller
from chromite.lib import cros_build_lib
from chromite.api import faux
from chromite.api import validate
from chromite.service import payload
_VALID_IMAGE_PAIRS = (('src_signed_image', 'tgt_signed_image'),
('src_unsigned_image', 'tgt_unsigned_image'),
('src_dlc_image', 'tgt_dlc_image'),
('full_update', 'tgt_unsigned_image'),
('full_update', 'tgt_signed_image'),
('full_update', 'tgt_dlc_image'))
_DEFAULT_PAYGEN_CACHE_DIR = '.paygen_cache'
# We have more fields we might validate however, they're either
# 'oneof' or allowed to be the empty value by design. If @validate
# gets more complex in the future we can add more here.
def GeneratePayload(input_proto, output_proto, config):
"""Generate a update payload ('do paygen').
input_proto (PayloadGenerationRequest): Input proto.
output_proto (PayloadGenerationResult): Output proto.
config (api.config.ApiConfig): The API call config.
A controller return code (e.g. controller.RETURN_CODE_SUCCESS).
# Resolve the tgt image oneof.
tgt_name = input_proto.WhichOneof('tgt_image_oneof')
tgt_image = getattr(input_proto, tgt_name)
except AttributeError:
cros_build_lib.Die('%s is not a known tgt image type' % (tgt_name,))
# Resolve the src image oneof.
src_name = input_proto.WhichOneof('src_image_oneof')
# If the source image is 'full_update' we lack a source entirely.
if src_name == 'full_update':
src_image = None
# Otherwise we have an image.
src_image = getattr(input_proto, src_name)
except AttributeError:
cros_build_lib.Die('%s is not a known src image type' % (src_name,))
# Ensure they are compatible oneofs.
if (src_name, tgt_name) not in _VALID_IMAGE_PAIRS:
cros_build_lib.Die('%s and %s are not valid image pairs' %
(src_image, tgt_image))
# Find the value of bucket or default to 'chromeos-releases'.
destination_bucket = input_proto.bucket or 'chromeos-releases'
if input_proto.dryrun:
keyset = ''
upload = False
keyset = input_proto.keyset
upload = True
# There's a potential that some paygen_lib library might raise here, but since
# we're still involved in config we'll keep it before the validate_only.
payload_config = payload.PayloadConfig(tgt_image, src_image,
destination_bucket, input_proto.verify,
keyset, upload,
# If configured for validation only we're done here.
if config.validate_only:
return controller.RETURN_CODE_VALID_INPUT
# Do payload generation.
local_path, remote_uri = payload_config.GeneratePayload()
_SetGeneratePayloadOutputProto(output_proto, local_path, remote_uri)
if remote_uri or not upload and local_path:
return controller.RETURN_CODE_SUCCESS
def _SetGeneratePayloadOutputProto(output_proto, local_path, remote_uri):
"""Set the output proto with the results from the service class.
output_proto (PayloadGenerationResult_pb2): The output proto.
local_path (str): set output_proto with the local path, or ''.
remote_uri (str): set output_proto with the remote uri, or ''.
output_proto.success = True
output_proto.local_path = local_path or ''
output_proto.remote_uri = remote_uri or ''