UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   / B: v; B) C) B, r' v  E% M2 O+ l
3. 
   6 I1 c/ }6 L0 y9 [: s4 t, ]. w/ V
4. #pragma   once
   
5. 
   3 y# ~* N0 Z+ ~' i
6. typedef   unsigned   long   ulong;
   
7. 
   , ?& U: ?6 ~# u' |2 m
8. class   MyUPnP
   
9. {
   
10. public:
    + b! w6 M- ^" c/ O
11. typedef   enum{
    ' s6 x) G: M4 K! i6 k9 }! p
12. UNAT_OK, //   Successfull
    / o; W4 F8 a, W4 `, c& J
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    7 E/ ^$ G- `, m& J
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    ) W6 E9 [3 B4 x% I6 c
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
    * J, c, s) `7 G/ O
17. }   UPNPNAT_RETURN;
    # ?1 W1 g6 s/ d! Y4 q
18. 
    3 v+ i& C( K! A- E6 X0 u2 F
19. typedef   enum{
    / r1 R; A+ M  {0 P1 u  p; ^
20. UNAT_TCP, //   TCP   Protocol
    0 W) `: E5 W1 b" g/ m* z$ R
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    0 B  N% M( y/ x8 W0 [
26. WORD   externalPort; //   Port   mapping   external   port
    ) Q$ C1 `& l* O
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    $ D) @* \# p" g
30. 
    3 L* j- h% g  T; B
31. MyUPnP();
    " c7 J  o/ S$ Q* y# \
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    8 }2 h/ I+ F- Q* |) D( [# K
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    . ~. V( Z  X2 b( u
39. CString GetLocalIPStr();
    9 \1 u! }7 C- j# }6 n
40. WORD GetLocalIP();
    & k% o% O' a; Y9 l$ p) Y4 H
41. bool IsLANIP(WORD   nIP);
    $ i# [+ `" J3 U8 `. a! a3 o8 Q
42. 
    
43. protected:
    
44. void InitLocalIP();
    1 t4 A6 h! I4 F# a9 a+ n
45. void SetLastError(CString   error);
    
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    1 ?: K6 M4 e+ ]' ]  w) K
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
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);
    ' v+ }  Y" h* x  A
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    6 S0 C0 O. |: ~  A
60. CString m_devicename;
    ' L  G0 |0 t+ w- A
61. CString m_name;
    % k* ^( \4 i4 Q  X
62. CString m_description;
    ) Y) B- J' U% X2 V- M
63. CString m_baseurl;
    
64. CString m_controlurl;
    5 S; F; a& y6 s) Y) ^- J# o8 Z- t
65. CString m_friendlyname;
    & M( }+ ~% @, ^( C+ a
66. CString m_modelname;
    
67. int m_version;
    
68. 
    7 u  g3 _5 y: x( k* o
69. private:
      J8 z. ~3 x0 t( n
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    0 ~4 ^" V& D; I. {% B
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    $ [% q1 D- D6 ^9 |( h8 y( M
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   + x+ X% o5 e! B$ n, A5 T
2. #include   "stdafx.h "
   0 ], m4 R+ P6 d9 d- r% a2 c, e; W1 {
3. 
   # B- |2 r% l' ~( e
4. #include   "upnp.h "
   
5. 
   3 _9 N! k# ^/ a& |$ h
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
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;
    # s4 W" f8 [* X: d+ m* i* U4 o" w
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    + U( |7 M" U/ s( Y! o2 @8 Z
15. 
    
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    ) }: U; I* I1 I/ h6 w8 }
19. 
    % ]# y+ W! z# N; I
20. s.Format(_T( "%d "),   i);
    - Y* ~5 l' m, M0 e' V+ Q
21. 
    ) h! Y0 a* f$ T5 m: q2 r
22. return   s;
    
23. }
    1 t9 A" G6 L- \" s, o- B" E% J
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    " ~* ^. C6 Y, R; M# d/ }5 |- f! z3 ~
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    . z1 I: C8 O+ _9 w
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    , D) `$ d$ r9 q8 e
33. }
    & k8 D& r! [1 o8 q% Y
34. 
    : \! {. p. W0 X# G4 i9 K
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
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));
    4 `- ?* C+ ~5 Y" J  t& u0 ]4 R
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    2 j& m( S/ S' s
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    6 W+ y  ^& G) F
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    4 _6 H- i0 q  c6 F( I
50. u_long   lv   =   1;
    $ Z/ d# _8 F6 _' s; @( g# m
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
      k% H7 e. t6 ]% ~: U
