UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   . z  D4 B" `5 z
8. class   MyUPnP
   
9. {
   
10. public:
    
11. typedef   enum{
    2 [  M9 L$ D/ P& C9 D& G# W
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    7 ^  ?7 S2 j, m0 o1 T" Q: e
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    # M; l1 X- i* p1 o) c: i
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. 
    ) T2 Z- V; a" T( }7 m" u8 }
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    8 [: f$ g. x+ C" H2 L5 Z
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    $ ]( r. K  O, X- f& }
24. typedef   struct{
    
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
    3 u1 a' |4 d3 U# y9 r
29. }   UPNPNAT_MAPPING;
    
30. 
    
31. MyUPnP();
    6 r4 A- v( X8 _- l
32. ~MyUPnP();
    
33. 
    - D2 U& d+ n" ]; E1 N! I0 x7 m
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
      z3 t- o! p1 y3 B
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    6 o3 H6 K# T$ @7 O3 P1 k! U
43. protected:
    
44. void InitLocalIP();
    , Z3 Q4 p& x8 p
45. void SetLastError(CString   error);
    ) `! j% @1 P7 G. Q' m1 ~
46. 
    3 U5 W7 k; l/ ^4 z  w$ ?( f* ?- i
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    ' |0 W4 Q; a- l, b, x2 f- U
49. bool   deletePortmap(int   eport,   const   CString&   type);
    ) Z, [4 F$ y4 I) Y/ q; U
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    
54. bool GetDescription();
    + |) M/ ~, B, n# |/ Z3 d( F
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    1 [8 }7 _' c$ e( u/ e4 c
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    " d! q6 i0 |  h/ g+ X, O
61. CString m_name;
    
62. CString m_description;
    7 b1 J8 E& _6 t$ F
63. CString m_baseurl;
    8 z* \/ x% R3 L2 W
64. CString m_controlurl;
    
65. CString m_friendlyname;
    4 ]  ?: ^  z4 G! z% E$ p8 b( X6 v
66. CString m_modelname;
    " L& F6 W7 S; n9 Y9 J. T, e& h, M
67. int m_version;
    
68. 
    ; |/ `/ a' _" E
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    6 T4 ^6 |5 S4 s
71. 
    
72. CString m_slocalIP;
    * ~/ |# p( ~3 U1 a( \& o
73. CString m_slastError;
      ?  L1 f6 J' s, y+ S& b3 h
74. WORD m_uLocalIP;
    
75. 
    # M8 L. r+ ?- t6 `$ \
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   6 T! ~1 q, f, j5 k3 j
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   ; t  V& z4 |3 q! U' L" N' r, l3 S
5. 
   7 }/ |" _2 |- Y. o) U6 Z, _
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   7 ?! }( U, y: F/ Q
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    ( t4 N  R7 b) u( A6 |5 z
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    " M5 p7 p, @  v' s. w# P" X& ]! @
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    
23. }
    & r0 U4 }3 B- M3 n# |7 U
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    ' r5 C0 @# _( l8 f6 n4 Q; E  ~* J
26. {
    # H4 f' ^% W- G: a
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    : S: v  G* m1 Q: N1 S, V) Z8 ^
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    : q0 z0 \/ k2 N7 w7 R: L
37. char   buffer[10240];
    / U( M# e8 t3 E
38. 
    - Z& T* d* O/ a6 }! P
