UPnP

天云上人

/*uPnP.h*/

1. 
   9 W9 ?8 @  {0 e" Y: H
2. #ifndef   MYUPNP_H_
   ; d7 Q# E0 W0 j# e2 ?
3. 
   ( a! {* S( L# F- m. {3 K6 M
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   , i: D- q" n  w9 V" K
7. 
   * D5 k- l2 G) I8 H. q
8. class   MyUPnP
   5 N/ j( M/ f  t& H
9. {
   
10. public:
    
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
    & o$ P1 r6 y, _9 J9 i( F1 g* {
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;
    ( O, C& d) g! n0 I: \# m% A
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    - E$ f& Q3 ?3 V1 U% Z, H; C$ V
21. UNAT_UDP //   UDP   Protocol
    ) Z0 J1 F! z' a  o# v
22. }   UPNPNAT_PROTOCOL;
    
23. 
    " g) N9 |9 Y/ d# h' y# e$ ]+ b3 g
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    % N8 I3 y) ?/ y9 U
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    - Z) [" V. g) r2 W) ]
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    : }6 E1 d0 o( ]9 @
33. 
    $ f+ K8 [% q' R5 L' k% H5 l9 R
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    1 e1 `$ j; a; a! Z2 ?2 s
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    6 O* H4 \6 e) M) N
43. protected:
    5 _( x; @, ]7 L, T
44. void InitLocalIP();
    1 K( ~' Y1 W: ~$ Y) _0 v. d
45. void SetLastError(CString   error);
    
46. 
    
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);
    
50. 
    7 B8 v2 \! v/ Q2 {7 T
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    5 _% _: i2 T; V' [7 E
52. 
    
53. bool Search(int   version=1);
    ! L, I; K9 x4 {: K! v
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    : T5 @* G$ v9 I
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);
      |0 l, Y! C% p! S. U
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
    : s( D# k" ]+ u# y& l2 y+ J/ y
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    & v2 c( M5 Q, ]) [
68. 
    
69. private:
    ; R, f  i! e- Y/ n" d
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    8 m5 R: k8 @: [. T( E0 B4 S
72. CString m_slocalIP;
    $ p0 r$ p# n' ^
73. CString m_slastError;
    2 y# n+ L+ o& [3 |: s3 F# F
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    % T: i- t" S1 Y3 P
77. };
    ! ^' F: F3 D: g# Y- Q7 Q
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   ' [. r7 `( a: h+ C+ w6 S# R
2. #include   "stdafx.h "
   / a- ?) U) A! |4 _( Q* U; @
3. 
   
4. #include   "upnp.h "
   7 V* K( S$ x8 R
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   2 E* h* ?# j3 g. c3 t2 x
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   3 [; `# I2 T6 [  X, }4 m0 E
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    $ n0 S2 {/ Y( v
11. 
    0 ?  i% u! w, E* l' }1 }0 O1 h1 V$ F
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    ' S; c! q$ N" N% }; P2 E
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    
17. {
    , ]" j+ ~4 Q& x$ W# _
18. CString   s;
    
19. 
    6 ^4 S+ u$ G3 Q' P; z
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    
23. }
    
24. 
    9 [9 R. D7 C6 q  o* n
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    0 s  h$ x- J# ~/ r- M- J
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    # C) Z9 N' ^+ A5 \+ Q
28. }
    
29. 
    # a8 V: b6 o. @9 B+ L) }, W# D
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    / j6 b1 l! Z; `% b: I
33. }
    
34. 
    8 N, S7 w8 \- J6 K7 I
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    * Q7 x! v6 s' _" U8 P
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    * R$ ^' S2 L5 o) L9 p
43. uint32   ip   =   inet_addr(CStringA(addr));
    - z7 s" p$ h* S$ Z" c5 x
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    ( \+ W# }( m- X; k" n1 m9 x  v
46. sockaddr.sin_family   =   AF_INET;
    " y, h. Q7 ]. R9 |3 ~/ i
47. sockaddr.sin_port   =   htons(port);
    0 h# ]" o" M. B* p
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    8 r% }7 ?  g2 R7 g( i
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    ' e7 ]' _, H9 d; s$ W1 H& ?6 P
50. u_long   lv   =   1;
    3 X9 P$ M; w4 A8 _9 B2 }6 M
51. ioctlsocket(s,   FIONBIO,   &lv);
    ' L! z, [+ F9 F6 L# L
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    $ i) v1 R3 `: ?  Q0 \) z- K+ u
54. int   n   =   send(s,   buffer,   length,   0);
    9 X$ m. S. b! B# V% ?" |
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    / q4 N/ @) r7 A" X9 b/ [& K8 j
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    9 N2 T7 p& ]9 w8 w" o' F
59. if   (!rlen)   return   false;
    " o# \5 G* b2 H5 x) l* ]  ?
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    + g4 J& g$ O- }
62. 
    + o  s6 \" n& g! B3 H. K' m' s
63. return   true;
    
64. }
    ; `, t9 ?% I9 Z0 r  O
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    * E6 N4 W8 z8 \" q3 |
67. {
    : Q# @# L/ ]9 B/ M0 t# U% L
68. char   buffer[10240];
    : R- K) w- O" t( ^( S
69. 
    3 Z5 j6 y2 l4 ]0 Q" V3 I
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    ( u* c4 z0 W7 u6 J) F
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    ' y! @5 y4 i  z
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    9 x1 J0 ?6 n# H
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    0 d& E- A% W3 ^& |/ D
89. result   =   response;
    
90. result.Delete(0,   pos);
    , m; x  \# V' Z0 K* X
91. 
    " n  G- ?3 T4 G8 F
92. pos   =   0;
    ; q1 y3 _# J" `0 g1 C
93. status.Tokenize(_T( "   "),   pos);
    ; N, `7 X& s; a- i7 r
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    - _+ {2 R. X- i3 ~
97. }
    ! ]2 B9 z* N8 J" U* _" I
98. 
    * X. O4 V/ F; |& \6 i
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     5 ~$ f8 z( ?5 x( I8 P3 _) i
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     5 Y% a  Z' s3 X0 w
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     # \; L6 v$ Z: ~# O' }
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     4 i' C4 F* M" o4 {/ o
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     ! V" k8 Y% N6 _
112. }
     : }% v* P. o4 M" s! ]2 V1 T
113. 
     & Q# o1 t  p1 u; c$ S9 b) }- h
114. MyUPnP::MyUPnP()
     8 w7 [+ \$ s: |/ t
115. :   m_version(1)
     
116. {
     2 s  k7 Z/ U' ]; @$ G% q. e- T
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     " p$ k) r/ e7 [3 b* Z6 w
120. 
     
121. MyUPnP::~MyUPnP()
     # `# U2 _/ U6 l- B; P; t) N
122. {
     ( W* i. v# ~) M
123. UPNPNAT_MAPPING   search;
     ! Z2 Q7 k- n4 \' u8 H% Y
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
       n; G5 k5 p# c  q: ]
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     . U, u( k: f  m& e" ~
127. RemoveNATPortMapping(search,   false);
     
128. }
     : r* O& c6 B2 ^2 S) v1 N9 ~
129. 
     
130. m_Mappings.RemoveAll();
     * N5 n7 E  |. _" s/ Q
131. }
     1 B' g- i; R: d9 S$ B& b
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     : t% a! s7 }0 x( t
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
       Y$ c0 i. j# P6 [7 K; w, l
141. {UPNPPORTMAP1,   _T( "service ")},
     3 C) L- Z5 E& B1 [
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     . b) h2 d( V5 ^
144. };
     & k0 ?- e- R+ b' h) ]5 H
145. 
     ' y& {3 W! Y, M
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     - }8 k' k( s# p. r+ }0 j
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     8 y5 M+ B1 F. d% P4 |
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     . z1 P- w9 g4 S" \% X% h
152. if   (!(i%100))   {
     / |4 p7 u/ O: o! @! l0 H8 D
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     , |. ~; E6 L$ m  X7 d6 b
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     : A! }# g) T/ F$ a( C
155. CString   request;
     
156. request.Format(_T( "M-SEARCH   *   HTTP/1.1\r\nHOST:   239.255.255.250:1900\r\nMAN:   \ "ssdp:discover\ "\r\nMX:   %d\r\nST:   %s\r\n\r\n "),
     . t( g" H7 R) O; p8 I9 B+ M
157. 6,   m_name);
     7 t* {3 y7 H2 m! I+ U
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     0 ^  e& X/ _5 `3 [( c, G8 K4 @4 g
159. }
     8 \0 k% r1 G$ b9 v4 |* `6 S4 Y
160. }
     
161. 
     0 s( j# K; r" |# u7 L) j6 f% n
162. Sleep(10);
     , m% b' P0 [4 v  H
163. 
     
164. char   buffer[10240];
     ; c( U6 ^6 A( {" c  P# @+ r3 g* A
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     / O' e: x" x' A; i  B
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     ( A) W$ C- T; D% B) X- F4 b
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     , A/ ]8 J! z8 `6 M7 ~# q
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     * n9 y- n& {1 N
172. 
     
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)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     . T: [+ Y  I. {% J" k4 g
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     " w& q0 n0 N4 U  V& n0 [. C
181. if   (name   ==   _T( "LOCATION: "))   {
     # i7 Q+ l" ~8 h
182. line.Delete(0,   9);
     ; U. a8 O) o) b  L, b7 Y. r" }
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
       `5 n5 l: K9 s, H; a# h
187. }
     
188. }
     
189. }
     , @( N  ?9 E% c; q( {* W) W
190. }
     
191. closesocket(s);
     
192. 
     
193. return   false;
     
194. }
     6 S8 d5 R, v/ M6 q6 w

复制代码

天云上人

以下有关upnp的接口来自emule，
) l) g6 F+ V1 O) b+ H1 L
& v: c# N1 K; B" H. c  t2 e+ J
///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


