UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   5 M8 f/ Z: ?. a6 |
3. 
   
4. #pragma   once
   
5. 
   6 ~* f5 r4 e6 X# U$ {6 n$ Y
6. typedef   unsigned   long   ulong;
   
7. 
   ( o& f! Y+ G# r: l2 n* D& Y# G
8. class   MyUPnP
   
9. {
   . G1 G' o: D6 p  c* l% [# |1 O
10. public:
    " s0 o* p/ q% J! q6 S- }7 K0 Q
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    , R' I; D/ j" X+ o3 i
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;
    / t0 c7 ^4 B$ O+ ]2 T; i
18. 
    . L8 I. G$ e% E
19. typedef   enum{
    6 g9 I$ F3 e0 ]9 }9 _# g4 g# l7 A
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    9 D$ m8 u1 P  q
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    , N9 H5 a* F( }# c' y' y
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    $ X( y1 j7 i+ a; N
36. void   clearNATPortMapping();
    5 z- s$ \6 e. j+ C  n4 K- e; d6 w
37. 
    
38. CString GetLastError();
    $ O; D( Y, d) i
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    0 ~/ j) r5 K. y' `. {
41. bool IsLANIP(WORD   nIP);
    
42. 
    8 J; {5 ^, u# K0 X* E" F. o; x
43. protected:
    
44. void InitLocalIP();
    # I' \9 b3 D0 B
45. void SetLastError(CString   error);
    2 ^$ ?( {! C/ h# }
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    7 S0 n2 J2 l( o4 p6 u8 m3 Z
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    . d5 i7 i8 l* R& N1 `
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    ; e9 B/ `% y/ w- v8 |$ m8 b
53. bool Search(int   version=1);
    
54. bool GetDescription();
    ( S% b4 J, O2 A" l# k: v% E( J2 g
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    7 ~" o. `0 W0 b8 A* j/ u1 Q) e. E+ ~) f
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    8 }. n2 D; C3 t& \; v
61. CString m_name;
    & Q- G8 _9 G7 S$ j
62. CString m_description;
    3 p4 o) e0 c  ~: s7 T" \$ Q
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    5 s& o. z; {' I8 A$ E4 x) T
66. CString m_modelname;
    
67. int m_version;
    ( ]+ w7 z4 ]! R/ U
68. 
    
69. private:
    1 W. Y& }  ?/ A& r
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    1 ^8 @/ V- s! E2 L9 r
71. 
    / P- A9 X' G+ [2 I2 H8 \& w& ~
72. CString m_slocalIP;
    " X/ S$ ?  \* L# B' l
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    * S, x2 d& v2 g. s9 D* d5 v
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   ( s, c, V- j  c8 I1 S% g( m3 ~
4. #include   "upnp.h "
   9 F8 B# P& L4 V! y  j( V( ?' ]
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   : g, y" B$ f- R; M/ ^1 N
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   9 r+ L, f: v3 E# o1 H$ d1 n
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   & {: J4 S+ A/ F0 b; E
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    9 h! L$ c% k2 B; o
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    : a1 D- i2 e  t/ H; p
13. static   const   int UPNPPORT   =   1900;
    ) ~+ s7 U& u; C
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    
19. 
    & P, l% B% W/ @! k
20. s.Format(_T( "%d "),   i);
    , m7 s+ D* v) |
21. 
    
22. return   s;
    
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    , l1 i7 B" c- v
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    ) Z2 F+ X3 x' e% u3 N2 Q
28. }
    4 _" ]' _" C5 U# T  I
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    5 P7 \  D6 l! F  s) F# H; J
31. {
    0 D# }4 q8 M( _& s
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    7 u2 P# m" R% `& w/ f
33. }
    " H5 Q: W3 q4 L2 K& o" [; J: Z( V9 Z
34. 
    + K& m% i) b6 N9 n% V
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();
    5 z2 c8 R* \. C1 ?. }8 r
41. strcpy(buffer,   (const   char*)sa);
    # a1 n. R: n+ q+ s8 g0 B
