UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   , i* H# d0 ^) D! }
3. 
   5 L7 @# {! h% {; d$ e
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   ! Q1 V4 c, ^" y# C
7. 
   
8. class   MyUPnP
   . _6 g/ L0 E7 k# `; V- v
9. {
   ( k/ l/ I- A" p, k& O7 p2 p
10. public:
    # V! T1 g/ ~8 V6 |! E& L+ K7 t! {3 ?
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    7 P0 b; Z6 v: A8 c$ u( K; V
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    , _1 f; U9 o+ B$ g" I5 ^5 \
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    1 a/ K9 E- f1 z8 J2 B+ I
17. }   UPNPNAT_RETURN;
    4 d+ x7 l- ^: E. e' ~
18. 
    9 G3 {: Q1 h( T% H
19. typedef   enum{
    ' n5 [. b, I5 C  J& E: Z# u
20. UNAT_TCP, //   TCP   Protocol
    ' Z. }7 Y. \/ t. b2 `
21. UNAT_UDP //   UDP   Protocol
    3 o, D9 u$ C" g
22. }   UPNPNAT_PROTOCOL;
    + c/ o% [! H) j+ Y' c
23. 
    
24. typedef   struct{
    ' E7 B' W% M6 `- g( k1 w# n
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    # t/ o+ u4 N1 }. w8 V& e+ m
28. CString   description; //   Port   mapping   description
    ! v8 T0 @! u) s0 i
29. }   UPNPNAT_MAPPING;
    , n* D2 i, D! M/ t( h
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    : j4 |% }; B5 j6 N( S
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    0 Z; U$ T/ u8 ~
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    ) j# |& ^9 p/ v5 R& J
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    * v( g4 X$ M% L. H$ @: s* @
46. 
    - w- p+ H. n& a8 c* S4 Q, U
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    # \3 ^9 `) q$ ?" b, a& {
49. bool   deletePortmap(int   eport,   const   CString&   type);
    ! s" m3 M+ s" b$ K' w
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    7 V$ F9 H: a; d, @- `% L4 v
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    , B( O, K: ^+ s  i) d5 x+ C, W+ e
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
    % F, q, r) h3 G9 k
62. CString m_description;
    % K7 L/ d- ^! G! E) F+ z6 ~' Q
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    ( q0 ^6 j, \! C/ f6 e
66. CString m_modelname;
    
67. int m_version;
    8 {# I& R1 x" e% b0 i9 Y
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    - T. P  @7 Y) B1 v/ O2 w
71. 
    
72. CString m_slocalIP;
    - `) U) }; ~, K  J9 M/ q
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
      C# [* e6 x4 x% q
76. bool isSearched;
    ' N) h. S2 l) ^0 G; y: y# b1 f
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   & e5 _0 \7 m# g: r" R5 v
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   5 z5 m' [# C! Z) U
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    4 `" Y1 c% Z7 ?; g* y7 Z
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    7 Q9 d# p: n! n. D/ o3 d& ^6 H3 {+ W
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    " [/ a2 }7 ]( }  _
16. const   CString   getString(int   i)
    
17. {
    4 q. c4 l" ~7 p3 `6 A5 s' Q
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    7 |6 m/ T" g' ?' O8 Y6 p8 h. c
21. 
    
22. return   s;
    7 z: t& }/ m1 N! H4 v$ q6 A
23. }
    
24. 
    : z. n' M1 D. e; O+ ^
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    9 s6 _& y& e9 C; i0 j- E" U8 Y1 ?3 A
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    * N" B; o  v# F) m. A
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    " i: M7 L' F9 f
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    3 D8 i. S$ Y. H
33. }
      R+ f: X3 \# u4 u7 p8 }1 }
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    * r2 K7 J$ V6 D  l! J$ _
36. {
    8 E/ B- k/ \! O( B4 \
37. char   buffer[10240];
    
38. 
    $ p6 V# Y0 |% u  _; t
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    1 A' s5 E) ]3 e' X9 @4 S
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    2 G3 S! _0 n' {# S8 B! J2 n' _
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    ) a& o# H: X5 u5 e. d- X. A7 k, z
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
      ~  U4 k7 F/ ?/ B
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    ' n! P0 m2 A# K9 f& d$ v
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    % m7 ?* S* F: T
59. if   (!rlen)   return   false;
    ; R" @; c; @* g! b
60. 
    5 ]5 _# ]" u* Y6 ?1 a
61. response   =   CString(CStringA(buffer,   rlen));
    5 w0 u* j9 a. P4 P9 c# m; F
62. 
    
63. return   true;
    
