UPnP

天云上人

/*uPnP.h*/

1. 
     n8 [# o6 z% x; Q; Q
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   , l. X7 d3 l) E. v( b* u
5. 
   9 n* U# n# O# f6 T0 }1 `: q- Z: N
6. typedef   unsigned   long   ulong;
   
7. 
   " f/ z( b+ I( W7 c& l
8. class   MyUPnP
   
9. {
   # E5 O! ?4 Q6 d/ d. T7 u/ S  N
10. public:
    ! R( L# C( z) n) Q/ H: \' T
11. typedef   enum{
    9 o* ?% H  _% Q/ G! h
12. UNAT_OK, //   Successfull
    4 z4 H2 ~/ f% F! i7 r* v
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    1 b1 p/ D) F* X6 Z# K6 ~& c: O
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
    6 O8 I8 Y) o6 ?7 `& H
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
      u3 n, _9 M, b9 c
17. }   UPNPNAT_RETURN;
    3 M9 X3 Q, i4 v) }6 `& k
18. 
    
19. typedef   enum{
    0 S0 u% M2 h  M3 M6 u2 e
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    2 i- P. L! }! U6 m" r2 t; m* i
25. WORD   internalPort; //   Port   mapping   internal   port
    0 U6 X' Q9 Z) n" y) V
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    8 u6 q" F( w# E. v( I) N& L
28. CString   description; //   Port   mapping   description
    . N4 ~& n8 J  X" Q8 w. s/ v7 F
29. }   UPNPNAT_MAPPING;
    
30. 
    0 I5 |3 |5 j' ^3 T
31. MyUPnP();
    3 ?& G- H  N4 F  S7 ^
32. ~MyUPnP();
    
33. 
    + ^) r) o$ Z7 p" D
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    $ [9 I5 c5 s- h8 v1 t$ N! m
39. CString GetLocalIPStr();
    / x, B6 k/ X9 m  q1 q
40. WORD GetLocalIP();
    ! ?) t' S* S( K, B# m
41. bool IsLANIP(WORD   nIP);
    
42. 
    % z! B' O* K; ~/ J; F% W: a
43. protected:
    " K8 ^% p5 ]; A! t2 X
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    : X: A0 X# W9 _' S2 d7 C7 O
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    6 `' Q2 A# t4 {8 E
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
      C7 ?- g8 A3 o4 z0 P
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    5 O5 I+ |% P/ O% A: x
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);
    8 Y9 k+ h6 h+ R& y4 k
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    - T9 a- j* C! [& q: m
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    $ X6 g! ?9 M2 o5 Z: ]  j
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    % p& v& F3 C3 B% y
66. CString m_modelname;
    
67. int m_version;
    & b2 l4 g, ^- N$ a' R; M, @
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    2 z$ w) r) Z/ [
71. 
    : P& h6 D" p' @8 C' k
72. CString m_slocalIP;
    
73. CString m_slastError;
    3 `% W& L1 F3 X