5 K# w9 O0 k, C#pragma once
#include <exception>

4 M7 D0 s4 [+ l4 Q$ m* y
5 a/ f4 ?7 ]% c% S  enum TRISTATE{
" ~% J$ m$ L2 A; u2 r! M3 \/ C        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
};


) O( ]% V0 q5 g6 kenum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
8 |" X4 p0 v& F4 L( t8 }4 x4 z1 C; C* s        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
* W  [6 e% o9 X- I" ]};

3 R# o9 j1 z% U: r
/ J& G- F, u1 W2 B8 Q* k
2 }# z/ X* S% j. [/ K8 e; `0 A
class CUPnPImpl
& e# G' S* y$ v, w. V, D{
public:
" h+ o* Y: `0 o3 M# J+ z        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
% E0 ^( p! D* a3 Z                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
* I! ^( q+ I1 f8 {( r        };
1 d% Q" n8 u* c6 l3 d$ V6 ~
! m8 ~, c4 W$ |, O7 Q) E3 i9 M
; I6 p/ f/ L+ ?! |9 {2 b        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
/ a6 P+ ]% E3 k' T2 p        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
' O5 F: r% u( _7 ]0 v       
& m$ f& `' P2 ^+ M/ O9 H* d3 ]! e. T        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
7 z' _) X3 ^4 _' o
- C8 b+ S9 I# W  B7 S  x$ J, Z. m
) s6 I- Y; J; g2 A        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
5 E( p) Q# z# v2 |        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
8 W+ Q" s2 x$ h2 K        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

# Z7 l- c6 u. y. K5 f$ b
! N8 m. p1 q1 e& Q// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
! p7 r: V$ z: |6 u$ X. M: O* x# c        void                                SendResultMessage();
/ }2 Z; x5 ]5 \4 d+ j/ O. N        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
& U7 j: M( @+ G4 {6 R+ k( \0 u+ \        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
" W3 t: p6 h# C* T
6 F5 v$ g& s$ G3 U
+ b" Z" W- T( H& n* ]* |& R$ Iprivate:
        HWND        m_hResultMessageWindow;
! u. t" \& M1 c  _! _2 [        UINT        m_nResultMessageID;
" f* a6 f1 b5 a! r$ F

3 U0 O( r! R9 }8 C* w};


5 `( _' {5 b/ d7 ~( N) t// Dummy Implementation to be used when no other implementation is available
  ~- M6 T7 i& B, qclass CUPnPImplNone: public CUPnPImpl
{
public:
* Z' H0 A( A  Z  A- n* ?        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
- i( E+ L+ k' ]' Q0 R% ]8 @$ t. M        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
2 F4 r* S, {! L1 Z7 X4 S3 r        virtual void        DeletePorts()                                                                                        { }
. ]! A* [% J: r3 b) S, B        virtual bool        IsReady()                                                                                                { return false; }
+ ^" P; O# I2 I. {: A& G. j' g        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
+ |& ]4 \9 C6 d2 V0 F! d# H};

! r, F( T+ V& w. G
; @0 N: x, w, _% x, z0 ?  }) K/////////////////////////////////////
0 {# D1 Y. Q* N7 G2 S//下面是使用windows操作系统自带的UPNP功能的子类
( K+ f2 k6 r2 ~

" e* l/ j- x# g( z; D#pragma once
#pragma warning( disable: 4355 )

( U* w+ }/ j7 p$ v
#include "UPnPImpl.h"
7 n! p0 R' Q; w0 V  L* t  |3 b#include <upnp.h>
6 |* Q  S) `8 \#include <iphlpapi.h>
0 |; e( k; u# Y9 j; T#include <comdef.h>
#include <winsvc.h>
- O! l& I9 y3 T3 H: L  P5 v
; v8 M: A! t; V# I7 k
#include <vector>
4 v; D  T7 o  D) v#include <exception>
#include <functional>
  ?) Y* E8 ]' x8 x6 d
1 _5 @" w- h/ m1 W0 G
$ o) K; u3 n. Z4 n+ S1 s- a

( \: ~/ ?7 z' P. S0 N) j; _typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
/ o+ Q8 a9 G6 \: F+ ltypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
" X2 `1 K+ B6 k5 jtypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

4 c+ e: B  w+ M2 I- V# f& w
typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);

$ f7 w8 H. \9 k3 a/ L! {
0 Z9 ^8 k& I2 G& H/ ]( o* o  ?  Ytypedef DWORD (WINAPI* TGetIpAddrTable) (
  b5 o# ^8 P. Z* z6 C  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
6 W' a" C/ L8 x- V8 Z: J% t+ a' J  BOOL bOrder
9 C0 o& h1 v( p0 J0 |);
- C/ ~+ j, p9 G+ Y

typedef DWORD (WINAPI* TGetIfEntry) (
/ N0 ^5 h  D) v5 a2 X  PMIB_IFROW pIfRow
);
( q7 K3 o) W' _0 }* a4 g% s

( ?8 d  m( U" wCString translateUPnPResult(HRESULT hr);
" p" Z" d$ Q' n' @HRESULT UPnPMessage(HRESULT hr);
" b  l9 E  R: h, d

3 D0 o( V1 r0 o0 fclass CUPnPImplWinServ: public CUPnPImpl
! `1 c3 [: I9 y{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
) E8 q( q! }* M. i// Construction
* P# [" E2 p0 b( P- `- Upublic:
$ s7 U. n4 v) I! e( D        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
# C. w- f7 I- n

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
. a, B& p4 y: }' r6 X& V( c        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
3 q1 t. [0 Y0 q$ P1 Q+ O        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

: N  P; {8 ^# p& ^" i9 M
0 Z2 a3 t4 E. ^" @9 Q( R        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };

7 F3 |( t3 ~7 L
( n) b: ^9 R7 lprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
4 q% d8 B% i7 a+ G& y        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
6 S- t+ ~' @2 j- C        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();
, ?0 g$ {% m- P4 r0 I: F) D3 x8 n

        inline bool IsAsyncFindRunning()
' ]7 R3 k) Z% e1 P" |* V        {
% C3 `7 [2 V1 J! T2 q                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
) K# {. Q8 v, C$ d/ [$ e                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
: v  I6 W. ]. T8 p' F- M4 q& `2 K                }
                MSG msg;
; U' L$ I; r$ _                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
, a" r* _3 k: G; i                {
4 ~: y+ k$ ], q# Q  y( w                        TranslateMessage( &msg );
% i" Y0 H. Y+ f( q( i" I- x                        DispatchMessage( &msg );
' I+ q# F/ ^6 k4 L) P                }
' [3 _" S8 Z; ]                return m_bAsyncFindRunning;
        }
+ K8 v9 g9 X; v4 @

        TRISTATE                        m_bUPnPDeviceConnected;
3 z  N: M( J2 a

! _1 Y$ E8 p; Q$ G% j. u/ X8 q) X// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
7 x1 A' g- f" U, Q        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
0 n' J- h; }3 Y' h9 A        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
" ~- ~& B7 V5 o' m        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
& N4 Y, S  e) l        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
9 ^( m+ P0 V% M; \2 n( Y# X        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
/ h  E' |$ z' r% I% R: i/ D5 R        TGetIfEntry                                m_pfGetIfEntry;
1 y9 e' z  A9 K4 K4 {

1 H0 r% [) e3 b        static FinderPointer CreateFinderInstance();
& x3 t1 j+ c3 }/ P! @' e        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
# v" ~& g  I3 n2 \! c& \: ~                result_type operator()(argument_type device) const
                {
3 }4 M+ t  K8 E; P5 e* v                        CComBSTR deviceName;
, g* `' ~9 Z* }                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
1 u& `1 f1 Q' G/ }- h$ ~, c

- V; D' g# J6 e7 {! U8 b- y" F                        if ( FAILED( hr ) )
3 [( Q% a9 m6 p3 B$ m  l                                return UPnPMessage( hr ), false;
, C! H# g0 ^  o, V3 m, b0 q: a# g
# i; a) r0 {' r6 E
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
- \. y3 H7 p; K3 _( r                }
% g& K1 ~( f* f7 T7 ^5 u3 s! P# A6 v                CComBSTR m_udn;
        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
" \% D6 r7 Y. d0 |9 l7 a+ f4 V* t        HRESULT        GetDeviceServices(DevicePointer pDevice);
7 B, I- R' g7 O$ ~        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
3 |( Q  W9 L1 S3 N: h8 o9 ^        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
9 W* V6 V" o) c# a( ?        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
/ s  V1 i8 ]& g3 a& f. c0 `4 o7 o2 n) U                LPCTSTR pszInArgString, CString& strResult);
! G/ b" G8 O6 ?6 V! A        void        StopUPnPService();
! N% f/ u/ b4 o) F9 m" W2 X2 `/ j/ Q

        // Utility functions
( X5 K/ U7 x& J8 A8 G; W        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
5 j" q2 y$ [3 T; a3 w* R( k3 ?        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
. Q" B  n0 X; j5 L0 K3 g/ N        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
9 {; ?' r. O' ^0 o- E2 K! u        CString        GetLocalRoutableIP(ServicePointer pService);
# o3 l5 z& O. a/ [2 L( C: J$ S" ~

: Q$ B+ _' e  I. H' A8 e: L// Private members
( Y6 i) r: \0 z; {( Aprivate:
/ L  W% i4 w& I( N' H7 ^9 u        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
% d# U7 {, _9 Z$ C8 a        FinderPointer                        m_pDeviceFinder;
/ T7 l4 j' S5 y- E* c8 Z        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
, A+ q! u2 S% A

! U$ \) [0 L) E9 x3 F. B' k        LONG        m_nAsyncFindHandle;
  p/ r0 L( y" f) r. h- l4 B) e! b3 c        bool        m_bCOM;
$ \1 T4 v9 o& h5 |# p  I* a        bool        m_bPortIsFree;
9 U+ r! b% j' r; I4 Q' P$ Y        CString m_sLocalIP;
- c3 g! g# R1 {# w        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
! Y0 x; k( M4 [4 V% g' p        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
3 \1 O0 K0 _5 n( `) p        bool        m_bInited;
4 U. q, j1 g7 E- t& X( i1 I        bool        m_bAsyncFindRunning;
$ h; c2 ], }2 J% m5 K2 l        HMODULE m_hADVAPI32_DLL;
" H5 N8 r1 R: ]6 c        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
/ ^3 K9 Y4 @8 Y# |        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;


};
- S. }$ K5 T/ l7 S. C, x% l4 B

// DeviceFinder Callback
& W0 w6 |6 A( Z0 ^1 hclass CDeviceFinderCallback
! o+ M0 j( W7 K        : public IUPnPDeviceFinderCallback
1 f4 F/ U! r" g{
public:
9 i4 G! o3 t9 ?! n( d3 M        CDeviceFinderCallback(CUPnPImplWinServ& instance)
: B# ]  `5 a* t" W  a                : m_instance( instance )
        { m_lRefCount = 0; }
+ P1 j, t- E; Y% q' \

   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
! r' i7 R1 N- j# h  p4 @   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
4 }$ k: V* Q% b
$ f4 J9 F) [' j3 W3 o8 s8 y
// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
6 T: j8 k2 g/ X. X0 i" V, I        HRESULT __stdcall SearchComplete(LONG nFindData);
2 A( ]; |+ }, w. L/ }

private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
$ t* `+ O  I; e
$ T- P! q1 @. V7 _! q
// Service Callback
class CServiceCallback
% N# `/ O1 V0 W  b8 s        : public IUPnPServiceCallback
{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
   
5 I  s3 \' k4 G( f4 K   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
9 C8 X# o% _% v   STDMETHODIMP_(ULONG) Release();

" H& i. T/ ~; @# t
- c, J) i# d: Y// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
) R" x" b8 m7 l        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


( q7 V. e3 k/ u" [5 Aprivate:
        CUPnPImplWinServ& m_instance;
+ S$ J( C0 R. Z" I2 Q        LONG m_lRefCount;
" q+ B; l' z( n! o. {};
$ u/ Y, \7 j+ H. Z' E

/////////////////////////////////////////////////

9 D% K" _* d7 @
使用时只需要使用抽象类的接口。
2 x7 ^1 ~% A' f3 \3 p5 y5 hCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
5 Q+ C9 y9 u/ b! J4 L: x! oCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
; g8 B/ K9 e9 p6 U7 B# cCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.