UPnP

天云上人

/*uPnP.h*/

1. 
   # N# ]  i4 g0 I% i. _+ m% p0 ]
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   ; X: |/ H* M0 Z- R' e- J
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   
9. {
   , E$ p7 d5 z, w  _
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    $ j9 i) }8 d5 [. e" Z
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    7 y: ~# k. J  r7 l+ T( n0 d* x
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    % {4 d; C" F& g: k4 S0 S$ R! l( w  j
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    $ x% T$ }9 U1 @* }# W+ z
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    7 T1 ^- }4 [/ b4 E
17. }   UPNPNAT_RETURN;
    ' T% O3 h3 H" v3 M
18. 
    
19. typedef   enum{
    7 U/ k& c* n9 p2 o0 k2 f; h, _: X% H
20. UNAT_TCP, //   TCP   Protocol
    . Q2 Y+ W4 D5 c; ]
21. UNAT_UDP //   UDP   Protocol
    4 l0 A1 n: ~( U" ?6 D2 j$ y. {
22. }   UPNPNAT_PROTOCOL;
    , \* c* z; Z  V2 B2 k
23. 
    ; I( V. o8 W7 _3 \
24. typedef   struct{
    ! G: B+ p0 V! ?8 z
25. WORD   internalPort; //   Port   mapping   internal   port
      }) B( p; m. N4 O% }9 g( F
26. WORD   externalPort; //   Port   mapping   external   port
    : t8 d4 M5 N# G
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
      ]5 @' {* X; w1 {: b2 n
29. }   UPNPNAT_MAPPING;
    # B5 B  P) N, n3 c# @
30. 
    
31. MyUPnP();
    ! S) @% ]) Q4 i3 V" n! B9 z
32. ~MyUPnP();
    
33. 
    9 t4 g3 O& P! k
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    6 B: |) }! S: t  O- A' m) s
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    ! S3 x& K' J3 M8 v" n: y, T
40. WORD GetLocalIP();
    0 d$ m9 |( {+ L. Y4 R" k9 F
41. bool IsLANIP(WORD   nIP);
    - f. L3 N, E  P3 n; S
42. 
    
43. protected:
    2 j! V6 b  X  K9 P0 `
44. void InitLocalIP();
    8 K6 k- q( l& b# z
45. void SetLastError(CString   error);
    
46. 
    7 [3 F. p9 R' t1 q8 o4 J8 D& @6 D( t
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();   }
    6 [- K  T2 S4 y4 Q2 A
52. 
    
53. bool Search(int   version=1);
    ' q0 N4 g8 O. L1 R
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;
    
62. CString m_description;
    * c2 ^/ Y; d) W8 b( ~
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    5 E" Z2 K5 A, Z2 X. P+ t
66. CString m_modelname;
    
67. int m_version;
    * K' h# G. C1 Y  \
68. 
    - d0 m" g. v! a# F$ L) _: b
69. private:
    1 l; G% `! s* o: O+ u1 ~2 q4 p
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    2 [( P- D  t/ n% d
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    $ o- G! F# j. I$ V* O( C; a) `
74. WORD m_uLocalIP;
    
75. 
    . }5 c8 P. @4 [" ^; \. V
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   ( X2 A7 P+ B) q* T
3. 
   + Z8 z9 A; f# ^+ O  z7 S# \% _
4. #include   "upnp.h "
   3 j7 w5 j. L" `1 v' t
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   2 U0 x% Q, F3 }& i' W" c
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   $ n9 Y9 d" _+ h4 a# e$ X9 y( Y
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    . n) l' f5 W/ X5 L4 s' T+ ?
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    ; m) f. U# X/ N8 @* A5 T& P
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    6 Y1 H$ r' g4 I+ K8 C
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    ' T1 y; U- B2 l7 _0 {
19. 
    
20. s.Format(_T( "%d "),   i);
    ; O0 U7 {; W5 X; s
21. 
    9 O5 q+ v! T* z# H
22. return   s;
    & k" n) I4 [, _9 J: Z& |, {# j8 l
23. }
    
24. 
    8 L, P4 L8 u7 \' R0 p' r
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    & F& j' j. f1 G1 o7 ?5 A$ |
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    0 ?  O# Z& t$ _
28. }
    ' b1 o- [4 }/ d. u& P
