UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   1 _. u" w2 o$ ~) O
3. 
   & {5 H# G7 S9 c' O1 i8 S8 N8 c
4. #pragma   once
   $ q) W6 z) y  E# V# h9 ]( S5 A/ V
5. 
   
6. typedef   unsigned   long   ulong;
   . W! ]; c( }6 ?/ Y( _
7. 
   2 y3 T3 w, q3 N1 E
8. class   MyUPnP
   9 I% l6 D2 F& D/ d
9. {
   
10. public:
    6 ]/ K* Q% G+ f* d
11. typedef   enum{
    + F3 ?+ ~. Q; g$ H' |
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    " y: X( p/ U' S
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    / `! ~& }6 `2 N! |7 `
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    , {# d1 s4 L. s1 }, e4 `
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    2 J' t, U$ t, x3 t0 h
19. typedef   enum{
    2 U- r; r) ^9 `$ i% L' R
20. UNAT_TCP, //   TCP   Protocol
    ) l3 n8 ]. u. M& F( k
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    # n/ i  C; f4 e8 j5 \
25. WORD   internalPort; //   Port   mapping   internal   port
    ( T6 s9 |0 y2 Z3 m* B$ g) |' s
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    
31. MyUPnP();
    ) B; A6 F7 V) k8 }8 v/ N/ t
32. ~MyUPnP();
    
33. 
    # Y+ A3 h7 e7 W0 x. i3 S& A
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    + U5 V7 u/ V( A3 @
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    / k) M  C5 e1 N5 x* B2 b
37. 
    4 ?' o0 I2 g# L- O1 @2 z& y
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    2 i1 B$ r3 h$ |+ L7 \+ v
43. protected:
    
44. void InitLocalIP();
    * `8 C& m+ `  k) w# b# _. `
45. void SetLastError(CString   error);
    
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    5 J1 \- ]8 F2 J. g. I
49. bool   deletePortmap(int   eport,   const   CString&   type);
    6 s( f8 |0 ~' \: F" u$ m+ [9 D( u
50. 
    2 `) Z. q  x- k( }' L9 d4 R# r- {# h
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    5 {: w# @4 ?7 u% i) w: J/ W* S- u
54. bool GetDescription();
    ' I6 o4 C. U5 Q8 s, g5 l; B1 c' K2 A
55. CString GetProperty(const   CString&   name,   CString&   response);
      _3 y/ q* ]* @0 _( }
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    , G3 ^1 y6 q9 f  `# F$ q
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    ! F7 }, E* C% W  m) S5 ~
60. CString m_devicename;
    1 V% P4 J8 w  ^
61. CString m_name;
    - V, I8 I# j1 q
62. CString m_description;
    3 o' n: Q8 H- L! S* N
63. CString m_baseurl;
    
