UPnP

天云上人

/*uPnP.h*/

1. 
   9 w$ Q! z/ u+ ~
2. #ifndef   MYUPNP_H_
   " i) Y; ~' K. f( C3 I) \2 e7 a/ b
3. 
   
4. #pragma   once
   * [0 Z, Z# b; v& r' e. s
5. 
   
6. typedef   unsigned   long   ulong;
   / f8 m" O! a( X" ~8 R5 Y
7. 
   + J& C  H0 n1 B( s$ \
8. class   MyUPnP
   5 C# Y) q  D/ m1 e# u- F8 g- q
9. {
   : H  V: X% u  E6 l3 i$ i
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    7 |9 Y$ J2 Z2 H( i; t& s6 f
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    6 S; Q# _2 ^0 P6 o# F
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    , q( D+ s! G- @' k4 z
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    ) D( v- @/ a! g. u
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    % S* e4 Y  K' L; ^. K% X
22. }   UPNPNAT_PROTOCOL;
    5 B! t0 m7 W) y6 j7 Y
23. 
    
24. typedef   struct{
    
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
    - I  c+ j! b9 c# H- O
29. }   UPNPNAT_MAPPING;
    
30. 
    
31. MyUPnP();
    8 P1 _$ X6 e# i* I6 P3 i
32. ~MyUPnP();
    
33. 
    , c: _" ~* a3 [
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    4 Q9 [4 H$ }) P* e1 M
36. void   clearNATPortMapping();
    9 ]) a- Y) K. |% p* o6 }
37. 
    
38. CString GetLastError();
    ( I8 X2 X9 @" A, X  {/ |
39. CString GetLocalIPStr();
    . J5 v& |* S- x/ c* \) x7 W! Z
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    * z1 J% O4 z9 {' H* w! |" o  k; J
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    : P  r4 P6 b+ f& }. [
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    $ Z% {$ D; N! e* R+ I2 q
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    9 n5 Q% ]1 x7 w' {6 D
50. 
    7 k5 j& K/ ?: F
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    * X7 t2 Z# a2 U
52. 
    
53. bool Search(int   version=1);
    / B- }4 m6 s) R7 H( R/ a
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    7 ?" W  v6 S+ l& v% C6 O4 j" c
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    4 i+ V( Y0 A5 C/ R, P
60. CString m_devicename;
    
61. CString m_name;
    1 L! y0 f/ h+ u* l
62. CString m_description;
    
63. CString m_baseurl;
    - I0 a. j: x8 j" _- q# q6 x* i
64. CString m_controlurl;
    + o2 E4 D) m" O+ P3 s
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    ! k  v- ]5 C) {. u  |8 s3 q
68. 
    3 W; y0 ~$ B+ P: C& f, d; F, N$ w- z1 O
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    4 ]  j# d# J( D
71. 
    & A+ v1 \" n% \( G# P$ P
72. CString m_slocalIP;
    1 \/ |! |& {. _4 @2 Z
73. CString m_slastError;
    - @% b4 H$ M$ O1 Z0 [, [7 n/ P- d
74. WORD m_uLocalIP;
    " \& w- O; f: j3 D9 X: B
75. 
    8 Y, @, W2 `4 J: i
76. bool isSearched;
    
77. };
    0 E+ Q, N8 t  f9 T; }8 ~
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   & j! M# P' @! c5 F8 _
5. 
   ; c, X1 N9 Q- P0 O1 H
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   7 O; q. ~0 [" K
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    ' ~# g; G2 o  J7 R# q
11. 
    % Q  C+ b) A$ ~' g$ X
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    + x6 S( J, c3 E+ b
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    ( R% Y0 l) a$ g1 Y
16. const   CString   getString(int   i)
    . Z& k& ]/ d3 H- _, n! a* K0 ?
17. {
    ' [8 {8 L& h; a7 O, m
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    : t: u  e! X7 b$ V! L
21. 
    
22. return   s;
    ' {& E* R( [, u: P6 [& G
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    ! A% U) L3 T2 q% L/ c( K
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
      U3 T- `0 ]- D/ W' `, _# ]
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    ( c0 M9 m' A9 S  }7 q
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    0 [: f3 P) S  I7 b2 K" ]* {
36. {
    
37. char   buffer[10240];
    + l9 _, @/ r' U5 m
38. 
      E2 n7 h! k6 `; _
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    , X; m! k! E- @
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ) A2 ?6 ^5 G% R  Z7 z6 i  W) ?# k/ L
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    * x2 p( M& p' B: E7 d
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    : ]9 O# _: C4 l. v) m! [- [
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    # {! r! U  G6 L$ A/ V: E, ]
63. return   true;
    
64. }
    % o* A) @- O/ z. P/ ]" R+ D
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    ; l  U" @0 ], ~) K# u7 }8 }: E
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    % {, x: V8 o; z7 K6 Q; k
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    ( W5 w" J/ p  Z! v
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    , j7 L& v: u" M. F; U# @
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    ; S/ `: c6 w7 C1 Y! |) d
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    6 }: o9 L4 E9 T4 j9 A
79. 
    1 ]# Z: |, g1 C9 |: t' Q2 j
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ( m, |7 c! s1 _% @, ?" H$ e
81. }
    2 p* ]3 q$ i! u7 T. @