64. }
    3 O& w2 [" J- e4 ^
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    # }: ~4 _  m2 N6 S5 U
67. {
    # @/ ?# Z) p6 b* _9 K4 H
68. char   buffer[10240];
    3 \; T) x9 C' k  J" m5 d7 P
69. 
    
70. const   CStringA   sa(request);
    8 r6 v% M0 @' d0 F' [' V% T
71. int   length   =   sa.GetLength();
    5 q- M# P  z( ]9 h& o
72. strcpy(buffer,   (const   char*)sa);
    ; R8 f) i4 m7 ^. L& d; p/ [
73. 
    
74. struct   sockaddr_in   sockaddr;
    ; R6 \' k1 s, ^% Q
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    # f+ [0 X5 e3 Q0 n9 t4 `
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    ) x9 A5 {9 k5 A( R) s' _! t% O
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    # R8 y9 }9 i2 G, B
82. 
    ' L8 K3 Z2 [) |+ g" t) _
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    & v- t* P1 V3 c8 o' |/ ^- G
93. status.Tokenize(_T( "   "),   pos);
    0 p% I6 @. W3 g, j+ [
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    4 U; g8 o& @% l% @
97. }
    + B% O- R5 r+ `% r" Q: i2 k
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    ; m, \4 g! v( T: j; F9 E
100. {
     - R6 B  i; Y# O* k/ a$ B$ b7 C; F
101. CString   startTag   =   ' < '   +   name   +   '> ';
     7 F3 i% Z* x: h. ^9 n
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     $ h/ @# M8 [3 L# a1 n
103. CString   property;
     
104. 
     ( a- J' E% L) i" b1 E
105. int   posStart   =   all.Find(startTag);
     " t! d! B0 |' x
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     " {6 ^! `5 U* G4 Z% F, F1 g% K  `! h
110. 
     ( {! N0 o$ N  e& B$ n# c" E5 Z
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     ; s3 d1 ^  u6 P1 q" [# Q2 p$ h
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     1 P! Z1 d8 v  n' {; r. K3 E9 O% ?
118. isSearched   =   false;
     
119. }
     ; `) U, T1 d8 K& D7 o1 F7 V
120. 
     - T! I5 Y9 V. K% E& ~2 e
121. MyUPnP::~MyUPnP()
     0 s3 N7 }$ x3 v0 `% M; f7 i$ i; t9 U
122. {
     
123. UPNPNAT_MAPPING   search;
     8 t5 \8 f0 B9 ^* q' s
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     . e3 e) s* W9 b) x/ C( u
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     ( f2 W  N' t( k  ?0 y+ f6 n# m$ ~
129. 
     
130. m_Mappings.RemoveAll();
     * M3 d% w& ?- R7 ~
131. }
     1 o% e" r- x  @6 T
132. 
     
133. 
     8 t2 |$ P* Q* T' s3 o
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     # b, G, K& v# W. ?' O+ ]- Q! Z. S' U
136. if(version <=0)version   =   1;
     2 }& l+ T5 C' i4 w/ H
137. m_version   =   version;
     ( k, N3 y7 P+ v; b8 j6 ?1 `
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     ' e- p. A- ?5 B3 @  \
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     ( x2 l* ~7 W- l) n4 K
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     4 |3 n: G- Q+ k8 f. m( f
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     6 ?- s! @# h- P
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     : O3 a" E) D) i) W. M
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     + a0 z8 l! g! F6 K
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     5 G6 y2 @# q2 R; X
156. request.Format(_T( "M-SEARCH   *   HTTP/1.1\r\nHOST:   239.255.255.250:1900\r\nMAN:   \ "ssdp:discover\ "\r\nMX:   %d\r\nST:   %s\r\n\r\n "),
     
157. 6,   m_name);
     6 K+ O+ G  v. ]4 {
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     5 J6 p0 ]$ J# n6 O5 r" h+ Q
159. }
     $ x; h+ t1 G2 o$ u
160. }
     
161. 
     
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     6 \' F: k; z( {( o1 _' W4 \
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     & Q/ @2 |" c9 m7 X4 J- h
167. closesocket(s);
     
168. 
     & e# ]% [6 a5 y0 `8 s( z4 w
169. CString   response   =   CString(CStringA(buffer,   rlen));
     5 u2 w% k$ Q% t  J2 W/ }5 M  K
