UPnP

天云上人

/*uPnP.h*/

1. 
   3 q0 Y3 F' C( d$ c: N; r
2. #ifndef   MYUPNP_H_
   
3. 
   ! \9 F7 |. [& \9 K; a+ u) c& V2 v
4. #pragma   once
   * l; w6 q3 T+ A8 n- d- R
5. 
   
6. typedef   unsigned   long   ulong;
   7 \+ B9 I# F3 o; `
7. 
   
8. class   MyUPnP
   
9. {
   ! w' f: W5 M! y* s- X: a; B
10. public:
    # t0 R; ~: b1 @/ k0 z
11. typedef   enum{
    6 q; y- |7 H# k( ~2 _  l) F
12. UNAT_OK, //   Successfull
    # L9 t8 T7 d7 `  e3 M$ S
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    , Y% s* ^5 W$ O7 g
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    9 ?* g0 H8 P  Z0 G3 o4 A3 a( ~6 Y- d3 @
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;
    
18. 
    8 {" a% b$ X2 o, ]. O. c# R& v
19. typedef   enum{
    ' y4 Y) c3 ~5 W
20. UNAT_TCP, //   TCP   Protocol
    3 H( B. o# C! v& S& U
21. UNAT_UDP //   UDP   Protocol
    : s5 I0 Q: n' d, {! B
22. }   UPNPNAT_PROTOCOL;
    & [5 j, S( `' l) q2 }/ s
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    / Q* c' T: N! [6 B: r- V! C# u
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
    9 |, e6 @- H# J9 U% R
29. }   UPNPNAT_MAPPING;
    5 v8 @1 \$ X8 N* ~( B0 A; K! A* H  |8 A. o
30. 
    ; N" C* B; d* [8 V8 R+ z# e
31. MyUPnP();
    ' E# W, _& ?/ |% t7 N
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    " D9 g  F; W3 F  N
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    * |* F0 u% }9 ]! G/ ]( N
36. void   clearNATPortMapping();
    5 O7 H# C, {# y) W) J  X  j7 p' g6 _
37. 
    
38. CString GetLastError();
    ; l+ f2 @% f# \4 `, }
39. CString GetLocalIPStr();
    . e8 P" z9 y/ n
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    $ n7 ?* J! Q) l1 y
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    : \. B0 v. y3 k# _9 J
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    0 |9 H" s+ m, @
48.       const   CString&   descri,   const   CString&   type);
    & ~8 i5 `4 x  F8 L9 Y% S$ g8 Y/ {
49. bool   deletePortmap(int   eport,   const   CString&   type);
    ; G+ \( m( }" Y- d+ ]( ~4 t% f
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    ( l+ z- D4 N, x
52. 
    & X, K7 F; t6 {
53. bool Search(int   version=1);
    , v% y( s8 m0 h2 b' [) r
54. bool GetDescription();
    . ^0 z& j+ o4 X4 I; z
55. CString GetProperty(const   CString&   name,   CString&   response);
    : Z$ K& N/ b' s: ]& R
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    - B# }" A- N  _/ E$ R
57. 
    0 b$ Z9 w# x5 B( t8 \2 q* K3 o1 y
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    8 o7 j& |$ Q- P9 s, C6 g" {9 y/ M0 K
59. bool InternalSearch(int   version);
    7 B% L! S! u- n/ c( j& Q
60. CString m_devicename;
    ; N( H6 ~" v, b+ j$ b6 X1 A
61. CString m_name;
    7 C; W7 T0 g- ^/ y
62. CString m_description;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    ! z# J6 l' P& t( l, l
68. 
    : A( i$ N) i  k( ?( C& J& M8 N
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    + D; B. x& l9 J: N5 ^' u
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    ) }1 F. l- d- D% ~) V- z
77. };
    & v& F" V' {& v* r8 K
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   
5. 
   ' ]8 c8 C. F# M5 s1 `3 i0 U' @
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   3 {% d$ d' ~0 d& i/ |' h- d8 _
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
      {5 k; i- F2 U* ?- N
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
      a+ D0 |3 L% u5 S
15. 
    
16. const   CString   getString(int   i)
    
17. {
    / ?" L7 b, U  o# p! S0 {7 V# f
18. CString   s;
    
19. 
    # m$ z. `: i7 Z  P
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    
23. }
    ! i4 ^% Q- g9 K. E3 _( }! q! v  y
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    9 _. [  e8 X5 a, s
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    0 R* K: F: v3 n. A, ]
33. }
    
34. 
    ' \( i3 A0 N: j1 ~
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    * W# X# a  N& w" P5 V4 K
36. {
    6 o) L3 c1 `5 b
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    ) W: x# L* J0 n% X0 V
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    $ `, i8 Z/ J" g  e* x$ U
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    , h3 Z, q: J8 c6 b# W
50. u_long   lv   =   1;
    / _. F0 K8 j/ |8 ]
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    4 y) }' j8 u  A+ R8 o2 _% D7 h
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    5 L* z# w1 _7 D, F: X$ F. d
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    6 o. \; D" V8 V' d* L
57. closesocket(s);
    ' n. v+ D# z7 x$ }& |) u) ^$ C
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    0 @6 F! `* y* g- z; T, }
60. 
    ; T+ A# K, p* [& T: W8 c
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    ( F% ~& z2 P8 m3 o& x
64. }
    / u' ]' l7 C( ~& O/ P0 U
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    
69. 
    4 \4 R/ `6 J/ T  F
70. const   CStringA   sa(request);
    ) R+ _# x* P4 I1 `$ Y, j2 w' {
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    2 t- P& h- H3 }* ^/ ]1 ?: f' H
74. struct   sockaddr_in   sockaddr;
    6 K( W1 v% B. [* k5 Y# c
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    ( |4 Q& b9 T, W2 r
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    * n, O. D" ]5 V  m1 `$ T# M* g
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    8 Y0 E) `2 G+ N0 M* o1 F% R
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    . F+ ^: t. Q8 n' V* `
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    ; g% G6 S; U8 X6 \' _* I
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    . E) j3 }5 B. X# a1 v" W
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    7 Z: u4 w0 T5 {; q2 r- p# }3 j
96. return   true;
    ( v$ F  y% M! m$ z/ o: W/ ]
97. }
    
