1 /*
2  * Copyright (C) 2001-2003 FhG FOKUS
3  * Copyright (C) 2006-2007 iptelorg GmbH
4  *
5  * This file is part of Kamailio, a free SIP server.
6  *
7  * Kamailio is free software; you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License as published by
9  * the Free Software Foundation; either version 2 of the License, or
10  * (at your option) any later version
11  *
12  * Kamailio is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  * GNU General Public License for more details.
16  *
17  * You should have received a copy of the GNU General Public License
18  * along with this program; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
20  */
21 
22 /** \ingroup DB_API
23  * @{
24  */
25 
26 #include "db_res.h"
27 #include "../../core/dprint.h"
28 #include "../../core/mem/mem.h"
29 #include <string.h>
30 
31 
db_res(db_cmd_t * cmd)32 db_res_t* db_res(db_cmd_t* cmd)
33 {
34     db_res_t* newp;
35 	int ret;
36 
37     newp = (db_res_t*)pkg_malloc(sizeof(db_res_t));
38     if (newp == NULL) goto err;
39 	memset(newp, '\0', sizeof(db_res_t));
40 	if (db_gen_init(&newp->gen) < 0) goto err;
41     newp->cmd = cmd;
42 	newp->field_count = cmd->result_count;
43 
44 	ret = db_drv_call(&cmd->ctx->con[db_payload_idx]->uri->scheme,
45 					  "db_res", newp, db_payload_idx);
46 	if (ret < 0) goto err;
47 
48 	newp->cur_rec = db_rec(newp, cmd->result);
49 	if (newp->cur_rec == NULL) goto err;
50     return newp;
51 
52  err:
53     ERR("db_res: Cannot create db_res structure\n");
54 	if (newp) {
55 		if (newp->cur_rec) db_rec_free(newp->cur_rec);
56 		db_gen_free(&newp->gen);
57 		pkg_free(newp);
58 	}
59     return NULL;
60 }
61 
62 
db_res_free(db_res_t * r)63 void db_res_free(db_res_t* r)
64 {
65     if (r == NULL) return;
66 	db_gen_free(&r->gen);
67 	if (r->cur_rec) db_rec_free(r->cur_rec);
68     pkg_free(r);
69 }
70 
71 
db_first(db_res_t * res)72 db_rec_t* db_first(db_res_t* res)
73 {
74 	if (res->cmd->first[0](res) != 0) {
75 		return NULL;
76 	}
77 	return res->cur_rec;
78 }
79 
80 
db_next(db_res_t * res)81 db_rec_t* db_next(db_res_t* res)
82 {
83 	if (res->cmd->next[0](res) != 0) {
84 		return NULL;
85 	}
86 	return res->cur_rec;
87 }
88 
89 /** @} */
90