54. int   n   =   send(s,   buffer,   length,   0);
    1 a2 i* p6 V$ P8 ^
55. Sleep(100);
    : f0 X& {" P# f- ~+ M& ]  J; d# ~2 }" c
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    ; Z" |3 V' _' m7 S
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    7 O' v% A/ @! u/ q9 o* n, w2 ^
62. 
    ( o7 i- B. A5 Q$ c( Y7 a, c
63. return   true;
    % U! i, \- }  f2 V; W* P% ?
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    ( O7 D$ f+ ~/ A9 A
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    # D0 Y2 P# A- S/ g
74. struct   sockaddr_in   sockaddr;
    # d3 b2 Z" `5 J. y/ [
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    0 ^7 F; d( j: f) B9 Q% u
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    " x9 G7 F' [7 c& l" @4 ]- r1 v
81. }
    ! s* G0 K/ K+ I/ c4 Z
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    0 f+ l/ @. C  B$ f/ v9 q) @
84. {
    
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    % I. [, S6 H0 S: O. q; f9 {
89. result   =   response;
    
90. result.Delete(0,   pos);
    ' Z! [5 W3 g% b& G3 a/ F2 P9 K/ A! u
91. 
    
92. pos   =   0;
    1 [, h  N0 N0 v' B
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    $ @( L' P( w% J( y2 M& M
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    9 E0 |: W+ H4 Q- p/ j* k
96. return   true;
    ' F4 D' K  J+ y& ]0 @: T& ]! D
97. }
    
98. 
    3 l4 Z8 B3 Q) l
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    & J! h9 H3 b; ]% J
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     8 B" n. z. m, p8 w$ G" A
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     3 s" [0 o) n  `* z% c  a; |; n- l
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     $ r  t, L8 _% x2 S
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     ! @7 O- Q2 o) e$ H3 R8 w
112. }
     
113. 
     ! e( ?3 ?( G( s4 V. R
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     & j* h$ E/ @' g5 R
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     ( B( g4 F! l$ |$ K
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     $ h% T/ P8 i1 q1 q, z1 i* B
123. UPNPNAT_MAPPING   search;
     7 N. g9 N6 K1 `' k1 q* T% `/ r4 X
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     7 g1 G) a7 l+ s, X
125. while(pos){
     # m, |5 _, @5 j7 \8 t* ^- g9 U
126. search   =   m_Mappings.GetNext(pos);
     ( Y, ~8 l# h$ t  C3 l
127. RemoveNATPortMapping(search,   false);
     
128. }
     2 g2 B& p; ?  Z/ N2 C
129. 
     2 y+ A" ]; t& E! T: X* s- v1 f
130. m_Mappings.RemoveAll();
     . V/ C9 E6 P) }2 D( G; Z3 a
131. }
     
132. 
     ' _8 [7 c; e  b4 z5 W' a
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     # O) ?& o6 r5 i
135. {
     
136. if(version <=0)version   =   1;
     * L8 x8 e. n2 T; j* i3 d% K7 b+ ~
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     ; n1 Q* @' [* x* U( B* M2 F
140. CString   devices[][2]   =   {
     0 s' P& B! `/ ?) T) ?
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     : \( `7 C! A& f; r6 P" \
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     7 [' C9 }; Y# X; ~
149. 
     
150. int   rlen   =   0;
     + d0 C2 z6 G1 B# b( A+ X6 S
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     $ j& x% T( ^% }, w
152. if   (!(i%100))   {
     $ q0 f$ _+ {* {3 H! {0 b
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);
     * c- J7 W- G3 x- j# h1 y0 h; P
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 "),
     # E* y& L, h+ ~9 ~- Z/ R
157. 6,   m_name);
       l' t" a( {; r5 l0 p% `; o
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     
161. 
     % e1 @8 ?$ g/ }' A  t0 b* H
162. Sleep(10);
     + d5 ]. J# M; L$ M& e. L
163. 
     
164. char   buffer[10240];
     " p% [4 ?% \. i/ J
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     6 c0 T: j( v, V- n
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     6 G  w8 Y7 Z) G0 U& J
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     # s# \4 m% B- g2 c; w) z
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     $ B! v) n5 d: `3 U. s
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
       v( T$ ~) y9 u! i0 k4 G
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     . l/ o7 @( _7 k( U' `1 H( R
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     ; R1 ]1 _& o* c. j. G1 P6 w
185. return   GetDescription();
     
186. }
     
187. }
     
188. }
     
