UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   & ]! ~3 H' Y2 m. K3 Z
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   $ k3 c0 j# H2 D1 p
9. {
   8 A1 ?& L8 Q2 G$ e) a
10. public:
    / ~- k+ Q5 S  E/ }9 ^7 H
11. typedef   enum{
    7 W/ [& j$ l* Q. j5 J
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    . t' [2 p  G; h/ a5 G0 q
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    0 ]* ?$ @* `1 r" \7 K1 I
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    & a# ]5 [- N, m7 ]$ n
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    - y. O6 E' N- O/ s: i
25. WORD   internalPort; //   Port   mapping   internal   port
    8 V# \. P1 H) y. p
26. WORD   externalPort; //   Port   mapping   external   port
    ; z' M9 e  W# [$ q4 t0 O/ L
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    ' i% \" B: K( s8 G' Y2 W: B: m. ^6 K% ]
29. }   UPNPNAT_MAPPING;
    
30. 
    6 z7 p3 {4 G8 ?; h- ^
31. MyUPnP();
    4 I- Q" u  E2 d- w$ }6 R
32. ~MyUPnP();
    : m% Q0 _4 O% t% v% l3 E
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    0 g, r# j' H! _6 f5 T; a7 X1 i
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    # [. k( _- |& d2 B( n7 g
37. 
    ' w' ]# Y8 e" v" |
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    $ m( t- G6 z) O' F; O; S/ E
40. WORD GetLocalIP();
    & ^/ Q2 S& ]7 @
41. bool IsLANIP(WORD   nIP);
    ; B+ e) W4 V) Q4 Z; a; F
42. 
    
43. protected:
    5 j2 T: F& c' N, w
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    " f9 b, @0 }& v1 d
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    6 G* d  n7 d" @0 M# ]
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
      W5 m# M3 r3 U( c) B1 Z' R
50. 
    ! ~' U! j) K5 h4 h7 V5 @/ R% p; C
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
      ?  O9 u) ^8 B  J1 m' a
52. 
    6 l. `5 a8 u! z: K/ |; }: _
53. bool Search(int   version=1);
    
54. bool GetDescription();
    # J; i1 x0 a$ L5 ]4 @4 e5 O
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    # }+ |" A8 L9 G. V  T+ G( U
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    3 `# {, `+ w5 N9 Z! _, V
61. CString m_name;
    2 }# K6 B0 v  i+ n
62. CString m_description;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
    # @' S- ]: ^" Y5 X
65. CString m_friendlyname;
    
66. CString m_modelname;
    , v2 ?1 t- N6 O; L
67. int m_version;
    9 E( G* g2 V+ c
68. 
    . P) C. e) h* A9 a% i
69. private:
    9 h* D6 A$ s0 N; V% I1 W
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    ( b+ ~3 Z5 ~2 I3 u% x
72. CString m_slocalIP;
    , Z3 `' q" T& _. {1 y5 O
73. CString m_slastError;
    6 W2 |$ X  _; `: H; w
74. WORD m_uLocalIP;
    2 c2 [. A; O0 U/ @6 L' X
75. 
    6 u7 a" u( f3 Z2 `
76. bool isSearched;
    : j4 j; w* k- [& Z. }. J
77. };
    $ x0 w  }- c8 d% O4 j2 B- F4 k8 g/ n
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   7 @5 L$ L8 t2 e4 A
2. #include   "stdafx.h "
   8 ?; T* M1 @% R% D9 t9 w- |6 K
3. 
   
4. #include   "upnp.h "
   7 U# F. H+ D( i0 v. C7 t1 K# r4 \
5. 
   / M! }9 ]! i3 O0 r, Q5 X
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   1 [: a; J5 {+ O4 d: {
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    ; s+ Y$ R1 o2 H! f
11. 
    3 p, q: g) C  q. z& \
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    & ]! D4 j0 E, i. y" B0 \* k
13. static   const   int UPNPPORT   =   1900;
    ) A8 q' @+ W! u  j
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    1 j: R; z9 k# q3 t( _7 ?4 r
15. 
    9 i. h$ F& z  u2 k1 _. u% ?0 x2 l
16. const   CString   getString(int   i)
    # f  q! I: x+ y: n6 p5 O7 ]- S/ b8 [( j
17. {
    
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    " @# f" i2 o) V
21. 
      x) T! D: @/ @, c
22. return   s;
    - W, O' ~+ r% k8 }
23. }
    
24. 
    4 |% ^' H1 R" _0 _
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    . e6 w' T# |# n  ^9 p
26. {
    3 r' w. `% s) X8 X" R9 I+ M
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    1 P4 Y% M+ l0 i' t+ T
29. 
    ! x! j+ d2 f1 \
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    9 ^2 a1 M8 Y0 b; X
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    : {" H( {  n& `' P' z" f( N
33. }
    ! r3 q, Q. E& H; \
34. 
    9 K4 K: C( X& i% w6 m- j
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    
38. 
    % H* P) w6 {% G1 s3 w
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    : h$ x' B9 ~; _
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    0 [0 L6 z) F3 h
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    1 M. s9 T% h9 x2 T' ~! L
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    6 ~6 B  `! E9 I, _! V. U3 P
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    9 ^+ O& f; J0 n* W4 [2 |+ R
57. closesocket(s);
    : d- P! ^7 {3 f
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    # v, g/ i/ P0 r9 N( B9 Z
59. if   (!rlen)   return   false;
    , c, j% R4 @5 C/ e% K" m
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    6 V3 E1 R- O1 e( i% T
63. return   true;
    
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    
69. 
    1 V9 p* d. R& E5 O  O
70. const   CStringA   sa(request);
    ( ]) Z+ L8 }( D* ?( F9 L
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    $ y3 X0 c8 @' Y) `3 ^4 `, ?
73. 
    : h2 u. N( ^; X- K: |: Y
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    $ N4 e7 u& V" H  ]3 D5 m0 j
76. sockaddr.sin_family   =   AF_INET;
    ( T2 p7 N0 r. [0 A5 X  [) X; p4 o0 I/ k
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    0 B* c# m) s" ?5 c& `, S2 T# r4 P
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    & D2 y; k* _+ J, y& ?! E4 O9 G
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    
86. 
    3 t9 L( T1 C; h5 x: n0 J& w0 E
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    ; x) A6 F  m2 S$ O
91. 
    ) ?9 E$ }7 X, D& a
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    ; i/ t8 Z; T' U7 E  j
94. status   =   status.Tokenize(_T( "   "),   pos);
    / ?4 B, j* w) [# r* U6 i% L
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    7 d0 i5 E& z4 M4 r
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    3 W; n" K$ |: G. M
100. {
     0 E6 i8 r9 g% b1 H8 y# T
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     $ p5 J- m2 a; s; C$ I7 Y
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     ; B7 t( w! m1 B  X# K8 U- s
108. int   posEnd   =   all.Find(endTag,   posStart);
     , |7 m3 J6 q$ b& H2 a0 y- S
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     ) ]/ K" j* i1 T1 Q) \6 A0 U
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     # t: L. H1 j5 T' X
114. MyUPnP::MyUPnP()
     ( c2 w$ k& e6 L( E1 |2 F; k
115. :   m_version(1)
     " T1 z' J. `# ?, T, a
116. {
     4 F/ Z) j! E8 T7 d7 a% K' e5 W
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     9 P* L( @8 t0 o  U& M
119. }
     8 {5 Y% _. q. \! ]% f7 v
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     + a& [% G$ p9 Q$ r- x+ g3 b2 v
128. }
     8 o( ^4 x3 _. d
129. 
     
130. m_Mappings.RemoveAll();
     
131. }
     * t" E; \( G1 {+ y# ?: v
132. 
     
133. 
     ! a7 E) F3 y  ~5 S2 t8 o4 R8 [
134. bool   MyUPnP::InternalSearch(int   version)
     7 m; s* k8 l: p$ g" w- A- h% e
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     & g0 G* j% _/ N" A3 v
138. 
     ) k7 R1 {0 a, a7 {, P
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     3 i8 S/ {6 W5 a6 r: J4 c0 b
141. {UPNPPORTMAP1,   _T( "service ")},
     7 k3 C5 {' k( q
142. {UPNPPORTMAP0,   _T( "service ")},
     ) F4 H- M# o" P! S( K! ~% R% U
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     & |' ?( v1 A5 F; w( Q3 b
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     
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);
     7 _) f+ C# U$ B. U
155. CString   request;
     7 T0 B5 A) [6 b# A
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 "),
     3 ]& y! ~- C" X/ ~4 p- V- D
157. 6,   m_name);
     0 A, ?/ g; C6 N2 s3 C: Z. x
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     : G" ]; D( H" _9 |# G
159. }
     
160. }
     
161. 
     ' i+ P. B4 O& D: ]" q
