UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   $ k7 {, D3 \# L, \0 _
4. #pragma   once
   / w# |! J( v( E' f, _8 O
5. 
   5 `9 O3 N" r# D
6. typedef   unsigned   long   ulong;
   
7. 
   9 r/ N4 Q0 x" K3 u4 T
8. class   MyUPnP
   
9. {
   # e' G: w1 c  @
10. public:
    
11. typedef   enum{
    + ]/ ?2 s4 w7 N, `0 f* A# X$ g
12. UNAT_OK, //   Successfull
    
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
    
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    6 H, N$ r0 f, h4 {9 `6 E' I
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    / D+ @4 M8 O4 [2 p% L5 y
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    + j  T% Q% H0 F1 `2 I
21. UNAT_UDP //   UDP   Protocol
    9 d; \5 g- x2 \( u
22. }   UPNPNAT_PROTOCOL;
    
23. 
    : l/ B) o! H5 r; _* H# ^- U
24. typedef   struct{
    2 `' I2 g8 u7 M2 F# q1 L
25. WORD   internalPort; //   Port   mapping   internal   port
    : }" C" r( q7 n# G! v+ ]
26. WORD   externalPort; //   Port   mapping   external   port
    1 e" N4 Q% Y. D  `
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    & h1 ~6 a1 Q3 V
28. CString   description; //   Port   mapping   description
    ; I. g; Q1 w* ?, V+ B3 p5 n9 `
29. }   UPNPNAT_MAPPING;
    ' \0 g7 d0 y+ m; A1 m% O) ?: e- J0 b
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    7 m9 K4 R: z$ n2 e8 y. `
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    + A; z2 n9 o. J3 O  f6 |: y
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    7 k+ Q. H7 r* K  z/ M) |" v
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    5 n0 y! d* G& V4 |. C
43. protected:
    
44. void InitLocalIP();
    2 @; s* w# f+ q7 M. [. x: v5 d
45. void SetLastError(CString   error);
    0 i" E; I6 K- F$ w- H
46. 
    ; J- r: s+ B# `* Q3 O7 c7 I
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    2 P1 z. v9 P8 M: o( t9 o  k7 M
48.       const   CString&   descri,   const   CString&   type);
    + y3 B# z# ^6 v
49. bool   deletePortmap(int   eport,   const   CString&   type);
    ! b8 u2 c8 ^' r( t' u, A- K
50. 
    2 k$ G  B' u6 `6 _/ i
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    9 d/ n. a, `4 a9 T
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);
    $ t  I; s( B' d8 \
57. 
    : J( \9 f9 ~( q/ e% H/ w; X
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    * z$ V; g. ]8 {; [6 {8 E+ d
61. CString m_name;
    
62. CString m_description;
    : l! Q6 J! K$ }* ?
63. CString m_baseurl;
    ' D( t) m; Z. K! x( w* @
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    % f# X: h4 |) U7 ?* W
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    . e0 b, S0 v2 i! X, r5 [2 U4 B
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    9 P& Y2 u4 X7 {- K. Y5 V$ ?( k1 G/ T9 n
77. };
    - L; [7 p  v: E2 k7 q! ~/ T: m
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   / W  ?8 i& g. T2 U  h$ J
3. 
   7 {! l9 F: t7 T" f; h/ n' G
4. #include   "upnp.h "
   
5. 
   1 k( h$ \* j0 I% [' t! l
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   ' Y) |* V% Q- T1 ]; p: E
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   * Z  ^8 C  V1 m5 Q$ j
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    + `; M+ l# {  B2 `. L# c# k% m0 u0 A
13. static   const   int UPNPPORT   =   1900;
    : h- ?* M  J6 N6 H4 D8 |8 x, w+ V
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    " Q& Q2 G  y* R
16. const   CString   getString(int   i)
    
17. {
    / z" s  o0 @; n
18. CString   s;
    
19. 
    8 h* u, \3 J& u1 ~
20. s.Format(_T( "%d "),   i);
    / D  E% b8 c% O) J
21. 
    
22. return   s;
    9 z% ~) l) W3 ]  B* F, C4 t
23. }
    