74. WORD m_uLocalIP;
    ! V6 [2 B+ e( B6 N. }% N( S
75. 
    0 q( n; q: P6 n: b, y" u% f: S
76. bool isSearched;
    # e; y8 F1 g+ n
77. };
    & E, z. j$ @/ M! c. B" f
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   : R1 M' D" @  O
3. 
   0 G* C( j8 v( a5 k5 m# z* P& g8 w
4. #include   "upnp.h "
   ' L$ f9 E9 [7 e' x4 t; v  Q6 M" g8 K: X
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   - L$ z8 x% Z9 G3 ]: L
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   : C: I$ l+ b$ l. b9 P
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   / [" y* @) E2 ?) N/ R8 i$ Z4 v, s
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   8 L, a8 D% k2 T5 G1 k) [
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    , G6 }* T0 X; ?  s& i
11. 
    . m; I$ E+ i- l/ I' U
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    2 a0 ^$ C/ h: a8 D  k; k; x* w
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    / s2 `/ D5 [3 ?3 F* u5 ?/ Y( \
16. const   CString   getString(int   i)
    
17. {
    ( O# v9 }' b: M1 Y2 e
18. CString   s;
    # p) ?* Y6 k  v7 b
19. 
    . A* h. `: W  @% c
20. s.Format(_T( "%d "),   i);
    % }$ d% f  t* G6 ?$ V& g0 X! I
21. 
    ; U% X7 _1 B5 y, S2 D2 F" F/ @
22. return   s;
    ! U3 w# m' T; \" O5 j
23. }
    
24. 
    + j' ^1 ~* R% y/ R, f3 M6 W7 L6 a; Q
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    + e# v8 g% Q2 [  L
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    8 Y+ b+ M: Q' A; t5 G( {
28. }
    + Z4 n- m7 _/ ]& S8 P. \+ m
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    # v/ a4 w. H) Y
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    ! @4 [" q' d3 e0 i
34. 
    " D- {9 G' W% h. x& V$ p+ |( q
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    - g' V3 Z! K' J( S! p1 ~1 `
36. {
    $ u6 o- \2 ^$ r% Z) D6 c
37. char   buffer[10240];
    
38. 
    ! C1 Z, T/ y7 v  J
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
      x$ Z; K* |$ q6 y
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    $ f. F' r( [4 M. k9 `* N
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    + A- W& z, ]- J8 w+ m; x* u& w
46. sockaddr.sin_family   =   AF_INET;
    % z1 a1 @7 u3 Y  D5 z- l
47. sockaddr.sin_port   =   htons(port);
    7 c& O' H# f* x7 E
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    - c" g% G' `0 o! g4 f) R* U
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    2 S4 K* V& [6 |
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    + |' o" Q- g/ E( O0 l
53. Sleep(20);
    $ M$ w. l" Q! m8 u1 k7 l$ W
54. int   n   =   send(s,   buffer,   length,   0);
    4 }) {7 j6 X- q( P# \1 ]( u
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    * R9 {6 ~# E' V$ @
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    ( N" y. K/ m2 m
59. if   (!rlen)   return   false;
      W# Q- ~4 Y4 G6 h2 P% K8 _$ c
60. 
    ; E0 {( I) k. R$ e. O8 x2 |
61. response   =   CString(CStringA(buffer,   rlen));
      Y# o3 s& ?: y7 Y+ x2 ~% p8 [
62. 
    ) }* Y# K( q4 K* q% [* ]4 F7 k
63. return   true;
    
64. }
    9 o6 }+ {: }: |% {
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    0 i& y0 q" x/ N1 d8 @0 y! O
68. char   buffer[10240];
    ' s. y. x! }$ b. ^
69. 
    
70. const   CStringA   sa(request);
    1 m2 z4 j* n' S) ]+ a
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    : Z. ~* b( t- C2 ~, V0 R+ U* |
73. 
    ; }( Q: O% k1 N; ~- d+ @0 b8 F' ?/ c
74. struct   sockaddr_in   sockaddr;
    ' M" s4 X+ V! o& x0 D" [; h5 M$ d
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    5 H. q1 N9 M! r8 V, t0 j
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    6 S6 R" ?3 a! u8 l$ n' e7 R
81. }
    1 Q4 B8 O$ N( B. ~& C, F
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    7 e# j% z5 P2 P
84. {
    7 r! v0 o: V9 j# V
85. int   pos   =   0;
    
86. 
    ' C, j; V( D: k$ F- g$ ?' m# l* N0 t
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    1 N0 w) k8 x# [# _. ~: n2 h
88. 
    
89. result   =   response;
    ' K+ X' I( J- I8 t7 n
90. result.Delete(0,   pos);
    
91. 
    3 Y3 V& N7 a; K* r, f& A; j
92. pos   =   0;
    8 w, _8 n4 j8 R; I* o/ K% n
93. status.Tokenize(_T( "   "),   pos);
    8 J2 ~- W% U: _+ b
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    , w0 Z9 O5 S. ?$ @' I% b
96. return   true;
    . x; }3 i: E# ?% h0 \( F
97. }
    % j( b! p6 l5 K1 g
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    . r. O+ \2 \  l6 l
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     ! I9 S. F7 K8 o. h% w  ?) H/ A- j
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     8 j# V4 b2 J& @7 ^- c
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
       a9 F6 c/ H/ z7 X4 a0 j( v6 N+ N
108. int   posEnd   =   all.Find(endTag,   posStart);
     0 E8 T' f0 ^& b4 s
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     ' g9 m" k: ~, M1 Q; I* a: F
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     3 l# a% l% i* l& n
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     9 Y& |5 l9 {  D
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     6 ~, x) i0 r- I
119. }
     ; o+ w2 F6 s) Q) U
