UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   + x4 D8 y+ f0 A* `. m6 X, @3 R) v
3. 
   ( \/ a& h% w6 E0 ^+ a+ P+ o
4. #pragma   once
   - @' ], L. e" K, s! t
5. 
   
6. typedef   unsigned   long   ulong;
   : Y9 X# ~- B2 u4 u$ {
7. 
   
8. class   MyUPnP
   3 \! V6 q% M) N/ N3 e/ h, l! l
9. {
   6 o% M' O* B1 K3 o) ?$ @( a5 F
10. public:
    2 M3 c6 i9 _% P4 v; h- q( w/ f2 m% K
11. typedef   enum{
    ( S: b2 q) B# ]- r3 T* Q% x
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
    
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    # }& i) o% e% f
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    9 e9 c' P7 `8 D
18. 
    
19. typedef   enum{
    9 ^3 h0 }) v' T; H9 k; }
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    ) }6 e5 T3 M! v
23. 
    3 e4 y2 v; d  i. x  R
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    ( Q2 t+ u/ B- m* z- G
26. WORD   externalPort; //   Port   mapping   external   port
    ) y5 l7 D8 u/ K
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    ) }1 U; Q/ V9 a) q
30. 
    ( L  a  K: @$ q, M: s2 F- U- P
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    # A  ?* \  W4 C; C8 ]. q( [
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    + J# O& f/ g( v# K
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    2 Q! j! i4 E1 t% s& ~
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    * A4 [: j6 ^8 g  [0 j" B
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    ; N% U9 G6 X2 k# n
46. 
    / o2 p3 ?* ]! v+ r
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);
    & G$ R$ k8 ]1 {) Y5 R$ ]