82. 
    0 J% M7 N: p3 h
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    
86. 
    . d5 M5 O1 q: A1 o8 R
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    + s5 x4 n: k! u
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    # Z! G" `0 i0 v, i5 y% y" F: H6 ?
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    ( {" r0 [( t. A  ?3 s. \
94. status   =   status.Tokenize(_T( "   "),   pos);
    * b2 f4 v6 z, c$ o
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    + F  p! g) d2 k) r3 {
98. 
    : w8 M7 O  Q* w2 e. N1 p2 v- _9 t, A
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    7 M9 T6 V: m4 {6 A( r) U* }. D
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     + M/ p5 m& ?7 W- E, z
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     5 T# F+ @4 v$ |
103. CString   property;
     9 g/ T3 \* w% p. ^! C9 l# K  i0 N
104. 
     ! y" ~* E0 P! c
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     , c" x. @. {6 K
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     9 e/ I! [+ V1 N8 O( B9 {0 V
110. 
     % k$ l5 B/ u# r  k, o5 d7 a
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     ! z0 o4 m2 D2 m$ e6 q! x3 f* B
112. }
     / F) |! ]: u$ x" Q: Z* D( l  T, z2 r
113. 
     
114. MyUPnP::MyUPnP()
     2 s/ \0 U( I3 O* X
115. :   m_version(1)
     9 H  X1 f. F4 y
116. {
     
117. m_uLocalIP   =   0;
     , T; b3 x- k8 Q: W. i9 }) [
118. isSearched   =   false;
     $ I) @9 r6 I8 _: f# u
119. }
     ) r0 S! J9 G& R, D2 r8 V5 D2 x: [
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
       M# ~8 |- N/ l+ f- C, a  B  U
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     % S: I& q" M9 A9 p" L
125. while(pos){
     $ T/ @! V2 V7 r1 d3 ]
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     + o* k( y! r5 l- E1 V
129. 
     6 k" t2 T0 @4 }, f) ]
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     ( m/ k8 s# f; o( U* E  X( a# Q/ L- w( \5 ]
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     2 f# v2 x$ t& j1 A# K
135. {
     / K# w( T  {& i) Y$ b1 w
136. if(version <=0)version   =   1;
     + e3 }& E4 R% E
137. m_version   =   version;
     
138. 
     , n3 v, q$ s$ k3 n5 }) G
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     % j9 }$ ?  ?8 x7 P0 o5 N; e% h
142. {UPNPPORTMAP0,   _T( "service ")},
     + v. K! y5 v; G# V: z  g* ?+ ]
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     : @9 A  e. j9 e2 W1 H+ {0 L3 ^
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     3 J! s  S! O# @
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     7 q0 k! o0 Z2 @1 h
150. int   rlen   =   0;
     # G  L7 G4 m9 D6 h$ B
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
       Z2 n+ s! L8 R
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     " h$ n( f1 j: X
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     * O8 }; j! R7 F' f5 [) ~6 y
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 "),
     - {8 |, f, F6 f3 n1 r. V
157. 6,   m_name);
     2 h) W7 v. [  n
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     " Z0 @' S& B5 b, X; b
159. }
     