120. 
     : Y9 t& M: u9 u1 |% u% I, j3 \
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     4 b* d; p4 K+ M8 W$ o
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     . ?/ [$ S0 f* k0 U4 W3 k# ~
128. }
     : O1 h7 u) {8 d
129. 
     - S! ^: u( D6 y  [7 _+ p
130. m_Mappings.RemoveAll();
     . M* n* @! `) D+ N0 E
131. }
     / J& `6 E1 h: h5 g
132. 
     $ |" ]0 J# ~- |& S1 k! R2 V
133. 
     ( {2 q; y0 ?5 B
134. bool   MyUPnP::InternalSearch(int   version)
     0 s7 f0 _; t" O
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     ) \" O; t& o$ m6 _9 V
138. 
     8 i" ^( d9 x! }4 b0 q) N
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
       L. `3 i0 P  d1 I( A2 p5 t: m/ _
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     ( B  {  j9 W* f1 h. O- y
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     * @+ I. D* ?$ n4 D: o3 }0 e: H1 J, x
147. u_long   lv   =   1;
     6 O7 G8 O' r5 x6 I2 `' @
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     $ D7 B! k7 A* E9 A
150. int   rlen   =   0;
     ( W/ P# B) ^1 u! b0 [* T2 V; g
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     " A* r9 n! T6 z: h
152. if   (!(i%100))   {
     3 @) W3 q) R, k% V! B  e" T' a# m
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;
     : N8 ~4 P/ s4 _5 ?8 v# r
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);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     ! ~0 w# ]) n* Y' |: C
159. }
     - C. l4 ^5 j! ~5 S/ B# F2 a6 {
160. }
     9 X- u3 M6 t$ M7 ]( Q" e6 l
161. 
     ) z3 i" n1 x# D, p2 [
162. Sleep(10);
     3 l# N5 f3 ?" o/ I. c9 y# M
