UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
     w& i* U7 ^$ y; ^
5. 
   
6. typedef   unsigned   long   ulong;
     c7 @7 Y, r  v3 w) M. }! s9 y
7. 
   2 A& B# |. z* i$ ]( W
8. class   MyUPnP
     V2 B" C4 I( s& R3 U6 Z- Y" Z" [
9. {
   
10. public:
    
11. typedef   enum{
    9 _3 S6 q- g6 B
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    * I6 w" z* n6 j& N4 t
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    ) q, Z0 b! q9 `; {1 N# d' t
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    ) G) r4 w+ g8 M, B" a1 W
19. typedef   enum{
    3 M8 e' s$ f' L
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    3 b& h& m6 I- ~' }* R
24. typedef   struct{
    ; n5 N; |" j! T! u+ h2 G
25. WORD   internalPort; //   Port   mapping   internal   port
    - [* ~; P" |( M5 x7 N, K1 c& Y
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    " z8 D4 T2 T# d- h
28. CString   description; //   Port   mapping   description
    5 ?2 ~2 G' q- S( ^/ H7 A# v
29. }   UPNPNAT_MAPPING;
    . M$ h: e$ [! v
30. 
    
31. MyUPnP();
    9 d$ Z5 s% U2 |) l8 g0 ^. B
32. ~MyUPnP();
    
33. 
      O% n6 B3 A! y! r
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    / ?# ?7 L! E5 R
36. void   clearNATPortMapping();
    - [: X% W7 n$ \" j% Z/ w, w
37. 
    - x, Y) _( s* m9 d6 E
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    $ u0 X" B; Z; H- E5 ~6 C
40. WORD GetLocalIP();
    $ T0 ~6 |) q. l0 e" W& U
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    ! S: T0 P9 n. T. ?( Q4 Q
44. void InitLocalIP();
      e* @$ t' z( ~
45. void SetLastError(CString   error);
    
46. 
    7 w! j* ^) |: R. W2 ]# `8 y
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    4 E" ~6 ?$ r# F! ?
48.       const   CString&   descri,   const   CString&   type);
    9 ^; ]  z2 l; J) [' x
49. bool   deletePortmap(int   eport,   const   CString&   type);
    7 c/ e1 F5 m, s
50. 
    9 X4 m& I1 {' G7 Z) s# `
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    5 z+ S, K4 Y, y5 M$ e7 f4 B$ b% ~
53. bool Search(int   version=1);
    
54. bool GetDescription();
    4 q* c0 Z1 Z3 O, C( ]. i& ?
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    0 p6 C; I  {: X- h3 V
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    5 O1 C) b1 P2 [  |3 B
59. bool InternalSearch(int   version);
    $ Z" K$ k1 p3 ]7 |. ^/ U
60. CString m_devicename;
    
61. CString m_name;
    9 y: j9 G, G. D" [! H& g0 Y8 R3 m( h) h
62. CString m_description;
    * N5 f- g7 n# H
63. CString m_baseurl;
    - |: n- y- n9 P. Q. _" |/ b" f+ Y
64. CString m_controlurl;
    ! V, m- x7 E, o% Q2 Z2 `2 l
65. CString m_friendlyname;
    8 y  Z- t; V/ ]9 t. o4 W
66. CString m_modelname;
    $ J8 }  X4 Z- F: D8 u2 l. f
67. int m_version;
    
68. 
    # h% I- b' z( o6 u% q
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    ) m5 _, I+ K# w6 @# m0 k6 C; {. B3 o
71. 
    
72. CString m_slocalIP;
    5 K2 ^" N5 f* g0 q* L. m8 m) N
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    8 M2 ~/ d5 _9 C3 V% J
75. 
    
76. bool isSearched;
    
77. };
    % B* N7 q% E0 x5 ]4 m3 j2 X2 P( K
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   $ O; f* b6 b: _* s
4. #include   "upnp.h "
   - h; K5 ^6 F4 O1 |5 g0 K
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    9 {; b: z/ W& I" a' \2 W: Y
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    ; \/ Z0 F: g( N1 D
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    ; N" L& d& ?$ x6 l
15. 
    
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    
19. 
    ) q$ o% o- N) ?0 \0 U" X
20. s.Format(_T( "%d "),   i);
    
21. 
    ; ^! J" H$ ~) O
22. return   s;
    
23. }
    0 U3 d% _' T' Y2 |1 i. F
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    & q0 c5 j" W* w& h( x
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    % ?6 Y! q6 L4 I2 A- G
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    $ \" f9 g1 O3 y5 W3 r/ w
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    6 q8 ]- y# i  i- T# l$ I. j7 Q7 h
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    ) v4 F8 |( j2 y
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    5 i- B; ^- d3 c" S" o6 a
42. 
    ; ?- s+ |+ S2 |" T  L# H' p
43. uint32   ip   =   inet_addr(CStringA(addr));
    " w2 Y. o$ Z8 A5 H' Y3 t4 g" h6 n4 F! s
44. struct   sockaddr_in   sockaddr;
    ! b! j. y. s: f
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    9 @% H/ t2 J. q$ H1 Z  v
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);
    
50. u_long   lv   =   1;
    ! n2 }: [% t9 h: H
51. ioctlsocket(s,   FIONBIO,   &lv);
    . x; H: C  ?  ~( \- b0 S. ^
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    : m- V( x( ^! [! g3 \/ ^5 y; i
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    & [7 j2 A/ g# d/ G( {0 A: [
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    : g& ~  ]1 \! Z# @2 B; t
59. if   (!rlen)   return   false;
    2 A/ a- V7 Y* |4 O3 |6 c. @% Q; I
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    6 r; A& d5 l# V/ A
62. 
    4 p4 S/ R7 @# Y/ ?( g
63. return   true;
    ' `9 W9 S7 {7 s+ y5 c1 Q5 J9 _  z$ t
64. }
    
65. 
    ) O" A3 t: W- E. g# x2 `
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    $ u$ G% T4 H! T& h) q8 o0 j
67. {
    
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    ( U$ o, W8 T2 ]" n
71. int   length   =   sa.GetLength();
    * I, P. j1 {* k0 \' b8 I( F5 t! @
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    & k) y) M: h0 @) p; P" t
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    ( x0 y: w5 p9 j# i1 V
76. sockaddr.sin_family   =   AF_INET;
    7 O5 m2 A6 p4 ^( A! M+ a$ J, N
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
      w8 M6 o7 c( k: q4 Z2 ]
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    8 Z2 i6 Q# G# d5 s3 ?
82. 
    , g0 J2 z, \3 D. w7 D
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    - W6 i- X6 W6 |! n, k& L4 k
84. {
    4 O3 m3 G( }9 }4 |" S
85. int   pos   =   0;
    % N: f/ B' _; O- F' N/ K
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    : q/ W0 ~; ]( @2 Q$ r0 A, [
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    & R4 s1 @+ Y# _
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    % {4 d! m) E; C) F* {& e
97. }
    8 m0 Q) n2 B( ~0 ^% [3 Z" v
98. 
    + z5 _: f, s) C
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     ! O$ F, u9 h/ a! g& i4 i
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     2 K' @: C9 e" O/ ~7 c7 l
103. CString   property;
     
104. 
     * w: x! z6 w) {( H# i9 O/ O/ z0 m5 N4 F
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     5 Q9 U$ m- g9 ?0 q
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     4 o$ O( D; @! \& ]
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     $ f4 E( |7 n% C
112. }
     
113. 
     ' j7 u! j+ X, v% G; h
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     " i! k% S* {7 Z) w8 i
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     ) n, j0 o# w7 L" q
120. 
     
121. MyUPnP::~MyUPnP()
     : L# K4 X0 v. g! j3 ~0 A& i
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     + c) {3 y) j7 g! u; m3 k! k+ n! h
128. }
     ' j  N( f/ I- F( z* v  ?
