370 lines
14 KiB
Python
370 lines
14 KiB
Python
#!/usr/bin/python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
# (c) 2013, Romeo Theriault <romeot () hawaii.edu>
|
|
#
|
|
# This file is part of Ansible
|
|
#
|
|
# Ansible is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# Ansible is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
# see examples/playbooks/uri.yml
|
|
|
|
import shutil
|
|
import tempfile
|
|
import base64
|
|
try:
|
|
import json
|
|
except ImportError:
|
|
import simplejson as json
|
|
|
|
DOCUMENTATION = '''
|
|
---
|
|
module: uri
|
|
short_description: Interacts with webservices
|
|
description:
|
|
- Interacts with HTTP and HTTPS webservices and supports Digest, Basic and WSSE
|
|
HTTP authentication mechanisms.
|
|
version_added: "1.1"
|
|
options:
|
|
url:
|
|
description:
|
|
- HTTP or HTTPS URL in the form (http|https)://host.domain[:port]/path
|
|
required: true
|
|
default: null
|
|
aliases: []
|
|
dest:
|
|
description:
|
|
- path of where to download the file to.
|
|
- If I(dest) is a directory, the basename of the file on the remote server will be used.
|
|
required: false
|
|
default: null
|
|
user:
|
|
description:
|
|
- username for the module to use for Digest, Basic or WSSE authentication.
|
|
required: false
|
|
default: null
|
|
password:
|
|
description:
|
|
- password for the module to use for Digest, Basic or WSSE authentication.
|
|
required: false
|
|
default: null
|
|
body:
|
|
description:
|
|
- The body of the http request/response to the webservice.
|
|
required: false
|
|
default: null
|
|
method:
|
|
description:
|
|
- The HTTP method of the request or response.
|
|
required: false
|
|
choices: [ "GET", "POST", "PUT", "HEAD", "DELETE", "OPTIONS" ]
|
|
default: "GET"
|
|
return_content:
|
|
description:
|
|
- Whether or not to return the body of the request in the "content" register variable.
|
|
required: false
|
|
choices: [ "yes", "no" ]
|
|
default: no
|
|
force_basic_auth:
|
|
description:
|
|
- httplib2, the library used by the uri module only sends authentication information when a webservice
|
|
responds to an initial request with a 401 status. Since some basic auth services do not properly
|
|
send a 401, logins will fail. This option forces the sending of the Basic authentication header
|
|
upon initial request.
|
|
required: false
|
|
choices: [ "yes", "no" ]
|
|
default: no
|
|
follow_redirects:
|
|
description:
|
|
- Whether or not the URI module should follow all redirects.
|
|
required: false
|
|
choices: [ "yes", "no" ]
|
|
default: no
|
|
creates:
|
|
description:
|
|
- a filename, when it already exists, this step will not be run.
|
|
required: false
|
|
removes:
|
|
description:
|
|
- a filename, when it does not exist, this step will not be run.
|
|
required: false
|
|
status_code:
|
|
description:
|
|
- A valid, numeric, HTTP status code that signifies success of the request.
|
|
required: false
|
|
default: 200
|
|
HEADER_:
|
|
description:
|
|
- Any parameter starting with "HEADER_" is a sent with your request as a header. For example,
|
|
HEADER_Content-Type="application/json" would send the header "Content-Type" along with your
|
|
request with a value of "application/json".
|
|
required: false
|
|
default: null
|
|
others:
|
|
description:
|
|
- all arguments accepted by the M(file) module also work here
|
|
required: false
|
|
examples:
|
|
- code: "uri: url=http://www.awesome.com method=GET"
|
|
description: "Check that you can connect to a page and it returns a status 200"
|
|
|
|
- code: |
|
|
uri: url=http://www.awesome.com method=GET return_content=yes
|
|
register: webpage
|
|
|
|
command: /usr/bin/uptime
|
|
when_string: '"AWESOME" in "${webpage.content}"'
|
|
|
|
description: Check that a page returns a status 200 and the word AWESOME is in the page contents.
|
|
|
|
- code: "uri: url=https://your.jira.server.com/rest/api/2/issue/ method=POST user=your_username password=your_pass body='$FILE(issue.json)' force_basic_auth=yes status_code=201 HEADER_Content-Type="application/json"
|
|
description: "Create a JIRA issue."
|
|
|
|
- code: |
|
|
uri: url=https://your.form.based.auth.app.com/index.php method=POST body="name=your_username&password=your_password&enter=Sign%20in" status_code=302 HEADER_Content-Type="application/x-www-form-urlencoded"
|
|
register: login
|
|
|
|
uri: url=https://your.form.based.auth.app.com/dashboard.php method=GET return_content=yes HEADER_Cookie="${login.set_cookie}"
|
|
description: "Login to a form based webpage, then use the cookie that got returned to access the app in later tasks."
|
|
|
|
notes:
|
|
- This module doesn't yet support configuration for proxies.
|
|
# informational: requirements for nodes
|
|
requirements: [ urlparse, httplib2 ]
|
|
author: Romeo Theriault
|
|
'''
|
|
|
|
HAS_HTTPLIB2 = True
|
|
try:
|
|
import httplib2
|
|
except ImportError:
|
|
HAS_HTTPLIB2 = False
|
|
|
|
HAS_URLPARSE = True
|
|
|
|
try:
|
|
import urlparse
|
|
import socket
|
|
except ImportError:
|
|
HAS_URLPARSE = False
|
|
|
|
|
|
def write_file(module, url, dest, content):
|
|
# create a tempfile with some test content
|
|
fd, tmpsrc = tempfile.mkstemp()
|
|
f = open(tmpsrc, 'wb')
|
|
try:
|
|
f.write(content)
|
|
except Exception, err:
|
|
os.remove(tmpsrc)
|
|
module.fail_json(msg="failed to create temporary content file: %s" % str(err))
|
|
f.close()
|
|
|
|
md5sum_src = None
|
|
md5sum_dest = None
|
|
|
|
# raise an error if there is no tmpsrc file
|
|
if not os.path.exists(tmpsrc):
|
|
os.remove(tmpsrc)
|
|
module.fail_json(msg="Source %s does not exist" % (tmpsrc))
|
|
if not os.access(tmpsrc, os.R_OK):
|
|
os.remove(tmpsrc)
|
|
module.fail_json( msg="Source %s not readable" % (tmpsrc))
|
|
md5sum_src = module.md5(tmpsrc)
|
|
|
|
# check if there is no dest file
|
|
if os.path.exists(dest):
|
|
# raise an error if copy has no permission on dest
|
|
if not os.access(dest, os.W_OK):
|
|
os.remove(tmpsrc)
|
|
module.fail_json( msg="Destination %s not writable" % (dest))
|
|
if not os.access(dest, os.R_OK):
|
|
os.remove(tmpsrc)
|
|
module.fail_json( msg="Destination %s not readable" % (dest))
|
|
md5sum_dest = module.md5(dest)
|
|
else:
|
|
if not os.access(os.path.dirname(dest), os.W_OK):
|
|
os.remove(tmpsrc)
|
|
module.fail_json( msg="Destination %s not writable" % (os.path.dirname(dest)))
|
|
|
|
if md5sum_src != md5sum_dest:
|
|
try:
|
|
shutil.copyfile(tmpsrc, dest)
|
|
except Exception, err:
|
|
os.remove(tmpsrc)
|
|
module.fail_json(msg="failed to copy %s to %s: %s" % (tmpsrc, dest, str(err)))
|
|
|
|
os.remove(tmpsrc)
|
|
|
|
|
|
def url_filename(url):
|
|
fn = os.path.basename(urlparse.urlsplit(url)[2])
|
|
if fn == '':
|
|
return 'index.html'
|
|
return fn
|
|
|
|
|
|
def uri(module, url, dest, user, password, body, method, headers, redirects):
|
|
# To debug
|
|
#httplib2.debug = 4
|
|
|
|
# Create a Http object and set some default options.
|
|
h = httplib2.Http(disable_ssl_certificate_validation=True)
|
|
h.follow_all_redirects = redirects
|
|
h.forward_authorization_headers = True
|
|
|
|
# If they have a username or password verify they have both, then add them to the request
|
|
if user is not None and password is None:
|
|
module.fail_json(msg="Both a username and password need to be set.")
|
|
if password is not None and user is None:
|
|
module.fail_json(msg="Both a username and password need to be set.")
|
|
if user is not None and password is not None:
|
|
h.add_credentials(user, password)
|
|
|
|
# Make the request, or try to :)
|
|
try:
|
|
resp, content = h.request(url, method=method, body=body, headers=headers)
|
|
return resp, content
|
|
except httplib2.RedirectMissingLocation:
|
|
module.fail_json(msg="A 3xx redirect response code was provided but no Location: header was provided to point to the new location.")
|
|
except httplib2.RedirectLimit:
|
|
module.fail_json(msg="The maximum number of redirections was reached without coming to a final URI.")
|
|
except httplib2.ServerNotFoundError:
|
|
module.fail_json(msg="Unable to resolve the host name given.")
|
|
except httplib2.RelativeURIError:
|
|
module.fail_json(msg="A relative, as opposed to an absolute URI, was passed in.")
|
|
except httplib2.FailedToDecompressContent:
|
|
module.fail_json(msg="The headers claimed that the content of the response was compressed but the decompression algorithm applied to the content failed.")
|
|
except httplib2.UnimplementedDigestAuthOptionError:
|
|
module.fail_json(msg="The server requested a type of Digest authentication that we are unfamiliar with.")
|
|
except httplib2.UnimplementedHmacDigestAuthOptionError:
|
|
module.fail_json(msg="The server requested a type of HMACDigest authentication that we are unfamiliar with.")
|
|
except httplib2.UnimplementedHmacDigestAuthOptionError:
|
|
module.fail_json(msg="The server requested a type of HMACDigest authentication that we are unfamiliar with.")
|
|
except socket.error, e:
|
|
module.fail_json(msg="Socket error: %s to %s" % (e, url))
|
|
|
|
def main():
|
|
if not HAS_HTTPLIB2:
|
|
module.fail_json(msg="httplib2 is not installed")
|
|
if not HAS_URLPARSE:
|
|
module.fail_json(msg="urlparse is not installed")
|
|
|
|
module = AnsibleModule(
|
|
argument_spec = dict(
|
|
url = dict(required=True),
|
|
dest = dict(required=False, default=None),
|
|
user = dict(required=False, default=None),
|
|
password = dict(required=False, default=None),
|
|
body = dict(required=False, default=None),
|
|
method = dict(required=False, default='GET', choices=['GET', 'POST', 'PUT', 'HEAD', 'DELETE', 'OPTIONS']),
|
|
return_content = dict(required=False, default='no', choices=BOOLEANS),
|
|
force_basic_auth = dict(required=False, default='no', choices=BOOLEANS),
|
|
follow_redirects = dict(required=False, default='no', choices=BOOLEANS),
|
|
creates = dict(required=False, default=None),
|
|
removes = dict(required=False, default=None),
|
|
status_code = dict(required=False, default="200"),
|
|
),
|
|
check_invalid_arguments=False,
|
|
add_file_common_args=True
|
|
)
|
|
|
|
url = module.params['url']
|
|
user = module.params['user']
|
|
password = module.params['password']
|
|
body = module.params['body']
|
|
method = module.params['method']
|
|
dest = module.params['dest']
|
|
return_content = module.boolean(module.params['return_content'])
|
|
force_basic_auth = module.boolean(module.params['force_basic_auth'])
|
|
follow_redirects = module.boolean(module.params['follow_redirects'])
|
|
creates = module.params['creates']
|
|
removes = module.params['removes']
|
|
status_code = module.params['status_code']
|
|
|
|
# Grab all the http headers. Need this hack since passing multi-values is currently a bit ugly. (e.g. headers='{"Content-Type":"application/json"}')
|
|
dict_headers = {}
|
|
for key, value in module.params.iteritems():
|
|
if key.startswith("HEADER_"):
|
|
skey = key.replace("HEADER_", "")
|
|
dict_headers[skey] = value
|
|
|
|
|
|
if creates is not None:
|
|
# do not run the command if the line contains creates=filename
|
|
# and the filename already exists. This allows idempotence
|
|
# of uri executions.
|
|
creates = os.path.expanduser(creates)
|
|
if os.path.exists(creates):
|
|
module.exit_json(stdout="skipped, since %s exists" % creates, skipped=True, changed=False, stderr=False, rc=0)
|
|
|
|
if removes is not None:
|
|
# do not run the command if the line contains removes=filename
|
|
# and the filename do not exists. This allows idempotence
|
|
# of uri executions.
|
|
v = os.path.expanduser(removes)
|
|
if not os.path.exists(removes):
|
|
module.exit_json(stdout="skipped, since %s does not exist" % removes, skipped=True, changed=False, stderr=False, rc=0)
|
|
|
|
|
|
# httplib2 only sends authentication after the server asks for it with a 401.
|
|
# Some 'basic auth' servies fail to send a 401 and require the authentication
|
|
# up front. This creates the Basic authentication header and sends it immediately.
|
|
if force_basic_auth:
|
|
dict_headers["Authorization"] = "Basic {0}".format(base64.b64encode("{0}:{1}".format(user, password)))
|
|
|
|
# Redirects
|
|
if follow_redirects:
|
|
redirects = True
|
|
else:
|
|
redirects = False
|
|
|
|
# If there is a dest, expand it and get the filename if one not explicitly set.
|
|
if dest is not None:
|
|
dest = os.path.expanduser(dest)
|
|
if os.path.isdir(dest):
|
|
dest = os.path.join(dest, url_filename(url))
|
|
|
|
# Make the request
|
|
resp, content = uri(module, url, dest, user, password, body, method, dict_headers, redirects)
|
|
|
|
# Write the file out if requested
|
|
if dest is not None:
|
|
write_file(module, url, dest, content)
|
|
|
|
# allow file attribute changes
|
|
changed = True
|
|
module.params['path'] = dest
|
|
file_args = module.load_file_common_arguments(module.params)
|
|
file_args['path'] = dest
|
|
changed = module.set_file_attributes_if_different(file_args, changed)
|
|
|
|
|
|
# Transmogrify the headers, replacing '-' with '_', since variables dont work with dashes.
|
|
uresp = {}
|
|
for key, value in resp.iteritems():
|
|
ukey = key.replace("-", "_")
|
|
uresp[ukey] = value
|
|
|
|
if resp['status'] != status_code:
|
|
module.fail_json(msg="Status code was not " + status_code, content=content, **uresp)
|
|
elif return_content:
|
|
module.exit_json(changed=True, content=content, **uresp)
|
|
else:
|
|
module.exit_json(changed=True, **uresp)
|
|
|
|
|
|
# this is magic, see lib/ansible/module_common.py
|
|
#<<INCLUDE_ANSIBLE_MODULE_COMMON>>
|
|
main()
|