modules/sq/mysql_driver.c

/* [<][>]
[^][v][top][bottom][index][help] */

FUNCTIONS

This source file includes following functions.
  1. log_query
  2. SQ_get_connection
  3. SQ_get_connection2
  4. SQ_execute_query
  5. SQ_get_column_count
  6. SQ_get_column_label
  7. SQ_get_column_max_length
  8. SQ_row_next
  9. SQ_get_column_string
  10. SQ_get_column_strings
  11. SQ_get_column_int
  12. SQ_result_to_string
  13. SQ_free_result
  14. SQ_close_connection
  15. SQ_num_rows
  16. SQ_info_to_string
  17. SQ_error
  18. SQ_errno

/***************************************
  $Revision: 1.12 $

  SQL module (sq) - this is a MySQL implementation of the SQL module.

  Status: NOT REVUED, NOT TESTED

  ******************/ /******************
  Filename            : mysql_driver.c
  Author              : ottrey@ripe.net
  OSs Tested          : Solaris
  ******************/ /******************
  Copyright (c) 1999                              RIPE NCC
 
  All Rights Reserved
  
  Permission to use, copy, modify, and distribute this software and its
  documentation for any purpose and without fee is hereby granted,
  provided that the above copyright notice appear in all copies and that
  both that copyright notice and this permission notice appear in
  supporting documentation, and that the name of the author not be
  used in advertising or publicity pertaining to distribution of the
  software without specific, written prior permission.
  
  THE AUTHOR DISCLAIMS ALL WARRANTIES WITH REGARD TO THIS SOFTWARE, INCLUDING
  ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS; IN NO EVENT SHALL
  AUTHOR BE LIABLE FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY
  DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN
  AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  ***************************************/
#include <stdlib.h>
#include <stdio.h>
#include "mysql_driver.h"
#include "constants.h"
#include <sys/timeb.h>

/* XXX NB.  The changes to SQ_row_next() and the removal of the Global variable Current_row */


/*+ String sizes +*/
#define STR_S   63
#define STR_M   255
#define STR_L   1023
#define STR_XL  4095
#define STR_XXL 16383


/* log_query() */
/*++++++++++++++++++++++++++++++++++++++
  Log the query.  This should/will get merged with a tracing module.

  More:
  +html+ <PRE>
  Authors:
        ottrey
  +html+ </PRE><DL COMPACT>
  +html+ <DT>Online References:
  +html+ <DD><UL>
  +html+ </UL></DL>

  ++++++++++++++++++++++++++++++++++++++*/
static void log_query(const char *logfile, const char *query, struct timeb *start, struct timeb *stop) {
/* [<][>][^][v][top][bottom][index][help] */
  FILE *logf;
  int seconds;
  int milliseconds;

  seconds = (int)(stop->time - start->time);
  milliseconds = (int)(stop->millitm - start->millitm);
  if (milliseconds < 0) {
    milliseconds += 1000;
    seconds--;
  }

  if (strcmp(logfile, "stdout") == 0) {
    printf("query=[%s] took %d sec %d msec\n", query, seconds, milliseconds);
  }
  else {
    logf = fopen(logfile, "a");
    fprintf(logf, "query=[%s] took %d sec %d msec\n", query, seconds, milliseconds);
    fclose(logf);
  }

} /* log_query() */

/* SQ_get_connection() */
/*++++++++++++++++++++++++++++++++++++++
  Get a connection to the database.

  const char *host
  
  unsigned int port

  const char *db
  
  const char *user
  
  const char *password
   
  More:
  +html+ <PRE>
  Authors:
        ottrey
  +html+ </PRE><DL COMPACT>
  +html+ <DT>Online References:
  +html+ <DD><UL>
  +html+     <LI><A HREF="http://www.tcx.se/Manual/manual.html#mysql_init">mysql_init()</A>
  +html+     <LI><A HREF="http://www.tcx.se/Manual/manual.html#mysql_real_connect">mysql_real_connect()</A>
  +html+ </UL></DL>

  ++++++++++++++++++++++++++++++++++++++*/
