/** * ircd-ratbox: A slightly useful ircd. * bantool.c: The ircd-ratbox database managment tool. * * Copyright (C) 1990 Jarkko Oikarinen and University of Oulu, Co Center * Copyright (C) 1996-2002 Hybrid Development Team * Copyright (C) 2002-2008 ircd-ratbox development team * Copyright (C) 2008 Daniel J Reidy * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 * USA * * The following server admins have either contributed various configs to test against, * or helped with debugging and feature requests. Many thanks to them. * stevoo / efnet.port80.se * AndroSyn / irc2.choopa.net, irc.igs.ca * Salvation / irc.blessed.net * JamesOff / efnet.demon.co.uk * * Thanks to AndroSyn for challenging me to learn C on the fly :) * BUGS Direct Question, Bug Reports, and Feature Requests to #ratbox on EFnet. * BUGS Complaints >/dev/null * */ #include #include "rsdb.h" using namespace ircd::defaults; #define BT_VERSION "0.4.1" typedef enum { BANDB_KLINE, BANDB_KLINE_PERM, BANDB_DLINE, BANDB_DLINE_PERM, BANDB_XLINE, BANDB_XLINE_PERM, BANDB_RESV, BANDB_RESV_PERM, LAST_BANDB_TYPE } bandb_type; static char bandb_letter[LAST_BANDB_TYPE] = { 'K', 'K', 'D', 'D', 'X', 'X', 'R', 'R' }; static const char *bandb_table[LAST_BANDB_TYPE] = { "kline", "kline", "dline", "dline", "xline", "xline", "resv", "resv" }; static const char *bandb_suffix[LAST_BANDB_TYPE] = { "", ".perm", "", ".perm", "", ".perm", "", ".perm" }; static char me[PATH_MAX]; /* *INDENT-OFF* */ /* report counters */ struct counter { unsigned int klines; unsigned int dlines; unsigned int xlines; unsigned int resvs; unsigned int error; } count = {0, 0, 0, 0, 0}; /* flags set by command line options */ struct flags { bool none; bool export_; bool import; bool verify; bool vacuum; bool pretend; bool verbose; bool wipe; bool dupes_ok; } flag = {true, false, false, false, false, false, false, false, false}; /* *INDENT-ON* */ static int table_has_rows(const char *table); static int table_exists(const char *table); static const char *clean_gecos_field(const char *gecos); static char *bt_smalldate(const char *string); static char *getfield(char *newline); static char *strip_quotes(const char *string); static char *mangle_reason(const char *string); static char *escape_quotes(const char *string); static void db_error_cb(const char *errstr); static void db_reclaim_slack(void); static void export_config(const char *conf, int id); static void import_config(const char *conf, int id); static void check_schema(void); static void print_help(int i_exit) __attribute__((noreturn)); static void wipe_schema(void); static void drop_dupes(const char *user, const char *host, const char *t); /** * swing your pants */ int main(int argc, char *argv[]) { char etc[PATH_MAX]; char conf[PATH_MAX]; int opt; int i; rb_strlcpy(me, argv[0], sizeof(me)); while((opt = getopt(argc, argv, "hieuspvwd")) != -1) { switch (opt) { case 'h': print_help(EXIT_SUCCESS); break; case 'i': flag.none = false; flag.import = true; break; case 'e': flag.none = false; flag.export_ = true; break; case 'u': flag.none = false; flag.verify = true; break; case 's': flag.none = false; flag.vacuum = true; break; case 'p': flag.pretend = true; break; case 'v': flag.verbose = true; break; case 'w': flag.wipe = true; break; case 'd': flag.dupes_ok = true; break; default: /* '?' */ print_help(EXIT_FAILURE); } } /* they should really read the help. */ if(flag.none) print_help(EXIT_FAILURE); if((flag.import && flag.export_) || (flag.export_ && flag.wipe) || (flag.verify && flag.pretend) || (flag.export_ && flag.pretend)) { fprintf(stderr, "* Error: Conflicting flags.\n"); if(flag.export_ && flag.pretend) fprintf(stderr, "* There is nothing to 'pretend' when exporting.\n"); fprintf(stderr, "* For an explination of commands, run: %s -h\n", me); exit(EXIT_FAILURE); } if(argv[optind] != NULL) rb_strlcpy(etc, argv[optind], sizeof(etc)); else rb_strlcpy(etc, ircd::fs::ETCPATH, sizeof(ircd::fs::ETCPATH)); fprintf(stdout, "* charybdis bantool v.%s\n", BT_VERSION); if(flag.pretend == false) { if(rsdb_init(db_error_cb) == -1) { fprintf(stderr, "* Error: Unable to open database\n"); exit(EXIT_FAILURE); } check_schema(); if(flag.vacuum) db_reclaim_slack(); if(flag.import && flag.wipe) { flag.dupes_ok = true; /* dont check for dupes if we are wiping the db clean */ for(i = 0; i < 3; i++) fprintf(stdout, "* WARNING: YOU ARE ABOUT TO WIPE YOUR DATABASE!\n"); fprintf(stdout, "* Press ^C to abort! "); fflush(stdout); rb_sleep(10, 0); fprintf(stdout, "Carrying on...\n"); wipe_schema(); } } if(flag.verbose && flag.dupes_ok == true) fprintf(stdout, "* Allowing duplicate bans...\n"); /* checking for our files to import or export */ for(i = 0; i < LAST_BANDB_TYPE; i++) { snprintf(conf, sizeof(conf), "%s/%s.conf%s", etc, bandb_table[i], bandb_suffix[i]); if(flag.import && flag.pretend == false) rsdb_transaction(RSDB_TRANS_START); if(flag.import) import_config(conf, i); if(flag.export_) export_config(conf, i); if(flag.import && flag.pretend == false) rsdb_transaction(RSDB_TRANS_END); } if(flag.import) { if(count.error && flag.verbose) fprintf(stderr, "* I was unable to locate %u config files to import.\n", count.error); fprintf(stdout, "* Import Stats: Klines: %u, Dlines: %u, Xlines: %u, Resvs: %u \n", count.klines, count.dlines, count.xlines, count.resvs); fprintf(stdout, "*\n* If your IRC server is currently running, newly imported bans \n* will not take effect until you issue the command: /quote rehash bans\n"); if(flag.pretend) fprintf(stdout, "* Pretend mode engaged. Nothing was actually entered into the database.\n"); } return 0; } /** * export the database to old-style flat files */ static void export_config(const char *conf, int id) { struct rsdb_table table; static char sql[BUFSIZE * 2]; static char buf[512]; FILE *fd = NULL; int j; /* for sanity sake */ const int mask1 = 0; const int mask2 = 1; const int reason = 2; const int oper = 3; const int ts = 4; /* const int perm = 5; */ if(!table_has_rows(bandb_table[id])) return; if(strstr(conf, ".perm") != 0) snprintf(sql, sizeof(sql), "SELECT DISTINCT mask1,mask2,reason,oper,time FROM %s WHERE perm = 1 ORDER BY time", bandb_table[id]); else snprintf(sql, sizeof(sql), "SELECT DISTINCT mask1,mask2,reason,oper,time FROM %s WHERE perm = 0 ORDER BY time", bandb_table[id]); rsdb_exec_fetch(&table, sql); if(table.row_count <= 0) { rsdb_exec_fetch_end(&table); return; } if(flag.verbose) fprintf(stdout, "* checking for %s: ", conf); /* debug */ /* open config for reading, or skip to the next */ if(!(fd = fopen(conf, "w"))) { if(flag.verbose) fprintf(stdout, "\tmissing.\n"); count.error++; return; } for(j = 0; j < table.row_count; j++) { switch (id) { case BANDB_DLINE: case BANDB_DLINE_PERM: snprintf(buf, sizeof(buf), "\"%s\",\"%s\",\"\",\"%s\",\"%s\",%s\n", table.row[j][mask1], mangle_reason(table.row[j][reason]), bt_smalldate(table.row[j][ts]), table.row[j][oper], table.row[j][ts]); break; case BANDB_XLINE: case BANDB_XLINE_PERM: snprintf(buf, sizeof(buf), "\"%s\",\"0\",\"%s\",\"%s\",%s\n", escape_quotes(table.row[j][mask1]), mangle_reason(table.row[j][reason]), table.row[j][oper], table.row[j][ts]); break; case BANDB_RESV: case BANDB_RESV_PERM: snprintf(buf, sizeof(buf), "\"%s\",\"%s\",\"%s\",%s\n", table.row[j][mask1], mangle_reason(table.row[j][reason]), table.row[j][oper], table.row[j][ts]); break; default: /* Klines */ snprintf(buf, sizeof(buf), "\"%s\",\"%s\",\"%s\",\"\",\"%s\",\"%s\",%s\n", table.row[j][mask1], table.row[j][mask2], mangle_reason(table.row[j][reason]), bt_smalldate(table.row[j][ts]), table.row[j][oper], table.row[j][ts]); break; } fprintf(fd, "%s", buf); } rsdb_exec_fetch_end(&table); if(flag.verbose) fprintf(stdout, "\twritten.\n"); fclose(fd); } /** * attempt to condense the individual conf functions into one */ static void import_config(const char *conf, int id) { FILE *fd; char line[BUFSIZE]; char *p; int i = 0; char f_perm = 0; const char *f_mask1 = NULL; const char *f_mask2 = NULL; const char *f_oper = NULL; const char *f_time = NULL; const char *f_reason = NULL; const char *f_oreason = NULL; char newreason[REASONLEN]; if(flag.verbose) fprintf(stdout, "* checking for %s: ", conf); /* debug */ /* open config for reading, or skip to the next */ if(!(fd = fopen(conf, "r"))) { if(flag.verbose) fprintf(stdout, "%*s", strlen(bandb_suffix[id]) > 0 ? 10 : 15, "missing.\n"); count.error++; return; } if(strstr(conf, ".perm") != 0) f_perm = 1; /* xline * "SYSTEM","0","banned","stevoo!stevoo@efnet.port80.se{stevoo}",1111080437 * resv * "OseK","banned nickname","stevoo!stevoo@efnet.port80.se{stevoo}",1111031619 * dline * "194.158.192.0/19","laptop scammers","","2005/3/17 05.33","stevoo!stevoo@efnet.port80.se{stevoo}",1111033988 */ while(fgets(line, sizeof(line), fd)) { if((p = strpbrk(line, "\r\n")) != NULL) *p = '\0'; if((*line == '\0') || (*line == '#')) continue; /* mask1 */ f_mask1 = getfield(line); if(EmptyString(f_mask1)) continue; /* mask2 */ switch (id) { case BANDB_XLINE: case BANDB_XLINE_PERM: f_mask1 = escape_quotes(clean_gecos_field(f_mask1)); getfield(NULL); /* empty field */ break; case BANDB_RESV: case BANDB_RESV_PERM: case BANDB_DLINE: case BANDB_DLINE_PERM: break; default: f_mask2 = getfield(NULL); if(EmptyString(f_mask2)) continue; break; } /* reason */ f_reason = getfield(NULL); if(EmptyString(f_reason)) continue; /* oper comment */ switch (id) { case BANDB_KLINE: case BANDB_KLINE_PERM: case BANDB_DLINE: case BANDB_DLINE_PERM: f_oreason = getfield(NULL); getfield(NULL); break; default: break; } f_oper = getfield(NULL); f_time = strip_quotes(f_oper + strlen(f_oper) + 2); if(EmptyString(f_oper)) f_oper = "unknown"; /* meh */ if(id == BANDB_KLINE || id == BANDB_KLINE_PERM) { if(strstr(f_mask1, "!") != NULL) { fprintf(stderr, "* SKIPPING INVALID KLINE %s@%s set by %s\n", f_mask1, f_mask2, f_oper); fprintf(stderr, " You may wish to re-apply it correctly.\n"); continue; } } /* append operreason_field to reason_field */ if(!EmptyString(f_oreason)) snprintf(newreason, sizeof(newreason), "%s | %s", f_reason, f_oreason); else snprintf(newreason, sizeof(newreason), "%s", f_reason); if(flag.pretend == false) { if(flag.dupes_ok == false) drop_dupes(f_mask1, f_mask2, bandb_table[id]); rsdb_exec(NULL, "INSERT INTO %s (mask1, mask2, oper, time, perm, reason) VALUES('%Q','%Q','%Q','%Q','%d','%Q')", bandb_table[id], f_mask1, f_mask2, f_oper, f_time, f_perm, newreason); } if(flag.pretend && flag.verbose) fprintf(stdout, "%s: perm(%d) mask1(%s) mask2(%s) oper(%s) reason(%s) time(%s)\n", bandb_table[id], f_perm, f_mask1, f_mask2, f_oper, newreason, f_time); i++; } switch (bandb_letter[id]) { case 'K': count.klines += i; break; case 'D': count.dlines += i; break; case 'X': count.xlines += i; break; case 'R': count.resvs += i; break; default: break; } if(flag.verbose) fprintf(stdout, "%*s\n", strlen(bandb_suffix[id]) > 0 ? 10 : 15, "imported."); fclose(fd); return; } /** * getfield * * inputs - input buffer * output - next field * side effects - field breakup for ircd.conf file. */ char * getfield(char *newline) { static char *line = NULL; char *end, *field; if(newline != NULL) line = newline; if(line == NULL) return (NULL); field = line; /* XXX make this skip to first " if present */ if(*field == '"') field++; else return (NULL); /* mal-formed field */ end = strchr(line, ','); while(1) { /* no trailing , - last field */ if(end == NULL) { end = line + strlen(line); line = NULL; if(*end == '"') { *end = '\0'; return field; } else return NULL; } else { /* look for a ", to mark the end of a field.. */ if(*(end - 1) == '"') { line = end + 1; end--; *end = '\0'; return field; } /* search for the next ',' */ end++; end = strchr(end, ','); } } return NULL; } /** * strip away "quotes" from around strings */ static char * strip_quotes(const char *string) { static char buf[14]; /* int(11) + 2 + \0 */ char *str = buf; if(string == NULL) return NULL; while(*string) { if(*string != '"') { *str++ = *string; } string++; } *str = '\0'; return buf; } /** * escape quotes in a string */ static char * escape_quotes(const char *string) { static char buf[BUFSIZE * 2]; char *str = buf; if(string == NULL) return NULL; while(*string) { if(*string == '"') { *str++ = '\\'; *str++ = '"'; } else { *str++ = *string; } string++; } *str = '\0'; return buf; } static char * mangle_reason(const char *string) { static char buf[BUFSIZE * 2]; char *str = buf; if(string == NULL) return NULL; while(*string) { switch (*string) { case '"': *str = '\''; break; case ':': *str = ' '; break; default: *str = *string; } string++; str++; } *str = '\0'; return buf; } /** * change spaces to \s in gecos field */ static const char * clean_gecos_field(const char *gecos) { static char buf[BUFSIZE * 2]; char *str = buf; if(gecos == NULL) return NULL; while(*gecos) { if(*gecos == ' ') { *str++ = '\\'; *str++ = 's'; } else *str++ = *gecos; gecos++; } *str = '\0'; return buf; } /** * verify the database integrity, and if necessary create apropriate tables */ static void check_schema(void) { int i, j; char type[8]; /* longest string is 'INTEGER\0' */ if(flag.verify || flag.verbose) fprintf(stdout, "* Verifying database.\n"); const char *columns[] = { "perm", "mask1", "mask2", "oper", "time", "reason", NULL }; for(i = 0; i < LAST_BANDB_TYPE; i++) { if(!table_exists(bandb_table[i])) { rsdb_exec(NULL, "CREATE TABLE %s (mask1 TEXT, mask2 TEXT, oper TEXT, time INTEGER, perm INTEGER, reason TEXT)", bandb_table[i]); } /* * i can't think of any better way to do this, other then attempt to * force the creation of column that may, or may not already exist. --dubkat */ else { for(j = 0; columns[j] != NULL; j++) { if(!strcmp(columns[j], "time") && !strcmp(columns[j], "perm")) rb_strlcpy(type, "INTEGER", sizeof(type)); else rb_strlcpy(type, "TEXT", sizeof(type)); /* attempt to add a column with extreme prejudice, errors are ignored */ rsdb_exec(NULL, "ALTER TABLE %s ADD COLUMN %s %s", bandb_table[i], columns[j], type); } } i++; /* skip over .perm */ } } static void db_reclaim_slack(void) { fprintf(stdout, "* Reclaiming free space.\n"); rsdb_exec(NULL, "VACUUM"); } /** * check that appropriate tables exist. */ static int table_exists(const char *dbtab) { struct rsdb_table table; rsdb_exec_fetch(&table, "SELECT name FROM sqlite_master WHERE type='table' AND name='%s'", dbtab); rsdb_exec_fetch_end(&table); return table.row_count; } /** * check that there are actual entries in a table */ static int table_has_rows(const char *dbtab) { struct rsdb_table table; rsdb_exec_fetch(&table, "SELECT * FROM %s", dbtab); rsdb_exec_fetch_end(&table); return table.row_count; } /** * completly wipes out an existing ban.db of all entries. */ static void wipe_schema(void) { int i; rsdb_transaction(RSDB_TRANS_START); for(i = 0; i < LAST_BANDB_TYPE; i++) { rsdb_exec(NULL, "DROP TABLE %s", bandb_table[i]); i++; /* double increment to skip over .perm */ } rsdb_transaction(RSDB_TRANS_END); check_schema(); } /** * remove pre-existing duplicate bans from the database. * we favor the new, imported ban over the one in the database */ void drop_dupes(const char *user, const char *host, const char *t) { rsdb_exec(NULL, "DELETE FROM %s WHERE mask1='%Q' AND mask2='%Q'", t, user, host); } static void db_error_cb(const char *errstr) { return; } /** * convert unix timestamp to human readable (small) date */ static char * bt_smalldate(const char *string) { static char buf[MAX_DATE_STRING]; struct tm *lt; time_t t; t = strtol(string, NULL, 10); lt = gmtime(&t); if(lt == NULL) return NULL; snprintf(buf, sizeof(buf), "%d/%d/%d %02d.%02d", lt->tm_year + 1900, lt->tm_mon + 1, lt->tm_mday, lt->tm_hour, lt->tm_min); return buf; } /** * you are here ->. */ void print_help(int i_exit) { fprintf(stderr, "bantool v.%s - the charybdis database tool.\n", BT_VERSION); fprintf(stderr, "Copyright (C) 2008 Daniel J Reidy \n"); fprintf(stderr, "This program is distributed in the hope that it will be useful,\n" "but WITHOUT ANY WARRANTY; without even the implied warranty of\n" "MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the\n" "GNU General Public License for more details.\n\n"); fprintf(stderr, "Usage: %s <-i|-e> [-p] [-v] [-h] [-d] [-w] [path]\n", me); fprintf(stderr, " -h : Display some slightly useful help.\n"); fprintf(stderr, " -i : Actually import configs into your database.\n"); fprintf(stderr, " -e : Export your database to old-style flat files.\n"); fprintf(stderr, " This is suitable for redistrubuting your banlists, or creating backups.\n"); fprintf(stderr, " -s : Reclaim empty slack space the database may be taking up.\n"); fprintf(stderr, " -u : Update the database tables to support any new features.\n"); fprintf(stderr, " This is automaticlly done if you are importing or exporting\n"); fprintf(stderr, " but should be run whenever you upgrade the ircd.\n"); fprintf(stderr, " -p : pretend, checks for the configs, and parses them, then tells you some data...\n"); fprintf(stderr, " but does not touch your database.\n"); fprintf(stderr, " -v : Be verbose... and it *is* very verbose! (intended for debugging)\n"); fprintf(stderr, " -d : Enable checking for redunant entries.\n"); fprintf(stderr, " -w : Completly wipe your database clean. May be used with -i \n"); fprintf(stderr, " path : An optional directory containing old ratbox configs for import, or export.\n"); fprintf(stderr, " If not specified, it looks in PREFIX/etc.\n"); exit(i_exit); }