external/bsd/bind/dist/contrib/sdb/sqlite/zone2sqlite.c

   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
  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
  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
  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
  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
 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
 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 }