git.sesse.net Git - nbtscanner/blob - mysql_interface.c

   1 /*
   2  * nbtscanner -- a tool for scanning large networks for SMB servers.
   3  *
   4  * mysql_interface.c: Interface routines for the MySQL table.
   5  * Copyright (C) 2000 Steinar H. Gunderson
   6  *
   7  * Large amounts of code adapted from Samba (http://www.samba.org/)
   8  * Copyright (C) Andrew Tridgell 1994-1998, and others.
   9  *
  10  * This program is free software; you can redistribute it and/or modify
  11  * it under the terms of the GNU General Public License as published by
  12  * the Free Software Foundation; either version 2 of the License, or
  13  * (at your option) any later version.
  14  *
  15  * This program is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  * GNU General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU General Public License
  21  * along with this program; if not, write to the Free Software
  22  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  23  */
  24
  25 #include <stdio.h>
  26 #include <string.h>
  27 #include <stdlib.h>
  28 #include <mysql.h>
  29
  30 #include "mysql_interface.h"
  31 #include "configfile.h"
  32
  33 MYSQL *mysql = NULL;
  34
  35 /* veeeery detailed error reporting here ;-) */
  36 void init_mysql(char *host, char *username, char *password)
  37 {
  38         if (mysql != NULL) return;      /* hack */
  39
  40         mysql = mysql_init(NULL);
  41         if (mysql == NULL) {
  42                 if (verbosity >= 1) {
  43                         fprintf(stderr, "mysql_init() failed\n");
  44                 }
  45                 exit(1);
  46         }
  47
  48         if (mysql_real_connect(mysql, mysql_host, username, password, "nbtscanner",
  49                 0, NULL, 0) == 0) {
  50                 if (verbosity >= 1) {
  51                         fprintf(stderr, "mysql_real_connect() failed\n");
  52                 }
  53                 exit(1);
  54         }
  55 }
  56
  57 void add_record_mysql(char *ip, char *hostname, char *fileservername, char *groupname)
  58 {
  59         char query[256];
  60         char hname[256], fsname[256], gname[256];
  61         int i;
  62
  63         mysql_real_escape_string(mysql, hname, hostname, strlen(hostname));
  64         mysql_real_escape_string(mysql, fsname, fileservername, strlen(fileservername));
  65         mysql_real_escape_string(mysql, gname, groupname, strlen(groupname));
  66
  67         snprintf(query, 256,
  68                 "REPLACE INTO nbtscanner.nbtscanner SET ip='%s', hostname='%s', fileservername='%s', groupname='%s';",
  69                 ip, hname, fsname, gname);
  70
  71         if (mysql_query(mysql, query) != 0) {
  72                 if (verbosity >= 1) {
  73                         fprintf(stderr, "mysql_query(`%s') failed\n", query);
  74                 }
  75                 exit(1);
  76         }
  77 }
  78
  79 void delete_record_mysql(char *ip)
  80 {
  81         char query[256];
  82         snprintf(query, 256,
  83                 "DELETE FROM nbtscanner.nbtscanner WHERE ip='%s';",
  84                 ip);
  85
  86         if (mysql_query(mysql, query) != 0) {
  87                 if (verbosity >= 1) {
  88                         fprintf(stderr, "mysql_query(`%s') failed\n", query);
  89                 }
  90                 exit(1);
  91         }
  92 }
  93
  94 void print_all_records_mysql()
  95 {
  96         MYSQL_RES *result;
  97         unsigned long *lengths;
  98         unsigned int num_fields;
  99         int i;
 100
 101         if (mysql_query(mysql, "SELECT ip,(CASE WHEN fileservername='-unknown-nbtscanner-' THEN hostname ELSE fileservername END),groupname,(fileservername <> '-unknown-nbtscanner-') FROM nbtscanner.nbtscanner;") != 0) {
 102                 if (verbosity >= 1) {
 103                         fprintf(stderr, "mysql_query('SELECT ip'...) failed\n");
 104                 }
 105                 exit(1);
 106         }
 107
 108         result = mysql_store_result(mysql);
 109         if (result == NULL) {
 110                 if (verbosity >= 1) {
 111                         fprintf(stderr, "mysql_store_result() failed\n");
 112                 }
 113                 exit(1);
 114         }
 115
 116         num_fields = mysql_num_fields(result);
 117         while (1) {
 118                 MYSQL_ROW row = mysql_fetch_row(result);
 119                 if (row == NULL) return;
 120
 121                 lengths = mysql_fetch_lengths(result);
 122
 123                 for (i = 0; i < num_fields; i++) {
 124                         int j;
 125
 126                         /* strip away CR/LFs */
 127                         for (j = 0; j < lengths[i]; j++) {
 128                                 if (row[i][j] == 0 || row[i][j] == 10 || row[i][j] == 13 || row[i][j] == ',') {
 129                                         row[i][j] = '?';
 130                                 }
 131                         }
 132                         printf("%.*s", (int) lengths[i], row[i] ? row[i] : "NULL");
 133                         if (i != num_fields - 1) printf(",");
 134                 }
 135                 printf("\r\n");
 136         }
 137 }
 138
 139 void finish_mysql()
 140 {
 141         /* don't care if this fails */
 142         mysql_query(mysql, "OPTIMIZE TABLE nbtscanner.nbtscanner;");
 143
 144         mysql_close(mysql);
 145 }
 146