main.sh 40 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318
  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. else
  558. echo 'MAILTO=""' > $crontab
  559. fi
  560. while read line; do
  561. parse_object_kv_list "$line"
  562. if [ "$SUSPENDED" = 'no' ]; then
  563. echo "$MIN $HOUR $DAY $MONTH $WDAY $CMD" |\
  564. sed -e "s/%quote%/'/g" -e "s/%dots%/:/g" \
  565. >> $crontab
  566. fi
  567. done < $USER_DATA/cron.conf
  568. chown $user:$user $crontab
  569. chmod 600 $crontab
  570. }
  571. # User format validator
  572. is_user_format_valid() {
  573. if [ ${#1} -eq 1 ]; then
  574. if ! [[ "$1" =~ ^^[[:alnum:]]$ ]]; then
  575. check_result $E_INVALID "invalid $2 format :: $1"
  576. fi
  577. else
  578. if ! [[ "$1" =~ ^[[:alnum:]][-|\.|_[:alnum:]]{0,28}[[:alnum:]]$ ]]
  579. then
  580. check_result $E_INVALID "invalid $2 format :: $1"
  581. fi
  582. fi
  583. }
  584. # Domain format validator
  585. is_domain_format_valid() {
  586. object_name=${2-domain}
  587. exclude="[!|@|#|$|^|&|*|(|)|+|=|{|}|:|,|<|>|?|_|/|\|\"|'|;|%|\`| ]"
  588. if [[ $1 =~ $exclude ]] || [[ $1 =~ ^[0-9]+$ ]] || [[ $1 =~ "\.\." ]] || [[ $1 =~ "$(printf '\t')" ]] || [[ "$1" = "www" ]]; then
  589. check_result $E_INVALID "invalid $object_name format :: $1"
  590. fi
  591. }
  592. # Alias forman validator
  593. is_alias_format_valid() {
  594. for object in ${1//,/ }; do
  595. exclude="[!|@|#|$|^|&|(|)|+|=|{|}|:|<|>|?|_|/|\|\"|'|;|%|\`| ]"
  596. if [[ "$object" =~ $exclude ]]; then
  597. check_result $E_INVALID "invalid alias format :: $object"
  598. fi
  599. if [[ "$object" =~ [*] ]] && ! [[ "$object" =~ ^[*]\..* ]]; then
  600. check_result $E_INVALID "invalid alias format :: $object"
  601. fi
  602. done
  603. }
  604. # IP format validator
  605. is_ip_format_valid() {
  606. object_name=${2-ip}
  607. ip_regex='([1-9]?[0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])'
  608. ip_clean=$(echo "${1%/*}")
  609. if ! [[ $ip_clean =~ ^$ip_regex\.$ip_regex\.$ip_regex\.$ip_regex$ ]]; then
  610. check_result $E_INVALID "invalid $object_name format :: $1"
  611. fi
  612. if [ $1 != "$ip_clean" ]; then
  613. ip_cidr="$ip_clean/"
  614. ip_cidr=$(echo "${1#$ip_cidr}")
  615. if [[ "$ip_cidr" -gt 32 ]] || [[ "$ip_cidr" =~ [:alnum:] ]]; then
  616. check_result $E_INVALID "invalid $object_name format :: $1"
  617. fi
  618. fi
  619. }
  620. # IPv6 format validator
  621. is_ipv6_format_valid() {
  622. object_name=${2-ip6}
  623. ip_regex='([1-9]?[0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])'
  624. t_ip=$(echo $1 |awk -F / '{print $1}')
  625. t_cidr=$(echo $1 |awk -F / '{print $2}')
  626. valid_cidr=1
  627. WORD="[0-9A-Fa-f]\{1,4\}"
  628. # flat address, no compressed words
  629. FLAT="^${WORD}\(:${WORD}\)\{7\}$"
  630. COMP2="^\(${WORD}:\)\{1,1\}\(:${WORD}\)\{1,6\}$"
  631. COMP3="^\(${WORD}:\)\{1,2\}\(:${WORD}\)\{1,5\}$"
  632. COMP4="^\(${WORD}:\)\{1,3\}\(:${WORD}\)\{1,4\}$"
  633. COMP5="^\(${WORD}:\)\{1,4\}\(:${WORD}\)\{1,3\}$"
  634. COMP6="^\(${WORD}:\)\{1,5\}\(:${WORD}\)\{1,2\}$"
  635. COMP7="^\(${WORD}:\)\{1,6\}\(:${WORD}\)\{1,1\}$"
  636. # trailing :: edge case, includes case of only :: (all 0's)
  637. EDGE_TAIL="^\(\(${WORD}:\)\{1,7\}\|:\):$"
  638. # leading :: edge case
  639. EDGE_LEAD="^:\(:${WORD}\)\{1,7\}$"
  640. echo $t_ip | grep --silent "\(${FLAT}\)\|\(${COMP2}\)\|\(${COMP3}\)\|\(${COMP4}\)\|\(${COMP5}\)\|\(${COMP6}\)\|\(${COMP7}\)\|\(${EDGE_TAIL}\)\|\(${EDGE_LEAD}\)"
  641. if [ $? -ne 0 ]; then
  642. check_result $E_INVALID "invalid $object_name format :: $1"
  643. fi
  644. if [ ! -z "$(echo $1|grep '/')" ]; then
  645. if [[ "$t_cidr" -lt 0 ]] || [[ "$t_cidr" -gt 128 ]]; then
  646. valid_cidr=0
  647. fi
  648. if ! [[ "$t_cidr" =~ ^[0-9]+$ ]]; then
  649. valid_cidr=0
  650. fi
  651. fi
  652. if [ "$valid_cidr" -eq 0 ]; then
  653. check_result $E_INVALID "invalid $object_name format :: $1"
  654. fi
  655. }
  656. is_ip46_format_valid() {
  657. t_ip=$(echo $1 |awk -F / '{print $1}')
  658. t_cidr=$(echo $1 |awk -F / '{print $2}')
  659. valid_octets=0
  660. valid_cidr=1
  661. for octet in ${t_ip//./ }; do
  662. if [[ $octet =~ ^[0-9]{1,3}$ ]] && [[ $octet -le 255 ]]; then
  663. ((++valid_octets))
  664. fi
  665. done
  666. if [ ! -z "$(echo $1|grep '/')" ]; then
  667. if [[ "$t_cidr" -lt 0 ]] || [[ "$t_cidr" -gt 32 ]]; then
  668. valid_cidr=0
  669. fi
  670. if ! [[ "$t_cidr" =~ ^[0-9]+$ ]]; then
  671. valid_cidr=0
  672. fi
  673. fi
  674. if [ "$valid_octets" -lt 4 ] || [ "$valid_cidr" -eq 0 ]; then
  675. #Check IPV6
  676. ipv6_valid=""
  677. WORD="[0-9A-Fa-f]\{1,4\}"
  678. # flat address, no compressed words
  679. FLAT="^${WORD}\(:${WORD}\)\{7\}$"
  680. COMP2="^\(${WORD}:\)\{1,1\}\(:${WORD}\)\{1,6\}$"
  681. COMP3="^\(${WORD}:\)\{1,2\}\(:${WORD}\)\{1,5\}$"
  682. COMP4="^\(${WORD}:\)\{1,3\}\(:${WORD}\)\{1,4\}$"
  683. COMP5="^\(${WORD}:\)\{1,4\}\(:${WORD}\)\{1,3\}$"
  684. COMP6="^\(${WORD}:\)\{1,5\}\(:${WORD}\)\{1,2\}$"
  685. COMP7="^\(${WORD}:\)\{1,6\}\(:${WORD}\)\{1,1\}$"
  686. # trailing :: edge case, includes case of only :: (all 0's)
  687. EDGE_TAIL="^\(\(${WORD}:\)\{1,7\}\|:\):$"
  688. # leading :: edge case
  689. EDGE_LEAD="^:\(:${WORD}\)\{1,7\}$"
  690. echo $t_ip | grep --silent "\(${FLAT}\)\|\(${COMP2}\)\|\(${COMP3}\)\|\(${COMP4}\)\|\(${COMP5}\)\|\(${COMP6}\)\|\(${COMP7}\)\|\(${EDGE_TAIL}\)\|\(${EDGE_LEAD}\)"
  691. if [ $? -ne 0 ]; then
  692. ipv6_valid="INVALID"
  693. fi
  694. if [ ! -z "$(echo $1|grep '/')" ]; then
  695. if [[ "$t_cidr" -lt 0 ]] || [[ "$t_cidr" -gt 128 ]]; then
  696. valid_cidr=0
  697. fi
  698. if ! [[ "$t_cidr" =~ ^[0-9]+$ ]]; then
  699. valid_cidr=0
  700. fi
  701. fi
  702. if [ ! -z "$ipv6_valid" ] || [ "$valid_cidr" -eq 0 ]; then
  703. check_result $E_INVALID "invalid IP format :: $1"
  704. fi
  705. fi
  706. }
  707. # Proxy extention format validator
  708. is_extention_format_valid() {
  709. exclude="[!|#|$|^|&|(|)|+|=|{|}|:|@|<|>|?|/|\|\"|'|;|%|\`| ]"
  710. if [[ "$1" =~ $exclude ]]; then
  711. check_result $E_INVALID "invalid proxy extention format :: $1"
  712. fi
  713. }
  714. # Number format validator
  715. is_number_format_valid() {
  716. object_name=${2-number}
  717. if ! [[ "$1" =~ ^[0-9]+$ ]] ; then
  718. check_result $E_INVALID "invalid $object_name format :: $1"
  719. fi
  720. }
  721. # Autoreply format validator
  722. is_autoreply_format_valid() {
  723. if [ 10240 -le ${#1} ]; then
  724. check_result $E_INVALID "invalid autoreply format :: $1"
  725. fi
  726. }
  727. # Boolean format validator
  728. is_boolean_format_valid() {
  729. if [ "$1" != 'yes' ] && [ "$1" != 'no' ]; then
  730. check_result $E_INVALID "invalid $2 format :: $1"
  731. fi
  732. }
  733. # Common format validator
  734. is_common_format_valid() {
  735. exclude="[!|#|$|^|&|(|)|+|=|{|}|:|<|>|?|/|\|\"|'|;|%|\`| ]"
  736. if [[ "$1" =~ $exclude ]]; then
  737. check_result $E_INVALID "invalid $2 format :: $1"
  738. fi
  739. if [ 400 -le ${#1} ]; then
  740. check_result $E_INVALID "invalid $2 format :: $1"
  741. fi
  742. if [[ "$1" =~ @ ]] && [ ${#1} -gt 1 ] ; then
  743. check_result $E_INVALID "invalid $2 format :: $1"
  744. fi
  745. if [[ $1 =~ \* ]]; then
  746. if [[ "$(echo $1 | grep -o '\*\.' |wc -l)" -eq 0 ]] && [[ $1 != '*' ]] ; then
  747. check_result $E_INVALID "invalid $2 format :: $1"
  748. fi
  749. fi
  750. if [[ $(echo -n "$1" | tail -c 1) =~ [^a-zA-Z0-9_*@] ]]; then
  751. check_result $E_INVALID "invalid $2 format :: $1"
  752. fi
  753. if [[ $(echo -n "$1" | grep -c '\.\.') -gt 0 ]];then
  754. check_result $E_INVALID "invalid $2 format :: $1"
  755. fi
  756. if [[ $(echo -n "$1" | head -c 1) =~ [^a-zA-Z0-9_*@] ]]; then
  757. check_result $E_INVALID "invalid $2 format :: $1"
  758. fi
  759. if [[ $(echo -n "$1" | grep -c '\-\-') -gt 0 ]]; then
  760. check_result $E_INVALID "invalid $2 format :: $1"
  761. fi
  762. if [[ $(echo -n "$1" | grep -c '\_\_') -gt 0 ]]; then
  763. check_result $E_INVALID "invalid $2 format :: $1"
  764. fi
  765. }
  766. # Database format validator
  767. is_database_format_valid() {
  768. exclude="[!|@|#|$|^|&|*|(|)|+|=|{|}|:|,|<|>|?|/|\|\"|'|;|%|\`| ]"
  769. if [[ "$1" =~ $exclude ]] || [ 64 -le ${#1} ]; then
  770. check_result $E_INVALID "invalid $2 format :: $1"
  771. fi
  772. }
  773. # Date format validator
  774. is_date_format_valid() {
  775. if ! [[ "$1" =~ ^[0-9][0-9][0-9][0-9]-[0-9][0-9]-[0-9][0-9]$ ]]; then
  776. check_result $E_INVALID "invalid date format :: $1"
  777. fi
  778. }
  779. # Database user validator
  780. is_dbuser_format_valid() {
  781. exclude="[!|@|#|$|^|&|*|(|)|+|=|{|}|:|,|<|>|?|/|\|\"|'|;|%|\`| ]"
  782. if [ 33 -le ${#1} ]; then
  783. check_result $E_INVALID "mysql username can be up to 32 characters long"
  784. fi
  785. if [[ "$1" =~ $exclude ]]; then
  786. check_result $E_INVALID "invalid $2 format :: $1"
  787. fi
  788. }
  789. # DNS record type validator
  790. is_dns_type_format_valid() {
  791. known_dnstype='A,AAAA,NS,CNAME,MX,TXT,SRV,DNSKEY,KEY,IPSECKEY,PTR,SPF,TLSA,CAA'
  792. if [ -z "$(echo $known_dnstype |grep -w $1)" ]; then
  793. check_result $E_INVALID "invalid dns record type format :: $1"
  794. fi
  795. }
  796. # DNS record validator
  797. is_dns_record_format_valid() {
  798. if [ "$rtype" = 'A' ]; then
  799. is_ip_format_valid "$1"
  800. fi
  801. if [ "$rtype" = 'NS' ]; then
  802. is_domain_format_valid "${1::-1}" 'ns_record'
  803. fi
  804. if [ "$rtype" = 'MX' ]; then
  805. is_domain_format_valid "${1::-1}" 'mx_record'
  806. is_int_format_valid "$priority" 'priority_record'
  807. fi
  808. }
  809. # Email format validator
  810. is_email_format_valid() {
  811. if [[ ! "$1" =~ ^[A-Za-z0-9._%+-]+@[[:alnum:].-]+\.[A-Za-z]{2,63}$ ]] ; then
  812. check_result $E_INVALID "invalid email format :: $1"
  813. fi
  814. }
  815. # Firewall action validator
  816. is_fw_action_format_valid() {
  817. if [ "$1" != "ACCEPT" ] && [ "$1" != 'DROP' ] ; then
  818. check_result $E_INVALID "invalid action format :: $1"
  819. fi
  820. }
  821. # Firewall protocol validator
  822. is_fw_protocol_format_valid() {
  823. if [ "$1" != "ICMP" ] && [ "$1" != 'UDP' ] && [ "$1" != 'TCP' ] ; then
  824. check_result $E_INVALID "invalid protocol format :: $1"
  825. fi
  826. }
  827. # Firewall port validator
  828. is_fw_port_format_valid() {
  829. if [ "${#1}" -eq 1 ]; then
  830. if ! [[ "$1" =~ [0-9] ]]; then
  831. check_result $E_INVALID "invalid port format :: $1"
  832. fi
  833. else
  834. if ! [[ "$1" =~ ^[0-9][-|,|:|0-9]{0,30}[0-9]$ ]]
  835. then
  836. check_result $E_INVALID "invalid port format :: $1"
  837. fi
  838. fi
  839. }
  840. # Integer validator
  841. is_int_format_valid() {
  842. if ! [[ "$1" =~ ^[0-9]+$ ]] ; then
  843. check_result $E_INVALID "invalid $2 format :: $1"
  844. fi
  845. }
  846. # Interface validator
  847. is_interface_format_valid() {
  848. netdevices=$(cat /proc/net/dev |grep : |cut -f 1 -d : |tr -d ' ')
  849. if [ -z $(echo "$netdevices" |grep -x $1) ]; then
  850. check_result $E_INVALID "invalid interface format :: $1"
  851. fi
  852. }
  853. # IP status validator
  854. is_ip_status_format_valid() {
  855. if [ -z "$(echo shared,dedicated | grep -w $1 )" ]; then
  856. check_result $E_INVALID "invalid status format :: $1"
  857. fi
  858. }
  859. # Cron validator
  860. is_cron_format_valid() {
  861. limit=59
  862. check_format=''
  863. if [ "$2" = 'hour' ]; then
  864. limit=23
  865. fi
  866. if [ "$2" = 'day' ]; then
  867. limit=31
  868. fi
  869. if [ "$2" = 'month' ]; then
  870. limit=12
  871. fi
  872. if [ "$2" = 'wday' ]; then
  873. limit=7
  874. fi
  875. if [ "$1" = '*' ]; then
  876. check_format='ok'
  877. fi
  878. if [[ "$1" =~ ^[\*]+[/]+[0-9] ]]; then
  879. if [ "$(echo $1 |cut -f 2 -d /)" -lt $limit ]; then
  880. check_format='ok'
  881. fi
  882. fi
  883. if [[ "$1" =~ ^[0-9][-|,|0-9]{0,70}[\/][0-9]$ ]]; then
  884. check_format='ok'
  885. crn_values=${1//,/ }
  886. crn_values=${crn_values//-/ }
  887. crn_values=${crn_values//\// }
  888. for crn_vl in $crn_values; do
  889. if [ "$crn_vl" -gt $limit ]; then
  890. check_format='invalid'
  891. fi
  892. done
  893. fi
  894. crn_values=$(echo $1 |tr "," " " | tr "-" " ")
  895. for crn_vl in $crn_values
  896. do
  897. if [[ "$crn_vl" =~ ^[0-9]+$ ]] && [ "$crn_vl" -le $limit ]; then
  898. check_format='ok'
  899. fi
  900. done
  901. if [ "$check_format" != 'ok' ]; then
  902. check_result $E_INVALID "invalid $2 format :: $1"
  903. fi
  904. }
  905. # Name validator
  906. is_name_format_valid() {
  907. if ! [[ "$1" =~ ^[[:alnum:]][-|\ |\.|_[:alnum:]]{0,28}[[:alnum:]]$ ]]; then
  908. check_result $E_INVALID "invalid $2 format :: $1"
  909. fi
  910. }
  911. # Object validator
  912. is_object_format_valid() {
  913. if ! [[ "$1" =~ ^[[:alnum:]][-|\.|_[:alnum:]]{0,64}[[:alnum:]]$ ]]; then
  914. check_result $E_INVALID "invalid $2 format :: $1"
  915. fi
  916. }
  917. # Role validator
  918. is_role_valid (){
  919. if ! [[ "$1" =~ ^admin|user$ ]]; then
  920. check_result $E_INVALID "invalid $2 format :: $1"
  921. fi
  922. }
  923. # Password validator
  924. is_password_format_valid() {
  925. if [ "${#1}" -lt '6' ]; then
  926. check_result $E_INVALID "invalid password format :: $1"
  927. fi
  928. }
  929. # Missing function -
  930. # Before: validate_format_shell
  931. # After: is_format_valid_shell
  932. is_format_valid_shell() {
  933. if [ -z "$(grep -w $1 /etc/shells)" ]; then
  934. echo "Error: shell $1 is not valid"
  935. log_event "$E_INVALID" "$EVENT"
  936. exit $E_INVALID
  937. fi
  938. }
  939. # Service name validator
  940. is_service_format_valid() {
  941. if ! [[ "$1" =~ ^[[:alnum:]][-|\.|_[:alnum:]]{0,64}$ ]]; then
  942. check_result $E_INVALID "invalid $2 format :: $1"
  943. fi
  944. }
  945. is_hash_format_valid() {
  946. if ! [[ "$1" =~ ^[-_A-Za-z0-9]{1,32}$ ]]; then
  947. check_result $E_INVALID "invalid $2 format :: $1"
  948. fi
  949. }
  950. # Format validation controller
  951. is_format_valid() {
  952. for arg_name in $*; do
  953. eval arg=\$$arg_name
  954. if [ ! -z "$arg" ]; then
  955. case $arg_name in
  956. account) is_user_format_valid "$arg" "$arg_name";;
  957. action) is_fw_action_format_valid "$arg";;
  958. active) is_boolean_format_valid "$arg" 'active' ;;
  959. aliases) is_alias_format_valid "$arg" ;;
  960. antispam) is_boolean_format_valid "$arg" 'antispam' ;;
  961. antivirus) is_boolean_format_valid "$arg" 'antivirus' ;;
  962. autoreply) is_autoreply_format_valid "$arg" ;;
  963. backup) is_object_format_valid "$arg" 'backup' ;;
  964. charset) is_object_format_valid "$arg" "$arg_name" ;;
  965. charsets) is_common_format_valid "$arg" 'charsets' ;;
  966. comment) is_object_format_valid "$arg" 'comment' ;;
  967. database) is_database_format_valid "$arg" 'database';;
  968. day) is_cron_format_valid "$arg" $arg_name ;;
  969. dbpass) is_password_format_valid "$arg" ;;
  970. dbuser) is_dbuser_format_valid "$arg" 'dbuser';;
  971. dkim) is_boolean_format_valid "$arg" 'dkim' ;;
  972. dkim_size) is_int_format_valid "$arg" ;;
  973. domain) is_domain_format_valid "$arg" ;;
  974. dvalue) is_dns_record_format_valid "$arg";;
  975. email) is_email_format_valid "$arg" ;;
  976. email_forward) is_email_format_valid "$arg" ;;
  977. exp) is_date_format_valid "$arg" ;;
  978. extentions) is_common_format_valid "$arg" 'extentions' ;;
  979. ftp_password) is_password_format_valid "$arg" ;;
  980. ftp_user) is_user_format_valid "$arg" "$arg_name" ;;
  981. hash) is_hash_format_valid "$arg" "$arg_name" ;;
  982. host) is_object_format_valid "$arg" "$arg_name" ;;
  983. hour) is_cron_format_valid "$arg" $arg_name ;;
  984. id) is_int_format_valid "$arg" 'id' ;;
  985. iface) is_interface_format_valid "$arg" ;;
  986. ip) is_ip_format_valid "$arg" ;;
  987. ipv6) is_ipv6_format_valid "$arg" ;;
  988. ip46) is_ip46_format_valid "$arg" ;;
  989. ip_name) is_domain_format_valid "$arg" 'IP name';;
  990. ip_status) is_ip_status_format_valid "$arg" ;;
  991. job) is_int_format_valid "$arg" 'job' ;;
  992. key) is_user_format_valid "$arg" "$arg_name" ;;
  993. malias) is_user_format_valid "$arg" "$arg_name" ;;
  994. max_db) is_int_format_valid "$arg" 'max db';;
  995. min) is_cron_format_valid "$arg" $arg_name ;;
  996. month) is_cron_format_valid "$arg" $arg_name ;;
  997. name) is_name_format_valid "$arg" "name" ;;
  998. nat_ip) is_ip_format_valid "$arg" ;;
  999. netmask) is_ip_format_valid "$arg" 'netmask' ;;
  1000. newid) is_int_format_valid "$arg" 'id' ;;
  1001. ns1) is_domain_format_valid "$arg" 'ns1' ;;
  1002. ns2) is_domain_format_valid "$arg" 'ns2' ;;
  1003. ns3) is_domain_format_valid "$arg" 'ns3' ;;
  1004. ns4) is_domain_format_valid "$arg" 'ns4' ;;
  1005. ns5) is_domain_format_valid "$arg" 'ns5' ;;
  1006. ns6) is_domain_format_valid "$arg" 'ns6' ;;
  1007. ns7) is_domain_format_valid "$arg" 'ns7' ;;
  1008. ns8) is_domain_format_valid "$arg" 'ns8' ;;
  1009. object) is_name_format_valid "$arg" 'object';;
  1010. package) is_object_format_valid "$arg" "$arg_name" ;;
  1011. password) is_password_format_valid "$arg" ;;
  1012. port) is_int_format_valid "$arg" 'port' ;;
  1013. port_ext) is_fw_port_format_valid "$arg";;
  1014. protocol) is_fw_protocol_format_valid "$arg" ;;
  1015. proxy_ext) is_extention_format_valid "$arg" ;;
  1016. quota) is_int_format_valid "$arg" 'quota' ;;
  1017. record) is_common_format_valid "$arg" 'record';;
  1018. restart) is_boolean_format_valid "$arg" 'restart' ;;
  1019. role) is_role_valid "$arg" 'role' ;;
  1020. rtype) is_dns_type_format_valid "$arg" ;;
  1021. rule) is_int_format_valid "$arg" "rule id" ;;
  1022. service) is_service_format_valid "$arg" "$arg_name" ;;
  1023. soa) is_domain_format_valid "$arg" 'SOA' ;;
  1024. #missing command: is_format_valid_shell
  1025. shell) is_format_valid_shell "$arg" ;;
  1026. stats_pass) is_password_format_valid "$arg" ;;
  1027. stats_user) is_user_format_valid "$arg" "$arg_name" ;;
  1028. template) is_object_format_valid "$arg" "$arg_name" ;;
  1029. ttl) is_int_format_valid "$arg" 'ttl';;
  1030. user) is_user_format_valid "$arg" $arg_name;;
  1031. wday) is_cron_format_valid "$arg" $arg_name ;;
  1032. esac
  1033. fi
  1034. done
  1035. }
  1036. # Domain argument formatting
  1037. format_domain() {
  1038. if [[ "$domain" = *[![:ascii:]]* ]]; then
  1039. if [[ "$domain" =~ [[:upper:]] ]]; then
  1040. domain=$(echo "$domain" |sed 's/[[:upper:]].*/\L&/')
  1041. fi
  1042. else
  1043. if [[ "$domain" =~ [[:upper:]] ]]; then
  1044. domain=$(echo "$domain" |tr '[:upper:]' '[:lower:]')
  1045. fi
  1046. fi
  1047. if [[ "$domain" =~ ^www\..* ]]; then
  1048. domain=$(echo "$domain" |sed -e "s/^www.//")
  1049. fi
  1050. if [[ "$domain" =~ .*\.$ ]]; then
  1051. domain=$(echo "$domain" |sed -e "s/[.]*$//g")
  1052. fi
  1053. if [[ "$domain" =~ ^\. ]]; then
  1054. domain=$(echo "$domain" |sed -e "s/^[.]*//")
  1055. fi
  1056. }
  1057. format_domain_idn() {
  1058. if [ -z "$domain_idn" ]; then
  1059. domain_idn=$domain
  1060. fi
  1061. if [[ "$domain_idn" = *[![:ascii:]]* ]]; then
  1062. domain_idn=$(idn -t --quiet -a $domain_idn)
  1063. fi
  1064. }
  1065. format_aliases() {
  1066. if [ ! -z "$aliases" ] && [ "$aliases" != 'none' ]; then
  1067. aliases=$(echo $aliases |tr '[:upper:]' '[:lower:]' |tr ',' '\n')
  1068. aliases=$(echo "$aliases" |sed -e "s/\.$//" |sort -u)
  1069. aliases=$(echo "$aliases" |tr -s '.')
  1070. aliases=$(echo "$aliases" |sed -e "s/[.]*$//g")
  1071. aliases=$(echo "$aliases" |sed -e "s/^[.]*//")
  1072. aliases=$(echo "$aliases" |sed -e "/^$/d")
  1073. aliases=$(echo "$aliases" |tr '\n' ',' |sed -e "s/,$//")
  1074. fi
  1075. }
  1076. check_backup_conditions() {
  1077. # Checking load average
  1078. la=$(cat /proc/loadavg |cut -f 1 -d ' ' |cut -f 1 -d '.')
  1079. # i=0
  1080. while [ "$la" -ge "$BACKUP_LA_LIMIT" ]; do
  1081. echo -e "$(date "+%F %T") Load Average $la"
  1082. sleep 60
  1083. la=$(cat /proc/loadavg |cut -f 1 -d ' ' |cut -f 1 -d '.')
  1084. done
  1085. }
  1086. # Define download function
  1087. download_file() {
  1088. local url=$1
  1089. local destination=$2
  1090. local force=$3
  1091. # Default destination is the curent working directory
  1092. local dstopt=""
  1093. if [ ! -z "$(echo "$url" | grep -E "\.(gz|gzip|bz2|zip|xz)$")" ]; then
  1094. # When an archive file is downloaded it will be first saved localy
  1095. dstopt="--directory-prefix=$ARCHIVE_DIR"
  1096. local is_archive="true"
  1097. local filename="${url##*/}"
  1098. if [ -z "$filename" ]; then
  1099. >&2 echo "[!] No filename was found in url, exiting ($url)"
  1100. exit 1
  1101. fi
  1102. if [ ! -z "$force" ] && [ -f "$ARCHIVE_DIR/$filename" ]; then
  1103. rm -f $ARCHIVE_DIR/$filename
  1104. fi
  1105. elif [ ! -z "$destination" ]; then
  1106. # Plain files will be written to specified location
  1107. dstopt="-O $destination"
  1108. fi
  1109. # check for corrupted archive
  1110. if [ -f "$ARCHIVE_DIR/$filename" ] && [ "$is_archive" = "true" ]; then
  1111. tar -tzf "$ARCHIVE_DIR/$filename" > /dev/null 2>&1
  1112. if [ $? -ne 0 ]; then
  1113. >&2 echo "[!] Archive $ARCHIVE_DIR/$filename is corrupted, redownloading"
  1114. rm -f $ARCHIVE_DIR/$filename
  1115. fi
  1116. fi
  1117. if [ ! -f "$ARCHIVE_DIR/$filename" ]; then
  1118. wget $url -q $dstopt --show-progress --progress=bar:force --limit-rate=3m
  1119. fi
  1120. if [ ! -z "$destination" ] && [ "$is_archive" = "true" ]; then
  1121. if [ "$destination" = "-" ]; then
  1122. cat "$ARCHIVE_DIR/$filename"
  1123. elif [ -d "$(dirname $destination)" ]; then
  1124. cp "$ARCHIVE_DIR/$filename" "$destination"
  1125. fi
  1126. fi
  1127. }
  1128. check_hestia_demo_mode() {
  1129. demo_mode=$(grep DEMO_MODE /usr/local/hestia/conf/hestia.conf | cut -d '=' -f2 | sed "s|'||g")
  1130. if [ ! -z "$demo_mode" ] && [ "$demo_mode" = "yes" ]; then
  1131. echo "ERROR: Unable to perform operation due to security restrictions that are in place."
  1132. exit 1
  1133. fi
  1134. }
  1135. multiphp_count() {
  1136. echo $(ls -d /etc/php/*/fpm/pool.d 2>/dev/null |wc -l)
  1137. }
  1138. multiphp_versions() {
  1139. local -a php_versions_list;
  1140. local php_ver;
  1141. if [ "$(multiphp_count)" -gt 0 ] ; then
  1142. for php_ver in $(ls -v /etc/php/); do
  1143. [ ! -d "/etc/php/$php_ver/fpm/pool.d/" ] && continue
  1144. php_versions_list+=($php_ver)
  1145. done
  1146. echo "${php_versions_list[@]}"
  1147. fi
  1148. }
  1149. multiphp_default_version() {
  1150. # Get system wide default php version (set by update-alternatives)
  1151. local sys_phpversion=$(php -r "echo (float)phpversion();")
  1152. # Check if the system php also has php-fpm enabled, otherwise return
  1153. # the most recent php version which does have it installed.
  1154. if [ ! -d "/etc/php/$sys_phpversion/fpm/pool.d/" ]; then
  1155. local all_versions="$(multiphp_versions)"
  1156. if [ ! -z "$all_versions" ]; then
  1157. sys_phpversion="${all_versions##*\ }";
  1158. fi
  1159. fi
  1160. echo "$sys_phpversion"
  1161. }
  1162. is_hestia_package(){
  1163. if [ -z "$(echo $1 | grep -w $2)" ]; then
  1164. check_result $E_INVALID "$2 package is not controlled by hestiacp"
  1165. fi
  1166. }
  1167. # Run arbitrary cli commands with dropped privileges
  1168. # Note: setpriv --init-groups is not available on debian9 (util-linux 2.29.2)
  1169. # Input:
  1170. # - $user : Vaild hestia user
  1171. user_exec() {
  1172. is_object_valid 'user' 'USER' "$user"
  1173. local user_groups=$(id -G "$user")
  1174. user_groups=${user_groups//\ /,}
  1175. setpriv --groups "$user_groups" --reuid "$user" --regid "$user" -- $@
  1176. }
  1177. # Simple chmod wrapper that skips symlink files after glob expand
  1178. no_symlink_chmod() {
  1179. local filemode=$1; shift;
  1180. for i in "$@"; do
  1181. [[ -L ${i} ]] && continue
  1182. chmod "${filemode}" "${i}"
  1183. done
  1184. }