Jacek Caban : urlmon: Added Abort support to BindProtocol.
Alexandre Julliard
julliard at winehq.org
Mon Oct 4 11:04:04 CDT 2010
Module: wine
Branch: master
Commit: 0d27b740b3b3c56ebba4e57ca2c6a92add3eef76
URL: http://source.winehq.org/git/wine.git/?a=commit;h=0d27b740b3b3c56ebba4e57ca2c6a92add3eef76
Author: Jacek Caban <jacek at codeweavers.com>
Date: Mon Oct 4 14:00:48 2010 +0200
urlmon: Added Abort support to BindProtocol.
---
dlls/urlmon/bindprot.c | 15 ++++++++++---
dlls/urlmon/tests/protocol.c | 43 ++++++++++++++++++++++++++++++++++++++++-
2 files changed, 52 insertions(+), 6 deletions(-)
diff --git a/dlls/urlmon/bindprot.c b/dlls/urlmon/bindprot.c
index 01443e3..3e1fb58 100644
--- a/dlls/urlmon/bindprot.c
+++ b/dlls/urlmon/bindprot.c
@@ -423,8 +423,10 @@ static HRESULT WINAPI BindProtocol_Abort(IInternetProtocol *iface, HRESULT hrRea
DWORD dwOptions)
{
BindProtocol *This = PROTOCOL_THIS(iface);
- FIXME("(%p)->(%08x %08x)\n", This, hrReason, dwOptions);
- return E_NOTIMPL;
+
+ TRACE("(%p)->(%08x %08x)\n", This, hrReason, dwOptions);
+
+ return IInternetProtocol_Abort(This->protocol_handler, hrReason, dwOptions);
}
static HRESULT WINAPI BindProtocol_Terminate(IInternetProtocol *iface, DWORD dwOptions)
@@ -653,8 +655,13 @@ static HRESULT WINAPI ProtocolHandler_Abort(IInternetProtocol *iface, HRESULT hr
DWORD dwOptions)
{
BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
- FIXME("(%p)->(%08x %08x)\n", This, hrReason, dwOptions);
- return E_NOTIMPL;
+
+ TRACE("(%p)->(%08x %08x)\n", This, hrReason, dwOptions);
+
+ if(This->protocol && !This->reported_result)
+ return IInternetProtocol_Abort(This->protocol, hrReason, dwOptions);
+
+ return S_OK;
}
static HRESULT WINAPI ProtocolHandler_Terminate(IInternetProtocol *iface, DWORD dwOptions)
diff --git a/dlls/urlmon/tests/protocol.c b/dlls/urlmon/tests/protocol.c
index a116cf3..15c3bc4 100644
--- a/dlls/urlmon/tests/protocol.c
+++ b/dlls/urlmon/tests/protocol.c
@@ -108,6 +108,7 @@ DEFINE_EXPECT(Read2);
DEFINE_EXPECT(SetPriority);
DEFINE_EXPECT(LockRequest);
DEFINE_EXPECT(UnlockRequest);
+DEFINE_EXPECT(Abort);
DEFINE_EXPECT(MimeFilter_CreateInstance);
DEFINE_EXPECT(MimeFilter_Start);
DEFINE_EXPECT(MimeFilter_ReportProgress);
@@ -1159,8 +1160,16 @@ static ULONG WINAPI Protocol_Release(IInternetProtocol *iface)
static HRESULT WINAPI Protocol_Abort(IInternetProtocol *iface, HRESULT hrReason,
DWORD dwOptions)
{
- ok(0, "unexpected call\n");
- return E_NOTIMPL;
+ HRESULT hres;
+
+ CHECK_EXPECT(Abort);
+
+ SET_EXPECT(ReportResult);
+ hres = IInternetProtocolSink_ReportResult(binding_sink, S_OK, ERROR_SUCCESS, NULL);
+ ok(hres == S_OK, "ReportResult failed: %08x\n", hres);
+ CHECK_CALLED(ReportResult);
+
+ return S_OK;
}
static HRESULT WINAPI Protocol_Suspend(IInternetProtocol *iface)
@@ -1262,6 +1271,11 @@ static DWORD WINAPI thread_proc(PVOID arg)
else
CHECK_CALLED(Switch);
+ if(test_abort) {
+ SetEvent(event_complete);
+ return 0;
+ }
+
prot_state = 2;
if(mimefilter_test)
SET_EXPECT(MimeFilter_Switch);
@@ -2999,6 +3013,18 @@ static void test_CreateBinding(void)
IInternetBindInfo_Release(prot_bind_info);
IInternetProtocol_Release(protocol);
+ hres = IInternetSession_CreateBinding(session, NULL, test_url, NULL, NULL, &protocol, 0);
+ ok(hres == S_OK, "CreateBinding failed: %08x\n", hres);
+ ok(protocol != NULL, "protocol == NULL\n");
+
+ hres = IInternetProtocol_Abort(protocol, E_ABORT, 0);
+ ok(hres == S_OK, "Abort failed: %08x\n", hres);
+
+ hres = IInternetProtocol_Abort(protocol, E_FAIL, 0);
+ ok(hres == S_OK, "Abort failed: %08x\n", hres);
+
+ IInternetProtocol_Release(protocol);
+
hres = IInternetSession_UnregisterNameSpace(session, &ClassFactory, wsz_test);
ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres);
@@ -3064,6 +3090,17 @@ static void test_binding(int prot, DWORD grf_pi, DWORD test_flags)
IInternetProtocol_Continue(protocol, pdata);
CHECK_CALLED(Continue);
}
+ if(test_abort && prot_state == 2) {
+ SET_EXPECT(Abort);
+ hres = IInternetProtocol_Abort(protocol, E_ABORT, 0);
+ ok(hres == S_OK, "Abort failed: %08x\n", hres);
+ CHECK_CALLED(Abort);
+
+ hres = IInternetProtocol_Abort(protocol, E_ABORT, 0);
+ ok(hres == S_OK, "Abort failed: %08x\n", hres);
+ SetEvent(event_complete2);
+ break;
+ }
SetEvent(event_complete2);
}
if(direct_read)
@@ -3179,6 +3216,8 @@ START_TEST(protocol)
test_binding(HTTP_TEST, PI_MIMEVERIFICATION, TEST_EMULATEPROT|TEST_FILTER);
trace("Testing http binding (mime verification, emulate prot, direct read)...\n");
test_binding(HTTP_TEST, PI_MIMEVERIFICATION, TEST_EMULATEPROT|TEST_DIRECT_READ);
+ trace("Testing http binding (mime verification, emulate prot, abort)...\n");
+ test_binding(HTTP_TEST, PI_MIMEVERIFICATION, TEST_EMULATEPROT|TEST_ABORT);
CloseHandle(event_complete);
CloseHandle(event_complete2);
More information about the wine-cvs
mailing list