/*
 * psql - the openGauss interactive terminal
 *
 * Copyright (c) 2000-2012, PostgreSQL Global Development Group
 *
 * src/bin/psql/copy.c
 */
#include "settings.h"
#include "postgres_fe.h"
#include "copy.h"

#include <signal.h>
#include <sys/stat.h>
#ifndef WIN32
#include <unistd.h> /* for isatty */
#else
#include <io.h> /* I think */
#endif

#include "libpq/libpq-fe.h"
#include "libpq/pqexpbuffer.h"
#include "pqsignal.h"
#include "dumputils.h"

#include "common.h"
#include "prompt.h"
#include "stringutils.h"
#include <cxxabi.h>

/*
 * parse_slash_copy
 * -- parses \copy command line
 *
 * The documented syntax is:
 *	\copy tablename [(columnlist)] from|to filename [options]
 *	\copy ( select stmt ) to filename [options]
 *
 * An undocumented fact is that you can still write BINARY before the
 * tablename; this is a hangover from the pre-7.3 syntax.  The options
 * syntax varies across backend versions, but we avoid all that mess
 * by just transmitting the stuff after the filename literally.
 *
 * table name can be double-quoted and can have a schema part.
 * column names can be double-quoted.
 * filename can be single-quoted like SQL literals.
 *
 * returns a malloc'ed structure with the options, or NULL on parsing error
 */

struct copy_options {
    char* before_tofrom; /* COPY string before TO/FROM */
    char* after_tofrom;  /* COPY string after TO/FROM filename */
    char* file;          /* NULL = stdin/stdout */
    bool psql_inout;     /* true = use psql stdin/stdout */
    bool from;           /* true = FROM, false = TO */
    int parallel;        /* Concurrent Number, only for copy from now */
    char* relName;       /* relation name. If enclosed in quotation marks, take out the value in the quotation marks,
                          * otherwise convert to lowercase. only for copy from */
    bool isFileBinary;   /* true if file format is binary, only for copy from */
    bool isEol;         /* true if 'eol' exists in copy options, only for copy from */
    bool hasHeader;     /* true if 'header' exists and value is true in copy options, only for copy from */
};

/* read chunk size for COPY IN - size is not critical */
#define COPYBUFSIZ 8192

static void free_copy_options(struct copy_options* ptr)
{
    if (ptr == NULL)
        return;
    free(ptr->before_tofrom);
    ptr->before_tofrom = NULL;
    free(ptr->after_tofrom);
    ptr->after_tofrom = NULL;
    free(ptr->file);
    ptr->file = NULL;
    free(ptr->relName);
    ptr->relName = NULL;
    free(ptr);
    ptr = NULL;
}

/* concatenate "more" onto "var", freeing the original value of *var */
static void xstrcat(char** var, const char* more)
{
    /* strlen will crash when NULL passed in */
    if (var == NULL || *var == NULL || more == NULL) {
        psql_error("trying to xstrcat a NULL \n");
        exit(EXIT_FAILURE);
        return;
    }

    errno_t err = EOK;
    char* newvar = NULL;
    size_t varlen = strlen(*var);
    size_t morelen = strlen(more);
    size_t newvarsz = varlen + morelen + 1;

    newvar = (char*)pg_malloc(newvarsz);
    err = strcpy_s(newvar, newvarsz, *var);
    check_strcpy_s(err);

    err = strcat_s(newvar, newvarsz, more);
    check_strcat_s(err);

    free(*var);
    *var = newvar;
}

static bool AtoiStrictly(const char *str, int *num)
{
    size_t i;
    for (i = 0; i < strlen(str); i++) {
        if (str[i] < '0' || str[i] > '9') {
            return false;
        }
    }

    *num = atoi(str);
    return true;
}

static void toUpper(char* dest, const char* source)
{
    size_t i;
    size_t len = strlen(source);
    for (i = 0; i < len; i++) {
        dest[i] = toupper(source[i]);
    }
    dest[len] = '\0';
}

