1 /* $NetBSD: zone2sqlite.c,v 1.5 2014/12/10 04:37:57 christos Exp $ */
2
3 /*
4 * Copyright (C) 2007 Internet Software Consortium.
5 *
6 * Permission to use, copy, modify, and distribute this software for any
7 * purpose with or without fee is hereby granted, provided that the above
8 * copyright notice and this permission notice appear in all copies.
9 *
10 * THE SOFTWARE IS PROVIDED "AS IS" AND INTERNET SOFTWARE CONSORTIUM
11 * DISCLAIMS ALL WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL
12 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL
13 * INTERNET SOFTWARE CONSORTIUM BE LIABLE FOR ANY SPECIAL, DIRECT,
14 * INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING
15 * FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT,
16 * NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION
17 * WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
18 */
19
20 /* Id: zone2sqlite.c,v 1.4 2010/08/16 05:32:44 marka Exp */
21
22 #include <stdlib.h>
23 #include <string.h>
24
25 #include <isc/buffer.h>
26 #include <isc/mem.h>
27 #include <isc/print.h>
28 #include <isc/result.h>
29
30 #include <dns/db.h>
31 #include <dns/dbiterator.h>
32 #include <isc/entropy.h>
33 #include <dns/fixedname.h>
34 #include <isc/hash.h>
35 #include <dns/name.h>
36 #include <dns/rdata.h>
37 #include <dns/rdataset.h>
38 #include <dns/rdatasetiter.h>
39 #include <dns/rdatatype.h>
40 #include <dns/result.h>
41
42 #include <sqlite3.h>
43
44 #ifndef UNUSED
45 #define UNUSED(x) (void)&(x)
46 #endif
47
48 /*
49 * Generate an SQLite table from a zone.
50 */
51
52 typedef struct _dbinfo {
53 sqlite3 *db;
54 char *filename;
55 char *table;
56 } dbinfo_t;
57
58 dbinfo_t dbi = { NULL, NULL, NULL };
59
60
61 static void
closeandexit(int status)62 closeandexit(int status)
63 {
64 if (dbi.db) {
65 sqlite3_close(dbi.db);
66 dbi.db = NULL;
67 }
68 exit(status);
69 }
70
71 static void
check_result(isc_result_t result,const char * message)72 check_result(isc_result_t result, const char *message)
73 {
74 if (result != ISC_R_SUCCESS) {
75 fprintf(stderr, "%s: %s\n", message,
76 isc_result_totext(result));
77 closeandexit(1);
78 }
79 }
80
81 static isc_result_t
db_connect(dbinfo_t * dbi)82 db_connect(dbinfo_t *dbi)
83 {
84 if (sqlite3_open(dbi->filename, &dbi->db) == SQLITE_OK) {
85 return (ISC_R_SUCCESS);
86 } else {
87 /* a connection is returned even if the open fails */
88 sqlite3_close(dbi->db);
89 dbi->db = NULL;
90 return (ISC_R_FAILURE);
91 }
92 }
93
94 static int
add_rdata_cb(void * parm,int cc,char ** cv,char ** cn)95 add_rdata_cb(void *parm, int cc, char **cv, char **cn)
96 {
97 UNUSED(parm);
98 UNUSED(cc);
99 UNUSED(cv);
100 UNUSED(cn);
101
102 return 0;
103 }
104
105
106 static void
addrdata(dns_name_t * name,dns_ttl_t ttl,dns_rdata_t * rdata)107 addrdata(dns_name_t *name, dns_ttl_t ttl, dns_rdata_t *rdata)
108 {
109 unsigned char namearray[DNS_NAME_MAXTEXT + 1];
110 unsigned char typearray[20];
111 unsigned char dataarray[2048];
112 isc_buffer_t b;
113 isc_result_t result;
114 char *sql;
115 char *errmsg = NULL;
116 int res;
117
118 isc_buffer_init(&b, namearray, sizeof(namearray) - 1);
119 result = dns_name_totext(name, ISC_TRUE, &b);
120 check_result(result, "dns_name_totext");
121 namearray[isc_buffer_usedlength(&b)] = 0;
122
123 isc_buffer_init(&b, typearray, sizeof(typearray) - 1);
124 result = dns_rdatatype_totext(rdata->type, &b);
125 check_result(result, "dns_rdatatype_totext");
126 typearray[isc_buffer_usedlength(&b)] = 0;
127
128 isc_buffer_init(&b, dataarray, sizeof(dataarray) - 1);
129 result = dns_rdata_totext(rdata, NULL, &b);
130 check_result(result, "dns_rdata_totext");
131 dataarray[isc_buffer_usedlength(&b)] = 0;
132
133 sql = sqlite3_mprintf(
134 "INSERT INTO %Q (NAME, TTL, RDTYPE, RDATA)"
135 " VALUES ('%q', %d, '%q', '%q') ",
136 dbi.table,
137 namearray, ttl, typearray, dataarray);
138 printf("%s\n", sql);
139 res = sqlite3_exec(dbi.db, sql, add_rdata_cb, NULL, &errmsg);
140 sqlite3_free(sql);
141
142 if (res != SQLITE_OK) {
143 fprintf(stderr, "INSERT failed: %s\n", errmsg);
144 closeandexit(1);
145 }
146 }
147
148 int
main(int argc,char * argv[])149 main(int argc, char *argv[])
150 {
151 char *sql;
152 int res;
153 char *errmsg = NULL;
154 char *porigin, *zonefile;
155 dns_fixedname_t forigin, fname;
156 dns_name_t *origin, *name;
157 dns_db_t *db = NULL;
158 dns_dbiterator_t *dbiter;
159 dns_dbnode_t *node;
160 dns_rdatasetiter_t *rdsiter;
161 dns_rdataset_t rdataset;
162 dns_rdata_t rdata = DNS_RDATA_INIT;
163 isc_mem_t *mctx = NULL;
164 isc_entropy_t *ectx = NULL;
165 isc_buffer_t b;
166 isc_result_t result;
167
168 if (argc != 5) {
169 printf("usage: %s <zone> <zonefile> <dbfile> <dbtable>\n", argv[0]);
170 exit(1);
171 }
172
173 porigin = argv[1];
174 zonefile = argv[2];
175
176 dbi.filename = argv[3];
177 dbi.table = argv[4];
178
179 dns_result_register();
180
181 result = isc_mem_create(0, 0, &mctx);
182 check_result(result, "isc_mem_create");
183 result = isc_entropy_create(mctx, &ectx);
184 check_result(result, "isc_entropy_create");
185 result = isc_hash_create(mctx, ectx, DNS_NAME_MAXWIRE);
186 check_result(result, "isc_hash_create");
187
188 isc_buffer_init(&b, porigin, strlen(porigin));
189 isc_buffer_add(&b, strlen(porigin));
190 dns_fixedname_init(&forigin);
191 origin = dns_fixedname_name(&forigin);
192 result = dns_name_fromtext(origin, &b, dns_rootname, 0, NULL);
193 check_result(result, "dns_name_fromtext");
194
195 db = NULL;
196 result = dns_db_create(mctx, "rbt", origin, dns_dbtype_zone,
197 dns_rdataclass_in, 0, NULL, &db);
198 check_result(result, "dns_db_create");
199
200 result = dns_db_load(db, zonefile);
201 if (result == DNS_R_SEENINCLUDE)
202 result = ISC_R_SUCCESS;
203 check_result(result, "dns_db_load");
204
205 printf("Connecting to '%s'\n", dbi.filename);
206
207 if ((result = db_connect(&dbi)) != ISC_R_SUCCESS) {
208 fprintf(stderr, "Connection to database '%s' failed\n",
209 dbi.filename);
210 closeandexit(1);
211 }
212
213 sql = sqlite3_mprintf("DROP TABLE %Q ", dbi.table);
214 printf("%s\n", sql);
215 res = sqlite3_exec(dbi.db, sql, NULL, NULL, &errmsg);
216 sqlite3_free(sql);
217 #if 0
218 if (res != SQLITE_OK) {
219 fprintf(stderr, "DROP TABLE %s failed: %s\n",
220 dbi.table, errmsg);
221 }
222 #endif
223
224 #if 0
225 sql = sqlite3_mprintf(sql, "BEGIN TRANSACTION");
226 printf("%s\n", sql);
227 res = sqlite3_exec(dbi.db, sql, NULL, NULL, &errmsg);
228 sqlite3_free(sql);
229 if (res != SQLITE_OK) {
230 fprintf(stderr, "BEGIN TRANSACTION failed: %s\n", errmsg);
231 closeandexit(1);
232 }
233 #endif
234
235 sql = sqlite3_mprintf(
236 "CREATE TABLE %Q "
237 "(NAME TEXT, TTL INTEGER, RDTYPE TEXT, RDATA TEXT) ",
238 dbi.table);
239 printf("%s\n", sql);
240 res = sqlite3_exec(dbi.db, sql, NULL, NULL, &errmsg);
241 sqlite3_free(sql);
242 if (res != SQLITE_OK) {
243 fprintf(stderr, "CREATE TABLE %s failed: %s\n",
244 dbi.table, errmsg);
245 closeandexit(1);
246 }
247
248 dbiter = NULL;
249 result = dns_db_createiterator(db, 0, &dbiter);
250 check_result(result, "dns_db_createiterator()");
251
252 result = dns_dbiterator_first(dbiter);
253 check_result(result, "dns_dbiterator_first");
254
255 dns_fixedname_init(&fname);
256 name = dns_fixedname_name(&fname);
257 dns_rdataset_init(&rdataset);
258 dns_rdata_init(&rdata);
259
260 while (result == ISC_R_SUCCESS) {
261 node = NULL;
262 result = dns_dbiterator_current(dbiter, &node, name);
263 if (result == ISC_R_NOMORE)
264 break;
265 check_result(result, "dns_dbiterator_current");
266
267 rdsiter = NULL;
268 result = dns_db_allrdatasets(db, node, NULL, 0, &rdsiter);
269 check_result(result, "dns_db_allrdatasets");
270
271 result = dns_rdatasetiter_first(rdsiter);
272
273 while (result == ISC_R_SUCCESS) {
274 dns_rdatasetiter_current(rdsiter, &rdataset);
275 result = dns_rdataset_first(&rdataset);
276 check_result(result, "dns_rdataset_first");
277 while (result == ISC_R_SUCCESS) {
278 dns_rdataset_current(&rdataset, &rdata);
279 addrdata(name, rdataset.ttl, &rdata);
280 dns_rdata_reset(&rdata);
281 result = dns_rdataset_next(&rdataset);
282 }
283 dns_rdataset_disassociate(&rdataset);
284 result = dns_rdatasetiter_next(rdsiter);
285 }
286 dns_rdatasetiter_destroy(&rdsiter);
287 dns_db_detachnode(db, &node);
288 result = dns_dbiterator_next(dbiter);
289 }
290
291 #if 0
292 sql = sqlite3_mprintf(sql, "COMMIT TRANSACTION ");
293 printf("%s\n", sql);
294 res = sqlite3_exec(dbi.db, sql, NULL, NULL, &errmsg);
295 sqlite3_free(sql);
296 if (res != SQLITE_OK) {
297 fprintf(stderr, "COMMIT TRANSACTION failed: %s\n", errmsg);
298 closeandexit(1);
299 }
300 #endif
301
302 dns_dbiterator_destroy(&dbiter);
303 dns_db_detach(&db);
304 isc_hash_destroy();
305 isc_entropy_detach(&ectx);
306 isc_mem_destroy(&mctx);
307
308 closeandexit(0);
309
310 exit(0);
311 }
312