170. CString   result;
     9 a0 H# E; V  b3 |9 C/ G+ x
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     ! d! ^7 d' }  P, b' P! k& n3 K
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     ! J9 U/ e+ Y2 e; `
175. if   (result.Find(m_name)   > =   0)   {
     8 q9 j; a& d. y% y/ G- b
176. for   (int   pos   =   0;;)   {
     . a! G) M5 b0 O  M3 m
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     2 a  @! h& M1 c# s* o
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     # K6 _  y8 J& o* }$ @1 v/ G
182. line.Delete(0,   9);
     1 v# ^! ]& e% w. b, Z- o1 s
183. m_description   =   line;
     
184. m_description.Trim();
     " k2 v5 g8 C6 A/ C8 @9 N
185. return   GetDescription();
     ( y* C& C+ i4 A0 C+ e2 W: I
186. }
     , L- d* G! Y/ p; e8 n8 S
187. }
     
188. }
     
189. }
     
190. }
     
191. closesocket(s);
     
192. 
     
193. return   false;
     
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，

8 `6 G9 }* W8 S! J3 J
///////////////////////////////////////////
5 A  l2 s7 W; E9 k" S7 g//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

$ ~2 f$ R4 [5 W( y
4 e. c7 u4 \0 c- @#pragma once
#include <exception>
  J7 f$ X6 x% c! _( C

9 d0 |# {" y1 F  enum TRISTATE{
. l0 k9 X: o  t: I4 R3 r" @$ }        TRIS_FALSE,
0 L1 y- ~9 ^, H9 \        TRIS_UNKNOWN,
9 Z, p( K$ g2 c# c        TRIS_TRUE
};
4 I& A. h: w2 u& g

enum UPNP_IMPLEMENTATION{
& O6 m+ E& w5 l3 B        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
! t7 O% r0 L6 G$ ^! \};
1 r" {' t. f8 q% J



5 ?+ ~" E0 e  ~$ c) i( `1 {class CUPnPImpl
{
public:
        CUPnPImpl();
- X% L9 h+ j- C8 E! s        virtual ~CUPnPImpl();
' j9 y% P+ m3 f$ K0 Q        struct UPnPError : std::exception {};
( e5 ~- m7 k* `8 s* S; A8 y0 G/ k5 F        enum {
                UPNP_OK,
2 ]5 p( b; e- r3 s4 A: ~                UPNP_FAILED,
                UPNP_TIMEOUT
9 t- ]$ C; c7 m9 l. J7 _        };
$ k' S8 {% m* H9 p$ P
$ |8 Y! |4 f- u8 j/ b6 O7 v
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
6 U# [# N. d* ?' I. d2 Q: j9 ~        virtual void        DeletePorts() = 0;
* w, @" @3 t: a1 Y+ l( I7 `        virtual bool        IsReady() = 0;
, e% Y) k) j+ j  k        virtual int                GetImplementationID() = 0;
+ v0 {. O& M1 J5 A1 c8 r       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


; V3 b. t" m% c; Z0 x, h; B        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
: |7 d: Z9 X0 Q9 E9 T
# a  N3 o; O0 }- q# n0 x
/ ]6 v9 A2 K9 G6 t// Implementation
protected:
& O! L+ f# n3 @4 w9 ?( x1 T/ G        volatile TRISTATE        m_bUPnPPortsForwarded;
$ A6 @% I4 }  g* Q6 k        void                                SendResultMessage();
        uint16                                m_nUDPPort;
, {. Y8 k$ ]: @3 q+ d        uint16                                m_nTCPPort;
% J* D- z( c$ x4 p        uint16                                m_nTCPWebPort;
* J( f  L( i6 o) N        bool                                m_bCheckAndRefresh;
1 k  _7 z) g. j* u4 A- h

( v2 J0 e! I0 d- e# A1 p% Jprivate:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
0 y  H! `. q4 ~
* |. n! G7 N* i0 b& x
};


; L/ \2 a% k& K+ }) T& G  s- z// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
' W/ S" \- C1 G+ I5 W{
public:
. p" z1 U) ^' c. I1 N        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
$ Y0 ^9 k0 S6 e* ?        virtual void        DeletePorts()                                                                                        { }
: q& g; |0 O, ]9 _* y8 p! B! s        virtual bool        IsReady()                                                                                                { return false; }
3 a8 e  C7 L9 P! ~, S" M" p6 b$ N) R        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
3 ?: Q+ Q, R8 @" y- M' R};


/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类


#pragma once
: g' k( M. K! J/ D8 V: ~% J" y#pragma warning( disable: 4355 )
& V- k" N$ B! \9 J3 x* S5 d& o- [

! n3 a. P- @% M2 W$ J8 N#include "UPnPImpl.h"
! x4 x0 T, B4 b" ~' a#include <upnp.h>
  }: G' X# t4 v8 T#include <iphlpapi.h>
#include <comdef.h>
; t9 y2 ~( H0 `6 L5 x6 }#include <winsvc.h>

  @2 E* y" C* |, p' y
) z9 p5 A( }6 U2 P& t* x" f  ?#include <vector>
! F" B; c5 j5 p4 T5 Q#include <exception>
' Z; z$ \. \8 [/ `, I#include <functional>
0 S) g2 f7 T5 {+ u

( Y( A- M3 P! r

typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
/ l' j1 s" n( D0 Ftypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
% [6 F  J9 \' Z  J6 Ftypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
! t2 n3 _: w; U6 G/ t/ jtypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
6 t/ I3 t0 K; l* E0 N1 S
& [/ R" g; E3 L" R, D
1 Q8 Y2 i3 N) otypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
- n' H! V0 z* s. c+ N! B  PDWORD pdwBestIfIndex
. M& K; M! h) D+ D);


' p$ ]6 P# P+ u/ z8 @! G1 _typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
: v1 e- ^1 B5 ?; G9 U  PULONG pdwSize,
3 g+ y6 q' j, a1 z6 `0 W  BOOL bOrder
  \% H9 ]: F; u# Q# V+ _" d. b- h);
* F: M7 F6 ?7 ~6 @1 L" o8 i

typedef DWORD (WINAPI* TGetIfEntry) (
; Q# L) c: e2 C) ~) ]5 Z3 n3 d  PMIB_IFROW pIfRow
);


8 @7 S4 ?" Y; SCString translateUPnPResult(HRESULT hr);
& |$ g5 v# _" z& f' Z  |1 C/ a" M5 LHRESULT UPnPMessage(HRESULT hr);


class CUPnPImplWinServ: public CUPnPImpl
5 Q6 |2 C# v3 ?$ B  w1 E{
. z! F) w, h- D, k        friend class CDeviceFinderCallback;
' S' m3 x3 B. ]: ^        friend class CServiceCallback;
// Construction
% g; z- q  ?! x8 W- K/ C# apublic:
        virtual ~CUPnPImplWinServ();