// check if copy options keyword key exists in source or not
// if ,key or key' exists in source, return true
static bool IsCopyOptionKeyWordExists(const char* source, const char* key)
{
    size_t sourceLen = strlen(source);
    size_t keyLen = strlen(key);

    if (sourceLen < keyLen) {
        return false;
    }

    if (pg_strcasecmp(source, key) == 0) {
        return true;
    }

    char* upperSource = (char*)pg_malloc(sourceLen + 1);
    // compare ,key or key', so malloc size is keyLen + 2
    char* upperKey = (char*)pg_malloc(keyLen + 2);

    *upperKey = ',';
    toUpper(upperSource, source);
    toUpper(upperKey + 1, key);
    if (strstr(upperSource, upperKey) != NULL) {
        free(upperKey);
        free(upperSource);
        return true;
    }

    toUpper(upperKey, key);
    upperKey[keyLen] = '\'';
    upperKey[keyLen + 1] = '\0';

    if (strstr(upperSource, upperKey) != NULL) {
        free(upperKey);
        free(upperSource);
        return true;
    }

    free(upperKey);
    free(upperSource);
    return false;
}

static void RemoveOptionInParallelCopy(struct copy_options* options, char *src, const char *key)
{
    if (src == nullptr) {
        return;
    }
    size_t srcLen = strlen(src);
    size_t keyLen = strlen(key);
    if (srcLen < keyLen) {
        return;
    }
    char* upperSrc = (char*)pg_malloc(srcLen + 1);
    char* upperKey = (char*)pg_malloc(keyLen + 1);
    toUpper(upperSrc, src);
    toUpper(upperKey, key);

    char *firstPos = strstr(upperSrc, upperKey);
    char *curPos = firstPos;
    size_t keyCnt = 0;
    /*
     * Check exist there are multiple keys or single keys.
     * When there are multiple headers, we will not process it,
     * the server will report an redundant options error.
     * because if there are two keys now, after remove one key,
     * there are only one key, an invalid SQL will become a valid SQL.
     */
    while (curPos != nullptr) {
        ++keyCnt;
        if (keyCnt > 1) {
            break;
        }
        curPos = curPos + keyLen;
        curPos = strstr(curPos, upperKey);
    }

    bool isKey = (keyCnt == 1);

    /* header option available only in CSV mode. */
    if (pg_strcasecmp(upperKey, "HEADER") == 0) {
        char *csvPos = strstr(upperSrc, "CSV");
        isKey = isKey && (csvPos != nullptr);
    }

    if (isKey) {
        const char* whitespace = " \t\n\r";
        char *firstPosStart = firstPos;
        char *firstPosEnd = firstPos + keyLen - 1;
        while (--firstPosStart >= upperSrc) {
            if (strchr(whitespace, *firstPosStart) == nullptr) {
                break;
            }
        }

        while (++firstPosEnd <= upperSrc + srcLen) {
            if (strchr(whitespace, *firstPosEnd) == nullptr) {
                break;
            }
        }

        /*
         * We will not handle situations where there are ","
         * before and after the key. eg: "with (csv, header)"
         */

        if ((firstPosStart >= upperSrc && *firstPosStart == ',') ||
             (firstPosEnd <= upperSrc + srcLen && *firstPosEnd == ',')) {
            isKey = false;
        }

        /*
         * We will handle situations like this:
         * eg: "with csv header;" or "with csv header ..."
         */
        isKey = isKey && (firstPos == upperSrc || strchr(whitespace, *(firstPos - 1)) != nullptr)  &&
                     (strchr(whitespace, *firstPosEnd) != nullptr || *firstPosEnd  == ';' || *firstPosEnd  == '\0');
    }

    if (keyCnt == 1 && isKey) {
        options->hasHeader = true;
        char *cpyDestPos = src + (firstPos - upperSrc);
        char *cpySrcPos = cpyDestPos + keyLen;
        errno_t rc = memmove_s(cpyDestPos, strlen(cpySrcPos) + 1, cpySrcPos, strlen(cpySrcPos) + 1);
        securec_check_c(rc, "\0", "\0");
    }
    free(upperKey);
    free(upperSrc);
}

