main.sh 40 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316
  1. #!/usr/bin/env bash
  2. # Internal variables
  3. HOMEDIR='/home'
  4. BACKUP='/backup'
  5. BACKUP_GZIP=9
  6. BACKUP_DISK_LIMIT=95
  7. BACKUP_LA_LIMIT=`cat /proc/cpuinfo | grep processor | wc -l`
  8. RRD_STEP=300
  9. BIN=$HESTIA/bin
  10. HESTIA_INSTALL_DIR=$HESTIA/install/deb
  11. HESTIA_BACKUP="/root/hst_backups/$(date +%d%m%Y%H%M)"
  12. USER_DATA=$HESTIA/data/users/$user
  13. WEBTPL=$HESTIA/data/templates/web
  14. MAILTPL=$HESTIA/data/templates/mail
  15. DNSTPL=$HESTIA/data/templates/dns
  16. RRD=$HESTIA/web/rrd
  17. SENDMAIL="$HESTIA/web/inc/mail-wrapper.php"
  18. HESTIA_GIT_REPO="https://raw.githubusercontent.com/hestiacp/hestiacp"
  19. HESTIA_THEMES="$HESTIA/web/css/themes"
  20. HESTIA_THEMES_CUSTOM="$HESTIA/web/css/themes/custom"
  21. SCRIPT="$(basename $0)"
  22. # Return codes
  23. OK=0
  24. E_ARGS=1
  25. E_INVALID=2
  26. E_NOTEXIST=3
  27. E_EXISTS=4
  28. E_SUSPENDED=5
  29. E_UNSUSPENDED=6
  30. E_INUSE=7
  31. E_LIMIT=8
  32. E_PASSWORD=9
  33. E_FORBIDEN=10
  34. E_DISABLED=11
  35. E_PARSING=12
  36. E_DISK=13
  37. E_LA=14
  38. E_CONNECT=15
  39. E_FTP=16
  40. E_DB=17
  41. E_RRD=18
  42. E_UPDATE=19
  43. E_RESTART=20
  44. # Detect operating system
  45. detect_os() {
  46. if [ -e "/etc/os-release" ]; then
  47. get_os_type=$(grep "^ID=" /etc/os-release | cut -f 2 -d '=')
  48. if [ "$get_os_type" = "ubuntu" ]; then
  49. if [ -e '/usr/bin/lsb_release' ]; then
  50. OS_VERSION="$(lsb_release -s -r)"
  51. OS_TYPE='Ubuntu'
  52. fi
  53. elif [ "$get_os_type" = "debian" ]; then
  54. OS_TYPE='Debian'
  55. OS_VERSION=$(cat /etc/debian_version|grep -o "[0-9]\{1,2\}"|head -n1)
  56. fi
  57. else
  58. OS_TYPE="Unsupported OS"
  59. OS_VERSION="Unknown"
  60. fi
  61. }
  62. # Generate time stamp
  63. new_timestamp() {
  64. time_n_date=$(date +'%T %F')
  65. time=$(echo "$time_n_date" |cut -f 1 -d \ )
  66. date=$(echo "$time_n_date" |cut -f 2 -d \ )
  67. }
  68. # Event string for logger
  69. ARGS=("$@")
  70. for ((I=1; I <= $# ; I++)); do
  71. if [[ "$HIDE" != $I ]]; then
  72. ARGUMENTS="$ARGUMENTS '${ARGS[${I}-1]}'"
  73. else
  74. ARGUMENTS="$ARGUMENTS '******'"
  75. fi
  76. done
  77. # Log event function
  78. log_event() {
  79. if [ -z "$time" ]; then
  80. LOG_TIME="$(date +'%F %T') $(basename $0)"
  81. else
  82. LOG_TIME="$date $time $(basename $0)"
  83. fi
  84. if [ "$1" -eq 0 ]; then
  85. echo "$LOG_TIME $2" >> $HESTIA/log/system.log
  86. else
  87. echo "$LOG_TIME $2 [Error $1]" >> $HESTIA/log/error.log
  88. fi
  89. }
  90. # Log user history
  91. log_history() {
  92. message=$1
  93. log_user=${3-$user}
  94. # Set default event level and category if not specified
  95. if [ -z "$event_level" ]; then
  96. event_level="Info"
  97. fi
  98. if [ -z "$event_category" ]; then
  99. event_category="System"
  100. fi
  101. # Log system events to system log file
  102. if [ "$log_user" = "system" ]; then
  103. log=$HESTIA/data/users/admin/system.log
  104. else
  105. if ! $BIN/v-list-user "$log_user" >/dev/null; then
  106. return $E_NOTEXIST
  107. fi
  108. log=$HESTIA/data/users/$log_user/history.log
  109. fi
  110. touch $log
  111. # TODO: Improve log pruning and pagination
  112. #
  113. #if [ '1000' -lt "$(wc -l $log |cut -f 1 -d ' ')" ]; then
  114. # tail -n 499 $log > $log.moved
  115. # mv -f $log.moved $log
  116. # chmod 660 $log
  117. #fi
  118. if [ -z "$date" ]; then
  119. time_n_date=$(date +'%T %F')
  120. time=$(echo "$time_n_date" |cut -f 1 -d \ )
  121. date=$(echo "$time_n_date" |cut -f 2 -d \ )
  122. fi
  123. curr_str=$(grep "ID=" $log | cut -f 2 -d \' | sort -n | tail -n1)
  124. id="$((curr_str +1))"
  125. echo "ID='$id' DATE='$date' TIME='$time' LEVEL='$event_level' CATEGORY='$event_category' MESSAGE='$message'" >> $log
  126. }
  127. # Result checker
  128. check_result() {
  129. if [ $1 -ne 0 ]; then
  130. echo "Error: $2"
  131. if [ ! -z "$3" ]; then
  132. log_event "$3" "$ARGUMENTS"
  133. exit $3
  134. else
  135. log_event "$1" "$ARGUMENTS"
  136. exit $1
  137. fi
  138. fi
  139. }
  140. # Argument list checker
  141. check_args() {
  142. if [ "$1" -gt "$2" ]; then
  143. echo "Usage: $(basename $0) $3"
  144. check_result $E_ARGS "not enought arguments" >/dev/null
  145. fi
  146. }
  147. # Subsystem checker
  148. is_system_enabled() {
  149. if [ -z "$1" ] || [ "$1" = no ]; then
  150. check_result $E_DISABLED "$2 is not enabled"
  151. fi
  152. }
  153. # User package check
  154. is_package_full() {
  155. case "$1" in
  156. WEB_DOMAINS) used=$(wc -l $USER_DATA/web.conf);;
  157. WEB_ALIASES) used=$(echo $aliases |tr ',' '\n' |wc -l);;
  158. DNS_DOMAINS) used=$(wc -l $USER_DATA/dns.conf);;
  159. DNS_RECORDS) used=$(wc -l $USER_DATA/dns/$domain.conf);;
  160. MAIL_DOMAINS) used=$(wc -l $USER_DATA/mail.conf);;
  161. MAIL_ACCOUNTS) used=$(wc -l $USER_DATA/mail/$domain.conf);;
  162. DATABASES) used=$(wc -l $USER_DATA/db.conf);;
  163. CRON_JOBS) used=$(wc -l $USER_DATA/cron.conf);;
  164. esac
  165. used=$(echo "$used"| cut -f 1 -d \ )
  166. limit=$(grep "^$1=" $USER_DATA/user.conf |cut -f 2 -d \')
  167. if [ "$1" = WEB_ALIASES ]; then
  168. # Used is always calculated with the new alias added
  169. if [ "$limit" != 'unlimited' ] && [[ "$used" -gt "$limit" ]]; then
  170. check_result $E_LIMIT "$1 limit is reached :: upgrade user package"
  171. fi
  172. else
  173. if [ "$limit" != 'unlimited' ] && [[ "$used" -ge "$limit" ]]; then
  174. check_result $E_LIMIT "$1 limit is reached :: upgrade user package"
  175. fi
  176. fi
  177. }
  178. # User owner for reseller plugin
  179. get_user_owner() {
  180. if [ -z "$RESELLER_KEY" ]; then
  181. owner='admin'
  182. else
  183. owner=$(grep "^OWNER" $USER_DATA/user.conf| cut -f 2 -d \')
  184. if [ -z "$owner" ]; then
  185. owner='admin'
  186. fi
  187. fi
  188. }
  189. # Random password generator
  190. generate_password() {
  191. matrix=$1
  192. length=$2
  193. if [ -z "$matrix" ]; then
  194. matrix="A-Za-z0-9"
  195. fi
  196. if [ -z "$length" ]; then
  197. length=16
  198. fi
  199. head /dev/urandom | tr -dc $matrix | head -c$length
  200. }
  201. # Package existence check
  202. is_package_valid() {
  203. if [ -z "$1" ]; then
  204. pkg_dir="$HESTIA/data/packages"
  205. fi
  206. if [ ! -e "$pkg_dir/$package.pkg" ]; then
  207. check_result $E_NOTEXIST "package $package doesn't exist"
  208. fi
  209. }
  210. # Validate system type
  211. is_type_valid() {
  212. if [ -z "$(echo $1 | grep -w $2)" ]; then
  213. check_result $E_INVALID "$2 type is invalid"
  214. fi
  215. }
  216. # Check user backup settings
  217. is_backup_enabled() {
  218. BACKUPS=$(grep "^BACKUPS=" $USER_DATA/user.conf | cut -f2 -d \')
  219. if [ -z "$BACKUPS" ] || [[ "$BACKUPS" -le '0' ]]; then
  220. check_result $E_DISABLED "user backup is disabled"
  221. fi
  222. }
  223. # Check user backup settings
  224. is_backup_scheduled() {
  225. if [ -e "$HESTIA/data/queue/backup.pipe" ]; then
  226. check_q=$(grep " $user " $HESTIA/data/queue/backup.pipe | grep $1)
  227. if [ ! -z "$check_q" ]; then
  228. check_result $E_EXISTS "$1 is already scheduled"
  229. fi
  230. fi
  231. }
  232. # Check if object is new
  233. is_object_new() {
  234. if [ $2 = 'USER' ]; then
  235. if [ -d "$USER_DATA" ]; then
  236. object="OK"
  237. fi
  238. else
  239. object=$(grep "$2='$3'" $USER_DATA/$1.conf)
  240. fi
  241. if [ ! -z "$object" ]; then
  242. check_result $E_EXISTS "$2=$3 already exists"
  243. fi
  244. }
  245. # Check if object is valid
  246. is_object_valid() {
  247. if [ $2 = 'USER' ]; then
  248. tstpath="$(readlink -f "$HESTIA/data/users/$3")"
  249. if [ "$(dirname "$tstpath")" != "$(readlink -f "$HESTIA/data/users")" ] || [ ! -d "$HESTIA/data/users/$3" ]; then
  250. check_result $E_NOTEXIST "$1 $3 doesn't exist"
  251. fi
  252. else
  253. object=$(grep "$2='$3'" $HESTIA/data/users/$user/$1.conf)
  254. if [ -z "$object" ]; then
  255. arg1=$(basename $1)
  256. arg2=$(echo $2 |tr '[:upper:]' '[:lower:]')
  257. check_result $E_NOTEXIST "$arg1 $arg2 $3 doesn't exist"
  258. fi
  259. fi
  260. }
  261. # Check if a object string with key values pairs has the correct format and load it afterwards
  262. parse_object_kv_list_non_eval() {
  263. local str
  264. local objkv obj_key obj_val
  265. local OLD_IFS="$IFS"
  266. str=${@//$'\n'/ }
  267. str=${str//\"/\\\"}
  268. str=${str//$/\\$}
  269. IFS=$'\n'
  270. # Extract and loop trough each key-value pair. (Regex test: https://regex101.com/r/eiMufk/5)
  271. for objkv in $( echo "$str" | perl -n -e "while(/\b([a-zA-Z]+[\w]*)='(.*?)'(\s|\$)/g) {print \$1.'='.\$2 . \"\n\" }" ); do
  272. if ! [[ "$objkv" =~ ^([[:alnum:]][_[:alnum:]]{0,64}[[:alnum:]])=(\'?[^\']+?\'?)?$ ]]; then
  273. check_result $E_INVALID "Invalid key value format [$objkv]"
  274. fi
  275. obj_key=${objkv%%=*} # strip everything after first '=' char
  276. obj_val=${objkv#*=} # strip everything before first '=' char
  277. declare -g $obj_key="$obj_val"
  278. done
  279. IFS="$OLD_IFS"
  280. }
  281. # Check if a object string with key values pairs has the correct format and load it afterwards
  282. parse_object_kv_list() {
  283. local str
  284. local objkv
  285. local suboutput
  286. local OLD_IFS="$IFS"
  287. str=${@//$'\n'/ }
  288. str=${str//\"/\\\"}
  289. str=${str//$/\\$}
  290. IFS=$'\n'
  291. suboutput=$(setpriv --clear-groups --reuid nobody --regid nogroup bash -c "PS4=''; set -xe; eval \"${str}\"" 2>&1)
  292. check_result $? "Invalid object format: ${str}" $E_INVALID
  293. for objkv in $suboutput; do
  294. if [[ "$objkv" =~ ^'eval ' ]]; then
  295. continue
  296. fi
  297. if ! [[ "$objkv" =~ ^([[:alnum:]][_[:alnum:]]{0,64}[[:alnum:]])=(\'?[^\']+?\'?)?$ ]]; then
  298. check_result $E_INVALID "Invalid key value format [$objkv]"
  299. fi
  300. eval "$objkv"
  301. done
  302. IFS="$OLD_IFS"
  303. }
  304. # Check if object is supended
  305. is_object_suspended() {
  306. if [ $2 = 'USER' ]; then
  307. spnd=$(cat $USER_DATA/$1.conf|grep "SUSPENDED='yes'")
  308. else
  309. spnd=$(grep "$2='$3'" $USER_DATA/$1.conf|grep "SUSPENDED='yes'")
  310. fi
  311. if [ -z "$spnd" ]; then
  312. check_result $E_UNSUSPENDED "$(basename $1) $3 is not suspended"
  313. fi
  314. }
  315. # Check if object is unsupended
  316. is_object_unsuspended() {
  317. if [ $2 = 'USER' ]; then
  318. spnd=$(cat $USER_DATA/$1.conf |grep "SUSPENDED='yes'")
  319. else
  320. spnd=$(grep "$2='$3'" $USER_DATA/$1.conf |grep "SUSPENDED='yes'")
  321. fi
  322. if [ ! -z "$spnd" ]; then
  323. check_result $E_SUSPENDED "$(basename $1) $3 is suspended"
  324. fi
  325. }
  326. # Check if object value is empty
  327. is_object_value_empty() {
  328. str=$(grep "$2='$3'" $USER_DATA/$1.conf)
  329. parse_object_kv_list "$str"
  330. eval value=$4
  331. if [ ! -z "$value" ] && [ "$value" != 'no' ]; then
  332. check_result $E_EXISTS "${4//$}=$value already exists"
  333. fi
  334. }
  335. # Check if object value is empty
  336. is_object_value_exist() {
  337. str=$(grep "$2='$3'" $USER_DATA/$1.conf)
  338. parse_object_kv_list "$str"
  339. eval value=$4
  340. if [ -z "$value" ] || [ "$value" = 'no' ]; then
  341. check_result $E_NOTEXIST "${4//$}=$value doesn't exist"
  342. fi
  343. }
  344. # Check if password is transmitted via file
  345. is_password_valid() {
  346. if [[ "$password" =~ ^/tmp/ ]]; then
  347. if [ -f "$password" ]; then
  348. password="$(head -n1 $password)"
  349. fi
  350. fi
  351. }
  352. # Check if hash is transmitted via file
  353. is_hash_valid() {
  354. if [[ "$hash" =~ ^/tmp/ ]]; then
  355. if [ -f "$hash" ]; then
  356. hash="$(head -n1 $hash)"
  357. fi
  358. fi
  359. }
  360. # Check if directory is a symlink
  361. is_dir_symlink() {
  362. if [[ -L "$1" ]]; then
  363. check_result $E_FORBIDEN "$1 directory is a symlink"
  364. fi
  365. }
  366. # Get object value
  367. get_object_value() {
  368. object=$(grep "$2='$3'" $USER_DATA/$1.conf)
  369. parse_object_kv_list "$object"
  370. eval echo $4
  371. }
  372. # Update object value
  373. update_object_value() {
  374. row=$(grep -nF "$2='$3'" $USER_DATA/$1.conf)
  375. lnr=$(echo $row | cut -f 1 -d ':')
  376. object=$(echo $row | sed "s/^$lnr://")
  377. parse_object_kv_list "$object"
  378. eval old="$4"
  379. old=$(echo "$old" | sed -e 's/\\/\\\\/g' -e 's/&/\\&/g' -e 's/\//\\\//g')
  380. new=$(echo "$5" | sed -e 's/\\/\\\\/g' -e 's/&/\\&/g' -e 's/\//\\\//g')
  381. sed -i "$lnr s/${4//$/}='${old//\*/\\*}'/${4//$/}='${new//\*/\\*}'/g" \
  382. $USER_DATA/$1.conf
  383. }
  384. # Add object key
  385. add_object_key() {
  386. row=$(grep -n "$2='$3'" $USER_DATA/$1.conf)
  387. lnr=$(echo $row | cut -f 1 -d ':')
  388. object=$(echo $row | sed "s/^$lnr://")
  389. if [ -z "$(echo $object |grep $4=)" ]; then
  390. eval old="$4"
  391. sed -i "$lnr s/$5='/$4='' $5='/" $USER_DATA/$1.conf
  392. fi
  393. }
  394. # Search objects
  395. search_objects() {
  396. OLD_IFS="$IFS"
  397. IFS=$'\n'
  398. for line in $(grep $2=\'$3\' $USER_DATA/$1.conf); do
  399. parse_object_kv_list "$line"
  400. eval echo \$$4
  401. done
  402. IFS="$OLD_IFS"
  403. }
  404. # Get user value
  405. get_user_value() {
  406. grep "^${1//$/}=" $USER_DATA/user.conf | head -1 | awk -F "'" '{print $2}'
  407. }
  408. # Update user value in user.conf
  409. update_user_value() {
  410. key="${2//$}"
  411. lnr=$(grep -n "^$key='" $HESTIA/data/users/$1/user.conf |cut -f 1 -d ':')
  412. if [ ! -z "$lnr" ]; then
  413. sed -i "$lnr d" $HESTIA/data/users/$1/user.conf
  414. sed -i "$lnr i\\$key='${3}'" $HESTIA/data/users/$1/user.conf
  415. fi
  416. }
  417. # Increase user counter
  418. increase_user_value() {
  419. key="${2//$}"
  420. factor="${3-1}"
  421. conf="$HESTIA/data/users/$1/user.conf"
  422. old=$(grep "$key=" $conf | cut -f 2 -d \')
  423. if [ -z "$old" ]; then
  424. old=0
  425. fi
  426. new=$((old + factor))
  427. sed -i "s/$key='$old'/$key='$new'/g" $conf
  428. }
  429. # Decrease user counter
  430. decrease_user_value() {
  431. key="${2//$}"
  432. factor="${3-1}"
  433. conf="$HESTIA/data/users/$1/user.conf"
  434. old=$(grep "$key=" $conf | cut -f 2 -d \')
  435. if [ -z "$old" ]; then
  436. old=0
  437. fi
  438. if [ "$old" -le 1 ]; then
  439. new=0
  440. else
  441. new=$((old - factor))
  442. fi
  443. if [ "$new" -lt 0 ]; then
  444. new=0
  445. fi
  446. sed -i "s/$key='$old'/$key='$new'/g" $conf
  447. }
  448. # Notify user
  449. send_notice() {
  450. topic=$1
  451. notice=$2
  452. if [ "$notify" = 'yes' ]; then
  453. touch $USER_DATA/notifications.conf
  454. chmod 660 $USER_DATA/notifications.conf
  455. time_n_date=$(date +'%T %F')
  456. time=$(echo "$time_n_date" |cut -f 1 -d \ )
  457. date=$(echo "$time_n_date" |cut -f 2 -d \ )
  458. nid=$(grep "NID=" $USER_DATA/notifications.conf |cut -f 2 -d \')
  459. nid=$(echo "$nid" |sort -n |tail -n1)
  460. if [ ! -z "$nid" ]; then
  461. nid="$((nid +1))"
  462. else
  463. nid=1
  464. fi
  465. str="NID='$nid' TOPIC='$topic' NOTICE='$notice' TYPE='$type'"
  466. str="$str ACK='no' TIME='$time' DATE='$date'"
  467. echo "$str" >> $USER_DATA/notifications.conf
  468. if [ -z "$(grep NOTIFICATIONS $USER_DATA/user.conf)" ]; then
  469. sed -i "s/^TIME/NOTIFICATIONS='yes'\nTIME/g" $USER_DATA/user.conf
  470. else
  471. update_user_value "$user" '$NOTIFICATIONS' "yes"
  472. fi
  473. fi
  474. }
  475. # Recalculate U_DISK value
  476. recalc_user_disk_usage() {
  477. u_usage=0
  478. if [ -f "$USER_DATA/web.conf" ]; then
  479. usage=0
  480. dusage=$(grep 'U_DISK=' $USER_DATA/web.conf |\
  481. awk -F "U_DISK='" '{print $2}' | cut -f 1 -d \')
  482. for disk_usage in $dusage; do
  483. usage=$((usage + disk_usage))
  484. done
  485. d=$(grep "U_DISK_WEB='" $USER_DATA/user.conf | cut -f 2 -d \')
  486. sed -i "s/U_DISK_WEB='$d'/U_DISK_WEB='$usage'/g" $USER_DATA/user.conf
  487. u_usage=$((u_usage + usage))
  488. fi
  489. if [ -f "$USER_DATA/mail.conf" ]; then
  490. usage=0
  491. dusage=$(grep 'U_DISK=' $USER_DATA/mail.conf |\
  492. awk -F "U_DISK='" '{print $2}' | cut -f 1 -d \')
  493. for disk_usage in $dusage; do
  494. usage=$((usage + disk_usage))
  495. done
  496. d=$(grep "U_DISK_MAIL='" $USER_DATA/user.conf | cut -f 2 -d \')
  497. sed -i "s/U_DISK_MAIL='$d'/U_DISK_MAIL='$usage'/g" $USER_DATA/user.conf
  498. u_usage=$((u_usage + usage))
  499. fi
  500. if [ -f "$USER_DATA/db.conf" ]; then
  501. usage=0
  502. dusage=$(grep 'U_DISK=' $USER_DATA/db.conf |\
  503. awk -F "U_DISK='" '{print $2}' | cut -f 1 -d \')
  504. for disk_usage in $dusage; do
  505. usage=$((usage + disk_usage))
  506. done
  507. d=$(grep "U_DISK_DB='" $USER_DATA/user.conf | cut -f 2 -d \')
  508. sed -i "s/U_DISK_DB='$d'/U_DISK_DB='$usage'/g" $USER_DATA/user.conf
  509. u_usage=$((u_usage + usage))
  510. fi
  511. usage=$(grep 'U_DISK_DIRS=' $USER_DATA/user.conf | cut -f 2 -d "'")
  512. u_usage=$((u_usage + usage))
  513. old=$(grep "U_DISK='" $USER_DATA/user.conf | cut -f 2 -d \')
  514. sed -i "s/U_DISK='$old'/U_DISK='$u_usage'/g" $USER_DATA/user.conf
  515. }
  516. # Recalculate U_BANDWIDTH value
  517. recalc_user_bandwidth_usage() {
  518. usage=0
  519. bandwidth_usage=$(grep 'U_BANDWIDTH=' $USER_DATA/web.conf |\
  520. awk -F "U_BANDWIDTH='" '{print $2}'|cut -f 1 -d \')
  521. for bandwidth in $bandwidth_usage; do
  522. usage=$((usage + bandwidth))
  523. done
  524. old=$(grep "U_BANDWIDTH='" $USER_DATA/user.conf | cut -f 2 -d \')
  525. sed -i "s/U_BANDWIDTH='$old'/U_BANDWIDTH='$usage'/g" $USER_DATA/user.conf
  526. }
  527. # Get next cron job id
  528. get_next_cronjob() {
  529. if [ -z "$job" ]; then
  530. curr_str=$(grep "JOB=" $USER_DATA/cron.conf|cut -f 2 -d \'|\
  531. sort -n|tail -n1)
  532. job="$((curr_str +1))"
  533. fi
  534. }
  535. # Sort cron jobs by id
  536. sort_cron_jobs() {
  537. cat $USER_DATA/cron.conf |sort -n -k 2 -t \' > $USER_DATA/cron.tmp
  538. mv -f $USER_DATA/cron.tmp $USER_DATA/cron.conf
  539. }
  540. # Sync cronjobs with system cron
  541. sync_cron_jobs() {
  542. source $USER_DATA/user.conf
  543. if [ -e "/var/spool/cron/crontabs" ]; then
  544. crontab="/var/spool/cron/crontabs/$user"
  545. else
  546. crontab="/var/spool/cron/$user"
  547. fi
  548. # remove file if exists
  549. if [ -e "$crontab" ]; then
  550. rm -f $crontab
  551. fi
  552. # touch new crontab file
  553. touch $crontab
  554. if [ "$CRON_REPORTS" = 'yes' ]; then
  555. echo "MAILTO=$CONTACT" > $crontab
  556. echo 'CONTENT_TYPE="text/plain; charset=utf-8"' >> $crontab
  557. fi
  558. while read line; do
  559. parse_object_kv_list "$line"
  560. if [ "$SUSPENDED" = 'no' ]; then
  561. echo "$MIN $HOUR $DAY $MONTH $WDAY $CMD" |\
  562. sed -e "s/%quote%/'/g" -e "s/%dots%/:/g" \
  563. >> $crontab
  564. fi
  565. done < $USER_DATA/cron.conf
  566. chown $user:$user $crontab
  567. chmod 600 $crontab
  568. }
  569. # User format validator
  570. is_user_format_valid() {
  571. if [ ${#1} -eq 1 ]; then
  572. if ! [[ "$1" =~ ^^[[:alnum:]]$ ]]; then
  573. check_result $E_INVALID "invalid $2 format :: $1"
  574. fi
  575. else
  576. if ! [[ "$1" =~ ^[[:alnum:]][-|\.|_[:alnum:]]{0,28}[[:alnum:]]$ ]]
  577. then
  578. check_result $E_INVALID "invalid $2 format :: $1"
  579. fi
  580. fi
  581. }
  582. # Domain format validator
  583. is_domain_format_valid() {
  584. object_name=${2-domain}
  585. exclude="[!|@|#|$|^|&|*|(|)|+|=|{|}|:|,|<|>|?|_|/|\|\"|'|;|%|\`| ]"
  586. if [[ $1 =~ $exclude ]] || [[ $1 =~ ^[0-9]+$ ]] || [[ $1 =~ "\.\." ]] || [[ $1 =~ "$(printf '\t')" ]] || [[ "$1" = "www" ]]; then
  587. check_result $E_INVALID "invalid $object_name format :: $1"
  588. fi
  589. }
  590. # Alias forman validator
  591. is_alias_format_valid() {
  592. for object in ${1//,/ }; do
  593. exclude="[!|@|#|$|^|&|(|)|+|=|{|}|:|<|>|?|_|/|\|\"|'|;|%|\`| ]"
  594. if [[ "$object" =~ $exclude ]]; then
  595. check_result $E_INVALID "invalid alias format :: $object"
  596. fi
  597. if [[ "$object" =~ [*] ]] && ! [[ "$object" =~ ^[*]\..* ]]; then
  598. check_result $E_INVALID "invalid alias format :: $object"
  599. fi
  600. done
  601. }
  602. # IP format validator
  603. is_ip_format_valid() {
  604. object_name=${2-ip}
  605. ip_regex='([1-9]?[0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])'
  606. ip_clean=$(echo "${1%/*}")
  607. if ! [[ $ip_clean =~ ^$ip_regex\.$ip_regex\.$ip_regex\.$ip_regex$ ]]; then
  608. check_result $E_INVALID "invalid $object_name format :: $1"
  609. fi
  610. if [ $1 != "$ip_clean" ]; then
  611. ip_cidr="$ip_clean/"
  612. ip_cidr=$(echo "${1#$ip_cidr}")
  613. if [[ "$ip_cidr" -gt 32 ]] || [[ "$ip_cidr" =~ [:alnum:] ]]; then
  614. check_result $E_INVALID "invalid $object_name format :: $1"
  615. fi
  616. fi
  617. }
  618. # IPv6 format validator
  619. is_ipv6_format_valid() {
  620. object_name=${2-ip6}
  621. ip_regex='([1-9]?[0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])'
  622. t_ip=$(echo $1 |awk -F / '{print $1}')
  623. t_cidr=$(echo $1 |awk -F / '{print $2}')
  624. valid_cidr=1
  625. WORD="[0-9A-Fa-f]\{1,4\}"
  626. # flat address, no compressed words
  627. FLAT="^${WORD}\(:${WORD}\)\{7\}$"
  628. COMP2="^\(${WORD}:\)\{1,1\}\(:${WORD}\)\{1,6\}$"
  629. COMP3="^\(${WORD}:\)\{1,2\}\(:${WORD}\)\{1,5\}$"
  630. COMP4="^\(${WORD}:\)\{1,3\}\(:${WORD}\)\{1,4\}$"
  631. COMP5="^\(${WORD}:\)\{1,4\}\(:${WORD}\)\{1,3\}$"
  632. COMP6="^\(${WORD}:\)\{1,5\}\(:${WORD}\)\{1,2\}$"
  633. COMP7="^\(${WORD}:\)\{1,6\}\(:${WORD}\)\{1,1\}$"
  634. # trailing :: edge case, includes case of only :: (all 0's)
  635. EDGE_TAIL="^\(\(${WORD}:\)\{1,7\}\|:\):$"
  636. # leading :: edge case
  637. EDGE_LEAD="^:\(:${WORD}\)\{1,7\}$"
  638. echo $t_ip | grep --silent "\(${FLAT}\)\|\(${COMP2}\)\|\(${COMP3}\)\|\(${COMP4}\)\|\(${COMP5}\)\|\(${COMP6}\)\|\(${COMP7}\)\|\(${EDGE_TAIL}\)\|\(${EDGE_LEAD}\)"
  639. if [ $? -ne 0 ]; then
  640. check_result $E_INVALID "invalid $object_name format :: $1"
  641. fi
  642. if [ ! -z "$(echo $1|grep '/')" ]; then
  643. if [[ "$t_cidr" -lt 0 ]] || [[ "$t_cidr" -gt 128 ]]; then
  644. valid_cidr=0
  645. fi
  646. if ! [[ "$t_cidr" =~ ^[0-9]+$ ]]; then
  647. valid_cidr=0
  648. fi
  649. fi
  650. if [ "$valid_cidr" -eq 0 ]; then
  651. check_result $E_INVALID "invalid $object_name format :: $1"
  652. fi
  653. }
  654. is_ip46_format_valid() {
  655. t_ip=$(echo $1 |awk -F / '{print $1}')
  656. t_cidr=$(echo $1 |awk -F / '{print $2}')
  657. valid_octets=0
  658. valid_cidr=1
  659. for octet in ${t_ip//./ }; do
  660. if [[ $octet =~ ^[0-9]{1,3}$ ]] && [[ $octet -le 255 ]]; then
  661. ((++valid_octets))
  662. fi
  663. done
  664. if [ ! -z "$(echo $1|grep '/')" ]; then
  665. if [[ "$t_cidr" -lt 0 ]] || [[ "$t_cidr" -gt 32 ]]; then
  666. valid_cidr=0
  667. fi
  668. if ! [[ "$t_cidr" =~ ^[0-9]+$ ]]; then
  669. valid_cidr=0
  670. fi
  671. fi
  672. if [ "$valid_octets" -lt 4 ] || [ "$valid_cidr" -eq 0 ]; then
  673. #Check IPV6
  674. ipv6_valid=""
  675. WORD="[0-9A-Fa-f]\{1,4\}"
  676. # flat address, no compressed words
  677. FLAT="^${WORD}\(:${WORD}\)\{7\}$"
  678. COMP2="^\(${WORD}:\)\{1,1\}\(:${WORD}\)\{1,6\}$"
  679. COMP3="^\(${WORD}:\)\{1,2\}\(:${WORD}\)\{1,5\}$"
  680. COMP4="^\(${WORD}:\)\{1,3\}\(:${WORD}\)\{1,4\}$"
  681. COMP5="^\(${WORD}:\)\{1,4\}\(:${WORD}\)\{1,3\}$"
  682. COMP6="^\(${WORD}:\)\{1,5\}\(:${WORD}\)\{1,2\}$"
  683. COMP7="^\(${WORD}:\)\{1,6\}\(:${WORD}\)\{1,1\}$"
  684. # trailing :: edge case, includes case of only :: (all 0's)
  685. EDGE_TAIL="^\(\(${WORD}:\)\{1,7\}\|:\):$"
  686. # leading :: edge case
  687. EDGE_LEAD="^:\(:${WORD}\)\{1,7\}$"
  688. echo $t_ip | grep --silent "\(${FLAT}\)\|\(${COMP2}\)\|\(${COMP3}\)\|\(${COMP4}\)\|\(${COMP5}\)\|\(${COMP6}\)\|\(${COMP7}\)\|\(${EDGE_TAIL}\)\|\(${EDGE_LEAD}\)"
  689. if [ $? -ne 0 ]; then
  690. ipv6_valid="INVALID"
  691. fi
  692. if [ ! -z "$(echo $1|grep '/')" ]; then
  693. if [[ "$t_cidr" -lt 0 ]] || [[ "$t_cidr" -gt 128 ]]; then
  694. valid_cidr=0
  695. fi
  696. if ! [[ "$t_cidr" =~ ^[0-9]+$ ]]; then
  697. valid_cidr=0
  698. fi
  699. fi
  700. if [ ! -z "$ipv6_valid" ] || [ "$valid_cidr" -eq 0 ]; then
  701. check_result $E_INVALID "invalid IP format :: $1"
  702. fi
  703. fi
  704. }
  705. # Proxy extention format validator
  706. is_extention_format_valid() {
  707. exclude="[!|#|$|^|&|(|)|+|=|{|}|:|@|<|>|?|/|\|\"|'|;|%|\`| ]"
  708. if [[ "$1" =~ $exclude ]]; then
  709. check_result $E_INVALID "invalid proxy extention format :: $1"
  710. fi
  711. }
  712. # Number format validator
  713. is_number_format_valid() {
  714. object_name=${2-number}
  715. if ! [[ "$1" =~ ^[0-9]+$ ]] ; then
  716. check_result $E_INVALID "invalid $object_name format :: $1"
  717. fi
  718. }
  719. # Autoreply format validator
  720. is_autoreply_format_valid() {
  721. if [ 10240 -le ${#1} ]; then
  722. check_result $E_INVALID "invalid autoreply format :: $1"
  723. fi
  724. }
  725. # Boolean format validator
  726. is_boolean_format_valid() {
  727. if [ "$1" != 'yes' ] && [ "$1" != 'no' ]; then
  728. check_result $E_INVALID "invalid $2 format :: $1"
  729. fi
  730. }
  731. # Common format validator
  732. is_common_format_valid() {
  733. exclude="[!|#|$|^|&|(|)|+|=|{|}|:|<|>|?|/|\|\"|'|;|%|\`| ]"
  734. if [[ "$1" =~ $exclude ]]; then
  735. check_result $E_INVALID "invalid $2 format :: $1"
  736. fi
  737. if [ 400 -le ${#1} ]; then
  738. check_result $E_INVALID "invalid $2 format :: $1"
  739. fi
  740. if [[ "$1" =~ @ ]] && [ ${#1} -gt 1 ] ; then
  741. check_result $E_INVALID "invalid $2 format :: $1"
  742. fi
  743. if [[ $1 =~ \* ]]; then
  744. if [[ "$(echo $1 | grep -o '\*\.' |wc -l)" -eq 0 ]] && [[ $1 != '*' ]] ; then
  745. check_result $E_INVALID "invalid $2 format :: $1"
  746. fi
  747. fi
  748. if [[ $(echo -n "$1" | tail -c 1) =~ [^a-zA-Z0-9_*@] ]]; then
  749. check_result $E_INVALID "invalid $2 format :: $1"
  750. fi
  751. if [[ $(echo -n "$1" | grep -c '\.\.') -gt 0 ]];then
  752. check_result $E_INVALID "invalid $2 format :: $1"
  753. fi
  754. if [[ $(echo -n "$1" | head -c 1) =~ [^a-zA-Z0-9_*@] ]]; then
  755. check_result $E_INVALID "invalid $2 format :: $1"
  756. fi
  757. if [[ $(echo -n "$1" | grep -c '\-\-') -gt 0 ]]; then
  758. check_result $E_INVALID "invalid $2 format :: $1"
  759. fi
  760. if [[ $(echo -n "$1" | grep -c '\_\_') -gt 0 ]]; then
  761. check_result $E_INVALID "invalid $2 format :: $1"
  762. fi
  763. }
  764. # Database format validator
  765. is_database_format_valid() {
  766. exclude="[!|@|#|$|^|&|*|(|)|+|=|{|}|:|,|<|>|?|/|\|\"|'|;|%|\`| ]"
  767. if [[ "$1" =~ $exclude ]] || [ 64 -le ${#1} ]; then
  768. check_result $E_INVALID "invalid $2 format :: $1"
  769. fi
  770. }
  771. # Date format validator
  772. is_date_format_valid() {
  773. if ! [[ "$1" =~ ^[0-9][0-9][0-9][0-9]-[0-9][0-9]-[0-9][0-9]$ ]]; then
  774. check_result $E_INVALID "invalid date format :: $1"
  775. fi
  776. }
  777. # Database user validator
  778. is_dbuser_format_valid() {
  779. exclude="[!|@|#|$|^|&|*|(|)|+|=|{|}|:|,|<|>|?|/|\|\"|'|;|%|\`| ]"
  780. if [ 33 -le ${#1} ]; then
  781. check_result $E_INVALID "mysql username can be up to 32 characters long"
  782. fi
  783. if [[ "$1" =~ $exclude ]]; then
  784. check_result $E_INVALID "invalid $2 format :: $1"
  785. fi
  786. }
  787. # DNS record type validator
  788. is_dns_type_format_valid() {
  789. known_dnstype='A,AAAA,NS,CNAME,MX,TXT,SRV,DNSKEY,KEY,IPSECKEY,PTR,SPF,TLSA,CAA'
  790. if [ -z "$(echo $known_dnstype |grep -w $1)" ]; then
  791. check_result $E_INVALID "invalid dns record type format :: $1"
  792. fi
  793. }
  794. # DNS record validator
  795. is_dns_record_format_valid() {
  796. if [ "$rtype" = 'A' ]; then
  797. is_ip_format_valid "$1"
  798. fi
  799. if [ "$rtype" = 'NS' ]; then
  800. is_domain_format_valid "${1::-1}" 'ns_record'
  801. fi
  802. if [ "$rtype" = 'MX' ]; then
  803. is_domain_format_valid "${1::-1}" 'mx_record'
  804. is_int_format_valid "$priority" 'priority_record'
  805. fi
  806. }
  807. # Email format validator
  808. is_email_format_valid() {
  809. if [[ ! "$1" =~ ^[A-Za-z0-9._%+-]+@[[:alnum:].-]+\.[A-Za-z]{2,63}$ ]] ; then
  810. check_result $E_INVALID "invalid email format :: $1"
  811. fi
  812. }
  813. # Firewall action validator
  814. is_fw_action_format_valid() {
  815. if [ "$1" != "ACCEPT" ] && [ "$1" != 'DROP' ] ; then
  816. check_result $E_INVALID "invalid action format :: $1"
  817. fi
  818. }
  819. # Firewall protocol validator
  820. is_fw_protocol_format_valid() {
  821. if [ "$1" != "ICMP" ] && [ "$1" != 'UDP' ] && [ "$1" != 'TCP' ] ; then
  822. check_result $E_INVALID "invalid protocol format :: $1"
  823. fi
  824. }
  825. # Firewall port validator
  826. is_fw_port_format_valid() {
  827. if [ "${#1}" -eq 1 ]; then
  828. if ! [[ "$1" =~ [0-9] ]]; then
  829. check_result $E_INVALID "invalid port format :: $1"
  830. fi
  831. else
  832. if ! [[ "$1" =~ ^[0-9][-|,|:|0-9]{0,30}[0-9]$ ]]
  833. then
  834. check_result $E_INVALID "invalid port format :: $1"
  835. fi
  836. fi
  837. }
  838. # Integer validator
  839. is_int_format_valid() {
  840. if ! [[ "$1" =~ ^[0-9]+$ ]] ; then
  841. check_result $E_INVALID "invalid $2 format :: $1"
  842. fi
  843. }
  844. # Interface validator
  845. is_interface_format_valid() {
  846. netdevices=$(cat /proc/net/dev |grep : |cut -f 1 -d : |tr -d ' ')
  847. if [ -z $(echo "$netdevices" |grep -x $1) ]; then
  848. check_result $E_INVALID "invalid interface format :: $1"
  849. fi
  850. }
  851. # IP status validator
  852. is_ip_status_format_valid() {
  853. if [ -z "$(echo shared,dedicated | grep -w $1 )" ]; then
  854. check_result $E_INVALID "invalid status format :: $1"
  855. fi
  856. }
  857. # Cron validator
  858. is_cron_format_valid() {
  859. limit=59
  860. check_format=''
  861. if [ "$2" = 'hour' ]; then
  862. limit=23
  863. fi
  864. if [ "$2" = 'day' ]; then
  865. limit=31
  866. fi
  867. if [ "$2" = 'month' ]; then
  868. limit=12
  869. fi
  870. if [ "$2" = 'wday' ]; then
  871. limit=7
  872. fi
  873. if [ "$1" = '*' ]; then
  874. check_format='ok'
  875. fi
  876. if [[ "$1" =~ ^[\*]+[/]+[0-9] ]]; then
  877. if [ "$(echo $1 |cut -f 2 -d /)" -lt $limit ]; then
  878. check_format='ok'
  879. fi
  880. fi
  881. if [[ "$1" =~ ^[0-9][-|,|0-9]{0,70}[\/][0-9]$ ]]; then
  882. check_format='ok'
  883. crn_values=${1//,/ }
  884. crn_values=${crn_values//-/ }
  885. crn_values=${crn_values//\// }
  886. for crn_vl in $crn_values; do
  887. if [ "$crn_vl" -gt $limit ]; then
  888. check_format='invalid'
  889. fi
  890. done
  891. fi
  892. crn_values=$(echo $1 |tr "," " " | tr "-" " ")
  893. for crn_vl in $crn_values
  894. do
  895. if [[ "$crn_vl" =~ ^[0-9]+$ ]] && [ "$crn_vl" -le $limit ]; then
  896. check_format='ok'
  897. fi
  898. done
  899. if [ "$check_format" != 'ok' ]; then
  900. check_result $E_INVALID "invalid $2 format :: $1"
  901. fi
  902. }
  903. # Name validator
  904. is_name_format_valid() {
  905. if ! [[ "$1" =~ ^[[:alnum:]][-|\ |\.|_[:alnum:]]{0,28}[[:alnum:]]$ ]]; then
  906. check_result $E_INVALID "invalid $2 format :: $1"
  907. fi
  908. }
  909. # Object validator
  910. is_object_format_valid() {
  911. if ! [[ "$1" =~ ^[[:alnum:]][-|\.|_[:alnum:]]{0,64}[[:alnum:]]$ ]]; then
  912. check_result $E_INVALID "invalid $2 format :: $1"
  913. fi
  914. }
  915. # Role validator
  916. is_role_valid (){
  917. if ! [[ "$1" =~ ^admin|user$ ]]; then
  918. check_result $E_INVALID "invalid $2 format :: $1"
  919. fi
  920. }
  921. # Password validator
  922. is_password_format_valid() {
  923. if [ "${#1}" -lt '6' ]; then
  924. check_result $E_INVALID "invalid password format :: $1"
  925. fi
  926. }
  927. # Missing function -
  928. # Before: validate_format_shell
  929. # After: is_format_valid_shell
  930. is_format_valid_shell() {
  931. if [ -z "$(grep -w $1 /etc/shells)" ]; then
  932. echo "Error: shell $1 is not valid"
  933. log_event "$E_INVALID" "$EVENT"
  934. exit $E_INVALID
  935. fi
  936. }
  937. # Service name validator
  938. is_service_format_valid() {
  939. if ! [[ "$1" =~ ^[[:alnum:]][-|\.|_[:alnum:]]{0,64}$ ]]; then
  940. check_result $E_INVALID "invalid $2 format :: $1"
  941. fi
  942. }
  943. is_hash_format_valid() {
  944. if ! [[ "$1" =~ ^[-_A-Za-z0-9]{1,32}$ ]]; then
  945. check_result $E_INVALID "invalid $2 format :: $1"
  946. fi
  947. }
  948. # Format validation controller
  949. is_format_valid() {
  950. for arg_name in $*; do
  951. eval arg=\$$arg_name
  952. if [ ! -z "$arg" ]; then
  953. case $arg_name in
  954. account) is_user_format_valid "$arg" "$arg_name";;
  955. action) is_fw_action_format_valid "$arg";;
  956. active) is_boolean_format_valid "$arg" 'active' ;;
  957. aliases) is_alias_format_valid "$arg" ;;
  958. antispam) is_boolean_format_valid "$arg" 'antispam' ;;
  959. antivirus) is_boolean_format_valid "$arg" 'antivirus' ;;
  960. autoreply) is_autoreply_format_valid "$arg" ;;
  961. backup) is_object_format_valid "$arg" 'backup' ;;
  962. charset) is_object_format_valid "$arg" "$arg_name" ;;
  963. charsets) is_common_format_valid "$arg" 'charsets' ;;
  964. comment) is_object_format_valid "$arg" 'comment' ;;
  965. database) is_database_format_valid "$arg" 'database';;
  966. day) is_cron_format_valid "$arg" $arg_name ;;
  967. dbpass) is_password_format_valid "$arg" ;;
  968. dbuser) is_dbuser_format_valid "$arg" 'dbuser';;
  969. dkim) is_boolean_format_valid "$arg" 'dkim' ;;
  970. dkim_size) is_int_format_valid "$arg" ;;
  971. domain) is_domain_format_valid "$arg" ;;
  972. dvalue) is_dns_record_format_valid "$arg";;
  973. email) is_email_format_valid "$arg" ;;
  974. email_forward) is_email_format_valid "$arg" ;;
  975. exp) is_date_format_valid "$arg" ;;
  976. extentions) is_common_format_valid "$arg" 'extentions' ;;
  977. ftp_password) is_password_format_valid "$arg" ;;
  978. ftp_user) is_user_format_valid "$arg" "$arg_name" ;;
  979. hash) is_hash_format_valid "$arg" "$arg_name" ;;
  980. host) is_object_format_valid "$arg" "$arg_name" ;;
  981. hour) is_cron_format_valid "$arg" $arg_name ;;
  982. id) is_int_format_valid "$arg" 'id' ;;
  983. iface) is_interface_format_valid "$arg" ;;
  984. ip) is_ip_format_valid "$arg" ;;
  985. ipv6) is_ipv6_format_valid "$arg" ;;
  986. ip46) is_ip46_format_valid "$arg" ;;
  987. ip_name) is_domain_format_valid "$arg" 'IP name';;
  988. ip_status) is_ip_status_format_valid "$arg" ;;
  989. job) is_int_format_valid "$arg" 'job' ;;
  990. key) is_user_format_valid "$arg" "$arg_name" ;;
  991. malias) is_user_format_valid "$arg" "$arg_name" ;;
  992. max_db) is_int_format_valid "$arg" 'max db';;
  993. min) is_cron_format_valid "$arg" $arg_name ;;
  994. month) is_cron_format_valid "$arg" $arg_name ;;
  995. name) is_name_format_valid "$arg" "name" ;;
  996. nat_ip) is_ip_format_valid "$arg" ;;
  997. netmask) is_ip_format_valid "$arg" 'netmask' ;;
  998. newid) is_int_format_valid "$arg" 'id' ;;
  999. ns1) is_domain_format_valid "$arg" 'ns1' ;;
  1000. ns2) is_domain_format_valid "$arg" 'ns2' ;;
  1001. ns3) is_domain_format_valid "$arg" 'ns3' ;;
  1002. ns4) is_domain_format_valid "$arg" 'ns4' ;;
  1003. ns5) is_domain_format_valid "$arg" 'ns5' ;;
  1004. ns6) is_domain_format_valid "$arg" 'ns6' ;;
  1005. ns7) is_domain_format_valid "$arg" 'ns7' ;;
  1006. ns8) is_domain_format_valid "$arg" 'ns8' ;;
  1007. object) is_name_format_valid "$arg" 'object';;
  1008. package) is_object_format_valid "$arg" "$arg_name" ;;
  1009. password) is_password_format_valid "$arg" ;;
  1010. port) is_int_format_valid "$arg" 'port' ;;
  1011. port_ext) is_fw_port_format_valid "$arg";;
  1012. protocol) is_fw_protocol_format_valid "$arg" ;;
  1013. proxy_ext) is_extention_format_valid "$arg" ;;
  1014. quota) is_int_format_valid "$arg" 'quota' ;;
  1015. record) is_common_format_valid "$arg" 'record';;
  1016. restart) is_boolean_format_valid "$arg" 'restart' ;;
  1017. role) is_role_valid "$arg" 'role' ;;
  1018. rtype) is_dns_type_format_valid "$arg" ;;
  1019. rule) is_int_format_valid "$arg" "rule id" ;;
  1020. service) is_service_format_valid "$arg" "$arg_name" ;;
  1021. soa) is_domain_format_valid "$arg" 'SOA' ;;
  1022. #missing command: is_format_valid_shell
  1023. shell) is_format_valid_shell "$arg" ;;
  1024. stats_pass) is_password_format_valid "$arg" ;;
  1025. stats_user) is_user_format_valid "$arg" "$arg_name" ;;
  1026. template) is_object_format_valid "$arg" "$arg_name" ;;
  1027. ttl) is_int_format_valid "$arg" 'ttl';;
  1028. user) is_user_format_valid "$arg" $arg_name;;
  1029. wday) is_cron_format_valid "$arg" $arg_name ;;
  1030. esac
  1031. fi
  1032. done
  1033. }
  1034. # Domain argument formatting
  1035. format_domain() {
  1036. if [[ "$domain" = *[![:ascii:]]* ]]; then
  1037. if [[ "$domain" =~ [[:upper:]] ]]; then
  1038. domain=$(echo "$domain" |sed 's/[[:upper:]].*/\L&/')
  1039. fi
  1040. else
  1041. if [[ "$domain" =~ [[:upper:]] ]]; then
  1042. domain=$(echo "$domain" |tr '[:upper:]' '[:lower:]')
  1043. fi
  1044. fi
  1045. if [[ "$domain" =~ ^www\..* ]]; then
  1046. domain=$(echo "$domain" |sed -e "s/^www.//")
  1047. fi
  1048. if [[ "$domain" =~ .*\.$ ]]; then
  1049. domain=$(echo "$domain" |sed -e "s/[.]*$//g")
  1050. fi
  1051. if [[ "$domain" =~ ^\. ]]; then
  1052. domain=$(echo "$domain" |sed -e "s/^[.]*//")
  1053. fi
  1054. }
  1055. format_domain_idn() {
  1056. if [ -z "$domain_idn" ]; then
  1057. domain_idn=$domain
  1058. fi
  1059. if [[ "$domain_idn" = *[![:ascii:]]* ]]; then
  1060. domain_idn=$(idn -t --quiet -a $domain_idn)
  1061. fi
  1062. }
  1063. format_aliases() {
  1064. if [ ! -z "$aliases" ] && [ "$aliases" != 'none' ]; then
  1065. aliases=$(echo $aliases |tr '[:upper:]' '[:lower:]' |tr ',' '\n')
  1066. aliases=$(echo "$aliases" |sed -e "s/\.$//" |sort -u)
  1067. aliases=$(echo "$aliases" |tr -s '.')
  1068. aliases=$(echo "$aliases" |sed -e "s/[.]*$//g")
  1069. aliases=$(echo "$aliases" |sed -e "s/^[.]*//")
  1070. aliases=$(echo "$aliases" |sed -e "/^$/d")
  1071. aliases=$(echo "$aliases" |tr '\n' ',' |sed -e "s/,$//")
  1072. fi
  1073. }
  1074. check_backup_conditions() {
  1075. # Checking load average
  1076. la=$(cat /proc/loadavg |cut -f 1 -d ' ' |cut -f 1 -d '.')
  1077. # i=0
  1078. while [ "$la" -ge "$BACKUP_LA_LIMIT" ]; do
  1079. echo -e "$(date "+%F %T") Load Average $la"
  1080. sleep 60
  1081. la=$(cat /proc/loadavg |cut -f 1 -d ' ' |cut -f 1 -d '.')
  1082. done
  1083. }
  1084. # Define download function
  1085. download_file() {
  1086. local url=$1
  1087. local destination=$2
  1088. local force=$3
  1089. # Default destination is the curent working directory
  1090. local dstopt=""
  1091. if [ ! -z "$(echo "$url" | grep -E "\.(gz|gzip|bz2|zip|xz)$")" ]; then
  1092. # When an archive file is downloaded it will be first saved localy
  1093. dstopt="--directory-prefix=$ARCHIVE_DIR"
  1094. local is_archive="true"
  1095. local filename="${url##*/}"
  1096. if [ -z "$filename" ]; then
  1097. >&2 echo "[!] No filename was found in url, exiting ($url)"
  1098. exit 1
  1099. fi
  1100. if [ ! -z "$force" ] && [ -f "$ARCHIVE_DIR/$filename" ]; then
  1101. rm -f $ARCHIVE_DIR/$filename
  1102. fi
  1103. elif [ ! -z "$destination" ]; then
  1104. # Plain files will be written to specified location
  1105. dstopt="-O $destination"
  1106. fi
  1107. # check for corrupted archive
  1108. if [ -f "$ARCHIVE_DIR/$filename" ] && [ "$is_archive" = "true" ]; then
  1109. tar -tzf "$ARCHIVE_DIR/$filename" > /dev/null 2>&1
  1110. if [ $? -ne 0 ]; then
  1111. >&2 echo "[!] Archive $ARCHIVE_DIR/$filename is corrupted, redownloading"
  1112. rm -f $ARCHIVE_DIR/$filename
  1113. fi
  1114. fi
  1115. if [ ! -f "$ARCHIVE_DIR/$filename" ]; then
  1116. wget $url -q $dstopt --show-progress --progress=bar:force --limit-rate=3m
  1117. fi
  1118. if [ ! -z "$destination" ] && [ "$is_archive" = "true" ]; then
  1119. if [ "$destination" = "-" ]; then
  1120. cat "$ARCHIVE_DIR/$filename"
  1121. elif [ -d "$(dirname $destination)" ]; then
  1122. cp "$ARCHIVE_DIR/$filename" "$destination"
  1123. fi
  1124. fi
  1125. }
  1126. check_hestia_demo_mode() {
  1127. demo_mode=$(grep DEMO_MODE /usr/local/hestia/conf/hestia.conf | cut -d '=' -f2 | sed "s|'||g")
  1128. if [ ! -z "$demo_mode" ] && [ "$demo_mode" = "yes" ]; then
  1129. echo "ERROR: Unable to perform operation due to security restrictions that are in place."
  1130. exit 1
  1131. fi
  1132. }
  1133. multiphp_count() {
  1134. echo $(ls -d /etc/php/*/fpm/pool.d 2>/dev/null |wc -l)
  1135. }
  1136. multiphp_versions() {
  1137. local -a php_versions_list;
  1138. local php_ver;
  1139. if [ "$(multiphp_count)" -gt 0 ] ; then
  1140. for php_ver in $(ls -v /etc/php/); do
  1141. [ ! -d "/etc/php/$php_ver/fpm/pool.d/" ] && continue
  1142. php_versions_list+=($php_ver)
  1143. done
  1144. echo "${php_versions_list[@]}"
  1145. fi
  1146. }
  1147. multiphp_default_version() {
  1148. # Get system wide default php version (set by update-alternatives)
  1149. local sys_phpversion=$(php -r "echo (float)phpversion();")
  1150. # Check if the system php also has php-fpm enabled, otherwise return
  1151. # the most recent php version which does have it installed.
  1152. if [ ! -d "/etc/php/$sys_phpversion/fpm/pool.d/" ]; then
  1153. local all_versions="$(multiphp_versions)"
  1154. if [ ! -z "$all_versions" ]; then
  1155. sys_phpversion="${all_versions##*\ }";
  1156. fi
  1157. fi
  1158. echo "$sys_phpversion"
  1159. }
  1160. is_hestia_package(){
  1161. if [ -z "$(echo $1 | grep -w $2)" ]; then
  1162. check_result $E_INVALID "$2 package is not controlled by hestiacp"
  1163. fi
  1164. }
  1165. # Run arbitrary cli commands with dropped privileges
  1166. # Note: setpriv --init-groups is not available on debian9 (util-linux 2.29.2)
  1167. # Input:
  1168. # - $user : Vaild hestia user
  1169. user_exec() {
  1170. is_object_valid 'user' 'USER' "$user"
  1171. local user_groups=$(id -G "$user")
  1172. user_groups=${user_groups//\ /,}
  1173. setpriv --groups "$user_groups" --reuid "$user" --regid "$user" -- $@
  1174. }
  1175. # Simple chmod wrapper that skips symlink files after glob expand
  1176. no_symlink_chmod() {
  1177. local filemode=$1; shift;
  1178. for i in "$@"; do
  1179. [[ -L ${i} ]] && continue
  1180. chmod "${filemode}" "${i}"
  1181. done
  1182. }