UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   " A# ]! a6 E" j' Q6 H) X3 A7 [$ O. F
3. 
   / d9 f2 r: p; d7 h/ `- v
4. #pragma   once
   % N, K& y; y' r1 h! j  \
5. 
   
6. typedef   unsigned   long   ulong;
   1 M/ m8 J( F% m: J8 W
7. 
   
8. class   MyUPnP
   ' R+ R1 F( c5 n
9. {
   
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    $ F$ n0 J- W( r4 Z( I% c
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    4 Y$ H/ d/ Y9 ^$ l0 x4 t8 x) D# k
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    
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;
    , H2 K) b. L) J  B
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    1 r% t/ y1 M$ B4 k0 ~/ D* a
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
      T0 j; Q- L! ]) X
30. 
    
31. MyUPnP();
    $ K, b- T1 Y) v5 \
32. ~MyUPnP();
    . K5 ^9 C* T/ w% w+ G: L$ u4 C; n
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    2 z  Y8 f% o3 t7 ?3 v
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    8 F3 S& H$ @; d  O0 n0 m
43. protected:
    
44. void InitLocalIP();
    : _2 i8 q$ T) U2 Q: {+ o+ E
45. void SetLastError(CString   error);
    2 I5 `  ?9 W8 V; ]4 ~
46. 
    9 B8 u* V5 Q* a+ _4 ]% {$ G
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    $ h/ B% W$ [, S5 s5 b* o* g
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    7 O5 k# n" j1 g
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);
      i, Q9 w4 \+ ~
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    2 y; K  H! h9 P/ E, ?# t" _& A6 X
57. 
    " G% J. ~4 |5 ?
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    $ x% [; d) Q1 I0 \% x! `
63. CString m_baseurl;
    
64. CString m_controlurl;
    " }. Y" n9 D. e3 a5 P" R& @- i
65. CString m_friendlyname;
    / t% r$ Z' Q' _7 |+ j& [5 ]8 Y* V7 I1 H
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    6 Y- w  q( d( v
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    " Q9 X* }! {# u+ H0 Z- j. t* d) |
75. 
    1 Y( K% {3 |) {9 a" n0 R, t
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   : j1 K' d  e) u, a& |9 v6 u  w: N
2. #include   "stdafx.h "
   
3. 
   & d  {2 [& K7 |3 B, Q
4. #include   "upnp.h "
   - c$ J6 ^9 Y# t0 Y; i. @
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   # D' b7 u% C1 P7 e
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   / o! x. Q% c0 a. `2 r4 X
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   4 U) c, y, G+ h$ ^9 H3 Z$ w
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    ' ], X; {7 D9 q0 s! N
11. 
    % |4 s9 J- V. S7 u6 n! ?5 C
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    1 M0 S4 f( q& v* J
13. static   const   int UPNPPORT   =   1900;
    ; P& g9 _! k9 B3 y( H# ]+ N# Z
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    : A! c! f$ W/ Q) E4 W
17. {
    4 E6 ~/ y! Q) D( b* S
18. CString   s;
    
19. 
    6 {9 ?( ~* B9 J4 U3 c( k0 z* k7 e& N
20. s.Format(_T( "%d "),   i);
    1 k" A+ {) N8 W6 r
21. 
    , ]: w* o9 c$ I. Z" @7 E2 X
22. return   s;
    
23. }
    " f' Q+ _  {8 d) j' S& O
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    , Q! f* `0 o" N. D
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    & U2 u( L' r! Q, H
28. }
    5 ]8 C  N& C+ s$ S
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    9 x! t. l# G( \  Z% R1 l( Z& `5 {
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    % N/ J( l7 a% f% Q4 q0 H
37. char   buffer[10240];
    
38. 
    3 @& |8 t) V+ P* o" \
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    4 u% O- M7 t8 Q9 j7 F# n
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    1 v1 F# k+ o1 W: R5 L( v7 [
44. struct   sockaddr_in   sockaddr;
    6 V7 v1 f# s/ A8 @2 n( W
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    2 w7 }& O9 e0 m7 G( B
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    $ r/ O9 A; C3 R7 S
50. u_long   lv   =   1;
    5 D. J0 R) N6 @8 ]: v; _! P
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    # ?; d( o& x. p% o/ }
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    ( [" g$ K3 j' m' c& Y, b2 E% x1 U
55. Sleep(100);
    ' {$ S' |4 j6 U6 Z5 g
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    . }& l+ B) k0 B% x; }
59. if   (!rlen)   return   false;
    0 X( ?( ~) `* h+ q" }. W" t
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    3 c, f/ ?+ I: y5 }) I
62. 
      U0 Z% p- b1 ]
63. return   true;
    
64. }
    6 |% E4 I4 J! W. p" V7 s  w
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    0 u% L) o9 Z1 J& L3 ]( c
68. char   buffer[10240];
    
69. 
    5 U: q( T* c' n% _# B4 p
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    * I  z5 j) y' e2 @' a: L; {. ~
73. 
    
74. struct   sockaddr_in   sockaddr;
    ) F7 i* g" J$ Q
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    ( J! T/ C2 U% r4 c0 K  p% k
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    9 x+ r. U+ f3 W  N& E; ~
79. 
    9 V0 H" O. m2 q9 @/ F: g' k+ L% d
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    $ M* Y. j$ V0 r
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    . `9 I  U( a: i/ O6 @% V
84. {
    
85. int   pos   =   0;
    2 Q) i4 s! M7 L0 t  R
86. 
    + X* w) C5 f) B
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    % W+ J& v1 z: U' w) l+ ?( M% ~( a
88. 
    ; k; l% L& u$ t' P, D# O) p+ X
89. result   =   response;
    * F. q* T% t' D9 ^% l
90. result.Delete(0,   pos);
    0 s+ B# h3 H: j6 A# R
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    2 e9 c$ |* |: N# u' l
97. }
    9 ?+ X  o9 e2 |7 Z
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    4 u% ]9 p& e, M* o9 }0 m
100. {
     9 S/ G" W) [" Y' q8 [% ^4 i
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     ( T/ o$ \$ O( h' ?
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     * u+ K3 l' {$ ]/ M* |/ e/ E
107. 
     : i/ f8 ]7 A# x$ q
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     1 e: R1 r) E7 K$ n
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
       [" ]* X0 f  v) p: w( }
114. MyUPnP::MyUPnP()
     3 A/ v# R( {) e# x; A
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     2 W/ h: w! M! b( P1 j7 i; K
118. isSearched   =   false;
     
119. }
     
120. 
     * a( \2 a6 ~: G8 o, V6 I% N' H5 v
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     - ]/ Y: A8 u0 ~
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     - |" U0 i5 Z: f, x
126. search   =   m_Mappings.GetNext(pos);
     8 a* U* w" u7 u1 @8 p
127. RemoveNATPortMapping(search,   false);
     
128. }
     # Y0 ]% S4 c; n: k6 M7 W. s
129. 
     1 E) n3 x6 a2 f& G
