UPnP

天云上人

/*uPnP.h*/

1. 
   8 E0 g& t0 _  u8 H
2. #ifndef   MYUPNP_H_
   
3. 
   : V6 w1 g+ G2 l& P( o2 `- D0 @
4. #pragma   once
   
5. 
     O. g/ h# U' o. c$ B: ]0 |* |
6. typedef   unsigned   long   ulong;
   : A% d; f4 ?, P7 J* K6 C: }  U
7. 
   , d) f/ i2 ^; D
8. class   MyUPnP
   1 O# [2 t: p; w# ]0 u- y
9. {
   # e. s3 v& A* J0 X2 B% _3 ]& I
10. public:
    ; C  d$ |- ^7 u/ Z; |, E" m
11. typedef   enum{
    ' |: f% Z3 `. Y* o& R( A
12. UNAT_OK, //   Successfull
    # l: i4 m. y  g9 Y, {
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    0 @3 v0 {7 J( k# k: ]2 T
19. typedef   enum{
    4 g9 C6 e; t% R0 F* U
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    4 N* k# ~+ [1 j
23. 
    5 o7 E2 h$ [" G. X4 |
24. typedef   struct{
    & V! A  \- J3 M$ T6 C- I: h
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
      c$ k* t* ]( X) t
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();
    & Y! O6 Q7 C! g& R
32. ~MyUPnP();
    
33. 
    . ~( |9 w% i% s  C6 I9 n' k
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    ( `8 H2 Z+ ?) |: c3 s( B
37. 
    2 l- L3 d0 c. E6 Q
38. CString GetLastError();
    & z3 ]4 O/ O# J- v$ ?2 e
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    " Z, S" V+ d. m: w5 c: u" ]4 E- i* }5 [
42. 
    
43. protected:
    - v# y3 `. A" r8 E
44. void InitLocalIP();
    " s3 r6 i2 p) p" E
45. void SetLastError(CString   error);
    
46. 
    
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);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    ( X6 U3 `9 _& C" |
52. 
    
53. bool Search(int   version=1);
    , I5 W, L& Z, @3 I+ j) P. b
54. bool GetDescription();
    6 z# B7 J- J, s; b6 t% [6 b# g8 w
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    8 S7 H; N) L- V) ^/ u& c
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    4 q+ R+ h" t0 s
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    + E) y& O8 b. _
61. CString m_name;
    + z3 M" E( x. P; w' i5 M% \, X
62. CString m_description;
    . s7 j( \0 H9 o9 u
63. CString m_baseurl;
    5 k$ A7 s7 A: M
64. CString m_controlurl;
    
65. CString m_friendlyname;
    # i  b2 X4 i! S2 ~; N
66. CString m_modelname;
    
67. int m_version;
    * Q, W/ n* c5 H' g! l. H
68. 
    
69. private:
    6 [0 [, h( A: L  j. M
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    1 {! M1 W6 n9 d) s
71. 
    
72. CString m_slocalIP;
    % j0 S2 c8 G) m  w9 P1 T
73. CString m_slastError;
    , e) {& H2 h2 q  U: P
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    0 R5 I% W: U& B% I" X
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   * \, W$ v" Y* H/ Y: S6 S+ k% c% g
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   - P" H) P- ?8 Z$ D* K# T
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    & |( u7 ?' O5 L8 q. z  d9 |5 X( v0 Y
11. 
    ( J. g$ P* v3 A
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    + ]5 _* f2 w6 R
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    ' n. U, Y( g* ^% q6 l9 ]
15. 
    % n* M( |8 m& A1 m1 {3 T" {
16. const   CString   getString(int   i)
    - ]4 n: q% O* H7 L7 s
17. {
    
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    6 V# ?, u# p4 D! Y
21. 
      E5 G- r6 \2 W& O9 D
22. return   s;
    
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    0 z, P% m7 O* f$ z; u2 n- ?
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    1 |8 A$ u5 m8 T% S" u
28. }
    
29. 
    9 N+ {- z! k! K6 o  i3 l
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    ; ^2 C9 s* Z# ~
33. }
    9 F9 E7 X3 i' L/ Q
34. 
    / C; J% K4 q# S% r5 x7 Z
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    , A3 D7 g4 w4 X- _
36. {
    % s9 u% X: t. h$ T# s- a( F" F
37. char   buffer[10240];
    , X. D  S/ G) k& E2 X
38. 
    * {; A$ O/ y% ^( T: h& g; l4 h
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    & n& X1 Q* w8 j) \
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    ! X" `4 Y9 ?  J" Q7 q! j" V4 o3 U6 Q
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    ; h7 B& ?4 S# J
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    & F& t. X! {0 O% T# b
50. u_long   lv   =   1;
    3 T' V: p/ F. d/ o
51. ioctlsocket(s,   FIONBIO,   &lv);
    8 K$ L" K& Z# |) B. v, Q1 y3 e
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    5 V" R+ W/ s% d! x0 g
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    ' W( B0 \/ }. b; a
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    $ f0 v( o$ n. U0 ^1 `
60. 
    # q0 i% s" j, i- U
61. response   =   CString(CStringA(buffer,   rlen));
    & {* N2 m2 r" }# q& b) r1 L0 A( w
62. 
    
63. return   true;
    
64. }
    
65. 
    2 _6 T9 l( J9 ]- V& Q
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    7 \* I) D. s1 T' H
67. {
    & c5 `; M5 x$ B6 r! S: v
68. char   buffer[10240];
    % R/ |$ U4 d( n" _: O5 a  H5 T' x# I
69. 
    ' l. w' ^. N6 P1 t* T
70. const   CStringA   sa(request);
    2 V0 X# Z, y5 S, {  l+ z% f
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    - N3 K$ x, T/ D2 V
74. struct   sockaddr_in   sockaddr;
    ( ~" e. o) L. c: x) D2 X1 I. g
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    : O: _" r$ f: J. ]- B
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
      i1 E% I" _/ N- }( o- T4 \
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    3 `# |! k' O* [* s6 F
81. }
    : |7 P9 Y( m8 `! |
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    ! g) \, X$ m( @. \( f, q
86. 
    9 P9 @9 [: _- m( a: t' V2 y" m
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    $ U2 N  M3 U; |9 q! M
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    * D  K+ N1 J. W8 Z( n
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    
98. 
    ( v, y6 k# V( I% c
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     4 u. q' g! y0 H& Q
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     ( w7 B% M& ~5 G4 n# e
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     / n/ |8 Q2 ?3 [0 L9 M- s
107. 
     0 M" W, d( e% c" c  Y' r* u$ F
108. int   posEnd   =   all.Find(endTag,   posStart);
     ! R- R) n+ U$ P* }
109. if   (posStart> =posEnd)   return   CString();
       V5 _$ K' Z9 C. i1 Y/ Z7 j! M
110. 
     & c0 {+ X8 w6 r. ~
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     2 `/ }9 p: V! p% L
113. 
     % E; ]9 q; m5 `
114. MyUPnP::MyUPnP()
     9 s* q5 n: e+ h8 D1 R- G% x
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     % H, H( Z3 m2 t- B7 y) }
118. isSearched   =   false;
     $ |) i* _; U4 i9 P* |; s6 p. [$ q+ |5 G
119. }
       k+ y5 L7 P6 R+ `
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     : x- E2 ~* g  H/ c" b: E# M
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     ! A8 x1 |! }5 K0 C4 t
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     % O3 f9 z! o8 b
127. RemoveNATPortMapping(search,   false);
     
128. }
     ! b# l: |/ v/ [: w( H
129. 
     3 t$ R# ~1 M, R8 D$ H8 Q% T3 t
130. m_Mappings.RemoveAll();
     
131. }
     ' n" }8 b; v: r/ q1 s; [  i
132. 
     / f* n/ ~9 r  s" W6 ^0 C# x+ n
133. 
     6 o4 `* k3 H  |- ]5 ]
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     . Q* F! S+ T: w* r" H
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     9 a( H' t6 k6 v% K& Y
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     # z8 L" T3 A4 X8 ?& b
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     " r4 x1 W: K! v! M" H: P- G
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     % p* J! ?# _' G% M
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     ' r' D4 q7 |  h
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     0 R* ?) F* D5 E# y& n& I+ f: K; E
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 q# i  t( `' C6 j! r2 o6 ?
157. 6,   m_name);
     ' D( _! W8 F/ u3 j  _" _
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     ) j" ?* m9 {( v5 ^2 r; w1 h/ f) z
160. }
     
161. 
     $ y3 e0 H1 v, s" k' |5 z. ?$ a
162. Sleep(10);
     # Z3 F9 e5 N4 D" E7 p4 G* G
163. 
     / b" R- L' [) C
164. char   buffer[10240];
     6 S# \. V' [& y7 D7 Q3 @
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     : U+ J3 {6 D) j" D- y( A
166. if   (rlen   <=   0)   continue;
     - Q- S  c2 v+ S
167. closesocket(s);
     
168. 
     8 u" N! k5 |& B5 J" r; p
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     ( E3 X2 @1 U9 b5 W# l
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)   {
     0 x  a0 o6 p# j* S
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
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;
     
184. m_description.Trim();
     " r& u% D6 E, [, f4 C, V( H  p
185. return   GetDescription();
     : ?* K. I* V! f0 p6 m# x. y
186. }
     
187. }
     ; X1 G, N; b7 k4 x: ~+ D) D
188. }
     # b( W. a3 ^( E- S: G. R
189. }
     
