UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   % t; B8 O) M9 T) m
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   8 ?. c  E, P" p0 L& l  ^
7. 
   
8. class   MyUPnP
   
9. {
     O9 Y- [. m- n( H! U
10. public:
    * h) b7 L1 v6 _7 R7 `; K7 j& b
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    % R3 C8 Y- t% O" b- H
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    - P9 V8 S* _5 x, {* C8 y: l
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    * o, H8 U' s$ ^4 x$ x
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    * H! R' i" Y+ z7 B; `& |6 B& p2 {
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    8 a9 y; c9 D9 n7 `3 a
18. 
    
19. typedef   enum{
    9 r7 ~! `& s0 [4 U  X* ~7 \
20. UNAT_TCP, //   TCP   Protocol
    & b7 W( x, f7 A
21. UNAT_UDP //   UDP   Protocol
    ( y% h' d, d5 f# T! d# F( x! N
22. }   UPNPNAT_PROTOCOL;
    
23. 
    " S5 E, e$ g# N( T
24. typedef   struct{
    + d7 q8 n, L. a
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    2 K" R" ]' o2 @* s( ], x3 K2 B
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    ( J0 }, T& v- V
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    % ]9 u& m: i6 a6 j
31. MyUPnP();
    4 h  a; _; Q. l! N: q; f7 C
32. ~MyUPnP();
    8 I8 R( y3 F, i( Q: B! I
33. 
    * h6 \+ T: ?2 f7 \; T+ [
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    7 J& D: W* i2 O# b( j6 y
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    8 T5 V, D7 y; M; e
43. protected:
    , |# M: F" A4 s
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    , M, p2 \3 Q, X8 i$ T, k8 ^5 s/ X8 j
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);
    ! I$ d) Z  d) ^% q9 R' m# {+ W
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    + _6 z! f3 }: m2 P- t
53. bool Search(int   version=1);
    
54. bool GetDescription();
    " ?5 A6 k8 I2 s3 r
55. CString GetProperty(const   CString&   name,   CString&   response);
    7 g1 |( Q$ G! b7 q- ~( e
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    6 V2 p: `+ P/ E
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    8 f. T9 H5 {! y3 _
59. bool InternalSearch(int   version);
    
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 D3 o2 d/ u, O8 N' o1 _+ k
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    0 c; n- c" x4 [5 T, q" Y  V( `# ?0 E) K
72. CString m_slocalIP;
    : p+ V3 o: |2 D  |/ e# B
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    . K( \1 ~% d, s6 s* k
75. 
    
76. bool isSearched;
    
77. };
      L& J- U$ n, D' r* f; N
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   / ~4 f7 p/ A9 A$ u- E. o
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   4 ]7 X# V( v, W9 u
5. 
   3 }" W' l. A; l/ K2 z
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   & p; k1 G! m% S8 F- ~
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   7 t) l' ^, Y2 A+ m2 h- j
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    ' i6 C9 M, C( e) ~  E# b2 f
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    & d5 A$ B8 J1 O& I% Z
13. static   const   int UPNPPORT   =   1900;
    : [6 [5 R/ `. e( c& d. w
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    8 h) C8 d( Z/ j/ r% J, A2 t
15. 
    8 Z3 [: O/ o$ G& Z) t0 R. p9 n
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    + _) a5 R* ~( |, v* k
21. 
    # y* M3 |7 P5 s3 ]+ e
22. return   s;
    
23. }
    . r' I' A! S; I3 Q+ ]" n0 K
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    ! R/ B: ]: E/ x3 p! C- N6 Z
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    1 G* A3 u4 i! }) ?/ O" r: U
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    5 H1 k( ]0 k$ d  B) Q6 c! V
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    # A# ]' o' k( @! G* O
40. int   length   =   sa.GetLength();
    7 b5 S' }0 ^( q+ C6 Q% N, p" k
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    ; [% a' y: Z$ K) P; L+ J
43. uint32   ip   =   inet_addr(CStringA(addr));
    . _; H' \8 S0 N, j+ p8 g% r  P
44. struct   sockaddr_in   sockaddr;
    $ b  d! f, a  D! o* U4 A. E+ b
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    - _5 c2 A6 E( H' E
46. sockaddr.sin_family   =   AF_INET;
    ! g; {" X( ^  Z. Y
47. sockaddr.sin_port   =   htons(port);
    3 |  h( _, A7 J
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    - o. o  y- {  Q1 R
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    + k" L) |7 F5 `. A
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    2 C) q' H: F5 F( x+ j. [& m* `
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    % `: s, U" j- N  L3 o
57. closesocket(s);
    ( j* [  q3 c9 j$ P3 j. t5 c: Q
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    % |1 E  \- M- ~* J7 f+ N
60. 
    ; n5 W7 [2 u9 J* d, @
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    * U! k, d6 }  h- ^! M- ?: ]( J
67. {
    
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    1 z2 N) e. n* y4 n
71. int   length   =   sa.GetLength();
    4 B9 W8 {' J! Z$ G
72. strcpy(buffer,   (const   char*)sa);
    " f/ g  W: r7 s5 N3 z2 B
73. 
    
74. struct   sockaddr_in   sockaddr;
    - P4 {9 t" w- _+ p4 U
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. 
    3 k7 @! ?, j! K. c5 ?
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ) J  [, v- j! ~5 V# O
81. }
    
82. 
    6 ], |- f3 p  Y& B( q
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    * O1 k8 n$ w$ E6 j0 `
84. {
    
85. int   pos   =   0;
    7 a2 ^2 A$ B' D) ]$ A! g' g$ `
86. 
    # T. O4 _* M9 s  r
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    4 h! K% g* F' T3 z
91. 
    9 [& g! Y. a% O7 [, J, u
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    / p; H  g( M' ~! x7 e
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    
98. 
    $ Y) ^9 A! O: {2 f) @. [
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     " `# L" P9 V2 D& s( w8 u" X; ^; A1 t; W
104. 
     ; C6 o/ V" S2 Z8 O: Y4 V& B# D$ |# J4 g
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     - d$ c- {' _( g
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     3 i9 `  `/ X6 O6 w6 V# p: @
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     / ^+ J1 a6 e: t# |0 M3 Z# {
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     1 S/ j# h2 E3 g$ w
118. isSearched   =   false;
     " l* C. n4 D  j7 }1 d, z+ P6 [
119. }
     7 U6 p& `& u2 J% j1 w8 c
120. 
     : S3 y! C( U9 y
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     * t" C0 K# v8 h: r4 |: x
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     9 S7 `  R# S: L3 O% l% Q
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     ) E" f2 D0 R& o8 m4 o
131. }
     - Q) `/ C2 q- |0 p! T+ R8 v
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     + S) d! X# E. \0 v9 M  `
136. if(version <=0)version   =   1;
     6 u: i* T& |* |3 f
137. m_version   =   version;
     
138. 
     3 i. ~/ }9 n: W; B, c
