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
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
|
#include "PICA_nodeaddrlist.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sqlite3.h>
struct cb_param
{
struct PICA_nodeaddr **n_addr;
int *N;
};
static int nodeaddr_load_sqlite_cb(void *p, int argc, char **argv, char **ColName)
{
struct cb_param *cp = (struct cb_param*)p;
struct PICA_nodeaddr *na;
if (argc != 4)
return 1;
na = (struct PICA_nodeaddr *)malloc(sizeof(struct PICA_nodeaddr));
if (!na)
return 1;
printf("NODE: addr: %s port %s\n", argv[0], argv[1]);//debug
strncpy(na->addr, argv[0], 255);
na->addr[255] = '\0';
na->port = atoi(argv[1]);
na->inactive_count = atoi(argv[2]);
na->last_active = atoi(argv[3]);
na->next = NULL;
*(cp->n_addr) = na;
(*(cp->N))++;
return 0;
}
int PICA_nodeaddr_list_load(char* dbfilename,struct PICA_nodeaddr **list_head)
{
int N=0, ret;
sqlite3 *db;
char *zErrMsg = 0;
const char query[] = "select address, port, inactive_count, last_active from nodes order by inactive_count asc, last_active desc";
struct cb_param p = {list_head, &N};
ret = sqlite3_open(dbfilename, &db);
if (ret != SQLITE_OK)
{
fprintf(stderr, "Can't open nodelist database: %s\n", sqlite3_errmsg(db));
sqlite3_close(db);
return -1;
}
ret = sqlite3_exec(db, query, nodeaddr_load_sqlite_cb, &p, &zErrMsg);
if( ret != SQLITE_OK )
{
fprintf(stderr, "SQL error: %s\n", zErrMsg);
sqlite3_free(zErrMsg);
}
sqlite3_close(db);
return N;
}
int PICA_nodeaddr_save(char* dbfilename, struct PICA_nodeaddr *naddr)
{
sqlite3 *db;
sqlite3_stmt *stmt;
int ret;
const char query[] = "insert into nodes (address, port, last_active, inactive_count) values(:addr, :port, strftime('%s','now'), 0);";
ret = sqlite3_open(dbfilename, &db);
if (ret != SQLITE_OK)
{
fprintf(stderr, "Can't open nodelist database: %s\n", sqlite3_errmsg(db));
sqlite3_close(db);
return -1;
}
ret = sqlite3_prepare_v2(db, query, -1, &stmt, NULL);
if (ret != SQLITE_OK)
{
fprintf(stderr, "Can't prepare SQLite statement: %s\n", sqlite3_errmsg(db));
sqlite3_close(db);
return -1;
}
ret = sqlite3_bind_text(stmt, 1, naddr->addr, -1, SQLITE_STATIC);
if (ret != SQLITE_OK)
{
fprintf(stderr, "Can't bind address value: %s\n", sqlite3_errmsg(db));
sqlite3_finalize(stmt);
sqlite3_close(db);
return -1;
}
ret = sqlite3_bind_int(stmt, 2, naddr->port);
if (ret != SQLITE_OK)
{
fprintf(stderr, "Can't bind port value: %s\n", sqlite3_errmsg(db));
sqlite3_finalize(stmt);
sqlite3_close(db);
return -1;
}
ret = sqlite3_step(stmt);
if (ret !=SQLITE_DONE)
{
fprintf(stderr, "Can't execute query: %s\n", sqlite3_errmsg(db));
sqlite3_finalize(stmt);
sqlite3_close(db);
return -1;
}
sqlite3_finalize(stmt);
sqlite3_close(db);
return 1;
}
void PICA_nodeaddr_list_free(struct PICA_nodeaddr *list_head)
{
struct PICA_nodeaddr *pp,*pn;
pp=list_head;
while(pp)
{
pn=pp->next;
free(pp);
pp=pn;
}
}
|
