Commit 32671b1d authored by Jacek Caban's avatar Jacek Caban Committed by Alexandre Julliard

urlmon: Added Seek implementations for streams using cache file.

parent df614332
......@@ -520,8 +520,40 @@ static HRESULT WINAPI ProtocolStream_Seek(IStream *iface, LARGE_INTEGER dlibMove
DWORD dwOrigin, ULARGE_INTEGER *plibNewPosition)
{
ProtocolStream *This = impl_from_IStream(iface);
FIXME("(%p)->(%d %08x %p)\n", This, dlibMove.u.LowPart, dwOrigin, plibNewPosition);
return E_NOTIMPL;
LARGE_INTEGER new_pos;
DWORD method;
TRACE("(%p)->(%d %08x %p)\n", This, dlibMove.u.LowPart, dwOrigin, plibNewPosition);
if(This->buf->file == INVALID_HANDLE_VALUE) {
/* We should probably call protocol handler's Seek. */
FIXME("no cache file, not supported\n");
return E_FAIL;
}
switch(dwOrigin) {
case STREAM_SEEK_SET:
method = FILE_BEGIN;
break;
case STREAM_SEEK_CUR:
method = FILE_CURRENT;
break;
case STREAM_SEEK_END:
method = FILE_END;
break;
default:
WARN("Invalid origin %x\n", dwOrigin);
return E_FAIL;
}
if(!SetFilePointerEx(This->buf->file, dlibMove, &new_pos, method)) {
FIXME("SetFilePointerEx failed: %u\n", GetLastError());
return E_FAIL;
}
if(plibNewPosition)
plibNewPosition->QuadPart = new_pos.QuadPart;
return S_OK;
}
static HRESULT WINAPI ProtocolStream_SetSize(IStream *iface, ULARGE_INTEGER libNewSize)
......
......@@ -1949,6 +1949,31 @@ static HRESULT WINAPI statusclb_GetBindInfo(IBindStatusCallbackEx *iface, DWORD
return S_OK;
}
static void test_stream_seek(IStream *stream)
{
ULARGE_INTEGER new_pos;
LARGE_INTEGER pos;
HRESULT hres;
pos.QuadPart = 0;
new_pos.QuadPart = 0xdeadbeef;
hres = IStream_Seek(stream, pos, STREAM_SEEK_SET, &new_pos);
ok(hres == S_OK, "Seek failed: %08x\n", hres);
ok(!new_pos.QuadPart, "new_pos.QuadPart != 0\n");
pos.QuadPart = 0;
new_pos.QuadPart = 0xdeadbeef;
hres = IStream_Seek(stream, pos, STREAM_SEEK_END, &new_pos);
ok(hres == S_OK, "Seek failed: %08x\n", hres);
ok(new_pos.QuadPart, "new_pos.QuadPart = 0\n");
pos.QuadPart = 0;
new_pos.QuadPart = 0xdeadbeef;
hres = IStream_Seek(stream, pos, 100, &new_pos);
ok(hres == E_FAIL, "Seek failed: %08x\n", hres);
ok(new_pos.QuadPart == 0xdeadbeef, "unexpected new_pos.QuadPart\n");
}
static HRESULT WINAPI statusclb_OnDataAvailable(IBindStatusCallbackEx *iface, DWORD grfBSCF,
DWORD dwSize, FORMATETC* pformatetc, STGMEDIUM* pstgmed)
{
......@@ -1995,24 +2020,28 @@ static HRESULT WINAPI statusclb_OnDataAvailable(IBindStatusCallbackEx *iface, DW
ok(pstgmed->pUnkForRelease != NULL, "pUnkForRelease == NULL\n");
switch(pstgmed->tymed) {
case TYMED_ISTREAM:
case TYMED_ISTREAM: {
IStream *stream = U(*pstgmed).pstm;
ok(stream != NULL, "U(*pstgmed).pstm == NULL\n");
if(grfBSCF & BSCF_FIRSTDATANOTIFICATION) {
STATSTG stat;
hres = IStream_Write(U(*pstgmed).pstm, buf, 10, NULL);
hres = IStream_Write(stream, buf, 10, NULL);
ok(hres == STG_E_ACCESSDENIED,
"Write failed: %08x, expected STG_E_ACCESSDENIED\n", hres);
hres = IStream_Commit(U(*pstgmed).pstm, 0);
hres = IStream_Commit(stream, 0);
ok(hres == E_NOTIMPL, "Commit failed: %08x, expected E_NOTIMPL\n", hres);
hres = IStream_Revert(U(*pstgmed).pstm);
hres = IStream_Revert(stream);
ok(hres == E_NOTIMPL, "Revert failed: %08x, expected E_NOTIMPL\n", hres);
hres = IStream_Stat(U(*pstgmed).pstm, NULL, STATFLAG_NONAME);
hres = IStream_Stat(stream, NULL, STATFLAG_NONAME);
ok(hres == E_FAIL, "hres = %x\n", hres);
if(use_cache_file && emulate_protocol) {
hres = IStream_Stat(U(*pstgmed).pstm, &stat, STATFLAG_DEFAULT);
hres = IStream_Stat(stream, &stat, STATFLAG_DEFAULT);
ok(hres == S_OK, "hres = %x\n", hres);
ok(!lstrcmpW(stat.pwcsName, cache_file_name),
"stat.pwcsName = %s, cache_file_name = %s\n",
......@@ -2021,7 +2050,7 @@ static HRESULT WINAPI statusclb_OnDataAvailable(IBindStatusCallbackEx *iface, DW
ok(U(stat.cbSize).LowPart == (bindf&BINDF_ASYNCHRONOUS?0:6500),
"stat.cbSize.LowPart = %u\n", U(stat.cbSize).LowPart);
} else {
hres = IStream_Stat(U(*pstgmed).pstm, &stat, STATFLAG_NONAME);
hres = IStream_Stat(stream, &stat, STATFLAG_NONAME);
ok(hres == S_OK, "hres = %x\n", hres);
ok(!stat.pwcsName || broken(stat.pwcsName!=NULL),
"stat.pwcsName = %s\n", wine_dbgstr_w(stat.pwcsName));
......@@ -2034,17 +2063,19 @@ static HRESULT WINAPI statusclb_OnDataAvailable(IBindStatusCallbackEx *iface, DW
ok(stat.reserved == 0, "stat.reserved = %x\n", stat.reserved);
}
ok(U(*pstgmed).pstm != NULL, "U(*pstgmed).pstm == NULL\n");
if(callback_read) {
do {
hres = IStream_Read(U(*pstgmed).pstm, buf, 512, &readed);
hres = IStream_Read(stream, buf, 512, &readed);
if(test_protocol == HTTP_TEST && emulate_protocol && readed)
ok(buf[0] == (use_cache_file && !(bindf&BINDF_ASYNCHRONOUS) ? 'X' : '?'), "buf[0] = '%c'\n", buf[0]);
}while(hres == S_OK);
ok(hres == S_FALSE || hres == E_PENDING, "IStream_Read returned %08x\n", hres);
}
break;
if(use_cache_file && (grfBSCF & BSCF_FIRSTDATANOTIFICATION) && !(bindf & BINDF_PULLDATA))
test_stream_seek(stream);
break;
}
case TYMED_FILE:
if(test_protocol == FILE_TEST)
ok(!lstrcmpW(pstgmed->u.lpszFileName, file_url+7),
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment