PostgreSQL
 sql >> Datenbank >  >> RDS >> PostgreSQL

Wechseln Sie die Rolle, nachdem Sie sich mit der Datenbank verbunden haben

--create a user that you want to use the database as:

create role neil;

--create the user for the web server to connect as:

create role webgui noinherit login password 's3cr3t';

--let webgui set role to neil:

grant neil to webgui; --this looks backwards but is correct.

webgui befindet sich jetzt im neil Gruppe, also webgui kann set role neil aufrufen . Allerdings webgui hat neil nicht geerbt die Berechtigungen von .

Melden Sie sich später als Webgui an:

psql -d some_database -U webgui
(enter s3cr3t as password)

set role neil;

webgui benötigt keinen superuser Erlaubnis dafür.

Sie möchten set role zu Beginn einer Datenbanksitzung und setzen sie am Ende der Sitzung zurück. In einer Web-App entspricht dies dem Abrufen einer Verbindung aus Ihrem Datenbankverbindungspool bzw. dem Freigeben dieser Verbindung. Hier ist ein Beispiel mit Tomcats Verbindungspool und Spring Security:

public class SetRoleJdbcInterceptor extends JdbcInterceptor {

    @Override
    public void reset(ConnectionPool connectionPool, PooledConnection pooledConnection) {

        Authentication authentication = SecurityContextHolder.getContext().getAuthentication();

        if(authentication != null) {
            try {

                /* 
                  use OWASP's ESAPI to encode the username to avoid SQL Injection. Can't use parameters with SET ROLE. Need to write PG codec.

                  Or use a whitelist-map approach
                */
                String username = ESAPI.encoder().encodeForSQL(MY_CODEC, authentication.getName());

                Statement statement = pooledConnection.getConnection().createStatement();
                statement.execute("set role \"" + username + "\"");
                statement.close();
            } catch(SQLException exp){
                throw new RuntimeException(exp);
            }
        }
    }

    @Override
    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {

        if("close".equals(method.getName())){
            Statement statement = ((Connection)proxy).createStatement();
            statement.execute("reset role");
            statement.close();
        }

        return super.invoke(proxy, method, args);
    }
}