UPnP

天云上人

/*uPnP.h*/

1. 
   ' u& A8 ?7 c& q# Y
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   8 g3 D  [9 ?$ c
8. class   MyUPnP
   
9. {
   
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
      p4 r# T, |1 Y: @% e# Y: V
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    5 n" U) w8 a4 ^8 p
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    : C; t6 e3 `  ^/ W5 K' Z
17. }   UPNPNAT_RETURN;
    ( G2 \5 R/ F# V4 O' j' {
18. 
    
19. typedef   enum{
    . e+ _8 }7 i" A) d3 @6 e; T2 @$ U2 N
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    + X* a# Z* Q9 {( G: W
24. typedef   struct{
    6 Y) |2 I* L0 X5 x5 @3 ]
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)
    
28. CString   description; //   Port   mapping   description
    - |  ?1 a! S" F& l
29. }   UPNPNAT_MAPPING;
    
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    7 `* i0 `5 s2 n8 i
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    ! S$ @. i9 j1 @5 F& [, M
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    3 R4 C$ R: j0 R6 g# _7 O
36. void   clearNATPortMapping();
    
37. 
    ; ]" A0 A2 N8 v# |' g! p
38. CString GetLastError();
    6 b; ^2 a% j; t2 }
39. CString GetLocalIPStr();
    : `. b, d0 u7 F% I
40. WORD GetLocalIP();
    ' Y7 V( ]$ l; B' U# \! k8 w2 Q
41. bool IsLANIP(WORD   nIP);
    - Y# K6 B# a) K8 V/ p
42. 
    
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    7 c! a; h% K+ s" \( ?) n9 ?
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    / Q5 k. z0 E: \  r+ A9 `
48.       const   CString&   descri,   const   CString&   type);
    & R. `6 E2 f! B0 L, q: y" N# e2 m3 O
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
      ?' o/ ~# E+ s9 ?) ~1 [
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    , `5 x) C" {* o3 a4 h+ x3 H
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    : n- B( \' n1 v/ {8 Z/ j: F( s/ D
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    ; w) O2 M  M3 h$ N3 z) d7 `9 @
59. bool InternalSearch(int   version);
    3 @& X2 c' x+ ^% i" X; g
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
      F- w- ~6 _$ T9 d* W0 e
63. CString m_baseurl;
    2 k& V# I( w& C& y2 D
64. CString m_controlurl;
    + y) C: s: s% p5 l/ j
65. CString m_friendlyname;
    1 h! r. m% L4 |1 S, g
66. CString m_modelname;
    
67. int m_version;
    ' C/ H8 d- m0 F' g2 ]7 d# B1 }
68. 
      B8 x+ Z7 d9 p2 ^) y$ J
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
      e, P8 h- ^2 \( l
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    6 c% S( L: ~! |& X
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   / o& D5 a, W2 x8 O
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   , c# t0 |! m8 H& g  }4 X% k
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   ! R4 F& X+ |+ j6 U4 F/ c
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   ( N4 b# m! ?% E! K( D
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    $ J; `0 h+ ]0 e+ r' B
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    * D3 v4 m; d4 W) ^. [$ n
13. static   const   int UPNPPORT   =   1900;
    ) e+ I+ Y/ l6 ^4 r# h( J$ y; w- E7 b; ], r
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    . f8 f. Q0 V4 {# N
17. {
    
18. CString   s;
    ; T% }. M* e" ^. v
19. 
    3 d5 x1 `+ W  F- l% u4 ?# [, n
20. s.Format(_T( "%d "),   i);
    ! T+ i/ H% E; A
21. 
    
22. return   s;
    
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    0 f7 e2 `" G7 J( Q7 A2 T+ }! u4 n
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    7 U# ~1 V( L7 B5 _1 m
31. {
    / n' o$ Q2 E; S3 s& Z; R
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    1 C2 p" t0 y, E. {) W# r) Y8 ^
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    & c# S. B9 E3 Q6 Z9 h1 K" L* g+ R  `
37. char   buffer[10240];
    7 M' z: L& b/ X* e+ F; ?
38. 
      b9 o- P8 L6 w
39. const   CStringA   sa(request);
    7 ^) P6 ~+ q9 }) g. s0 w6 [+ H' L
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    $ @9 v6 I: v/ ~9 w. b
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    ( q4 R' u. j1 K8 m0 l4 Z
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    2 A8 C+ E  S) R) S8 S  u8 o
46. sockaddr.sin_family   =   AF_INET;
    - E9 Z. O4 A# k0 n' M. }8 j
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    $ e7 i9 f3 }) x7 B. e- Z
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    % l* x& a3 V7 C6 @5 J9 v$ ]' B
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;
    
