UPnP

天云上人

/*uPnP.h*/

1. 
   % @; f  u4 }6 h3 F2 N7 {( R
2. #ifndef   MYUPNP_H_
   9 _0 L  y9 q$ \
3. 
   : m0 v( O- K2 f7 {1 f& D7 w, H% @
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   " r2 q: T" s3 ?: K, x
7. 
   ( q* u: M7 b  w0 B6 i- s
8. class   MyUPnP
   
9. {
   % i8 q6 }3 y- @8 y1 M6 ~
10. public:
    - x7 @" e" h1 @$ S
11. typedef   enum{
    + k8 T$ f* Y- f' k- R+ \
12. UNAT_OK, //   Successfull
    1 r/ e$ H# y! t* f% t
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
    # {: O% F" d4 n5 K7 f
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
    5 x9 D: l9 G( c! ?' w* p8 t1 s+ W
17. }   UPNPNAT_RETURN;
    ! g/ E# ~7 A; i8 e
18. 
    
19. typedef   enum{
    " K9 b- s7 I" y
20. UNAT_TCP, //   TCP   Protocol
    % |- Z4 }7 L3 X8 G! V, y
21. UNAT_UDP //   UDP   Protocol
    5 o% v3 M5 R$ ~0 P
22. }   UPNPNAT_PROTOCOL;
    0 [! c7 I. q- h: f! I8 h- n
23. 
    & j$ [- D1 S3 w' P' V
24. typedef   struct{
    , r3 T/ b+ j$ n0 r# D7 M, n2 c
25. WORD   internalPort; //   Port   mapping   internal   port
      x( X# G  K+ K* P
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    ) J  f6 w2 h$ H4 R1 N" W( @1 e
30. 
    9 j+ [3 @5 b* D! |: @% P$ K) w
31. MyUPnP();
    
32. ~MyUPnP();
    + A: n, L7 y+ a. o8 l  c
33. 
    4 K# r7 c8 {: t% ]. y
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    5 R0 E3 s, f1 ~
36. void   clearNATPortMapping();
    
37. 
    $ z+ W# x- m$ W. e! Z( ], H
38. CString GetLastError();
    - v% l- l( `9 O1 x' S; L
39. CString GetLocalIPStr();
    ; \. N1 r& n9 k0 S- J( \+ w
40. WORD GetLocalIP();
    : u+ r( [6 h7 H' q4 P$ d) F
41. bool IsLANIP(WORD   nIP);
    1 J; `! _8 s: _/ s- C
42. 
    
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    # q* Z7 u6 y! v! W; Y
46. 
    ; A' U- @  g: @3 N, r& J6 G
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);
    " r3 h, n4 a4 P8 S1 U
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
      X, h  [9 ~% e5 h) t% y
53. bool Search(int   version=1);
    / ?6 V, Y: i4 S; @" @. p, Z' `! m
54. bool GetDescription();
    ) {$ ?. q7 e+ l2 D. k7 [' \& Q6 n. H
55. CString GetProperty(const   CString&   name,   CString&   response);
    . O7 S% `" ^3 b2 w1 A# M; P& V
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    , K7 P* n2 |1 X/ r3 i2 o# U2 \, X) V
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    4 Z+ }3 M7 m, H
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    ; f1 {# D$ m2 g
61. CString m_name;
    
62. CString m_description;
    % y  N  u/ _$ ?; U3 c
63. CString m_baseurl;
    4 z4 q4 P5 D7 X. Z' x- C! \
64. CString m_controlurl;
    . g5 k  E* ~7 M: F; w! R! r& H
65. CString m_friendlyname;
    
66. CString m_modelname;
    3 C; o2 @6 E$ v
67. int m_version;
    
68. 
    . V4 |: R4 Y2 @7 q, J
69. private:
    , }6 u) m: }: [, |. _
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    # N- |: i" ]6 k' `
71. 
      F2 W+ r: |+ n- T( z/ t
72. CString m_slocalIP;
    ) f9 a+ L) ^4 x  l
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    2 C; \: G1 [, ^1 z
76. bool isSearched;
    ( [1 b2 b! B, r! s# B2 W
77. };
    5 p( Q$ x% s. i1 C+ P
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   2 n- R4 ~. j# w5 z9 b
2. #include   "stdafx.h "
     D: }0 S& z0 ^$ S5 i$ _3 D9 A, B5 ]& s
3. 
   0 G8 c  d" W3 x: ]# E
4. #include   "upnp.h "
   ; E1 d/ i& P9 b4 U. g* i! p
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   6 {( S* i- F4 Y" ]: S
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    5 T  S# k2 ]; {" q# h
11. 
    9 J7 L+ k: D3 t. ^$ \9 {) ~# Y
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    : @! r# Y3 ~2 ]: o0 b# h" o: j. @
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    5 q% c5 i$ B; O% S
17. {
    
18. CString   s;
    ' q- M5 F$ G' G0 S) O% D
19. 
    % W" c+ G; J% x0 C  w7 o- U5 L0 }9 C
20. s.Format(_T( "%d "),   i);
    , h" G) [% p3 _( g: F7 m' B8 \
21. 
      I! S" I/ Y. T. B1 x
22. return   s;
    1 c; Y/ N7 R- D: t
23. }
    0 S' K' ]- h7 ~7 n; w. Y# Z. \/ H+ b
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    / I+ i& T! Z8 `5 Z( s2 o
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    $ x0 |3 A( p$ A
30. const   CString   GetArgString(const   CString&   name,   int   value)
    , I( v" I3 C. ^5 s* a/ p+ Q% X3 o
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
      F: W% O$ |- n2 b
33. }
    4 F) n) g) K. A7 i7 n0 `
34. 
    $ Q/ q. v0 u/ ^4 O: H: D
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    3 K5 a( O5 q8 y! L% |9 x
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    ! k/ }, ?  ~, e! i# O( F% n
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    ; E* V- [, R( {! V0 W" a
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    5 U; X" S  i0 ?" ^
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    4 r1 C* H" \# {9 t" q
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    " i5 G. B2 N, b  l, x; ^6 l
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    ( Q) L2 J; ^' S3 X) S5 r
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    1 L, V- g$ `/ g7 x1 r
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    ) N* o  ^2 i6 |3 P5 w( I* I% N) i9 ?
59. if   (!rlen)   return   false;
    . D9 A7 A7 K. }9 z, _4 c
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    
64. }
    & X$ E2 @( \. C( {' x
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    
69. 
    : h9 @' r, x; M( }  O
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    3 E- H7 R9 y3 F) o# Q, D
74. struct   sockaddr_in   sockaddr;
    + j" }; q  [8 C+ G  |
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    ( D' W- ^% T8 n$ T
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    6 @% _- K& s9 V# M) a, R
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ; c4 e& l! P7 B6 A) x1 ~! |
79. 
    ( s  w9 d5 K$ d7 C& |
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ) C# h, }* ~- \( m
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    9 o& W0 ^. s' H* e6 @4 \& ]! t
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    ; z/ |, b) W( E( ~8 r& P  W
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    " g( A" k  S" O- L6 ~
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    1 u* Q6 D# X1 G1 q$ A- a  d+ n
100. {
       m& {4 t1 ]1 B( {1 r
101. CString   startTag   =   ' < '   +   name   +   '> ';
     # [) C0 W( |8 g% z% x/ d" j1 _9 E% M
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     7 G, _+ w% ~9 s7 F& y  l
104. 
     % l8 p/ n$ p. B/ |: s+ r
105. int   posStart   =   all.Find(startTag);
     * l% @; ?$ e4 D  S9 \
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     ' l+ _6 T+ P' W+ w) F
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     3 ^5 U1 u! o$ n9 A% L
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     8 Z" H  d. P- D0 d! P1 o
114. MyUPnP::MyUPnP()
     7 s( C3 O) A' X3 `3 C; k
115. :   m_version(1)
     8 X3 e  ^% X; p7 N; l) v3 A
116. {
     ; d9 R/ R) q: U
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     
120. 
     ; |9 R" _" A) q
121. MyUPnP::~MyUPnP()
     : X" ]% m4 s( C& F1 k+ M+ N2 v
122. {
     3 I0 s& L3 X% {( g3 c- |) z' N* P# B
123. UPNPNAT_MAPPING   search;
     0 f$ [( n0 j4 V  V$ B' x
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     # q$ N5 k, Q# K
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     8 K" ~9 |- k+ `$ w& m
128. }
     
129. 
     * [% K% B9 K: d/ x( e. M* k
130. m_Mappings.RemoveAll();
     ' {; o# A8 c3 u' r
131. }
     
132. 
     % X8 v( J& c1 i( w
133. 
     " z2 Y- Q  G$ L. Y' b: a2 F
134. bool   MyUPnP::InternalSearch(int   version)
     0 Q( P' e3 F( Y
135. {
     ) \: f# h1 q8 A2 y4 U/ a  Y4 l( g" g
136. if(version <=0)version   =   1;
     ; B6 s  [& q, B1 I, ^& {
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     8 @' B1 o, h2 {
140. CString   devices[][2]   =   {
     5 g$ @# I3 O, U8 f& w
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     / Y# I$ D" V! P+ D
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     ( i/ e  j; w( f  o
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     7 B: @; t- e( G* v4 p- S; C0 L
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     % K# Y% ]+ W: b5 ?  B
150. int   rlen   =   0;
     8 g1 x% H6 a& @4 t8 F' s
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     7 T! L& T0 X6 x2 Q1 ~' n, ~! S
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     * |; }% Y7 m. [( X2 G$ w
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     7 \$ h8 O! `8 i4 B
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 "),
     # g2 A: O/ b0 Z) t
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     5 I/ Q6 R& o4 w9 n# m9 l
159. }
     ! ~0 q! {6 Q2 C5 Y4 ~
160. }
     
161. 
     5 [! F1 H5 P  o6 A5 H
162. Sleep(10);
     / b3 Y- M" j; d5 s/ n* ~
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     4 R# l, X7 u7 v
166. if   (rlen   <=   0)   continue;
     3 B! X" Y* A! i/ {6 t9 E
167. closesocket(s);
     
168. 
     8 t6 l1 M4 e3 _: _; p$ _% G5 s
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     # Y2 U( @  f' N/ M) a! f
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     1 \: D5 P1 u( v  @8 M0 ^( D  \
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     3 v. X- Q. m' {+ ^
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     ) q8 F! Z$ N9 h: E9 H3 u
175. if   (result.Find(m_name)   > =   0)   {
     ) Q  G1 l; p/ f9 y# A: R7 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: "))   {
     ( _8 U3 d* o7 C' B9 c$ b, s
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     0 j  T' ]( F4 a$ f" x2 t
186. }
     ( E3 N& s& N3 {: m- A- q
187. }
     8 G2 H* D2 q9 J! u. r  \
188. }
     
189. }
     5 q( d' z  t* O) p" L, L
190. }
     ' Z# d, J" I, g7 S- p0 B
191. closesocket(s);
     
192. 
     % J+ R" K2 S5 H& W) Y4 [! y
193. return   false;
     
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
/ v: _6 Q2 D( M. c. T+ ?# Z% d
7 U  s3 _' X: |( ]
///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

' ]; B3 T" w  t2 i5 Q3 O
#pragma once
#include <exception>
1 n2 d5 n- [6 m9 ?. o* U1 e
% k$ R  I7 f' M7 u0 T. K* W; x6 W# W
( z: t& y0 f& o% `! y$ }  enum TRISTATE{
' S& [" U! i. T- R        TRIS_FALSE,
. G) s% t' ^$ }1 o2 o7 N        TRIS_UNKNOWN,
; o4 S; q) B+ ]  O        TRIS_TRUE
};


' n: \! u" @8 T" T9 Y) B2 ?( Benum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
5 e" C) G7 ]  U$ I& N; G( B. A$ U        UPNP_IMPL_MINIUPNPLIB,
2 E+ ~7 V( q/ Y) q& J4 f( W        UPNP_IMPL_NONE /*last*/
};


) K5 R" R* \, y* H; i' g( ~! w  y3 y

% W0 ], D1 n! H/ T0 Vclass CUPnPImpl
{
public:
        CUPnPImpl();
0 v  ?( z! r; Y        virtual ~CUPnPImpl();
( ?: u( `# o) l/ E& ^* p. I        struct UPnPError : std::exception {};
$ \, w/ W( b3 f# p. L  M, i        enum {
                UPNP_OK,
                UPNP_FAILED,
# I; [* \* j& E* k$ C2 E# Z% y                UPNP_TIMEOUT
4 G8 a7 q2 U+ T% ^3 o  D4 E/ |        };
, X9 s1 w# L+ H" S2 v& C- }4 h
. q9 i8 [  t5 p! o7 e, ^
7 V0 z1 j1 j3 r5 n4 l, \        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
- p( d% q# T8 k. T. g% L, g$ z5 L        virtual bool        CheckAndRefresh() = 0;
7 O) Y6 u8 K; t: f        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
7 ^, R, [, R9 N, S- @! O  M        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
# [+ i, {4 D8 w* I% I- @        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

. A. Q# }6 A* \+ f' n
' }5 F6 y1 R6 h; l8 w$ S/ ?        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
" }; j. f9 `% T6 z5 r: N        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


// Implementation
4 O3 u. S, Z; X' z) iprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
" i' n6 m" e- J' f9 r        uint16                                m_nUDPPort;
; _: ]1 a% f2 g0 {$ a& j        uint16                                m_nTCPPort;
6 L6 I0 M8 [- I0 v$ t        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;

! W& |  ~! n1 R9 D, x
; }$ R+ g: K: ]* r% Y4 M$ Bprivate:
        HWND        m_hResultMessageWindow;
5 y  c3 }5 o0 A/ E7 x$ Z3 _7 [        UINT        m_nResultMessageID;
" J6 L! M4 n/ P5 H9 U  y

( g% \5 I/ C+ Q7 S};
: b+ Y4 b. }0 F1 K+ Q. W" ~. X) n  U
# Q$ H# ?" T# ?: V4 E% G7 [: o% O
- p" p, E7 W4 u  C: T// Dummy Implementation to be used when no other implementation is available
; N$ R# @& L9 f3 U2 M$ h; dclass CUPnPImplNone: public CUPnPImpl
{
1 ]! P3 E: S7 a! x# U3 d8 i& Z; Zpublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
) o/ b0 X% [& k5 y        virtual void        StopAsyncFind()                                                                                        { }
' ^9 H. B8 r2 [  j3 U& ?5 h2 T* F        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
" Z3 U; ~, E: }) }        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
# }) b+ l; s, J};
( z4 v! ]: B5 {+ g' @+ F
# V( S8 n; N" R% F) ^- e  @
3 B, o9 g8 n# V9 t1 B# R/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
1 \, E4 }" O( G6 u. C

% d& `0 e* s  z$ x8 u7 J% @. j#pragma once
#pragma warning( disable: 4355 )


' P2 R4 f, N8 M. l% n#include "UPnPImpl.h"
/ e1 {, C2 c# p- \#include <upnp.h>
  d; k8 j8 K% c6 d, _6 r#include <iphlpapi.h>
* E2 a" N  L6 t- L8 l2 @- X" s: j#include <comdef.h>
1 z5 q4 G& `! x, q#include <winsvc.h>
( u% A. v: c7 w8 ]$ |0 _

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

$ t3 x+ u4 f- }7 ]% V2 d0 A4 w


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;
3 `7 ]/ e6 ]: e$ d6 P3 ztypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
3 w7 f; @1 D, A  ktypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
: I' K/ f5 [* h* r+ n) j" t4 I& T
4 O4 ?1 @) R9 {8 s- B# T. Z
typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
& B* U, `6 k, R* t! |);


, W. A( u& ^$ Jtypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
: W7 w5 L. @$ Z$ ^  PULONG pdwSize,
  BOOL bOrder
);
. b8 Z3 u, p- r4 ?' R, X. r
* D0 M5 S6 L, g
typedef DWORD (WINAPI* TGetIfEntry) (
  J4 y' \4 S$ V- ~+ J0 x) d  PMIB_IFROW pIfRow
' |" p# ]" N) c# p);
! n- @, B1 ]. p9 d" i( e, ^6 f! S0 A

5 l* K% Y* @" x2 d6 eCString translateUPnPResult(HRESULT hr);
4 j, p, k8 A7 `6 |1 k# ], RHRESULT UPnPMessage(HRESULT hr);


class CUPnPImplWinServ: public CUPnPImpl
. k2 s5 V& l/ C) h{
        friend class CDeviceFinderCallback;
# G1 c/ M2 _  z9 Y, a2 S  Z5 G        friend class CServiceCallback;
// Construction
& S8 X( T4 T$ p( I- u' Tpublic:
( h* U# b4 B+ h& e2 j/ Y( c/ f        virtual ~CUPnPImplWinServ();
8 ^$ y6 b; X4 l        CUPnPImplWinServ();
7 o8 \/ m1 D" S. Y  F* Y

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
6 @& w- p" _! `6 A% @7 N        virtual bool        IsReady();
1 o! Z1 L4 l! K& c9 f4 l( `! B        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)
        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };

7 L0 _9 W2 F+ x4 r+ J6 U/ _
9 q3 I* G, p& U, l  |3 ]protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
% `7 S' m7 |! M, D( |        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
1 Q7 S5 p0 G; f  t9 n        void        RemoveDevice(CComBSTR bsUDN);
: i' r* x6 b7 m# n; I: T4 r* l        bool        OnSearchComplete();
" I8 E# a  M( p. J) Z        void        Init();

  S6 x% }! q3 k9 u! K0 B  Q, B
7 y/ \6 }5 X9 R, ?6 K( k" T& u5 g        inline bool IsAsyncFindRunning()
% i' K3 F7 P) j* ?/ N$ U        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
7 x- p- `7 `& z2 M; Y' Y/ o* b/ q                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
                MSG msg;
' L5 X! J: I$ |5 @# e$ g! y3 X                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
7 S0 t( {) m2 g" z" C                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
2 F5 |& M& y0 m, l                }
                return m_bAsyncFindRunning;
- r% X4 P8 q  K2 ~        }
% `  n  ]" P0 o5 a, Y
' D9 Q; I4 N& j; f2 F! b! ?5 ^
        TRISTATE                        m_bUPnPDeviceConnected;
5 s6 P; j1 S2 P+ ~( Y- U

// Implementation
; N& m7 c$ c1 R3 S# h) l& U        // API functions
: I" w4 \! `/ R0 j        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
' u' w  b: b) m- E- C! ^$ Z4 s: n        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
' o2 f6 f5 f  s( U+ b        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
; z& F7 X2 k  E0 X3 h4 f$ M& U        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;


        static FinderPointer CreateFinderInstance();
6 ~0 \( N9 k3 L) T        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
7 _/ Y7 t) {) c9 t* ?' t, O                result_type operator()(argument_type device) const
7 P1 O$ x- a. B, F& f                {
                        CComBSTR deviceName;
* f3 e) \4 J/ T; ]0 L: S# T9 Z  S                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
1 r1 J) f* m9 ~7 j( m$ K
. m- z, n4 U8 X* U7 T4 m' b
                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;

2 E( D7 Z) u* ^' x! ?' N
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
5 o2 i- p0 s' ?                }
                CComBSTR m_udn;
/ ?, l1 W5 {. B& r4 w) g' y        };
1 O5 ^. `2 B5 w9 h* P       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
  D! {" }! z  E" K        void        DeleteExistingPortMappings(ServicePointer pService);
. [8 H5 Z+ p: G1 ]8 t        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
) h' i7 Z  K- X% c        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
5 J+ l$ r0 b+ Y3 S; @        void        StopUPnPService();


# ?9 H$ o. ^) `; s" R( i3 T" b        // Utility functions
2 g( C4 R0 {4 m' @2 v        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
, Z3 |" l3 W  I        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
0 n, f+ E7 C, f* J        CString        GetLocalRoutableIP(ServicePointer pService);
* @" l7 ^% |6 |; J# K
8 z" r5 }9 S! O: |
// Private members
private:
$ k+ ^7 M' b' M! R2 M        DWORD        m_tLastEvent;        // When the last event was received?
4 I. C! J( X. G( ^# |        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
! N( n9 t0 N# c# m; a        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
+ K: }+ v% O. b* \+ B        ServiceCallback                        m_pServiceCallback;


3 e! d8 q. v% e4 `        LONG        m_nAsyncFindHandle;
* L4 {1 E3 v* P# p" D* v        bool        m_bCOM;
2 V. \5 w7 Q* s/ y        bool        m_bPortIsFree;
        CString m_sLocalIP;
2 b1 M" J. w, m2 A* X3 W' X        CString m_sExternalIP;
+ z2 j: d( S) l& E' m( G# {- k        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
" }# n; p& A$ x8 N8 m* r. a- f        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
+ s4 j# e5 x4 B# q( w2 F        HMODULE        m_hIPHLPAPI_DLL;
+ r/ @/ F- ?; C; x! h9 o        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
0 e3 g1 S0 V/ b7 \' l        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
$ c7 y3 s" T/ g: r! c. P$ v
/ E5 k' S* I4 j/ i7 w
};

) V9 ^: L4 Z+ f" ^
// DeviceFinder Callback
3 ^/ f" U& F. V8 W, u4 s1 Gclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
public:
2 H5 ~: i  L" A- m8 P; Z# e        CDeviceFinderCallback(CUPnPImplWinServ& instance)
6 u7 A$ C! E' V8 z                : m_instance( instance )
        { m_lRefCount = 0; }


) [* o! t8 V. C2 u3 n' E# m' C9 ~   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
" t# D( A$ w, V* Z) `8 {6 Y6 K- `3 {   STDMETHODIMP_(ULONG) AddRef();
/ h' f3 Y' I# A) X" ~3 i   STDMETHODIMP_(ULONG) Release();
; l* ~8 ^5 Z; n$ N
' o( ^! z$ G- F6 F9 \' d. L
# o5 ~! Z" _1 C6 x. E* v" b* ]2 E// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
, }2 s% s, H. l4 t5 Y! a& k+ ^. k4 s

private:
        CUPnPImplWinServ& m_instance;
. H- A) o8 M# R. B- U; {, s' C4 Y        LONG m_lRefCount;
};

* M; {. G# E/ @' m* E
// Service Callback
1 ~1 e* }2 w1 jclass CServiceCallback
        : public IUPnPServiceCallback
{
; s: d7 g; G  q4 Q+ W3 J: ]& _6 o. {public:
0 ]# [+ d0 x, p* P9 s5 E        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
+ p7 ~: ~0 X/ n% v; i+ ]) Y$ w        { m_lRefCount = 0; }
' ~) l8 i% ]8 j* g, b   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
+ J( l: c* n9 `4 P. u
& `* O0 b0 [3 }) }: ?: r. v
// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

& U4 R9 D) v* ^- F
0 I5 J. k- D% z- J# U) P+ z0 t, @* bprivate:
, y8 O3 ^. e* l0 P        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
) X5 _7 t4 i: p( {, Y- I' q};
8 G# z$ u) i1 a3 v/ ?8 N+ t

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


" s0 F1 w8 }2 G2 j6 Q: d使用时只需要使用抽象类的接口。
  a1 s" a; u7 }: A7 g5 w$ [5 lCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
2 j% s0 I& W: `& ZCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
  \! Z2 Y' _% g5 K' p8 wCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.