PostgreSQL Source Code  git master
output.c
Go to the documentation of this file.
1 /* src/interfaces/ecpg/preproc/output.c */
2 
3 #include "postgres_fe.h"
4 
5 #include "preproc_extern.h"
6 
7 static void output_escaped_str(char *cmd, bool quoted);
8 
9 void
11 {
12  char *line = hashline_number();
13 
14  fprintf(base_yyout, "%s", line);
15  free(line);
16 }
17 
18 void
19 output_simple_statement(char *stmt, int whenever_mode)
20 {
21  output_escaped_str(stmt, false);
22  if (whenever_mode)
23  whenever_action(whenever_mode);
25  free(stmt);
26 }
27 
28 
29 /*
30  * store the whenever action here
31  */
32 struct when when_error,
33  when_nf,
34  when_warn;
35 
36 static void
37 print_action(struct when *w)
38 {
39  switch (w->code)
40  {
41  case W_SQLPRINT:
42  fprintf(base_yyout, "sqlprint();");
43  break;
44  case W_GOTO:
45  fprintf(base_yyout, "goto %s;", w->command);
46  break;
47  case W_DO:
48  fprintf(base_yyout, "%s;", w->command);
49  break;
50  case W_STOP:
51  fprintf(base_yyout, "exit (1);");
52  break;
53  case W_BREAK:
54  fprintf(base_yyout, "break;");
55  break;
56  case W_CONTINUE:
57  fprintf(base_yyout, "continue;");
58  break;
59  default:
60  fprintf(base_yyout, "{/* %d not implemented yet */}", w->code);
61  break;
62  }
63 }
64 
65 void
67 {
68  if ((mode & 1) == 1 && when_nf.code != W_NOTHING)
69  {
71  fprintf(base_yyout, "\nif (sqlca.sqlcode == ECPG_NOT_FOUND) ");
72  print_action(&when_nf);
73  }
74  if (when_warn.code != W_NOTHING)
75  {
77  fprintf(base_yyout, "\nif (sqlca.sqlwarn[0] == 'W') ");
79  }
80  if (when_error.code != W_NOTHING)
81  {
83  fprintf(base_yyout, "\nif (sqlca.sqlcode < 0) ");
84  print_action(&when_error);
85  }
86 
87  if ((mode & 2) == 2)
88  fputc('}', base_yyout);
89 
91 }
92 
93 char *
95 {
96  /* do not print line numbers if we are in debug mode */
97  if (input_filename
98 #ifdef YYDEBUG
99  && !base_yydebug
100 #endif
101  )
102  {
103  /* "* 2" here is for escaping '\' and '"' below */
104  char *line = mm_alloc(strlen("\n#line %d \"%s\"\n") + sizeof(int) * CHAR_BIT * 10 / 3 + strlen(input_filename) * 2);
105  char *src,
106  *dest;
107 
108  sprintf(line, "\n#line %d \"", base_yylineno);
109  src = input_filename;
110  dest = line + strlen(line);
111  while (*src)
112  {
113  if (*src == '\\' || *src == '"')
114  *dest++ = '\\';
115  *dest++ = *src++;
116  }
117  *dest = '\0';
118  strcat(dest, "\"\n");
119 
120  return line;
121  }
122 
123  return EMPTY;
124 }
125 
126 static char *ecpg_statement_type_name[] = {
127  "ECPGst_normal",
128  "ECPGst_execute",
129  "ECPGst_exec_immediate",
130  "ECPGst_prepnormal",
131  "ECPGst_prepare",
132  "ECPGst_exec_with_exprlist"
133 };
134 
135 void
136 output_statement(char *stmt, int whenever_mode, enum ECPG_statement_type st)
137 {
138  fprintf(base_yyout, "{ ECPGdo(__LINE__, %d, %d, %s, %d, ", compat, force_indicator, connection ? connection : "NULL", questionmarks);
139 
140  if (st == ECPGst_prepnormal && !auto_prepare)
141  st = ECPGst_normal;
142 
143  /*
144  * In following cases, stmt is CSTRING or char_variable. They must be
145  * output directly. - prepared_name of EXECUTE without exprlist -
146  * execstring of EXECUTE IMMEDIATE
147  */
149  if (st == ECPGst_execute || st == ECPGst_exec_immediate)
150  fprintf(base_yyout, "%s, ", stmt);
151  else
152  {
153  fputs("\"", base_yyout);
154  output_escaped_str(stmt, false);
155  fputs("\", ", base_yyout);
156  }
157 
158  /* dump variables to C file */
160  fputs("ECPGt_EOIT, ", base_yyout);
162  fputs("ECPGt_EORT);", base_yyout);
163  reset_variables();
164 
165  whenever_action(whenever_mode | 2);
166  free(stmt);
167 }
168 
169 void
170 output_prepare_statement(char *name, char *stmt)
171 {
172  fprintf(base_yyout, "{ ECPGprepare(__LINE__, %s, %d, ", connection ? connection : "NULL", questionmarks);
173  output_escaped_str(name, true);
174  fputs(", ", base_yyout);
175  output_escaped_str(stmt, true);
176  fputs(");", base_yyout);
177  whenever_action(2);
178  free(name);
179 }
180 
181 void
183 {
184  const char *con = connection ? connection : "NULL";
185 
186  if (strcmp(name, "all") != 0)
187  {
188  fprintf(base_yyout, "{ ECPGdeallocate(__LINE__, %d, %s, ", compat, con);
189  output_escaped_str(name, true);
190  fputs(");", base_yyout);
191  }
192  else
193  fprintf(base_yyout, "{ ECPGdeallocate_all(__LINE__, %d, %s);", compat, con);
194 
195  whenever_action(2);
196  free(name);
197 }
198 
199 static void
200 output_escaped_str(char *str, bool quoted)
201 {
202  int i = 0;
203  int len = strlen(str);
204 
205  if (quoted && str[0] == '"' && str[len - 1] == '"') /* do not escape quotes
206  * at beginning and end
207  * if quoted string */
208  {
209  i = 1;
210  len--;
211  fputs("\"", base_yyout);
212  }
213 
214  /* output this char by char as we have to filter " and \n */
215  for (; i < len; i++)
216  {
217  if (str[i] == '"')
218  fputs("\\\"", base_yyout);
219  else if (str[i] == '\n')
220  fputs("\\\n", base_yyout);
221  else if (str[i] == '\\')
222  {
223  int j = i;
224 
225  /*
226  * check whether this is a continuation line if it is, do not
227  * output anything because newlines are escaped anyway
228  */
229 
230  /* accept blanks after the '\' as some other compilers do too */
231  do
232  {
233  j++;
234  } while (str[j] == ' ' || str[j] == '\t');
235 
236  if ((str[j] != '\n') && (str[j] != '\r' || str[j + 1] != '\n')) /* not followed by a
237  * newline */
238  fputs("\\\\", base_yyout);
239  }
240  else if (str[i] == '\r' && str[i + 1] == '\n')
241  {
242  fputs("\\\r\n", base_yyout);
243  i++;
244  }
245  else
246  fputc(str[i], base_yyout);
247  }
248 
249  if (quoted && str[0] == '"' && str[len] == '"')
250  fputs("\"", base_yyout);
251 }
static PgChecksumMode mode
Definition: pg_checksums.c:65
static void print_action(struct when *w)
Definition: output.c:37
Definition: type.h:73
static char * ecpg_statement_type_name[]
Definition: output.c:126
void output_statement(char *stmt, int whenever_mode, enum ECPG_statement_type st)
Definition: output.c:136
void output_line_number(void)
Definition: output.c:10
#define fprintf
Definition: port.h:221
enum WHEN_TYPE code
Definition: type.h:84
void whenever_action(int mode)
Definition: output.c:66
void reset_variables(void)
Definition: variable.c:369
ECPG_statement_type
Definition: ecpgtype.h:95
#define sprintf
Definition: port.h:219
char * hashline_number(void)
Definition: output.c:94
static void output_escaped_str(char *cmd, bool quoted)
Definition: output.c:200
void dump_variables(struct arguments *, int)
Definition: variable.c:438
char * command
Definition: type.h:85
Definition: type.h:82
bool questionmarks
Definition: ecpg.c:19
enum COMPAT_MODE compat
Definition: ecpg.c:25
int base_yylineno
void output_prepare_statement(char *name, char *stmt)
Definition: output.c:170
Definition: type.h:79
Definition: type.h:78
#define free(a)
Definition: header.h:65
void output_deallocate_prepare_statement(char *name)
Definition: output.c:182
Definition: type.h:75
bool auto_prepare
Definition: ecpg.c:21
#define EMPTY
Definition: regcomp.c:285
const char * name
Definition: encode.c:561
bool force_indicator
Definition: ecpg.c:18
void * mm_alloc(size_t)
Definition: type.c:13
int i
char * input_filename
Definition: type.h:77
struct arguments * argsresult
Definition: variable.c:366
char * connection
struct arguments * argsinsert
Definition: variable.c:365
void output_simple_statement(char *stmt, int whenever_mode)
Definition: output.c:19
struct when when_error when_nf when_warn
Definition: output.c:32
FILE * base_yyout