162. Sleep(10);
     + L0 |% l8 z+ A% |9 `
163. 
     * u6 H6 }: `* L6 G9 @  L* u
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     ; X$ `( M; y& b8 _% A
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
       d2 R/ Z$ F7 @& o
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     % E! j5 k9 x8 X) J- }' k
171. if   (!parseHTTPResponse(response,   result))   return   false;
     ' B0 `% Q7 `2 l- |
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;;)   {
       ]4 O5 [7 G9 Y/ `" O5 G$ a7 j
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     + S- d, i. D/ k3 k4 C# Y; I1 a
181. if   (name   ==   _T( "LOCATION: "))   {
     : x- a0 n5 W7 w* Q1 `
182. line.Delete(0,   9);
     . A9 b/ W: W2 Y
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     " C  g% C* D0 ~* s% g6 ~
186. }
     " o" [8 j1 k6 U3 n# V3 Y1 S
187. }
     $ C6 w9 {  {; ]% [% T( H+ j* t" M
188. }
     + U6 p8 p( M: Z9 N! x
189. }
     $ N0 B5 I4 X" Q$ c
190. }
     0 j+ c2 z( X& j- {% y
191. closesocket(s);
     
192. 
     
193. return   false;
     / \1 E" |/ b" M( G
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
) ~+ b9 R& ^: Q+ e; w

4 Z3 j9 v. L  a" m* V/ ^3 z* e$ M///////////////////////////////////////////
7 b8 i) U) M' {) ~% f! I/ C2 W//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

# U) L1 m& c% y
2 S! P' D" @, r7 i: S#pragma once
#include <exception>


  enum TRISTATE{
5 U  E1 [/ e% u" H- n+ t        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
};


4 k( y1 ~0 {5 ^" q) W3 Aenum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
! F8 U4 ]7 p, P% L2 c/ g3 O        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};
: C9 ?; o9 o8 F0 x; b


( z' c2 c: B2 \6 |' ]! {
* }* y4 Y% U% Jclass CUPnPImpl
{
, d' @$ B: R0 @- [) Y5 ^public:
  v9 D  u- R+ ^; T! o1 u        CUPnPImpl();
        virtual ~CUPnPImpl();
' ^% A  Q; @( [# M8 V/ S        struct UPnPError : std::exception {};
        enum {
, `  D6 a4 h8 D) r1 I                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
6 K' V" }& K6 i$ x( z        };
6 C& j: Q/ l- J6 n6 Z" r" W
" b& Z0 Q. A* Z
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
: q/ H; ^3 n% y. M        virtual bool        CheckAndRefresh() = 0;
) D. S, b3 k8 t" P        virtual void        StopAsyncFind() = 0;
7 _9 z9 B1 g+ d: l+ s% j# N        virtual void        DeletePorts() = 0;
+ X! i! [8 r7 p  M$ S" u- f        virtual bool        IsReady() = 0;
0 [$ L0 ?! g% e1 G. |9 O. h. c        virtual int                GetImplementationID() = 0;
6 U$ ~- {9 X; E, w* r       
7 J1 K) `+ p  n/ w5 d6 M$ M3 u$ z        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
8 z$ j9 ?. H4 o0 Y& G1 ^1 E        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
. L5 S  e  X* ~        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
- g- n  V& `5 |& S5 c        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
! v! I! }3 L# {  v0 y9 q

% S' ~7 I# ?. [// Implementation
# n) D4 S/ D& o& r0 lprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
( m6 N# u: B8 g        void                                SendResultMessage();
2 G6 \) d& G7 l: d8 F- K        uint16                                m_nUDPPort;
0 Q( w) o; _* ~& y" _        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;


private:
" {: `% S' A0 V" `! ^        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
) F6 K3 k7 \0 G8 s) Y) {' g0 f
& y% V2 @$ ?. {7 R# {' n# P
+ s( R3 }1 D/ H" I};
$ C3 b2 A  o+ r/ x- J3 u$ m
( }4 q' }" Q& Z" e
$ k; w# _% B  ?// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
, ~, x( g* p% M* j  {! Y! |{
, {3 ^! I& u9 xpublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
: i9 q% N) D0 b! E9 v6 z        virtual void        StopAsyncFind()                                                                                        { }
4 h. f; ^7 n. _8 L        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
1 y" y. H! I( L! z: _& R3 W        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};


/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
" u; B& ]' G5 ]- I
  ~- Z% s* a/ E2 I
# N& Z; i/ n3 N#pragma once
$ F1 O* H# q# _, b#pragma warning( disable: 4355 )
4 l! a/ |0 F0 z7 f

% F  K% v5 ~# T/ C% Q#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>
: U5 x( O+ u- u

#include <vector>
1 P! x: r3 r( C9 f$ N! Y/ R#include <exception>
* L8 P' n: R+ J. z4 o. l: _) H' r#include <functional>
7 U' P1 K, e2 \5 x) T, U  n: c  H


3 c5 f- Q& D" t& M( H- \6 o
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
  y8 `7 o6 i9 D7 \/ Ntypedef        _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;
6 n+ w1 B5 a: |- T- U: w3 Z' t/ I
/ V6 v5 R2 l) [+ D
  c4 M5 A* c6 o  p5 v4 ^' {typedef DWORD (WINAPI* TGetBestInterface) (
( e- J4 Y4 f) W  G5 ~6 r  IPAddr dwDestAddr,
: D$ V: m* U0 ?: v; L- g7 n3 H  L  PDWORD pdwBestIfIndex
/ S- e7 g2 C, P6 B);


typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
9 V& a: w4 e  \6 t) _' w; r);
0 `$ t% }2 a! y, |8 i1 w3 ~

% a7 n0 a& I* D9 [+ Etypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
& _. [& z' R5 ?4 y) Y7 d);

6 C' |' N/ w9 V9 I) p1 \0 ]2 D
CString translateUPnPResult(HRESULT hr);
' }4 r- ?( C, h6 @9 `HRESULT UPnPMessage(HRESULT hr);


class CUPnPImplWinServ: public CUPnPImpl
{
+ F$ q/ Y, B# g' i2 [  b        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
- J- U& M1 _$ o: G) }' f// Construction
public:
        virtual ~CUPnPImplWinServ();
0 C$ ]# R; G0 c$ z/ e3 `4 ~        CUPnPImplWinServ();
2 K- j2 X+ X" W! j- {5 n

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
: n% z# _; I' p8 a6 a) [        virtual void        StopAsyncFind();
7 m3 L! a2 U' a& ?) V: m; F        virtual void        DeletePorts();
( T7 g! Z/ h: C6 D8 J$ R$ |        virtual bool        IsReady();
# k$ S% G* ~( {        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
6 D# n* r- V+ H$ ?5 Y1 A

        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
0 X$ Q: L/ e" w  F: B        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };

& ]+ [% P0 w$ `- A5 x6 E5 N
: X, U9 y& T/ O' iprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
5 x8 J6 b. o4 X8 p& P1 c        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();
  k; F) g; B( }, x

        inline bool IsAsyncFindRunning()
+ {/ G# `  b) d( ]" N        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
2 b  k* q) I3 L5 s4 F- w8 l" w1 ?" \                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
" s" V: }7 o3 [1 d                        m_bAsyncFindRunning = false;
                }
                MSG msg;
; |1 W& R) ~: h' I7 m8 A                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
6 I' K/ Q, i" H0 ^9 Z  h: V4 I                }
                return m_bAsyncFindRunning;
        }
; u- [% |  j$ w: k" ?
8 ]1 d, e: x4 {, l, p
        TRISTATE                        m_bUPnPDeviceConnected;
* A. r& o2 W8 v6 U" ^
2 w+ c* _$ H% o3 u' P! ]
// Implementation
% e6 i/ u+ f" K0 C- M- J9 Q        // API functions
% m: }( ]+ d+ R8 E8 _9 s        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
* r* C1 O: ^$ l% T( Q        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
- f: W+ r! a9 N8 {# U; W; o; o' }        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
6 I! T* C: h+ R( O' @        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
) J" U  D  L: G. R* M1 E! T        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
4 @4 }) I1 N0 u& b  }  l        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
' J2 a% C0 @; L( [6 `
  @5 _$ \# n4 I4 N* ], x
        TGetBestInterface                m_pfGetBestInterface;
, k* T+ e5 \* [+ X        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
( Q4 z" U3 Q- I$ J' a) n8 x

        static FinderPointer CreateFinderInstance();
- B, ]0 m5 I4 E8 i. b        struct FindDevice : std::unary_function< DevicePointer, bool >
; \: j. f: p* H: M        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
- v, A5 K- `. D/ X8 {0 U1 P                result_type operator()(argument_type device) const
                {
' j* j1 X& Q" ?" u                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

) L7 T; \! r7 U2 n4 d! P0 v1 e- C
                        if ( FAILED( hr ) )
% r9 z5 ^. p3 Q                                return UPnPMessage( hr ), false;

& w4 L9 r$ [) |3 a" ^4 S7 ^. K3 B1 k/ H0 ]
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
0 G1 P2 m: R7 `, x# T; N8 v$ o                CComBSTR m_udn;
$ W7 F1 c5 u+ \1 s5 F        };
       