189. }
     
190. }
     
191. closesocket(s);
     
192. 
     . _- a! a7 Q, N, Y, h9 Q$ q
193. return   false;
     
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
  `5 p$ b0 G; }

///////////////////////////////////////////
8 U" C( @" Z: ~: Z//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

" I) G4 ~4 y9 F* ?
( ]+ F3 [/ J; X" E6 p- V- L#pragma once
#include <exception>
* Q, n& x% D7 d/ f, u
. G) w( v4 x# ~& B
  enum TRISTATE{
, k, T, k% a* u6 S: x! _! M) ~2 K        TRIS_FALSE,
$ {6 I4 ?! ^% d) X3 f; h5 y2 f# Z- h        TRIS_UNKNOWN,
% h" U3 ]! g( C  J; `6 N        TRIS_TRUE
" _: a" u) V) c$ r- O* q% F/ _/ B};
4 Y  e, M  D! K) ]

. y7 {; z9 v1 J! @enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
0 n9 x7 j+ R5 J9 Y) j. T3 {) M};
5 V# W/ g" Y! u0 ^+ X9 y4 b; h$ H9 H
- W6 y+ `$ ?1 }+ r
) T( [; W+ \) Z4 T: s( y4 L

4 n- v2 Y$ [2 K2 O9 Wclass CUPnPImpl
{
: ?# x5 b  c: Upublic:
, W% K: s+ C; o' _4 t        CUPnPImpl();
0 s( i- g/ a! O$ w        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
4 L% ]1 q+ K$ a% K& B6 |        enum {
                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
        };

3 T7 w5 S( D1 L( v, h
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
( C' m2 k6 K$ w! F. l- }4 a  N' }        virtual void        DeletePorts() = 0;
. f' R  K( \% {7 l2 S: l9 a# I        virtual bool        IsReady() = 0;
% u2 X/ g/ n, I' M. x) R        virtual int                GetImplementationID() = 0;
       
5 F- ?0 I+ A' V$ C; }5 q8 n        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
2 C( j! E2 V7 _        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


3 w7 ?3 Y  u5 u5 a/ V, x& G// Implementation
protected:
  L8 `1 f. ]+ s" ?1 |3 S        volatile TRISTATE        m_bUPnPPortsForwarded;
2 Z. t  r5 p% S. V) U5 b        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
9 ]3 Y, s$ B! A/ `, f2 E! ]" {9 |        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
) k* M8 Q6 p! y

: P3 H, |- k8 m4 k8 O$ f; X, f8 wprivate:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
& |; @4 u$ W4 H$ \# o9 o
0 ~- T, C8 F: i$ h6 x- ^
( Q* J+ x3 r% t9 L, Y};

' z; _. M' v; C
// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
) n. C* d# b( p/ x% M{
, N& u2 F/ Y  c8 C/ p  Rpublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
& l. d: X+ t5 H* f& v2 M. c        virtual void        StopAsyncFind()                                                                                        { }
1 n% E% b- \/ h  F        virtual void        DeletePorts()                                                                                        { }
1 @# `8 k3 Q, ?0 H, F- U4 g$ f        virtual bool        IsReady()                                                                                                { return false; }
& v2 X4 g5 r: n: ~! ]" H; r        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
, ~) F5 u  W$ e' [+ j1 D};


/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
* Y2 D1 Q: V- N, k; z1 w) B
2 C5 g: `. H# }' M
' v/ Z& f8 J& J#pragma once
#pragma warning( disable: 4355 )
0 r7 S, U1 d' V! V5 g8 L
" @0 f; g6 X. [; [
* l' d  I: O* a. |#include "UPnPImpl.h"
# X. B: a3 \0 }& y2 U#include <upnp.h>
! |% W/ y+ I: p; t( n" @  {#include <iphlpapi.h>
" A& n5 o9 N, m6 h3 V8 }+ g#include <comdef.h>
#include <winsvc.h>
; F, D2 ]  A0 o! `: j% j

#include <vector>
#include <exception>
#include <functional>


