~drizzle-trunk/drizzle/development

/* Copyright (C) 2004 MySQL AB

   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; version 2 of the License.

   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 */

/*

  Written by Anjuta Widenius

*/

/*

  Creates one include file and multiple language-error message files from one

  multi-language text file.

*/

#include <my_global.h>

#include <m_ctype.h>

#include <my_sys.h>

#include <m_string.h>

#include <my_getopt.h>

#include <assert.h>

#include <my_dir.h>

#define MAX_ROWS  1000

#define HEADER_LENGTH 32                /* Length of header in errmsg.sys */

#define DEFAULT_CHARSET_DIR "../sql/share/charsets"

#define ER_PREFIX "ER_"

#define WARN_PREFIX "WARN_"

static char *OUTFILE= (char*) "errmsg.sys";

static char *HEADERFILE= (char*) "mysqld_error.h";

static char *NAMEFILE= (char*) "mysqld_ername.h";

static char *STATEFILE= (char*) "sql_state.h";

static char *TXTFILE= (char*) "../sql/share/errmsg.txt";

static char *DATADIRECTORY= (char*) "../sql/share/";

/* Header for errmsg.sys files */

uchar file_head[]= { 254, 254, 2, 1 };

/* Store positions to each error message row to store in errmsg.sys header */

uint file_pos[MAX_ROWS];

const char *empty_string= "";			/* For empty states */

/*

  Default values for command line options. See getopt structure for definitions

  for these.

*/

const char *default_language= "eng";

int er_offset= 1000;

my_bool info_flag= 0;

/* Storage of one error message row (for one language) */

struct message

{

  char *lang_short_name;

  char *text;

};

/* Storage for languages and charsets (from start of error text file) */

struct languages

{

  char *lang_long_name;				/* full name of the language */

  char *lang_short_name;			/* abbreviation of the lang. */

  char *charset;				/* Character set name */

  struct languages *next_lang;			/* Pointer to next language */

};

/* Name, code and  texts (for all lang) for one error message */

struct errors

{

  const char *er_name;			/* Name of the error (ER_HASHCK) */

  int d_code;                           /* Error code number */

  const char *sql_code1;		/* sql state */

  const char *sql_code2;		/* ODBC state */

  struct errors *next_error;            /* Pointer to next error */

  DYNAMIC_ARRAY msg;                    /* All language texts for this error */

};

static struct my_option my_long_options[]=

{

  {"debug-info", 'T', "Print some debug info at exit.", (char**) & info_flag,

   (char**) & info_flag, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},

  {"help", '?', "Displays this help and exits.", 0, 0, 0, GET_NO_ARG,

   NO_ARG, 0, 0, 0, 0, 0, 0},

  {"version", 'V', "Prints version", 0, 0, 0, GET_NO_ARG,

   NO_ARG, 0, 0, 0, 0, 0, 0},

  {"charset", 'C', "Charset dir", (char**) & charsets_dir,

   (char**) & charsets_dir,

   0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},

  {"in_file", 'F', "Input file", (char**) & TXTFILE, (char**) & TXTFILE,

   0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},

  {"out_dir", 'D', "Output base directory", (char**) & DATADIRECTORY,

   (char**) & DATADIRECTORY,

   0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},

  {"out_file", 'O', "Output filename (errmsg.sys)", (char**) & OUTFILE,

   (char**) & OUTFILE, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},

  {"header_file", 'H', "mysqld_error.h file ", (char**) & HEADERFILE,

   (char**) & HEADERFILE, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},

  {"name_file", 'N', "mysqld_ername.h file ", (char**) & NAMEFILE,

   (char**) & NAMEFILE, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},

  {"state_file", 'S', "sql_state.h file", (char**) & STATEFILE,

   (char**) & STATEFILE, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},

  {0, 0, 0, 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0}

};

static struct languages *parse_charset_string(char *str);

static struct errors *parse_error_string(char *ptr, int er_count);

static struct message *parse_message_string(struct message *new_message,

					    char *str);

static struct message *find_message(struct errors *err, const char *lang,

                                    my_bool no_default);

static int check_message_format(struct errors *err,

                                const char* mess);

