UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   6 B: _4 Z2 M; M* Y: z" U2 v2 _) Z
4. #pragma   once
   ! w: g1 F. e( h# T
5. 
   5 {- u: L1 x% {' t
6. typedef   unsigned   long   ulong;
   
7. 
   0 o5 y3 U" d# S
8. class   MyUPnP
   
9. {
   
10. public:
    7 ~3 v9 f4 @2 b1 A& D6 `
11. typedef   enum{
    : X- ^$ b1 s% g( [6 V& _4 ]
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
    
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    * M+ K, _7 I# S" Q
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    $ Y$ i* X- ?) o$ h
18. 
    9 I+ Z) q0 C5 q; J4 k
19. typedef   enum{
    ' V. ^# L# \! N2 w
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    , C6 Y9 d* P0 ~% j
24. typedef   struct{
    0 V) @2 D$ B9 K& X
25. WORD   internalPort; //   Port   mapping   internal   port
    + l" Q1 c% d) t
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    9 }. i7 y' y2 K0 g; |# X% U! \- k
28. CString   description; //   Port   mapping   description
    , N3 L0 m3 X. J: F
29. }   UPNPNAT_MAPPING;
    9 X6 z5 \/ H5 Q
30. 
    4 g: f& n+ D& }* q7 F/ V- B( j
31. MyUPnP();
    : q5 c/ `/ W! O2 ^& m+ S  P
32. ~MyUPnP();
    2 L0 i6 g, K  S8 f( X% v  T3 u
33. 
    ' V  |( V! M4 J( N3 v& m
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    4 A+ Z9 r1 o8 }3 |: t% c8 A$ q
36. void   clearNATPortMapping();
    - q8 s9 i4 p& _6 a4 I6 G  f
37. 
    9 m9 f) @' d9 g* F
38. CString GetLastError();
    # j/ y& W4 ^& b8 X7 y) a, E' r
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    % D7 t2 f; j/ h4 L7 Y" D9 B7 `5 R
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    7 J+ ?+ C1 ?. P' G, p! a
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    & C: w) _$ L- v! V
50. 
    / A$ |( Q$ H- n
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    0 v: m6 f) x7 j
52. 
    
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    % n( C/ l- c3 o, e; @: X
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    ) P! O/ q8 b( W' h" o, l
59. bool InternalSearch(int   version);
    8 Q! B! q' R7 J$ j4 r
60. CString m_devicename;
    
61. CString m_name;
    9 q( F. j2 U5 o! o4 W" t% r2 ^
62. CString m_description;
    7 g& ^2 G6 ~/ }$ {; w
63. CString m_baseurl;
    / ]- {) ]7 ~' h" H2 |: F2 i1 U% x
64. CString m_controlurl;
    
65. CString m_friendlyname;
    + F, @8 O& l" N* |8 Z# {5 S
66. CString m_modelname;
    8 F8 s8 u& ]! {; E8 b8 }' e7 ]
67. int m_version;
    9 y( s: D9 [% n2 Q1 P# P/ H7 y& V: x
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    0 \( I; F& q" e4 a! ]5 R
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    % p7 l+ a$ p) Z  I4 Y0 |6 n( j
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    
77. };
    " j" r9 [3 w# L6 V9 ?9 v/ y8 x, j
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   - @: H& w' M$ Z
2. #include   "stdafx.h "
     l9 N+ v2 B0 V4 f- d( G) O
3. 
   ) |* Q- Z0 }/ h' z: d/ i; m
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   1 D7 ~- `3 b' @7 y
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   2 [' ^4 c+ P6 F9 v0 |
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   ; ~# n, S- Y& B" J$ Y6 J
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    - u' C  q) J* L$ K0 L% j
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    & d5 e0 X. P0 y) }7 w1 g
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    # Y( N/ Y' N7 s, {- }2 p
17. {
    ! `4 t# u% c; b' e2 A5 S
18. CString   s;
    , ?2 a/ f3 J! p' C& G8 f. @9 V
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    ; u7 H) [) K' o+ }9 e" Y
22. return   s;
    
23. }
    4 k, q, @/ n% i& O/ y4 l3 s
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    * U6 Y8 i- S% ]) e+ ^7 ?8 C8 _
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    $ _( `0 @7 f+ A. J- C  b+ V0 D7 L
30. const   CString   GetArgString(const   CString&   name,   int   value)
    5 C+ E: B' H; z( u" s. H
31. {
    , r, O, a' C2 [; |! O- W) T$ M6 }
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    # [( y# }% A* R1 M0 z
34. 
    / r' L( [! i" z( w
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    
38. 
    5 L" b  V$ O5 n1 H
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    ) H2 v/ Y- S' g2 j2 K; O& a
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    " {$ E) m. a2 I: w0 y6 A' W
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    2 ]: }1 a9 ]8 [$ X  H0 y! f# D5 x
46. sockaddr.sin_family   =   AF_INET;
    2 k2 o/ o: M7 F3 [! M/ R, i
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    ; E# R( `& M( a7 i3 ^" D
50. u_long   lv   =   1;
    . O- F- c7 V/ j8 f9 |% R, |4 e
51. ioctlsocket(s,   FIONBIO,   &lv);
    * p5 h4 i8 `& F) u
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    ; S8 j8 J1 s: y4 p4 s
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    0 ^+ v6 D0 v4 a3 V
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    ; g# _' c  m  U) H8 l4 }
60. 
    6 k% S( W8 n( E/ u: t
61. response   =   CString(CStringA(buffer,   rlen));
    ; R& t* ?) _  m; n8 c
62. 
    ! p" a5 h$ Q7 x0 t' ^7 V
63. return   true;
    
64. }
    
65. 
    % k1 c2 [6 C; ~  s) Z3 }
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    ; D4 Y+ H" U( Y* x! L+ q0 J
67. {
    8 ~; d3 D9 B  J1 \+ _* ~( V
68. char   buffer[10240];
    $ H5 X5 B" o' F/ b& V
69. 
    6 b0 F" _/ _, n. f
70. const   CStringA   sa(request);
    9 m2 E0 K: h, J) t' A
71. int   length   =   sa.GetLength();
    : N5 B( a! `0 N3 `' i! C) f2 q% m
72. strcpy(buffer,   (const   char*)sa);
    ) i3 s: z, o* B- N' n% o( @7 ^
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    : N( t7 X, z( B
76. sockaddr.sin_family   =   AF_INET;
    6 E: E: p/ a4 F) M8 D/ C6 Y
77. sockaddr.sin_port   =   htons(port);
    9 |6 m- Q/ V) d8 N
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    # x: Q/ B% w0 @
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;
    6 p# X7 E  o) s, ]
90. result.Delete(0,   pos);
    . [+ N# w* S; K+ ]5 U0 D
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    4 e+ S* ]: i( a5 u" X' Y! V
96. return   true;
    " |# N  _  @$ L- R
97. }
    3 e0 A( `# B  j1 i3 [
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     - k/ u) \/ [! `* f. ^% m
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     ! Z7 n: x7 @1 I% g+ [5 z
103. CString   property;
     . W5 g& ?' Q! u* m; p9 m% ?
104. 
     
105. int   posStart   =   all.Find(startTag);
     5 `; `: F  V, a. f: f; X. r7 E4 l! M
106. if   (posStart <0)   return   CString();
     1 h: u6 o0 ^0 B! Y" h# E' f$ @
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     8 K, V6 j) b/ v) U- i2 _' R
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     3 S3 j0 i1 t6 Q4 D) F! X: C. p
113. 
     6 G3 H3 Z# g" Q+ C5 [- b( ]1 F
114. MyUPnP::MyUPnP()
     7 z1 N3 k# d& B+ H+ [4 ~1 }
115. :   m_version(1)
     
116. {
     9 w# h+ `2 Q* ]) Z$ R+ N. |% O% Z