% b2 b9 G! G5 ]2 @( K
/ Z: T# g5 x! I3 M
+ s. W/ x  w" mtypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
: X: N6 v  [( b3 t# y% T4 }9 Rtypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
8 E3 c6 O* ~8 x+ J5 z7 B' |" J5 w' Ytypedef        _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;

6 ?  a. r6 }" T0 _) S
typedef DWORD (WINAPI* TGetBestInterface) (
: |8 d- N( M& d, U. ?- q' O. ~# O  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
7 o. H: I% T2 f; v" J);

7 V+ A( t' y" a& o4 ?/ m
$ H( g: k2 ~, T2 n  O5 A* y+ Htypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
/ S. A  ]2 U% D7 m: w3 C1 }( v1 v2 p  PULONG pdwSize,
* n& _- _3 ]" k2 w* ~8 [3 h  BOOL bOrder
);
* y" f# |) z% ?( P+ `) b7 w+ O. ^+ h9 D

( F. N0 C( {) o- p, c6 ttypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
, ?2 h5 g9 ]% M6 l);


3 t; U; ?; U! v8 @+ \7 O+ N! kCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);


8 i0 b0 W! O3 M9 `! M: e+ K6 a) |class CUPnPImplWinServ: public CUPnPImpl
7 f/ n" |0 J3 _; B! M  I{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
4 p, {" y: s3 `4 J/ L: e// Construction
public:
) P' ~! Q( E2 n/ E# C        virtual ~CUPnPImplWinServ();
' |- T9 d+ y& e, H        CUPnPImplWinServ();
/ n5 b! z% N0 I, ?4 }3 n% L6 y
3 t; i3 C3 p& q- @6 U, o4 j
* F1 @! ?4 \& ^$ C        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
& o8 J! j4 y) j# F& j  c" e, Q        virtual void        StopAsyncFind();
! f& Q& E1 |; p, E        virtual void        DeletePorts();
1 X" j$ }0 O- v3 v) c* D  s7 S        virtual bool        IsReady();
0 a# E  M: p7 z3 L  q( \4 u" m; i3 K        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
" x5 }0 v  S$ j
: X& K% s$ E+ ]) b% h
  H5 j- p- a4 V        // 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
8 P6 U/ ]2 j' E/ i& {0 w' f        virtual bool        CheckAndRefresh()                                                                                { return false; };
+ i( w9 }+ Y3 w  U

protected:
$ T! U  ]) c! q' c8 i$ |        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
% x, p* H! N4 ^1 [1 Q        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
4 o1 ~0 N( f" V4 z        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();

5 j: s9 m. }/ T9 q/ e
/ p* I! a9 b7 h+ I, t) I        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
, a$ t+ }4 E: ^) O8 |                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
& }9 W( i9 G0 r1 r/ F8 Y                }
, I5 r% M+ d9 F0 T                MSG msg;
$ N8 b2 G# B; V3 B                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
/ z& \; G8 }1 ~$ `6 t2 I) d0 P                {
                        TranslateMessage( &msg );
2 E6 S' A1 l3 F                        DispatchMessage( &msg );
2 G, y2 J. X6 C0 w/ i$ ~  ~# W+ `* F                }
' \6 T9 c" E( }' v" A. K                return m_bAsyncFindRunning;
        }

6 h8 S6 @( u3 A9 \  l0 s6 J3 D
* S1 j% |; Q7 v        TRISTATE                        m_bUPnPDeviceConnected;
" A" b+ }% F9 p

0 h8 `6 o) d6 R2 N0 z// Implementation
4 S  e% a( o8 t1 h3 K5 _$ N        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
3 L4 N0 W3 z0 F        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
3 {5 E7 e7 [3 k6 \  a/ I' |: k        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
7 t/ d: k+ W+ N; r/ ]: W9 }; W        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
* _! W0 X3 J7 p- \( \) u        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
, T) ]& H7 ~  i4 I, p$ m) Y' u        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


        TGetBestInterface                m_pfGetBestInterface;
( L: a: x8 N: h. y  r        TGetIpAddrTable                        m_pfGetIpAddrTable;
4 E  i9 d% _5 I9 \9 f) f3 `        TGetIfEntry                                m_pfGetIfEntry;
6 a2 U8 z2 B" z, u

        static FinderPointer CreateFinderInstance();
