UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   " q! e+ c1 }0 |! P/ {( u3 V! [
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   2 a, ~$ Y9 r1 M8 `( Z3 W" M9 [- t
9. {
   
10. public:
    ) _7 a: F3 }: S# T7 k" e/ Q+ f
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    ' S: i/ O# }; g& p& z# [
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    ; Q% l6 J* A# O2 X" _
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    9 A3 u$ D% R, f
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    4 D  a+ E5 K1 H3 g/ F% y: d; G0 n
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    3 D8 [+ m& v  p' ?8 M9 I
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    / v' Y0 ]4 K5 d% [: @$ f
30. 
    ( G5 ?2 N/ z% h: F/ f2 g& K5 C6 ]
31. MyUPnP();
    ( ]; V: g1 p0 ]- |
32. ~MyUPnP();
    . m9 j: {" K$ K+ l2 o% c
33. 
    ' }( e6 K. r% m
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    ! V' o1 M& J* t
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    " v& X/ b4 Z7 o9 E5 j: Q
36. void   clearNATPortMapping();
    . \& p; T# J: K' z0 t
37. 
    1 e+ F- Q0 s- S: Y
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    ; t( Q9 r8 |( m; i' Z$ _; o
41. bool IsLANIP(WORD   nIP);
    
42. 
      i2 @4 S, c5 ?' \
43. protected:
    % U% j! b( _; U0 @. h* i+ @9 R
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    # S9 ^3 o) b3 W7 K8 f
46. 
    & i" y, o; c6 A, H" R
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    " d" q1 ~; F9 N
48.       const   CString&   descri,   const   CString&   type);
    2 U* [- ^9 C3 z5 @8 S
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    + u2 U& }9 c! W' H- D7 y
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    # |! O- f* e* b2 f
52. 
    ) o; V/ R1 u# \$ e0 J
53. bool Search(int   version=1);
    
54. bool GetDescription();
    1 c9 n% k( X1 A3 j
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());}
    6 b) M/ g# S7 m; m. f, j! p
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
    * s1 M' j' e( \0 W# k7 q* u) m
62. CString m_description;
    - H( ~( `+ L+ K! Y
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    % _; E. w+ R/ Q
66. CString m_modelname;
    0 X% L; j$ ~6 b  Y9 i3 q
67. int m_version;
    % |" ^! q, [. s  }2 k: F# j; K
68. 
    2 J4 G  m: }* d# H  V4 J. _/ c
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    , ~9 @. l* N$ F. b
71. 
    $ E+ F. P. n# d) i2 K: ~6 y' s
72. CString m_slocalIP;
    
73. CString m_slastError;
    2 x! N' Z8 z& I) n
