| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200 |
- #!/bin/bash
- # Licensed to Cloudera, Inc. under one
- # or more contributor license agreements. See the NOTICE file
- # distributed with this work for additional information
- # regarding copyright ownership. Cloudera, Inc. licenses this file
- # to you 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.
- #parse command line arguments
- parse_arguments()
- {
- # Test that we're using compatible getopt version.
- getopt -T > /dev/null
- if [[ $? -ne 4 ]]; then
- echo "Incompatible getopt version."
- exit 1
- fi
- HUE_AUTH_SERVER="LDAP"
- GETOPT=`getopt -n $0 -o u:,w:,s:,p:,e,v,h \
- -l user:,password:,server:,port:,enablessl,verbose,help \
- -- "$@"`
- eval set -- "$GETOPT"
- while true;
- do
- case "$1" in
- -u|--user)
- HUE_USER=$2
- shift 2
- ;;
- -w|--password)
- HUE_PASSWORD=$2
- shift 2
- ;;
- -s|--server)
- HUE_SERVER=$2
- shift 2
- ;;
- -p|--port)
- HUE_PORT=$2
- shift 2
- ;;
- -e|--enablessl)
- ENABLESSL=1
- shift
- ;;
- -v|--verbose)
- VERBOSE=1
- shift
- ;;
- --)
- shift
- break
- ;;
- *)
- usage
- exit 1
- ;;
- esac
- done
- #
- if [[ -z ${HUE_USER} ]]
- then
- HUE_USER="admin"
- fi
- if [[ -z ${HUE_PASSWORD} ]]
- then
- HUE_PASSWORD="admin"
- fi
- if [[ -z ${HUE_SERVER} ]]
- then
- HUE_SERVER="localhost"
- fi
- if [[ -z ${HUE_PORT} ]]
- then
- HUE_PORT="8888"
- fi
- }
- usage()
- {
- cat << EOF
- usage: $0 [options]
- Collects troubleshooting data from Hue server:
- OPTIONS
- -u|--user Hue username - default admin.
- -w|--password Hue password - default admin.
- -s|--server Hue server host - localhost.
- -p|--port Hue server port - 8888.
- -h|--help Show this message.
- EOF
- }
- main()
- {
- parse_arguments "$@"
-
- if [[ -z ${ENABLESSL} ]]
- then
- HUE_HTTP="http"
- else
- HUE_HTTP="https"
- fi
- HUE_PASS_URL="${HUE_HTTP}://${HUE_SERVER}:${HUE_PORT}/accounts/login/"
- HUE_IMPALA_PARAMS_URL="${HUE_HTTP}://${HUE_SERVER}:${HUE_PORT}/impala/api/query/parameters"
- POST_STRING="query-query=drop+table+test1%3B&query-database=default&settings-next_form_id=0&file_resources-next_form_id=0&functions-next_form_id=0&query-email_notify=false&query-is_parameterized=true"
-
- hue_login
- echo "Posting Impala Params:"
- do_curl \
- POST \
- "${HUE_IMPALA_PARAMS_URL}" \
- -H "content-type:application/x-www-form-urlencoded" -d "${POST_STRING}"
- }
- function do_curl() {
- METHOD=$1
- shift
- URL=$1
- shift
- ARGS=$@
- CURL=$(which curl)
- if [ -z ${COOKIE_JAR} ]
- then
- COOKIE_JAR=/tmp/cookie.jar
- fi
- if [ -f ${COOKIE_JAR} ]
- then
- CSRF_TOKEN=`grep ${HUE_SERVER} ${COOKIE_JAR} | grep csrftoken | cut -f 7`
- fi
- if [ ! -f ${CURL} ]
- then
- echo "curl not found, unable to run any curl commands"
- else
- debug "Connecting to ${URL}"
- debug "${CURL} \
- ${CURL_OPTS} \
- -k \
- -e \"${HUE_HTTP}://${HUE_SERVER}:${HUE_PORT}/\" \
- -b @${COOKIE_JAR} \
- -c ${COOKIE_JAR} \
- -H \"X-CSRFToken: ${CSRF_TOKEN}\" \
- -X ${METHOD} \
- -f \
- ${URL} \
- ${ARGS}"
- ${CURL} \
- ${CURL_OPTS} \
- -k \
- -e "${HUE_HTTP}://${HUE_SERVER}:${HUE_PORT}/" \
- -b @${COOKIE_JAR} \
- -c ${COOKIE_JAR} \
- -H "X-CSRFToken: ${CSRF_TOKEN}" \
- -X ${METHOD} \
- -f \
- ${URL} \
- ${ARGS}
- fi
- }
- function hue_login() {
- echo "Login to Hue to get Cookie:"
- do_curl \
- GET \
- "${HUE_PASS_URL}" \
- -L 2>&1 > /dev/null
- do_curl \
- POST \
- "${HUE_PASS_URL}" \
- -F username=${HUE_USER} -F password="${HUE_PASSWORD}" -F server="${HUE_AUTH_SERVER}" 2>&1 > /dev/null
- }
- debug()
- {
- if [[ ! -z $VERBOSE ]]
- then
- echo "$1"
- fi
- }
- main "$@"
|