29. 
    & q' H: W8 \2 @. t# |4 a7 D* E
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
      @3 k! x+ Z. @$ E! _- x" m
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    2 x: p7 e( G9 _7 i9 W* Q
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    . r5 N+ @1 Q3 G: ]
37. char   buffer[10240];
    ) B4 M( r% I: U" y1 N
38. 
    2 h8 c1 Q# B9 P6 ^. ~6 n& `
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    . w7 i" u4 n" P5 g
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    2 m. O  n! Y+ m8 T0 s9 @' ]" U
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    ! I7 J& U: x' E5 r  e, R/ [
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    5 p* ?4 W" _9 ~
53. Sleep(20);
    0 \) F3 {' n& ~
54. int   n   =   send(s,   buffer,   length,   0);
    , W# w0 ?  {' u, T; e- j% u
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    ! q* y2 W1 D6 r& G
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    . e: _2 p& i' `5 d: x
59. if   (!rlen)   return   false;
    ; m: O0 {0 F$ x8 b& \8 K
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    : X; }8 p5 {8 ~" Z) d
62. 
    
63. return   true;
    - K; Q: T' c, ^- a& h! b4 Q
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    . n, U, Q) j( E
67. {
    
68. char   buffer[10240];
    ' b  e" g7 b# Q: R7 j( k
69. 
    
70. const   CStringA   sa(request);
    ; ~* u8 h. e9 @8 R+ J
71. int   length   =   sa.GetLength();
    $ Y% J* F1 g6 q; H$ ^7 h
72. strcpy(buffer,   (const   char*)sa);
    # l% g0 Q2 [* C. O
73. 
    
74. struct   sockaddr_in   sockaddr;
    ) @  w* j( e+ ]: k
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    / K7 R9 \( r) q5 ^0 s8 O" v  `2 [
77. sockaddr.sin_port   =   htons(port);
    5 S" r4 r' T2 r" Y+ Q, Z
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    4 k3 w8 \8 a+ Y
79. 
    ' {3 p2 G2 U3 ?  u$ u/ e
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    5 F1 t4 c. h" L2 q7 x# S& b' |0 G  P6 I
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    " G9 ]8 w1 ~/ z, v  Q3 i6 `# e  l
88. 
    9 ^9 V+ F. [  J3 V' ]( f
89. result   =   response;
    
90. result.Delete(0,   pos);
      Y# T7 }  h) Z% v$ ?# l* a
91. 
    $ U0 f" Z) s) I' _0 B
92. pos   =   0;
    4 f  p: O4 b/ W' ^$ ]1 \
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    6 v$ S& K4 l7 \2 {/ m
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    ; U/ R5 K2 b0 D. A
97. }
    1 E: T7 g2 L6 f- ?: ~6 V
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    & c, {/ G. M+ y+ e& M  v2 H
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     2 b1 n! X2 {$ s$ n
103. CString   property;
     
104. 
     7 s2 F1 U9 g% u
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     " t9 g- H' S4 N
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     3 w% ~: o4 {* v
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     5 k* T: ~5 X- L1 n, [* J: z
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     , Y  G8 {. s- N: R9 y+ G4 _: M% M$ v
116. {
     ) j6 l% s! j8 ^9 D0 J
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     0 Y* J7 t, O1 S4 s0 A. L5 z
122. {
     , O  C$ u3 @2 p
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     # ]5 p; ?6 ]; D$ C; a: y9 O. P
125. while(pos){
     ) i$ Z4 q+ _) `7 x& |
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     5 B; a1 I7 w# t( W
129. 
     % s# y, ~/ A1 T& a, R) P
130. m_Mappings.RemoveAll();
     & N, y4 o& Q( k0 y/ y% C
131. }
     
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     & A3 L, a8 c/ a: Q6 S5 u* c, h5 k
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     4 u' n/ ~" U& y0 E
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     6 Q. Y+ h8 O: h, o& c& C$ O
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     % u2 o+ L* o  Q5 d0 D, R! E
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     7 }* M" e3 X# m
149. 
     
150. int   rlen   =   0;
     
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);
     " N: I$ W9 D1 [3 p; c0 ?
155. CString   request;
     5 p) S$ \! B3 n1 ]7 y
156. request.Format(_T( "M-SEARCH   *   HTTP/1.1\r\nHOST:   239.255.255.250:1900\r\nMAN:   \ "ssdp:discover\ "\r\nMX:   %d\r\nST:   %s\r\n\r\n "),
     
157. 6,   m_name);
     4 Q  g4 M$ |$ s
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     - Y( J  g& i, ?3 E/ E0 k1 Y
159. }
     
160. }
     & e2 a/ O7 W2 R3 a! t
161. 
     
162. Sleep(10);
     
163. 
     : S$ a" b5 u8 r, T* d$ u" }
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     * d' G/ P) R, c  J& [
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     ( l! s4 F% l. u/ s8 m- T
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     " c- j& y" B$ [
171. if   (!parseHTTPResponse(response,   result))   return   false;
     3 v1 n! N8 }, M+ }* _7 Z
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     % W% y. s" _$ H3 L
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     ! o: W1 m9 o9 u# x
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     - T; ~& j- u/ R! y3 J* @5 f5 f2 \, D# g
179. CString   name   =   line.Mid(0,   9);
     5 Q$ T/ r2 p' V3 o
180. name.MakeUpper();
     4 u7 v+ o. D! _/ r) q4 h& _
181. if   (name   ==   _T( "LOCATION: "))   {
     % D' \) p% A1 L6 P& {( D6 t' G  t: r! C
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     9 y: F$ x. f; ~# {
186. }
     
187. }
     
188. }
     
189. }
     
190. }
     , u7 Y) W% Q! a; n
191. closesocket(s);
     ! ~) ]9 i  g4 e6 H: c$ t1 e
192. 
     . M7 P( M+ D+ q! l; i
193. return   false;
     
194. }
     + a2 t: v4 l% p

复制代码

天云上人

以下有关upnp的接口来自emule，
# g# C" `+ ]+ A

- N/ \; [6 O8 J" K///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
# F( g2 Y6 r: m& ~7 v: o' t0 A

! R( H" |4 T' r#pragma once
6 |* m$ F* z5 |) Q3 e+ g#include <exception>
: P+ ?8 L# N  D, ?, x3 ^) b& \4 [, m

  enum TRISTATE{
        TRIS_FALSE,
; P/ R+ _& W. q. [, V! I9 o9 n        TRIS_UNKNOWN,
$ Y- o9 p4 |6 r2 G8 D: P        TRIS_TRUE
};
' z/ r6 O2 i. e9 C
( w" n% {. S) c8 p8 u& ]3 n2 c
enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
  T& @6 w" l+ \, w8 {- o7 [5 a        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};
  J* d- M4 B+ Q+ x  h



class CUPnPImpl
{
* x5 A- S% b7 A9 V: u" ~  z9 }% fpublic:
        CUPnPImpl();
1 Z- r0 q3 _: f$ {6 K- `# P        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
+ s" r. t4 ]) }- d) Z3 Y* m* r                UPNP_FAILED,
! t+ O) i" I+ m4 x+ K+ i* P                UPNP_TIMEOUT
  b" k# b4 A! }  j0 n: @& ^        };

$ z9 ~6 }& Q# C( u
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
/ X3 d" k0 |* ]5 b        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
# Q/ V5 q0 Q( ]4 v: M        virtual void        DeletePorts() = 0;
$ U) R: W0 y" `9 {5 Z3 w9 k, H) o1 @        virtual bool        IsReady() = 0;
/ H- I8 h3 Y8 |2 p. ~% s4 F        virtual int                GetImplementationID() = 0;
: O$ S( z/ z3 @2 u0 j6 }       
$ s3 {* T, E7 T3 \# Q9 }( L5 l6 m        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

0 ], M. l8 d8 T
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
0 J: E7 m7 ^* P        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
8 I" y+ g7 D, b3 h        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
) Q+ D, W. v: n2 Y% j

// Implementation
protected:
/ E* p- f) B& W8 J. s9 k" x        volatile TRISTATE        m_bUPnPPortsForwarded;
8 ?9 F3 B1 L5 w. Z0 R        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
/ ]# n6 ]3 ~1 h) C$ ~
, X; s7 v' b* d/ m
private:
1 l; H: o% L/ q# `  D; I        HWND        m_hResultMessageWindow;
9 B3 V7 k( T$ k( Y, s        UINT        m_nResultMessageID;
( [5 X  u+ B. T
# ~( V3 ~+ N+ T0 Y
};

' S2 M: Z; F* K) q, H3 h
// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
- x) d: T7 Y# u+ ?{
public:
- ?) r/ E: b- w  y$ c: c4 S        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
" A. f4 G6 w9 u+ E        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
  ~. E- c, M" w# P0 X& e2 b

/////////////////////////////////////
/ \$ i: Y/ L! y//下面是使用windows操作系统自带的UPNP功能的子类

! j0 Q% \& O! z8 u. P4 b0 L4 c- e
7 l/ x# w2 L' U8 g, A& s* }' V, g#pragma once
3 j9 j9 R/ @9 y. q+ P#pragma warning( disable: 4355 )
5 S6 A/ w: M/ O; o9 s- k7 f
$ p! J$ W1 b; z5 ~& I- C
#include "UPnPImpl.h"
#include <upnp.h>
! G$ s9 A9 p' M/ l0 _2 ^( B8 K9 Z#include <iphlpapi.h>
  `1 s7 C* K$ ?0 i+ U" M#include <comdef.h>
4 g6 ]$ m  l; q/ \2 K: A( D  v#include <winsvc.h>
4 ~5 J! j! K" O9 H" s! p; G  ~
2 N: W( [8 a  }
) t0 y: ?6 {& {( f7 L$ F#include <vector>
#include <exception>
' e- q/ H( z5 _$ t4 R#include <functional>




" W+ v6 \" v; O7 ntypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
6 c% |7 q- l; X9 v0 X. Ttypedef _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;
3 F6 \- x0 R/ i) p- ]0 Ttypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
+ U" l2 ]% _/ }
" N3 u/ P5 N" c4 E4 n9 J( L2 A
typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);
' h( i7 o; N& B0 V' c
7 K  f* X, _$ }2 O; n# R) N  H0 p
typedef DWORD (WINAPI* TGetIpAddrTable) (
/ P  x$ X+ v2 g" W9 z% K  PMIB_IPADDRTABLE pIpAddrTable,
' |. i) \( U1 `9 D: `  PULONG pdwSize,
- {! K8 U/ L0 `. c/ W- \  `) l  BOOL bOrder
& O( M$ N! w$ H);


: i5 S$ a& ]% a# ^* Ttypedef DWORD (WINAPI* TGetIfEntry) (
0 O3 B. v0 o7 A( F4 b  PMIB_IFROW pIfRow
9 x  S4 A. D8 E. V);


CString translateUPnPResult(HRESULT hr);
: Q0 U/ j% w4 s; A$ U! {6 ~5 @HRESULT UPnPMessage(HRESULT hr);
* h; V7 m9 t; o3 O# H" Q

) t3 [8 G# r' v; i5 S: |class CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
! @" i. l( ]' `  p+ w# r% C        friend class CServiceCallback;
// Construction
public:
2 ?2 b) o& N: g, m+ q5 `( D/ X0 i        virtual ~CUPnPImplWinServ();
% i& B5 M- g0 v6 V: ^        CUPnPImplWinServ();

; H  V& a: R) O
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
# q: n9 |- E+ j" D- v: B        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
* K: c$ c( Q7 R3 k( _7 J1 L
6 W+ Y% Y8 x( B  I( f
        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
; K& P5 f& p$ r& d6 f6 a" o2 Z, {        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
4 h# |, ]* j+ @2 R# ]        virtual bool        CheckAndRefresh()                                                                                { return false; };
7 I$ A5 c; U% e4 ^8 e  K( R

! p' L* w$ C+ ]: g3 {' v0 {protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
) ~! ~1 i5 p) l- i) K, x0 a( d, V' J        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();

, O5 E2 N  S+ H* z& Z
: ^9 O4 l# t1 p4 A        inline bool IsAsyncFindRunning()
        {
* e! W4 R  b, Y                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
; E6 H; y3 p, ^' e3 W! z; j                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
* p$ E' u4 s0 d& F                        m_bAsyncFindRunning = false;
                }
& K6 y9 V) d  z/ R, B$ X, T                MSG msg;
0 I  W+ z5 p1 R: e' u; Z. L% W                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
8 X! l6 M+ O9 {4 p                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
2 M& H) p8 N" G- v& C5 D5 l                return m_bAsyncFindRunning;
* Z) R: c8 D5 s' ~# y/ G: R        }

$ ~' ]* v  a. |+ c/ B
        TRISTATE                        m_bUPnPDeviceConnected;
1 ?( N6 }* l  ^0 \
+ a) z. v& Y4 M$ g2 [9 C
// Implementation
9 d. t5 S! C( e& C& D        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
0 {5 X! t' ?) J- m! e3 M8 a        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
" c& R7 O! ~! b        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
; U- {3 }9 F) T7 E+ x        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


' U- r" |/ ^& i4 G        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
- T# y; A1 y- @        TGetIfEntry                                m_pfGetIfEntry;
4 W! ^( i. I1 }& V8 L
& G# J3 c; |6 S
        static FinderPointer CreateFinderInstance();
# w: B/ u0 i* R% o4 [/ l6 y5 \        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
' h) w3 |$ g6 P8 k                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
! K- H% K- F3 d- P6 `
; v% L9 `9 L2 }4 J9 `
                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
+ Y6 {& p. I4 u% \. h

                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
5 ]9 b0 ~* u* _9 c% O* t; U- G* U% a                CComBSTR m_udn;
        };
) d" J0 v$ S7 i+ ?' q/ T: e       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
2 N5 \1 y* F' x; z# P( y        void        StartPortMapping();
8 W' W1 {3 S0 v  V- W+ T- w4 ]- G, [- M        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,
; o$ I' ?5 R- ?2 u0 L! h& S  w& w                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
* e  ]5 M; o5 y7 e" u7 p: r2 U

$ u+ F& w1 |' c7 w        // Utility functions
: J' _+ q/ h! N; q4 g        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
  i7 M! }) Z# e% ?. Q        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
6 I0 E! z, I3 U- W: b6 w1 A        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
1 F- `7 T& l- I- D( N        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
3 U* T% o( w$ r" ~* \        CString        GetLocalRoutableIP(ServicePointer pService);
! \: e! n: F: R& Q! N+ d# i
* f5 Q7 L3 y* y( D! ~
, X- e4 T) }. B! X( K/ D8 l// Private members
: r3 I1 ~& C) e" I4 |private:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
$ R7 H& q0 s& Z* Z! m        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
) e! T5 @, d  h6 e+ b

! I2 G" ~1 {, s4 o* J% Y/ L        LONG        m_nAsyncFindHandle;
; Y1 K' S( o  l) @        bool        m_bCOM;
        bool        m_bPortIsFree;
- u5 A) d" }' o$ Q1 Q4 B6 N( T        CString m_sLocalIP;
, I3 ?4 W  V- o& e9 \7 P        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
$ j" T& }. [, [+ B8 ^8 K        bool        m_bInited;
  ?4 d: R* M% ~$ Q. }6 Z        bool        m_bAsyncFindRunning;
1 Z, x9 ]. @# Y7 Y! [5 \        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
4 [6 R9 k$ |9 _        bool        m_bSecondTry;
6 h0 E" b& L" ~( F0 q' k7 }; K- d; n        bool        m_bServiceStartedByEmule;
( A: v- E( \) E+ I+ o        bool        m_bDisableWANIPSetup;
' |: R2 s4 R3 N  Y: z$ G) V; V) b        bool        m_bDisableWANPPPSetup;

% d1 t, G) `9 r
};


// DeviceFinder Callback
" \0 ^6 r4 y0 Jclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
public:
& N' k  p, O  w7 d# N        CDeviceFinderCallback(CUPnPImplWinServ& instance)
+ D5 P, c% i( T3 K6 {( s# W+ M# x                : m_instance( instance )
        { m_lRefCount = 0; }
/ ?% L: e  c  b7 J- _  y
0 u  A) J* t# J! X5 D
9 j8 O6 |" v2 O& }# V. W9 Y( ]   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
- _6 \3 Y1 G3 \% c* J4 K! ~& C   STDMETHODIMP_(ULONG) AddRef();
# @+ A# A1 k$ n, D% Z! }   STDMETHODIMP_(ULONG) Release();
$ g4 l8 }3 O% z' L5 j! M

7 o9 T' D. @* e! I6 Z// implementation
6 F( M  h$ K, r* |$ n) i. a$ Qprivate:
3 u+ u) U: b5 \9 @4 V        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
. T' _1 Z: r- c* e        HRESULT __stdcall SearchComplete(LONG nFindData);
( V$ c1 O0 A) s0 S  O& X; ^

private:
        CUPnPImplWinServ& m_instance;
; V# i5 h$ n* U- ^$ ^        LONG m_lRefCount;
};


// Service Callback
0 H3 E( ~( M( jclass CServiceCallback
+ d+ Y& J4 l% n" j        : public IUPnPServiceCallback
{
4 Y; @2 S1 ~- X# A8 d5 I' Wpublic:
        CServiceCallback(CUPnPImplWinServ& instance)
+ {( s: {: ~! K9 T8 v                : m_instance( instance )
        { m_lRefCount = 0; }
) W: s: E5 U- T- W; P& [" K: l. F   
" q: v: |. V% ^, E' [+ z/ @   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
$ v( q- [) T0 T! [   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
# e: b% T7 k! \0 f
+ W& |4 A1 b' a' O
5 f3 l& D% b' ~8 D// implementation
4 \" A8 F, ^8 {" Q* f  B7 rprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
6 b" K2 n% }( N9 Y' V; Q- d

private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
  `, g5 r* B) f" u};


/////////////////////////////////////////////////
0 g+ q0 N# J9 a3 U. ^" P7 d/ h$ I: P/ S

3 c) b# G( v: R% t8 k8 _使用时只需要使用抽象类的接口。
( S; k7 j# I1 p& `! X4 t+ aCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
' _& D/ d( ?# e+ L8 h; f& WCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
9 [1 _4 R5 S, F8 r3 n! tCUPnPImpl::StopAsyncFind停止设备查找.
! P' Y3 r0 L0 D2 u6 Q0 ZCUPnPImpl::DeletePorts删除端口映射.