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
  62         mysql_real_escape_string(mysql, hname, hostname, strlen(hostname));
  63         mysql_real_escape_string(mysql, fsname, fileservername, strlen(fileservername));
  64         mysql_real_escape_string(mysql, gname, groupname, strlen(groupname));
  65
  66         snprintf(query, 256,
  67                 "REPLACE INTO nbtscanner.nbtscanner SET ip='%s', hostname='%s', fileservername='%s', groupname='%s';",
  68                 ip, hname, fsname, gname);
  69
  70         if (mysql_query(mysql, query) != 0) {
  71                 if (verbosity >= 1) {
  72                         fprintf(stderr, "mysql_query(`%s') failed\n", query);
  73                 }
  74                 exit(1);
  75         }
  76 }
  77
  78 void delete_record_mysql(char *ip)
  79 {
  80         char query[256];
  81         snprintf(query, 256,
  82                 "DELETE FROM nbtscanner.nbtscanner WHERE ip='%s';",
  83                 ip);
  84
  85         if (mysql_query(mysql, query) != 0) {
  86                 if (verbosity >= 1) {
  87                         fprintf(stderr, "mysql_query(`%s') failed\n", query);
  88                 }
  89                 exit(1);
  90         }
  91 }
  92
  93 void print_all_records_mysql()
  94 {
  95         MYSQL_RES *result;
  96         unsigned long *lengths;
  97         unsigned int num_fields;
  98         int i;
  99
 100         if (mysql_query(mysql, "SELECT ip,(CASE WHEN fileservername='-unknown-nbtscanner-' THEN hostname ELSE fileservername END),groupname,(fileservername <> '-unknown-nbtscanner-') FROM nbtscanner.nbtscanner;") != 0) {
 101                 if (verbosity >= 1) {
 102                         fprintf(stderr, "mysql_query('SELECT ip'...) failed\n");
 103                 }
 104                 exit(1);
 105         }
 106
 107         result = mysql_store_result(mysql);
 108         if (result == NULL) {
 109                 if (verbosity >= 1) {
 110                         fprintf(stderr, "mysql_store_result() failed\n");
 111                 }
 112                 exit(1);
 113         }
 114
 115         num_fields = mysql_num_fields(result);
 116         while (1) {
 117                 MYSQL_ROW row = mysql_fetch_row(result);
 118                 if (row == NULL) return;
 119
 120                 lengths = mysql_fetch_lengths(result);
 121
 122                 for (i = 0; i < num_fields; i++) {
 123                         int j;
 124
 125                         /* strip away CR/LFs */
 126                         for (j = 0; j < lengths[i]; j++) {
 127                                 if (row[i][j] == 0 || row[i][j] == 10 || row[i][j] == 13 || row[i][j] == ',') {
 128                                         row[i][j] = '?';
 129                                 }
 130                         }
 131                         printf("%.*s", (int) lengths[i], row[i] ? row[i] : "NULL");
 132                         if (i != num_fields - 1) printf(",");
 133                 }
 134                 printf("\r\n");
 135         }
 136 }
 137
 138 void finish_mysql()
 139 {
 140         /* don't care if this fails */
 141         mysql_query(mysql, "OPTIMIZE TABLE nbtscanner.nbtscanner;");
 142
 143         mysql_close(mysql);
 144 }
 145