forked from premiere/premiere-libtorrent
352 lines
12 KiB
C++
352 lines
12 KiB
C++
/*
|
|
|
|
Copyright (c) 2012, Arvid Norberg
|
|
All rights reserved.
|
|
|
|
Redistribution and use in source and binary forms, with or without
|
|
modification, are permitted provided that the following conditions
|
|
are met:
|
|
|
|
* Redistributions of source code must retain the above copyright
|
|
notice, this list of conditions and the following disclaimer.
|
|
* Redistributions in binary form must reproduce the above copyright
|
|
notice, this list of conditions and the following disclaimer in
|
|
the documentation and/or other materials provided with the distribution.
|
|
* Neither the name of the author nor the names of its
|
|
contributors may be used to endorse or promote products derived
|
|
from this software without specific prior written permission.
|
|
|
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
|
LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
|
CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
|
SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
|
CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
|
ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
*/
|
|
|
|
#include "libtorrent/xml_parse.hpp"
|
|
#include "libtorrent/upnp.hpp"
|
|
#include "test.hpp"
|
|
#include <iostream>
|
|
|
|
char upnp_xml[] =
|
|
"<root>"
|
|
"<specVersion>"
|
|
"<major>1</major>"
|
|
"<minor>0</minor>"
|
|
"</specVersion>"
|
|
"<URLBase>http://192.168.0.1:5678</URLBase>"
|
|
"<device>"
|
|
"<deviceType>"
|
|
"urn:schemas-upnp-org:device:InternetGatewayDevice:1"
|
|
"</deviceType>"
|
|
"<presentationURL>http://192.168.0.1:80</presentationURL>"
|
|
"<friendlyName>D-Link Router</friendlyName>"
|
|
"<manufacturer>D-Link</manufacturer>"
|
|
"<manufacturerURL>http://www.dlink.com</manufacturerURL>"
|
|
"<modelDescription>Internet Access Router</modelDescription>"
|
|
"<modelName>D-Link Router</modelName>"
|
|
"<UDN>uuid:upnp-InternetGatewayDevice-1_0-12345678900001</UDN>"
|
|
"<UPC>123456789001</UPC>"
|
|
"<serviceList>"
|
|
"<service>"
|
|
"<serviceType>urn:schemas-upnp-org:service:Layer3Forwarding:1</serviceType>"
|
|
"<serviceId>urn:upnp-org:serviceId:L3Forwarding1</serviceId>"
|
|
"<controlURL>/Layer3Forwarding</controlURL>"
|
|
"<eventSubURL>/Layer3Forwarding</eventSubURL>"
|
|
"<SCPDURL>/Layer3Forwarding.xml</SCPDURL>"
|
|
"</service>"
|
|
"</serviceList>"
|
|
"<deviceList>"
|
|
"<device>"
|
|
"<deviceType>urn:schemas-upnp-org:device:WANDevice:1</deviceType>"
|
|
"<friendlyName>WANDevice</friendlyName>"
|
|
"<manufacturer>D-Link</manufacturer>"
|
|
"<manufacturerURL>http://www.dlink.com</manufacturerURL>"
|
|
"<modelDescription>Internet Access Router</modelDescription>"
|
|
"<modelName>D-Link Router</modelName>"
|
|
"<modelNumber>1</modelNumber>"
|
|
"<modelURL>http://support.dlink.com</modelURL>"
|
|
"<serialNumber>12345678900001</serialNumber>"
|
|
"<UDN>uuid:upnp-WANDevice-1_0-12345678900001</UDN>"
|
|
"<UPC>123456789001</UPC>"
|
|
"<serviceList>"
|
|
"<service>"
|
|
"<serviceType>"
|
|
"urn:schemas-upnp-org:service:WANCommonInterfaceConfig:1"
|
|
"</serviceType>"
|
|
"<serviceId>urn:upnp-org:serviceId:WANCommonInterfaceConfig</serviceId>"
|
|
"<controlURL>/WANCommonInterfaceConfig</controlURL>"
|
|
"<eventSubURL>/WANCommonInterfaceConfig</eventSubURL>"
|
|
"<SCPDURL>/WANCommonInterfaceConfig.xml</SCPDURL>"
|
|
"</service>"
|
|
"</serviceList>"
|
|
"<deviceList>"
|
|
"<device>"
|
|
"<deviceType>urn:schemas-upnp-org:device:WANConnectionDevice:1</deviceType>"
|
|
"<friendlyName>WAN Connection Device</friendlyName>"
|
|
"<manufacturer>D-Link</manufacturer>"
|
|
"<manufacturerURL>http://www.dlink.com</manufacturerURL>"
|
|
"<modelDescription>Internet Access Router</modelDescription>"
|
|
"<modelName>D-Link Router</modelName>"
|
|
"<modelNumber>1</modelNumber>"
|
|
"<modelURL>http://support.dlink.com</modelURL>"
|
|
"<serialNumber>12345678900001</serialNumber>"
|
|
"<UDN>uuid:upnp-WANConnectionDevice-1_0-12345678900001</UDN>"
|
|
"<UPC>123456789001</UPC>"
|
|
"<serviceList>"
|
|
"<service>"
|
|
"<serviceType>urn:schemas-upnp-org:service:WANIPConnection:1</serviceType>"
|
|
"<serviceId>urn:upnp-org:serviceId:WANIPConnection</serviceId>"
|
|
"<controlURL>/WANIPConnection</controlURL>"
|
|
"<eventSubURL>/WANIPConnection</eventSubURL>"
|
|
"<SCPDURL>/WANIPConnection.xml</SCPDURL>"
|
|
"</service>"
|
|
"</serviceList>"
|
|
"</device>"
|
|
"</deviceList>"
|
|
"</device>"
|
|
"</deviceList>"
|
|
"</device>"
|
|
"</root>";
|
|
|
|
char upnp_xml2[] =
|
|
"<root>"
|
|
"<specVersion>"
|
|
"<major>1</major>"
|
|
"<minor>0</minor>"
|
|
"</specVersion>"
|
|
"<URLBase>http://192.168.1.1:49152</URLBase>"
|
|
"<device>"
|
|
"<deviceType>"
|
|
"urn:schemas-upnp-org:device:InternetGatewayDevice:1"
|
|
"</deviceType>"
|
|
"<friendlyName>LINKSYS WAG200G Gateway</friendlyName>"
|
|
"<manufacturer>LINKSYS</manufacturer>"
|
|
"<manufacturerURL>http://www.linksys.com</manufacturerURL>"
|
|
"<modelDescription>LINKSYS WAG200G Gateway</modelDescription>"
|
|
"<modelName>Wireless-G ADSL Home Gateway</modelName>"
|
|
"<modelNumber>WAG200G</modelNumber>"
|
|
"<modelURL>http://www.linksys.com</modelURL>"
|
|
"<serialNumber>123456789</serialNumber>"
|
|
"<UDN>uuid:8d401597-1dd2-11b2-a7d4-001ee5947cac</UDN>"
|
|
"<UPC>WAG200G</UPC>"
|
|
"<serviceList>"
|
|
"<service>"
|
|
"<serviceType>urn:schemas-upnp-org:service:Layer3Forwarding:1</serviceType>"
|
|
"<serviceId>urn:upnp-org:serviceId:L3Forwarding1</serviceId>"
|
|
"<controlURL>/upnp/control/L3Forwarding1</controlURL>"
|
|
"<eventSubURL>/upnp/event/L3Forwarding1</eventSubURL>"
|
|
"<SCPDURL>/l3frwd.xml</SCPDURL>"
|
|
"</service>"
|
|
"</serviceList>"
|
|
"<deviceList>"
|
|
"<device>"
|
|
"<deviceType>urn:schemas-upnp-org:device:WANDevice:1</deviceType>"
|
|
"<friendlyName>WANDevice</friendlyName>"
|
|
"<manufacturer>LINKSYS</manufacturer>"
|
|
"<manufacturerURL>http://www.linksys.com/</manufacturerURL>"
|
|
"<modelDescription>Residential Gateway</modelDescription>"
|
|
"<modelName>Internet Connection Sharing</modelName>"
|
|
"<modelNumber>1</modelNumber>"
|
|
"<modelURL>http://www.linksys.com/</modelURL>"
|
|
"<serialNumber>0000001</serialNumber>"
|
|
"<UDN>uuid:8d401596-1dd2-11b2-a7d4-001ee5947cac</UDN>"
|
|
"<UPC>WAG200G</UPC>"
|
|
"<serviceList>"
|
|
"<service>"
|
|
"<serviceType>"
|
|
"urn:schemas-upnp-org:service:WANCommonInterfaceConfig:1"
|
|
"</serviceType>"
|
|
"<serviceId>urn:upnp-org:serviceId:WANCommonIFC1</serviceId>"
|
|
"<controlURL>/upnp/control/WANCommonIFC1</controlURL>"
|
|
"<eventSubURL>/upnp/event/WANCommonIFC1</eventSubURL>"
|
|
"<SCPDURL>/cmnicfg.xml</SCPDURL>"
|
|
"</service>"
|
|
"</serviceList>"
|
|
"<deviceList>"
|
|
"<device>"
|
|
"<deviceType>urn:schemas-upnp-org:device:WANConnectionDevice:1</deviceType>"
|
|
"<friendlyName>WANConnectionDevice</friendlyName>"
|
|
"<manufacturer>LINKSYS</manufacturer>"
|
|
"<manufacturerURL>http://www.linksys.com/</manufacturerURL>"
|
|
"<modelDescription>Residential Gateway</modelDescription>"
|
|
"<modelName>Internet Connection Sharing</modelName>"
|
|
"<modelNumber>1</modelNumber>"
|
|
"<modelURL>http://www.linksys.com/</modelURL>"
|
|
"<serialNumber>0000001</serialNumber>"
|
|
"<UDN>uuid:8d401597-1dd2-11b2-a7d3-001ee5947cac</UDN>"
|
|
"<UPC>WAG200G</UPC>"
|
|
"<serviceList>"
|
|
"<service>"
|
|
"<serviceType>"
|
|
"urn:schemas-upnp-org:service:WANEthernetLinkConfig:1"
|
|
"</serviceType>"
|
|
"<serviceId>urn:upnp-org:serviceId:WANEthLinkC1</serviceId>"
|
|
"<controlURL>/upnp/control/WANEthLinkC1</controlURL>"
|
|
"<eventSubURL>/upnp/event/WANEthLinkC1</eventSubURL>"
|
|
"<SCPDURL>/wanelcfg.xml</SCPDURL>"
|
|
"</service>"
|
|
"<service>"
|
|
"<serviceType>urn:schemas-upnp-org:service:WANPPPConnection:1</serviceType>"
|
|
"<serviceId>urn:upnp-org:serviceId:WANPPPConn1</serviceId>"
|
|
"<controlURL>/upnp/control/WANPPPConn1</controlURL>"
|
|
"<eventSubURL>/upnp/event/WANPPPConn1</eventSubURL>"
|
|
"<SCPDURL>/pppcfg.xml</SCPDURL>"
|
|
"</service>"
|
|
"</serviceList>"
|
|
"</device>"
|
|
"</deviceList>"
|
|
"</device>"
|
|
"<device>"
|
|
"<deviceType>urn:schemas-upnp-org:device:LANDevice:1</deviceType>"
|
|
"<friendlyName>LANDevice</friendlyName>"
|
|
"<manufacturer>LINKSYS</manufacturer>"
|
|
"<manufacturerURL>http://www.linksys.com/</manufacturerURL>"
|
|
"<modelDescription>Residential Gateway</modelDescription>"
|
|
"<modelName>Residential Gateway</modelName>"
|
|
"<modelNumber>1</modelNumber>"
|
|
"<modelURL>http://www.linksys.com/</modelURL>"
|
|
"<serialNumber>0000001</serialNumber>"
|
|
"<UDN>uuid:8d401596-1dd2-11b2-a7d3-001ee5947cac</UDN>"
|
|
"<UPC>WAG200G</UPC>"
|
|
"<serviceList>"
|
|
"<service>"
|
|
"<serviceType>"
|
|
"urn:schemas-upnp-org:service:LANHostConfigManagement:1"
|
|
"</serviceType>"
|
|
"<serviceId>urn:upnp-org:serviceId:LANHostCfg1</serviceId>"
|
|
"<controlURL>/upnp/control/LANHostCfg1</controlURL>"
|
|
"<eventSubURL>/upnp/event/LANHostCfg1</eventSubURL>"
|
|
"<SCPDURL>/lanhostc.xml</SCPDURL>"
|
|
"</service>"
|
|
"</serviceList>"
|
|
"</device>"
|
|
"</deviceList>"
|
|
"<presentationURL>http://192.168.1.1/index.htm</presentationURL>"
|
|
"</device>"
|
|
"</root>";
|
|
|
|
using namespace libtorrent;
|
|
|
|
struct parse_state
|
|
{
|
|
parse_state(): in_service(false) {}
|
|
void reset(char const* st)
|
|
{
|
|
in_service = false;
|
|
service_type = st;
|
|
tag_stack.clear();
|
|
control_url.clear();
|
|
model.clear();
|
|
url_base.clear();
|
|
}
|
|
bool in_service;
|
|
std::list<std::string> tag_stack;
|
|
std::string control_url;
|
|
char const* service_type;
|
|
std::string model;
|
|
std::string url_base;
|
|
};
|
|
|
|
TORRENT_EXPORT void find_control_url(int type, char const* string, parse_state& state);
|
|
|
|
void parser_callback(std::string& out, int token, char const* s, char const* val)
|
|
{
|
|
switch (token)
|
|
{
|
|
case xml_start_tag: out += "B"; break;
|
|
case xml_end_tag: out += "F"; break;
|
|
case xml_empty_tag: out += "E"; break;
|
|
case xml_declaration_tag: out += "D"; break;
|
|
case xml_comment: out += "C"; break;
|
|
case xml_string: out += "S"; break;
|
|
case xml_attribute: out += "A"; break;
|
|
case xml_parse_error: out += "P"; break;
|
|
case xml_tag_content: out += "T"; break;
|
|
default: TEST_CHECK(false);
|
|
}
|
|
out += s;
|
|
if (token == xml_attribute)
|
|
{
|
|
TEST_CHECK(val != 0);
|
|
out += "V";
|
|
out += val;
|
|
}
|
|
else
|
|
{
|
|
TEST_CHECK(val == 0);
|
|
}
|
|
}
|
|
|
|
int test_main()
|
|
{
|
|
// test upnp xml parser
|
|
|
|
parse_state xml_s;
|
|
xml_s.reset("urn:schemas-upnp-org:service:WANIPConnection:1");
|
|
xml_parse(upnp_xml, upnp_xml + sizeof(upnp_xml)
|
|
, boost::bind(&find_control_url, _1, _2, boost::ref(xml_s)));
|
|
|
|
std::cerr << "namespace " << xml_s.service_type << std::endl;
|
|
std::cerr << "url_base: " << xml_s.url_base << std::endl;
|
|
std::cerr << "control_url: " << xml_s.control_url << std::endl;
|
|
std::cerr << "model: " << xml_s.model << std::endl;
|
|
TEST_CHECK(xml_s.url_base == "http://192.168.0.1:5678");
|
|
TEST_CHECK(xml_s.control_url == "/WANIPConnection");
|
|
TEST_CHECK(xml_s.model == "D-Link Router");
|
|
|
|
xml_s.reset("urn:schemas-upnp-org:service:WANPPPConnection:1");
|
|
xml_parse(upnp_xml2, upnp_xml2 + sizeof(upnp_xml2)
|
|
, boost::bind(&find_control_url, _1, _2, boost::ref(xml_s)));
|
|
|
|
std::cerr << "namespace " << xml_s.service_type << std::endl;
|
|
std::cerr << "url_base: " << xml_s.url_base << std::endl;
|
|
std::cerr << "control_url: " << xml_s.control_url << std::endl;
|
|
std::cerr << "model: " << xml_s.model << std::endl;
|
|
TEST_CHECK(xml_s.url_base == "http://192.168.1.1:49152");
|
|
TEST_CHECK(xml_s.control_url == "/upnp/control/WANPPPConn1");
|
|
TEST_CHECK(xml_s.model == "Wireless-G ADSL Home Gateway");
|
|
|
|
// test xml parser
|
|
char xml1[] = "<a>foo<b/>bar</a>";
|
|
std::string out1;
|
|
|
|
xml_parse(xml1, xml1 + sizeof(xml1) - 1, boost::bind(&parser_callback
|
|
, boost::ref(out1), _1, _2, _3));
|
|
std::cerr << out1 << std::endl;
|
|
TEST_CHECK(out1 == "BaSfooEbSbarFa");
|
|
|
|
char xml2[] = "<?xml version = \"1.0\"?><c x=\"1\" \t y=\"3\"/><d foo='bar'></d boo='foo'><!--comment-->";
|
|
std::string out2;
|
|
|
|
xml_parse(xml2, xml2 + sizeof(xml2) - 1, boost::bind(&parser_callback
|
|
, boost::ref(out2), _1, _2, _3));
|
|
std::cerr << out2 << std::endl;
|
|
TEST_CHECK(out2 == "DxmlAversionV1.0EcAxV1AyV3BdAfooVbarFdAbooVfooCcomment");
|
|
|
|
char xml3[] = "<a f=1>foo</a f='b>";
|
|
std::string out3;
|
|
|
|
xml_parse(xml3, xml3 + sizeof(xml3) - 1, boost::bind(&parser_callback
|
|
, boost::ref(out3), _1, _2, _3));
|
|
std::cerr << out3 << std::endl;
|
|
TEST_CHECK(out3 == "BaPunquoted attribute valueSfooFaPmissing end quote on attribute");
|
|
|
|
char xml4[] = "<a f>foo</a v >";
|
|
std::string out4;
|
|
|
|
xml_parse(xml4, xml4 + sizeof(xml4) - 1, boost::bind(&parser_callback
|
|
, boost::ref(out4), _1, _2, _3));
|
|
std::cerr << out4 << std::endl;
|
|
TEST_CHECK(out4 == "BaTfSfooFaTv ");
|
|
return 0;
|
|
}
|
|
|