160. }
     . f0 `) [( r- ^7 o( g9 Q' E$ z# [5 U9 ^
161. 
     + [9 @& h4 S- d' o4 s! v9 R
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     ( `/ T" P- w8 q/ s4 |% `- D' y
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     2 \1 Y' {, a2 ?  h  f- o' X
166. if   (rlen   <=   0)   continue;
     ( @7 E% ^' `% x! D
167. closesocket(s);
     8 |9 T; m' Y% S  ~$ T( s+ i$ H
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++)   {
     
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     8 ]0 X: x2 E! `! [
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     9 R2 E- a: R9 z5 }: _
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     / B) W8 ~' F4 D/ s5 M, A* X' w
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     $ ]6 ^. G3 R7 k# M4 V# {  R" Q
186. }
     ) Y: F& g2 e* p% ^  b
187. }
     
188. }
     9 _2 t* X- }: X
189. }
     # d" J' T3 ~6 m/ F5 t
190. }
     
191. closesocket(s);
     : a+ E+ b/ x) ^4 {1 r
192. 
     
193. return   false;
     + W. B) e1 n3 p
194. }
     

复制代码

天云上人

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


% Z1 A8 J0 c- n8 P///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
5 M4 P% j  e" K

; z1 I; d. N6 {$ O/ Q" b1 H#pragma once
#include <exception>
1 |- h5 d" g4 z( x
' v( x7 @* l9 j* {: x
  enum TRISTATE{
        TRIS_FALSE,
2 K. \% U# q; G        TRIS_UNKNOWN,
2 W% [9 Z- C  e* k2 q- m        TRIS_TRUE
2 W; M& C: y/ t+ y+ e$ G4 Y! h};
8 M- \4 y' Z4 \, J* w
+ f  H/ ]7 K# D& O- f$ p. @
2 i! ^& I2 S" ^7 s: benum UPNP_IMPLEMENTATION{
9 l5 [9 D: ?: H' o        UPNP_IMPL_WINDOWSERVICE = 0,
* d* G' J4 J5 s% z        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};

- b, ~  S+ n! ^' k: R
/ d3 M- q9 U1 `9 @
) ?5 X2 }7 ]6 @! k; m- t
class CUPnPImpl
{
public:
! h$ J9 k- ]( A0 E        CUPnPImpl();
        virtual ~CUPnPImpl();
. T. x0 L  e- V8 o        struct UPnPError : std::exception {};
        enum {
/ _* J$ m" ~8 |" J5 }( ]# H                UPNP_OK,
                UPNP_FAILED,
: ~) s/ q, G4 l* }# {+ }                UPNP_TIMEOUT
1 e8 P4 n7 N& ?  w        };

5 j. Q0 ^6 H6 P* W
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
8 x2 @+ q+ M; m6 ~4 O) _$ I) M2 |; u        virtual bool        CheckAndRefresh() = 0;
! j2 M- t6 L; q* Q& K+ k( v        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
, f4 E0 S1 j! h$ h        virtual bool        IsReady() = 0;
9 c# q5 `$ |* Y: A2 t' g        virtual int                GetImplementationID() = 0;
       
1 d$ K3 x: J: f2 |        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
! w& M! @/ X6 k; `9 Z: z5 s* \

        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
2 u% t0 r( s  K/ u        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
9 _' i- ^- F: Y' m" R/ n

// Implementation
protected:
0 s% P% L, e. X  a        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
; M2 Q1 \6 i0 t% z0 x, U- q0 G        uint16                                m_nTCPPort;
  K7 `0 M7 h5 G7 Q8 @, Q        uint16                                m_nTCPWebPort;
) D/ l' A! `' ^0 Y9 |        bool                                m_bCheckAndRefresh;
- |3 r; K& K* N

# E" {6 X% V  t* U, i% C3 L% lprivate:
) ~/ ~# g& ~) B( c8 x1 A        HWND        m_hResultMessageWindow;
" y0 i- _2 E+ Z" `* T        UINT        m_nResultMessageID;


/ o- l# s/ Y5 W: o+ y: v};
  @% I2 R, ]" O. Z+ G4 f, O
2 y- A. `$ U1 n* Q. t' U
// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
0 ]6 I: w4 v; j" a; f5 t' b9 Opublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
3 K' Q1 k/ P! ~$ c* ~0 f5 i        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
+ v2 ^$ \  I1 c( D. V- z2 q        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};


