UPnP

天云上人

/*uPnP.h*/

1. 
   1 t- |1 Z6 k. W3 l0 I( l4 H
2. #ifndef   MYUPNP_H_
   
3. 
   , z0 o& X3 j2 k9 {! a/ c' [" v
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   ' a  f& v* h- L+ i
7. 
   1 s& Q+ F0 s' Q1 a/ l
8. class   MyUPnP
   
9. {
   4 F* B) S9 U# ~  o, i. w
10. public:
    
11. typedef   enum{
    ) R2 I( \. v7 U
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
    9 Y$ {+ E5 e7 N! ~' s5 Z/ N
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;
    - \+ C0 a5 |1 V6 ~3 J; }
18. 
    ( P5 w2 F% u8 L4 O& I4 E$ B) w
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    3 d- v9 q* T! T$ T! s; T8 e
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
      U5 j# k4 x/ V# [( F
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
    5 _3 ]- \0 Q0 M, v  K' j, m" M; H* J
29. }   UPNPNAT_MAPPING;
      o2 n) g) R+ {% e& P
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    8 H  A/ P- H( ]- d$ E1 }8 @
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    1 Z5 U, I; T6 Z0 T$ w
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    3 E. h- B0 ]" Y$ a
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    3 u" E, p* |1 Y5 O0 m5 J; S* ?8 E
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    . L& f# {% c/ Q* G
43. protected:
    8 ~* ?' J* G  G, E! A
44. void InitLocalIP();
    2 T+ ?; X" d) j8 t  j
45. void SetLastError(CString   error);
    + s% c: u5 s+ n. p/ f0 k
46. 
    1 W, o& v6 b" p+ z$ ^0 ?
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    ; I# n8 K( n+ Y! X" P
50. 
    $ N$ g; d% D+ F. e' i
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
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);
    
57. 
    % K; L+ h, A3 \: I$ i$ S
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    & l1 `5 b! P. ]. Y
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    1 Z9 T& q: U+ d5 F
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    & Y' w- X3 }8 S1 Y' Y8 l' a; `
66. CString m_modelname;
    
67. int m_version;
    + \: S8 J2 L& {/ Y9 R$ l. g7 i8 P" R; [. Z
68. 
    9 s+ G: Q4 x5 G
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    ! I  E( s; V5 Z, ~" k
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    8 e8 J+ j" L3 j* B
74. WORD m_uLocalIP;
    0 W. `9 Z# N( P/ b" t. f6 x. `( s% k1 J
75. 
    7 v8 z, U( S1 C  }6 ^
76. bool isSearched;
    
77. };
    2 |% |; ]" _7 R( b
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   0 n6 M; e& G' H% o
2. #include   "stdafx.h "
   
3. 
   + ~: ^' ^0 o* E
4. #include   "upnp.h "
   
5. 
   " G' j2 s% V% S5 n7 z; O5 L9 W
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   # f+ v+ Y# D0 {" ]7 X" h: l$ q% Y% d
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    7 Y5 K% T0 P1 @. r: J. o
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    ; Z: z7 H$ [; a1 j# @2 o! L8 _
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    4 X6 H' O* `1 B$ K& q- [& I
21. 
    ! h/ P2 d- g" r' w2 u6 Y
22. return   s;
    
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    1 ~& F. D$ O  K: }
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
      {* L9 R: B, j4 u) P& U% O# s
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    " g4 y4 g" z( j1 w, u5 _' H
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    , t+ @) r  C2 J  t
37. char   buffer[10240];
    
38. 
    $ h) O/ S$ z& E. d- ~1 g0 v
39. const   CStringA   sa(request);
    * N: R* X! j+ e- v# u
40. int   length   =   sa.GetLength();
    $ w: ]7 j6 T; v0 [
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    $ J& b7 F0 P5 J# g
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    ! U% D8 Z. I: G3 j! C) V
46. sockaddr.sin_family   =   AF_INET;
    " k3 ]: F7 ?( S% j$ w8 D
47. sockaddr.sin_port   =   htons(port);
    , C2 x2 p# F* _! L, W
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    . z* S! s# Y2 v  T9 c
50. u_long   lv   =   1;
    ; S4 b) |+ Y% L) H& R
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    $ r0 P0 l8 s( a5 _, F
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;
    # l1 a7 G8 O  H( H
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    * }1 p: g7 p& M" O0 d  M! q' Y! R
62. 
    ' D2 D% @$ R0 T1 g
