#!/bin/sh
# PCP QA Test No. 766
# Test using the pmfind app to find PCP servers using the active probing
# discovery mechanism. Probe inet only.
#
# Copyright (c) 2014 Red Hat.  All Rights Reserved.
#

seq=`basename $0`
echo "QA output created by $seq"

# get standard environment, filters and checks
. ./common.product
. ./common.filter
. ./common.check

_get_libpcp_config
$service_discovery || _notrun "No support for service discovery"

signal=$PCP_BINADM_DIR/pmsignal
$sudo $signal -a pmproxy pmwebd

status=1	# failure is the default!
$sudo rm -rf $tmp.* $seq.full
trap "cd $here; rm -rf $tmp.*; exit \$status" 0 1 2 3 15

_sought_filter()
{
    # Only pmcd is guaranteed to be running, but other services may also be.
    # Transform two cases - no servers vs found servers - into deterministic
    # output for the --all invocations
    sed \
	-e 's/No \(pmproxy servers\) discovered/Sought \1/g' \
	-e 's/No \(pmwebd servers\) discovered/Sought \1/g' \
	-e 's/Discovered \(pmproxy servers\):/Sought \1/g' \
	-e 's/Discovered \(pmwebd servers\):/Sought \1/g' \
    # end
}

_unresolved_filter()
{
    sed -e '/  pcp:/d;
	    /  proxy:/d;
	    /  http:/d' \
    | _sought_filter
}

_resolved_filter()
{
    # Pass unresolved urls, filter the resolved ones.
    sed -e '/  pcp:\/\/[0-9]\+\.[0-9]\+\.[0-9]\+\.[0-9]\+/{p;b};
	    /  proxy:\/\/[0-9]\+\.[0-9]\+\.[0-9]\+\.[0-9]\+/{p;b};
	    /  http:\/\/[0-9]\+\.[0-9]\+\.[0-9]\+\.[0-9]\+/{p;b};
	    /  pcp:/d;
	    /  proxy:/d;
	    /  http:/d' \
    | _sought_filter
}

# real QA test starts here

# Obtain the inet address of an active network interface
# We're extracting a metric of the form:
#
#    inst [0 or "eth0"] value "172.31.0.12"
#
# and then extracting the address
#
addr=`pminfo -f network.interface.inet_addr | \
  tail -n +3 | \
  grep -v 127.0.0.1 | \
  head -n1 | \
  awk '{ print $6 }' | \
  sed s/\"//g`

[ -z addr ] && _notrun "no active inet interfaces"

# Probe the obtained network.
# Test various combinations of service queries and subnet sizes. Keep the
# subnet size small -- say max 4 bits.
echo "-m probe=$addr/32" >> $seq.full
echo "-m probe=INET_ADDR/32"
pmfind -m probe=$addr/32 | _unresolved_filter
echo "Exit status: $?" | tee -a $seq.full

echo "-s pmcd -m probe=$addr/31" -r >> $seq.full
echo "-s pmcd -m probe=INET_ADDR/31 -r"
pmfind -s pmcd -m probe=$addr/30 -r | _resolved_filter
echo "Exit status: $?" | tee -a $seq.full

echo "-q -m probe=$addr/30" >> $seq.full
echo "-q -m probe=INET_ADDR/30"
pmfind -q -m probe=$addr/29 | _unresolved_filter
echo "Exit status: $?" | tee -a $seq.full

echo "-q -s pmcd -m probe=$addr/29 --resolve" >> $seq.full
echo "-q -s pmcd -m probe=INET_ADDR/29 --resolve"
pmfind -q -s pmcd -m probe=$addr/28 --resolve | _resolved_filter
echo "Exit status: $?" | tee -a $seq.full

echo "-q -s pmcd -m probe=$addr/28,maxThreads=8" >> $seq.full
echo "-q -s pmcd -m probe=INET_ADDR/28,maxThreads=8"
pmfind -q -s pmcd -m probe=$addr/28,maxThreads=8 | _unresolved_filter
echo "Exit status: $?" | tee -a $seq.full

# success, all done
status=0

exit