main.sh 47 KB

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