74. WORD m_uLocalIP;
    3 Y: W, v$ [7 b1 `, d% C
75. 
    
76. bool isSearched;
    
77. };
    2 a3 S( ~) l+ e; U0 Q
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   2 Y$ G4 t. w1 b0 _5 e" O
3. 
   3 W  P+ |7 H8 V* M
4. #include   "upnp.h "
   7 F4 W5 h9 x; X. v- d( z+ A$ p$ S
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   * ^* S7 V+ L0 i5 n: z6 N
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   : k4 j$ }) g) Z
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    0 E! c( c- Q& p2 f7 p6 p
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    / ?! D1 `; y0 K7 E9 B5 i# X/ k
16. const   CString   getString(int   i)
    
17. {
    ( n! e4 q5 ^2 C' s! t( ~0 B3 [7 u$ _
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    6 x6 @5 F; L2 Q+ R  Z
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    + W: ^0 K/ m6 l) I1 {, m8 ~
26. {
    6 ~. e  c& W* @0 q; U
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    % J- ^& }$ D/ g: k7 _: H( i
30. const   CString   GetArgString(const   CString&   name,   int   value)
    : a3 l1 h4 `% J3 |% ]& d0 _! ?
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    1 Y# q5 H4 [% s. j$ h8 Y
33. }
    5 H* k5 E4 J7 o3 p# }
34. 
    7 Z$ t9 A# b7 k
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    ; A; A! Q. Q; Q. _  d
36. {
    ( w+ ^  M% @$ B" y9 _
37. char   buffer[10240];
    
38. 
    + S% F1 G: d: G% B+ O
39. const   CStringA   sa(request);
    3 g( a/ u: D3 X
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    ' R5 y3 c7 G1 l" w
43. uint32   ip   =   inet_addr(CStringA(addr));
    9 B6 Z; p0 Q/ R. N
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    3 H; Q: l* R' N8 T0 V& a9 x
46. sockaddr.sin_family   =   AF_INET;
    # e* x& ^0 z- K1 Z! p6 F2 l
47. sockaddr.sin_port   =   htons(port);
    ' B8 {3 ]+ T* q
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    ; _& R& s7 d2 s  r- v
51. ioctlsocket(s,   FIONBIO,   &lv);
    $ S( n9 t* ]7 q/ ?1 w8 T2 V
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    % i5 D* Q/ t1 `  K: E8 Q% Z
53. Sleep(20);
    4 {; E4 B4 d8 C# G, W! }
54. int   n   =   send(s,   buffer,   length,   0);
    ) x: O5 J. w3 |7 J& V0 x
55. Sleep(100);
    ' N& g( I- ]; a9 g9 d- N1 j
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    & k" e6 a+ `7 I6 r' e5 G& s
59. if   (!rlen)   return   false;
    
60. 
    % j3 C/ S  W1 Z$ J
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    ) S2 v6 X& M) ?+ Z# ~
64. }
    
65. 
    6 i8 O9 a/ ^. W5 H1 @9 G$ _& }
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    - i6 k( w( C' Q! I) S) b
67. {
    
68. char   buffer[10240];
    2 I& ]+ K+ R/ p
69. 
    " r) s# W$ a; }% ^* @. J
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    / l9 d- E. k) O/ b
72. strcpy(buffer,   (const   char*)sa);
    ( K8 Q& Q# V, c' @) j  x
73. 
    % @' h7 d% i( _3 A
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    % x. }( \7 Z0 {' v
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    : h' p. u' q0 s4 x! Q+ T" ~$ f
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    + M/ r$ _3 Y1 E+ A2 ^7 h) ^
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    # J; w3 y% w, ~0 K' i9 x! U
81. }
    : B1 d0 p, _  I
82. 
    6 R0 `" A) |+ ~
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
      t% W$ ^3 ~3 H. c$ b+ w
84. {
    * G; ^1 ]; M; E' n# J* P" R
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    + O; I9 e: @; u8 |$ ~) Q
89. result   =   response;
    8 c: ~( k$ K  z% e: H' g) V
90. result.Delete(0,   pos);
    
91. 
      i0 Y/ J  q1 s. @+ }
92. pos   =   0;
    5 a8 U' n; \1 R5 m; r* E  z
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    # C. f0 B$ C2 V: @2 a
96. return   true;
    " n+ @0 Z1 s9 }# L8 i: v
97. }
    / V) T# @3 ?4 K
98. 
    6 j- V2 n  p1 C1 S- n( _' c- a
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    * a, o  \( y2 A$ B5 L1 q) r
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     ( e, [1 ?' v7 f- }
103. CString   property;
     : H2 z/ |+ w3 }& z8 ?2 l1 S
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     : d0 Z# z( |6 U. P5 J8 D
107. 
     5 Q/ _9 n- Y) v2 p' V2 Y7 E6 G
108. int   posEnd   =   all.Find(endTag,   posStart);
     # p+ P! A; N* {
109. if   (posStart> =posEnd)   return   CString();
     ! f5 z. Y% I4 a$ q* P, o
110. 
     8 K9 ^8 _% c7 X* K, r
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     1 U& i- ], H/ N! E5 p7 n* N- }
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     9 m# q9 |! V& C2 Z# a
115. :   m_version(1)
     1 R/ B% M' G4 i6 I
116. {
     
117. m_uLocalIP   =   0;
     ( b) e5 |2 I9 C. b7 a+ C7 ]
118. isSearched   =   false;
     * s2 n* r, s/ [& p1 b/ V4 ?9 h0 w" [
119. }
     . P7 }6 d9 Q8 I6 {/ W0 R
120. 
     / q/ e, c0 h0 e6 g3 m
121. MyUPnP::~MyUPnP()
     1 a% V' s- t  I7 ^: k3 g1 r  n1 M
122. {
     / C& q5 ^$ V( a9 x
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     7 O: S8 v: w8 Q8 X  ~
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     " C4 [6 H8 b: Q  a, ~4 |$ G
128. }
     4 u: x7 N( F; p, r. ~4 c
129. 
     
130. m_Mappings.RemoveAll();
     ! k5 O" M$ W7 Y& F' E# t: `4 Z# d0 Q) R
131. }
     # a$ v" v+ A9 {7 _/ D
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     ) t6 S8 H1 R. \1 u2 K- G4 L0 f9 d
135. {
     - v+ o* v: o1 C2 [' P
136. if(version <=0)version   =   1;
     - X2 ?6 ~& h8 ^6 i, V6 X6 Y$ f
137. m_version   =   version;
     
138. 
     3 j/ M9 _% K! Y- b9 N; Y( w; S
139. #define   NUMBEROFDEVICES 2
     1 R) x) D3 ]& B9 p) d( r1 x! j+ i& @
140. CString   devices[][2]   =   {
     ! [0 u& O" j# N7 a* G
141. {UPNPPORTMAP1,   _T( "service ")},
     % J% t: n6 R6 V' L5 `# R2 k1 h
142. {UPNPPORTMAP0,   _T( "service ")},
     1 a0 q) A) Y. b! Q& J$ e7 ?
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     ) u; `, C3 J1 I/ K
144. };
     % S  h% e0 l/ c: Z" t
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     ) R. I" a" y2 @. b3 o. @* o4 D9 U
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);
     $ h6 [/ i0 B. q, D# o2 t# ^
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 "),
     
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     % u# P% S: _' h* j. Z$ y" v
159. }
     2 Z# Y/ l2 v* I- L% _+ r
160. }
     3 E3 K5 M$ s: ]4 P- H" v
161. 
     
162. Sleep(10);
     0 J0 {. l7 i) ^8 R
163. 
     ! C* D" v* x) A. @
164. char   buffer[10240];
     6 ?  t; I' ]) x: e
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
       H; G$ E7 g. x2 H& P/ w
166. if   (rlen   <=   0)   continue;
     3 o" u) j! v/ |
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     : N9 M- K6 i5 z4 D# Q" [6 f
171. if   (!parseHTTPResponse(response,   result))   return   false;
     , L& O) K6 N0 X  \; M. l8 L
172. 
       r7 i" C' ]: K  p; \& I
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
       u( `& g* b. x- w
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     ' ~2 B" P8 ~8 k& ?& n
176. for   (int   pos   =   0;;)   {
     1 @8 c) j" F* r/ N4 n
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     % N3 c( C8 L- t$ K! ^9 R  R' W
179. CString   name   =   line.Mid(0,   9);
     $ b! C0 i6 h2 j+ G3 E8 j  W1 w% Z
180. name.MakeUpper();
     - C/ j' j) t" d6 P5 F7 O
181. if   (name   ==   _T( "LOCATION: "))   {
     $ f3 y+ F- V6 H- S1 D0 G( u
182. line.Delete(0,   9);
     8 n/ m) X8 ^' K# b0 {5 k
183. m_description   =   line;
     
184. m_description.Trim();
     / y: j% |# W2 i( c' K/ _
185. return   GetDescription();
     ; K) w4 E/ o5 E, l
186. }
     