% q8 M  ]0 c( i$ o9 N# e  W5 X        CUPnPImplWinServ();


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
" U4 Y/ M6 p4 R9 H1 W        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
, c% C; n  b* E        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
# V. Y& Z7 o% F- [3 j! o1 F

# I. f2 K! W1 n1 x" `6 T( T# _; t        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };


protected:
* G, m$ Z: O( Q' e8 I# m" I& R9 K        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
7 k& T  y5 R- w% @7 w4 d. v# H7 `        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
/ Y; l2 P; E% ?3 S; W7 |        void        RemoveDevice(CComBSTR bsUDN);
# R; S9 E& u6 A) a" w        bool        OnSearchComplete();
        void        Init();
0 y( `2 w9 N' m; B8 N+ i

, q; a$ k- P  O$ k1 t# `        inline bool IsAsyncFindRunning()
! j! E6 G5 I% t$ B0 P/ C3 F7 w        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
7 P( N/ M8 d( \0 @                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
' n: S. `6 \: Z$ P- |7 Y4 \6 R& s                }
% T0 r6 I9 m6 s, ^6 V8 V0 m6 u                MSG msg;
' W/ ~  v' n& _' a# t- S, P9 f4 _                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
2 l. H1 M) e, P8 Y* u                        TranslateMessage( &msg );
  T! |- n& s) B2 N( b4 Y+ K                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
        }
2 t4 i2 q" }/ P( b

        TRISTATE                        m_bUPnPDeviceConnected;
4 c. I3 T6 E0 f& h

// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
& M0 ?- [, |* \! Q) Z5 m        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
) `, _0 i( w1 W9 b        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
6 W3 H: O, @7 O        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
, `$ O/ C7 L" U( S8 p, J' }        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


3 M& M3 }3 k5 d8 j! J5 b        TGetBestInterface                m_pfGetBestInterface;
; f1 P8 a0 w9 o0 u3 v        TGetIpAddrTable                        m_pfGetIpAddrTable;
  T0 \8 I/ E. [/ a        TGetIfEntry                                m_pfGetIfEntry;
( t. j: D2 B/ w/ q; a' }
4 {. h1 R3 R2 |7 [' E: L
- l: `2 R6 [& v# h0 V        static FinderPointer CreateFinderInstance();
4 h4 |. F$ ^/ M$ i        struct FindDevice : std::unary_function< DevicePointer, bool >
% q9 J1 e: G3 J+ J7 H        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
) T' k4 y1 _5 B0 `6 G                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
* ]9 W' [0 y: w! f) W: e- F
0 M/ E7 s' l* R7 e2 X# O! n
                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;

* @4 ^* @+ [3 Y
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
) V' m0 b0 G4 ]' b0 o                }
1 m7 N( k. F  y/ L- Q: l                CComBSTR m_udn;
4 Y  f! g/ z& k        };
3 d  R) Y3 z) L. t1 H$ i' T3 j       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
+ M1 H& }7 W8 H) O5 s        HRESULT        GetDeviceServices(DevicePointer pDevice);
* ^8 R! E$ G  _! _+ h0 @        void        StartPortMapping();
; Y* {) L. M' N8 [        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
5 q. Z# \3 q8 r: {8 @! u7 e, q        void        CreatePortMappings(ServicePointer pService);
: F& X- p' o% g. C& k        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();

) n5 M2 R7 B! C( N) x8 ~7 g2 ]3 l
0 u) r: j6 g% \        // Utility functions
3 j! e& N0 W. B& _: Y        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
, `" @" U3 {! W% B        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
6 k' j) E6 R' I' f  @/ ~        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
" p) n% |6 H3 A6 O0 \. s        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
) o' I5 R6 S: R6 O/ r        CString        GetLocalRoutableIP(ServicePointer pService);
' E  v' w4 a$ V  m

5 }, F/ H; M  G0 e" W) ~2 |+ k// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
) S: y$ o1 X5 x" H        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
( A  x3 D; F. f% `        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
4 }. A) j9 \" n1 T/ \

        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
  m) u7 U! h; X; m3 }7 v        bool        m_bPortIsFree;
