oss-fuzz/infra/gcb/build_base_images.py

74 lines
1.5 KiB
Python
Executable File

#!/usr/bin/python2
"""Build base images on Google Cloud Builder.
Usage: build_base_images.py
"""
import os
import yaml
from oauth2client.client import GoogleCredentials
from googleapiclient.discovery import build
BASE_IMAGES = [
'base-image',
'base-clang',
'base-builder',
'base-runner',
'base-runner-debug',
'base-msan-builder',
]
TAG_PREFIX = 'gcr.io/oss-fuzz-base/'
def get_steps(images):
steps = [{
'args': [
'clone', 'https://github.com/google/oss-fuzz.git',
],
'name': 'gcr.io/cloud-builders/git',
}]
for base_image in images:
steps.append({
'args': [
'build',
'-t',
TAG_PREFIX + base_image,
'.',
],
'dir': 'oss-fuzz/infra/base-images/' + base_image,
'name': 'gcr.io/cloud-builders/docker',
})
return steps
def main():
options = {}
if "GCB_OPTIONS" in os.environ:
options = yaml.safe_load(os.environ["GCB_OPTIONS"])
build_body = {
'steps': get_steps(BASE_IMAGES),
'timeout': str(4 * 3600) + 's',
'options': options,
'images': [
TAG_PREFIX + base_image for base_image in BASE_IMAGES
],
}
credentials = GoogleCredentials.get_application_default()
cloudbuild = build('cloudbuild', 'v1', credentials=credentials)
build_info = cloudbuild.projects().builds().create(projectId='oss-fuzz-base', body=build_body).execute()
build_id = build_info['metadata']['build']['id']
print build_id
if __name__ == "__main__":
main()