myvesta/func/db.sh
2024-05-05 20:28:05 +02:00

580 lines
18 KiB
Bash

# MySQL
mysql_connect() {
host_str=$(grep "HOST='$1'" $VESTA/conf/mysql.conf)
eval $host_str
if [ -z $HOST ] || [ -z $USER ] || [ -z $PASSWORD ]; then
echo "Error: mysql config parsing failed"
log_event "$E_PARSING" "$ARGUMENTS"
exit $E_PARSING
fi
mycnf="$VESTA/conf/.mysql.$HOST"
if [ ! -e "$mycnf" ]; then
echo "[client]">$mycnf
echo "host='$HOST'" >> $mycnf
echo "user='$USER'" >> $mycnf
echo "password='$PASSWORD'" >> $mycnf
chmod 600 $mycnf
else
mypw=$(grep password $mycnf|cut -f 2 -d \')
if [ "$mypw" != "$PASSWORD" ]; then
echo "[client]">$mycnf
echo "host='$HOST'" >> $mycnf
echo "user='$USER'" >> $mycnf
echo "password='$PASSWORD'" >> $mycnf
chmod 660 $mycnf
fi
fi
mysql_out=$(mktemp)
mysql --defaults-file=$mycnf -e 'SELECT VERSION()' > $mysql_out 2>&1
if [ '0' -ne "$?" ]; then
if [ "$notify" != 'no' ]; then
subj="Error: Connection to $HOST failed"
email=$($BIN/v-get-user-value admin CONTACT)
echo -e "Can't connect to MySQL $HOST\n$(cat $mysql_out)" |\
$SENDMAIL -s "$subj" $email
fi
rm -f $mysql_out
echo "Error: Connection to $HOST failed"
log_event "$E_CONNECT" "$ARGUMENTS"
exit $E_CONNECT
fi
mysql_ver=$(cat $mysql_out |tail -n1 |cut -f 1 -d -)
mysql_fork="mysql"
check_mysql_fork=$(grep "MariaDB" $mysql_out)
if [ ! -z "$check_mysql_fork" ]; then
mysql_fork="mariadb"
fi
rm -f $mysql_out
}
mysql_query() {
sql_tmp=$(mktemp)
echo "$1" > $sql_tmp
mysql --defaults-file=$mycnf < "$sql_tmp" 2>/dev/null
rm -f "$sql_tmp"
}
mysql_dump() {
err="/tmp/e.mysql"
mysqldump --defaults-file=$mycnf --single-transaction --max_allowed_packet=100M -r $1 $2 2> $err
if [ '0' -ne "$?" ]; then
rm -rf $tmpdir
if [ "$notify" != 'no' ]; then
subj="Error: dump $database failed"
email=$($BIN/v-get-user-value admin CONTACT)
echo -e "Can't dump database $database\n$(cat $err)" |\
$SENDMAIL -s "$subj" $email
fi
echo "Error: dump $database failed\n$(cat $err)"
log_event "$E_DB" "$ARGUMENTS"
exit $E_DB
fi
}
# PostgreSQL
psql_connect() {
host_str=$(grep "HOST='$1'" $VESTA/conf/pgsql.conf)
eval $host_str
export PGPASSWORD="$PASSWORD"
if [ -z $HOST ] || [ -z $USER ] || [ -z $PASSWORD ] || [ -z $TPL ]; then
echo "Error: postgresql config parsing failed"
log_event "$E_PARSING" "$ARGUMENTS"
exit $E_PARSING
fi
psql -h $HOST -U $USER -c "SELECT VERSION()" > /dev/null 2>/tmp/e.psql
if [ '0' -ne "$?" ]; then
if [ "$notify" != 'no' ]; then
subj="Error: Connection to $HOST failed"
email=$($BIN/v-get-user-value admin CONTACT)
echo -e "Can't connect to PostgreSQL $HOST\n$(cat /tmp/e.psql)" |\
$SENDMAIL -s "$subj" $email
fi
echo "Error: Connection to $HOST failed"
log_event "$E_CONNECT" "$ARGUMENTS"
exit $E_CONNECT
fi
}
psql_query() {
sql_tmp=$(mktemp)
echo "$1" > $sql_tmp
psql -h $HOST -U $USER -f "$sql_tmp" 2>/dev/null
rm -f $sql_tmp
}
psql_dump() {
pg_dump -h $HOST -U $USER -c --inserts -O -x -f $1 $2 2>/tmp/e.psql
if [ '0' -ne "$?" ]; then
rm -rf $tmpdir
if [ "$notify" != 'no' ]; then
subj="Error: dump $database failed"
email=$($BIN/v-get-user-value admin CONTACT)
echo -e "Can't dump database $database\n$(cat /tmp/e.psql)" |\
$SENDMAIL -s "$subj" $email
fi
echo "Error: dump $database failed"
log_event "$E_DB" "$ARGUMENTS"
exit $E_DB
fi
}
# Get database host
get_next_dbhost() {
if [ -z "$host" ] || [ "$host" == 'default' ]; then
IFS=$'\n'
host='EMPTY_DB_HOST'
config="$VESTA/conf/$type.conf"
host_str=$(grep "SUSPENDED='no'" $config)
check_row=$(echo "$host_str"|wc -l)
if [ 0 -lt "$check_row" ]; then
if [ 1 -eq "$check_row" ]; then
for db in $host_str; do
eval $db
if [ "$MAX_DB" -gt "$U_DB_BASES" ]; then
host=$HOST
fi
done
else
old_weight='100'
for db in $host_str; do
eval $db
let weight="$U_DB_BASES * 100 / $MAX_DB" >/dev/null 2>&1
if [ "$old_weight" -gt "$weight" ]; then
host="$HOST"
old_weight="$weight"
fi
done
fi
fi
fi
}
# Database charset validation
is_charset_valid() {
host_str=$(grep "HOST='$host'" $VESTA/conf/$type.conf)
eval $host_str
if [ -z "$(echo $CHARSETS | grep -wi $charset )" ]; then
echo "Error: charset $charset not exist"
log_event "$E_NOTEXIST" "$ARGUMENTS"
exit $E_NOTEXIST
fi
}
# Increase database host value
increase_dbhost_values() {
host_str=$(grep "HOST='$host'" $VESTA/conf/$type.conf)
eval $host_str
old_dbbases="U_DB_BASES='$U_DB_BASES'"
new_dbbases="U_DB_BASES='$((U_DB_BASES + 1))'"
if [ -z "$U_SYS_USERS" ]; then
old_users="U_SYS_USERS=''"
new_users="U_SYS_USERS='$user'"
else
old_users="U_SYS_USERS='$U_SYS_USERS'"
new_users="U_SYS_USERS='$U_SYS_USERS'"
if [ -z "$(echo $U_SYS_USERS|sed "s/,/\n/g"|grep -w $user)" ]; then
old_users="U_SYS_USERS='$U_SYS_USERS'"
new_users="U_SYS_USERS='$U_SYS_USERS,$user'"
fi
fi
sed -i "s/$old_dbbases/$new_dbbases/g" $VESTA/conf/$type.conf
sed -i "s/$old_users/$new_users/g" $VESTA/conf/$type.conf
}
# Decrease database host value
decrease_dbhost_values() {
host_str=$(grep "HOST='$HOST'" $VESTA/conf/$TYPE.conf)
eval $host_str
old_dbbases="U_DB_BASES='$U_DB_BASES'"
new_dbbases="U_DB_BASES='$((U_DB_BASES - 1))'"
old_users="U_SYS_USERS='$U_SYS_USERS'"
U_SYS_USERS=$(echo "$U_SYS_USERS" |\
sed "s/,/\n/g"|\
sed "s/^$user$//g"|\
sed "/^$/d"|\
sed ':a;N;$!ba;s/\n/,/g')
new_users="U_SYS_USERS='$U_SYS_USERS'"
sed -i "s/$old_dbbases/$new_dbbases/g" $VESTA/conf/$TYPE.conf
sed -i "s/$old_users/$new_users/g" $VESTA/conf/$TYPE.conf
}
# Create MySQL database
add_mysql_database() {
mysql_connect $host
mysql_ver_sub=$(echo $mysql_ver |cut -d '.' -f1)
mysql_ver_sub_sub=$(echo $mysql_ver |cut -d '.' -f2)
query="CREATE DATABASE \`$database\` CHARACTER SET $charset"
mysql_query "$query" > /dev/null
if [ "$mysql_fork" = "mysql" ] && [ "$mysql_ver_sub" -ge 8 ] ; then
query="CREATE USER \`$dbuser\`@\`%\`
IDENTIFIED BY '$dbpass'"
mysql_query "$query" > /dev/null
query="CREATE USER \`$dbuser\`@localhost
IDENTIFIED BY '$dbpass'"
mysql_query "$query" > /dev/null
query="GRANT ALL ON \`$database\`.* TO \`$dbuser\`@\`%\`"
mysql_query "$query" > /dev/null
query="GRANT ALL ON \`$database\`.* TO \`$dbuser\`@localhost"
mysql_query "$query" > /dev/null
else
query="GRANT ALL ON \`$database\`.* TO \`$dbuser\`@\`%\`
IDENTIFIED BY '$dbpass'"
mysql_query "$query" > /dev/null
query="GRANT ALL ON \`$database\`.* TO \`$dbuser\`@localhost
IDENTIFIED BY '$dbpass'"
mysql_query "$query" > /dev/null
fi
if [ "$mysql_fork" = "mysql" ]; then
# mysql
if [ "$mysql_ver_sub" -ge 8 ] || { [ "$mysql_ver_sub" -eq 5 ] && [ "$mysql_ver_sub_sub" -ge 7 ]; } then
if [ "$mysql_ver_sub" -ge 8 ]; then
# mysql >= 8
md5=$(mysql_query "SHOW CREATE USER \`$dbuser\`" 2>/dev/null)
# echo $md5
md5=$(echo "$md5" |grep password |cut -f4 -d \')
# echo $md5
else
# mysql < 8
md5=$(mysql_query "SHOW CREATE USER \`$dbuser\`" 2>/dev/null)
md5=$(echo "$md5" |grep password |cut -f8 -d \')
fi
else
# mysql < 5.7
md5=$(mysql_query "SHOW GRANTS FOR \`$dbuser\`" 2>/dev/null)
md5=$(echo "$md5" |grep PASSW|tr ' ' '\n' |tail -n1 |cut -f 2 -d \')
fi
else
# mariadb
md5=$(mysql_query "SHOW GRANTS FOR \`$dbuser\`" 2>/dev/null)
md5=$(echo "$md5" |grep PASSW|tr ' ' '\n' |tail -n1 |cut -f 2 -d \')
fi
}
# Create PostgreSQL database
add_pgsql_database() {
psql_connect $host
query="CREATE ROLE $dbuser WITH LOGIN PASSWORD '$dbpass'"
psql_query "$query" > /dev/null
query="CREATE DATABASE $database OWNER $dbuser"
if [ "$TPL" = 'template0' ]; then
query="$query ENCODING '$charset' TEMPLATE $TPL"
else
query="$query TEMPLATE $TPL"
fi
psql_query "$query" > /dev/null
query="GRANT ALL PRIVILEGES ON DATABASE $database TO $dbuser"
psql_query "$query" > /dev/null
query="GRANT CONNECT ON DATABASE template1 to $dbuser"
psql_query "$query" > /dev/null
query="SELECT rolpassword FROM pg_authid WHERE rolname='$dbuser';"
md5=$(psql_query "$query" | grep md5 | cut -f 2 -d \ )
}
# Check if database host do not exist in config
is_dbhost_new() {
if [ -e "$VESTA/conf/$type.conf" ]; then
check_host=$(grep "HOST='$host'" $VESTA/conf/$type.conf)
if [ ! -z "$check_host" ]; then
echo "Error: db host exist"
log_event "$E_EXISTS" "$ARGUMENTS"
exit $E_EXISTS
fi
fi
}
# Get database values
get_database_values() {
eval $(grep "DB='$database'" $USER_DATA/db.conf)
}
# Change MySQL database password
change_mysql_password() {
mysql_connect $HOST
mysql_ver_sub=$(echo $mysql_ver |cut -d '.' -f1)
mysql_ver_sub_sub=$(echo $mysql_ver |cut -d '.' -f2)
if [ "$mysql_fork" = "mysql" ]; then
# mysql
if [ "$mysql_ver_sub" -ge 8 ]; then
# mysql >= 8
query="SET PASSWORD FOR \`$DBUSER\`@\`%\` = '$dbpass'"
mysql_query "$query" > /dev/null
query="SET PASSWORD FOR \`$DBUSER\`@localhost = '$dbpass'"
mysql_query "$query" > /dev/null
else
# mysql < 8
query="GRANT ALL ON \`$database\`.* TO \`$DBUSER\`@\`%\`
IDENTIFIED BY '$dbpass'"
mysql_query "$query" > /dev/null
query="GRANT ALL ON \`$database\`.* TO \`$DBUSER\`@localhost
IDENTIFIED BY '$dbpass'"
mysql_query "$query" > /dev/null
fi
else
# mariadb
query="GRANT ALL ON \`$database\`.* TO \`$DBUSER\`@\`%\`
IDENTIFIED BY '$dbpass'"
mysql_query "$query" > /dev/null
query="GRANT ALL ON \`$database\`.* TO \`$DBUSER\`@localhost
IDENTIFIED BY '$dbpass'"
mysql_query "$query" > /dev/null
fi
if [ "$mysql_fork" = "mysql" ]; then
# mysql
if [ "$mysql_ver_sub" -ge 8 ] || { [ "$mysql_ver_sub" -eq 5 ] && [ "$mysql_ver_sub_sub" -ge 7 ]; } then
if [ "$mysql_ver_sub" -ge 8 ]; then
# mysql >= 8
md5=$(mysql_query "SHOW CREATE USER \`$DBUSER\`" 2>/dev/null)
# echo $md5
md5=$(echo "$md5" |grep password |cut -f4 -d \')
# echo $md5
else
# mysql < 8
md5=$(mysql_query "SHOW CREATE USER \`$DBUSER\`" 2>/dev/null)
md5=$(echo "$md5" |grep password |cut -f8 -d \')
fi
else
# mysql < 5.7
md5=$(mysql_query "SHOW GRANTS FOR \`$DBUSER\`" 2>/dev/null)
md5=$(echo "$md5" |grep PASSW|tr ' ' '\n' |tail -n1 |cut -f 2 -d \')
fi
else
# mariadb
md5=$(mysql_query "SHOW GRANTS FOR \`$DBUSER\`" 2>/dev/null)
md5=$(echo "$md5" |grep PASSW|tr ' ' '\n' |tail -n1 |cut -f 2 -d \')
fi
}
# Change PostgreSQL database password
change_pgsql_password() {
psql_connect $HOST
query="ALTER ROLE $DBUSER WITH LOGIN PASSWORD '$dbpass'"
psql_query "$query" > /dev/null
query="SELECT rolpassword FROM pg_authid WHERE rolname='$DBUSER';"
md5=$(psql_query "$query" | grep md5 |cut -f 2 -d \ )
}
# Delete MySQL database
delete_mysql_database() {
mysql_connect $HOST
query="DROP DATABASE \`$database\`"
mysql_query "$query" > /dev/null
query="REVOKE ALL ON \`$database\`.* FROM \`$DBUSER\`@\`%\`"
mysql_query "$query" > /dev/null
query="REVOKE ALL ON \`$database\`.* FROM \`$DBUSER\`@localhost"
mysql_query "$query" > /dev/null
if [ "$(grep "DBUSER='$DBUSER'" $USER_DATA/db.conf |wc -l)" -lt 2 ]; then
query="DROP USER '$DBUSER'@'%'"
mysql_query "$query" > /dev/null
query="DROP USER '$DBUSER'@'localhost'"
mysql_query "$query" > /dev/null
fi
}
# Delete PostgreSQL database
delete_pgsql_database() {
psql_connect $HOST
query="REVOKE ALL PRIVILEGES ON DATABASE $database FROM $DBUSER"
psql_query "$query" > /dev/null
query="DROP DATABASE $database"
psql_query "$query" > /dev/null
if [ "$(grep "DBUSER='$DBUSER'" $USER_DATA/db.conf |wc -l)" -lt 2 ]; then
query="REVOKE CONNECT ON DATABASE template1 FROM $db_user"
psql_query "$query" > /dev/null
query="DROP ROLE $db_user"
psql_query "$query" > /dev/null
fi
}
# Dump MySQL database
dump_mysql_database() {
mysql_connect $HOST
mysql_dump $dump $database
query="SHOW GRANTS FOR '$DBUSER'@'localhost'"
mysql_query "$query" | grep -v "Grants for" > $grants
query="SHOW GRANTS FOR '$DBUSER'@'%'"
mysql_query "$query" | grep -v "Grants for" > $grants
}
# Dump PostgreSQL database
dump_pgsql_database() {
psql_connect $HOST
psql_dump $dump $database
query="SELECT rolpassword FROM pg_authid WHERE rolname='$DBUSER';"
md5=$(psql_query "$query" | head -n1 | cut -f 2 -d \ )
pw_str="UPDATE pg_authid SET rolpassword='$md5' WHERE rolname='$DBUSER';"
gr_str="GRANT ALL PRIVILEGES ON DATABASE $database to '$DBUSER'"
echo -e "$pw_str\n$gr_str" >> $grants
}
# Check if database server is in use
is_dbhost_free() {
host_str=$(grep "HOST='$host'" $VESTA/conf/$type.conf)
eval $host_str
if [ 0 -ne "$U_DB_BASES" ]; then
echo "Error: host $HOST is used"
log_event "$E_INUSE" "$ARGUMENTS"
exit $E_INUSE
fi
}
# Suspend MySQL database
suspend_mysql_database() {
mysql_connect $HOST
query="REVOKE ALL ON \`$database\`.* FROM \`$DBUSER\`@\`%\`"
mysql_query "$query" > /dev/null
query="REVOKE ALL ON \`$database\`.* FROM \`$DBUSER\`@localhost"
mysql_query "$query" > /dev/null
}
# Suspend PostgreSQL database
suspend_pgsql_database() {
psql_connect $HOST
query="REVOKE ALL PRIVILEGES ON $database FROM $DBUSER"
psql_query "$query" > /dev/null
}
# Unsuspend MySQL database
unsuspend_mysql_database() {
mysql_connect $HOST
query="GRANT ALL ON \`$database\`.* TO \`$DBUSER\`@\`%\`"
mysql_query "$query" > /dev/null
query="GRANT ALL ON \`$database\`.* TO \`$DBUSER\`@localhost"
mysql_query "$query" > /dev/null
}
# Unsuspend PostgreSQL database
unsuspend_pgsql_database() {
psql_connect $HOST
query="GRANT ALL PRIVILEGES ON DATABASE $database TO $DBUSER"
psql_query "$query" > /dev/null
}
# Get MySQL disk usage
get_mysql_disk_usage() {
mysql_connect $HOST
query="SELECT SUM( data_length + index_length ) / 1024 / 1024 'Size'
FROM information_schema.TABLES WHERE table_schema='$database'"
usage=$(mysql_query "$query" |tail -n1)
if [ "$usage" == '' ] || [ "$usage" == 'NULL' ] || [ "${usage:0:1}" -eq '0' ]; then
usage=1
fi
export LC_ALL=C
usage=$(printf "%0.f\n" $usage)
}
# Get PostgreSQL disk usage
get_pgsql_disk_usage() {
psql_connect $HOST
query="SELECT pg_database_size('$database');"
usage=$(psql_query "$query")
usage=$(echo "$usage" | grep -v "-" | grep -v 'row' | sed "/^$/d")
usage=$(echo "$usage" | grep -v "pg_database_size" | awk '{print $1}')
if [ -z "$usage" ]; then
usage=0
fi
usage=$(($usage / 1048576))
if [ "$usage" -eq '0' ]; then
usage=1
fi
}
# Delete MySQL user
delete_mysql_user() {
mysql_connect $HOST
query="REVOKE ALL ON \`$database\`.* FROM \`$old_dbuser\`@\`%\`"
mysql_query "$query" > /dev/null
query="REVOKE ALL ON \`$database\`.* FROM \`$old_dbuser\`@localhost"
mysql_query "$query" > /dev/null
query="DROP USER '$old_dbuser'@'%'"
mysql_query "$query" > /dev/null
query="DROP USER '$old_dbuser'@'localhost'"
mysql_query "$query" > /dev/null
}
# Delete PostgreSQL user
delete_pgsql_user() {
psql_connect $HOST
query="REVOKE ALL PRIVILEGES ON DATABASE $database FROM $old_dbuser"
psql_query "$query" > /dev/null
query="REVOKE CONNECT ON DATABASE template1 FROM $old_dbuser"
psql_query "$query" > /dev/null
query="DROP ROLE $old_dbuser"
psql_query "$query" > /dev/null
}
# Get database name without user prefix
get_database_name_without_user_prefix() {
user=$1
db=$2
user_len=${#user}
user_len=$((user_len+1))
echo ${db:user_len}
}
# Check if database exists
check_if_database_exists() {
USER_DATA_PATH=$VESTA/data/users/$1
if [ ! -d "$USER_DATA_PATH" ]; then
echo "no"
return;
fi
counter=$(grep -c "DB='$2'" $USER_DATA_PATH/db.conf)
if [ "$counter" = "0" ]; then
echo "no"
else
echo "yes"
fi
}