( d( ^4 t4 L1 }$ W6 F- ]1 J        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
0 U8 G9 L& m2 x2 x8 k. ?7 N                {
% x2 R! [* ^( b4 g+ G3 F                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


, V# d$ M, E, ?5 \3 L8 r                        if ( FAILED( hr ) )
3 Q7 ^( l: ]+ |: j                                return UPnPMessage( hr ), false;
' ?1 q8 W' C6 \

                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
# \: `; n& k; u; n4 g                CComBSTR m_udn;
# X% w/ i, B) R4 C        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
( O  b& W$ t& e: U/ b/ I+ k        void        CreatePortMappings(ServicePointer pService);
3 y( T8 q1 B2 j6 q- l' O        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
+ n' \  F, V: @" M  y; E

0 ]$ y2 H7 e8 C7 M( q, l6 L9 {8 I" p        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
) n$ _0 [. e5 v9 r" E" X        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
$ o* r0 z/ W" k4 _: T$ W7 f  x        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
; d: b5 r( V8 M4 V) D3 H        CString        GetLocalRoutableIP(ServicePointer pService);
# l0 e7 o' x: X2 v" i; y! G5 h

1 @6 G, S4 P& v// Private members
- G- q7 Z8 h" p+ Oprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
* C3 }+ @8 D, x( T- C3 s        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
" Y. [( J0 d2 |# G/ p3 u& A        ServiceCallback                        m_pServiceCallback;

; D8 o2 r: z  w- I/ Q9 T
        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
7 I3 l8 R" ]% }" |        bool        m_bPortIsFree;
' S+ a8 E& M8 `5 d        CString m_sLocalIP;
* @0 V3 D- D6 \. U6 ^5 L9 R        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
3 k* D! m  T1 I" _7 ^4 v/ E% O, Z        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
, a6 H) F  M; o. H% v        bool        m_bServiceStartedByEmule;
  ~! }2 |1 O. k) x" b5 E' X/ u. }        bool        m_bDisableWANIPSetup;
" Q- P& t- K- D        bool        m_bDisableWANPPPSetup;

5 I& @- I0 e3 }3 C7 b
- D4 @8 A+ v! l! r. Q# c1 |};


: ^1 x$ w$ ]4 n4 _  k// DeviceFinder Callback
class CDeviceFinderCallback
% M4 w( B. D1 G/ N, A$ X        : public IUPnPDeviceFinderCallback
* ?' A. r* ^6 k  z9 z1 F+ `# D8 d{
public:
/ c' v9 j/ T# g, m+ z& ]* g5 m% ~        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
3 S) ]+ Z1 H" Z$ W7 C( M! c        { m_lRefCount = 0; }
+ W# q- Y, q' }$ t) Y; k9 k2 P
0 y9 F' z- E1 u$ @
5 X/ y* \7 P+ [0 [, @6 J   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
3 y, U# n! H2 X        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
" C2 {* t0 s( I1 J# O1 C/ B1 |/ D        HRESULT __stdcall SearchComplete(LONG nFindData);


private:
        CUPnPImplWinServ& m_instance;
9 }7 o' ~8 I% d, ]; b- N) F  H+ e        LONG m_lRefCount;
6 J- ]; S7 c. ?4 D5 D0 P1 ~- Y/ T};

: h- I/ g! ]; F: ?8 Q! L& Q
// Service Callback
, x/ N% F$ ^) X0 A$ N6 I; H# zclass CServiceCallback
7 u% K/ f) P- f% a2 t1 ?. E        : public IUPnPServiceCallback
{
; I  i# U! X& o5 a' `; i+ ?: Upublic:
        CServiceCallback(CUPnPImplWinServ& instance)
4 r% A) R# M6 d4 q                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
- V6 n( f0 l4 h   STDMETHODIMP_(ULONG) Release();

0 |* z/ U1 k' x6 t9 S8 |; {& H" J! @
/ O# [1 |) P7 i8 E9 O, B; ^// implementation
- \3 Q8 L, q4 w: q/ L: L/ Dprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
- R: X0 R- n0 A3 w: ?" W: u+ r        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
+ g7 {2 [, ^9 E6 y  m2 Y

private:
, N- {0 M6 A5 _$ w3 f2 j8 P, R        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
7 D' h9 V9 m! ~/ i4 D: |};
& Y$ L& J: ^; E% j
' F. z; j( I9 b0 o( @
% @# c$ c$ c3 ?; t/////////////////////////////////////////////////

8 d) I5 A# E" l+ Q
% d: S2 j" ~# _3 o. M  {' Y使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
  f! n$ [5 Q* G2 M4 M& d6 d8 HCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
4 P/ K( \8 W( B* T5 JCUPnPImpl::StopAsyncFind停止设备查找.
! g% `9 M4 h; }. Y  P; ^CUPnPImpl::DeletePorts删除端口映射.