1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
/*!
\file db/driver/postgres/listdb.c
\brief DBMI - Low Level PostgreSQL database driver - list databases
This program is free software under the GNU General Public License
(>=v2). Read the file COPYING that comes with GRASS for details.
\author Radim Blazek
\author Updated for GRASS 7 by Martin Landa <landa.martin gmail.com>
*/
#include <grass/dbmi.h>
#include <grass/glocale.h>
#include "globals.h"
#include "proto.h"
int db__driver_list_databases(dbString *dbpath, int npaths, dbHandle **dblist,
int *dbcount)
{
int i;
const char *user, *passwd, *host, *port;
PGCONN pgconn;
PGresult *res;
int rec_num = 0;
dbHandle *list;
*dblist = NULL;
*dbcount = 0;
/* TODO: the solution below is not good as user usually does not
* have permissions for "template1" */
if (npaths < 1) {
db_d_append_error(_("No path given"));
db_d_report_error();
return DB_FAILED;
}
if (parse_conn(db_get_string(dbpath), &pgconn) == DB_FAILED) {
db_d_report_error();
return DB_FAILED;
}
G_debug(1,
"db = %s, user = %s, pass = %s, "
"host = %s, port = %s, options = %s, tty = %s",
pgconn.dbname, pgconn.user, pgconn.password, pgconn.host,
pgconn.port, pgconn.options, pgconn.tty);
db_get_login2("pg", NULL, &user, &passwd, &host, &port);
G_debug(1, "user = %s, passwd = %s", user, passwd ? "xxx" : "");
if (user || passwd) {
pg_conn = PQsetdbLogin(host, port, pgconn.options, pgconn.tty,
"template1", user, passwd);
}
else {
pg_conn = PQsetdb(host, port, pgconn.options, pgconn.tty, "template1");
}
if (PQstatus(pg_conn) == CONNECTION_BAD) {
db_d_append_error("%s\n%s", _("Unable to connect to Postgres:"),
PQerrorMessage(pg_conn));
db_d_report_error();
PQfinish(pg_conn);
return DB_FAILED;
}
res = PQexec(pg_conn, "select datname from pg_database");
if (!res || PQresultStatus(res) != PGRES_TUPLES_OK) {
db_d_append_error("%s\n%s", _("Unable to select from Postgres:"),
PQerrorMessage(pg_conn));
db_d_report_error();
PQclear(res);
PQfinish(pg_conn);
return DB_FAILED;
}
rec_num = PQntuples(res);
list = db_alloc_handle_array(rec_num);
if (list == NULL) {
db_d_append_error(_("Out of memory"));
db_d_report_error();
return DB_FAILED;
}
for (i = 0; i < rec_num; i++) {
db_init_handle(&list[i]);
if (db_set_handle(&list[i], PQgetvalue(res, i, 0), NULL) != DB_OK) {
db_d_append_error(_("Unable to set handle"));
db_d_report_error();
db_free_handle_array(list, rec_num);
return DB_FAILED;
}
}
PQclear(res);
PQfinish(pg_conn);
*dblist = list;
*dbcount = rec_num;
return DB_OK;
}
|
