functions

#!/bin/bash
###################################################################
# Prey Windows Specific Functions - by Tomas Pollak (bootlog.org)
# URL: http://preyproject.com
# License: GPLv3
####################################################################

# in windows, prey always runs as the local user
run_as_current_user(){
	$1
}

get_pid(){
	processes | grep "$1" | sed 's/[a-z\. ]*\([0-9]*\).*/\1/'
}

############################################
# verification stuff
############################################

verify_installation(){

	# check prey service on
	log " -- Checking if Prey system service is running..."
	local cron_running=`is_process_running "cronsvc"`
	if [ -n "$cron_running" ]; then
		echo " -- Cron system service up and running. Good."
	else
		echo " !! Cron system service not found! Please make sure no antivirus is blocking Cron Service (cronsvc) from running. Optionally reinstall. :)"
	fi

	# check net framework
	log ' -- Checking for installed .NET Framework...'
	if [ "`is_net_framework_installed`" == 1 ]; then
		log " -- Required .NET version $required_net_version or newer is installed. All good."
	else
		log " !! .NET framework version $required_net_version or newer not installed!"
	fi
}

# returns 1 if it is
is_net_framework_installed() {
	local version=`echo $required_net_version | sed 's/\..\./\./'`
	local results=`reg query "HKLM\Software\Microsoft\NET Framework Setup\NDP" | sed -e '1,2d' -e 's/[^0-9.]//g' | grep -v "^$"`

	for result in `echo -e "$results"`; do
		if [[ "$result" == "$version" || `is_bigger_than $result $version` == 1 ]]; then
			echo 1
			break
		fi
	done
	return 0
}

############################################
# core network functions
############################################

get_gateway_ip() {
	if [ -z "$gateway_ip" ]; then
		gateway_ip=`ipconfig | grep Gateway | sed 's/^.*: //g' | grep -v '^$' | head -1`
	fi
}

get_internal_ip() {
	if [ -z "$internal_ip" ]; then
		internal_ip=`ipconfig | grep "Address" | grep -v "local" | sed 's/^.*: //g' | grep -v '^$' | head -1`
	fi
}

get_wifi_info() {
	if [ -n "$winxp" ]; then
		wifi_info=`autowcxp.exe -list`
	else
		if [ "`is_net_framework_installed`" == 1 ]; then
			wifi_info=`autowc.exe -list`
		else
			log " !! .NET framework version $required_net_version or newer not installed!"
		fi
	fi
}

try_to_connect() {
	if [ "`is_net_framework_installed`" == 1 ]; then
		autowc.exe -connect 2> /dev/null
	else
		log " !! .NET framework version $required_net_version or newer not installed!"
	fi
}

############################################
# module functions
############################################

get_delay_for(){
	local delay_var=$(($1*1))
	if [ "$delay_var" == "$1" ]; then # integer, minute
		echo $(($1*60*1000))
	else
		case "$1" in
		"hourly")
			echo 3600000 # 60 * 60 * 1000
			;;
		"daily")
			echo 86400000 # 24 * 60 * 60 * 1000
			;;
		"weekly")
			echo 604800000 # 7 * 24 * 60 * 60 * 1000
			;;
		# "monthly")
		# 	echo 18144000000 # 30 * 7 * 24 * 60 * 60 * 1000
		# 	;;
		esac
	fi
}

get_current_delay(){
	reg query "$prey_reg_key" //v "Delay" | grep Delay | sed "s/[^0-9]*//"
}

update_execution_delay(){
	reg add "$prey_reg_key" //v "Delay" //d "$new_delay" //f > /dev/null
}

############################################
# module functions
############################################

is_module_active(){
	cat "$base_path/modules/$1/active" 2> /dev/null
}

############################################
# updater-specific functions
############################################

reset_permissions(){
	log ' -- Resetting permissions...'
	if [ -n "$winxp" ]; then
		local command="echo y| cacls.exe"
		local readonly="//t //c //p Everyone:r //g Administrators:f"
		local fullaccess="//t //c //p Everyone:f"
	else
		local command='Icacls.exe'
		local readonly="//T //C //grant Everyone:R"
		local fullaccess="//T //C //grant Everyone:F"
	fi
	# perhaps we should iterate over the diff file and only apply this on the
	# modified files, but seems like overkill to me -- Tomas
	eval $command \"$base_path/*\" "$readonly" &> /dev/null
	eval $command \"$base_path/prey.log\" "$fullaccess" &> /dev/null
	# eval $command \"$platform_path\tmp\" "$fullaccess" &> /dev/null
}

pre_update_hook(){

	# stop the cron windows service
	TASKKILL //F //IM cronsvc.exe //T &> /dev/null

	# we also need to make sure the configurator is not running, so we can
	# also update it in case its necessary
	local configurator_running=`is_process_running 'prey-config.exe'`
	if [ -n "$configurator_running" ]; then
		log ' -- Forcing shutdown of Prey Configurator... Sorry!'
		TASKKILL //F //IM prey-config.exe //T &> /dev/null
	fi

	return $?
}

# lets restart prey so it refreshes with the new status
post_update_hook(){
	reset_permissions
	log ' -- Reloading Prey...'
	net start "Cron Service for Prey"
}

############################################
# Device creation parameters
############################################
get_pc_info(){

	pc_name=`hostname`
	if [ -n "$winxp" ]; then
		pc_os_version="xp"
	else
		local pc_os_number=`echo 'os get caption' | wmic 2> /dev/null | grep -i 'microsoft'`

		if [[ -n `echo "$pc_os_number" | grep -i 'vista'` ]]; then
			pc_os_version="vista"
		else
			pc_os_version="7"
		fi

	fi

  local chassis_type=`echo 'systemenclosure get ChassisTypes' | wmic 2> /dev/null | grep '{' | sed 's/[^0-9]//g'`
	if [[ $chassis_type == "8" || $chassis_type == "9" || $chassis_type == "10" || $chassis_type == "11" || $chassis_type == "14" ]]; then
		pc_type="Portable"
	else
		pc_type="Desktop"
	fi

}