UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   
5. 
   : _. X0 T" g$ N' @3 c+ V
6. typedef   unsigned   long   ulong;
   
7. 
   * m. e9 c* y/ Z8 Q) O7 ?. h( M
8. class   MyUPnP
   2 Y3 X, i4 S8 ?7 I4 Z
9. {
   
10. public:
    6 [0 i7 l& ^, v3 f% D
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
      j3 H" h: p5 C6 s( z' L
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
    7 Q0 G1 j$ O  I' D/ w. a$ I
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    $ q/ u  Y7 Q& ^! Q# i. i
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    1 L+ q! P- K' Y; @5 r4 c
22. }   UPNPNAT_PROTOCOL;
    " E) _' m$ a$ ~/ c) q$ v! J
23. 
    
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
    
29. }   UPNPNAT_MAPPING;
    2 `! J. e' z2 x, K
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    9 H: n- c% ^. S; M, D
36. void   clearNATPortMapping();
    7 n4 B; I; x1 ~
37. 
    + Q9 @& e0 \' C! P: G& e* F" c  m
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    3 b4 r1 O: g2 e: f
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    - ]3 R) n. q; H0 y9 s" S
44. void InitLocalIP();
    - P0 y/ N0 G7 [( ?0 c1 c
45. void SetLastError(CString   error);
    
46. 
    - m+ g! ]" G" e) x& \/ n% d
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);
    6 P' N# D  |+ R7 i9 `# g
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    & q- B) i# a( H7 K
53. bool Search(int   version=1);
    
54. bool GetDescription();
    # y+ y, G( F! I4 {
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    
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;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    8 y! a- ?3 B+ |( r$ m! o; t* y: ?' k
66. CString m_modelname;
    & |8 x- x5 O6 r) l7 e7 N
67. int m_version;
      z6 E' r" ?+ S/ \: K' d) d
68. 
    
69. private:
    9 b; S" _/ t; E2 l. V, m: {6 f
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    0 u( N! V# f% {$ N' K" o4 c% m
72. CString m_slocalIP;
    9 z7 V8 Q& I& a/ ~9 Q, \1 O$ v/ o
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    " Y$ i- `, f' y1 J
76. bool isSearched;
    
77. };
    9 q6 Q+ _5 B/ Q; z! f) P! v$ P
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   ( v/ T- O) f8 T$ }* ]9 K
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    ( a5 }- |+ y6 Q
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    # \1 c% x6 J) a+ ]: |9 t2 N
13. static   const   int UPNPPORT   =   1900;
    / l5 \& B  V7 P+ X
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    + f& A6 _) S* E9 X) ^5 R
15. 
    
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    0 E  E$ D* i* B- J
19. 
    ' |0 g- f" u! ?" A+ Z
20. s.Format(_T( "%d "),   i);
    
21. 
    ; v- p& G2 h4 _
22. return   s;
    - y; M; m) o  j
