UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   - ?; L# G( T0 t
4. #pragma   once
   
5. 
   $ p/ g+ b7 [' S* P  n; W& L
6. typedef   unsigned   long   ulong;
   1 H6 j, V* J9 J2 z# |5 H
7. 
   ! [) ^0 x/ o/ ~, Z
8. class   MyUPnP
   
9. {
   
10. public:
    ; B* e! |7 ^: ^" f
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    ; n' L1 e2 R4 c
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
    9 C2 V2 d$ m2 h# k$ T. w% `
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    : e; F" {) k9 U. S1 V" o1 S: F
18. 
    / e, r) i, s7 @8 p" B0 @
19. typedef   enum{
      H. K3 u! Z+ a1 \, S
20. UNAT_TCP, //   TCP   Protocol
    ; o* a& U/ w: Q$ a) c
21. UNAT_UDP //   UDP   Protocol
    ; e" g( C& S2 M5 q& E* _
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    1 \' A$ z" B5 w+ Q7 q2 G, y! a
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    2 U# o* G- l6 ?" P8 b( H' x
28. CString   description; //   Port   mapping   description
    9 m" H" c0 `/ N; k! Z2 Z4 L% U8 t
29. }   UPNPNAT_MAPPING;
    ( R' _3 Z( f* {# n9 J( U  _9 V
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    ' d  C1 s" b. {& ~0 m
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    # d$ c! O$ A, n" g* R
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    1 G1 \) |6 _; A0 o- c
42. 
    
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    & F$ [/ X9 r" j6 p
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    9 e& h; h. c) u2 L. J4 v; w: ?( {+ Z; D
49. bool   deletePortmap(int   eport,   const   CString&   type);
    ( h* X* L! a6 e( \
50. 
    " @, e" n! t8 q, c( q0 o& J
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    " z) I9 R) L8 X1 z9 @- D
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    4 ^/ \( S: ~+ T+ c
57. 
    " {5 l6 i* \  |
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    4 O" \; u4 d% H3 p
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    7 {6 u" E! ~  i7 G) m
61. CString m_name;
    
62. CString m_description;
    % i" N3 l8 ?. V8 c* \
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    $ }5 l' K# ^/ u5 P9 e& M1 e9 L
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    , T, @3 e  E  o" y
71. 
    
72. CString m_slocalIP;
    2 {  \* h/ h" r; N8 C
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    9 q6 e. i; a. T: J3 r! ]! z) C
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   ' {; U: D7 [& C! [! N6 Q
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   
5. 
   0 z% ?9 C  X* U: n! {
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   5 J, O8 c7 G8 R9 U
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    ; _6 j; Z" A3 |6 K1 q6 _
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    + Y/ h9 f5 R/ M' y( N# U6 Q
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    , x$ ?5 W0 [' F
16. const   CString   getString(int   i)
    1 h  c2 M) q* b' s
17. {
    
18. CString   s;
    6 E& D) L  {# S7 W
19. 
    " w& y" f" l- c: P9 T
20. s.Format(_T( "%d "),   i);
    ' j( h" l7 F1 M" w
21. 
    
22. return   s;
    + C1 z2 a( V/ K3 h4 \) l. j  z
23. }
    
24. 
      [( d4 G. X+ R  W
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    # K+ z5 X& i; ~& b4 d8 E, S
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    / g" d2 y$ U* I. o
28. }
    
29. 
    ( _0 l& @( y$ C( `# z& z! i
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    : n: O' k+ }6 G' b1 v% v; g" b3 H3 Q2 A; S
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
      {6 {( i& H/ y, A+ J: G$ O/ I# _0 e
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    ( x$ f; z  B% y9 u; F
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    5 P: R) a0 N; a! p+ d0 e
41. strcpy(buffer,   (const   char*)sa);
    7 ?  B) P# X9 R
42. 
      W% I2 {& o3 K) u, h( y
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    % a. u, n3 i/ y- [; A: a( }
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    7 v; A1 O2 P; a  H% A9 E4 @7 ]
46. sockaddr.sin_family   =   AF_INET;
    . c5 Q" e, w# f  `& T- U
47. sockaddr.sin_port   =   htons(port);
    # o3 h& `$ M2 n* Q4 T. q
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    1 `- O* i: _2 b
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    - }, Y( S: d$ u5 L
53. Sleep(20);
    ) t( b) b) d# V3 T2 u" t' L3 ~! V
54. int   n   =   send(s,   buffer,   length,   0);
    # H" M8 F  o" A2 y
55. Sleep(100);
    9 U: [* r$ k. x5 y) q
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    4 t5 i9 q9 I9 W' K
59. if   (!rlen)   return   false;
    7 V, j4 [9 f* L
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    
64. }
    
65. 
    , Q. n7 q" I5 \9 E' Q
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    7 x6 c  N. @. K8 G8 d: T9 W# j
68. char   buffer[10240];
    8 z8 K1 y5 \' Y% q4 @8 U# H
69. 
    ' W5 l% a# Y$ f" E9 t- e: f; ^' h! o
70. const   CStringA   sa(request);
    ' J  ]; ]# q' W9 {7 C2 J' p
71. int   length   =   sa.GetLength();
    4 ?! p# V! g5 Q2 C1 t3 m1 a' a
72. strcpy(buffer,   (const   char*)sa);
    ! q& k, s6 p- B# {8 R8 b7 i
73. 
    , m# R, @- B; u. |
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    + f5 D$ k" D( g9 C7 f+ p+ ^! M
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    % g. k$ H: l7 T/ E( U
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    7 Q/ ]6 _: Z; M
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    , X+ t$ ?7 a5 w' e
84. {
    
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    / k$ \4 K' H) c
88. 
    
89. result   =   response;
    $ P( J) J. N; y8 C2 K& I
90. result.Delete(0,   pos);
    3 {1 K7 N& Q0 \3 ^( B+ l5 t, A
91. 
    3 A" O' X; K! Y7 U1 b0 {
92. pos   =   0;
    5 x; f: Y5 k6 s
93. status.Tokenize(_T( "   "),   pos);
    / l# v5 B) E! d
94. status   =   status.Tokenize(_T( "   "),   pos);
    6 @# c$ s% z) Z4 k2 }* e4 H
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    + Q' s9 c2 `& ^& c3 G& o
96. return   true;
    
97. }
    . H' r9 w1 l$ I* `/ G# s
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    * r; [% Q) W1 W0 e' b
100. {
     7 h& V- h7 y1 c0 x4 v0 }/ v" |
101. CString   startTag   =   ' < '   +   name   +   '> ';
     ; S- V6 p6 u) A& N( Q+ R
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     2 c1 Y% I% T$ z$ E5 t  U; [
103. CString   property;
     
104. 
     5 \+ P! A& t  h) Y. C
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     / G* c. O9 k8 X" \( ^9 @" q3 v% y) h
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     6 v% {+ |* ]' u3 g3 e! g5 M3 o
109. if   (posStart> =posEnd)   return   CString();
     # f. K# B3 T; N4 A$ `9 A
110. 
       [! p, S2 t% k6 D
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     2 j7 S" @1 q+ u+ N) R
114. MyUPnP::MyUPnP()
     $ h: D1 a* B) U$ {7 p  d: E1 ]0 N
115. :   m_version(1)
     0 r* S. D" F1 M9 u
116. {
     
117. m_uLocalIP   =   0;
     ' Y* V* j0 r0 Z  ?2 L* R3 W
118. isSearched   =   false;
     
119. }
     
120. 
     $ J7 K; J6 y2 X* [8 f- c/ m
121. MyUPnP::~MyUPnP()
     
122. {
     ' {1 A6 E( n- S  X
123. UPNPNAT_MAPPING   search;
     2 z. ^+ W' Q+ g# g! k* T  R6 h2 G
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     ' c# s) V( l4 J1 G* f3 l
125. while(pos){
     $ V- d# K3 h1 W
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     1 B5 O1 M4 C/ [" u
129. 
     
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     
133. 
     - u9 a; N* `7 {) m0 W' j0 {
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     , F& Q6 f7 Q, U  ^4 E3 [6 A$ V
136. if(version <=0)version   =   1;
     % h. Z# N) E4 g
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     ; {! v. q5 _/ g' w$ q0 K9 D
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     1 [: n/ F9 G# S' ?8 ]
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     ! a' h  n) I5 K$ t4 z8 |- _
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     ( V3 {* T6 x. F8 N/ F3 p1 t8 V7 T
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     , Q) s" u7 C: ]# U# Z0 V
150. int   rlen   =   0;
     5 t0 c5 K# N' g: ~7 s7 ?
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     - h8 {+ L- e8 Z7 Q  _
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     8 q" Q8 y3 R7 H6 W1 r8 i/ R. 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 "),
     - Q0 n' R$ d  ?# K
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     5 h; f5 @6 y9 ]- W* F8 V
160. }
     * D% G# H7 A- f" b7 k: ~+ |+ r
161. 
     
162. Sleep(10);
     
163. 
     ) W/ S5 {" X' F- p. S
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     ! x6 W6 ]2 _' G( {
166. if   (rlen   <=   0)   continue;
     ! g! @1 e5 m0 O' B! w' E' ^
167. closesocket(s);
     ; Y& ?7 k/ V6 n/ v& @* y
168. 
     9 K! C% T$ b, W) ?5 d
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     # r: c) V1 ?; u/ z4 g* Q. C2 }
171. if   (!parseHTTPResponse(response,   result))   return   false;
     ; S: J  V/ y/ [8 V" u) z( E
172. 
     6 m, w$ R, I, d# W5 f* l
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)   {
     3 o7 r! s' f  n5 m, Z; {7 K* Y
176. for   (int   pos   =   0;;)   {
     ! L4 `, W, w- h, \) v" y
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     0 y, k6 U1 z# y( q4 U
178. if   (line.IsEmpty())   return   false;
     ! J) `' s  @$ C" U7 o" c! n$ X
179. CString   name   =   line.Mid(0,   9);
     7 e+ E6 a) y8 ?" s& \, C
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     1 a  l. A0 k  @
182. line.Delete(0,   9);
     4 F& R8 e. D9 w+ C
183. m_description   =   line;
     6 B% O! [# s% c) }% B
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     
187. }
     
188. }
     ( K  g6 u8 V% H0 {2 I
189. }
     9 j+ H2 ~- e4 j% h" Y: M3 N& \
190. }
     
191. closesocket(s);
     4 g* x$ Z+ ~$ p8 ]; @8 C* A
192. 
     
193. return   false;
     4 q% w/ ]; E# n
194. }
     8 u2 b" E9 I! h; ~* S' Q5 L

复制代码

天云上人

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


///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
7 u6 f; T0 ~* f& W0 z- J

: R& v% D& M: H% L" q  l#pragma once
7 E1 G9 D$ e" H5 A! |#include <exception>


  enum TRISTATE{
* s" u* K4 r9 `3 P: T        TRIS_FALSE,
        TRIS_UNKNOWN,
5 m( n/ C  V0 i! p% P  Z5 P        TRIS_TRUE
/ r* \$ U5 ^8 m+ Q, j$ t};


- ]$ V' V0 ?$ M1 a# B0 oenum UPNP_IMPLEMENTATION{
/ w6 X$ d0 F  I- i0 t        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
3 m. X' b6 W, v! }        UPNP_IMPL_NONE /*last*/
};
3 D& f4 V8 W! ?* ^* @
; T, i/ {; z6 q/ [! h8 r* k' f

$ A0 b. P" v! ?  U5 q1 ~' x
class CUPnPImpl
5 F) C1 T8 D4 t+ y3 x5 U; M+ R{
# K2 D/ I1 q: R1 Fpublic:
        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
5 b; ^/ C& v1 q: F4 S$ e1 p* I        enum {
                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
$ ?7 r- W8 t6 v; k        };


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
+ X' x! h9 E9 F9 V9 {6 D/ F3 {0 f        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
4 x% P$ ?2 H& {# X# l' a) v1 ?        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
+ f+ |, |0 {6 p; s8 M7 r        virtual int                GetImplementationID() = 0;
       
% h& i6 b6 \. d' D: M, X        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
9 u. [* _1 u! s; z, V% [2 r        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
" j$ p. y; i( p, A' N& k2 i

// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
6 r& y, m; @  T2 \7 H        uint16                                m_nUDPPort;
0 p$ t. j. Z: \        uint16                                m_nTCPPort;
! P  X( K3 g5 r' @7 v        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
3 A& [2 G5 d$ S# g2 W
' m* X, x7 ?4 ~. P+ K; e: W; F
# `' y" n1 L7 R$ `3 W- @& W& ?private:
        HWND        m_hResultMessageWindow;
4 |( W6 c% f% B/ _' Q  q# h. K        UINT        m_nResultMessageID;
5 G  o, s( u" O) {) H" M! G

  g( j* a2 O# I};
0 {+ b3 I9 Y; J) y, A* Z5 v4 N

// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
; A* Q+ w* j- O5 u6 L        virtual bool        CheckAndRefresh()                                                                                { return false; }
) g' A# |% a, e        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
1 k9 Z) D7 }! r        virtual bool        IsReady()                                                                                                { return false; }
" F9 {1 O5 h' o* S) I* u        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};

# u' z  b) c' G/ a4 a) O/ d
% n5 z( x; y9 q/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类


#pragma once
+ {/ b8 [6 J6 f1 p#pragma warning( disable: 4355 )


+ ~/ D) l$ ~, c; H9 [2 D1 _#include "UPnPImpl.h"
#include <upnp.h>
5 W3 S3 H* z) m1 `#include <iphlpapi.h>
/ E4 i1 P- y. E$ m( u( v#include <comdef.h>
#include <winsvc.h>

- Q; Z. S8 o" U) w7 T% P
#include <vector>
#include <exception>
#include <functional>


1 W" I" r4 v1 ?9 O7 ~. J

typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
* `6 n" z; `2 w5 Ttypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
4 F8 e1 f: ?' F1 Z( ]: Etypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
5 s4 t$ m" I. J" ]( O6 k9 Ltypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
0 y( m' w* x3 c8 F! G' j0 ptypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


typedef DWORD (WINAPI* TGetBestInterface) (
' f1 B) P( k  X  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
/ {2 ^5 J& N! b: ~8 E* f$ ^);

6 z3 w2 g( x  z3 q
typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
);
0 l* o5 X9 D8 I. B4 j( q+ w
1 V( @- V5 T' p
. t6 R1 |1 y' M+ ]" h# r8 Ttypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
! W9 N% k5 W! x+ x0 J. M);


) G% K7 n* [5 J( G; w9 sCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
' ^5 i: E. ~" S$ Y

, a9 I: I2 j5 sclass CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
- l4 e- J0 K; A. W. r5 \7 w2 m8 \public:
        virtual ~CUPnPImplWinServ();
5 ]( e9 ?- S4 i- W$ [8 l+ x! x        CUPnPImplWinServ();
' a! w6 E7 C9 i$ j1 I5 H0 I, E
" f- b* H% @# ?  h0 \
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
( t8 y5 X# u. g5 Y: c  m8 H. ~8 `. l        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
1 @  O# [4 _7 |        virtual bool        IsReady();
; F1 W/ U8 R1 ]        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


        // 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
0 F3 O: J0 [9 {( @( P0 `$ A        virtual bool        CheckAndRefresh()                                                                                { return false; };
* T: d  w. d4 C& ?
* ]' B1 T$ z/ l/ Y  T
protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
& p7 ]6 }4 A3 b& ?8 w2 g! J5 @        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
( A5 y/ K- R% [3 Z6 a1 f9 A        void        RemoveDevice(CComBSTR bsUDN);
, w' f4 r$ S+ n8 |8 @+ [& c/ a! k0 h        bool        OnSearchComplete();
) C; x& c( i: q+ P. Q        void        Init();

8 \- p5 f0 d& N  E- d+ N$ m* X; M
& T8 U5 P+ b" u2 u& k5 U        inline bool IsAsyncFindRunning()
8 r! C; P7 n: V9 p$ Z* g        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
; ?! S/ M: f/ t& r                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
7 R) M* l# \0 F1 n) p/ ^                MSG msg;
8 y+ y% n5 \& G+ @! S; X0 g. v( r( C                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
3 U1 d! o' ^: M0 c$ q                {
                        TranslateMessage( &msg );
1 h; f* g6 C6 a$ e& y' y                        DispatchMessage( &msg );
' D3 x  y" Q1 a" X  T                }
                return m_bAsyncFindRunning;
) O" ^0 k9 s6 S' l" T        }
4 V4 h7 x, w, C2 F

        TRISTATE                        m_bUPnPDeviceConnected;
7 Q( T0 l/ y8 S) x
0 A, y8 Q3 `% m9 T: y* i
/ Q( q* R0 x) u9 d4 ^, |+ N7 k) ?7 `5 N// Implementation
        // API functions
! D/ i6 U: o/ Y: ]' ^        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);
: J; I- x- ]2 R& o; e1 e        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
! Y) m/ T2 ^) H. O        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
6 W& C. Z& `; \$ q# B

        TGetBestInterface                m_pfGetBestInterface;
" v$ q" s) \6 b1 G6 o        TGetIpAddrTable                        m_pfGetIpAddrTable;
& J" i  g" K6 u        TGetIfEntry                                m_pfGetIfEntry;

# A# n- F0 p9 \, S: I
        static FinderPointer CreateFinderInstance();
- G1 n3 X$ l7 S' g! N9 Y        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
6 N; [+ e7 O+ ]5 U& n; q$ v: T% w$ h                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
: i- e9 [1 y+ r  r; h% V                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
$ n6 w0 @1 Z) l) y
9 w0 V: `1 J, T
0 i5 A) n1 n" \* _                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;


                        return wcscmp( deviceName.m_str, m_udn ) == 0;
2 h! n! J- t+ N, j& l+ f& Y                }
                CComBSTR m_udn;
1 ]) o/ W8 h, o2 ^+ h        };
2 m' g) G( P: L% @% [5 c/ A       
* K2 Q, h1 s% x. D! c9 V        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
# ?, D- @4 }1 H7 l4 X4 i        void        StartPortMapping();
5 h- o! p% N8 V5 j7 l        HRESULT        MapPort(const ServicePointer& service);
# _0 b2 u8 ^# y% y* V: t, d: _        void        DeleteExistingPortMappings(ServicePointer pService);
/ w3 A2 E- e5 X/ q3 c        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
7 L# Z% h( P& s' y) a& d                LPCTSTR pszInArgString, CString& strResult);
1 [; v& j0 w0 i5 H. R        void        StopUPnPService();
3 ~, c9 M' @! ]8 _; ~/ R! M8 \
- p9 n7 ]$ [7 |& ~; z+ I) ?
* `# ?- B% F- t" [        // Utility functions
- v/ I5 x9 s! d5 `$ ]* a        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
7 v( Y& k5 F- t; n5 v        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
9 ?7 L+ ?" ^% g% ^9 Y8 e  ?4 a        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
5 e- Y+ N3 u1 b+ y        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
7 O$ Q1 u6 h3 M7 N        CString        GetLocalRoutableIP(ServicePointer pService);


// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
( t7 X) H: r% Q7 t. [( N* D$ L' c        DeviceFinderCallback        m_pDeviceFinderCallback;
" I& u$ C( X7 J        ServiceCallback                        m_pServiceCallback;
) {. w( G0 W4 b* c1 q
7 M+ L# Y1 W7 ?$ C8 g; S
$ t, A& A" G2 L8 A/ a$ `5 N* Z. }        LONG        m_nAsyncFindHandle;
5 k% X( f0 o$ H- W        bool        m_bCOM;
- f7 D) J) b( }2 f) d        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
! q  T" a$ J7 T' T' s: T        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
" f/ h$ m3 Q; Q. q* |$ R% o7 b1 K        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
4 f( G0 e- k9 y' R# W        HMODULE        m_hIPHLPAPI_DLL;
2 Z9 G3 A5 [' k) E# t1 ^        bool        m_bSecondTry;
3 y  w) b& ]6 x        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
" E0 u- ?( y* @  s' e5 ]# Z- ?        bool        m_bDisableWANPPPSetup;
/ g- C! F( T  A) E/ I7 V0 l. a  }
" f3 ~7 }2 ], @7 S5 d
};
. Q+ v/ ~7 ]" z. \6 O- r$ V% A

3 `  M' ~+ z; R; U5 x" g3 x// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
# U5 D4 _2 a3 T! a$ e: u                : m_instance( instance )
( T$ `' ]6 i! }9 f        { m_lRefCount = 0; }
( x* ]9 k, N' C9 Q' U8 W: v' H3 z" K, @
$ l+ C" O  Z! J8 j
* ]4 u/ k! a2 P! U0 P! p# n   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
, t! C- Q* ^: c: G& B. S$ {# [   STDMETHODIMP_(ULONG) Release();
- n3 x/ d9 Q) ]6 P& ^

2 N6 Q1 {% I) D6 t// implementation
private:
6 ]( E) [' o& N/ H$ L; F& I# V, ^        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
5 k1 t8 H# `& K3 z        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
  d+ r( v. i8 @) |        HRESULT __stdcall SearchComplete(LONG nFindData);

" a9 S3 T7 l% F
private:
7 {$ W% P9 l7 d8 ~        CUPnPImplWinServ& m_instance;
+ r+ H* ^! u% ^: L6 h5 F        LONG m_lRefCount;
};
7 D! u5 r9 R( w+ r- r; Y# c# K$ l8 L

4 O( K3 P1 K9 z2 p  A6 `& |& q// Service Callback
class CServiceCallback
2 t; X2 w, l* i5 y" V1 a        : public IUPnPServiceCallback
) M3 d' o' N$ A/ \' N% ^{
public:
! O9 I" M( E: I- T        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
! R; l# h9 S; y   
: F0 D8 ~! @: }( v$ M   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
0 X# Y& [7 X$ a$ Q/ L, N% x# H" x   STDMETHODIMP_(ULONG) Release();
5 ^; s* L. E! e4 o1 ]
0 I0 `+ w6 Z( g- J
// implementation
* c: H) W. u. T3 I) d) w+ mprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


private:
( s  l* A( I' {9 j" S5 @0 S, @0 C        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
  ?+ H/ F" |, \};


/////////////////////////////////////////////////
+ V3 {8 \, a& Q( R0 i$ p
) a& N0 }* q8 n
7 Y7 S* T% \4 s3 B! o( R- C' g使用时只需要使用抽象类的接口。
9 S6 a2 s7 S1 V7 z. ICUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
  ^, N5 ?+ ~4 UCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
8 p" O& ]0 X, bCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.