129. 
     * h8 D' k4 P& t
130. m_Mappings.RemoveAll();
     
131. }
     " |" P, Y6 i4 D/ d( F( @
132. 
     ) O. |0 w8 ~* J0 e( w
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     1 S7 @  R3 S' B3 S) e* o
138. 
     3 G. U6 v2 [5 s- C3 Q" n4 e! k
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     ' A3 {1 d: j8 U: L% i1 \
141. {UPNPPORTMAP1,   _T( "service ")},
     9 o/ r( b3 i% s; Z# A2 p
142. {UPNPPORTMAP0,   _T( "service ")},
     " o8 |2 T& y8 g5 B' N; c3 ~
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     % K! P4 ^+ [5 B1 t2 O- }! S4 {4 n
144. };
     9 D, @  C0 @, U1 A5 M3 L2 V
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     1 P* [1 s  r: h" f) X( j
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     # |* V2 f$ a" u9 k
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     - o. X5 H: I: A6 H# m2 H
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     ( Y( m: i( w) D" c& [. G# C
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     5 P# k2 N2 v8 i3 \" t  h* c
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 "),
     ) j2 R3 ^- w6 _
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     7 N9 t% a1 {/ Z/ v6 P) F
160. }
     3 ?- q4 f5 ?, e. \! u8 G; \
161. 
     
162. Sleep(10);
     & s  f4 h9 h7 P6 ~( z5 }
163. 
     , V# o! F' q0 M- }. ?
164. char   buffer[10240];
     $ M7 E; o4 r. \  r0 n
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     * @; d# d. v4 t6 p( _! [/ o7 G
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     3 @  I7 N# {9 X* ?, d; y/ X8 ?* U
170. CString   result;
     $ I7 h8 {  F5 W' H" b2 W
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     ' ?7 Z" N- v/ @. g7 y8 f- Q
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);
     , `2 D( T+ ]3 d6 ?2 s
175. if   (result.Find(m_name)   > =   0)   {
     8 `$ d4 P. N3 i; Q4 g. P
176. for   (int   pos   =   0;;)   {
     ! g0 f2 B" E. w" K
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     / L7 R. }# F& V: e0 e$ s! H6 A. H
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     * Z8 T# ~: {" G0 t5 s- F# \* n
181. if   (name   ==   _T( "LOCATION: "))   {
     0 ]8 |' }6 q9 d6 W! U9 H
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     6 x' ]1 V- Y  e% a+ G$ ^; F1 P
186. }
     
187. }
     
188. }
     ( L3 Q4 p* V* I0 U1 R8 M# H! D& \7 A
189. }
     
190. }
     ( q  D9 u! r5 h. I8 X$ ]
191. closesocket(s);
     
192. 
     
193. return   false;
     5 o& Y4 M0 ?+ w* r' x
194. }
     

