| /* |
| * Copyright 2016 Hans Leidekker for CodeWeavers |
| * |
| * This library is free software; you can redistribute it and/or |
| * modify it under the terms of the GNU Lesser General Public |
| * License as published by the Free Software Foundation; either |
| * version 2.1 of the License, or (at your option) any later version. |
| * |
| * This library is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| * Lesser General Public License for more details. |
| * |
| * You should have received a copy of the GNU Lesser General Public |
| * License along with this library; if not, write to the Free Software |
| * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA |
| */ |
| |
| #include <stdarg.h> |
| |
| #include "windef.h" |
| #include "winbase.h" |
| #include "winuser.h" |
| #include "winsock2.h" |
| #include "webservices.h" |
| |
| #include "wine/debug.h" |
| #include "wine/list.h" |
| #include "wine/unicode.h" |
| #include "webservices_private.h" |
| |
| WINE_DEFAULT_DEBUG_CHANNEL(webservices); |
| |
| static const struct prop_desc channel_props[] = |
| { |
| { sizeof(ULONG), FALSE }, /* WS_CHANNEL_PROPERTY_MAX_BUFFERED_MESSAGE_SIZE */ |
| { sizeof(UINT64), FALSE }, /* WS_CHANNEL_PROPERTY_MAX_STREAMED_MESSAGE_SIZE */ |
| { sizeof(ULONG), FALSE }, /* WS_CHANNEL_PROPERTY_MAX_STREAMED_START_SIZE */ |
| { sizeof(ULONG), FALSE }, /* WS_CHANNEL_PROPERTY_MAX_STREAMED_FLUSH_SIZE */ |
| { sizeof(WS_ENCODING), FALSE }, /* WS_CHANNEL_PROPERTY_ENCODING */ |
| { sizeof(WS_ENVELOPE_VERSION), FALSE }, /* WS_CHANNEL_PROPERTY_ENVELOPE_VERSION */ |
| { sizeof(WS_ADDRESSING_VERSION), FALSE }, /* WS_CHANNEL_PROPERTY_ADDRESSING_VERSION */ |
| { sizeof(ULONG), FALSE }, /* WS_CHANNEL_PROPERTY_MAX_SESSION_DICTIONARY_SIZE */ |
| { sizeof(WS_CHANNEL_STATE), TRUE }, /* WS_CHANNEL_PROPERTY_STATE */ |
| { sizeof(WS_CALLBACK_MODEL), FALSE }, /* WS_CHANNEL_PROPERTY_ASYNC_CALLBACK_MODEL */ |
| { sizeof(WS_IP_VERSION), FALSE }, /* WS_CHANNEL_PROPERTY_IP_VERSION */ |
| { sizeof(ULONG), FALSE }, /* WS_CHANNEL_PROPERTY_RESOLVE_TIMEOUT */ |
| { sizeof(ULONG), FALSE }, /* WS_CHANNEL_PROPERTY_CONNECT_TIMEOUT */ |
| { sizeof(ULONG), FALSE }, /* WS_CHANNEL_PROPERTY_SEND_TIMEOUT */ |
| { sizeof(ULONG), FALSE }, /* WS_CHANNEL_PROPERTY_RECEIVE_RESPONSE_TIMEOUT */ |
| { sizeof(ULONG), FALSE }, /* WS_CHANNEL_PROPERTY_RECEIVE_TIMEOUT */ |
| { sizeof(ULONG), FALSE }, /* WS_CHANNEL_PROPERTY_CLOSE_TIMEOUT */ |
| { sizeof(BOOL), FALSE }, /* WS_CHANNEL_PROPERTY_ENABLE_TIMEOUTS */ |
| { sizeof(WS_TRANSFER_MODE), FALSE }, /* WS_CHANNEL_PROPERTY_TRANSFER_MODE */ |
| { sizeof(ULONG), FALSE }, /* WS_CHANNEL_PROPERTY_MULTICAST_INTERFACE */ |
| { sizeof(ULONG), FALSE }, /* WS_CHANNEL_PROPERTY_MULTICAST_HOPS */ |
| { sizeof(WS_ENDPOINT_ADDRESS), TRUE }, /* WS_CHANNEL_PROPERTY_REMOTE_ADDRESS */ |
| { sizeof(SOCKADDR_STORAGE), TRUE }, /* WS_CHANNEL_PROPERTY_REMOTE_IP_ADDRESS */ |
| { sizeof(ULONGLONG), TRUE }, /* WS_CHANNEL_PROPERTY_HTTP_CONNECTION_ID */ |
| { sizeof(WS_CUSTOM_CHANNEL_CALLBACKS), FALSE }, /* WS_CHANNEL_PROPERTY_CUSTOM_CHANNEL_CALLBACKS */ |
| { 0, FALSE }, /* WS_CHANNEL_PROPERTY_CUSTOM_CHANNEL_PARAMETERS */ |
| { sizeof(void *), FALSE }, /* WS_CHANNEL_PROPERTY_CUSTOM_CHANNEL_INSTANCE */ |
| { sizeof(WS_STRING), TRUE }, /* WS_CHANNEL_PROPERTY_TRANSPORT_URL */ |
| { sizeof(BOOL), FALSE }, /* WS_CHANNEL_PROPERTY_NO_DELAY */ |
| { sizeof(BOOL), FALSE }, /* WS_CHANNEL_PROPERTY_SEND_KEEP_ALIVES */ |
| { sizeof(ULONG), FALSE }, /* WS_CHANNEL_PROPERTY_KEEP_ALIVE_TIME */ |
| { sizeof(ULONG), FALSE }, /* WS_CHANNEL_PROPERTY_KEEP_ALIVE_INTERVAL */ |
| { sizeof(ULONG), FALSE }, /* WS_CHANNEL_PROPERTY_MAX_HTTP_SERVER_CONNECTIONS */ |
| { sizeof(BOOL), TRUE }, /* WS_CHANNEL_PROPERTY_IS_SESSION_SHUT_DOWN */ |
| { sizeof(WS_CHANNEL_TYPE), TRUE }, /* WS_CHANNEL_PROPERTY_CHANNEL_TYPE */ |
| { sizeof(ULONG), FALSE }, /* WS_CHANNEL_PROPERTY_TRIM_BUFFERED_MESSAGE_SIZE */ |
| { sizeof(WS_CHANNEL_ENCODER), FALSE }, /* WS_CHANNEL_PROPERTY_ENCODER */ |
| { sizeof(WS_CHANNEL_DECODER), FALSE }, /* WS_CHANNEL_PROPERTY_DECODER */ |
| { sizeof(WS_PROTECTION_LEVEL), TRUE }, /* WS_CHANNEL_PROPERTY_PROTECTION_LEVEL */ |
| { sizeof(WS_COOKIE_MODE), FALSE }, /* WS_CHANNEL_PROPERTY_COOKIE_MODE */ |
| { sizeof(WS_HTTP_PROXY_SETTING_MODE), FALSE }, /* WS_CHANNEL_PROPERTY_HTTP_PROXY_SETTING_MODE */ |
| { sizeof(WS_CUSTOM_HTTP_PROXY), FALSE }, /* WS_CHANNEL_PROPERTY_CUSTOM_HTTP_PROXY */ |
| { sizeof(WS_HTTP_MESSAGE_MAPPING), FALSE }, /* WS_CHANNEL_PROPERTY_HTTP_MESSAGE_MAPPING */ |
| { sizeof(BOOL), FALSE }, /* WS_CHANNEL_PROPERTY_ENABLE_HTTP_REDIRECT */ |
| { sizeof(WS_HTTP_REDIRECT_CALLBACK_CONTEXT), FALSE }, /* WS_CHANNEL_PROPERTY_HTTP_REDIRECT_CALLBACK_CONTEXT */ |
| { sizeof(BOOL), FALSE }, /* WS_CHANNEL_PROPERTY_FAULTS_AS_ERRORS */ |
| { sizeof(BOOL), FALSE }, /* WS_CHANNEL_PROPERTY_ALLOW_UNSECURED_FAULTS */ |
| { sizeof(WCHAR *), TRUE }, /* WS_CHANNEL_PROPERTY_HTTP_SERVER_SPN */ |
| { sizeof(WCHAR *), TRUE }, /* WS_CHANNEL_PROPERTY_HTTP_PROXY_SPN */ |
| { sizeof(ULONG), FALSE } /* WS_CHANNEL_PROPERTY_MAX_HTTP_REQUEST_HEADERS_BUFFER_SIZE */ |
| }; |
| |
| struct channel |
| { |
| WS_CHANNEL_TYPE type; |
| WS_CHANNEL_BINDING binding; |
| WS_CHANNEL_STATE state; |
| WS_ENDPOINT_ADDRESS addr; |
| WS_XML_WRITER *writer; |
| WS_XML_READER *reader; |
| HINTERNET http_session; |
| HINTERNET http_connect; |
| HINTERNET http_request; |
| ULONG prop_count; |
| struct prop prop[sizeof(channel_props)/sizeof(channel_props[0])]; |
| }; |
| |
| static struct channel *alloc_channel(void) |
| { |
| static const ULONG count = sizeof(channel_props)/sizeof(channel_props[0]); |
| struct channel *ret; |
| ULONG size = sizeof(*ret) + prop_size( channel_props, count ); |
| |
| if (!(ret = heap_alloc_zero( size ))) return NULL; |
| prop_init( channel_props, count, ret->prop, &ret[1] ); |
| ret->prop_count = count; |
| return ret; |
| } |
| |
| static void free_channel( struct channel *channel ) |
| { |
| if (!channel) return; |
| WsFreeWriter( channel->writer ); |
| WsFreeReader( channel->reader ); |
| WinHttpCloseHandle( channel->http_request ); |
| WinHttpCloseHandle( channel->http_connect ); |
| WinHttpCloseHandle( channel->http_session ); |
| heap_free( channel->addr.url.chars ); |
| heap_free( channel ); |
| } |
| |
| static HRESULT create_channel( WS_CHANNEL_TYPE type, WS_CHANNEL_BINDING binding, |
| const WS_CHANNEL_PROPERTY *properties, ULONG count, struct channel **ret ) |
| { |
| struct channel *channel; |
| ULONG i, msg_size = 65536; |
| HRESULT hr; |
| |
| if (!(channel = alloc_channel())) return E_OUTOFMEMORY; |
| |
| prop_set( channel->prop, channel->prop_count, WS_CHANNEL_PROPERTY_MAX_BUFFERED_MESSAGE_SIZE, |
| &msg_size, sizeof(msg_size) ); |
| |
| for (i = 0; i < count; i++) |
| { |
| hr = prop_set( channel->prop, channel->prop_count, properties[i].id, properties[i].value, |
| properties[i].valueSize ); |
| if (hr != S_OK) |
| { |
| free_channel( channel ); |
| return hr; |
| } |
| } |
| |
| channel->type = type; |
| channel->binding = binding; |
| |
| *ret = channel; |
| return S_OK; |
| } |
| |
| /************************************************************************** |
| * WsCreateChannel [webservices.@] |
| */ |
| HRESULT WINAPI WsCreateChannel( WS_CHANNEL_TYPE type, WS_CHANNEL_BINDING binding, |
| const WS_CHANNEL_PROPERTY *properties, ULONG count, |
| const WS_SECURITY_DESCRIPTION *desc, WS_CHANNEL **handle, |
| WS_ERROR *error ) |
| { |
| struct channel *channel; |
| HRESULT hr; |
| |
| TRACE( "%u %u %p %u %p %p %p\n", type, binding, properties, count, desc, handle, error ); |
| if (error) FIXME( "ignoring error parameter\n" ); |
| if (desc) FIXME( "ignoring security description\n" ); |
| |
| if (!handle) return E_INVALIDARG; |
| |
| if (type != WS_CHANNEL_TYPE_REQUEST) |
| { |
| FIXME( "channel type %u not implemented\n", type ); |
| return E_NOTIMPL; |
| } |
| if (binding != WS_HTTP_CHANNEL_BINDING) |
| { |
| FIXME( "channel binding %u not implemented\n", binding ); |
| return E_NOTIMPL; |
| } |
| |
| if ((hr = create_channel( type, binding, properties, count, &channel )) != S_OK) |
| return hr; |
| |
| *handle = (WS_CHANNEL *)channel; |
| return S_OK; |
| } |
| |
| /************************************************************************** |
| * WsFreeChannel [webservices.@] |
| */ |
| void WINAPI WsFreeChannel( WS_CHANNEL *handle ) |
| { |
| struct channel *channel = (struct channel *)handle; |
| |
| TRACE( "%p\n", handle ); |
| free_channel( channel ); |
| } |
| |
| /************************************************************************** |
| * WsGetChannelProperty [webservices.@] |
| */ |
| HRESULT WINAPI WsGetChannelProperty( WS_CHANNEL *handle, WS_CHANNEL_PROPERTY_ID id, void *buf, |
| ULONG size, WS_ERROR *error ) |
| { |
| struct channel *channel = (struct channel *)handle; |
| |
| TRACE( "%p %u %p %u %p\n", handle, id, buf, size, error ); |
| if (error) FIXME( "ignoring error parameter\n" ); |
| |
| if (!handle) return E_INVALIDARG; |
| return prop_get( channel->prop, channel->prop_count, id, buf, size ); |
| } |
| |
| /************************************************************************** |
| * WsSetChannelProperty [webservices.@] |
| */ |
| HRESULT WINAPI WsSetChannelProperty( WS_CHANNEL *handle, WS_CHANNEL_PROPERTY_ID id, const void *value, |
| ULONG size, WS_ERROR *error ) |
| { |
| struct channel *channel = (struct channel *)handle; |
| |
| TRACE( "%p %u %p %u\n", handle, id, value, size ); |
| if (error) FIXME( "ignoring error parameter\n" ); |
| |
| if (!handle) return E_INVALIDARG; |
| return prop_set( channel->prop, channel->prop_count, id, value, size ); |
| } |
| |
| static HRESULT open_channel( struct channel *channel, const WS_ENDPOINT_ADDRESS *endpoint ) |
| { |
| if (endpoint->headers || endpoint->extensions || endpoint->identity) |
| { |
| FIXME( "headers, extensions or identity not supported\n" ); |
| return E_NOTIMPL; |
| } |
| |
| if (!(channel->addr.url.chars = heap_alloc( endpoint->url.length * sizeof(WCHAR) ))) return E_OUTOFMEMORY; |
| memcpy( channel->addr.url.chars, endpoint->url.chars, endpoint->url.length * sizeof(WCHAR) ); |
| channel->addr.url.length = endpoint->url.length; |
| |
| channel->state = WS_CHANNEL_STATE_OPEN; |
| return S_OK; |
| } |
| |
| /************************************************************************** |
| * WsOpenChannel [webservices.@] |
| */ |
| HRESULT WINAPI WsOpenChannel( WS_CHANNEL *handle, const WS_ENDPOINT_ADDRESS *endpoint, |
| const WS_ASYNC_CONTEXT *ctx, WS_ERROR *error ) |
| { |
| struct channel *channel = (struct channel *)handle; |
| |
| TRACE( "%p %p %p %p\n", handle, endpoint, ctx, error ); |
| if (error) FIXME( "ignoring error parameter\n" ); |
| if (ctx) FIXME( "ignoring ctx parameter\n" ); |
| |
| if (!handle || !endpoint) return E_INVALIDARG; |
| if (channel->state != WS_CHANNEL_STATE_CREATED) return WS_E_INVALID_OPERATION; |
| |
| return open_channel( channel, endpoint ); |
| } |
| |
| static HRESULT close_channel( struct channel *channel ) |
| { |
| WinHttpCloseHandle( channel->http_request ); |
| channel->http_request = NULL; |
| WinHttpCloseHandle( channel->http_connect ); |
| channel->http_connect = NULL; |
| WinHttpCloseHandle( channel->http_session ); |
| channel->http_session = NULL; |
| |
| heap_free( channel->addr.url.chars ); |
| channel->addr.url.chars = NULL; |
| channel->addr.url.length = 0; |
| |
| channel->state = WS_CHANNEL_STATE_CLOSED; |
| return S_OK; |
| } |
| |
| /************************************************************************** |
| * WsCloseChannel [webservices.@] |
| */ |
| HRESULT WINAPI WsCloseChannel( WS_CHANNEL *handle, const WS_ASYNC_CONTEXT *ctx, WS_ERROR *error ) |
| { |
| struct channel *channel = (struct channel *)handle; |
| |
| TRACE( "%p %p %p\n", handle, ctx, error ); |
| if (error) FIXME( "ignoring error parameter\n" ); |
| if (ctx) FIXME( "ignoring ctx parameter\n" ); |
| |
| if (!handle) return E_INVALIDARG; |
| return close_channel( channel ); |
| } |
| |
| static HRESULT parse_url( const WCHAR *url, ULONG len, URL_COMPONENTS *uc ) |
| { |
| HRESULT hr = E_OUTOFMEMORY; |
| WCHAR *tmp; |
| DWORD err; |
| |
| memset( uc, 0, sizeof(*uc) ); |
| uc->dwStructSize = sizeof(*uc); |
| uc->dwHostNameLength = 128; |
| uc->lpszHostName = heap_alloc( uc->dwHostNameLength * sizeof(WCHAR) ); |
| uc->dwUrlPathLength = 128; |
| uc->lpszUrlPath = heap_alloc( uc->dwUrlPathLength * sizeof(WCHAR) ); |
| uc->dwExtraInfoLength = 128; |
| uc->lpszExtraInfo = heap_alloc( uc->dwExtraInfoLength * sizeof(WCHAR) ); |
| if (!uc->lpszHostName || !uc->lpszUrlPath || !uc->lpszExtraInfo) goto error; |
| |
| if (!WinHttpCrackUrl( url, len, ICU_DECODE, uc )) |
| { |
| if ((err = GetLastError()) != ERROR_INSUFFICIENT_BUFFER) |
| { |
| hr = HRESULT_FROM_WIN32( err ); |
| goto error; |
| } |
| if (!(tmp = heap_realloc( uc->lpszHostName, uc->dwHostNameLength * sizeof(WCHAR) ))) goto error; |
| uc->lpszHostName = tmp; |
| if (!(tmp = heap_realloc( uc->lpszUrlPath, uc->dwUrlPathLength * sizeof(WCHAR) ))) goto error; |
| uc->lpszUrlPath = tmp; |
| if (!(tmp = heap_realloc( uc->lpszExtraInfo, uc->dwExtraInfoLength * sizeof(WCHAR) ))) goto error; |
| uc->lpszExtraInfo = tmp; |
| WinHttpCrackUrl( url, len, ICU_DECODE, uc ); |
| } |
| |
| return S_OK; |
| |
| error: |
| heap_free( uc->lpszHostName ); |
| heap_free( uc->lpszUrlPath ); |
| heap_free( uc->lpszExtraInfo ); |
| return hr; |
| } |
| |
| static HRESULT connect_channel( struct channel *channel, WS_MESSAGE *msg ) |
| { |
| static const WCHAR agentW[] = |
| {'M','S','-','W','e','b','S','e','r','v','i','c','e','s','/','1','.','0',0}; |
| static const WCHAR postW[] = |
| {'P','O','S','T',0}; |
| HINTERNET ses = NULL, con = NULL, req = NULL; |
| WCHAR *path; |
| URL_COMPONENTS uc; |
| DWORD flags = 0; |
| HRESULT hr; |
| |
| if ((hr = parse_url( channel->addr.url.chars, channel->addr.url.length, &uc )) != S_OK) return hr; |
| if (!uc.dwExtraInfoLength) path = uc.lpszUrlPath; |
| else if (!(path = heap_alloc( (uc.dwUrlPathLength + uc.dwExtraInfoLength + 1) * sizeof(WCHAR) ))) |
| { |
| hr = E_OUTOFMEMORY; |
| goto done; |
| } |
| else |
| { |
| strcpyW( path, uc.lpszUrlPath ); |
| strcatW( path, uc.lpszExtraInfo ); |
| } |
| |
| switch (uc.nScheme) |
| { |
| case INTERNET_SCHEME_HTTP: break; |
| case INTERNET_SCHEME_HTTPS: |
| flags |= WINHTTP_FLAG_SECURE; |
| break; |
| |
| default: |
| FIXME( "scheme %u not supported\n", uc.nScheme ); |
| hr = E_NOTIMPL; |
| goto done; |
| } |
| |
| if (!(ses = WinHttpOpen( agentW, 0, NULL, NULL, 0 ))) |
| { |
| hr = HRESULT_FROM_WIN32( GetLastError() ); |
| goto done; |
| } |
| if (!(con = WinHttpConnect( ses, uc.lpszHostName, uc.nPort, 0 ))) |
| { |
| hr = HRESULT_FROM_WIN32( GetLastError() ); |
| goto done; |
| } |
| if (!(req = WinHttpOpenRequest( con, postW, path, NULL, NULL, NULL, flags ))) |
| { |
| hr = HRESULT_FROM_WIN32( GetLastError() ); |
| goto done; |
| } |
| |
| if ((hr = message_insert_http_headers( msg, req )) != S_OK) goto done; |
| |
| channel->http_session = ses; |
| channel->http_connect = con; |
| channel->http_request = req; |
| |
| done: |
| if (hr != S_OK) |
| { |
| WinHttpCloseHandle( req ); |
| WinHttpCloseHandle( con ); |
| WinHttpCloseHandle( ses ); |
| } |
| heap_free( uc.lpszHostName ); |
| heap_free( uc.lpszUrlPath ); |
| heap_free( uc.lpszExtraInfo ); |
| if (path != uc.lpszUrlPath) heap_free( path ); |
| return hr; |
| } |
| |
| HRESULT set_output( WS_XML_WRITER *writer ) |
| { |
| WS_XML_WRITER_TEXT_ENCODING text = { {WS_XML_WRITER_ENCODING_TYPE_TEXT}, WS_CHARSET_UTF8 }; |
| WS_XML_WRITER_BUFFER_OUTPUT buf = { {WS_XML_WRITER_OUTPUT_TYPE_BUFFER} }; |
| return WsSetOutput( writer, &text.encoding, &buf.output, NULL, 0, NULL ); |
| } |
| |
| static HRESULT write_message( WS_MESSAGE *handle, WS_XML_WRITER *writer, const WS_ELEMENT_DESCRIPTION *desc, |
| WS_WRITE_OPTION option, const void *body, ULONG size ) |
| { |
| HRESULT hr; |
| if ((hr = WsWriteEnvelopeStart( handle, writer, NULL, NULL, NULL )) != S_OK) return hr; |
| if ((hr = WsWriteBody( handle, desc, option, body, size, NULL )) != S_OK) return hr; |
| return WsWriteEnvelopeEnd( handle, NULL ); |
| } |
| |
| static HRESULT send_message( struct channel *channel, BYTE *data, ULONG len ) |
| { |
| if (!WinHttpSendRequest( channel->http_request, NULL, 0, data, len, len, 0 )) |
| return HRESULT_FROM_WIN32( GetLastError() ); |
| |
| if (!WinHttpReceiveResponse( channel->http_request, NULL )) |
| return HRESULT_FROM_WIN32( GetLastError() ); |
| return S_OK; |
| } |
| |
| HRESULT channel_send_message( WS_CHANNEL *handle, WS_MESSAGE *msg ) |
| { |
| struct channel *channel = (struct channel *)handle; |
| WS_XML_WRITER *writer; |
| WS_BYTES buf; |
| HRESULT hr; |
| |
| if ((hr = connect_channel( channel, msg )) != S_OK) return hr; |
| WsGetMessageProperty( msg, WS_MESSAGE_PROPERTY_BODY_WRITER, &writer, sizeof(writer), NULL ); |
| WsGetWriterProperty( writer, WS_XML_WRITER_PROPERTY_BYTES, &buf, sizeof(buf), NULL ); |
| return send_message( channel, buf.bytes, buf.length ); |
| } |
| |
| /************************************************************************** |
| * WsSendMessage [webservices.@] |
| */ |
| HRESULT WINAPI WsSendMessage( WS_CHANNEL *handle, WS_MESSAGE *msg, const WS_MESSAGE_DESCRIPTION *desc, |
| WS_WRITE_OPTION option, const void *body, ULONG size, const WS_ASYNC_CONTEXT *ctx, |
| WS_ERROR *error ) |
| { |
| struct channel *channel = (struct channel *)handle; |
| HRESULT hr; |
| |
| TRACE( "%p %p %p %08x %p %u %p %p\n", handle, msg, desc, option, body, size, ctx, error ); |
| if (error) FIXME( "ignoring error parameter\n" ); |
| if (ctx) FIXME( "ignoring ctx parameter\n" ); |
| |
| if (!handle || !msg || !desc) return E_INVALIDARG; |
| |
| if ((hr = WsInitializeMessage( msg, WS_REQUEST_MESSAGE, NULL, NULL )) != S_OK) return hr; |
| if ((hr = WsAddressMessage( msg, &channel->addr, NULL )) != S_OK) return hr; |
| if ((hr = message_set_action( msg, desc->action )) != S_OK) return hr; |
| |
| if (!channel->writer && (hr = WsCreateWriter( NULL, 0, &channel->writer, NULL )) != S_OK) return hr; |
| if ((hr = set_output( channel->writer )) != S_OK) return hr; |
| if ((hr = write_message( msg, channel->writer, desc->bodyElementDescription, option, body, size )) != S_OK) |
| return hr; |
| |
| return channel_send_message( handle, msg ); |
| } |
| |
| #define INITIAL_READ_BUFFER_SIZE 4096 |
| static HRESULT receive_message( struct channel *channel, ULONG max_len, char **ret, ULONG *ret_len ) |
| { |
| DWORD len, bytes_read, offset = 0, size = INITIAL_READ_BUFFER_SIZE; |
| char *buf; |
| |
| if (!(buf = heap_alloc( size ))) return E_OUTOFMEMORY; |
| *ret_len = 0; |
| for (;;) |
| { |
| if (!WinHttpQueryDataAvailable( channel->http_request, &len )) |
| { |
| heap_free( buf ); |
| return HRESULT_FROM_WIN32( GetLastError() ); |
| } |
| if (!len) break; |
| if (*ret_len + len > max_len) |
| { |
| heap_free( buf ); |
| return WS_E_QUOTA_EXCEEDED; |
| } |
| if (*ret_len + len > size) |
| { |
| char *tmp; |
| DWORD new_size = max( *ret_len + len, size * 2 ); |
| if (!(tmp = heap_realloc( buf, new_size ))) |
| { |
| heap_free( buf ); |
| return E_OUTOFMEMORY; |
| } |
| buf = tmp; |
| size = new_size; |
| } |
| if (!WinHttpReadData( channel->http_request, buf + offset, len, &bytes_read )) |
| { |
| heap_free( buf ); |
| return HRESULT_FROM_WIN32( GetLastError() ); |
| } |
| if (!bytes_read) break; |
| *ret_len += bytes_read; |
| offset += bytes_read; |
| } |
| |
| *ret = buf; |
| return S_OK; |
| } |
| |
| HRESULT channel_receive_message( WS_CHANNEL *handle, char **buf, ULONG *len ) |
| { |
| struct channel *channel = (struct channel *)handle; |
| ULONG max_len; |
| |
| WsGetChannelProperty( handle, WS_CHANNEL_PROPERTY_MAX_BUFFERED_MESSAGE_SIZE, &max_len, sizeof(max_len), NULL ); |
| return receive_message( channel, max_len, buf, len ); |
| } |
| |
| HRESULT set_input( WS_XML_READER *reader, char *data, ULONG size ) |
| { |
| WS_XML_READER_TEXT_ENCODING text = {{WS_XML_READER_ENCODING_TYPE_TEXT}, WS_CHARSET_UTF8}; |
| WS_XML_READER_BUFFER_INPUT buf; |
| |
| buf.input.inputType = WS_XML_READER_INPUT_TYPE_BUFFER; |
| buf.encodedData = data; |
| buf.encodedDataSize = size; |
| return WsSetInput( reader, &text.encoding, &buf.input, NULL, 0, NULL ); |
| } |
| |
| static HRESULT read_message( WS_MESSAGE *handle, WS_XML_READER *reader, const WS_ELEMENT_DESCRIPTION *desc, |
| WS_READ_OPTION option, WS_HEAP *heap, void *body, ULONG size ) |
| { |
| HRESULT hr; |
| if ((hr = WsReadEnvelopeStart( handle, reader, NULL, NULL, NULL )) != S_OK) return hr; |
| if ((hr = WsReadBody( handle, desc, option, heap, body, size, NULL )) != S_OK) return hr; |
| return WsReadEnvelopeEnd( handle, NULL ); |
| } |
| |
| /************************************************************************** |
| * WsReceiveMessage [webservices.@] |
| */ |
| HRESULT WINAPI WsReceiveMessage( WS_CHANNEL *handle, WS_MESSAGE *msg, const WS_MESSAGE_DESCRIPTION **desc, |
| ULONG count, WS_RECEIVE_OPTION option, WS_READ_OPTION read_option, WS_HEAP *heap, |
| void *value, ULONG size, ULONG *index, const WS_ASYNC_CONTEXT *ctx, WS_ERROR *error ) |
| { |
| struct channel *channel = (struct channel *)handle; |
| char *buf = NULL; |
| ULONG len; |
| HRESULT hr; |
| |
| TRACE( "%p %p %p %u %08x %08x %p %p %u %p %p %p\n", handle, msg, desc, count, option, read_option, heap, |
| value, size, index, ctx, error ); |
| if (error) FIXME( "ignoring error parameter\n" ); |
| if (ctx) FIXME( "ignoring ctx parameter\n" ); |
| if (index) |
| { |
| FIXME( "index parameter not supported\n" ); |
| return E_NOTIMPL; |
| } |
| if (count != 1) |
| { |
| FIXME( "no support for multiple descriptions\n" ); |
| return E_NOTIMPL; |
| } |
| if (option != WS_RECEIVE_REQUIRED_MESSAGE) |
| { |
| FIXME( "receive option %08x not supported\n", option ); |
| return E_NOTIMPL; |
| } |
| |
| if (!handle || !msg || !desc || !count) return E_INVALIDARG; |
| |
| if ((hr = channel_receive_message( handle, &buf, &len )) != S_OK) return hr; |
| if (!channel->reader && (hr = WsCreateReader( NULL, 0, &channel->reader, NULL )) != S_OK) goto done; |
| if ((hr = set_input( channel->reader, buf, len )) != S_OK) goto done; |
| hr = read_message( msg, channel->reader, desc[0]->bodyElementDescription, read_option, heap, value, size ); |
| |
| done: |
| heap_free( buf ); |
| return hr; |
| } |