63. return   true;
    8 ]) p# a( x, N  y
64. }
    
65. 
    , s0 N1 E; I4 s! N+ m/ s# G# Y; @
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    % V. d" n: a; N/ V7 T! q) F, m) x
67. {
    
68. char   buffer[10240];
    # \( r8 x5 C. Y
69. 
    ; v( v8 i8 ]. K+ m
70. const   CStringA   sa(request);
    3 g4 D; m, x7 y8 O
71. int   length   =   sa.GetLength();
    * l% _9 ^8 m) z* a  e
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    & d( M- z4 ~( D1 ^: S( L% B. U! z
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    " B& I7 n8 Y$ y+ K/ Q2 l2 H7 `9 p
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    ) y0 a, l% X4 u" }" [  B
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    9 W2 V" N$ N! k9 f& _1 a
89. result   =   response;
    ' Q3 K# Z/ N) Z0 E) l
90. result.Delete(0,   pos);
    $ _* o7 n. p" C  o% ?5 A8 L0 y
91. 
    4 q; q; Z6 w, X
92. pos   =   0;
    ) i0 [/ |& j, h( b" {5 k
93. status.Tokenize(_T( "   "),   pos);
    * ~" O7 M$ Y% r- [% @
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    + E" B' z8 J' x- J  z
98. 
    $ a: o0 k# U! ^& W& c
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    ! V6 J3 i3 _. a6 @8 \  P: z# i
100. {
     & @$ f8 E5 j% X" a8 ^% k: B
101. CString   startTag   =   ' < '   +   name   +   '> ';
     ) b2 r3 B( F  {" l( H2 o+ V7 r( }
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     ' k+ X3 u+ g# d3 C0 Y% ]+ x1 `. G
103. CString   property;
     $ v7 Y& j2 M& G- W
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     % K( G* B7 v& [5 `
107. 
     $ Y. t) F7 r# b3 n! r
108. int   posEnd   =   all.Find(endTag,   posStart);
     : P1 Y4 n! {5 R8 X
109. if   (posStart> =posEnd)   return   CString();
     0 v" Q+ w# ?2 C$ J+ c0 A/ p
110. 
     % A1 e* p9 j# x- X
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     * Q" N7 t0 f, N; C
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     6 a1 o" Y8 G3 E% d1 B4 ]0 s
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     
120. 
     6 J0 `/ Q0 @9 _# i
121. MyUPnP::~MyUPnP()
     
122. {
     6 q% F( |! F! l" s9 \+ X
123. UPNPNAT_MAPPING   search;
     / Y  s5 f# h$ L/ P
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     $ |* s5 k& w5 _9 r) R
128. }
     4 }3 F5 K4 }" u: y: s. }& F- t; q3 T
129. 
     , k+ [0 m5 a2 p6 g. y" U
130. m_Mappings.RemoveAll();
     
131. }
     ' Q% W" ~- `3 l3 ]7 Y$ Y
132. 
     & W3 F' C- t6 \; P! T9 w
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     1 A" C4 s8 a9 n* z2 r  x& f
140. CString   devices[][2]   =   {
       V! i& x4 a8 R% }4 D/ l
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     4 o, b* Q/ }" t% y/ A/ E& o: ~
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     8 x  x6 j! F+ S/ `# C( |) V
148. ioctlsocket(s,   FIONBIO,   &lv);
     : T. F& r9 x6 l  f
149. 
     + C" _) ?8 U, n# T  u8 E, w
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     - a9 \! h  F- Y1 e, P
152. if   (!(i%100))   {
     " R7 G+ `. o8 O7 I: o- O
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 "),
     
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     
161. 
     
162. Sleep(10);
       ^0 R+ @7 U  H& f
163. 
     5 u- T1 V. {9 e1 X2 y. ?
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     ( h* d! u& s: ~4 g( d
167. closesocket(s);
     - q1 U% D4 H& M$ n
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     # c- Y9 B( ]2 P, t. s6 R7 O- n
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);
     8 Q, T2 E" H( s# H0 f
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     # ^: s* W: b9 {7 u
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);
     
183. m_description   =   line;
     + h9 j4 ~$ V4 k4 M0 Q& j- o  q/ I
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     ! X8 ~9 v: f! ^6 ^" Q
187. }
     
188. }
     
189. }
     
190. }
     
191. closesocket(s);
     
192. 
     
193. return   false;
       L+ X  \' H8 l. G5 R. b( {! k8 y# D
194. }
     

复制代码

天云上人

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


7 W) ^$ t; @2 [3 V& j" I///////////////////////////////////////////
+ K3 K  c0 U0 o//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

4 l, R* P5 i, Q* B8 j
( |+ X+ E' [! G#pragma once
#include <exception>
+ d  C: P! l/ }2 Y9 q) z
3 ~  I* b" y9 O  k
$ i; `2 {) ~1 b+ I' r' J0 `  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
1 h' k' l) D- {+ J0 f2 x        TRIS_TRUE
! E$ ]# }) `/ c5 z% R( m, E0 ~};

