UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   3 n/ |( f9 d, p0 q& [
3. 
   ; [8 P  L4 Z) `" k' V2 i
4. #pragma   once
   
5. 
   7 Z  ?6 f( D' P
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   
9. {
   
10. public:
    ! ~5 I/ A/ a) c
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    9 K  m0 c% v; w6 c3 I
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    " q7 G5 X4 n$ V' q6 N
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    " J8 `) O! B; e& L0 g+ T" s
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    & `* y( U$ o: h4 j  o0 _, Z9 D
17. }   UPNPNAT_RETURN;
    
18. 
    ) N2 z) w' U$ L( b6 Y* X/ h
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    . x9 }8 i9 O  a- g; ?
22. }   UPNPNAT_PROTOCOL;
    
23. 
    : m" l2 d2 |" d
24. typedef   struct{
    4 i, e% D) M2 K& ^
25. WORD   internalPort; //   Port   mapping   internal   port
    , r( H" o, z! ]' O* @
26. WORD   externalPort; //   Port   mapping   external   port
    * f5 L& i3 p( K0 U9 {8 }
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    ; S3 z0 }: `9 s8 g4 J- h
29. }   UPNPNAT_MAPPING;
    
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    & ^, Z  C9 u3 A9 J7 O1 s( }! v
33. 
    6 ?8 T3 L9 Q$ q0 s& X
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    ! w+ U; a" l& b, \1 z
36. void   clearNATPortMapping();
    
37. 
    1 N+ o9 U8 ?4 l6 f2 J; s. W3 o
38. CString GetLastError();
    . A' f5 L' `; i6 |0 A# t
39. CString GetLocalIPStr();
      D, R4 b, h) Z) t# G- x; U7 o
40. WORD GetLocalIP();
    " M% y, K3 ~0 `2 g7 ?3 Q6 J8 }5 v
41. bool IsLANIP(WORD   nIP);
    $ e) A2 ^1 A6 G% A
42. 
    
43. protected:
    6 B9 S4 N9 P2 F* S* {% w" ^8 J
44. void InitLocalIP();
    : A2 C$ ^  b/ |0 U; S
45. void SetLastError(CString   error);
    8 `/ j" o" Q* W) _- r: S9 ?
46. 
    1 \& }1 H! m* A  U# ^: h
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);
    ; Y  ]! C3 h; f' P* L3 T. D, ]9 T# u
50. 
    $ E4 n( Z# b( q3 |
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
      N# Y. T, I5 ~9 I( W9 a; }
52. 
      z# k) ~2 }' `0 J; P9 X4 w
53. bool Search(int   version=1);
    
54. bool GetDescription();
    * G) r# K' D% a9 g( A# e
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    ! ]" W& ^2 j  d/ _
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    : Z! p% i: x. z
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
    - K+ s# B4 \! O5 D7 x/ b
62. CString m_description;
    
63. CString m_baseurl;
    7 w; y+ p% y9 V* g" B- d2 U
64. CString m_controlurl;
    8 C# H0 Z2 j9 a, O- S
65. CString m_friendlyname;
    4 C2 g/ ?! |% }5 k) G: A1 P
66. CString m_modelname;
    
67. int m_version;
    * [# D. Z, w. Q8 T2 ^
68. 
    
69. private:
    + B* V8 i: @3 [! a
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    ) e' ~4 [) N3 T& j' N* o
71. 
    
72. CString m_slocalIP;
    8 J. O. z, ^' p
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    
77. };
    . Y# Z) g8 P& b5 h9 N
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   5 ~5 P; W3 t* n  Z7 ^7 i) W5 W
2. #include   "stdafx.h "
   8 }3 F  k1 A! k8 M3 X% s
3. 
   
4. #include   "upnp.h "
   