163. 
     + {7 W: j- j8 t0 k
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     & i0 D" W) s- G' Q; y" N( l& {- H
166. if   (rlen   <=   0)   continue;
     * _7 N  D" V+ a+ m, W8 f/ Q
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     ' q: u! C: C! u. M* {+ N
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     5 N2 `% Z, P0 U7 @
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     * x' y6 ]* u4 d% i6 _0 O# j
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;
     ! |5 ?0 a% W; L& l
184. m_description.Trim();
     ; ]- A1 @3 s# D+ X' i" d0 X9 L
185. return   GetDescription();
     
186. }
     
187. }
     3 r: v" ~: A: J6 L0 F: d- B1 I
188. }
     2 P4 H0 k! g3 p
189. }
     5 y" Z$ _6 }$ ^& [
190. }
     
191. closesocket(s);
     , d4 B) |% l" l
192. 
     
193. return   false;
     
194. }
     ( x7 i+ d# F/ y( ~* p$ {" ?

复制代码

天云上人

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


///////////////////////////////////////////
% t2 L" @( K5 H9 c4 B. d2 q//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
9 x! f5 @' }5 ~) T8 k# v

3 }4 J. x; v4 W6 X#pragma once
#include <exception>
. p4 O3 ~! U  D3 W6 F! Z# i
5 {7 u' k7 i9 F3 P0 _; J
  enum TRISTATE{
8 E3 A1 I* v/ V6 u2 X, x        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
};
/ e0 O- i' S6 h5 y
: K6 J6 l5 h2 M4 g: d0 v' M$ \; `( `
enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
2 a$ Y: _7 j5 W$ z$ s        UPNP_IMPL_NONE /*last*/
% _* H( }: [1 _- O};
) j. T) I2 h* Z. F



. H( ]% L4 S' ]  [4 eclass CUPnPImpl
& V% ]2 r! c8 b8 @0 F{
4 R' B0 b( ]$ h( \/ ~: \' }public:
$ F% v4 M, Q& Z% L        CUPnPImpl();
: R& j/ F0 Z$ a        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
2 P. W7 P- R, b# V( T3 ?        enum {
                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
2 D' y( W: n, b7 [- N& L8 v* X        };
+ j6 `# I, p6 z+ {
. Z. ?& @$ |& P" [: @: C" b: s  o
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
4 }% v& X7 K; {! I+ N! x: s        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
, N2 F( v7 h. u        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
+ ^6 @* m% ]: D; k4 J       
3 f- f+ o# S* f        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
) T" w. C; ]9 J& _6 I
5 g, |2 `- z1 C; J3 ~9 w' Q7 }4 B
, u( m) l  U# R# K" y/ E        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
2 B! K- i8 A% a# r6 |

. l9 C' \) k' X4 N( |// Implementation
* u% L. Q% B$ b4 aprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
' f! f) z( a: n* V7 z2 |$ N        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
9 V) h' u4 L4 f2 Z( m6 c& e: W        uint16                                m_nTCPWebPort;
; x' e' m$ l9 q# b$ V* K        bool                                m_bCheckAndRefresh;

% f' Z# t# I. e  ]9 Q+ V- Y
. j1 Z( Y0 W; R! D- ~private:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
3 z3 \: L* T% Z% ]

};

6 R5 M! F* n# E- p5 s& T
- ^6 j) v- C+ y6 W8 X7 F4 y) j( ?// Dummy Implementation to be used when no other implementation is available
3 R" J0 F9 z' A" h; gclass CUPnPImplNone: public CUPnPImpl
{
3 w& k( J' ^- z+ N* hpublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
! \9 Y: R: D) P% _- V2 v+ n        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
" N6 _- s" |& L1 G# u3 u        virtual void        DeletePorts()                                                                                        { }
6 @, L+ o7 v3 _, C4 y! `        virtual bool        IsReady()                                                                                                { return false; }
4 T  g% `2 R2 {! _        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
  o0 P  W2 U# v6 Z6 j) z0 ~};


/////////////////////////////////////
6 P; x, J6 O4 c. c7 q8 h' g( t4 F//下面是使用windows操作系统自带的UPNP功能的子类


1 k7 [; ?9 i& l8 @4 ]9 R& _9 D#pragma once
#pragma warning( disable: 4355 )
" n$ s5 }, ?& G
2 Z( a% o7 Y& y- Q) J1 c: {
#include "UPnPImpl.h"
* y2 E3 p6 x6 m8 J: i0 |#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>
) }" c' h) S9 o4 v9 K; P& z7 x
( ^" t/ `; l; X& c
1 m/ y$ t6 f* q2 O# r4 o3 i: ?! J#include <vector>
#include <exception>
5 O0 y* ?* C5 W" c#include <functional>
6 j& X6 ?7 \; K

: b( L5 Y3 P0 W; s
* ~4 r+ k4 v! M( k
8 {1 u2 W5 Z4 a5 _6 r# [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;
/ X7 h: }6 |9 {typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
9 S& a0 J6 i6 e' r, v$ htypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
8 p- C8 R& {3 E1 F0 B% y
3 v- w+ L% \$ F: v( A- {
typedef DWORD (WINAPI* TGetBestInterface) (
4 ?$ J  \) c% g2 @; d+ A) |  IPAddr dwDestAddr,
6 @- C8 w7 C! Y  PDWORD pdwBestIfIndex
);
, j3 ]7 B, @% C6 ]2 F) e

) Z6 g4 {5 K4 p2 jtypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
: `9 x4 H5 M+ w8 r  BOOL bOrder
);
2 k+ Q3 J" n4 P: F+ [
! H! {* x- p+ s8 I9 ^- h
3 F3 E2 |& }9 l! T/ q  H; A4 S5 n# y% ztypedef DWORD (WINAPI* TGetIfEntry) (
& v. o* }1 ^6 `+ v9 a* c; `  PMIB_IFROW pIfRow
);


# N: Y. _, E9 n4 QCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
6 ?6 w% U$ J+ |1 j$ ?. r9 s7 |
# _1 [3 s8 v1 h' D4 x3 o( }
! Y1 I- |- ^7 }9 g' yclass CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
9 F, E) I) U1 Q* D) P% K        friend class CServiceCallback;
3 k7 }4 q$ H: [* Z* g, T$ p6 |1 E// Construction
( I4 c2 d( \8 \, \  o4 V( P0 ]. Lpublic:
, `* x, s# ?' u7 U% e8 h        virtual ~CUPnPImplWinServ();
) w, C7 L8 U1 ~$ ?        CUPnPImplWinServ();
8 h; d0 \: @# T/ h1 [* l, i7 Y- I
8 n8 U" O1 V/ d
; P& a4 E; h* K7 d9 e        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
. d. N% n% n9 w+ z/ w9 @        virtual void        StopAsyncFind();
% Y" S6 I3 q" x; J; W" q( y# @$ u        virtual void        DeletePorts();
; D! t7 K3 P/ Q; V8 q; `7 y6 R% T; V        virtual bool        IsReady();
# G2 }5 L' _" h& K7 Q  ^        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
9 z/ p3 ]! J+ d, c
, T6 H  G. l) H3 {
        // 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
  S- M! o. R& x& |1 X- p        virtual bool        CheckAndRefresh()                                                                                { return false; };
6 e) c. Q% W, j9 L* c
0 q0 O" w! w' g" S
protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
* q9 ~( U$ r6 l        void        Init();


# A) \6 _: S" M) J( x/ I/ n1 s        inline bool IsAsyncFindRunning()
        {
- M. y7 z9 R# q* b; ~4 w* V                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
7 \/ T# {! x1 P, ^& i; }                {
# O9 ?# L! B5 v9 s0 L( y/ \% @2 w                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
. o: m3 v% V. \& y* ~& f                {
5 ?& {& b, e; V0 y* ^  ^9 w8 w4 L; [                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
        }
5 u, c1 `, |; ]
! h- r- I- _  Z6 U% T% Y, B5 B
, m# d- L/ E  n& B. P$ C        TRISTATE                        m_bUPnPDeviceConnected;
  T, s& v: O( O* Z9 c# b$ C
3 v% M- _; U! ]0 F& X
// Implementation
        // API functions
0 ^/ z8 `2 o  o$ o9 z9 o5 e" q) z  u        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
8 C/ G; G3 I- Y( y( y        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*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
" H; k2 l6 |7 |6 K2 I
7 c: n; z- u! _/ {8 |& E, x
# w( Q7 _$ u, @        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
1 B0 h& t1 g+ e$ z2 T9 \* y        TGetIfEntry                                m_pfGetIfEntry;

: \+ x% f1 S  n  `
        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
- z# U; Q2 o5 x/ r1 x( h+ W. _6 C6 k  A                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


                        if ( FAILED( hr ) )
0 @) b7 m4 s1 u4 u4 a* V! c- I                                return UPnPMessage( hr ), false;


                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
        };
       