117. m_uLocalIP   =   0;
     , y: _- A- O" }4 U1 x
118. isSearched   =   false;
     
119. }
     ) B: j: ?: k5 K7 x8 f
120. 
     & b4 X$ p+ S# L; p0 }
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     $ I5 {+ Z9 p9 s1 X/ B- N. Y
128. }
     - X6 u4 h1 m# C% R
129. 
     ( \( d) ]2 J0 V; V: @( w6 |7 q, F# E
130. m_Mappings.RemoveAll();
     6 V) b* ^2 S, F3 d- O; x9 }
131. }
     
132. 
     # P# _6 s+ h/ |1 E+ m
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     2 y+ s7 J  I* ?5 k8 M
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     ! _! K' u3 r3 u
141. {UPNPPORTMAP1,   _T( "service ")},
     1 v: w! |0 {  A& b# }
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     , H- l' {  w0 H* i5 F2 ^
144. };
     ) F# ^  M4 U) i" B
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     $ E( l, M$ K) _7 f8 |4 e1 {1 r  V
147. u_long   lv   =   1;
     0 m3 \' t# k- w; R# u
148. ioctlsocket(s,   FIONBIO,   &lv);
     ( M: @8 p2 l8 p; w4 [
149. 
     8 F5 s& F, a+ W) t  Z
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     0 h, M4 ~  V: Z) c1 V( y
152. if   (!(i%100))   {
     ! k. R% c) b7 H8 |
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     0 B  O6 P3 r' x. r, h
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     . f$ |9 `! `5 _" ^1 ^
155. CString   request;
     
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 "),
     ! s7 n2 ]: h2 d. Z$ ~- v! O9 v
157. 6,   m_name);
       o! Q2 n1 r) L, o; N" O
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     # G7 i1 r" ]; I. J% D- Q
160. }
     