130. m_Mappings.RemoveAll();
     6 B& ?% }  Z  C0 @
131. }
     
132. 
     
133. 
     7 ?% v# _7 u4 ~: N* m
134. bool   MyUPnP::InternalSearch(int   version)
     ( _$ [" D+ e0 ^" w& D7 k* B
135. {
     1 S* E& Z, }# B, p, ]
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     & Z) Q% ]- Z5 d. k) ?/ }4 Q3 N
139. #define   NUMBEROFDEVICES 2
     4 w  ~# j* x0 Q3 V
140. CString   devices[][2]   =   {
     / L4 y3 a' I+ C/ F
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     : A2 c: k1 t8 c/ I
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     7 e8 c) n& d0 D. g6 V
144. };
     
145. 
     5 h  u; Q' N; M+ N. g2 P5 h
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     $ ~+ ~' M) q! g9 F9 M4 z4 ]% Y' b
148. ioctlsocket(s,   FIONBIO,   &lv);
     % V3 ?, r8 J. s1 K6 c; h
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     $ G) ^4 I8 X5 c6 {  s
152. if   (!(i%100))   {
     2 ]6 K* I4 H2 _" j5 M
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;
     ) U& ^9 d0 d6 e! u
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);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     ! `: D' `8 {" g2 y
159. }
     
160. }
     
