UPnP

天云上人

/*uPnP.h*/

1. 
   1 h6 \  X8 a6 o& k
2. #ifndef   MYUPNP_H_
   + l! I9 z- J+ D: t
3. 
   . o; R; e4 I/ F2 `& U4 L
4. #pragma   once
   + l9 q3 _: W0 u. S3 i# h. V8 }
5. 
   7 _( }' [& @5 T! v9 i
6. typedef   unsigned   long   ulong;
   - l5 o! h7 s9 e8 Y* P/ }+ B0 f  {
7. 
   , X2 l3 ?# ?. H
8. class   MyUPnP
     U, c9 E5 g. Z. f% @) j
9. {
   
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    , s$ l5 _% d0 c
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    + G* T: J2 A# X- U
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    7 m1 D0 j# y0 {4 g
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    4 F6 b, \9 g: A+ c9 [) |4 Z% R+ w
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    + J( C$ B% w2 g' E) L* E
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    $ H" t5 Z% C7 k! n( s/ d
22. }   UPNPNAT_PROTOCOL;
    8 g) B6 d  m9 `. v2 n9 ]9 V4 r
23. 
    
24. typedef   struct{
    ) H; A/ d" y5 h/ K0 O
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    0 }* c) n/ ]( E4 w
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    + `( q3 w0 j) D* d; K
29. }   UPNPNAT_MAPPING;
    
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    : u2 S; b, z" v
33. 
    ! u3 `( g! Q4 G* g
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    ) D( I7 [# a3 z% n, B
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    4 I' K* b; T& H' u
37. 
    2 M8 [+ J4 ^" g- h/ [
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
      u/ R/ @' m: }2 _# ?
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    $ v0 M; r- P& I% T: q
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    ' Y' X& `; e# C
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    ! g8 g8 k, U# W8 i( p7 B
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
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. 
    ! g& E5 X3 l, k: f6 C+ g8 o# W- j
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    * p% W" _; |/ P7 Z
59. bool InternalSearch(int   version);
    + N4 R: Q; q0 a) v: i3 x. h
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    5 J" ^# B( R$ _" m% j
66. CString m_modelname;
    ! P9 X9 M' E; @) C
67. int m_version;
    
68. 
    8 }9 ~+ `& ~+ Q  I
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    2 R' f9 |: w8 w+ w
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    % d7 n8 E8 h0 \' J; r* N
74. WORD m_uLocalIP;
    4 X# Q0 A( N- F! ?+ k
75. 
    
76. bool isSearched;
    7 g% v1 {3 w4 V1 Y3 P
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   . X" r! z6 G, A0 q& E
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   % }+ @% c! m" j, U9 s
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   6 x" [) j6 l! P6 v/ [3 B
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   $ w; ~8 m/ [0 x+ ?& L' i" e
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   " J) D, s/ V0 j: K3 D# f8 M2 [
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    % L% E; l' Z  a
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    " W; P* |8 k- A" s  b7 z
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    / T- ~/ o* E! U, f5 g  o
15. 
    ) M1 W; Z! y7 X+ T8 J& a' c/ u6 d
16. const   CString   getString(int   i)
    / L& N+ G; n7 t" I: A* H
17. {
    
18. CString   s;
    " Z% R4 j. H& {6 u# t: h5 t
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    4 j1 A/ h7 h* K+ G
22. return   s;
    
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    / r% m! G, b. H7 i; n1 j, t
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    " A8 z" T, q+ T9 t
28. }
    
29. 
    ' \( T/ d4 D! |, C
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    1 Y2 O1 ~# \9 q0 R
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    & Z# r9 O4 ?  g
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    + }! E# `* O* K  h) ]2 B
38. 
    ) H9 H( E/ f0 Y( s5 j, P
39. const   CStringA   sa(request);
    & t* x# y/ i, h' r
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    % S+ o( u# z$ }
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    3 X- G8 {- P. N1 X% l* z
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    ; L# P9 w! ^3 j- F6 X' u# B* p
50. u_long   lv   =   1;
    6 }8 w; C5 e+ D) Y
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
      x# a6 B, P, A" h" v5 U& ^" \  X5 w$ H
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    ' l, a; I5 |- b2 O
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
      O1 N% K" {" |) [3 [, O
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    . P+ L" v7 x0 Y5 J2 n
59. if   (!rlen)   return   false;
    9 m  ~1 m4 {6 @" s' p5 o2 `* @
60. 
    / H3 N% j8 ]0 v' ?4 `
61. response   =   CString(CStringA(buffer,   rlen));
    / j) Y5 ^. h9 Z7 c$ p+ v& H( [
62. 
    ' @4 v: A/ `3 u
63. return   true;
    # j5 {8 z' X; H/ T
64. }
    
65. 
    - e8 s2 q# R+ N( R
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    4 Y( T2 w; S) u7 w- p, j- |
69. 
    : Z. E9 F/ ]2 }/ s$ L1 h' c
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
      T8 f  s2 _$ |1 {, V
73. 
    
74. struct   sockaddr_in   sockaddr;
    + g) l% N# w' v6 \% A
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    - V' b1 B& A' L& }( K
76. sockaddr.sin_family   =   AF_INET;
    3 ]' J7 o) b+ ]  x7 z5 X; G
77. sockaddr.sin_port   =   htons(port);
    / d9 _+ c0 C- E% y' B- t0 d
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    - g5 ?7 Y2 n* d/ A/ }  l* \
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    2 P2 E& X* o9 |
81. }
    0 ?& o( {( U$ C6 B: y; A9 I
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    0 j: ^5 n/ q! b; m
84. {
    
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    ( {4 j( @  ^) c5 a) V
91. 
    ) N" l. _/ o9 U
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    $ L$ J% |" M9 e4 b& I
94. status   =   status.Tokenize(_T( "   "),   pos);
    / r4 J+ [5 X( k! x! \
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    ! M2 P6 U" z0 D, H
96. return   true;
    # N* \) ?3 K  \3 A( g: B
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    # B4 G1 }) F% G# G1 X
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     0 x1 x4 [9 Z! s, {( Z9 k
103. CString   property;
     6 U) U4 o9 g, n" ?1 V/ l# M
104. 
     # f8 o) t) W8 y9 w6 \6 r' X
105. int   posStart   =   all.Find(startTag);
     : s) V8 w: O/ g- {
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     - a* C- [3 z1 q; E7 ]
109. if   (posStart> =posEnd)   return   CString();
     7 b$ S1 X2 W7 U8 X) m5 G
110. 
     1 D! K' X. T6 k( Y' h) c
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     8 @) l3 |' _) W9 N; X: J4 Y
112. }
     6 X7 _; r% M7 {% J! {! `
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     
120. 
     ( k' x9 l1 T2 H! Z: P
121. MyUPnP::~MyUPnP()
     / Q4 U9 J5 R" L
122. {
     + b! Z$ t* n, m/ R
123. UPNPNAT_MAPPING   search;
     + ]7 }$ u/ V, K+ O* B' Q+ F" R6 D
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     ' G1 x) c7 V% j) P& ?, K8 k
127. RemoveNATPortMapping(search,   false);
     ) ^2 v, M' I7 b# T7 s1 E! L* Q
128. }
     : b% L$ \/ W- U  G0 [# V# Y
129. 
     7 y! Z5 v) x4 {
130. m_Mappings.RemoveAll();
     ; y5 ]% ^! X% Z' W
131. }
     
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     1 O% C- A( o, }$ S
137. m_version   =   version;
     
138. 
     % A: l; k! S& E7 h+ O- f
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     1 B5 l1 Q5 ^( u( |2 }& d; V
141. {UPNPPORTMAP1,   _T( "service ")},
     ! K  q0 }: ?. z
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     2 H" q! K) l0 S. b: [+ r3 w. |0 N$ b
145. 
     / I; ]  ?' m. H* {/ B8 J
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     : Z  W( N0 m7 ?* P- P9 r
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++)   {
     9 h7 m8 ]% |) d* h6 U
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     # o/ c' H, L- C; J4 M- L4 P
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 "),
     6 _- @; v2 K" k- @
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     5 S7 ~$ k+ A' y& r" v" `- i
160. }
     
161. 
     
162. Sleep(10);
     $ O: f( c- E5 _  f( r0 Q6 ^/ Y
163. 
     6 H5 }' P  E; d, s3 n: s
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     7 B1 [! k0 Y! m
169. CString   response   =   CString(CStringA(buffer,   rlen));
     ( m4 a  U4 e3 N/ ^6 U" g. q
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     , k6 H. b, V3 u
172. 
     6 y% v3 j4 k9 l% K1 h: a' `% J
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     3 {( b$ z1 F( J
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     * C8 W7 I7 ~$ o: U6 D% m
176. for   (int   pos   =   0;;)   {
     
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();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     0 @, }1 L  F; s1 x% H& P
182. line.Delete(0,   9);
     ' w! j1 f$ _& R- \) b
183. m_description   =   line;
     9 h: ~4 {! I% S3 I. D
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     # E& S2 Y, f7 U9 p- M+ F. Q  d( I
187. }
     
188. }
     7 `% ?; d, [+ }$ _$ z# V
189. }
     - y3 q1 ~/ Q: c% Y, J9 h' ?( x9 T
190. }
     
191. closesocket(s);
     0 s# ~/ A+ Q8 G2 V- ^1 t
192. 
     
193. return   false;
     - Y, X9 Y* l2 ]$ u+ a% V7 T' o2 C
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
8 ~# x9 @( ^4 v

: u; C# x- i6 `7 W7 Y# }///////////////////////////////////////////
# A- y. w% ~9 A9 _' K//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

/ y' I7 i; C; w( D
1 p. w* Q. A1 [! [8 f#pragma once
#include <exception>
5 o' z2 M9 h( g9 a) t, k( |) @+ f2 p

8 T' k% ~* a2 n9 ?  enum TRISTATE{
        TRIS_FALSE,
% m5 i' r# e0 l        TRIS_UNKNOWN,
. H+ R) R$ N: e9 ]' `        TRIS_TRUE
};

( E# P1 i0 `3 |, M0 b4 ^8 d
' `  \  y' S5 D; I2 ?, O& Fenum UPNP_IMPLEMENTATION{
9 @: A! \! T+ m  F* p8 B4 k9 ^        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
3 m8 ?0 ?7 W; {5 }        UPNP_IMPL_NONE /*last*/
8 D% O3 s+ @# A4 g9 C; a  ?};

0 T" M; Y' `' \$ k0 y6 `6 X
7 F" }) w% V0 @1 D' M1 `
9 a, K. a) Y4 I( O3 z8 }
class CUPnPImpl
. W% f" W$ E% B! _0 u6 }{
public:
        CUPnPImpl();
        virtual ~CUPnPImpl();
' Y4 v5 z$ y1 r4 [2 ]        struct UPnPError : std::exception {};
4 m! r$ \% f) U3 j. p$ g; {# e        enum {
                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
        };


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
, y/ g7 N9 v5 O, K        virtual void        StopAsyncFind() = 0;
. a/ P& p7 I7 ~        virtual void        DeletePorts() = 0;
& C) G- V5 W9 a, n+ ~        virtual bool        IsReady() = 0;
0 |8 l6 T( J9 n2 {        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
4 `: h5 ~6 }) E/ [# E" M        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
3 p) R- j# i* V8 J% T  w0 Z        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
3 P" {: O0 }8 ], u
3 m- g7 \6 ?3 b" V# u6 V
+ U: o3 U! d( T3 y5 u2 v// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
4 [9 n( w5 X* d4 J( |        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
. J- h. q5 S9 u1 J# T, I        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;


private:
# x0 c4 z. Z8 _# @! b) M6 a' F! d' w& d        HWND        m_hResultMessageWindow;
" [$ |) x4 n* o" J+ C4 Y/ c        UINT        m_nResultMessageID;
3 P  F" e7 N- O5 n$ ?. w7 h
4 U  _4 |- _. R9 E/ M; ]
" f( M) U+ m) H$ V0 u};
: }0 l% A9 \+ `) ]0 L3 O( U

// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
1 G) a; C0 O0 r: _: h" k% N        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
! {( \) M. ^% }* x0 [        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};

; o6 v; q  L0 K" |. t8 Q
/////////////////////////////////////
% V$ m, _, q6 a1 ]: k7 \//下面是使用windows操作系统自带的UPNP功能的子类
. I7 I7 B1 v; p1 l1 Q( T5 y0 j$ P0 N
# k3 u. u! R% s
#pragma once
' l! q. q; [5 g. `) g& s* ^: E#pragma warning( disable: 4355 )
  Z% R  d. Q' h
$ a9 l3 s+ g. o' H9 V
#include "UPnPImpl.h"
+ j' \; ?! b# W3 t$ F9 h/ {#include <upnp.h>
1 j$ M8 x5 R3 B% b# @4 k2 z- K7 V#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>
; n4 ]. h( `5 s4 D* ^

! b* `5 B9 F! d#include <vector>
#include <exception>
/ M, H& I3 v8 |! z* X& D( a  Q4 h#include <functional>
4 n# W2 b2 n# H8 d1 |3 D
' q7 Y7 d% \5 o+ @& U$ i


0 r# y. Y2 _* e# c5 Z9 Y, ftypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
, v0 j- L4 s3 h9 M  Etypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
  K# Y  t5 H; `- q; {! ctypedef        _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;
' G7 f/ `7 F, _5 x9 n: Utypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

( _+ I$ W: {2 {% r$ c
typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
3 l) E5 w5 A+ O2 ?5 Z- ]);


5 [* e/ y' v; M: h# etypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
4 J: ^* u' _; h  PULONG pdwSize,
  BOOL bOrder
3 {& m8 m  y+ H. y: O' t/ g* \);
( C! I6 h: l' O. @3 V$ p: k9 g
2 b! f2 F, f7 J
. g( C* |# D5 z8 l9 Vtypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);

% {) O; B& u* F: ~( Z* h
CString translateUPnPResult(HRESULT hr);
& K" q! u+ n1 }; ZHRESULT UPnPMessage(HRESULT hr);
& t4 q) B/ J3 `# Z3 e% z, j
$ J2 A- C7 \' n; @% {* t" S* u
class CUPnPImplWinServ: public CUPnPImpl
{
9 D( @; _4 s1 X8 w/ r. w        friend class CDeviceFinderCallback;
! p( ^" T% a- a9 F5 e. H% N+ d. [        friend class CServiceCallback;
// Construction
0 {. B) W5 K/ ^* ?0 r9 W% Npublic:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();


" J! E! X& @! f+ l" T' [0 p) s        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
9 K/ m  R7 [8 _8 F* h( g3 o/ |        virtual void        StopAsyncFind();
/ P+ P- l/ l% ~: o% g% A        virtual void        DeletePorts();
: V7 \% o( o- ^7 P% x        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

$ K% f0 ]8 {& `0 ~8 o" C, w
7 ?) p2 l, b' Q, U5 n; w        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
$ Q9 D8 @" D2 r5 R$ _/ s/ l! ~2 }        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
$ p/ J1 q9 |$ b; I        virtual bool        CheckAndRefresh()                                                                                { return false; };
) c( v$ T7 e. K. }  _! x4 A4 {
! d3 a. _  }2 j; H
; ]( }: u" C$ a: }3 }protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
6 a& C2 ~+ ^; n        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
  l; V! E# F5 l: O        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
8 ^* P) T5 s: M        void        Init();
; q. f0 j9 V6 w- a3 t

        inline bool IsAsyncFindRunning()
$ t" F# N# d0 ~" s! t9 d5 U        {
: E( X& H8 Y) N6 G                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
. d! |6 ~9 G  h+ N% ^7 R- S                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
5 C* ~( i! [5 O8 j, E% x* B                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
& D9 K2 n7 B1 i                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
) j& \: _1 L5 O/ e, m9 f        }
" `& W1 Y3 R. }( q9 U- Y
: O$ p' C$ K! P* ]" O8 k3 S
        TRISTATE                        m_bUPnPDeviceConnected;

8 F0 A9 T( @% \4 A8 H
# X; b2 U* i' v1 M1 s: y" u// Implementation
  a% M; J1 @6 |. T. |* Y        // 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);
' A) _2 o. X5 j! b        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
" I6 B' P7 Q4 p( h7 c! T# @        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
0 _* R" G5 g1 N. Z        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

6 U* b" O! A- A! ^" m( v
( Q5 c- I9 S, \- x' M; V- A        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
6 m/ N6 o0 D* F! i% i  ]' E$ E1 g

        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
  Q' @5 s/ e/ o. \2 v& R/ O8 {( t        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
4 d0 u0 T! e( B3 q9 L! T

4 U/ o4 L0 ^( m# c                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
        };
. c8 p0 y2 ]2 U. `( o       
  i& T6 D( g. M0 W' ^% m7 l  o9 T7 P        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
( t! o& J+ C; E        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
/ I) }$ l1 f3 a6 o% p: T, O( A        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
6 V7 `# J$ n3 d% G                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
* K: p) W: o5 O4 V$ H
$ ]) T; E3 F. t9 {
- C. A. k. {8 ~& Z0 z8 n" r7 s        // Utility functions
% u& h5 @! L: S9 _+ W: m  V        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
  j6 c% ]7 ^  {6 M; w: k4 i        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
8 c6 [8 f5 @4 l8 e8 X        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
2 F* Y" D* f0 f7 _) A% b        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
% ]% ]# \: |7 o        CString        GetLocalRoutableIP(ServicePointer pService);

0 R% W$ F6 e) [; P2 a( w
// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
& s- Z, M4 r- z: g0 G( r        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
, v9 K3 y8 @. g1 |! b        ServiceCallback                        m_pServiceCallback;
- ?8 M! V* N6 a0 f
" v  {3 a& w/ T0 V' f: a" y
; l$ b. }- p& {8 {) \0 ^9 S        LONG        m_nAsyncFindHandle;
# k2 s/ r, }1 B        bool        m_bCOM;
/ S1 o1 c: O+ G% a" d        bool        m_bPortIsFree;
3 S8 a! l) u7 X% N8 n/ k: `  A        CString m_sLocalIP;
* a4 P/ u$ D; s4 f3 B        CString m_sExternalIP;
! U/ J+ q, u# j, c# {        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
$ j) r6 ?! Z4 k5 p3 S        bool        m_bAsyncFindRunning;
8 \1 V6 w- ~  N: r; ^7 T        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
$ v% U% Y  p) G6 \, A! d. B        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;


% j8 h  O% m0 y};

2 N8 y1 e3 l9 y, r9 J5 g. N% Z5 I
3 r& T  U% n1 }8 W* \. R// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
/ B5 a) M) x; j) U$ s{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
8 s) k# x; e$ g$ v                : m_instance( instance )
2 s; ^& n" t9 U0 K* o7 r* G        { m_lRefCount = 0; }


' [+ v' m6 G7 [* A   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
, e1 G5 ]8 M9 u- J5 X+ X) G6 ?   STDMETHODIMP_(ULONG) Release();


/ ~$ U9 ], B' y0 q( `) x9 E// implementation
% ^0 v( S0 `% y  W" b" z& H* v6 `# `8 Fprivate:
% m+ _8 d7 T6 F" n2 q+ u, ~9 z8 d$ w        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
7 }0 X0 \/ a( g' {" |  y. J1 v        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
/ A( @8 w$ j5 q2 N; d, V" ]

private:
3 _8 {% C& Q6 n2 g        CUPnPImplWinServ& m_instance;
3 W" D: B2 F' U2 {9 Z+ p% B        LONG m_lRefCount;
};
4 [$ c! Q, u, h+ C
- G) S5 x' w. h9 l% z1 Q
// Service Callback
) x9 R0 y& w) i8 M! [. y2 @class CServiceCallback
        : public IUPnPServiceCallback
{
public:
1 B" }" h# q2 G- Y        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
. J# b. }& m4 H, M" c        { m_lRefCount = 0; }
7 u' [! T$ W. M7 G9 @   
/ c1 j- z1 Z( S2 M5 T$ C3 M   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


+ o; ?5 s* S; K  h8 [0 C" T! @  W6 u// implementation
9 K" Y: S5 A2 D9 Pprivate:
: B6 z5 {3 M' E+ E7 j        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
; K: V4 L' s" n' _* J
/ l8 a) Q8 R' d# l
2 U; y8 t1 B+ A' _private:
1 Q% y; X4 E+ m7 ~        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
, e, q. {( ^  a9 R) o( N; z" p

9 M+ s9 u2 z. E- n) ]/////////////////////////////////////////////////
$ l$ m1 n3 g6 Z! M  j2 J

使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
6 ]' S# N4 c+ |6 a' BCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
/ Q6 e; q1 Q  Y4 c: I5 a8 X7 ACUPnPImpl::DeletePorts删除端口映射.