main.sh 24 KB

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