23. }
    ! ?) N, ^& W( w& W, u% N( o
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    $ m7 P8 S6 \) ^
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    " N3 R& k5 @+ H
34. 
    . O2 @5 N6 o5 a6 M
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    ' Z' w3 l8 P7 T# M  U" }
40. int   length   =   sa.GetLength();
    * u* C6 B, M7 ?0 s. D
41. strcpy(buffer,   (const   char*)sa);
    
42. 
      ~  y6 f: m. G  k
43. uint32   ip   =   inet_addr(CStringA(addr));
    3 v5 T) [2 L- N0 I7 ?
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    . Z9 t6 r3 D* l+ b- W
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    # g; a6 q4 c! n
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    2 S7 |9 O; G& \$ C+ _  q( N
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    ' O: l1 G( h  C, z
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    9 g1 v' [$ U1 T2 B/ o3 k
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    6 \% Q- j& v8 L# G
57. closesocket(s);
    : H9 M/ K3 T) y! V( F$ }. Y
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    6 A/ j+ h+ I! T( E
61. response   =   CString(CStringA(buffer,   rlen));
    ' m* X9 t& B% H
62. 
    & ?  G/ R& z2 I5 d3 Q$ I+ |
63. return   true;
    
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    ' ?: G7 Y" }6 u
67. {
    : Y( |9 r* w1 c1 g0 n* Q
68. char   buffer[10240];
    $ w) z/ S# ^! r  ?4 N
69. 
    
70. const   CStringA   sa(request);
    " _7 S3 Q& J( D+ D" |3 w2 ~
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    1 t. ?' W: N% s
73. 
    7 W( `) |) E" Y8 r; Q
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    ! S0 w  |2 f" h" C* u! p& E
76. sockaddr.sin_family   =   AF_INET;
    3 R' P% ]$ V. N! C* X; [$ t$ |
77. sockaddr.sin_port   =   htons(port);
    6 |5 a3 m7 J1 N0 s+ g3 P6 S* a
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    2 _+ y; x0 g5 j0 a1 d) }
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    1 f1 H; E3 M1 `7 B
81. }
    ) ^3 R) M0 l, ^! N
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    / i. @8 |: N6 L9 a
84. {
    4 k/ _& ?8 J4 l/ l" I. K; V) @" \8 P0 W
85. int   pos   =   0;
    
86. 
    $ `8 f- \( g" W1 x* k- `
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    * H. K  ]8 j( C5 n0 _! E* Z" o9 e
90. result.Delete(0,   pos);
    # U$ T. W( q1 L: B" R/ s
91. 
    / ]) t  g* B2 x6 |# m6 T0 n0 B
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    + z! Q/ @3 @  D2 C. Q7 _& ~+ ?
97. }
    2 [( x& w0 j( s9 j1 c4 X
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     + ~% ^- b0 u8 R# k% E; r3 h3 q- K! F
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     + o6 q7 [" N5 R" n
103. CString   property;
     % |% U5 I. W" }( R* j
104. 
     # h5 Q1 M* s9 i
105. int   posStart   =   all.Find(startTag);
     . G! A. ~- q' |- p8 u1 ]! r2 {
106. if   (posStart <0)   return   CString();
     ; ~1 c5 b4 K) v
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     3 e6 r; R, R4 ]/ d3 O
110. 
     8 [3 z% g' B6 D- l9 N4 j7 Z
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     8 m0 Q) l4 a, R! H1 Q
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     ! i" K' ~& R" p/ E( f
116. {
     
117. m_uLocalIP   =   0;
     ' R0 L. I  j% \3 b
118. isSearched   =   false;
     + G' |- U1 n" B  T4 ?
119. }
     - B# n. }; y7 |! o
120. 
     
121. MyUPnP::~MyUPnP()
     6 q5 h! P0 F2 `/ u5 m
122. {
     
123. UPNPNAT_MAPPING   search;
     ) B/ m& g! O( q1 u
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     8 e: y" |/ ~/ p6 c
125. while(pos){
     + J& E3 _& @9 X2 Y( T* S9 i
126. search   =   m_Mappings.GetNext(pos);
     2 c; u* I1 s. f/ D4 s9 x( T6 T
127. RemoveNATPortMapping(search,   false);
     
128. }
     + _3 l+ l% h( }5 _$ G3 p) h
129. 
     6 H5 w% p6 @& x( W+ L3 i$ k3 H" C5 L
130. m_Mappings.RemoveAll();
     4 a0 @1 j/ U" x% E0 _6 Z" z! u( G0 [
131. }
     
132. 
     
133. 
     ) T# {4 e9 o& L) j
134. bool   MyUPnP::InternalSearch(int   version)
     ( H6 U: J5 Y( Q* {; g$ ]/ ]
135. {
       y/ v( a9 l; q7 u) g
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     ' z; L5 d% K' l1 H6 y! I  [
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     & a- F) ~3 i% r2 {; S, [' h  ^
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     9 g- A# L9 v6 C
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
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 "),
     8 S# }. R9 t+ I: o+ u: W
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     2 z8 n  D5 F: P6 P; I& x
160. }
     8 N! p$ V2 X+ N
161. 
     ( j) p7 {  ~0 v# S! @: o/ h6 [" E2 z; g
162. Sleep(10);
     
163. 
     9 S9 b5 [. }  i. \8 M" M( u8 y
164. char   buffer[10240];
     ) ?7 A! @: Z9 [; f
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     8 Z3 L9 d0 o( [; j
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     & M% f  L% P9 u- W* i; k
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     * `* \2 p! t- B5 N9 \
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     # x6 W$ Q6 G$ f* Q' v
176. for   (int   pos   =   0;;)   {
     3 f7 ^9 f  P5 ~1 @: h
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     + X# g' ]) I2 A0 [0 N
178. if   (line.IsEmpty())   return   false;
     * K3 K8 N  Y* X# F0 K& C
179. CString   name   =   line.Mid(0,   9);
     4 g2 _+ X% p0 {" D
180. name.MakeUpper();
     : t9 y4 ?" g7 s2 T
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     0 R; G6 K0 }4 N
185. return   GetDescription();
     + W& F2 A: T3 h* E# _3 a( p# a% v% M
186. }
     
187. }
     
188. }
     
189. }
     