5. 
   - ~: X0 B6 U7 L: y% W% ]
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   6 d1 A: G5 B  T' ~
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   " S: }* A& I' \) H
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   0 U9 G- u' z$ e* K+ E
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    - f! d0 n. s( k$ u/ b0 m- R- ?* ]
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    * S3 _+ A( u2 ?3 \( |6 H% W; w0 m( k! C
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    . z( f& L, i9 j3 Y3 K1 ^0 L
15. 
    
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    ; u- f6 g) K1 C8 ]# b
19. 
    1 o4 P9 m# ]$ }0 y0 m, @6 e0 t2 X6 q
20. s.Format(_T( "%d "),   i);
    " z5 |# R( s3 ]; I: ?2 c2 ]$ a
21. 
    2 [* T8 k2 r( o5 U" d
22. return   s;
    2 u7 t/ O/ V3 [" N0 c/ X4 E
23. }
    6 a# y9 Z6 O# B2 l
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
      o  O2 u7 q6 s: @6 k' {$ @' U
26. {
    ' l1 b* D* J( G) g/ I* _
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    , S" {+ `# o7 U, B, ~
28. }
    9 A0 ?( d1 [3 y% X  |
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    + u7 y) @& W7 u- A" X, \
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    5 Z1 C+ x) t3 [$ x7 s
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    + ]$ `: o5 {( d$ @% C
36. {
    4 A" K9 \7 ?7 b( W2 T* P
37. char   buffer[10240];
    1 s9 D$ R1 n$ \- X
38. 
    
39. const   CStringA   sa(request);
    ) e5 ~" u- y( s+ b) F( y+ `9 u
40. int   length   =   sa.GetLength();
    # m* c- g3 L# U. x) K' i7 Z
41. strcpy(buffer,   (const   char*)sa);
    ' k0 n1 E- ?. {% ?- d8 }! S- [
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    # f. l3 \# s# X. u9 j9 F
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    8 T1 w2 u, ]+ {- u& d! j; ]+ Q" c- h
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    ' i4 X3 G4 `4 Z$ }1 b" l
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    1 X% O$ f, |7 [- _. [8 l
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    2 ~0 D3 ^: D0 X0 l
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    " e% q8 [- d) |+ h, w
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    7 b5 U* F  \4 J! q5 V
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    * R: `) ~) \$ \9 c. o( P
64. }
    7 e5 D. X3 f$ `1 p3 X. [
65. 
      c. k# x/ Q  u- ^5 [
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    # B; q8 K, j+ e1 S. E
71. int   length   =   sa.GetLength();
    ' B- L! N! {- K" }6 z( K
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    4 S3 ~9 h! }: q- H6 U
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    " }- K0 E" n. m  e# ~( H1 U
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ; N! W+ Y% n( B* m. n# G, D
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    , [) e9 g- [( z9 e+ D/ b
81. }
    % P! ?1 D8 @8 G' ]+ g! C' Y
82. 
    0 q) [+ ~: u  [- [' Q, X
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    ' c: L; e, A6 ]
85. int   pos   =   0;
    5 C$ E7 |( O- D4 I7 d7 n0 @2 X
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    2 \# V+ c; y1 o) A# Y+ c) t
88. 
    : y- C6 A3 O7 [5 ?* a0 E: A$ i
89. result   =   response;
    
90. result.Delete(0,   pos);
    ! U% x% i/ f( B( W0 _" l; q
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    . t/ s' u& K+ G. l8 D
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    2 c( M  \) e& {1 P: V( H; f
96. return   true;
    7 i' {+ @% P% i1 h  E3 f# y
97. }
    - }* S: F, s3 R% ^* N9 \
98. 
    0 S/ B. x# E# Y6 v
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     " C! Q1 d/ V3 N, S& c1 \
103. CString   property;
     1 T$ E1 I7 \7 S' J4 m
104. 
     
105. int   posStart   =   all.Find(startTag);
     5 p9 f; h6 `( Z4 S" d4 i: M
106. if   (posStart <0)   return   CString();
     . A2 O" v* Y' X9 m! q& `- @
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     3 `$ T+ G6 l4 x; W( A
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     6 c$ A  O6 A. u7 ^$ ^* D0 C) P* z5 K0 P
112. }
     / K5 q! N2 f% Z* f, _+ q  a( \
113. 
     + `: M  U/ X, u
114. MyUPnP::MyUPnP()
     ( x0 X/ ^  I! W2 t& a3 J) `$ a
115. :   m_version(1)
     $ ]. t4 ?6 T" x% h. w! w$ r
116. {
     3 j7 _5 r( F8 w, U
117. m_uLocalIP   =   0;
     ; F' Q) R, Z( F) ]7 j
118. isSearched   =   false;
     ' W3 Z' K5 e' P* h# T2 \  p" v
119. }
     2 O2 e. X) ?7 T+ y, I4 T
120. 
     $ Z7 d% ~9 H7 {9 n3 Z, ~5 O# @
121. MyUPnP::~MyUPnP()
     
122. {
     ' {& k8 D3 D. t% e+ v9 m
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     0 f, F7 W! e: W& X8 N
125. while(pos){
     + G0 e( v5 H' C# d
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     # ~. W1 \- i2 i: K4 [' x( ]
129. 
     1 k: N5 A4 {8 v( A
130. m_Mappings.RemoveAll();
     , `7 [1 L1 ?. P+ t5 x  E" j
131. }
     $ ]* {. |" u- l! L0 Q# E! ]8 ^
132. 
     
133. 
     ' \  |9 i1 V9 Q
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     & S- H6 H8 X4 T
138. 
     
139. #define   NUMBEROFDEVICES 2
     1 T9 r8 L6 n" k* _7 c, x6 n
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     2 F5 @9 ?, O8 _, Q" c
144. };
     5 ^( I& n& ^6 w" P8 p2 I; Q7 K
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     9 q. ?4 t7 ]9 f7 x, H3 @
150. int   rlen   =   0;
     9 y3 `+ d! I# I" O# v
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     7 x& }. |) R0 V
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     3 C5 a5 c+ ^' l3 [9 d! K/ u
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);
     . d- A/ _1 N4 I9 m" {- U& y
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     
161. 
     " h% v* h& ?7 e2 v  O4 v
162. Sleep(10);
     3 C! x! G3 G1 n# L$ f5 A
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     # ?& H/ L* D' a/ W. B% X
167. closesocket(s);
     
168. 
     : b3 T2 d( \' i& R
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     ( p0 U1 b0 Q: W* V& q' p+ Q
172. 
     ! l* Q" _( ?/ M6 O6 b9 l
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     2 l8 V0 M6 \0 P5 y  ?
175. if   (result.Find(m_name)   > =   0)   {
     ! u9 E1 O  L7 |! i
176. for   (int   pos   =   0;;)   {
     & g& ^1 S5 G/ ]. o3 `
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     / L" L# v) x* u: R2 @3 ?* {
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     * ^& j1 w5 e0 |# r
182. line.Delete(0,   9);
     
183. m_description   =   line;
     : r9 }/ b* ?8 k! u) L
184. m_description.Trim();
     
185. return   GetDescription();
     4 {) j- e4 g0 T# }/ [" `
186. }
     
187. }
     
188. }
     & {/ M, a) b3 _; F
189. }
     
190. }
     
191. closesocket(s);
     
192. 
     + o: B4 o# V, j# ~+ `# `5 K
193. return   false;
     ! N; ^; b3 G7 p
194. }
     

复制代码

天云上人

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

9 Q4 O6 a2 _5 b: ]0 H" y' C3 Z
///////////////////////////////////////////
0 h  s' V) C) f- ~5 n& {7 L9 p//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
1 T% @. b7 A+ p/ y( |+ i

. Y5 k- {% r- D5 k3 S8 @( w#pragma once
' u) E  V% K, G! u5 Y#include <exception>

5 f" O- \5 B8 X6 E9 [
! M9 o' ?7 V# z  enum TRISTATE{
/ |6 Y; j7 W% Q3 u1 o        TRIS_FALSE,
" h. E  N8 d! S& [: }0 p" \        TRIS_UNKNOWN,
        TRIS_TRUE
};
4 G1 j$ R% ?  d& ~1 @1 \
2 u% P5 j1 s$ C7 h# N$ S3 \
enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
+ b6 d: }) Q! i& D        UPNP_IMPL_NONE /*last*/
};

) Z( m* c! @* R: \

% i  H  ]. G7 x, S
class CUPnPImpl
{
public:
7 B/ @) G" Q/ J& s, E6 J        CUPnPImpl();
9 O2 l; _: D. Z/ s6 i% k4 N' h        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
, M$ b0 ]+ }" [- x8 X& k        enum {
) }) x9 ?4 h3 D  ?" R                UPNP_OK,
( g1 }* Q  ^7 N7 c                UPNP_FAILED,
' U6 X) v0 F( ?! q% c7 j4 U4 s0 n2 H                UPNP_TIMEOUT
        };


) q* \; j9 H/ E1 M' [) |        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
, {9 j7 a( f+ ~. A  j$ ~        virtual bool        CheckAndRefresh() = 0;
! i& J4 {+ @# q- J4 e0 r        virtual void        StopAsyncFind() = 0;
- K+ z& N5 B6 N0 p. C        virtual void        DeletePorts() = 0;
" P  u6 n/ s" T7 ^- g        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
# N8 d5 |' X) J2 A8 U* y) ^        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
5 P7 Q! ~- N1 p" D% t+ A

2 t6 R5 K3 p  {6 R        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
* H2 n4 P! V( Y" Q3 Q9 t# t; x. p# K        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


// Implementation
9 a' F6 S( @/ i6 kprotected:
+ A9 \0 }! w1 w3 R        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
0 T6 Y: s6 R+ i, p        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;

( n5 u8 g  Y0 n
5 e7 q2 e9 A/ Q1 j5 ^" gprivate:
* t1 ?) i" n, ]% |        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
! f3 A0 w3 i' W6 V0 ~
/ X  ^3 n* ^* C6 w
};
' J: N; d1 Q4 M! e
: ?9 {+ n, y' ^% b' z) E
5 A) J' ^, `6 J4 @' G% R9 K, K3 ^/ `3 i// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
5 a2 t; r$ U0 c5 V* |. O  }public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
% W) ?7 ~" g/ ?; ^        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
8 ?# u. s0 q: c2 K& U- e        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
7 B$ z( y( k, E2 U3 J. B9 ]        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};


; [2 ~5 L: z: n4 N( W7 r8 u# {9 [/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类


) T% b7 Z5 }% H5 u9 a4 d" t#pragma once
#pragma warning( disable: 4355 )


/ Y2 h2 N- c: }% g9 [#include "UPnPImpl.h"
#include <upnp.h>
% J! ~8 \6 Y9 }#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>

! L- A% |* ^: a8 @) V/ M
: e% ^. j; y$ \2 d! r0 \#include <vector>
  g  `. E: E% d, I, m& y  I! a#include <exception>
( V! i$ D: E) G$ O2 w0 g# T1 `#include <functional>
) G7 |5 C4 }) Z8 T5 s
3 J6 G. s2 ]( w
  p" D7 Z8 j  r9 W4 Y& |

typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
. K/ l- E  t$ o3 Y4 L' g2 d) Xtypedef _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;
' o3 d  m# F: Itypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
/ _  i5 ?/ R' l5 S( A7 ~* v
* D% c$ w4 `. g+ n6 Q% l
typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
) {" K6 r9 Z8 \& p7 R) u8 C( J);
# \2 d* I) C. a6 F, a

typedef DWORD (WINAPI* TGetIpAddrTable) (
5 g$ g/ n+ H: f, B  PMIB_IPADDRTABLE pIpAddrTable,
7 R/ M- ]5 g2 _& m8 r9 P* c  PULONG pdwSize,
  BOOL bOrder
);


; |( E3 ], N5 t( S* A" xtypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);

4 n' p% J; _2 a2 H0 x
CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
- v* M2 {6 V/ h! I" ]8 s7 p) b/ u7 S

0 P  Q) y7 V, X/ g6 ^6 Gclass CUPnPImplWinServ: public CUPnPImpl
1 t2 q7 j" w) t! a1 i{
3 r/ v/ T, ]- q        friend class CDeviceFinderCallback;
8 P- K! `* ], ~; }+ B        friend class CServiceCallback;
// Construction
public:
+ l8 Q9 X& d/ [# X, m. E        virtual ~CUPnPImplWinServ();
- O* W! Z8 V" d9 i) c6 V( _1 L        CUPnPImplWinServ();


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
; x1 ^" r  u1 y1 ^8 G        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
, R( @5 b) Z) n) e
. ^; ]' G3 N& j5 S, I. m9 x
2 }: b! m5 c/ O* ]        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
3 e  v9 }& D) Z  z5 X        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
- z* K: v, R' R0 A6 [        virtual bool        CheckAndRefresh()                                                                                { return false; };
* F1 Q4 V& e" ^' P4 F: W
, V& R6 U2 ^7 h) q* U8 Z# H; G& }5 D0 a
protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
$ c; h0 }8 c& m9 d3 r- }        void        RemoveDevice(CComBSTR bsUDN);
' `! a5 J; W* A8 q( x        bool        OnSearchComplete();
        void        Init();