42. 
    & N3 e' ]. ^1 c; Z0 Z  m  F% ]
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    8 j. c, T/ w5 N. m
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    6 M! _( r: _( A. G* Z
47. sockaddr.sin_port   =   htons(port);
    * C3 g) @& `3 L( r
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ! E/ J" X- K0 `$ }0 L2 k7 t* p9 A; S
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    * N/ F- i: T$ K6 Z4 i, N% @
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    , z* D9 T4 ?0 H1 e1 B, z( F* G- k
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    * m6 @% S& H8 v3 `6 A* Z  b! `
59. if   (!rlen)   return   false;
    8 Y5 X2 n/ r# M" p
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    % G7 A1 C6 [+ }- a
64. }
    * s( j# k; x' P, a6 n
65. 
    2 A& e' I7 i: o8 c/ c5 f
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    
69. 
    5 L8 O% ^9 E' q& m2 a4 c4 J( o1 i- N
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    9 n" D' P7 E" A& t; l2 ^
73. 
    6 x% J7 f! g8 }8 _3 _; k
74. struct   sockaddr_in   sockaddr;
    : {/ Z8 T6 `2 r! t& m% t, \
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    0 q3 _- g: F) K4 ^, L. k- d
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ! Z  w' l. {" g" b5 f
79. 
    3 R+ L$ ?& }9 N9 h2 Y. |
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    7 O$ n; n6 N/ v
84. {
    
85. int   pos   =   0;
    0 @( y' y$ U/ H% A" |' d
86. 
    7 k1 z1 s) N- L# q2 |. N' D; u
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    , d! l% a! `: U4 j( Z  W
88. 
    
89. result   =   response;
    / d% I7 C, i7 y; u1 |
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    . `& l( K  D4 x* \
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    9 d$ M9 N0 v, e% f0 o1 ^7 m" H9 L9 ]
97. }
    / G  Z! z, |  T; L& M
98. 
    ' P' S4 |' s: \* n; x: U. R/ `
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     6 k/ z' R/ r$ B) ?& I  K6 H: O4 J
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     " U; G& [  j! k# p
103. CString   property;
     
104. 
     ) D$ u) M. E) v  c* X1 P& U/ ^
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
       L% c5 t4 m8 F
112. }
     % |$ |# [  v0 _' Z% o- t
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     / Q3 H! C4 ^- q+ O2 N. _- N; U, _
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     6 b+ \1 I% {$ q4 h1 \
119. }
     : T7 a1 m( \) J( |( W3 R
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     0 q, T4 Y( \: Z2 }* {
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     " @6 [" \1 u: T
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     1 @6 @7 f% Y$ e0 d$ C! p( B
131. }
     8 `. b% O- o8 l. V" P1 a: G
132. 
     # d. ]6 j/ F5 X" s+ S" Z
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     + X& y  \' J! [/ s; ?* y
135. {
     % w) ^8 l( W% v. {* T% \2 U" x0 i
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     6 I! f  o% S3 m& J% S
139. #define   NUMBEROFDEVICES 2
     3 I* o( U9 y8 o# E
140. CString   devices[][2]   =   {
     1 x/ |: ?9 o' T
141. {UPNPPORTMAP1,   _T( "service ")},
     9 L- p% d+ A! Q7 s3 A( {
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     6 ]6 }5 C+ Q3 C, e" L$ s
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     3 B, ]+ ~' p" _0 p* a. f
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     . X9 g# l6 X+ D, m7 n
150. int   rlen   =   0;
     1 J3 s, q4 y, e
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     5 Q4 n6 G) q( K. z
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);
     
155. CString   request;
     , B) z* L' H4 z( d5 p! v
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 "),
     6 d  w- m( K$ |& c' ?; b
157. 6,   m_name);
     4 z0 Z; H$ y) q- i
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     ; L! M$ T- J( ^3 M! ?! x# X
161. 
     
162. Sleep(10);
     ! k  G! ~. C! c% d. V
163. 
     + y/ r0 m/ B& Y# l% S8 r
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     $ J3 p9 f# X' G- ~% k& V
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     * w, N6 R2 }1 S! `9 F2 r
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     / I: U1 i% ]& l* x2 A+ N! D
171. if   (!parseHTTPResponse(response,   result))   return   false;
     . g2 m6 ]4 j' D. t; `5 h# C
172. 
     4 f% Z7 t# e0 \
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     8 r- O* S/ i1 ]' T
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     / }2 @/ e6 Y4 H, y! o( }; h% t
175. if   (result.Find(m_name)   > =   0)   {
     ) f2 F0 _$ Z4 {- x9 [8 _" f
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: "))   {
     - B0 P! ^, V( K" G# J! M% \
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     * Q* E" ?9 N7 C5 }" M: ~' r9 k
185. return   GetDescription();
     * E+ }, R/ ]8 @# m# o; c2 g* v$ z