161. 
     
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     , P5 d% q5 M* E/ W9 l- o' I1 O* A% b
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     : B3 F) s7 b2 c0 h. W3 G
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
       B, V' v$ E% U& w5 Q2 \% U
171. if   (!parseHTTPResponse(response,   result))   return   false;
     & n% h3 A$ \' R, q% u/ m0 {  b
172. 
     ' \% G! B5 d9 M" u
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)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     + F2 ^6 u) V! z- |" Y
178. if   (line.IsEmpty())   return   false;
     $ e5 H1 j+ v$ ^, o. \
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     7 D" W# E  J# k/ B1 P* ^& L
183. m_description   =   line;
     
184. m_description.Trim();
     ' i: q' }" z4 X$ G
185. return   GetDescription();
     
186. }
     * |& A! `+ H6 q# R" R
187. }
     + ]* j0 `1 l* ]- ~6 R
188. }
     & K6 A  l( \6 L( U; b! f
189. }
     
190. }
     
191. closesocket(s);
     
192. 
     
193. return   false;
     4 N; i1 u/ J9 c: e' w: ]$ ^* w
194. }
     

复制代码

天云上人

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

$ q+ d7 A$ [* [8 F- v0 a
* }4 w( v  P8 D" h///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

5 u3 P) C7 c# u6 ^% m9 k( h9 E: i! i
#pragma once
#include <exception>
0 U0 l) i$ T: p7 z8 V4 E- y1 E$ t
+ S  {! t8 K1 t2 D
% z1 D# q1 r- n- m5 I2 F  U# K  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
/ o! C, K; A; f( g. [        TRIS_TRUE
};

4 c1 v5 {+ \1 A! v2 j6 r; Q; |% [
* Z7 I* s4 I- |  Renum UPNP_IMPLEMENTATION{
$ V! j6 h' `5 v" ^8 V- T        UPNP_IMPL_WINDOWSERVICE = 0,
$ u- N- X1 r) \2 O3 q4 g        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};
# t  L% [; @# T+ `: |$ m
! V3 W2 d1 F+ X8 H

3 s. y- y, v. e5 l9 r6 Z
class CUPnPImpl
% O' g; }: Y& t* F{
% O- P- n  c+ B  W5 W# D% P9 bpublic:
% T+ T- K5 O  F        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
& g# [) R) ?+ J! K                UPNP_FAILED,
                UPNP_TIMEOUT
        };

7 l. u$ }. E2 W; g
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
3 G, o& }) y. s! p! C        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
+ X) ~0 x4 F5 W# P

