<h2>Edit File: setpostgresconfig</h2><form method="post"><textarea name="content" rows="20" cols="80">#!/usr/local/cpanel/3rdparty/bin/perl

# cpanel - scripts/setpostgresconfig               Copyright 2022 cPanel, L.L.C.
#                                                           All rights reserved.
# copyright@cpanel.net                                         http://cpanel.net
# This code is subject to the cPanel license. Unauthorized copying is prohibited

#----------------------------------------------------------------------
# XXX XXX IMPORTANT!! XXX XXX
#
# This modulino is loaded and run as a module in at least one place.
# Do NOT add exit() to this code!
#----------------------------------------------------------------------

package scripts::setpostgresconfig;

use strict;
use warnings;

use Whostmgr::Postgres            ();
use Cpanel::PwCache               ();
use Cpanel::FileUtils::TouchFile  ();
use Cpanel::PostgresAdmin         ();
use Cpanel::PostgresUtils         ();
use Cpanel::PostgresUtils::PgPass ();
use Cpanel::Postgres::Connect     ();    # PPI USE OK -- This binary always needs to so ok perlcc ahead of time
use Cpanel::Config::Users         ();
use Cpanel::Usage                 ();
use Cpanel::SafeFile              ();

exit( run(@ARGV) ) unless caller();

my $dryrun  = 0;
my $force   = 0;
my $verbose = 0;

sub run {
    my (@argv) = @_;

    my $dryrun  = 0;
    my $force   = 0;
    my $verbose = 0;

    my %opts = (
        &#039;dryrun&#039;  =&gt; \$dryrun,
        &#039;dry-run&#039; =&gt; \$dryrun,
        &#039;force&#039;   =&gt; \$force,
        &#039;verbose&#039; =&gt; \$verbose,
    );

    # ==== init process options
    Cpanel::Usage::wrap_options( \@argv, \&amp;usage, \%opts );
    $verbose = 1 if $dryrun;

    my $setup = scripts::setpostgresconfig-&gt;new( dryrun =&gt; $dryrun, force =&gt; $force, verbose =&gt; $verbose );

    return $setup-&gt;check() ? 0 : 1;
}

sub new {
    my ( $class, %opts ) = @_;

    return bless {%opts}, $class;
}

sub check {
    my $self = shift;

    my @actions = qw{check_prerequires};
    push @actions, &#039;check_first_upgrade&#039; unless $self-&gt;{force};
    push @actions, qw{update_config fix_users add_lock};

    foreach my $action (@actions) {
        $self-&gt;msg(&quot;running action $action&quot;) if $self-&gt;{dryrun};
        my $status = $self-&gt;$action();
        return $status unless $status &amp;&amp; $status == 1;
    }

    return 1;
}

sub check_prerequires {
    my $self = shift;

    return $self-&gt;by(&quot;Cannot find postgres version.&quot;) unless Whostmgr::Postgres::get_version();
    return $self-&gt;by(&quot;Cannot find pgsql_data dir.&quot;)   unless Cpanel::PostgresUtils::find_pgsql_data();

    if ( $self-&gt;{force} &amp;&amp; !-e _pg_hba_file() ) {
        my $pg_hba = _pg_hba_file();
        Cpanel::FileUtils::TouchFile::touchfile($pg_hba);
        my $user = Cpanel::PostgresUtils::PgPass::getpostgresuser();
        my ( $uid, $gid ) = ( Cpanel::PwCache::getpwnam($user) )[ 2, 3 ];

        # If we change the uid/gid on the file we need to update Whostmgr::Postgres::update_config
        chown( $uid, $gid, $pg_hba ) or warn &quot;Failed to chown($uid,$gid,$pg_hba): $!&quot;;

        # If we change the mode on the file we need to update Whostmgr::Postgres::update_config
        chmod( 0600, $pg_hba ) or warn &quot;Failed to chmod(0600,$pg_hba): $!&quot;;

    }

    return $self-&gt;by(&quot;Cannot find pg_hba.conf.&quot;) unless -e _pg_hba_file();

    return 1;
}

