ReceiveTempSMS
/
Sweden-Number
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

shcore: Don't return E_FAIL in IStream::SetSize() for read-only mode. 

Signed-off-by: Jactry Zeng <jzeng@codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>
This commit is contained in:
Jactry Zeng 2019-02-05 23:39:04 +08:00 committed by Alexandre Julliard
parent d4dc48041b
commit 13e0b4839b
2 changed files with 132 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
dlls/shcore/main.c
View File

@ -913,15 +913,26 @@ static HRESULT WINAPI filestream_Seek(IStream *iface, LARGE_INTEGER move, DWORD
static HRESULT WINAPI filestream_SetSize(IStream *iface, ULARGE_INTEGER size)
{
struct shstream *stream = impl_from_IStream(iface);
LARGE_INTEGER origin, move;
TRACE("(%p, %s)\n", stream, wine_dbgstr_longlong(size.QuadPart));
if (!SetFilePointer(stream->u.file.handle, size.QuadPart, NULL, FILE_BEGIN))
move.QuadPart = 0;
if (!SetFilePointerEx(stream->u.file.handle, move, &origin, FILE_CURRENT))
return E_FAIL;
if (!SetEndOfFile(stream->u.file.handle))
move.QuadPart = size.QuadPart;
if (!SetFilePointerEx(stream->u.file.handle, move, NULL, FILE_BEGIN))
return E_FAIL;
if (stream->u.file.mode != STGM_READ)
{
if (!SetEndOfFile(stream->u.file.handle))
return E_FAIL;
if (!SetFilePointerEx(stream->u.file.handle, origin, NULL, FILE_BEGIN))
return E_FAIL;
}
return S_OK;
}

119
dlls/shcore/tests/shcore.c
View File

@ -41,6 +41,8 @@ static LSTATUS (WINAPI *pSHRegGetValueA)(HKEY, const char *, const char *, SRRF,
static DWORD (WINAPI *pSHQueryValueExA)(HKEY, const char *, DWORD *, DWORD *, void *buff, DWORD *buff_len);
static DWORD (WINAPI *pSHRegGetPathA)(HKEY, const char *, const char *, char *, DWORD);
static DWORD (WINAPI *pSHCopyKeyA)(HKEY, const char *, HKEY, DWORD);
static HRESULT (WINAPI *pSHCreateStreamOnFileA)(const char *path, DWORD mode, IStream **stream);
static HRESULT (WINAPI *pIStream_Size)(IStream *stream, ULARGE_INTEGER *size);
/* Keys used for testing */
#define REG_TEST_KEY "Software\\Wine\\Test"
@ -73,6 +75,8 @@ static void init(HMODULE hshcore)
X(SHQueryValueExA);
X(SHRegGetPathA);
X(SHCopyKeyA);
X(SHCreateStreamOnFileA);
X(IStream_Size);
#undef X
}
@ -671,6 +675,120 @@ static void test_SHCopyKey(void)
delete_key( hkey, "Software\\Wine", "Test" );
}
#define CHECK_FILE_SIZE(filename,exp_size) _check_file_size(filename, exp_size, __LINE__)
static void _check_file_size(const CHAR *filename, LONG exp_size, int line)
{
HANDLE handle;
DWORD file_size = 0xdeadbeef;
handle = CreateFileA(filename, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, 0);
file_size = GetFileSize(handle, NULL);
ok_(__FILE__,line)(file_size == exp_size, "got wrong file size: %d.\n", file_size);
CloseHandle(handle);
}
#define CHECK_STREAM_SIZE(obj,exp_size) _check_stream_size(obj, exp_size, __LINE__)
static void _check_stream_size(IStream *obj, LONG exp_size, int line)
{
ULARGE_INTEGER stream_size;
STATSTG stat;
HRESULT hr;
stream_size.QuadPart = 0xdeadbeef;
hr = pIStream_Size(obj, &stream_size);
ok_(__FILE__,line)(hr == S_OK, "IStream_Size failed: 0x%08x.\n", hr);
ok_(__FILE__,line)(stream_size.QuadPart == exp_size, "Size(): got wrong size of stream: %s.\n",
wine_dbgstr_longlong(stream_size.QuadPart));
hr = IStream_Stat(obj, &stat, STATFLAG_NONAME);
ok_(__FILE__,line)(hr == S_OK, "IStream_Stat failed: 0x%08x.\n", hr);
ok_(__FILE__,line)(stat.cbSize.QuadPart == exp_size, "Stat(): got wrong size of stream: %s.\n",
wine_dbgstr_longlong(stat.cbSize.QuadPart));
}
#define CHECK_STREAM_POS(obj,exp_pos) _check_stream_pos(obj, exp_pos, __LINE__)
static void _check_stream_pos(IStream *obj, LONG exp_pos, int line)
{
LARGE_INTEGER move;
ULARGE_INTEGER pos;
HRESULT hr;
move.QuadPart = 0;
pos.QuadPart = 0xdeadbeef;
hr = IStream_Seek(obj, move, STREAM_SEEK_CUR, &pos);
ok_(__FILE__,line)(hr == S_OK, "IStream_Seek failed: 0x%08x.\n", hr);
ok_(__FILE__,line)(pos.QuadPart == exp_pos, "got wrong position: %s.\n",
wine_dbgstr_longlong(pos.QuadPart));
}
static void test_stream_size(void)
{
static const byte test_data[] = {0x1,0x2,0x3,0x4,0x5,0x6};
static const CHAR filename[] = "test_file";
IStream *stream, *stream2;
HANDLE handle;
DWORD written = 0;
ULARGE_INTEGER stream_size;
HRESULT hr;
handle = CreateFileA(filename, GENERIC_READ|GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, 0);
ok(handle != INVALID_HANDLE_VALUE, "File creation failed: 0x%08x.\n", GetLastError());
WriteFile(handle, test_data, sizeof(test_data), &written, NULL);
ok(written == sizeof(test_data), "Failed to write data into file.\n");
CloseHandle(handle);
/* in read-only mode, SetSize() will success but it has no effect on Size() and the file */
hr = pSHCreateStreamOnFileA(filename, STGM_FAILIFTHERE|STGM_READ, &stream);
ok(hr == S_OK, "SHCreateStreamOnFileA failed: 0x%08x.\n", hr);
CHECK_STREAM_SIZE(stream, sizeof(test_data));
stream_size.QuadPart = 0;
hr = IStream_SetSize(stream, stream_size);
ok(hr == S_OK, "IStream_SetSize failed: 0x%08x.\n", hr);
CHECK_STREAM_SIZE(stream, sizeof(test_data));
CHECK_STREAM_POS(stream, 0);
stream_size.QuadPart = 100;
hr = IStream_SetSize(stream, stream_size);
ok(hr == S_OK, "IStream_SetSize failed: 0x%08x.\n", hr);
CHECK_STREAM_SIZE(stream, sizeof(test_data));
CHECK_STREAM_POS(stream, 100);
IStream_Release(stream);
CHECK_FILE_SIZE(filename, sizeof(test_data));
hr = pSHCreateStreamOnFileA(filename, STGM_FAILIFTHERE|STGM_WRITE, &stream);
ok(hr == S_OK, "SHCreateStreamOnFileA failed: 0x%08x.\n", hr);
hr = pSHCreateStreamOnFileA(filename, STGM_FAILIFTHERE|STGM_READ, &stream2);
ok(hr == S_OK, "SHCreateStreamOnFileA failed: 0x%08x.\n", hr);
CHECK_STREAM_SIZE(stream, sizeof(test_data));
CHECK_STREAM_SIZE(stream2, sizeof(test_data));
CHECK_STREAM_POS(stream, 0);
CHECK_STREAM_POS(stream2, 0);
stream_size.QuadPart = 0;
hr = IStream_SetSize(stream, stream_size);
ok(hr == S_OK, "IStream_SetSize failed: 0x%08x.\n", hr);
CHECK_STREAM_SIZE(stream, 0);
CHECK_STREAM_SIZE(stream2, 0);
CHECK_STREAM_POS(stream, 0);
CHECK_STREAM_POS(stream2, 0);
stream_size.QuadPart = 100;
hr = IStream_SetSize(stream, stream_size);
ok(hr == S_OK, "IStream_SetSize failed: 0x%08x.\n", hr);
CHECK_STREAM_SIZE(stream, 100);
CHECK_STREAM_SIZE(stream2, 100);
CHECK_STREAM_POS(stream, 0);
CHECK_STREAM_POS(stream2, 0);
stream_size.QuadPart = 90;
hr = IStream_SetSize(stream2, stream_size);
ok(hr == S_OK, "IStream_SetSize failed: 0x%08x.\n", hr);
CHECK_STREAM_SIZE(stream, 100);
CHECK_STREAM_SIZE(stream2, 100);
CHECK_STREAM_POS(stream, 0);
CHECK_STREAM_POS(stream2, 90);
IStream_Release(stream);
IStream_Release(stream2);
CHECK_FILE_SIZE(filename, 100);
DeleteFileA(filename);
}
START_TEST(shcore)
{
HMODULE hshcore = LoadLibraryA("shcore.dll");
@ -695,4 +813,5 @@ START_TEST(shcore)
test_SHQueryValueEx();
test_SHRegGetPath();
test_SHCopyKey();
test_stream_size();
}