main.sh 28 KB

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