farrokhi/pg_repack
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Added PG 10 compatibility wrapper to simple_prompt()

Browse Source 
Also checking for malloc failure into previous version's simple_prompt()
calls.
This commit is contained in:
Daniele Varrazzo 2017-04-25 12:24:12 +01:00
parent b9b8e60bda
commit 19e34a3202
1 changed files with 20 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
bin/pgut/pgut.c
View File

@ -434,7 +434,26 @@ simple_string_list_size(SimpleStringList list)
static char *
prompt_for_password(void)
{
return simple_prompt("Password: ", 100, false);
char *buf;
#define BUFSIZE 100
#if PG_VERSION_NUM < 100000
buf = simple_prompt("Password: ", BUFSIZE, false);
#else
buf = (char *)malloc(BUFSIZE);
if (buf != NULL)
simple_prompt("Password: ", buf, BUFSIZE, false);
#endif
if (buf == NULL)
ereport(FATAL,
(errcode_errno(),
errmsg("could not allocate memory (" UINT64_FORMAT " bytes): ",
(unsigned long) BUFSIZE)));
return buf;
#undef BUFSIZE
}