UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   - u+ R/ A0 N  M- w+ y, ?7 Z8 {
3. 
   
4. #pragma   once
   / l9 S$ y: I$ A! t: k; g) L
5. 
   5 D( r4 V% ]& C$ U
6. typedef   unsigned   long   ulong;
   
7. 
   - c4 |4 g3 n' a! f; W
8. class   MyUPnP
   7 U) t% ~6 p: t7 @6 _4 m
9. {
   / c& Z* _+ m" V6 D  T
10. public:
      B  m! Z! c% U3 y' D, v7 D1 a7 ?
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    % J3 y8 N0 j: _1 b7 G
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    ; `9 F: D: e: c: k5 c  J+ {# D
21. UNAT_UDP //   UDP   Protocol
    2 r8 J; c  W: a- T! c
22. }   UPNPNAT_PROTOCOL;
    
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)
    $ A  L1 ?! q4 o; e2 G$ Q7 Z
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    
31. MyUPnP();
    $ B/ K* R! U- F5 w
32. ~MyUPnP();
    
33. 
    6 F5 E, d! ]% l( A1 l: N* ]. W& ]
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    ; F6 T0 p6 X( a3 i8 J/ j: b8 [
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    ! Z9 R8 t2 I& M6 @2 u
36. void   clearNATPortMapping();
    , j5 Y5 O4 K2 \  s+ ^4 c, K
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    % T6 K4 l/ e% x" ~0 P3 ^
40. WORD GetLocalIP();
    + y$ b8 i) F2 _  k( f( l) i& s# ~
41. bool IsLANIP(WORD   nIP);
    ) q% l5 K* Y' Q6 v3 f
42. 
    
43. protected:
    ' A0 J. l0 v; g
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    ( O) [# |( E* a5 r% x* N
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    9 x/ H: s3 L3 D
48.       const   CString&   descri,   const   CString&   type);
    % p/ t0 q, q5 b& Y: A7 K
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    4 G) `+ N+ E$ ~
53. bool Search(int   version=1);
    
54. bool GetDescription();
    / J0 [+ t1 X; k; s# e
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);
    . \/ S* d* \% ^& I- x
60. CString m_devicename;
    - |- ?6 i! U6 @
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    $ m7 v# u0 B$ c7 u: x4 \
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    0 ~" M6 o! e( G( P6 Z  G
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    . x/ y  `. ?4 V
72. CString m_slocalIP;
    - b! l: O" m/ i; n- I
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    # _6 h6 G: r. A( C4 U/ P: i
75. 
    
76. bool isSearched;
    ; t6 @) G" ^- ~; p) t: |
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   ) {* L: }. I7 c' f
2. #include   "stdafx.h "
   
3. 
   * z3 L) b( F" L% |- q2 t' J
4. #include   "upnp.h "
   ' f# }' i8 P  W; D& z2 @: X
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. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    / L' H+ `% j8 F6 k/ L0 H; y  P
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    7 [, i, V- a$ t
15. 
    " _9 q% H; P. O2 B
16. const   CString   getString(int   i)
    
17. {
    - U3 ]# S5 d# I: O) h4 E0 Z
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    8 T  x; \8 @2 n* X% j& e. A
21. 
    
22. return   s;
    
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    * M+ F2 n& l- H' r' z' G# V
26. {
    ; O' t1 Z* R5 l5 A4 G
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    ) {3 y4 K! K' @! w1 T7 o
28. }
    9 P6 P( j5 x6 k9 X( b& u- _+ \5 M: Q
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    4 {; b" K! o' G# Q' F3 d
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    6 T+ L: B! p2 M* M/ q
33. }
    
34. 
    6 b$ e$ |7 Z( ~' f  l3 H/ v
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
      P  v5 @) q5 ~) Y4 G
36. {
    
37. char   buffer[10240];
    ( R, ^( t8 d8 F* S' P! O0 u
38. 
    9 r* m$ [- z. d
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    / m8 s5 p+ ^) U; g6 t7 J
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    4 ~( F% M3 U; E. b  t# c
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    . D4 s4 ^, A$ I1 x
46. sockaddr.sin_family   =   AF_INET;
    # R; E. Z% }- ^, {/ a. h( N
47. sockaddr.sin_port   =   htons(port);
    $ x3 V0 G. [. n2 i: c
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    ; H! p2 d" |% |4 R
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    7 k  m( p6 `4 q
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    + V/ E; ]. A# K9 w- ]' t  }2 I
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    ! o+ n0 u0 `* F5 _  S, i
55. Sleep(100);
    + S. c7 q0 [  }0 x
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    ' l# [% g2 |8 X, y5 {
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    : r  `# o& a+ {% I- B- p
62. 
    2 ^! U/ i1 p" i
63. return   true;
    
64. }
    
65. 
    $ {0 T% i8 O( R8 A1 P
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    4 X# V/ ~/ p- U) o
67. {
    * P1 p" q% |+ ]
68. char   buffer[10240];
    . O$ Z6 O! O) ]% x$ L: T
69. 
    
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    $ l# K$ e# b6 `5 C  t1 n
73. 
      b  J1 t3 C. B# X
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    % s( e9 w  U( U; t! E4 X
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    # ]1 B$ b& |# ?: P
79. 
    , r( Q* J: f4 `
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    * z7 s6 u0 @& a' U* j
82. 
    3 z2 m8 p4 o8 X& ]4 r
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    6 h4 O: p$ d% G4 n  h
84. {
    
85. int   pos   =   0;
    ! s( R3 F3 _# f9 b& B
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    % g, {( _0 @/ Y( k, Y( ]
91. 
    , L5 h# @1 M( I# w) f, K
92. pos   =   0;
    5 m" R6 {" `' P9 o- \; |. e5 n! o
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     8 m& d* @" R: V6 d5 U5 _- a0 E
101. CString   startTag   =   ' < '   +   name   +   '> ';
     ; U; G+ z- E# \; O8 E, W6 A
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     / r$ r: n6 ?& {
103. CString   property;
     ) E+ V) b" o; d7 a. i
104. 
     , ]1 `9 t+ z: e7 ~2 V( r; Q1 d
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     # O$ D3 m3 c" z: H
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     ) Z" O9 Q7 f9 ]8 Q' {3 u7 |3 _/ y
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     % i6 i0 \& R: F
112. }
     0 J) ]- L0 x4 i( A
113. 
     : W3 L+ u* a3 E% a! {, {
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     $ U8 o/ |) u1 k4 b6 |% M
116. {
     
117. m_uLocalIP   =   0;
     / S6 b7 I* e" Q3 I. [
118. isSearched   =   false;
     
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     1 B+ y+ ^1 ~2 S/ ?' ^7 Y: b
123. UPNPNAT_MAPPING   search;
     , N& p$ U( n5 W4 C
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     * ]6 P3 a. W* [: k
129. 
     
130. m_Mappings.RemoveAll();
     5 f, E8 I, E9 f. _0 ~
131. }
     
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     , }% [/ d0 P# G6 P6 D
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     3 S; Q- W8 L5 w0 u$ P2 f: j
138. 
     
139. #define   NUMBEROFDEVICES 2
     " e# k8 D! A" a- b: L$ ?1 \; i8 `3 `
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     9 x$ w' P+ l. L! P- H
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     ) B& h# S2 {& f2 k' B3 v  Z
144. };
     1 A# ]& @& j3 X* H2 H7 v% i/ d0 x' R
145. 
     ! m2 `7 q0 G# h1 ?
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     + A! x& ]# Q! P3 v
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     # \2 x  e& X/ k( m" J
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;
     / k% v9 q' K% R, I
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);
     & r7 {" H2 I, a
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     % R/ E) C7 g' p, Z
159. }
     
160. }
     
161. 
     
162. Sleep(10);
     1 |$ M5 V0 a* n$ J
163. 
     
164. char   buffer[10240];
     1 Z6 ]6 l# X% j( L  C4 N; l7 ?
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     0 h5 |# y, v+ ]. ~
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     ! w& v3 X  ~9 f& X) Z) P
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     8 j! |0 n) d* V& ^5 N+ Q3 F4 v0 O4 J  v& e
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     % V) W: u( ], L9 d; J
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     1 R1 _; H0 Y" @4 m# Y5 |% b$ E
180. name.MakeUpper();
     0 a8 l( h+ L5 x( q
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     # s- Z4 }7 l" C, M/ b, M! \
183. m_description   =   line;
     / S8 |6 g3 e) i' n5 T$ w1 i4 |
184. m_description.Trim();
     ( s. Q# a1 w6 ?2 C5 k
185. return   GetDescription();
     
186. }
     
187. }
     5 h- ?6 y( B3 g
188. }
     8 @' Q. D% [% d
189. }
     2 q) X+ P+ `; m/ Y% O
190. }
     
191. closesocket(s);
     3 o* Q2 v. K% C" J! q. m$ _
192. 
     , f5 n7 u& }9 d' Z, ]) u% z. G4 D
193. return   false;
     
194. }
     % S" v! V2 |5 |* j$ O; N

复制代码

天云上人

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

2 |- M5 D: \6 U6 E8 j7 S
///////////////////////////////////////////
* c9 t1 {2 v) H- T3 p8 {! ?//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
) z: ^$ P# Z* `( I6 B' s
# U5 [' p2 i% c9 J+ i
' ?' p! _( f" `5 W$ k7 N3 C#pragma once
. [" A8 B  Y9 O( D& q#include <exception>
% I2 J7 b# ~+ l) {6 p8 c1 _! G
" d4 ?1 |4 K' _7 d, V/ o0 `
  enum TRISTATE{
        TRIS_FALSE,
" A" ~* _0 O" a/ d$ T7 Z) I( i5 @6 z        TRIS_UNKNOWN,
        TRIS_TRUE
};
) K0 J$ C- Q+ b/ o& d" J6 `9 ^/ a+ K
# O$ Z+ c" V  g0 C7 ]5 A
enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
- P$ |' v5 m( H# H# S4 b; U        UPNP_IMPL_NONE /*last*/
};


* n' y9 X! B) a0 h% k

class CUPnPImpl
0 I6 Y3 P  z! l" C5 j' ]( h{
6 h( l4 S. C0 _8 i% g" a4 kpublic:
        CUPnPImpl();
- `( t2 W: B7 {$ L  a/ i        virtual ~CUPnPImpl();
% P7 l" Z% h! k) y        struct UPnPError : std::exception {};
, ?, ~4 V  ?  i5 @$ _        enum {
                UPNP_OK,
' M# @" \3 L2 O9 J! V                UPNP_FAILED,
5 ?. Y. U: |0 e; O0 ~5 q7 M/ ^                UPNP_TIMEOUT
4 Y: H. m' c% t6 e        };

1 e+ O. s) B" u6 K
" U; ?1 y2 g4 J/ v& a. s% N        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
- j: h  |1 }5 O        virtual void        DeletePorts() = 0;
4 l* B. h  g! ]- U& a& K; {$ U, p% e        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
, L& I- u! U3 G& ]4 j( H        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

8 }$ ]2 B7 |' B, G8 q5 \4 a
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
9 c! e& n6 r; b3 x5 L0 Q' g        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
6 ^7 X8 r5 Q3 O5 y9 ^' K
3 ^8 p- u$ k7 P! f
# l. _4 l5 c7 h) A" s// Implementation
protected:
( I/ J1 w; K9 [, U' B        volatile TRISTATE        m_bUPnPPortsForwarded;
7 F7 \* t9 v+ y: k        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;


8 v: N# t. }0 t& b$ r% {, Gprivate:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
6 G/ I( G4 p/ u1 H4 t3 E& A# w

};

) B' f* \( ~0 ^6 u2 ?% j
3 V- a  [  R% q4 r' W// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
/ A7 A7 C1 l9 p& v: s5 n$ a{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
1 v8 z: ]8 O7 y7 M        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
2 V. H1 p2 x0 S; L4 V        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
9 b, C% [1 K: e1 r" t        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
' \+ R! l: r) d4 Q$ e) N};
9 M/ k- _( {, a% p; s% p
$ C- n) f1 V5 a
/////////////////////////////////////
# s( _: G& H1 F! G. U& J//下面是使用windows操作系统自带的UPNP功能的子类
0 m" G: L+ J6 q, X8 W3 S9 l
7 A- Y& C5 g  x0 ^+ D
#pragma once
#pragma warning( disable: 4355 )


& c' R' c; e  R& `3 O/ F#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
, ~6 k$ Y% `! f0 |+ H* d#include <winsvc.h>

  [+ o2 j, H) G. [' ]8 p
#include <vector>
#include <exception>
#include <functional>
6 p3 J# }8 _1 O

. l; t$ W1 h3 b& i1 ~) b" P( `  P
* V! J( `8 u% F
4 s1 u  S0 j0 T5 R0 Itypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
& A  z, _( R' N: l  O5 [6 S) |typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
9 x( r  a+ G, W: Ttypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
9 g) A0 {/ u, v( ~9 ]typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
$ x! m5 _+ B- utypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
! H. T/ Y; h* b4 Z1 l$ Htypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
5 `0 ~3 J8 }: t* j/ ~: ^* u- }  rtypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

4 R' @* {: L( J  n9 f3 C/ u
typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
% _4 `4 I+ A7 w);
' _# L* {, W0 z; C- Y+ q, j
& S6 R2 j6 h3 ]( `3 y( H( a" C
% t( c  D7 [0 H+ ltypedef DWORD (WINAPI* TGetIpAddrTable) (
2 U. e# ^, R) M1 L3 g  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
);

4 q, @" k) b" t4 l( H0 s. _9 u
: d3 j" F* ~% \6 y7 v; Rtypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);

; B# R8 U. ]: r1 S; C
4 r& r4 ]. b3 ]7 F5 b& n! r# ZCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);


class CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
public:
        virtual ~CUPnPImplWinServ();
7 l7 S4 f  ?$ e: ]  Q1 V        CUPnPImplWinServ();


3 S4 h0 @6 K4 {- n7 Q9 k1 J' Q' C: j        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

/ D6 o& k" v, `/ e& ]
2 X" o% _! f0 s" f, }        // 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; };
- x( C" Z" _; p3 b

* I5 g1 B7 ?4 E/ o& |* ]5 hprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
' E0 t/ U2 X% {' Q* Z        void        Init();

, R. e  x& K; H' {& D0 }
3 _* _! z# s% f4 }: @9 J% X        inline bool IsAsyncFindRunning()
        {
8 I6 ~! A" N$ C1 W% e6 p& c9 L1 [( V                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
: L) [* W) i, c* q                {
6 p* E, c  K" X3 j4 s6 U' i                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
+ a2 a+ o. \2 y                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
$ o/ `- n: Z3 f                }
+ s: S0 Q- X. W                return m_bAsyncFindRunning;
        }

9 E# f, r0 X2 b0 X" w
% M* _4 N$ i4 F3 j        TRISTATE                        m_bUPnPDeviceConnected;


' @! P/ ~- \* ^$ C$ U2 [// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
# Y* c5 b/ X8 v" m! N        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
$ e2 f' H5 ]+ h; _' z        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
* y( t3 t$ d' q8 p9 y/ [        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
$ F/ ]3 r# K: {8 X; q7 o5 j

/ x, f, {* ?  T/ L/ o) |9 R        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
; J) J2 e5 W6 J; E8 H1 R

        static FinderPointer CreateFinderInstance();
0 P$ y; I! @% d; h" p        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
) L) b5 [, d  X) Q! M                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
% x  ]! V7 U5 R; m0 z- ?) r. O                result_type operator()(argument_type device) const
! m1 @# a1 T  S6 |+ f+ ^5 |" i( F                {
                        CComBSTR deviceName;
0 U; B4 X- W  c                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
& @. d# q0 w( r+ C1 M; A# h

0 O7 [3 j$ E1 g9 }" G" T* D                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
" V+ q7 `3 v+ O4 V$ v
+ s% R1 X1 X2 q( p5 w8 Q0 ~6 O) s
! y4 x" G7 {1 c6 O                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
4 h& |( a  r4 B$ j( V6 k- ?' H3 |                CComBSTR m_udn;
        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
2 f! ^  A& Y% N1 Z6 G9 i! I1 S4 E        HRESULT        MapPort(const ServicePointer& service);
" e9 X8 @0 X$ s4 t8 k        void        DeleteExistingPortMappings(ServicePointer pService);
3 H5 l( {2 }* W7 a3 G: _        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
8 z6 i( _; T$ z) `        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();

  O. B- ^' w- F$ C7 E6 |1 D7 h
$ e) D+ {7 S  R0 t8 Z        // Utility functions
$ t+ j. F3 W, n& y; v        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
  K* c0 |5 U: i: f2 ]3 X        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
" ^0 ^8 E' R& m7 A9 v( o8 z        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        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);
; j. |; l2 o6 y0 p" {$ O

8 c* i3 D, R# N2 W2 @* O// Private members
private:
1 B' n" @' m/ j( l: Y3 w        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
/ n' C8 W% i1 b- A" C* _        std::vector< ServicePointer > m_pServices;
1 ]% N- H7 `9 A" O+ `' G: b        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
6 `  x2 p) B' n* G0 ~+ c, k        ServiceCallback                        m_pServiceCallback;


        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
" p3 L* W( e" r: R        CString m_sLocalIP;
  A& b) \2 e$ m1 K7 U, d  b  O( ^        CString m_sExternalIP;
. g  P% A- @, \' ]4 L        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
8 t4 x  C4 A5 D        bool        m_bInited;
# J* `3 U% U, b+ R( O( J3 M        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
0 |1 r9 E: T5 X2 i" v: d        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
( A5 k0 t- A9 x5 c6 A2 o
8 K$ J- t+ c5 l6 {! _" R0 v
8 J. _# e6 D9 J: L2 ~$ t};
6 S* i$ a, w$ X% i9 y4 W
+ T5 j' n* X% l: S) ]2 [
* R7 d) J( B7 X& s; }6 T// DeviceFinder Callback
( h" [# `1 \5 r# fclass CDeviceFinderCallback
3 }8 \! _% m/ X5 H+ Q        : public IUPnPDeviceFinderCallback
{
) w& X9 N, e; V; F' a/ B4 ?8 Epublic:
! @+ x9 d: x1 |1 g& ?% T" e        CDeviceFinderCallback(CUPnPImplWinServ& instance)
. c( A( e* N# A! }2 E4 K9 d                : m_instance( instance )
        { m_lRefCount = 0; }

' c5 o+ y; e9 t6 j- @: ^1 m  l
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
; B0 F% k# Z1 w) ]2 h   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

; r# x, P0 d  X* C
// implementation
; a; M. v3 h/ L) `$ h$ Zprivate:
( H$ c7 U: j; \4 T/ b( g  S3 ^        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
- T2 l2 P: J4 w        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
- J3 ~3 h% a: q% T

private:
& r- J% P* {# B4 Z        CUPnPImplWinServ& m_instance;
# W# e* n1 l, u        LONG m_lRefCount;
};


; ]7 w+ n/ p" k( [( B8 Q// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
{
- |0 p# \( P7 R( q: Zpublic:
9 o* K! C% N" v% Z- Y/ N! e$ V# p& w        CServiceCallback(CUPnPImplWinServ& instance)
- P8 D  l2 z, r                : m_instance( instance )
; \; e2 s; H, n+ l+ L: h* [        { m_lRefCount = 0; }
4 @1 F' M4 C! y7 X1 r   
1 d6 @5 }1 Q6 I6 P0 V   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
- y2 U% H  w5 P8 s  k5 R( n   STDMETHODIMP_(ULONG) Release();

* F: W7 ]: D: G: ~5 h1 \" q
) u6 N9 N9 t' h6 g) T0 s// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
8 p$ C7 ]+ t" U3 V1 x

( }4 T, J" }! l8 q0 }6 iprivate:
4 a, W  k! {! I, L        CUPnPImplWinServ& m_instance;
0 ^% _( b! J- X; V- r6 G        LONG m_lRefCount;
};
8 r: I% l! l; H$ X+ m8 j. m7 i) y
& D( ^  T: V% l
/////////////////////////////////////////////////
$ l) a% b8 t5 L' E' ?
2 N% a: ?3 R' B( e/ J
使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
4 @$ O8 M0 @; }( VCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
% G, i6 c; ?7 A' J$ m! i! h: VCUPnPImpl::StopAsyncFind停止设备查找.
4 M/ ]( U6 I! b) [& G# Q5 U9 n* e( mCUPnPImpl::DeletePorts删除端口映射.