复制代码

天云上人

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

0 L4 |" M( }$ P- p3 @8 g
/ P  w% A% v+ H9 h" L///////////////////////////////////////////
8 v$ Z' q% C- x# {. Z( N//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


#pragma once
#include <exception>


8 }- W: Y/ w- d4 j8 O( u1 j0 o/ W. @6 O  enum TRISTATE{
8 {% [; B2 N. K        TRIS_FALSE,
        TRIS_UNKNOWN,
! g; ]6 Z5 n- u: f; I% J' [' w2 O; ]        TRIS_TRUE
};


enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
" [3 i  ?! A' G2 b/ Q, [: @        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};
- [( L- [3 A9 L- R+ u" N, Z) b

7 S7 e* b+ |/ E  p# m. D

! F. s" f, F5 B! i; y' Gclass CUPnPImpl
2 U& z* ^9 @- l( @" m{
' y7 G2 X, B0 k% g4 X- h% cpublic:
        CUPnPImpl();
9 k; R  @2 T0 a4 R/ E6 t        virtual ~CUPnPImpl();
! t9 ?" K& X. H        struct UPnPError : std::exception {};
" [6 y$ ]# q/ }) z        enum {
                UPNP_OK,
                UPNP_FAILED,
# ]5 m6 X9 D* p  J2 s$ ~                UPNP_TIMEOUT
        };
' t3 \3 h5 c2 v: H

6 s* Y3 I+ A* s  _        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
! D- Y% X! E  g% S3 }        virtual void        StopAsyncFind() = 0;
- U- B8 I6 {& e: R& X8 W) ]        virtual void        DeletePorts() = 0;
6 H" [! z# U' T1 D        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
7 P7 Q, J* V5 b0 y; l; S0 }7 u       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

1 H( m; W0 @4 [
! ~0 f! T5 h7 o6 R        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
5 H6 J7 N' j3 U        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
* A& w# S. M6 t  I& p* n( B/ K        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
0 z% M. K! W9 w9 y$ e
/ Y) E% Q4 e! N2 \
+ F5 x& r1 k* U  qprivate:
        HWND        m_hResultMessageWindow;
) K4 I5 D2 I) n        UINT        m_nResultMessageID;

, U0 [0 e- T0 [  \# i/ g: ?3 C$ g
};
2 ~! \8 k! D7 J7 p- i* }; p
% {! y# e8 v+ g( n, L: b# d
% }/ i/ w# W2 k& }/ s% V// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
# r: F) _1 g6 |) r) g8 u; Y4 M9 K        virtual bool        CheckAndRefresh()                                                                                { return false; }
) r. p7 P$ E4 G/ ^        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
- s  `. z, q+ \        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
8 D# A: K  v; O0 K9 \: r- L8 l

9 }0 P8 |+ v# x/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类


5 y1 y+ B8 t6 O1 H#pragma once
; I: I) i  A2 k' k+ o. H5 d#pragma warning( disable: 4355 )

5 ]5 j. ?) u* q# R) ?
7 m( J( R1 }  \#include "UPnPImpl.h"
#include <upnp.h>
' v6 q  F/ k5 C. d. P#include <iphlpapi.h>
) I9 `9 N3 P  B# L#include <comdef.h>
' n/ M1 [) l4 e$ v* v#include <winsvc.h>