/ I# J4 c6 @& f% ^+ O/////////////////////////////////////
9 j# K1 v  I7 J3 u  S9 U0 z//下面是使用windows操作系统自带的UPNP功能的子类
# [1 _9 V" U. w- L; N  [
+ A, M+ E0 Y! c7 J9 {; L
9 c! j: S  f1 k#pragma once
#pragma warning( disable: 4355 )

5 C! m( V: A1 O
0 h7 ?6 m2 n$ j' Y# s& {#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
# [; I, n$ m1 Y#include <comdef.h>
#include <winsvc.h>


, T/ N* f* f1 M( S' I#include <vector>
4 b2 ?4 T& W% B8 @/ w6 g#include <exception>
#include <functional>
% b( f" M' z% O( |. N" O

( M/ v$ D* v! @& V3 Z* x
! M9 ~9 b+ ?! D1 ]( T& I9 ?
$ g8 q2 D/ }  g6 ^* ]" t2 xtypedef _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;
  `% m. d4 m# A$ h" X. {typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
) p9 u2 {+ S' E0 F: htypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
- b7 M$ n; W$ h3 M9 R; t- jtypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
: B- _9 M( t, }. u

typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);
) D4 O4 Q" h4 W

typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
5 f/ l+ \& z+ w  BOOL bOrder
8 S( @" `* ~& e9 o" P);

% `# ~2 ~' _, [# G
typedef DWORD (WINAPI* TGetIfEntry) (
( P4 I6 }4 m' w4 T  ]  PMIB_IFROW pIfRow
7 O; I' J8 t9 ?8 [9 {( P) E0 e);

2 M% J3 }- z6 @0 \, U+ b' G
6 B, G" k0 y- U1 v* c6 Z- RCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);

0 H8 W' W$ Y/ T9 @
+ [. z' h4 U4 b! [5 |" y* nclass CUPnPImplWinServ: public CUPnPImpl
{
0 D6 _) q8 y+ y0 E: m        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
3 n! w6 z5 m: f. J5 n7 e// Construction
, z4 l% E* q5 n. Qpublic:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
$ O6 e8 ^% S  `+ o$ ]3 c/ L! ^! s& l
' {' E8 m8 e1 O  \: ]
( P3 C' T/ t% f% F( r, ^* J        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
- Y) q5 ~, N2 _9 U' }        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
* \+ D( j8 d4 U, |        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
% z0 F0 B; t5 b4 o

) [3 w. p# ~2 e        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
0 q* F8 V. ~! p: {- ^' a        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
* p- ]6 n' f- @- X4 y& G* a
& d, N* A4 y0 S; P; i; }6 H
protected:
& {3 ~- j* }. M4 ?# u( X4 N        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
# D) N/ t. n. T        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
! N) k0 m1 {' r: M        bool        OnSearchComplete();
        void        Init();
4 _' T. x; G8 C. B: `6 r
% J* b; h+ J. w+ k" L% x: c$ `4 p2 ]
9 \# m* X% q. Q) K& n) s" H        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
4 h% |! L8 U/ W6 g0 W                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
+ r* T- @0 I# e, F2 T7 K% U                        m_bAsyncFindRunning = false;
                }
                MSG msg;
* V9 Q' g1 X5 d) u1 _8 a/ Q) u: O                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
; `* r% [% s$ Z' s% w                return m_bAsyncFindRunning;
        }


* j( w" @3 }6 m: _        TRISTATE                        m_bUPnPDeviceConnected;
. ~& I( z( u; L* A+ ~6 C* `8 v
6 l' c9 ~( P+ c! |
  ]) J  F( W/ g" I9 U( Q2 j// Implementation
  i3 E2 X0 z- _, A- O/ v3 @        // API functions
  N! d9 u- Y' v% a        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
: p- R. ^% X9 A1 @- Q7 u        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
7 R8 R) i5 L* W/ u; Z$ S" l& p        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
" g% b, v5 u' U) ~, T* F        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
* I/ w7 ?: c* W6 @1 E        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
( W. r$ ]3 R3 b$ T% h

        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