/* parse parallel settings */
static bool ParseParallelOption(struct copy_options* result, char** errToken)
{
    const char* whitespace = " \t\n\r";
    char *token = strtokx(nullptr, whitespace, ".,()", "\"", 0, false, false, pset.encoding);

    if (token == nullptr) {
        return true;
    }

    result->after_tofrom = pg_strdup("");

    if (pg_strcasecmp(token, "with") == 0) {
        /* Do not copy "with" to result->after_tofrom */
        token = strtokx(nullptr, whitespace, NULL, "\"", 0, false, false, pset.encoding);
        if (token == nullptr)
            return false;
    }

    for (;;) {
        if (pg_strcasecmp(token, "parallel") == 0) {
            token = strtokx(nullptr, whitespace, ";", "\"", 0, false, false, pset.encoding);
            if (token == nullptr)
                return false;

            if (!AtoiStrictly(token, &result->parallel)) {
                *errToken = token;
                return false;
            }

            token = strtokx(nullptr, "", NULL, NULL, 0, false, false, pset.encoding);
            RemoveOptionInParallelCopy(result, token, "header");
            if (token != nullptr) {
                xstrcat(&result->after_tofrom, " ");
                xstrcat(&result->after_tofrom, token);
            }
            return true;
        }

        if (IsCopyOptionKeyWordExists(token, "binary")) {
            result->isFileBinary = true;
        }

        if (IsCopyOptionKeyWordExists(token, "eol")) {
            result->isEol = true;
        }

        if (IsCopyOptionKeyWordExists(token, "header")) {
            // if header keyword exists, read a token forward
            // when header is true, copy "header false" to result->after_tofrom
            xstrcat(&result->after_tofrom, " ");
            xstrcat(&result->after_tofrom, token);

            token = strtokx(nullptr, whitespace, ",()", NULL, 0, false, false, pset.encoding);
            if (token != nullptr) {
                bool res = (pg_strcasecmp(token, "true") == 0 || pg_strcasecmp(token, "on") == 0);
                result->hasHeader = (result->hasHeader || res);
                const char *resToken = res ? "false" : token;
                xstrcat(&result->after_tofrom, " ");
                xstrcat(&result->after_tofrom, resToken);
            }
        } else {
            xstrcat(&result->after_tofrom, " ");
            xstrcat(&result->after_tofrom, token);
        }

        token = strtokx(nullptr, whitespace, NULL, NULL, 0, false, false, pset.encoding);
        if (token == nullptr) {
            return true;
        }
    }
    return true;
}

/* if token is \"Xab\", then get Xab; if token is Xab, then get xab */
static void GetRelNameWithCase(char** dest, const char* token)
{
    int len = strlen(token);
    errno_t err = EOK;
    
    char* tokenCopy = (char*)pg_malloc(len + 1);
    err = strcpy_s(tokenCopy, len + 1, token);
    check_strcpy_s(err);
    
    if (token[0] == '\"') {
        tokenCopy[len - 1] = '\0';
        xstrcat(dest, tokenCopy + 1);
    } else {
        xstrcat(dest, pg_strtolower(tokenCopy));
    }

    free(tokenCopy);
}

static struct copy_options* parse_slash_copy(const char* args)
{
    struct copy_options* result = NULL;
    char* token = NULL;
    const char* whitespace = " \t\n\r";
    char nonstd_backslash = standard_strings() ? 0 : '\\';

    if (args == NULL) {
        psql_error("\\copy: arguments required\n");
        return NULL;
    }

    result = (struct copy_options*)pg_calloc(1, sizeof(struct copy_options));

    result->before_tofrom = pg_strdup(""); /* initialize for appending */
    result->parallel = 0;
    result->isFileBinary = false;
    result->isEol = false;

    token = strtokx(args, whitespace, ".,()", "\"", 0, false, false, pset.encoding);
    if (token == NULL)
        goto error;