0 w6 t  G: M5 e8 A6 v1 x8 e0 d% n        void        ProcessAsyncFind(CComBSTR bsSearchType);
( G$ ^7 Y# `1 E        HRESULT        GetDeviceServices(DevicePointer pDevice);
) c8 s% Q! V1 }5 j        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
7 R" O( B( m6 y3 Q5 e4 B, q7 ^# v5 ]        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
7 Y3 R9 f" S; l3 @                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();

9 ^: T' ?) ]3 {: G9 S! ]: v7 V
. {8 T( V2 d2 R9 m        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
- h7 S% U) X; X/ O. ^* X1 L7 d3 c        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
4 P, U+ ~1 v5 K$ E2 B        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
. D- H& J: q. c' z( g        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
7 B, y" W0 [7 Z+ {
+ j) @2 E4 P$ Y) s. f9 ^1 y
// Private members
5 L; ~* r1 Y- H/ @: u/ G: Bprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
# O1 l7 L+ `' L7 B8 E0 \' r! X5 {        std::vector< ServicePointer > m_pServices;
3 x, h" O* C2 H: E. _# E! p+ Z        FinderPointer                        m_pDeviceFinder;
  q6 |0 p4 h& ^9 }8 x        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
: G# y! P# P9 I) y8 H
, e) x% _) k4 s: v0 P6 k5 a7 D' v
        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