187. }
     ' }  v5 a* Q6 W/ ]/ T
188. }
     
189. }
     ; w  ^1 {5 g! m  y
190. }
     " k8 ~1 i) e8 h* W8 G! k$ }
191. closesocket(s);
     : Z" S+ @: j% s( Q5 P7 x3 }+ b
192. 
     ' h8 n6 i$ ?6 q+ o. o% ^/ v
193. return   false;
     
194. }
     

复制代码

天云上人

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


///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

9 r% w6 i- @+ Z# \6 t- X
#pragma once
#include <exception>
( V. b6 b6 n! a% H0 f) H) R! M: t
8 E+ d* |) P9 w& \- b1 P" B$ A1 n* e
  enum TRISTATE{
# ?: I- W- h+ N        TRIS_FALSE,
7 D  k0 n7 {6 N' D, i: V, r$ ^. o        TRIS_UNKNOWN,
        TRIS_TRUE
4 g" |; x, ~: `4 W};


enum UPNP_IMPLEMENTATION{
. H8 D" Y2 m0 W; v2 V2 y        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
. D6 s: G) |6 q};

. h1 x: C, g2 U1 j) r$ S

! {2 O5 h7 @9 Q# Q/ {" }
! B2 @5 P" l7 J& }9 Xclass CUPnPImpl
{
  h5 m  m6 O4 _& Dpublic:
        CUPnPImpl();
        virtual ~CUPnPImpl();
4 \7 u; S/ c2 [9 X2 F0 W3 Q( K! d        struct UPnPError : std::exception {};
& _  ~- E* j. D, y3 a8 o        enum {
                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
9 `0 x1 }+ K( M# B+ i        };

8 Z$ z6 s( Z" I8 W; `- A5 h8 @
5 V$ X0 C$ T: ^# z" G* q3 U  w1 V        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
' W* v, v- X! K+ o        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
: E- @0 ?5 F  j        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
6 q2 ?, f. h# o7 J5 ?( Z1 L6 B        virtual int                GetImplementationID() = 0;
4 C; J& y! {0 T, H$ R# M9 h9 k+ M4 k. \       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

& Z' {! J% I0 t% x: t1 s
2 `( ~8 h9 i6 h: I; k! N        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
5 {. M. t. C4 G' r" |1 [0 N# ]: T0 R, K' K        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
0 h' f, T8 z% Q6 J3 p

// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
& C, N7 S/ B. L  s% C- `7 }/ A        void                                SendResultMessage();
- q4 ^" H: N; E        uint16                                m_nUDPPort;
& n: h( |9 q6 F/ e, q5 N7 O        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;


2 P' x. c1 ]% mprivate:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;


};


// Dummy Implementation to be used when no other implementation is available
8 F& y7 T2 T# e* B; l4 @" X* U' jclass CUPnPImplNone: public CUPnPImpl
5 p( |2 B0 X0 e" i( A{
public:
8 r+ B+ Y5 \+ ~9 \6 \! f0 n        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
8 M& B; v+ B( b4 R* J        virtual bool        CheckAndRefresh()                                                                                { return false; }
3 y& r5 _0 l+ U- ?        virtual void        StopAsyncFind()                                                                                        { }
3 g# T; s: D1 n2 H        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
2 H. S4 f* T9 Y& ]$ y0 O* w        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
6 i: }$ h3 p% K& X2 }1 {5 L  c

/////////////////////////////////////
. t1 l$ @4 V2 J# f//下面是使用windows操作系统自带的UPNP功能的子类
2 i- `8 m" _9 N8 C0 M* {  M6 B
" \) I. B3 p" ^$ R1 Z2 X
#pragma once
) r& o3 |9 ?( b$ `#pragma warning( disable: 4355 )

4 c5 ^8 h2 X/ K" U
5 a6 R+ f, b) Z8 N4 P) v8 X#include "UPnPImpl.h"
#include <upnp.h>
+ B2 k3 O8 ^) S$ a0 W#include <iphlpapi.h>
% Y4 g5 }# b# H1 q* q  I#include <comdef.h>
' q, B% K; S# E9 F( [#include <winsvc.h>
! W! i* H9 v! _  O; H5 [% Y

: O: e; [. `* }#include <vector>
, R! H* y6 h3 o/ P; |/ a# z#include <exception>
#include <functional>
4 B. H/ G9 r' m5 i# s7 S
; k/ }$ C7 S) y


typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
/ G' {& E2 z" z+ s; \9 u  y: ctypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
- d, g2 q/ M2 D% T2 G5 g. Mtypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

/ ]6 Y) ]! Y( P4 F/ x
typedef DWORD (WINAPI* TGetBestInterface) (
% n( U4 M1 c5 m/ n+ Q3 h* g  IPAddr dwDestAddr,
5 A" t0 H* J) r- j( T  z% S  PDWORD pdwBestIfIndex
);
4 m+ [% w3 o9 ^' |% @
1 d+ A$ r& W3 ?% {5 n! z! ~
typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
; O% ]% A- k3 ~: M, d! a3 \  Y7 j  BOOL bOrder
4 S  Q) r! X" b* _4 w);
' h& y0 R4 z# c7 x% b

' R3 s9 n2 x7 [7 K0 J; \typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);


CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);

+ }$ I8 G9 W6 u7 A
  S4 J+ w7 V. @3 v8 D- o$ pclass CUPnPImplWinServ: public CUPnPImpl
{
. I9 g# M) e% Y- R        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
; k% J9 [# ?" b/ j& |// Construction
) e& M6 q+ W4 Qpublic:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
2 b! O4 Q! O5 ?/ R0 d. w
- t5 Q& J# _( b3 {
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
! n& {% r! _4 Z, R* r3 x" j9 L+ o        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
8 L6 S7 w4 \% }9 U        virtual bool        CheckAndRefresh()                                                                                { return false; };


' t) M+ c1 @" T) b  l, fprotected:
3 o1 k3 f6 G% w4 `3 C        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
- A2 N- N& M2 @' O7 x        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
% N8 u9 |/ V, [5 i5 ]        bool        OnSearchComplete();
        void        Init();
, o1 ?1 o( m1 Z0 ]+ j" T

) Z6 f8 _8 i+ F9 @  w2 l        inline bool IsAsyncFindRunning()
! [% D2 |6 K1 \$ P# Y. g8 E3 Q: W        {
" l- M9 ^6 p; _- _                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
6 D* C) ]1 |7 u0 M$ i0 a                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
6 \7 @/ K6 u* u: g                }
' A+ B. c1 `& Z1 t                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
( Z4 c" H4 k! V* c                {
                        TranslateMessage( &msg );
# [& i0 l  v1 E! R# K. R6 R                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
        }