/ S9 G9 L% a" C% G; G# A1 H
$ _, W0 P- h+ v3 V0 v6 j! T- R2 P#include <vector>
#include <exception>
2 a; _. q& ~9 D#include <functional>
# ]* C- q8 ~) d( T



typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
) f! V' U; N" g( K, }$ z( @typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
+ S2 v  v+ S. z' g+ Btypedef        _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;
  l7 P% E3 ~3 A/ M: Z
+ }! @6 A5 D* I, p! p) ~5 Y7 w
typedef DWORD (WINAPI* TGetBestInterface) (
! ]. d, ~/ t& \' N* a4 w: \2 ]" \, D  IPAddr dwDestAddr,
1 y% ^' a9 M( Q: r9 u5 J  PDWORD pdwBestIfIndex
8 ^& O7 l' A. N7 v# m8 U);
/ ]* W3 u/ A" F  G& ]' H- W# k

typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
6 t( a; R7 H7 \) R" Q  BOOL bOrder
);

1 U, }, J9 V6 j/ c1 v, h" y. I
' W& z( [0 L) k9 B4 V. Ztypedef DWORD (WINAPI* TGetIfEntry) (
4 Z1 }5 K) S! I/ W! r! I" ~0 Z  PMIB_IFROW pIfRow
% f+ r$ G* r* D0 s);


3 d3 G) k! ~4 ]1 ZCString translateUPnPResult(HRESULT hr);
) r3 c! M% A9 x. `, M/ BHRESULT UPnPMessage(HRESULT hr);
7 {/ L- U2 y2 x4 g: A8 Y

class CUPnPImplWinServ: public CUPnPImpl
5 G7 j0 [' x' y0 {2 M, `& m{
1 c, w2 c% X. p& U$ t        friend class CDeviceFinderCallback;
% R$ L, v0 @- I' b, o        friend class CServiceCallback;
// Construction
& n# V& M; `# kpublic:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();

/ x9 s) z) q- \" ^- Y4 \7 Z" d
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
* W) y+ f" n" ^4 O4 t0 V        virtual bool        IsReady();
6 J% B3 c9 Z) |1 ~8 x' `* U        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
: @3 K) d2 j/ S; Y& c
8 b; T$ }5 \( E
  g! S5 b+ L5 r# r8 Q8 r        // 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
$ }- N2 H( j: j! X& M        virtual bool        CheckAndRefresh()                                                                                { return false; };

* v: y" q! q! W* q% i' @  s6 T2 t
  a9 S% |+ I) dprotected:
