aboutsummaryrefslogtreecommitdiffstats
path: root/ansible/library/fetch_url_and_verify.py
diff options
context:
space:
mode:
authorRoss Brattain <ross.b.brattain@intel.com>2017-08-11 17:47:29 +0000
committerGerrit Code Review <gerrit@opnfv.org>2017-08-11 17:47:29 +0000
commit0949884ab431a16dc1841118e13c0b893fa11b85 (patch)
treeba8588f8fbaf0a17179b339f5512d949386a77a7 /ansible/library/fetch_url_and_verify.py
parent43bf12d6ab7bcaea16dc75ed4ccbe3895cf51da3 (diff)
parentc7c51d5100e8eba93337c34bd9eb101ec4cf70df (diff)
Merge "yardstick setup ansible, including load_images"
Diffstat (limited to 'ansible/library/fetch_url_and_verify.py')
-rw-r--r--ansible/library/fetch_url_and_verify.py80
1 files changed, 80 insertions, 0 deletions
diff --git a/ansible/library/fetch_url_and_verify.py b/ansible/library/fetch_url_and_verify.py
new file mode 100644
index 000000000..6c5c0a8c2
--- /dev/null
+++ b/ansible/library/fetch_url_and_verify.py
@@ -0,0 +1,80 @@
+#!/usr/bin/env python
+# Copyright (c) 2017 Intel Corporation
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+DOCUMENTATION = '''
+---
+module: fetch_url_and_verify
+short_description: Fetch image and verify against a SHA256SUM URL
+description:
+ - Download a URL and check it against a remote SHA256SUMS file
+options:
+ url: Image URL
+ image_dest: Image filename
+ sha256_url: SHA256SUMS URL
+ dest: python file mode (w, wb, a, ab)
+ retries: fetch retries
+'''
+
+
+def main():
+ module = AnsibleModule(
+ argument_spec={
+ 'url': {'required': True, 'type': 'str'},
+ 'sha256url': {'required': True, 'type': 'str'},
+ 'dest': {'required': True, 'type': 'path'},
+ 'retries': {'required': False, 'type': 'int', 'default': 3},
+ }
+ )
+ params = module.params
+ url = params['url']
+ dest = params['dest']
+ sha256url = params['sha256url']
+ retries = params['retries']
+
+ image_dir, image_filename = os.path.split(dest)
+ rc, stdout, stderr = module.run_command(['curl', '-sS', sha256url])
+ if rc == 0 and stdout:
+ sha256line = next(
+ (l for l in stdout.splitlines() if image_filename in l), "")
+ if not sha256line:
+ module.fail_json(
+ msg="Unable to find SHA256SUMS line for file {}".format(
+ image_filename))
+ rc = \
+ module.run_command(['sha256sum', '-c'], data=sha256line, cwd=image_dir)[0]
+ if rc == 0:
+ sha256sum = sha256line.split()[0]
+ module.exit_json(changed=False, dest=dest, url=url,
+ sha256sum=sha256sum)
+
+ for retry in range(retries):
+ curl_rc, stdout, stderr = module.run_command(
+ ['curl', '-sS', '-o', dest, url], cwd=image_dir)
+ if curl_rc == 0:
+ sha256_rc, stdout, stderr = module.run_command(['sha256sum', '-c'],
+ data=sha256line,
+ cwd=image_dir)
+ if sha256_rc == 0:
+ module.exit_json(changed=True)
+
+ module.fail_json(msg="Unable to download {}".format(url), stdout=stdout,
+ stderr=stderr)
+
+
+# <<INCLUDE_ANSIBLE_MODULE_COMMON>>
+from ansible.module_utils.basic import * # noqa
+
+if __name__ == '__main__':
+ main()