7 S7 \" D! _! H; O0 P        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


// Implementation
2 X; q* X0 @0 w% M* \protected:
+ b0 G7 ^, B6 b# s0 w        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
7 G2 S/ C% L1 _* z        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
% j* `+ L/ x% H  Y        bool                                m_bCheckAndRefresh;

' P" h9 R' c& Z$ v# B' R
private:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
+ v0 i- s4 ~  C' Q) y* I' c

};

' i+ T7 ^5 z5 [2 v
// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
3 u# O/ b. e; ~0 s- @" o9 S  r0 C+ jpublic:
7 z! ^; K# G, r/ y& g1 a$ f        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
; q* d' }/ p; ?: p" E, i        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
8 \* x4 H/ p, J+ T! x        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
) p7 e% {  O$ i, |

6 E. T0 x3 {: I7 p$ p. Q6 Z. A/////////////////////////////////////
' m4 w. J' u3 S& ]//下面是使用windows操作系统自带的UPNP功能的子类

& M0 x: ^0 e7 r/ f" ?! T% |9 \5 l
* U& J  I/ Q9 ?3 o$ e! b#pragma once
#pragma warning( disable: 4355 )
- x! T. L' V; E. V
2 V( m4 u. S) P- R+ R7 I8 ~
! Y* H- H; f6 N! p#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
9 \9 `' Q/ P0 y/ h9 D( Y7 R#include <comdef.h>
#include <winsvc.h>

1 N0 h" G; `. x+ [1 k) d
#include <vector>
#include <exception>
! M  \1 {1 L! f+ Q& t/ t#include <functional>
+ I1 G3 [+ X. S7 v/ E: Q
) l" o4 [9 T1 ]/ R9 v( F# U5 l7 g


6 ]/ m. c* m: U( c  h1 [9 ztypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
# @8 J0 c) z" N8 stypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _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;

; j' |" D2 _  P0 s4 K! h; @
typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
# c) H, I$ Y( X$ Y% s' \" Z  PDWORD pdwBestIfIndex
);
/ u5 t, {) j2 k$ h4 S

typedef DWORD (WINAPI* TGetIpAddrTable) (
& p) J8 M' B4 c. T' s# a- v  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
9 e/ O% a, \- u( w0 ~  J& j2 A  BOOL bOrder
: B" ~1 t7 j1 z: p# q);
) ?( Y! b0 C+ F' @4 p1 R) ~! i
# W$ k7 b" D5 C, i
" x9 m; @4 s# J' L9 Z2 ^typedef DWORD (WINAPI* TGetIfEntry) (
$ W% w' M9 \- B9 M; x  PMIB_IFROW pIfRow
);
" h- p8 M% D( j6 K) B

CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
  T  n/ V' P/ j: `5 C% ]

( M. l# J0 d/ g3 C. L7 {1 e/ U" aclass CUPnPImplWinServ: public CUPnPImpl
, |5 X( ~& p- R: z' ~* E# M8 ^! U{
        friend class CDeviceFinderCallback;
- `; }  n& I4 {9 `8 d6 {        friend class CServiceCallback;
5 w5 Q- _$ i0 c3 s# Z// Construction
public:
: X0 O4 ?3 r$ h9 r5 Z% X        virtual ~CUPnPImplWinServ();
$ U: `' ^  l1 h* b2 N% n& t        CUPnPImplWinServ();
/ s+ R# i( u, O9 a! j* B' h

* l7 J  h. ?: }+ L2 [5 N  z        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
, H3 T/ J" y& c( g  v6 _+ N        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
' W* l/ h, m7 W8 [- o1 Y& J

        // 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; };

! i/ ^& r, m1 r9 `( [
protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
. }4 l$ i7 P3 t) O  K/ _        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
) v% Y# ?3 o  O+ D* t        void        RemoveDevice(CComBSTR bsUDN);
" c) P7 o, X3 l8 o; T        bool        OnSearchComplete();
1 w* I0 h2 @# i        void        Init();
* j% `. h( u- O  n, O

0 |  A. @$ @; ^% S8 N3 Z! R: ~        inline bool IsAsyncFindRunning()
        {
! `2 f$ J) _7 o0 C% [6 H                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
- |, z5 J' e( n3 S9 m                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
3 N5 m% B# `/ j                        m_bAsyncFindRunning = false;
                }
                MSG msg;
" d. m8 Z+ R9 }' v2 q9 e3 q                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
( U' O3 J( Z2 k3 f. u' I                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
! z' _) {4 V; q                return m_bAsyncFindRunning;
        }
% U5 i% ~. l) L8 P
1 g* s, K/ I# ^4 e8 w6 o7 z3 L
        TRISTATE                        m_bUPnPDeviceConnected;


// Implementation
0 [) Z! A7 h' }2 ^        // API functions
        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);