% t' N" ]0 d0 |, Q  N- h6 Q& K        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
  g$ ?7 i0 P0 R. M0 k0 Z( b& l        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
8 ]6 i; l+ E9 Q  Y        bool        m_bInited;
        bool        m_bAsyncFindRunning;
2 I, z' ?/ I: [, }        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
' v0 Z" x: d. u; |, E5 P        bool        m_bServiceStartedByEmule;
6 h2 s/ @* c% w; Z) V% j        bool        m_bDisableWANIPSetup;
: `' ^+ N7 N0 t7 \, g        bool        m_bDisableWANPPPSetup;


  o$ f) Y& O/ Y6 l6 u5 }9 m};

' S2 k! v' u5 O. i  B& m
// DeviceFinder Callback
class CDeviceFinderCallback
  C# d7 h1 o7 s! k2 k- |( l        : public IUPnPDeviceFinderCallback
: L9 X& {* y0 _1 D{
. \% ^2 e) b) @4 h$ g5 Ipublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
4 L# }. A  N7 B4 |! J                : m_instance( instance )
1 H5 [5 o" G" h- Z$ F( n        { m_lRefCount = 0; }
# H  F8 }2 A: m3 g0 K

   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

/ R$ ^7 S2 S$ c! f* f
// implementation
1 z0 q( W/ o6 B% Y9 e7 Uprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
0 j2 W$ f0 Z5 N3 m' p9 i+ U4 S        HRESULT __stdcall SearchComplete(LONG nFindData);
# d+ {- E2 J  M% L4 J: x. d
% s, |! O2 z& |# [" v$ o
private:
* o2 q% @$ k  k1 U- }  R3 T4 _% B        CUPnPImplWinServ& m_instance;
! v4 t( O" ~* S# f        LONG m_lRefCount;
$ C$ O0 [7 h' c# z};

: Q. T* ~, S7 n  H/ _
" M3 ~5 Y! c  }// Service Callback
# W( V! q9 j8 t0 Lclass CServiceCallback
        : public IUPnPServiceCallback
$ M( T" H# @/ O# [! }9 ?{
) ?9 C/ [8 J) G, Zpublic:
        CServiceCallback(CUPnPImplWinServ& instance)
$ b% v6 ^& ?& [# H                : m_instance( instance )
; F( I( f& R5 T% p3 _        { m_lRefCount = 0; }
+ F* m4 ~$ c+ U6 W8 v2 u   
; x- @6 y' @/ b* x* Z) J   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
& q. a% L4 E) G1 `; `  Z0 g   STDMETHODIMP_(ULONG) Release();
. I, \6 |. a5 c* u( `' c

4 W! c- b" J3 L/ K3 V4 F4 Y// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
! K* U1 t9 z7 @3 \- s4 W9 Q

private:
        CUPnPImplWinServ& m_instance;
& T4 b2 g2 W0 M+ g        LONG m_lRefCount;
};

2 K+ i" d: I; R9 M4 c$ s
/////////////////////////////////////////////////


; f  P' M) n) k& r$ w+ T4 x使用时只需要使用抽象类的接口。
, q5 x9 t4 @1 v7 X5 K$ mCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.