main.sh 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941
  1. #!/usr/bin/env bash
  2. # Internal variables
  3. HOMEDIR='/home'
  4. BACKUP='/backup'
  5. BACKUP_GZIP=9
  6. BACKUP_DISK_LIMIT=95
  7. BACKUP_LA_LIMIT=5
  8. RRD_STEP=300
  9. BIN=$VESTA/bin
  10. USER_DATA=$VESTA/data/users/$user
  11. WEBTPL=$VESTA/data/templates/web
  12. DNSTPL=$VESTA/data/templates/dns
  13. RRD=$VESTA/web/rrd
  14. SENDMAIL="$VESTA/web/inc/mail-wrapper.php"
  15. # Return codes
  16. OK=0
  17. E_ARGS=1
  18. E_INVALID=2
  19. E_NOTEXIST=3
  20. E_EXISTS=4
  21. E_SUSPENDED=5
  22. E_UNSUSPENDED=6
  23. E_INUSE=7
  24. E_LIMIT=8
  25. E_PASSWORD=9
  26. E_FORBIDEN=10
  27. E_DISABLED=11
  28. E_PARSING=12
  29. E_DISK=13
  30. E_LA=14
  31. E_CONNECT=15
  32. E_FTP=16
  33. E_DB=17
  34. E_RRD=18
  35. E_UPDATE=19
  36. E_RESTART=20
  37. E_TEAPOT=418
  38. # Event string for logger
  39. for ((I=1; I <= $# ; I++)); do
  40. if [[ "$HIDE" != $I ]]; then
  41. ARGUMENTS="$ARGUMENTS '$(eval echo \$${I})'"
  42. else
  43. ARGUMENTS="$ARGUMENTS '******'"
  44. fi
  45. done
  46. # Log event function
  47. log_event() {
  48. if [ -z "$time" ]; then
  49. LOG_TIME="$(date +'%F %T') $(basename $0)"
  50. else
  51. LOG_TIME="$date $time $(basename $0)"
  52. fi
  53. if [ "$1" -eq 0 ]; then
  54. echo "$LOG_TIME $2" >> $VESTA/log/system.log
  55. else
  56. echo "$LOG_TIME $2 [Error $1]" >> $VESTA/log/error.log
  57. fi
  58. }
  59. # Log user history
  60. log_history() {
  61. cmd=$1
  62. undo=${2-no}
  63. log_user=${3-$user}
  64. log=$VESTA/data/users/$log_user/history.log
  65. touch $log
  66. if [ '99' -lt "$(wc -l $log |cut -f 1 -d ' ')" ]; then
  67. tail -n 49 $log > $log.moved
  68. mv -f $log.moved $log
  69. chmod 660 $log
  70. fi
  71. if [ -z "$date" ]; then
  72. time_n_date=$(date +'%T %F')
  73. time=$(echo "$time_n_date" |cut -f 1 -d \ )
  74. date=$(echo "$time_n_date" |cut -f 2 -d \ )
  75. fi
  76. curr_str=$(grep "ID=" $log | cut -f 2 -d \' | sort -n | tail -n1)
  77. id="$((curr_str +1))"
  78. echo "ID='$id' DATE='$date' TIME='$time' CMD='$cmd' UNDO='$undo'" >> $log
  79. }
  80. # Result checker
  81. check_result() {
  82. if [ $1 -ne 0 ]; then
  83. echo "Error: $2"
  84. if [ ! -z "$3" ]; then
  85. log_event "$3" "$ARGUMENTS"
  86. exit $3
  87. else
  88. log_event "$1" "$ARGUMENTS"
  89. exit $1
  90. fi
  91. fi
  92. }
  93. # Argument list checker
  94. check_args() {
  95. if [ "$1" -gt "$2" ]; then
  96. echo "Usage: $(basename $0) $3"
  97. check_result $E_ARGS "not enought arguments" >/dev/null
  98. fi
  99. }
  100. # Subsystem checker
  101. is_system_enabled() {
  102. if [ -z "$1" ] || [ "$1" = no ]; then
  103. check_result $E_DISABLED "$2 is not enabled"
  104. fi
  105. }
  106. # User package check
  107. is_package_full() {
  108. case "$1" in
  109. WEB_DOMAINS) used=$(wc -l $USER_DATA/web.conf);;
  110. WEB_ALIASES) used=$(echo $aliases |tr ',' '\n' |wc -l);;
  111. DNS_DOMAINS) used=$(wc -l $USER_DATA/dns.conf);;
  112. DNS_RECORDS) used=$(wc -l $USER_DATA/dns/$domain.conf);;
  113. MAIL_DOMAINS) used=$(wc -l $USER_DATA/mail.conf);;
  114. MAIL_ACCOUNTS) used=$(wc -l $USER_DATA/mail/$domain.conf);;
  115. DATABASES) used=$(wc -l $USER_DATA/db.conf);;
  116. CRON_JOBS) used=$(wc -l $USER_DATA/cron.conf);;
  117. esac
  118. used=$(echo "$used"| cut -f 1 -d \ )
  119. limit=$(grep "^$1=" $USER_DATA/user.conf |cut -f 2 -d \')
  120. if [ "$limit" != 'unlimited' ] && [[ "$used" -ge "$limit" ]]; then
  121. check_result $E_LIMIT "$1 limit is reached :: upgrade user package"
  122. fi
  123. }
  124. # User owner for reseller plugin
  125. get_user_owner() {
  126. if [ -z "$RESELLER_KEY" ]; then
  127. owner='admin'
  128. else
  129. owner=$(grep "^OWNER" $USER_DATA/user.conf| cut -f 2 -d \')
  130. if [ -z "$owner" ]; then
  131. owner='admin'
  132. fi
  133. fi
  134. }
  135. # Random password generator
  136. generate_password() {
  137. matrix=$1
  138. lenght=$2
  139. if [ -z "$matrix" ]; then
  140. matrix=0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz
  141. fi
  142. if [ -z "$lenght" ]; then
  143. lenght=10
  144. fi
  145. i=1
  146. while [ $i -le $lenght ]; do
  147. pass="$pass${matrix:$(($RANDOM%${#matrix})):1}"
  148. ((i++))
  149. done
  150. echo "$pass"
  151. }
  152. # Package existence check
  153. is_package_valid() {
  154. if [ -z "$1" ]; then
  155. pkg_dir="$VESTA/data/packages"
  156. fi
  157. if [ ! -e "$pkg_dir/$package.pkg" ]; then
  158. check_result $E_NOTEXIST "package $package doesn't exist"
  159. fi
  160. }
  161. # Validate system type
  162. is_type_valid() {
  163. if [ -z "$(echo $1 | grep -w $2)" ]; then
  164. check_result $E_INVALID "$2 type is invalid"
  165. fi
  166. }
  167. # Check user backup settings
  168. is_backup_enabled() {
  169. BACKUPS=$(grep "^BACKUPS=" $USER_DATA/user.conf | cut -f2 -d \')
  170. if [ -z "$BACKUPS" ] || [[ "$BACKUPS" -le '0' ]]; then
  171. check_result $E_DISABLED "user backup is disabled"
  172. fi
  173. }
  174. # Check user backup settings
  175. is_backup_scheduled() {
  176. if [ -e "$VESTA/data/queue/backup.pipe" ]; then
  177. check_q=$(grep " $user " $VESTA/data/queue/backup.pipe | grep $1)
  178. if [ ! -z "$check_q" ]; then
  179. check_result $E_EXISTS "$1 is already scheduled"
  180. fi
  181. fi
  182. }
  183. # Check if object is new
  184. is_object_new() {
  185. if [ $2 = 'USER' ]; then
  186. if [ -d "$USER_DATA" ]; then
  187. object="OK"
  188. fi
  189. else
  190. object=$(grep "$2='$3'" $USER_DATA/$1.conf)
  191. fi
  192. if [ ! -z "$object" ]; then
  193. check_result $E_EXISTS "$2=$3 is already exists"
  194. fi
  195. }
  196. # Check if object is valid
  197. is_object_valid() {
  198. if [ $2 = 'USER' ]; then
  199. user_vst_dir=$(basename $3)
  200. if [ ! -d "$VESTA/data/users/$user_vst_dir" ]; then
  201. check_result $E_NOTEXIST "$1 $3 doesn't exist"
  202. fi
  203. else
  204. object=$(grep "$2='$3'" $VESTA/data/users/$user/$1.conf)
  205. if [ -z "$object" ]; then
  206. arg1=$(basename $1)
  207. arg2=$(echo $2 |tr '[:upper:]' '[:lower:]')
  208. check_result $E_NOTEXIST "$arg1 $arg2 $3 doesn't exist"
  209. fi
  210. fi
  211. }
  212. # Check if object is supended
  213. is_object_suspended() {
  214. if [ $2 = 'USER' ]; then
  215. spnd=$(cat $USER_DATA/$1.conf|grep "SUSPENDED='yes'")
  216. else
  217. spnd=$(grep "$2='$3'" $USER_DATA/$1.conf|grep "SUSPENDED='yes'")
  218. fi
  219. if [ -z "$spnd" ]; then
  220. check_result $E_UNSUSPENDED "$(basename $1) $3 is not suspended"
  221. fi
  222. }
  223. # Check if object is unsupended
  224. is_object_unsuspended() {
  225. if [ $2 = 'USER' ]; then
  226. spnd=$(cat $USER_DATA/$1.conf |grep "SUSPENDED='yes'")
  227. else
  228. spnd=$(grep "$2='$3'" $USER_DATA/$1.conf |grep "SUSPENDED='yes'")
  229. fi
  230. if [ ! -z "$spnd" ]; then
  231. check_result $E_SUSPENDED "$(basename $1) $3 is suspended"
  232. fi
  233. }
  234. # Check if object value is empty
  235. is_object_value_empty() {
  236. str=$(grep "$2='$3'" $USER_DATA/$1.conf)
  237. eval $str
  238. eval value=$4
  239. if [ ! -z "$value" ] && [ "$value" != 'no' ]; then
  240. check_result $E_EXISTS "${4//$}=$value is already exists"
  241. fi
  242. }
  243. # Check if object value is empty
  244. is_object_value_exist() {
  245. str=$(grep "$2='$3'" $USER_DATA/$1.conf)
  246. eval $str
  247. eval value=$4
  248. if [ -z "$value" ] || [ "$value" = 'no' ]; then
  249. check_result $E_NOTEXIST "${4//$}=$value doesn't exist"
  250. fi
  251. }
  252. # Check if password is transmitted via file
  253. is_password_valid() {
  254. if [[ "$password" =~ ^/tmp/ ]]; then
  255. if [ -f "$password" ]; then
  256. password="$(head -n1 $password)"
  257. fi
  258. fi
  259. }
  260. # Check if hash is transmitted via file
  261. is_hash_valid() {
  262. if [[ "$hash" =~ ^/tmp/ ]]; then
  263. if [ -f "$hash" ]; then
  264. hash="$(head -n1 $hash)"
  265. fi
  266. fi
  267. }
  268. # Check if directory is a symlink
  269. is_dir_symlink() {
  270. if [[ -L "$1" ]]; then
  271. check_result $E_FORBIDEN "$1 directory is a symlink"
  272. fi
  273. }
  274. # Get object value
  275. get_object_value() {
  276. object=$(grep "$2='$3'" $USER_DATA/$1.conf)
  277. eval "$object"
  278. eval echo $4
  279. }
  280. # Update object value
  281. update_object_value() {
  282. row=$(grep -nF "$2='$3'" $USER_DATA/$1.conf)
  283. lnr=$(echo $row | cut -f 1 -d ':')
  284. object=$(echo $row | sed "s/^$lnr://")
  285. eval "$object"
  286. eval old="$4"
  287. old=$(echo "$old" | sed -e 's/\\/\\\\/g' -e 's/&/\\&/g' -e 's/\//\\\//g')
  288. new=$(echo "$5" | sed -e 's/\\/\\\\/g' -e 's/&/\\&/g' -e 's/\//\\\//g')
  289. sed -i "$lnr s/${4//$/}='${old//\*/\\*}'/${4//$/}='${new//\*/\\*}'/g" \
  290. $USER_DATA/$1.conf
  291. }
  292. # Add object key
  293. add_object_key() {
  294. row=$(grep -n "$2='$3'" $USER_DATA/$1.conf)
  295. lnr=$(echo $row | cut -f 1 -d ':')
  296. object=$(echo $row | sed "s/^$lnr://")
  297. if [ -z "$(echo $object |grep $4=)" ]; then
  298. eval old="$4"
  299. sed -i "$lnr s/$5='/$4='' $5='/" $USER_DATA/$1.conf
  300. fi
  301. }
  302. # Search objects
  303. search_objects() {
  304. OLD_IFS="$IFS"
  305. IFS=$'\n'
  306. for line in $(grep $2=\'$3\' $USER_DATA/$1.conf); do
  307. eval $line
  308. eval echo \$$4
  309. done
  310. IFS="$OLD_IFS"
  311. }
  312. # Get user value
  313. get_user_value() {
  314. grep "^${1//$/}=" $USER_DATA/user.conf |awk -F "'" '{print $2}'
  315. }
  316. # Update user value in user.conf
  317. update_user_value() {
  318. key="${2//$}"
  319. lnr=$(grep -n "^$key='" $VESTA/data/users/$1/user.conf |cut -f 1 -d ':')
  320. if [ ! -z "$lnr" ]; then
  321. sed -i "$lnr d" $VESTA/data/users/$1/user.conf
  322. sed -i "$lnr i\\$key='${3}'" $VESTA/data/users/$1/user.conf
  323. fi
  324. }
  325. # Increase user counter
  326. increase_user_value() {
  327. key="${2//$}"
  328. factor="${3-1}"
  329. conf="$VESTA/data/users/$1/user.conf"
  330. old=$(grep "$key=" $conf | cut -f 2 -d \')
  331. if [ -z "$old" ]; then
  332. old=0
  333. fi
  334. new=$((old + factor))
  335. sed -i "s/$key='$old'/$key='$new'/g" $conf
  336. }
  337. # Decrease user counter
  338. decrease_user_value() {
  339. key="${2//$}"
  340. factor="${3-1}"
  341. conf="$VESTA/data/users/$1/user.conf"
  342. old=$(grep "$key=" $conf | cut -f 2 -d \')
  343. if [ -z "$old" ]; then
  344. old=0
  345. fi
  346. if [ "$old" -le 1 ]; then
  347. new=0
  348. else
  349. new=$((old - factor))
  350. fi
  351. if [ "$new" -lt 0 ]; then
  352. new=0
  353. fi
  354. sed -i "s/$key='$old'/$key='$new'/g" $conf
  355. }
  356. # Notify user
  357. send_notice() {
  358. topic=$1
  359. notice=$2
  360. if [ "$notify" = 'yes' ]; then
  361. touch $USER_DATA/notifications.conf
  362. chmod 660 $USER_DATA/notifications.conf
  363. time_n_date=$(date +'%T %F')
  364. time=$(echo "$time_n_date" |cut -f 1 -d \ )
  365. date=$(echo "$time_n_date" |cut -f 2 -d \ )
  366. nid=$(grep "NID=" $USER_DATA/notifications.conf |cut -f 2 -d \')
  367. nid=$(echo "$nid" |sort -n |tail -n1)
  368. if [ ! -z "$nid" ]; then
  369. nid="$((nid +1))"
  370. else
  371. nid=1
  372. fi
  373. str="NID='$nid' TOPIC='$topic' NOTICE='$notice' TYPE='$type'"
  374. str="$str ACK='no' TIME='$time' DATE='$date'"
  375. echo "$str" >> $USER_DATA/notifications.conf
  376. if [ -z "$(grep NOTIFICATIONS $USER_DATA/user.conf)" ]; then
  377. sed -i "s/^TIME/NOTIFICATIONS='yes'\nTIME/g" $USER_DATA/user.conf
  378. else
  379. update_user_value "$user" '$NOTIFICATIONS' "yes"
  380. fi
  381. fi
  382. }
  383. # Recalculate U_DISK value
  384. recalc_user_disk_usage() {
  385. u_usage=0
  386. if [ -f "$USER_DATA/web.conf" ]; then
  387. usage=0
  388. dusage=$(grep 'U_DISK=' $USER_DATA/web.conf |\
  389. awk -F "U_DISK='" '{print $2}' | cut -f 1 -d \')
  390. for disk_usage in $dusage; do
  391. usage=$((usage + disk_usage))
  392. done
  393. d=$(grep "U_DISK_WEB='" $USER_DATA/user.conf | cut -f 2 -d \')
  394. sed -i "s/U_DISK_WEB='$d'/U_DISK_WEB='$usage'/g" $USER_DATA/user.conf
  395. u_usage=$((u_usage + usage))
  396. fi
  397. if [ -f "$USER_DATA/mail.conf" ]; then
  398. usage=0
  399. dusage=$(grep 'U_DISK=' $USER_DATA/mail.conf |\
  400. awk -F "U_DISK='" '{print $2}' | cut -f 1 -d \')
  401. for disk_usage in $dusage; do
  402. usage=$((usage + disk_usage))
  403. done
  404. d=$(grep "U_DISK_MAIL='" $USER_DATA/user.conf | cut -f 2 -d \')
  405. sed -i "s/U_DISK_MAIL='$d'/U_DISK_MAIL='$usage'/g" $USER_DATA/user.conf
  406. u_usage=$((u_usage + usage))
  407. fi
  408. if [ -f "$USER_DATA/db.conf" ]; then
  409. usage=0
  410. dusage=$(grep 'U_DISK=' $USER_DATA/db.conf |\
  411. awk -F "U_DISK='" '{print $2}' | cut -f 1 -d \')
  412. for disk_usage in $dusage; do
  413. usage=$((usage + disk_usage))
  414. done
  415. d=$(grep "U_DISK_DB='" $USER_DATA/user.conf | cut -f 2 -d \')
  416. sed -i "s/U_DISK_DB='$d'/U_DISK_DB='$usage'/g" $USER_DATA/user.conf
  417. u_usage=$((u_usage + usage))
  418. fi
  419. usage=$(grep 'U_DISK_DIRS=' $USER_DATA/user.conf | cut -f 2 -d "'")
  420. u_usage=$((u_usage + usage))
  421. old=$(grep "U_DISK='" $USER_DATA/user.conf | cut -f 2 -d \')
  422. sed -i "s/U_DISK='$old'/U_DISK='$u_usage'/g" $USER_DATA/user.conf
  423. }
  424. # Recalculate U_BANDWIDTH value
  425. recalc_user_bandwidth_usage() {
  426. usage=0
  427. bandwidth_usage=$(grep 'U_BANDWIDTH=' $USER_DATA/web.conf |\
  428. awk -F "U_BANDWIDTH='" '{print $2}'|cut -f 1 -d \')
  429. for bandwidth in $bandwidth_usage; do
  430. usage=$((usage + bandwidth))
  431. done
  432. old=$(grep "U_BANDWIDTH='" $USER_DATA/user.conf | cut -f 2 -d \')
  433. sed -i "s/U_BANDWIDTH='$old'/U_BANDWIDTH='$usage'/g" $USER_DATA/user.conf
  434. }
  435. # Get next cron job id
  436. get_next_cronjob() {
  437. if [ -z "$job" ]; then
  438. curr_str=$(grep "JOB=" $USER_DATA/cron.conf|cut -f 2 -d \'|\
  439. sort -n|tail -n1)
  440. job="$((curr_str +1))"
  441. fi
  442. }
  443. # Sort cron jobs by id
  444. sort_cron_jobs() {
  445. cat $USER_DATA/cron.conf |sort -n -k 2 -t \' > $USER_DATA/cron.tmp
  446. mv -f $USER_DATA/cron.tmp $USER_DATA/cron.conf
  447. }
  448. # Sync cronjobs with system cron
  449. sync_cron_jobs() {
  450. source $USER_DATA/user.conf
  451. if [ -e "/var/spool/cron/crontabs" ]; then
  452. crontab="/var/spool/cron/crontabs/$user"
  453. else
  454. crontab="/var/spool/cron/$user"
  455. fi
  456. rm -f $crontab
  457. if [ "$CRON_REPORTS" = 'yes' ]; then
  458. echo "MAILTO=$CONTACT" > $crontab
  459. echo 'CONTENT_TYPE="text/plain; charset=utf-8"' >> $crontab
  460. fi
  461. while read line; do
  462. eval $line
  463. if [ "$SUSPENDED" = 'no' ]; then
  464. echo "$MIN $HOUR $DAY $MONTH $WDAY $CMD" |\
  465. sed -e "s/%quote%/'/g" -e "s/%dots%/:/g" \
  466. >> $crontab
  467. fi
  468. done < $USER_DATA/cron.conf
  469. chown $user:$user $crontab
  470. chmod 600 $crontab
  471. }
  472. # User format validator
  473. is_user_format_valid() {
  474. if [ ${#1} -eq 1 ]; then
  475. if ! [[ "$1" =~ ^^[[:alnum:]]$ ]]; then
  476. check_result $E_INVALID "invalid $2 format :: $1"
  477. fi
  478. else
  479. if ! [[ "$1" =~ ^[[:alnum:]][-|\.|_[:alnum:]]{0,28}[[:alnum:]]$ ]]
  480. then
  481. check_result $E_INVALID "invalid $2 format :: $1"
  482. fi
  483. fi
  484. }
  485. # Domain format validator
  486. is_domain_format_valid() {
  487. object_name=${2-domain}
  488. exclude="[!|@|#|$|^|&|*|(|)|+|=|{|}|:|,|<|>|?|_|/|\|\"|'|;|%|\`| ]"
  489. if [[ $1 =~ $exclude ]] || [[ $1 =~ ^[0-9]+$ ]] || [[ $1 =~ "\.\." ]] || [[ $1 =~ "$(printf '\t')" ]]; then
  490. check_result $E_INVALID "invalid $object_name format :: $1"
  491. fi
  492. }
  493. # Alias forman validator
  494. is_alias_format_valid() {
  495. for object in ${1//,/ }; do
  496. exclude="[!|@|#|$|^|&|(|)|+|=|{|}|:|<|>|?|_|/|\|\"|'|;|%|\`| ]"
  497. if [[ "$object" =~ $exclude ]]; then
  498. check_result $E_INVALID "invalid alias format :: $object"
  499. fi
  500. if [[ "$object" =~ [*] ]] && ! [[ "$object" =~ ^[*]\..* ]]; then
  501. check_result $E_INVALID "invalid alias format :: $object"
  502. fi
  503. done
  504. }
  505. # IP format validator
  506. is_ip_format_valid() {
  507. object_name=${2-ip}
  508. ip_regex='([1-9]?[0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])'
  509. ip_clean=$(echo "${1%/*}")
  510. if ! [[ $ip_clean =~ ^$ip_regex\.$ip_regex\.$ip_regex\.$ip_regex$ ]]; then
  511. check_result $E_INVALID "invalid $object_name format :: $1"
  512. fi
  513. if [ $1 != "$ip_clean" ]; then
  514. ip_cidr="$ip_clean/"
  515. ip_cidr=$(echo "${1#$ip_cidr}")
  516. if [[ "$ip_cidr" -gt 32 ]] || [[ "$ip_cidr" =~ [:alnum:] ]]; then
  517. check_result $E_INVALID "invalid $object_name format :: $1"
  518. fi
  519. fi
  520. }
  521. # Proxy extention format validator
  522. is_extention_format_valid() {
  523. exclude="[!|#|$|^|&|(|)|+|=|{|}|:|@|<|>|?|/|\|\"|'|;|%|\`| ]"
  524. if [[ "$1" =~ $exclude ]]; then
  525. check_result $E_INVALID "invalid proxy extention format :: $1"
  526. fi
  527. }
  528. # Number format validator
  529. is_number_format_valid() {
  530. object_name=${2-number}
  531. if ! [[ "$1" =~ ^[0-9]+$ ]] ; then
  532. check_result $E_INVALID "invalid $object_name format :: $1"
  533. fi
  534. }
  535. # Autoreply format validator
  536. is_autoreply_format_valid() {
  537. if [[ "$1" =~ [$|\`] ]] || [ 10240 -le ${#1} ]; then
  538. check_result $E_INVALID "invalid autoreply format :: $1"
  539. fi
  540. }
  541. # Boolean format validator
  542. is_boolean_format_valid() {
  543. if [ "$1" != 'yes' ] && [ "$1" != 'no' ]; then
  544. check_result $E_INVALID "invalid $2 format :: $1"
  545. fi
  546. }
  547. # Common format validator
  548. is_common_format_valid() {
  549. exclude="[!|#|$|^|&|(|)|+|=|{|}|:|<|>|?|/|\|\"|'|;|%|\`| ]"
  550. if [[ "$1" =~ $exclude ]]; then
  551. check_result $E_INVALID "invalid $2 format :: $1"
  552. fi
  553. if [ 400 -le ${#1} ]; then
  554. check_result $E_INVALID "invalid $2 format :: $1"
  555. fi
  556. if [[ "$1" =~ @ ]] && [ ${#1} -gt 1 ] ; then
  557. check_result $E_INVALID "invalid $2 format :: $1"
  558. fi
  559. if [[ $1 =~ \* ]]; then
  560. if [[ "$(echo $1 | grep -o '\*\.' |wc -l)" -eq 0 ]] && [[ $1 != '*' ]] ; then
  561. check_result $E_INVALID "invalid $2 format :: $1"
  562. fi
  563. fi
  564. if [[ $(echo -n "$1" | tail -c 1) =~ [^a-zA-Z0-9_*@] ]]; then
  565. check_result $E_INVALID "invalid $2 format :: $1"
  566. fi
  567. if [[ $(echo -n "$1" | grep -c '\.\.') -gt 0 ]];then
  568. check_result $E_INVALID "invalid $2 format :: $1"
  569. fi
  570. if [[ $(echo -n "$1" | head -c 1) =~ [^a-zA-Z0-9_*@] ]]; then
  571. check_result $E_INVALID "invalid $2 format :: $1"
  572. fi
  573. if [[ $(echo -n "$1" | grep -c '\-\-') -gt 0 ]]; then
  574. check_result $E_INVALID "invalid $2 format :: $1"
  575. fi
  576. if [[ $(echo -n "$1" | grep -c '\_\_') -gt 0 ]]; then
  577. check_result $E_INVALID "invalid $2 format :: $1"
  578. fi
  579. }
  580. # Database format validator
  581. is_database_format_valid() {
  582. exclude="[!|@|#|$|^|&|*|(|)|+|=|{|}|:|,|<|>|?|/|\|\"|'|;|%|\`| ]"
  583. if [[ "$1" =~ $exclude ]] || [ 65 -le ${#1} ]; then
  584. check_result $E_INVALID "invalid $2 format :: $1"
  585. fi
  586. }
  587. # Date format validator
  588. is_date_format_valid() {
  589. if ! [[ "$1" =~ ^[0-9][0-9][0-9][0-9]-[0-9][0-9]-[0-9][0-9]$ ]]; then
  590. check_result $E_INVALID "invalid date format :: $1"
  591. fi
  592. }
  593. # Database user validator
  594. is_dbuser_format_valid() {
  595. exclude="[!|@|#|$|^|&|*|(|)|+|=|{|}|:|,|<|>|?|/|\|\"|'|;|%|\`| ]"
  596. if [ 17 -le ${#1} ]; then
  597. check_result $E_INVALID "mysql username can be up to 16 characters long"
  598. fi
  599. if [[ "$1" =~ $exclude ]]; then
  600. check_result $E_INVALID "invalid $2 format :: $1"
  601. fi
  602. }
  603. # DNS record type validator
  604. is_dns_type_format_valid() {
  605. known_dnstype='A,AAAA,NS,CNAME,MX,TXT,SRV,DNSKEY,KEY,IPSECKEY,PTR,SPF,TLSA,CAA'
  606. if [ -z "$(echo $known_dnstype |grep -w $1)" ]; then
  607. check_result $E_INVALID "invalid dns record type format :: $1"
  608. fi
  609. }
  610. # DNS record validator
  611. is_dns_record_format_valid() {
  612. if [ "$rtype" = 'A' ]; then
  613. is_ip_format_valid "$1"
  614. fi
  615. if [ "$rtype" = 'NS' ]; then
  616. is_domain_format_valid "${1::-1}" 'ns_record'
  617. fi
  618. if [ "$rtype" = 'MX' ]; then
  619. is_domain_format_valid "${1::-1}" 'mx_record'
  620. is_int_format_valid "$priority" 'priority_record'
  621. fi
  622. }
  623. # Email format validator
  624. is_email_format_valid() {
  625. if [[ ! "$1" =~ ^[A-Za-z0-9._%+-]+@[[:alnum:].-]+\.[A-Za-z]{2,63}$ ]] ; then
  626. check_result $E_INVALID "invalid email format :: $1"
  627. fi
  628. }
  629. # Firewall action validator
  630. is_fw_action_format_valid() {
  631. if [ "$1" != "ACCEPT" ] && [ "$1" != 'DROP' ] ; then
  632. check_result $E_INVALID "invalid action format :: $1"
  633. fi
  634. }
  635. # Firewall protocol validator
  636. is_fw_protocol_format_valid() {
  637. if [ "$1" != "ICMP" ] && [ "$1" != 'UDP' ] && [ "$1" != 'TCP' ] ; then
  638. check_result $E_INVALID "invalid protocol format :: $1"
  639. fi
  640. }
  641. # Firewall port validator
  642. is_fw_port_format_valid() {
  643. if [ "${#1}" -eq 1 ]; then
  644. if ! [[ "$1" =~ [0-9] ]]; then
  645. check_result $E_INVALID "invalid port format :: $1"
  646. fi
  647. else
  648. if ! [[ "$1" =~ ^[0-9][-|,|:|0-9]{0,30}[0-9]$ ]]
  649. then
  650. check_result $E_INVALID "invalid port format :: $1"
  651. fi
  652. fi
  653. }
  654. # Integer validator
  655. is_int_format_valid() {
  656. if ! [[ "$1" =~ ^[0-9]+$ ]] ; then
  657. check_result $E_INVALID "invalid $2 format :: $1"
  658. fi
  659. }
  660. # Interface validator
  661. is_interface_format_valid() {
  662. netdevices=$(cat /proc/net/dev |grep : |cut -f 1 -d : |tr -d ' ')
  663. if [ -z $(echo "$netdevices" |grep -x $1) ]; then
  664. check_result $E_INVALID "invalid interface format :: $1"
  665. fi
  666. }
  667. # IP status validator
  668. is_ip_status_format_valid() {
  669. if [ -z "$(echo shared,dedicated | grep -w $1 )" ]; then
  670. check_result $E_INVALID "invalid status format :: $1"
  671. fi
  672. }
  673. # Cron validator
  674. is_cron_format_valid() {
  675. limit=59
  676. check_format=''
  677. if [ "$2" = 'hour' ]; then
  678. limit=23
  679. fi
  680. if [ "$2" = 'day' ]; then
  681. limit=31
  682. fi
  683. if [ "$2" = 'month' ]; then
  684. limit=12
  685. fi
  686. if [ "$2" = 'wday' ]; then
  687. limit=7
  688. fi
  689. if [ "$1" = '*' ]; then
  690. check_format='ok'
  691. fi
  692. if [[ "$1" =~ ^[\*]+[/]+[0-9] ]]; then
  693. if [ "$(echo $1 |cut -f 2 -d /)" -lt $limit ]; then
  694. check_format='ok'
  695. fi
  696. fi
  697. if [[ "$1" =~ ^[0-9][-|,|0-9]{0,70}[\/][0-9]$ ]]; then
  698. check_format='ok'
  699. crn_values=${1//,/ }
  700. crn_values=${crn_values//-/ }
  701. crn_values=${crn_values//\// }
  702. for crn_vl in $crn_values; do
  703. if [ "$crn_vl" -gt $limit ]; then
  704. check_format='invalid'
  705. fi
  706. done
  707. fi
  708. crn_values=$(echo $1 |tr "," " " | tr "-" " ")
  709. for crn_vl in $crn_values
  710. do
  711. if [[ "$crn_vl" =~ ^[0-9]+$ ]] && [ "$crn_vl" -le $limit ]; then
  712. check_format='ok'
  713. fi
  714. done
  715. if [ "$check_format" != 'ok' ]; then
  716. check_result $E_INVALID "invalid $2 format :: $1"
  717. fi
  718. }
  719. # Name validator
  720. is_name_format_valid() {
  721. if ! [[ "$1" =~ ^[[:alnum:]][-|\ |\.|_[:alnum:]]{0,28}[[:alnum:]]$ ]]; then
  722. check_result $E_INVALID "invalid $2 format :: $1"
  723. fi
  724. }
  725. # Object validator
  726. is_object_format_valid() {
  727. if ! [[ "$1" =~ ^[[:alnum:]][-|\.|_[:alnum:]]{0,64}[[:alnum:]]$ ]]; then
  728. check_result $E_INVALID "invalid $2 format :: $1"
  729. fi
  730. }
  731. # Password validator
  732. is_password_format_valid() {
  733. if [ "${#1}" -lt '6' ]; then
  734. check_result $E_INVALID "invalid password format :: $1"
  735. fi
  736. }
  737. # Missing function -
  738. # Before: validate_format_shell
  739. # After: is_format_valid_shell
  740. is_format_valid_shell() {
  741. if [ -z "$(grep -w $1 /etc/shells)" ]; then
  742. echo "Error: shell $1 is not valid"
  743. log_event "$E_INVALID" "$EVENT"
  744. exit $E_INVALID
  745. fi
  746. }
  747. # Format validation controller
  748. is_format_valid() {
  749. for arg_name in $*; do
  750. eval arg=\$$arg_name
  751. if [ ! -z "$arg" ]; then
  752. case $arg_name in
  753. account) is_user_format_valid "$arg" "$arg_name";;
  754. action) is_fw_action_format_valid "$arg";;
  755. aliases) is_alias_format_valid "$arg" ;;
  756. antispam) is_boolean_format_valid "$arg" 'antispam' ;;
  757. antivirus) is_boolean_format_valid "$arg" 'antivirus' ;;
  758. autoreply) is_autoreply_format_valid "$arg" ;;
  759. backup) is_object_format_valid "$arg" 'backup' ;;
  760. charset) is_object_format_valid "$arg" "$arg_name" ;;
  761. charsets) is_common_format_valid "$arg" 'charsets' ;;
  762. comment) is_object_format_valid "$arg" 'comment' ;;
  763. database) is_database_format_valid "$arg" 'database';;
  764. day) is_cron_format_valid "$arg" $arg_name ;;
  765. dbpass) is_password_format_valid "$arg" ;;
  766. dbuser) is_dbuser_format_valid "$arg" 'dbuser';;
  767. dkim) is_boolean_format_valid "$arg" 'dkim' ;;
  768. dkim_size) is_int_format_valid "$arg" ;;
  769. domain) is_domain_format_valid "$arg" ;;
  770. dvalue) is_dns_record_format_valid "$arg";;
  771. email) is_email_format_valid "$arg" ;;
  772. exp) is_date_format_valid "$arg" ;;
  773. extentions) is_common_format_valid "$arg" 'extentions' ;;
  774. fname) is_name_format_valid "$arg" "first name" ;;
  775. ftp_password) is_password_format_valid "$arg" ;;
  776. ftp_user) is_user_format_valid "$arg" "$arg_name" ;;
  777. host) is_object_format_valid "$arg" "$arg_name" ;;
  778. hour) is_cron_format_valid "$arg" $arg_name ;;
  779. id) is_int_format_valid "$arg" 'id' ;;
  780. ip) is_ip_format_valid "$arg" ;;
  781. ip_name) is_domain_format_valid "$arg" 'IP name';;
  782. ip_status) is_ip_status_format_valid "$arg" ;;
  783. job) is_int_format_valid "$arg" 'job' ;;
  784. key) is_user_format_valid "$arg" "$arg_name" ;;
  785. lname) is_name_format_valid "$arg" "last name" ;;
  786. malias) is_user_format_valid "$arg" "$arg_name" ;;
  787. max_db) is_int_format_valid "$arg" 'max db';;
  788. min) is_cron_format_valid "$arg" $arg_name ;;
  789. month) is_cron_format_valid "$arg" $arg_name ;;
  790. nat_ip) is_ip_format_valid "$arg" ;;
  791. netmask) is_ip_format_valid "$arg" 'netmask' ;;
  792. newid) is_int_format_valid "$arg" 'id' ;;
  793. ns1) is_domain_format_valid "$arg" 'ns1' ;;
  794. ns2) is_domain_format_valid "$arg" 'ns2' ;;
  795. ns3) is_domain_format_valid "$arg" 'ns3' ;;
  796. ns4) is_domain_format_valid "$arg" 'ns4' ;;
  797. ns5) is_domain_format_valid "$arg" 'ns5' ;;
  798. ns6) is_domain_format_valid "$arg" 'ns6' ;;
  799. ns7) is_domain_format_valid "$arg" 'ns7' ;;
  800. ns8) is_domain_format_valid "$arg" 'ns8' ;;
  801. object) is_name_format_valid "$arg" 'object';;
  802. package) is_object_format_valid "$arg" "$arg_name" ;;
  803. password) is_password_format_valid "$arg" ;;
  804. port) is_int_format_valid "$arg" 'port' ;;
  805. port_ext) is_fw_port_format_valid "$arg";;
  806. protocol) is_fw_protocol_format_valid "$arg" ;;
  807. proxy_ext) is_extention_format_valid "$arg" ;;
  808. quota) is_int_format_valid "$arg" 'quota' ;;
  809. record) is_common_format_valid "$arg" 'record';;
  810. restart) is_boolean_format_valid "$arg" 'restart' ;;
  811. rtype) is_dns_type_format_valid "$arg" ;;
  812. rule) is_int_format_valid "$arg" "rule id" ;;
  813. soa) is_domain_format_valid "$arg" 'SOA' ;;
  814. #missing command: is_format_valid_shell
  815. shell) is_format_valid_shell "$arg" ;;
  816. stats_pass) is_password_format_valid "$arg" ;;
  817. stats_user) is_user_format_valid "$arg" "$arg_name" ;;
  818. template) is_object_format_valid "$arg" "$arg_name" ;;
  819. ttl) is_int_format_valid "$arg" 'ttl';;
  820. user) is_user_format_valid "$arg" $arg_name;;
  821. wday) is_cron_format_valid "$arg" $arg_name ;;
  822. esac
  823. fi
  824. done
  825. }
  826. # Domain argument formatting
  827. format_domain() {
  828. if [[ "$domain" = *[![:ascii:]]* ]]; then
  829. if [[ "$domain" =~ [[:upper:]] ]]; then
  830. domain=$(echo "$domain" |sed 's/[[:upper:]].*/\L&/')
  831. fi
  832. else
  833. if [[ "$domain" =~ [[:upper:]] ]]; then
  834. domain=$(echo "$domain" |tr '[:upper:]' '[:lower:]')
  835. fi
  836. fi
  837. if [[ "$domain" =~ ^www\..* ]]; then
  838. domain=$(echo "$domain" |sed -e "s/^www.//")
  839. fi
  840. if [[ "$domain" =~ .*\.$ ]]; then
  841. domain=$(echo "$domain" |sed -e "s/[.]*$//g")
  842. fi
  843. if [[ "$domain" =~ ^\. ]]; then
  844. domain=$(echo "$domain" |sed -e "s/^[.]*//")
  845. fi
  846. }
  847. format_domain_idn() {
  848. if [ -z "$domain_idn" ]; then
  849. domain_idn=$domain
  850. fi
  851. if [[ "$domain_idn" = *[![:ascii:]]* ]]; then
  852. domain_idn=$(idn -t --quiet -a $domain_idn)
  853. fi
  854. }
  855. format_aliases() {
  856. if [ ! -z "$aliases" ] && [ "$aliases" != 'none' ]; then
  857. aliases=$(echo $aliases |tr '[:upper:]' '[:lower:]' |tr ',' '\n')
  858. aliases=$(echo "$aliases" |sed -e "s/\.$//" |sort -u)
  859. aliases=$(echo "$aliases" |tr -s '.')
  860. aliases=$(echo "$aliases" |sed -e "s/[.]*$//g")
  861. aliases=$(echo "$aliases" |sed -e "s/^[.]*//")
  862. aliases=$(echo "$aliases" |sed -e "/^$/d")
  863. aliases=$(echo "$aliases" |tr '\n' ',' |sed -e "s/,$//")
  864. fi
  865. }