main.sh 49 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780
  1. #!/usr/bin/env bash
  2. #===========================================================================#
  3. # #
  4. # Hestia Control Panel - Core Function Library #
  5. # #
  6. #===========================================================================#
  7. # Source conf function for correct variable initialisation
  8. source_conf() {
  9. while IFS='= ' read -r lhs rhs; do
  10. if [[ ! $lhs =~ ^\ *# && -n $lhs ]]; then
  11. rhs="${rhs%%^\#*}" # Del in line right comments
  12. rhs="${rhs%%*( )}" # Del trailing spaces
  13. rhs="${rhs%\'*}" # Del opening string quotes
  14. rhs="${rhs#\'*}" # Del closing string quotes
  15. declare -g $lhs="$rhs"
  16. fi
  17. done < $1
  18. }
  19. if [ -z "$user" ]; then
  20. if [ -z "$ROOT_USER" ]; then
  21. if [ -z "$HESTIA" ]; then
  22. # shellcheck source=/etc/hestiacp/hestia.conf
  23. source /etc/hestiacp/hestia.conf
  24. fi
  25. source_conf "$HESTIA/conf/hestia.conf" # load config file
  26. fi
  27. user="$ROOT_USER"
  28. fi
  29. # Internal variables
  30. HOMEDIR='/home'
  31. BACKUP='/backup'
  32. BACKUP_GZIP=9
  33. BACKUP_DISK_LIMIT=95
  34. BACKUP_LA_LIMIT=$(cat /proc/cpuinfo | grep processor | wc -l)
  35. RRD_STEP=300
  36. BIN=$HESTIA/bin
  37. HESTIA_INSTALL_DIR="$HESTIA/install/deb"
  38. HESTIA_COMMON_DIR="$HESTIA/install/common"
  39. HESTIA_BACKUP="/root/hst_backups/$(date +%d%m%Y%H%M)"
  40. HESTIA_PHP="$HESTIA/php/bin/php"
  41. USER_DATA=$HESTIA/data/users/$user
  42. WEBTPL=$HESTIA/data/templates/web
  43. MAILTPL=$HESTIA/data/templates/mail
  44. DNSTPL=$HESTIA/data/templates/dns
  45. RRD=$HESTIA/web/rrd
  46. SENDMAIL="$HESTIA/web/inc/mail-wrapper.php"
  47. HESTIA_GIT_REPO="https://raw.githubusercontent.com/hestiacp/hestiacp"
  48. HESTIA_THEMES="$HESTIA/web/css/themes"
  49. HESTIA_THEMES_CUSTOM="$HESTIA/web/css/themes/custom"
  50. SCRIPT="$(basename $0)"
  51. CHECK_RESULT_CALLBACK=""
  52. # Return codes
  53. OK=0
  54. E_ARGS=1
  55. E_INVALID=2
  56. E_NOTEXIST=3
  57. E_EXISTS=4
  58. E_SUSPENDED=5
  59. E_UNSUSPENDED=6
  60. E_INUSE=7
  61. E_LIMIT=8
  62. E_PASSWORD=9
  63. E_FORBIDEN=10
  64. E_DISABLED=11
  65. E_PARSING=12
  66. E_DISK=13
  67. E_LA=14
  68. E_CONNECT=15
  69. E_FTP=16
  70. E_DB=17
  71. E_RRD=18
  72. E_UPDATE=19
  73. E_RESTART=20
  74. # Detect operating system
  75. detect_os() {
  76. if [ -e "/etc/os-release" ]; then
  77. get_os_type=$(grep "^ID=" /etc/os-release | cut -f 2 -d '=')
  78. if [ "$get_os_type" = "ubuntu" ]; then
  79. if [ -e '/usr/bin/lsb_release' ]; then
  80. OS_VERSION="$(lsb_release -s -r)"
  81. OS_TYPE='Ubuntu'
  82. fi
  83. elif [ "$get_os_type" = "debian" ]; then
  84. OS_TYPE='Debian'
  85. OS_VERSION=$(cat /etc/debian_version | grep -o "[0-9]\{1,2\}" | head -n1)
  86. fi
  87. else
  88. OS_TYPE="Unsupported OS"
  89. OS_VERSION="Unknown"
  90. fi
  91. }
  92. # Generate time stamp
  93. new_timestamp() {
  94. time_n_date=$(date +'%T %F')
  95. time=$(echo "$time_n_date" | cut -f 1 -d \ )
  96. date=$(echo "$time_n_date" | cut -f 2 -d \ )
  97. }
  98. # Event string for logger
  99. ARGS=("$@")
  100. for ((I = 1; I <= $#; I++)); do
  101. if [[ "$HIDE" != "$I" ]]; then
  102. ARGUMENTS="$ARGUMENTS '${ARGS[${I} - 1]}'"
  103. else
  104. ARGUMENTS="$ARGUMENTS '******'"
  105. fi
  106. done
  107. # Log event function
  108. log_event() {
  109. if [ -z "$time" ]; then
  110. LOG_TIME="$(date +'%F %T') $(basename $0)"
  111. else
  112. LOG_TIME="$date $time $(basename $0)"
  113. fi
  114. if [ "$1" -eq 0 ]; then
  115. echo "$LOG_TIME $2" >> $HESTIA/log/system.log
  116. else
  117. echo "$LOG_TIME $2 [Error $1]" >> $HESTIA/log/error.log
  118. fi
  119. }
  120. # Log user history
  121. log_history() {
  122. message=${1//\'/\´} # Avoid single quotes broken the log
  123. evt_level=${2:-$event_level}
  124. log_user=${3-$user}
  125. evt_category=${4:-$event_category}
  126. # Set default event level and category if not specified
  127. if [ -z "$evt_level" ]; then
  128. evt_level="Info"
  129. fi
  130. if [ -z "$evt_category" ]; then
  131. evt_category="System"
  132. fi
  133. # Log system events to system log file
  134. if [ "$log_user" = "system" ]; then
  135. log=$HESTIA/log/activity.log
  136. else
  137. if ! $BIN/v-list-user "$log_user" > /dev/null; then
  138. return $E_NOTEXIST
  139. fi
  140. log=$HESTIA/data/users/$log_user/history.log
  141. fi
  142. touch $log
  143. if [ '300' -lt "$(wc -l $log | cut -f 1 -d ' ')" ]; then
  144. tail -n 250 $log > $log.moved
  145. mv -f $log.moved $log
  146. chmod 660 $log
  147. fi
  148. if [ -z "$date" ]; then
  149. time_n_date=$(date +'%T %F')
  150. time=$(echo "$time_n_date" | cut -f 1 -d \ )
  151. date=$(echo "$time_n_date" | cut -f 2 -d \ )
  152. fi
  153. curr_str=$(tail -n1 $log | grep "ID=" --text | cut -f2 -d \')
  154. id="$((curr_str + 1))"
  155. echo "ID='$id' DATE='$date' TIME='$time' LEVEL='$evt_level' CATEGORY='$evt_category' MESSAGE='$message'" >> $log
  156. }
  157. # Result checker
  158. check_result() {
  159. if [ $1 -ne 0 ]; then
  160. local err_code="${3:-$1}"
  161. if [[ -n "$CHECK_RESULT_CALLBACK" && "$(type -t "$CHECK_RESULT_CALLBACK")" == 'function' ]]; then
  162. $CHECK_RESULT_CALLBACK "$err_code" "$2"
  163. else
  164. echo "Error: $2"
  165. log_event "$err_code" "$ARGUMENTS"
  166. fi
  167. exit $err_code
  168. fi
  169. }
  170. # Argument list checker
  171. check_args() {
  172. if [ "$1" -gt "$2" ]; then
  173. echo "Usage: $(basename $0) $3"
  174. check_result "$E_ARGS" "not enought arguments" > /dev/null
  175. fi
  176. }
  177. # Define version check function
  178. version_ge() { test "$(printf '%s\n' "$@" | sort -V | head -n 1)" != "$1" -o -n "$1" -a "$1" = "$2"; }
  179. # Subsystem checker
  180. is_system_enabled() {
  181. if [ -z "$1" ] || [ "$1" = no ]; then
  182. check_result "$E_DISABLED" "$2 is not enabled"
  183. fi
  184. }
  185. # User package check
  186. is_package_full() {
  187. case "$1" in
  188. WEB_DOMAINS) used=$(wc -l $USER_DATA/web.conf) ;;
  189. WEB_ALIASES) used=$(echo $aliases | tr ',' '\n' | wc -l) ;;
  190. DNS_DOMAINS) used=$(wc -l $USER_DATA/dns.conf) ;;
  191. DNS_RECORDS) used=$(wc -l $USER_DATA/dns/$domain.conf) ;;
  192. MAIL_DOMAINS) used=$(wc -l $USER_DATA/mail.conf) ;;
  193. MAIL_ACCOUNTS) used=$(wc -l $USER_DATA/mail/$domain.conf) ;;
  194. DATABASES) used=$(wc -l $USER_DATA/db.conf) ;;
  195. CRON_JOBS) used=$(wc -l $USER_DATA/cron.conf) ;;
  196. esac
  197. used=$(echo "$used" | cut -f 1 -d \ )
  198. limit=$(grep "^$1=" $USER_DATA/user.conf | cut -f 2 -d \')
  199. if [ "$1" = WEB_ALIASES ]; then
  200. # Used is always calculated with the new alias added
  201. if [ "$limit" != 'unlimited' ] && [[ "$used" -gt "$limit" ]]; then
  202. check_result "$E_LIMIT" "$1 limit is reached :: upgrade user package"
  203. fi
  204. else
  205. if [ "$limit" != 'unlimited' ] && [[ "$used" -ge "$limit" ]]; then
  206. check_result "$E_LIMIT" "$1 limit is reached :: upgrade user package"
  207. fi
  208. fi
  209. }
  210. # User owner for reseller plugin
  211. get_user_owner() {
  212. if [ -z "$RESELLER_KEY" ]; then
  213. owner="$ROOT_USER"
  214. else
  215. owner=$(grep "^OWNER" $USER_DATA/user.conf | cut -f 2 -d \')
  216. if [ -z "$owner" ]; then
  217. owner="$ROOT_USER"
  218. fi
  219. fi
  220. }
  221. # Random password generator
  222. generate_password() {
  223. matrix=$1
  224. length=$2
  225. if [ -z "$matrix" ]; then
  226. matrix="A-Za-z0-9"
  227. fi
  228. if [ -z "$length" ]; then
  229. length=16
  230. fi
  231. head /dev/urandom | tr -dc $matrix | head -c$length
  232. }
  233. # Package existence check
  234. is_package_valid() {
  235. if [ -z $1 ]; then
  236. if [ ! -e "$HESTIA/data/packages/$package.pkg" ]; then
  237. check_result "$E_NOTEXIST" "package $package doesn't exist"
  238. fi
  239. else
  240. if [ ! -e "$HESTIA/data/packages/$1.pkg" ]; then
  241. check_result "$E_NOTEXIST" "package $1 doesn't exist"
  242. fi
  243. fi
  244. }
  245. is_package_new() {
  246. if [ -e "$HESTIA/data/packages/$1.pkg" ]; then
  247. echo "Error: package $1 already exists."
  248. log_event "$E_EXISTS" "$ARGUMENTS"
  249. exit "$E_EXISTS"
  250. fi
  251. }
  252. # Validate system type
  253. is_type_valid() {
  254. if [ -z "$(echo $1 | grep -w $2)" ]; then
  255. check_result "$E_INVALID" "$2 type is invalid"
  256. fi
  257. }
  258. # Check user backup settings
  259. is_backup_enabled() {
  260. BACKUPS=$(grep "^BACKUPS=" $USER_DATA/user.conf | cut -f2 -d \')
  261. if [ -z "$BACKUPS" ] || [[ "$BACKUPS" -le '0' ]]; then
  262. check_result "$E_DISABLED" "user backup is disabled"
  263. fi
  264. }
  265. # Check user backup settings
  266. is_backup_scheduled() {
  267. if [ -e "$HESTIA/data/queue/backup.pipe" ]; then
  268. check_q=$(grep " $user " $HESTIA/data/queue/backup.pipe | grep $1)
  269. if [ -n "$check_q" ]; then
  270. check_result "$E_EXISTS" "$1 is already scheduled"
  271. fi
  272. fi
  273. }
  274. # Check if object is new
  275. is_object_new() {
  276. if [ $2 = 'USER' ]; then
  277. if [ -d "$USER_DATA" ]; then
  278. object="OK"
  279. fi
  280. else
  281. object=$(grep "$2='$3'" $USER_DATA/$1.conf)
  282. fi
  283. if [ -n "$object" ]; then
  284. check_result "$E_EXISTS" "$2=$3 already exists"
  285. fi
  286. }
  287. # Check if object is valid
  288. is_object_valid() {
  289. if [ $2 = 'USER' ]; then
  290. tstpath="$(readlink -f "$HESTIA/data/users/$3")"
  291. if [ "$(dirname "$tstpath")" != "$(readlink -f "$HESTIA/data/users")" ] || [ ! -d "$HESTIA/data/users/$3" ]; then
  292. check_result "$E_NOTEXIST" "$1 $3 doesn't exist"
  293. fi
  294. elif [ $2 = 'KEY' ]; then
  295. local key="$(basename "$3")"
  296. if [[ -z "$key" || ${#key} -lt 16 ]] || [[ ! -f "$HESTIA/data/access-keys/${key}" && ! -f "$HESTIA/data/access-keys/$key" ]]; then
  297. check_result "$E_NOTEXIST" "$1 $3 doesn't exist"
  298. fi
  299. else
  300. object=$(grep "$2='$3'" $HESTIA/data/users/$user/$1.conf)
  301. if [ -z "$object" ]; then
  302. arg1=$(basename $1)
  303. arg2=$(echo $2 | tr '[:upper:]' '[:lower:]')
  304. check_result "$E_NOTEXIST" "$arg1 $arg2 $3 doesn't exist"
  305. fi
  306. fi
  307. }
  308. # Check if a object string with key values pairs has the correct format and load it afterwards
  309. parse_object_kv_list_non_eval() {
  310. local str
  311. local objkv obj_key obj_val
  312. local OLD_IFS="$IFS"
  313. str=${@//$'\n'/ }
  314. str=${str//\"/\\\"}
  315. str=${str//$/\\$}
  316. IFS=$'\n'
  317. # Extract and loop trough each key-value pair. (Regex test: https://regex101.com/r/eiMufk/5)
  318. for objkv in $(echo "$str" | perl -n -e "while(/\b([a-zA-Z]+[\w]*)='(.*?)'(\s|\$)/g) {print \$1.'='.\$2 . \"\n\" }"); do
  319. if ! [[ "$objkv" =~ ^([[:alnum:]][_[:alnum:]]{0,64}[[:alnum:]])=(\'?[^\']+?\'?)?$ ]]; then
  320. check_result "$E_INVALID" "Invalid key value format [$objkv]"
  321. fi
  322. obj_key=${objkv%%=*} # strip everything after first '=' char
  323. obj_val=${objkv#*=} # strip everything before first '=' char
  324. declare -g $obj_key="$obj_val"
  325. done
  326. IFS="$OLD_IFS"
  327. }
  328. # Check if a object string with key values pairs has the correct format and load it afterwards
  329. parse_object_kv_list() {
  330. local str
  331. local objkv
  332. local suboutput
  333. local OLD_IFS="$IFS"
  334. str=${@//$'\n'/ }
  335. str=${str//\"/\\\"}
  336. str=${str//$/\\$}
  337. IFS=$'\n'
  338. suboutput=$(setpriv --clear-groups --reuid nobody --regid nogroup bash -c "PS4=''; set -xe; eval \"${str}\"" 2>&1)
  339. check_result $? "Invalid object format: ${str}" $E_INVALID
  340. for objkv in $suboutput; do
  341. if [[ "$objkv" =~ ^'eval ' ]]; then
  342. continue
  343. fi
  344. if ! [[ "$objkv" =~ ^([[:alnum:]][_[:alnum:]]{0,64}[[:alnum:]])=(\'?[^\']+?\'?)?$ ]]; then
  345. check_result "$E_INVALID" "Invalid key value format [$objkv]"
  346. fi
  347. eval "$objkv"
  348. done
  349. IFS="$OLD_IFS"
  350. }
  351. # Check if object is supended
  352. is_object_suspended() {
  353. if [ $2 = 'USER' ]; then
  354. spnd=$(cat $USER_DATA/$1.conf | grep "SUSPENDED='yes'")
  355. else
  356. spnd=$(grep "$2='$3'" $USER_DATA/$1.conf | grep "SUSPENDED='yes'")
  357. fi
  358. if [ -z "$spnd" ]; then
  359. check_result "$E_UNSUSPENDED" "$(basename $1) $3 is not suspended"
  360. fi
  361. }
  362. # Check if object is unsupended
  363. is_object_unsuspended() {
  364. if [ $2 = 'USER' ]; then
  365. spnd=$(cat $USER_DATA/$1.conf | grep "SUSPENDED='yes'")
  366. else
  367. spnd=$(grep "$2='$3'" $USER_DATA/$1.conf | grep "SUSPENDED='yes'")
  368. fi
  369. if [ -n "$spnd" ]; then
  370. check_result "$E_SUSPENDED" "$(basename $1) $3 is suspended"
  371. fi
  372. }
  373. # Check if object value is empty
  374. is_object_value_empty() {
  375. str=$(grep "$2='$3'" $USER_DATA/$1.conf)
  376. parse_object_kv_list "$str"
  377. eval value=$4
  378. if [ -n "$value" ] && [ "$value" != 'no' ]; then
  379. check_result "$E_EXISTS" "${4//$/}=$value already exists"
  380. fi
  381. }
  382. # Check if object value is empty
  383. is_object_value_exist() {
  384. str=$(grep "$2='$3'" $USER_DATA/$1.conf)
  385. parse_object_kv_list "$str"
  386. eval value=$4
  387. if [ -z "$value" ] || [ "$value" = 'no' ]; then
  388. check_result "$E_NOTEXIST" "${4//$/}=$value doesn't exist"
  389. fi
  390. }
  391. # Check if password is transmitted via file
  392. is_password_valid() {
  393. if [[ "$password" =~ ^/tmp/ ]]; then
  394. if ! [[ "$password" == *../* ]]; then
  395. if [ -f "$password" ]; then
  396. password="$(head -n1 $password)"
  397. fi
  398. fi
  399. fi
  400. }
  401. # Check if hash is transmitted via file
  402. is_hash_valid() {
  403. if [[ "$hash" =~ ^/tmp/ ]]; then
  404. if ! [[ "$hash" == *../* ]]; then
  405. if [ -f "$hash" ]; then
  406. hash="$(head -n1 $hash)"
  407. fi
  408. fi
  409. fi
  410. }
  411. # Check if directory is a symlink
  412. is_dir_symlink() {
  413. if [[ -L "$1" ]]; then
  414. check_result "$E_FORBIDEN" "$1 directory is a symlink"
  415. fi
  416. }
  417. # Get object value
  418. get_object_value() {
  419. object=$(grep "$2='$3'" $USER_DATA/$1.conf)
  420. parse_object_kv_list "$object"
  421. eval echo $4
  422. }
  423. get_object_values() {
  424. parse_object_kv_list $(grep "$2='$3'" $USER_DATA/$1.conf)
  425. }
  426. # Update object value
  427. update_object_value() {
  428. row=$(grep -nF "$2='$3'" $USER_DATA/$1.conf)
  429. lnr=$(echo $row | cut -f 1 -d ':')
  430. object=$(echo $row | sed "s/^$lnr://")
  431. parse_object_kv_list "$object"
  432. eval old="$4"
  433. old=$(echo "$old" | sed -e 's/\\/\\\\/g' -e 's/&/\\&/g' -e 's/\//\\\//g')
  434. new=$(echo "$5" | sed -e 's/\\/\\\\/g' -e 's/&/\\&/g' -e 's/\//\\\//g')
  435. sed -i "$lnr s/${4//$/}='${old//\*/\\*}'/${4//$/}='${new//\*/\\*}'/g" \
  436. $USER_DATA/$1.conf
  437. }
  438. # Add object key
  439. add_object_key() {
  440. row=$(grep -n "$2='$3'" $USER_DATA/$1.conf)
  441. lnr=$(echo $row | cut -f 1 -d ':')
  442. object=$(echo $row | sed "s/^$lnr://")
  443. if [ -z "$(echo $object | grep $4=)" ]; then
  444. eval old="$4"
  445. sed -i "$lnr s/$5='/$4='' $5='/" $USER_DATA/$1.conf
  446. fi
  447. }
  448. # Search objects
  449. search_objects() {
  450. OLD_IFS="$IFS"
  451. IFS=$'\n'
  452. if [ -f $USER_DATA/$1.conf ]; then
  453. for line in $(grep $2=\'$3\' $USER_DATA/$1.conf); do
  454. parse_object_kv_list "$line"
  455. eval echo \$$4
  456. done
  457. fi
  458. IFS="$OLD_IFS"
  459. }
  460. # Get user value
  461. get_user_value() {
  462. grep "^${1//$/}=" $USER_DATA/user.conf | head -1 | awk -F "'" '{print $2}'
  463. }
  464. # Update user value in user.conf
  465. update_user_value() {
  466. key="${2//$/}"
  467. lnr=$(grep -n "^$key='" $HESTIA/data/users/$1/user.conf | cut -f 1 -d ':')
  468. if [ -n "$lnr" ]; then
  469. sed -i "$lnr d" $HESTIA/data/users/$1/user.conf
  470. sed -i "$lnr i\\$key='${3}'" $HESTIA/data/users/$1/user.conf
  471. fi
  472. }
  473. # Increase user counter
  474. increase_user_value() {
  475. key="${2//$/}"
  476. factor="${3-1}"
  477. conf="$HESTIA/data/users/$1/user.conf"
  478. old=$(grep "$key=" $conf | cut -f 2 -d \')
  479. if [ -z "$old" ]; then
  480. old=0
  481. fi
  482. new=$((old + factor))
  483. sed -i "s/$key='$old'/$key='$new'/g" $conf
  484. }
  485. # Decrease user counter
  486. decrease_user_value() {
  487. key="${2//$/}"
  488. factor="${3-1}"
  489. conf="$HESTIA/data/users/$1/user.conf"
  490. old=$(grep "$key=" $conf | cut -f 2 -d \')
  491. if [ -z "$old" ]; then
  492. old=0
  493. fi
  494. if [ "$old" -le 1 ]; then
  495. new=0
  496. else
  497. new=$((old - factor))
  498. fi
  499. if [ "$new" -lt 0 ]; then
  500. new=0
  501. fi
  502. sed -i "s/$key='$old'/$key='$new'/g" $conf
  503. }
  504. # Notify user
  505. send_notice() {
  506. topic=$1
  507. notice=$2
  508. if [ "$notify" = 'yes' ]; then
  509. touch $USER_DATA/notifications.conf
  510. chmod 660 $USER_DATA/notifications.conf
  511. time_n_date=$(date +'%T %F')
  512. time=$(echo "$time_n_date" | cut -f 1 -d \ )
  513. date=$(echo "$time_n_date" | cut -f 2 -d \ )
  514. nid=$(grep "NID=" $USER_DATA/notifications.conf | cut -f 2 -d \')
  515. nid=$(echo "$nid" | sort -n | tail -n1)
  516. if [ -n "$nid" ]; then
  517. nid="$((nid + 1))"
  518. else
  519. nid=1
  520. fi
  521. str="NID='$nid' TOPIC='$topic' NOTICE='$notice' TYPE='$type'"
  522. str="$str ACK='no' TIME='$time' DATE='$date'"
  523. echo "$str" >> $USER_DATA/notifications.conf
  524. if [ -z "$(grep NOTIFICATIONS $USER_DATA/user.conf)" ]; then
  525. sed -i "s/^TIME/NOTIFICATIONS='yes'\nTIME/g" $USER_DATA/user.conf
  526. else
  527. update_user_value "$user" '$NOTIFICATIONS' "yes"
  528. fi
  529. fi
  530. }
  531. # Recalculate U_DISK value
  532. recalc_user_disk_usage() {
  533. u_usage=0
  534. if [ -f "$USER_DATA/web.conf" ]; then
  535. usage=0
  536. dusage=$(grep 'U_DISK=' $USER_DATA/web.conf \
  537. | awk -F "U_DISK='" '{print $2}' | cut -f 1 -d \')
  538. for disk_usage in $dusage; do
  539. usage=$((usage + disk_usage))
  540. done
  541. d=$(grep "U_DISK_WEB='" $USER_DATA/user.conf | cut -f 2 -d \')
  542. sed -i "s/U_DISK_WEB='$d'/U_DISK_WEB='$usage'/g" $USER_DATA/user.conf
  543. u_usage=$((u_usage + usage))
  544. fi
  545. if [ -f "$USER_DATA/mail.conf" ]; then
  546. usage=0
  547. dusage=$(grep 'U_DISK=' $USER_DATA/mail.conf \
  548. | awk -F "U_DISK='" '{print $2}' | cut -f 1 -d \')
  549. for disk_usage in $dusage; do
  550. usage=$((usage + disk_usage))
  551. done
  552. d=$(grep "U_DISK_MAIL='" $USER_DATA/user.conf | cut -f 2 -d \')
  553. sed -i "s/U_DISK_MAIL='$d'/U_DISK_MAIL='$usage'/g" $USER_DATA/user.conf
  554. u_usage=$((u_usage + usage))
  555. fi
  556. if [ -f "$USER_DATA/db.conf" ]; then
  557. usage=0
  558. dusage=$(grep 'U_DISK=' $USER_DATA/db.conf \
  559. | awk -F "U_DISK='" '{print $2}' | cut -f 1 -d \')
  560. for disk_usage in $dusage; do
  561. usage=$((usage + disk_usage))
  562. done
  563. d=$(grep "U_DISK_DB='" $USER_DATA/user.conf | cut -f 2 -d \')
  564. sed -i "s/U_DISK_DB='$d'/U_DISK_DB='$usage'/g" $USER_DATA/user.conf
  565. u_usage=$((u_usage + usage))
  566. fi
  567. usage=$(grep 'U_DISK_DIRS=' $USER_DATA/user.conf | cut -f 2 -d "'")
  568. u_usage=$((u_usage + usage))
  569. old=$(grep "U_DISK='" $USER_DATA/user.conf | cut -f 2 -d \')
  570. sed -i "s/U_DISK='$old'/U_DISK='$u_usage'/g" $USER_DATA/user.conf
  571. }
  572. # Recalculate U_BANDWIDTH value
  573. recalc_user_bandwidth_usage() {
  574. usage=0
  575. bandwidth_usage=$(grep 'U_BANDWIDTH=' $USER_DATA/web.conf \
  576. | awk -F "U_BANDWIDTH='" '{print $2}' | cut -f 1 -d \')
  577. for bandwidth in $bandwidth_usage; do
  578. usage=$((usage + bandwidth))
  579. done
  580. old=$(grep "U_BANDWIDTH='" $USER_DATA/user.conf | cut -f 2 -d \')
  581. sed -i "s/U_BANDWIDTH='$old'/U_BANDWIDTH='$usage'/g" $USER_DATA/user.conf
  582. }
  583. # Get next cron job id
  584. get_next_cronjob() {
  585. if [ -z "$job" ]; then
  586. curr_str=$(grep "JOB=" $USER_DATA/cron.conf | cut -f 2 -d \' \
  587. | sort -n | tail -n1)
  588. job="$((curr_str + 1))"
  589. fi
  590. }
  591. # Sort cron jobs by id
  592. sort_cron_jobs() {
  593. cat $USER_DATA/cron.conf | sort -n -k 2 -t \' > $USER_DATA/cron.tmp
  594. mv -f $USER_DATA/cron.tmp $USER_DATA/cron.conf
  595. }
  596. # Sync cronjobs with system cron
  597. sync_cron_jobs() {
  598. source_conf "$USER_DATA/user.conf"
  599. if [ -e "/var/spool/cron/crontabs" ]; then
  600. crontab="/var/spool/cron/crontabs/$user"
  601. else
  602. crontab="/var/spool/cron/$user"
  603. fi
  604. # remove file if exists
  605. if [ -e "$crontab" ]; then
  606. rm -f $crontab
  607. fi
  608. # touch new crontab file
  609. touch $crontab
  610. if [ "$CRON_REPORTS" = 'yes' ]; then
  611. echo "MAILTO=$CONTACT" > $crontab
  612. echo 'CONTENT_TYPE="text/plain; charset=utf-8"' >> $crontab
  613. else
  614. echo 'MAILTO=""' > $crontab
  615. fi
  616. while read line; do
  617. parse_object_kv_list "$line"
  618. if [ "$SUSPENDED" = 'no' ]; then
  619. echo "$MIN $HOUR $DAY $MONTH $WDAY $CMD" \
  620. | sed -e "s/%quote%/'/g" -e "s/%dots%/:/g" \
  621. >> $crontab
  622. fi
  623. done < $USER_DATA/cron.conf
  624. chown $user:$user $crontab
  625. chmod 600 $crontab
  626. }
  627. # User format validator
  628. is_user_format_valid() {
  629. if [ ${#1} -eq 1 ]; then
  630. if ! [[ "$1" =~ ^^[[:alnum:]]$ ]]; then
  631. check_result "$E_INVALID" "invalid $2 format :: $1"
  632. fi
  633. else
  634. if [ -n "$3" ]; then
  635. maxlenght=$(($3 - 2))
  636. if ! [[ "$1" =~ ^[[:alnum:]][-|\.|_[:alnum:]]{0,$maxlenght}[[:alnum:]]$ ]]; then
  637. check_result "$E_INVALID" "invalid $2 format :: $1"
  638. fi
  639. else
  640. if ! [[ "$1" =~ ^[[:alnum:]][-|\.|_[:alnum:]]{0,28}[[:alnum:]]$ ]]; then
  641. check_result "$E_INVALID" "invalid $2 format :: $1"
  642. fi
  643. fi
  644. fi
  645. if [ "$1" != "${1//[^[:ascii:]]/}" ]; then
  646. check_result "$E_INVALID" "invalid $2 format :: $1"
  647. fi
  648. }
  649. # Domain format validator
  650. is_domain_format_valid() {
  651. object_name=${2-domain}
  652. exclude="[!|@|#|$|^|&|*|(|)|+|=|{|}|:|,|<|>|?|_|/|\|\"|'|;|%|\`| ]"
  653. if [[ $1 =~ $exclude ]] || [[ $1 =~ ^[0-9]+$ ]] || [[ $1 =~ \.\. ]] || [[ $1 =~ $(printf '\t') ]] || [[ "$1" = "www" ]]; then
  654. check_result "$E_INVALID" "invalid $object_name format :: $1"
  655. fi
  656. is_no_new_line_format "$1"
  657. }
  658. # Alias forman validator
  659. is_alias_format_valid() {
  660. for object in ${1//,/ }; do
  661. exclude="[!|@|#|$|^|&|(|)|+|=|{|}|:|<|>|?|_|/|\|\"|'|;|%|\`| ]"
  662. if [[ "$object" =~ $exclude ]]; then
  663. check_result "$E_INVALID" "invalid alias format :: $object"
  664. fi
  665. if [[ "$object" =~ [*] ]] && ! [[ "$object" =~ ^[*]\..* ]]; then
  666. check_result "$E_INVALID" "invalid alias format :: $object"
  667. fi
  668. done
  669. }
  670. # IP format validator
  671. is_ip_format_valid() {
  672. object_name=${2-ip}
  673. ip_regex='([1-9]?[0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])'
  674. ip_clean=$(echo "${1%/*}")
  675. if ! [[ $ip_clean =~ ^$ip_regex\.$ip_regex\.$ip_regex\.$ip_regex$ ]]; then
  676. check_result "$E_INVALID" "invalid $object_name format :: $1"
  677. fi
  678. if [ $1 != "$ip_clean" ]; then
  679. ip_cidr="$ip_clean/"
  680. ip_cidr=$(echo "${1#$ip_cidr}")
  681. if [[ "$ip_cidr" -gt 32 ]] || [[ "$ip_cidr" =~ [:alnum:] ]]; then
  682. check_result "$E_INVALID" "invalid $object_name format :: $1"
  683. fi
  684. fi
  685. }
  686. # IPv6 format validator
  687. is_ipv6_format_valid() {
  688. object_name=${2-ipv6}
  689. ip_regex='([1-9]?[0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])'
  690. t_ip=$(echo $1 | awk -F / '{print $1}')
  691. t_cidr=$(echo $1 | awk -F / '{print $2}')
  692. valid_cidr=1
  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. check_result "$E_INVALID" "invalid $object_name format :: $1"
  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 [ "$valid_cidr" -eq 0 ]; then
  719. check_result "$E_INVALID" "invalid $object_name format :: $1"
  720. fi
  721. }
  722. is_ip46_format_valid() {
  723. t_ip=$(echo $1 | awk -F / '{print $1}')
  724. t_cidr=$(echo $1 | awk -F / '{print $2}')
  725. valid_octets=0
  726. valid_cidr=1
  727. for octet in ${t_ip//./ }; do
  728. if [[ $octet =~ ^[0-9]{1,3}$ ]] && [[ $octet -le 255 ]]; then
  729. ((++valid_octets))
  730. fi
  731. done
  732. if [ -n "$(echo $1 | grep '/')" ]; then
  733. if [[ "$t_cidr" -lt 0 ]] || [[ "$t_cidr" -gt 32 ]]; then
  734. valid_cidr=0
  735. fi
  736. if ! [[ "$t_cidr" =~ ^[0-9]+$ ]]; then
  737. valid_cidr=0
  738. fi
  739. fi
  740. if [ "$valid_octets" -lt 4 ] || [ "$valid_cidr" -eq 0 ]; then
  741. #Check IPV6
  742. ipv6_valid=""
  743. WORD="[0-9A-Fa-f]\{1,4\}"
  744. # flat address, no compressed words
  745. FLAT="^${WORD}\(:${WORD}\)\{7\}$"
  746. COMP2="^\(${WORD}:\)\{1,1\}\(:${WORD}\)\{1,6\}$"
  747. COMP3="^\(${WORD}:\)\{1,2\}\(:${WORD}\)\{1,5\}$"
  748. COMP4="^\(${WORD}:\)\{1,3\}\(:${WORD}\)\{1,4\}$"
  749. COMP5="^\(${WORD}:\)\{1,4\}\(:${WORD}\)\{1,3\}$"
  750. COMP6="^\(${WORD}:\)\{1,5\}\(:${WORD}\)\{1,2\}$"
  751. COMP7="^\(${WORD}:\)\{1,6\}\(:${WORD}\)\{1,1\}$"
  752. # trailing :: edge case, includes case of only :: (all 0's)
  753. EDGE_TAIL="^\(\(${WORD}:\)\{1,7\}\|:\):$"
  754. # leading :: edge case
  755. EDGE_LEAD="^:\(:${WORD}\)\{1,7\}$"
  756. echo $t_ip | grep --silent "\(${FLAT}\)\|\(${COMP2}\)\|\(${COMP3}\)\|\(${COMP4}\)\|\(${COMP5}\)\|\(${COMP6}\)\|\(${COMP7}\)\|\(${EDGE_TAIL}\)\|\(${EDGE_LEAD}\)"
  757. if [ $? -ne 0 ]; then
  758. ipv6_valid="INVALID"
  759. fi
  760. if [ -n "$(echo $1 | grep '/')" ]; then
  761. if [[ "$t_cidr" -lt 0 ]] || [[ "$t_cidr" -gt 128 ]]; then
  762. valid_cidr=0
  763. fi
  764. if ! [[ "$t_cidr" =~ ^[0-9]+$ ]]; then
  765. valid_cidr=0
  766. fi
  767. fi
  768. if [ -n "$ipv6_valid" ] || [ "$valid_cidr" -eq 0 ]; then
  769. check_result "$E_INVALID" "invalid IP format :: $1"
  770. fi
  771. fi
  772. }
  773. # Proxy extention format validator
  774. is_extention_format_valid() {
  775. exclude="[!|#|$|^|&|(|)|+|=|{|}|:|@|<|>|?|/|\|\"|'|;|%|\`| ]"
  776. if [[ "$1" =~ $exclude ]]; then
  777. check_result "$E_INVALID" "invalid proxy extention format :: $1"
  778. fi
  779. is_no_new_line_format "$1"
  780. }
  781. # Number format validator
  782. is_number_format_valid() {
  783. object_name=${2-number}
  784. if ! [[ "$1" =~ ^[0-9]+$ ]]; then
  785. check_result "$E_INVALID" "invalid $object_name format :: $1"
  786. fi
  787. }
  788. # Autoreply format validator
  789. is_autoreply_format_valid() {
  790. if [ 10240 -le ${#1} ]; then
  791. check_result "$E_INVALID" "invalid autoreply format :: $1"
  792. fi
  793. }
  794. # Boolean format validator
  795. is_boolean_format_valid() {
  796. if [ "$1" != 'yes' ] && [ "$1" != 'no' ]; then
  797. check_result "$E_INVALID" "invalid $2 format :: $1"
  798. fi
  799. }
  800. # Refresh IPset format validator
  801. is_refresh_ipset_format_valid() {
  802. if [ "$1" != 'load' ] && [ "$1" != 'yes' ] && [ "$1" != 'no' ]; then
  803. check_result "$E_INVALID" "invalid $2 format :: $1"
  804. fi
  805. }
  806. # Common format validator
  807. is_common_format_valid() {
  808. exclude="[!|#|$|^|&|(|)|+|=|{|}|:|<|>|?|/|\|\"|'|;|%|\`| ]"
  809. if [[ "$1" =~ $exclude ]]; then
  810. check_result "$E_INVALID" "invalid $2 format :: $1"
  811. fi
  812. if [ 400 -le ${#1} ]; then
  813. check_result "$E_INVALID" "invalid $2 format :: $1"
  814. fi
  815. if [[ "$1" =~ @ ]] && [ ${#1} -gt 1 ]; then
  816. check_result "$E_INVALID" "invalid $2 format :: $1"
  817. fi
  818. if [[ $1 =~ \* ]]; then
  819. if [[ "$(echo $1 | grep -o '\*\.' | wc -l)" -eq 0 ]] && [[ $1 != '*' ]]; then
  820. check_result "$E_INVALID" "invalid $2 format :: $1"
  821. fi
  822. fi
  823. if [[ $(echo -n "$1" | tail -c 1) =~ [^a-zA-Z0-9_*@.] ]]; then
  824. check_result "$E_INVALID" "invalid $2 format :: $1"
  825. fi
  826. if [[ $(echo -n "$1" | grep -c '\.\.') -gt 0 ]]; then
  827. check_result "$E_INVALID" "invalid $2 format :: $1"
  828. fi
  829. if [[ $(echo -n "$1" | head -c 1) =~ [^a-zA-Z0-9_*@] ]]; then
  830. check_result "$E_INVALID" "invalid $2 format :: $1"
  831. fi
  832. if [[ $(echo -n "$1" | grep -c '\-\-') -gt 0 ]]; then
  833. check_result "$E_INVALID" "invalid $2 format :: $1"
  834. fi
  835. if [[ $(echo -n "$1" | grep -c '\_\_') -gt 0 ]]; then
  836. check_result "$E_INVALID" "invalid $2 format :: $1"
  837. fi
  838. is_no_new_line_format "$1"
  839. }
  840. is_no_new_line_format() {
  841. test=$(echo "$1" | head -n1)
  842. if [[ "$test" != "$1" ]]; then
  843. check_result "$E_INVALID" "invalid value :: $1"
  844. fi
  845. }
  846. is_string_format_valid() {
  847. exclude="[!|#|$|^|&|(|)|+|=|{|}|:|<|>|?|/|\|\"|'|;|%|\`]"
  848. if [[ "$1" =~ $exclude ]]; then
  849. check_result "$E_INVALID" "invalid $2 format :: $1"
  850. fi
  851. is_no_new_line_format "$1"
  852. }
  853. # Database format validator
  854. is_database_format_valid() {
  855. exclude="[!|@|#|$|^|&|*|(|)|+|=|{|}|:|,|<|>|?|/|\|\"|'|;|%|\`| ]"
  856. if [[ "$1" =~ $exclude ]] || [ 64 -le ${#1} ]; then
  857. check_result "$E_INVALID" "invalid $2 format :: $1"
  858. fi
  859. is_no_new_line_format "$1"
  860. }
  861. # Date format validator
  862. is_date_format_valid() {
  863. if ! [[ "$1" =~ ^[0-9][0-9][0-9][0-9]-[0-9][0-9]-[0-9][0-9]$ ]]; then
  864. check_result "$E_INVALID" "invalid date format :: $1"
  865. fi
  866. }
  867. # Database user validator
  868. is_dbuser_format_valid() {
  869. exclude="[!|@|#|$|^|&|*|(|)|+|=|{|}|:|,|<|>|?|/|\|\"|'|;|%|\`| ]"
  870. if [ 33 -le ${#1} ]; then
  871. check_result "$E_INVALID" "mysql username can be up to 32 characters long"
  872. fi
  873. if [[ "$1" =~ $exclude ]]; then
  874. check_result "$E_INVALID" "invalid $2 format :: $1"
  875. fi
  876. is_no_new_line_format "$1"
  877. }
  878. # DNS record type validator
  879. is_dns_type_format_valid() {
  880. known_dnstype='A,AAAA,NS,CNAME,MX,TXT,SRV,DNSKEY,KEY,IPSECKEY,PTR,SPF,TLSA,CAA,DS'
  881. if [ -z "$(echo $known_dnstype | grep -w $1)" ]; then
  882. check_result "$E_INVALID" "invalid dns record type format :: $1"
  883. fi
  884. }
  885. # DNS record validator
  886. is_dns_record_format_valid() {
  887. if [ "$rtype" = 'A' ]; then
  888. is_ip_format_valid "$1"
  889. fi
  890. if [ "$rtype" = 'NS' ]; then
  891. is_domain_format_valid "${1::-1}" 'ns_record'
  892. fi
  893. if [ "$rtype" = 'MX' ]; then
  894. is_domain_format_valid "${1::-1}" 'mx_record'
  895. is_int_format_valid "$priority" 'priority_record'
  896. fi
  897. if [ "$rtype" = 'SRV' ]; then
  898. format_no_quotes "$priority" 'priority_record'
  899. fi
  900. is_no_new_line_format "$1"
  901. }
  902. # Email format validator
  903. is_email_format_valid() {
  904. if [[ ! "$1" =~ ^[A-Za-z0-9._%+-]+@[[:alnum:].-]+\.[A-Za-z]{2,63}$ ]]; then
  905. if [[ ! "$1" =~ ^[A-Za-z0-9._%+-]+@[[:alnum:].-]+\.(xn--)[[:alnum:]]{2,63}$ ]]; then
  906. check_result "$E_INVALID" "invalid email format :: $1"
  907. fi
  908. fi
  909. }
  910. # Firewall action validator
  911. is_fw_action_format_valid() {
  912. if [ "$1" != "ACCEPT" ] && [ "$1" != 'DROP' ]; then
  913. check_result "$E_INVALID" "invalid action format :: $1"
  914. fi
  915. }
  916. # Firewall protocol validator
  917. is_fw_protocol_format_valid() {
  918. if [ "$1" != "ICMP" ] && [ "$1" != 'UDP' ] && [ "$1" != 'TCP' ]; then
  919. check_result "$E_INVALID" "invalid protocol format :: $1"
  920. fi
  921. }
  922. # Firewall port validator
  923. is_fw_port_format_valid() {
  924. if [ "${#1}" -eq 1 ]; then
  925. if ! [[ "$1" =~ [0-9] ]]; then
  926. check_result "$E_INVALID" "invalid port format :: $1"
  927. fi
  928. else
  929. if ! [[ "$1" =~ ^[0-9][-|,|:|0-9]{0,76}[0-9]$ ]]; then
  930. check_result "$E_INVALID" "invalid port format and/or more than 78 chars used :: $1"
  931. fi
  932. fi
  933. }
  934. # DNS record id validator
  935. is_id_format_valid() {
  936. if ! echo "$1" | grep -qE '^[1-9][0-9]{0,}$'; then
  937. check_result "$E_INVALID" "invalid $2 format :: $1"
  938. fi
  939. }
  940. # Integer validator
  941. is_int_format_valid() {
  942. if ! [[ "$1" =~ ^[0-9]+$ ]]; then
  943. check_result "$E_INVALID" "invalid $2 format :: $1"
  944. fi
  945. }
  946. # Interface validator
  947. is_interface_format_valid() {
  948. nic_names="$(ip -d -j link show | jq -r '.[] | if .link_type == "loopback" then empty else .ifname, if .altnames then .altnames[] else empty end end')"
  949. if [ -z "$(echo "$nic_names" | grep -x "$1")" ]; then
  950. check_result "$E_INVALID" "invalid interface format :: $1"
  951. fi
  952. }
  953. # IP status validator
  954. is_ip_status_format_valid() {
  955. if [ -z "$(echo shared,dedicated | grep -w "$1")" ]; then
  956. check_result "$E_INVALID" "invalid status format :: $1"
  957. fi
  958. }
  959. # Cron validator
  960. is_cron_format_valid() {
  961. limit=59
  962. check_format=''
  963. if [ "$2" = 'hour' ]; then
  964. limit=23
  965. fi
  966. if [ "$2" = 'day' ]; then
  967. limit=31
  968. fi
  969. if [ "$2" = 'month' ]; then
  970. limit=12
  971. fi
  972. if [ "$2" = 'wday' ]; then
  973. limit=7
  974. fi
  975. if [ "$1" = '*' ]; then
  976. check_format='ok'
  977. fi
  978. if [[ "$1" =~ ^[\*]+[/]+[0-9] ]]; then
  979. if [ "$(echo $1 | cut -f 2 -d /)" -lt $limit ]; then
  980. check_format='ok'
  981. fi
  982. fi
  983. if [[ "$1" =~ ^[0-9][-|,|0-9]{0,70}[\/][0-9]$ ]]; then
  984. check_format='ok'
  985. crn_values=${1//,/ }
  986. crn_values=${crn_values//-/ }
  987. crn_values=${crn_values//\// }
  988. for crn_vl in $crn_values; do
  989. if [ "$crn_vl" -gt $limit ]; then
  990. check_format='invalid'
  991. fi
  992. done
  993. fi
  994. crn_values=$(echo $1 | tr "," " " | tr "-" " ")
  995. for crn_vl in $crn_values; do
  996. if [[ "$crn_vl" =~ ^[0-9]+$ ]] && [ "$crn_vl" -le $limit ]; then
  997. check_format='ok'
  998. fi
  999. done
  1000. if [ "$check_format" != 'ok' ]; then
  1001. check_result "$E_INVALID" "invalid $2 format :: $1"
  1002. fi
  1003. }
  1004. is_object_name_format_valid() {
  1005. if ! [[ "$1" =~ ^[-|\ |\.|_[:alnum:]]{0,50}$ ]]; then
  1006. check_result "$E_INVALID" "invalid $2 format :: $1"
  1007. fi
  1008. }
  1009. # Name validator
  1010. is_name_format_valid() {
  1011. exclude="['|\"|<|>]"
  1012. if [[ "$1" =~ $exclude ]]; then
  1013. check_result "$E_INVALID" "Invalid $2 contains qoutes (\" or ') :: $1"
  1014. fi
  1015. is_no_new_line_format "$1"
  1016. }
  1017. # Object validator
  1018. is_object_format_valid() {
  1019. if ! [[ "$1" =~ ^[[:alnum:]][-|\.|_[:alnum:]]{0,64}[[:alnum:]]$ ]]; then
  1020. check_result "$E_INVALID" "invalid $2 format :: $1"
  1021. fi
  1022. }
  1023. # Role validator
  1024. is_role_valid() {
  1025. if ! [[ "$1" =~ ^admin$|^user$|^dns-cluster$ ]]; then
  1026. check_result "$E_INVALID" "invalid $2 format :: $1"
  1027. fi
  1028. }
  1029. # Password validator
  1030. is_password_format_valid() {
  1031. if [ "${#1}" -lt '6' ]; then
  1032. check_result "$E_INVALID" "invalid password format :: $1"
  1033. fi
  1034. }
  1035. # Missing function -
  1036. # Before: validate_format_shell
  1037. # After: is_format_valid_shell
  1038. is_format_valid_shell() {
  1039. if [ -z "$(grep -w $1 /etc/shells)" ]; then
  1040. echo "Error: shell $1 is not valid"
  1041. log_event "$E_INVALID" "$EVENT"
  1042. exit $E_INVALID
  1043. fi
  1044. }
  1045. # Service name validator
  1046. is_service_format_valid() {
  1047. if ! [[ "$1" =~ ^[[:alnum:]][-|\.|_[:alnum:]]{0,64}$ ]]; then
  1048. check_result "$E_INVALID" "invalid $2 format :: $1"
  1049. fi
  1050. }
  1051. is_hash_format_valid() {
  1052. if ! [[ "$1" =~ ^[[:alnum:]|\:|\=|_|-]{1,80}$ ]]; then
  1053. check_result "$E_INVALID" "invalid $2 format :: $1"
  1054. fi
  1055. }
  1056. # Format validation controller
  1057. is_format_valid() {
  1058. for arg_name in $*; do
  1059. eval arg=\$$arg_name
  1060. if [ -n "$arg" ]; then
  1061. case $arg_name in
  1062. access_key_id) is_access_key_id_format_valid "$arg" "$arg_name" ;;
  1063. account) is_user_format_valid "$arg" "$arg_name" '64' ;;
  1064. action) is_fw_action_format_valid "$arg" ;;
  1065. active) is_boolean_format_valid "$arg" 'active' ;;
  1066. aliases) is_alias_format_valid "$arg" ;;
  1067. alias) is_alias_format_valid "$arg" ;;
  1068. antispam) is_boolean_format_valid "$arg" 'antispam' ;;
  1069. antivirus) is_boolean_format_valid "$arg" 'antivirus' ;;
  1070. autoreply) is_autoreply_format_valid "$arg" ;;
  1071. backup) is_object_format_valid "$arg" 'backup' ;;
  1072. charset) is_object_format_valid "$arg" "$arg_name" ;;
  1073. charsets) is_common_format_valid "$arg" 'charsets' ;;
  1074. chain) is_object_format_valid "$arg" 'chain' ;;
  1075. comment) is_object_format_valid "$arg" 'comment' ;;
  1076. database) is_database_format_valid "$arg" 'database' ;;
  1077. day) is_cron_format_valid "$arg" $arg_name ;;
  1078. dbpass) is_password_format_valid "$arg" ;;
  1079. dbuser) is_dbuser_format_valid "$arg" 'dbuser' ;;
  1080. dkim) is_boolean_format_valid "$arg" 'dkim' ;;
  1081. dkim_size) is_int_format_valid "$arg" ;;
  1082. domain) is_domain_format_valid "$arg" ;;
  1083. dom_alias) is_alias_format_valid "$arg" ;;
  1084. dvalue) is_dns_record_format_valid "$arg" ;;
  1085. email) is_email_format_valid "$arg" ;;
  1086. email_forward) is_email_format_valid "$arg" ;;
  1087. exp) is_date_format_valid "$arg" ;;
  1088. extentions) is_common_format_valid "$arg" 'extentions' ;;
  1089. format) is_type_valid 'plain json shell csv' "$arg" ;;
  1090. ftp_password) is_password_format_valid "$arg" ;;
  1091. ftp_user) is_user_format_valid "$arg" "$arg_name" ;;
  1092. hash) is_hash_format_valid "$arg" "$arg_name" ;;
  1093. host) is_object_format_valid "$arg" "$arg_name" ;;
  1094. hour) is_cron_format_valid "$arg" $arg_name ;;
  1095. id) is_id_format_valid "$arg" 'id' ;;
  1096. iface) is_interface_format_valid "$arg" ;;
  1097. ip) is_ip_format_valid "$arg" ;;
  1098. ipv6) is_ipv6_format_valid "$arg" ;;
  1099. ip46) is_ip46_format_valid "$arg" ;;
  1100. ip_name) is_domain_format_valid "$arg" 'IP name' ;;
  1101. ip_status) is_ip_status_format_valid "$arg" ;;
  1102. job) is_int_format_valid "$arg" 'job' ;;
  1103. key) is_common_format_valid "$arg" "$arg_name" ;;
  1104. malias) is_user_format_valid "$arg" "$arg_name" '64' ;;
  1105. max_db) is_int_format_valid "$arg" 'max db' ;;
  1106. min) is_cron_format_valid "$arg" $arg_name ;;
  1107. month) is_cron_format_valid "$arg" $arg_name ;;
  1108. name) is_name_format_valid "$arg" "name" ;;
  1109. nat_ip) is_ip_format_valid "$arg" ;;
  1110. netmask) is_ip_format_valid "$arg" 'netmask' ;;
  1111. newid) is_int_format_valid "$arg" 'id' ;;
  1112. ns1) is_domain_format_valid "$arg" 'ns1' ;;
  1113. ns2) is_domain_format_valid "$arg" 'ns2' ;;
  1114. ns3) is_domain_format_valid "$arg" 'ns3' ;;
  1115. ns4) is_domain_format_valid "$arg" 'ns4' ;;
  1116. ns5) is_domain_format_valid "$arg" 'ns5' ;;
  1117. ns6) is_domain_format_valid "$arg" 'ns6' ;;
  1118. ns7) is_domain_format_valid "$arg" 'ns7' ;;
  1119. ns8) is_domain_format_valid "$arg" 'ns8' ;;
  1120. object) is_object_name_format_valid "$arg" 'object' ;;
  1121. package) is_object_format_valid "$arg" "$arg_name" ;;
  1122. password) is_password_format_valid "$arg" ;;
  1123. priority) is_int_format_valid $arg ;;
  1124. port) is_int_format_valid "$arg" 'port' ;;
  1125. port_ext) is_fw_port_format_valid "$arg" ;;
  1126. protocol) is_fw_protocol_format_valid "$arg" ;;
  1127. proxy_ext) is_extention_format_valid "$arg" ;;
  1128. quota) is_int_format_valid "$arg" 'quota' ;;
  1129. rate) is_int_format_valid "$arg" 'rate' ;;
  1130. record) is_common_format_valid "$arg" 'record' ;;
  1131. reject) is_boolean_format_valid "$arg" 'reject' ;;
  1132. restart) is_restart_format_valid "$arg" 'restart' ;;
  1133. role) is_role_valid "$arg" 'role' ;;
  1134. rtype) is_dns_type_format_valid "$arg" ;;
  1135. rule) is_int_format_valid "$arg" "rule id" ;;
  1136. service) is_service_format_valid "$arg" "$arg_name" ;;
  1137. secret_access_key) is_secret_access_key_format_valid "$arg" "$arg_name" ;;
  1138. soa) is_domain_format_valid "$arg" 'SOA' ;;
  1139. #missing command: is_format_valid_shell
  1140. shell) is_format_valid_shell "$arg" ;;
  1141. ssl_dir) is_folder_exists "$arg" "$arg_name" ;;
  1142. stats_pass) is_password_format_valid "$arg" ;;
  1143. stats_user) is_user_format_valid "$arg" "$arg_name" ;;
  1144. template) is_object_format_valid "$arg" "$arg_name" ;;
  1145. theme) is_common_format_valid "$arg" "$arg_name" ;;
  1146. ttl) is_int_format_valid "$arg" 'ttl' ;;
  1147. user) is_user_format_valid "$arg" $arg_name ;;
  1148. wday) is_cron_format_valid "$arg" $arg_name ;;
  1149. value) is_common_format_valid "$arg" $arg_name ;;
  1150. esac
  1151. fi
  1152. done
  1153. }
  1154. is_folder_exists() {
  1155. if [ ! -d "$1" ]; then
  1156. check_result "$E_NOTEXIST" "folder $1 does not exist"
  1157. fi
  1158. }
  1159. is_command_valid_format() {
  1160. if [[ ! "$1" =~ ^v-[[:alnum:]][-|\.|_[:alnum:]]{0,64}[[:alnum:]]$ ]]; then
  1161. check_result "$E_INVALID" "Invalid command format"
  1162. fi
  1163. if [[ -n $(echo "$1" | grep -e '\-\-') ]]; then
  1164. check_result "$E_INVALID" "Invalid command format"
  1165. fi
  1166. }
  1167. # Check access_key_id name
  1168. # Don't work with legacy key format
  1169. is_access_key_id_format_valid() {
  1170. local hash="$1"
  1171. # ACCESS_KEY_ID format validation
  1172. if ! [[ "$hash" =~ ^[[:alnum:]]{20}$ ]]; then
  1173. check_result "$E_INVALID" "invalid $2 format :: $hash"
  1174. fi
  1175. }
  1176. # SECRET_ACCESS_KEY format validation
  1177. is_secret_access_key_format_valid() {
  1178. local hash="$1"
  1179. if ! [[ "$hash" =~ ^[[:alnum:]|_|\.|\+|/|\^|~|=|%|\-]{40}$ ]]; then
  1180. check_result "$E_INVALID" "invalid $2 format"
  1181. fi
  1182. }
  1183. # Checks if the secret belongs to the access key
  1184. check_access_key_secret() {
  1185. local access_key_id="$(basename "$1")"
  1186. local secret_access_key=$2
  1187. local -n key_user=$3
  1188. if [[ -z "$access_key_id" || ! -f "$HESTIA/data/access-keys/${access_key_id}" ]]; then
  1189. check_result "$E_PASSWORD" "Access key $access_key_id doesn't exist"
  1190. fi
  1191. if [[ -z "$secret_access_key" ]]; then
  1192. check_result "$E_PASSWORD" "Secret key not provided for key $access_key_id"
  1193. elif ! [[ "$secret_access_key" =~ ^[[:alnum:]|_|\.|\+|/|\^|~|=|%|\-]{40}$ ]]; then
  1194. check_result "$E_PASSWORD" "Invalid secret key for key $access_key_id"
  1195. else
  1196. SECRET_ACCESS_KEY=""
  1197. source_conf "$HESTIA/data/access-keys/${access_key_id}"
  1198. if [[ -z "$SECRET_ACCESS_KEY" || "$SECRET_ACCESS_KEY" != "$secret_access_key" ]]; then
  1199. check_result "$E_PASSWORD" "Invalid secret key for key $access_key_id"
  1200. fi
  1201. fi
  1202. key_user="$USER"
  1203. }
  1204. # Checks if the key belongs to the user
  1205. check_access_key_user() {
  1206. local access_key_id="$(basename "$1")"
  1207. local user=$2
  1208. if [[ -z "$access_key_id" || ! -f "$HESTIA/data/access-keys/${access_key_id}" ]]; then
  1209. check_result "$E_FORBIDEN" "Access key $access_key_id doesn't exist"
  1210. fi
  1211. if [[ -z "$user" ]]; then
  1212. check_result "$E_FORBIDEN" "User not provided"
  1213. else
  1214. USER=""
  1215. source_conf "$HESTIA/data/access-keys/${access_key_id}"
  1216. if [[ -z "$USER" || "$USER" != "$user" ]]; then
  1217. check_result "$E_FORBIDEN" "key $access_key_id does not belong to the user $user"
  1218. fi
  1219. fi
  1220. }
  1221. # Checks if the key is allowed to run the command
  1222. check_access_key_cmd() {
  1223. local access_key_id="$(basename "$1")"
  1224. local cmd=$2
  1225. local -n user_arg_position=$3
  1226. if [[ "$DEBUG_MODE" = "true" ]]; then
  1227. new_timestamp
  1228. echo "[$date:$time] $1 $2" >> /var/log/hestia/api.log
  1229. fi
  1230. if [[ -z "$access_key_id" || ! -f "$HESTIA/data/access-keys/${access_key_id}" ]]; then
  1231. check_result "$E_FORBIDEN" "Access key $access_key_id doesn't exist"
  1232. fi
  1233. if [[ -z "$cmd" ]]; then
  1234. check_result "$E_FORBIDEN" "Command not provided"
  1235. elif [[ "$cmd" = 'v-make-tmp-file' ]]; then
  1236. USER="" PERMISSIONS=""
  1237. source_conf "${HESTIA}/data/access-keys/${access_key_id}"
  1238. local allowed_commands
  1239. if [[ -n "$PERMISSIONS" ]]; then
  1240. allowed_commands="$(get_apis_commands "$PERMISSIONS")"
  1241. if [[ -z "$(echo ",${allowed_commands}," | grep ",${hst_command},")" ]]; then
  1242. check_result "$E_FORBIDEN" "Key $access_key_id don't have permission to run the command $hst_command"
  1243. fi
  1244. elif [[ -z "$PERMISSIONS" && "$USER" != "$ROOT_USER" ]]; then
  1245. check_result "$E_FORBIDEN" "Key $access_key_id don't have permission to run the command $hst_command"
  1246. fi
  1247. user_arg_position="0"
  1248. elif [[ ! -e "$BIN/$cmd" ]]; then
  1249. check_result "$E_FORBIDEN" "Command $cmd not found"
  1250. else
  1251. USER="" PERMISSIONS=""
  1252. source_conf "${HESTIA}/data/access-keys/${access_key_id}"
  1253. local allowed_commands
  1254. if [[ -n "$PERMISSIONS" ]]; then
  1255. allowed_commands="$(get_apis_commands "$PERMISSIONS")"
  1256. if [[ -z "$(echo ",${allowed_commands}," | grep ",${hst_command},")" ]]; then
  1257. check_result "$E_FORBIDEN" "Key $access_key_id don't have permission to run the command $hst_command"
  1258. fi
  1259. elif [[ -z "$PERMISSIONS" && "$USER" != "$ROOT_USER" ]]; then
  1260. check_result "$E_FORBIDEN" "Key $access_key_id don't have permission to run the command $hst_command"
  1261. fi
  1262. if [[ "$USER" == "$ROOT_USER" ]]; then
  1263. # Admin can run commands for any user
  1264. user_arg_position="0"
  1265. else
  1266. user_arg_position="$(search_command_arg_position "$hst_command" "USER")"
  1267. if ! [[ "$user_arg_position" =~ ^[0-9]+$ ]]; then
  1268. check_result "$E_FORBIDEN" "Command $hst_command not found"
  1269. fi
  1270. fi
  1271. fi
  1272. }
  1273. # Domain argument formatting
  1274. format_domain() {
  1275. if [[ "$domain" = *[![:ascii:]]* ]]; then
  1276. if [[ "$domain" =~ [[:upper:]] ]]; then
  1277. domain=$(echo "$domain" | sed 's/[[:upper:]].*/\L&/')
  1278. fi
  1279. else
  1280. if [[ "$domain" =~ [[:upper:]] ]]; then
  1281. domain=$(echo "$domain" | tr '[:upper:]' '[:lower:]')
  1282. fi
  1283. fi
  1284. if [[ "$domain" =~ ^www\..* ]]; then
  1285. domain=$(echo "$domain" | sed -e "s/^www.//")
  1286. fi
  1287. if [[ "$domain" =~ .*\.$ ]]; then
  1288. domain=$(echo "$domain" | sed -e "s/[.]*$//g")
  1289. fi
  1290. if [[ "$domain" =~ ^\. ]]; then
  1291. domain=$(echo "$domain" | sed -e "s/^[.]*//")
  1292. fi
  1293. # Remove white spaces
  1294. domain=$(echo $domain | sed 's/^[ \t]*//;s/[ \t]*$//')
  1295. }
  1296. format_domain_idn() {
  1297. if [ -z "$domain_idn" ]; then
  1298. domain_idn=$domain
  1299. fi
  1300. if [[ "$domain_idn" = *[![:ascii:]]* ]]; then
  1301. domain_idn=$(idn2 --quiet $domain_idn)
  1302. fi
  1303. }
  1304. format_aliases() {
  1305. if [ -n "$aliases" ] && [ "$aliases" != 'none' ]; then
  1306. aliases=$(echo $aliases | tr '[:upper:]' '[:lower:]' | tr ',' '\n')
  1307. aliases=$(echo "$aliases" | sed -e "s/\.$//" | sort -u)
  1308. aliases=$(echo "$aliases" | tr -s '.')
  1309. aliases=$(echo "$aliases" | sed -e "s/[.]*$//g")
  1310. aliases=$(echo "$aliases" | sed -e "s/^[.]*//")
  1311. aliases=$(echo "$aliases" | sed -e "/^$/d")
  1312. aliases=$(echo "$aliases" | tr '\n' ',' | sed -e "s/,$//")
  1313. fi
  1314. }
  1315. is_restart_format_valid() {
  1316. if [ "$1" != 'yes' ] && [ "$1" != 'no' ] && [ "$1" != 'ssl' ] && [ "$1" != 'reload' ] && [ "$1" != 'updatessl' ]; then
  1317. check_result "$E_INVALID" "invalid $2 format :: $1"
  1318. fi
  1319. }
  1320. check_backup_conditions() {
  1321. # Checking load average
  1322. la=$(cat /proc/loadavg | cut -f 1 -d ' ' | cut -f 1 -d '.')
  1323. # i=0
  1324. while [ "$la" -ge "$BACKUP_LA_LIMIT" ]; do
  1325. echo -e "$(date "+%F %T") Load Average $la"
  1326. sleep 60
  1327. la=$(cat /proc/loadavg | cut -f 1 -d ' ' | cut -f 1 -d '.')
  1328. done
  1329. }
  1330. # Define download function
  1331. download_file() {
  1332. local url=$1
  1333. local destination=$2
  1334. local force=$3
  1335. # Default destination is the curent working directory
  1336. local dstopt=""
  1337. if [ -n "$(echo "$url" | grep -E "\.(gz|gzip|bz2|zip|xz)$")" ]; then
  1338. # When an archive file is downloaded it will be first saved localy
  1339. dstopt="--directory-prefix=$ARCHIVE_DIR"
  1340. local is_archive="true"
  1341. local filename="${url##*/}"
  1342. if [ -z "$filename" ]; then
  1343. echo >&2 "[!] No filename was found in url, exiting ($url)"
  1344. exit 1
  1345. fi
  1346. if [ -n "$force" ] && [ -f "$ARCHIVE_DIR/$filename" ]; then
  1347. rm -f $ARCHIVE_DIR/$filename
  1348. fi
  1349. elif [ -n "$destination" ]; then
  1350. # Plain files will be written to specified location
  1351. dstopt="-O $destination"
  1352. fi
  1353. # check for corrupted archive
  1354. if [ -f "$ARCHIVE_DIR/$filename" ] && [ "$is_archive" = "true" ]; then
  1355. tar -tzf "$ARCHIVE_DIR/$filename" > /dev/null 2>&1
  1356. if [ $? -ne 0 ]; then
  1357. echo >&2 "[!] Archive $ARCHIVE_DIR/$filename is corrupted, redownloading"
  1358. rm -f $ARCHIVE_DIR/$filename
  1359. fi
  1360. fi
  1361. if [ ! -f "$ARCHIVE_DIR/$filename" ]; then
  1362. wget $url -q $dstopt --show-progress --progress=bar:force --limit-rate=3m
  1363. fi
  1364. if [ -n "$destination" ] && [ "$is_archive" = "true" ]; then
  1365. if [ "$destination" = "-" ]; then
  1366. cat "$ARCHIVE_DIR/$filename"
  1367. elif [ -d "$(dirname $destination)" ]; then
  1368. cp "$ARCHIVE_DIR/$filename" "$destination"
  1369. fi
  1370. fi
  1371. }
  1372. check_hestia_demo_mode() {
  1373. demo_mode=$(grep DEMO_MODE /usr/local/hestia/conf/hestia.conf | cut -d '=' -f2 | sed "s|'||g")
  1374. if [ -n "$demo_mode" ] && [ "$demo_mode" = "yes" ]; then
  1375. echo "ERROR: Unable to perform operation due to security restrictions that are in place."
  1376. exit 1
  1377. fi
  1378. }
  1379. multiphp_count() {
  1380. $BIN/v-list-sys-php plain | wc -l
  1381. }
  1382. multiphp_versions() {
  1383. local -a php_versions_list
  1384. local php_ver
  1385. if [ "$(multiphp_count)" -gt 0 ]; then
  1386. for php_ver in $($BIN/v-list-sys-php plain); do
  1387. [ ! -d "/etc/php/$php_ver/fpm/pool.d/" ] && continue
  1388. php_versions_list+=($php_ver)
  1389. done
  1390. echo "${php_versions_list[@]}"
  1391. fi
  1392. }
  1393. multiphp_default_version() {
  1394. # Get system wide default php version (set by update-alternatives)
  1395. local sys_phpversion=$(php -r "echo substr(phpversion(),0,3);")
  1396. # Check if the system php also has php-fpm enabled, otherwise return
  1397. # the most recent php version which does have it installed.
  1398. if [ ! -d "/etc/php/$sys_phpversion/fpm/pool.d/" ]; then
  1399. local all_versions="$(multiphp_versions)"
  1400. if [ -n "$all_versions" ]; then
  1401. sys_phpversion="${all_versions##*\ }"
  1402. fi
  1403. fi
  1404. echo "$sys_phpversion"
  1405. }
  1406. is_hestia_package() {
  1407. check=false
  1408. for pkg in $1; do
  1409. if [ "$pkg" == "$2" ]; then
  1410. check="true"
  1411. fi
  1412. done
  1413. if [ "$check" != "true" ]; then
  1414. check_result $E_INVALID "$2 package is not controlled by hestiacp"
  1415. fi
  1416. }
  1417. # Run arbitrary cli commands with dropped privileges
  1418. # Note: setpriv --init-groups is not available on debian9 (util-linux 2.29.2)
  1419. # Input:
  1420. # - $user : Vaild hestia user
  1421. user_exec() {
  1422. is_object_valid 'user' 'USER' "$user"
  1423. local user_groups=$(id -G "$user")
  1424. user_groups=${user_groups//\ /,}
  1425. setpriv --groups "$user_groups" --reuid "$user" --regid "$user" -- "${@}"
  1426. }
  1427. # Simple chmod wrapper that skips symlink files after glob expand
  1428. no_symlink_chmod() {
  1429. local filemode=$1
  1430. shift
  1431. for i in "$@"; do
  1432. [[ -L ${i} ]] && continue
  1433. chmod "${filemode}" "${i}"
  1434. done
  1435. }
  1436. format_no_quotes() {
  1437. exclude="['|\"]"
  1438. if [[ "$1" =~ $exclude ]]; then
  1439. check_result "$E_INVALID" "Invalid $2 contains qoutes (\" or ' or | ) :: $1"
  1440. fi
  1441. is_no_new_line_format "$1"
  1442. }
  1443. is_username_format_valid() {
  1444. if [[ ! "$1" =~ ^[A-Za-z0-9._%+-]+@[[:alnum:].-]+\.[A-Za-z]{2,63}$ ]]; then
  1445. is_string_format_valid "$1" "$2"
  1446. fi
  1447. }
  1448. change_sys_value() {
  1449. check_ckey=$(grep "^$1='" "$HESTIA/conf/hestia.conf")
  1450. if [ -z "$check_ckey" ]; then
  1451. echo "$1='$2'" >> "$HESTIA/conf/hestia.conf"
  1452. else
  1453. sed -i "s|^$1=.*|$1='$2'|g" "$HESTIA/conf/hestia.conf"
  1454. fi
  1455. }
  1456. # Checks the format of APIs that will be allowed for the key
  1457. is_key_permissions_format_valid() {
  1458. local permissions="$1"
  1459. local user="$2"
  1460. if [[ "$user" != "$ROOT_USER" && -z "$permissions" ]]; then
  1461. check_result "$E_INVALID" "Non-admin users need a permission list"
  1462. fi
  1463. while IFS=',' read -ra permissions_arr; do
  1464. for permission in "${permissions_arr[@]}"; do
  1465. permission="$(basename "$permission" | sed -E "s/^\s*|\s*$//g")"
  1466. # if [[ -z "$(echo "$permission" | grep -E "^v-")" ]]; then
  1467. if [[ ! -e "$HESTIA/data/api/$permission" ]]; then
  1468. check_result "$E_NOTEXIST" "API $permission doesn't exist"
  1469. fi
  1470. source_conf "$HESTIA/data/api/$permission"
  1471. if [ "$ROLE" = "admin" ] && [ "$user" != "$ROOT_USER" ]; then
  1472. check_result "$E_INVALID" "Only the admin can run this API"
  1473. fi
  1474. # elif [[ ! -e "$BIN/$permission" ]]; then
  1475. # check_result "$E_NOTEXIST" "Command $permission doesn't exist"
  1476. # fi
  1477. done
  1478. done <<< "$permissions"
  1479. }
  1480. # Remove whitespaces, and bin path from commands
  1481. cleanup_key_permissions() {
  1482. local permissions="$1"
  1483. local final quote
  1484. while IFS=',' read -ra permissions_arr; do
  1485. for permission in "${permissions_arr[@]}"; do
  1486. permission="$(basename "$permission" | sed -E "s/^\s*|\s*$//g")"
  1487. # Avoid duplicate items
  1488. if [[ -z "$(echo ",${final}," | grep ",${permission},")" ]]; then
  1489. final+="${quote}${permission}"
  1490. quote=','
  1491. fi
  1492. done
  1493. done <<< "$permissions"
  1494. echo "$final"
  1495. }
  1496. # Extract all allowed commands from a permission list
  1497. get_apis_commands() {
  1498. local permissions="$1"
  1499. local allowed_commands quote commands_to_add
  1500. while IFS=',' read -ra permissions_arr; do
  1501. for permission in "${permissions_arr[@]}"; do
  1502. permission="$(basename "$permission" | sed -E "s/^\s*|\s*$//g")"
  1503. commands_to_add=""
  1504. # if [[ -n "$(echo "$permission" | grep -E "^v-")" ]]; then
  1505. # commands_to_add="$permission"
  1506. # el
  1507. if [[ -e "$HESTIA/data/api/$permission" ]]; then
  1508. source_conf "$HESTIA/data/api/$permission"
  1509. commands_to_add="$COMMANDS"
  1510. fi
  1511. if [[ -n "$commands_to_add" ]]; then
  1512. allowed_commands+="${quote}${commands_to_add}"
  1513. quote=','
  1514. fi
  1515. done
  1516. done <<< "$permissions"
  1517. cleanup_key_permissions "$allowed_commands"
  1518. }
  1519. # Get the position of an argument by name in a hestia command using the command's documentation comment.
  1520. #
  1521. # Return:
  1522. # * 0: It doesn't have the argument;
  1523. # * 1-9: The position of the argument in the command.
  1524. search_command_arg_position() {
  1525. local hst_command="$(basename "$1")"
  1526. local arg_name="$2"
  1527. local command_path="$BIN/$hst_command"
  1528. if [[ -z "$hst_command" || ! -e "$command_path" ]]; then
  1529. echo "-1"
  1530. return
  1531. fi
  1532. local position=0
  1533. local count=0
  1534. local command_options="$(sed -En 's/^# options: (.+)/\1/p' "$command_path")"
  1535. while IFS=' ' read -ra options_arr; do
  1536. for option in "${options_arr[@]}"; do
  1537. count=$((count + 1))
  1538. option_name="$(echo " $option " | sed -E 's/^(\s|\[)*|(\s|\])*$//g')"
  1539. if [[ "${option_name^^}" == "$arg_name" ]]; then
  1540. position=$count
  1541. fi
  1542. done
  1543. done <<< "$command_options"
  1544. echo "$position"
  1545. }
  1546. add_chroot_jail() {
  1547. local user=$1
  1548. mkdir -p /srv/jail/$user
  1549. chown 0:0 /srv /srv/jail /srv/jail/$user
  1550. chmod 755 /srv /srv/jail /srv/jail/$user
  1551. if [ ! -d /srv/jail/$user/home ]; then
  1552. mkdir -p /srv/jail/$user/home
  1553. chown 0:0 /srv/jail/$user/home
  1554. chmod 755 /srv/jail/$user/home
  1555. fi
  1556. systemd=$(systemd-escape -p --suffix=mount "/srv/jail/$user/home")
  1557. cat > "/etc/systemd/system/$systemd" << EOF
  1558. [Unit]
  1559. Description=Mount $user's home directory to the jail chroot
  1560. Before=local-fs.target
  1561. [Mount]
  1562. What=$(getent passwd $user | cut -d : -f 6)
  1563. Where=/srv/jail/$user/home
  1564. Type=none
  1565. Options=bind
  1566. LazyUnmount=yes
  1567. [Install]
  1568. RequiredBy=local-fs.target
  1569. EOF
  1570. systemctl daemon-reload > /dev/null 2>&1
  1571. systemctl enable "$systemd" > /dev/null 2>&1
  1572. systemctl start "$systemd" > /dev/null 2>&1
  1573. }
  1574. delete_chroot_jail() {
  1575. local user=$1
  1576. systemd=$(systemd-escape -p --suffix=mount "/srv/jail/$user/home")
  1577. systemctl stop "$systemd" > /dev/null 2>&1
  1578. systemctl disable "$systemd" > /dev/null 2>&1
  1579. rm -f "/etc/systemd/system/$systemd"
  1580. systemctl daemon-reload > /dev/null 2>&1
  1581. rmdir /srv/jail/$user/home > /dev/null 2>&1
  1582. rmdir /srv/jail/$user > /dev/null 2>&1
  1583. }