190. }
     + I( j% G3 i+ {' I
191. closesocket(s);
     
192. 
     
193. return   false;
     
194. }
     

复制代码

天云上人

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

) k  T; \: h! _( v9 H
///////////////////////////////////////////
& r6 t, I: G+ ~( h/ _5 `//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

) C0 ~) |/ p1 R
; m- o- X; d( y& x' a% o#pragma once
1 W2 V7 g( C; k/ g$ t#include <exception>
0 U& s( l' Q  i, P. X

  enum TRISTATE{
+ j: D2 o% H/ n1 ?        TRIS_FALSE,
6 m) t0 ]% ^! s5 F; m% A        TRIS_UNKNOWN,
3 Y- b2 |: E! `8 M  K) R        TRIS_TRUE
};


enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
) L  X4 ]! ?# R7 v8 M8 p};
8 n$ x( U! M9 E& b& m
+ E8 E( p9 v  i2 {# D( A; u3 `+ M
" q, O( g1 x4 a* x! L7 A0 ~, I1 r

class CUPnPImpl
{
public:
        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
. S0 ~9 M/ i% G7 @        enum {
                UPNP_OK,
! V7 @% h/ T4 R$ o: F                UPNP_FAILED,
                UPNP_TIMEOUT
+ S3 |. x( ^5 H: j) P& C        };


% k* }( I) n3 w3 {# v. D        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
( F3 L4 W* ?7 C4 M( @5 w        virtual bool        CheckAndRefresh() = 0;
% X. E. t/ G. x- B+ l- M3 W        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
  S1 f; k  X; {4 ?/ a' n        virtual bool        IsReady() = 0;
: k9 q) {8 h- P5 Q7 f( [        virtual int                GetImplementationID() = 0;
  M) o( [% \% G% C& x4 a       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
; F& m4 r) q9 o6 k) Y- q- K# @. |

        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

0 T" ?$ c6 D! a* @% _& w4 ~% H3 Y
// Implementation
& t$ u$ z# D. f6 @protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
$ f; o& C! ~& R' f" Y6 [+ z9 \        void                                SendResultMessage();
) {9 `) `- ~% p3 L& t; {- b        uint16                                m_nUDPPort;
# l6 q1 T) k# [        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
3 G; ]" t( G, h: Q" V        bool                                m_bCheckAndRefresh;

6 G9 m$ z& P$ N" F; n6 L' k4 i( X
private:
/ N  M) G( @9 z        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;


3 Y& I% ~% O& X8 K};


// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
& x$ Q  X! ?( Q! f- O' A3 D7 Npublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
1 B) p: l7 m: q) _+ k6 }. H' B        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
2 ?2 q! E  g& w& v, w        virtual void        DeletePorts()                                                                                        { }
" F" `% s: _. y0 ]        virtual bool        IsReady()                                                                                                { return false; }
" L0 u9 j% }! `# i& ]* P        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
0 K. k3 X) h; t0 C4 F: y3 n% e0 p};


; K3 s0 Z* `. ^8 U0 i/////////////////////////////////////
" h4 _* y' E& ]* m//下面是使用windows操作系统自带的UPNP功能的子类

3 L5 p9 W1 Y, Y, c: @' y2 e9 a
#pragma once
% I) d: Q6 r% K6 ^% i#pragma warning( disable: 4355 )
; p3 r+ Q. Z" R4 \, n" C  \9 R
9 Y2 w1 ?. P7 s% O
' U! s: V) N0 P+ S0 V) G; x#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
0 d# R' x. J, D+ M3 ~- `/ p+ B#include <comdef.h>
# r8 O* s& \  s, J4 x#include <winsvc.h>


" n6 Y. h' V8 |; T' r#include <vector>
#include <exception>
) G/ \  P& N* ~#include <functional>

5 n4 v% [' p! e/ h0 y' [

" U5 n$ r- H% P; t- k
) c/ M. M. H3 k( W  O# |typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
6 q8 T% _- ]1 K' r( Ptypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
: ]+ \- ^! L3 ]4 o3 ztypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

/ @- m  x  z& K7 \  B
' j3 Y/ q  u# a; `+ ]typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);

3 z% d3 c9 c: I1 i0 [
7 W1 Y6 j: M( q1 j; k6 [typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
) Q9 Q& o5 D, o% R  PULONG pdwSize,
  BOOL bOrder
);


) X& e) I# g( ~typedef DWORD (WINAPI* TGetIfEntry) (
' D2 p7 D3 c4 w  PMIB_IFROW pIfRow
7 c3 A/ h! _# B& Z# L);
/ e" ~' R/ s" g. P& B# [
, m& {1 {6 B$ ~! Z
1 p+ o6 Q5 z1 a1 k) X/ E0 dCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);