59. if   (!rlen)   return   false;
    
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    
64. }
      `1 _: f* u6 I7 P/ @3 H; X1 [
65. 
    $ [: @' f4 U, @+ Y
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    ( F9 z/ c1 G5 m0 M3 k$ h: m
67. {
    
68. char   buffer[10240];
    1 C2 J, J/ T8 ^7 s
69. 
    
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    5 L0 w4 W, T' S" u
73. 
    
74. struct   sockaddr_in   sockaddr;
    ; x; Q0 _9 {- N9 w4 @, ~5 R! l
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    6 ~! ~; G7 H7 {: P9 {3 o
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    0 B+ s: e) c6 N* h! W7 K
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
      b! g8 e! b% f
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    - E1 y! n+ H. a$ L6 {
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    
86. 
    ' I( h% m8 D1 B5 P) f0 Y+ ?* ]) P5 `
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    % j% g7 D1 L. q9 s9 i
88. 
    
89. result   =   response;
    , q1 C/ f8 b) Q( P6 `; d# m5 C4 B: ^
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    # z( Y+ U" K! e9 ~1 [; y
94. status   =   status.Tokenize(_T( "   "),   pos);
    % m. [: V/ a' B2 o. q
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    
98. 
    3 a: m, R1 R/ f/ z
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     . i. S& }: Z4 X4 f3 E
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     4 S6 _( a/ M$ k! K
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     & l* J+ F  ?, X6 k5 F  N$ T7 O% \. W
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     6 {" s1 Z! T( X  e
115. :   m_version(1)
     - d+ P7 }' ^: m. ?2 R7 ?
116. {
     3 k6 m# o8 I, f
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     . v/ j. W+ K; d# d( x+ g) r
122. {
     ; r$ h; L  t5 \6 U! g4 V
123. UPNPNAT_MAPPING   search;
     ) ?) d* P1 t8 A( `) J
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     8 l3 a$ {0 \6 R5 ^: E
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     & p7 D8 U! t2 w* {* B! A1 m
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     
131. }
       v- A" S  m# s: o( A
132. 
     + w9 k( \. h- b. F/ ~, Q: w) L
133. 
     " N5 u$ A, e( v
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     % r. a( F8 s, a+ _
140. CString   devices[][2]   =   {
     ; c0 Q' F" _; p; d, A
141. {UPNPPORTMAP1,   _T( "service ")},
     : m4 J7 `( B7 _# }4 ^! i( c% ]
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     ' T* G3 Y+ K; D# n8 i- p/ W
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     7 h8 n( S4 I' ~, `% \; }- q
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     # D6 L% N3 \5 o6 Y# D4 C
149. 
     ) s7 T8 z+ A) K# [: h  t' t
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     ! n" o/ b8 M+ ]) O" l) @4 X
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     + @' Z2 B% @( ?# \  B, C8 g. j
155. CString   request;
     . X0 h4 |0 q3 k& m, u" W& Q  {
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);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     9 V, A$ q$ B8 Y- s. x6 W- d% Q' i3 a0 ?
159. }
     6 g$ b+ J4 b/ L& y9 H/ b
160. }
     
161. 
     
162. Sleep(10);
     8 Q; B7 b/ K2 V' f
163. 
     ) f8 t5 E) t5 H
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     , }0 G* e8 c$ q0 n0 \
168. 
       H: m% g! I2 I- p
169. CString   response   =   CString(CStringA(buffer,   rlen));
     ; k5 L- \  Z& l$ q& q, \
170. CString   result;
     ; N3 h3 H6 S" d. m5 h5 S; H4 X7 D
171. if   (!parseHTTPResponse(response,   result))   return   false;
     ! F7 Q! c' p$ D: T6 z
