UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   , i# J% n" N6 o" `/ G2 A& ?. G
4. #pragma   once
   2 [* q. Y) m8 s( P$ V3 m5 q6 ^
5. 
   ! s0 q) O. c: b" s
6. typedef   unsigned   long   ulong;
   
7. 
   & I  @8 l5 w. T' ~; {/ h' @
8. class   MyUPnP
   
9. {
   $ n4 B4 S0 @" B( ^4 ~8 z
10. public:
    ' f7 p& x( z* l! _) X
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    $ v3 O( z8 C/ l5 ~$ E! B( Y
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
    + a/ f6 q! G4 U2 o% T
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    2 y  N  A5 r* }
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    ! a% c% S; U; i
18. 
    ( \6 c& l( p, \1 v
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    . u, x) l  U9 `6 k, P/ ]* c  F
22. }   UPNPNAT_PROTOCOL;
    9 K0 O  k/ Q& Q+ r6 b
23. 
    $ @2 Q; P8 r7 {2 e$ ?. j
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    , L8 F. y3 y5 _! ?
26. WORD   externalPort; //   Port   mapping   external   port
      n$ d3 G; ~8 C; q3 D! ]
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    % y4 n. J% K) Z% X, [* i9 {7 r
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    1 z6 K) N- v6 N! t& O5 t! r1 g
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    6 X6 T+ b9 U6 b2 _) ]' j
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    $ D* g9 ?. ]$ m7 Y0 D1 }# B+ l2 u
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    - H" A* k2 B5 m6 Z' ?
36. void   clearNATPortMapping();
    
37. 
    ! F0 |! `/ `& O! T; }; m
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    
44. void InitLocalIP();
    7 O8 L/ ?( s* c8 w
45. void SetLastError(CString   error);
    / ^( q$ z4 }, O" M# f- I
46. 
    ' j/ u  ~. n. A' q1 w
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);
    / z, ]/ H! h  [  l
50. 
    ! ?1 D5 c, o+ n; }+ D& g. h
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    # Q! B, S' t" ^8 o6 ~7 v9 B- k
52. 
    
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    1 x) j' B& {" [9 h1 p3 v
60. CString m_devicename;
    2 u* X( R$ T4 @; Y  Q
61. CString m_name;
    / A- F0 K, c' p5 V6 e( m' V% A' q
62. CString m_description;
    
63. CString m_baseurl;
    % o4 y6 b; R0 I, S/ i
64. CString m_controlurl;
    + E* S; P0 [9 ~) [" a0 t+ }
65. CString m_friendlyname;
    
66. CString m_modelname;
    % }. k6 B7 ?+ X7 F, d, W
67. int m_version;
    
68. 
    * j& C* d$ k! W" ^! d5 C1 a8 G
69. private:
    . ?% N! w: Y0 r/ R
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    & l/ C# ~+ x! r/ ^
75. 
    , C- s" {  {! |% K
76. bool isSearched;
    * Z& y) W5 W5 ]/ _, ?  H
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   
5. 
   ' o$ h$ `# h0 T) H5 _" U
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   8 c1 }5 A9 p' b+ o
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   * I2 s$ c# J2 N: b+ W
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    7 `& i4 C4 y, x) ?7 [8 Z( O3 F( i
11. 
    6 r. d, F( [5 X: N& [2 b
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    # i, |8 n) @! c, h; l# B
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    + @. e5 i! p! f& C' P3 i% ^6 g
21. 
    / x' }  @3 x. J* \
22. return   s;
    
23. }
    
24. 
    : \8 D" e8 W' G# T1 j( M
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    % C: T$ B& z# w  a" D0 D$ p
28. }
    $ s9 X- r3 ?; J
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    8 ^6 Q; f5 }2 u2 a9 u! F' \
34. 
    7 x: @$ U- L' D+ ~
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    5 b5 e6 R* X8 G: |
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    . _% V3 J' L' A, V( O
43. uint32   ip   =   inet_addr(CStringA(addr));
    5 p% s. Q( n& R+ v0 m
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    7 Y0 J! s+ W5 c
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    ' I, J- J! ?) h5 L
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    # Y$ p) H/ L8 z1 h. A
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    ' y! [# e% N+ {' i& {% A! q4 x
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    $ O( N- W7 C% x- j6 x
57. closesocket(s);
    # U6 Y! ^* m' w+ |0 Y
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    3 v/ `/ s3 \) j
59. if   (!rlen)   return   false;
    
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    - x+ o0 l2 L  t) b) e3 k
62. 
    # i' @2 `  h9 ?; B0 V, S" x