24. 
    $ `2 g$ e* f( w: a( I1 [& e/ l
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    2 n" F7 U& a! I0 [8 ~0 r$ P
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    % e' v" R' u+ j% n4 Z
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    8 f- l% n9 Y$ i( ~& r0 i
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    ' q0 d5 W& H* t
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    0 F2 i6 l$ N3 v; ^4 H" w' ]' h
37. char   buffer[10240];
    
38. 
    * h- O9 U! M& Y
39. const   CStringA   sa(request);
    : [. w; W( {2 X8 O+ X
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    + C9 T+ b' q+ X5 R
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    / g0 ^! E$ M' U8 C% L
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    5 s4 ^3 z9 ], k; C7 W' f. o) l
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    % L0 ^2 L( u5 H. e: n8 y; M# v, v
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    8 N/ `' k7 L) T3 P
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    ' ~) S) Z. g* d& @' |8 W
50. u_long   lv   =   1;
      B( v! B' _7 ~* G( u
51. ioctlsocket(s,   FIONBIO,   &lv);
    : ^5 ^' c3 ]# H$ ?4 N4 v
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    - s7 L0 X$ A1 K* e" c$ l
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    / ?0 W0 q" ]8 L# m5 X; r
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    7 E& R' w$ @0 c3 K/ z
59. if   (!rlen)   return   false;
    
60. 
    ; i. S' E0 W: \4 C) ~
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    + f' T; [  m3 P; d9 v5 i
63. return   true;
    1 N/ g) f0 R" t, a+ j! o