98. 
    4 U) Q! U! w3 C: K: S
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
      e! c/ P8 L  X; j& z
100. {
     . Q# Q4 s6 _/ r  X3 e
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     $ z1 Q; j' I9 G: a
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     , n0 T: Q: y  w& N, H- Q
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     2 I  ^: R! x+ Z$ R5 F* R  [
117. m_uLocalIP   =   0;
     6 ~9 p+ \, w- U8 Y
118. isSearched   =   false;
     
119. }
     - |+ c# Y! q4 a; ^  f
120. 
     
121. MyUPnP::~MyUPnP()
     0 S7 g) D) C: d6 {6 j
122. {
     
123. UPNPNAT_MAPPING   search;
     8 V0 X! v- q5 n+ g) H2 w0 P. R
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     5 z& U4 W/ [4 S- y, u# R0 u1 j+ N
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     9 s, Z7 C- `) _/ k
128. }
     2 x' \$ X* _8 T5 ^# d# w
129. 
     , o8 j9 ~" T4 Z6 g7 W' Z6 z. @
130. m_Mappings.RemoveAll();
     - M- ^5 g* W. V
131. }
     5 K' g  s) Y! Y# w8 \2 D
132. 
     ( ?# q$ H5 L* z" R/ I
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     ' H- k+ S' A) s! s& _/ t. P& n# t
135. {
     & [  p' J. l. h# h0 B3 |
136. if(version <=0)version   =   1;
     # i% e# C; c3 c6 j: Y2 t( C' Q
137. m_version   =   version;
     / m8 L/ x8 D( t  E7 x& p
138. 
     5 U. E- _6 a7 x( H3 n
139. #define   NUMBEROFDEVICES 2
     ) \1 _' r0 t0 ?7 J
140. CString   devices[][2]   =   {
     3 `3 E. p6 k7 O& G$ h/ J+ O: |
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     - v# w- B0 W9 M' t' ^- K! Y
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     ' Q: G% ^/ X( ]8 f6 O" q" c
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
       k- a7 u9 T& v1 ^  a  Z* [  B
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 [& C( O. [+ d, p# c7 S/ z
157. 6,   m_name);
     ! a+ v' D  t' ]( q$ V) B4 r
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     / b: P! O+ m/ o& L/ B3 K+ B2 C
160. }
     
161. 
     6 v3 {/ ]: n( |0 Q' i+ P
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
       |' B9 u* e) _' V: W; M
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     $ r" n7 R4 ]8 f$ X" U
166. if   (rlen   <=   0)   continue;
     ! g7 v$ E8 Y/ F6 ?1 R. u7 ^! q9 J
167. closesocket(s);
     
168. 
     ; q, I- I. Y  R/ O! X6 a
169. CString   response   =   CString(CStringA(buffer,   rlen));
     8 }& H& ?. @& S% |3 e
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     # @8 |" x# N( U8 Q6 n# z
172. 
     - j1 v9 d& H/ x
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)   {
     9 D! a) N% D, k, R" w% C# s
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     9 o8 K2 i2 Q# ~4 W' d
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     8 h* X9 m: |4 b' M" `
181. if   (name   ==   _T( "LOCATION: "))   {
     6 a( W3 j' H# R2 W7 @
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     9 E5 R" a$ Q; ?7 Z& w) R
186. }
     
187. }
     
188. }
     : T  Z5 {5 H/ _
189. }
     
190. }
     