139. #define   NUMBEROFDEVICES 2
     3 w4 Z' {: o3 _# C: j/ U4 W7 g: I
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     % u6 @5 E, s9 V2 g/ A5 N8 n
144. };
     5 ^* w1 f! S% z# @- z
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     7 b! B0 J. }/ l) @2 Q- h" p$ V
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     / h& i: [  \+ j, M; _0 h0 P7 a6 K5 q
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
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 "),
     5 l! l0 l, M  b
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     . X. g0 M4 n% d- c5 N/ g/ Z9 h" h
161. 
     7 ^5 X; m4 U) ^) t+ A/ \
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     ' Z0 B/ N0 I, T4 E
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     & @& f1 e$ h' x  {5 k: n
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     2 ]- g1 l6 E3 b9 ]# u- S
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: "))   {
       F7 f" \  ~/ H* j
182. line.Delete(0,   9);
     ! N* v/ a1 S. s
183. m_description   =   line;
     
184. m_description.Trim();
     8 h; s2 j  \+ j- d& B
185. return   GetDescription();
     
186. }
     * y2 v5 ]0 P& ~
187. }
     
188. }
     ! g9 C; X9 g. M+ z3 ]- P
189. }
     
190. }
     
191. closesocket(s);
     
192. 
     
193. return   false;
     : W6 R# u' P; Q
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
* I8 t) u3 @4 H& s

5 t/ e6 ^3 s0 i4 t///////////////////////////////////////////
. k1 I, \. J. l+ p# J//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