64. CString m_controlurl;
    ( a* M* Z3 w; @  i
65. CString m_friendlyname;
    # _( e% P1 m+ P! v! g% W
66. CString m_modelname;
    
67. int m_version;
    3 {. P; V/ x/ X9 h& `1 _) i9 Q* ~) j% L
68. 
    9 f2 m9 c; f" d6 F' r
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    % _: k  z8 j; i+ t+ E( D
75. 
    ' ~6 `) C" z/ a1 J  h
76. bool isSearched;
    4 C: x8 _- x. T% p1 D
77. };
    - ?7 ?/ i4 M7 X2 s0 B! q' A
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   5 R+ e3 {. R' @' V
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   1 R: G& M7 t( P7 |
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   ! [$ [1 p: D: R
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   0 c) C# z- T$ H
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    $ J. n* E% B4 o% u
16. const   CString   getString(int   i)
    ; M7 E' y3 m$ H* ^* Y& n- X
17. {
    
18. CString   s;
    " V3 `6 [1 }8 C
19. 
    
20. s.Format(_T( "%d "),   i);
    , h9 f4 z4 I4 |. t9 r
21. 
    
22. return   s;
    
23. }
    8 z1 d3 A$ x& x( D: |- i
24. 
    * L% ^7 z3 r( n& v# q  i
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    9 Z2 d# v6 j. g1 m5 V
26. {
    2 e" f% q' |* p* n& W  p! Z
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    6 S% Q7 ?, Z( L8 k% i- f
28. }
    
29. 
    2 C' g+ m2 h2 y' [# N9 Y/ q
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    . \8 P- t& O* W+ W/ }$ E/ |4 p3 a
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    9 c" r9 U: F; N
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    ( j5 c: f# ?; r: G- A/ p
36. {
    * T+ |2 k1 \: p7 ?
37. char   buffer[10240];
    ) n5 H4 [  n3 ]5 t& W/ R
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    1 b6 U* s8 \, U0 V
41. strcpy(buffer,   (const   char*)sa);
    8 E8 A. }, i* ]8 K
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    : `9 H9 A% |# R2 r: H0 s1 T( \
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;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    0 U0 f0 f4 K/ }9 h4 N
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ; E  p- D2 \  O; S
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    . f3 j2 B+ K' m2 {$ Z  d7 x
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    7 ?4 x4 f+ U. v6 t0 E
59. if   (!rlen)   return   false;
    1 F. i0 U2 V+ j3 o) ?5 q
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    ( U; O1 m& }5 |9 o
64. }
    1 H# ~$ V3 `+ H8 g- K: R
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    4 A1 H/ S% z5 J/ a' u8 g
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    # D9 A9 U# C; P- ~5 |6 h1 P" n
72. strcpy(buffer,   (const   char*)sa);
    ) _  v7 M4 p: S: @6 j0 ?$ w9 p6 @' Y
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    5 V1 N! H9 L0 w, G
77. sockaddr.sin_port   =   htons(port);
      @( G! o4 \1 p9 R8 l) z* v
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    5 \1 c7 K& }1 q/ p
86. 
    0 P% r# s$ U; a
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    ; w3 w2 ~5 \8 m7 |4 Z
90. result.Delete(0,   pos);
      F! p- B6 O, u8 b
91. 
    " I1 _7 y1 c; R6 J% u: Y1 ^8 V2 D5 L3 Q
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
      y# f0 J: k% S
94. status   =   status.Tokenize(_T( "   "),   pos);
    9 M/ u- c& R8 X/ H! w8 R1 ]9 t
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    ) x2 \, i1 g% T7 u; i; N5 A
97. }
    
98. 
    . Q2 Z2 k- u5 I0 C
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     * s) q/ q5 T" M0 n
101. CString   startTag   =   ' < '   +   name   +   '> ';
     0 c$ Y2 A  Z) c5 D
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     , ]0 ^* ~6 @$ Z' ^, c2 F  ~  |& F
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. }
     
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
       E' o- q0 j) k" H; W
117. m_uLocalIP   =   0;
     ) [/ J4 v; E9 _/ X
118. isSearched   =   false;
     2 D2 f* l. m# w& t
119. }
     
120. 
     - f  Y2 ^5 N# T! Z; d' ]
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     / j* A7 q" ^( _( |/ t* a+ E
128. }
     - M7 `1 k# L' T( J
129. 
     & g$ ]4 N% r4 K* w
130. m_Mappings.RemoveAll();
     
131. }
     * g8 o0 s+ {* |) M* N$ q
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     2 f  c/ B  [! I/ K
136. if(version <=0)version   =   1;
     & z1 y: I! [# f1 r# {3 R
137. m_version   =   version;
     
138. 
     2 K4 r2 |+ M8 H, T' s
139. #define   NUMBEROFDEVICES 2
     7 T: j, ~5 k" ~1 R* s
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     " Z7 i8 c& @& }5 ?4 |9 z
142. {UPNPPORTMAP0,   _T( "service ")},
       g! l5 w% {! j( Y7 _
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     " Q( X. ~* l( v  U' {
144. };
     2 {- u, p# a2 A; @% P
145. 
     . S3 B2 z! `  g
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     7 \8 N% @- \5 D( l; k5 N% Z5 m
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     6 a: {+ S7 h2 ]9 J' Y" Y; U
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     4 R$ d) `1 c2 r
152. if   (!(i%100))   {
     4 d) s# x6 p0 b
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     % k9 A8 F* J  {. R3 u7 B; @
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     ( n2 M& g* z5 c" Z& g  R
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 "),
     
157. 6,   m_name);
     * J1 c2 t& M; @
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     8 Y/ Y" z/ Y  `( N+ N4 o: g3 y' @: x
159. }
     ' U% p) s& o. q& v; |# Y2 X1 u) U
160. }
     
161. 
     
162. Sleep(10);
     & M' @: X( u( g# m( l% ^$ j- Q
