98dbb43878e4219ac34cf447d4d5bf30aac073cd
[sip-router] / modules / db_postgres / km_pg_con.c
1 /* 
2  * $Id$
3  *
4  * Copyright (C) 2001-2004 iptel.org
5  * Copyright (C) 2008 1&1 Internet AG
6  *
7  * This file is part of Kamailio, a free SIP server.
8  *
9  * Kamailio is free software; you can redistribute it and/or modify
10  * it under the terms of the GNU General Public License as published by
11  * the Free Software Foundation; either version 2 of the License, or
12  * (at your option) any later version
13  *
14  * Kamailio is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17  * GNU General Public License for more details.
18  *
19  * You should have received a copy of the GNU General Public License 
20  * along with this program; if not, write to the Free Software 
21  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
22  */
23
24 /*! \file
25  *  \brief DB_POSTGRES :: Core
26  *  \ingroup db_postgres
27  *  Module: \ref db_postgres
28  */
29
30 #include "km_pg_con.h"
31 #include "../../mem/mem.h"
32 #include "../../dprint.h"
33 #include "../../ut.h"
34 #include <string.h>
35 #include <time.h>
36
37
38 /*!
39  * \brief Create a new connection
40  *
41  * Create a new connection structure in private memory, open the PostgreSQL
42  * connection and set reference count to 1
43  * \param id database id
44  * \return postgres connection structure, 0 on error
45  */
46 struct pg_con* db_postgres_new_connection(struct db_id* id)
47 {
48         struct pg_con* ptr;
49         char *ports;
50
51         LM_DBG("db_id = %p\n", id);
52  
53         if (!id) {
54                 LM_ERR("invalid db_id parameter value\n");
55                 return 0;
56         }
57
58         ptr = (struct pg_con*)pkg_malloc(sizeof(struct pg_con));
59         if (!ptr) {
60                 LM_ERR("failed trying to allocated %lu bytes for connection structure."
61                                 "\n", (unsigned long)sizeof(struct pg_con));
62                 return 0;
63         }
64         LM_DBG("%p=pkg_malloc(%lu)\n", ptr, (unsigned long)sizeof(struct pg_con));
65
66         memset(ptr, 0, sizeof(struct pg_con));
67         ptr->ref = 1;
68
69         if (id->port) {
70                 ports = int2str(id->port, 0);
71                 LM_DBG("opening connection: postgres://xxxx:xxxx@%s:%d/%s\n", ZSW(id->host),
72                         id->port, ZSW(id->database));
73         } else {
74                 ports = NULL;
75                 LM_DBG("opening connection: postgres://xxxx:xxxx@%s/%s\n", ZSW(id->host),
76                         ZSW(id->database));
77         }
78
79         ptr->con = PQsetdbLogin(id->host, ports, NULL, NULL, id->database, id->username, id->password);
80         LM_DBG("PQsetdbLogin(%p)\n", ptr->con);
81
82         if( (ptr->con == 0) || (PQstatus(ptr->con) != CONNECTION_OK) )
83         {
84                 LM_ERR("%s\n", PQerrorMessage(ptr->con));
85                 PQfinish(ptr->con);
86                 goto err;
87         }
88
89         ptr->connected = 1;
90         ptr->timestamp = time(0);
91         ptr->id = id;
92
93         return ptr;
94
95  err:
96         if (ptr) {
97                 LM_ERR("cleaning up %p=pkg_free()\n", ptr);
98                 pkg_free(ptr);
99         }
100         return 0;
101 }
102
103
104 /*!
105  * \brief Close the connection and release memory
106  * \param con connection
107  */
108 void db_postgres_free_connection(struct pool_con* con)
109 {
110
111         struct pg_con * _c;
112         
113         if (!con) return;
114
115         _c = (struct pg_con*)con;
116
117         if (_c->res) {
118                 LM_DBG("PQclear(%p)\n", _c->res);
119                 PQclear(_c->res);
120                 _c->res = 0;
121         }
122         if (_c->id) free_db_id(_c->id);
123         if (_c->con) {
124                 LM_DBG("PQfinish(%p)\n", _c->con);
125                 PQfinish(_c->con);
126                 _c->con = 0;
127         }
128         LM_DBG("pkg_free(%p)\n", _c);
129         pkg_free(_c);
130 }