UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   1 v  o1 w# E- {9 V, }- Z( g
3. 
   
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   ' X5 a  Z+ [6 J
7. 
     A' X& v- w  X. \; a2 d
8. class   MyUPnP
   4 i. w# A7 e+ a& _
9. {
   ) c5 ~, S: `' i) ], K: Q$ N$ \8 ~
10. public:
    
11. typedef   enum{
    : L0 \& |( D& @) J" Q! Y- @6 Z
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
      f0 k( [. {( C$ F! t
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
    # r, ]( z* P0 F9 W  |. t
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    7 _) O1 x9 Z5 ^2 v2 v
17. }   UPNPNAT_RETURN;
    
18. 
    6 S. B  n% d3 ]" Z
19. typedef   enum{
    0 d: l3 s8 c1 `- W
20. UNAT_TCP, //   TCP   Protocol
    ' A. o# \7 r' E; x; S# d
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    * P/ j, C. t; T' e
24. typedef   struct{
    . y4 P; c7 }$ `/ P3 I( ?' C
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
    - R; Q5 b" U' R" S9 ^5 b
29. }   UPNPNAT_MAPPING;
    
30. 
    ; V8 r" e0 L) a: Y3 O' {
31. MyUPnP();
    & J5 F) R; H, o. R- N$ X# E
32. ~MyUPnP();
    9 i1 j4 `9 {# A9 }( a) U* y
33. 
    1 i8 q( R5 y& ]
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    : r1 U/ c" n# m) y
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    1 ~, a( x3 F0 F- [( b' L1 A
41. bool IsLANIP(WORD   nIP);
    ; A: B7 u. Y' v8 P
42. 
    
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    , G0 E  m0 C+ {" }
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    ! A" A7 j; j6 ]# R! V. n) P
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    8 w4 c. g% I* b
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    8 h8 G8 Z9 C& t
53. bool Search(int   version=1);
    
54. bool GetDescription();
    3 E. g6 w; y3 c& z; Y3 F
55. CString GetProperty(const   CString&   name,   CString&   response);
    0 U+ ?8 I2 _; l7 E8 b* ]
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    . A: p; `6 P: m3 M3 C
57. 
    9 Q2 {. Y  U' P
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    , _3 ]/ x/ I+ U0 b& v. b! M9 }/ @
63. CString m_baseurl;
    1 T) f1 r8 e! B* i5 n
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    % |$ k$ L" S* C
67. int m_version;
    ! ~" l8 O7 Z+ F* n# F3 B, R
68. 
    
69. private:
    ' x4 X( M& @, U3 a) Z9 M" X
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    / M' J& d. j5 i" Z
72. CString m_slocalIP;
    9 ?- \8 p9 g8 k$ N, G6 f( j
73. CString m_slastError;
    ( G8 @: n. m; @2 [' j, y* T! K
74. WORD m_uLocalIP;
    ( e5 K0 D7 o7 v# E' v: ^1 n2 u
75. 
    
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   0 a3 u6 t' X+ D2 a, I5 Z5 T2 ~
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   
5. 
   / w: W$ A% x7 ]/ ]( k4 P3 k
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   + B$ t. P9 I1 x& C7 |+ ~  @
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    / t) D9 F+ }0 q5 \( q
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    # O' T) C8 u  @5 a
13. static   const   int UPNPPORT   =   1900;
    ( E  v7 N- \2 |8 \, U
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    
17. {
    , Z- s" x# {& k) D5 F4 a" V- o
18. CString   s;
    $ L( Y7 v8 m8 M9 w/ q/ u2 e
19. 
    
20. s.Format(_T( "%d "),   i);
    & c7 q8 o# k# p5 m
21. 
    8 P; w2 ^0 y4 f. w
22. return   s;
    7 C2 y. n& Z" ]& C! l
23. }
    , b: I! y4 e7 E+ L
24. 
    6 \6 r2 x) i/ n8 S
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    2 ~7 V! c" `  N7 D  x6 m
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    % `8 D) [  U8 I2 A' K7 o) i& E
28. }
    5 k4 c+ L1 ?6 _/ m4 _# |
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    7 ~1 I0 ?* G4 D8 f+ ~. V
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    ( l" O( v' w7 B+ _
33. }
    
34. 
    4 i) ^3 P/ K# o9 }- _) g
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    * d  @# m6 R6 Z: p
36. {
    
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    1 E  b! _0 M. ~6 l+ J; u) J3 J
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    ' G6 ^! n- P5 F, I) C$ t. n
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    + R: f$ ~* V! ~  Z
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    ; l/ v& b6 h- I
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    8 J( F# N* R" K6 p5 z
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    4 F/ A8 y- V* z+ F
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    + A( b! m, v5 n$ z1 O
60. 
    4 _0 h3 `7 V1 U  D
61. response   =   CString(CStringA(buffer,   rlen));
    ! T7 u' ]# ~: e% e8 E
62. 
    
63. return   true;
    
64. }
    
65. 
    % {8 ]1 A( i% x, L3 I- I5 E$ c. E
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    % T9 P+ {8 d$ M1 N
67. {
    0 @/ N% s( c- V, D
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    , i3 k- A0 s/ p" V( c8 e
71. int   length   =   sa.GetLength();
    ( t. h- ^5 Z: F9 N4 A5 B, m  Y! E
72. strcpy(buffer,   (const   char*)sa);
    : ?7 p9 ~* J$ c& R: ~' [; b
73. 
    " E! a8 @& K, t0 ~+ s
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    ' Q  U) O$ {7 h- n; N
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
      A, F/ d( }4 g7 s
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    0 o: @7 N  t  c8 b
81. }
    7 y1 |; f0 N' J/ n
82. 
    3 m$ u) R- a& [$ b0 m
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    $ v) S* N) g! _# \5 ?% c% C1 ?
84. {
    
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    4 Q4 W- J6 U& y5 b1 Z
90. result.Delete(0,   pos);
    
91. 
    . U  G6 C  Y+ {& R
92. pos   =   0;
    & o! C$ u2 c+ S; U) |
93. status.Tokenize(_T( "   "),   pos);
    * d, V8 d6 b, o# v. _+ ~
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    1 s6 b% @) y* y9 R
97. }
    6 O, V. @( b/ M' P* m, Y, Z
98. 
    . p6 w+ }9 W: @& Z* `& C
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    " q6 |3 [, K2 e9 w) [
100. {
     $ f; h. c" s$ R( x
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     , z  @% a5 z6 p( Y! c5 v
104. 
     $ P7 ]! d% O* s4 A9 H2 l* J
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     . ]6 x6 V  E5 h: ~2 N: X
107. 
     : D- A5 {9 f% [& ^3 o
108. int   posEnd   =   all.Find(endTag,   posStart);
     2 u2 B& B8 ?# o/ K9 j+ p
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     # E( q8 N# l0 S$ B
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     5 j. [+ t1 V$ z, @& J  ~. h& }
116. {
     * W3 C. U1 V4 k9 q+ ~
117. m_uLocalIP   =   0;
     ) H9 ~: J7 {+ a0 Y
118. isSearched   =   false;
     : @8 q! |% v% @
119. }
     2 _4 {- q! c' `% Q! u+ [( E2 y8 E
120. 
     3 p1 E% l( s# C& d2 V7 C
121. MyUPnP::~MyUPnP()
     0 Y  h8 J5 p; Z& V0 j2 [5 B
122. {
     
123. UPNPNAT_MAPPING   search;
     6 l9 O( b  O- a) ]9 w$ A% ?
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     ; I' I& e- F' m% W+ `# f2 i
125. while(pos){
     " I, o0 F& s; ~# b( g
126. search   =   m_Mappings.GetNext(pos);
     & _& w# Q: x1 ~# s3 Z+ D
127. RemoveNATPortMapping(search,   false);
     ; W8 P+ E/ u' Y/ ~! ]9 C
128. }
     6 C4 S, L: _$ [2 B
129. 
     4 x& d0 w8 K) k
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     1 _, m1 j- P6 c5 l/ ?
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     + R6 Y/ Q* s. N5 u2 o& d. R
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     , b3 A. f8 l; C+ `
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     : o2 a) J/ P6 G5 \, W6 b
147. u_long   lv   =   1;
     0 H+ v; L+ v7 _; P3 C  n" u! `
148. ioctlsocket(s,   FIONBIO,   &lv);
     ) z) q9 Y7 y; f" M& K; G
149. 
     : W) ?) A+ Y, k) S
150. int   rlen   =   0;
     * d+ u6 X$ j1 F5 E1 p
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     + X" `. o) Y: b( o
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     3 Y0 T9 c! Y; A$ F7 X
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     , v0 q1 a/ b6 g
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);
     / A2 S2 L% z8 [) ~% e
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     
161. 
     ; y& \! @! B+ z) K+ Y8 V" ]" _
