#!/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
#
# 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

	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
	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 &

			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`
				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

				# Following section convert into bits/s
				SPEED=`echo $RAW_SPEED | awk '{print $1}' | tr '(' ' '`
				UNIT=`echo $RAW_SPEED | grep KB`
				if [ -n "$UNIT" ]
				then
					# KB
					WGET_SPEED=`echo $SPEED*1024 | bc`
				else
					UNIT=`echo $RAW_SPEED | grep MB`
					if [ -n "$UNIT" ]
					then
						WGET_SPEED=`echo $SPEED*1048576 | bc`
					fi
				fi

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

	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`
		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

		# Following section convert into bits/s
		SPEED=`echo $RAW_SPEED | awk '{print $1}' | tr '(' ' '`
		UNIT=`echo $RAW_SPEED | grep KB`
		if [ -n "$UNIT" ]
		then
			# KB
			WGET_SPEED=`echo $SPEED*1024 | bc`
		else
			UNIT=`echo $RAW_SPEED | grep MB`
			if [ -n "$UNIT" ]
			then
				WGET_SPEED=`echo $SPEED*1048576 | bc`
			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`
		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

		# Following section convert into bits/s
		SPEED=`echo $RAW_SPEED | awk '{print $1}' | tr '(' ' '`
		UNIT=`echo $RAW_SPEED | grep KB`
		if [ -n "$UNIT" ]
		then
			# KB
			WGET_SPEED=`echo $SPEED*1024 | bc`
		else
			UNIT=`echo $RAW_SPEED | grep MB`
			if [ -n "$UNIT" ]
			then
				WGET_SPEED=`echo $SPEED*1048576 | bc`
			fi
		fi

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

	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"

	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`
                                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

				# Following section convert into bits/s
                                SPEED=`echo $RAW_SPEED | awk '{print $1}' | tr '(' ' '`
                                UNIT=`echo $RAW_SPEED | grep KB`
                                if [ -n "$UNIT" ]
                                then
                                        # KB
                                        WGET_SPEED=`echo $SPEED*1024 | bc`
                                else
                                        UNIT=`echo $RAW_SPEED | grep MB`
                                        if [ -n "$UNIT" ]
                                        then
                                                WGET_SPEED=`echo $SPEED*1048576 | bc`
                                        fi
                                fi

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