161. 
     
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     - \& V: \4 o& M6 z- G! Z  q4 [
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     9 j7 @2 A9 H# I- l, L4 N
166. if   (rlen   <=   0)   continue;
     ! t. @  m! c. N1 e; c
167. closesocket(s);
     
168. 
     9 a" P6 l$ n$ F, \0 r' {2 d9 e
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     / I- }4 Q2 F2 a
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
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);
     ; P, T, k; L& B7 H
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     ! H& w( `2 V) m7 v8 X% v
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: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     4 U1 S& C) R: R6 a. |; G
184. m_description.Trim();
     5 ~7 r& z9 P# P: J+ e/ U9 x
185. return   GetDescription();
     
186. }
     9 `! U0 [+ G0 {, S: O
187. }
     
188. }
     
189. }
       B6 L8 D' L" I' j
190. }
     ! c; |+ h# @1 C: P( z7 b
191. closesocket(s);
     
192. 
     : z) k( r- g& U3 s, ~. }) p1 e
193. return   false;
     
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
# y3 G0 S) k$ w% p4 x; P' Z8 w

///////////////////////////////////////////
- p9 m8 i) r. S& T//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
9 N# e6 f- j" p2 Z

#pragma once
#include <exception>

8 \* R8 @+ R; s8 z3 f5 U
  enum TRISTATE{
        TRIS_FALSE,
/ Y$ `% u% x. Y. h        TRIS_UNKNOWN,
        TRIS_TRUE
" |! y  l: A' m6 r5 g};
8 Q& o6 y; }& ]; k
: t7 I9 M, ^; Q* m
enum UPNP_IMPLEMENTATION{
- \* A) E8 Q0 p, j) ~* p        UPNP_IMPL_WINDOWSERVICE = 0,
2 @) K% T# G% b        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};


- p7 V5 z' @) h3 ^2 J( V& k4 [
, o1 i1 W9 u# J# \; _9 Z
, O5 h  \0 M: ~$ Z  N5 U9 R  Lclass CUPnPImpl
3 w' a6 b. w/ P* K" o; j( N" h: b{
6 C7 U( r$ ^$ l. Ppublic:
        CUPnPImpl();
5 b' z/ ?! A5 d) l! v: a        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
  J# d0 H0 j' b6 X& W        enum {
                UPNP_OK,
1 c4 J$ i: u0 M+ A                UPNP_FAILED,
                UPNP_TIMEOUT
6 q  \3 M0 |2 K/ G! n/ C0 K        };


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
" r4 C9 r. X7 W4 V  a        virtual bool        CheckAndRefresh() = 0;
# }" |4 Z( F) {- n8 [$ n, z/ ^  {        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
0 H7 C9 {& T  h7 |( X        virtual bool        IsReady() = 0;
2 P- u, A! M: S7 l" E        virtual int                GetImplementationID() = 0;
       
/ ^4 ?" @, x# v2 X0 X  t        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
; }; Q4 S' \' m7 x
0 \1 c* M9 A- A6 C
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
8 p1 J3 b, M+ k0 _+ ^: _: M! J. ^4 I4 H        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
+ }' d3 S( f5 D. L+ A        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

2 e/ r# Q7 L  }6 h7 N
! I0 p8 Z, p! J& b0 s% ~0 k' ~// Implementation
/ K) b+ I) }+ `& `protected:
! v) K) b+ [* D0 @        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
" v, ^# \  O6 ]        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;

5 y- D' R7 W* c# t. C+ }- F
0 r1 |+ N$ k4 n: Wprivate:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
4 z' T1 _, S" s9 F0 e

};


// Dummy Implementation to be used when no other implementation is available
4 y6 ^% S/ r0 ]+ Hclass CUPnPImplNone: public CUPnPImpl
{
6 h. X& K. _5 }8 o( e  Rpublic:
& k/ _' x7 K0 `8 o: \" `! t  H        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
) M# E8 d- c# E9 R7 y        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
! q! B; c6 S% v7 ~/ g4 R        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};


/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类


* S6 q8 l  _' W/ F! c8 W6 Y+ m#pragma once
#pragma warning( disable: 4355 )


#include "UPnPImpl.h"
% [  k2 u# s- G7 K3 n# S# h4 X/ y5 s#include <upnp.h>
0 z' V& `1 G+ F3 ~#include <iphlpapi.h>
; m9 g: t3 g1 Z2 j- d6 Z#include <comdef.h>
#include <winsvc.h>


#include <vector>
#include <exception>
4 V2 T7 \5 T% v#include <functional>



9 n9 ~' g" a/ `5 Z  u
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;
3 r; }" X$ p( |typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
# F" x1 G. K. E" G2 i

