~ubuntu-branches/ubuntu/trusty/mariadb-5.5/trusty-proposed

« back to all changes in this revision

Viewing changes to scripts/comp_sql.c

  • Committer: Package Import Robot
  • Author(s): Otto Kekäläinen
  • Date: 2013-12-22 10:27:05 UTC
  • Revision ID: package-import@ubuntu.com-20131222102705-mndw7s12mz0szrcn
Tags: upstream-5.5.32
Import upstream version 5.5.32

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
/* Copyright (C) 2004 MySQL AB
 
2
 
 
3
   This program is free software; you can redistribute it and/or modify
 
4
   it under the terms of the GNU General Public License as published by
 
5
   the Free Software Foundation; version 2 of the License.
 
6
 
 
7
   This program is distributed in the hope that it will be useful,
 
8
   but WITHOUT ANY WARRANTY; without even the implied warranty of
 
9
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
10
   GNU General Public License for more details.
 
11
 
 
12
   You should have received a copy of the GNU General Public License
 
13
   along with this program; if not, write to the Free Software
 
14
   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA */
 
15
 
 
16
/*
 
17
  Written by Magnus Svensson
 
18
*/
 
19
 
 
20
/*
 
21
  Converts a SQL file into a C file that can be compiled and linked
 
22
  into other programs
 
23
*/
 
24
 
 
25
#include <stdarg.h>
 
26
#include <stdlib.h>
 
27
#include <stdio.h>
 
28
#include <sys/stat.h>
 
29
 
 
30
/* Compiler-dependent constant for maximum string constant */
 
31
#define MAX_STRING_CONSTANT_LENGTH 65535
 
32
 
 
33
FILE *in, *out;
 
34
 
 
35
static void die(const char *fmt, ...)
 
36
{
 
37
  va_list args;
 
38
 
 
39
  /* Print the error message */
 
40
  fprintf(stderr, "FATAL ERROR: ");
 
41
  if (fmt)
 
42
  {
 
43
    va_start(args, fmt);
 
44
    vfprintf(stderr, fmt, args);
 
45
    va_end(args);
 
46
  }
 
47
  else
 
48
    fprintf(stderr, "unknown error");
 
49
  fprintf(stderr, "\n");
 
50
  fflush(stderr);
 
51
 
 
52
  /* Close any open files */
 
53
  if (in)
 
54
    fclose(in);
 
55
  if (out)
 
56
    fclose(out);
 
57
 
 
58
  exit(1);
 
59
}
 
60
 
 
61
 
 
62
int main(int argc, char *argv[])
 
63
{
 
64
  char buff[512];
 
65
  struct stat st;
 
66
  char* struct_name= argv[1];
 
67
  char* infile_name= argv[2];
 
68
  char* outfile_name= argv[3];
 
69
 
 
70
 
 
71
  if (argc != 4)
 
72
    die("Usage: comp_sql <struct_name> <sql_filename> <c_filename>");
 
73
 
 
74
  /* Open input and output file */
 
75
  if (!(in= fopen(infile_name, "r")))
 
76
    die("Failed to open SQL file '%s'", infile_name);
 
77
 
 
78
  
 
79
  if (!(out= fopen(outfile_name, "w")))
 
80
    die("Failed to open output file '%s'", outfile_name);
 
81
  fprintf(out, "const char %s[]={\n",struct_name);
 
82
 
 
83
  /* 
 
84
    Some compilers have limitations how long a string constant can be.
 
85
    We'll output very long strings as hexadecimal arrays, and short ones
 
86
    as strings (prettier)
 
87
  */
 
88
  stat(infile_name, &st);
 
89
  if (st.st_size > MAX_STRING_CONSTANT_LENGTH)
 
90
  {
 
91
    int cnt=0;
 
92
    int c;
 
93
    for(cnt=0;;cnt++)
 
94
    {
 
95
      c= fgetc(in);
 
96
      if (c== -1)
 
97
        break;
 
98
 
 
99
      if(cnt != 0)
 
100
        fputc(',', out);
 
101
 
 
102
      /* Put line break after each 16 hex characters */
 
103
      if(cnt && (cnt%16 == 0))
 
104
        fputc('\n', out);
 
105
 
 
106
      fprintf(out,"0x%02x",c);
 
107
    }
 
108
    fprintf(out,",0x00");
 
109
  }
 
110
  else
 
111
  {
 
112
    fprintf(out,"\"");
 
113
    while (fgets(buff, sizeof(buff), in))
 
114
    {
 
115
      char *curr= buff;
 
116
      while (*curr)
 
117
      {
 
118
        if (*curr == '\n')
 
119
        {
 
120
          /*
 
121
            Reached end of line, add escaped newline, escaped
 
122
            backslash and a newline to outfile
 
123
          */
 
124
          fprintf(out, "\\n \"\n\"");
 
125
          curr++;
 
126
        }
 
127
        else if (*curr == '\r')
 
128
        {
 
129
          curr++; /* Skip */
 
130
        }
 
131
        else
 
132
        {
 
133
          if (*curr == '"')
 
134
          {
 
135
            /* Needs escape */
 
136
            fputc('\\', out);
 
137
          }
 
138
 
 
139
          fputc(*curr, out);
 
140
          curr++;
 
141
        }
 
142
      }
 
143
      if (*(curr-1) != '\n')
 
144
      {
 
145
        /*
 
146
          Some compilers have a max string length,
 
147
          insert a newline at every 512th char in long
 
148
          strings
 
149
        */
 
150
        fprintf(out, "\"\n\"");
 
151
      }
 
152
    }
 
153
    fprintf(out, "\\\n\"");
 
154
  }
 
155
  
 
156
  fprintf(out, "};\n");
 
157
  fclose(in);
 
158
  fclose(out);
 
159
 
 
160
  exit(0);
 
161
 
 
162
}
 
163