#!/bin/sh
#
# The Qubes OS Project, http://www.qubes-os.org
#
# Copyright (C) 2010  Rafal Wojtczuk  <rafal@invisiblethingslab.com>
#
# This program 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 2
# of the License, or (at your option) any later version.
#
# This program 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 this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
#
#

if [ x"$1" = "x--with-progress" ] ; then
	DO_PROGRESS=1
	shift
else
	DO_PROGRESS=0
fi

if [ $# -lt 2 ] ; then
	echo usage: $0 '[--with-progress] dest_vmname file [file]+'
	exit 1
fi

VM="$1"
shift

if [ $DO_PROGRESS = 1 ] ; then
	SIZE=$(du --apparent-size -c "$@" | tail -1 | cut -f 1)
fi

export PROGRESS_FILE=$(mktemp)
/usr/lib/qubes/qvm-trigger-copy-to-vm $VM "$@"
while ! [ -s $PROGRESS_FILE ] ; do
	sleep 0.1
done

while true ; do
	read agentpid sentsize agentstatus < $PROGRESS_FILE
	if [ "x"$agentstatus = x ] ; then continue ; fi
	if ! [ -e /proc/$agentpid ] ; then break ; fi
	if [ "x"$agentstatus = xDONE ] ; then break ; fi
	CURRSIZE=$(($sentsize/1024))
	if [ $DO_PROGRESS = 1 ] ; then 
		echo -ne "\r sent $CURRSIZE/$SIZE KB                   " 
	fi
	sleep 0.4
done

rm -f $PROGRESS_FILE
if [ $DO_PROGRESS = 1 ] ; then
	echo
fi

if ! [ "x"$agentstatus = xDONE ] ; then 
	exit 1
else
	exit 0
fi