$ @  X9 u( o) I7 c# m1 X7 {7 _
! P! Z- P' g5 R; z3 _( P
        inline bool IsAsyncFindRunning()
        {
. X  ~  C& f9 G  E1 D# ^7 I. G                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
( n: I) p$ P8 E1 D3 m/ `/ ~- R                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
& M  |# `; H+ W  b; s7 K                }
                MSG msg;
! d. E- |# ~  o+ V$ y                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
# s8 |" _1 R( ]8 ]$ e  b5 g5 z                        DispatchMessage( &msg );
% l' Y" m. R1 I                }
$ k  X- b" c0 z" B9 B0 ~                return m_bAsyncFindRunning;
        }
  }: o& W2 f, l9 s% O" E6 k6 q
. e) ?5 Q% U$ h- j; x% Q
& }: T) R& l# h+ D2 R        TRISTATE                        m_bUPnPDeviceConnected;
$ Y- g. @7 m0 K
2 P6 @+ N+ J9 J2 S' n! t
// Implementation
0 q8 a. V% J8 J7 t( ]3 ~5 n        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        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*);
+ H, s7 a2 b5 s& D6 `/ n( [        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
4 d+ ^6 b6 m5 R$ H
9 R/ `( L* z; Q- _
% F/ h  ?: q2 V        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
8 g6 }5 M6 ^, L+ t% e. n! G4 C        TGetIfEntry                                m_pfGetIfEntry;