SQ_connection_t *SQ_get_connection(const char *host, unsigned int port, const char *db, const char *user, const char *password) {
/* [<][>][^][v][top][bottom][index][help] */

  SQ_connection_t *sql_connection;

  sql_connection = mysql_init(NULL);
  if (!sql_connection) {
/* Check for errors */
          printf("Connection init error\n");
  }

  sql_connection = mysql_real_connect(sql_connection, host, user, password, db, port, NULL, 0);
  if (!sql_connection) {
/* Check for errors */
          printf("Connection error: Failed to connect to database.... %s\n", db);
    /* XXX Don't be so harsh!
    exit(-1);
    */
  }

  return sql_connection;

} /* SQ_get_connection() */

SQ_connection_t *SQ_get_connection2(void) {
/* [<][>][^][v][top][bottom][index][help] */
  return SQ_get_connection(CO_get_host(),
                           CO_get_database_port(),
                           CO_get_database(),
                           CO_get_user(),
                           CO_get_password()
                          );
} /* SQ_get_connection() */

/* SQ_execute_query() */
/*++++++++++++++++++++++++++++++++++++++
  Execute the sql query.

  SQ_connection_t *sql_connection Connection to database.
  
  const char *query SQL query.
  
  More:
  +html+ <PRE>
  Authors:
        ottrey
  +html+ </PRE><DL COMPACT>
  +html+ <DT>Online References:
  +html+ <DD><UL>
  +html+     <LI><A HREF="http://www.tcx.se/Manual/manual.html#mysql_query">mysql_query()</A>
  +html+     <LI><A HREF="http://www.tcx.se/Manual/manual.html#mysql_use_result">mysql_use_result()</A>
  +html+ </UL></DL>

  ++++++++++++++++++++++++++++++++++++++*/
SQ_result_set_t *SQ_execute_query(SQ_connection_t *sql_connection, const char *query) {
/* [<][>][^][v][top][bottom][index][help] */
  struct timeb *start_time;
  struct timeb *stop_time;
  int err;
  SQ_result_set_t *result;

  if (CO_get_query_logging() == 1) {
    start_time=(struct timeb *)calloc(1, sizeof(struct timeb)+1);
    stop_time=(struct timeb *)calloc(1, sizeof(struct timeb)+1);

    ftime(start_time);
    err = mysql_query(sql_connection, query);
    ftime(stop_time);

    log_query(CO_get_query_logfile(), query, start_time, stop_time);
    
    free(start_time);
    free(stop_time);
  }
  else {
    err = mysql_query(sql_connection, query);
  }

  if (err == 0) {
    result = mysql_use_result(sql_connection);
  }
  else {
    result = NULL;
  }

  return result;

} /* SQ_execute_query() */

/* SQ_get_column_count() */
/*++++++++++++++++++++++++++++++++++++++
  Get the column count.

  SQ_result_set_t *result The results from the query.
  
  More:
  +html+ <PRE>
  Authors:
        ottrey
  +html+ </PRE><DL COMPACT>
  +html+ <DT>Online References:
  +html+ <DD><UL>
  +html+     <LI><A HREF="http://www.tcx.se/Manual/manual.html#mysql_num_fields">mysql_num_fields()</A>
  +html+ </UL></DL>

  ++++++++++++++++++++++++++++++++++++++*/
int SQ_get_column_count(SQ_result_set_t *result) {
/* [<][>][^][v][top][bottom][index][help] */
  int cols;

  cols = mysql_num_fields(result);

  return cols;

} /* SQ_get_column_count() */

/* SQ_get_column_label() */
/*++++++++++++++++++++++++++++++++++++++
  Get the column label.

  SQ_result_set_t *result The results from the query.
  
  unsigned int column The column index.

  More:
  +html+ <PRE>
  Authors:
        ottrey
  +html+ </PRE><DL COMPACT>
  +html+ <DT>Online References:
  +html+ <DD><UL>
  +html+     <LI><A HREF="http://www.tcx.se/Manual/manual.html#mysql_fetch_field_direct">mysql_fetch_field_direct()</A>
  +html+ </UL></DL>

  ++++++++++++++++++++++++++++++++++++++*/