; [2 L3 x4 l+ a        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
' J( K0 }+ }  g8 m3 y        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();


6 {3 V. L& s' a6 R8 b+ W        inline bool IsAsyncFindRunning()
        {
7 D/ l9 {, ~3 _) U7 M8 |4 X                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
  x6 i, z% k' }- k                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
9 q( g, ^+ q3 s# K                        m_bAsyncFindRunning = false;
# w, E; B5 m3 O& X2 m. @                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
3 M# Q& y& d7 V' C                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
' Z7 r. s5 g0 l                return m_bAsyncFindRunning;
        }
0 ^- V) w4 q9 p- Q, L, \

        TRISTATE                        m_bUPnPDeviceConnected;

& O: K: T; c5 s* x- K' B7 l
1 p+ b9 V9 u# d! Q// Implementation
3 `4 g& G: \/ O( ?1 p; q        // API functions
/ D/ l. h6 H, ]9 j, T        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);
/ A# b1 N% E# w        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


6 ?/ X) F6 Q: V, S        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
) j* {4 D1 ~: M0 N5 P; e  [        TGetIfEntry                                m_pfGetIfEntry;


        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
! _; e+ J% w- A* \% \  {# n- h& ?                        CComBSTR deviceName;
9 D6 f5 y% ?& ?' A4 C3 z; \                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
( x+ m! C8 T: c! T- e
9 Q, B% R7 i; r0 [
: e0 u. D8 p+ G* W4 y6 T                        if ( FAILED( hr ) )
5 A8 ~8 E$ o' U. S2 [# r! K' ?                                return UPnPMessage( hr ), false;
, O+ k/ D7 }- H- i" M4 b# w! p

2 [1 j' L: H4 A; C                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
6 }' O% y1 m: o8 C  f4 d                CComBSTR m_udn;
. k0 @9 S8 }7 f; P$ N$ y        };
+ ^2 h5 Q8 f0 |0 l4 q$ l       
4 c7 |4 E( |! i5 Y. |% M/ v: o        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
% [- z3 `9 Z+ u9 y/ u5 ?        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
+ q" Q# u$ l9 ~        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
' ]2 l& `. P; `. u) K        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
" F: C- k) X0 J( T$ x% @                LPCTSTR pszInArgString, CString& strResult);
$ y% {* X( [% O3 H        void        StopUPnPService();

/ I0 C4 w8 Q$ o5 `: D8 F( |
' D. _$ R6 s* G        // Utility functions
$ \6 g) e$ O0 i2 [0 ?3 K; A& Y        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
; ?2 C$ x' m# ~5 k: X! \0 _6 B. l        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
) a' J( m+ d8 u9 x" ]        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
8 I" j/ u1 ]) ~3 [; T* y+ j' p        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
& K3 T& C3 l3 h  X9 K* S
, L! A: C# e# B  `/ N
9 I+ d6 Y1 f. X// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
! t4 @  h: R" ]; j8 w# `$ x# ?        std::vector< DevicePointer >  m_pDevices;
2 d, N1 B8 ?. Q/ q" h& _        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
; a% u' |) f; [0 }- M        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;


        LONG        m_nAsyncFindHandle;
& [% @/ }4 [" ~. W$ c1 K        bool        m_bCOM;
        bool        m_bPortIsFree;
8 u  V  \' g  y3 d9 p1 `2 ]        CString m_sLocalIP;
* O+ _/ T4 i7 d% j) |8 H% k        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
0 M' @% X0 f1 P& u* r- s        bool        m_bInited;
1 u! C  a. ~* I9 t' d        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;

" W  M4 y/ k' K6 O9 g2 G& Q' m$ \
+ \' O- n# N0 {# L};


// DeviceFinder Callback
2 Q* V. `9 l2 I4 @" {, U4 b& kclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
( W6 T* }  x6 c. `{
" c; ~- X$ S5 x, O6 W8 Y/ I. }! y  opublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
8 X1 N0 I" p/ \6 n                : m_instance( instance )
1 C) V' T/ R& B+ P* l        { m_lRefCount = 0; }
; r" ^4 J+ |# [3 b) V' ~# J

   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
3 d  U0 S; U4 I   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
- ^3 V- B8 `8 ^

// implementation
6 h7 ?0 z7 y; @4 }+ k5 D3 ]# p, r0 Rprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
- ^9 j; B+ y0 T* q        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
6 G/ k$ e; z0 Z( S( y1 y8 d7 D        HRESULT __stdcall SearchComplete(LONG nFindData);
- w9 b9 r1 U5 D; x/ ^8 ]& ?6 }: @
5 p" K/ g0 }. |5 N9 Z
private:
. ]! k* X" L9 h9 ]        CUPnPImplWinServ& m_instance;
5 H! O0 H2 ^* y2 A1 c: _6 `        LONG m_lRefCount;
};

/ A! w- p; {% s1 g0 F
/ g4 [$ K5 a, S# `7 {. U( T// Service Callback
5 L# M' {! F8 b% P, a4 b& ?class CServiceCallback
$ |6 {1 v& M% B: t& n$ u& [        : public IUPnPServiceCallback
5 \/ x1 [0 f6 w! F, r{
; t  x+ N0 T8 |public:
7 E8 A1 |5 l* e+ r        CServiceCallback(CUPnPImplWinServ& instance)
- Z2 l5 Z% G& r7 M                : m_instance( instance )
& l. g2 D3 ?: h* K* \: e        { m_lRefCount = 0; }
   
7 b, z& R! x3 D   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
2 `9 ?2 f6 g2 S5 l' x' T   STDMETHODIMP_(ULONG) Release();

4 G: J) x5 p* W, R* \9 c
// implementation
/ I- Z; o2 [+ o3 yprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
" l# p3 l- c" O4 c};
: B& s5 [" }' N- ?% A" F: Q1 o
5 f% e$ C. j) I* C8 _! Q; b
9 a# f" J1 V( a4 J. W& x: X5 ^/////////////////////////////////////////////////
" s: f+ k& C/ k/ _8 ~  ^; d& I
3 {  e5 F& d$ Q$ A" ~: ^% f3 ^
1 g0 W0 ]# q0 y& s, E( {& A% s使用时只需要使用抽象类的接口。
& o6 V6 f$ M7 FCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
* u+ Z; f& R+ PCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.