191. closesocket(s);
     
192. 
     
193. return   false;
     
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
3 ~) m3 [. h) f  m& D
: P- O& s' {5 g4 T! W' U  `0 h: ?
///////////////////////////////////////////
7 D7 E  p' V' N/ l* X2 Y6 q# l5 q//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


#pragma once
  Y) G$ I5 K' l( E$ G) e#include <exception>
. R* a% D  {1 ~. z1 `
! v. Q0 U" B; `% }3 r) Z
+ b4 F* Q1 h6 s& r% p: W  enum TRISTATE{
$ ]* a1 D9 @& ?! J, X) E1 A+ q        TRIS_FALSE,
7 u$ D, {) U' l$ T1 J        TRIS_UNKNOWN,
        TRIS_TRUE
};


enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
4 I, c1 J! _( D& L* v        UPNP_IMPL_MINIUPNPLIB,
7 ?! j! z4 P. }3 t+ K9 E        UPNP_IMPL_NONE /*last*/
6 U7 y* v8 o1 v  R};

" a3 D; W" l3 c  M/ o6 x

$ _1 G" l( R8 }' n( L+ {0 y$ J
class CUPnPImpl
{
public:
1 _3 ?3 A- X) f) i3 l* L        CUPnPImpl();
; U/ h& B4 [+ P$ Y9 V        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
! z. e0 s8 f4 T! y        enum {
+ v- ]0 Z2 V3 [" l: {0 [' u7 r                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
        };

3 S1 A* s8 F% s9 _
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
" J8 D; i5 H3 z$ w        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
; f- D; S$ l% H3 I8 u$ W        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
$ _, T* S# n$ j4 l1 [6 e
' S) S9 y: P% g
4 e- r; W  l7 i" I2 D4 g. e        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
" F  ~8 H; B+ S$ y        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
9 `9 X2 y1 g& G: q        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
' Y( i: S+ p5 N! n
1 o# x: T! D* _3 x: B3 ~' a
// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
% p: v4 f) Q2 ~0 [; r) C        uint16                                m_nUDPPort;
9 K8 l% ~2 S6 k4 v% b$ I+ w" h        uint16                                m_nTCPPort;
, R5 H8 E4 D. c1 v* ]  X        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;

" J. R1 \% Z" m6 f! ^
private:
        HWND        m_hResultMessageWindow;
: [) k8 i. Y. a5 e" g% e/ z        UINT        m_nResultMessageID;

9 b6 l7 v  }+ o2 E; O
2 I" A4 [- N" ^0 M& s' {, Q* `- x};

1 j$ r9 N+ s7 w! A& G
+ h" m2 C, z/ m% G+ M// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
6 ?- M5 \  b% k' I7 T' G  @+ F6 Vpublic:
* O/ M% A* ?5 @  I$ }# |/ S9 v        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
* u9 H) J5 o' y1 h, T        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};