    /* The following can be removed when we drop 7.3 syntax support */
    if (pg_strcasecmp(token, "binary") == 0) {
        result->isFileBinary = true;
        xstrcat(&result->before_tofrom, token);
        token = strtokx(NULL, whitespace, ".,()", "\"", 0, false, false, pset.encoding);
        if (token == NULL)
            goto error;
    }

    /* Handle COPY (SELECT) case */
    if (token[0] == '(') {
        int parens = 1;

        while (parens > 0) {
            xstrcat(&result->before_tofrom, " ");
            xstrcat(&result->before_tofrom, token);
            token = strtokx(NULL, whitespace, "()", "\"'", nonstd_backslash, true, false, pset.encoding);
            if (token == NULL)
                goto error;
            if (token[0] == '(') {
                parens++;
            } else if (token[0] == ')') {
                parens--;
            }
        }
    }

    result->relName = pg_strdup("");
    
    xstrcat(&result->before_tofrom, " ");
    xstrcat(&result->before_tofrom, token);
    GetRelNameWithCase(&result->relName, token);
    token = strtokx(NULL, whitespace, ".,()", "\"", 0, false, false, pset.encoding);
    if (token == NULL)
        goto error;

    /*
     * strtokx() will not have returned a multi-character token starting with
     * '.', so we don't need strcmp() here.  Likewise for '(', etc, below.
     */
    if (token[0] == '.') {
        /* handle schema . table */
        xstrcat(&result->before_tofrom, token);
        xstrcat(&result->relName, token);
        token = strtokx(NULL, whitespace, ".,()", "\"", 0, false, false, pset.encoding);
        if (token == NULL)
            goto error;
        xstrcat(&result->before_tofrom, token);
        GetRelNameWithCase(&result->relName, token);
        token = strtokx(NULL, whitespace, ".,()", "\"", 0, false, false, pset.encoding);
        if (token == NULL)
            goto error;
    }

    if (token[0] == '(') {
        /* handle parenthesized column list */
        for (;;) {
            xstrcat(&result->before_tofrom, " ");
            xstrcat(&result->before_tofrom, token);
            token = strtokx(NULL, whitespace, "()", "\"", 0, false, false, pset.encoding);
            if (token == NULL) {
                goto error;
            }
            if (token[0] == ')') {
                break;
            }
        }
        xstrcat(&result->before_tofrom, " ");
        xstrcat(&result->before_tofrom, token);
        token = strtokx(NULL, whitespace, ".,()", "\"", 0, false, false, pset.encoding);
        if (token == NULL)
            goto error;
    }

    if (pg_strcasecmp(token, "from") == 0)
        result->from = true;
    else if (pg_strcasecmp(token, "to") == 0)
        result->from = false;
    else
        goto error;

    token = strtokx(NULL, whitespace, ";", "'", 0, false, true, pset.encoding);
    if (token == NULL)
        goto error;

    if (pg_strcasecmp(token, "stdin") == 0 || pg_strcasecmp(token, "stdout") == 0) {
        result->psql_inout = false;
        result->file = NULL;
    } else if (pg_strcasecmp(token, "pstdin") == 0 || pg_strcasecmp(token, "pstdout") == 0) {
        result->psql_inout = true;
        result->file = NULL;
    } else {
        result->psql_inout = false;
        result->file = pg_strdup(token);
        expand_tilde(&result->file);
    }

    /* Subtract the parallel setting */
    token = nullptr;
    if (!ParseParallelOption(result, &token)) {
        goto error;
    }

    return result;

error:
    if (NULL != token)
        psql_error("\\copy: parse error at \"%s\"\n", token);
    else
        psql_error("\\copy: parse error at end of line\n");
    free_copy_options(result);

    return NULL;
}