39. const   CStringA   sa(request);
    2 A' L$ A( g2 u0 f: a6 v
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    0 M$ v. p* w1 r% I  r: X/ ~2 l
42. 
    7 @! }5 I8 H: A# ~: x* U
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    ( i7 H0 {7 u& @3 V% w
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;
    0 ]! P0 P( O; i! r
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    , V! ]7 K* e, w8 R" B- x
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    " f- M  a9 U- ?: X# ~* _( H
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    ; G# T+ M, ^3 ?$ ^1 F# K) P" x, T
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
      ^& g& H2 r3 [2 n7 J+ h
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    ) s7 f- j5 i7 k$ S6 z
57. closesocket(s);
    / L4 S( f8 I0 R5 y
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    + v2 ^/ t4 m" W7 [# ^6 H
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    . G& P- n4 z4 l2 Y
64. }
    + F  m; X# Z. Y/ `
65. 
    7 F: z- b- ^, E4 U
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    , {  S8 P2 U' t
67. {
    
68. char   buffer[10240];
    
69. 
    1 _. `' M; |2 _4 H
70. const   CStringA   sa(request);
    : P8 v: }4 d7 {3 o/ F% ?2 b0 D
71. int   length   =   sa.GetLength();
    & ~& E, m0 q, Q' |# R
72. strcpy(buffer,   (const   char*)sa);
    7 L+ }& K" q6 @
73. 
    1 {8 |+ j* Y; H1 j" c4 u
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    2 a; _# Q% P+ k
76. sockaddr.sin_family   =   AF_INET;
    - k, B, B; I% Q  ~
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    2 n- X, p0 V) [7 h, E# ]
79. 
    6 l5 A' _5 D, ?. e
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    % l7 O+ B% S/ S5 K4 n4 ]
81. }
    
82. 
    2 x2 z* H6 W5 n: x( V
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    5 c% j& a. N4 y' ~4 ^, [; A; g
84. {
    9 T- }6 e9 y9 e# ]
85. int   pos   =   0;
    8 o: J* B5 Y: _* Z. m
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    " h- |( j+ f, g5 l
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    8 a" M5 q( |, d( v, p
92. pos   =   0;
    $ f% ]0 h9 j* A, `: p2 r- \" ?
93. status.Tokenize(_T( "   "),   pos);
    0 e0 X) e; I+ u& Y( n: X
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    9 H2 B5 W9 i/ A( X
97. }
    ( O$ }6 q5 ^$ j0 ?
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     6 \! x# D1 p  a
104. 
     
105. int   posStart   =   all.Find(startTag);
     ! Y1 ~8 Y; @( Q
106. if   (posStart <0)   return   CString();
     . j; I- s9 c% ?" K( U
107. 
     ( u8 e& [! j4 O" U
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     7 k$ i+ `0 H) E; {
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     8 C& T& v5 c  C. z) s
112. }
     
113. 
     0 l2 s+ {7 R4 p, H1 @
114. MyUPnP::MyUPnP()
     " ]2 E, h5 b8 A
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     + M8 Z9 u) c4 m! O) ?1 Y
118. isSearched   =   false;
     # o8 _. N3 S! D
119. }
     ) N% t  x7 _$ d7 ~
120. 
     % @- E; l' k, u, A* P" C- z2 G9 r
121. MyUPnP::~MyUPnP()
     
122. {
     : [+ }5 b) R, {2 _2 l
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     + v2 I; B' v7 T8 S* ~
125. while(pos){
     6 I, Z+ j5 \7 N3 \( E6 F7 ]
126. search   =   m_Mappings.GetNext(pos);
     : ?2 B' X; i; ~9 p6 H
127. RemoveNATPortMapping(search,   false);
     8 L6 \. l% S2 D5 a" ]9 z7 ~. x
128. }
     8 o; p* Q' T4 u
129. 
     
130. m_Mappings.RemoveAll();
     + o- {1 E* s( ?" p3 [6 R( y& d$ \
131. }
     , K* J7 l7 B" o0 b
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     : C' w3 m' r  m( Q" b: q0 l
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     $ |+ o+ O# N/ Z% f$ q
140. CString   devices[][2]   =   {
     # h. l5 |4 K( A! B' k" L
141. {UPNPPORTMAP1,   _T( "service ")},
     ) r  y0 X+ X3 c; b4 |1 N( a8 e
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     + v: `1 b- b. S: {0 t6 U
144. };
     5 [# A$ j: x8 J
145. 
     7 y% N. M8 i0 t- s$ S' V
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     ' J9 J) ]$ a5 ]7 m( Z
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     + g7 i# W! j0 z% v% t2 E
150. int   rlen   =   0;
     # ?8 K9 F0 j  o8 a' F; M( @0 g
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     3 w9 |" m$ |" C0 g5 s
152. if   (!(i%100))   {
     2 C; p- v3 k- e! {9 Z
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     : U- Z' b7 p' d. F  }1 g8 Q
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     6 b7 v. j- R! m& O
155. CString   request;
     8 u" \5 a2 r2 C% d$ r9 p
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);
     0 D( c- s9 C# v1 y& s5 R# Q* z
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     ( }0 Y. h' {8 v8 b. O
159. }
     ) f! V6 ~  J2 D$ ~3 T
160. }
     
161. 
     
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     $ _( T! t5 ?* j
167. closesocket(s);
     
168. 
     ) C0 F' o% j, z3 [6 ^- T5 b+ I/ L
169. CString   response   =   CString(CStringA(buffer,   rlen));
     + H6 G# {- S/ A' `( L8 |! x
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     & Y. |/ r$ o  }8 L5 y
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);
     * K1 E$ O& `, L1 K/ p
175. if   (result.Find(m_name)   > =   0)   {
     / B& @, f) M9 D0 [/ r: k) [/ n
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     % _+ z; S. x: C, D
178. if   (line.IsEmpty())   return   false;
     2 z* o& J! p9 m4 O/ t4 E
179. CString   name   =   line.Mid(0,   9);
     ; J  b1 F( F, p9 D+ E
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     & I6 B" i9 ^. Y+ D; o8 L5 Y
185. return   GetDescription();
     % v; t. O$ B& E+ J
186. }
     % h8 T% b4 E/ t7 I8 s
187. }
     
188. }
     
189. }
     / n# Z9 w  E7 r) a1 @2 q2 q
190. }
     
191. closesocket(s);
     9 ~/ J6 ?) p: [9 Z/ w5 f; R3 _
192. 
     
193. return   false;
     0 |: h1 s' Q$ `- s+ a
194. }
     ; s% w: a2 Q) s) H