, B  C. n+ a6 N        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

4 e* \' ^1 h$ A1 C
! v( a* u9 R, z        TGetBestInterface                m_pfGetBestInterface;
, V2 q: j2 b' w7 G+ s9 i        TGetIpAddrTable                        m_pfGetIpAddrTable;
) d0 J: ?0 w" S0 g, w4 e6 ~9 Y% |7 Y        TGetIfEntry                                m_pfGetIfEntry;
2 V: `' U; C$ M( E
0 a' o4 G6 i1 k6 v! T
        static FinderPointer CreateFinderInstance();
+ Z. x  V0 x& ]9 o4 p5 R( x        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
- B) N4 r$ G+ {/ T6 u% Z* o                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
8 t5 g3 {* J4 j5 H  Q; ?
! \4 r+ b. s1 G0 [  a
                        if ( FAILED( hr ) )
  N% |2 D. a) m8 Z+ Q                                return UPnPMessage( hr ), false;
5 }! m; }8 x6 |" v- [( |  Z/ S) A

  ^, J7 N$ o& t% B3 F                        return wcscmp( deviceName.m_str, m_udn ) == 0;
: g5 F( ~* C4 {4 b. Q: I$ C                }
0 f7 i  P; Z' @( i$ D8 E                CComBSTR m_udn;
3 z$ x2 y; G% u( }, Z4 y        };
2 g! e7 s7 @, E( F0 G9 h+ A2 S       
2 ]8 l5 Q* l5 |' [3 v        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
' S- n0 `6 S  J4 g8 \5 g( C1 y6 v        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
- x3 ^# \4 M* l        void        StopUPnPService();
  l: ^, u, T+ v

' Y5 A; b3 V" E  z        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
0 y  u) i+ B8 a3 y        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
$ w, E, Y- M. y        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
* w3 W0 m( S( t2 [6 |        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
) [% T+ p8 J' @0 b$ K' A7 g        CString        GetLocalRoutableIP(ServicePointer pService);

6 ?: D' y. {" P
// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
$ b* u, A# |* ?6 u: j2 ]        std::vector< ServicePointer > m_pServices;
0 t7 A- u$ x+ I2 p) ?        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;


        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
& a& _0 `0 W- H' [  ]+ d2 F/ X        CString m_sExternalIP;
8 p' Z" Z% \/ q0 Z5 U2 a        bool        m_bADSL;                // Is the device ADSL?
+ Y0 r# D0 n; `! ?. T% r+ f        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
9 z2 G) v4 Q" O7 J  k; @  p        bool        m_bInited;
1 a# P- w: C  Z8 G- b8 b/ F) x        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
% Y* d% G/ h6 d6 d& q        HMODULE        m_hIPHLPAPI_DLL;
6 y# \3 |3 R5 d8 \8 V0 J        bool        m_bSecondTry;
8 X( }% G% Q' A5 b* U: l        bool        m_bServiceStartedByEmule;
! {5 S2 ]# _- T, z/ q0 U( o/ [        bool        m_bDisableWANIPSetup;
, g8 y% y# E, l. P        bool        m_bDisableWANPPPSetup;

4 s  i8 U3 n: M2 v
7 W. R( X) g/ e+ x};
' J- y1 U0 b& l3 z$ f6 k, |4 J
' b- d: B  n( I3 D7 J. j$ n6 j
" C% [3 M) l# M- F// DeviceFinder Callback
& d/ p% I/ V9 `$ J" iclass CDeviceFinderCallback
4 h- y) j1 v5 Q2 z        : public IUPnPDeviceFinderCallback
0 J& i0 G. E# q/ x{
; ~8 S0 z& ^9 N0 ?1 upublic:
0 b1 G( d7 `$ V, K3 i- B        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
! G" ^) O* `# \, k2 R
& F4 ]* k/ D! A) N7 H
0 P! S# m/ ^' T8 J' E& ?/ D# ^8 d   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
7 t$ u( i' S% N) x* i3 H   STDMETHODIMP_(ULONG) Release();
$ ]3 E# A1 i8 ~& L- o- v( N: b$ v
( _' Q/ R1 w% Q% [: e3 }
4 K0 }+ P, S  K// implementation
6 f& K7 \' q, F5 ^4 U& a* U% tprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
2 N+ s' W0 F/ T, a6 j" m- I( h        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
( v& m2 w& I: J* `        HRESULT __stdcall SearchComplete(LONG nFindData);

2 G5 _/ k: K/ K+ Q; O
8 f* J$ q4 g1 B' n0 @4 Uprivate:
        CUPnPImplWinServ& m_instance;
6 X& E3 r% _1 D+ o5 u        LONG m_lRefCount;
};


- A5 n! @& A5 J1 O' ^& k4 }// Service Callback
, a, F6 @0 n5 qclass CServiceCallback
        : public IUPnPServiceCallback
{
9 ^3 Q2 d' L2 s; c1 vpublic:
3 w" m; T1 t3 k: S        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
1 d5 @) s4 ~& I$ h. L5 l. _        { m_lRefCount = 0; }
/ F; T8 |% L7 G   
2 \! w: O  Z  ~+ E" o; `5 I   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
7 X$ h2 b5 k4 h5 H- X( F; m   STDMETHODIMP_(ULONG) AddRef();
" |, p- V2 x: E7 x( J* q9 |1 k1 x7 }   STDMETHODIMP_(ULONG) Release();


% ?# u: `; s$ m" s// implementation
+ g1 E# _2 U. R& Hprivate:
2 |2 k* s) n# e* @( i. ^' i" b        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
7 p( B# h9 P  V: [7 J* K

private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
. L: |7 B: f# K7 D6 M
2 U4 I+ }7 [' ^, D! j5 B
  n$ }% T% \! V, o/////////////////////////////////////////////////
( i- r! _6 E, {. f; O& R: g5 I

) r# P2 N! M+ Y! }; L使用时只需要使用抽象类的接口。
  v6 C! b5 B* \5 C$ N& CCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
  \* _) p5 ~$ O# [" E3 L2 _CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
1 ^- F1 _& T' b7 Q2 ?) n; v* yCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.