50. 
    0 A" X, R5 q4 E' Y& @) J0 C
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    3 I8 x- N. o4 M) d5 y
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. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
    , T& [- `/ B' R9 F6 s  ~
62. CString m_description;
    8 v0 J6 ~8 G, r2 g; }. K
63. CString m_baseurl;
    # N: n2 {' W2 k
64. CString m_controlurl;
    
65. CString m_friendlyname;
    % i( d0 ?. f$ Z$ o' }# H, G1 U
66. CString m_modelname;
    
67. int m_version;
    
68. 
    9 ?2 Y3 D) D& r( a# E& X  @  N
69. private:
    , g* J# Q: N4 V, r  p% f: Z
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    * m4 A& Z( L- x4 G! V! p
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    ! h2 B0 t7 E8 }& k" i+ ^
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   * N) L" h/ z) w+ F; W: L# n" P
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   ' D# s9 E/ ]! y9 Y( t
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   ; ^& Y* y0 i0 {5 e+ X
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    5 C, j$ R, w) p  L/ q. k
11. 
    5 X3 Q4 g* S( e3 ~
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    1 _! t. Q  `3 t2 r2 `; T
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    , M9 m2 @, X/ _6 g! A  D
15. 
    
16. const   CString   getString(int   i)
    , r2 X" s: f+ t% }
17. {
    
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    # @. U, t+ z6 a
22. return   s;
    " [8 }* D7 z% w0 O: Q% o
23. }
    ' b- `- @+ \8 c) F: q2 ^
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    ( a1 Y0 s# S6 ?( G& L& G- h
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    + |5 u+ @0 L$ \( @. J* {
28. }
    $ O4 D* L# r4 \5 [
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    3 N! O5 K- P+ C) r" M" D1 |- f
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    ; x3 v/ @( t  P! u& n% s# Q
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    / {* [: v6 e% b; t0 n5 S, P
41. strcpy(buffer,   (const   char*)sa);
    0 N- o4 b! T8 k( c& P/ U4 E. y
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    5 f) e' P$ e2 C4 c
44. struct   sockaddr_in   sockaddr;
    0 G4 f) Y  O+ O8 X6 |1 O5 D
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    ' y* Q- p7 M/ o- V$ i& [9 E) {3 {
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    ; N* B  f3 N3 o; q* S& q+ ^
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    7 [1 |$ g* U* z# ^0 L9 a
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    5 y0 r& ]/ C6 }" w) R# T
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    6 `( n5 o0 e4 p  H- l% g- ^$ R; ?
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    * U& R$ Y  z6 N8 A* R
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    " M& D# b7 x& Y
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    8 F+ [" S: x1 Y0 f1 N
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    ! c, P" A( x  w( o, e1 N
62. 
    
63. return   true;
    
64. }
    # U6 Z2 d% ~1 j1 J* C
65. 
    + m! w6 @7 g! p% W7 t( p0 J2 k
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    / j" o3 g4 c1 S# l3 |/ Y
67. {
    
68. char   buffer[10240];
    / c7 Y& }% i( ~: G
69. 
      e0 ^4 n8 A1 t4 I
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
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);
    8 z  O# {" w- a( q
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    + g4 \. k, w5 E
81. }
    % `: S9 @% p* i  D* G
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    - a& j& M( b1 N, m) A/ o
84. {
    ! ]6 N( P) ~8 p# u
85. int   pos   =   0;
    ' d4 y1 ^( \0 ?/ T4 k5 t
86. 
    % g# \- {; S9 Z" g
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    ( B2 K  g1 n! A
88. 
    % Q" U5 S) u8 C: Z" K
89. result   =   response;
    
90. result.Delete(0,   pos);
    7 k3 ?+ n3 V0 H  u+ J6 u
91. 
    ! ^9 c: ^/ B- ?+ g- O0 G: ]: P
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    3 X4 @' q. K8 O) Q
94. status   =   status.Tokenize(_T( "   "),   pos);
    ! k3 c; ^( r7 o5 A$ @+ M$ u0 Y
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    
98. 
    - |/ v  J# ^$ f5 _! f
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     4 l) @" g) j9 N4 A/ l# e) @7 d
103. CString   property;
     * U+ b" L/ m% _- `0 O
104. 
     
105. int   posStart   =   all.Find(startTag);
     : Q! q) n% d4 i
106. if   (posStart <0)   return   CString();
     
107. 
     . M6 _1 n0 R; A# b7 b# k" a
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     0 K4 P1 b, ~7 p2 \& A, i6 ^; Q
110. 
     2 F4 S. f% g5 u' P# o, {7 f
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     1 [6 K$ }8 C" T. w* m
112. }
     / q% |$ {- Q9 m6 h8 T7 |7 U3 E
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     ! O: G: a$ G, ?0 C7 z  f
116. {
     
117. m_uLocalIP   =   0;
       _8 X7 f1 Z! q8 `
118. isSearched   =   false;
     
119. }
     
120. 
     + u3 S4 j  S8 H! d; j& i$ `- j/ j
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     3 R" a+ r+ s) R
125. while(pos){
     : a" c3 i8 N1 k0 y
126. search   =   m_Mappings.GetNext(pos);
     * Y( ^, k. n$ w" B
127. RemoveNATPortMapping(search,   false);
     
128. }
     1 B$ A* S9 _, S: U5 J+ a
129. 
     
130. m_Mappings.RemoveAll();
     
131. }
     0 s) h7 b9 A! C
132. 
     & C5 T- x. M; [- o4 J9 X
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     ) c$ U% Q* }  H, A; w0 P
136. if(version <=0)version   =   1;
     - S, x( K0 y! ~# ^
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     / G! S$ ~" t. `
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     " j8 X' C1 c8 f! X9 o+ L
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     ! c" ]1 E8 C. o$ J" G; [5 U. }
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     7 @: o2 {! X+ T' o6 ]. U
147. u_long   lv   =   1;
     ' Z" R0 P7 U& j2 t. K2 z
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     ) B4 U) F4 a0 }0 k
150. int   rlen   =   0;
     + [3 U0 I" w% q& m6 w
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     6 z3 }$ n6 t8 k) q+ ^# a
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 "),
     : ?3 C/ N0 V9 S$ P6 _; K/ d  s. _
157. 6,   m_name);
     + d$ p( J6 @0 h/ H& ?
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     % c; K0 _4 N+ Y# M# g
159. }
     8 i" I# U7 Q3 s" e0 [7 X
160. }
     
161. 
     7 ~, A- g0 W% @9 Z& B) q
162. Sleep(10);
     " o3 J& \) J3 @
163. 
     
164. char   buffer[10240];
     7 h! C- E- D" l8 A% t" E
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     ; h: s7 I% |! K
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     ) }- O6 e8 k- R# G# |
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)   {
     * ]( u0 M9 `) r0 ^9 n
176. for   (int   pos   =   0;;)   {
     3 p# ~$ J9 p- H9 |- I
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     9 j8 L# K8 M5 V1 P7 `: s0 ]
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();
     
