db.sh 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640
  1. #!/bin/bash
  2. #===========================================================================#
  3. # #
  4. # Hestia Control Panel - Domain Function Library #
  5. # #
  6. #===========================================================================#
  7. # Global
  8. database_set_default_ports() {
  9. # Set default ports for MySQL and PostgreSQL
  10. mysql_default="3306"
  11. pgsql_default="5432"
  12. # Handle missing values for both $PORT and $port
  13. # however don't override both at once or custom ports will be overridden.
  14. if [ -z "$PORT" ]; then
  15. if [ "$type" = 'mysql' ]; then
  16. PORT="$mysql_default"
  17. fi
  18. if [ "$type" = 'pgsql' ]; then
  19. PORT="$pgsql_default"
  20. fi
  21. fi
  22. if [ -z "$port" ]; then
  23. if [ "$type" = 'mysql' ]; then
  24. port="$mysql_default"
  25. fi
  26. if [ "$type" = 'pgsql' ]; then
  27. port="$pgsql_default"
  28. fi
  29. fi
  30. }
  31. # MySQL
  32. mysql_connect() {
  33. unset PORT
  34. host_str=$(grep "HOST='$1'" $HESTIA/conf/mysql.conf)
  35. parse_object_kv_list "$host_str"
  36. if [ -z $PORT ]; then PORT=3306; fi
  37. if [ -z $HOST ] || [ -z $USER ] || [ -z $PASSWORD ]; then
  38. echo "Error: mysql config parsing failed"
  39. log_event "$E_PARSING" "$ARGUMENTS"
  40. exit $E_PARSING
  41. fi
  42. mycnf="$HESTIA/conf/.mysql.$HOST"
  43. if [ ! -e "$mycnf" ]; then
  44. echo "[client]" > $mycnf
  45. echo "host='$HOST'" >> $mycnf
  46. echo "user='$USER'" >> $mycnf
  47. echo "password='$PASSWORD'" >> $mycnf
  48. echo "port='$PORT'" >> $mycnf
  49. chmod 600 $mycnf
  50. else
  51. mypw=$(grep password $mycnf | cut -f 2 -d \')
  52. if [ "$mypw" != "$PASSWORD" ]; then
  53. echo "[client]" > $mycnf
  54. echo "host='$HOST'" >> $mycnf
  55. echo "user='$USER'" >> $mycnf
  56. echo "password='$PASSWORD'" >> $mycnf
  57. echo "port='$PORT'" >> $mycnf
  58. chmod 660 $mycnf
  59. fi
  60. fi
  61. mysql_out=$(mktemp)
  62. if [ -f '/usr/bin/mariadb' ]; then
  63. mariadb --defaults-file=$mycnf -e 'SELECT VERSION()' > $mysql_out 2>&1
  64. else
  65. mysql --defaults-file=$mycnf -e 'SELECT VERSION()' > $mysql_out 2>&1
  66. fi
  67. if [ '0' -ne "$?" ]; then
  68. if [ "$notify" != 'no' ]; then
  69. email=$(grep CONTACT $HESTIA/data/users/admin/user.conf | cut -f 2 -d \')
  70. subj="MySQL connection error on $(hostname)"
  71. echo -e "Can't connect to MySQL $HOST\n$(cat $mysql_out)" \
  72. | $SENDMAIL -s "$subj" $email
  73. fi
  74. rm -f $mysql_out
  75. echo "Error: Connection to $HOST failed"
  76. log_event "$E_CONNECT" "$ARGUMENTS"
  77. exit $E_CONNECT
  78. fi
  79. mysql_ver=$(cat $mysql_out | tail -n1 | cut -f 1 -d -)
  80. mysql_fork="mysql"
  81. check_mysql_fork=$(grep "MariaDB" $mysql_out)
  82. if [ "$check_mysql_fork" ]; then
  83. mysql_fork="mariadb"
  84. fi
  85. rm -f $mysql_out
  86. }
  87. mysql_query() {
  88. sql_tmp=$(mktemp)
  89. echo "$1" > $sql_tmp
  90. if [ -f '/usr/bin/mariadb' ]; then
  91. mariadb --defaults-file=$mycnf < "$sql_tmp" 2> /dev/null
  92. return_code=$?
  93. else
  94. mysql --defaults-file=$mycnf < "$sql_tmp" 2> /dev/null
  95. return_code=$?
  96. fi
  97. rm -f "$sql_tmp"
  98. return $return_code
  99. }
  100. mysql_dump() {
  101. err="/tmp/e.mysql"
  102. mysqldmp="mysqldump"
  103. if [ -f '/usr/bin/mariadb-dump' ]; then
  104. mysqldmp="/usr/bin/mariadb-dump"
  105. fi
  106. $mysqldmp --defaults-file=$mycnf --single-transaction --routines -r $1 $2 2> $err
  107. if [ '0' -ne "$?" ]; then
  108. $mysqldmp --defaults-extra-file=$mycnf --single-transaction --routines -r $1 $2 2> $err
  109. if [ '0' -ne "$?" ]; then
  110. rm -rf $tmpdir
  111. if [ "$notify" != 'no' ]; then
  112. email=$(grep CONTACT $HESTIA/data/users/admin/user.conf | cut -f 2 -d \')
  113. subj="MySQL error on $(hostname)"
  114. echo -e "Can't dump database $database\n$(cat $err)" \
  115. | $SENDMAIL -s "$subj" $email
  116. fi
  117. echo "Error: dump $database failed"
  118. log_event "$E_DB" "$ARGUMENTS"
  119. exit "$E_DB"
  120. fi
  121. fi
  122. }
  123. # PostgreSQL
  124. psql_connect() {
  125. unset PORT
  126. host_str=$(grep "HOST='$1'" $HESTIA/conf/pgsql.conf)
  127. parse_object_kv_list "$host_str"
  128. export PGPASSWORD="$PASSWORD"
  129. if [ -z $PORT ]; then PORT=5432; fi
  130. if [ -z $HOST ] || [ -z $USER ] || [ -z $PASSWORD ] || [ -z $TPL ]; then
  131. echo "Error: postgresql config parsing failed"
  132. log_event "$E_PARSING" "$ARGUMENTS"
  133. exit $E_PARSING
  134. fi
  135. psql -h $HOST -U $USER -p $PORT -c "SELECT VERSION()" > /dev/null 2> /tmp/e.psql
  136. if [ '0' -ne "$?" ]; then
  137. if [ "$notify" != 'no' ]; then
  138. email=$(grep CONTACT $HESTIA/data/users/admin/user.conf | cut -f 2 -d \')
  139. subj="PostgreSQL connection error on $(hostname)"
  140. echo -e "Can't connect to PostgreSQL $HOST\n$(cat /tmp/e.psql)" \
  141. | $SENDMAIL -s "$subj" $email
  142. fi
  143. echo "Error: Connection to $HOST failed"
  144. log_event "$E_CONNECT" "$ARGUMENTS"
  145. exit "$E_CONNECT"
  146. fi
  147. }
  148. psql_query() {
  149. sql_tmp=$(mktemp)
  150. echo "$1" > $sql_tmp
  151. psql -h $HOST -U $USER -f "$sql_tmp" 2> /dev/null
  152. rm -f $sql_tmp
  153. }
  154. psql_dump() {
  155. pg_dump -h $HOST -U $USER -c --inserts -O -x -f $1 $2 2> /tmp/e.psql
  156. if [ '0' -ne "$?" ]; then
  157. rm -rf $tmpdir
  158. if [ "$notify" != 'no' ]; then
  159. email=$(grep CONTACT $HESTIA/data/users/admin/user.conf | cut -f 2 -d \')
  160. subj="PostgreSQL error on $(hostname)"
  161. echo -e "Can't dump database $database\n$(cat /tmp/e.psql)" \
  162. | $SENDMAIL -s "$subj" $email
  163. fi
  164. echo "Error: dump $database failed"
  165. log_event "$E_DB" "$ARGUMENTS"
  166. exit "$E_DB"
  167. fi
  168. }
  169. # Get database host
  170. get_next_dbhost() {
  171. if [ -z "$host" ] || [ "$host" == 'default' ]; then
  172. IFS=$'\n'
  173. host='EMPTY_DB_HOST'
  174. config="$HESTIA/conf/$type.conf"
  175. host_str=$(grep "SUSPENDED='no'" $config)
  176. check_row=$(echo "$host_str" | wc -l)
  177. if [ 0 -lt "$check_row" ]; then
  178. if [ 1 -eq "$check_row" ]; then
  179. for db in $host_str; do
  180. parse_object_kv_list "$db"
  181. if [ "$MAX_DB" -gt "$U_DB_BASES" ]; then
  182. host=$HOST
  183. fi
  184. done
  185. else
  186. old_weight='100'
  187. for db in $host_str; do
  188. parse_object_kv_list "$db"
  189. let weight="$U_DB_BASES * 100 / $MAX_DB" > /dev/null 2>&1
  190. if [ "$old_weight" -gt "$weight" ]; then
  191. host="$HOST"
  192. old_weight="$weight"
  193. fi
  194. done
  195. fi
  196. fi
  197. fi
  198. }
  199. # Database charset validation
  200. is_charset_valid() {
  201. host_str=$(grep "HOST='$host'" $HESTIA/conf/$type.conf)
  202. parse_object_kv_list "$host_str"
  203. if [ -z "$(echo $CHARSETS | grep -wi $charset)" ]; then
  204. echo "Error: charset $charset not exist"
  205. log_event "$E_NOTEXIST" "$ARGUMENTS"
  206. exit $E_NOTEXIST
  207. fi
  208. }
  209. # Increase database host value
  210. increase_dbhost_values() {
  211. host_str=$(grep "HOST='$host'" $HESTIA/conf/$type.conf)
  212. parse_object_kv_list "$host_str"
  213. old_dbbases="U_DB_BASES='$U_DB_BASES'"
  214. new_dbbases="U_DB_BASES='$((U_DB_BASES + 1))'"
  215. if [ -z "$U_SYS_USERS" ]; then
  216. old_users="U_SYS_USERS=''"
  217. new_users="U_SYS_USERS='$user'"
  218. else
  219. old_users="U_SYS_USERS='$U_SYS_USERS'"
  220. new_users="U_SYS_USERS='$U_SYS_USERS'"
  221. if [ -z "$(echo $U_SYS_USERS | sed "s/,/\n/g" | grep -w $user)" ]; then
  222. old_users="U_SYS_USERS='$U_SYS_USERS'"
  223. new_users="U_SYS_USERS='$U_SYS_USERS,$user'"
  224. fi
  225. fi
  226. sed -i "s/$old_dbbases/$new_dbbases/g" $HESTIA/conf/$type.conf
  227. sed -i "s/$old_users/$new_users/g" $HESTIA/conf/$type.conf
  228. }
  229. # Decrease database host value
  230. decrease_dbhost_values() {
  231. host_str=$(grep "HOST='$HOST'" $HESTIA/conf/$TYPE.conf)
  232. parse_object_kv_list "$host_str"
  233. old_dbbases="U_DB_BASES='$U_DB_BASES'"
  234. new_dbbases="U_DB_BASES='$((U_DB_BASES - 1))'"
  235. old_users="U_SYS_USERS='$U_SYS_USERS'"
  236. U_SYS_USERS=$(echo "$U_SYS_USERS" \
  237. | sed "s/,/\n/g" \
  238. | sed "s/^$user$//g" \
  239. | sed "/^$/d" \
  240. | sed ':a;N;$!ba;s/\n/,/g')
  241. new_users="U_SYS_USERS='$U_SYS_USERS'"
  242. sed -i "s/$old_dbbases/$new_dbbases/g" $HESTIA/conf/$TYPE.conf
  243. sed -i "s/$old_users/$new_users/g" $HESTIA/conf/$TYPE.conf
  244. }
  245. # Create MySQL database
  246. add_mysql_database() {
  247. mysql_connect $host
  248. mysql_ver_sub=$(echo $mysql_ver | cut -d '.' -f1)
  249. mysql_ver_sub_sub=$(echo $mysql_ver | cut -d '.' -f2)
  250. query="CREATE DATABASE \`$database\` CHARACTER SET $charset"
  251. mysql_query "$query"
  252. check_result $? "Unable to create database $database"
  253. if [ "$mysql_fork" = "mysql" ] && [ "$mysql_ver_sub" -ge 8 ]; then
  254. query="CREATE USER \`$dbuser\`@\`%\`
  255. IDENTIFIED BY '$dbpass'"
  256. mysql_query "$query" > /dev/null
  257. query="CREATE USER \`$dbuser\`@localhost
  258. IDENTIFIED BY '$dbpass'"
  259. mysql_query "$query" > /dev/null
  260. query="GRANT ALL ON \`$database\`.* TO \`$dbuser\`@\`%\`"
  261. mysql_query "$query" > /dev/null
  262. query="GRANT ALL ON \`$database\`.* TO \`$dbuser\`@localhost"
  263. mysql_query "$query" > /dev/null
  264. else
  265. query="GRANT ALL ON \`$database\`.* TO \`$dbuser\`@\`%\`
  266. IDENTIFIED BY '$dbpass'"
  267. mysql_query "$query" > /dev/null
  268. query="GRANT ALL ON \`$database\`.* TO \`$dbuser\`@localhost
  269. IDENTIFIED BY '$dbpass'"
  270. mysql_query "$query" > /dev/null
  271. fi
  272. if [ "$mysql_fork" = "mysql" ]; then
  273. # mysql
  274. if [ "$mysql_ver_sub" -ge 8 ] || { [ "$mysql_ver_sub" -eq 5 ] && [ "$mysql_ver_sub_sub" -ge 7 ]; }; then
  275. if [ "$mysql_ver_sub" -ge 8 ]; then
  276. # mysql >= 8
  277. # This query will be proceeding with the usage of Print identified with as hex feature
  278. md5=$(mysql_query "SET print_identified_with_as_hex=ON; SHOW CREATE USER \`$dbuser\`" 2> /dev/null)
  279. # echo $md5
  280. if [[ "$md5" =~ 0x([^ ]+) ]]; then
  281. md5=$(echo "$md5" | grep password | grep -E -o '0x([^ ]+)')
  282. else
  283. md5=$(echo "$md5" | grep password | cut -f4 -d \')
  284. fi
  285. # echo $md5
  286. else
  287. # mysql < 8
  288. md5=$(mysql_query "SHOW CREATE USER \`$dbuser\`" 2> /dev/null)
  289. md5=$(echo "$md5" | grep password | cut -f8 -d \')
  290. fi
  291. else
  292. # mysql < 5.7
  293. md5=$(mysql_query "SHOW GRANTS FOR \`$dbuser\`" 2> /dev/null)
  294. md5=$(echo "$md5" | grep PASSW | tr ' ' '\n' | tail -n1 | cut -f 2 -d \')
  295. fi
  296. else
  297. # mariadb
  298. md5=$(mysql_query "SHOW GRANTS FOR \`$dbuser\`" 2> /dev/null)
  299. md5=$(echo "$md5" | grep PASSW | tr ' ' '\n' | tail -n1 | cut -f 2 -d \')
  300. fi
  301. }
  302. # Create PostgreSQL database
  303. add_pgsql_database() {
  304. psql_connect $host
  305. query="CREATE ROLE $dbuser WITH LOGIN PASSWORD '$dbpass'"
  306. psql_query "$query" > /dev/null
  307. query="CREATE DATABASE $database OWNER $dbuser"
  308. if [ "$TPL" = 'template0' ]; then
  309. query="$query ENCODING '$charset' TEMPLATE $TPL"
  310. else
  311. query="$query TEMPLATE $TPL"
  312. fi
  313. psql_query "$query" > /dev/null
  314. query="GRANT ALL PRIVILEGES ON DATABASE $database TO $dbuser"
  315. psql_query "$query" > /dev/null
  316. query="GRANT CONNECT ON DATABASE template1 to $dbuser"
  317. psql_query "$query" > /dev/null
  318. query="SELECT rolpassword FROM pg_authid WHERE rolname='$dbuser'"
  319. md5=$(psql_query "$query" | grep md5 | cut -f 2 -d \ )
  320. }
  321. add_mysql_database_temp_user() {
  322. mysql_connect $host
  323. query="GRANT ALL ON \`$database\`.* TO \`$dbuser\`@localhost
  324. IDENTIFIED BY '$dbpass'"
  325. mysql_query "$query" > /dev/null
  326. }
  327. delete_mysql_database_temp_user() {
  328. mysql_connect $host
  329. query="REVOKE ALL ON \`$database\`.* FROM \`$dbuser\`@localhost"
  330. mysql_query "$query" > /dev/null
  331. query="DROP USER '$dbuser'@'localhost'"
  332. mysql_query "$query" > /dev/null
  333. }
  334. # Check if database host do not exist in config
  335. is_dbhost_new() {
  336. if [ -e "$HESTIA/conf/$type.conf" ]; then
  337. check_host=$(grep "HOST='$host'" $HESTIA/conf/$type.conf)
  338. if [ "$check_host" ]; then
  339. echo "Error: db host exist"
  340. log_event "$E_EXISTS" "$ARGUMENTS"
  341. exit $E_EXISTS
  342. fi
  343. fi
  344. }
  345. # Get database values
  346. get_database_values() {
  347. parse_object_kv_list $(grep "DB='$database'" $USER_DATA/db.conf)
  348. }
  349. # Change MySQL database password
  350. change_mysql_password() {
  351. mysql_connect $HOST
  352. mysql_ver_sub=$(echo $mysql_ver | cut -d '.' -f1)
  353. mysql_ver_sub_sub=$(echo $mysql_ver | cut -d '.' -f2)
  354. if [ "$mysql_fork" = "mysql" ]; then
  355. # mysql
  356. if [ "$mysql_ver_sub" -ge 8 ]; then
  357. # mysql >= 8
  358. query="SET PASSWORD FOR \`$DBUSER\`@\`%\` = '$dbpass'"
  359. mysql_query "$query" > /dev/null
  360. query="SET PASSWORD FOR \`$DBUSER\`@localhost = '$dbpass'"
  361. mysql_query "$query" > /dev/null
  362. else
  363. # mysql < 8
  364. query="GRANT ALL ON \`$database\`.* TO \`$DBUSER\`@\`%\`
  365. IDENTIFIED BY '$dbpass'"
  366. mysql_query "$query" > /dev/null
  367. query="GRANT ALL ON \`$database\`.* TO \`$DBUSER\`@localhost
  368. IDENTIFIED BY '$dbpass'"
  369. mysql_query "$query" > /dev/null
  370. fi
  371. else
  372. # mariadb
  373. query="GRANT ALL ON \`$database\`.* TO \`$DBUSER\`@\`%\`
  374. IDENTIFIED BY '$dbpass'"
  375. mysql_query "$query" > /dev/null
  376. query="GRANT ALL ON \`$database\`.* TO \`$DBUSER\`@localhost
  377. IDENTIFIED BY '$dbpass'"
  378. mysql_query "$query" > /dev/null
  379. fi
  380. if [ "$mysql_fork" = "mysql" ]; then
  381. # mysql
  382. if [ "$mysql_ver_sub" -ge 8 ] || { [ "$mysql_ver_sub" -eq 5 ] && [ "$mysql_ver_sub_sub" -ge 7 ]; }; then
  383. if [ "$mysql_ver_sub" -ge 8 ]; then
  384. # mysql >= 8
  385. # This query will be proceeding with the usage of Print identified with as hex feature
  386. md5=$(mysql_query "SET print_identified_with_as_hex=ON; SHOW CREATE USER \`$DBUSER\`" 2> /dev/null)
  387. # echo $md5
  388. if [[ "$md5" =~ 0x([^ ]+) ]]; then
  389. md5=$(echo "$md5" | grep password | grep -E -o '0x([^ ]+)')
  390. else
  391. md5=$(echo "$md5" | grep password | cut -f4 -d \')
  392. fi
  393. # echo $md5
  394. else
  395. # mysql < 8
  396. md5=$(mysql_query "SHOW CREATE USER \`$DBUSER\`" 2> /dev/null)
  397. md5=$(echo "$md5" | grep password | cut -f8 -d \')
  398. fi
  399. else
  400. # mysql < 5.7
  401. md5=$(mysql_query "SHOW GRANTS FOR \`$DBUSER\`" 2> /dev/null)
  402. md5=$(echo "$md5" | grep PASSW | tr ' ' '\n' | tail -n1 | cut -f 2 -d \')
  403. fi
  404. else
  405. # mariadb
  406. md5=$(mysql_query "SHOW GRANTS FOR \`$DBUSER\`" 2> /dev/null)
  407. md5=$(echo "$md5" | grep PASSW | tr ' ' '\n' | tail -n1 | cut -f 2 -d \')
  408. fi
  409. }
  410. # Change PostgreSQL database password
  411. change_pgsql_password() {
  412. psql_connect $HOST
  413. query="ALTER ROLE $DBUSER WITH LOGIN PASSWORD '$dbpass'"
  414. psql_query "$query" > /dev/null
  415. query="SELECT rolpassword FROM pg_authid WHERE rolname='$DBUSER'"
  416. md5=$(psql_query "$query" | grep md5 | cut -f 2 -d \ )
  417. }
  418. # Delete MySQL database
  419. delete_mysql_database() {
  420. mysql_connect $HOST
  421. query="DROP DATABASE \`$database\`"
  422. mysql_query "$query"
  423. query="REVOKE ALL ON \`$database\`.* FROM \`$DBUSER\`@\`%\`"
  424. mysql_query "$query" > /dev/null
  425. query="REVOKE ALL ON \`$database\`.* FROM \`$DBUSER\`@localhost"
  426. mysql_query "$query" > /dev/null
  427. if [ "$(grep "DBUSER='$DBUSER'" $USER_DATA/db.conf | wc -l)" -lt 2 ]; then
  428. query="DROP USER '$DBUSER'@'%'"
  429. mysql_query "$query" > /dev/null
  430. query="DROP USER '$DBUSER'@'localhost'"
  431. mysql_query "$query" > /dev/null
  432. fi
  433. }
  434. # Delete PostgreSQL database
  435. delete_pgsql_database() {
  436. psql_connect $HOST
  437. query="REVOKE ALL PRIVILEGES ON DATABASE $database FROM $DBUSER"
  438. psql_query "$query" > /dev/null
  439. query="DROP DATABASE $database"
  440. psql_query "$query" > /dev/null
  441. if [ "$(grep "DBUSER='$DBUSER'" $USER_DATA/db.conf | wc -l)" -lt 2 ]; then
  442. query="REVOKE CONNECT ON DATABASE template1 FROM $DBUSER"
  443. psql_query "$query" > /dev/null
  444. query="DROP ROLE $DBUSER"
  445. psql_query "$query" > /dev/null
  446. fi
  447. }
  448. # Dump MySQL database
  449. dump_mysql_database() {
  450. mysql_connect $HOST
  451. mysql_dump $dump $database
  452. query="SHOW GRANTS FOR '$DBUSER'@'localhost'"
  453. mysql_query "$query" | grep -v "Grants for" > $grants
  454. query="SHOW GRANTS FOR '$DBUSER'@'%'"
  455. mysql_query "$query" | grep -v "Grants for" > $grants
  456. }
  457. # Dump PostgreSQL database
  458. dump_pgsql_database() {
  459. psql_connect $HOST
  460. psql_dump $dump $database
  461. query="SELECT rolpassword FROM pg_authid WHERE rolname='$DBUSER'"
  462. md5=$(psql_query "$query" | head -n1 | cut -f 2 -d \ )
  463. pw_str="UPDATE pg_authid SET rolpassword='$md5' WHERE rolname='$DBUSER'"
  464. gr_str="GRANT ALL PRIVILEGES ON DATABASE $database to '$DBUSER'"
  465. echo -e "$pw_str\n$gr_str" >> $grants
  466. }
  467. # Check if database server is in use
  468. is_dbhost_free() {
  469. host_str=$(grep "HOST='$host'" $HESTIA/conf/$type.conf)
  470. parse_object_kv_list "$host_str"
  471. if [ 0 -ne "$U_DB_BASES" ]; then
  472. echo "Error: host $HOST is used"
  473. log_event "$E_INUSE" "$ARGUMENTS"
  474. exit $E_INUSE
  475. fi
  476. }
  477. # Suspend MySQL database
  478. suspend_mysql_database() {
  479. mysql_connect $HOST
  480. query="REVOKE ALL ON \`$database\`.* FROM \`$DBUSER\`@\`%\`"
  481. mysql_query "$query" > /dev/null
  482. query="REVOKE ALL ON \`$database\`.* FROM \`$DBUSER\`@localhost"
  483. mysql_query "$query" > /dev/null
  484. }
  485. # Suspend PostgreSQL database
  486. suspend_pgsql_database() {
  487. psql_connect $HOST
  488. query="REVOKE ALL PRIVILEGES ON $database FROM $DBUSER"
  489. psql_query "$query" > /dev/null
  490. }
  491. # Unsuspend MySQL database
  492. unsuspend_mysql_database() {
  493. mysql_connect $HOST
  494. query="GRANT ALL ON \`$database\`.* TO \`$DBUSER\`@\`%\`"
  495. mysql_query "$query" > /dev/null
  496. query="GRANT ALL ON \`$database\`.* TO \`$DBUSER\`@localhost"
  497. mysql_query "$query" > /dev/null
  498. }
  499. # Unsuspend PostgreSQL database
  500. unsuspend_pgsql_database() {
  501. psql_connect $HOST
  502. query="GRANT ALL PRIVILEGES ON DATABASE $database TO $DBUSER"
  503. psql_query "$query" > /dev/null
  504. }
  505. # Get MySQL disk usage
  506. get_mysql_disk_usage() {
  507. mysql_connect $HOST
  508. query="SELECT SUM( data_length + index_length ) / 1024 / 1024 'Size'
  509. FROM information_schema.TABLES WHERE table_schema='$database'"
  510. usage=$(mysql_query "$query" | tail -n1)
  511. if [ "$usage" == '' ] || [ "$usage" == 'NULL' ] || [ "${usage:0:1}" -eq '0' ]; then
  512. usage=1
  513. fi
  514. export LC_ALL=C
  515. usage=$(printf "%0.f\n" $usage)
  516. }
  517. # Get PostgreSQL disk usage
  518. get_pgsql_disk_usage() {
  519. psql_connect $HOST
  520. query="SELECT pg_database_size('$database');"
  521. usage=$(psql_query "$query")
  522. usage=$(echo "$usage" | grep -v "-" | grep -v 'row' | sed "/^$/d")
  523. usage=$(echo "$usage" | grep -v "pg_database_size" | awk '{print $1}')
  524. if [ -z "$usage" ]; then
  525. usage=0
  526. fi
  527. usage=$(($usage / 1048576))
  528. if [ "$usage" -eq '0' ]; then
  529. usage=1
  530. fi
  531. }
  532. # Delete MySQL user
  533. delete_mysql_user() {
  534. mysql_connect $HOST
  535. query="REVOKE ALL ON \`$database\`.* FROM \`$old_dbuser\`@\`%\`"
  536. mysql_query "$query" > /dev/null
  537. query="REVOKE ALL ON \`$database\`.* FROM \`$old_dbuser\`@localhost"
  538. mysql_query "$query" > /dev/null
  539. query="DROP USER '$old_dbuser'@'%'"
  540. mysql_query "$query" > /dev/null
  541. query="DROP USER '$old_dbuser'@'localhost'"
  542. mysql_query "$query" > /dev/null
  543. }
  544. # Delete PostgreSQL user
  545. delete_pgsql_user() {
  546. psql_connect $HOST
  547. query="REVOKE ALL PRIVILEGES ON DATABASE $database FROM $old_dbuser"
  548. psql_query "$query" > /dev/null
  549. query="REVOKE CONNECT ON DATABASE template1 FROM $old_dbuser"
  550. psql_query "$query" > /dev/null
  551. query="DROP ROLE $old_dbuser"
  552. psql_query "$query" > /dev/null
  553. }