Branch data Line data Source code
1 : : /*-------------------------------------------------------------------------
2 : : *
3 : : * copy.c
4 : : * Implements the COPY utility command
5 : : *
6 : : * Portions Copyright (c) 1996-2026, PostgreSQL Global Development Group
7 : : * Portions Copyright (c) 1994, Regents of the University of California
8 : : *
9 : : *
10 : : * IDENTIFICATION
11 : : * src/backend/commands/copy.c
12 : : *
13 : : *-------------------------------------------------------------------------
14 : : */
15 : : #include "postgres.h"
16 : :
17 : : #include <ctype.h>
18 : : #include <unistd.h>
19 : : #include <sys/stat.h>
20 : :
21 : : #include "access/sysattr.h"
22 : : #include "access/table.h"
23 : : #include "access/xact.h"
24 : : #include "catalog/pg_authid.h"
25 : : #include "commands/copy.h"
26 : : #include "commands/defrem.h"
27 : : #include "executor/executor.h"
28 : : #include "mb/pg_wchar.h"
29 : : #include "miscadmin.h"
30 : : #include "nodes/makefuncs.h"
31 : : #include "nodes/miscnodes.h"
32 : : #include "optimizer/optimizer.h"
33 : : #include "parser/parse_coerce.h"
34 : : #include "parser/parse_collate.h"
35 : : #include "parser/parse_expr.h"
36 : : #include "parser/parse_relation.h"
37 : : #include "utils/acl.h"
38 : : #include "utils/builtins.h"
39 : : #include "utils/lsyscache.h"
40 : : #include "utils/rel.h"
41 : : #include "utils/rls.h"
42 : :
43 : : /*
44 : : * DoCopy executes the SQL COPY statement
45 : : *
46 : : * Either unload or reload contents of table <relation>, depending on <from>.
47 : : * (<from> = true means we are inserting into the table.) In the "TO" case
48 : : * we also support copying the output of an arbitrary SELECT, INSERT, UPDATE
49 : : * or DELETE query.
50 : : *
51 : : * If <pipe> is false, transfer is between the table and the file named
52 : : * <filename>. Otherwise, transfer is between the table and our regular
53 : : * input/output stream. The latter could be either stdin/stdout or a
54 : : * socket, depending on whether we're running under Postmaster control.
55 : : *
56 : : * Do not allow a Postgres user without the 'pg_read_server_files' or
57 : : * 'pg_write_server_files' role to read from or write to a file.
58 : : *
59 : : * Do not allow the copy if user doesn't have proper permission to access
60 : : * the table or the specifically requested columns.
61 : : */
62 : : void
63 : 6682 : DoCopy(ParseState *pstate, const CopyStmt *stmt,
64 : : int stmt_location, int stmt_len,
65 : : uint64 *processed)
66 : : {
67 : 6682 : bool is_from = stmt->is_from;
68 : 6682 : bool pipe = (stmt->filename == NULL);
69 : : Relation rel;
70 : : Oid relid;
71 : 6682 : RawStmt *query = NULL;
72 : 6682 : Node *whereClause = NULL;
73 : :
74 : : /*
75 : : * Disallow COPY to/from file or program except to users with the
76 : : * appropriate role.
77 : : */
78 [ + + ]: 6682 : if (!pipe)
79 : : {
80 [ - + ]: 310 : if (stmt->is_program)
81 : : {
82 [ # # ]: 0 : if (!has_privs_of_role(GetUserId(), ROLE_PG_EXECUTE_SERVER_PROGRAM))
83 [ # # ]: 0 : ereport(ERROR,
84 : : (errcode(ERRCODE_INSUFFICIENT_PRIVILEGE),
85 : : errmsg("permission denied to COPY to or from an external program"),
86 : : errdetail("Only roles with privileges of the \"%s\" role may COPY to or from an external program.",
87 : : "pg_execute_server_program"),
88 : : errhint("Anyone can COPY to stdout or from stdin. "
89 : : "psql's \\copy command also works for anyone.")));
90 : : }
91 : : else
92 : : {
93 [ + + - + ]: 310 : if (is_from && !has_privs_of_role(GetUserId(), ROLE_PG_READ_SERVER_FILES))
94 [ # # ]: 0 : ereport(ERROR,
95 : : (errcode(ERRCODE_INSUFFICIENT_PRIVILEGE),
96 : : errmsg("permission denied to COPY from a file"),
97 : : errdetail("Only roles with privileges of the \"%s\" role may COPY from a file.",
98 : : "pg_read_server_files"),
99 : : errhint("Anyone can COPY to stdout or from stdin. "
100 : : "psql's \\copy command also works for anyone.")));
101 : :
102 [ + + - + ]: 310 : if (!is_from && !has_privs_of_role(GetUserId(), ROLE_PG_WRITE_SERVER_FILES))
103 [ # # ]: 0 : ereport(ERROR,
104 : : (errcode(ERRCODE_INSUFFICIENT_PRIVILEGE),
105 : : errmsg("permission denied to COPY to a file"),
106 : : errdetail("Only roles with privileges of the \"%s\" role may COPY to a file.",
107 : : "pg_write_server_files"),
108 : : errhint("Anyone can COPY to stdout or from stdin. "
109 : : "psql's \\copy command also works for anyone.")));
110 : : }
111 : : }
112 : :
113 [ + + ]: 6682 : if (stmt->relation)
114 : : {
115 [ + + ]: 6323 : LOCKMODE lockmode = is_from ? RowExclusiveLock : AccessShareLock;
116 : : ParseNamespaceItem *nsitem;
117 : : RTEPermissionInfo *perminfo;
118 : : TupleDesc tupDesc;
119 : : List *attnums;
120 : : ListCell *cur;
121 : :
122 : : Assert(!stmt->query);
123 : :
124 : : /* Open and lock the relation, using the appropriate lock type. */
125 : 6323 : rel = table_openrv(stmt->relation, lockmode);
126 : :
127 : 6318 : relid = RelationGetRelid(rel);
128 : :
129 : 6318 : nsitem = addRangeTableEntryForRelation(pstate, rel, lockmode,
130 : : NULL, false, false);
131 : :
132 : 6318 : perminfo = nsitem->p_perminfo;
133 [ + + ]: 6318 : perminfo->requiredPerms = (is_from ? ACL_INSERT : ACL_SELECT);
134 : :
135 [ + + ]: 6318 : if (stmt->whereClause)
136 : : {
137 : 52 : Bitmapset *expr_attrs = NULL;
138 : : int i;
139 : :
140 : : /* Add nsitem to query namespace */
141 : 52 : addNSItemToQuery(pstate, nsitem, false, true, true);
142 : :
143 : : /* Transform the raw expression tree */
144 : 52 : whereClause = transformExpr(pstate, stmt->whereClause, EXPR_KIND_COPY_WHERE);
145 : :
146 : : /* Make sure it yields a boolean result */
147 : 32 : whereClause = coerce_to_boolean(pstate, whereClause, "WHERE");
148 : :
149 : : /* We have to fix its collations too */
150 : 32 : assign_expr_collations(pstate, whereClause);
151 : :
152 : : /*
153 : : * Identify all columns used in the WHERE clause's expression. If
154 : : * there's a whole-row reference, replace it with a range of all
155 : : * the user columns (caution: that'll include dropped columns).
156 : : */
157 : 32 : pull_varattnos(whereClause, 1, &expr_attrs);
158 [ + + ]: 32 : if (bms_is_member(0 - FirstLowInvalidHeapAttributeNumber, expr_attrs))
159 : : {
160 : 16 : expr_attrs = bms_add_range(expr_attrs,
161 : : 1 - FirstLowInvalidHeapAttributeNumber,
162 : 8 : RelationGetNumberOfAttributes(rel) - FirstLowInvalidHeapAttributeNumber);
163 : 8 : expr_attrs = bms_del_member(expr_attrs, 0 - FirstLowInvalidHeapAttributeNumber);
164 : : }
165 : :
166 : : /* Now we can scan each column needed in the WHERE clause */
167 : 32 : i = -1;
168 [ + + ]: 52 : while ((i = bms_next_member(expr_attrs, i)) >= 0)
169 : : {
170 : 40 : AttrNumber attno = i + FirstLowInvalidHeapAttributeNumber;
171 : : Form_pg_attribute att;
172 : :
173 : : Assert(attno != 0); /* removed above */
174 : :
175 : : /*
176 : : * Prohibit system columns in the WHERE clause. They won't
177 : : * have been filled yet when the filtering happens. (We could
178 : : * allow tableoid, but right now it isn't really useful: it
179 : : * will read as the target table's OID. Any conceivable use
180 : : * for such a WHERE clause would probably wish it to read as
181 : : * the target partition's OID, which is not known yet.
182 : : * Disallow it to keep flexibility to change that sometime.)
183 : : */
184 [ + + ]: 40 : if (attno < 0)
185 [ + - ]: 4 : ereport(ERROR,
186 : : errcode(ERRCODE_INVALID_COLUMN_REFERENCE),
187 : : errmsg("system columns are not supported in COPY FROM WHERE conditions"),
188 : : errdetail("Column \"%s\" is a system column.",
189 : : get_attname(RelationGetRelid(rel), attno, false)));
190 : :
191 : : /*
192 : : * Prohibit generated columns in the WHERE clause. Stored
193 : : * generated columns are not yet computed when the filtering
194 : : * happens. Virtual generated columns could probably work (we
195 : : * would need to expand them somewhere around here), but for
196 : : * now we keep them consistent with the stored variant.
197 : : */
198 : 36 : att = TupleDescAttr(RelationGetDescr(rel), attno - 1);
199 [ + + + - ]: 36 : if (att->attgenerated && !att->attisdropped)
200 [ + - ]: 16 : ereport(ERROR,
201 : : errcode(ERRCODE_INVALID_COLUMN_REFERENCE),
202 : : errmsg("generated columns are not supported in COPY FROM WHERE conditions"),
203 : : errdetail("Column \"%s\" is a generated column.",
204 : : get_attname(RelationGetRelid(rel), attno, false)));
205 : : }
206 : :
207 : : /* Reduce WHERE clause to standard list-of-AND-terms form */
208 : 12 : whereClause = eval_const_expressions(NULL, whereClause);
209 : :
210 : 12 : whereClause = (Node *) canonicalize_qual((Expr *) whereClause, false);
211 : 12 : whereClause = (Node *) make_ands_implicit((Expr *) whereClause);
212 : : }
213 : :
214 : 6278 : tupDesc = RelationGetDescr(rel);
215 : 6278 : attnums = CopyGetAttnums(tupDesc, rel, stmt->attlist);
216 [ + + + + : 27163 : foreach(cur, attnums)
+ + ]
217 : : {
218 : : int attno;
219 : : Bitmapset **bms;
220 : :
221 : 20941 : attno = lfirst_int(cur) - FirstLowInvalidHeapAttributeNumber;
222 [ + + ]: 20941 : bms = is_from ? &perminfo->insertedCols : &perminfo->selectedCols;
223 : :
224 : 20941 : *bms = bms_add_member(*bms, attno);
225 : : }
226 : 6222 : ExecCheckPermissions(pstate->p_rtable, list_make1(perminfo), true);
227 : :
228 : : /*
229 : : * Permission check for row security policies.
230 : : *
231 : : * check_enable_rls will ereport(ERROR) if the user has requested
232 : : * something invalid and will otherwise indicate if we should enable
233 : : * RLS (returns RLS_ENABLED) or not for this COPY statement.
234 : : *
235 : : * If the relation has a row security policy and we are to apply it
236 : : * then perform a "query" copy and allow the normal query processing
237 : : * to handle the policies.
238 : : *
239 : : * If RLS is not enabled for this, then just fall through to the
240 : : * normal non-filtering relation handling.
241 : : */
242 [ + + ]: 6166 : if (check_enable_rls(relid, InvalidOid, false) == RLS_ENABLED)
243 : : {
244 : : SelectStmt *select;
245 : : ColumnRef *cr;
246 : : ResTarget *target;
247 : : RangeVar *from;
248 : 56 : List *targetList = NIL;
249 : :
250 [ + + ]: 56 : if (is_from)
251 [ + - ]: 4 : ereport(ERROR,
252 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
253 : : errmsg("COPY FROM not supported with row-level security"),
254 : : errhint("Use INSERT statements instead.")));
255 : :
256 : : /*
257 : : * Build target list
258 : : *
259 : : * If no columns are specified in the attribute list of the COPY
260 : : * command, then the target list is 'all' columns. Therefore, '*'
261 : : * should be used as the target list for the resulting SELECT
262 : : * statement.
263 : : *
264 : : * In the case that columns are specified in the attribute list,
265 : : * create a ColumnRef and ResTarget for each column and add them
266 : : * to the target list for the resulting SELECT statement.
267 : : */
268 [ + + ]: 52 : if (!stmt->attlist)
269 : : {
270 : 28 : cr = makeNode(ColumnRef);
271 : 28 : cr->fields = list_make1(makeNode(A_Star));
272 : 28 : cr->location = -1;
273 : :
274 : 28 : target = makeNode(ResTarget);
275 : 28 : target->name = NULL;
276 : 28 : target->indirection = NIL;
277 : 28 : target->val = (Node *) cr;
278 : 28 : target->location = -1;
279 : :
280 : 28 : targetList = list_make1(target);
281 : : }
282 : : else
283 : : {
284 : : ListCell *lc;
285 : :
286 [ + - + + : 68 : foreach(lc, stmt->attlist)
+ + ]
287 : : {
288 : : /*
289 : : * Build the ColumnRef for each column. The ColumnRef
290 : : * 'fields' property is a String node that corresponds to
291 : : * the column name respectively.
292 : : */
293 : 44 : cr = makeNode(ColumnRef);
294 : 44 : cr->fields = list_make1(lfirst(lc));
295 : 44 : cr->location = -1;
296 : :
297 : : /* Build the ResTarget and add the ColumnRef to it. */
298 : 44 : target = makeNode(ResTarget);
299 : 44 : target->name = NULL;
300 : 44 : target->indirection = NIL;
301 : 44 : target->val = (Node *) cr;
302 : 44 : target->location = -1;
303 : :
304 : : /* Add each column to the SELECT statement's target list */
305 : 44 : targetList = lappend(targetList, target);
306 : : }
307 : : }
308 : :
309 : : /*
310 : : * Build RangeVar for from clause, fully qualified based on the
311 : : * relation which we have opened and locked. Use "ONLY" so that
312 : : * COPY retrieves rows from only the target table not any
313 : : * inheritance children, the same as when RLS doesn't apply.
314 : : *
315 : : * However, when copying data from a partitioned table, we don't
316 : : * use "ONLY", since we need to retrieve rows from its descendant
317 : : * tables too.
318 : : */
319 : 52 : from = makeRangeVar(get_namespace_name(RelationGetNamespace(rel)),
320 : 52 : pstrdup(RelationGetRelationName(rel)),
321 : : -1);
322 : 52 : from->inh = (rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE);
323 : :
324 : : /* Build query */
325 : 52 : select = makeNode(SelectStmt);
326 : 52 : select->targetList = targetList;
327 : 52 : select->fromClause = list_make1(from);
328 : :
329 : 52 : query = makeNode(RawStmt);
330 : 52 : query->stmt = (Node *) select;
331 : 52 : query->stmt_location = stmt_location;
332 : 52 : query->stmt_len = stmt_len;
333 : :
334 : : /*
335 : : * Close the relation for now, but keep the lock on it to prevent
336 : : * changes between now and when we start the query-based COPY.
337 : : *
338 : : * We'll reopen it later as part of the query-based COPY.
339 : : */
340 : 52 : table_close(rel, NoLock);
341 : 52 : rel = NULL;
342 : : }
343 : : }
344 : : else
345 : : {
346 : : Assert(stmt->query);
347 : :
348 : 359 : query = makeNode(RawStmt);
349 : 359 : query->stmt = stmt->query;
350 : 359 : query->stmt_location = stmt_location;
351 : 359 : query->stmt_len = stmt_len;
352 : :
353 : 359 : relid = InvalidOid;
354 : 359 : rel = NULL;
355 : : }
356 : :
357 [ + + ]: 6509 : if (is_from)
358 : : {
359 : : CopyFromState cstate;
360 : :
361 : : Assert(rel);
362 : :
363 : : /* check read-only transaction and parallel mode */
364 [ - + - - ]: 1134 : if (XactReadOnly && !rel->rd_islocaltemp)
365 : 0 : PreventCommandIfReadOnly("COPY FROM");
366 : :
367 : 1134 : cstate = BeginCopyFrom(pstate, rel, whereClause,
368 : 1134 : stmt->filename, stmt->is_program,
369 : 1134 : NULL, stmt->attlist, stmt->options);
370 : 938 : *processed = CopyFrom(cstate); /* copy from file to database */
371 : 758 : EndCopyFrom(cstate);
372 : : }
373 : : else
374 : : {
375 : : CopyToState cstate;
376 : :
377 : 5375 : cstate = BeginCopyTo(pstate, rel, query, relid,
378 : 5375 : stmt->filename, stmt->is_program,
379 : 5375 : NULL, stmt->attlist, stmt->options);
380 : 5177 : *processed = DoCopyTo(cstate); /* copy from database to file */
381 : 5175 : EndCopyTo(cstate);
382 : : }
383 : :
384 [ + + ]: 5933 : if (rel != NULL)
385 : 5615 : table_close(rel, NoLock);
386 : 5933 : }
387 : :
388 : : /*
389 : : * Extract the CopyFormatOptions.header_line value from a DefElem.
390 : : *
391 : : * Parses the HEADER option for COPY, which can be a boolean, an integer greater
392 : : * than or equal to zero (number of lines to skip), or the special value
393 : : * "match".
394 : : */
395 : : static int
396 : 184 : defGetCopyHeaderOption(DefElem *def, bool is_from)
397 : : {
398 : 184 : int ival = COPY_HEADER_FALSE;
399 : :
400 : : /*
401 : : * If no parameter value given, assume "true" is meant.
402 : : */
403 [ + + ]: 184 : if (def->arg == NULL)
404 : 28 : return COPY_HEADER_TRUE;
405 : :
406 : : /*
407 : : * Allow an integer value greater than or equal to zero (integers
408 : : * specified as strings are also accepted, mainly for file_fdw foreign
409 : : * table options), "true", "false", "on", "off", or "match".
410 : : */
411 [ + + ]: 156 : switch (nodeTag(def->arg))
412 : : {
413 : 24 : case T_Integer:
414 : 24 : ival = intVal(def->arg);
415 : 24 : break;
416 : 132 : default:
417 : : {
418 : 132 : char *sval = defGetString(def);
419 : :
420 : : /*
421 : : * The set of strings accepted here should match up with the
422 : : * grammar's opt_boolean_or_string production.
423 : : */
424 [ + + ]: 132 : if (pg_strcasecmp(sval, "true") == 0)
425 : 37 : return COPY_HEADER_TRUE;
426 [ - + ]: 95 : if (pg_strcasecmp(sval, "false") == 0)
427 : 0 : return COPY_HEADER_FALSE;
428 [ - + ]: 95 : if (pg_strcasecmp(sval, "on") == 0)
429 : 0 : return COPY_HEADER_TRUE;
430 [ + + ]: 95 : if (pg_strcasecmp(sval, "off") == 0)
431 : 4 : return COPY_HEADER_FALSE;
432 [ + + ]: 91 : if (pg_strcasecmp(sval, "match") == 0)
433 : : {
434 [ + + ]: 56 : if (!is_from)
435 [ + - ]: 4 : ereport(ERROR,
436 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
437 : : errmsg("cannot use \"%s\" with HEADER in COPY TO",
438 : : sval)));
439 : 52 : return COPY_HEADER_MATCH;
440 : : }
441 : : else
442 : : {
443 : 35 : ErrorSaveContext escontext = {T_ErrorSaveContext};
444 : :
445 : : /* Check if the header is a valid integer */
446 : 35 : ival = pg_strtoint32_safe(sval, (Node *) &escontext);
447 [ + + ]: 35 : if (escontext.error_occurred)
448 [ + - ]: 14 : ereport(ERROR,
449 : : (errcode(ERRCODE_SYNTAX_ERROR),
450 : : /*- translator: first %s is the name of a COPY option, e.g. ON_ERROR,
451 : : second %s is the special value "match" for that option */
452 : : errmsg("%s requires a Boolean value, an integer "
453 : : "value greater than or equal to zero, "
454 : : "or the string \"%s\"",
455 : : def->defname, "match")));
456 : : }
457 : : }
458 : 21 : break;
459 : : }
460 : :
461 [ + + ]: 45 : if (ival < 0)
462 [ + - ]: 9 : ereport(ERROR,
463 : : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
464 : : errmsg("a negative integer value cannot be "
465 : : "specified for %s", def->defname)));
466 : :
467 [ + + + + ]: 36 : if (!is_from && ival > 1)
468 [ + - ]: 8 : ereport(ERROR,
469 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
470 : : errmsg("cannot use multi-line header in COPY TO")));
471 : :
472 : 28 : return ival;
473 : : }
474 : :
475 : : /*
476 : : * Extract a CopyOnErrorChoice value from a DefElem.
477 : : */
478 : : static CopyOnErrorChoice
479 : 122 : defGetCopyOnErrorChoice(DefElem *def, ParseState *pstate, bool is_from)
480 : : {
481 : 122 : char *sval = defGetString(def);
482 : :
483 [ + + ]: 122 : if (!is_from)
484 [ + - ]: 12 : ereport(ERROR,
485 : : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
486 : : /*- translator: first %s is the name of a COPY option, e.g. ON_ERROR,
487 : : second %s is a COPY with direction, e.g. COPY TO */
488 : : errmsg("COPY %s cannot be used with %s", "ON_ERROR", "COPY TO"),
489 : : parser_errposition(pstate, def->location)));
490 : :
491 [ + + ]: 110 : if (pg_strcasecmp(sval, "stop") == 0)
492 : 4 : return COPY_ON_ERROR_STOP;
493 [ + + ]: 106 : if (pg_strcasecmp(sval, "ignore") == 0)
494 : 59 : return COPY_ON_ERROR_IGNORE;
495 [ + + ]: 47 : if (pg_strcasecmp(sval, "set_null") == 0)
496 : 42 : return COPY_ON_ERROR_SET_NULL;
497 : :
498 [ + - ]: 5 : ereport(ERROR,
499 : : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
500 : : /*- translator: first %s is the name of a COPY option, e.g. ON_ERROR */
501 : : errmsg("COPY %s \"%s\" not recognized", "ON_ERROR", sval),
502 : : parser_errposition(pstate, def->location)));
503 : : return COPY_ON_ERROR_STOP; /* keep compiler quiet */
504 : : }
505 : :
506 : : /*
507 : : * Extract REJECT_LIMIT value from a DefElem.
508 : : *
509 : : * REJECT_LIMIT can be specified in two ways: as an int64 for the COPY command
510 : : * option or as a single-quoted string for the foreign table option using
511 : : * file_fdw. Therefore this function needs to handle both formats.
512 : : */
513 : : static int64
514 : 31 : defGetCopyRejectLimitOption(DefElem *def)
515 : : {
516 : : int64 reject_limit;
517 : :
518 [ - + ]: 31 : if (def->arg == NULL)
519 [ # # ]: 0 : ereport(ERROR,
520 : : (errcode(ERRCODE_SYNTAX_ERROR),
521 : : errmsg("%s requires a numeric value",
522 : : def->defname)));
523 [ + + ]: 31 : else if (IsA(def->arg, String))
524 : 7 : reject_limit = pg_strtoint64(strVal(def->arg));
525 : : else
526 : 24 : reject_limit = defGetInt64(def);
527 : :
528 [ + + ]: 31 : if (reject_limit <= 0)
529 [ + - ]: 5 : ereport(ERROR,
530 : : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
531 : : errmsg("REJECT_LIMIT (%" PRId64 ") must be greater than zero",
532 : : reject_limit)));
533 : :
534 : 26 : return reject_limit;
535 : : }
536 : :
537 : : /*
538 : : * Extract a CopyLogVerbosityChoice value from a DefElem.
539 : : */
540 : : static CopyLogVerbosityChoice
541 : 32 : defGetCopyLogVerbosityChoice(DefElem *def, ParseState *pstate)
542 : : {
543 : : char *sval;
544 : :
545 : : /*
546 : : * Allow "silent", "default", or "verbose" values.
547 : : */
548 : 32 : sval = defGetString(def);
549 [ + + ]: 32 : if (pg_strcasecmp(sval, "silent") == 0)
550 : 11 : return COPY_LOG_VERBOSITY_SILENT;
551 [ + + ]: 21 : if (pg_strcasecmp(sval, "default") == 0)
552 : 4 : return COPY_LOG_VERBOSITY_DEFAULT;
553 [ + + ]: 17 : if (pg_strcasecmp(sval, "verbose") == 0)
554 : 12 : return COPY_LOG_VERBOSITY_VERBOSE;
555 : :
556 [ + - ]: 5 : ereport(ERROR,
557 : : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
558 : : /*- translator: first %s is the name of a COPY option, e.g. ON_ERROR */
559 : : errmsg("COPY %s \"%s\" not recognized", "LOG_VERBOSITY", sval),
560 : : parser_errposition(pstate, def->location)));
561 : : return COPY_LOG_VERBOSITY_DEFAULT; /* keep compiler quiet */
562 : : }
563 : :
564 : : /*
565 : : * Process the statement option list for COPY.
566 : : *
567 : : * Scan the options list (a list of DefElem) and transpose the information
568 : : * into *opts_out, applying appropriate error checking.
569 : : *
570 : : * If 'opts_out' is not NULL, it is assumed to be filled with zeroes initially.
571 : : *
572 : : * This is exported so that external users of the COPY API can sanity-check
573 : : * a list of options. In that usage, 'opts_out' can be passed as NULL and
574 : : * the collected data is just leaked until CurrentMemoryContext is reset.
575 : : *
576 : : * Note that additional checking, such as whether column names listed in FORCE
577 : : * QUOTE actually exist, has to be applied later. This just checks for
578 : : * self-consistency of the options list.
579 : : */
580 : : void
581 : 6803 : ProcessCopyOptions(ParseState *pstate,
582 : : CopyFormatOptions *opts_out,
583 : : bool is_from,
584 : : List *options)
585 : : {
586 : 6803 : bool format_specified = false;
587 : 6803 : bool freeze_specified = false;
588 : 6803 : bool header_specified = false;
589 : 6803 : bool on_error_specified = false;
590 : 6803 : bool log_verbosity_specified = false;
591 : 6803 : bool reject_limit_specified = false;
592 : 6803 : bool force_array_specified = false;
593 : : ListCell *option;
594 : :
595 : : /* Support external use for option sanity checking */
596 [ + + ]: 6803 : if (opts_out == NULL)
597 : 58 : opts_out = palloc0_object(CopyFormatOptions);
598 : :
599 : 6803 : opts_out->file_encoding = -1;
600 : : /* default format */
601 : 6803 : opts_out->format = COPY_FORMAT_TEXT;
602 : :
603 : : /* Extract options from the statement node tree */
604 [ + + + + : 8461 : foreach(option, options)
+ + ]
605 : : {
606 : 1789 : DefElem *defel = lfirst_node(DefElem, option);
607 : :
608 [ + + ]: 1789 : if (strcmp(defel->defname, "format") == 0)
609 : : {
610 : 607 : char *fmt = defGetString(defel);
611 : :
612 [ + + ]: 607 : if (format_specified)
613 : 4 : errorConflictingDefElem(defel, pstate);
614 : 603 : format_specified = true;
615 [ + + ]: 603 : if (strcmp(fmt, "text") == 0)
616 : 52 : opts_out->format = COPY_FORMAT_TEXT;
617 [ + + ]: 551 : else if (strcmp(fmt, "csv") == 0)
618 : 348 : opts_out->format = COPY_FORMAT_CSV;
619 [ + + ]: 203 : else if (strcmp(fmt, "binary") == 0)
620 : 46 : opts_out->format = COPY_FORMAT_BINARY;
621 [ + + ]: 157 : else if (strcmp(fmt, "json") == 0)
622 : 156 : opts_out->format = COPY_FORMAT_JSON;
623 : : else
624 [ + - ]: 1 : ereport(ERROR,
625 : : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
626 : : errmsg("COPY format \"%s\" not recognized", fmt),
627 : : parser_errposition(pstate, defel->location)));
628 : : }
629 [ + + ]: 1182 : else if (strcmp(defel->defname, "freeze") == 0)
630 : : {
631 [ + + ]: 53 : if (freeze_specified)
632 : 4 : errorConflictingDefElem(defel, pstate);
633 : 49 : freeze_specified = true;
634 : 49 : opts_out->freeze = defGetBoolean(defel);
635 : : }
636 [ + + ]: 1129 : else if (strcmp(defel->defname, "delimiter") == 0)
637 : : {
638 [ + + ]: 196 : if (opts_out->delim)
639 : 4 : errorConflictingDefElem(defel, pstate);
640 : 192 : opts_out->delim = defGetString(defel);
641 : : }
642 [ + + ]: 933 : else if (strcmp(defel->defname, "null") == 0)
643 : : {
644 [ + + ]: 89 : if (opts_out->null_print)
645 : 4 : errorConflictingDefElem(defel, pstate);
646 : 85 : opts_out->null_print = defGetString(defel);
647 : : }
648 [ + + ]: 844 : else if (strcmp(defel->defname, "default") == 0)
649 : : {
650 [ - + ]: 63 : if (opts_out->default_print)
651 : 0 : errorConflictingDefElem(defel, pstate);
652 : 63 : opts_out->default_print = defGetString(defel);
653 : : }
654 [ + + ]: 781 : else if (strcmp(defel->defname, "header") == 0)
655 : : {
656 [ + + ]: 188 : if (header_specified)
657 : 4 : errorConflictingDefElem(defel, pstate);
658 : 184 : header_specified = true;
659 : 184 : opts_out->header_line = defGetCopyHeaderOption(defel, is_from);
660 : : }
661 [ + + ]: 593 : else if (strcmp(defel->defname, "quote") == 0)
662 : : {
663 [ + + ]: 65 : if (opts_out->quote)
664 : 4 : errorConflictingDefElem(defel, pstate);
665 : 61 : opts_out->quote = defGetString(defel);
666 : : }
667 [ + + ]: 528 : else if (strcmp(defel->defname, "escape") == 0)
668 : : {
669 [ + + ]: 60 : if (opts_out->escape)
670 : 4 : errorConflictingDefElem(defel, pstate);
671 : 56 : opts_out->escape = defGetString(defel);
672 : : }
673 [ + + ]: 468 : else if (strcmp(defel->defname, "force_quote") == 0)
674 : : {
675 [ + + - + ]: 56 : if (opts_out->force_quote || opts_out->force_quote_all)
676 : 4 : errorConflictingDefElem(defel, pstate);
677 [ + - + + ]: 52 : if (defel->arg && IsA(defel->arg, A_Star))
678 : 24 : opts_out->force_quote_all = true;
679 [ + - + - ]: 28 : else if (defel->arg && IsA(defel->arg, List))
680 : 28 : opts_out->force_quote = castNode(List, defel->arg);
681 : : else
682 [ # # ]: 0 : ereport(ERROR,
683 : : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
684 : : errmsg("argument to option \"%s\" must be a list of column names",
685 : : defel->defname),
686 : : parser_errposition(pstate, defel->location)));
687 : : }
688 [ + + ]: 412 : else if (strcmp(defel->defname, "force_not_null") == 0)
689 : : {
690 [ + + + + ]: 62 : if (opts_out->force_notnull || opts_out->force_notnull_all)
691 : 8 : errorConflictingDefElem(defel, pstate);
692 [ + - + + ]: 54 : if (defel->arg && IsA(defel->arg, A_Star))
693 : 24 : opts_out->force_notnull_all = true;
694 [ + - + - ]: 30 : else if (defel->arg && IsA(defel->arg, List))
695 : 30 : opts_out->force_notnull = castNode(List, defel->arg);
696 : : else
697 [ # # ]: 0 : ereport(ERROR,
698 : : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
699 : : errmsg("argument to option \"%s\" must be a list of column names",
700 : : defel->defname),
701 : : parser_errposition(pstate, defel->location)));
702 : : }
703 [ + + ]: 350 : else if (strcmp(defel->defname, "force_null") == 0)
704 : : {
705 [ + + + + ]: 62 : if (opts_out->force_null || opts_out->force_null_all)
706 : 8 : errorConflictingDefElem(defel, pstate);
707 [ + - + + ]: 54 : if (defel->arg && IsA(defel->arg, A_Star))
708 : 24 : opts_out->force_null_all = true;
709 [ + - + - ]: 30 : else if (defel->arg && IsA(defel->arg, List))
710 : 30 : opts_out->force_null = castNode(List, defel->arg);
711 : : else
712 [ # # ]: 0 : ereport(ERROR,
713 : : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
714 : : errmsg("argument to option \"%s\" must be a list of column names",
715 : : defel->defname),
716 : : parser_errposition(pstate, defel->location)));
717 : : }
718 [ + + ]: 288 : else if (strcmp(defel->defname, "convert_selectively") == 0)
719 : : {
720 : : /*
721 : : * Undocumented, not-accessible-from-SQL option: convert only the
722 : : * named columns to binary form, storing the rest as NULLs. It's
723 : : * allowed for the column list to be NIL.
724 : : */
725 [ + + ]: 11 : if (opts_out->convert_selectively)
726 : 4 : errorConflictingDefElem(defel, pstate);
727 : 7 : opts_out->convert_selectively = true;
728 [ + + + - ]: 7 : if (defel->arg == NULL || IsA(defel->arg, List))
729 : 7 : opts_out->convert_select = castNode(List, defel->arg);
730 : : else
731 [ # # ]: 0 : ereport(ERROR,
732 : : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
733 : : errmsg("argument to option \"%s\" must be a list of column names",
734 : : defel->defname),
735 : : parser_errposition(pstate, defel->location)));
736 : : }
737 [ + + ]: 277 : else if (strcmp(defel->defname, "encoding") == 0)
738 : : {
739 [ + + ]: 52 : if (opts_out->file_encoding >= 0)
740 : 4 : errorConflictingDefElem(defel, pstate);
741 : 48 : opts_out->file_encoding = pg_char_to_encoding(defGetString(defel));
742 [ - + ]: 48 : if (opts_out->file_encoding < 0)
743 [ # # ]: 0 : ereport(ERROR,
744 : : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
745 : : errmsg("argument to option \"%s\" must be a valid encoding name",
746 : : defel->defname),
747 : : parser_errposition(pstate, defel->location)));
748 : : }
749 [ + + ]: 225 : else if (strcmp(defel->defname, "force_array") == 0)
750 : : {
751 [ - + ]: 28 : if (force_array_specified)
752 : 0 : errorConflictingDefElem(defel, pstate);
753 : 28 : force_array_specified = true;
754 : 28 : opts_out->force_array = defGetBoolean(defel);
755 : : }
756 [ + + ]: 197 : else if (strcmp(defel->defname, "on_error") == 0)
757 : : {
758 [ + + ]: 130 : if (on_error_specified)
759 : 8 : errorConflictingDefElem(defel, pstate);
760 : 122 : on_error_specified = true;
761 : 122 : opts_out->on_error = defGetCopyOnErrorChoice(defel, pstate, is_from);
762 : : }
763 [ + + ]: 67 : else if (strcmp(defel->defname, "log_verbosity") == 0)
764 : : {
765 [ + + ]: 36 : if (log_verbosity_specified)
766 : 4 : errorConflictingDefElem(defel, pstate);
767 : 32 : log_verbosity_specified = true;
768 : 32 : opts_out->log_verbosity = defGetCopyLogVerbosityChoice(defel, pstate);
769 : : }
770 [ + - ]: 31 : else if (strcmp(defel->defname, "reject_limit") == 0)
771 : : {
772 [ - + ]: 31 : if (reject_limit_specified)
773 : 0 : errorConflictingDefElem(defel, pstate);
774 : 31 : reject_limit_specified = true;
775 : 31 : opts_out->reject_limit = defGetCopyRejectLimitOption(defel);
776 : : }
777 : : else
778 [ # # ]: 0 : ereport(ERROR,
779 : : (errcode(ERRCODE_SYNTAX_ERROR),
780 : : errmsg("option \"%s\" not recognized",
781 : : defel->defname),
782 : : parser_errposition(pstate, defel->location)));
783 : : }
784 : :
785 : : /*
786 : : * Check for incompatible options (must do these three before inserting
787 : : * defaults)
788 : : */
789 [ + + ]: 6672 : if (opts_out->delim &&
790 [ + + ]: 188 : (opts_out->format == COPY_FORMAT_BINARY ||
791 [ + + ]: 184 : opts_out->format == COPY_FORMAT_JSON))
792 [ + - + + ]: 8 : ereport(ERROR,
793 : : errcode(ERRCODE_SYNTAX_ERROR),
794 : : opts_out->format == COPY_FORMAT_BINARY
795 : : ? errmsg("cannot specify %s in BINARY mode", "DELIMITER")
796 : : : errmsg("cannot specify %s in JSON mode", "DELIMITER"));
797 : :
798 [ + + ]: 6664 : if (opts_out->null_print &&
799 [ + + ]: 81 : (opts_out->format == COPY_FORMAT_BINARY ||
800 [ + + ]: 77 : opts_out->format == COPY_FORMAT_JSON))
801 [ + - + + ]: 8 : ereport(ERROR,
802 : : errcode(ERRCODE_SYNTAX_ERROR),
803 : : opts_out->format == COPY_FORMAT_BINARY
804 : : ? errmsg("cannot specify %s in BINARY mode", "NULL")
805 : : : errmsg("cannot specify %s in JSON mode", "NULL"));
806 : :
807 [ + + ]: 6656 : if (opts_out->default_print &&
808 [ + + ]: 63 : (opts_out->format == COPY_FORMAT_BINARY ||
809 [ + + ]: 59 : opts_out->format == COPY_FORMAT_JSON))
810 [ + - + + ]: 8 : ereport(ERROR,
811 : : errcode(ERRCODE_SYNTAX_ERROR),
812 : : opts_out->format == COPY_FORMAT_BINARY
813 : : ? errmsg("cannot specify %s in BINARY mode", "DEFAULT")
814 : : : errmsg("cannot specify %s in JSON mode", "DEFAULT"));
815 : :
816 : : /* Set defaults for omitted options */
817 [ + + ]: 6648 : if (!opts_out->delim)
818 [ + + ]: 6468 : opts_out->delim = (opts_out->format == COPY_FORMAT_CSV) ? "," : "\t";
819 : :
820 [ + + ]: 6648 : if (!opts_out->null_print)
821 [ + + ]: 6575 : opts_out->null_print = (opts_out->format == COPY_FORMAT_CSV) ? "" : "\\N";
822 : 6648 : opts_out->null_print_len = strlen(opts_out->null_print);
823 : :
824 [ + + ]: 6648 : if (opts_out->format == COPY_FORMAT_CSV)
825 : : {
826 [ + + ]: 336 : if (!opts_out->quote)
827 : 285 : opts_out->quote = "\"";
828 [ + + ]: 336 : if (!opts_out->escape)
829 : 291 : opts_out->escape = opts_out->quote;
830 : : }
831 : :
832 : : /* Only single-byte delimiter strings are supported. */
833 [ + + ]: 6648 : if (strlen(opts_out->delim) != 1)
834 [ + - ]: 1 : ereport(ERROR,
835 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
836 : : errmsg("COPY delimiter must be a single one-byte character")));
837 : :
838 : : /* Disallow end-of-line characters */
839 [ + - ]: 6647 : if (strchr(opts_out->delim, '\r') != NULL ||
840 [ + + ]: 6647 : strchr(opts_out->delim, '\n') != NULL)
841 [ + - ]: 1 : ereport(ERROR,
842 : : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
843 : : errmsg("COPY delimiter cannot be newline or carriage return")));
844 : :
845 [ + - ]: 6646 : if (strchr(opts_out->null_print, '\r') != NULL ||
846 [ + + ]: 6646 : strchr(opts_out->null_print, '\n') != NULL)
847 [ + - ]: 1 : ereport(ERROR,
848 : : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
849 : : errmsg("COPY null representation cannot use newline or carriage return")));
850 : :
851 [ + + ]: 6645 : if (opts_out->default_print)
852 : : {
853 : 55 : opts_out->default_print_len = strlen(opts_out->default_print);
854 : :
855 [ + + ]: 55 : if (strchr(opts_out->default_print, '\r') != NULL ||
856 [ + + ]: 51 : strchr(opts_out->default_print, '\n') != NULL)
857 [ + - ]: 8 : ereport(ERROR,
858 : : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
859 : : errmsg("COPY default representation cannot use newline or carriage return")));
860 : : }
861 : :
862 : : /*
863 : : * Disallow unsafe delimiter characters in non-CSV mode. We can't allow
864 : : * backslash because it would be ambiguous. We can't allow the other
865 : : * cases because data characters matching the delimiter must be
866 : : * backslashed, and certain backslash combinations are interpreted
867 : : * non-literally by COPY IN. Disallowing all lower case ASCII letters is
868 : : * more than strictly necessary, but seems best for consistency and
869 : : * future-proofing. Likewise we disallow all digits though only octal
870 : : * digits are actually dangerous.
871 : : */
872 [ + + ]: 6637 : if (opts_out->format != COPY_FORMAT_CSV &&
873 : 6304 : strchr("\\.abcdefghijklmnopqrstuvwxyz0123456789",
874 [ + + ]: 6304 : opts_out->delim[0]) != NULL)
875 [ + - ]: 5 : ereport(ERROR,
876 : : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
877 : : errmsg("COPY delimiter cannot be \"%s\"", opts_out->delim)));
878 : :
879 : : /* Check header */
880 [ + + ]: 6632 : if (opts_out->header_line != COPY_HEADER_FALSE &&
881 [ + + ]: 141 : (opts_out->format == COPY_FORMAT_BINARY ||
882 [ + + ]: 140 : opts_out->format == COPY_FORMAT_JSON))
883 [ + - + + ]: 9 : ereport(ERROR,
884 : : errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
885 : : /*- translator: %s is the name of a COPY option, e.g. ON_ERROR */
886 : : opts_out->format == COPY_FORMAT_BINARY
887 : : ? errmsg("cannot specify %s in BINARY mode", "HEADER")
888 : : : errmsg("cannot specify %s in JSON mode", "HEADER"));
889 : :
890 : : /* Check quote */
891 [ + + + + ]: 6623 : if (opts_out->format != COPY_FORMAT_CSV && opts_out->quote != NULL)
892 [ + - ]: 6 : ereport(ERROR,
893 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
894 : : /*- translator: %s is the name of a COPY option, e.g. ON_ERROR */
895 : : errmsg("COPY %s requires CSV mode", "QUOTE")));
896 : :
897 [ + + + + ]: 6617 : if (opts_out->format == COPY_FORMAT_CSV && strlen(opts_out->quote) != 1)
898 [ + - ]: 1 : ereport(ERROR,
899 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
900 : : errmsg("COPY quote must be a single one-byte character")));
901 : :
902 [ + + + + ]: 6616 : if (opts_out->format == COPY_FORMAT_CSV && opts_out->delim[0] == opts_out->quote[0])
903 [ + - ]: 1 : ereport(ERROR,
904 : : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
905 : : errmsg("COPY delimiter and quote must be different")));
906 : :
907 : : /* Check escape */
908 [ + + + + ]: 6615 : if (opts_out->format != COPY_FORMAT_CSV && opts_out->escape != NULL)
909 [ + - ]: 7 : ereport(ERROR,
910 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
911 : : /*- translator: %s is the name of a COPY option, e.g. ON_ERROR */
912 : : errmsg("COPY %s requires CSV mode", "ESCAPE")));
913 : :
914 [ + + + + ]: 6608 : if (opts_out->format == COPY_FORMAT_CSV && strlen(opts_out->escape) != 1)
915 [ + - ]: 1 : ereport(ERROR,
916 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
917 : : errmsg("COPY escape must be a single one-byte character")));
918 : :
919 : : /* Check force_quote */
920 [ + + + + : 6607 : if (opts_out->format != COPY_FORMAT_CSV && (opts_out->force_quote || opts_out->force_quote_all))
+ + ]
921 [ + - ]: 12 : ereport(ERROR,
922 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
923 : : /*- translator: %s is the name of a COPY option, e.g. ON_ERROR */
924 : : errmsg("COPY %s requires CSV mode", "FORCE_QUOTE")));
925 [ + + + + : 6595 : if ((opts_out->force_quote || opts_out->force_quote_all) && is_from)
+ + ]
926 [ + - ]: 8 : ereport(ERROR,
927 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
928 : : /*- translator: first %s is the name of a COPY option, e.g. ON_ERROR,
929 : : second %s is a COPY with direction, e.g. COPY TO */
930 : : errmsg("COPY %s cannot be used with %s", "FORCE_QUOTE",
931 : : "COPY FROM")));
932 : :
933 : : /* Check force_notnull */
934 [ + + + + ]: 6587 : if (opts_out->format != COPY_FORMAT_CSV && (opts_out->force_notnull != NIL ||
935 [ + + ]: 6260 : opts_out->force_notnull_all))
936 [ + - ]: 13 : ereport(ERROR,
937 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
938 : : /*- translator: %s is the name of a COPY option, e.g. ON_ERROR */
939 : : errmsg("COPY %s requires CSV mode", "FORCE_NOT_NULL")));
940 [ + + + + ]: 6574 : if ((opts_out->force_notnull != NIL || opts_out->force_notnull_all) &&
941 [ + + ]: 33 : !is_from)
942 [ + - ]: 8 : ereport(ERROR,
943 : : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
944 : : /*- translator: first %s is the name of a COPY option, e.g. ON_ERROR,
945 : : second %s is a COPY with direction, e.g. COPY TO */
946 : : errmsg("COPY %s cannot be used with %s", "FORCE_NOT_NULL",
947 : : "COPY TO")));
948 : :
949 : : /* Check force_null */
950 [ + + + + ]: 6566 : if (opts_out->format != COPY_FORMAT_CSV && (opts_out->force_null != NIL ||
951 [ + + ]: 6248 : opts_out->force_null_all))
952 [ + - ]: 12 : ereport(ERROR,
953 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
954 : : /*- translator: %s is the name of a COPY option, e.g. ON_ERROR */
955 : : errmsg("COPY %s requires CSV mode", "FORCE_NULL")));
956 : :
957 [ + + + + ]: 6554 : if ((opts_out->force_null != NIL || opts_out->force_null_all) &&
958 [ + + ]: 33 : !is_from)
959 [ + - ]: 8 : ereport(ERROR,
960 : : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
961 : : /*- translator: first %s is the name of a COPY option, e.g. ON_ERROR,
962 : : second %s is a COPY with direction, e.g. COPY TO */
963 : : errmsg("COPY %s cannot be used with %s", "FORCE_NULL",
964 : : "COPY TO")));
965 : :
966 : : /* Don't allow the delimiter to appear in the null string. */
967 [ + + ]: 6546 : if (strchr(opts_out->null_print, opts_out->delim[0]) != NULL)
968 [ + - ]: 1 : ereport(ERROR,
969 : : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
970 : : /*- translator: %s is the name of a COPY option, e.g. NULL */
971 : : errmsg("COPY delimiter character must not appear in the %s specification",
972 : : "NULL")));
973 : :
974 : : /* Don't allow the CSV quote char to appear in the null string. */
975 [ + + ]: 6545 : if (opts_out->format == COPY_FORMAT_CSV &&
976 [ + + ]: 305 : strchr(opts_out->null_print, opts_out->quote[0]) != NULL)
977 [ + - ]: 1 : ereport(ERROR,
978 : : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
979 : : /*- translator: %s is the name of a COPY option, e.g. NULL */
980 : : errmsg("CSV quote character must not appear in the %s specification",
981 : : "NULL")));
982 : :
983 : : /* Check freeze */
984 [ + + - + ]: 6544 : if (opts_out->freeze && !is_from)
985 [ # # ]: 0 : ereport(ERROR,
986 : : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
987 : : /*- translator: first %s is the name of a COPY option, e.g. ON_ERROR,
988 : : second %s is a COPY with direction, e.g. COPY TO */
989 : : errmsg("COPY %s cannot be used with %s", "FREEZE",
990 : : "COPY TO")));
991 : :
992 : : /* Check json format */
993 [ + + + + ]: 6544 : if (opts_out->format == COPY_FORMAT_JSON && is_from)
994 [ + - ]: 4 : ereport(ERROR,
995 : : errcode(ERRCODE_INVALID_PARAMETER_VALUE),
996 : : errmsg("COPY %s is not supported for %s", "FORMAT JSON", "COPY FROM"));
997 : :
998 [ + + + + ]: 6540 : if (opts_out->format != COPY_FORMAT_JSON && opts_out->force_array)
999 [ + - ]: 4 : ereport(ERROR,
1000 : : errcode(ERRCODE_INVALID_PARAMETER_VALUE),
1001 : : errmsg("COPY %s can only be used with JSON mode", "FORCE_ARRAY"));
1002 : :
1003 [ + + ]: 6536 : if (opts_out->default_print)
1004 : : {
1005 [ + + ]: 47 : if (!is_from)
1006 [ + - ]: 4 : ereport(ERROR,
1007 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
1008 : : /*- translator: first %s is the name of a COPY option, e.g. ON_ERROR,
1009 : : second %s is a COPY with direction, e.g. COPY TO */
1010 : : errmsg("COPY %s cannot be used with %s", "DEFAULT",
1011 : : "COPY TO")));
1012 : :
1013 : : /* Don't allow the delimiter to appear in the default string. */
1014 [ + + ]: 43 : if (strchr(opts_out->default_print, opts_out->delim[0]) != NULL)
1015 [ + - ]: 4 : ereport(ERROR,
1016 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
1017 : : /*- translator: %s is the name of a COPY option, e.g. NULL */
1018 : : errmsg("COPY delimiter character must not appear in the %s specification",
1019 : : "DEFAULT")));
1020 : :
1021 : : /* Don't allow the CSV quote char to appear in the default string. */
1022 [ + + ]: 39 : if (opts_out->format == COPY_FORMAT_CSV &&
1023 [ + + ]: 19 : strchr(opts_out->default_print, opts_out->quote[0]) != NULL)
1024 [ + - ]: 4 : ereport(ERROR,
1025 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
1026 : : /*- translator: %s is the name of a COPY option, e.g. NULL */
1027 : : errmsg("CSV quote character must not appear in the %s specification",
1028 : : "DEFAULT")));
1029 : :
1030 : : /* Don't allow the NULL and DEFAULT string to be the same */
1031 [ + + ]: 35 : if (opts_out->null_print_len == opts_out->default_print_len &&
1032 : 16 : strncmp(opts_out->null_print, opts_out->default_print,
1033 [ + + ]: 16 : opts_out->null_print_len) == 0)
1034 [ + - ]: 4 : ereport(ERROR,
1035 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
1036 : : errmsg("NULL specification and DEFAULT specification cannot be the same")));
1037 : : }
1038 : : /* Check on_error */
1039 [ + + + + ]: 6520 : if (opts_out->format == COPY_FORMAT_BINARY && opts_out->on_error != COPY_ON_ERROR_STOP)
1040 [ + - ]: 9 : ereport(ERROR,
1041 : : (errcode(ERRCODE_SYNTAX_ERROR),
1042 : : errmsg("only ON_ERROR STOP is allowed in BINARY mode")));
1043 : :
1044 [ + + + + ]: 6511 : if (opts_out->reject_limit && opts_out->on_error != COPY_ON_ERROR_IGNORE)
1045 [ + - ]: 13 : ereport(ERROR,
1046 : : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
1047 : : /*- translator: first and second %s are the names of COPY option, e.g.
1048 : : * ON_ERROR, third is the value of the COPY option, e.g. IGNORE */
1049 : : errmsg("COPY %s requires %s to be set to %s",
1050 : : "REJECT_LIMIT", "ON_ERROR", "IGNORE")));
1051 : 6498 : }
1052 : :
1053 : : /*
1054 : : * CopyGetAttnums - build an integer list of attnums to be copied
1055 : : *
1056 : : * The input attnamelist is either the user-specified column list,
1057 : : * or NIL if there was none (in which case we want all the non-dropped
1058 : : * columns).
1059 : : *
1060 : : * We don't include generated columns in the generated full list and we don't
1061 : : * allow them to be specified explicitly. They don't make sense for COPY
1062 : : * FROM, but we could possibly allow them for COPY TO. But this way it's at
1063 : : * least ensured that whatever we copy out can be copied back in.
1064 : : *
1065 : : * rel can be NULL ... it's only used for error reports.
1066 : : */
1067 : : List *
1068 : 12710 : CopyGetAttnums(TupleDesc tupDesc, Relation rel, List *attnamelist)
1069 : : {
1070 : 12710 : List *attnums = NIL;
1071 : :
1072 [ + + ]: 12710 : if (attnamelist == NIL)
1073 : : {
1074 : : /* Generate default column list */
1075 : 2656 : int attr_count = tupDesc->natts;
1076 : : int i;
1077 : :
1078 [ + + ]: 9616 : for (i = 0; i < attr_count; i++)
1079 : : {
1080 : 6960 : CompactAttribute *attr = TupleDescCompactAttr(tupDesc, i);
1081 : :
1082 [ + + + + ]: 6960 : if (attr->attisdropped || attr->attgenerated)
1083 : 209 : continue;
1084 : 6751 : attnums = lappend_int(attnums, i + 1);
1085 : : }
1086 : : }
1087 : : else
1088 : : {
1089 : : /* Validate the user-supplied list and extract attnums */
1090 : : ListCell *l;
1091 : :
1092 [ + - + + : 44988 : foreach(l, attnamelist)
+ + ]
1093 : : {
1094 : 34990 : char *name = strVal(lfirst(l));
1095 : : int attnum;
1096 : : int i;
1097 : :
1098 : : /* Lookup column name */
1099 : 34990 : attnum = InvalidAttrNumber;
1100 [ + + ]: 4971381 : for (i = 0; i < tupDesc->natts; i++)
1101 : : {
1102 : 4971361 : Form_pg_attribute att = TupleDescAttr(tupDesc, i);
1103 : :
1104 [ + + ]: 4971361 : if (att->attisdropped)
1105 : 448 : continue;
1106 [ + + ]: 4970913 : if (namestrcmp(&(att->attname), name) == 0)
1107 : : {
1108 [ + + ]: 34970 : if (att->attgenerated)
1109 [ + - ]: 32 : ereport(ERROR,
1110 : : (errcode(ERRCODE_INVALID_COLUMN_REFERENCE),
1111 : : errmsg("column \"%s\" is a generated column",
1112 : : name),
1113 : : errdetail("Generated columns cannot be used in COPY.")));
1114 : 34938 : attnum = att->attnum;
1115 : 34938 : break;
1116 : : }
1117 : : }
1118 [ + + ]: 34958 : if (attnum == InvalidAttrNumber)
1119 : : {
1120 [ + - ]: 20 : if (rel != NULL)
1121 [ + - ]: 20 : ereport(ERROR,
1122 : : (errcode(ERRCODE_UNDEFINED_COLUMN),
1123 : : errmsg("column \"%s\" of relation \"%s\" does not exist",
1124 : : name, RelationGetRelationName(rel))));
1125 : : else
1126 [ # # ]: 0 : ereport(ERROR,
1127 : : (errcode(ERRCODE_UNDEFINED_COLUMN),
1128 : : errmsg("column \"%s\" does not exist",
1129 : : name)));
1130 : : }
1131 : : /* Check for duplicates */
1132 [ + + ]: 34938 : if (list_member_int(attnums, attnum))
1133 [ + - ]: 4 : ereport(ERROR,
1134 : : (errcode(ERRCODE_DUPLICATE_COLUMN),
1135 : : errmsg("column \"%s\" specified more than once",
1136 : : name)));
1137 : 34934 : attnums = lappend_int(attnums, attnum);
1138 : : }
1139 : : }
1140 : :
1141 : 12654 : return attnums;
1142 : : }
|