0 d4 q; _7 F  h- B6 n+ N                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
: }6 S7 U; e" I% [& l1 F                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

0 p  D6 \& ]1 W- |
                        if ( FAILED( hr ) )
" G9 V. @2 F* f8 I- `) U                                return UPnPMessage( hr ), false;
6 v0 ?7 c8 M% G0 A$ d

                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
1 b/ o( e: Y! k5 s        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
% D8 D9 _# z! r6 G' M* z7 E$ O        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();

4 D- R9 l% T5 b! z2 Y& N8 r3 N
6 l: z2 x, n( v! a1 m0 w+ u# D        // Utility functions
% _+ k) N: \6 ]" y& y        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
6 j' k0 s$ [4 e3 m& {6 ]. X6 c        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
2 j) [/ W1 }0 u9 [        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
$ M/ B7 l! o  p: O: |: ]        CString        GetLocalRoutableIP(ServicePointer pService);
  f4 }, \9 e3 l( R

" a1 o9 C5 |7 s" M$ e+ y/ I// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
) e. |' e( N# _9 y        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
, g, u% O0 j+ }: _3 G! A4 Z        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
. @! Y: o4 Z' q6 w& Z( _# H# N! q) r        ServiceCallback                        m_pServiceCallback;

8 u) x3 d1 \' w0 d1 U
: P: r& _$ g3 ?! H        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
2 n$ s9 ]' E4 y) b* [7 n        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
9 m+ R& @6 u& C, c        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
  ~* _" Z& l9 H: D3 h  S1 t        HMODULE        m_hIPHLPAPI_DLL;
6 q. Y+ Y- @+ \8 K! c( B        bool        m_bSecondTry;
6 E- X9 \! n7 u) F" V  E3 k        bool        m_bServiceStartedByEmule;
9 z5 ?) m' |5 q# c, B) _' T        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
  g1 k# u* U) p0 W% [
8 B9 `1 Z: a( L: }5 g
; W7 J9 j( p, O% T: _+ _};
  ^7 V3 }: Y; [+ Q" ^. ]0 s; ^
$ T1 P% L# }% p( Z- v( v
$ ~6 s2 A2 A# ]& Z8 b: c// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
/ R0 Q( ]- Y8 U( I/ V  d  a, O                : m_instance( instance )
6 ~' d, f( L9 {" _8 K1 n/ ^        { m_lRefCount = 0; }

% v1 t8 L" s' j; }3 g  Q5 a
# A+ w, q; @7 [. ?   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
: A- @' r! T4 d% X6 ~& i   STDMETHODIMP_(ULONG) Release();

) ?0 l# Y$ P5 h& Q4 \' R6 E+ s, t
// implementation
" C4 h6 S7 Q# gprivate:
4 N4 @' V9 V" }: x* N/ G+ F        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
9 c' l6 \$ Y8 k- F# [; Q. u/ Y$ y        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);


1 H" v4 V, I* E0 lprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};

( s" o5 L8 T" a
' H% |+ j; v& D& _+ i* {- ^// Service Callback
" I! m1 ?: I" h" k1 Q1 x( xclass CServiceCallback
        : public IUPnPServiceCallback
{
( A7 g0 F& g, p1 M9 H# i; a  f7 mpublic:
) i4 R" S8 Y7 d4 G3 ^        CServiceCallback(CUPnPImplWinServ& instance)
+ n; i3 w6 R% f% C3 f: u$ b                : m_instance( instance )
        { m_lRefCount = 0; }
. B3 C, k- k  z- W: S, V( Y" J   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
( W( H: E6 b: U% V3 X

// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

5 e( k6 N( m+ v' R7 c/ T
# q. S" y; y  c* e' R6 T( [. Oprivate:
; U/ t' F5 ~5 z$ ]        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
; w  F3 B" u/ h. K  |; J

/////////////////////////////////////////////////
# l7 P* p) V& [0 w$ }! ?

9 ~+ j/ B7 ]% W( O  z/ K) V" U使用时只需要使用抽象类的接口。
% Z, n) U- r# P9 L# t4 ?9 t/ QCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.