8 K: F  I, k. `8 m. K9 ]+ }        bool        m_bPortIsFree;
        CString m_sLocalIP;
/ W1 }" o1 v- p3 E" z3 Z        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
# H% g' E9 s# r& {6 Z* c, u: G/ \4 o0 F        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
. s9 K- H4 N' m/ F  Z, j0 u        bool        m_bInited;
        bool        m_bAsyncFindRunning;
) A+ A; ]; C; X        HMODULE m_hADVAPI32_DLL;
) @0 V% a9 U6 e) Y/ n9 c8 K! q        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
4 z, w( p1 e5 V. g        bool        m_bDisableWANPPPSetup;


};


+ Z3 @& }5 R# p- c+ P3 r// DeviceFinder Callback
2 K* ?7 _: s! I; Y4 j6 Yclass CDeviceFinderCallback
) ]2 k) X" `- _# p  e& n        : public IUPnPDeviceFinderCallback
{
& A. t1 D. S% ~" |4 Qpublic:
5 a# R+ _' t* s. q2 W        CDeviceFinderCallback(CUPnPImplWinServ& instance)
0 M8 ^$ |" c; p/ s2 T3 }                : m_instance( instance )
5 h* z9 b! [4 T4 J        { m_lRefCount = 0; }
1 n3 W% o: r' p& B+ W/ b

   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
2 N8 P7 p( N: I: E% k$ \   STDMETHODIMP_(ULONG) Release();
7 |/ V9 b0 [& S) v  p& E9 n
2 v) f9 C% m/ G
// implementation
private:
3 x% X* ^9 K' P" K+ o, q) A        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
2 w( L4 ~1 G3 s# f

private:
        CUPnPImplWinServ& m_instance;
1 O6 I$ c* A4 r        LONG m_lRefCount;
0 H1 k7 x1 f$ e4 O4 F8 T8 X5 ^( P};
' M  g) v8 r* b* ^, A6 S! w2 t
9 U, K+ E7 h) ?6 ?" H7 \: ^; l$ i
// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
{
+ J1 \: h9 o! E5 k3 ]$ J) k/ _; hpublic:
        CServiceCallback(CUPnPImplWinServ& instance)
$ B. i+ C  a- a: k                : m_instance( instance )
4 N' R6 t# G% C: A        { m_lRefCount = 0; }
   
- c4 G  N0 `' k+ N$ U/ W# W6 [   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
; n! _" n- E+ z$ h: g6 g! n' n

" W6 {: ^3 h) f( f: u1 V. L// implementation
' V7 ]1 ]' m( E" x' r, E5 e- J" Vprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
" T- P% O; S) E1 v% a. [/ U' {* @

. S4 i7 e& ?9 ~0 q6 m+ S7 c/ O, cprivate:
' T( f! V# [/ j$ l        CUPnPImplWinServ& m_instance;
* `, c2 a# ]/ X; D$ o- A        LONG m_lRefCount;
" r5 D" F1 L5 i) B% ~2 w' K};
$ N6 G; F$ a/ z! O/ v

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

3 D9 Q- o; q$ l8 V, ]
/ D7 Y+ I" z; S/ H" o- L1 L使用时只需要使用抽象类的接口。
% F  X% s$ c" Q4 j1 Y% E" BCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
% z% o1 u- _7 s. K9 g6 m8 ACUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.