static bool IsParallelCopyFrom(const struct copy_options* options)
{
    PGresult* result = nullptr;
    char* query = nullptr;
    char* val = nullptr;
    
    if (PQtransactionStatus(pset.db) != PQTRANS_IDLE) {
        return false;
    }
    
    if (!options->from || options->parallel == 0 || options->isEol
        || options->isFileBinary || pset.decryptInfo.encryptInclude) {
        return false;
    }

    /* check if option->relName is temp table */
    query = pg_strdup("SELECT relpersistence FROM pg_class WHERE relname=\'");
    xstrcat(&query, options->relName);
    xstrcat(&query, "\';");

    result = PQexec(pset.db, query);
    free(query);
    if (PQresultStatus(result) != PGRES_TUPLES_OK) {
        psql_error("%s", PQerrorMessage(pset.db));
        PQclear(result);
        return false;
    }

    if (PQntuples(result) == 0) {
        PQclear(result);
        return true;
    }
    
    val = PQgetvalue(result, 0, 0);
    if (pg_strcasecmp(val, "t") == 0) {
        PQclear(result);
        return false;
    }

    PQclear(result);
    
    return true;
}

static void skip_first_line_when_has_header(const struct copy_options* options, FILE* copystream)
{
    if (!options->hasHeader) {
        return;
    }

    char buf[COPYBUFSIZ];
    fgets(buf, sizeof(buf), copystream);
}

/*
 * Execute a \copy command (frontend copy). We have to open a file, then
 * submit a COPY query to the backend and either feed it data from the
 * file or route its response into the file.
 */
bool do_copy(const char* args)
{
    PQExpBufferData query;
    FILE* copystream = NULL;
    struct copy_options* options = NULL;
    bool success = true;
    struct stat st;

    /* parse options */
    options = parse_slash_copy(args);

    if (options == NULL)
        return false;

    /* prepare to read or write the target file */
    if (NULL != options->file)
        canonicalize_path(options->file);

    if (options->from) {
        if (NULL != options->file)
            copystream = fopen(options->file, PG_BINARY_R);
        else if (!options->psql_inout)
            copystream = pset.cur_cmd_source;
        else
            copystream = stdin;
    } else {
        if (options->file != NULL) {
            canonicalize_path(options->file);
            copystream = fopen(options->file, PG_BINARY_W);
        } else if (!options->psql_inout)
            copystream = pset.queryFout;
        else
            copystream = stdout;
    }

    if (copystream == NULL) {
        psql_error("%s: %s\n", options->file, strerror(errno));
        free_copy_options(options);
        return false;
    }

    // if header = true, skip first line
    skip_first_line_when_has_header(options, copystream);

    /* make sure the specified file is not a directory */
    fstat(fileno(copystream), &st);
    if (S_ISDIR(st.st_mode)) {
        fclose(copystream);
        psql_error("%s: cannot copy from/to a directory\n", options->file);
        free_copy_options(options);
        return false;
    }

    /* build the command we will send to the backend */
    initPQExpBuffer(&query);
    printfPQExpBuffer(&query, "COPY ");
    appendPQExpBufferStr(&query, options->before_tofrom);
    if (options->from)
        appendPQExpBuffer(&query, " FROM STDIN ");
    else
        appendPQExpBuffer(&query, " TO STDOUT ");
    if (NULL != options->after_tofrom)
        appendPQExpBufferStr(&query, options->after_tofrom);

    /* Run it like a user command, interposing the data source or sink. */
    pset.copyStream = copystream;
    if (IsParallelCopyFrom(options)) {
        success = MakeCopyWorker(query.data, options->parallel);
    } else {
        success = SendQuery(query.data);
    }
    pset.copyStream = NULL;
    termPQExpBuffer(&query);

    if (options->file != NULL) {
        if (fclose(copystream) != 0) {
            psql_error("%s: %s\n", options->file, strerror(errno));
            success = false;
        }
    }
    free_copy_options(options);
    return success;
}

/*
 * Functions for handling COPY IN/OUT data transfer.
 *
 * If you want to use COPY TO STDOUT/FROM STDIN in your application,
 * this is the code to steal ;)
 */