$ W. y; u% n$ D$ P1 r
" }# g( u' B0 J$ D7 M  B7 ~enum UPNP_IMPLEMENTATION{
: l, E# Y6 R! T( W        UPNP_IMPL_WINDOWSERVICE = 0,
. |0 M# k% P: U        UPNP_IMPL_MINIUPNPLIB,
) |* S  i8 Q! H        UPNP_IMPL_NONE /*last*/
  q. Z& W' {, B& ?3 h};


4 G% n$ O& X& ?
6 n! h- U0 f% M
class CUPnPImpl
$ ]8 ?; S9 o$ n$ B' B' `4 k- X$ o{
$ @; E- w7 Q# {, o: c+ X" K: q$ hpublic:
        CUPnPImpl();
        virtual ~CUPnPImpl();
1 W' g0 I% s/ D( T        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
3 z) {: U  A, M- [        };


3 u* S# Z& s" g/ l" I: ^( Z- k        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
- w- V/ P1 @0 O. w, G, J        virtual bool        CheckAndRefresh() = 0;
, I! d! ?- j3 }, b1 u        virtual void        StopAsyncFind() = 0;
6 }, h6 D, p4 \" k        virtual void        DeletePorts() = 0;
: K9 @  h) g  X7 X% A1 |9 u! D; ^        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
1 @% p8 `7 ~2 A' e

. [! V3 D, C5 k9 U) p* c1 T        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
; G+ p0 c7 Z1 |9 x, y$ N        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
) b0 l. D! G+ |$ G
7 ]' p" c' L+ D2 o! j0 I$ }$ u
( N0 ?# R+ a2 F! W( [" j5 {7 y# S// Implementation
" N3 R% E! A: z1 B% h9 Qprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
7 h' e- X$ |# Y! d; O: U        uint16                                m_nTCPWebPort;
  A) w* e9 g2 ~5 q: o3 c        bool                                m_bCheckAndRefresh;