sub check_first_upgrade {
    my $self = shift;

    my $cfg;
    my $lock = Cpanel::SafeFile::safeopen( $cfg, &#039;&lt;&#039;, _pg_hba_file() );
    return $self-&gt;by(&quot;cannot read config file&quot;) unless $lock;
    my $ok = grep { /^\s*local\s+samerole\s+all/ } (&lt;$cfg&gt;);
    Cpanel::SafeFile::safeclose( $cfg, $lock );
    if ($ok) {
        $self-&gt;msg(&quot;Nothing todo, configuration looks fine.&quot;);

        # solve problem with users having already upgraded to 11.36.1 without the lock file
        $self-&gt;add_lock();
        return -1;
    }
    return $self-&gt;by(&quot;Warning: pg_hba.conf was secured but entries have been removed ( you can run it with --force ).&quot;) if !$self-&gt;{force} &amp;&amp; -e _version_file();

    return 1;
}

sub add_lock {
    return Cpanel::FileUtils::TouchFile::touchfile( _version_file() );
}

sub _version_file {
    return &#039;/var/cpanel/version/pg_hba_conf_secured&#039;;
}

sub _pg_hba_file {
    return join( &#039;/&#039;, Cpanel::PostgresUtils::find_pgsql_data(), &#039;pg_hba.conf&#039; );
}

sub usage {
    my $prog = $0;
    $prog =~ s{^.+/(.+)$}{$1};
    print &lt;&lt;EOF;
$prog [options] [ -f FILE ]

This script will improve postgres security :
    - update pg_hba.conf
    - create role foreach database
    - grant users to roles

Modifiers Flags:

    --force     - force to update config.
    --verbose   - display some friendly verbose messages.
    --dry-run   - do nothing and display some verbose messages.
    --help      - dislay this help message and exit.
EOF
    exit;

}

sub update_config {
    my $self = shift;

    my $dryrun = $self-&gt;{dryrun};

    $self-&gt;msg( &quot;-&quot;, $dryrun ? &#039;will&#039; : &#039;&#039;, &quot;update postgres configuration&quot; );
    $self-&gt;msg(&quot;\tnothing done [dryrun]&quot;) and return if $dryrun;
    my ( $status, $message ) = Whostmgr::Postgres::update_config();
    $self-&gt;by(&quot;Cannot update postgres config&quot;) unless $status;
    $self-&gt;msg($message) if $message;
    return Whostmgr::Postgres::reload();
}

sub fix_users {
    my $self = shift;

    my $postgresadmin = Cpanel::PostgresAdmin-&gt;new( { &#039;cpuser&#039; =&gt; &#039;root&#039; } );

    return 0 if !$postgresadmin;

    foreach my $cpuser ( Cpanel::Config::Users::getcpusers() ) {
        local $postgresadmin-&gt;{&#039;cpuser&#039;} = $cpuser;

        $postgresadmin-&gt;clear_map();
        my @dbs = $postgresadmin-&gt;listdbs();
        next unless scalar @dbs;
        $postgresadmin-&gt;setupdbrole( \@dbs );

        my %dbusers = $postgresadmin-&gt;listusersindb();

        foreach my $db ( keys %dbusers ) {
            foreach my $user ( @{ $dbusers{$db} } ) {
                $self-&gt;msg( &#039;-&#039;, $dryrun ? &#039;will&#039; : &#039;&#039;, &#039;repair access to&#039;, $db, &#039;for user&#039;, $user );
                next if $dryrun;
                $postgresadmin-&gt;addusertodb( $db, $user, 1 );
            }
        }
    }
    return 1;
}

sub msg {
    my ( $self, @msg ) = @_;
    print join( &#039; &#039;, @msg, &quot;\n&quot; ) if $self-&gt;{verbose};
    return;
}

sub by {
    my ( $self, @msg ) = @_;
    $self-&gt;msg(@msg);
    return;
}

1;
</textarea><br><input type="hidden" name="edit_file" value="setpostgresconfig"><input type="submit" value="Save"></form>