186. }
     ) u, i1 }& p6 f+ U+ l
187. }
     # c  f& F8 X/ u" ]+ v' p
188. }
     2 Y9 _# R7 U# H) f0 z
189. }
     
190. }
     
191. closesocket(s);
       B" {$ Q. T; @7 g1 J
192. 
     
193. return   false;
     2 v9 {+ d8 V# q. |
194. }
     9 s7 i' r# f( I5 w- W- ?1 q

复制代码

天云上人

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

9 ~2 b; |0 A- Q8 h
///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


; V6 c" w5 j! C( G: z#pragma once
#include <exception>
- }% u: \3 c, [9 q) n: a5 Z  {' a5 ^
( G% z' \$ r0 D' i8 W$ C# D: _
, `& A, x- u6 [1 h1 z* r  enum TRISTATE{
        TRIS_FALSE,
& ?- f" O( @5 O5 ^! f* }0 R        TRIS_UNKNOWN,
' r: i5 L9 g) U3 l0 Y        TRIS_TRUE
};
- @4 b& n+ M4 _0 D# r

enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
7 w" ~/ O: w3 h' B7 @, R9 p  {        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};
: @& M5 P( e; `1 d$ w  `/ R
: j% P8 a  q# x, Z- N1 V+ k

9 {8 N. \  H2 B1 d5 z& Z& z" e
class CUPnPImpl
' @7 y, v3 B" Y+ S, J! K{
6 P" `/ J# |/ \/ opublic:
- q7 o/ y" a3 {9 o# c        CUPnPImpl();
* z2 ?2 P$ ]! f# z        virtual ~CUPnPImpl();
7 v2 i7 N' B0 o8 R' a        struct UPnPError : std::exception {};
$ U1 o  y# H: p9 Y7 I# l# ]        enum {
                UPNP_OK,
, b& ]: \9 x! V7 f2 c                UPNP_FAILED,
                UPNP_TIMEOUT