/*
 * handleCopyOut
 * receives data as a result of a COPY ... TO STDOUT command
 *
 * conn should be a database connection that you just issued COPY TO on
 * and got back a PGRES_COPY_OUT result.
 * copystream is the file stream for the data to go to.
 *
 * result is true if successful, false if not.
 */
bool handleCopyOut(PGconn* conn, FILE* copystream)
{
    bool OK = true;
    char* buf = NULL;
    int ret;
    PGresult* res = NULL;

    for (;;) {
        ret = PQgetCopyData(conn, &buf, 0);

        if (ret < 0) {
            break; /* done or error */
        }

        if (NULL != buf) {
            if (fwrite(buf, 1, ret, copystream) != (unsigned int)(ret)) {
                if (OK) /* complain only once, keep reading data */
                    psql_error("could not write COPY data: %s\n", strerror(errno));
                OK = false;
            }
            PQfreemem(buf);
        }
    }

    if (OK && fflush(copystream)) {
        psql_error("could not write COPY data: %s\n", strerror(errno));
        OK = false;
    }

    if (ret == -2) {
        psql_error("COPY data transfer failed: %s", PQerrorMessage(conn));
        OK = false;
    }

    /*
     * Check command status and return to normal libpq state.  After a
     * client-side error, the server will remain ready to deliver data.  The
     * cleanest thing is to fully drain and discard that data.	If the
     * client-side error happened early in a large file, this takes a long
     * time.  Instead, take advantage of the fact that PQexec() will silently
     * end any ongoing PGRES_COPY_OUT state.  This does cause us to lose the
     * results of any commands following the COPY in a single command string.
     * It also only works for protocol version 3.  XXX should we clean up
     * using the slow way when the connection is using protocol version 2?
     *
     * We must not ever return with the status still PGRES_COPY_OUT.  Our
     * caller is unable to distinguish that situation from reaching the next
     * COPY in a command string that happened to contain two consecutive COPY
     * TO STDOUT commands.	We trust that no condition can make PQexec() fail
     * indefinitely while retaining status PGRES_COPY_OUT.
     */
    while (res = PQgetResult(conn), PQresultStatus(res) == PGRES_COPY_OUT) {
        OK = false;
        PQclear(res);

        PQexec(conn, "-- clear PGRES_COPY_OUT state");
    }
    if (PQresultStatus(res) != PGRES_COMMAND_OK) {
        psql_error("%s", PQerrorMessage(conn));
        OK = false;
    }
    PQclear(res);

    return OK;
}

/*
 * handleCopyIn
 * sends data to complete a COPY ... FROM STDIN command
 *
 * conn should be a database connection that you just issued COPY FROM on
 * and got back a PGRES_COPY_IN result.
 * copystream is the file stream to read the data from.
 * isbinary can be set from PQbinaryTuples().
 *
 * result is true if successful, false if not.
 */

