#!/bin/bash
#
# FM TESTER - Main file
#
# Copyright 2014-2015 - Olivier BEDOUET pour lafibre.info
#
# ------------------------------------------------------------------------
#    This file is part of FM TESTER.
#
#    FM TESTER 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.
#
#    FM TESTER 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 FM TESTER.  If not, see <http://www.gnu.org/licenses/>.
# ------------------------------------------------------------------------
#
# History
# 0.01 (13/08/2015): Version initiale
# 0.02 (14/08/2015): 
#  - suppression de LANG
#  - ajout de tests vers Free
# 0.03 (14/08/2015)
#  - tests HTTPS
#  - changement de la methode de kill
#
# Usage:
#  user@linux$ ./fm_tester.sh
#

BASE_URL=http://1.testdebit.info/fichiers
SIZES="10ko 100ko 1Mo"
EXTENSIONS="iso pdf txt doc docx odt xls xlsx ods ppt pptx odp html jpg jpeg png gif svg webp woff2 avi mpg mp4 mov swf wmv divx xvid mkv mka mks flv rmvb mp3 wma wav aif webm raw au asf aac vqf zip 7z rar tar tar.xz tar.bz2 gz bz2 xz tgz sit hqx sea uu msi exe iso bin apk deb rpm ova dmg rnd vivien"

if [ -n "$1" -a -n "$2" ]
then
	# Lance le polling
	unset LANG
	echo $$ > /tmp/wget.lck
	wget -4 -O /dev/null ${BASE_URL}/$1/$1.$2 > /tmp/wget.result 2>&1 </dev/null

	rm -f /tmp/wget.lck
elif [ -n "$1" ]
then
	# Lance le polling d'une URL specifique
	unset LANG
	echo $$ > /tmp/wget.lck
	wget -4 -O /dev/null $1 > /tmp/wget.result 2>&1

	rm -f /tmp/wget.lck