6 u9 Q0 R7 d# o# \        static FinderPointer CreateFinderInstance();
3 f  p; D7 }2 E) A        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
+ E6 z) i- V3 G, N5 \( l                result_type operator()(argument_type device) const
1 k# K7 o+ f0 Z* p; U                {
1 W' g& k) Z. B# G6 [' z                        CComBSTR deviceName;
- B( w+ D9 l. K- ~1 M! }                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
, S/ j! W; V2 j3 e) c
: I* I7 G+ X, L7 D' b
                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;


                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
% Y- D7 i4 Y6 N4 \        };
6 s% J, c4 [6 G# S5 C       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
; B: I: A- ^8 q/ [% {/ E+ i        void        StartPortMapping();
: R5 }8 y5 t1 I. \: L) O: Y# O        HRESULT        MapPort(const ServicePointer& service);
+ @1 s' `2 k" a' a( b8 s$ V7 o        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
. ~3 h* c$ ^2 M        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
' M) h$ Z4 X. {' Z& j+ i, T" O  \* J        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
; B! j& s( b2 Y0 i" ?1 R                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
8 S5 A2 N$ w# F& a  w! U' Y# Z
5 D4 ~: a) N+ n, {) Z; ~. M
        // Utility functions
4 g3 \! L4 U$ ~        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
/ `! \! N! U0 h5 H, _, H        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
4 K2 f0 o0 a9 X" x7 f        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
7 B& o- T1 u! {% l, i        CString        GetLocalRoutableIP(ServicePointer pService);

& v# h& {4 w  J/ x4 Z( n2 }( j- ~
// Private members
% i& m- P0 D! v1 U- |private:
$ o- z  K5 G- o$ e6 ?8 g$ _        DWORD        m_tLastEvent;        // When the last event was received?
5 K7 R: ]8 L/ N, p, z7 b        std::vector< DevicePointer >  m_pDevices;
, g* ~" R* l, q9 ~) |  h5 x        std::vector< ServicePointer > m_pServices;
5 x, ^& I9 \. d7 E% P! _        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
. P6 C6 a+ n, Q; Q, V. Z+ w        ServiceCallback                        m_pServiceCallback;

5 `: B( e1 b+ \, n# \5 p7 h
        LONG        m_nAsyncFindHandle;
* N9 |4 H, v. |) O        bool        m_bCOM;
; ]7 ^' g' J& S% t; b1 U4 T) ?        bool        m_bPortIsFree;
, l; i6 |+ h2 ]: z0 [1 i        CString m_sLocalIP;
        CString m_sExternalIP;
% \9 T5 F9 z* }6 m: i) t% w        bool        m_bADSL;                // Is the device ADSL?
1 _9 N3 W! T$ \6 A7 V$ h0 k3 U9 u5 Y        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
% \  k0 s7 i& _; e- m0 b        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
+ S" p2 f9 [  K( d' G* }8 [        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
' E6 Z+ H) B7 A4 D        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;


