UPnP

天云上人

/*uPnP.h*/

1. 
   0 z  {' E1 F" A) ?) f" l: Y1 m& ^
2. #ifndef   MYUPNP_H_
   7 C: x" m  i9 `: {# N3 b
3. 
   
4. #pragma   once
   0 z4 W1 E! K4 Z
5. 
   / l8 n; V, I0 F
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   
9. {
   ! p/ X0 _7 u1 \3 A, M7 p( \; q1 u
10. public:
    ! p+ w% O* n8 T% \- Q3 U0 Y
11. typedef   enum{
    " S9 D7 U6 o  N  H
12. UNAT_OK, //   Successfull
    " o) _( @9 s8 [6 a  p
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    % L0 `, t1 O  {1 Q; z; ~
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    , M3 h2 E7 r' a4 M
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. 
    9 m1 f- D) d$ z2 T1 Y% {2 j! U
19. typedef   enum{
    : V9 J  t% w9 w5 W* {
20. UNAT_TCP, //   TCP   Protocol
      L3 }7 t) \* @% s, G/ ]6 F' a1 l
21. UNAT_UDP //   UDP   Protocol
    ' O$ k6 h: ^7 l
22. }   UPNPNAT_PROTOCOL;
    1 L+ I! V8 L* S( ^
23. 
    " g0 X3 m! E) K" O9 i
24. typedef   struct{
    7 z2 F1 S- V1 ^, _/ ?
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    6 A- J( O+ j9 L" w2 D
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    1 ~1 y* W/ E' c4 Q
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    . c+ b4 `5 [$ K" C* z$ W
31. MyUPnP();
    6 S  ?; B2 M4 `0 a4 Z' O. m
32. ~MyUPnP();
    . U8 M" o& }* ]4 N* `4 u+ [
33. 
    ( R0 C3 i+ R9 ^6 n
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    5 P+ F- d; m. L+ E( u; M
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    9 V; P5 w( N  |9 k
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    ; X$ |# @4 `; V2 H0 r  ]5 x
40. WORD GetLocalIP();
    2 \, c6 G% R$ c* M% e2 C5 e
41. bool IsLANIP(WORD   nIP);
    
42. 
    * e8 O* b: [9 d
43. protected:
    
44. void InitLocalIP();
    8 R0 s7 h2 ^* w
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);
    $ o5 L/ i+ P3 k8 m* @. R
50. 
    4 j: n3 q, u0 \9 U  `  I
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    / D5 J' R/ `. S! x) f; Z; w
54. bool GetDescription();
    ( b7 n3 `9 R6 k3 H5 o  R9 f
55. CString GetProperty(const   CString&   name,   CString&   response);
    5 A& A. o/ u- D5 P( w
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    6 p& y# t( A9 G" c) U2 j
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    5 x" S9 T, Q: `( f- t" S
59. bool InternalSearch(int   version);
    5 k: Y% G, t7 K+ |: H
60. CString m_devicename;
    ! }7 I/ `8 D/ j2 [- Z
61. CString m_name;
    ( ?' R- ?4 M7 J9 N7 u, M& b
62. CString m_description;
    2 K( i$ q. j5 z0 J, P
63. CString m_baseurl;
    0 n, F: E/ \6 U9 W6 F3 \
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    % Y4 S3 e2 I9 c  R# A" u$ ^
72. CString m_slocalIP;
    " A1 g9 u+ {0 Q4 D" Z
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    . N: N8 j# t2 a
75. 
    1 m7 F  K( w& i) z5 s. u: V
76. bool isSearched;
    0 W1 @; @9 q+ ]+ s: n
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   . e: J" S' k0 x' M
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   7 x& `- o2 C) t% ]8 i; a
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    # T  P0 A  s! n- N
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    % d; F9 ]; B4 G
17. {
    2 Y; w: z( l4 l* L  ~+ P8 H7 W
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    ' B# x* u7 y" o$ B, j; D8 O
22. return   s;
    + _9 S7 o' d0 o/ ~. Z% [
23. }
    
24. 
    5 ^0 J  g8 P. w% f; ?  X& K) I* Q
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    1 O( e* ~1 X! Z, E% J
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    ; X" f, u1 t8 P7 p' ~% a
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    , o0 t. z6 @( E* Q
33. }
    7 u# a# k' e) Q1 s- t
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    1 W. F' ~  U1 ^+ Q! b
37. char   buffer[10240];
    ! ~/ k+ N1 E5 j
38. 
    $ V5 J; ~4 v$ D/ l6 A$ g
39. const   CStringA   sa(request);
    6 o- G1 Q; N; G& S
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    ! P, C( \+ P) f- u! {
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    7 l0 Z( J* i8 N" p/ h9 E
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    ! z; f- W. |' I( y" L' O/ R
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    8 E& I, F  o0 f/ y- \
51. ioctlsocket(s,   FIONBIO,   &lv);
    5 m' i0 q4 F( g% g
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    . v0 [( b3 t$ F9 |0 ]2 [8 N; V
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    ' a1 b% k4 Z: Y, r9 p! {" m
59. if   (!rlen)   return   false;
    
60. 
    , c: N: h; G9 G
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    % T. S8 ], x3 Y9 i1 j5 j1 L# E
63. return   true;
    
64. }
    7 v0 W6 p# U$ D) G
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    5 _- ~. E; q. s- E9 ?/ t. i- D
67. {
    
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    ( h, @5 }  O+ J' ~) k
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    & r5 k0 B: c9 c' A
73. 
    2 V- j+ o; k1 x& R3 s
74. struct   sockaddr_in   sockaddr;
    
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. 
    - G( C5 I' ?4 @) a! j. J
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
      K: H- l' n2 I/ Q" ^; a5 z
84. {
    
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    " V% i2 P. Z9 N6 N7 c& K/ W
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    : u% {  r% a. e, t" i" |
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    " B0 U4 c8 w8 j% f" }
97. }
    ' L. m( R. a; A: {5 W' L( B+ F
98. 
    1 {6 R3 w' C) r# N
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    , i$ l* L1 b% X+ m' w) s
100. {
     + H3 X& M. J8 |. G' [2 d
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     ! C* m. R7 r+ V! Q' \& X/ u3 ^
104. 
     
105. int   posStart   =   all.Find(startTag);
     8 W+ t/ Q& i9 u- c& j9 G& R0 f1 M
106. if   (posStart <0)   return   CString();
     / o% t$ l) |4 S( I, Q1 Z
107. 
     ; B6 N/ I9 U' o1 M' c0 P3 J# x6 T& ^
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     9 j  ?( t4 D8 O3 B' Z
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     # U8 K6 C/ y, |- N) H" J7 E
118. isSearched   =   false;
     0 h5 R# o  f7 E$ E- J& s0 r9 V, R
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     ( ~9 `4 X5 u* @) H( a
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     5 `1 O) }' u6 f1 m2 ?+ h( T/ F, s
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     3 k! G$ S1 d3 @  ^* E7 u
131. }
     ( C; F) Z+ V$ z' {* q# b
132. 
     
133. 
     5 H5 [( [* R, C' Q+ V5 p
134. bool   MyUPnP::InternalSearch(int   version)
     8 [* l3 @! x$ B9 [# z3 ]  c0 y
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     ) r" k1 ^- I- i  K- z/ ~! G
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     8 ~8 @* t' U* l5 o) O7 R$ n
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     0 ^+ a8 x, s0 }: c& w$ w
144. };
     ! F. J& Z, m/ m0 D( i5 r
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     ) c/ ?  P9 u" N7 t! V! ~. d- H- G
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     / K0 T5 s. r, L* l' s( X
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     1 c1 ^% w/ I- F" c, h, f
152. if   (!(i%100))   {
     " ?% Y! i* I9 R; {+ A6 a
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;
     
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 "),
     2 ~2 M6 X5 o' q+ r0 _, Y
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     ; t+ C+ s7 R5 A4 }* ^
160. }
     : }3 V0 b& P/ ^6 f+ i( i
161. 
     
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     4 \; S$ Z( C( W9 B  T$ P
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     % [/ u1 l) G5 x. C
172. 
     ! A  h& ?* \1 ]) z1 a
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     " q1 u, |! }% A6 ~% N
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     ; g7 J# l3 E2 T  h, M  S
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     3 a8 _2 A0 w# s8 W7 v
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     7 ^, ?2 f; [$ ]1 `# q* Q4 C5 u
178. if   (line.IsEmpty())   return   false;
       E+ ]; ~6 ~6 F/ v$ [" d6 E/ J
179. CString   name   =   line.Mid(0,   9);
     $ l# J( {; u2 ?) X8 p$ O3 O! H
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     . E6 b: H# h& }% A. _9 Q8 p+ w  O5 j
184. m_description.Trim();
     
185. return   GetDescription();
     ) W+ C" Y# X  Z3 t$ O
186. }
     3 B7 Q& L" B, }; i/ j+ y2 f  @
187. }
     
188. }
     
189. }
     
190. }
     
191. closesocket(s);
     
192. 
     % R% b" c( q8 m$ w; I. s
193. return   false;
     
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
/ c9 I6 f" e  p4 }$ R& @) s

7 ]) u* P% @5 G3 W( ~///////////////////////////////////////////
& H. n, E' P1 K) |" g//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


$ o7 c0 `1 I8 Y- B#pragma once
$ C- }! t+ W3 V, v7 l# H+ a. v, l#include <exception>

4 m5 q" b: R' v5 e
  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
8 p! y6 i+ ^/ ?# f/ K4 m4 V1 g) {* [        TRIS_TRUE
- k8 S( p6 d( p- j6 I" J};


/ `% y6 x0 G! q" \enum UPNP_IMPLEMENTATION{
* m2 D% a2 e1 Z9 }$ i; e1 L        UPNP_IMPL_WINDOWSERVICE = 0,
# `; m! L  L# S6 o0 }9 U        UPNP_IMPL_MINIUPNPLIB,
$ M$ T7 p: |; O        UPNP_IMPL_NONE /*last*/
};




: R5 ^3 Y' e5 L% U2 c7 ?class CUPnPImpl
( n6 M. _, r0 ?% O{
public:
        CUPnPImpl();
        virtual ~CUPnPImpl();
2 q' t' q8 ?* |* H8 V        struct UPnPError : std::exception {};
6 _3 u$ X- @4 @# R4 h        enum {
, f$ ^0 |  w1 x4 f0 R                UPNP_OK,
% c, t% ]9 j  H' v1 r+ o1 y9 V  u2 c                UPNP_FAILED,
( `: s! V. o1 N% w                UPNP_TIMEOUT
1 D1 @# I3 h- \  f; r2 O( H! j+ C        };

( ]% v3 _' N0 ~9 O. e* S' z
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
7 {/ X/ O- G  [        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
9 O6 x# ~4 K1 t: [- {/ E4 t       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

5 @& x8 H. V* V2 y9 b) f
$ l: l! _7 F" D# g6 P1 ?$ D# {        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
4 A+ C! S9 A: W6 J) I; J        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
& P# r2 `: W. a. O        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
+ u* r; `$ o3 q" {& o+ t, d6 S

// Implementation
2 n) n- S& p' S: I! o) }protected:
# J# f6 ~$ r" Q8 |) p  U        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
6 j: i0 [# u* t6 B& y" P        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
" [9 C0 N' D, C) C! Y        bool                                m_bCheckAndRefresh;
! o. m- G: n6 J' n4 {

private:
9 i5 L5 h) V4 ]$ }' u6 {  Q" c. v        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
; @9 ^% K; n( u/ p7 d% O6 V
1 T; \* O. A& E, t; @2 w
) W  z6 w6 P* }4 Q3 T- c};
) v* Q- y. s& r& j( K/ N9 k1 }! U9 E
- d8 p! P7 d6 G, h
// Dummy Implementation to be used when no other implementation is available
0 x) a9 _, G2 B' S1 iclass CUPnPImplNone: public CUPnPImpl
{
public:
: C/ Y: p9 h: B1 v' J7 ]' ?        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
3 G: z- v: }2 [        virtual void        StopAsyncFind()                                                                                        { }
- I: L& a- O) X2 R        virtual void        DeletePorts()                                                                                        { }
3 G" j) Y% `+ u, Z2 C9 @8 I        virtual bool        IsReady()                                                                                                { return false; }
5 U  B6 s/ z. ~5 P" z        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
+ P  Q- h) h& L; J  ?};
! U% r% e6 Q" V% o% `0 _! r
$ l, s, [+ U# ]0 t  j/ p
/////////////////////////////////////
% _6 V- w4 e7 L& x! }0 ?//下面是使用windows操作系统自带的UPNP功能的子类

) W' f: r' _4 a- A. G
#pragma once
/ B) k! K  ^8 A0 M#pragma warning( disable: 4355 )
+ T2 u- c/ g: o, Y3 @

; E/ Y$ E! M9 p& C2 G$ G#include "UPnPImpl.h"
#include <upnp.h>
$ b  r% a9 W2 K' a( H; |; k1 B% W#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>

/ v8 H5 j: [0 I5 H
% }- }3 E1 C9 E$ O) X#include <vector>
#include <exception>
#include <functional>
8 ?1 n: o; p: ~
: t7 N3 Z3 d' V' G5 {0 ^
" d2 {8 ~" f' J- B+ {
+ O9 _; T/ p, j2 T" ~
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;
4 _; c3 O; Z& @typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
2 W$ s% _  J+ v1 Ktypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
$ ]- ?/ d% ~1 Z: ^( Ntypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

) D1 Z4 z5 S/ Q2 V* i) l; i/ _
typedef DWORD (WINAPI* TGetBestInterface) (
& S9 A6 K# n! c0 f  IPAddr dwDestAddr,
: y6 [9 x- l8 t+ k- @+ H" z* l( L  PDWORD pdwBestIfIndex
);


typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
# @- g) s2 @) h3 N1 T0 f);

- M+ e* A) |6 N0 x$ g
4 _4 s. J5 b1 V  d1 V  V  E. ]typedef DWORD (WINAPI* TGetIfEntry) (
' p  n/ I7 G, X& Y) `& u3 P  PMIB_IFROW pIfRow
8 x' Y5 w8 M2 u, G/ Y$ ]9 P);

: a9 T3 \0 H/ O( k& Y* A' A1 u( C; j
, t4 W, w, ^9 e# k' uCString translateUPnPResult(HRESULT hr);
. H& u9 r4 n% d; HHRESULT UPnPMessage(HRESULT hr);


# e" s& F) s: s. W/ z& s9 c) d5 }class CUPnPImplWinServ: public CUPnPImpl
$ u2 ^8 H, R7 I2 H' }; ?3 E, }{
+ w; N4 E4 U' f6 m7 J        friend class CDeviceFinderCallback;
7 f& h! B$ n# Y6 t' }/ X" I) _$ F        friend class CServiceCallback;
// Construction
public:
        virtual ~CUPnPImplWinServ();
+ n1 i0 j& g" X( H/ u        CUPnPImplWinServ();
' `, c- \: Q# ?- o) d5 x/ H

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
+ T/ @$ d6 b$ p        virtual bool        IsReady();
  H% t+ t: W2 \9 x  z# k" x; n/ f3 `        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

& L$ s' m2 H! S
        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
0 F* a; x4 L; @1 ^, M$ T) C. {        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
) K( a) ]3 R' i5 ?- [5 s& i$ ~        virtual bool        CheckAndRefresh()                                                                                { return false; };
7 s: g1 F: P# B9 `8 L5 r

protected:
. ^; K; t0 v" f        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
, y; {( n: _2 g( u6 c+ _1 {        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();


: [; s" v# C% u) z1 [( W* x        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
) Z: \1 R4 j. V  b9 i                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
% a+ i8 J( S$ ]# o                }
                MSG msg;
; E0 o9 q/ _( i7 `0 Y4 c                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
. B, I7 ]+ A. |# R% S0 Z6 I0 S, \) Z* v                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
/ j' g1 E- b5 F                }
- X) C& a; j2 e; U* ?1 N                return m_bAsyncFindRunning;
        }

! }9 @, X$ V1 P7 j0 z
- c( }" t2 S. a, {2 e        TRISTATE                        m_bUPnPDeviceConnected;
# G7 O" W  c6 W, a- E. g! T$ S3 P

// Implementation
  `9 ]7 J: \/ F$ x! k        // API functions
* P7 b" u" d3 [. |- ]- y8 n1 r+ H        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);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
* f7 H0 Z2 s8 v- Y8 _        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
7 W  u/ [9 ]2 z& v/ W) @, J; U        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
7 {: w% g$ [/ w8 ^

        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;

+ \/ F# r2 h9 I  f% l' Y! f, r
        static FinderPointer CreateFinderInstance();
$ B) L5 q( _! U) {6 s4 p        struct FindDevice : std::unary_function< DevicePointer, bool >
( c; k. N" {5 {9 d, M        {
1 k. j/ q/ S3 C, `1 S  g                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
- c: J8 I1 q+ z* d/ P6 a+ s# c                result_type operator()(argument_type device) const
8 M& T/ y! d9 W( `( V$ d                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

0 g7 i2 ?7 s( L3 W) d/ Y
4 V  ~+ Y4 k* }5 H3 P+ w+ K                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
- |8 }" t" H! u; U& r  k
6 N! B9 M; s7 C0 }# g5 Z% d: N2 B) h
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
3 U2 a, p! a0 ]                }
6 s- q2 G8 m( ^+ ~6 T6 f+ ^                CComBSTR m_udn;
0 g/ a' W4 w& M0 f" x. Y        };
7 u! P3 A' K8 v  N1 \% B       
& j. G* C! A9 Z, D        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
( l. d7 M' |9 |2 K8 E* b+ E        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
$ y" p: F7 h! ?: Z3 Y6 m                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
4 a  d/ `2 b' K+ c( f7 e. H4 e1 Y
  D: S/ w# S6 y" T% A+ S
0 _. }9 f( j- e        // Utility functions
: D. S& m7 Y3 E0 p0 P        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
# |8 ~- \$ E2 e& M* u* j- T3 }        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
5 }1 J* S( h+ @0 o( A' S        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
! r. {" ~- U  k- h- J        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
: Y+ Z) `0 v+ G' w        CString        GetLocalRoutableIP(ServicePointer pService);


// Private members
private:
0 d9 ?: z. e( P0 w        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
2 X6 S$ B( Y2 L- h5 `4 j        std::vector< ServicePointer > m_pServices;
; d/ _, W; H1 Z; [1 X        FinderPointer                        m_pDeviceFinder;
, l- c: ]* U$ y% ~        DeviceFinderCallback        m_pDeviceFinderCallback;
& g4 }9 p/ q/ V2 C6 [/ h        ServiceCallback                        m_pServiceCallback;

" `7 b4 }7 ~$ x( [- h
        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
( d0 C. U; k" s        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
: x9 V$ k& |" D1 l& U9 }& @        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
' L5 U4 l9 Z7 p  {7 c/ g: s        bool        m_bSecondTry;
2 X  g6 M, `0 a. s2 C" R        bool        m_bServiceStartedByEmule;
/ y% M  A# G2 B! _/ t        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;

9 A& d' V& l: x  s
/ O/ h+ N# t. y9 c! A};
( ?) _& Z: E" z( D- b; X

// DeviceFinder Callback
! Q5 H9 U3 `: a' E/ K0 Tclass CDeviceFinderCallback
( @. P. s3 _8 K. v: y' n        : public IUPnPDeviceFinderCallback
{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
  U& ?% W5 c+ a( z8 _1 M: s" L        { m_lRefCount = 0; }
- ?7 z/ m2 A: b0 N+ `& l
! c6 l) G( N# V9 C- s; w1 X
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
3 A: h, }" }" G; [+ I. N+ U- Q$ [6 Y
2 @) p% K# I. q) p; x" p
6 c  J- |# L( r// implementation
3 D4 @5 X! n( J+ A9 I. Yprivate:
+ R2 O" l* m, g# N) e        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
" k  I  [& i4 L        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);

) m+ ^0 s; d7 B/ Y/ A6 H
private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
+ C- i( n& A2 k2 m};
8 n* z- B( R% P
  _5 ]; x& X' [
// Service Callback
class CServiceCallback
+ ?* Y! d: S5 g! B' R3 r        : public IUPnPServiceCallback
{
public:
: d4 m/ J  Z, P! l% ]2 B        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
! }# q" s# g7 J7 O+ P0 v' d   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
! o5 q. b  G. S8 J  r   STDMETHODIMP_(ULONG) Release();
) p' g5 |% K2 m/ {" n
( H/ U  I+ Q! A7 R% M1 U
// implementation
private:
( H2 J8 H- ]- X. B9 y+ I! }        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
4 b$ K. \5 v+ Y        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

- y( s6 S+ g/ x% @2 h( b" ^7 K
private:
: y; i3 O! p, G1 ?7 c: e  I" r        CUPnPImplWinServ& m_instance;
! r* r( ^' @. U9 n: c. a7 v        LONG m_lRefCount;
, T7 |- x% s# [9 j- T" i};


) v/ e7 m' z4 V% F7 Z& v/////////////////////////////////////////////////

3 |6 D& O5 f& l, }4 _3 M" T
使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
" W  w' H  ]1 F8 [5 T9 D1 M$ _CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
0 V- b+ M2 i8 n$ _CUPnPImpl::StopAsyncFind停止设备查找.
& O: W; X1 G( n& Q. K7 rCUPnPImpl::DeletePorts删除端口映射.