char *SQ_get_column_label(SQ_result_set_t *result, unsigned int column) {
/* [<][>][^][v][top][bottom][index][help] */
  char *str;
  MYSQL_FIELD field;

  field = mysql_fetch_field_direct(result, column);

/*
  printf("column=%d\n", column);
  printf("field.name=%s\n", field.name);
  printf("field.table=%s\n", field.table);
*/
  /*
  printf("field.def=%s\n", field.def);
  */
/*
  printf("field.type=%d\n", field.type);
  printf("field.length=%d\n", field.length);
  printf("field.max_length=%d\n", field.max_length);
  printf("field.flags=%d\n", field.flags);
  printf("field.decimals=%d\n", field.decimals);
*/

  str = (char *)calloc(1, strlen(field.name)+1);
  strcpy(str, field.name);

  return str;

} /* SQ_get_column_label() */

/* SQ_get_column_max_length() */
/*++++++++++++++++++++++++++++++++++++++
  Get the max length of the column.

  SQ_result_set_t *result The results from the query.
  
  unsigned int column The column index.

  More:
  +html+ <PRE>
  Authors:
        ottrey
  +html+ </PRE><DL COMPACT>
  +html+ <DT>Online References:
  +html+ <DD><UL>
  +html+     <LI><A HREF="http://www.tcx.se/Manual/manual.html#mysql_fetch_field_direct">mysql_fetch_field_direct()</A>
  +html+ </UL></DL>

  ++++++++++++++++++++++++++++++++++++++*/
unsigned int SQ_get_column_max_length(SQ_result_set_t *result, unsigned int column) {
/* [<][>][^][v][top][bottom][index][help] */
  MYSQL_FIELD field;

  field = mysql_fetch_field_direct(result, column);

  return field.length;

} /* SQ_get_column_max_length() */

/* SQ_row_next() */
/*++++++++++++++++++++++++++++++++++++++
  Get the next row.

  SQ_result_set_t *result The results from the query.
  
  unsigned int column The column index.

  More:
  +html+ <PRE>
  Authors:
        ottrey
  +html+ </PRE><DL COMPACT>
  +html+ <DT>Online References:
  +html+ <DD><UL>
  +html+     <LI><A HREF="http://www.tcx.se/Manual/manual.html#mysql_fetch_row">mysql_fetch_row()</A>
  +html+ </UL></DL>

  ++++++++++++++++++++++++++++++++++++++*/
SQ_row_t *SQ_row_next(SQ_result_set_t *result) {
/* [<][>][^][v][top][bottom][index][help] */

  return (SQ_row_t *)mysql_fetch_row(result);

} /* SQ_row_next() */

/* SQ_get_column_string() */
/*++++++++++++++++++++++++++++++++++++++
  Get the column string.

  SQ_row_t *current_row The current row (obtained from a SQ_row_next() ).
  
  unsigned int column The column index.

  More:
  +html+ <PRE>
  Authors:
        ottrey
  +html+ </PRE><DL COMPACT>
  +html+ <DT>Online References:
  +html+ <DD><UL>
  +html+ </UL></DL>

  ++++++++++++++++++++++++++++++++++++++*/
char *SQ_get_column_string(SQ_result_set_t *result, SQ_row_t *current_row, unsigned int column) {
/* [<][>][^][v][top][bottom][index][help] */
  char *str=NULL;
  int length = mysql_fetch_lengths(result)[column];


  if (current_row[column] != NULL) {
    str = (char *)malloc(length + 1);
    if (str != NULL) {
      memcpy(str, current_row[column], length );
      str[length] = '\0';
    }
  }

  return str;
  
} /* SQ_get_column_string() */

/* SQ_get_column_strings() */
/*++++++++++++++++++++++++++++++++++++++
  Get the all the strings in one column.

  SQ_result_set_t *result The results.
  
  unsigned int column The column index.

  More:
  +html+ <PRE>
  Authors:
        ottrey
  +html+ </PRE><DL COMPACT>
  +html+ <DT>Online References:
  +html+ <DD><UL>
  +html+ </UL></DL>

  ++++++++++++++++++++++++++++++++++++++*/