185. return   GetDescription();
     
186. }
     7 x, V% b3 o9 O# P. v, p
187. }
     
188. }
     5 m6 {7 E5 {6 j  i- y. y
189. }
     
190. }
     
191. closesocket(s);
     
192. 
     + m) e, @0 j3 H6 [. o+ c. \' N
193. return   false;
     
194. }
       Z( S- ?4 R" F2 s& \8 K

复制代码

天云上人

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

4 ?" c" B# C( i5 `
/ `( @3 Z, t' d3 H3 I" F9 d///////////////////////////////////////////
+ B: z! [0 P  }7 c) j1 ]+ O" o( E//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
% q3 e* F  a9 c+ b  ~* ^
: @0 F% S6 C! [# v
, C, P" J+ _$ v) @; h1 M* @#pragma once
3 y2 \" L9 K+ |/ r0 ~& m#include <exception>

9 x. E" q6 N) }% K
  enum TRISTATE{
/ a" G) A' Z" I( n        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
6 `$ ]" f+ R- _7 n0 {};

6 m8 ^$ w- ]6 ~. S$ B% b. z
# R+ N$ \& H" x: u9 benum UPNP_IMPLEMENTATION{
* s* R- p+ g7 }4 K        UPNP_IMPL_WINDOWSERVICE = 0,
6 u, |. d. ^- d6 b        UPNP_IMPL_MINIUPNPLIB,
6 W" P1 f' T5 a, x& _' P        UPNP_IMPL_NONE /*last*/
};
  h: r& L% h" T. x' F5 P

% P/ I9 M" }6 S

class CUPnPImpl
{
public:
        CUPnPImpl();
        virtual ~CUPnPImpl();
) ?8 \7 B# ~7 N; h+ Z        struct UPnPError : std::exception {};
        enum {
, p, a- M+ `/ D7 ^( Z0 K1 h                UPNP_OK,
                UPNP_FAILED,
; r  R$ _8 ?, l                UPNP_TIMEOUT
        };
. }* a( f7 }0 a. o; d; c- s0 i

8 F' y! i) E- w        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
& Y( B, f" f# a        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
7 }( m$ b5 t$ E; N- |: N  A4 C; h       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
9 a/ ?  i. N1 g% s; R  n; z: K
7 {& H2 j  U0 ], H) W
7 E0 l& q) \6 k- n! I' u        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
* e3 @* X* m$ m        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
8 j3 s- `6 N! E        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
$ x, {; Q' L; o* ?! v( U# F& F
9 D0 F. C# a$ d: U0 o7 O: Q
// Implementation
2 i3 Z, E' Z+ X/ J6 _: n. J1 ?protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
1 J4 U5 ~: f, U3 }8 q; W# h2 R9 q        void                                SendResultMessage();
& O2 Q& t# v3 u$ {9 r- @% @4 C        uint16                                m_nUDPPort;
  q1 v5 E# h, ?! m; _; Y        uint16                                m_nTCPPort;
/ R: A1 u! q9 R+ z0 s: ^! Y2 q+ J        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;

% k$ ?( {+ F( w/ Y/ E% ^
& g$ C0 F+ g3 K+ W- ~private:
* A4 N# K( h; g& u) z; D/ N# G; ]        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;


};
2 s) c( q0 ~% F: z2 [/ D! C

2 ?  a8 D5 s' [( x// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
" U$ I# G- _1 W5 ?& d8 O3 d7 @& O{
0 `0 \# Q/ r5 O* \) M  W/ hpublic:
. N+ f3 J/ w. N1 i4 a        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
0 R5 ?0 G' [. P/ y. ^/ O        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
: |/ }2 W5 v$ E        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
& J2 H% }5 t6 V" \};
8 J- c! I# k8 N2 i

/////////////////////////////////////
* H0 Z. v1 G2 q& a//下面是使用windows操作系统自带的UPNP功能的子类
, n! S7 p( B: O! r( m9 i4 A. H9 K

#pragma once
% j, p6 C! E5 N( x% X#pragma warning( disable: 4355 )

, R' b- C- Q: Y, q( O) \
#include "UPnPImpl.h"
#include <upnp.h>
3 o& z3 E! |8 s+ }. e#include <iphlpapi.h>
& C' }; @6 f4 r3 \$ v6 N#include <comdef.h>
#include <winsvc.h>
& m: S- g& g, @( M
& b4 `, ?; |5 r
#include <vector>
$ G- c# d6 `: W' [#include <exception>
# q% j" \& A  r7 I, @0 C: d#include <functional>
: n3 Z& o0 G7 b, J! b
: S) A  I" j: A. G


' I' y. c2 w3 M- x0 Ttypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
! A4 F) j/ i+ W0 W' i3 Dtypedef _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;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
0 w) p" j0 Y# z% x* Ctypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);

. O( N2 i  X3 Z% X) E* D2 j; s
( X( X7 Y1 `4 \7 }7 @! j, jtypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
);


0 }" B( v& u0 X* a$ dtypedef DWORD (WINAPI* TGetIfEntry) (
) P  Z. Y5 n, p: |' `/ ~  PMIB_IFROW pIfRow
);


  ~9 P; q2 ?4 s* ^, ?' s! YCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
2 D9 O0 V& i+ R$ n% a& ^3 N; H
) i" O) G# o" _+ L, v. @7 y
# j, c' F# }: Y* T7 iclass CUPnPImplWinServ: public CUPnPImpl
4 L2 V2 u5 u& i{
2 _) u. M7 X4 ?8 z        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
: d8 |# W$ N( C! }2 ?  cpublic:
6 k4 Y! ]' X" P2 r2 N        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
, E: M# [6 N1 C) M: V' M/ i
# x% i' K( |: b2 ^% }
7 v0 R& A; {: ~. m$ L$ p9 X. @: y+ z1 e        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
0 K/ b' O- @/ m        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

/ n- W$ ~. k' B) X7 Z! b  o: {3 Y
        // 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; };

. X0 s% p( @7 {5 s& g# a' b
: x) w; l* ~( u9 _2 n; A" k' eprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
& C/ ^8 P. O3 t9 j: @        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
4 z0 _1 ^. }. h4 ^, g        void        RemoveDevice(CComBSTR bsUDN);
, \# s- Z9 c# q3 m6 ]- ?* ^        bool        OnSearchComplete();
        void        Init();
8 s( ?' P0 Z& ~7 H' D

        inline bool IsAsyncFindRunning()
# J$ F6 @+ N0 c; T, [2 `        {
: S9 n) f  J) L; C                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
+ \7 s1 v5 e  @1 Y                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
4 e3 M) E( ]7 W; ~                        m_bAsyncFindRunning = false;
                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
