pg_query 1.0.2 → 2.0.0
Sign up to get free protection for your applications and to get access to all the features.
- checksums.yaml +5 -5
- data/CHANGELOG.md +162 -40
- data/README.md +80 -69
- data/Rakefile +85 -4
- data/ext/pg_query/extconf.rb +4 -32
- data/ext/pg_query/guc-file.c +0 -0
- data/ext/pg_query/pg_query.c +104 -0
- data/ext/pg_query/pg_query.pb-c.c +37628 -0
- data/ext/pg_query/pg_query_deparse.c +9953 -0
- data/ext/pg_query/pg_query_fingerprint.c +292 -0
- data/ext/pg_query/pg_query_fingerprint.h +8 -0
- data/ext/pg_query/pg_query_internal.h +24 -0
- data/ext/pg_query/pg_query_json_plpgsql.c +738 -0
- data/ext/pg_query/pg_query_json_plpgsql.h +9 -0
- data/ext/pg_query/pg_query_normalize.c +437 -0
- data/ext/pg_query/pg_query_outfuncs.h +10 -0
- data/ext/pg_query/pg_query_outfuncs_json.c +297 -0
- data/ext/pg_query/pg_query_outfuncs_protobuf.c +237 -0
- data/ext/pg_query/pg_query_parse.c +148 -0
- data/ext/pg_query/pg_query_parse_plpgsql.c +460 -0
- data/ext/pg_query/pg_query_readfuncs.h +11 -0
- data/ext/pg_query/pg_query_readfuncs_protobuf.c +142 -0
- data/ext/pg_query/pg_query_ruby.c +108 -12
- data/ext/pg_query/pg_query_scan.c +173 -0
- data/ext/pg_query/pg_query_split.c +221 -0
- data/ext/pg_query/protobuf-c.c +3660 -0
- data/ext/pg_query/src_backend_catalog_namespace.c +1051 -0
- data/ext/pg_query/src_backend_catalog_pg_proc.c +142 -0
- data/ext/pg_query/src_backend_commands_define.c +117 -0
- data/ext/pg_query/src_backend_libpq_pqcomm.c +651 -0
- data/ext/pg_query/src_backend_nodes_bitmapset.c +513 -0
- data/ext/pg_query/src_backend_nodes_copyfuncs.c +6013 -0
- data/ext/pg_query/src_backend_nodes_equalfuncs.c +4003 -0
- data/ext/pg_query/src_backend_nodes_extensible.c +99 -0
- data/ext/pg_query/src_backend_nodes_list.c +922 -0
- data/ext/pg_query/src_backend_nodes_makefuncs.c +417 -0
- data/ext/pg_query/src_backend_nodes_nodeFuncs.c +1363 -0
- data/ext/pg_query/src_backend_nodes_value.c +84 -0
- data/ext/pg_query/src_backend_parser_gram.c +47456 -0
- data/ext/pg_query/src_backend_parser_parse_expr.c +313 -0
- data/ext/pg_query/src_backend_parser_parser.c +497 -0
- data/ext/pg_query/src_backend_parser_scan.c +7091 -0
- data/ext/pg_query/src_backend_parser_scansup.c +160 -0
- data/ext/pg_query/src_backend_postmaster_postmaster.c +2230 -0
- data/ext/pg_query/src_backend_storage_ipc_ipc.c +192 -0
- data/ext/pg_query/src_backend_storage_lmgr_s_lock.c +370 -0
- data/ext/pg_query/src_backend_tcop_postgres.c +776 -0
- data/ext/pg_query/src_backend_utils_adt_datum.c +326 -0
- data/ext/pg_query/src_backend_utils_adt_expandeddatum.c +98 -0
- data/ext/pg_query/src_backend_utils_adt_format_type.c +136 -0
- data/ext/pg_query/src_backend_utils_adt_ruleutils.c +1683 -0
- data/ext/pg_query/src_backend_utils_error_assert.c +74 -0
- data/ext/pg_query/src_backend_utils_error_elog.c +1748 -0
- data/ext/pg_query/src_backend_utils_fmgr_fmgr.c +570 -0
- data/ext/pg_query/src_backend_utils_hash_dynahash.c +1086 -0
- data/ext/pg_query/src_backend_utils_init_globals.c +168 -0
- data/ext/pg_query/src_backend_utils_mb_mbutils.c +839 -0
- data/ext/pg_query/src_backend_utils_misc_guc.c +1831 -0
- data/ext/pg_query/src_backend_utils_mmgr_aset.c +1560 -0
- data/ext/pg_query/src_backend_utils_mmgr_mcxt.c +1006 -0
- data/ext/pg_query/src_common_encnames.c +158 -0
- data/ext/pg_query/src_common_keywords.c +39 -0
- data/ext/pg_query/src_common_kwlist_d.h +1081 -0
- data/ext/pg_query/src_common_kwlookup.c +91 -0
- data/ext/pg_query/src_common_psprintf.c +158 -0
- data/ext/pg_query/src_common_string.c +86 -0
- data/ext/pg_query/src_common_stringinfo.c +336 -0
- data/ext/pg_query/src_common_wchar.c +1651 -0
- data/ext/pg_query/src_pl_plpgsql_src_pl_comp.c +1133 -0
- data/ext/pg_query/src_pl_plpgsql_src_pl_funcs.c +877 -0
- data/ext/pg_query/src_pl_plpgsql_src_pl_gram.c +6533 -0
- data/ext/pg_query/src_pl_plpgsql_src_pl_handler.c +107 -0
- data/ext/pg_query/src_pl_plpgsql_src_pl_reserved_kwlist_d.h +123 -0
- data/ext/pg_query/src_pl_plpgsql_src_pl_scanner.c +671 -0
- data/ext/pg_query/src_pl_plpgsql_src_pl_unreserved_kwlist_d.h +255 -0
- data/ext/pg_query/src_port_erand48.c +127 -0
- data/ext/pg_query/src_port_pg_bitutils.c +246 -0
- data/ext/pg_query/src_port_pgsleep.c +69 -0
- data/ext/pg_query/src_port_pgstrcasecmp.c +83 -0
- data/ext/pg_query/src_port_qsort.c +240 -0
- data/ext/pg_query/src_port_random.c +31 -0
- data/ext/pg_query/src_port_snprintf.c +1449 -0
- data/ext/pg_query/src_port_strerror.c +324 -0
- data/ext/pg_query/src_port_strnlen.c +39 -0
- data/ext/pg_query/xxhash.c +43 -0
- data/lib/pg_query.rb +7 -4
- data/lib/pg_query/constants.rb +21 -0
- data/lib/pg_query/deparse.rb +16 -991
- data/lib/pg_query/filter_columns.rb +86 -85
- data/lib/pg_query/fingerprint.rb +122 -87
- data/lib/pg_query/json_field_names.rb +1402 -0
- data/lib/pg_query/node.rb +31 -0
- data/lib/pg_query/param_refs.rb +42 -37
- data/lib/pg_query/parse.rb +220 -200
- data/lib/pg_query/parse_error.rb +1 -1
- data/lib/pg_query/pg_query_pb.rb +3211 -0
- data/lib/pg_query/scan.rb +23 -0
- data/lib/pg_query/treewalker.rb +24 -40
- data/lib/pg_query/truncate.rb +64 -43
- data/lib/pg_query/version.rb +2 -2
- metadata +102 -11
- data/ext/pg_query/pg_query_ruby.h +0 -10
- data/lib/pg_query/deep_dup.rb +0 -16
- data/lib/pg_query/deparse/alter_table.rb +0 -42
- data/lib/pg_query/deparse/interval.rb +0 -105
- data/lib/pg_query/legacy_parsetree.rb +0 -109
- data/lib/pg_query/node_types.rb +0 -282
@@ -0,0 +1,148 @@
|
|
1
|
+
#include "pg_query.h"
|
2
|
+
#include "pg_query_internal.h"
|
3
|
+
#include "pg_query_outfuncs.h"
|
4
|
+
|
5
|
+
#include "parser/parser.h"
|
6
|
+
#include "parser/scanner.h"
|
7
|
+
#include "parser/scansup.h"
|
8
|
+
|
9
|
+
#include <unistd.h>
|
10
|
+
#include <fcntl.h>
|
11
|
+
|
12
|
+
PgQueryInternalParsetreeAndError pg_query_raw_parse(const char* input)
|
13
|
+
{
|
14
|
+
PgQueryInternalParsetreeAndError result = {0};
|
15
|
+
MemoryContext parse_context = CurrentMemoryContext;
|
16
|
+
|
17
|
+
char stderr_buffer[STDERR_BUFFER_LEN + 1] = {0};
|
18
|
+
#ifndef DEBUG
|
19
|
+
int stderr_global;
|
20
|
+
int stderr_pipe[2];
|
21
|
+
#endif
|
22
|
+
|
23
|
+
#ifndef DEBUG
|
24
|
+
// Setup pipe for stderr redirection
|
25
|
+
if (pipe(stderr_pipe) != 0) {
|
26
|
+
PgQueryError* error = malloc(sizeof(PgQueryError));
|
27
|
+
|
28
|
+
error->message = strdup("Failed to open pipe, too many open file descriptors")
|
29
|
+
|
30
|
+
result.error = error;
|
31
|
+
|
32
|
+
return result;
|
33
|
+
}
|
34
|
+
|
35
|
+
fcntl(stderr_pipe[0], F_SETFL, fcntl(stderr_pipe[0], F_GETFL) | O_NONBLOCK);
|
36
|
+
|
37
|
+
// Redirect stderr to the pipe
|
38
|
+
stderr_global = dup(STDERR_FILENO);
|
39
|
+
dup2(stderr_pipe[1], STDERR_FILENO);
|
40
|
+
close(stderr_pipe[1]);
|
41
|
+
#endif
|
42
|
+
|
43
|
+
PG_TRY();
|
44
|
+
{
|
45
|
+
result.tree = raw_parser(input);
|
46
|
+
|
47
|
+
#ifndef DEBUG
|
48
|
+
// Save stderr for result
|
49
|
+
read(stderr_pipe[0], stderr_buffer, STDERR_BUFFER_LEN);
|
50
|
+
#endif
|
51
|
+
|
52
|
+
result.stderr_buffer = strdup(stderr_buffer);
|
53
|
+
}
|
54
|
+
PG_CATCH();
|
55
|
+
{
|
56
|
+
ErrorData* error_data;
|
57
|
+
PgQueryError* error;
|
58
|
+
|
59
|
+
MemoryContextSwitchTo(parse_context);
|
60
|
+
error_data = CopyErrorData();
|
61
|
+
|
62
|
+
// Note: This is intentionally malloc so exiting the memory context doesn't free this
|
63
|
+
error = malloc(sizeof(PgQueryError));
|
64
|
+
error->message = strdup(error_data->message);
|
65
|
+
error->filename = strdup(error_data->filename);
|
66
|
+
error->funcname = strdup(error_data->funcname);
|
67
|
+
error->context = NULL;
|
68
|
+
error->lineno = error_data->lineno;
|
69
|
+
error->cursorpos = error_data->cursorpos;
|
70
|
+
|
71
|
+
result.error = error;
|
72
|
+
FlushErrorState();
|
73
|
+
}
|
74
|
+
PG_END_TRY();
|
75
|
+
|
76
|
+
#ifndef DEBUG
|
77
|
+
// Restore stderr, close pipe
|
78
|
+
dup2(stderr_global, STDERR_FILENO);
|
79
|
+
close(stderr_pipe[0]);
|
80
|
+
close(stderr_global);
|
81
|
+
#endif
|
82
|
+
|
83
|
+
return result;
|
84
|
+
}
|
85
|
+
|
86
|
+
PgQueryParseResult pg_query_parse(const char* input)
|
87
|
+
{
|
88
|
+
MemoryContext ctx = NULL;
|
89
|
+
PgQueryInternalParsetreeAndError parsetree_and_error;
|
90
|
+
PgQueryParseResult result = {0};
|
91
|
+
char *tree_json = NULL;
|
92
|
+
|
93
|
+
ctx = pg_query_enter_memory_context();
|
94
|
+
|
95
|
+
parsetree_and_error = pg_query_raw_parse(input);
|
96
|
+
|
97
|
+
// These are all malloc-ed and will survive exiting the memory context, the caller is responsible to free them now
|
98
|
+
result.stderr_buffer = parsetree_and_error.stderr_buffer;
|
99
|
+
result.error = parsetree_and_error.error;
|
100
|
+
|
101
|
+
tree_json = pg_query_nodes_to_json(parsetree_and_error.tree);
|
102
|
+
result.parse_tree = strdup(tree_json);
|
103
|
+
pfree(tree_json);
|
104
|
+
|
105
|
+
pg_query_exit_memory_context(ctx);
|
106
|
+
|
107
|
+
return result;
|
108
|
+
}
|
109
|
+
|
110
|
+
PgQueryProtobufParseResult pg_query_parse_protobuf(const char* input)
|
111
|
+
{
|
112
|
+
MemoryContext ctx = NULL;
|
113
|
+
PgQueryInternalParsetreeAndError parsetree_and_error;
|
114
|
+
PgQueryProtobufParseResult result = {};
|
115
|
+
|
116
|
+
ctx = pg_query_enter_memory_context();
|
117
|
+
|
118
|
+
parsetree_and_error = pg_query_raw_parse(input);
|
119
|
+
|
120
|
+
// These are all malloc-ed and will survive exiting the memory context, the caller is responsible to free them now
|
121
|
+
result.stderr_buffer = parsetree_and_error.stderr_buffer;
|
122
|
+
result.error = parsetree_and_error.error;
|
123
|
+
result.parse_tree = pg_query_nodes_to_protobuf(parsetree_and_error.tree);
|
124
|
+
|
125
|
+
pg_query_exit_memory_context(ctx);
|
126
|
+
|
127
|
+
return result;
|
128
|
+
}
|
129
|
+
|
130
|
+
void pg_query_free_parse_result(PgQueryParseResult result)
|
131
|
+
{
|
132
|
+
if (result.error) {
|
133
|
+
pg_query_free_error(result.error);
|
134
|
+
}
|
135
|
+
|
136
|
+
free(result.parse_tree);
|
137
|
+
free(result.stderr_buffer);
|
138
|
+
}
|
139
|
+
|
140
|
+
void pg_query_free_protobuf_parse_result(PgQueryProtobufParseResult result)
|
141
|
+
{
|
142
|
+
if (result.error) {
|
143
|
+
pg_query_free_error(result.error);
|
144
|
+
}
|
145
|
+
|
146
|
+
free(result.parse_tree.data);
|
147
|
+
free(result.stderr_buffer);
|
148
|
+
}
|
@@ -0,0 +1,460 @@
|
|
1
|
+
#define _GNU_SOURCE // Necessary to get asprintf (which is a GNU extension)
|
2
|
+
#include <stdio.h>
|
3
|
+
|
4
|
+
#include "pg_query.h"
|
5
|
+
#include "pg_query_internal.h"
|
6
|
+
#include "pg_query_json_plpgsql.h"
|
7
|
+
|
8
|
+
#include <assert.h>
|
9
|
+
|
10
|
+
#include <catalog/pg_type.h>
|
11
|
+
#include <catalog/objectaddress.h>
|
12
|
+
#include <catalog/pg_proc.h>
|
13
|
+
#include <nodes/parsenodes.h>
|
14
|
+
#include <nodes/nodeFuncs.h>
|
15
|
+
|
16
|
+
typedef struct {
|
17
|
+
PLpgSQL_function *func;
|
18
|
+
PgQueryError* error;
|
19
|
+
} PgQueryInternalPlpgsqlFuncAndError;
|
20
|
+
|
21
|
+
static PgQueryInternalPlpgsqlFuncAndError pg_query_raw_parse_plpgsql(CreateFunctionStmt* stmt);
|
22
|
+
|
23
|
+
static void add_dummy_return(PLpgSQL_function *function)
|
24
|
+
{
|
25
|
+
/*
|
26
|
+
* If the outer block has an EXCEPTION clause, we need to make a new outer
|
27
|
+
* block, since the added RETURN shouldn't act like it is inside the
|
28
|
+
* EXCEPTION clause.
|
29
|
+
*/
|
30
|
+
if (function->action->exceptions != NULL)
|
31
|
+
{
|
32
|
+
PLpgSQL_stmt_block *new;
|
33
|
+
|
34
|
+
new = palloc0(sizeof(PLpgSQL_stmt_block));
|
35
|
+
new->cmd_type = PLPGSQL_STMT_BLOCK;
|
36
|
+
new->body = list_make1(function->action);
|
37
|
+
|
38
|
+
function->action = new;
|
39
|
+
}
|
40
|
+
if (function->action->body == NIL ||
|
41
|
+
((PLpgSQL_stmt *) llast(function->action->body))->cmd_type != PLPGSQL_STMT_RETURN)
|
42
|
+
{
|
43
|
+
PLpgSQL_stmt_return *new;
|
44
|
+
|
45
|
+
new = palloc0(sizeof(PLpgSQL_stmt_return));
|
46
|
+
new->cmd_type = PLPGSQL_STMT_RETURN;
|
47
|
+
new->expr = NULL;
|
48
|
+
new->retvarno = function->out_param_varno;
|
49
|
+
|
50
|
+
function->action->body = lappend(function->action->body, new);
|
51
|
+
}
|
52
|
+
}
|
53
|
+
|
54
|
+
static void plpgsql_compile_error_callback(void *arg)
|
55
|
+
{
|
56
|
+
if (arg)
|
57
|
+
{
|
58
|
+
/*
|
59
|
+
* Try to convert syntax error position to reference text of original
|
60
|
+
* CREATE FUNCTION or DO command.
|
61
|
+
*/
|
62
|
+
if (function_parse_error_transpose((const char *) arg))
|
63
|
+
return;
|
64
|
+
|
65
|
+
/*
|
66
|
+
* Done if a syntax error position was reported; otherwise we have to
|
67
|
+
* fall back to a "near line N" report.
|
68
|
+
*/
|
69
|
+
}
|
70
|
+
|
71
|
+
if (plpgsql_error_funcname)
|
72
|
+
errcontext("compilation of PL/pgSQL function \"%s\" near line %d",
|
73
|
+
plpgsql_error_funcname, plpgsql_latest_lineno());
|
74
|
+
}
|
75
|
+
|
76
|
+
static PLpgSQL_function *compile_create_function_stmt(CreateFunctionStmt* stmt)
|
77
|
+
{
|
78
|
+
char *func_name;
|
79
|
+
char *proc_source = NULL;
|
80
|
+
PLpgSQL_function *function;
|
81
|
+
ErrorContextCallback plerrcontext;
|
82
|
+
PLpgSQL_variable *var;
|
83
|
+
int parse_rc;
|
84
|
+
MemoryContext func_cxt;
|
85
|
+
int i;
|
86
|
+
PLpgSQL_rec *rec;
|
87
|
+
const ListCell *lc, *lc2, *lc3;
|
88
|
+
bool is_trigger = false;
|
89
|
+
bool is_setof = false;
|
90
|
+
|
91
|
+
assert(IsA(stmt, CreateFunctionStmt));
|
92
|
+
|
93
|
+
func_name = strVal(linitial(stmt->funcname));
|
94
|
+
|
95
|
+
foreach(lc, stmt->options)
|
96
|
+
{
|
97
|
+
DefElem* elem = (DefElem*) lfirst(lc);
|
98
|
+
|
99
|
+
if (strcmp(elem->defname, "as") == 0) {
|
100
|
+
const ListCell *lc2;
|
101
|
+
|
102
|
+
assert(IsA(elem->arg, List));
|
103
|
+
|
104
|
+
foreach(lc2, (List*) elem->arg)
|
105
|
+
{
|
106
|
+
proc_source = strVal(lfirst(lc2));
|
107
|
+
}
|
108
|
+
}
|
109
|
+
}
|
110
|
+
|
111
|
+
assert(proc_source);
|
112
|
+
|
113
|
+
if (stmt->returnType != NULL) {
|
114
|
+
foreach(lc3, stmt->returnType->names)
|
115
|
+
{
|
116
|
+
char* val = strVal(lfirst(lc3));
|
117
|
+
|
118
|
+
if (strcmp(val, "trigger") == 0) {
|
119
|
+
is_trigger = true;
|
120
|
+
}
|
121
|
+
}
|
122
|
+
|
123
|
+
if (stmt->returnType->setof) {
|
124
|
+
is_setof = true;
|
125
|
+
}
|
126
|
+
}
|
127
|
+
|
128
|
+
/*
|
129
|
+
* Setup the scanner input and error info. We assume that this function
|
130
|
+
* cannot be invoked recursively, so there's no need to save and restore
|
131
|
+
* the static variables used here.
|
132
|
+
*/
|
133
|
+
plpgsql_scanner_init(proc_source);
|
134
|
+
|
135
|
+
plpgsql_error_funcname = func_name;
|
136
|
+
|
137
|
+
/*
|
138
|
+
* Setup error traceback support for ereport()
|
139
|
+
*/
|
140
|
+
plerrcontext.callback = plpgsql_compile_error_callback;
|
141
|
+
plerrcontext.arg = proc_source;
|
142
|
+
plerrcontext.previous = error_context_stack;
|
143
|
+
error_context_stack = &plerrcontext;
|
144
|
+
|
145
|
+
/* Do extra syntax checking if check_function_bodies is on */
|
146
|
+
plpgsql_check_syntax = true;
|
147
|
+
|
148
|
+
/* Function struct does not live past current statement */
|
149
|
+
function = (PLpgSQL_function *) palloc0(sizeof(PLpgSQL_function));
|
150
|
+
|
151
|
+
plpgsql_curr_compile = function;
|
152
|
+
|
153
|
+
/*
|
154
|
+
* All the rest of the compile-time storage (e.g. parse tree) is kept in
|
155
|
+
* its own memory context, so it can be reclaimed easily.
|
156
|
+
*/
|
157
|
+
func_cxt = AllocSetContextCreate(CurrentMemoryContext,
|
158
|
+
"PL/pgSQL pg_query context",
|
159
|
+
ALLOCSET_DEFAULT_SIZES);
|
160
|
+
plpgsql_compile_tmp_cxt = MemoryContextSwitchTo(func_cxt);
|
161
|
+
|
162
|
+
function->fn_signature = pstrdup(func_name);
|
163
|
+
function->fn_is_trigger = PLPGSQL_NOT_TRIGGER;
|
164
|
+
function->fn_input_collation = InvalidOid;
|
165
|
+
function->fn_cxt = func_cxt;
|
166
|
+
function->out_param_varno = -1; /* set up for no OUT param */
|
167
|
+
function->resolve_option = plpgsql_variable_conflict;
|
168
|
+
function->print_strict_params = plpgsql_print_strict_params;
|
169
|
+
|
170
|
+
/*
|
171
|
+
* don't do extra validation for inline code as we don't want to add spam
|
172
|
+
* at runtime
|
173
|
+
*/
|
174
|
+
function->extra_warnings = 0;
|
175
|
+
function->extra_errors = 0;
|
176
|
+
|
177
|
+
plpgsql_ns_init();
|
178
|
+
plpgsql_ns_push(func_name, PLPGSQL_LABEL_BLOCK);
|
179
|
+
plpgsql_DumpExecTree = false;
|
180
|
+
plpgsql_start_datums();
|
181
|
+
|
182
|
+
/* Set up as though in a function returning VOID */
|
183
|
+
function->fn_rettype = VOIDOID;
|
184
|
+
function->fn_retset = is_setof;
|
185
|
+
function->fn_retistuple = false;
|
186
|
+
function->fn_retisdomain = false;
|
187
|
+
function->fn_prokind = PROKIND_FUNCTION;
|
188
|
+
/* a bit of hardwired knowledge about type VOID here */
|
189
|
+
function->fn_retbyval = true;
|
190
|
+
function->fn_rettyplen = sizeof(int32);
|
191
|
+
|
192
|
+
/*
|
193
|
+
* Remember if function is STABLE/IMMUTABLE. XXX would it be better to
|
194
|
+
* set this TRUE inside a read-only transaction? Not clear.
|
195
|
+
*/
|
196
|
+
function->fn_readonly = false;
|
197
|
+
|
198
|
+
/*
|
199
|
+
* Create the magic FOUND variable.
|
200
|
+
*/
|
201
|
+
var = plpgsql_build_variable("found", 0,
|
202
|
+
plpgsql_build_datatype(BOOLOID,
|
203
|
+
-1,
|
204
|
+
InvalidOid,
|
205
|
+
NULL),
|
206
|
+
true);
|
207
|
+
function->found_varno = var->dno;
|
208
|
+
|
209
|
+
if (is_trigger) {
|
210
|
+
/* Add the record for referencing NEW */
|
211
|
+
rec = plpgsql_build_record("new", 0, NULL, RECORDOID, true);
|
212
|
+
function->new_varno = rec->dno;
|
213
|
+
|
214
|
+
/* Add the record for referencing OLD */
|
215
|
+
rec = plpgsql_build_record("old", 0, NULL, RECORDOID, true);
|
216
|
+
function->old_varno = rec->dno;
|
217
|
+
}
|
218
|
+
|
219
|
+
/*
|
220
|
+
* Now parse the function's text
|
221
|
+
*/
|
222
|
+
parse_rc = plpgsql_yyparse();
|
223
|
+
if (parse_rc != 0)
|
224
|
+
elog(ERROR, "plpgsql parser returned %d", parse_rc);
|
225
|
+
function->action = plpgsql_parse_result;
|
226
|
+
|
227
|
+
plpgsql_scanner_finish();
|
228
|
+
|
229
|
+
/*
|
230
|
+
* If it returns VOID (always true at the moment), we allow control to
|
231
|
+
* fall off the end without an explicit RETURN statement.
|
232
|
+
*/
|
233
|
+
if (function->fn_rettype == VOIDOID)
|
234
|
+
add_dummy_return(function);
|
235
|
+
|
236
|
+
/*
|
237
|
+
* Complete the function's info
|
238
|
+
*/
|
239
|
+
function->fn_nargs = 0;
|
240
|
+
|
241
|
+
plpgsql_finish_datums(function);
|
242
|
+
|
243
|
+
/*
|
244
|
+
* Pop the error context stack
|
245
|
+
*/
|
246
|
+
error_context_stack = plerrcontext.previous;
|
247
|
+
plpgsql_error_funcname = NULL;
|
248
|
+
|
249
|
+
plpgsql_check_syntax = false;
|
250
|
+
|
251
|
+
MemoryContextSwitchTo(plpgsql_compile_tmp_cxt);
|
252
|
+
plpgsql_compile_tmp_cxt = NULL;
|
253
|
+
return function;
|
254
|
+
}
|
255
|
+
|
256
|
+
PgQueryInternalPlpgsqlFuncAndError pg_query_raw_parse_plpgsql(CreateFunctionStmt* stmt)
|
257
|
+
{
|
258
|
+
PgQueryInternalPlpgsqlFuncAndError result = {0};
|
259
|
+
MemoryContext cctx = CurrentMemoryContext;
|
260
|
+
|
261
|
+
char stderr_buffer[STDERR_BUFFER_LEN + 1] = {0};
|
262
|
+
#ifndef DEBUG
|
263
|
+
int stderr_global;
|
264
|
+
int stderr_pipe[2];
|
265
|
+
#endif
|
266
|
+
|
267
|
+
#ifndef DEBUG
|
268
|
+
// Setup pipe for stderr redirection
|
269
|
+
if (pipe(stderr_pipe) != 0) {
|
270
|
+
PgQueryError* error = malloc(sizeof(PgQueryError));
|
271
|
+
|
272
|
+
error->message = strdup("Failed to open pipe, too many open file descriptors")
|
273
|
+
|
274
|
+
result.error = error;
|
275
|
+
|
276
|
+
return result;
|
277
|
+
}
|
278
|
+
|
279
|
+
fcntl(stderr_pipe[0], F_SETFL, fcntl(stderr_pipe[0], F_GETFL) | O_NONBLOCK);
|
280
|
+
|
281
|
+
// Redirect stderr to the pipe
|
282
|
+
stderr_global = dup(STDERR_FILENO);
|
283
|
+
dup2(stderr_pipe[1], STDERR_FILENO);
|
284
|
+
close(stderr_pipe[1]);
|
285
|
+
#endif
|
286
|
+
|
287
|
+
PG_TRY();
|
288
|
+
{
|
289
|
+
result.func = compile_create_function_stmt(stmt);
|
290
|
+
|
291
|
+
#ifndef DEBUG
|
292
|
+
// Save stderr for result
|
293
|
+
read(stderr_pipe[0], stderr_buffer, STDERR_BUFFER_LEN);
|
294
|
+
#endif
|
295
|
+
|
296
|
+
if (strlen(stderr_buffer) > 0) {
|
297
|
+
PgQueryError* error = malloc(sizeof(PgQueryError));
|
298
|
+
error->message = strdup(stderr_buffer);
|
299
|
+
error->filename = "";
|
300
|
+
error->funcname = "";
|
301
|
+
error->context = "";
|
302
|
+
result.error = error;
|
303
|
+
}
|
304
|
+
}
|
305
|
+
PG_CATCH();
|
306
|
+
{
|
307
|
+
ErrorData* error_data;
|
308
|
+
PgQueryError* error;
|
309
|
+
|
310
|
+
MemoryContextSwitchTo(cctx);
|
311
|
+
error_data = CopyErrorData();
|
312
|
+
|
313
|
+
// Note: This is intentionally malloc so exiting the memory context doesn't free this
|
314
|
+
error = malloc(sizeof(PgQueryError));
|
315
|
+
error->message = strdup(error_data->message);
|
316
|
+
error->filename = strdup(error_data->filename);
|
317
|
+
error->funcname = strdup(error_data->funcname);
|
318
|
+
error->context = strdup(error_data->context);
|
319
|
+
error->lineno = error_data->lineno;
|
320
|
+
error->cursorpos = error_data->cursorpos;
|
321
|
+
|
322
|
+
result.error = error;
|
323
|
+
FlushErrorState();
|
324
|
+
}
|
325
|
+
PG_END_TRY();
|
326
|
+
|
327
|
+
#ifndef DEBUG
|
328
|
+
// Restore stderr, close pipe
|
329
|
+
dup2(stderr_global, STDERR_FILENO);
|
330
|
+
close(stderr_pipe[0]);
|
331
|
+
close(stderr_global);
|
332
|
+
#endif
|
333
|
+
|
334
|
+
return result;
|
335
|
+
}
|
336
|
+
|
337
|
+
typedef struct createFunctionStmts
|
338
|
+
{
|
339
|
+
CreateFunctionStmt **stmts;
|
340
|
+
int stmts_buf_size;
|
341
|
+
int stmts_count;
|
342
|
+
} createFunctionStmts;
|
343
|
+
|
344
|
+
static bool create_function_stmts_walker(Node *node, createFunctionStmts *state)
|
345
|
+
{
|
346
|
+
bool result;
|
347
|
+
MemoryContext ccxt = CurrentMemoryContext;
|
348
|
+
|
349
|
+
if (node == NULL) return false;
|
350
|
+
|
351
|
+
if (IsA(node, CreateFunctionStmt))
|
352
|
+
{
|
353
|
+
if (state->stmts_count >= state->stmts_buf_size)
|
354
|
+
{
|
355
|
+
state->stmts_buf_size *= 2;
|
356
|
+
state->stmts = (CreateFunctionStmt**) repalloc(state->stmts, state->stmts_buf_size * sizeof(CreateFunctionStmt*));
|
357
|
+
}
|
358
|
+
state->stmts[state->stmts_count] = (CreateFunctionStmt *) node;
|
359
|
+
state->stmts_count++;
|
360
|
+
} else if (IsA(node, RawStmt)) {
|
361
|
+
return create_function_stmts_walker((Node *) ((RawStmt *) node)->stmt, state);
|
362
|
+
}
|
363
|
+
|
364
|
+
PG_TRY();
|
365
|
+
{
|
366
|
+
result = raw_expression_tree_walker(node, create_function_stmts_walker, (void*) state);
|
367
|
+
}
|
368
|
+
PG_CATCH();
|
369
|
+
{
|
370
|
+
MemoryContextSwitchTo(ccxt);
|
371
|
+
FlushErrorState();
|
372
|
+
result = false;
|
373
|
+
}
|
374
|
+
PG_END_TRY();
|
375
|
+
|
376
|
+
return result;
|
377
|
+
}
|
378
|
+
|
379
|
+
PgQueryPlpgsqlParseResult pg_query_parse_plpgsql(const char* input)
|
380
|
+
{
|
381
|
+
MemoryContext ctx = NULL;
|
382
|
+
PgQueryPlpgsqlParseResult result = {0};
|
383
|
+
PgQueryInternalParsetreeAndError parse_result;
|
384
|
+
createFunctionStmts statements;
|
385
|
+
size_t i;
|
386
|
+
|
387
|
+
ctx = pg_query_enter_memory_context();
|
388
|
+
|
389
|
+
parse_result = pg_query_raw_parse(input);
|
390
|
+
result.error = parse_result.error;
|
391
|
+
if (result.error != NULL) {
|
392
|
+
pg_query_exit_memory_context(ctx);
|
393
|
+
return result;
|
394
|
+
}
|
395
|
+
|
396
|
+
statements.stmts_buf_size = 100;
|
397
|
+
statements.stmts = (CreateFunctionStmt**) palloc(statements.stmts_buf_size * sizeof(CreateFunctionStmt*));
|
398
|
+
statements.stmts_count = 0;
|
399
|
+
|
400
|
+
create_function_stmts_walker((Node*) parse_result.tree, &statements);
|
401
|
+
|
402
|
+
if (statements.stmts_count == 0) {
|
403
|
+
result.plpgsql_funcs = strdup("[]");
|
404
|
+
pg_query_exit_memory_context(ctx);
|
405
|
+
return result;
|
406
|
+
}
|
407
|
+
|
408
|
+
result.plpgsql_funcs = strdup("[\n");
|
409
|
+
|
410
|
+
for (i = 0; i < statements.stmts_count; i++) {
|
411
|
+
PgQueryInternalPlpgsqlFuncAndError func_and_error;
|
412
|
+
|
413
|
+
func_and_error = pg_query_raw_parse_plpgsql(statements.stmts[i]);
|
414
|
+
|
415
|
+
// These are all malloc-ed and will survive exiting the memory context, the caller is responsible to free them now
|
416
|
+
result.error = func_and_error.error;
|
417
|
+
|
418
|
+
if (result.error != NULL) {
|
419
|
+
pg_query_exit_memory_context(ctx);
|
420
|
+
return result;
|
421
|
+
}
|
422
|
+
|
423
|
+
if (func_and_error.func != NULL) {
|
424
|
+
char *func_json;
|
425
|
+
char *new_out;
|
426
|
+
|
427
|
+
func_json = plpgsqlToJSON(func_and_error.func);
|
428
|
+
plpgsql_free_function_memory(func_and_error.func);
|
429
|
+
|
430
|
+
int err = asprintf(&new_out, "%s%s,\n", result.plpgsql_funcs, func_json);
|
431
|
+
if (err == -1) {
|
432
|
+
PgQueryError* error = malloc(sizeof(PgQueryError));
|
433
|
+
error->message = strdup("Failed to output PL/pgSQL functions due to asprintf failure");
|
434
|
+
result.error = error;
|
435
|
+
} else {
|
436
|
+
free(result.plpgsql_funcs);
|
437
|
+
result.plpgsql_funcs = new_out;
|
438
|
+
}
|
439
|
+
|
440
|
+
pfree(func_json);
|
441
|
+
}
|
442
|
+
}
|
443
|
+
|
444
|
+
result.plpgsql_funcs[strlen(result.plpgsql_funcs) - 2] = '\n';
|
445
|
+
result.plpgsql_funcs[strlen(result.plpgsql_funcs) - 1] = ']';
|
446
|
+
|
447
|
+
free(parse_result.stderr_buffer);
|
448
|
+
pg_query_exit_memory_context(ctx);
|
449
|
+
|
450
|
+
return result;
|
451
|
+
}
|
452
|
+
|
453
|
+
void pg_query_free_plpgsql_parse_result(PgQueryPlpgsqlParseResult result)
|
454
|
+
{
|
455
|
+
if (result.error) {
|
456
|
+
pg_query_free_error(result.error);
|
457
|
+
}
|
458
|
+
|
459
|
+
free(result.plpgsql_funcs);
|
460
|
+
}
|