" ]# ^9 E' {2 n/ A
' F0 |) q6 U! D6 M  N, S
        TRISTATE                        m_bUPnPDeviceConnected;


5 L! A. e( D- R& ?& B% K0 _// Implementation
. \% f. V' M. N) g% n2 f        // API functions
6 b. l( g/ N& N$ V4 b% W        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
# _: S+ `* P6 }* y; ~        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
7 q) U2 n" O  u+ _        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

% ^( V6 |9 |, u+ J) c
        TGetBestInterface                m_pfGetBestInterface;
( f! ]1 |7 k* H4 L$ J        TGetIpAddrTable                        m_pfGetIpAddrTable;
+ Q0 R4 l* n& J1 X9 Z2 V        TGetIfEntry                                m_pfGetIfEntry;
0 h1 g3 X& I# |( J3 H4 L6 q, }

6 v+ ]9 E5 s$ ^% r* U4 e        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
* ~% F- ^, t5 I6 w        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
) v, ?7 c# ]- t                {
                        CComBSTR deviceName;
5 U$ v% A4 X0 @7 E0 H6 K                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
$ ]  Y- X1 _6 R/ P" s, V! i
3 U8 h! N# N: T- |, V: h
                        if ( FAILED( hr ) )
/ s. _) G( }! H( J6 f% H( Q9 }4 |                                return UPnPMessage( hr ), false;


                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
        };
       