63. return   true;
    
64. }
    
65. 
    3 Q* M: _9 t2 y3 A2 U( L5 U
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    9 b3 Q; n% a  H: I% k# w* M) S- D
69. 
    . M; h; P( V! |
70. const   CStringA   sa(request);
    & n& C, C* o& C% \1 G
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    : Q. P  j, g  A  S
73. 
    2 o4 A% s7 t# w$ o% P, K
74. struct   sockaddr_in   sockaddr;
    / }( r/ R) s$ b& t' h/ g" |
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;
    3 F- a2 l' X% u
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    4 n2 u% G1 z7 @
81. }
    8 D0 p  I; j+ J- w. J5 j
82. 
    4 V! {: Q  |8 {, N3 Z/ D, s" D
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    + B9 ?/ G, Z  ^( X6 X
84. {
    
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    & U9 f& x) M$ N, ~; C7 Y
90. result.Delete(0,   pos);
    ! L4 q0 e. B0 a/ K4 C0 c
91. 
    
92. pos   =   0;
    * L  j0 n" J  j+ p! e
93. status.Tokenize(_T( "   "),   pos);
    4 w+ s; g2 r  l8 \3 J! }7 W
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    6 B1 C( V! }# X; {0 [
96. return   true;
    . e" b* _0 R0 M& v: s6 L$ L7 A: w
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    7 P% `  p( q* D% D
100. {
     + ^" t- E. {7 G0 S: \# B
101. CString   startTag   =   ' < '   +   name   +   '> ';
     / O4 {' w! I4 j
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     4 W: m: q$ l& ^! Y+ L
106. if   (posStart <0)   return   CString();
     ' A& X! T' k" h
107. 
     % |- A- ]+ i9 w8 l8 b
108. int   posEnd   =   all.Find(endTag,   posStart);
     7 ^# p% D4 w4 f
109. if   (posStart> =posEnd)   return   CString();
     " n! ?. g' T% m# E  [2 S
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     0 \* h+ d# k+ N1 n; s3 J
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     1 F8 _9 ?, X, v- Z6 a- f0 e
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     # U8 J8 i0 ]! Z: I. Z3 z
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     + u! @4 l6 j, p. {& }' D
122. {
     ' E9 P+ i8 E; ^, ?% L( \
123. UPNPNAT_MAPPING   search;
     ' c5 U9 \. Y: X8 u
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     8 d* Y5 A$ o% s
125. while(pos){
     6 E* Q5 _6 p6 x' p
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     
131. }
       T& m) J4 K3 \- d
132. 
     ' a# }) _  ^" X; F
133. 
     4 W6 g% Y8 x8 d& C! P) f( Z
134. bool   MyUPnP::InternalSearch(int   version)
     8 w3 I6 A/ y/ f) z
135. {
     
136. if(version <=0)version   =   1;
     & e& {! E* }0 z0 _
137. m_version   =   version;
     & X; g/ A: f' Z! P. Y# Y
138. 
     1 j7 u# j' v9 p1 h
139. #define   NUMBEROFDEVICES 2
     8 t2 u" A/ ~( V- v* a8 |
140. CString   devices[][2]   =   {
     + M- _' I6 p3 a9 a
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     ' s- S( [2 d& m) w) @+ |* V/ z9 G
144. };
     # r5 I. k2 C; Y( Z6 N
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
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++)   {
       l( V# K+ e! R" v; P
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);
     ) j# F1 U! D) y2 m" Z# {# d* @
155. CString   request;
     9 f- k0 z; J) H
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);
     % g( D1 f) T4 b) T
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     5 ]6 W2 N  f& F- p1 f2 @  ^
159. }
     1 O3 w& P: v5 l
160. }
     ( l3 l  @: W* l  e. q( b- i. Y# K
161. 
     $ d, O/ z& [# Y# W) Y. c$ q5 U, i
162. Sleep(10);
       o. G/ l' E' V: Y1 n
163. 
     
164. char   buffer[10240];
     * Q" |! M# L" Q& y- R, B: O8 s
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     ( j" g* u: r0 |* i
169. CString   response   =   CString(CStringA(buffer,   rlen));
     8 ]1 u9 ?+ M. d9 J
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     6 P2 L# B0 B8 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)   {
     % u  o+ S$ l* C1 Y0 B
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     3 J: k) Z. W1 {1 Y  z
178. if   (line.IsEmpty())   return   false;
     # l4 C: W2 X, @
179. CString   name   =   line.Mid(0,   9);
     % G( F" c0 A6 F1 ?. K6 Y! s
180. name.MakeUpper();
     & c( o( w- N8 U# N, w
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     - u6 r* O2 }/ _2 a
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     5 ]  s$ ~' m% b, g, @
186. }
     : f' `- h, b+ {  c
187. }
     ( ^  z2 a4 b5 w# f$ u; q7 b
188. }
     8 C4 W. m3 a- b3 }9 D6 `
189. }
     ) p3 i1 w9 O3 A
190. }
     ! y0 k. B8 |9 d8 ~% z; a* j
191. closesocket(s);
     $ K  U3 s! S6 I( h3 @3 z  l
192. 
     
193. return   false;
     
194. }
     * m% A3 ~5 z' L

复制代码

天云上人

以下有关upnp的接口来自emule，
3 r: {8 u+ C, d/ L7 ^9 V
$ v' K8 o& j  {1 ?
///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

+ K0 W9 H0 f1 }* X' z, y' P
$ |* U  {) x2 g( D#pragma once
#include <exception>

5 Z. `3 }; B" @( W) T* w/ O
: K/ O) S( h' B% [2 z( N  enum TRISTATE{
        TRIS_FALSE,
" j7 o6 R  u$ w$ U& p        TRIS_UNKNOWN,
0 ^1 C. f" e/ f) |  P7 ^1 X4 J        TRIS_TRUE
};
2 _* d! C' q. \* d! G/ C: j  E% {