8 n% P9 @8 @9 S4 B( F/ x+ p8 V
class CUPnPImplWinServ: public CUPnPImpl
7 r! F1 q: c0 l' L{
        friend class CDeviceFinderCallback;
% O" w! W; l, f+ X! {        friend class CServiceCallback;
// Construction
+ U0 q1 g6 h: cpublic:
        virtual ~CUPnPImplWinServ();
( j: t/ N- M2 u  ]2 D% Q        CUPnPImplWinServ();
  v0 q: ]. A  n9 L2 c9 d

( {2 m0 z) |7 U2 Y: |2 z. O        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
/ d9 o# z1 G7 C* O        virtual void        DeletePorts();
+ ?$ B$ I/ x/ \/ n& c. o        virtual bool        IsReady();
5 J7 E8 T) Z7 O  L& P        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

: x7 S6 {* w1 r) ^! V! |
        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
$ ]! _9 o9 G: z, L! s" @5 i; l        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
$ j0 z5 ~1 h* J& q        virtual bool        CheckAndRefresh()                                                                                { return false; };


2 @$ N+ M& F+ O  |! Xprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
- d- X: v' m( V! i/ k, h        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
9 O. e' K( b/ n* \7 ~2 I- v1 f        bool        OnSearchComplete();
        void        Init();
