1 files changed, 19 insertions, 7 deletions
diff --git a/byteback-receive b/byteback-receive
index 1c8f4ae..9756978 100755
--- a/byteback-receive
+++ b/byteback-receive
@@ -3,13 +3,13 @@
 # Program to receive backups and run rsync in receive mode.  Must check that
 # user as authorised by SSH is allowed to access particular directory.
 
-#STDERR.print ARGV.inspect + "\n"
-
 $LOAD_PATH << '/usr/lib/byteback'
+
 require 'trollop'
 require 'byteback'
 include Byteback::Log
 
+
 if ENV['SSH_ORIGINAL_COMMAND']
 	ARGV.concat(ENV['SSH_ORIGINAL_COMMAND'].split(" "))
 end
@@ -20,11 +20,22 @@ fatal("BYTEBACK_HOST environment not set") unless byteback_host
 byteback_root = ENV['HOME'] + "/" + ENV["BYTEBACK_HOST"]
 fatal("#{byteback_root} does not exist") unless File.directory?(byteback_root)
 
-# force destination to be where we expect
 #
-if ARGV[0] == 'rsync'
-	ARGV[-1] = "#{byteback_root}/current"
-	exec(*ARGV)
+#  Force restores to be limited to the hostname we're connecting form
+#
+if ( ARGV[0] == 'restore' )
+     ARGV[0] = 'rsync'
+     a = []
+     ARGV.each do |tmp|
+        if ( tmp =~ /^\/(.*)/ )
+           tmp = "#{byteback_host}/#{$1.dup}"
+        end
+        a.push(tmp)
+     end
+     exec(*a)
+eslif ARGV[0] == 'rsync'
+        ARGV[-1] = "#{byteback_root}/current"
+  exec(*ARGV)
 elsif ARGV[0] == 'byteback-snapshot'
 	ARGV.concat(["--root", "#{byteback_root}"])
 	exec(*ARGV)
@@ -34,6 +45,7 @@ opts = Trollop::options do
 	opt :verbose, "Print diagnostics"
 	opt :ping, "Check connection parameters and exit"
 	opt :list, "Show backed up files matching the given pattern", :type => :string
+	opt :restore, "Perform a restoration operation", :type => :string
 	opt :complete, "Mark current backup as complete"
 end
 
@@ -41,7 +53,7 @@ error("Please only choose one mode") if opts[:ping] && opts[:complete]
 if opts[:complete]
 	system("byteback-snapshot", "--root", byteback_root)
 elsif opts[:list]
-	system( "find #{byteback_root} -print | grep #{opts[:list]}" )
+	system( "cd #{byteback_root} && find . -print | grep #{opts[:list]}" )
 	exit(0 )
 elsif opts[:ping]
 	exit 0