7 F( J! a/ y* ~1 Y" D# o& J/ V* v#pragma once
* s! M. M  B2 K# q; [) n#include <exception>


8 }( b- X1 D! V4 z' Q* Y  enum TRISTATE{
( a1 |& Y; l+ a  \: w6 Q7 l        TRIS_FALSE,
        TRIS_UNKNOWN,
0 X2 j3 X& u0 `$ z) H* j. N        TRIS_TRUE
};
( X' h& H. m4 v6 c, K2 ~: t
2 U7 T! n8 P! X+ \  _+ s
" F3 `& S0 F9 L) Genum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
4 r2 H+ ?, m" y( }. @9 D8 @        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
3 U8 A2 Z, R5 A2 d8 ~" E};
) ~2 H2 X; o, z$ x



) s) Y- m3 l. T6 lclass CUPnPImpl
2 g! ?: E- h+ ?{
public:
        CUPnPImpl();
( T+ L+ s1 j% L, I: ]% F* B        virtual ~CUPnPImpl();
( j! ], r3 O1 |* V9 N        struct UPnPError : std::exception {};
0 P# w8 ?* D: i3 q5 @& P        enum {
                UPNP_OK,
7 G) P; ~, T( G1 {+ Z2 \7 h                UPNP_FAILED,
3 r! v) w/ [$ E: d                UPNP_TIMEOUT
8 E4 d+ g3 b2 O2 @: J) e) ~        };
, W( C8 q% h& E$ f9 n

        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;
% c% s- P5 ~7 i0 t  k$ }        virtual int                GetImplementationID() = 0;
       
/ a- G/ f% C3 t7 l4 U4 G. @2 T        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
) @: l( o0 I4 M5 R2 {
# R# D/ T* v: D" C0 e2 l$ \
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
/ Z; o& n0 N0 A& k+ _        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
. l* _6 i, x3 z3 m( |        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
' T* s4 B: f( R
) J: k+ r4 P7 k3 D$ ?" H* C1 h
// Implementation
protected:
2 m% e7 }: c- |" B. [2 e: m        volatile TRISTATE        m_bUPnPPortsForwarded;
, R, b* K% v0 a5 s2 a7 j4 n: J        void                                SendResultMessage();
+ {& k! y5 d& X) S3 {0 b# o        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
/ F* }) _0 A" ^/ }; w- j2 w+ I        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
3 w3 B4 S4 v8 R4 e3 Q

private:
0 q- O& F9 O/ Z; l& z        HWND        m_hResultMessageWindow;
( ~+ Q8 m; |; T% F& G        UINT        m_nResultMessageID;


};

4 R8 r4 H2 D, w9 Y
, \" _+ I7 b3 ^6 q' j// Dummy Implementation to be used when no other implementation is available
, X; Y- R0 N' Rclass CUPnPImplNone: public CUPnPImpl
{
public:
. a0 `* m" @: R! G        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
- r3 \; K0 w( R7 N8 F) H        virtual bool        CheckAndRefresh()                                                                                { return false; }
* D& h# ]6 z7 g  N0 C9 N        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
- r# \9 c3 O8 h+ R2 ~" P+ A        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
( u. X9 [  ]" i3 O$ @};


$ r; ?# i* j0 ?7 u' ], _/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
) @# L) {6 p& N) @
" ?+ L( f/ i* {9 S0 I' A# s
#pragma once
#pragma warning( disable: 4355 )


#include "UPnPImpl.h"
) e2 W+ Z' ^0 U; l# ]+ o#include <upnp.h>
- z9 P+ F; F% A: x, F- b' k% S#include <iphlpapi.h>
/ Y6 y0 m+ n6 ^& d: k" \#include <comdef.h>
#include <winsvc.h>
' q% |; U7 F5 w# I, }. H1 Y

#include <vector>
#include <exception>
#include <functional>

# P2 R$ J" a9 P( ~7 v0 K9 Q
: P8 p9 |9 s% W1 J) I

) T' W8 ]4 N$ R- Htypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
2 y* Y  E8 P% _( B0 \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;
0 e0 t% W' Q) b/ X3 atypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
" a- M0 ]5 |' ^typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
' R' a/ t* Y9 D$ r# {: H& X* B
$ ~! D. L' [- l' @* B
! e& y' D3 f- Y7 P. Rtypedef DWORD (WINAPI* TGetBestInterface) (
2 c% m& _% u3 A1 }0 {2 u  IPAddr dwDestAddr,
1 J4 A7 C' o3 X' c" z6 a  PDWORD pdwBestIfIndex
);


) [1 m4 r8 _# Z9 n8 s6 itypedef DWORD (WINAPI* TGetIpAddrTable) (
- k* }# ]) m( }4 {5 U- a0 k  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
: L" Z3 z. A* T% U  BOOL bOrder
5 N7 J/ p: D5 [! h+ B);
7 T! j2 `& N% A' @

6 h# \; ~# }# ?# Xtypedef DWORD (WINAPI* TGetIfEntry) (
7 B3 Z2 d- F8 k+ T. j2 ]" b  PMIB_IFROW pIfRow
);
- D5 U+ t! z/ |3 x7 ?

