main.sh 40 KB

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