diff options
| -rw-r--r-- | src/backend/commands/copy.c | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/src/backend/commands/copy.c b/src/backend/commands/copy.c index c7537687640..15829d9fc94 100644 --- a/src/backend/commands/copy.c +++ b/src/backend/commands/copy.c @@ -738,7 +738,7 @@ DoCopy(const CopyStmt *stmt, const char *queryString) bool pipe = (stmt->filename == NULL); Relation rel; uint64 processed; - RangeTblEntry *rte; + List *range_table = NIL; /* Disallow file COPY except to superusers. */ if (!pipe && !superuser()) @@ -754,6 +754,7 @@ DoCopy(const CopyStmt *stmt, const char *queryString) AclMode required_access = (is_from ? ACL_INSERT : ACL_SELECT); List *attnums; ListCell *cur; + RangeTblEntry *rte; Assert(!stmt->query); @@ -766,6 +767,7 @@ DoCopy(const CopyStmt *stmt, const char *queryString) rte->relid = RelationGetRelid(rel); rte->relkind = rel->rd_rel->relkind; rte->requiredPerms = required_access; + range_table = list_make1(rte); tupDesc = RelationGetDescr(rel); attnums = CopyGetAttnums(tupDesc, rel, stmt->attlist); @@ -779,7 +781,7 @@ DoCopy(const CopyStmt *stmt, const char *queryString) else rte->selectedCols = bms_add_member(rte->selectedCols, attno); } - ExecCheckRTPerms(list_make1(rte), true); + ExecCheckRTPerms(range_table, true); } else { @@ -796,7 +798,7 @@ DoCopy(const CopyStmt *stmt, const char *queryString) cstate = BeginCopyFrom(rel, stmt->filename, stmt->attlist, stmt->options); - cstate->range_table = list_make1(rte); + cstate->range_table = range_table; processed = CopyFrom(cstate); /* copy from file to database */ EndCopyFrom(cstate); } @@ -804,7 +806,6 @@ DoCopy(const CopyStmt *stmt, const char *queryString) { cstate = BeginCopyTo(rel, stmt->query, queryString, stmt->filename, stmt->attlist, stmt->options); - cstate->range_table = list_make1(rte); processed = DoCopyTo(cstate); /* copy from database to file */ EndCopyTo(cstate); } |