7 V6 t$ B5 a0 S* c& I- d, r2 C$ O        void        ProcessAsyncFind(CComBSTR bsSearchType);
' e, M2 E2 _9 \. V; G        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
* V" r6 H: x" b        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
4 B; P6 A5 Q7 d% {2 r. B                LPCTSTR pszInArgString, CString& strResult);
0 {% H  E' i% T" S* W        void        StopUPnPService();


8 D4 p+ D4 B( v1 o" L        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
( N. b, m2 N3 z$ S2 I        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
" o- f5 `5 ^/ ]+ F. I" |$ I        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
0 I9 l" j0 m) G$ p3 ]        CString        GetLocalRoutableIP(ServicePointer pService);
3 L/ E- w2 i# W0 }+ U
8 p: l- R2 O* e7 S  b: g
7 n: s  V. S5 `$ l) ^) p3 C// Private members
private:
1 g6 L5 ~* M4 _) l. a3 ^4 D        DWORD        m_tLastEvent;        // When the last event was received?
- b; k% o6 `7 M  k5 I5 b* Y7 a        std::vector< DevicePointer >  m_pDevices;
3 o" d6 w3 c1 c        std::vector< ServicePointer > m_pServices;
. `, U7 G$ y: V        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
2 {; e, i6 o) [& N7 x        ServiceCallback                        m_pServiceCallback;
/ L2 Q. X) e7 K( G  Z+ F% T, i
3 {2 M" o8 v  i/ d) v+ F& K9 b$ i
        LONG        m_nAsyncFindHandle;
3 D* K, Q- }) ~& x& l( x3 n        bool        m_bCOM;
1 U9 P9 k9 u" u. ^        bool        m_bPortIsFree;
+ F7 j! H* [* c9 K1 v; h1 L" R        CString m_sLocalIP;
& C" v# Z% e* I# y        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
9 R' I: k! e- Q        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
* f& q/ r7 S. ^' \+ w        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
7 y; g& Q( K" ^) i        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
6 t/ S* C2 ?& \        bool        m_bDisableWANPPPSetup;

; N# e" y  _% }4 s+ w( T" T& P
};
& r: t5 q; C& a$ @) P$ ^
; ?+ \8 k( {  r7 e3 b1 s) e  x- c
$ m0 C9 O1 ]& ]. v5 l1 D// DeviceFinder Callback
3 x% L3 ]6 s9 }2 Lclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
  N4 F' p$ W; d, k  i! vpublic:
$ _7 R0 T' X! @2 S1 z1 B        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
# l) n$ I' Q0 S* \  C# r0 ^: W: j        { m_lRefCount = 0; }
' ?- U8 P% Q  Q8 C5 V

& x* G" U. f! e6 K1 x' g4 V   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
2 R& |! o; [, I& w: q( H# ~4 K   STDMETHODIMP_(ULONG) Release();
4 B2 x0 Z- i9 ]; J/ j. Y+ S

// implementation
2 y9 P8 L- U2 B. e+ e( oprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
4 P: }6 W" o: B# u8 ], h+ Y        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
1 O1 N  n* ~( I        HRESULT __stdcall SearchComplete(LONG nFindData);
9 G" T. {1 j1 }- M4 ?) G
2 ?5 d. J( i2 h& O8 H
0 C) W  j! B8 I" z/ O1 g& Qprivate:
        CUPnPImplWinServ& m_instance;
* F4 |" j3 {4 D$ v4 A        LONG m_lRefCount;
};

, g2 w7 ]5 P" M
// Service Callback
- i4 l1 Q+ Z  d' @- Jclass CServiceCallback
        : public IUPnPServiceCallback
3 d- T+ q9 L2 g" d  W{
/ \4 q- r# c5 R) G9 t7 apublic:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
+ S# j# h+ U, W   STDMETHODIMP_(ULONG) Release();

. O. K& r; w0 k& |
// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
0 C. ]5 R8 p; a" e! E7 Q

private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
3 T) b& U- @8 K9 F5 L. c7 _};


& p4 e& s  t2 _/////////////////////////////////////////////////


使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
- e) K" j( a8 qCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
; F$ x9 d! Q8 MCUPnPImpl::DeletePorts删除端口映射.