# ?4 J7 @6 h% }, o. V) Renum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
9 n" t' V) i& S1 s" s- a# b        UPNP_IMPL_MINIUPNPLIB,
6 q4 A1 `! Z! G& ]2 @7 {        UPNP_IMPL_NONE /*last*/
" {  Y1 w4 o, Z  X; M};
- J% U: C; X4 B3 u$ j# R: I


. Y; c+ Z. L* A" h
8 z' \/ e$ A: N8 k  M2 W; L4 A- Xclass CUPnPImpl
# `/ q7 `' Y* b2 _{
' f$ Z2 j  C( t9 R- t+ b2 Epublic:
        CUPnPImpl();
5 R5 D# r  P9 }& p9 g        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
0 o6 M. \% N1 |# y3 j$ U        enum {
                UPNP_OK,
                UPNP_FAILED,
/ g. z% {; o. U2 Q9 n                UPNP_TIMEOUT
        };

6 y2 D, {3 x+ ?* L
0 q9 K! i+ S# k- d" `        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
, k6 V. e7 w2 b% r# Z; h        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
( t  _) m7 t& o1 P( D! s7 R! l

* J: j. j( l6 J- i1 n3 Y        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
) l, W3 n( H- }! {9 p. Z' Z        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

" z8 h8 e  }* f" A
9 O' F# k# R1 c6 X# Z( ?- r# i// Implementation
protected:
& n0 M. T  T/ g$ N8 }+ x# |8 R1 [/ s        volatile TRISTATE        m_bUPnPPortsForwarded;
2 h& L* }5 m3 I        void                                SendResultMessage();
        uint16                                m_nUDPPort;
% o, ]# C/ v8 S6 t* I        uint16                                m_nTCPPort;
  I2 u3 Q5 z; }- C) [, J7 m8 ^* l        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
* G7 T% A0 N* H& M1 [
* i4 |  Z, R$ w+ w2 d- d
private:
2 _1 g+ I" l9 h5 }$ ^* |  f        HWND        m_hResultMessageWindow;
' X  V4 J% ]% F/ g8 L& R6 f        UINT        m_nResultMessageID;
& A: M- }) w7 M% a! I
& _2 H: X! b2 e: H7 V, T. w
  S/ O2 U- c8 M: |2 _};