bool handleCopyIn(PGconn* conn, FILE* copystream, bool isbinary)
{
    bool OK = false;
    const char* prompt = NULL;
    char buf[COPYBUFSIZ];
    PGresult* res = NULL;

    /*
     * Establish longjmp destination for exiting from wait-for-input. (This is
     * only effective while sigint_interrupt_enabled is TRUE.)
     */
    if (sigsetjmp(sigint_interrupt_jmp, 1) != 0) {
        /* got here with longjmp */
        /* Terminate data transfer */
        (void)PQputCopyEnd(conn, _("canceled by user"));

        OK = false;
        goto copyin_cleanup;
    }

    /* Prompt if interactive input */
    if (isatty(fileno(copystream))) {
        if (!pset.quiet)
            puts(_("Enter data to be copied followed by a newline.\n"
                   "End with a backslash and a period on a line by itself."));
        prompt = get_prompt(PROMPT_COPY);
    } else
        prompt = NULL;

    OK = true;

    if (isbinary) {
        /* interactive input probably silly, but give one prompt anyway */
        if (NULL != prompt) {
            fputs(prompt, stdout);
            fflush(stdout);
        }

        for (;;) {
            int buflen;

            /* enable longjmp while waiting for input */
            sigint_interrupt_enabled = true;

            buflen = fread(buf, 1, COPYBUFSIZ, copystream);

            sigint_interrupt_enabled = false;

            if (buflen <= 0) {
                break;
            }

            if (PQputCopyData(conn, buf, buflen) <= 0) {
                OK = false;
                break;
            }
        }
    } else {
        bool copydone = false;

        while (!copydone) { /* for each input line ... */
            bool firstload = false;
            bool linedone = false;

            if (NULL != prompt) {
                fputs(prompt, stdout);
                fflush(stdout);
            }

            firstload = true;
            linedone = false;

            while (!linedone) { /* for each bufferload in line ... */
                size_t linelen;
                char* fgresult = NULL;

                /* enable longjmp while waiting for input */
                sigint_interrupt_enabled = true;
                /* Database Security: Data importing/dumping support AES128. */
                if (!pset.decryptInfo.encryptInclude) {
                    fgresult = fgets(pset.decryptInfo.currLine, sizeof(pset.decryptInfo.currLine), copystream);
                } else {
                    fgresult = getLineFromAesEncryptFile(copystream, &pset.decryptInfo);
                }
                sigint_interrupt_enabled = false;

                if (fgresult == NULL) {
                    copydone = true;
                    break;
                }

                if (!pset.decryptInfo.encryptInclude) {
                    linelen = strlen(pset.decryptInfo.currLine);

                    /* current line is done? */
                    if (linelen > 0 && pset.decryptInfo.currLine[linelen - 1] == '\n')
                        linedone = true;

                    /* check for EOF marker, but not on a partial line */
                    if (firstload) {
                        if (strcmp(pset.decryptInfo.currLine, "\\.\n") == 0 ||
                            strcmp(pset.decryptInfo.currLine, "\\.\r\n") == 0) {
                            copydone = true;
                            break;
                        }

                        firstload = false;
                    }

                    if (PQputCopyData(conn, pset.decryptInfo.currLine, linelen) <= 0) {
                        OK = false;
                        copydone = true;
                        break;
                    }
                } else {
                    linelen = strlen((const char*)pset.decryptInfo.decryptBuff);

                    /* current line is done? */
                    if (linelen > 0 && pset.decryptInfo.decryptBuff[linelen - 1] == '\n')
                        linedone = true;

                    /* check for EOF marker, but not on a partial line */
                    if (firstload) {
                        if (strncmp((const char*)pset.decryptInfo.decryptBuff, "\\.\n", strlen("\\.\n")) == 0 ||
                            strncmp((const char*)pset.decryptInfo.decryptBuff, "\\.\r\n", strlen("\\.\n")) == 0) {
                            copydone = true;
                            free(pset.decryptInfo.decryptBuff);
                            pset.decryptInfo.decryptBuff = NULL;
                            break;
                        }

                        firstload = false;
                    }

                    if (PQputCopyData(conn, (const char*)pset.decryptInfo.decryptBuff, linelen) <= 0) {
                        OK = false;
                        copydone = true;
                        free(pset.decryptInfo.decryptBuff);
                        pset.decryptInfo.decryptBuff = NULL;
                        break;
                    }

                    free(pset.decryptInfo.decryptBuff);
                    pset.decryptInfo.decryptBuff = NULL;
                }
            }
            if (copystream == pset.cur_cmd_source)
                pset.lineno++;
        }
    }

    if (pset.decryptInfo.decryptBuff != NULL) {
        free(pset.decryptInfo.decryptBuff);
        pset.decryptInfo.decryptBuff = NULL;
    }

    /* Check for read error */
    if (ferror(copystream))
        OK = false;

    /* Terminate data transfer */
    if (PQputCopyEnd(conn, OK ? NULL : _("aborted because of read failure")) <= 0)
        OK = false;

copyin_cleanup:

    /*
     * Check command status and return to normal libpq state
     *
     * We must not ever return with the status still PGRES_COPY_IN.  Our
     * caller is unable to distinguish that situation from reaching the next
     * COPY in a command string that happened to contain two consecutive COPY
     * FROM STDIN commands.  XXX if something makes PQputCopyEnd() fail
     * indefinitely while retaining status PGRES_COPY_IN, we get an infinite
     * loop.  This is more realistic than handleCopyOut()'s counterpart risk.
     */
    while (res = PQgetResult(conn), PQresultStatus(res) == PGRES_COPY_IN) {
        OK = false;
        PQclear(res);

        PQputCopyEnd(pset.db, _("trying to exit copy mode"));
    }
    if (PQresultStatus(res) != PGRES_COMMAND_OK) {
        psql_error("%s", PQerrorMessage(conn));
        OK = false;
    }
    PQclear(res);

    return OK;
}