typedef DWORD (WINAPI* TGetBestInterface) (
/ ^  g9 `3 @3 r3 V7 R  IPAddr dwDestAddr,
0 y4 j' x/ N& {* e: Y  L+ }  PDWORD pdwBestIfIndex
; |" W4 y9 ^; J7 v8 c3 d6 [  d);
% N0 P$ N8 s* O6 `+ K0 a
$ d, A0 e% s4 _9 C  G( ^
! @: f! _3 {5 ~- ?, V0 ttypedef DWORD (WINAPI* TGetIpAddrTable) (
+ U, Q/ @# Y& ]) L$ c, a. n3 W' G% i  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
6 i  A$ d0 ?& G  _7 y  BOOL bOrder
& H5 v  a2 g" n. k);


typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
* |" f# K+ U8 R);
0 R( o0 x% F5 `+ h4 }
2 o/ Z9 v0 J) d; a# h% |" P
CString translateUPnPResult(HRESULT hr);
6 V) L$ \8 W. r' h0 ZHRESULT UPnPMessage(HRESULT hr);


9 V9 E8 E9 g% ], f' M6 Iclass CUPnPImplWinServ: public CUPnPImpl
+ l. m# G6 x4 w" J9 [6 ?{
2 x/ C0 P1 x, d        friend class CDeviceFinderCallback;
* B) F& }( H6 V        friend class CServiceCallback;
// Construction
6 ~8 o9 g! |0 w( \0 _1 K- Ppublic:
  R2 T# f; I- l: t/ ~6 T        virtual ~CUPnPImplWinServ();
- X1 r& s: Q0 l8 U- g  z        CUPnPImplWinServ();
$ \  }. N5 ?) T" `

2 f% X& n# W& r- g/ C        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
" x- X* w# U. f3 L8 e0 R' c! H' x/ o        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
6 H( j$ B! P6 h4 T  {

        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
: q$ m) I' d' S! p0 B9 y6 v2 D        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
! X0 H. H. D' ?* ]1 k" w        virtual bool        CheckAndRefresh()                                                                                { return false; };
9 i: y1 }; v2 j* \
" v' c3 T0 t  D8 x
protected:
; X7 k+ z) T& N1 M        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();


        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
  ?+ N0 Q: x# T8 n- a7 P                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
- M6 b# z: F& k1 @) X( L  Y                        m_bAsyncFindRunning = false;
8 U0 q7 ]) G2 Z  B( y' ~: j                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
$ y3 f$ I! m: D* D; L) p                        DispatchMessage( &msg );
4 P2 y4 ~' q1 C# s- \- q* L                }
4 P& L& `3 d( p3 f8 `: A. E                return m_bAsyncFindRunning;
        }
3 D% W5 h' D( g: L
& u( q7 K3 V* i$ F
' y' j  L3 P3 x3 Y- [7 P        TRISTATE                        m_bUPnPDeviceConnected;
( J  z3 A- i) _! u% X

. @# `% F3 ]3 |0 `% H' {1 s! w// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
/ _3 Q8 u7 k: G6 [0 x/ U; A, i; C        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
. o! J( }7 j: U: S" B4 U1 ]9 i        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*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
0 n6 {* q4 z7 E! J1 A' z6 o3 t/ Z

* {$ D7 W' Q3 K" b; y) V        TGetBestInterface                m_pfGetBestInterface;
/ U( C8 I+ @: S) T        TGetIpAddrTable                        m_pfGetIpAddrTable;
& |( F+ \6 l( \7 _1 }        TGetIfEntry                                m_pfGetIfEntry;


        static FinderPointer CreateFinderInstance();
& l9 G& j& t+ l+ p4 w# q! O2 i        struct FindDevice : std::unary_function< DevicePointer, bool >
; \( r# i. {! i* k% ?, m        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
% T3 G& f) u. ]9 o2 q% h                result_type operator()(argument_type device) const
6 I  N7 m8 N/ B3 {7 [) S0 e                {
# V2 z' H4 g4 Q/ Q. Z                        CComBSTR deviceName;
9 ^9 U+ ]3 K8 z" L  o; k                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

7 I  M: {9 m: o9 H
                        if ( FAILED( hr ) )
9 M! S1 t+ k, g% @. d3 Y" C" Y2 `. E                                return UPnPMessage( hr ), false;
* W" F+ I+ Y. j3 `3 c9 ^+ v
1 r7 v8 L7 L8 I5 l. n# K
0 B  a7 A4 d% O- _2 T                        return wcscmp( deviceName.m_str, m_udn ) == 0;
! ~7 Z6 T8 H$ J9 E1 V# S                }
                CComBSTR m_udn;
        };
       
6 C. E& f- z+ s) Y        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
6 A5 V1 }- R2 ]; g: |        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
- l' s6 P/ u& Y& j  {        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
% r% K. @) p* J        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
; B( q1 }5 M0 _! R# S        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
, I% O, w0 @% Z0 s! [                LPCTSTR pszInArgString, CString& strResult);
3 A4 C% j; j4 @0 l        void        StopUPnPService();
9 \) G, t# `9 F4 y+ {) a
' U, d; L; b* r* i, E/ o
        // Utility functions
2 b* X5 N( B  K! L) j+ P2 N' `        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
% u0 e! A0 @5 M' t# y        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
9 i  ~) m, U# Q% p; K        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
( J0 W5 X9 i! g# C/ H1 V        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
4 H1 U5 ~9 R+ l. S- ^        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
3 Y, z9 H3 l3 n4 W' K! T

4 i* w& g, E2 a// Private members
! [. M) z( K- w( g( k( J. Zprivate:
8 l9 ^6 _* U9 b/ e7 H; u6 S        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
) _  e- p$ F( L. o7 a        std::vector< ServicePointer > m_pServices;
3 ~/ ?+ x8 p) B        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
0 }- d! s; j6 B& b& K% R  a% G

' E4 [8 @( P9 M6 V. U- o# e        LONG        m_nAsyncFindHandle;
- {& X3 B" V  J- z) L4 ]3 T        bool        m_bCOM;
, h) `7 x5 v! K7 j9 L% Q        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
( S% Z, w! z! u/ V4 U8 F1 |        bool        m_bADSL;                // Is the device ADSL?
' v* S0 H  B/ X        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
+ s( n! b' X) _0 S6 T* E        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
6 u' b$ t$ p, R, j$ g4 v        bool        m_bDisableWANIPSetup;
& ~' _; }: W, L& G, E: i        bool        m_bDisableWANPPPSetup;
& ~& k2 w$ l8 d3 x) v

};

# r2 _/ g1 B# ?% S
3 _; b% v2 {9 r/ _: P// DeviceFinder Callback
5 g. o% G1 ?; _( Q4 n9 q8 \0 Oclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
! y- R3 u# b6 h                : m_instance( instance )
8 z8 Z; [- @* [# W1 J: D% h        { m_lRefCount = 0; }


2 N( l3 _5 J9 R) l- I& d   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
$ M, s& J& x% r$ @2 J   STDMETHODIMP_(ULONG) AddRef();
' g& Y2 Z$ ]" l- \   STDMETHODIMP_(ULONG) Release();
" B: W: B: G! M) Q. r( I
1 I6 ^. m3 _3 P9 M8 c  h" S
$ o* \) ?, q- }6 {) M; ^& f- l$ B// implementation
0 W) l( ~( v! E+ w* ~( E# Kprivate:
7 z! S% W' M$ v        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
' E( L$ U, U. \$ O0 t        HRESULT __stdcall SearchComplete(LONG nFindData);
) d. j5 D) @: e

private:
        CUPnPImplWinServ& m_instance;
6 F; G- d8 u0 r+ {        LONG m_lRefCount;
};

+ Q& ^$ ^* a: Q# Z
& R* g9 M' m, Q# v1 M// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
{
public:
; P! k; f* E% k3 G& b0 j* z6 k- j7 W        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
9 b  B0 E1 ?, f2 Y! J. I   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

; T) N) ~. ]7 t$ ~' x
. c- Z( K# [2 V, Q9 s; S1 a1 Q// implementation
. c0 j! W" P* P: N5 ~; B1 b5 lprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
7 y/ R) v( p6 o) O8 [* @
/ Y1 R1 y  J6 p* C" |: q
private:
  f# h& L9 H' X        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
0 t5 R  N. j! z: m};
7 l7 E- j0 K+ O6 h! X% }

/////////////////////////////////////////////////
" I/ @* V" C3 l6 G1 Z

使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
) N% m$ a( E; C3 w2 KCUPnPImpl::StopAsyncFind停止设备查找.
3 T% s' F' ], l1 B, aCUPnPImpl::DeletePorts删除端口映射.