复制代码

天云上人

以下有关upnp的接口来自emule，
7 M$ E' o/ _$ j/ c( c
. a8 F% Y% K( w* U
///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
9 O0 J! U8 I4 r' z* u8 b8 z4 ?
  C7 }+ h$ o' D+ \: O2 V
#pragma once
#include <exception>
* I: q: n% Z  a( k
* t. |% P5 x& |0 Q) F4 t0 f' l; W2 F
( j: x+ y% k, i: g1 I. D  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
};


enum UPNP_IMPLEMENTATION{
7 s% u' I: {/ F3 B! d6 W& [        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
( ^1 [0 F( }  I; F, e# Y};


/ q0 O0 u. O3 @" D
5 \; U! j7 y3 u. a8 H
( b. [2 O% H* e8 P  X6 dclass CUPnPImpl
{
' i" e, M4 Z6 i6 J  H) b$ Apublic:
        CUPnPImpl();
" }; b7 Q: d* C7 ~& b        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
; {: l6 R! l- a7 [                UPNP_FAILED,
                UPNP_TIMEOUT
        };
# a9 B  m" A3 o8 K8 F- h8 S
  v- e% |- [+ R0 Y2 i) M
! M6 O2 T) j2 N        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
/ V4 l8 Y$ q3 t& K3 q        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
' Z& X- C4 f% w, C       
* M0 l1 m8 c5 y1 s, ^        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; }
+ a: T+ |% H: Y) S5 p, o5 X        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
! X8 ^6 [3 [7 C/ J- k3 d5 D) `+ O

// Implementation
" R+ q+ g( S0 k% W$ Hprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
4 b9 _& |3 s) A' Z% ]( f        uint16                                m_nUDPPort;
) q. D7 ~, Z) J: t        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
' |- q0 l2 V" x2 M        bool                                m_bCheckAndRefresh;


private:
        HWND        m_hResultMessageWindow;
4 E, y0 [- L. U3 N+ e& I        UINT        m_nResultMessageID;

+ M* P; d' P7 L' f  ^: A6 I
4 I- F* F3 v' V: z};
& o+ ~3 @9 @$ W* a  Q" j# b7 d
9 r& o' Q  U$ G8 C1 o
// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
. R6 ]8 X) R; C2 \% |/ Cpublic:
' c# V2 r+ D4 l: y* H& i. y8 v        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
/ c: }' X/ P$ U! I' D+ c& Z        virtual bool        CheckAndRefresh()                                                                                { return false; }
2 X; A, y: D+ G& T; t( _" ]2 k        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
3 H/ n# y$ D4 ]- \: w& I7 A0 x        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
+ R7 n& z3 e; Q& Q" X; t};
' `! R% z: X# P0 u

6 _& X! `9 J+ e7 h, V- K/////////////////////////////////////
$ i7 h3 Y: d6 f//下面是使用windows操作系统自带的UPNP功能的子类
% r( m  M; f- ]9 o9 r

. |: B& ?. J- P' _0 d8 j#pragma once
#pragma warning( disable: 4355 )

; u8 P! ?; }/ o2 ~$ w# t( Z
  H- _$ X; j* V3 j* i4 V#include "UPnPImpl.h"
1 \* \$ ~/ }$ m1 a  _7 A0 E#include <upnp.h>
0 G* w/ a) \( i* ?# _#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>
; H+ H+ N" ~, y2 U; T) i
" G' O' q5 m5 A$ k
#include <vector>
#include <exception>
4 W4 I: n8 r7 t: `( |5 d#include <functional>

0 u. H. }) {4 T. _% i: }


typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
# j" q5 U5 M- B0 N) ]typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
3 H$ `. `4 k% l+ C( _typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
3 }* w+ x5 \2 f6 H) qtypedef        _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;


typedef DWORD (WINAPI* TGetBestInterface) (
+ Z2 i( H4 H- Y" r  IPAddr dwDestAddr,
: R1 |; ?$ i! X) c4 q8 W  PDWORD pdwBestIfIndex
2 }- t2 @: g* J- O. }: Q  G. p6 Y4 p);
/ `- \, Z4 U& E
! z8 I2 y* \, C
typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
7 a, t$ h: O4 F& h$ b( s  PULONG pdwSize,
  BOOL bOrder
1 }5 l& E- ]$ Q);

  O  M+ J' b3 Y+ w
typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
, X# s7 j' l2 i( O, w* O);

# v- ^) ^1 A) h% t4 d3 |# w/ B
7 j* q, \" b, b% K4 X8 sCString translateUPnPResult(HRESULT hr);
" `, @! W. ?7 ~5 `( m4 qHRESULT UPnPMessage(HRESULT hr);
% l" O/ P8 J  Z: d
5 B& @! X* H$ T- F
; L' ]5 a/ K) u% r6 B% Eclass CUPnPImplWinServ: public CUPnPImpl
6 E1 B# x& [5 a8 ]4 s: J{
6 Q3 ~+ e4 ^7 J& r3 Y, E, W3 i. a        friend class CDeviceFinderCallback;
) _/ N" S; w% E9 c2 }3 G) C, M        friend class CServiceCallback;
# O4 \1 W, B" g6 P// Construction
  x) U$ E' e& T% M/ Kpublic:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();

: C. x( _! u6 d, ?2 D0 n2 y
! b5 p# f9 |5 r1 n        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
% \3 e# S& r$ t$ r) e* A/ ?# v8 ~2 y" E        virtual void        DeletePorts();
3 v! N! w4 @$ K5 f+ ]* T, G& a        virtual bool        IsReady();
2 q* |9 ?9 e, f# R6 X- }: N        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
" Z; m  b/ G0 O, g9 S) A! a

2 e4 p+ x# ]7 F3 y        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
4 o- Z" u) k  j- E  o/ U. ?$ s        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };


  ]5 y, z  ~) ]( K# k; d# D% e+ Vprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
0 k+ m/ _# k5 w        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
3 h3 y& F% F; L; B        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();
0 L0 j" I0 {5 v8 `2 W' o

" X: `- S& H. C7 m6 V) \1 _8 V        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
; \0 t7 d! r% e( ]: z                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
$ T( t$ Z3 {/ \; N: ?                MSG msg;
! h) c8 _0 ~1 c, Z* I; U                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
  Y; k$ |& ^4 ?  d                return m_bAsyncFindRunning;
        }
, B( b# u* u0 d$ n

# a& [! ^$ H2 b# Q) {: V        TRISTATE                        m_bUPnPDeviceConnected;
( w) m! G* l! E' c6 G  A

// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
' z2 M7 X4 N! i9 @- P6 B        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
7 H2 K; {' h& Y3 k+ H8 ?. b% A+ I        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
$ t# `5 s$ _8 j7 [2 D9 r% p! I        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


        TGetBestInterface                m_pfGetBestInterface;
, y0 r' s7 t6 N/ O        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
" I8 C9 Z) [' D0 `
. |) ?: `% l" u( q9 m# B8 Q
        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
( b: L- _( Z! t- l$ P, S        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
  S0 N: [. k' r1 G                result_type operator()(argument_type device) const
                {
3 d4 P" c7 U# K4 |: F" K                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
. N, T# ]$ s1 m# q6 w5 m

                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
# ^3 b1 z0 D( [: K: T4 H

                        return wcscmp( deviceName.m_str, m_udn ) == 0;
# Y$ G; m" Q' M* y                }
% [) {& S$ P- [' z7 U5 h1 y( L                CComBSTR m_udn;
5 S) o+ g; U. ~9 K% Y        };
       
+ I5 l# b2 z: Z* Q! F5 o' @        void        ProcessAsyncFind(CComBSTR bsSearchType);
5 B$ [# |5 x! X        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
/ o  K3 s' _( r! ?4 g        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
- P5 ]+ B8 P+ b1 k0 F4 A# d        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();


        // Utility functions
) G+ U. e) m4 _1 Z. Z8 ]  d' e        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
( Q1 D* Q2 b0 X9 S        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
% {$ A& V3 r' l1 s        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
) t: K+ z( ?+ [* F        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);
. J3 q5 {# z/ l1 E

3 @; z# r" ^9 m" a, _) A// Private members
private:
7 d7 P6 i6 ?6 O4 g# ~& t        DWORD        m_tLastEvent;        // When the last event was received?
9 H8 z7 d  \7 J: N, H# U8 F6 i) S        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
) c8 x* g6 O8 b! l. x        FinderPointer                        m_pDeviceFinder;
! \1 W/ u; w% N; f: @        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;


        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
+ Z" U8 y- k; s( i2 f2 j" j        bool        m_bPortIsFree;
        CString m_sLocalIP;
  D) n- k2 `% h2 k0 ^5 T2 q, T        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
; N4 H) v9 w- G' p        HMODULE m_hADVAPI32_DLL;
) v2 P5 m3 v: C! ?+ y6 e        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
, `$ F) g/ Z+ u        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
- J# I# F# `; f) A" d        bool        m_bDisableWANPPPSetup;

