1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
#!/usr/bin/ruby
# encoding: UTF-8
#
# Program to receive backups and run rsync in receive mode. Must check that
# user as authorised by SSH is allowed to access particular directory.
$LOAD_PATH << '/usr/lib/byteback'
require 'trollop'
require 'byteback'
require 'byteback/restore'
include Byteback::Log
if ENV['SSH_ORIGINAL_COMMAND']
ARGV.concat(ENV['SSH_ORIGINAL_COMMAND'].split(' '))
end
byteback_host = ENV['BYTEBACK_HOST']
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 restores to be limited to the hostname we're connecting form
#
if (ARGV[0] == 'byteback-restore')
args = ["rsync"]
revision = nil
while((arg = ARGV.shift) != ".")
break if arg.nil?
verbose = arg if arg == "--verbose"
if arg == "--revision"
revision = ARGV.shift
else
args << arg
end
end
restore = Byteback::Restore.new(byteback_root)
restore.revision = revision if revision
restore.find(Byteback::Restore.decode_args(ARGV))
Dir.chdir(byteback_host)
restore.results.each do |r|
args << File.join(".", r.snapshot, r.path)
end
info(args.join(" "))
exec(*args)
elsif ARGV[0] == 'rsync'
ARGV[-1] = "#{byteback_root}/current"
exec(*ARGV)
elsif ARGV[0] == 'byteback-snapshot'
ARGV.concat(['--root', "#{byteback_root}"])
exec(*ARGV)
elsif ARGV[0] == "restore"
puts "** Your byteback package needs to be updated. Please update and try again."
exit(1)
end
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'
opt :list_all, 'Show all stored versions of a file'
opt :revision, 'Show backed up files in a certain revision.', :default => '*'
opt :restore, 'Perform a restoration operation', :type => :string
opt :complete, 'Mark current backup as complete'
end
error('Please only choose one mode') if opts[:ping] && opts[:complete]
if opts[:complete]
system('byteback-snapshot', '--root', byteback_root)
elsif opts[:list]
args = Byteback::Restore.decode_args(ARGV[1..-1])
restore = Byteback::Restore.new(byteback_root)
restore.revision = opts[:revision]
restore.find(args, :all => opts[:list_all], :verbose => opts[:verbose])
if restore.results.empty?
puts "** Sorry. There were no files matching:"
puts "--> "+args.join("\n--> ")
else
puts restore.list
end
exit(0)
elsif opts[:ping]
exit 0
else
STDERR.print "byteback-receive failed\n"
exit 9
end
|