172. 
     , p/ v- S* T* X0 E; l
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);
     & |* t, Z1 m4 Z/ g0 K
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     ! ^# K  Y! N& B; _
180. name.MakeUpper();
     6 ^% O8 L9 Z+ |: T& B5 O, G
181. if   (name   ==   _T( "LOCATION: "))   {
     / ~& O  R, N/ `( }9 [
182. line.Delete(0,   9);
     ' ?/ W4 P/ w. a
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     
187. }
     
188. }
     
189. }
     
190. }
     ' p) E! ^' r0 S; {+ l6 c  J# t
191. closesocket(s);
     
192. 
     % l4 c; d7 Q% k' y
193. return   false;
     
194. }
     . p0 Q; _; ^4 h9 M  N# T$ ]8 u$ g

复制代码

天云上人

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

1 Q% n6 V! v/ \0 B# k
///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
/ M6 E1 t, V+ V" K
* Y. e2 i2 ?) U, c# s3 V
3 k) e! D* h: c' g#pragma once
5 g/ J) V: o7 G# V; A5 G#include <exception>
' K+ j0 {  A4 l/ N" t
( c, @' K( f/ f( f  Z3 D$ d; a
# `' k  Z6 X3 @  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
1 F) T8 P7 e7 H+ L$ L+ N" @        TRIS_TRUE
};
8 K2 }7 F: f; {+ O* v

enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
& W& B. e! g1 d' i0 e  y6 i! m6 D        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
+ I% F# Z; W, L};
; t, {/ _' @, H4 {$ z

9 p, V: ]- S3 v$ X
+ o1 u4 u7 F0 t
  G" b+ o: U& k8 V; ^, ]class CUPnPImpl
{
1 z& P( \/ `5 z. jpublic:
        CUPnPImpl();
( ?* r; T3 R% _3 ^1 @( |6 A* ^        virtual ~CUPnPImpl();
8 ]9 g4 r( U: e( M% m. K        struct UPnPError : std::exception {};
& s; X+ q, S6 H. a        enum {
                UPNP_OK,
! \5 ~0 V, c! y8 T* K2 _# `0 t                UPNP_FAILED,
% [9 A  l6 R0 S6 X# t                UPNP_TIMEOUT
" f. z, N. b9 u# Z. B% X9 W1 F        };
  L, Z- s, ^2 I( j* D( i" I

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
- @" y: e9 I- d1 ^, y        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
3 J! g/ h/ i0 [# Y. E% z        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
  D$ {6 _+ {2 B* ?# M9 b3 G  `        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

: ?3 m2 x" G: v, e0 G! P/ G
3 L( ^  Q  n' o  D// Implementation
protected:
& L; Y( e$ l% [        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
7 Y6 ^* R6 f6 ~- M+ a        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;

6 l( y# L- G" x: O" B
private:
        HWND        m_hResultMessageWindow;
, \. g4 I; Y; i  G8 _6 U% d        UINT        m_nResultMessageID;

& Z. h  \. q0 Q8 o
};

" x) [' \' V" O6 I2 p: \
// Dummy Implementation to be used when no other implementation is available
$ Z3 P1 C% }0 N$ Zclass CUPnPImplNone: public CUPnPImpl
" l6 C4 d3 S8 d  g; q1 ?{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
3 U* c* y8 e# C# a4 f& h; {        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
5 r% ?" E; t# J3 A- g};


5 e* W0 [7 A. O0 Z. [, x3 H$ W+ f/////////////////////////////////////
- N1 [- F, d1 m8 A0 O2 ?) n( W//下面是使用windows操作系统自带的UPNP功能的子类

* D7 D& y7 m$ L) V9 p2 s4 _
#pragma once
' Q# d- L3 Q6 j* `- @' t#pragma warning( disable: 4355 )
4 k4 @! Q6 d* a1 ?6 w

#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
& E$ G5 B) S, e5 C! j1 ~% W#include <winsvc.h>
# v* i& g, T6 n: S! U' o- m

#include <vector>
4 v. W5 D( w3 ~2 l" W5 Q$ K#include <exception>
#include <functional>


6 W7 T9 y9 p7 p; P
5 d1 g6 l' W0 U, t
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
6 K" D3 O4 r7 \- O. [! Ytypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
) R: L) K2 }# `+ ctypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
2 }7 ^  ?- l% H5 q/ X' Jtypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
0 n+ H% I4 b9 c0 Z" S+ H2 ?typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
; O: \! E$ \$ s) P' a

typedef DWORD (WINAPI* TGetBestInterface) (
) p; v+ F6 W3 v  IPAddr dwDestAddr,
+ A- b, h5 _- R  PDWORD pdwBestIfIndex
& P  k2 k9 q7 H+ H! b6 m7 [# [8 m);

' N) w- x) b, a/ U
8 E6 z$ O  R! m3 ?7 @4 etypedef DWORD (WINAPI* TGetIpAddrTable) (
% `6 {! g$ L) Z0 m0 q2 x  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
+ E2 g0 R8 q0 ]);
4 G8 Y( Q/ S$ \9 d5 f7 _

typedef DWORD (WINAPI* TGetIfEntry) (
9 T/ D: V* z# W* K1 d4 ?% ^  PMIB_IFROW pIfRow
);

- g; i% Y% [3 f! c' u
" T" n( O/ V; h) n4 p$ A, N# }CString translateUPnPResult(HRESULT hr);
" t& w& l. ~* P$ D0 s6 f$ AHRESULT UPnPMessage(HRESULT hr);
( z6 H$ A8 V% ^, N; N

class CUPnPImplWinServ: public CUPnPImpl
* h- R0 Z7 u- d- c{
2 E. W; h5 H# n9 t; |        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
9 x7 _' G1 N. n* ]; @: B( h// Construction
8 W, Z. o! w8 ^% a; q" H# x& y9 ^9 Ipublic:
4 S  P9 z  j! m6 b        virtual ~CUPnPImplWinServ();
+ |: w, R) \4 ^: F' x( A% [' C        CUPnPImplWinServ();
3 [6 _% O' H# j

0 L1 S, J; y* R$ O5 U        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
! k; l& w# @. D# X: ~9 W9 G0 U        virtual bool        IsReady();
' Y6 M3 C. M* L* w6 u        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

: f% Y7 o( C& Q- n  B  z6 Q1 F" y
( X3 O. o# z4 h6 ~        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
$ V# k! v# W7 S: t        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
# l+ u  Y6 ]: |, |

protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
& n/ r' V. N. |1 l2 B        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
. n' x/ B2 b, o) l8 ]/ p  a        void        Init();
+ F2 t' Q1 ^$ _1 S9 p6 T

        inline bool IsAsyncFindRunning()
        {
6 d5 M' \; M% v- Q                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
4 n  K5 q% {. O2 W$ X. t                {
) c9 J/ Y0 ^0 z/ r4 T( C: _                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
0 F& a, U& a( r                        m_bAsyncFindRunning = false;
                }
3 o: g* n$ S, w  b  D                MSG msg;
, i) N: k2 X, k                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
* y1 s4 r, K$ D                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
7 _; |; t5 E/ Y0 z3 Z4 S" H5 x                return m_bAsyncFindRunning;
# r( M3 I- C: I        }
. I- r! n* W  \4 _

        TRISTATE                        m_bUPnPDeviceConnected;

& x! v7 V( w. |2 n" F
// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
! P3 ^$ k) ]* D5 Y        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
% m* _3 f5 C( z. a        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


        TGetBestInterface                m_pfGetBestInterface;
" k$ z5 Y$ b) `+ V: `        TGetIpAddrTable                        m_pfGetIpAddrTable;
3 o( }4 ]; e# D, y5 v/ Q        TGetIfEntry                                m_pfGetIfEntry;
2 U4 y6 _/ q4 X8 [

  `/ F4 O7 J! B- B        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
( \. m' H8 C: E. {                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
8 ?8 u9 S- H5 k  p6 q                result_type operator()(argument_type device) const
, g/ _6 s5 S9 x! e, o( D* F8 G                {
& V( t: l5 X- t) b* v( I                        CComBSTR deviceName;
, N) [4 U0 G! ?' g0 p                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
# P! p5 @2 L" [- a. z) o8 Y9 a# |

% M! I% ]$ s' |* K$ F8 \                        if ( FAILED( hr ) )
1 n0 m/ ^; A4 h1 b7 r4 T                                return UPnPMessage( hr ), false;
6 [1 f- ^6 a, j8 P, ?

                        return wcscmp( deviceName.m_str, m_udn ) == 0;
5 h$ w# J& K7 W8 X' u  s+ r" t- [                }
                CComBSTR m_udn;
' z; u1 L" ?) R        };
$ g' H: ^2 S: x6 d       
% G9 r4 e% |" U8 C$ Z# @1 Y        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
+ S& @4 m+ `, C3 L        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
3 g- y- L5 Q! n( w7 i# R. s        void        DeleteExistingPortMappings(ServicePointer pService);
  Z  G# l! @, q" M        void        CreatePortMappings(ServicePointer pService);
8 e6 z3 j9 z; d1 a/ L1 A+ N        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
# j9 ?) a9 C3 D, h. M- G0 q
8 G( W7 o1 M7 o% L* V/ S; e
% j0 H& J1 F6 B2 ~+ A$ v" E2 m        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
1 {: O1 f4 t7 P        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
7 _0 k/ s1 D* B

$ w3 H% k, s+ R4 i// Private members
private:
$ o) w7 L$ s9 X9 A' n& ~# x9 [- ~        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
* W: M/ x1 j4 O4 w* ?2 Z        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
) Y3 O. c4 [3 J2 t" G; Q- B        DeviceFinderCallback        m_pDeviceFinderCallback;
& Z$ Q" W& @+ i7 d        ServiceCallback                        m_pServiceCallback;
- v2 M' q( ?& j# o0 ^

        LONG        m_nAsyncFindHandle;
1 ~7 \3 S8 v  N) i. S- o        bool        m_bCOM;
4 i! r, H8 U& Z7 \# a9 b  J2 c0 d        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
* M' @9 [' Q1 x# c        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
5 d5 o0 P; S9 y1 o9 B. U& U        bool        m_bInited;
7 q( p1 j  b9 j. p! L0 K        bool        m_bAsyncFindRunning;
1 l. I1 P5 Q6 I+ G3 m+ a        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
3 {) M  x2 K4 }2 f        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;

2 S, N* Z' Y# _3 ]2 z9 u7 Z; l
};
  P/ x* D7 T# j: b1 L$ r
6 i5 z) u5 t) Z
: `% Q: h+ P8 h& f: n// DeviceFinder Callback
- X# \! Q9 b) H% C8 y. x. i6 Q; Bclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
0 }( M5 o" r9 {. L; K) opublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
) O; d# ~* c0 y# [        { m_lRefCount = 0; }
, o0 n3 B' Q4 z8 j% L+ m. l0 O

: f- ?- J1 G; [' g3 h$ c   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


// implementation
( n3 p- e& g; E2 j5 }; ~6 Uprivate:
& U) E% l$ b5 b% H, M" {) C        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
: o5 y& g7 g0 \$ r1 h' ]        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
' V2 H* K) h2 H1 f- ?        HRESULT __stdcall SearchComplete(LONG nFindData);
2 s5 \0 [) X+ Y

private:
% G, ?7 O- q9 |" }0 P        CUPnPImplWinServ& m_instance;
0 j  s: B$ v( v! E2 L4 _* I        LONG m_lRefCount;
( R4 n7 M1 J$ h2 v/ J};


. Y: ~) n. j" z4 |9 {' F// Service Callback
class CServiceCallback
5 F. {1 Z9 h2 y7 i        : public IUPnPServiceCallback
2 \3 k/ d1 y, [, F{
2 _7 @- h& u! e+ d/ H; Tpublic:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
) ~/ Q6 C: B  @* ~   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
. |7 t7 k; c7 O   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


/ V2 z- E1 i  W+ G! [; z  O" }// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
! x2 C, ^1 R5 t6 ^9 r! S        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};


4 w0 v! @0 n7 j; @- z- k: k2 r2 _/////////////////////////////////////////////////


2 p, X: l# k0 J; P使用时只需要使用抽象类的接口。
1 |: b. f& S$ f9 e1 W) CCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.