From a6478045da8ad1c50aebdde5dd1e34579a18ba40 Mon Sep 17 00:00:00 2001
From: fredrik <fredrik@959494ce-11ee-0310-bf91-de5d638817bd>
Date: Sat, 24 Jun 2006 02:25:06 +0000
Subject: [PATCH] TTH support and some other things.

git-svn-id: svn+ssh://svn.dolda2000.com/srv/svn/repos/src/doldaconnect@649 959494ce-11ee-0310-bf91-de5d638817bd
---
 config/dc-filter    | 59 ++++++++++++++++++++++++++++++++++-------------------
 config/dc-filtercmd |  6 +++---
 2 files changed, 41 insertions(+), 24 deletions(-)

diff --git a/config/dc-filter b/config/dc-filter
index 103f007..e3d0a26 100644
--- a/config/dc-filter
+++ b/config/dc-filter
@@ -1,25 +1,24 @@
-#!/bin/sh
+#!/bin/bash
+
 trap "" SIGHUP SIGPIPE
+
 if [ ! -d $HOME/dc ]; then mkdir $HOME/dc; fi
 if [ ! -d $HOME/dc/done ]; then mkdir $HOME/dc/done; fi
 if [ ! -d $HOME/dc/resume ]; then mkdir $HOME/dc/resume; fi
 if [ ! -d $HOME/dc/users ]; then mkdir $HOME/dc/users; fi
 exec 2>>$HOME/dc/filterlog
 cd $HOME/dc
-unset speedrec
-if which speedrec >/dev/null 2>&1; then
-    speedrec=speedrec
-fi
-if [ -z "$speedrec"]; then
-    for dir in /usr/libexec /usr/local/libexec; do
-	if [ -x "${dir}/speedrec" ]; then
-	    speedrec="${dir}/speedrec"
-	fi
-    done
-fi
-if [ -z "$speedrec" ]; then
+
+unset speedrecpath
+for dir in /usr/libexec /usr/local/libexec; do
+    if [ -x "${dir}/speedrec" ]; then
+	speedrecpath="${dir}/speedrec"
+    fi
+done
+if [ -z "$speedrecpath" ]; then
     echo "could not find speedrec - using cat instead" >&2
 fi
+
 maxsize=0
 unset resfile
 unset infofile
@@ -48,10 +47,12 @@ while [ -z "$resfile" -a "$found" = y ]; do
 	fi
     fi
 done
+
 unset "${!dcarg_@}"
 if [ -n "$resfile" ]; then
     . "${resfile}.info"
 fi
+
 origname="$1"
 shift
 filesize="$1"
@@ -65,6 +66,7 @@ while [ $# -gt 1 ]; do
     shift
     declare "dcarg_$rec"="$val"
 done
+
 if [ -z "$resfile" ]; then
     resfile="$(mktemp resume/resXXXXXX)"
     chmod 644 "$resfile"
@@ -72,23 +74,38 @@ if [ -z "$resfile" ]; then
     >"${resfile}.lock"
 fi
 declare -p origname filesize "${!dcarg_@}" >"${resfile}.info"
-echo "resume $maxsize"
-if [ -z "$speedrec" ]; then
-    cat >>"$resfile"
+
+echo "resume $maxsize" 2>/dev/null   # Avoid EPIPE warning if the server has already closed the transfer.
+
+if [ -z "$speedrecpath" ]; then
+    speedrec=(cat)
 else
+    speedrec=("$speedrecpath" "$HOME/dc/users/$(tr / _ <<<"$peername")")
     peerfile="$(tr / _ <<<"$peername")"
-    "$speedrec" "$HOME/dc/users/$peerfile" >>"$resfile"
 fi
+if ! which tthsum >/dev/null 2>&1; then
+    tthsum=(cat)
+else
+    tthsum=(tthsum -6 -F 3 -s "${resfile}.tthstate" -f)
+fi
+"${speedrec[@]}" | "${tthsum[@]}" 3>"${resfile}.tthfinal" >>"$resfile"
+
 size="$(wc -c <"$resfile")"
+
 if [ "$size" -eq 0 ]; then
-    rm -f "$resfile" "${resfile}.info" "${resfile}.lock"
+    rm -f "$resfile" "${resfile}.info" "${resfile}.lock" "${resfile}.tthstate" "${resfile}.tthfinal"
     exit 1
 fi
-rm -f "${resfile}.lock"
+
 if [ "$size" -lt "$filesize" ]; then
+    rm -f "${resfile}.lock" "${resfile}.tthfinal"
     exit 1   # Exit code != 0 means restart transfer
 fi
-rm -f "${resfile}.info"
+
+tth="$(cat "${resfile}.tthfinal")"
+if [ -z "$tth" ]; then unset tth; fi
+
+rm -f "${resfile}.info" "${resfile}.lock" "${resfile}.tthstate" "${resfile}.tthfinal"
 
 destdir="$HOME/dc/done"
 
@@ -98,7 +115,7 @@ if [ -e "$newname" ]; then
 fi
 mv "$resfile" "$newname"
 if [ -x $HOME/dc/complete ]; then
-    export "${!dcarg_@}"
+    export "${!dcarg_@}" tth
     exec $HOME/dc/complete "$newname"
 fi
 exit 0
diff --git a/config/dc-filtercmd b/config/dc-filtercmd
index b452e7d..e5e812b 100755
--- a/config/dc-filtercmd
+++ b/config/dc-filtercmd
@@ -34,8 +34,8 @@ if [ "$cmd" = rmtag ]; then
 	    unset dcarg_tag
 	    . "$f"
 	    if [ "$dcarg_tag" = "$1" ]; then
-		rm "$f"
-		rm "${f%.info}"
+		base="${f%.info}"
+		rm -f "$base" "${base}.info" "${base}.tthstate"
 	    fi
 	fi
     done
@@ -110,6 +110,6 @@ if [ "$cmd" = rmres ]; then
 	echo "err:locked"
 	exit
     fi
-    rm -f "$resfile" "${resfile}.info"
+    rm -f "$resfile" "${resfile}.info" "${resfile}.tthstate"
     echo "ok"
 fi
-- 
2.11.0