main.sh 43 KB

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