main.sh 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859
  1. # Internal variables
  2. DATE=$(date +%F)
  3. TIME=$(date +%T)
  4. SCRIPT=$(basename $0)
  5. A1=$1
  6. A2=$2
  7. A3=$3
  8. A4=$4
  9. A5=$5
  10. A6=$6
  11. A7=$7
  12. A8=$8
  13. A9=$9
  14. EVENT="$DATE $TIME $SCRIPT $A1 $A2 $A3 $A4 $A5 $A6 $A7 $A8 $A9"
  15. HOMEDIR='/home'
  16. BACKUP='/backup'
  17. BACKUP_GZIP=5
  18. BACKUP_DISK_LIMIT=95
  19. BACKUP_LA_LIMIT=5
  20. RRD_STEP=300
  21. RRD_IFACE_EXCLUDE=lo
  22. BIN=$VESTA/bin
  23. USER_DATA=$VESTA/data/users/$user
  24. WEBTPL=$VESTA/data/templates/web
  25. DNSTPL=$VESTA/data/templates/dns
  26. RRD=$VESTA/web/rrd
  27. # Return codes
  28. OK=0
  29. E_ARGS=1
  30. E_INVALID=2
  31. E_NOTEXIST=3
  32. E_EXISTS=4
  33. E_SUSPENDED=5
  34. E_UNSUSPENDED=6
  35. E_INUSE=7
  36. E_LIMIT=8
  37. E_PASSWORD=9
  38. E_FORBIDEN=10
  39. E_DISABLED=11
  40. E_PARSING=12
  41. E_DISK=13
  42. E_LA=14
  43. E_CONNECT=15
  44. E_FTP=16
  45. E_DB=17
  46. E_RRD=18
  47. E_UPDATE=19
  48. E_RESTART=20
  49. # Log event function
  50. log_event() {
  51. if [ "$1" -eq 0 ]; then
  52. echo "$2" >> $VESTA/log/system.log
  53. else
  54. echo "$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. curr_str=$(grep "ID=" $log | cut -f 2 -d \' | sort -n | tail -n1)
  70. id="$((curr_str +1))"
  71. echo "ID='$id' DATE='$DATE' TIME='$TIME' CMD='$cmd' UNDO='$undo'" >> $log
  72. }
  73. # Argument list checker
  74. check_args() {
  75. if [ "$1" -gt "$2" ]; then
  76. echo "Error: not enought arguments"
  77. echo "Usage: $SCRIPT $3"
  78. log_event "$E_ARGS" "$EVENT"
  79. exit $E_ARGS
  80. fi
  81. }
  82. # Subsystem checker
  83. is_system_enabled() {
  84. if [ -z "$1" ] || [ "$1" = no ]; then
  85. echo "Error: $2 is disabled in the vesta.conf"
  86. log_event "$E_DISABLED" "$EVENT"
  87. exit $E_DISABLED
  88. fi
  89. }
  90. # User package check
  91. is_package_full() {
  92. case "$1" in
  93. WEB_DOMAINS) used=$(wc -l $USER_DATA/web.conf|cut -f1 -d \ );;
  94. WEB_ALIASES) used=$(grep "DOMAIN='$domain'" $USER_DATA/web.conf |\
  95. awk -F "ALIAS='" '{print $2}' | cut -f 1 -d \' | tr ',' '\n' |\
  96. wc -l );;
  97. DNS_DOMAINS) used=$(wc -l $USER_DATA/dns.conf |cut -f1 -d \ );;
  98. DNS_RECORDS) used=$(wc -l $USER_DATA/dns/$domain.conf |cut -f1 -d \ );;
  99. MAIL_DOMAINS) used=$(wc -l $USER_DATA/mail.conf |cut -f1 -d \ );;
  100. MAIL_ACCOUNTS) used=$(wc -l $USER_DATA/mail/$domain.conf |\
  101. cut -f1 -d \ );;
  102. DATABASES) used=$(wc -l $USER_DATA/db.conf |cut -f1 -d \ );;
  103. CRON_JOBS) used=$(wc -l $USER_DATA/cron.conf |cut -f1 -d \ );;
  104. esac
  105. limit=$(grep "^$1=" $USER_DATA/user.conf | cut -f 2 -d \' )
  106. if [ "$used" -ge "$limit" ]; then
  107. echo "Error: Limit reached / Upgrade package"
  108. log_event "$E_LIMIT" "$EVENT"
  109. exit $E_LIMIT
  110. fi
  111. }
  112. # Random password generator
  113. gen_password() {
  114. matrix='0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz'
  115. lenght=10
  116. while [ ${n:=1} -le $lenght ]; do
  117. pass="$pass${matrix:$(($RANDOM%${#matrix})):1}"
  118. let n+=1
  119. done
  120. echo "$pass"
  121. }
  122. # Package existance check
  123. is_package_valid() {
  124. if [ -z "$1" ]; then
  125. pkg_dir="$VESTA/data/packages"
  126. fi
  127. if [ ! -e "$pkg_dir/$package.pkg" ]; then
  128. echo "Error: package $package doesn't exist"
  129. log_event "$E_NOTEXIST $EVENT"
  130. exit $E_NOTEXIST
  131. fi
  132. }
  133. # Validate system type
  134. is_type_valid() {
  135. if [ -z "$(echo $1 | grep -w $2)" ]; then
  136. echo "Error: $2 is unknown type"
  137. log_event "$E_INVALID" "$EVENT"
  138. exit $E_INVALID
  139. fi
  140. }
  141. # Check if backup is available for user
  142. is_backup_available() {
  143. if [ "$user" != "$(echo $backup | cut -f 1 -d '.')" ]; then
  144. echo "Error: User $user don't have permission to use $backup"
  145. log_event "$E_FORBIDEN" "$EVENT"
  146. exit $E_FORBIDEN
  147. fi
  148. }
  149. # Check user backup settings
  150. is_backup_enabled() {
  151. BACKUPS=$(grep "^BACKUPS=" $USER_DATA/user.conf | cut -f2 -d \')
  152. if [ -z "$BACKUPS" ] || [[ "$BACKUPS" -le '0' ]]; then
  153. echo "Error: user backup disabled"
  154. log_event "$E_DISABLED" "$EVENT"
  155. exit $E_DISABLED
  156. fi
  157. }
  158. # Check user backup settings
  159. is_backup_scheduled() {
  160. if [ -e "$VESTA/data/queue/backup.pipe" ]; then
  161. check_q=$(grep " $user " $VESTA/data/queue/backup.pipe | grep $1)
  162. if [ ! -z "$check_q" ]; then
  163. echo "Error: $1 is already scheduled"
  164. log_event "$E_EXISTS" "$EVENT"
  165. exit $E_EXISTS
  166. fi
  167. fi
  168. }
  169. # Check if object is new
  170. is_object_new() {
  171. if [ $2 = 'USER' ]; then
  172. if [ -d "$USER_DATA" ]; then
  173. object="OK"
  174. fi
  175. else
  176. object=$(grep "$2='$3'" $USER_DATA/$1.conf)
  177. fi
  178. if [ ! -z "$object" ]; then
  179. echo "Error: $2 with value $3 exists"
  180. log_event "$E_EXISTS" "$EVENT"
  181. exit $E_EXISTS
  182. fi
  183. }
  184. # Check if object exists and can be used
  185. is_object_valid() {
  186. if [ $2 = 'USER' ]; then
  187. if [ -d "$VESTA/data/users/$user" ]; then
  188. sobject="OK"
  189. fi
  190. else
  191. if [ $2 = 'DBHOST' ]; then
  192. sobject=$(grep "HOST='$host'" $VESTA/conf/$type.conf)
  193. else
  194. sobject=$(grep "$2='$3'" $VESTA/data/users/$user/$1.conf)
  195. fi
  196. fi
  197. if [ -z "$sobject" ]; then
  198. echo "Error: $2 $3 doesn't exist"
  199. log_event "$E_NOTEXIST" "$EVENT"
  200. exit $E_NOTEXIST
  201. fi
  202. }
  203. # Check if object is supended
  204. is_object_suspended() {
  205. if [ $2 = 'USER' ]; then
  206. spnd=$(cat $USER_DATA/$1.conf|grep "SUSPENDED='yes'")
  207. else
  208. spnd=$(grep "$2='$3'" $USER_DATA/$1.conf|grep "SUSPENDED='yes'")
  209. fi
  210. if [ -z "$spnd" ]; then
  211. echo "Error: $(basename $1) $3 is not suspended"
  212. log_event "$E_SUSPENDED" "$EVENT"
  213. exit $E_SUSPENDED
  214. fi
  215. }
  216. # Check if object is unsupended
  217. is_object_unsuspended() {
  218. if [ $2 = 'USER' ]; then
  219. spnd=$(cat $USER_DATA/$1.conf|grep "SUSPENDED='yes'")
  220. else
  221. spnd=$(grep "$2='$3'" $USER_DATA/$1.conf|grep "SUSPENDED='yes'")
  222. fi
  223. if [ ! -z "$spnd" ]; then
  224. echo "Error: $(basename $1) $3 is already suspended"
  225. log_event "$E_UNSUSPENDED" "$EVENT"
  226. exit $E_UNSUSPENDED
  227. fi
  228. }
  229. # Check if object value is empty
  230. is_object_value_empty() {
  231. str=$(grep "$2='$3'" $USER_DATA/$1.conf)
  232. eval $str
  233. eval value=$4
  234. if [ ! -z "$value" ] && [ "$value" != 'no' ]; then
  235. echo "Error: ${4//$}=$value (not empty)"
  236. log_event "$E_EXISTS" "$EVENT"
  237. exit $E_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. echo "Error: ${4//$}=$value (doesn't exist)"
  247. log_event "$E_NOTEXIST" "$EVENT"
  248. exit $E_NOTEXIST
  249. fi
  250. }
  251. # Get object value
  252. get_object_value() {
  253. object=$(grep "$2='$3'" $USER_DATA/$1.conf)
  254. eval "$object"
  255. eval echo $4
  256. }
  257. # Update object value
  258. update_object_value() {
  259. row=$(grep -n "$2='$3'" $USER_DATA/$1.conf)
  260. lnr=$(echo $row | cut -f 1 -d ':')
  261. object=$(echo $row | sed "s/^$lnr://")
  262. eval "$object"
  263. eval old="$4"
  264. old=$(echo "$old" | sed -e 's/\\/\\\\/g' -e 's/&/\\&/g' -e 's/\//\\\//g')
  265. new=$(echo "$5" | sed -e 's/\\/\\\\/g' -e 's/&/\\&/g' -e 's/\//\\\//g')
  266. sed -i "$lnr s/${4//$/}='${old//\*/\\*}'/${4//$/}='${new//\*/\\*}'/g" \
  267. $USER_DATA/$1.conf
  268. }
  269. # Add object key
  270. add_object_key() {
  271. row=$(grep -n "$2='$3'" $USER_DATA/$1.conf)
  272. lnr=$(echo $row | cut -f 1 -d ':')
  273. object=$(echo $row | sed "s/^$lnr://")
  274. if [ -z "$(echo $object |grep $4=)" ]; then
  275. eval old="$4"
  276. sed -i "$lnr s/$5='/$4='' $5='/" $USER_DATA/$1.conf
  277. fi
  278. }
  279. # Search objects
  280. search_objects() {
  281. OLD_IFS="$IFS"
  282. IFS=$'\n'
  283. for line in $(grep $2=\'$3\' $USER_DATA/$1.conf); do
  284. eval $line
  285. eval echo \$$4
  286. done
  287. IFS="$OLD_IFS"
  288. }
  289. # Get user value
  290. get_user_value() {
  291. grep "^${1//$/}=" $USER_DATA/user.conf| cut -f 2 -d \'
  292. }
  293. # Update user value in user.conf
  294. update_user_value() {
  295. key="${2//$}"
  296. lnr=$(grep -n "^$key='" $VESTA/data/users/$1/user.conf |cut -f 1 -d ':')
  297. if [ ! -z "$lnr" ]; then
  298. sed -i "$lnr d" $VESTA/data/users/$1/user.conf
  299. sed -i "$lnr i\\$key='${3}'" $VESTA/data/users/$1/user.conf
  300. fi
  301. }
  302. # Increase user counter
  303. increase_user_value() {
  304. key="${2//$}"
  305. factor="${3-1}"
  306. conf="$VESTA/data/users/$1/user.conf"
  307. old=$(grep "$key=" $conf | cut -f 2 -d \')
  308. if [ -z "$old" ]; then
  309. old=0
  310. fi
  311. new=$((old + factor))
  312. sed -i "s/$key='$old'/$key='$new'/g" $conf
  313. }
  314. # Decrease user counter
  315. decrease_user_value() {
  316. key="${2//$}"
  317. factor="${3-1}"
  318. conf="$VESTA/data/users/$1/user.conf"
  319. old=$(grep "$key=" $conf | cut -f 2 -d \')
  320. if [ -z "$old" ]; then
  321. old=0
  322. fi
  323. if [ "$old" -le 1 ]; then
  324. new=0
  325. else
  326. new=$((old - factor))
  327. fi
  328. sed -i "s/$key='$old'/$key='$new'/g" $conf
  329. }
  330. # Json listing function
  331. json_list() {
  332. echo '{'
  333. fileds_count=$(echo $fields| wc -w )
  334. #for line in $(cat $conf); do
  335. while read line; do
  336. eval $line
  337. if [ -n "$data_output" ]; then
  338. echo -e ' },'
  339. fi
  340. i=1
  341. for field in $fields; do
  342. eval value=$field
  343. if [ $i -eq 1 ]; then
  344. (( ++i))
  345. echo -e "\t\"$value\": {"
  346. else
  347. if [ $i -lt $fileds_count ]; then
  348. (( ++i))
  349. echo -e "\t\t\"${field//$/}\": \"$value\","
  350. else
  351. echo -e "\t\t\"${field//$/}\": \"$value\""
  352. data_output=yes
  353. fi
  354. fi
  355. done
  356. done < $conf
  357. if [ "$data_output" = 'yes' ]; then
  358. echo -e ' }'
  359. fi
  360. echo -e '}'
  361. }
  362. # Shell listing function
  363. shell_list() {
  364. if [ -z "$nohead" ] ; then
  365. echo "${fields//$/}"
  366. for a in $fields; do
  367. echo -e "------ \c"
  368. done
  369. echo
  370. fi
  371. while read line ; do
  372. eval $line
  373. for field in $fields; do
  374. eval value=$field
  375. if [ -z "$value" ]; then
  376. value='NULL'
  377. fi
  378. echo -n "$value "
  379. done
  380. echo
  381. done < $conf
  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_DIR_DISK=' $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. sys_cron="/var/spool/cron/crontabs/$user"
  453. else
  454. sys_cron="/var/spool/cron/$user"
  455. fi
  456. rm -f $sys_cron
  457. if [ "$CRON_REPORTS" = 'yes' ]; then
  458. echo "MAILTO=$CONTACT" > $sys_cron
  459. fi
  460. while read line; do
  461. eval $line
  462. if [ "$SUSPENDED" = 'no' ]; then
  463. echo "$MIN $HOUR $DAY $MONTH $WDAY $CMD" |\
  464. sed -e "s/%quote%/'/g" -e "s/%dots%/:/g" \
  465. >> $sys_cron
  466. fi
  467. done < $USER_DATA/cron.conf
  468. # Set proper permissions
  469. chown $user:$user $sys_cron
  470. chmod 600 $sys_cron
  471. }
  472. ### Format Validators ###
  473. # Shell
  474. validate_format_shell() {
  475. if [ -z "$(grep -w $1 /etc/shells)" ]; then
  476. echo "Error: shell $1 is not valid"
  477. log_event "$E_INVALID" "$EVENT"
  478. exit $E_INVALID
  479. fi
  480. }
  481. # Password
  482. validate_format_password() {
  483. if [ "${#1}" -lt '6' ]; then
  484. echo "Error: password is too short"
  485. log_event "$E_INVALID" "$EVENT"
  486. exit $E_INVALID
  487. fi
  488. }
  489. # Integer
  490. validate_format_int() {
  491. if ! [[ "$1" =~ ^[0-9]+$ ]] ; then
  492. echo "Error: $2 $1 is not valid"
  493. log_event "$E_INVALID" "$EVENT"
  494. exit $E_INVALID
  495. fi
  496. }
  497. # Boolean
  498. validate_format_boolean() {
  499. if [ "$1" != 'yes' ] && [ "$1" != 'no' ]; then
  500. echo "Error: $2 $1 is not valid"
  501. log_event "$E_INVALID" "$EVENT"
  502. exit $E_INVALID
  503. fi
  504. }
  505. # Network interface
  506. validate_format_interface() {
  507. netdevices=$(cat /proc/net/dev | grep : | cut -f 1 -d : | tr -d ' ')
  508. if [ -z $(echo "$netdevices"| grep -x $1) ]; then
  509. echo "Error: intreface $1 is not valid"
  510. log_event "$E_INVALID" "$EVENT"
  511. exit $E_INVALID
  512. fi
  513. }
  514. # IP address
  515. validate_format_ip() {
  516. valid_octets=0
  517. for octet in ${1//./ }; do
  518. if [[ $octet =~ ^[0-9]{1,3}$ ]] && [[ $octet -le 255 ]]; then
  519. ((++valid_octets))
  520. fi
  521. done
  522. if [ "$valid_octets" -lt 4 ]; then
  523. echo "Error: ip $1 is not valid"
  524. log_event "$E_INVALID" "$EVENT"
  525. exit $E_INVALID
  526. fi
  527. }
  528. # IP address status
  529. validate_format_ip_status() {
  530. if [ -z "$(echo shared,dedicated | grep -w $1 )" ]; then
  531. echo "Error: ip_status $1 is not valid"
  532. log_event "$E_INVALID" "$EVENT"
  533. exit $E_INVALID
  534. fi
  535. }
  536. # Email address
  537. validate_format_email() {
  538. local_part=$(echo $1 | cut -s -f1 -d\@)
  539. remote_host=$(echo $1 | cut -s -f2 -d\@)
  540. mx_failed=1
  541. if [ ! -z "$remote_host" ] && [ ! -z "$local_part" ]; then
  542. /usr/bin/host -t mx "$remote_host" &> /dev/null
  543. mx_failed="$?"
  544. fi
  545. if [ "$mx_failed" -eq 1 ]; then
  546. echo "Error: email $1 is not valid"
  547. log_event "$E_INVALID" "$EVENT"
  548. exit $E_INVALID
  549. fi
  550. }
  551. # Name
  552. validate_format_name() {
  553. if ! [[ "$1" =~ ^[[:alnum:]][-|\.|_[:alnum:]]{0,28}[[:alnum:]]$ ]]; then
  554. echo "Error: $2 $1 is not valid"
  555. log_event "$E_INVALID" "$EVENT"
  556. exit $E_INVALID
  557. fi
  558. }
  559. # Name with space
  560. validate_format_name_s() {
  561. if ! [[ "$1" =~ ^[[:alnum:]][-|\ |\.|_[:alnum:]]{0,28}[[:alnum:]]$ ]]; then
  562. echo "Error: $2 $1 is not valid"
  563. log_event "$E_INVALID" "$EVENT"
  564. exit $E_INVALID
  565. fi
  566. }
  567. # Username
  568. validate_format_username() {
  569. if [ "${#1}" -eq 1 ]; then
  570. if ! [[ "$1" =~ [a-z] ]]; then
  571. echo "Error: $2 $1 is not valid"
  572. exit 1
  573. fi
  574. else
  575. if ! [[ "$1" =~ ^[a-zA-Z0-9][-|\.|_|a-zA-Z0-9]{0,28}[a-zA-Z0-9]$ ]]
  576. then
  577. echo "Error: $2 $1 is not valid"
  578. exit 1
  579. fi
  580. fi
  581. }
  582. # Domain
  583. validate_format_domain() {
  584. exclude="[!|@|#|$|^|&|*|(|)|+|=|{|}|:|,|<|>|?|_|/|\|\"|'|;|%|\`| ]"
  585. if [[ "$1" =~ $exclude ]] || [[ "$1" =~ "^[0-9]+$" ]]; then
  586. echo "Error: $2 $1 is not valid"
  587. log_event "$E_INVALID" "$EVENT"
  588. exit $E_INVALID
  589. fi
  590. }
  591. # Domain alias
  592. validate_format_domain_alias() {
  593. exclude="[!|@|#|$|^|&|(|)|+|=|{|}|:|,|<|>|?|_|/|\|\"|'|;|%|\`| ]"
  594. if [[ "$1" =~ $exclude ]] || [[ "$1" =~ "^[0-9]+$" ]]; then
  595. echo "Error: domain alias $1 is not valid"
  596. log_event "$E_INVALID" "$EVENT"
  597. exit $E_INVALID
  598. fi
  599. }
  600. # Database
  601. validate_format_database() {
  602. exclude="[!|@|#|$|^|&|*|(|)|+|=|{|}|:|,|<|>|?|/|\|\"|'|;|%|\`| ]"
  603. if [[ "$1" =~ $exclude ]] || [ 65 -le ${#1} ]; then
  604. echo "Error: $2 $1 is not valid"
  605. log_event "$E_INVALID" "$EVENT"
  606. exit $E_INVALID
  607. fi
  608. }
  609. # Database user
  610. validate_format_dbuser() {
  611. exclude="[!|@|#|$|^|&|*|(|)|+|=|{|}|:|,|<|>|?|/|\|\"|'|;|%|\`| ]"
  612. if [[ "$1" =~ $exclude ]] || [ 17 -le ${#1} ]; then
  613. echo "Error: $2 $1 is not valid"
  614. log_event "$E_INVALID" "$EVENT"
  615. exit $E_INVALID
  616. fi
  617. }
  618. # DNS type
  619. validate_format_dns_type() {
  620. known_dnstype='A,AAAA,NS,CNAME,MX,TXT,SRV,DNSKEY,KEY,IPSECKEY,PTR,SPF'
  621. if [ -z "$(echo $known_dnstype | grep -w $1)" ]; then
  622. echo "Error: dnstype $1 is not valid"
  623. log_event "$E_INVALID" "$EVENT"
  624. exit $E_INVALID
  625. fi
  626. }
  627. # DKIM key size
  628. validate_format_key_size() {
  629. known_size='128,256,512,768,1024,2048'
  630. if [ -z "$(echo $known_size | grep -w $1)" ]; then
  631. echo "Error: key_size $1 is not valid"
  632. log_event "$E_INVALID" "$EVENT"
  633. exit $E_INVALID
  634. fi
  635. }
  636. # Minute / Hour / Day / Month / Day of Week
  637. validate_format_mhdmw() {
  638. limit=60
  639. check_format=''
  640. if [ "$2" = 'day' ]; then
  641. limit=31
  642. fi
  643. if [ "$2" = 'month' ]; then
  644. limit=12
  645. fi
  646. if [ "$2" = 'wday' ]; then
  647. limit=7
  648. fi
  649. if [ "$1" = '*' ]; then
  650. check_format='ok'
  651. fi
  652. if [[ "$1" =~ ^[\*]+[/]+[0-9] ]]; then
  653. if [ "$(echo $1 |cut -f 2 -d /)" -lt $limit ]; then
  654. check_format='ok'
  655. fi
  656. fi
  657. if [[ "$1" =~ ^[0-9][-|,|0-9]{0,28}[0-9]$ ]]; then
  658. check_format='ok'
  659. crn_values=${1//,/ }
  660. crn_values=${crn_values//-/ }
  661. for crn_vl in $crn_values; do
  662. if [ "$crn_vl" -gt $limit ]; then
  663. check_format='invalid'
  664. fi
  665. done
  666. fi
  667. if [[ "$1" =~ ^[0-9]+$ ]] && [ "$1" -lt $limit ]; then
  668. check_format='ok'
  669. fi
  670. if [ "$check_format" != 'ok' ]; then
  671. echo "Error: $2 $1 is not valid"
  672. log_event "$E_INVALID" "$EVENT"
  673. exit $E_INVALID
  674. fi
  675. }
  676. # proxy extention or DNS record
  677. validate_format_common() {
  678. exclude="[!|#|$|^|&|(|)|+|=|{|}|:|<|>|?|/|\|\"|'|;|%|\`| ]"
  679. if [[ "$1" =~ $exclude ]] || [ 200 -le ${#1} ]; then
  680. echo "Error: $2 $1 is not valid"
  681. log_event "$E_INVALID" "$EVENT"
  682. exit $E_INVALID
  683. fi
  684. }
  685. # DNS record value
  686. validate_format_dvalue() {
  687. record_types="$(echo A,AAAA,NS,CNAME | grep -w "$rtype")"
  688. if [[ "$1" =~ [\ ] ]] && [ ! -z "$record_types" ]; then
  689. echo "Error: dvalue $1 is not valid"
  690. log_event "$E_INVALID" "$EVENT"
  691. exit $E_INVALID
  692. fi
  693. if [ "$rtype" = 'A' ]; then
  694. validate_format_ip "$1"
  695. fi
  696. if [ "$rtype" = 'NS' ]; then
  697. validate_format_domain "$1" 'ns_record'
  698. fi
  699. if [ "$rtype" = 'MX' ]; then
  700. validate_format_domain "$1" 'mx_record'
  701. validate_format_int "$priority" 'priority_record'
  702. fi
  703. }
  704. # Date
  705. validate_format_date() {
  706. if ! [[ "$1" =~ ^[0-9][0-9][0-9][0-9]-[0-9][0-9]-[0-9][0-9]$ ]]; then
  707. echo "Error: date $1 is not valid"
  708. log_event "$E_INVALID" "$EVENT"
  709. exit $E_INVALID
  710. fi
  711. }
  712. # Autoreply
  713. validate_format_autoreply() {
  714. exclude="[$|\`]"
  715. if [[ "$1" =~ $exclude ]] || [ 10240 -le ${#1} ]; then
  716. echo "Error: autoreply is not valid"
  717. log_event "$E_INVALID" "$EVENT"
  718. exit $E_INVALID
  719. fi
  720. }
  721. # Format validation controller
  722. validate_format(){
  723. for arg_name in $*; do
  724. eval arg=\$$arg_name
  725. if [ -z "$arg" ]; then
  726. echo "Error: argument $arg_name is not valid (empty)"
  727. log_event "$E_INVALID" "$EVENT"
  728. exit $E_INVALID
  729. fi
  730. case $arg_name in
  731. account) validate_format_username "$arg" "$arg_name" ;;
  732. antispam) validate_format_boolean "$arg" 'antispam' ;;
  733. antivirus) validate_format_boolean "$arg" 'antivirus' ;;
  734. autoreply) validate_format_autoreply "$arg" ;;
  735. backup) validate_format_domain "$arg" 'backup' ;;
  736. charset) validate_format_name "$arg" "$arg_name" ;;
  737. charsets) validate_format_common "$arg" 'charsets' ;;
  738. database) validate_format_database "$arg" 'database';;
  739. day) validate_format_mhdmw "$arg" $arg_name ;;
  740. dbpass) validate_format_password "$arg" ;;
  741. dbuser) validate_format_dbuser "$arg" 'db_user';;
  742. dkim) validate_format_boolean "$arg" 'dkim' ;;
  743. dkim_size) validate_format_key_size "$arg" ;;
  744. domain) validate_format_domain "$arg" 'domain';;
  745. dom_alias) validate_format_domain_alias "$arg" 'alias';;
  746. dvalue) validate_format_dvalue "$arg";;
  747. email) validate_format_email "$arg" ;;
  748. exp) validate_format_date "$arg" ;;
  749. extentions) validate_format_common "$arg" 'extentions' ;;
  750. fname) validate_format_name_s "$arg" "$arg_name" ;;
  751. forward) validate_format_email "$arg" ;;
  752. ftp_password) validate_format_password "$arg" ;;
  753. ftp_user) validate_format_username "$arg" "$arg_name" ;;
  754. host) validate_format_domain "$arg" "$arg_name" 'host';;
  755. hour) validate_format_mhdmw "$arg" $arg_name ;;
  756. id) validate_format_int "$arg" 'id' ;;
  757. interface) validate_format_interface "$arg" ;;
  758. ip) validate_format_ip "$arg" ;;
  759. ip_name) validate_format_domain "$arg" 'domain';;
  760. ip_status) validate_format_ip_status "$arg" ;;
  761. job) validate_format_int "$arg" 'job' ;;
  762. key) validate_format_username "$arg" "$arg_name" ;;
  763. lname) validate_format_name_s "$arg" "$arg_name" ;;
  764. malias) validate_format_username "$arg" "$arg_name" ;;
  765. mask) validate_format_ip "$arg" ;;
  766. max_db) validate_format_int "$arg" 'max db';;
  767. min) validate_format_mhdmw "$arg" $arg_name ;;
  768. month) validate_format_mhdmw "$arg" $arg_name ;;
  769. nat_ip) validate_format_ip "$arg" ;;
  770. newid) validate_format_int "$arg" 'id' ;;
  771. ns1) validate_format_domain "$arg" 'name_server';;
  772. ns2) validate_format_domain "$arg" 'name_server';;
  773. ns3) validate_format_domain "$arg" 'name_server';;
  774. ns4) validate_format_domain "$arg" 'name_server';;
  775. package) validate_format_name "$arg" "$arg_name" ;;
  776. password) validate_format_password "$arg" ;;
  777. port) validate_format_int "$arg" 'port' ;;
  778. quota) validate_format_int "$arg" 'quota' ;;
  779. restart) validate_format_boolean "$arg" 'restart' ;;
  780. record) validate_format_common "$arg" 'record';;
  781. rtype) validate_format_dns_type "$arg" ;;
  782. shell) validate_format_shell "$arg" ;;
  783. soa) validate_format_domain "$arg" 'soa_record';;
  784. stats_pass) validate_format_password "$arg" ;;
  785. stats_user) validate_format_username "$arg" "$arg_name" ;;
  786. template) validate_format_name "$arg" "$arg_name" ;;
  787. ttl) validate_format_int "$arg" 'ttl';;
  788. user) validate_format_username "$arg" "$arg_name" ;;
  789. wday) validate_format_mhdmw "$arg" $arg_name ;;
  790. esac
  791. done
  792. }