static int parse_input_file(const char *file_name, struct errors **top_error,

			    struct languages **top_language);

static int get_options(int *argc, char ***argv);

static void print_version(void);

static void usage(void);

static bool get_one_option(int optid, const struct my_option *opt,

			      char *argument);

static char *parse_text_line(char *pos);

static int copy_rows(FILE * to, char *row, int row_nr, long start_pos);

static char *parse_default_language(char *str);

static uint parse_error_offset(char *str);

static char *skip_delimiters(char *str);

static char *get_word(char **str);

static char *find_end_of_word(char *str);

static void clean_up(struct languages *lang_head, struct errors *error_head);

static int create_header_files(struct errors *error_head);

static int create_sys_files(struct languages *lang_head,

			    struct errors *error_head, uint row_count);

int main(int argc, char *argv[])

{

  MY_INIT(argv[0]);

  {

    uint row_count;

    struct errors *error_head;

    struct languages *lang_head;

    charsets_dir= DEFAULT_CHARSET_DIR;

    my_umask_dir= 0777;

    if (get_options(&argc, &argv))

      return(1);

    if (!(row_count= parse_input_file(TXTFILE, &error_head, &lang_head)))

    {

      fprintf(stderr, "Failed to parse input file %s\n", TXTFILE);

      return(1);

    }

    if (lang_head == NULL || error_head == NULL)

    {

      fprintf(stderr, "Failed to parse input file %s\n", TXTFILE);

      return(1);

    }

    if (create_header_files(error_head))

    {

      fprintf(stderr, "Failed to create header files\n");

      return(1);

    }

    if (create_sys_files(lang_head, error_head, row_count))

    {

      fprintf(stderr, "Failed to create sys files\n");

      return(1);

    }

    clean_up(lang_head, error_head);

    my_end(info_flag ? MY_CHECK_ERROR | MY_GIVE_INFO : 0);

    return 0;

  }

}

static int create_header_files(struct errors *error_head)

{

  uint er_last;

  FILE *er_definef, *sql_statef, *er_namef;

  struct errors *tmp_error;

  if (!(er_definef= my_fopen(HEADERFILE, O_WRONLY, MYF(MY_WME))))

  {

    return(1);

  }

  if (!(sql_statef= my_fopen(STATEFILE, O_WRONLY, MYF(MY_WME))))

  {

    my_fclose(er_definef, MYF(0));

    return(1);

  }

  if (!(er_namef= my_fopen(NAMEFILE, O_WRONLY, MYF(MY_WME))))

  {

    my_fclose(er_definef, MYF(0));

    my_fclose(sql_statef, MYF(0));

    return(1);

  }

  fprintf(er_definef, "/* Autogenerated file, please don't edit */\n\n");

  fprintf(sql_statef, "/* Autogenerated file, please don't edit */\n\n");

  fprintf(er_namef, "/* Autogenerated file, please don't edit */\n\n");

  fprintf(er_definef, "#define ER_ERROR_FIRST %d\n", error_head->d_code);

  for (tmp_error= error_head; tmp_error; tmp_error= tmp_error->next_error)

  {

    /*

       generating mysqld_error.h

       fprintf() will automatically add \r on windows

    */

    fprintf(er_definef, "#define %s %d\n", tmp_error->er_name,

	    tmp_error->d_code);

    er_last= tmp_error->d_code;

    /* generating sql_state.h file */

    if (tmp_error->sql_code1[0] || tmp_error->sql_code2[0])

      fprintf(sql_statef,

	      "{ %-40s,\"%s\", \"%s\" },\n", tmp_error->er_name,

	      tmp_error->sql_code1, tmp_error->sql_code2);

    /*generating er_name file */

    fprintf(er_namef, "{ \"%s\", %d },\n", tmp_error->er_name,

	    tmp_error->d_code);

  }

  /* finishing off with mysqld_error.h */

  fprintf(er_definef, "#define ER_ERROR_LAST %d\n", er_last);

  my_fclose(er_definef, MYF(0));

  my_fclose(sql_statef, MYF(0));

  my_fclose(er_namef, MYF(0));

  return(0);

}