+ c4 F  A  o3 \3 x4 t! I. K; Y5 Z; h& c8 v
, i2 x+ Y; V4 f, }" E, `( Hprivate:
; Z. z% w2 D) L  C% [  Z        HWND        m_hResultMessageWindow;
9 E: U8 |* _( E9 g8 V        UINT        m_nResultMessageID;

$ c: w* e8 a% _3 j3 p# r
1 h6 t8 t+ b/ h# K8 H% T};
1 V. D4 p" z9 ~/ W9 h0 D
5 i9 d9 G4 U& {5 b% o* g
// Dummy Implementation to be used when no other implementation is available
; e  U- q& w6 W# D) G' V( G: Wclass CUPnPImplNone: public CUPnPImpl
{
public:
$ {5 i: m2 K( a8 M0 i        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
. o7 F* |& V* H. V. R        virtual bool        CheckAndRefresh()                                                                                { return false; }
; e$ X3 t1 W2 j8 C# [- @. u        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
  _8 n% R0 s) X( j2 h, N        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};

5 m+ z2 `# g" X' k) z
/////////////////////////////////////
8 z1 i3 n! B: w% h6 L//下面是使用windows操作系统自带的UPNP功能的子类
3 L" [' _! G$ U( t$ e5 ~
/ k! ~1 j$ ?) E4 Z, r( d
#pragma once
# w- F) [6 o; i, E% U- o9 K* U#pragma warning( disable: 4355 )


6 y! j( L7 a' n* n6 u#include "UPnPImpl.h"
$ U/ m( `: t! W! g; E#include <upnp.h>
#include <iphlpapi.h>
8 o. q) _4 `9 i) k8 J#include <comdef.h>
4 B0 N9 |! d  l5 s0 c: X( C# P#include <winsvc.h>
+ @& }- C' M0 e+ I0 X9 {
4 T& A3 y3 A8 j" r  B% e
, D' w! _7 R$ z# B# u#include <vector>
#include <exception>
#include <functional>


- l0 O  f% B. S! v

: Y  ]# B2 ?  r, S' H( t# Z, P: o  I4 Stypedef _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;
* P' R" d* M7 b6 K/ \9 _typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
' s, V1 p9 W0 D$ g2 Stypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
$ `# T5 |" X, F3 Jtypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
  v9 w$ Z2 T- k

typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
! ~. P& ^6 d; l7 }, I  PDWORD pdwBestIfIndex
' ^2 U. ]8 O3 I);
* b5 q0 H* [# L4 b/ v0 `
6 C$ }, ^- w- p" R8 V% z
: e# ^  v. V6 X, a/ J/ Ctypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
* J. g, U; F- N9 o; X' E  PULONG pdwSize,
' v+ d. C6 B4 O9 @% E  BOOL bOrder
);
# i+ U2 z6 B# F- c0 `

typedef DWORD (WINAPI* TGetIfEntry) (
5 j1 Z, \( f4 _0 W: A& t  `  PMIB_IFROW pIfRow
);

& ^' p7 e% a% [
CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);

$ G% v% X7 U5 j+ Q: R/ U: F5 z
class CUPnPImplWinServ: public CUPnPImpl
{
; e3 a& t& F) u5 o  H$ d7 c/ v        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
public:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
1 r. Z: S/ ]% s' G& M: o3 T
2 I5 x; y, m" i& s  C/ y8 P: u4 \* v
$ Z. d9 _& X  f/ ~# r% u  w$ |        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
7 P6 U0 W6 E6 X8 Q5 A6 L        virtual void        StopAsyncFind();
4 A0 P1 W9 R# L7 i* v" J        virtual void        DeletePorts();
! \/ L( L6 B" S; f        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
9 ^- j! i$ j( _9 C
7 [# I, {) _" J/ T
; p$ j$ x* D* T        // 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
' z! L* {. q3 `" I( n        virtual bool        CheckAndRefresh()                                                                                { return false; };
' c6 Q& t& z' K0 N+ }, E4 A

protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
/ [; c" d% p& |& O' g/ n        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
5 B) C0 A1 H7 b! Y; i1 |        bool        OnSearchComplete();
9 `8 e5 K  m! S" A8 Z* Y        void        Init();


        inline bool IsAsyncFindRunning()
! H6 S# k7 W- a        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
* ~6 J" W$ K. W  o                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
  K0 \9 s. K$ c1 V( P3 D6 R+ \                }
                return m_bAsyncFindRunning;
        }

$ y- q% G3 u, o! `
7 z5 z7 K2 X3 ?9 u, H: l+ \6 h        TRISTATE                        m_bUPnPDeviceConnected;

1 b. N* t1 f4 X
6 \( {, i: M; j/ r// Implementation
; h$ @  S" H: ~& X% v        // API functions
' c% C5 b( y' ]" z        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
( c. z6 d* g1 D0 Y: c+ V/ }$ E+ z        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
% U7 _' {- @' l: V" \2 w        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
3 a+ p; X2 c+ G) f: ]6 @        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
2 N! |/ D: y* |" ^# b9 i3 w  G' C

. j5 @, B' S9 ]1 ?6 H) T2 b        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;


( _% B7 ]( x+ v: Z/ P7 ?        static FinderPointer CreateFinderInstance();
- [6 V- ~: F+ v        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
( @1 ]/ @7 r* Q! w5 D: m- }5 w                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
, O( s: A' i( q0 B2 ]5 o                result_type operator()(argument_type device) const
6 W' J% j! A- j. r! z: d+ j                {
0 k8 s; P& x+ L$ N) y/ a  b                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

' g/ q& j% L2 {. |& O
                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
2 R& E/ m  a; e& a9 I3 l, l

% T0 y& J! f1 `6 A) j0 ~                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
0 s* f- c' S( Q" @! u6 v3 s: e' e# E2 A        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
. b$ p  y, _0 a0 k: B) @        HRESULT        GetDeviceServices(DevicePointer pDevice);
  T) D  w" V% }( C        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
" X. k) {/ g# E        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
; ^" g$ Q6 P4 l& K9 V+ q        void        StopUPnPService();


6 T" j: L* U4 x1 q) V        // Utility functions
/ Q1 D" R: a( t9 l4 c        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
: O* n5 @- d! ]% |2 @9 m        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
$ H& M6 ^/ Q0 L7 N, `7 c3 V1 V  u        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
# _  i! @4 u) M  [$ {/ i- C. J* o        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
5 F0 Q6 z9 w1 ]1 G, n
; N4 S0 j0 |4 \
// Private members
private:
4 x( O5 l) k1 @3 J$ ~$ ^0 G$ j2 R        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
( i/ v* ]8 n4 n: n3 D' E        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
, l+ \. k% ~1 {7 s

        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
( T  V8 x$ s0 x% A- F' M        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
6 L, u' H: X, j" x, Y  [        bool        m_bAsyncFindRunning;
( W  X) L4 s- t        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
* S4 p$ G: f5 f. p8 ]- C& G* X* @        bool        m_bSecondTry;
9 Y" ~) q, x4 K8 ~6 z9 k3 y        bool        m_bServiceStartedByEmule;
* n( |% u! g' {! G% X        bool        m_bDisableWANIPSetup;
$ a7 f7 l1 S- s) J        bool        m_bDisableWANPPPSetup;
/ Z. l. E5 d' \# J+ R+ E
* Z# [. p5 S5 D, x2 h
};
' W5 h( \: v- D* O

' H; l5 m5 i2 L9 \// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
0 p* J+ ^9 e7 s' X9 l7 T{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
1 t0 B7 }0 V4 O2 }  Q                : m_instance( instance )
  m$ z3 v7 R6 {% W        { m_lRefCount = 0; }


: ]; m% `) t6 s   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
6 R2 p% J! T9 Y& f. h. p   STDMETHODIMP_(ULONG) AddRef();
5 N9 W* p9 j5 u8 U0 [% H3 R   STDMETHODIMP_(ULONG) Release();

" p- a( ?5 u* `3 I% z& l
// implementation
private:
3 ?5 g+ I  L3 q; g' f4 k        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
# L/ [( p( J4 o0 v. D7 t! Y        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
7 x; L7 Q, O% e1 K8 ]! h( U        HRESULT __stdcall SearchComplete(LONG nFindData);
. [" {1 G* F9 }  H8 l; }1 ]/ ~. s: h

3 }) W3 Q  ~/ L& A8 P8 |5 cprivate:
5 l" T& \) P1 ~        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};

$ L9 u3 s1 n( _/ G4 q
& [% g5 |3 I2 b! j9 I0 a% ^// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
{
  R# i- f9 O2 spublic:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
; d& H+ P2 }, H& j% s) y& E, B9 O/ C   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
0 ~# [( R% x/ y- u) Q. l/ M8 f5 W9 U   STDMETHODIMP_(ULONG) AddRef();
* S2 _; P7 z* ?+ u   STDMETHODIMP_(ULONG) Release();
# _1 b4 l: l4 c# h

2 A: T& v; M4 f5 Q2 X$ G0 R// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

/ ], N8 ?) a6 B+ t, S7 y
private:
3 s, l4 V1 |" X! I7 v- ?        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
9 U% ?& S1 j4 \: W. `$ g+ ?8 O
: j2 D2 G4 {% w
: h0 H: p1 g; s* D% X/////////////////////////////////////////////////
6 E1 O$ Z4 N& V. @7 O9 p, {5 e
7 N* t  m( l0 L  r" d8 R' z$ H- J5 p6 t
使用时只需要使用抽象类的接口。
) g/ X. D% S3 zCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
( f+ c1 g$ a6 @CUPnPImpl::DeletePorts删除端口映射.