( _$ \- @% J+ O" `// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
+ M) R8 v: |8 jpublic:
8 j, v3 |9 j* m5 H! m4 q, n, q        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
( B# d+ q' Q, r, f4 ^        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
( w( k$ M8 d: a4 q7 U        virtual void        DeletePorts()                                                                                        { }
0 F* t9 G# A4 Z; x0 W        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
/ d/ |" H! e5 ^2 d8 o0 s

* Y' G' y4 u7 f9 ^/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
. `6 Z2 A# b# ^8 g6 y

#pragma once
, F* b+ C! h) y) J' |  [+ g/ S6 X# S#pragma warning( disable: 4355 )

( a  k- \, X; R' z! w
# ^* s# a  \. S9 P+ @#include "UPnPImpl.h"
#include <upnp.h>
  W6 J& a# v+ y; k- w#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>

7 r1 a' Q  X% S% [
#include <vector>
#include <exception>
#include <functional>

7 A3 `4 m: Y  b6 t$ \

1 {' F7 G! j+ w
4 v" f4 A! J4 L% p! L- t. Wtypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
& f* P8 G4 k$ Z# ?# x1 \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;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
  j( w& m/ I6 }1 c" i0 \typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
( k  ?4 Q* o  |, A% M) G/ H. i
  V  P$ n0 ^6 B+ v' K
9 j# W4 @" k# i* Btypedef DWORD (WINAPI* TGetBestInterface) (
8 Z0 a; i$ E# j" b( {  IPAddr dwDestAddr,
, ?* m" L, Q, j& `0 C! ]" h  PDWORD pdwBestIfIndex
7 T& ~% j. a1 M);


typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
! C$ P& p; v. J3 k8 X& Q9 P  PULONG pdwSize,
& v$ e! s; G+ C! \6 N) _  BOOL bOrder
);


typedef DWORD (WINAPI* TGetIfEntry) (
8 W) g5 N* }0 N. T2 e% x* {& w4 M/ ]  PMIB_IFROW pIfRow
);
9 B0 b2 g, h- i; `3 a1 G- M
. V7 p" ]) T0 S
CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);

9 n3 j* ?1 L& h1 t# r/ x2 j
class CUPnPImplWinServ: public CUPnPImpl
" s- F% w9 i, T{
$ v# ~' \* H/ k8 w) O3 l  |        friend class CDeviceFinderCallback;
% ^/ M' t% N+ v4 z5 Q3 |' t% C. ^        friend class CServiceCallback;
' r& ?6 ^3 x( f& p6 u// Construction
public:
        virtual ~CUPnPImplWinServ();
, n! i) p# G" l; s; p* i# l        CUPnPImplWinServ();


- _+ m6 a5 K% o$ Y! u6 [* G  a        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
, a2 [7 c/ k/ r/ E        virtual void        DeletePorts();
. J) ]- `* Q& G  L/ r        virtual bool        IsReady();
- V& b. c; i0 L* Z) A        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


. p- P) L: r# r9 o        // 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; };