else
	# Appel sans argument: lance le polling

	# ETAPE1: essaye toutes les extensions avec differentes tailles ==============
	echo "Testing HTTP -------------"
	for SIZE in $SIZES
	do

		echo "Testing size=$SIZE"

		case $SIZE in
		1ko ) TIMEOUT=5;;
		10ko ) TIMEOUT=10;;
		100ko ) TIMEOUT=10;;
		1Mo ) TIMEOUT=20;;
		2Mo ) TIMEOUT=40;;
		esac

		for EXTENSION in $EXTENSIONS
		do
			echo -e "$EXTENSION=\c"
			./fm_tester.sh $SIZE $EXTENSION 2>/dev/null &

			i=0;
			while [ $i -lt $TIMEOUT ]
			do
				sleep 1
				if [ -f /tmp/wget.lck ]
				then
					i=`expr $i + 1`
				else
					i=$TIMEOUT
				fi
			done

			if [ $i -eq $TIMEOUT -a -f /tmp/wget.lck ]
			then
				#kill `cat /tmp/wget.lck`
				WGET_PID=`ps ux | grep wget | grep -v grep | awk '{print $2}'`
				[ -n "$WGET_PID" ] && kill $WGET_PID
				rm -f /tmp/wget.lck
				echo TIMEOUT
				# Pour DEBUG
				#cat /tmp/wget.result
			elif [ -f /tmp/wget.result ]
			then
				RAW_SPEED=`cat /tmp/wget.result | grep saved | awk '{print $3,$4}'`
				if [ -z "$RAW_SPEED" ]
				then
					# Fallback test
					RAW_SPEED=`cat /tmp/wget.result | grep enregistr | awk '{print $3,$4}'`

					if [ -z "$RAW_SPEED" ]
					then
						RAW_SPEED=`cat /tmp/wget.result | grep sauvegar | awk '{print $3,$4}'`
					fi
				fi

				if [ -z "$RAW_SPEED" ]
				then
					echo "No Value (see /tmp/wget.result)"
				else
					echo $RAW_SPEED
				fi

				# Pour DEBUG
				#cat /tmp/wget.result
			fi
		done
	done

	# Etape 2: test avec les sites de Free ===================================
	echo "Testing Free Servers -------------"
	echo -e "Testing http://test-debit.free.fr/512.rnd=\c"
	./fm_tester.sh http://test-debit.free.fr/512.rnd &
	
	i=0;
	while [ $i -lt $TIMEOUT ]
	do
		sleep 1
		if [ -f /tmp/wget.lck ]
		then
			i=`expr $i + 1`
		else
			i=$TIMEOUT
		fi
	done

	if [ $i -eq $TIMEOUT -a -f /tmp/wget.lck ]
	then
		#kill `cat /tmp/wget.lck`
		WGET_PID=`ps ux | grep wget | grep -v grep | awk '{print $2}'`
		[ -n "$WGET_PID" ] && kill $WGET_PID
		rm -f /tmp/wget.lck
		echo TIMEOUT
	elif [ -f /tmp/wget.result ]
	then
		RAW_SPEED=`cat /tmp/wget.result | grep saved | awk '{print $3,$4}'`
		if [ -z "$RAW_SPEED" ]
		then
			# Fallback test pour version localisee
			RAW_SPEED=`cat /tmp/wget.result | grep enregistr | awk '{print $3,$4}'`

			if [ -z "$RAW_SPEED" ]
			then
				RAW_SPEED=`cat /tmp/wget.result | grep sauvegar | awk '{print $3,$4}'`
			fi
		fi

		if [ -z "$RAW_SPEED" ]
		then
			echo "No Value (see /tmp/wget.result)"
		else
			echo $RAW_SPEED
		fi
	fi

	echo -e "Testing http://test-debit.free.fr/1024.rnd=\c"
	./fm_tester.sh http://test-debit.free.fr/1024.rnd &
	
	i=0;
	while [ $i -lt $TIMEOUT ]
	do
		sleep 1
		if [ -f /tmp/wget.lck ]
		then
			i=`expr $i + 1`
		else
			i=$TIMEOUT
		fi
	done

	if [ $i -eq $TIMEOUT -a -f /tmp/wget.lck ]
	then
		#kill `cat /tmp/wget.lck`
		WGET_PID=`ps ux | grep wget | grep -v grep | awk '{print $2}'`
		[ -n "$WGET_PID" ] && kill $WGET_PID
		rm -f /tmp/wget.lck
		echo TIMEOUT
	elif [ -f /tmp/wget.result ]
	then
		RAW_SPEED=`cat /tmp/wget.result | grep saved | awk '{print $3,$4}'`
		if [ -z "$RAW_SPEED" ]
		then
			# Fallback test
			RAW_SPEED=`cat /tmp/wget.result | grep enregistr | awk '{print $3,$4}'`

			if [ -z "$RAW_SPEED" ]
			then
				RAW_SPEED=`cat /tmp/wget.result | grep sauvegar | awk '{print $3,$4}'`
			fi
		fi

		if [ -z "$RAW_SPEED" ]
		then
			echo "No Value (see /tmp/wget.result)"
		else
			echo $RAW_SPEED
		fi
	fi

	# Etape 3 : tests HTTPS avec quelques extensions ===============================
	BASE_URL=https://1.testdebit.info/fichiers
	SIZES="10ko 100ko 1Mo"
	EXTENSIONS="iso txt doc docx xls xlsx html jpg jpeg png gif avi mpg mp4 vivien"

	echo "Testing HTTPS ---------------------"

	for SIZE in $SIZES
        do

                echo "Testing size=$SIZE"

                case $SIZE in
                1ko ) TIMEOUT=5;;
                10ko ) TIMEOUT=10;;
                100ko ) TIMEOUT=10;;
                1Mo ) TIMEOUT=20;;
                2Mo ) TIMEOUT=40;;
                esac

                for EXTENSION in $EXTENSIONS
                do
                        echo -e "$EXTENSION=\c"
                        ./fm_tester.sh ${BASE_URL}/${SIZE}/${SIZE}.${EXTENSION} &

                        i=0;
                        while [ $i -lt $TIMEOUT ]
                        do
                                sleep 1
                                if [ -f /tmp/wget.lck ]
                                then
                                        i=`expr $i + 1`
                                else
                                        i=$TIMEOUT
                                fi
                        done

                        if [ $i -eq $TIMEOUT -a -f /tmp/wget.lck ]
                        then
                                #kill `cat /tmp/wget.lck`
				WGET_PID=`ps ux | grep wget | grep -v grep | awk '{print $2}'`
				[ -n "$WGET_PID" ] && kill $WGET_PID
                                rm -f /tmp/wget.lck
                                echo TIMEOUT
                        elif [ -f /tmp/wget.result ]
                        then
                                RAW_SPEED=`cat /tmp/wget.result | grep saved | awk '{print $3,$4}'`
                                if [ -z "$RAW_SPEED" ]
                                then
                                        # Fallback test
                                        RAW_SPEED=`cat /tmp/wget.result | grep enregistr | awk '{print $3,$4}'`

                                        if [ -z "$RAW_SPEED" ]
                                        then
                                                RAW_SPEED=`cat /tmp/wget.result | grep sauvegar | awk '{print $3,$4}'`
                                        fi
                                fi

				if [ -z "$RAW_SPEED" ]
				then
					echo "No Value (see /tmp/wget.result)"
				else
					echo $RAW_SPEED
				fi
                        fi
                done
        done
fi