162. Sleep(10);
     
163. 
     $ ]: P6 p- k' R5 Y
164. char   buffer[10240];
     9 b8 n" g. N: L, Q) D+ k" n5 q4 r
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     4 D; M* O, l1 }% l$ ?: M
167. closesocket(s);
     - f" Q, {/ P$ c
168. 
     ) C, ~8 @6 p: j' D4 [% N. c) s$ \( B
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     
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);
     
175. if   (result.Find(m_name)   > =   0)   {
     # P! V: k. }* }8 r
176. for   (int   pos   =   0;;)   {
     7 k/ \& [& G2 a: {; ]1 c8 T) {
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     6 S. F1 G; d7 p1 P9 R8 ^
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     2 @* \% w, T4 O4 W3 `
182. line.Delete(0,   9);
     8 |* b  }  S$ I, d; j. R- S4 {  e
183. m_description   =   line;
     
184. m_description.Trim();
     ) W+ \/ k& r+ R
185. return   GetDescription();
     
186. }
     
187. }
     
188. }
     5 F% h7 s: p$ h: P& T3 L0 ~
189. }
     
190. }
     0 e& c! h0 _# S4 p; n" [* j
191. closesocket(s);
     - Z, I* p  ?7 p4 l4 @0 W* A$ y; j
192. 
     
193. return   false;
     4 D9 U. l- \4 w3 J4 F, A+ l$ i
194. }
     " D1 O$ H4 A9 ]" e  Y

复制代码

天云上人

以下有关upnp的接口来自emule，
' G) g! d' ~8 X3 n# b7 }+ [! N
: Z" Y& g; A2 G. Y
///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
& G- ~4 i( v/ \$ h; d

#pragma once
3 S9 u3 c6 A0 d#include <exception>
9 |) K& o2 Z+ k
3 `/ y7 H; \# Z3 N6 ~
+ @* T; C, T5 j* Y  enum TRISTATE{
        TRIS_FALSE,
9 c- t8 \# @" b( }7 Z        TRIS_UNKNOWN,
& T6 J( h4 t* l        TRIS_TRUE
};
1 b$ ]7 ~3 e& c
0 B4 h. f! a) S& ~; A8 ]
: R0 ?) q9 v# Z8 y0 qenum UPNP_IMPLEMENTATION{
* T  _: `; \: R0 Y) j* \/ }        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};



5 x5 {. B+ t( ~! f
class CUPnPImpl
{
$ H  d; x3 ?- k  zpublic:
& M! M  B# F, Z$ W        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
: e0 `- P7 ]! S        enum {
/ m- u$ ?, {, q% d                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
        };
& X: {3 m& s9 D% j0 ^% X& R  e

8 Q' q; k" t) d$ {- }1 n        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
* A5 z; ~1 a7 X. _        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
1 q" ?' {! c5 w        virtual void        DeletePorts() = 0;
3 y6 F+ k4 F+ Y) _- b4 L5 h0 T        virtual bool        IsReady() = 0;
4 [' o# s2 S! D0 f. K        virtual int                GetImplementationID() = 0;
       
: f) h! j7 u# s3 O        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
: z- _- O& F* R, e, \0 x# i# E- K

        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
) G. {0 D- H9 w5 G        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


8 Z4 t- O) l( @// Implementation
protected:
, q4 q4 ~: e4 v1 H) C        volatile TRISTATE        m_bUPnPPortsForwarded;
/ j* n/ `! b- U        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
+ }6 R' f* [; ~/ r8 z. h        uint16                                m_nTCPWebPort;
( X. A% {% R! d3 O# K        bool                                m_bCheckAndRefresh;

2 M. R/ [$ N+ `9 w0 W) x
private:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
$ b, P) ?; c  U  ^9 `

};


// Dummy Implementation to be used when no other implementation is available
8 H$ |7 V  `5 x0 Kclass CUPnPImplNone: public CUPnPImpl
) ~8 f* N4 F7 x) ^8 _{
public:
( l- w& e) x: W0 a        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
" V+ p' ?8 R. E( q& P4 V$ Q        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
. \7 ?3 `( Z$ T9 z        virtual void        DeletePorts()                                                                                        { }
+ N) M: b' ?( L  C5 ~. q, V0 ]0 I        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
1 E5 g0 A" v* I. f  ^9 ~$ ^+ A# F};


0 W; C; v+ ~4 r- F/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类


, c8 h( y" a: y4 s#pragma once
) Z/ h* P3 K3 g#pragma warning( disable: 4355 )
/ y8 }$ K$ F: w) }( V

' {8 v- X& F' N7 J* ^9 `) o#include "UPnPImpl.h"
% f" {; S1 I9 i1 F! g2 X# U#include <upnp.h>
. t1 Z6 c7 V1 U8 r#include <iphlpapi.h>
8 g' z: I/ R5 @#include <comdef.h>
9 s, k- {- T( s  _4 C#include <winsvc.h>
" C6 z! Z; J% |) `
* M* y# U+ x& [$ j" ~5 G
#include <vector>
#include <exception>
; o; A5 Q  P! T#include <functional>

9 Y& l- v( q* R$ m
2 j0 B9 h9 c  N% F7 ?

4 o: u. _1 N6 J4 E. v' |typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
9 J! m0 q! f: `: i% n. z6 P% H* \typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
/ A6 M/ ^( |' M* j0 g* htypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
4 D& x; N/ C& b8 a. u( V7 Ytypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
* D; b+ ~6 N* D& c* s6 @typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
* M0 G" d3 y2 d& X: }% ?typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

' i, x) C5 {$ g4 O3 v! b2 \
- O$ K+ {  q# ]typedef DWORD (WINAPI* TGetBestInterface) (
$ h9 V( i* N+ z  IPAddr dwDestAddr,
* \! z# m8 |* ~3 k! Q2 |6 c4 Z  PDWORD pdwBestIfIndex
);
$ U# S) J8 ^( f$ e
) S6 V( g7 k5 M; L" n& C6 E
4 m1 V4 i! H. d3 F6 r6 V8 ttypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
5 ]8 i4 ~+ I. ^; O% p. G, Y  PULONG pdwSize,
, _1 E. c3 M0 L5 D- Y$ {; H  BOOL bOrder
  Q$ @" k, u& [0 O$ Y);
0 J& P6 J. Y5 U$ x

. f# _  ?* O4 o1 c& z$ N2 i& a. Ktypedef DWORD (WINAPI* TGetIfEntry) (
; a3 ^8 w, f5 f+ k" r  PMIB_IFROW pIfRow
1 p! [6 F3 v% l0 ]/ g);

9 k: x7 z) \0 Y2 Y" U
CString translateUPnPResult(HRESULT hr);
" T) B  K) y3 Y5 h, t! a0 LHRESULT UPnPMessage(HRESULT hr);
# n& i; T* X: R, E2 V7 G6 ?
! a: n1 i/ F' ]* d
; u' S0 n. Y. L4 D1 [class CUPnPImplWinServ: public CUPnPImpl
{
; U, T4 e4 s7 }; @" K4 D" a        friend class CDeviceFinderCallback;
3 `- V6 `- b, }* O        friend class CServiceCallback;
// Construction
' j5 R% v% l, P  _1 Gpublic:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();

! N: x% F5 ^9 f) f+ l- m
2 z- g6 P: n/ ^: |- z. r        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
, W/ q7 G2 |$ P( Y6 x: p        virtual void        StopAsyncFind();
$ O5 v- U& v% w$ ^; X        virtual void        DeletePorts();
5 _5 u2 s. b  R  ^: L# h        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
8 S' Z5 S, l5 J

0 {0 S) ], N" n, W+ Q4 i        // 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
8 ^- `, f8 S: G        virtual bool        CheckAndRefresh()                                                                                { return false; };
8 J8 p/ c# v  V' t0 E1 e
% S& ~9 \8 ~- p8 \5 A- ]
protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
8 _" R# u: I# ^. }' |8 B        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
! O% z( [# ]9 \7 B4 w, p        void        RemoveDevice(CComBSTR bsUDN);
. e% ^- b( n1 ~) E        bool        OnSearchComplete();
7 I8 M: s% j  [9 H& X  _1 g        void        Init();
% I9 e; v. y7 G3 f6 i
) ~3 Z- `! s$ }( g2 |1 \0 t# ?
        inline bool IsAsyncFindRunning()
4 ?! V5 M( o* z4 H2 a        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
" Q, t" r( J" V# n8 F7 C2 k                {
. L3 e; F8 A2 K4 A6 ]; E                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
" j2 O9 H$ j1 b# U$ C- V                }
                MSG msg;
& S; d1 \/ J& M4 M4 H! m. H                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
" F' M$ D( L6 {8 C1 J9 q                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
        }
" I3 z7 c- B8 F1 i

        TRISTATE                        m_bUPnPDeviceConnected;
, Q1 `$ n0 }& M) D
1 n& e4 f3 i* w8 W
// Implementation
        // API functions
1 @: G8 S+ W; }; O) J        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
2 }8 @/ ]2 p9 o- M/ f" Y# R        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
! p" {% _3 R! o0 g/ q        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