static int create_sys_files(struct languages *lang_head,

			    struct errors *error_head, uint row_count)

{

  FILE *to;

  uint csnum= 0, length, i, row_nr;

  uchar head[32];

  char outfile[FN_REFLEN], *outfile_end;

  long start_pos;

  struct message *tmp;

  struct languages *tmp_lang;

  struct errors *tmp_error;

  struct stat stat_info;

  /*

     going over all languages and assembling corresponding error messages

  */

  for (tmp_lang= lang_head; tmp_lang; tmp_lang= tmp_lang->next_lang)

  {

    /* setting charset name */

    if (!(csnum= get_charset_number(tmp_lang->charset, MY_CS_PRIMARY)))

    {

      fprintf(stderr, "Unknown charset '%s' in '%s'\n", tmp_lang->charset,

	      TXTFILE);

      return(1);

    }

    outfile_end= strxmov(outfile, DATADIRECTORY, 

                         tmp_lang->lang_long_name, NullS);

    if (stat(outfile, &stat_info))

    {

      if (my_mkdir(outfile, 0777,MYF(0)) < 0)

      {

        fprintf(stderr, "Can't create output directory for %s\n", 

                outfile);

        return(1);

      }

    }

    strxmov(outfile_end, FN_ROOTDIR, OUTFILE, NullS);

    if (!(to= my_fopen(outfile, O_WRONLY | FILE_BINARY, MYF(MY_WME))))

      return(1);

    /* 2 is for 2 bytes to store row position / error message */

    start_pos= (long) (HEADER_LENGTH + row_count * 2);

    fseek(to, start_pos, 0);

    row_nr= 0;

    for (tmp_error= error_head; tmp_error; tmp_error= tmp_error->next_error)

    {

      /* dealing with messages */

      tmp= find_message(tmp_error, tmp_lang->lang_short_name, false);

      if (!tmp)

      {

	fprintf(stderr,

		"Did not find message for %s neither in %s nor in default "

		"language\n", tmp_error->er_name, tmp_lang->lang_short_name);

	goto err;

      }

      if (copy_rows(to, tmp->text, row_nr, start_pos))

      {

	fprintf(stderr, "Failed to copy rows to %s\n", outfile);

	goto err;

      }

      row_nr++;

    }

    /* continue with header of the errmsg.sys file */

    length= ftell(to) - HEADER_LENGTH - row_count * 2;

    bzero((uchar*) head, HEADER_LENGTH);

    bmove((uchar *) head, (uchar *) file_head, 4);

    head[4]= 1;

    int2store(head + 6, length);

    int2store(head + 8, row_count);

    head[30]= csnum;

    my_fseek(to, 0l, MY_SEEK_SET, MYF(0));

    if (my_fwrite(to, (uchar*) head, HEADER_LENGTH, MYF(MY_WME | MY_FNABP)))

      goto err;

    for (i= 0; i < row_count; i++)

    {

      int2store(head, file_pos[i]);

      if (my_fwrite(to, (uchar*) head, 2, MYF(MY_WME | MY_FNABP)))

	goto err;

    }

    my_fclose(to, MYF(0));

  }

  return(0);

err:

  my_fclose(to, MYF(0));

  return(1);

}

static void clean_up(struct languages *lang_head, struct errors *error_head)

{

  struct languages *tmp_lang, *next_language;

  struct errors *tmp_error, *next_error;

  uint count, i;

  my_free((uchar*) default_language, MYF(0));

  for (tmp_lang= lang_head; tmp_lang; tmp_lang= next_language)

  {

    next_language= tmp_lang->next_lang;

    my_free(tmp_lang->lang_short_name, MYF(0));

    my_free(tmp_lang->lang_long_name, MYF(0));

    my_free(tmp_lang->charset, MYF(0));

    my_free((uchar*) tmp_lang, MYF(0));

  }

  for (tmp_error= error_head; tmp_error; tmp_error= next_error)

  {

    next_error= tmp_error->next_error;

    count= (tmp_error->msg).elements;

    for (i= 0; i < count; i++)

    {

      struct message *tmp;

      tmp= dynamic_element(&tmp_error->msg, i, struct message*);

      my_free((uchar*) tmp->lang_short_name, MYF(0));

      my_free((uchar*) tmp->text, MYF(0));

    }

    delete_dynamic(&tmp_error->msg);

    if (tmp_error->sql_code1[0])

      my_free((uchar*) tmp_error->sql_code1, MYF(0));

    if (tmp_error->sql_code2[0])

      my_free((uchar*) tmp_error->sql_code2, MYF(0));

    my_free((uchar*) tmp_error->er_name, MYF(0));

    my_free((uchar*) tmp_error, MYF(0));

  }

}