bool ParallelCopyIn(const CopyInArgs* copyarg, const char** errMsg)
{
    PGconn* conn = copyarg->conn;
    FILE* copystream = pset.copyStream ? pset.copyStream : pset.cur_cmd_source;
    bool OK = true;
    char buf[COPYBUFSIZ];
    PGresult* res = NULL;
    bool copydone = false;

    while (!copydone) { /* for each input line ... */
        bool firstload = false;
        bool linedone = false;

        firstload = true;
        linedone = false;

        while (!linedone) { /* for each bufferload in line ... */
            size_t linelen;
            char* fgresult = NULL;

            pthread_mutex_lock(copyarg->stream_mutex);

            if (pset.parallelCopyDone) {
                copydone = true;
                OK = pset.parallelCopyOk;
                pthread_mutex_unlock(copyarg->stream_mutex);
                break;
            }

            fgresult = fgets(buf, sizeof(buf), copystream);

            if (fgresult == NULL) {
                copydone = true;
                pset.parallelCopyDone = true;
                pthread_mutex_unlock(copyarg->stream_mutex);
                break;
            }

            /* check for EOF marker, but not on a partial line */
            if (firstload) {
                if (strcmp(buf, "\\.\n") == 0 ||
                    strcmp(buf, "\\.\r\n") == 0) {
                    copydone = true;
                    pset.parallelCopyDone = true;
                    pthread_mutex_unlock(copyarg->stream_mutex);
                    break;
                }

                firstload = false;
            }

            pthread_mutex_unlock(copyarg->stream_mutex);

            linelen = strlen(buf);

            /* current line is done? */
            if (linelen > 0 && buf[linelen - 1] == '\n') {
                linedone = true;
            }

            if (PQputCopyData(conn, buf, linelen) <= 0) {
                OK = false;
                copydone = true;
                break;
            }
        }

        if (copystream == pset.cur_cmd_source)
            pset.lineno++;
    }

    /* Check for read error */
    if (ferror(copystream))
        OK = false;
    
    pthread_mutex_lock(copyarg->stream_mutex);
    pset.parallelCopyDone = true;
    pset.parallelCopyOk = OK;
    pthread_mutex_unlock(copyarg->stream_mutex);

    /* Terminate data transfer */
    if (PQputCopyEnd(conn, OK ? NULL : _("aborted because of read failure")) <= 0)
        OK = false;

    /*
     * Check command status and return to normal libpq state
     *
     * We must not ever return with the status still PGRES_COPY_IN.  Our
     * caller is unable to distinguish that situation from reaching the next
     * COPY in a command string that happened to contain two consecutive COPY
     * FROM STDIN commands.  XXX if something makes PQputCopyEnd() fail
     * indefinitely while retaining status PGRES_COPY_IN, we get an infinite
     * loop.  This is more realistic than handleCopyOut()'s counterpart risk.
     */
    while (res = PQgetResult(conn), PQresultStatus(res) == PGRES_COPY_IN) {
        OK = false;
        PQclear(res);

        PQputCopyEnd(conn, _("trying to exit copy mode"));
    }
    if (PQresultStatus(res) != PGRES_COMMAND_OK) {
        *errMsg = PQerrorMessage(conn);
        OK = false;
    }
    PQclear(res);

    return OK;
}