char *SQ_get_column_strings(SQ_result_set_t *result, unsigned int column) {
/* [<][>][^][v][top][bottom][index][help] */
  MYSQL_ROW row;
  char str_buffer[STR_XXL];
  char str_buffer_tmp[STR_L];
  char *str;

  int l;

  strcpy(str_buffer, "");

  while ((row = mysql_fetch_row(result)) != NULL) {
    if (row[column] != NULL) {
      sprintf(str_buffer_tmp, "%s\n", row[column]);
    }
    strcat(str_buffer, str_buffer_tmp);

    if (strlen(str_buffer) >= (STR_XXL - STR_XL) ) {
      strcat(str_buffer, "And some more stuff...\n");
      break;
    }
  }

  if (strcmp(str_buffer, "") != 0) {
    str = (char *)calloc(1, strlen(str_buffer)+1);
    strcpy(str, str_buffer);
  }
  else {
    str = NULL;
  }

  return str;

} /* SQ_get_column_strings() */

/* SQ_get_column_int() */
/*++++++++++++++++++++++++++++++++++++++
  Get an integer from the column.

  SQ_result_set_t *result The results.
  
  SQ_row_t *current_row The current row.

  unsigned int column The column index.

  This uses atoi.  So it may be advisable not to use it.

  More:
  +html+ <PRE>
  Authors:
        ottrey
  +html+ </PRE><DL COMPACT>
  +html+ <DT>Online References:
  +html+ <DD><UL>
  +html+ </UL></DL>

  ++++++++++++++++++++++++++++++++++++++*/
int SQ_get_column_int(SQ_result_set_t *result, SQ_row_t *current_row, unsigned int column) {
/* [<][>][^][v][top][bottom][index][help] */
  int ret_val=-1;

  if (*current_row[column] != NULL) {
    ret_val = atoi(*current_row[column]);
  }
  else {
    ;
  }

  return ret_val;
  
} /* SQ_get_column_int() */


/* SQ_result_to_string() */
/*++++++++++++++++++++++++++++++++++++++
  Convert the result set to a string.

  SQ_result_set_t *result The results.
  
  More:
  +html+ <PRE>
  Authors:
        ottrey
  +html+ </PRE><DL COMPACT>
  +html+ <DT>Online References:
  +html+ <DD><UL>
  +html+ </UL></DL>

  ++++++++++++++++++++++++++++++++++++++*/
char *SQ_result_to_string(SQ_result_set_t *result) {
/* [<][>][^][v][top][bottom][index][help] */
  MYSQL_ROW row;
  unsigned int no_cols;
  unsigned int i, j;
  char str_buffer[STR_XXL];
  char str_buffer_tmp[STR_L];
  char border[STR_L];
  char *str;

  char *label;

  unsigned int length[STR_S];
  int l;

  strcpy(str_buffer, "");

  no_cols = mysql_num_fields(result);

  /* Determine the maximum column widths */
  /* XXX Surely MySQL should keep note of this for me! */
  strcpy(border, "");
  for (i=0; i < no_cols; i++) {
    length[i] = SQ_get_column_max_length(result, i);
    /* Make sure the lenghts don't get too long */
    if (length[i] > STR_M) {
      length[i] = STR_M;
    }
    strcat(border, "*");
    for (j=0; (j <= length[i]) && (j < STR_L); j++) {
      strcat(border, "-");
    }
  }
  strcat(border, "*\n");
  /*
  for (i=0; i < no_cols; i++) {
    printf("length[%d]=%d\n", i, length[i]);
  }
  */

  strcat(str_buffer, border);

  for (i=0; i < no_cols; i++) {
    label = SQ_get_column_label(result, i);
    if (label != NULL) {
      sprintf(str_buffer_tmp, "| %-*s", length[i], label);
      strcat(str_buffer, str_buffer_tmp);
    }
  }
  strcat(str_buffer, "|\n");
  
  strcat(str_buffer, border);


  while ((row = mysql_fetch_row(result)) != NULL) {
    for (i=0; i < no_cols; i++) {
      if (row[i] != NULL) {
        sprintf(str_buffer_tmp, "| %-*s", length[i], row[i]);
      }
      else {
        sprintf(str_buffer_tmp, "| %-*s", length[i], "NuLL");
      }
      strcat(str_buffer, str_buffer_tmp);
    }
    strcat(str_buffer, "|\n");

    if (strlen(str_buffer) >= (STR_XXL - STR_XL) ) {
      strcat(str_buffer, "And some more stuff...\n");
      break;
    }
  }

  strcat(str_buffer, border);
  
  str = (char *)calloc(1, strlen(str_buffer)+1);
  strcpy(str, str_buffer);

  return str;

} /* SQ_result_to_string() */