64. }
    2 D& R( f0 E: ?$ t$ ?+ c8 W
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    ' ~# N: f/ |. c* M
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    ' t* T9 s" H  @! ?3 w
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    8 G- a" e, \; d' K- U' E' G
76. sockaddr.sin_family   =   AF_INET;
    " b# @7 s5 l4 B  E
77. sockaddr.sin_port   =   htons(port);
    % `/ r$ w# U  [6 h8 f. P
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    . ]- p* X, r4 M/ Y1 a
81. }
    
82. 
      x9 f  s# U7 v6 `' p$ k
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    . }3 m! Q" o3 E. k5 _
84. {
    
85. int   pos   =   0;
    3 g0 M0 t5 n7 @# ^- `+ |+ N
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    3 G6 E, X$ b% T9 {* S" j' O
88. 
    8 q: ~& `  {! P. m
89. result   =   response;
    - J* g5 b' ^4 X
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    5 t/ w" M5 O1 h, I
94. status   =   status.Tokenize(_T( "   "),   pos);
    / E$ V4 k0 B; B  M( r$ |. }  f
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    . a" v6 I: Y4 y0 R% d
98. 
    5 R; G: B! ~) o% g: i/ l
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     & Y/ G; ^, C' `7 F. s
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     $ h+ d1 r+ r4 V3 c- s
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     & o" g- ?0 T6 W! @1 u7 f2 K7 F$ v' T2 J
107. 
       `2 O8 F3 v% ^' m$ y$ R/ z4 `$ r
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     0 u2 r6 F8 o" g" G1 A
110. 
     ; N+ D4 G( w; Y! ~. k6 L) `" q" C
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     - u' R7 T! ?& D4 Y( a, T
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     4 ^# j# V  G- n$ _; G5 `
118. isSearched   =   false;
     5 X2 n6 }+ B0 @3 @! [
119. }
     1 e! e$ K& V( C1 y, b$ N$ f
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     $ ?- |  @' Y# p+ J! g% T+ R
125. while(pos){
     * n* t# f$ z4 }. ~4 i
126. search   =   m_Mappings.GetNext(pos);
     " U& ~3 q; g) w. K2 ]0 \) E- y/ k, P, u
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     ; O* Y, B/ n7 e- R
138. 
     
139. #define   NUMBEROFDEVICES 2
     9 T( g$ _  V# L, B# \/ I( _
140. CString   devices[][2]   =   {
     1 ]$ Z9 s8 r) Z! y! f; K
141. {UPNPPORTMAP1,   _T( "service ")},
     & p1 I. B2 d- x5 D
142. {UPNPPORTMAP0,   _T( "service ")},
     * ?4 W0 t4 t* O7 f) n
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     * N6 ?. b* e6 [7 s
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     : s9 m$ V" r/ V0 o% h. F
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     * d9 c2 m9 j) k* t
149. 
     / E6 I; \& [) \. G: T
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     9 f& X* I* J% A- j9 J% @* C' k
152. if   (!(i%100))   {
     : T, W9 Z$ [3 ~0 C' W/ W, e, q2 j
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;
     # N$ g# r: f+ m0 e  a2 J/ N
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);
     8 Q& o" q6 C/ `/ G- w' N
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     ; k5 \' K. t# ?! a; v/ ]
159. }
     4 k, Q8 u+ S9 P$ W0 p7 U" g! e, u
160. }
     
161. 
     
162. Sleep(10);
     
163. 
     ; L; y1 `: A4 t! w
164. char   buffer[10240];
     1 K8 E* d$ C/ h% B. _& {/ Q- C
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     8 G- a5 E" O* G5 _9 X, T
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     3 {) M4 b0 R# z
168. 
     & R# H7 Q, r" y$ ?; Z
169. CString   response   =   CString(CStringA(buffer,   rlen));
     ( ^1 E) `$ W$ [5 M$ O- m
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     ( k+ {8 Z8 c& ^5 i% C
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);
     & c1 \, m9 S0 d9 f% a& u
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     # f6 c; J6 s- t, n" x1 c6 D6 G
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     8 r4 z' P1 P3 Y7 ^
180. name.MakeUpper();
     0 C7 n2 U3 D% r
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     , i  _  e0 G1 H) J  {1 K
183. m_description   =   line;
     / i5 C/ [0 M( B9 v. G
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     2 u* J9 o  g, O: t% g- i3 D& H
187. }
     ; W( w( \# n. q! ?
188. }
     
189. }
     
190. }
     
191. closesocket(s);
     , A( |5 I! X4 W9 u
192. 
     ( \! r( g9 u3 c# K+ |
193. return   false;
     * j/ |/ m1 }, Q' D- d; I
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，


///////////////////////////////////////////
% L; |. G6 G/ a  U+ |8 z  G; X- p//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


#pragma once
! y/ u0 V$ j7 ]1 i+ F. b& C" d#include <exception>


1 {% I1 d5 D& A9 \) W5 \  enum TRISTATE{
        TRIS_FALSE,
6 x8 J( [' F! x        TRIS_UNKNOWN,
        TRIS_TRUE
};

! ?9 a. ]. V3 G! F6 Y" H2 l) e
$ X8 d  ]) S& g1 h) {+ nenum UPNP_IMPLEMENTATION{
5 k* G* U$ r/ F0 C  R% i2 Z& _        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
' W) I& o; C0 G) E) I};



. |, M% F+ e* [% i5 j$ Z# x/ h2 o
class CUPnPImpl
  z5 l( a- @+ u" G1 J{
" h0 D0 Z: s- v2 e$ {public:
        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
: U2 V- T4 \1 j# `        enum {
/ e) `* l0 K4 d7 b3 `( I                UPNP_OK,
: P! Q. t  G/ o# p9 c2 T  p! b                UPNP_FAILED,
4 z! V& _# ?% g5 @7 T  P                UPNP_TIMEOUT
' w" y1 v$ S9 M# C6 J2 E        };
& L' D* v9 d3 e* T
- Y( J7 _; d; l- h
: r8 Z# a; Y# t2 C' ~        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
8 b/ ]+ T" x3 s; A# y        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
* _9 o5 ]" I$ f, b" k        virtual int                GetImplementationID() = 0;
+ L6 {4 K! ^/ {% p       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


1 R1 V# |- V+ W: z        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
9 g& a; E0 b4 W6 S        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
$ G0 [) {6 ~3 E. l* ?        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
5 y% H- A2 ~1 j

// Implementation
- Y& d- G( e7 f- bprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
% X7 C6 y7 f$ Y* O        void                                SendResultMessage();
2 K" N2 i+ o% N* h; C  `        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
8 n$ ~, n# g1 `- V7 G, _0 T0 p8 r        uint16                                m_nTCPWebPort;
+ T- _' ]4 x! m. o  n        bool                                m_bCheckAndRefresh;

# l. a% A* U  A4 G
# G6 Q' g0 p9 u9 L! Z- gprivate:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;


8 [- p$ d. Y( j/ _3 J8 a5 b+ z2 A, P};
" c3 v; O% j$ N# j% \* J

- h- M3 P& ]0 I$ \// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
% |: T" r8 }% Mpublic:
/ u6 {6 p% r. N8 P$ t' ~: y- U        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
: b( ?  n1 G& R" Z        virtual bool        IsReady()                                                                                                { return false; }
& _8 a$ n# Z0 [0 N9 r! k, t4 M        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
% o# p* F3 Q, F! \, Z  F
& Z- J- z* G; l% p8 i5 D' U
/////////////////////////////////////
; P* o1 s/ j) d1 {3 x: k& K! r//下面是使用windows操作系统自带的UPNP功能的子类
. T2 }3 s1 U* x9 D' y& s

#pragma once
#pragma warning( disable: 4355 )
% K  X2 |: @. F4 ?8 k! n- P7 {

) y! U( a, a( ?% m#include "UPnPImpl.h"
& ?$ t/ R6 M# c. n$ [! Y#include <upnp.h>
9 c- A1 g5 W* o$ ~  O  h#include <iphlpapi.h>
  n4 G$ Q* }/ n& ?+ }/ W#include <comdef.h>
#include <winsvc.h>
$ \9 a+ S( d2 N# J0 i
$ Y6 O5 m9 e0 f' o; n7 q- b" G! y
#include <vector>
- Y+ K' S% @5 c4 T( j% v# ~#include <exception>
#include <functional>

, q3 m. x& n  Z
* w' S, C( p. b% R8 x- c
0 y' V" x) Q7 h
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;
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;
) V7 @4 E$ w& L* p% \typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

: [3 n# A2 ?% Q: t* h% U
% H5 d% O) F& `5 J* \typedef DWORD (WINAPI* TGetBestInterface) (
9 f' W, s! p% ]3 [! v0 X  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
! I5 R# v4 W6 b# h5 F: e) Y$ H# B. j);


typedef DWORD (WINAPI* TGetIpAddrTable) (
8 \  B: B. y  B' a( Z8 a: r  PMIB_IPADDRTABLE pIpAddrTable,
6 S% u- B/ }7 x  PULONG pdwSize,
# Q: g0 |* o  [! G# H2 J  BOOL bOrder
);


typedef DWORD (WINAPI* TGetIfEntry) (
( C) \& d% M* a: R! O6 B  PMIB_IFROW pIfRow
8 I* r1 G) X- n* J1 ]);
' L3 E1 a, s6 ^. W- b
2 S% P" x, \) `/ S5 h4 {( e
/ i9 |8 w" e6 ]  m+ B3 eCString translateUPnPResult(HRESULT hr);
/ s. X8 [! E( l( m& H4 tHRESULT UPnPMessage(HRESULT hr);


class CUPnPImplWinServ: public CUPnPImpl
6 V4 `0 _! V2 t( I" T{
6 O% R5 ]$ R4 k/ @1 j        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
/ ?1 L) y& E8 @# v7 [+ W. S$ tpublic:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
5 b0 I& v% Z- M- n$ N, t# j  n7 w
3 ^) A6 n& Y( d/ _
* f9 v1 [. m+ \        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
7 `  H8 n; \. r% J9 k# S        virtual void        DeletePorts();
& G, f% e. X6 i  F; x        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


: `6 m& F" m( _+ w% a5 ?! ]$ f        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
* \6 X% r8 A3 f* ^        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
: A% u: c  ^- _8 d7 ~        virtual bool        CheckAndRefresh()                                                                                { return false; };
- V# o) O+ V$ P- ]/ K3 K9 G

protected:
: W% f( E- R1 ?. }1 B+ e  ~0 a        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
0 c9 n" c/ D4 y: r% A        void        RemoveDevice(CComBSTR bsUDN);
8 [* \) Z! `! M& }        bool        OnSearchComplete();
        void        Init();
0 T0 t# s: C$ U& F" z& P

        inline bool IsAsyncFindRunning()
        {
* Z$ m$ K; D6 _% j' h                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
2 N& k. P) _2 q) _) v0 B' I+ k                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
) ^( _7 g- ]5 R; s. X                        m_bAsyncFindRunning = false;
4 w# s! j; ]- x& U# a' V5 d. |4 @                }
                MSG msg;
3 a, }  [" v! V. ^4 ~3 {                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
5 M! X( S% v" V0 t2 y                        DispatchMessage( &msg );
( X' S8 Z9 ?/ L. U) h                }
                return m_bAsyncFindRunning;
' L0 n- A+ j  ^6 ^% c" d        }

* |! [( R6 S1 \6 r6 m, L" K
        TRISTATE                        m_bUPnPDeviceConnected;


; k- K, W0 ^7 o, x' f// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
+ z% U, _8 k  f        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
2 C2 ~) n" K( f        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);
+ w: _: v+ ~) ]1 ~! j
4 j1 R1 C+ Y$ [6 u+ |$ g
9 `) m  b- F. E+ s$ z9 E        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;

- m* U/ Y% u2 ~  q8 O! g! S
. b( R- P. `! e. }  F6 o        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
" `- v$ n7 ?! K" U% x% e# ~' B                {
  c$ L. V! ]% f2 t, i1 v- ^                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

" T$ V7 k9 S7 ?3 u; y
& ]4 B7 G% b- q6 y, @3 W, e                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
' T) v9 a( r0 ?2 f4 I% B

                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
) ^  H& K0 ^4 N9 r# S                CComBSTR m_udn;
        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
, E# Y6 R+ N" h! f; n% B  h        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
& g$ J7 z) a- a        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
+ J* C" E, {' @& ?        void        CreatePortMappings(ServicePointer pService);
5 s) `% H% j! W3 Z        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
$ `; y1 N; }, u& \                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
& x# Y4 m; B8 S' Z% I: }1 Y

        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
& D! x, t" L1 R9 a  @$ J) d+ d        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
8 x. [% L; \; ^' o  r  Z3 r        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);


// Private members
private:
6 N! g7 P+ H$ Q' b9 [        DWORD        m_tLastEvent;        // When the last event was received?
( ~! i0 k1 i+ B- {0 w& c! s        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;

% \4 H, U% @, m  _. R: L
) J% _9 S7 c8 O, ^        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
+ j' j* c+ e2 _# N4 m        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
( x: _  D+ @" Z! r! z! Y, ^8 E        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
. l4 c" _1 `3 v+ q! W9 h        HMODULE m_hADVAPI32_DLL;
% [/ j6 o2 y1 F+ G9 K7 v% W        HMODULE        m_hIPHLPAPI_DLL;
6 r% u2 {/ i# \9 {        bool        m_bSecondTry;
2 S! e1 T* Y* ~$ h  p2 \8 v        bool        m_bServiceStartedByEmule;
% C' C! b% a# k        bool        m_bDisableWANIPSetup;
5 \  ^3 `9 d# q! b$ {  D9 O        bool        m_bDisableWANPPPSetup;

: C& E: |  K2 W; B6 x" H6 W
# a$ q: t5 Z6 ?5 D- r. I2 M};

; \+ s8 I  T+ z, I
$ }, y7 q6 o2 o3 U6 B& f- o0 U, P// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
0 C+ n- s" ^8 j* S! O; k9 spublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
/ o! g& Y6 B9 B0 I                : m_instance( instance )
! _4 x# q7 @; Q$ I/ t# G8 o        { m_lRefCount = 0; }
- ]( O: J7 k) J5 P' n  D
: m' t$ b. c( X% j
4 O9 M! |- Z& a  |" N- o   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
  l# X* v: f) r$ _, V, t2 z   STDMETHODIMP_(ULONG) Release();
; G( w) O8 G5 I' M2 l! m

// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
% V9 Y: K0 X1 i: {4 u1 I: G        HRESULT __stdcall SearchComplete(LONG nFindData);


1 W, u3 H" e2 \, U! lprivate:
        CUPnPImplWinServ& m_instance;
" T* ~$ n; A+ n+ [2 B8 c) L        LONG m_lRefCount;
0 I) O7 e: A. X# f8 S};


# d, n+ Z  S) A, s// Service Callback
class CServiceCallback
% S" Z# T# ]& w: o" T        : public IUPnPServiceCallback
% K1 w2 a$ H4 n{
- e, J* i& ?2 ], k3 [8 a! O2 m& ppublic:
5 F' N# @* o" X5 G! h        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
+ ^1 `" i* h; s% \, f( R" k' o   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
$ [2 r* @$ S" ]% Y   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
( D* j4 B1 ?4 A" h1 m

  I5 Y  W1 d0 P) g1 X) k// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
( _3 m3 G4 S4 Q4 k
! {( Y, @7 ^) ?0 k
) d$ ?( h) K$ d, U3 ?* e; xprivate:
        CUPnPImplWinServ& m_instance;
  o# @+ O1 F+ m/ p  n        LONG m_lRefCount;
};


/////////////////////////////////////////////////


使用时只需要使用抽象类的接口。
8 w& i! p4 |; \8 Q1 S1 W! m7 Z+ A: `CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
3 G2 }) b& H4 x% s. _* ~CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.