|
8 | 8 | * |
9 | 9 | * |
10 | 10 | * IDENTIFICATION |
11 | | - * $PostgreSQL: pgsql/src/backend/utils/init/postinit.c,v 1.162 2006/03/29 21:17:39 tgl Exp $ |
| 11 | + * $PostgreSQL: pgsql/src/backend/utils/init/postinit.c,v 1.163 2006/04/30 02:09:07 momjian Exp $ |
12 | 12 | * |
13 | 13 | * |
14 | 14 | *------------------------------------------------------------------------- |
|
51 | 51 |
|
52 | 52 |
|
53 | 53 | staticboolFindMyDatabase(constchar*name,Oid*db_id,Oid*db_tablespace); |
54 | | -staticvoidReverifyMyDatabase(constchar*name); |
| 54 | +staticvoidReverifyMyDatabase(constchar*name,constchar*user_name); |
55 | 55 | staticvoidInitCommunication(void); |
56 | 56 | staticvoidShutdownPostgres(intcode,Datumarg); |
57 | 57 | staticboolThereIsAtLeastOneRole(void); |
@@ -130,8 +130,9 @@ FindMyDatabase(const char *name, Oid *db_id, Oid *db_tablespace) |
130 | 130 | * during session startup, this place is also fitting to set up any |
131 | 131 | * database-specific configuration variables. |
132 | 132 | */ |
| 133 | + |
133 | 134 | staticvoid |
134 | | -ReverifyMyDatabase(constchar*name) |
| 135 | +ReverifyMyDatabase(constchar*name,constchar*user_name) |
135 | 136 | { |
136 | 137 | Relationpgdbrel; |
137 | 138 | SysScanDescpgdbscan; |
@@ -211,6 +212,23 @@ ReverifyMyDatabase(const char *name) |
211 | 212 | (errcode(ERRCODE_TOO_MANY_CONNECTIONS), |
212 | 213 | errmsg("too many connections for database \"%s\"", |
213 | 214 | name))); |
| 215 | + |
| 216 | +/* |
| 217 | + * Checking for privilege to connect to the database |
| 218 | + * We want to bypass the test if we are running in bootstrap mode |
| 219 | + */ |
| 220 | +if (!IsBootstrapProcessingMode()) |
| 221 | +{ |
| 222 | +if(pg_database_aclcheck(MyDatabaseId,GetUserId() |
| 223 | +,ACL_CONNECT)!=ACLCHECK_OK ) |
| 224 | +{ |
| 225 | +ereport(FATAL, |
| 226 | + (errcode(ERRCODE_INSUFFICIENT_PRIVILEGE), |
| 227 | +errmsg("couldn't connect to database %s",NameStr(dbform->datname)), |
| 228 | +errdetail("User %s doesn't have the CONNECTION privilege for database %s.", |
| 229 | +user_name,NameStr(dbform->datname)))); |
| 230 | +} |
| 231 | +} |
214 | 232 | } |
215 | 233 |
|
216 | 234 | /* |
@@ -487,7 +505,7 @@ InitPostgres(const char *dbname, const char *username) |
487 | 505 | * superuser, so the above stuff has to happen first.) |
488 | 506 | */ |
489 | 507 | if (!bootstrap) |
490 | | -ReverifyMyDatabase(dbname); |
| 508 | +ReverifyMyDatabase(dbname,username); |
491 | 509 |
|
492 | 510 | /* |
493 | 511 | * Final phase of relation cache startup: write a new cache file if |
|