6 `9 G/ X) N0 ~+ i9 `5 U! f                        TranslateMessage( &msg );
5 x: f9 R( _& O* m                        DispatchMessage( &msg );
                }
3 }* y) z7 p# p5 k+ Q5 \* ~% g                return m_bAsyncFindRunning;
        }
% A6 i/ F( Q7 j) g0 z( C
) t: y. E7 {$ Q' n- E
        TRISTATE                        m_bUPnPDeviceConnected;
8 H8 @+ d. h& G5 }" ?9 A
3 l8 |; {: C& [
$ c( d: ?9 ?, w# B// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
2 F5 L  @, j( v- ?: Y6 T9 V- `+ W        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
$ [7 Q! ]. y- C# [        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
: k- H4 z% j8 m5 A        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
/ g+ N+ _2 ~$ z; d& k# K

, @2 N; w1 A/ `  d$ _        TGetBestInterface                m_pfGetBestInterface;
2 N! e- f1 m! j/ _        TGetIpAddrTable                        m_pfGetIpAddrTable;
5 Y. U; k" D  ~" z; E- _' E        TGetIfEntry                                m_pfGetIfEntry;
& V3 `: ~8 k$ S( ?$ j( O

3 Y, \* o# _8 [$ a2 S        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
' H  A  ~" O! n3 @# k! z+ C* Y( w6 `                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
- {. X/ O( z" I2 R

                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
' g3 p% r5 A8 O: ~" u

- @1 E/ _  ?" S# s5 `- Q                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
4 y$ l5 t5 T" ?        };
, \. w6 P, U8 d  p) o3 E       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
4 j& X0 I$ g! z( d7 Y( Y$ K+ [: t        HRESULT        GetDeviceServices(DevicePointer pDevice);
+ Z. @* \% ?* o        void        StartPortMapping();
! `; g. r( i8 l% K2 w$ R+ e        HRESULT        MapPort(const ServicePointer& service);
0 }7 @& B5 g5 h# t) x/ E        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
; S' M7 C; H' [
5 |$ |/ q# I3 J4 _1 u
9 x0 z" `- h; Q- i+ d# l5 V! U        // Utility functions
/ u+ a* k# B# M" M& o+ q        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
. ]. i: k  e  \! e- Y( ]        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
6 P- v) u# a$ a7 ~        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
+ k' Q0 w% u" B9 z/ j        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
0 R6 g: f+ |" s' a( f6 Z# j: [        CString        GetLocalRoutableIP(ServicePointer pService);
% k6 \. b2 r6 |# O6 g6 ~0 ~
& H5 ?; X; k/ V. v) ], d% u' W6 z8 c
// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
: z* A# P2 p* M, s        std::vector< DevicePointer >  m_pDevices;
: W$ U9 `# X7 Y" v- b        std::vector< ServicePointer > m_pServices;
% l. w* b% s: c6 ?        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
  @" }0 h8 j" n( L
6 v; p2 y# O5 B, b( n
6 K( b  J6 j$ Q' K. X        LONG        m_nAsyncFindHandle;
5 e* ^1 l& U- ?        bool        m_bCOM;
        bool        m_bPortIsFree;
8 C: L9 M/ |  j* y, s# Z) |! W        CString m_sLocalIP;
" K0 n. U% {. m. a0 P; Z# {; B        CString m_sExternalIP;
, }9 \- d/ E$ w7 p* ^        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
- F8 ~+ F5 b& h0 e$ h0 k        bool        m_bInited;
, z9 I3 \* w1 U, U) x        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
2 O; V" d/ d7 Z3 n        HMODULE        m_hIPHLPAPI_DLL;
/ `, ?$ H5 [3 c8 |" f6 I        bool        m_bSecondTry;
/ L' z0 i  w- n4 ]9 V        bool        m_bServiceStartedByEmule;
: E$ e4 u# r9 S; L, t        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
3 ~2 [$ d. _/ R  d+ g  x. W
" u: p2 `) P. _9 }
};

3 l0 b8 \# ~" k! H  K% ^
6 r$ o* h7 Z8 Y  R) J7 G// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
' f% d0 `5 W5 o, u; @5 d& }/ o; qpublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
7 {* P( n7 P" q

   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
0 Y" t5 A$ @; ~   STDMETHODIMP_(ULONG) AddRef();
; g/ d% }" q! ~% Q& e7 M/ l, @   STDMETHODIMP_(ULONG) Release();


8 V7 x! q7 u9 w0 C4 e: W// implementation
/ {. y# T# a; h& pprivate:
2 Y. J1 v' A4 l. M  h- r& g' e7 J; c        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
2 e, D8 u) X* @/ p' l        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
, s# Q! f, R( ^& r
* I, m4 h( p" H) w: S$ F6 R; m, I
# N& |. U* j4 |& G% L5 zprivate:
, C* n6 F; R+ E* A4 j2 {; o0 {        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
" o8 R' K0 f5 B( d3 V
: F2 [3 B, x& i  D
// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
8 U. r9 C' t3 Y{
public:
1 D* F& ?8 E( N* \0 m5 H( w        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
! `) Y+ |' }4 D/ B   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
% s! }/ B- G7 S) z  v   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

4 H  A4 n( ~+ e% t- J
2 Q8 c" R3 F3 P) ^8 G// implementation
, Z1 e& L# g, |: ~7 M; Vprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


9 b+ `! T& E. A! Yprivate:
! I2 z! m* U; w. E9 z        CUPnPImplWinServ& m_instance;
8 k+ y: S- |) N& e( U9 m        LONG m_lRefCount;
};
7 N% |* L& g) Z1 z

+ T1 u/ k& f6 {- @% `/////////////////////////////////////////////////

* g, T* @" w& ?( {' x  E8 ]
# s9 Y5 C- A9 M1 q使用时只需要使用抽象类的接口。
, O" P' I' X$ A! I8 F1 q, OCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
1 i( k3 o6 Z2 m5 P: J% jCUPnPImpl::DeletePorts删除端口映射.