190. }
     " m* V. Y2 K( v% g& i
191. closesocket(s);
     ' M& S% J9 [/ r! T
192. 
     
193. return   false;
     
194. }
     . ], q, \* \! a) L" v) B

复制代码

天云上人

以下有关upnp的接口来自emule，
# U7 W& K. P, w* d6 j

' J$ Q9 i* V5 [% W, x///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


6 ^+ f. V+ q4 o6 B1 \4 H#pragma once
#include <exception>

: h9 h; q. v! d
& ^2 H8 W1 E" x; o9 h) R  enum TRISTATE{
        TRIS_FALSE,
9 d3 R6 y2 F- X6 A        TRIS_UNKNOWN,
        TRIS_TRUE
};
: `; L6 s0 l0 Z: O
9 L8 |9 `; u1 K" ?$ |5 t8 t
& o( a! U8 M$ s% ?, Oenum UPNP_IMPLEMENTATION{
; r, D( @" ^, `- b# C        UPNP_IMPL_WINDOWSERVICE = 0,
# q# @  f: i( Y' L% d. l' D& ^$ t        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
% M$ J8 O7 r6 v3 P- l5 r};
3 Z' a0 B3 S" u1 h( p# b
8 C1 b5 `5 `' Z6 @1 w: V
+ H- A, f5 {4 M2 l

class CUPnPImpl
{
public:
+ M- L+ R1 @# V7 ^+ z$ X        CUPnPImpl();
/ A; M% L  w  I5 w4 R, t3 U        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
5 \/ t: s4 g9 p1 C: \5 u        enum {
                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
        };
2 ]8 B$ l( Z8 p
% H3 |, P0 G( y6 i% d
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
" s% r  o  [$ [& L5 C        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
4 N: w3 _0 K. O7 q& f8 u        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
, _& ^+ u1 P* P# s7 _       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


+ L' r5 }5 w% J% o, C+ r, N        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
5 t- a. j5 M8 ~9 C% b        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
- p. R( @" y" B. ~        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
. z& a6 D% M9 f, ^
) C/ u, L$ _0 v
# Q$ l( T$ g8 F6 W+ I// Implementation
protected:
; d5 r* k! ^. d. {2 S0 b        volatile TRISTATE        m_bUPnPPortsForwarded;
5 E5 \9 v) t$ `& s+ ]* ^        void                                SendResultMessage();
" G8 D( Y/ k, T( m$ h6 }        uint16                                m_nUDPPort;
( I$ b9 t* b! l+ y, Y& F  ~        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
2 O+ H8 m  R. n5 _( i, E5 Q        bool                                m_bCheckAndRefresh;
5 Q, p" `, C- f2 R

% v! W' m1 m5 v; M+ T" yprivate:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
- e  k: o: b3 K% m' j  X. H/ J
4 t2 `) h, J& }' ^; h) A" `, C+ L
};


// Dummy Implementation to be used when no other implementation is available
- y3 }: g- O% @% k- i4 Bclass CUPnPImplNone: public CUPnPImpl
% Y8 X$ {, C9 Q# {{
; n% ~% d$ ]6 x$ m% R; y- Gpublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
7 z( j. ]% j" M1 m        virtual void        StopAsyncFind()                                                                                        { }
, S8 t5 K5 A3 O3 _  v5 t        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
  S; H% {) J/ E4 o' |};

5 T+ s5 H7 U9 T9 k
. T- K7 B- e, s6 j) k- d6 k2 i' ]/////////////////////////////////////
+ W0 Z' `+ V2 P$ H/ Q' j! Z1 _//下面是使用windows操作系统自带的UPNP功能的子类
- l  r' q. C% ~: l+ V
4 m) ?4 K/ ]7 e% v9 l  q8 J: u! w
( {& b$ D2 T5 x- U- O. e2 ^#pragma once
#pragma warning( disable: 4355 )
( e2 G: l* K# W9 p* X/ T9 U

#include "UPnPImpl.h"
' S) I, d$ ]& @! ~% p#include <upnp.h>
! N( Y3 A- @: {+ h3 B#include <iphlpapi.h>
7 E, _' g0 _! p. \0 q  ]#include <comdef.h>
; P$ Q' F2 `# }#include <winsvc.h>

! A, x3 t5 X- v. P
#include <vector>
& b; L; J- n- a& o# {6 y#include <exception>
#include <functional>
% f+ G, Q" d% a: o9 q! G

0 B' {, g, e9 |* V; @+ b

" l& _" @. B' D5 ]. rtypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
! Z( k7 u8 J( V5 V( t7 ftypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
- S# z% J$ g. I, L9 R: w' k% ktypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
3 {% y( O* w  L- m* l" Ntypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
/ J4 X* w' b# R, L5 mtypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


typedef DWORD (WINAPI* TGetBestInterface) (
9 A0 y; F% h0 j# t( `  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);
$ m0 v( [1 i, @" U/ [5 c
2 S+ b# X) v3 N6 d4 S+ A
0 o3 ?1 {' P8 z8 B) F1 l" s9 ~$ ftypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
( J, f. p+ {, F* j) F5 Y1 u5 A  PULONG pdwSize,
  BOOL bOrder
);
* O2 X5 @! M/ B$ `3 d6 Z
7 O4 u7 s  v( ?. m0 j
typedef DWORD (WINAPI* TGetIfEntry) (
! s9 y* U. ]' H& y9 g  PMIB_IFROW pIfRow
);
" U1 [1 Z7 b/ H; R9 q" g" {
" z) r; D0 B: i- E  x2 G4 H7 E  [
! \. _# A6 S& z: v4 \CString translateUPnPResult(HRESULT hr);
# _" O- f% N% N/ ^0 y+ ZHRESULT UPnPMessage(HRESULT hr);

" p/ Q- \5 H5 Q% Z& k
class CUPnPImplWinServ: public CUPnPImpl
{
6 P" d$ k8 y0 F$ ]# j1 O  @, Q        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
) Y$ D& M# B* s+ i  Bpublic:
        virtual ~CUPnPImplWinServ();
4 c6 p0 x. f4 Y# H        CUPnPImplWinServ();
3 [9 c+ J% l0 @- B

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
/ c5 u: ?  {! ^- \6 x4 Y        virtual void        StopAsyncFind();
6 G, w7 E8 @! H" ?, \        virtual void        DeletePorts();
/ r0 [! N- ^* G% b0 M        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

$ |% e% z) x0 a! C
# v& u  J7 U, c. Z- o        // 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; };


3 a! F/ K1 X$ [' S1 Uprotected:
3 ~- t1 u. }$ f7 C        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
8 d2 N5 F5 N3 J: ^        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
/ K! b. ^3 b2 W2 B# X        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
7 m2 ^' d5 w6 x' z+ p& e        void        Init();
2 N# n2 s8 I- n1 U) ~6 t
$ I" L- W6 n! x. l. S" S" |
        inline bool IsAsyncFindRunning()
4 d4 D9 p) ~  G$ {        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
1 s. e% U/ n  B( D                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
; W0 Q0 l8 h2 n- Q+ q6 i, z                        m_bAsyncFindRunning = false;
                }
+ z, @9 @1 o( @                MSG msg;
5 f  N& V8 Z) O4 X8 P                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
0 @: i8 }6 X3 k( _5 P0 K                {
2 Q* `7 @+ \# }+ r- G5 Q6 A                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
; j3 ]" s# Q* C) ^                return m_bAsyncFindRunning;
        }
% Z$ F" Q. R/ w# k

' [+ x" i( @/ y+ [0 R        TRISTATE                        m_bUPnPDeviceConnected;
& a- G% Y0 l" O% ]( b
" P1 p. h% d7 D, w
' n0 R, l1 Q3 N" u& G2 U+ f// Implementation
        // API functions
1 u% M" T) I  o4 J/ L* a! m        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
! |' o1 q+ J8 i+ O" B2 {# L        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
5 I6 `- n) v, L0 N2 @% R        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
3 M6 @- N- I' o% i* I( Q
& l2 z5 D/ @& O1 r) _
        static FinderPointer CreateFinderInstance();
# s* x2 Y6 `' _        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
* w& M5 D9 s) H( R                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
) L$ O; \: [8 L6 a                result_type operator()(argument_type device) const
, S) [6 X+ G2 Z                {
5 @* O9 ?- T' w$ s0 T                        CComBSTR deviceName;
# F7 S( I3 s9 H5 @  ?8 s: s5 [                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
6 @+ j4 t8 w( q! k7 ?% c- k
- b8 C* Q# z+ f+ v& i2 g
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
/ f+ r1 S; i8 d& H* L                }
                CComBSTR m_udn;
/ c- n9 t% U, y/ `        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
  p+ I3 W! A8 h- j' T- O        void        StartPortMapping();
; X7 R. \5 c6 J/ l, j0 q5 _) \        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
6 U0 X/ D# V0 y" }1 Z( B7 I        void        StopUPnPService();
3 W+ F9 M' E3 [) X1 I5 g1 n
; @( i6 \! H5 E
        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
* u% D) B0 }7 [7 B0 o        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
! d! c% k4 P/ v* i% d. g        CString        GetLocalRoutableIP(ServicePointer pService);
* S- B4 J, Q/ X

// Private members
  U  N- Q+ P  d6 c$ ?( `private:
        DWORD        m_tLastEvent;        // When the last event was received?
6 ~5 m  `0 F! `5 O3 g$ L        std::vector< DevicePointer >  m_pDevices;
. Q, s* W7 j+ Q/ a        std::vector< ServicePointer > m_pServices;
  t  T4 \: i3 S$ B" n        FinderPointer                        m_pDeviceFinder;
4 C, Z3 l- D3 o9 H0 U/ w& @        DeviceFinderCallback        m_pDeviceFinderCallback;
3 \! B8 C0 s  R$ q( X' V' T1 T* I9 G- [        ServiceCallback                        m_pServiceCallback;

. k: a* W, `" m- m( u
  T) z9 `$ s1 F        LONG        m_nAsyncFindHandle;
  c( H7 g# j) ~        bool        m_bCOM;
& ^2 [, x) I5 e' U& a* C. R        bool        m_bPortIsFree;
        CString m_sLocalIP;
5 p( V. y4 V2 P8 Z! C0 {, |        CString m_sExternalIP;
. u% d5 s( Q3 l% m* v4 C; x        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
% h, g3 i2 n# b) l- h        bool        m_bAsyncFindRunning;
! R4 d2 H+ `1 ^+ Y        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
0 A; ?; {) m5 p9 s5 `        bool        m_bSecondTry;
: O( o0 T9 n% U, ?        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
$ E3 R+ D5 p; K7 ~% s$ V' a$ D; v        bool        m_bDisableWANPPPSetup;


};
  j  e% n8 x: j# v( G
1 I3 q: P9 D) @/ E
// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
5 h: l5 i( b; ?. D) [2 a{
- _. z  J+ f- M/ V- K" Tpublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }


8 Y. @8 H& ^! a2 E% Z; g8 x   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


6 f6 s) s  ?: [  Q// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
+ r$ B  P$ V  W8 P' z, W  Z& Z8 C        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
8 w* v5 z# j  ~4 G" ]1 W        HRESULT __stdcall SearchComplete(LONG nFindData);
' t" _% ?0 k; s  x( s" T/ b/ }) k
: J& I( k! R  Q' K6 v
. s; x! K  B- Sprivate:
        CUPnPImplWinServ& m_instance;
: ?" T# }; w, ]        LONG m_lRefCount;
};
# Y' l* m# p# e, ]$ A7 Q6 K9 w

// Service Callback
5 ^9 y! G$ Z( rclass CServiceCallback
        : public IUPnPServiceCallback
{
4 F" }. z( T5 O* ^: H' M  f$ g* w* \public:
        CServiceCallback(CUPnPImplWinServ& instance)
  c3 D0 o  M: c6 \, A) G' @                : m_instance( instance )
4 Q3 j2 |9 s/ ^+ V        { m_lRefCount = 0; }
+ z7 t; o8 E6 V   
0 ^, d! C1 S9 P) Y' j' q9 z   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
: _0 M8 H# Y7 l+ ?6 b   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

/ W7 N$ U) m5 u/ J6 R
0 u) d) H% x) _& r1 p// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

9 x  B9 W. Z3 T% e' E0 @
private:
        CUPnPImplWinServ& m_instance;
6 e# V6 E# ?8 s. W        LONG m_lRefCount;
};
! r$ G$ v$ o4 W0 F0 b5 l

/////////////////////////////////////////////////
4 g2 G4 T7 a2 E

& {) G6 m; w0 L5 u使用时只需要使用抽象类的接口。
3 U3 D# |5 H9 v9 o6 l6 X# |- K, KCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
$ d* ^" u8 P5 R1 I7 `! V  ECUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
: X" s# a1 I) bCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.