$ j" e9 a6 ~1 e/ q& J};

3 e( a4 h  z/ z* `1 @* N
* R( p# A' s# c- w; K// DeviceFinder Callback
class CDeviceFinderCallback
5 z( X) B, d) J) o        : public IUPnPDeviceFinderCallback
{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
% v$ S& @; n6 _! O& s& c) P9 U/ |                : m_instance( instance )
        { m_lRefCount = 0; }


5 F6 K3 j- g6 p, q3 A: L- B   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
" F3 Y0 b8 x4 L4 \6 o- g   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
8 l* e3 f, _# r  U* E$ M
+ A3 }7 v8 i7 |6 ]
// implementation
private:
7 v* }# ~$ Q- V' o8 b        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);


private:
        CUPnPImplWinServ& m_instance;
. c2 T$ Q9 M- D: y" G5 F& F# O        LONG m_lRefCount;
};

  }) W+ W6 S/ u$ Z0 ?' P# @( Z3 U
8 K2 V% A4 C% I2 y// Service Callback
( H& a, r% ^7 ?' U" G. N, Uclass CServiceCallback
5 S# S& D# B: J& i3 K        : public IUPnPServiceCallback
6 Y" X5 [/ S( ~3 Y: o$ l7 }: P{
public:
3 K7 ?7 e( |0 s$ a% i. Q4 Z' f        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
" y, d- a- v! J   STDMETHODIMP_(ULONG) AddRef();
5 \* F" \9 R5 O   STDMETHODIMP_(ULONG) Release();


// implementation
! j2 B8 Z9 `2 f4 u: ]private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
7 H  N: D4 \4 P5 a$ `};

" e% J+ S$ b! D2 {4 q) g, t6 R6 t6 A, y
/////////////////////////////////////////////////

8 g+ f6 h/ g% C* K4 a" ?
使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
! R  V; u1 b6 Q8 F' K, r! }( l" lCUPnPImpl::DeletePorts删除端口映射.