' U, m6 O# {0 z% uCString translateUPnPResult(HRESULT hr);
( q0 A$ z" r# j' W0 V$ K+ D- r) |HRESULT UPnPMessage(HRESULT hr);


1 q" Z# F/ z& U" ?2 W2 l; _class CUPnPImplWinServ: public CUPnPImpl
{
, J& W% W" S0 X        friend class CDeviceFinderCallback;
: m/ `: i5 R" |        friend class CServiceCallback;
" E+ p+ _, ?: i( |// Construction
8 j! N, ^& L! v, gpublic:
" l. o9 q$ Q! z6 S$ l        virtual ~CUPnPImplWinServ();
6 o* k5 T5 ^8 T. ^* t        CUPnPImplWinServ();

9 b( I, K1 S0 ^% \+ d( C' u
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
# v; d& D2 @6 Y% G  M$ j        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
; \' u# k5 ?' @: H& k+ f2 e        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
( M" R+ j; [; [- a! l: _  ?
* K3 v! x- d5 V2 e0 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
% o7 i3 B0 j$ T; T* Y- r- B; D        virtual bool        CheckAndRefresh()                                                                                { return false; };
1 z7 J( w* z$ V% H/ N1 K1 J% C/ G3 B
, c4 }% o6 O8 h- S5 }! G$ D
protected:
; q( e4 s; E1 g0 R7 L- T        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
, b, h2 C; K" B. M        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();

4 z. ?9 w, X3 i( W5 J0 ~5 j% }
        inline bool IsAsyncFindRunning()
        {
3 g/ h# g, i5 b+ X6 A                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
$ c0 f  R( `) W( r$ _5 G                        m_bAsyncFindRunning = false;
                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
& Y5 t4 L" S: `6 A7 T$ W  p                        TranslateMessage( &msg );
7 s: m( h: d( w0 L7 q+ y! F! h2 Q                        DispatchMessage( &msg );
/ N; d, @" _7 b5 p                }
                return m_bAsyncFindRunning;
2 t+ _0 a0 R9 P/ y* D! N7 z        }
6 ^+ Q" e, F, R& M4 Z
' T+ w- L, X/ q3 |; `9 X; f4 m
        TRISTATE                        m_bUPnPDeviceConnected;
$ ]9 P2 U" `- v' j

/ ~" S+ e) l- _+ E) u// 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);
- n- F9 d; s/ c! l% }        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
8 z: M/ c/ ~- _( L' ?( r, }        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
9 P) D* i+ G" n' q7 o9 T& s, S2 R9 M' N

2 F" t$ }5 z2 s7 M" o2 N+ j8 }8 j        TGetBestInterface                m_pfGetBestInterface;
5 j( b# z- w' m. j' _        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
  v0 m& m; k1 v# ^% G& ?4 k% B

3 A( y. A  Z4 w        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
, M: E; Z% Z) K0 X" V4 h8 O# N                result_type operator()(argument_type device) const
                {
$ @; V7 w) p* u6 V# ~. a5 Z                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

' |/ r* {1 L5 s$ P6 h( o) Q, J2 W3 O8 e
                        if ( FAILED( hr ) )
5 G* W. @& a/ f9 Y# `                                return UPnPMessage( hr ), false;

7 ]6 J6 s1 w) B9 f8 U
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
; S! C# x1 [* [        };
# l7 U5 l$ E  {& D       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
) ]* g( X  L) P2 I$ D, q        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
  [" A$ {/ J8 H0 |3 Y( x; S        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
! @* ~' n4 I$ X8 [4 r        void        StopUPnPService();
$ c& i% b6 U( ]: k
: L4 I& a5 g$ b- j  Y- s; n: p
        // Utility functions
( [( m1 {4 z$ Y% w3 H( m3 a        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
; E* H, u6 ~& n8 L9 s( \6 M        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
! O  \( Z$ V, H        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
  k# X. W4 p& Y8 X+ o# e9 F" v

, O  V2 L) n% O0 j. ?6 Z8 T- H// Private members
$ A0 i+ H% d( _/ z, Q! Z6 V" K% `private:
        DWORD        m_tLastEvent;        // When the last event was received?
* u. n9 B" B& r: L4 r/ ?" w2 }        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
2 u' \5 e7 ^, T) L1 w% ~        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;


5 V& {  I* p" A7 g+ C* \        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
9 U5 L# w5 C4 }7 }        bool        m_bPortIsFree;
, w8 B- c4 g# x$ H        CString m_sLocalIP;
: F0 W5 W: v" l5 ]9 c! c/ j        CString m_sExternalIP;
! h; J* a, k% k0 ^; L        bool        m_bADSL;                // Is the device ADSL?
- d+ a" T9 m+ d$ p( u& B        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
5 v9 q( R8 M5 `: ^3 m        bool        m_bInited;
: V" b) F. s; `2 b) Q( V1 u4 q2 `        bool        m_bAsyncFindRunning;
. {* b. W1 k: `        HMODULE m_hADVAPI32_DLL;
5 K& p* X, @2 e0 D1 B7 Z# M        HMODULE        m_hIPHLPAPI_DLL;
2 c( g! A. V* q9 }( p' P# r  ~. p' a        bool        m_bSecondTry;
9 H, V' j! ?4 D7 Z6 J! O4 |: q        bool        m_bServiceStartedByEmule;
( z) b9 U2 m6 i) ]& e3 ]        bool        m_bDisableWANIPSetup;
1 C. D5 o+ {2 G7 n( C2 x        bool        m_bDisableWANPPPSetup;

- B% P! x% R2 A7 a- ~( t
};

8 t/ X% N9 h  @! e, D- a3 U
9 Z" y/ U; @9 z: k$ u// DeviceFinder Callback
6 S  E# V% h) [0 v+ Gclass CDeviceFinderCallback
6 B, H' _! q: V6 E7 {        : public IUPnPDeviceFinderCallback
{
public:
' A/ _! s$ d) k; Z        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
$ E4 ^! C) z* J5 I/ K& {        { m_lRefCount = 0; }


   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
+ r: L& n; P; Q- M- m* n" X0 n   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


* f$ Z, b+ @% x3 C7 F// implementation
private:
5 j% M0 Q4 F7 h  H5 z" i% ]& }        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
+ ]3 E/ o8 ]4 Y
% s  n1 B  }: Q1 M- v. G! a4 n
private:
        CUPnPImplWinServ& m_instance;
0 E9 R3 N+ T' x, @- k3 i/ z( f( W        LONG m_lRefCount;
9 t6 k% \, \2 u! o- H};
' |' E0 t. y  {
7 _0 q* E0 a) O! ]" f
// Service Callback
/ F& i% E: J9 r) Rclass CServiceCallback
# [! _+ W+ L/ i7 r- P* m( H        : public IUPnPServiceCallback
{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
0 f2 j, {+ k% n; |# s7 j   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
" Z* J! ?, O/ e   STDMETHODIMP_(ULONG) AddRef();
5 k% X, {/ q0 @$ G9 G! ?   STDMETHODIMP_(ULONG) Release();

5 H/ W9 Z( ~& I4 G( I
// implementation
private:
0 c/ R8 S6 t5 W# O        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


; C" S! c$ J: c: [1 e8 h3 Hprivate:
* j- W5 z4 Q6 i, x5 D2 j- G/ p        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
: _$ p/ e) u% v( V; D};

; t6 Y* f$ f" ~! B
/////////////////////////////////////////////////


使用时只需要使用抽象类的接口。
) Z( ]( ~) C+ A5 I8 ?; J) BCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.