static int parse_input_file(const char *file_name, struct errors **top_error,

			    struct languages **top_lang)

{

  FILE *file;

  char *str, buff[1000];

  struct errors *current_error= 0, **tail_error= top_error;

  struct message current_message;

  int rcount= 0;

  *top_error= 0;

  *top_lang= 0;

  if (!(file= my_fopen(file_name, O_RDONLY | O_SHARE, MYF(MY_WME))))

    return(0);

  while ((str= fgets(buff, sizeof(buff), file)))

  {

    if (is_prefix(str, "language"))

    {

      if (!(*top_lang= parse_charset_string(str)))

      {

	fprintf(stderr, "Failed to parse the charset string!\n");

	return(0);

      }

      continue;

    }

    if (is_prefix(str, "start-error-number"))

    {

      if (!(er_offset= parse_error_offset(str)))

      {

	fprintf(stderr, "Failed to parse the error offset string!\n");

	return(0);

      }

      continue;

    }

    if (is_prefix(str, "default-language"))

    {

      if (!(default_language= parse_default_language(str)))

      {

	fprintf(stderr,

		"Failed to parse the default language line. Aborting\n");

	return(0);

      }

      continue;

    }

    if (*str == '\t' || *str == ' ')

    {

      /* New error message in another language for previous error */

      if (!current_error)

      {

	fprintf(stderr, "Error in the input file format\n");

	return(0);

      }

      if (!parse_message_string(&current_message, str))

      {

	fprintf(stderr, "Failed to parse message string for error '%s'",

		current_error->er_name);

	return(0);

      }

      if (find_message(current_error, current_message.lang_short_name, true))

      {

	fprintf(stderr, "Duplicate message string for error '%s'"

                        " in language '%s'\n",

		current_error->er_name, current_message.lang_short_name);

	return(0);

      }

      if (check_message_format(current_error, current_message.text))

      {

	fprintf(stderr, "Wrong formatspecifier of error message string"

                        " for error '%s' in language '%s'\n",

		current_error->er_name, current_message.lang_short_name);

	return(0);

      }

      if (insert_dynamic(&current_error->msg, (uchar *) & current_message))

	return(0);

      continue;

    }

    if (is_prefix(str, ER_PREFIX) || is_prefix(str, WARN_PREFIX))

    {

      if (!(current_error= parse_error_string(str, rcount)))

      {

	fprintf(stderr, "Failed to parse the error name string\n");

	return(0);

      }

      rcount++;                         /* Count number of unique errors */

      /* add error to the list */

      *tail_error= current_error;

      tail_error= &current_error->next_error;

      continue;

    }

    if (*str == '#' || *str == '\n')

      continue;					/* skip comment or empty lines */

    fprintf(stderr, "Wrong input file format. Stop!\nLine: %s\n", str);

    return(0);

  }

  *tail_error= 0;				/* Mark end of list */

  my_fclose(file, MYF(0));

  return(rcount);

}

static uint parse_error_offset(char *str)

{

  char *soffset, *end;

  int error;

  uint ioffset;

  /* skipping the "start-error-number" keyword and spaces after it */

  str= find_end_of_word(str);

  str= skip_delimiters(str);

  if (!*str)

    return(0);     /* Unexpected EOL: No error number after the keyword */

  /* reading the error offset */

  if (!(soffset= get_word(&str)))

    return(0);				/* OOM: Fatal error */

  /* skipping space(s) and/or tabs after the error offset */

  str= skip_delimiters(str);

  if (*str)

  {

    /* The line does not end with the error offset -> error! */

    fprintf(stderr, "The error offset line does not end with an error offset");

    return(0);

  }

  end= 0;

  ioffset= (uint) my_strtoll10(soffset, &end, &error);

  my_free((uchar*) soffset, MYF(0));

  return(ioffset);

}