protected:
( k* b8 h$ J& D" a8 n! f        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
; A9 E- m  G0 B, n: ^        bool        OnSearchComplete();
        void        Init();

# Y( ]. [. C$ X# V- \$ F5 @
" J6 m7 q: V5 D; u5 l4 E        inline bool IsAsyncFindRunning()
* H. |7 _' K) H5 _7 ^- Z; A) N5 P        {
2 f7 l* [$ _2 x3 v/ _3 @                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
6 l5 K& f6 ?. e2 r1 D  }                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
* h6 s" u+ ?! P$ e# j  m( {                        m_bAsyncFindRunning = false;
                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
$ S; ^2 p$ e7 Y+ c8 F                        TranslateMessage( &msg );
9 E( V- O: ~8 d: y* K( `2 }                        DispatchMessage( &msg );
' {5 M; V6 u; q: o                }
                return m_bAsyncFindRunning;
' n2 b2 w, N5 G, ~  U; g        }


; V( ~8 Q# g- _        TRISTATE                        m_bUPnPDeviceConnected;


// Implementation
        // API functions
        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);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
  V4 K. w6 }- a( m0 A        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

4 a. q! ^: u& C: v7 ~5 ~: c9 q, v" T' K
        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
1 l$ n6 w6 O& U! P5 Q3 F
* L4 B$ o( ~% k' f+ N
        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
* Z" u! T' {+ H        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
# U+ q$ G+ ^" h6 G7 T: I' R8 d                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
: z; Q) |. o9 Z, B                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

9 X% {& ^0 o  O% p/ o/ }4 w  q
                        if ( FAILED( hr ) )
! p. }: `( @8 m& D$ U8 K                                return UPnPMessage( hr ), false;


                        return wcscmp( deviceName.m_str, m_udn ) == 0;
  A1 H* k) R' f0 t' ?) L% {                }
. j4 W9 I( w$ a* o                CComBSTR m_udn;
  b5 Y, M, M8 u* [5 [5 k+ u        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
8 t  M4 ^( q! z        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
6 _% S  l% O2 I3 ?        void        DeleteExistingPortMappings(ServicePointer pService);
* Y7 e% @$ `/ B. I        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
# a9 S3 d% `* L. [        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
5 w" s( g( s$ S7 ?                LPCTSTR pszInArgString, CString& strResult);
- H9 }# \" L+ T. C* D        void        StopUPnPService();

( {; \2 T7 t4 E# D6 m& U. z2 @
        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
# I- g0 k* I) |3 S        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
. X" {6 v/ t2 e6 h* ~        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);


// Private members
private:
4 [1 [2 q- l( W0 V# k5 g, [        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
0 x) M$ t" o( n6 H. v        FinderPointer                        m_pDeviceFinder;
4 \$ \9 P8 T3 |; c- u# ~: k6 E) {  w        DeviceFinderCallback        m_pDeviceFinderCallback;
( I4 a' ]% k, ^9 J& ^  r        ServiceCallback                        m_pServiceCallback;
1 S9 _; [  n9 |
% Q- N, Q. R0 F. W( `; Y8 o
        LONG        m_nAsyncFindHandle;
; H# j$ h2 K  X) X3 E& i9 r        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
; y6 z0 q& j# |0 t6 r        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
6 V9 L; @8 e( u$ C% `
" [$ |7 n" u2 A
) Z- ]& t. K, a8 k$ q, d};
7 Y. Y, T! [$ ?$ {1 G

// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
) E& Q" N3 g) ?; j1 F6 }$ D1 ~% q                : m_instance( instance )
1 o( p7 _! d8 h        { m_lRefCount = 0; }
& E- }2 z* |$ H: W
5 k1 V6 G: N4 D
9 G5 p' a7 k$ N6 D8 t5 n2 N9 z   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
& v- ]" K6 Z; c/ `5 ^! V   STDMETHODIMP_(ULONG) Release();


// implementation
private:
% {& o+ n2 a) @8 e        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);


* f5 A- T* [1 ^# c3 _- A& U; K% Aprivate:
; ?! H  z7 L: a$ `        CUPnPImplWinServ& m_instance;
; s! p4 B4 D/ |- I. g        LONG m_lRefCount;
- d; c, l6 U$ Y+ E};
3 j& Q2 t4 Q3 `8 o. z& z0 Y

/ a( M9 o" g- w3 j4 n// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
{
' m8 w* y2 h; g' u& E. Bpublic:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
* l9 i" Y$ \+ J  d8 L- ]* J8 [        { m_lRefCount = 0; }
   
# Q- R. V6 x9 r" i9 l2 J2 W2 ^   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
6 R3 d4 i6 r4 Q0 Z   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


% n1 E3 s4 A8 [: q1 B// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
$ y- r8 w4 o; y' A) W. _1 F
: R; u  Y$ p  O1 U
# L, o2 ?& G5 }) `$ ^) ~* aprivate:
1 a3 B3 D3 T' `: s        CUPnPImplWinServ& m_instance;
1 }0 R$ i* {' r8 z4 @2 E+ |        LONG m_lRefCount;
; X6 I3 z$ e: j, f0 R! S. m' k};

4 ?5 t& G4 X% E9 v
/////////////////////////////////////////////////
2 k$ z' ~0 S) G

2 D8 q' C, O( W/ i使用时只需要使用抽象类的接口。
$ v) U" |( ^& cCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
' y. ]$ _) p: a  b  PCUPnPImpl::StopAsyncFind停止设备查找.
( p6 J; i' }8 B) dCUPnPImpl::DeletePorts删除端口映射.