/* SQ_free_result() */
/*++++++++++++++++++++++++++++++++++++++
  Free the result set.

  SQ_result_set_t *result The results.
  
  More:
  +html+ <PRE>
  Authors:
        ottrey
  +html+ </PRE><DL COMPACT>
  +html+ <DT>Online References:
  +html+ <DD><UL>
  +html+     <LI><A HREF="http://www.tcx.se/Manual/manual.html#mysql_free_result">mysql_free_result()</A>
  +html+ </UL></DL>

  ++++++++++++++++++++++++++++++++++++++*/
void SQ_free_result(SQ_result_set_t *result) {
/* [<][>][^][v][top][bottom][index][help] */
  mysql_free_result(result);
} /* SQ_free_result() */


/* SQ_close_connection() */
/*++++++++++++++++++++++++++++++++++++++
  Call this function to close a connection to the server

  SQ_connection_t *sql_connection The connection to the database.
  
  More:
  +html+ <PRE>
  Authors:
        ottrey
  +html+ </PRE><DL COMPACT>
  +html+ <DT>Online References:
  +html+ <DD><UL>
  +html+     <LI><A HREF="http://www.tcx.se/Manual/manual.html#mysql_close">mysql_close()</A>
  +html+ </UL></DL>

  ++++++++++++++++++++++++++++++++++++++*/
void SQ_close_connection(SQ_connection_t *sql_connection) {
/* [<][>][^][v][top][bottom][index][help] */

  mysql_close(sql_connection);

}

/* SQ_num_rows() */
/*++++++++++++++++++++++++++++++++++++++
  Call this function to find out how many rows are in a query result

  SQ_result_set_t *result The results.
  
  More:
  +html+ <PRE>
  Authors:
        ottrey
  +html+ </PRE><DL COMPACT>
  +html+ <DT>Online References:
  +html+ <DD><UL>
  +html+     <LI><A HREF="http://www.tcx.se/Manual/manual.html#mysql_num_rows">mysql_num_rows()</A>
  +html+ </UL></DL>

  ++++++++++++++++++++++++++++++++++++++*/
int SQ_num_rows(SQ_result_set_t *result) {
/* [<][>][^][v][top][bottom][index][help] */

  int rows;

  rows = mysql_num_rows(result);

  return rows;
}

/* SQ_info_to_string() */
/*++++++++++++++++++++++++++++++++++++++
  Convert all available information about the sql server into a string.

  SQ_connection_t *sql_connection The connection to the database.

  More:
  +html+ <PRE>
  Authors:
        ottrey
  +html+ </PRE><DL COMPACT>
  +html+ <DT>Online References:
  +html+ <DD><UL>
  +html+ </UL></DL>

  ++++++++++++++++++++++++++++++++++++++*/
