Sweden-Number/dlls/hnetcfg/port.c

2331 lines
67 KiB
C

/*
* Copyright 2009 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 <stdio.h>
#define COBJMACROS
#include "windef.h"
#include "winbase.h"
#include "winuser.h"
#include "string.h"
#include "initguid.h"
#include "assert.h"
#include "winsock2.h"
#include "winhttp.h"
#include "shlwapi.h"
#include "xmllite.h"
#include "ole2.h"
#include "netfw.h"
#include "natupnp.h"
#include "wine/debug.h"
#include "hnetcfg_private.h"
WINE_DEFAULT_DEBUG_CHANNEL(hnetcfg);
struct port_mapping
{
BSTR external_ip;
LONG external;
BSTR protocol;
LONG internal;
BSTR client;
VARIANT_BOOL enabled;
BSTR descr;
};
struct xml_value_desc
{
const WCHAR *name;
BSTR value;
};
static struct
{
LONG refs;
BOOL winsock_initialized;
WCHAR locationW[256];
HINTERNET session, connection;
WCHAR desc_urlpath[128];
WCHAR control_url[256];
unsigned int version;
struct port_mapping *mappings;
unsigned int mapping_count;
}
upnp_gateway_connection;
static SRWLOCK upnp_gateway_connection_lock = SRWLOCK_INIT;
static void free_port_mapping( struct port_mapping *mapping )
{
SysFreeString( mapping->external_ip );
SysFreeString( mapping->protocol );
SysFreeString( mapping->client );
SysFreeString( mapping->descr );
}
static void free_mappings(void)
{
unsigned int i;
for (i = 0; i < upnp_gateway_connection.mapping_count; ++i)
free_port_mapping( &upnp_gateway_connection.mappings[i] );
free( upnp_gateway_connection.mappings );
upnp_gateway_connection.mappings = NULL;
upnp_gateway_connection.mapping_count = 0;
}
static BOOL copy_port_mapping( struct port_mapping *dst, const struct port_mapping *src )
{
memset( dst, 0, sizeof(*dst) );
#define COPY_BSTR_CHECK(name) if (src->name && !(dst->name = SysAllocString( src->name ))) \
{ \
free_port_mapping( dst ); \
return FALSE; \
}
COPY_BSTR_CHECK( external_ip );
COPY_BSTR_CHECK( protocol );
COPY_BSTR_CHECK( client );
COPY_BSTR_CHECK( descr );
#undef COPY_BSTR_CHECK
dst->external = src->external;
dst->internal = src->internal;
dst->enabled = src->enabled;
return TRUE;
}
static BOOL parse_search_response( char *response, WCHAR *locationW, unsigned int location_size )
{
char *saveptr = NULL, *tok, *tok2;
unsigned int status;
tok = strtok_s( response, "\n", &saveptr );
if (!tok) return FALSE;
/* HTTP/1.1 200 OK */
tok2 = strtok( tok, " " );
if (!tok2) return FALSE;
tok2 = strtok( NULL, " " );
if (!tok2) return FALSE;
status = atoi( tok2 );
if (status != HTTP_STATUS_OK)
{
WARN( "status %u.\n", status );
return FALSE;
}
while ((tok = strtok_s( NULL, "\n", &saveptr )))
{
tok2 = strtok( tok, " " );
if (!tok2) continue;
if (!stricmp( tok2, "LOCATION:" ))
{
tok2 = strtok( NULL, " \r" );
if (!tok2)
{
WARN( "Error parsing location.\n" );
return FALSE;
}
return !!MultiByteToWideChar( CP_UTF8, 0, tok2, -1, locationW, location_size / 2 );
}
}
return FALSE;
}
static BOOL parse_desc_xml( const char *desc_xml )
{
static const WCHAR urn_wanipconnection[] = L"urn:schemas-upnp-org:service:WANIPConnection:";
WCHAR control_url[ARRAY_SIZE(upnp_gateway_connection.control_url)];
BOOL service_type_matches, control_url_found, found = FALSE;
unsigned int version = 0;
XmlNodeType node_type;
IXmlReader *reader;
const WCHAR *value;
BOOL ret = FALSE;
IStream *stream;
HRESULT hr;
if (!(stream = SHCreateMemStream( (BYTE *)desc_xml, strlen( desc_xml ) + 1 ))) return FALSE;
if (FAILED(hr = CreateXmlReader( &IID_IXmlReader, (void **)&reader, NULL )))
{
IStream_Release( stream );
return FALSE;
}
if (FAILED(hr = IXmlReader_SetInput( reader, (IUnknown*)stream ))) goto done;
while (SUCCEEDED(IXmlReader_Read( reader, &node_type )) && node_type != XmlNodeType_None)
{
if (node_type != XmlNodeType_Element) continue;
if (FAILED(IXmlReader_GetLocalName( reader, &value, NULL ))) goto done;
if (wcsicmp( value, L"service" )) continue;
control_url_found = service_type_matches = FALSE;
while (SUCCEEDED(IXmlReader_Read( reader, &node_type )))
{
if (node_type != XmlNodeType_Element && node_type != XmlNodeType_EndElement) continue;
if (FAILED(IXmlReader_GetLocalName( reader, &value, NULL )))
{
WARN( "IXmlReader_GetLocalName failed.\n" );
goto done;
}
if (node_type == XmlNodeType_EndElement)
{
if (!wcsicmp( value, L"service" )) break;
continue;
}
if (!wcsicmp( value, L"serviceType" ))
{
if (FAILED(IXmlReader_Read(reader, &node_type ))) goto done;
if (node_type != XmlNodeType_Text) goto done;
if (FAILED(IXmlReader_GetValue( reader, &value, NULL ))) goto done;
if (wcsnicmp( value, urn_wanipconnection, ARRAY_SIZE(urn_wanipconnection) - 1 )) break;
version = _wtoi( value + ARRAY_SIZE(urn_wanipconnection) - 1 );
service_type_matches = version >= 1;
}
else if (!wcsicmp( value, L"controlURL" ))
{
if (FAILED(IXmlReader_Read(reader, &node_type ))) goto done;
if (node_type != XmlNodeType_Text) goto done;
if (FAILED(IXmlReader_GetValue( reader, &value, NULL ))) goto done;
if (wcslen( value ) + 1 > ARRAY_SIZE(control_url)) goto done;
wcscpy( control_url, value );
control_url_found = TRUE;
}
}
if (service_type_matches && control_url_found)
{
if (found)
{
FIXME( "Found another WANIPConnection service, ignoring.\n" );
continue;
}
found = TRUE;
wcscpy( upnp_gateway_connection.control_url, control_url );
upnp_gateway_connection.version = version;
}
}
ret = found;
done:
IXmlReader_Release( reader );
IStream_Release( stream );
return ret;
}
static BOOL get_xml_elements( const char *desc_xml, struct xml_value_desc *values, unsigned int value_count )
{
XmlNodeType node_type;
IXmlReader *reader;
const WCHAR *value;
BOOL ret = FALSE;
IStream *stream;
unsigned int i;
HRESULT hr;
for (i = 0; i < value_count; ++i) assert( !values[i].value );
if (!(stream = SHCreateMemStream( (BYTE *)desc_xml, strlen( desc_xml ) + 1 ))) return FALSE;
if (FAILED(hr = CreateXmlReader( &IID_IXmlReader, (void **)&reader, NULL )))
{
IStream_Release( stream );
return FALSE;
}
if (FAILED(hr = IXmlReader_SetInput( reader, (IUnknown*)stream ))) goto done;
while (SUCCEEDED(IXmlReader_Read( reader, &node_type )) && node_type != XmlNodeType_None)
{
if (node_type != XmlNodeType_Element) continue;
if (FAILED(IXmlReader_GetQualifiedName( reader, &value, NULL ))) goto done;
for (i = 0; i < value_count; ++i)
if (!wcsicmp( value, values[i].name )) break;
if (i == value_count) continue;
if (FAILED(IXmlReader_Read(reader, &node_type ))) goto done;
if (node_type != XmlNodeType_Text)
{
if (node_type == XmlNodeType_EndElement) value = L"";
else goto done;
}
else
{
if (FAILED(IXmlReader_GetValue( reader, &value, NULL ))) goto done;
}
if (values[i].value)
{
WARN( "Duplicate value %s.\n", debugstr_w(values[i].name) );
goto done;
}
if (!(values[i].value = SysAllocString( value ))) goto done;
}
ret = TRUE;
done:
if (!ret)
{
for (i = 0; i < value_count; ++i)
{
SysFreeString( values[i].value );
values[i].value = NULL;
}
}
IXmlReader_Release( reader );
IStream_Release( stream );
return ret;
}
static BOOL open_gateway_connection(void)
{
static const int timeout = 3000;
WCHAR hostname[64];
URL_COMPONENTS url;
memset( &url, 0, sizeof(url) );
url.dwStructSize = sizeof(url);
url.lpszHostName = hostname;
url.dwHostNameLength = ARRAY_SIZE(hostname);
url.lpszUrlPath = upnp_gateway_connection.desc_urlpath;
url.dwUrlPathLength = ARRAY_SIZE(upnp_gateway_connection.desc_urlpath);
if (!WinHttpCrackUrl( upnp_gateway_connection.locationW, 0, 0, &url )) return FALSE;
upnp_gateway_connection.session = WinHttpOpen( L"hnetcfg", WINHTTP_ACCESS_TYPE_NO_PROXY,
WINHTTP_NO_PROXY_NAME, WINHTTP_NO_PROXY_BYPASS, 0 );
if (!upnp_gateway_connection.session) return FALSE;
if (!WinHttpSetTimeouts( upnp_gateway_connection.session, timeout, timeout, timeout, timeout ))
return FALSE;
TRACE( "hostname %s, urlpath %s, port %u.\n",
debugstr_w(hostname), debugstr_w(upnp_gateway_connection.desc_urlpath), url.nPort );
upnp_gateway_connection.connection = WinHttpConnect ( upnp_gateway_connection.session, hostname, url.nPort, 0 );
if (!upnp_gateway_connection.connection)
{
WARN( "WinHttpConnect error %lu.\n", GetLastError() );
return FALSE;
}
return TRUE;
}
static BOOL get_control_url(void)
{
static const WCHAR *accept_types[] =
{
L"text/xml",
NULL
};
unsigned int desc_xml_size, offset;
DWORD size, status = 0;
HINTERNET request;
char *desc_xml;
BOOL ret;
request = WinHttpOpenRequest( upnp_gateway_connection.connection, NULL, upnp_gateway_connection.desc_urlpath, NULL,
WINHTTP_NO_REFERER, accept_types, 0 );
if (!request) return FALSE;
if (!WinHttpSendRequest( request, WINHTTP_NO_ADDITIONAL_HEADERS, 0, NULL, 0, 0, 0 ))
{
WARN( "Error sending request %lu.\n", GetLastError() );
WinHttpCloseHandle( request );
return FALSE;
}
if (!WinHttpReceiveResponse(request, NULL))
{
WARN( "Error receiving response %lu.\n", GetLastError() );
WinHttpCloseHandle( request );
return FALSE;
}
size = sizeof(status);
if (!WinHttpQueryHeaders( request, WINHTTP_QUERY_STATUS_CODE | WINHTTP_QUERY_FLAG_NUMBER,
NULL, &status, &size, NULL) || status != HTTP_STATUS_OK )
{
WARN( "Error response from server, error %lu, http status %lu.\n", GetLastError(), status );
WinHttpCloseHandle( request );
return FALSE;
}
desc_xml_size = 1024;
desc_xml = malloc( desc_xml_size );
offset = 0;
while (WinHttpReadData( request, desc_xml + offset, desc_xml_size - offset - 1, &size ) && size)
{
offset += size;
if (offset + 1 == desc_xml_size)
{
char *new;
desc_xml_size *= 2;
if (!(new = realloc( desc_xml, desc_xml_size )))
{
ERR( "No memory.\n" );
break;
}
desc_xml = new;
}
}
desc_xml[offset] = 0;
WinHttpCloseHandle( request );
ret = parse_desc_xml( desc_xml );
free( desc_xml );
return ret;
}
static void gateway_connection_cleanup(void)
{
TRACE( ".\n" );
free_mappings();
WinHttpCloseHandle( upnp_gateway_connection.connection );
WinHttpCloseHandle( upnp_gateway_connection.session );
if (upnp_gateway_connection.winsock_initialized) WSACleanup();
memset( &upnp_gateway_connection, 0, sizeof(upnp_gateway_connection) );
}
static BOOL request_service( const WCHAR *function, const struct xml_value_desc *request_param,
unsigned int request_param_count, struct xml_value_desc *result,
unsigned int result_count, DWORD *http_status, BSTR *server_error_code_str )
{
static const char request_template_header[] =
"<?xml version=\"1.0\"?>\r\n"
"<SOAP-ENV:Envelope xmlns:SOAP-ENV=\"http://schemas.xmlsoap.org/soap/envelope/\" "
"SOAP-ENV:encodingStyle=\"http://schemas.xmlsoap.org/soap/encoding/\">\r\n"
" <SOAP-ENV:Body>\r\n"
" <m:%S xmlns:m=\"urn:schemas-upnp-org:service:WANIPConnection:%u\">\r\n";
static const char request_template_footer[] =
" </m:%S>\r\n"
" </SOAP-ENV:Body>\r\n"
"</SOAP-ENV:Envelope>\r\n";
unsigned int request_data_size, request_len, offset, i, reply_buffer_size;
char *request_data, *reply_buffer = NULL, *ptr;
struct xml_value_desc error_value_desc;
WCHAR request_headers[1024];
HINTERNET request = NULL;
BOOL ret = FALSE;
DWORD size;
*server_error_code_str = NULL;
request_data_size = strlen(request_template_header) + strlen(request_template_footer) + 2 * wcslen( function )
+ 9 /* version + zero terminator */;
for (i = 0; i < request_param_count; ++i)
{
request_data_size += 13 + 2 * wcslen( request_param[i].name ) + wcslen( request_param[i].value );
}
if (!(request_data = malloc( request_data_size ))) return FALSE;
request = WinHttpOpenRequest( upnp_gateway_connection.connection, L"POST", upnp_gateway_connection.control_url,
NULL, WINHTTP_NO_REFERER, WINHTTP_DEFAULT_ACCEPT_TYPES, 0 );
if (!request) goto done;
ptr = request_data;
snprintf( ptr, request_data_size, request_template_header, function, upnp_gateway_connection.version );
offset = strlen( ptr );
ptr += offset;
request_data_size -= offset;
for (i = 0; i < request_param_count; ++i)
{
snprintf( ptr, request_data_size, " <%S>%S</%S>\r\n",
request_param[i].name, request_param[i].value, request_param[i].name);
offset = strlen( ptr );
ptr += offset;
request_data_size -= offset;
}
snprintf( ptr, request_data_size, request_template_footer, function );
request_len = strlen( request_data );
swprintf( request_headers, ARRAY_SIZE(request_headers),
L"SOAPAction: \"urn:schemas-upnp-org:service:WANIPConnection:%u#%s\"\r\n"
L"Content-Type: text/xml",
upnp_gateway_connection.version, function );
if (!WinHttpSendRequest( request, request_headers, -1, request_data, request_len, request_len, 0 ))
{
WARN( "Error sending request %lu.\n", GetLastError() );
goto done;
}
if (!WinHttpReceiveResponse(request, NULL))
{
WARN( "Error receiving response %lu.\n", GetLastError() );
goto done;
}
size = sizeof(*http_status);
if (!WinHttpQueryHeaders( request, WINHTTP_QUERY_STATUS_CODE | WINHTTP_QUERY_FLAG_NUMBER,
NULL, http_status, &size, NULL) || *http_status != HTTP_STATUS_OK )
{
if (*http_status != HTTP_STATUS_SERVER_ERROR)
{
ret = TRUE;
goto done;
}
}
offset = 0;
reply_buffer_size = 1024;
if (!(reply_buffer = malloc( reply_buffer_size ))) goto done;
while ((ret = WinHttpReadData( request, reply_buffer + offset, reply_buffer_size - offset - 1, &size )) && size)
{
offset += size;
if (offset + 1 == reply_buffer_size)
{
char *new;
reply_buffer_size *= 2;
if (!(new = realloc( reply_buffer, reply_buffer_size ))) goto done;
reply_buffer = new;
}
}
reply_buffer[offset] = 0;
if (*http_status == HTTP_STATUS_OK) ret = get_xml_elements( reply_buffer, result, result_count );
else
{
error_value_desc.name = L"errorCode";
error_value_desc.value = NULL;
if ((ret = get_xml_elements( reply_buffer, &error_value_desc, 1 )))
*server_error_code_str = error_value_desc.value;
}
done:
free( reply_buffer );
free( request_data );
WinHttpCloseHandle( request );
return ret;
}
enum port_mapping_parameter
{
PM_EXTERNAL_IP,
PM_EXTERNAL,
PM_PROTOCOL,
PM_INTERNAL,
PM_CLIENT,
PM_ENABLED,
PM_DESC,
PM_LEASE_DURATION,
PM_LAST,
PM_REMOVE_PORT_LAST = PM_INTERNAL,
};
static struct xml_value_desc port_mapping_template[] =
{
{ L"NewRemoteHost" },
{ L"NewExternalPort" },
{ L"NewProtocol" },
{ L"NewInternalPort" },
{ L"NewInternalClient" },
{ L"NewEnabled" },
{ L"NewPortMappingDescription" },
{ L"NewLeaseDuration" },
};
static LONG long_from_bstr( BSTR s )
{
if (!s) return 0;
return _wtoi( s );
}
static BSTR mapping_move_bstr( BSTR *s )
{
BSTR ret;
if (*s)
{
ret = *s;
*s = NULL;
}
else if (!(ret = SysAllocString( L"" )))
{
ERR( "No memory.\n" );
}
return ret;
}
static void update_mapping_list(void)
{
struct xml_value_desc mapping_desc[ARRAY_SIZE(port_mapping_template)];
struct xml_value_desc index_param;
struct port_mapping *new_mappings;
unsigned int i, index;
WCHAR index_str[9];
BSTR error_str;
DWORD status;
BOOL ret;
free_mappings();
index_param.name = L"NewPortMappingIndex";
index = 0;
while (1)
{
new_mappings = realloc( upnp_gateway_connection.mappings, (index + 1) * sizeof(*new_mappings) );
if (!new_mappings) break;
upnp_gateway_connection.mappings = new_mappings;
memcpy( mapping_desc, port_mapping_template, sizeof(mapping_desc) );
swprintf( index_str, ARRAY_SIZE(index_str), L"%u", index );
index_param.value = SysAllocString( index_str );
ret = request_service( L"GetGenericPortMappingEntry", &index_param, 1,
mapping_desc, ARRAY_SIZE(mapping_desc), &status, &error_str );
SysFreeString( index_param.value );
if (!ret) break;
if (status != HTTP_STATUS_OK)
{
if (error_str)
{
if (long_from_bstr( error_str ) != 713)
WARN( "Server returned error %s.\n", debugstr_w(error_str) );
SysFreeString( error_str );
}
break;
}
new_mappings[index].external_ip = mapping_move_bstr( &mapping_desc[PM_EXTERNAL_IP].value );
new_mappings[index].external = long_from_bstr( mapping_desc[PM_EXTERNAL].value );
new_mappings[index].protocol = mapping_move_bstr( &mapping_desc[PM_PROTOCOL].value );
new_mappings[index].internal = long_from_bstr( mapping_desc[PM_INTERNAL].value );
new_mappings[index].client = mapping_move_bstr( &mapping_desc[PM_CLIENT].value );
if (mapping_desc[PM_ENABLED].value && (!wcsicmp( mapping_desc[PM_ENABLED].value, L"true" )
|| long_from_bstr( mapping_desc[PM_ENABLED].value )))
new_mappings[index].enabled = VARIANT_TRUE;
else
new_mappings[index].enabled = VARIANT_FALSE;
new_mappings[index].descr = mapping_move_bstr( &mapping_desc[PM_DESC].value );
TRACE( "%s %s %s:%lu -> %s:%lu, enabled %d.\n", debugstr_w(new_mappings[index].descr),
debugstr_w(new_mappings[index].protocol), debugstr_w(new_mappings[index].external_ip),
new_mappings[index].external, debugstr_w(new_mappings[index].client),
new_mappings[index].internal, new_mappings[index].enabled );
for (i = 0; i < ARRAY_SIZE(mapping_desc); ++i)
SysFreeString( mapping_desc[i].value );
upnp_gateway_connection.mappings = new_mappings;
upnp_gateway_connection.mapping_count = ++index;
}
}
static BOOL remove_port_mapping( LONG port, BSTR protocol )
{
struct xml_value_desc mapping_desc[PM_REMOVE_PORT_LAST];
DWORD status = 0;
BSTR error_str;
WCHAR portW[6];
BOOL ret;
AcquireSRWLockExclusive( &upnp_gateway_connection_lock );
memcpy( mapping_desc, port_mapping_template, sizeof(mapping_desc) );
swprintf( portW, ARRAY_SIZE(portW), L"%u", port );
mapping_desc[PM_EXTERNAL_IP].value = SysAllocString( L"" );
mapping_desc[PM_EXTERNAL].value = SysAllocString( portW );
mapping_desc[PM_PROTOCOL].value = protocol;
ret = request_service( L"DeletePortMapping", mapping_desc, PM_REMOVE_PORT_LAST,
NULL, 0, &status, &error_str );
if (ret && status != HTTP_STATUS_OK)
{
WARN( "status %lu, server returned error %s.\n", status, debugstr_w(error_str) );
SysFreeString( error_str );
ret = FALSE;
}
else if (!ret)
{
WARN( "Request failed.\n" );
}
update_mapping_list();
ReleaseSRWLockExclusive( &upnp_gateway_connection_lock );
SysFreeString( mapping_desc[PM_EXTERNAL_IP].value );
SysFreeString( mapping_desc[PM_EXTERNAL].value );
return ret;
}
static BOOL add_port_mapping( LONG external, BSTR protocol, LONG internal, BSTR client,
VARIANT_BOOL enabled, BSTR description )
{
struct xml_value_desc mapping_desc[PM_LAST];
WCHAR externalW[6], internalW[6];
DWORD status = 0;
BSTR error_str;
BOOL ret;
AcquireSRWLockExclusive( &upnp_gateway_connection_lock );
memcpy( mapping_desc, port_mapping_template, sizeof(mapping_desc) );
swprintf( externalW, ARRAY_SIZE(externalW), L"%u", external );
swprintf( internalW, ARRAY_SIZE(internalW), L"%u", internal );
mapping_desc[PM_EXTERNAL_IP].value = SysAllocString( L"" );
mapping_desc[PM_EXTERNAL].value = SysAllocString( externalW );
mapping_desc[PM_PROTOCOL].value = protocol;
mapping_desc[PM_INTERNAL].value = SysAllocString( internalW );
mapping_desc[PM_CLIENT].value = client;
mapping_desc[PM_ENABLED].value = SysAllocString( enabled ? L"1" : L"0" );
mapping_desc[PM_DESC].value = description;
mapping_desc[PM_LEASE_DURATION].value = SysAllocString( L"0" );
ret = request_service( L"AddPortMapping", mapping_desc, PM_LAST,
NULL, 0, &status, &error_str );
if (ret && status != HTTP_STATUS_OK)
{
WARN( "status %lu, server returned error %s.\n", status, debugstr_w(error_str) );
SysFreeString( error_str );
ret = FALSE;
}
else if (!ret)
{
WARN( "Request failed.\n" );
}
update_mapping_list();
ReleaseSRWLockExclusive( &upnp_gateway_connection_lock );
SysFreeString( mapping_desc[PM_EXTERNAL_IP].value );
SysFreeString( mapping_desc[PM_EXTERNAL].value );
SysFreeString( mapping_desc[PM_INTERNAL].value );
SysFreeString( mapping_desc[PM_ENABLED].value );
SysFreeString( mapping_desc[PM_LEASE_DURATION].value );
return ret;
}
static BOOL init_gateway_connection(void)
{
static const char upnp_search_request[] =
"M-SEARCH * HTTP/1.1\r\n"
"HOST:239.255.255.250:1900\r\n"
"ST:upnp:rootdevice\r\n"
"MX:2\r\n"
"MAN:\"ssdp:discover\"\r\n"
"\r\n";
const DWORD timeout = 1000;
int len, address_len;
char buffer[2048];
SOCKADDR_IN addr;
WSADATA wsa_data;
unsigned int i;
SOCKET s;
upnp_gateway_connection.winsock_initialized = WSAStartup( MAKEWORD( 2, 2 ), &wsa_data );
if ((s = socket( AF_INET, SOCK_DGRAM, IPPROTO_UDP )) == -1)
{
ERR( "Failed to create socket, error %u.\n", WSAGetLastError() );
return FALSE;
}
if (setsockopt(s, SOL_SOCKET, SO_RCVTIMEO, (const char *)&timeout, sizeof(timeout)) == SOCKET_ERROR)
{
ERR( "setsockopt(SO_RCVTIME0) failed, error %u.\n", WSAGetLastError() );
closesocket( s );
return FALSE;
}
addr.sin_family = AF_INET;
addr.sin_port = htons( 1900 );
addr.sin_addr.S_un.S_addr = inet_addr( "239.255.255.250" );
if (sendto( s, upnp_search_request, strlen( upnp_search_request ), 0, (SOCKADDR *)&addr, sizeof(addr) )
== SOCKET_ERROR)
{
ERR( "sendto failed, error %u\n", WSAGetLastError() );
closesocket( s );
return FALSE;
}
/* Windows has a dedicated SSDP discovery service which maintains gateway device info and does
* not usually delay in get_StaticPortMappingCollection(). Although it may still delay depending
* on network connection state and always delays in IUPnPNAT_get_NATEventManager(). */
FIXME( "Waiting for reply from router.\n" );
for (i = 0; i < 2; ++i)
{
address_len = sizeof(addr);
len = recvfrom( s, buffer, sizeof(buffer) - 1, 0, (SOCKADDR *)&addr, &address_len );
if (len == -1)
{
if (WSAGetLastError() != WSAETIMEDOUT)
{
WARN( "recvfrom error %u.\n", WSAGetLastError() );
closesocket( s );
return FALSE;
}
}
else break;
}
closesocket( s );
if (i == 2)
{
TRACE( "No reply from router.\n" );
return FALSE;
}
TRACE( "Received reply from gateway, len %d.\n", len );
buffer[len] = 0;
if (!parse_search_response( buffer, upnp_gateway_connection.locationW, sizeof(upnp_gateway_connection.locationW) ))
{
WARN( "Error parsing response.\n" );
return FALSE;
}
TRACE( "Gateway description location %s.\n", debugstr_w(upnp_gateway_connection.locationW) );
if (!open_gateway_connection())
{
WARN( "Error opening gateway connection.\n" );
return FALSE;
}
TRACE( "Opened gateway connection.\n" );
if (!get_control_url())
{
WARN( "Could not get_control URL.\n" );
gateway_connection_cleanup();
return FALSE;
}
TRACE( "control_url %s, version %u.\n", debugstr_w(upnp_gateway_connection.control_url),
upnp_gateway_connection.version );
update_mapping_list();
return TRUE;
}
static BOOL grab_gateway_connection(void)
{
AcquireSRWLockExclusive( &upnp_gateway_connection_lock );
if (!upnp_gateway_connection.refs && !init_gateway_connection())
{
gateway_connection_cleanup();
ReleaseSRWLockExclusive( &upnp_gateway_connection_lock );
return FALSE;
}
++upnp_gateway_connection.refs;
ReleaseSRWLockExclusive( &upnp_gateway_connection_lock );
return TRUE;
}
static void release_gateway_connection(void)
{
AcquireSRWLockExclusive( &upnp_gateway_connection_lock );
assert( upnp_gateway_connection.refs );
if (!--upnp_gateway_connection.refs)
gateway_connection_cleanup();
ReleaseSRWLockExclusive( &upnp_gateway_connection_lock );
}
static BOOL find_port_mapping( LONG port, BSTR protocol, struct port_mapping *ret )
{
unsigned int i;
BOOL found;
AcquireSRWLockExclusive( &upnp_gateway_connection_lock );
for (i = 0; i < upnp_gateway_connection.mapping_count; ++i)
{
if (upnp_gateway_connection.mappings[i].external == port
&& !wcscmp( upnp_gateway_connection.mappings[i].protocol, protocol ))
break;
}
found = i < upnp_gateway_connection.mapping_count;
if (found) copy_port_mapping( ret, &upnp_gateway_connection.mappings[i] );
ReleaseSRWLockExclusive( &upnp_gateway_connection_lock );
return found;
}
static unsigned int get_port_mapping_range( unsigned int index, unsigned int count, struct port_mapping *ret )
{
unsigned int i;
AcquireSRWLockExclusive( &upnp_gateway_connection_lock );
for (i = 0; i < count && index + i < upnp_gateway_connection.mapping_count; ++i)
if (!copy_port_mapping( &ret[i], &upnp_gateway_connection.mappings[index + i] ))
{
ERR( "No memory.\n" );
break;
}
ReleaseSRWLockExclusive( &upnp_gateway_connection_lock );
return i;
}
static unsigned int get_port_mapping_count(void)
{
unsigned int ret;
AcquireSRWLockExclusive( &upnp_gateway_connection_lock );
ret = upnp_gateway_connection.mapping_count;
ReleaseSRWLockExclusive( &upnp_gateway_connection_lock );
return ret;
}
static BOOL is_valid_protocol( BSTR protocol )
{
if (!protocol) return FALSE;
return !wcscmp( protocol, L"UDP" ) || !wcscmp( protocol, L"TCP" );
}
struct static_port_mapping
{
IStaticPortMapping IStaticPortMapping_iface;
LONG refs;
struct port_mapping data;
};
static inline struct static_port_mapping *impl_from_IStaticPortMapping( IStaticPortMapping *iface )
{
return CONTAINING_RECORD(iface, struct static_port_mapping, IStaticPortMapping_iface);
}
static ULONG WINAPI static_port_mapping_AddRef(
IStaticPortMapping *iface )
{
struct static_port_mapping *mapping = impl_from_IStaticPortMapping( iface );
return InterlockedIncrement( &mapping->refs );
}
static ULONG WINAPI static_port_mapping_Release(
IStaticPortMapping *iface )
{
struct static_port_mapping *mapping = impl_from_IStaticPortMapping( iface );
LONG refs = InterlockedDecrement( &mapping->refs );
if (!refs)
{
TRACE("destroying %p\n", mapping);
free_port_mapping( &mapping->data );
free( mapping );
}
return refs;
}
static HRESULT WINAPI static_port_mapping_QueryInterface(
IStaticPortMapping *iface,
REFIID riid,
void **ppvObject )
{
struct static_port_mapping *mapping = impl_from_IStaticPortMapping( iface );
TRACE("%p %s %p\n", mapping, debugstr_guid( riid ), ppvObject );
if ( IsEqualGUID( riid, &IID_IStaticPortMapping ) ||
IsEqualGUID( riid, &IID_IDispatch ) ||
IsEqualGUID( riid, &IID_IUnknown ) )
{
*ppvObject = iface;
}
else
{
FIXME("interface %s not implemented\n", debugstr_guid(riid));
return E_NOINTERFACE;
}
IStaticPortMapping_AddRef( iface );
return S_OK;
}
static HRESULT WINAPI static_port_mapping_GetTypeInfoCount(
IStaticPortMapping *iface,
UINT *pctinfo )
{
struct static_port_mapping *mapping = impl_from_IStaticPortMapping( iface );
TRACE("%p %p\n", mapping, pctinfo);
*pctinfo = 1;
return S_OK;
}
static HRESULT WINAPI static_port_mapping_GetTypeInfo(
IStaticPortMapping *iface,
UINT iTInfo,
LCID lcid,
ITypeInfo **ppTInfo )
{
struct static_port_mapping *mapping = impl_from_IStaticPortMapping( iface );
TRACE("%p %u %lu %p\n", mapping, iTInfo, lcid, ppTInfo);
return get_typeinfo( IStaticPortMapping_tid, ppTInfo );
}
static HRESULT WINAPI static_port_mapping_GetIDsOfNames(
IStaticPortMapping *iface,
REFIID riid,
LPOLESTR *rgszNames,
UINT cNames,
LCID lcid,
DISPID *rgDispId )
{
struct static_port_mapping *mapping = impl_from_IStaticPortMapping( iface );
ITypeInfo *typeinfo;
HRESULT hr;
TRACE("%p %s %p %u %lu %p\n", mapping, debugstr_guid(riid), rgszNames, cNames, lcid, rgDispId);
hr = get_typeinfo( IStaticPortMapping_tid, &typeinfo );
if (SUCCEEDED(hr))
{
hr = ITypeInfo_GetIDsOfNames( typeinfo, rgszNames, cNames, rgDispId );
ITypeInfo_Release( typeinfo );
}
return hr;
}
static HRESULT WINAPI static_port_mapping_Invoke(
IStaticPortMapping *iface,
DISPID dispIdMember,
REFIID riid,
LCID lcid,
WORD wFlags,
DISPPARAMS *pDispParams,
VARIANT *pVarResult,
EXCEPINFO *pExcepInfo,
UINT *puArgErr )
{
struct static_port_mapping *mapping = impl_from_IStaticPortMapping( iface );
ITypeInfo *typeinfo;
HRESULT hr;
TRACE("%p %ld %s %ld %d %p %p %p %p\n", mapping, dispIdMember, debugstr_guid(riid),
lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
hr = get_typeinfo( IStaticPortMapping_tid, &typeinfo );
if (SUCCEEDED(hr))
{
hr = ITypeInfo_Invoke( typeinfo, &mapping->IStaticPortMapping_iface, dispIdMember,
wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr );
ITypeInfo_Release( typeinfo );
}
return hr;
}
static HRESULT WINAPI static_port_mapping_get_ExternalIPAddress(
IStaticPortMapping *iface,
BSTR *value )
{
struct static_port_mapping *mapping = impl_from_IStaticPortMapping( iface );
TRACE( "iface %p, value %p.\n", iface, value );
if (!value) return E_POINTER;
*value = SysAllocString( mapping->data.external_ip );
if (mapping->data.external_ip && !*value) return E_OUTOFMEMORY;
return S_OK;
}
static HRESULT WINAPI static_port_mapping_get_ExternalPort(
IStaticPortMapping *iface,
LONG *value )
{
struct static_port_mapping *mapping = impl_from_IStaticPortMapping( iface );
TRACE( "iface %p, value %p.\n", iface, value );
if (!value) return E_POINTER;
*value = mapping->data.external;
return S_OK;
}
static HRESULT WINAPI static_port_mapping_get_InternalPort(
IStaticPortMapping *iface,
LONG *value )
{
struct static_port_mapping *mapping = impl_from_IStaticPortMapping( iface );
TRACE( "iface %p, value %p.\n", iface, value );
if (!value) return E_POINTER;
*value = mapping->data.internal;
return S_OK;
}
static HRESULT WINAPI static_port_mapping_get_Protocol(
IStaticPortMapping *iface,
BSTR *value )
{
struct static_port_mapping *mapping = impl_from_IStaticPortMapping( iface );
TRACE( "iface %p, value %p.\n", iface, value );
if (!value) return E_POINTER;
*value = SysAllocString( mapping->data.protocol );
if (mapping->data.protocol && !*value) return E_OUTOFMEMORY;
return S_OK;
}
static HRESULT WINAPI static_port_mapping_get_InternalClient(
IStaticPortMapping *iface,
BSTR *value )
{
struct static_port_mapping *mapping = impl_from_IStaticPortMapping( iface );
TRACE( "iface %p, value %p.\n", iface, value );
if (!value) return E_POINTER;
*value = SysAllocString( mapping->data.client );
if (mapping->data.client && !*value) return E_OUTOFMEMORY;
return S_OK;
}
static HRESULT WINAPI static_port_mapping_get_Enabled(
IStaticPortMapping *iface,
VARIANT_BOOL *value )
{
struct static_port_mapping *mapping = impl_from_IStaticPortMapping( iface );
TRACE( "iface %p, value %p.\n", iface, value );
if (!value) return E_POINTER;
*value = mapping->data.enabled;
return S_OK;
}
static HRESULT WINAPI static_port_mapping_get_Description(
IStaticPortMapping *iface,
BSTR *value )
{
struct static_port_mapping *mapping = impl_from_IStaticPortMapping( iface );
TRACE( "iface %p, value %p.\n", iface, value );
if (!value) return E_POINTER;
*value = SysAllocString( mapping->data.descr );
if (mapping->data.descr && !*value) return E_OUTOFMEMORY;
return S_OK;
}
static HRESULT WINAPI static_port_mapping_EditInternalClient(
IStaticPortMapping *iface,
BSTR value )
{
FIXME( "iface %p, value %s stub.\n", iface, debugstr_w(value) );
return E_NOTIMPL;
}
static HRESULT WINAPI static_port_mapping_Enable(
IStaticPortMapping *iface,
VARIANT_BOOL value )
{
FIXME( "iface %p, value %d stub.\n", iface, value );
return E_NOTIMPL;
}
static HRESULT WINAPI static_port_mapping_EditDescription(
IStaticPortMapping *iface,
BSTR value )
{
FIXME( "iface %p, value %s stub.\n", iface, debugstr_w(value) );
return E_NOTIMPL;
}
static HRESULT WINAPI static_port_mapping_EditInternalPort(
IStaticPortMapping *iface,
LONG value )
{
FIXME( "iface %p, value %ld stub.\n", iface, value );
return E_NOTIMPL;
}
static const IStaticPortMappingVtbl static_port_mapping_vtbl =
{
static_port_mapping_QueryInterface,
static_port_mapping_AddRef,
static_port_mapping_Release,
static_port_mapping_GetTypeInfoCount,
static_port_mapping_GetTypeInfo,
static_port_mapping_GetIDsOfNames,
static_port_mapping_Invoke,
static_port_mapping_get_ExternalIPAddress,
static_port_mapping_get_ExternalPort,
static_port_mapping_get_InternalPort,
static_port_mapping_get_Protocol,
static_port_mapping_get_InternalClient,
static_port_mapping_get_Enabled,
static_port_mapping_get_Description,
static_port_mapping_EditInternalClient,
static_port_mapping_Enable,
static_port_mapping_EditDescription,
static_port_mapping_EditInternalPort,
};
static HRESULT static_port_mapping_create( const struct port_mapping *mapping_data, IStaticPortMapping **ret )
{
struct static_port_mapping *mapping;
if (!(mapping = calloc( 1, sizeof(*mapping) ))) return E_OUTOFMEMORY;
mapping->refs = 1;
mapping->IStaticPortMapping_iface.lpVtbl = &static_port_mapping_vtbl;
mapping->data = *mapping_data;
*ret = &mapping->IStaticPortMapping_iface;
return S_OK;
}
struct port_mapping_enum
{
IEnumVARIANT IEnumVARIANT_iface;
LONG refs;
unsigned int index;
};
static inline struct port_mapping_enum *impl_from_IEnumVARIANT( IEnumVARIANT *iface )
{
return CONTAINING_RECORD(iface, struct port_mapping_enum, IEnumVARIANT_iface);
}
static ULONG WINAPI port_mapping_enum_AddRef(
IEnumVARIANT *iface )
{
struct port_mapping_enum *mapping_enum = impl_from_IEnumVARIANT( iface );
return InterlockedIncrement( &mapping_enum->refs );
}
static ULONG WINAPI port_mapping_enum_Release(
IEnumVARIANT *iface )
{
struct port_mapping_enum *mapping_enum = impl_from_IEnumVARIANT( iface );
LONG refs = InterlockedDecrement( &mapping_enum->refs );
if (!refs)
{
TRACE("destroying %p\n", mapping_enum);
free( mapping_enum );
release_gateway_connection();
}
return refs;
}
static HRESULT WINAPI port_mapping_enum_QueryInterface(
IEnumVARIANT *iface,
REFIID riid,
void **ppvObject )
{
struct port_mapping_enum *mapping_enum = impl_from_IEnumVARIANT( iface );
TRACE("%p %s %p\n", mapping_enum, debugstr_guid( riid ), ppvObject );
if ( IsEqualGUID( riid, &IID_IEnumVARIANT ) ||
IsEqualGUID( riid, &IID_IUnknown ) )
{
*ppvObject = iface;
}
else
{
FIXME("interface %s not implemented\n", debugstr_guid(riid));
return E_NOINTERFACE;
}
IEnumVARIANT_AddRef( iface );
return S_OK;
}
static HRESULT WINAPI port_mapping_enum_Next( IEnumVARIANT *iface, ULONG celt, VARIANT *var, ULONG *fetched )
{
struct port_mapping_enum *mapping_enum = impl_from_IEnumVARIANT( iface );
struct port_mapping *data;
IStaticPortMapping *pm;
unsigned int i, count;
HRESULT ret;
TRACE( "iface %p, celt %lu, var %p, fetched %p.\n", iface, celt, var, fetched );
if (fetched) *fetched = 0;
if (!celt) return S_OK;
if (!var) return E_POINTER;
if (!(data = calloc( 1, celt * sizeof(*data) ))) return E_OUTOFMEMORY;
count = get_port_mapping_range( mapping_enum->index, celt, data );
TRACE( "count %u.\n", count );
for (i = 0; i < count; ++i)
{
if (FAILED(static_port_mapping_create( &data[i], &pm ))) break;
V_VT(&var[i]) = VT_DISPATCH;
V_DISPATCH(&var[i]) = (IDispatch *)pm;
}
mapping_enum->index += i;
if (fetched) *fetched = i;
ret = (i < celt) ? S_FALSE : S_OK;
for ( ; i < count; ++i)
{
free_port_mapping( &data[i] );
VariantInit( &var[i] );
}
for ( ; i < celt; ++i)
VariantInit( &var[i] );
free( data );
return ret;
}
static HRESULT WINAPI port_mapping_enum_Skip( IEnumVARIANT *iface, ULONG celt )
{
struct port_mapping_enum *mapping_enum = impl_from_IEnumVARIANT( iface );
unsigned int count = get_port_mapping_count();
TRACE( "iface %p, celt %lu.\n", iface, celt );
mapping_enum->index += celt;
return mapping_enum->index <= count ? S_OK : S_FALSE;
}
static HRESULT WINAPI port_mapping_enum_Reset( IEnumVARIANT *iface )
{
struct port_mapping_enum *mapping_enum = impl_from_IEnumVARIANT( iface );
TRACE( "iface %p.\n", iface );
mapping_enum->index = 0;
return S_OK;
}
static HRESULT create_port_mapping_enum( IUnknown **ret );
static HRESULT WINAPI port_mapping_enum_Clone( IEnumVARIANT *iface, IEnumVARIANT **ret )
{
struct port_mapping_enum *mapping_enum = impl_from_IEnumVARIANT( iface );
HRESULT hr;
TRACE( "iface %p, ret %p.\n", iface, ret );
if (!ret) return E_POINTER;
*ret = NULL;
if (FAILED(hr = create_port_mapping_enum( (IUnknown **)ret ))) return hr;
impl_from_IEnumVARIANT( *ret )->index = mapping_enum->index;
return S_OK;
}
static const IEnumVARIANTVtbl port_mapping_enum_vtbl =
{
port_mapping_enum_QueryInterface,
port_mapping_enum_AddRef,
port_mapping_enum_Release,
port_mapping_enum_Next,
port_mapping_enum_Skip,
port_mapping_enum_Reset,
port_mapping_enum_Clone,
};
static HRESULT create_port_mapping_enum( IUnknown **ret )
{
struct port_mapping_enum *mapping_enum;
if (!(mapping_enum = calloc( 1, sizeof(*mapping_enum) ))) return E_OUTOFMEMORY;
grab_gateway_connection();
mapping_enum->refs = 1;
mapping_enum->IEnumVARIANT_iface.lpVtbl = &port_mapping_enum_vtbl;
mapping_enum->index = 0;
*ret = (IUnknown *)&mapping_enum->IEnumVARIANT_iface;
return S_OK;
}
struct static_port_mapping_collection
{
IStaticPortMappingCollection IStaticPortMappingCollection_iface;
LONG refs;
};
static inline struct static_port_mapping_collection *impl_from_IStaticPortMappingCollection
( IStaticPortMappingCollection *iface )
{
return CONTAINING_RECORD(iface, struct static_port_mapping_collection, IStaticPortMappingCollection_iface);
}
static ULONG WINAPI static_ports_AddRef(
IStaticPortMappingCollection *iface )
{
struct static_port_mapping_collection *ports = impl_from_IStaticPortMappingCollection( iface );
return InterlockedIncrement( &ports->refs );
}
static ULONG WINAPI static_ports_Release(
IStaticPortMappingCollection *iface )
{
struct static_port_mapping_collection *ports = impl_from_IStaticPortMappingCollection( iface );
LONG refs = InterlockedDecrement( &ports->refs );
if (!refs)
{
TRACE("destroying %p\n", ports);
release_gateway_connection();
free( ports );
}
return refs;
}
static HRESULT WINAPI static_ports_QueryInterface(
IStaticPortMappingCollection *iface,
REFIID riid,
void **ppvObject )
{
struct static_port_mapping_collection *ports = impl_from_IStaticPortMappingCollection( iface );
TRACE("%p %s %p\n", ports, debugstr_guid( riid ), ppvObject );
if ( IsEqualGUID( riid, &IID_IStaticPortMappingCollection ) ||
IsEqualGUID( riid, &IID_IDispatch ) ||
IsEqualGUID( riid, &IID_IUnknown ) )
{
*ppvObject = iface;
}
else
{
FIXME("interface %s not implemented\n", debugstr_guid(riid));
return E_NOINTERFACE;
}
IStaticPortMappingCollection_AddRef( iface );
return S_OK;
}
static HRESULT WINAPI static_ports_GetTypeInfoCount(
IStaticPortMappingCollection *iface,
UINT *pctinfo )
{
struct static_port_mapping_collection *ports = impl_from_IStaticPortMappingCollection( iface );
TRACE("%p %p\n", ports, pctinfo);
*pctinfo = 1;
return S_OK;
}
static HRESULT WINAPI static_ports_GetTypeInfo(
IStaticPortMappingCollection *iface,
UINT iTInfo,
LCID lcid,
ITypeInfo **ppTInfo )
{
struct static_port_mapping_collection *ports = impl_from_IStaticPortMappingCollection( iface );
TRACE("%p %u %lu %p\n", ports, iTInfo, lcid, ppTInfo);
return get_typeinfo( IStaticPortMappingCollection_tid, ppTInfo );
}
static HRESULT WINAPI static_ports_GetIDsOfNames(
IStaticPortMappingCollection *iface,
REFIID riid,
LPOLESTR *rgszNames,
UINT cNames,
LCID lcid,
DISPID *rgDispId )
{
struct static_port_mapping_collection *ports = impl_from_IStaticPortMappingCollection( iface );
ITypeInfo *typeinfo;
HRESULT hr;
TRACE("%p %s %p %u %lu %p\n", ports, debugstr_guid(riid), rgszNames, cNames, lcid, rgDispId);
hr = get_typeinfo( IStaticPortMappingCollection_tid, &typeinfo );
if (SUCCEEDED(hr))
{
hr = ITypeInfo_GetIDsOfNames( typeinfo, rgszNames, cNames, rgDispId );
ITypeInfo_Release( typeinfo );
}
return hr;
}
static HRESULT WINAPI static_ports_Invoke(
IStaticPortMappingCollection *iface,
DISPID dispIdMember,
REFIID riid,
LCID lcid,
WORD wFlags,
DISPPARAMS *pDispParams,
VARIANT *pVarResult,
EXCEPINFO *pExcepInfo,
UINT *puArgErr )
{
struct static_port_mapping_collection *ports = impl_from_IStaticPortMappingCollection( iface );
ITypeInfo *typeinfo;
HRESULT hr;
TRACE("%p %ld %s %ld %d %p %p %p %p\n", ports, dispIdMember, debugstr_guid(riid),
lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
hr = get_typeinfo( IStaticPortMappingCollection_tid, &typeinfo );
if (SUCCEEDED(hr))
{
hr = ITypeInfo_Invoke( typeinfo, &ports->IStaticPortMappingCollection_iface, dispIdMember,
wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr );
ITypeInfo_Release( typeinfo );
}
return hr;
}
static HRESULT WINAPI static_ports__NewEnum(
IStaticPortMappingCollection *iface,
IUnknown **ret )
{
TRACE( "iface %p, ret %p.\n", iface, ret );
if (!ret) return E_POINTER;
*ret = NULL;
return create_port_mapping_enum( ret );
}
static HRESULT WINAPI static_ports_get_Item(
IStaticPortMappingCollection *iface,
LONG port,
BSTR protocol,
IStaticPortMapping **mapping )
{
struct port_mapping mapping_data;
HRESULT ret;
TRACE( "iface %p, port %ld, protocol %s.\n", iface, port, debugstr_w(protocol) );
if (!mapping) return E_POINTER;
*mapping = NULL;
if (!is_valid_protocol( protocol )) return E_INVALIDARG;
if (port < 0 || port > 65535) return E_INVALIDARG;
if (!find_port_mapping( port, protocol, &mapping_data )) return HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND);
if (FAILED(ret = static_port_mapping_create( &mapping_data, mapping ))) free_port_mapping( &mapping_data );
return ret;
}
static HRESULT WINAPI static_ports_get_Count(
IStaticPortMappingCollection *iface,
LONG *count )
{
TRACE( "iface %p, count %p.\n", iface, count );
if (!count) return E_POINTER;
*count = get_port_mapping_count();
return S_OK;
}
static HRESULT WINAPI static_ports_Remove(
IStaticPortMappingCollection *iface,
LONG port,
BSTR protocol )
{
TRACE( "iface %p, port %ld, protocol %s.\n", iface, port, debugstr_w(protocol) );
if (!is_valid_protocol( protocol )) return E_INVALIDARG;
if (port < 0 || port > 65535) return E_INVALIDARG;
if (!remove_port_mapping( port, protocol )) return E_FAIL;
return S_OK;
}
static HRESULT WINAPI static_ports_Add(
IStaticPortMappingCollection *iface,
LONG external,
BSTR protocol,
LONG internal,
BSTR client,
VARIANT_BOOL enabled,
BSTR description,
IStaticPortMapping **mapping )
{
struct port_mapping mapping_data;
HRESULT ret;
TRACE( "iface %p, external %ld, protocol %s, internal %ld, client %s, enabled %d, description %s, mapping %p.\n",
iface, external, debugstr_w(protocol), internal, debugstr_w(client), enabled, debugstr_w(description),
mapping );
if (!mapping) return E_POINTER;
*mapping = NULL;
if (!is_valid_protocol( protocol )) return E_INVALIDARG;
if (external < 0 || external > 65535) return E_INVALIDARG;
if (internal < 0 || internal > 65535) return E_INVALIDARG;
if (!client || !description) return E_INVALIDARG;
if (!add_port_mapping( external, protocol, internal, client, enabled, description )) return E_FAIL;
mapping_data.external_ip = NULL;
mapping_data.external = external;
mapping_data.protocol = SysAllocString( protocol );
mapping_data.internal = internal;
mapping_data.client = SysAllocString( client );
mapping_data.enabled = enabled;
mapping_data.descr = SysAllocString( description );
if (!mapping_data.protocol || !mapping_data.client || !mapping_data.descr)
{
free_port_mapping( &mapping_data );
return E_OUTOFMEMORY;
}
if (FAILED(ret = static_port_mapping_create( &mapping_data, mapping ))) free_port_mapping( &mapping_data );
return ret;
}
static const IStaticPortMappingCollectionVtbl static_ports_vtbl =
{
static_ports_QueryInterface,
static_ports_AddRef,
static_ports_Release,
static_ports_GetTypeInfoCount,
static_ports_GetTypeInfo,
static_ports_GetIDsOfNames,
static_ports_Invoke,
static_ports__NewEnum,
static_ports_get_Item,
static_ports_get_Count,
static_ports_Remove,
static_ports_Add,
};
static HRESULT static_port_mapping_collection_create(IStaticPortMappingCollection **object)
{
struct static_port_mapping_collection *ports;
if (!object) return E_POINTER;
if (!grab_gateway_connection())
{
*object = NULL;
return S_OK;
}
if (!(ports = calloc( 1, sizeof(*ports) )))
{
release_gateway_connection();
return E_OUTOFMEMORY;
}
ports->refs = 1;
ports->IStaticPortMappingCollection_iface.lpVtbl = &static_ports_vtbl;
*object = &ports->IStaticPortMappingCollection_iface;
return S_OK;
}
typedef struct fw_port
{
INetFwOpenPort INetFwOpenPort_iface;
LONG refs;
BSTR name;
NET_FW_IP_PROTOCOL protocol;
LONG port;
} fw_port;
static inline fw_port *impl_from_INetFwOpenPort( INetFwOpenPort *iface )
{
return CONTAINING_RECORD(iface, fw_port, INetFwOpenPort_iface);
}
static ULONG WINAPI fw_port_AddRef(
INetFwOpenPort *iface )
{
fw_port *fw_port = impl_from_INetFwOpenPort( iface );
return InterlockedIncrement( &fw_port->refs );
}
static ULONG WINAPI fw_port_Release(
INetFwOpenPort *iface )
{
fw_port *fw_port = impl_from_INetFwOpenPort( iface );
LONG refs = InterlockedDecrement( &fw_port->refs );
if (!refs)
{
TRACE("destroying %p\n", fw_port);
SysFreeString( fw_port->name );
free( fw_port );
}
return refs;
}
static HRESULT WINAPI fw_port_QueryInterface(
INetFwOpenPort *iface,
REFIID riid,
void **ppvObject )
{
fw_port *This = impl_from_INetFwOpenPort( iface );
TRACE("%p %s %p\n", This, debugstr_guid( riid ), ppvObject );
if ( IsEqualGUID( riid, &IID_INetFwOpenPort ) ||
IsEqualGUID( riid, &IID_IDispatch ) ||
IsEqualGUID( riid, &IID_IUnknown ) )
{
*ppvObject = iface;
}
else
{
FIXME("interface %s not implemented\n", debugstr_guid(riid));
return E_NOINTERFACE;
}
INetFwOpenPort_AddRef( iface );
return S_OK;
}
static HRESULT WINAPI fw_port_GetTypeInfoCount(
INetFwOpenPort *iface,
UINT *pctinfo )
{
fw_port *This = impl_from_INetFwOpenPort( iface );
TRACE("%p %p\n", This, pctinfo);
*pctinfo = 1;
return S_OK;
}
static HRESULT WINAPI fw_port_GetTypeInfo(
INetFwOpenPort *iface,
UINT iTInfo,
LCID lcid,
ITypeInfo **ppTInfo )
{
fw_port *This = impl_from_INetFwOpenPort( iface );
TRACE("%p %u %lu %p\n", This, iTInfo, lcid, ppTInfo);
return get_typeinfo( INetFwOpenPort_tid, ppTInfo );
}
static HRESULT WINAPI fw_port_GetIDsOfNames(
INetFwOpenPort *iface,
REFIID riid,
LPOLESTR *rgszNames,
UINT cNames,
LCID lcid,
DISPID *rgDispId )
{
fw_port *This = impl_from_INetFwOpenPort( iface );
ITypeInfo *typeinfo;
HRESULT hr;
TRACE("%p %s %p %u %lu %p\n", This, debugstr_guid(riid), rgszNames, cNames, lcid, rgDispId);
hr = get_typeinfo( INetFwOpenPort_tid, &typeinfo );
if (SUCCEEDED(hr))
{
hr = ITypeInfo_GetIDsOfNames( typeinfo, rgszNames, cNames, rgDispId );
ITypeInfo_Release( typeinfo );
}
return hr;
}
static HRESULT WINAPI fw_port_Invoke(
INetFwOpenPort *iface,
DISPID dispIdMember,
REFIID riid,
LCID lcid,
WORD wFlags,
DISPPARAMS *pDispParams,
VARIANT *pVarResult,
EXCEPINFO *pExcepInfo,
UINT *puArgErr )
{
fw_port *This = impl_from_INetFwOpenPort( iface );
ITypeInfo *typeinfo;
HRESULT hr;
TRACE("%p %ld %s %ld %d %p %p %p %p\n", This, dispIdMember, debugstr_guid(riid),
lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
hr = get_typeinfo( INetFwOpenPort_tid, &typeinfo );
if (SUCCEEDED(hr))
{
hr = ITypeInfo_Invoke( typeinfo, &This->INetFwOpenPort_iface, dispIdMember,
wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr );
ITypeInfo_Release( typeinfo );
}
return hr;
}
static HRESULT WINAPI fw_port_get_Name(
INetFwOpenPort *iface,
BSTR *name)
{
fw_port *This = impl_from_INetFwOpenPort( iface );
FIXME("%p %p\n", This, name);
return E_NOTIMPL;
}
static HRESULT WINAPI fw_port_put_Name(
INetFwOpenPort *iface,
BSTR name)
{
fw_port *This = impl_from_INetFwOpenPort( iface );
TRACE("%p %s\n", This, debugstr_w(name));
if (!(name = SysAllocString( name )))
return E_OUTOFMEMORY;
SysFreeString( This->name );
This->name = name;
return S_OK;
}
static HRESULT WINAPI fw_port_get_IpVersion(
INetFwOpenPort *iface,
NET_FW_IP_VERSION *ipVersion)
{
fw_port *This = impl_from_INetFwOpenPort( iface );
FIXME("%p %p\n", This, ipVersion);
return E_NOTIMPL;
}
static HRESULT WINAPI fw_port_put_IpVersion(
INetFwOpenPort *iface,
NET_FW_IP_VERSION ipVersion)
{
fw_port *This = impl_from_INetFwOpenPort( iface );
FIXME("%p %u\n", This, ipVersion);
return E_NOTIMPL;
}
static HRESULT WINAPI fw_port_get_Protocol(
INetFwOpenPort *iface,
NET_FW_IP_PROTOCOL *ipProtocol)
{
fw_port *This = impl_from_INetFwOpenPort( iface );
FIXME("%p %p\n", This, ipProtocol);
return E_NOTIMPL;
}
static HRESULT WINAPI fw_port_put_Protocol(
INetFwOpenPort *iface,
NET_FW_IP_PROTOCOL ipProtocol)
{
fw_port *This = impl_from_INetFwOpenPort( iface );
TRACE("%p %u\n", This, ipProtocol);
if (ipProtocol != NET_FW_IP_PROTOCOL_TCP && ipProtocol != NET_FW_IP_PROTOCOL_UDP)
return E_INVALIDARG;
This->protocol = ipProtocol;
return S_OK;
}
static HRESULT WINAPI fw_port_get_Port(
INetFwOpenPort *iface,
LONG *portNumber)
{
fw_port *This = impl_from_INetFwOpenPort( iface );
FIXME("%p %p\n", This, portNumber);
return E_NOTIMPL;
}
static HRESULT WINAPI fw_port_put_Port(
INetFwOpenPort *iface,
LONG portNumber)
{
fw_port *This = impl_from_INetFwOpenPort( iface );
TRACE("%p %ld\n", This, portNumber);
This->port = portNumber;
return S_OK;
}
static HRESULT WINAPI fw_port_get_Scope(
INetFwOpenPort *iface,
NET_FW_SCOPE *scope)
{
fw_port *This = impl_from_INetFwOpenPort( iface );
FIXME("%p %p\n", This, scope);
return E_NOTIMPL;
}
static HRESULT WINAPI fw_port_put_Scope(
INetFwOpenPort *iface,
NET_FW_SCOPE scope)
{
fw_port *This = impl_from_INetFwOpenPort( iface );
FIXME("%p %u\n", This, scope);
return E_NOTIMPL;
}
static HRESULT WINAPI fw_port_get_RemoteAddresses(
INetFwOpenPort *iface,
BSTR *remoteAddrs)
{
fw_port *This = impl_from_INetFwOpenPort( iface );
FIXME("%p %p\n", This, remoteAddrs);
return E_NOTIMPL;
}
static HRESULT WINAPI fw_port_put_RemoteAddresses(
INetFwOpenPort *iface,
BSTR remoteAddrs)
{
fw_port *This = impl_from_INetFwOpenPort( iface );
FIXME("%p %s\n", This, debugstr_w(remoteAddrs));
return E_NOTIMPL;
}
static HRESULT WINAPI fw_port_get_Enabled(
INetFwOpenPort *iface,
VARIANT_BOOL *enabled)
{
fw_port *This = impl_from_INetFwOpenPort( iface );
FIXME("%p %p\n", This, enabled);
*enabled = VARIANT_TRUE;
return S_OK;
}
static HRESULT WINAPI fw_port_put_Enabled(
INetFwOpenPort *iface,
VARIANT_BOOL enabled)
{
fw_port *This = impl_from_INetFwOpenPort( iface );
FIXME("%p %d\n", This, enabled);
return S_OK;
}
static HRESULT WINAPI fw_port_get_BuiltIn(
INetFwOpenPort *iface,
VARIANT_BOOL *builtIn)
{
fw_port *This = impl_from_INetFwOpenPort( iface );
FIXME("%p %p\n", This, builtIn);
return E_NOTIMPL;
}
static const struct INetFwOpenPortVtbl fw_port_vtbl =
{
fw_port_QueryInterface,
fw_port_AddRef,
fw_port_Release,
fw_port_GetTypeInfoCount,
fw_port_GetTypeInfo,
fw_port_GetIDsOfNames,
fw_port_Invoke,
fw_port_get_Name,
fw_port_put_Name,
fw_port_get_IpVersion,
fw_port_put_IpVersion,
fw_port_get_Protocol,
fw_port_put_Protocol,
fw_port_get_Port,
fw_port_put_Port,
fw_port_get_Scope,
fw_port_put_Scope,
fw_port_get_RemoteAddresses,
fw_port_put_RemoteAddresses,
fw_port_get_Enabled,
fw_port_put_Enabled,
fw_port_get_BuiltIn
};
HRESULT NetFwOpenPort_create( IUnknown *pUnkOuter, LPVOID *ppObj )
{
fw_port *fp;
TRACE("(%p,%p)\n", pUnkOuter, ppObj);
fp = malloc( sizeof(*fp) );
if (!fp) return E_OUTOFMEMORY;
fp->INetFwOpenPort_iface.lpVtbl = &fw_port_vtbl;
fp->refs = 1;
fp->name = NULL;
fp->protocol = NET_FW_IP_PROTOCOL_TCP;
fp->port = 0;
*ppObj = &fp->INetFwOpenPort_iface;
TRACE("returning iface %p\n", *ppObj);
return S_OK;
}
typedef struct fw_ports
{
INetFwOpenPorts INetFwOpenPorts_iface;
LONG refs;
} fw_ports;
static inline fw_ports *impl_from_INetFwOpenPorts( INetFwOpenPorts *iface )
{
return CONTAINING_RECORD(iface, fw_ports, INetFwOpenPorts_iface);
}
static ULONG WINAPI fw_ports_AddRef(
INetFwOpenPorts *iface )
{
fw_ports *fw_ports = impl_from_INetFwOpenPorts( iface );
return InterlockedIncrement( &fw_ports->refs );
}
static ULONG WINAPI fw_ports_Release(
INetFwOpenPorts *iface )
{
fw_ports *fw_ports = impl_from_INetFwOpenPorts( iface );
LONG refs = InterlockedDecrement( &fw_ports->refs );
if (!refs)
{
TRACE("destroying %p\n", fw_ports);
free( fw_ports );
}
return refs;
}
static HRESULT WINAPI fw_ports_QueryInterface(
INetFwOpenPorts *iface,
REFIID riid,
void **ppvObject )
{
fw_ports *This = impl_from_INetFwOpenPorts( iface );
TRACE("%p %s %p\n", This, debugstr_guid( riid ), ppvObject );
if ( IsEqualGUID( riid, &IID_INetFwOpenPorts ) ||
IsEqualGUID( riid, &IID_IDispatch ) ||
IsEqualGUID( riid, &IID_IUnknown ) )
{
*ppvObject = iface;
}
else
{
FIXME("interface %s not implemented\n", debugstr_guid(riid));
return E_NOINTERFACE;
}
INetFwOpenPorts_AddRef( iface );
return S_OK;
}
static HRESULT WINAPI fw_ports_GetTypeInfoCount(
INetFwOpenPorts *iface,
UINT *pctinfo )
{
fw_ports *This = impl_from_INetFwOpenPorts( iface );
TRACE("%p %p\n", This, pctinfo);
*pctinfo = 1;
return S_OK;
}
static HRESULT WINAPI fw_ports_GetTypeInfo(
INetFwOpenPorts *iface,
UINT iTInfo,
LCID lcid,
ITypeInfo **ppTInfo )
{
fw_ports *This = impl_from_INetFwOpenPorts( iface );
TRACE("%p %u %lu %p\n", This, iTInfo, lcid, ppTInfo);
return get_typeinfo( INetFwOpenPorts_tid, ppTInfo );
}
static HRESULT WINAPI fw_ports_GetIDsOfNames(
INetFwOpenPorts *iface,
REFIID riid,
LPOLESTR *rgszNames,
UINT cNames,
LCID lcid,
DISPID *rgDispId )
{
fw_ports *This = impl_from_INetFwOpenPorts( iface );
ITypeInfo *typeinfo;
HRESULT hr;
TRACE("%p %s %p %u %lu %p\n", This, debugstr_guid(riid), rgszNames, cNames, lcid, rgDispId);
hr = get_typeinfo( INetFwOpenPorts_tid, &typeinfo );
if (SUCCEEDED(hr))
{
hr = ITypeInfo_GetIDsOfNames( typeinfo, rgszNames, cNames, rgDispId );
ITypeInfo_Release( typeinfo );
}
return hr;
}
static HRESULT WINAPI fw_ports_Invoke(
INetFwOpenPorts *iface,
DISPID dispIdMember,
REFIID riid,
LCID lcid,
WORD wFlags,
DISPPARAMS *pDispParams,
VARIANT *pVarResult,
EXCEPINFO *pExcepInfo,
UINT *puArgErr )
{
fw_ports *This = impl_from_INetFwOpenPorts( iface );
ITypeInfo *typeinfo;
HRESULT hr;
TRACE("%p %ld %s %ld %d %p %p %p %p\n", This, dispIdMember, debugstr_guid(riid),
lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
hr = get_typeinfo( INetFwOpenPorts_tid, &typeinfo );
if (SUCCEEDED(hr))
{
hr = ITypeInfo_Invoke( typeinfo, &This->INetFwOpenPorts_iface, dispIdMember,
wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr );
ITypeInfo_Release( typeinfo );
}
return hr;
}
static HRESULT WINAPI fw_ports_get_Count(
INetFwOpenPorts *iface,
LONG *count)
{
fw_ports *This = impl_from_INetFwOpenPorts( iface );
FIXME("%p, %p\n", This, count);
*count = 0;
return S_OK;
}
static HRESULT WINAPI fw_ports_Add(
INetFwOpenPorts *iface,
INetFwOpenPort *port)
{
fw_ports *This = impl_from_INetFwOpenPorts( iface );
FIXME("%p, %p\n", This, port);
return S_OK;
}
static HRESULT WINAPI fw_ports_Remove(
INetFwOpenPorts *iface,
LONG portNumber,
NET_FW_IP_PROTOCOL ipProtocol)
{
fw_ports *This = impl_from_INetFwOpenPorts( iface );
FIXME("%p, %ld, %u\n", This, portNumber, ipProtocol);
return E_NOTIMPL;
}
static HRESULT WINAPI fw_ports_Item(
INetFwOpenPorts *iface,
LONG portNumber,
NET_FW_IP_PROTOCOL ipProtocol,
INetFwOpenPort **openPort)
{
HRESULT hr;
fw_ports *This = impl_from_INetFwOpenPorts( iface );
FIXME("%p, %ld, %u, %p\n", This, portNumber, ipProtocol, openPort);
hr = NetFwOpenPort_create( NULL, (void **)openPort );
if (SUCCEEDED(hr))
{
INetFwOpenPort_put_Protocol( *openPort, ipProtocol );
INetFwOpenPort_put_Port( *openPort, portNumber );
}
return hr;
}
static HRESULT WINAPI fw_ports_get__NewEnum(
INetFwOpenPorts *iface,
IUnknown **newEnum)
{
fw_ports *This = impl_from_INetFwOpenPorts( iface );
FIXME("%p, %p\n", This, newEnum);
return E_NOTIMPL;
}
static const struct INetFwOpenPortsVtbl fw_ports_vtbl =
{
fw_ports_QueryInterface,
fw_ports_AddRef,
fw_ports_Release,
fw_ports_GetTypeInfoCount,
fw_ports_GetTypeInfo,
fw_ports_GetIDsOfNames,
fw_ports_Invoke,
fw_ports_get_Count,
fw_ports_Add,
fw_ports_Remove,
fw_ports_Item,
fw_ports_get__NewEnum
};
HRESULT NetFwOpenPorts_create( IUnknown *pUnkOuter, LPVOID *ppObj )
{
fw_ports *fp;
TRACE("(%p,%p)\n", pUnkOuter, ppObj);
fp = malloc( sizeof(*fp) );
if (!fp) return E_OUTOFMEMORY;
fp->INetFwOpenPorts_iface.lpVtbl = &fw_ports_vtbl;
fp->refs = 1;
*ppObj = &fp->INetFwOpenPorts_iface;
TRACE("returning iface %p\n", *ppObj);
return S_OK;
}
typedef struct _upnpnat
{
IUPnPNAT IUPnPNAT_iface;
LONG ref;
} upnpnat;
static inline upnpnat *impl_from_IUPnPNAT( IUPnPNAT *iface )
{
return CONTAINING_RECORD(iface, upnpnat, IUPnPNAT_iface);
}
static HRESULT WINAPI upnpnat_QueryInterface(IUPnPNAT *iface, REFIID riid, void **object)
{
upnpnat *This = impl_from_IUPnPNAT( iface );
TRACE("%p %s %p\n", This, debugstr_guid( riid ), object );
if ( IsEqualGUID( riid, &IID_IUPnPNAT ) ||
IsEqualGUID( riid, &IID_IDispatch ) ||
IsEqualGUID( riid, &IID_IUnknown ) )
{
*object = iface;
}
else if(IsEqualGUID( riid, &IID_IProvideClassInfo))
{
TRACE("IProvideClassInfo not supported.\n");
return E_NOINTERFACE;
}
else
{
FIXME("interface %s not implemented\n", debugstr_guid(riid));
return E_NOINTERFACE;
}
IUPnPNAT_AddRef( iface );
return S_OK;
}
static ULONG WINAPI upnpnat_AddRef(IUPnPNAT *iface)
{
upnpnat *This = impl_from_IUPnPNAT( iface );
return InterlockedIncrement( &This->ref );
}
static ULONG WINAPI upnpnat_Release(IUPnPNAT *iface)
{
upnpnat *This = impl_from_IUPnPNAT( iface );
LONG refs = InterlockedDecrement( &This->ref );
if (!refs)
{
free( This );
}
return refs;
}
static HRESULT WINAPI upnpnat_GetTypeInfoCount(IUPnPNAT *iface, UINT *pctinfo)
{
upnpnat *This = impl_from_IUPnPNAT( iface );
TRACE("%p %p\n", This, pctinfo);
*pctinfo = 1;
return S_OK;
}
static HRESULT WINAPI upnpnat_GetTypeInfo(IUPnPNAT *iface, UINT iTInfo, LCID lcid, ITypeInfo **ppTInfo)
{
upnpnat *This = impl_from_IUPnPNAT( iface );
TRACE("%p %u %lu %p\n", This, iTInfo, lcid, ppTInfo);
return get_typeinfo( IUPnPNAT_tid, ppTInfo );
}
static HRESULT WINAPI upnpnat_GetIDsOfNames(IUPnPNAT *iface, REFIID riid, LPOLESTR *rgszNames,
UINT cNames, LCID lcid, DISPID *rgDispId)
{
upnpnat *This = impl_from_IUPnPNAT( iface );
ITypeInfo *typeinfo;
HRESULT hr;
TRACE("%p %s %p %u %lu %p\n", This, debugstr_guid(riid), rgszNames, cNames, lcid, rgDispId);
hr = get_typeinfo( IUPnPNAT_tid, &typeinfo );
if (SUCCEEDED(hr))
{
hr = ITypeInfo_GetIDsOfNames( typeinfo, rgszNames, cNames, rgDispId );
ITypeInfo_Release( typeinfo );
}
return hr;
}
static HRESULT WINAPI upnpnat_Invoke(IUPnPNAT *iface, DISPID dispIdMember, REFIID riid, LCID lcid,
WORD wFlags, DISPPARAMS *pDispParams, VARIANT *pVarResult, EXCEPINFO *pExcepInfo,
UINT *puArgErr)
{
upnpnat *This = impl_from_IUPnPNAT( iface );
ITypeInfo *typeinfo;
HRESULT hr;
TRACE("%p %ld %s %ld %d %p %p %p %p\n", This, dispIdMember, debugstr_guid(riid),
lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
hr = get_typeinfo( IUPnPNAT_tid, &typeinfo );
if (SUCCEEDED(hr))
{
hr = ITypeInfo_Invoke( typeinfo, &This->IUPnPNAT_iface, dispIdMember,
wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr );
ITypeInfo_Release( typeinfo );
}
return hr;
}
static HRESULT WINAPI upnpnat_get_StaticPortMappingCollection(IUPnPNAT *iface, IStaticPortMappingCollection **collection)
{
upnpnat *This = impl_from_IUPnPNAT( iface );
TRACE("%p, %p\n", This, collection);
return static_port_mapping_collection_create( collection );
}
static HRESULT WINAPI upnpnat_get_DynamicPortMappingCollection(IUPnPNAT *iface, IDynamicPortMappingCollection **collection)
{
upnpnat *This = impl_from_IUPnPNAT( iface );
FIXME("%p, %p\n", This, collection);
if(collection)
*collection = NULL;
return E_NOTIMPL;
}
static HRESULT WINAPI upnpnat_get_NATEventManager(IUPnPNAT *iface, INATEventManager **manager)
{
upnpnat *This = impl_from_IUPnPNAT( iface );
FIXME("%p, %p\n", This, manager);
if(manager)
*manager = NULL;
return E_NOTIMPL;
}
const static IUPnPNATVtbl upnpnat_vtbl =
{
upnpnat_QueryInterface,
upnpnat_AddRef,
upnpnat_Release,
upnpnat_GetTypeInfoCount,
upnpnat_GetTypeInfo,
upnpnat_GetIDsOfNames,
upnpnat_Invoke,
upnpnat_get_StaticPortMappingCollection,
upnpnat_get_DynamicPortMappingCollection,
upnpnat_get_NATEventManager
};
HRESULT IUPnPNAT_create(IUnknown *outer, void **object)
{
upnpnat *nat;
TRACE("(%p,%p)\n", outer, object);
nat = malloc( sizeof(*nat) );
if (!nat) return E_OUTOFMEMORY;
nat->IUPnPNAT_iface.lpVtbl = &upnpnat_vtbl;
nat->ref = 1;
*object = &nat->IUPnPNAT_iface;
TRACE("returning iface %p\n", *object);
return S_OK;
}