Improve WIN32 port of fstat() to detect more file types

The current implementation of _pgfstat64() is ineffective in detecting a
terminal handle or an anonymous named pipe.  This commit improves our
port of fstat() to detect more efficiently such cases by relying on
GetFileType(), and returning more correct data when the type found is
either a FILE_TYPE_PIPE (_S_IFIFO) or a FILE_TYPE_CHAR (_S_IFCHR).

This is part of a more global fix to address failures when feeding the
output generated by pg_dump to pg_restore through a pipe, for example,
but not all of it.   We are also going to need to do something about
fseek() and ftello() which are not reliable on WIN32 for the same cases
where fstat() was incorrect.  Fixing fstat() is independent of the rest,
though, which is why both fixes are handled separately, and this is the
first part of it.

Reported-by: Daniel Watzinger
Author: Daniel Watzinger, Juan José Santamaría Flecha
Discussion: https://postgr.es/m/b1448cd7-871e-20e3-8398-895e2d1d3bf9@gmail.com
Backpatch-through: 14
This commit is contained in:
Michael Paquier 2023-03-15 12:55:51 +09:00
parent 89e46da5e5
commit 765f5df726
1 changed files with 49 additions and 17 deletions

View File

@ -257,34 +257,66 @@ int
_pgfstat64(int fileno, struct stat *buf)
{
HANDLE hFile = (HANDLE) _get_osfhandle(fileno);
BY_HANDLE_FILE_INFORMATION fiData;
DWORD fileType = FILE_TYPE_UNKNOWN;
DWORD lastError;
unsigned short st_mode;
if (hFile == INVALID_HANDLE_VALUE || buf == NULL)
/*
* When stdin, stdout, and stderr aren't associated with a stream the
* special value -2 is returned:
* https://learn.microsoft.com/en-us/cpp/c-runtime-library/reference/get-osfhandle
*/
if (hFile == INVALID_HANDLE_VALUE || hFile == (HANDLE) -2 || buf == NULL)
{
errno = EINVAL;
return -1;
}
fileType = GetFileType(hFile);
lastError = GetLastError();
/*
* Check if the fileno is a data stream. If so, unless it has been
* redirected to a file, getting information through its HANDLE will fail,
* so emulate its stat information in the most appropriate way and return
* it instead.
* Invoke GetLastError in order to distinguish between a "valid" return of
* FILE_TYPE_UNKNOWN and its return due to a calling error. In case of
* success, GetLastError returns NO_ERROR.
*/
if ((fileno == _fileno(stdin) ||
fileno == _fileno(stdout) ||
fileno == _fileno(stderr)) &&
!GetFileInformationByHandle(hFile, &fiData))
if (fileType == FILE_TYPE_UNKNOWN && lastError != NO_ERROR)
{
memset(buf, 0, sizeof(*buf));
buf->st_mode = _S_IFCHR;
buf->st_dev = fileno;
buf->st_rdev = fileno;
buf->st_nlink = 1;
return 0;
_dosmaperr(lastError);
return -1;
}
return fileinfo_to_stat(hFile, buf);
switch (fileType)
{
/* The specified file is a disk file */
case FILE_TYPE_DISK:
return fileinfo_to_stat(hFile, buf);
/*
* The specified file is a socket, a named pipe, or an anonymous
* pipe.
*/
case FILE_TYPE_PIPE:
st_mode = _S_IFIFO;
break;
/* The specified file is a character file */
case FILE_TYPE_CHAR:
st_mode = _S_IFCHR;
break;
/* Unused flag and unknown file type */
case FILE_TYPE_REMOTE:
case FILE_TYPE_UNKNOWN:
default:
errno = EINVAL;
return -1;
}
memset(buf, 0, sizeof(*buf));
buf->st_mode = st_mode;
buf->st_dev = fileno;
buf->st_rdev = fileno;
buf->st_nlink = 1;
return 0;
}
#endif /* WIN32 */