( c; p) Q# l: V8 A5 L' C% b        void        ProcessAsyncFind(CComBSTR bsSearchType);
. y% O, r1 {: F; B" ^" z' C1 o        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
" j# j' ~' `) p* o8 P; ~        void        DeleteExistingPortMappings(ServicePointer pService);
  A( D' C4 @7 K$ r. s9 c        void        CreatePortMappings(ServicePointer pService);
1 S6 C- e. j4 ^. F5 z        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
( `( g  X( J, l! D8 M: E0 }3 Z7 ?                LPCTSTR pszInArgString, CString& strResult);
( s1 Q* c- O# e        void        StopUPnPService();
7 V9 r& w9 T$ E. L
* o' a& W5 x, [
        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
0 [* m( r. Z( H  ?        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
; `2 [. {4 S+ x% [        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
7 ]' s4 R& G+ F8 Y        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
  d8 q- ^) m) W) K7 w, b8 \( T        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
# C/ k9 d! S% b2 w4 p& N0 c        CString        GetLocalRoutableIP(ServicePointer pService);
" i- l: @( }5 ]6 A" S) I! @
# v7 G& U1 Y7 D+ @. ~
: d/ N8 L8 c# f% c// Private members
private:
0 |2 T/ A! D$ q3 o" Z8 j* f        DWORD        m_tLastEvent;        // When the last event was received?
- p5 P/ T1 l9 p3 W. Q        std::vector< DevicePointer >  m_pDevices;
! \/ b7 w9 U# _/ r  L2 B8 _$ o        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
* f* I$ W: k% O        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
* N: k, g- T& s  y6 r
, @3 {3 A1 M) p
        LONG        m_nAsyncFindHandle;
3 n& ?$ v* h$ A- E* M1 y. W, y        bool        m_bCOM;
! U7 Q: b6 d- R, B0 q        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
+ ~) ~! w9 T6 v% v) ^- p% \. [        bool        m_bADSL;                // Is the device ADSL?
% u4 E3 F- M4 A  O' `6 b" M0 l        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
/ h" C0 l  p* p: \( L0 y+ F        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
# ?+ b, v* h$ N! F% _+ F        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
4 j1 J- s. s  o, t; k  h        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
* z3 d  |$ i  D6 r3 z! w        bool        m_bDisableWANPPPSetup;
6 u/ Q4 Q2 H9 ^( d3 M4 O
4 I% |+ E5 C  J6 {
};
2 y2 Y8 ~$ ~7 |1 c2 [  a

// DeviceFinder Callback
/ B+ X% c0 `% |2 o3 y, gclass CDeviceFinderCallback
9 h9 ?& a$ K2 y4 u! U( i, C        : public IUPnPDeviceFinderCallback
5 t: p0 I; U2 K. A: g, _& t  U{
6 V, K& N1 ~1 q/ [0 f5 \& Wpublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
  N5 O1 C  \6 d9 @" K$ f4 }                : m_instance( instance )
        { m_lRefCount = 0; }

+ v5 o; Z' I& K/ W6 s- G# U0 P9 ]
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
0 F9 B5 L9 E9 ~& y/ `0 t  }   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
4 _* Q6 c. E5 H& l. ]9 u9 i) d
# w5 Y" v, T! w) A0 Z" P5 k
// implementation
! p7 S. C" L: V. J; mprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
' W6 D6 |8 Y3 T        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
6 U- g& |% T% f        HRESULT __stdcall SearchComplete(LONG nFindData);
7 j$ u- h8 c+ G% n7 E( l  E5 z
8 T$ R) j7 R8 R
private:
( S0 K+ K3 V5 Q  D& [        CUPnPImplWinServ& m_instance;
6 U0 x' q3 x) D$ v6 J        LONG m_lRefCount;
, G2 i, ^. N  A- D! {. @};
% O' _" o, n* n

// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
{
public:
# w1 z  G) C6 A$ k        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
" N1 K! T, ^7 ]        { m_lRefCount = 0; }
6 d4 @: r' c8 {3 \$ a8 w, Q9 s   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
3 c! D- s0 r) Z/ t2 k: c   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
) w4 E# {: d; V  H1 j

+ b& p+ t6 W& g// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
) i. ^" \3 u& M        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

  ~# w4 K) L1 v9 c* [0 a
/ k$ Y: i. U( Y0 v/ b% Fprivate:
- L( D: F& W# J% G        CUPnPImplWinServ& m_instance;
- E, I5 a  r' T9 D        LONG m_lRefCount;
};
/ v/ n0 y' S1 ]% e' ~
- B2 Y5 T; ]  f" J2 ^- ]6 V
+ M- z" k' ^9 A+ X/////////////////////////////////////////////////


使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.