wcp/bin/v-add-backup-sftp-host at bbffe9fffb1f74e007b011f0e9b7d11f7bb6f736 · ghzhost/wcp · GitHub

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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
#!/bin/bash
# info: add backup sftp host
# options: HOST USERNAME [PASSWORD] [PATH] [PORT]
#
# The function adds sftp host for system backups

#----------------------------------------------------------#
#                    Variable&Function                     #
#----------------------------------------------------------#

# Argument defenition
sftp_host=$1
sftp_user=$2
sftp_password=${3-******}
sftp_path=${4-backup}
sftp_port=${5-22}

A3='******'

# Includes
source $VESTA/func/main.sh
source $VESTA/conf/vesta.conf

# Replace password with ******
if [[ $A3 != '******' ]]
then
	EVENT="${EVENT/$sftp_password/******}"
fi

# sftp command function
sftpc() {
expect -f "-" <<EOF "$@"
	set count 0
	spawn /usr/bin/sftp -o StrictHostKeyChecking=no -o Port=$sftp_port $sftp_user@$sftp_host
	expect {
		"password:" {
	    	send "$sftp_password\r"
	    	exp_continue
	  	}

	  	-re "Couldn't|(.*)disconnect|(.*)stalled|(.*)not found" {
			set count \$argc
			set output "Disconnected."
			set rc $E_FTP
			exp_continue
		}

		-re ".*denied.*(publickey|password)." {
			set output "Permission denied, wrong publickey or password."
			set rc $E_CONNECT
		}

	  	"sftp>" {
	  		if {\$count < \$argc} {
	  			set arg [lindex \$argv \$count]
	  			send "\$arg\r"
	  			incr count
	  		} else {
	  			send "exit\r"
	  			set output "Disconnected."
	  			if {[info exists rc] != 1} {
	  				set rc $OK
	  			}
	  		}
	  		exp_continue
		}

		timeout {
			set output "Connection timeout."
			set rc $E_CONNECT
		}
  	}

  	if {[info exists output] == 1} {
  		puts "\$output"
  	}

  	exit \$rc
EOF
}

#----------------------------------------------------------#
#                    Verifications                         #
#----------------------------------------------------------#

check_expect=$(which expect)
if [[ ! -n $check_expect ]]
then
	echo "Error: \"expect\" utility not found"
	log_event "$E_NOTEXIST" "$EVENT"
	exit $E_NOTEXIST
fi

check_args '2' "$#" "HOST USERNAME [PASSWORD] [PATH] [PORT]"

#----------------------------------------------------------#
#                       Action                             #
#----------------------------------------------------------#

# Checking network connection and write permissions
sftmpdir="$sftp_path/vst.bK76A9SUkt"
sftpc "mkdir $sftp_path" > /dev/null 2>&1
sftpc "mkdir $sftmpdir" "rmdir $sftmpdir"
rc=$?
if [[ "$rc" != 0 ]]
then
	case $rc in
        $E_CONNECT) echo "Error: can't login to sftp host";;
		$E_FTP) echo "Error: can't create temp folder on the sftp host";;
	esac
	log_event "$rc" "$EVENT"
	exit "$rc"
fi

# Adding sftp backup config file
echo "HOST='$sftp_host'
USERNAME='$sftp_user'
PASSWORD='$sftp_password'
BPATH='$sftp_path'
PORT='$sftp_port'
TIME='$TIME'
DATE='$DATE'" > $VESTA/conf/sftp.backup.conf
chmod 660 $VESTA/conf/sftp.backup.conf

#----------------------------------------------------------#
#                       Vesta                              #
#----------------------------------------------------------#

# Update vesta.conf
if [ -z "$(grep LANGUAGE $VESTA/conf/vesta.conf)" ]; then
    echo "BACKUP_SYSTEM='ftp'" >> $VESTA/conf/vesta.conf
else
    bckp=$(echo "$BACKUP_SYSTEM,sftp" |\
        sed "s/,/\n/g"|\
        sort -r -u |\
        sed "/^$/d"|\
        sed ':a;N;$!ba;s/\n/,/g')
    sed -i "s/BACKUP_SYSTEM=.*/BACKUP_SYSTEM='$bckp'/g" $VESTA/conf/vesta.conf
fi

# Logging
echo "$sftp_host/$sftp_path successfully added as backup destination"
log_event "$OK" "$EVENT"

exit