1296
|
1 /* Copyright (C) 2003 Alex Howansky, Timo Sirainen */
|
|
2
|
|
3 #include "config.h"
|
|
4 #undef HAVE_CONFIG_H
|
|
5
|
|
6 #ifdef USERDB_PGSQL
|
|
7
|
|
8 #include "common.h"
|
|
9 #include "str.h"
|
|
10 #include "var-expand.h"
|
|
11 #include "db-pgsql.h"
|
|
12 #include "userdb.h"
|
|
13
|
|
14 #include <libpq-fe.h>
|
|
15 #include <stdlib.h>
|
|
16 #include <string.h>
|
|
17
|
|
18 struct userdb_pgsql_connection {
|
|
19 struct pgsql_connection *conn;
|
|
20 };
|
|
21
|
|
22 struct userdb_pgsql_request {
|
|
23 struct pgsql_request request;
|
|
24 userdb_callback_t *userdb_callback;
|
|
25 };
|
|
26
|
|
27 static struct userdb_pgsql_connection *userdb_pgsql_conn;
|
|
28
|
|
29 static int is_result_valid(PGresult *res)
|
|
30 {
|
|
31 if (PQresultStatus(res) != PGRES_TUPLES_OK) {
|
|
32 i_error("PGSQL: Query failed");
|
|
33 return FALSE;
|
|
34 }
|
|
35
|
|
36 if (PQntuples(res) == 0) {
|
|
37 if (verbose)
|
|
38 i_error("PGSQL: Authenticated user not found");
|
|
39 return FALSE;
|
|
40 }
|
|
41
|
|
42 if (PQfnumber(res, "home") == -1) {
|
|
43 i_error("PGSQL: User query did not return 'home' field");
|
|
44 return FALSE;
|
|
45 }
|
|
46
|
|
47 if (PQfnumber(res, "uid") == -1) {
|
|
48 i_error("PGSQL: User query did not return 'uid' field");
|
|
49 return FALSE;
|
|
50 }
|
|
51
|
|
52 if (PQfnumber(res, "gid") == -1) {
|
|
53 i_error("PGSQL: User query did not return 'gid' field");
|
|
54 return FALSE;
|
|
55 }
|
|
56
|
|
57 return TRUE;
|
|
58 }
|
|
59
|
|
60 static void pgsql_handle_request(struct pgsql_connection *conn __attr_unused__,
|
|
61 struct pgsql_request *request, PGresult *res)
|
|
62 {
|
|
63 struct userdb_pgsql_request *urequest =
|
|
64 (struct userdb_pgsql_request *) request;
|
|
65 struct user_data user;
|
|
66
|
|
67 if (res != NULL && is_result_valid(res)) {
|
|
68 memset(&user, 0, sizeof(user));
|
|
69 user.home = PQgetvalue(res, 0, PQfnumber(res, "home"));
|
|
70 user.uid = atoi(PQgetvalue(res, 0, PQfnumber(res, "uid")));
|
|
71 user.gid = atoi(PQgetvalue(res, 0, PQfnumber(res, "gid")));
|
|
72 urequest->userdb_callback(&user, request->context);
|
|
73 } else {
|
|
74 urequest->userdb_callback(NULL, request->context);
|
|
75 }
|
|
76 }
|
|
77
|
|
78 static void userdb_pgsql_lookup(const char *user, userdb_callback_t *callback,
|
|
79 void *context)
|
|
80 {
|
|
81 struct pgsql_connection *conn = userdb_pgsql_conn->conn;
|
|
82 struct userdb_pgsql_request *request;
|
|
83 const char *query;
|
|
84 string_t *str;
|
|
85
|
|
86 str = t_str_new(512);
|
|
87 var_expand(str, conn->set.user_query, user, NULL);
|
|
88 query = str_c(str);
|
|
89
|
|
90 request = i_new(struct userdb_pgsql_request, 1);
|
|
91 request->request.callback = pgsql_handle_request;
|
|
92 request->request.context = context;
|
|
93 request->userdb_callback = callback;
|
|
94
|
|
95 if (db_pgsql_is_valid_username(conn, user))
|
|
96 db_pgsql_query(conn, query, &request->request);
|
|
97 else {
|
|
98 if (verbose)
|
|
99 i_info("pgsql(%s): Invalid username", user);
|
|
100 pgsql_handle_request(conn, &request->request, NULL);
|
|
101 }
|
|
102 }
|
|
103
|
|
104 static void userdb_pgsql_init(const char *args)
|
|
105 {
|
|
106 struct pgsql_connection *conn;
|
|
107
|
|
108 userdb_pgsql_conn = i_new(struct userdb_pgsql_connection, 1);
|
|
109 userdb_pgsql_conn->conn = conn = db_pgsql_init(args);
|
|
110 }
|
|
111
|
|
112 static void userdb_pgsql_deinit(void)
|
|
113 {
|
|
114 db_pgsql_unref(userdb_pgsql_conn->conn);
|
|
115 i_free(userdb_pgsql_conn);
|
|
116 }
|
|
117
|
|
118 struct userdb_module userdb_pgsql = {
|
|
119 userdb_pgsql_init,
|
|
120 userdb_pgsql_deinit,
|
|
121
|
|
122 userdb_pgsql_lookup
|
|
123 };
|
|
124
|
|
125 #endif
|