/* Parsing of the default language line. e.g. "default-language eng" */

static char *parse_default_language(char *str)

{

  char *slang;

  /* skipping the "default-language" keyword */

  str= find_end_of_word(str);

  /* skipping space(s) and/or tabs after the keyword */

  str= skip_delimiters(str);

  if (!*str)

  {

    fprintf(stderr,

	    "Unexpected EOL: No short language name after the keyword\n");

    return(0);

  }

  /* reading the short language tag */

  if (!(slang= get_word(&str)))

    return(0);				/* OOM: Fatal error */

  str= skip_delimiters(str);

  if (*str)

  {

    fprintf(stderr,

	    "The default language line does not end with short language "

	    "name\n");

    return(0);

  }

  return(slang);

}

/*

  Find the message in a particular language

  SYNOPSIS

    find_message()

    err             Error to find message for

    lang            Language of message to find

    no_default      Don't return default (English) if does not exit

  RETURN VALUE

    Returns the message structure if one is found, or NULL if not.

*/

static struct message *find_message(struct errors *err, const char *lang,

                                    my_bool no_default)

{

  struct message *tmp, *return_val= 0;

  uint i, count;

  count= (err->msg).elements;

  for (i= 0; i < count; i++)

  {

    tmp= dynamic_element(&err->msg, i, struct message*);

    if (!strcmp(tmp->lang_short_name, lang))

      return(tmp);

    if (!strcmp(tmp->lang_short_name, default_language))

    {

      assert(tmp->text[0] != 0);

      return_val= tmp;

    }

  }

  return(no_default ? NULL : return_val);

}

/*

  Check message format specifiers against error message for

  previous language

  SYNOPSIS

    checksum_format_specifier()

    msg            String for which to generate checksum

                   for the format specifiers

  RETURN VALUE

    Returns the checksum for all the characters of the

    format specifiers

    Ex.

     "text '%-64.s' text part 2 %d'"

            ^^^^^^              ^^

            characters will be xored to form checksum

    NOTE:

      Does not support format specifiers with positional args

      like "%2$s" 

*/

static ha_checksum checksum_format_specifier(const char* msg)

{

  ha_checksum chksum= 0;

  const uchar* p= (uchar *)msg;

  const uchar* start= 0;

  int num_format_specifiers= 0;

  while (*p)

  {

    if (*p == '%')

    {

      start= p+1; /* Entering format specifier */

      num_format_specifiers++;

    }

    else if (start)

    {

      switch(*p)

      {

      case 'd':

      case 'u':

      case 'x':

      case 's':

        chksum= my_checksum(chksum, start, p-start);

        start= 0; /* Not in format specifier anymore */

        break;

      default:

        break;

      }

    }

    p++;

  }

  if (start)

  {

    /* Still inside a format specifier after end of string */

    fprintf(stderr, "Still inside formatspecifier after end of string"

                    " in'%s'\n", msg);

    assert(start==0);

  }

  /* Add number of format specifiers to checksum as extra safeguard */

  chksum+= num_format_specifiers;

  return chksum;

}

/*

  Check message format specifiers against error message for

  previous language

  SYNOPSIS

    check_message_format()

    err             Error to check message for

    mess            Message to check

  RETURN VALUE

    Returns 0 if no previous error message or message format is ok

*/

static int check_message_format(struct errors *err,

                                const char* mess)

{

  struct message *first;

  /*  Get first message(if any) */

  if ((err->msg).elements == 0)

    return(0); /* No previous message to compare against */

  first= dynamic_element(&err->msg, 0, struct message*);

  assert(first != NULL);

  if (checksum_format_specifier(first->text) !=

      checksum_format_specifier(mess))

  {

    /* Check sum of format specifiers failed, they should be equal */

    return(1);

  }

  return(0);

}

/*

  Skips spaces and or tabs till the beginning of the next word

  Returns pointer to the beginning of the first character of the word

*/

static char *skip_delimiters(char *str)