163. 
     1 R$ s8 m. {1 N; R9 _# Q
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     # E, L1 k2 }5 n- w: Q
166. if   (rlen   <=   0)   continue;
     ( h9 O& ?: P# l2 @/ L+ p. u
167. closesocket(s);
     * S( d+ ^! M) J9 O
168. 
     5 O- D. ?  p' b
169. CString   response   =   CString(CStringA(buffer,   rlen));
     ! w$ ~+ y- b; Z2 i. o
170. CString   result;
     8 \3 |4 m7 k0 e' ?
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);
     ; H& C- a+ t8 D1 ]
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     / F2 o$ Q$ F7 I
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     3 o9 x* E8 U8 h5 [' Z- y  Z( J8 l
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     % {, ^3 m- t9 N( I/ I; L: t
187. }
     ( g7 e  `9 S0 Z
188. }
     
189. }
     
190. }
     
191. closesocket(s);
     
192. 
     ! L& T' M& J7 L8 Z' H/ T% t5 U: ^
193. return   false;
     4 {: ^& C: t4 z- B' s% Z" J# R
194. }
     " \4 h5 L# G( s7 F+ q/ d3 ?

复制代码

天云上人

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


///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
& M7 ^' J% [1 @
' D2 |( o# s$ k$ H7 N( l3 H6 \) ?
#pragma once
#include <exception>
/ X' G6 v) `$ X7 R1 N5 D

  enum TRISTATE{
        TRIS_FALSE,
: O2 j) A! f+ t! k9 ]" ]1 u        TRIS_UNKNOWN,
2 j1 N! w7 f; Z        TRIS_TRUE
. l8 x( g2 C0 K7 ?};
' [3 I5 g) v6 e: O

1 B! |: t& v" w! J: Q1 ~% c6 ?% Xenum UPNP_IMPLEMENTATION{
1 v% `5 p" s/ _4 N4 U9 ], h, s        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
9 ~' W; O5 A& B$ [* |};
8 {1 w% H, _8 @
4 ]2 w2 ~4 E5 I+ |/ d0 {" D
* S9 Z. B5 M6 d* J: Y  ~

class CUPnPImpl
{
public:
        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
        };

& Y: b+ V5 Y( l8 B8 r6 b1 J
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
7 d+ Y! L( f8 c8 K* i! u3 b3 I# @* W        virtual void        StopAsyncFind() = 0;
* a# P) `/ @* c- O  I$ z        virtual void        DeletePorts() = 0;
# S" a! R9 J. Q1 F) _. z1 J        virtual bool        IsReady() = 0;
0 M0 Z2 Y7 Z/ p/ w$ k        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
$ b  f. E7 Q, T+ w# w
2 t$ a) ~( q* ?; R
, T3 z# N# `/ s0 |3 |4 ?* k* ?        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
' T0 `# c" b( F5 Y3 n  ?; k        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


9 W3 W, g1 f* t# c// Implementation
$ t% A! |% K" s. Xprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
% b- d6 ]& u9 }9 {  X. y        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
5 j( L. c6 r( ^! \( m: {; W( m

1 _# {- z6 u9 W( S* `4 J6 Xprivate:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;

/ M: }/ n1 Y1 L3 @% h" ?! q- N
};
& g' Z7 l/ L1 B. k9 ?# I2 H! H0 f

' Y* h+ [# U* r- l+ S// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
% c2 F. K& L" r5 W, v* G% \{
" T) i# x3 W6 ]& t  S6 D9 cpublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
: W( Z7 \; `, M9 b2 H* h        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
+ Q% _! N- g* C! W        virtual bool        IsReady()                                                                                                { return false; }
! j- W  h: r  E% `! a) y        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
8 w% e$ q3 `) ?) J};
% Q* i$ s6 E2 L' z7 Q& S, A9 \

9 A; n' D$ H, t- V: B( Y0 Z6 g# `/////////////////////////////////////
  \4 ~" s# l  ~0 }  M9 W//下面是使用windows操作系统自带的UPNP功能的子类

- b$ H# \, e" w2 y# j6 J6 w7 x6 {) b
1 O# X% ~- T% Y, y#pragma once
* K3 Q, q5 ]% ^#pragma warning( disable: 4355 )
. D# D- i3 {  H4 n2 ~$ c+ P

#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
! K, z  S9 `( s2 Z- n; i% ]7 a#include <comdef.h>
/ o% o/ j! K# D#include <winsvc.h>

2 g* K& `; n1 G  ~4 k- V( H1 F% q
#include <vector>
#include <exception>
#include <functional>
9 m0 D: z; c" x! Z1 e



typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
- j! l4 Z+ _4 s0 a" \" y9 l( m0 d$ P7 etypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
3 O- Z. a- I& C7 dtypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
! ^& Z; T3 R* B- ?9 p& g8 Q/ ?typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
% f  I- `! i; Jtypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

  E+ p5 F5 V# H# F9 w4 |
1 {4 Y# L% u6 P$ gtypedef DWORD (WINAPI* TGetBestInterface) (
% c4 a6 c; D8 {% C! \  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
8 K. R  V. b8 {0 c( j" q);

/ c! A, v2 Y+ r2 C# h9 z4 M
typedef DWORD (WINAPI* TGetIpAddrTable) (
, T4 V; p' a; D6 s& L) v  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
);
" |' Z, t9 F( S4 Y

typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
. P% v7 `* q$ ]' B$ t( `2 Q);

% ?8 H9 w9 H- u! ~3 s; V5 N
CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);

9 Z( [% g2 `; o
) B2 s% r2 ?4 s- A# p9 P1 Rclass CUPnPImplWinServ: public CUPnPImpl
{
8 q# [! k& t! p: [5 M/ O+ V; z% b3 P+ s        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
% J. F2 F/ w# @" l; }// Construction
# t" Y5 s/ @8 |& mpublic:
        virtual ~CUPnPImplWinServ();
9 A! S+ {$ X3 F4 j0 c7 S        CUPnPImplWinServ();


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


! {8 E! U& }% w$ i+ \        // 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
        virtual bool        CheckAndRefresh()                                                                                { return false; };
  B/ c9 Y. i7 t, @( C$ _. G
$ ]1 b, {& G5 Q6 N& C1 ^1 j8 A
# w. h4 E+ z2 ~. Q* Y* E# Y" Uprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
5 u  ?) [& i# W: @        void        RemoveDevice(CComBSTR bsUDN);
9 x# F' z  g2 N- v# @( u        bool        OnSearchComplete();
' B: B" r1 M9 M; h9 ]        void        Init();

! z2 K8 o9 U5 i! }1 ^4 a
/ Y* t) ]+ @5 t) s% B. J        inline bool IsAsyncFindRunning()
. U% ?' A0 B1 \& N& P" A        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
! f' ?3 i8 i& t! Y, _& K" l- P' w                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
/ L6 y5 x8 b* E                        m_bAsyncFindRunning = false;
5 d" v* ~6 a! Q. X; b                }
1 a# G. q9 t4 @                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
; r+ A% q& S% f& s& Q/ H; w' G                        TranslateMessage( &msg );
; n3 J! O9 [% g3 k& B+ J/ H                        DispatchMessage( &msg );
                }
% u: Y8 O$ M' }- M3 j                return m_bAsyncFindRunning;
0 I7 }5 V* n/ {, |# ^        }

  A1 f! H/ G* F# v) |
        TRISTATE                        m_bUPnPDeviceConnected;


, f, a) l# H1 }- f6 h// Implementation
. Q1 @/ @9 d$ Z9 a7 h1 ~% `        // API functions
8 q4 Q9 V+ B% o) g, j* f        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
4 W+ L& ~  B. b# ~: ~, B        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
6 [+ ?; t4 @" B! [9 u6 N        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
% [2 ]0 |; X; l2 ~- t8 d: z- A. B        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
& Y- T: }8 V- [" ~% d& _0 X
% R' e- }( A% x9 V
        TGetBestInterface                m_pfGetBestInterface;
; _& L" I- S- t5 G/ W        TGetIpAddrTable                        m_pfGetIpAddrTable;
& z$ j( d: U" H. b6 D- _1 L        TGetIfEntry                                m_pfGetIfEntry;

, K7 M' [) i+ a8 m9 I  |
        static FinderPointer CreateFinderInstance();
0 y* }& v  R: C' X& v* Y6 \) R        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
: G8 F8 X4 _8 q- T8 ^% E, B                result_type operator()(argument_type device) const
7 R! U. D+ m3 }! @+ T. H                {
                        CComBSTR deviceName;
) \( {' Y! q8 ?6 P" c, @% B0 D) C                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
1 t  f; x5 c! ^
2 m0 u  C8 l* u% p
, ]5 ~- a1 E6 l2 Y& J                        if ( FAILED( hr ) )
+ K: B5 }: ]( a2 M/ ?4 o% W1 R                                return UPnPMessage( hr ), false;

: F% K1 g0 x! b$ L$ m
) f% t7 g0 S9 C  R/ o7 @                        return wcscmp( deviceName.m_str, m_udn ) == 0;
) X/ p8 l' s2 C) _* T                }
$ U" I* j" \) \$ J2 J4 F                CComBSTR m_udn;
) i0 E$ K/ g5 S% z8 z9 c' N- X7 _5 q        };
: t9 x' X$ ~6 ^% Y; z1 J       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
1 i$ j+ v2 o9 D. i) o        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
5 S4 I0 q+ q$ ~/ p- `        void        DeleteExistingPortMappings(ServicePointer pService);
7 Q5 C- y. Y3 B% H3 C        void        CreatePortMappings(ServicePointer pService);
! i7 i& Y6 X+ E, C" ]        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
* |0 G1 {2 x  G# J$ v$ a1 Q- A/ [                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();

) V+ H, g# b: |( ~. d/ x$ Q
; C9 n# l) U$ A        // Utility functions
        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);
        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);
4 c: t* c8 M# S; L* |* ]4 |) D

// Private members
private:
4 B0 q8 b+ Y# c        DWORD        m_tLastEvent;        // When the last event was received?
3 v) u5 w/ ]* ?3 j( ~% H5 {4 E& T        std::vector< DevicePointer >  m_pDevices;
0 p/ ^- j5 X: m3 u        std::vector< ServicePointer > m_pServices;
+ k$ e$ L/ C4 P: a$ v        FinderPointer                        m_pDeviceFinder;
5 j5 {4 Z& Y4 J        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;


- t2 \& S5 X% l2 r* w8 I; y8 K        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
0 e, |/ r' }" Q/ q        CString m_sLocalIP;
        CString m_sExternalIP;
" I. o. v* u, ?7 B& P5 j9 h        bool        m_bADSL;                // Is the device ADSL?
1 {  n3 f& |6 x7 X$ D        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
5 `2 S+ V! x, q, `3 b: F3 a! ~        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
, m* s. [, e5 G1 w; u        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
* G- J: A% W% @+ ]* h        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;

- \- G  T; P# L6 ~
};


% P+ @% T( j5 v% Y- y// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
1 l0 K5 j7 ]& U* S{
public:
7 R4 o" R$ C; x! G% A        CDeviceFinderCallback(CUPnPImplWinServ& instance)
. O* P) q8 J8 j0 E' Z  T                : m_instance( instance )
7 a) c5 b7 x5 J) P0 n- ]. ~) [: h        { m_lRefCount = 0; }
1 m. Z# Z9 z  v& Q% F: I

   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
6 b+ p& }3 X# t, n5 q   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

; @- w, C% [8 Y2 U2 j* o
$ \# y1 u$ f6 {6 i$ j// implementation
5 `) \  c8 [( Q( r. {2 I# @private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
0 C% J3 P' d( \5 `: G) @3 T6 \/ Z& @        HRESULT __stdcall SearchComplete(LONG nFindData);
# ]% f, }" _0 r0 I5 X0 C# W

: ]1 Y+ @5 U) J8 Hprivate:
% ~  b" u7 \$ y. P; ^6 p: b        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
5 \) X; s4 `" b$ G' i* T. y3 A5 O};
1 ~8 ^" U0 C% t( L
" J+ H, t6 h5 U. ?& N* ]; ?
// Service Callback
$ N: U: {: o$ Y# qclass CServiceCallback
        : public IUPnPServiceCallback
! ^* P7 h3 H! R$ V{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
; p- }- M* d. X! B   
7 I4 L2 p2 D! R5 `. y1 b   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
1 j. j9 R. l1 ~: F   STDMETHODIMP_(ULONG) Release();
/ Z1 N) }# z! U
* O' }: Z* G; ~8 K9 M1 E
// implementation
* Y9 ]  x7 }! Xprivate:
' }1 r2 l5 C' D" d/ [! H9 r& a- D        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
+ k8 [1 K& J* ]- ~9 n2 B

private:
7 o' `5 @6 p6 \+ a  u2 I# ~  y        CUPnPImplWinServ& m_instance;
$ _& V( v3 \$ s2 G" {: n        LONG m_lRefCount;
};

8 S; o4 F% A; Z8 }. U
1 G0 F5 M# v: c' V; t8 J) a/////////////////////////////////////////////////


9 X9 e+ }( V# a4 t5 p7 Y; |  X使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
: A% d2 x# c0 W3 z' M9 eCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
& ^  d& u8 r1 p3 mCUPnPImpl::StopAsyncFind停止设备查找.
& G) K  `) N! c, w% D% v: R- H8 y! lCUPnPImpl::DeletePorts删除端口映射.