oss-fuzz/infra/auto-setup/sync.py

89 lines
2.2 KiB
Python
Executable File

#!/usr/bin/env python
"""Script to sync CF and Jenkins jobs."""
import json
import os
import sys
import urllib2
import xml.etree.ElementTree as ET
import jenkins
BUILD_BUCKET = 'clusterfuzz-builds'
JENKINS_SERVER = ('localhost', 8080)
SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
def main():
# Connect to jenkins server.
jenkins_login = get_jenkins_login()
server = jenkins.Jenkins('http://%s:%d' % JENKINS_SERVER,
username=jenkins_login[0], password=jenkins_login[1])
for library in get_libraries():
print 'syncing configs for', library
try:
# Create/update jenkins build job.
sync_jenkins_job(server, library)
except Exception as e:
print >>sys.stderr, 'Failed to setup job with exception', e
def get_libraries():
"""Return list of libraries for oss-fuzz."""
OSSFUZZ_TREE_URL = ('https://api.github.com/repos/google/oss-fuzz/'
'git/trees/master')
tree = json.loads(urllib2.urlopen(OSSFUZZ_TREE_URL).read())
libraries = []
targets_url = None
for item in tree['tree']:
if item['path'] == 'targets':
targets_url = item['url']
break
if not targets_url:
print >>sys.stderr, 'No libraries found.'
return []
tree = json.loads(urllib2.urlopen(targets_url).read())
for item in tree['tree']:
if item['type'] != 'tree':
continue
libraries.append(item['path'])
return libraries
def get_jenkins_login():
"""Returns (username, password) for jenkins."""
username = os.getenv('JENKINS_USER')
password = os.getenv('JENKINS_PASS')
return username, password
def sync_jenkins_job(server, library):
"""Sync the config with jenkins."""
job_name = 'targets/' + library
if server.job_exists(job_name):
# Job already set up.
# TODO(ochang): Also update jobs if the definition is different.
return
job_definition = ET.parse(os.path.join(SCRIPT_DIR, 'jenkins_config',
'base_job.xml'))
jenkinsfile_location = job_definition.findall('.//definition/scriptPath')[0]
jenkinsfile_location.text = 'targets/%s/Jenkinsfile' % library
server.create_job(job_name, ET.tostring(job_definition.getroot()))
server.build_job(job_name)
if __name__ == '__main__':
main()