3 G0 H# P. @# E( G
/////////////////////////////////////
$ F* l- `( Q. t( r9 D//下面是使用windows操作系统自带的UPNP功能的子类
5 F- I6 ?% ^# x; M2 R' }
2 [7 s" C3 F9 f+ }+ v
#pragma once
' x  Q3 I* ?. F$ U! y& u* h#pragma warning( disable: 4355 )
4 O8 \( d7 w7 Y7 t# p1 y

#include "UPnPImpl.h"
& f% i. H* F* i4 a4 s#include <upnp.h>
#include <iphlpapi.h>
3 s; m( v% D5 B1 P; a) E#include <comdef.h>
; @) e  i* J3 B5 P#include <winsvc.h>


7 _$ R* ]; _: \  d) X#include <vector>
$ u. ^* R# ^* F& e8 {1 H& P- G  c8 y#include <exception>
#include <functional>

: ~% d1 _4 f" c) s


' c, h% g+ _7 A' D6 Utypedef _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;
* C  z0 i) \- E$ X# atypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
& @' Y! h0 @% D* G( A  C: ^. _0 Ytypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
( \" p0 M0 R& ~) k+ ^+ mtypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

4 D, G3 J3 x3 v1 q9 c
typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
6 V/ i; q5 b+ ?0 f: l  PDWORD pdwBestIfIndex
);

, y" j1 B" m$ i" U& {) }  X9 o
typedef DWORD (WINAPI* TGetIpAddrTable) (
' w! ?, L) O) a; b6 Q0 X+ I2 Y( O5 G  PMIB_IPADDRTABLE pIpAddrTable,
. m0 f" u" l- f) ]0 I% L% ], u: Z  PULONG pdwSize,
  BOOL bOrder
);

% c. e: I: p1 R
; y9 ^8 k5 n% N8 r4 L$ Ltypedef DWORD (WINAPI* TGetIfEntry) (
5 h+ b: O& [; F; {9 o  PMIB_IFROW pIfRow
; u+ u, |- c2 i( D8 }" H);
2 k1 ]1 J- N# w7 M! R" m" X

& e1 y: s3 X2 l  E( FCString translateUPnPResult(HRESULT hr);
* i7 }6 q3 N* @4 ]! J  u- bHRESULT UPnPMessage(HRESULT hr);


class CUPnPImplWinServ: public CUPnPImpl
) |! }4 m8 P$ q  i+ Z) _{
        friend class CDeviceFinderCallback;
7 V; l: X) h$ A% g2 v" `        friend class CServiceCallback;
, a* c! f" Z3 p0 N; L// Construction
public:
        virtual ~CUPnPImplWinServ();
