如何从Windows上的程序传入psql密码

时间:2012-02-17 10:43:32

标签: postgresql winapi psql

我目前正在为我的项目开发PostgreSQL备份和还原功能。我已经阅读了这篇http://www.codeproject.com/Articles/37154/PostgreSQL-PostGis-Operations文章并按照这种方法来做到这一点。它工作正常但最近我已将PostgreSQL身份验证方法更改为password文件中的pg_hba.con。因此,每当我执行psql.exepg_dump.exepg_restore.exe时,它就会开始提示输入密码。为了通过我的项目提供密码,我使用了“RedirectStandardInput”方法。但它不起作用,psqlpg_dump仍然提示输入密码。但是“RedirectStandardOutput”和错误方法工作正常。

我浏览了PostgreSQL源代码,发现GetConsoleModeSetConsoleMode用于删除回声。我希望(不确定)可能是原因,这就是我无法重定向输入的原因。

PostgreSQL源代码提示密码

simple_prompt(const char *prompt, int maxlen, bool echo)
{
    int         length;
    char       *destination;
    FILE       *termin,
               *termout;
#ifdef HAVE_TERMIOS_H
    struct termios t_orig,
                t;
#else
#ifdef WIN32
    HANDLE      t = NULL;
    LPDWORD     t_orig = NULL;
#endif
#endif
    destination = (char *) malloc(maxlen + 1);
    if (!destination)
        return NULL;

    /*
     * Do not try to collapse these into one "w+" mode file. Doesn't work on
     * some platforms (eg, HPUX 10.20).
     */
    termin = fopen(DEVTTY, "r");
    termout = fopen(DEVTTY, "w");
    if (!termin || !termout
#ifdef WIN32
    /* See DEVTTY comment for msys */
        || (getenv("OSTYPE") && strcmp(getenv("OSTYPE"), "msys") == 0)
#endif
        )
    {
        if (termin)
            fclose(termin);
        if (termout)
            fclose(termout);
        termin = stdin;
        termout = stderr;
    }

#ifdef HAVE_TERMIOS_H
    if (!echo)
    {
        tcgetattr(fileno(termin), &t);
        t_orig = t;
        t.c_lflag &= ~ECHO;
        tcsetattr(fileno(termin), TCSAFLUSH, &t);
    }
#else
#ifdef WIN32
    if (!echo)
    {
        /* get a new handle to turn echo off */
        t_orig = (LPDWORD) malloc(sizeof(DWORD));
        t = GetStdHandle(STD_INPUT_HANDLE);

        /* save the old configuration first */
        GetConsoleMode(t, t_orig);

        /* set to the new mode */
        SetConsoleMode(t, ENABLE_LINE_INPUT | ENABLE_PROCESSED_INPUT);
    }
#endif
#endif
    if (prompt)
    {
        fputs(_(prompt), termout);
        fflush(termout);
    }

    if (fgets(destination, maxlen + 1, termin) == NULL)
        destination[0] = '\0';

    length = strlen(destination);
    if (length > 0 && destination[length - 1] != '\n')
    {
        /* eat rest of the line */
        char        buf[128];
        int         buflen;

        do
        {
            if (fgets(buf, sizeof(buf), termin) == NULL)
                break;
            buflen = strlen(buf);
        } while (buflen > 0 && buf[buflen - 1] != '\n');
    }

    if (length > 0 && destination[length - 1] == '\n')
        /* remove trailing newline */
        destination[length - 1] = '\0';
#ifdef HAVE_TERMIOS_H
    if (!echo)
    {
        tcsetattr(fileno(termin), TCSAFLUSH, &t_orig);
        fputs("\n", termout);
        fflush(termout);
    }
#else
#ifdef WIN32
    if (!echo)
    {
        /* reset to the original console mode */
        SetConsoleMode(t, *t_orig);
        fputs("\n", termout);
        fflush(termout);
        free(t_orig);
    }
#endif
#endif
    if (termin != stdin)
    {
        fclose(termin);
        fclose(termout);
    }

    return destination;
}

请在这里帮助我,如何通过C#代码将密码发送到psqlpg_dump

1 个答案:

答案 0 :(得分:0)

由于这是应用程序的本地,最好的办法是设置%PGPASSWORD%,然后psql不会要求输入密码。如果你想避免提供密码,可以使用.pgpass。通常,您不希望从命令行指定环境变量,因为它们可能会为其他用户显示,但这不是一个问题。