char *SQ_info_to_string(SQ_connection_t *sql_connection) {
/* [<][>][^][v][top][bottom][index][help] */
  char str_buffer[STR_XXL];
  char str_buffer_tmp[STR_L];
  char *str;
  char *str_tmp;

  strcpy(str_buffer, "");

  /* Makes the server dump debug information to the log. */
  sprintf(str_buffer_tmp, "mysql_dump_debug_info()=%d\n", mysql_dump_debug_info(sql_connection));
  strcat(str_buffer, str_buffer_tmp);

  /* Returns the error number from the last MySQL function. */
  sprintf(str_buffer_tmp, "mysql_errno()=%d\n", mysql_errno(sql_connection));
  strcat(str_buffer, str_buffer_tmp);

  /* Returns the error message from the last MySQL function. */
  sprintf(str_buffer_tmp, "mysql_error()=%s\n", mysql_error(sql_connection));
  strcat(str_buffer, str_buffer_tmp);

  /* Returns client version information. */
  sprintf(str_buffer_tmp, "mysql_get_client_info()=%s\n", mysql_get_client_info() );
  strcat(str_buffer, str_buffer_tmp);

  /* Returns a string describing the connection. */
  sprintf(str_buffer_tmp, "mysql_get_host_info()=%s\n", mysql_get_host_info(sql_connection));
  strcat(str_buffer, str_buffer_tmp);

  /* Returns the protocol version used by the connection. */
  sprintf(str_buffer_tmp, "mysql_get_proto_info()=%d\n", mysql_get_proto_info(sql_connection));
  strcat(str_buffer, str_buffer_tmp);

  /* Returns the server version number. */
  sprintf(str_buffer_tmp, "mysql_get_server_info()=%s\n", mysql_get_server_info(sql_connection));
  strcat(str_buffer, str_buffer_tmp);

  /* Information about the most recently executed query. */
  /* XXX Check for NULL */
  str_tmp = mysql_info(sql_connection);
  if (str_tmp != NULL) {
    sprintf(str_buffer_tmp, "mysql_info()=%s\n", str_tmp);
  }
  else {
    sprintf(str_buffer_tmp, "mysql_info()=%s\n", "NulL");
  }
  strcat(str_buffer, str_buffer_tmp);


  /* Returns a list of the current server threads. */
  sprintf(str_buffer_tmp, "mysql_list_processes()=%d\n", mysql_list_processes(sql_connection));
  strcat(str_buffer, str_buffer_tmp);

  /* Checks if the connection to the server is working. */
  sprintf(str_buffer_tmp, "mysql_ping()=%d\n", mysql_ping(sql_connection));
  strcat(str_buffer, str_buffer_tmp);

  /* Returns the server status as a string. */
  sprintf(str_buffer_tmp, "mysql_stat()=%s\n", mysql_stat(sql_connection));
  strcat(str_buffer, str_buffer_tmp);

  /* Returns the current thread id. */
  sprintf(str_buffer_tmp, "mysql_thread_id()=%d\n", mysql_thread_id(sql_connection));
  strcat(str_buffer, str_buffer_tmp);


  str = (char *)calloc(1, strlen(str_buffer)+1);
  strcpy(str, str_buffer);

  return str;

} /* SQ_info_to_string() */

/* SQ_error() */
/*++++++++++++++++++++++++++++++++++++++
  Get the error string for the last error.

  SQ_connection_t *sql_connection The connection to the database.

  More:
  +html+ <PRE>
  Authors:
        ottrey
  +html+ </PRE><DL COMPACT>
  +html+ <DT>Online References:
  +html+ <DD><UL>
  +html+     <LI><A HREF="http://www.tcx.se/Manual/manual.html#mysql_error">mysql_error()</A>
  +html+ </UL></DL>

  ++++++++++++++++++++++++++++++++++++++*/
char *SQ_error(SQ_connection_t *sql_connection) {
/* [<][>][^][v][top][bottom][index][help] */

  return mysql_error(sql_connection);

} /* SQ_error() */

/* SQ_errno() */
/*++++++++++++++++++++++++++++++++++++++
  Get the error number for the last error.

  SQ_connection_t *sql_connection The connection to the database.

  More:
  +html+ <PRE>
  Authors:
        ottrey
  +html+ </PRE><DL COMPACT>
  +html+ <DT>Online References:
  +html+ <DD><UL>
  +html+     <LI><A HREF="http://www.tcx.se/Manual/manual.html#mysql_free_result">mysql_free_result()</A>
  +html+ </UL></DL>

  ++++++++++++++++++++++++++++++++++++++*/
int SQ_errno(SQ_connection_t *sql_connection) {
/* [<][>][^][v][top][bottom][index][help] */

  return mysql_errno(sql_connection);

} /* SQ_errno() */


/* [<][>][^][v][top][bottom][index][help] */