$ `' }3 W& R0 C/ x  l' t  E
& J& W0 B0 ~' f$ J. ]0 `+ {};
& T: u0 U6 C# G; m! s$ M) I2 U

// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
: q( I6 o; s* G  \% d$ s{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
1 @$ _. X7 p2 [                : m_instance( instance )
        { m_lRefCount = 0; }
+ l# c! k1 Z/ u" |7 K

, N8 e  z# W( {. P   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
1 e6 E; h( }# F% |% t2 }0 t   STDMETHODIMP_(ULONG) AddRef();
0 u1 F7 r. x2 I/ [& z   STDMETHODIMP_(ULONG) Release();


$ |2 e5 x9 i- \; Q% \, }" k0 I// implementation
private:
% f, E0 U8 Q) L+ M. x        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
9 }) m& ]" }- P3 P        HRESULT __stdcall SearchComplete(LONG nFindData);
. D, v1 [. ~8 ~6 t: J- `- n* G

private:
        CUPnPImplWinServ& m_instance;
9 T8 _1 L/ C: R9 l+ T& |" V' P9 I        LONG m_lRefCount;
. L0 Z! l! g5 B7 g2 b4 y};

. w4 f: N; q4 Z( |
// Service Callback
class CServiceCallback
$ N# ]4 q) p# b4 E        : public IUPnPServiceCallback
: z( C$ h3 y2 @3 ?# c  J: ^{
% p1 T: E. d) c6 I4 r# d5 Gpublic:
        CServiceCallback(CUPnPImplWinServ& instance)
" `. P* {; i6 d                : m_instance( instance )
) {) f' P6 d9 s6 o        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
2 I, Q! S/ @, l
0 B9 S& G4 j4 R3 C9 j
7 J- D: o4 N+ N. \/ m* G// implementation
, |$ ~& ]- j6 c) b0 vprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
* d; E$ L  ]% ^' q        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
9 f$ Y# j* p& k: N5 S7 K$ c8 A+ I( x
) h) s- @% _, K; D
private:
        CUPnPImplWinServ& m_instance;
" \" R/ H! X! ]* w, F$ Z        LONG m_lRefCount;
1 j6 z) p3 ]. `: T! Z8 U6 J  @};

, j6 v) X: l! F+ r$ l
! }' H9 e6 X/ [" F/////////////////////////////////////////////////

3 C: |+ Z( W& g1 _* e
- i1 W7 k. j$ ~9 X& o使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
2 u' G  Z6 y3 A0 k$ o- ?  Z# }3 @CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.