3 s6 D5 s5 t: m7 R; r, n
( o# K- r7 m6 u( b        TGetBestInterface                m_pfGetBestInterface;
( Y( ]  ?1 F3 J/ ~$ }# w# a, g; F        TGetIpAddrTable                        m_pfGetIpAddrTable;
+ n$ r7 {0 v! L/ P; U( y5 I& B& |; R        TGetIfEntry                                m_pfGetIfEntry;
0 q  j% ?5 I, U0 W5 p

        static FinderPointer CreateFinderInstance();
% @+ O9 T$ ~4 O- q        struct FindDevice : std::unary_function< DevicePointer, bool >
" q; v( B/ f& h0 f/ V/ ]3 }7 O        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
9 b: P3 {' w8 r) @                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
' t5 d& y! ^0 E+ g" X
3 N9 m7 y( B9 _: M- P, z, D
                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
3 D6 \% B8 l$ o

                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
        };
+ i! z9 |" S$ l# k       
5 Q$ _2 |% r, G2 a* h; C$ ]        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
6 X: c" J" O3 j. X/ O. _" C* O, D        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
$ ~% w- ?0 _6 R' Y        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
" S1 F, H6 H8 I8 R, F' O        void        StopUPnPService();

; |3 Q' w4 m8 h2 t/ X
5 ?9 b8 i" H3 j! f        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
+ y% i3 g9 w+ ]9 |+ s5 G) ^        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
1 B. o+ M9 A+ ]( O  C        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
4 d. J( h6 f/ h3 F        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);


: ~2 i! m* g: X- U/ k; V( Y// Private members
4 ~& D; w  t; Hprivate:
8 L6 h  i6 i* G/ R9 {        DWORD        m_tLastEvent;        // When the last event was received?
/ n# P5 L1 s, z8 a  d. ?4 Y% a        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
" u0 E5 Z7 ?0 n        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
8 p6 m. j9 c) e' G* Z
! e$ e3 \  h; t% R9 U. I
$ r. \. z5 H5 l# W" E        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
  t6 k9 `0 w$ t2 t8 P, K        bool        m_bPortIsFree;
        CString m_sLocalIP;
9 [5 T1 [  ~  s, f        CString m_sExternalIP;
* o% j. b$ |0 p  v: @        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
: Z' P4 v6 F- t0 T' L        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
+ Z6 b' j# D' a# O        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
' O( {) Z: G6 j: H        bool        m_bServiceStartedByEmule;
4 g7 f  I$ U9 Q% a* |! M) r7 d        bool        m_bDisableWANIPSetup;
* |2 s. k$ |: K( W' w# {        bool        m_bDisableWANPPPSetup;


( f5 s! k: z$ b2 I};
; y% m* J! z0 R4 l' t" ?4 x- D
2 f- S6 q; V9 \, H
9 J7 m- I8 ^2 ]// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
2 T" r, b- }7 ^$ x* n" @{
5 h0 Z9 u" f" y& }# b7 Epublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
! i2 `& i* m+ a+ H  W& b                : m_instance( instance )
& A0 k' T" h$ ?. S# C        { m_lRefCount = 0; }
0 K) K7 f6 K# b% R& Q6 {; Z

8 W  @  G" W1 h& v9 X   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
5 F7 F7 Y! {1 }/ D/ m. L9 _   STDMETHODIMP_(ULONG) Release();


// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
; C  u" Y$ x4 H
# `2 g& A" p$ j3 M# E7 Q/ U! Z
private:
# H8 }" k' C# o        CUPnPImplWinServ& m_instance;
  b5 R2 _$ m  c! R+ M        LONG m_lRefCount;
7 q7 l2 j! |8 Q9 S) R  d* r% m+ S};
, f' W6 L2 k# z1 Y% D- V
; T; K4 \  N5 C2 b' i
// Service Callback
0 q8 [2 o8 x) j0 H; {4 ^; fclass CServiceCallback
: f% \. Y. X3 @3 h        : public IUPnPServiceCallback
{
, T: s) L+ D  h3 Cpublic:
4 G% k: J6 x2 k- z0 \        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
' j  x+ a: I) {) Q        { m_lRefCount = 0; }
& ?: P8 l* G3 k& J7 k  q" E   
1 F$ I( O7 r  M+ W! k$ w! D   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
! q; z4 `% f9 v. @6 O   STDMETHODIMP_(ULONG) Release();
$ b6 I" V  ]3 Q2 H
+ p  r9 q. V$ m
0 q2 B/ X* i& }$ q! ]// implementation
# n' `- z2 L4 v3 h# C. eprivate:
8 y- ?4 w9 }2 K2 e) e' e        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

% d5 O1 W1 ]' n6 I
2 j+ s2 Z# j7 g- ]( }1 Kprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};

8 |  [  B- d2 e( ~% K7 L7 N
/////////////////////////////////////////////////
" p* O7 d& r9 P7 }% G+ p# J+ Y

) i. l! U! o7 P- B: R- ^使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
# u1 h% W1 N4 F. D/ M  A% z: |CUPnPImpl::DeletePorts删除端口映射.