2 Y$ V( c* d# s- M9 V5 T        };
2 p# ~7 F+ W' ]3 V; X8 D8 l. S+ b- N

9 m& F0 ~. u: a9 l$ X8 n        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
/ @" [8 K3 L& ^7 ~( a( x        virtual void        DeletePorts() = 0;
  p- x* b& h# r+ N5 Q8 Z        virtual bool        IsReady() = 0;
2 X& U+ w  f& ~8 p, m3 \        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


* \4 R0 a6 J- z/ ~5 B/ P/ O- g        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
; r' j# S3 x& O2 x2 k0 |        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
- z' M, v5 B2 a. d! ^        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
8 M5 i  v! e5 x3 G: J" x

// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
5 c: G( k+ s: O: j# Y        void                                SendResultMessage();
+ U- b7 m' |8 P# f9 d2 |2 i0 K/ I        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
( J& r9 \9 P) z" a; b+ J5 p        bool                                m_bCheckAndRefresh;


1 f" v+ x9 c: Yprivate:
6 H7 ^4 S9 O& H' A5 Q1 T: X/ i        HWND        m_hResultMessageWindow;
9 ]1 Y- {7 Y8 M+ l7 l        UINT        m_nResultMessageID;
" V. N& `4 _5 c- U- A

. R, J6 R; l" q& W9 _$ Z};
2 H3 c0 W+ h" y! n6 T5 p& B; ?$ r

// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
; z6 I- t) d. P& {  a{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
# J) A4 k# _6 m6 m3 l+ ?! b" ^        virtual bool        IsReady()                                                                                                { return false; }
; y. {1 ?7 z. ]4 a1 W/ ^! W        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
. T0 i4 m1 I; M0 R

! c4 I3 z( T. j$ n/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类


, W3 V7 u6 y( H5 r1 C2 b#pragma once
#pragma warning( disable: 4355 )
' C$ V' j7 l2 d, s* t- [2 U& R. s, R
$ z) k2 N( B: q5 }+ n
* [; h  \- {2 A7 C#include "UPnPImpl.h"
#include <upnp.h>
5 H, b9 Q, v4 Y6 {* {" w#include <iphlpapi.h>
9 A" j  e0 [/ Z7 P) b: K#include <comdef.h>
6 f# l) `2 P  b) g  t#include <winsvc.h>


#include <vector>
#include <exception>
#include <functional>
! B1 Z4 J* P. i7 b

5 s5 |4 o: i; o0 C3 P" z- s, x

typedef _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;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
3 k5 V" |# {/ }" c1 ^% D. a3 ctypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


7 J/ B# F# Z/ v3 qtypedef DWORD (WINAPI* TGetBestInterface) (
, u0 Y; E* {9 f9 v& n  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
4 G6 ]1 A# ^# H, B" _: a);

+ Y( _2 x  d/ |6 p3 ~+ f# `
typedef DWORD (WINAPI* TGetIpAddrTable) (
2 e7 Y7 |6 B6 ]4 X  PMIB_IPADDRTABLE pIpAddrTable,
2 T; C$ B' D3 R' o3 Y& {  PULONG pdwSize,
" R0 z6 Q! Y5 O4 d6 ?  BOOL bOrder
);


typedef DWORD (WINAPI* TGetIfEntry) (
' W9 k: O2 X! J4 L, _8 [1 u' s! V  PMIB_IFROW pIfRow
  d( a. Z" L+ R6 {);
) R$ S2 k1 n# {% X" B
: X7 {4 v. W7 z+ X. ]
CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);

4 j! f5 x5 S) A( n- v6 m+ S
; |7 T7 w1 s  nclass CUPnPImplWinServ: public CUPnPImpl
{
, [$ S/ i; w* O% Y. v        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
public:
        virtual ~CUPnPImplWinServ();
3 f7 `& Q0 w- X9 o4 E        CUPnPImplWinServ();
0 q2 t* C8 |2 E* C+ \

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
* t1 t: i) [$ a* h& O        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
* e3 {* V3 K: b! a- _* ~        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
( E& F5 @8 j4 R: |5 N

protected:
! }8 f' g! ^3 }        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
6 l# g" O% C: V- g" f+ V" D        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();

7 @1 |" N9 b8 |/ o
5 N* C9 p, G8 W2 ]6 }) s        inline bool IsAsyncFindRunning()
        {
' o. @/ {4 S7 _/ r' S                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
) s. v! A$ C' p* k1 `                {
% l. g0 E8 B3 s9 y& g  N                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
) `! I1 ~4 [  ]% Q3 K% {/ R, O' D( t                }
                MSG msg;
- ^2 d7 u; S, T% D4 h8 U$ _, Z+ U                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
+ l  ]5 m1 }, T: R/ v                {
                        TranslateMessage( &msg );
/ F1 V3 H3 p5 _2 \+ X- G# Z                        DispatchMessage( &msg );
9 N1 W# s5 q2 |7 l& S/ J                }
& g5 k, p+ a) Y9 \0 Q& T. S                return m_bAsyncFindRunning;
        }

) H& e6 z) u0 r0 p3 G
5 c* }2 g7 s$ H; }) `        TRISTATE                        m_bUPnPDeviceConnected;
, }0 b( x; D$ L- B1 c$ x

& x2 A' d+ g2 r) T// Implementation
$ @5 Y) K. w7 t1 D        // API functions
$ Y- m# Y+ J& f* V: i        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
4 t+ E5 d! P, x# W0 c8 W        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
2 }' r' h8 c3 {        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
; L% V6 F6 h3 t+ T: e/ L        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

; f7 b1 `' d. `( e0 `, W
        TGetBestInterface                m_pfGetBestInterface;
) q2 Q7 I2 b$ g6 H+ o        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;


+ Z" ~1 b9 ?' j- d8 g: m        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
- @9 @- `4 I5 k                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
1 a6 M/ u1 T* D( O  P                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
2 \! H$ ]5 j; n4 E8 p

% c- i% O# K6 ]  w                        if ( FAILED( hr ) )
. r$ i) {# R4 ]6 @" Q                                return UPnPMessage( hr ), false;
1 a. I  ?8 b6 T. X

                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
) f8 a  i4 t. l        };
       
3 g9 ]/ O. ^; `! P3 B        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
- W) u- I$ X1 V6 E$ j        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
3 ~0 W6 k0 ]0 x  Q: b7 q. B! n+ {8 X        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
$ ^/ T4 Y9 M4 V/ k. e                LPCTSTR pszInArgString, CString& strResult);
# v9 _6 i! h+ {8 o, X        void        StopUPnPService();
7 g) |" d# S. ]. C. b2 c; `

9 Y* O0 S( S) x" b' y+ `- L        // Utility functions
1 D0 O) K  {/ x- q. u4 ]* n- P        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
' G, [5 b5 M! @9 ?0 W% W; o: F8 `$ q6 g        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
( A  m/ Z3 N: u: e4 j' Y        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
+ o' C' Q& x8 ~. _        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
1 @% V, d" j$ Y% p2 S, K) G

// Private members
$ C5 a8 f& F  {* Vprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
0 N; d, v. s/ o1 O* [& ]5 x        std::vector< ServicePointer > m_pServices;
2 L& e# Z" x, I2 v0 R        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;


        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        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;
  l; i9 g" ?$ Q  p# ~9 q. i        bool        m_bAsyncFindRunning;
$ U1 N" @1 I$ L        HMODULE m_hADVAPI32_DLL;
6 M+ \8 @* i+ _# O; s) K9 E: G        HMODULE        m_hIPHLPAPI_DLL;
, E) s. u* t6 i% t8 F; b" e        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
$ v% O4 V2 o9 Q6 [# t        bool        m_bDisableWANPPPSetup;

  y( s9 M+ Q2 S& q5 {6 r* j. k. v; w
};
2 B5 }9 j+ I" a& y  P
% m& @3 k) v( P
// DeviceFinder Callback
class CDeviceFinderCallback
  J3 n0 q1 s7 l! z" c4 L        : public IUPnPDeviceFinderCallback
3 f# o8 `. i+ O; R, G{
' }$ X. c7 s7 b! mpublic:
+ v/ y) ^5 `4 r1 N        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }

8 G8 L1 _$ i: w: b, H/ m
( x$ h7 n" o9 n( S" ]: Q+ X( q   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
+ b5 z4 C) ~6 y9 a   STDMETHODIMP_(ULONG) Release();

$ E) l" E0 e6 a; {
// implementation
private:
8 f7 Z2 ^* b  L( z' f1 z/ |/ `        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
, ~$ @6 }9 D8 ^2 A: N        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
  i, @7 L% S) r# N2 l. S2 `7 U8 r        HRESULT __stdcall SearchComplete(LONG nFindData);
( _4 |% w; c" m$ D# U  _1 G9 I5 D
7 s. N; ~7 C# o: w! [
private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
- ^( C9 `. i% ]" |# i};
8 S7 h) j; n$ T9 \5 H& [* e: a

// Service Callback
% b# l* N! T+ Q2 jclass CServiceCallback
        : public IUPnPServiceCallback
{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
/ k- ~8 W$ {1 z5 G                : m_instance( instance )
        { m_lRefCount = 0; }
   
; m/ u" }* c, I" T" _   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
, M/ c! D5 b! K% t( A
" Q  P6 |, ?0 M
5 w$ V  B$ \) Z7 N, `/ f  W// implementation
0 }" n9 B& y9 t& K) a) Xprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


- ^2 G. Y+ X% O4 Q" x' ]  Qprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
6 C( k% I2 x- V; R  i* U

/////////////////////////////////////////////////


5 w% B7 _' }, D# H使用时只需要使用抽象类的接口。
: z- V  s! P* L" p' ]& q' @/ Y. z5 nCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
3 t* @0 w/ K% N0 \# Q! i/ qCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
% ~0 z# r8 x5 l7 f# {6 GCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.