$ G3 U6 v! K5 {. Y0 {. f2 w3 ]% y

: J6 b& d$ ?) `+ B; a. D        inline bool IsAsyncFindRunning()
        {
% p- u7 n6 p+ X! ]                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
+ o0 x! W3 k% O5 r. T                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
8 u& C/ ^8 q7 n$ m8 E9 f                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
  P% q% d5 r% b: u                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
+ v$ ]" G7 O0 X, g                }
                return m_bAsyncFindRunning;
        }
# n) j4 t+ s0 S6 M4 B3 g& P
# J9 x( F# W2 l! N) H0 o
        TRISTATE                        m_bUPnPDeviceConnected;


$ R8 K$ W, _" n6 h% g// Implementation
        // API functions
/ l0 C! \2 X8 ?. Y) k        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
  P0 e1 s5 ~6 P  [* w+ R        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
+ ~4 ~( P1 b: p3 v" X$ {/ a        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
! t0 ^$ b5 {8 b& t# S        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
; R, a1 B8 ?1 S        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


        TGetBestInterface                m_pfGetBestInterface;
( m, h( y4 ?: D8 t- }4 v9 C        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
7 }* {( z! ]0 W1 ]

        static FinderPointer CreateFinderInstance();
; h: w$ {) O3 |+ N+ E1 T% x. e3 v        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
0 i- h7 }5 _4 `, w$ S                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
/ X4 B- J4 g; I" V) H3 u                result_type operator()(argument_type device) const
                {
( i* g5 ^4 ?+ v& \7 P; e% J/ F                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

( {- o5 P. M9 t6 \
9 f. V4 g% C% K; y1 n9 m                        if ( FAILED( hr ) )
: l0 D* S$ o: ~  Y/ h% v                                return UPnPMessage( hr ), false;
6 R- a& T$ L4 W4 u( R
- J7 [# M7 v. |  J
3 }+ G( ^* \$ ]. Z9 p4 q                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
' E8 b1 O5 S+ G" E1 Z' M, B4 P                CComBSTR m_udn;
' |: `# @( j/ z8 f; p3 R        };
' j, @; m0 J9 V) m# w       
' T3 ?; p- d0 @# r. d$ D8 R/ A        void        ProcessAsyncFind(CComBSTR bsSearchType);
7 Q# y3 O' w9 R, K        HRESULT        GetDeviceServices(DevicePointer pDevice);
4 r% ^7 O; j; W3 y. S3 O/ q1 m8 g        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
# {0 |" U% b3 I- m5 w# _                LPCTSTR pszInArgString, CString& strResult);
- X! d, m- l; s        void        StopUPnPService();


- h7 s" Q+ B: h: X' W" B, w        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
: n, e+ R# ~% s6 m5 V9 W) P4 A        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
# S$ U! P6 n* Q$ P+ r/ L        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);
# G0 p6 Y! ~$ v4 ?# [
6 C; L9 [2 x- ]4 ?/ `* O: y
! n4 Y: t* r8 a# S2 N4 P$ R// Private members
- q% t8 g8 L1 U/ Aprivate:
+ C( c/ y( ?, J2 z1 c        DWORD        m_tLastEvent;        // When the last event was received?
3 b) ]! n: C* a4 J+ ?7 z        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
) p2 Q: U: R4 M. q        FinderPointer                        m_pDeviceFinder;
! o6 `# V( K, `) \3 U9 G) }        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
! X- m1 a! r1 m. {, a
/ y1 h/ l) M; F: |4 G
7 V7 r; q. r  o/ Z! J        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
& t) [3 t7 j4 L; j# M9 _        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
4 s! g4 r' ]  c% ~        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
& F$ q* E7 }1 f$ a        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;

4 m; |" }1 z1 P. a
9 A$ B! f" L* F8 W& h* z1 A$ `  q};


// DeviceFinder Callback
class CDeviceFinderCallback
; \6 j+ R3 n( _; v        : public IUPnPDeviceFinderCallback
# I+ b1 _  S5 m- ^3 |& `; W{
public:
( p2 \' K! A2 p        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
( }. v) U! I. x

   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
2 G3 E( b( W# N   STDMETHODIMP_(ULONG) Release();


// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
# P7 c% V# o* Y1 G        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
1 v+ K" B; a) {( Y% M# w# f8 V        HRESULT __stdcall SearchComplete(LONG nFindData);
$ \0 W$ @' C% y( l+ B

private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};

3 m* }- @8 Q% C
// Service Callback
+ X' i0 b9 l+ x% h0 Kclass CServiceCallback
( G& |& q( U7 w5 D. ^% D9 Y        : public IUPnPServiceCallback
{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
2 H+ r) Q9 c% S% V( S   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


  Z& ~- H: E$ J  m6 c$ Q% u// implementation
9 q/ [, @' m) `/ Q  z/ K9 kprivate:
1 ^6 }# K0 @0 Q' Y        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

- z% l5 {6 U; m( r7 F
private:
8 [! a- T6 U4 g9 x& P        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
" |* q* t! a: q' m, v% y6 S( N( A};


/////////////////////////////////////////////////
3 S1 i$ b: A2 i( U9 n/ e
2 c, V1 {/ X/ k7 o
6 k1 k+ y9 O! [使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
' `# ]+ w; v& U  F) Q2 uCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
& K5 T8 a  l( Y9 D+ S. X# ICUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.