/ k( a+ S: ^; r. J" {' K( S        CUPnPImplWinServ();

/ X9 m! d) h& ~# h$ |3 N. U! d
6 p5 J/ @/ T1 ?2 F3 v7 [        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
6 T; f8 b4 i2 d, N2 ?7 W5 o        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


# v* x  k/ q- R( U/ b4 s* D# u% B        // 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
& i& _+ v8 |( ?2 @  J        virtual bool        CheckAndRefresh()                                                                                { return false; };

) O( ~$ D7 T8 j6 ?4 B! a+ p
% q; d$ p6 r2 J6 a% Q! |; Zprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
% m+ o. v$ O0 W3 p        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();
3 K( h. D4 |; t2 k& I
" b0 ?" `# L" N0 R# f7 Y* P: v7 I7 K
        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
' v! _, V. a$ Q8 N                }
$ |' ]6 u1 T# _2 w2 n9 j5 e                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
+ W* |/ v9 X+ U0 G                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
        }

+ \( ]& c* A5 Q/ h0 z7 a, |# p
# N0 d7 E) E, b7 v7 F, ^        TRISTATE                        m_bUPnPDeviceConnected;
- a1 v% Y1 n% Y0 u: }

// Implementation
        // API functions
4 ~& m/ ~% T# m) @        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
7 Q" g; S. y3 _- _8 u9 b6 z/ V        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
% j1 ]& l, T7 p$ w$ ]/ T' @' G        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*);
$ K8 ?& j/ g4 _' O2 P0 z& f        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


        TGetBestInterface                m_pfGetBestInterface;
: i1 U, S, t2 {        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
) P: m5 x# ~9 A) r
. N! g1 X  f$ |/ W
0 {4 Q, p" Y/ Z3 ?- \        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
5 Y% z: k5 D; {& {6 h                result_type operator()(argument_type device) const
% Q# [* b5 @, d, y5 J/ u* H                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
4 Y( A9 p# `3 e' w: X

                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
, c4 e% e/ H, d8 r( Y0 D

5 [' W  b3 P9 e. j                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
% q: c, n5 X# H, X3 P9 G3 p        };
  }5 F0 b9 ~: i/ m       
7 i5 V! \8 ^  l4 w9 v% ^) L  Z/ v        void        ProcessAsyncFind(CComBSTR bsSearchType);
/ O* x, m7 }0 K8 @' h        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
7 R4 t; K' S! n, i        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
2 _/ N' ?" W0 J, {* k* H        void        CreatePortMappings(ServicePointer pService);
' x/ i0 q% q+ g; J        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
0 U; w4 ~6 [& c# ~( k' X7 Y        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();


        // Utility functions
! |. O0 z  A1 z* w" X        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
) e, F3 p! [4 z        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
7 ?5 @1 f( o, k4 a. ]. s        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
! v  k4 X' \  ]  f) r& k) m        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
; ^" Y3 P/ o1 F' O* s        CString        GetLocalRoutableIP(ServicePointer pService);
( o; e) c: Y9 N/ L

8 v; ?" ^( J5 A  B// Private members
4 J9 X! {# t% N3 Bprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
2 S. @1 x' M" ^/ W' v        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
- @  v' \! I1 u/ b4 I/ J* O- k        ServiceCallback                        m_pServiceCallback;

1 R3 h2 f, M+ N. N+ p
' ?' u- D5 \( B- n        LONG        m_nAsyncFindHandle;
1 A5 a- i$ l' t( i: m% r+ _        bool        m_bCOM;
' O* ~) E/ L0 T( {6 m, F        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
* }3 }% m+ t4 Y& r  J        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
$ [/ l" R5 G8 {, E1 U        bool        m_bInited;
. ~2 E; B2 \( z& r' d        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
# ?7 O, v3 q& `( a0 P        bool        m_bDisableWANPPPSetup;
( g! E( s% T1 p# Q& o  n, W. B

% c5 ]2 G) f9 u};

% L4 [3 d1 f$ z( i8 S# ?! f
# K. `3 F9 \. V1 I# \4 I% A// DeviceFinder Callback
. C8 F- g( l/ J& Oclass CDeviceFinderCallback
* a8 H1 F' D3 V# ^8 L        : public IUPnPDeviceFinderCallback
{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
) B: n+ ?# J' v5 u1 l" a                : m_instance( instance )
        { m_lRefCount = 0; }
& @# {: J" i& T# v3 |

   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
3 C/ N- c: b, ^' y+ w/ C0 a* B   STDMETHODIMP_(ULONG) AddRef();
3 x9 O3 |# S( J1 T; }   STDMETHODIMP_(ULONG) Release();


4 w8 O6 ], O9 h1 ^$ e% h5 g// implementation
private:
6 A  I* s2 r3 A" a$ N, Y        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
3 D+ L' J9 n  p- S2 u$ n. l        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
5 j8 Y) u! ~' D7 }4 F
2 F  X5 `, o' G( F0 C/ [0 O
private:
/ R1 J. z6 ~8 z, u        CUPnPImplWinServ& m_instance;
* O0 g7 e1 |7 V: E( W3 V( d/ P  y& V        LONG m_lRefCount;
};

2 M' ^" j1 @# \6 V9 B
// Service Callback
4 A0 E3 G' x! H+ a  v" fclass CServiceCallback
/ X  y6 b" B/ h% a) w        : public IUPnPServiceCallback
{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
   
" ^3 e0 F; `* b# {7 Q   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
' f' r- P7 B0 g$ H1 V   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

, p# h! M4 d2 i: p# b0 o
, T& F4 r0 s3 P, `. Y3 a0 c! A// implementation
private:
+ ~; d# N1 t! n        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
7 S. J$ K" `8 i: H2 K
' o( E7 N/ N' k. Z3 Q2 Y; i
' m  h3 @' J& Z  {private:
$ V3 m; h0 A- R" b        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
) U5 G  D+ ]4 z- r};

6 f9 l0 W% z: T0 I& d5 v0 K* e
/////////////////////////////////////////////////

) H3 ?& f  F7 B* d7 |' `
+ z' c6 w6 ^) `6 \& m使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
$ t* j  S1 s$ J" s2 ?9 H. vCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
7 \8 A" i6 c. vCUPnPImpl::StopAsyncFind停止设备查找.
: q# W' a2 j. n/ L' r* e% ]% `CUPnPImpl::DeletePorts删除端口映射.