UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   5 b( X  y/ m/ }
3. 
   
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   ' Y5 J6 e/ j4 q6 M( S) J
7. 
   
8. class   MyUPnP
   9 [4 B$ j% {3 N$ C. L* {1 c/ I
9. {
   & P5 m* p# C$ k3 {& i% h, i- a
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    " ?; d/ k0 d- d2 r; U/ s
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
    1 P! w# X( l  l4 A: P# F3 q0 z" T
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
    # r  [2 ~/ F2 B
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    : x, J* t3 o& D: z
21. UNAT_UDP //   UDP   Protocol
    4 ~4 b6 C! D5 ?8 B( @
22. }   UPNPNAT_PROTOCOL;
    
23. 
    " n3 c/ N5 G0 u2 _4 }9 M
24. typedef   struct{
    " ~& s1 _7 Z8 t9 b8 Z* f4 L
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    8 a' z) ?" {1 G( b
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    , J( u6 h. H3 s$ r! n
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    , @: K& q: i/ F5 o8 r8 B8 ^  _6 t
30. 
    
31. MyUPnP();
    * x4 t; {: y; S/ C
32. ~MyUPnP();
    1 ^3 r7 ]- U2 [, Q, {/ ^
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    # E( X& D8 f& R4 L" K! O
36. void   clearNATPortMapping();
    * Q# n6 x% m4 r3 J. n) N6 M' Z* l. L
37. 
    " d8 g' v5 O% k: N5 l) d* E' O
38. CString GetLastError();
    & U) W% H, J9 |8 l7 A" T
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    0 k6 C3 ^1 i0 n1 ?
43. protected:
    / d3 Q" c, V5 x/ |- `; L7 b
44. void InitLocalIP();
    6 r( [: s% Z9 {9 q- f1 U  }7 S/ Q: r
45. void SetLastError(CString   error);
    
46. 
    : I6 A0 W& d% g2 a2 L3 l+ B2 u
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    * ~5 K9 d. P5 ^4 b9 \
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    ! S' t' b1 G- h
50. 
    ' n' ^8 x) L+ r4 |
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    * s) x6 K" T3 F) y; B7 U- O
52. 
    ; u7 Q. `& Z2 E/ [+ m" K
53. bool Search(int   version=1);
    
54. bool GetDescription();
    7 c2 ^+ s1 Y9 ^) m  N/ ]
55. CString GetProperty(const   CString&   name,   CString&   response);
    8 t: Y% E: R3 O, \6 a4 R4 T
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    # p0 F/ d/ t4 D9 N5 k( s
57. 
    2 R8 Z- [/ m- X
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    9 B( b. W6 u5 }0 |  J
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
      ]; Q0 b3 g$ [# W: O
62. CString m_description;
    
63. CString m_baseurl;
      x. Z9 q7 ?. X: F4 R" h
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    ; F* T: ]6 c1 z! L& P6 T; m. o
73. CString m_slastError;
    " z* U. ]+ z% S7 `1 v& r
74. WORD m_uLocalIP;
    
75. 
    - Z6 H- q) |# [# a- V/ \* V4 D1 N; j0 q
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   " i. E) Q2 w4 b* s& t) o3 e! q
4. #include   "upnp.h "
   * O5 k, z/ m6 P2 F9 t3 f0 k
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   / L% k. u- V: W2 G' x$ D; m
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   $ @4 {9 |+ P: a  C# K& o
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    1 v" V+ t: o! R" H
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    + Y2 }$ w5 M5 ~% \$ H3 E( z
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    - G3 F  m2 z5 D1 q" {0 N4 T* J4 s
15. 
    
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    : Z2 w, F. _3 u4 s) g7 P
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    4 h4 ~/ ]9 b$ E# y- {' d5 t+ i
22. return   s;
    8 @. E; D6 _* L" O$ b  _/ @5 Q
23. }
    
24. 
    % S% ?; @$ E+ U* s2 q4 n. K
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    $ Q' Q' i9 g- i* R
26. {
    9 D( q' N1 o( `5 G1 L) z8 @/ x! D6 r
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    8 s! W! e4 g4 E. }5 i  K
28. }
    8 e9 M1 L: x0 c: i2 }! t
29. 
    $ P* f7 ^  Q' U+ Y6 h
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    & s; |! s3 L: V; d
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    3 O2 f4 @0 g0 |& r4 O" ]1 D0 ?
33. }
      a+ S" [# F! Q( b/ X2 [
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    8 E& m) }/ {0 C% \
36. {
    6 ^, B8 `  ^& W0 u1 |
37. char   buffer[10240];
    3 g* y5 d2 W6 E
38. 
    + H% m2 |. J2 u% n) F- `- A0 Z7 h  o
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    3 _* x% b5 J! @! S: W
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
      u+ p0 Z' M6 X8 F  p) P
44. struct   sockaddr_in   sockaddr;
    . Q5 B$ u. y& [$ K; k
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    : ?5 _' J# b1 H2 r/ h- w
46. sockaddr.sin_family   =   AF_INET;
    ) g4 F$ I/ Q2 d4 _, c: R
47. sockaddr.sin_port   =   htons(port);
    4 K% `/ Z, V& `8 v% s
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    0 S7 R% R1 I  C4 ^; R$ Y" n# e' {
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    : v+ O+ l% I0 z* \) f/ ]  G8 H/ }' z
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    - R9 y9 e0 i# S/ Y  U+ m
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    4 `- g; e0 \3 D6 c3 K
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    . f7 d9 m) G% w! `
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    3 F. |' v& |" E8 B1 L2 i
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    . n8 |) r4 }: v  [" V& [/ _
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    4 \  x: O9 y5 X8 z0 [, H* l0 W
62. 
    , o) Q; s- {, J5 m
63. return   true;
    
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    . @/ u0 ?; A: _
67. {
    
68. char   buffer[10240];
    ; X" u2 P) S6 e* V5 l
69. 
    8 }; Y# b6 W4 W" b& M. T. s
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    $ P5 j. |! V- o6 B
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
      G* P: o6 ?* W% Q  ^
77. sockaddr.sin_port   =   htons(port);
    " e6 ^6 n9 R/ p3 i7 E0 W. s0 T& U
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    * Y6 }7 D9 c' o5 `, ^  c5 G
79. 
    2 S1 P( B8 g* e$ s
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    2 B1 |( A7 V9 {+ n  f0 i) ?
81. }
    
82. 
    $ U! w6 h: g4 E! E
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    " m# S2 e% G3 x2 x- ?% I8 f
85. int   pos   =   0;
    : c( _+ K* i: Z8 D7 ~7 M' t1 n5 J
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    & F# h5 Q! i$ K8 R8 |
88. 
    
89. result   =   response;
    & N( {6 y8 x: t, l+ m" x7 }
90. result.Delete(0,   pos);
    
91. 
    7 U2 B4 |& t4 I  e. D% `
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    # J& G  e; S' k+ b, x1 v
94. status   =   status.Tokenize(_T( "   "),   pos);
    7 ?( m; p5 i  {* W# {8 J
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    - l* q% k% |* K+ h' p( V& h
96. return   true;
    ; }5 H) K4 Z$ X% Z, u' @
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     * K# o1 `' x1 A, {6 \' w
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     6 H7 n: x. L! p) T
103. CString   property;
     : r' `. u. Q; e4 {
104. 
     4 F+ N3 C5 K" [/ u( y/ Z1 G
105. int   posStart   =   all.Find(startTag);
     & U/ S5 c# ^% ?/ z
106. if   (posStart <0)   return   CString();
     9 a# T9 b* u, }6 ~
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     , _3 v( M- L& J
112. }
     & G- P0 E7 f* \4 E9 g& _
113. 
     ' p" B1 r, n- t
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     4 d& q( U6 J; C
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     " l9 b. h, Y7 d
119. }
       I/ R8 Y3 n2 C& r+ k3 c0 J
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     $ g0 J- q* P, ^
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     + \' \5 G5 K! t- L3 T2 ~2 @
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     9 o5 G) u$ f5 K5 g& r4 b
127. RemoveNATPortMapping(search,   false);
     
128. }
     : \5 A" L0 q$ K" f
129. 
     5 n/ b3 _1 P; `* G  e
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     
133. 
     ! s8 @' [' R  S! `4 p8 V" \/ m8 i
134. bool   MyUPnP::InternalSearch(int   version)
     - x8 J' P: k: g+ g
135. {
     
136. if(version <=0)version   =   1;
     ! f) k+ j2 ]5 S8 d9 v
137. m_version   =   version;
     . q6 m& ]1 g3 x* c+ L
138. 
     
139. #define   NUMBEROFDEVICES 2
     / [6 k% s  O9 v- i2 S8 e
140. CString   devices[][2]   =   {
     # Q2 B# Z% v; E7 J1 [
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     1 ~4 L" V$ L) X/ t: z( O, s1 d8 ^
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     / R' i6 D' H$ B' ^9 d9 a9 p
147. u_long   lv   =   1;
     3 |6 R, f7 r5 X* Y0 \, g- B2 Z/ I* y
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
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 "),
     7 \+ a. ?, R( |# Y4 D. p9 @/ c! F% @
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
       Q3 S; D( u$ V# M
159. }
     5 U% ?' h  ~2 q, q* x4 M
160. }
     
161. 
     
162. Sleep(10);
     
163. 
     , H! I. v1 M$ o* s' v7 M
164. char   buffer[10240];
     4 l7 B9 H+ @8 D. l
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     2 N5 t! T$ U+ l# a/ G  e& A
166. if   (rlen   <=   0)   continue;
     3 r+ e: n6 {5 Q) C% w4 P
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     3 b. u, D' y: f5 j5 j; Z; y* V
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     ' s' R4 W7 H4 _9 p9 ^: j: M% [3 b
172. 
     
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);
     ) r0 p2 r# k* I( x% Q
175. if   (result.Find(m_name)   > =   0)   {
     , S5 S3 ]3 h: p6 Z4 z
176. for   (int   pos   =   0;;)   {
     / r- \! {) {4 A
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     $ y2 K0 @1 v6 x$ o: \8 A
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     ( ?: R2 Z3 I* \2 Y
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     ( A3 y" H$ E) h. h- b# z
186. }
     
187. }
     6 C$ F9 m% m/ o  H- G
188. }
     
189. }
     
190. }
     ( D( |& B7 T, J# |1 K
191. closesocket(s);
     
192. 
     
193. return   false;
     ' w2 q+ W' G! w
194. }
     

复制代码

天云上人

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

& A: G' [, s0 I! p
///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
5 [  ]+ L! j0 {7 Z9 r& s: q/ F0 K

#pragma once
#include <exception>


! q% D% `3 _3 M6 }9 e7 e  enum TRISTATE{
' l# S+ T& ~# ?: E3 H        TRIS_FALSE,
        TRIS_UNKNOWN,
; }7 d7 A, i" o* r6 l9 c+ c% u        TRIS_TRUE
1 K' u" h5 x, `( C};
/ O2 q4 x" n8 f0 N* y, u
9 {* V: S9 l: M# Y9 [) Z
enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
$ C; A" C8 D$ ^8 t        UPNP_IMPL_MINIUPNPLIB,
9 p7 R: M& O4 N) f        UPNP_IMPL_NONE /*last*/
4 d2 ?. f8 }3 j( K};
' n$ X3 L) T3 V6 e
& x( q2 e  a0 g! D( C
  Q6 C8 b# N5 @: b4 ]+ [4 e& j8 ]
0 Z2 q( x( y7 A4 M' g5 S8 f4 [
+ e8 p. O% U  v9 t4 wclass CUPnPImpl
{
public:
        CUPnPImpl();
3 o4 H% g; C4 F) _  |" N        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
2 W0 y% ?  b1 G; m& ?5 z  B: s% |                UPNP_OK,
- \5 v( K5 ]& }+ W% |. q                UPNP_FAILED,
" k- O3 @' @9 T4 k- {                UPNP_TIMEOUT
        };
1 g1 k0 D. [  k
; M, K9 K2 X* ~& s
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
/ j9 F" m$ ]6 ~; v3 y) t- _+ `6 G        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
  {5 K: `9 L7 Z$ U       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


: o9 N' \8 u* `: u        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
- {# U1 B3 q& z* z8 j1 @        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

2 Y7 \7 H0 ~6 p6 u! }/ C- ?7 a
// Implementation
0 N- }# j# w% j+ }2 Rprotected:
+ o- y0 E, f- |        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
/ I) u( e; x7 r" k        uint16                                m_nUDPPort;
' W/ W. I7 \6 S, l2 i        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;

) f, q3 X5 u+ D" J+ Q2 ]5 [+ S! t- I
$ w& R5 _4 H8 X6 `( L. nprivate:
# p, h! ~+ X0 {; j) P        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;

$ }2 k/ A" Y* p1 H: z9 s- A+ x# w
};

. S/ Y/ y& G0 \$ `/ P/ [  D
// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
9 B/ F1 N4 W2 U+ F{
" `9 q: P5 L$ [" y$ Tpublic:
5 p1 j4 \5 S- _3 |! U; p- v& f        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
" X# l. l; ]) J- i6 W* F  K  h        virtual bool        CheckAndRefresh()                                                                                { return false; }
6 H' W/ O4 o% x0 O9 w- [- s* E0 \        virtual void        StopAsyncFind()                                                                                        { }
; C6 L2 ?, L9 A5 d: w! j7 P  G        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
; C8 j# {- b  M8 k7 G1 F        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};

$ G! B3 r0 J4 I6 m3 l" \, {" v
9 f7 k2 g% M1 i/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类


#pragma once
/ ~0 x+ `% x: M1 h2 s- H+ b* x#pragma warning( disable: 4355 )


#include "UPnPImpl.h"
( o0 y4 [% J' w# R- t; P$ _. |#include <upnp.h>
) R/ T! _5 A6 [. ^8 C#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>

* j/ F9 c8 ~& A9 w
+ v& S8 A0 Z3 F#include <vector>
* c5 B2 C( K6 C* W1 t# `& g#include <exception>
#include <functional>

4 O/ A; c" T4 J( [. `7 x( {; w


6 \& `3 L" G- ytypedef _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;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
: g: \, R% `  C; p  y- f, D9 qtypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
; X" N* s8 a2 l3 W9 }typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
1 ^, K2 p( g, v/ D9 h1 ?+ S
% G; u2 }  x6 u" N% K. u/ L
3 F1 H0 U" I" r5 i/ Z4 ytypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
; G1 w8 y' r9 H/ K$ w) y$ t  PDWORD pdwBestIfIndex
);
- S6 {6 j. Q( }4 r

  z5 e9 ~2 H5 R- ?5 f9 gtypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
" h( o: F4 A8 {9 F6 L% O. q, N  PULONG pdwSize,
7 U% ~$ z- c/ c$ d& \$ ?' R; h  BOOL bOrder
, E$ b  O# C  G( {);
  x" M  `; N3 s1 P

/ X+ K# {2 A' X7 `- {  @( atypedef DWORD (WINAPI* TGetIfEntry) (
: S; ^, G- ?  w4 s6 o$ d" J' b0 Y8 k  PMIB_IFROW pIfRow
9 j6 X: o! ^& q4 J. F);
# @3 f' d! z5 T8 q( T
; @6 j; v; v1 S# _, Z2 U$ x3 f
CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
2 Z1 B: b2 h' N2 \* I1 i7 [2 `
4 Z/ a( V+ h3 |" }7 t" U: s
class CUPnPImplWinServ: public CUPnPImpl
. D9 H1 Q& q' ?{
% ~! B6 F6 F. p. ^' }        friend class CDeviceFinderCallback;
' t1 ?; h9 u8 g        friend class CServiceCallback;
9 M: Y: [' g: U% C& B" m6 Z0 h// Construction
public:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();


0 z5 a, @) B$ U, e$ X4 u. @        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
9 l* u8 i) i  G4 c- u5 D        virtual void        DeletePorts();
        virtual bool        IsReady();
  X: E5 O' e( y, G6 n        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
  G6 C- [, ?  }- T2 Q, I1 D* n2 D

        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
# d* ?5 h2 \- \* X- }" N! a2 F8 n        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };


  I6 \* r. s. y! A, dprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
& w4 Y0 @. U% u' e* }        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
1 T9 y1 g" a+ C' B        bool        OnSearchComplete();
        void        Init();
6 W: Y, ^' x$ V, s9 \

        inline bool IsAsyncFindRunning()
        {
- V1 Y& P8 b) G2 |' W5 B                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
3 S5 d& f% f0 I9 L  x                {
' V3 d$ j2 t0 k) v! X& e                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
; z6 N$ s+ K  K! m2 Y                        m_bAsyncFindRunning = false;
4 [  O4 G: c( {+ S9 {9 S. V( Q$ M                }
                MSG msg;
# `2 D6 X5 \' W* t- B2 Y                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
4 `) k" X& t- m8 K7 N$ ?                        TranslateMessage( &msg );
2 f8 \3 B/ B; x                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
  B9 j" _8 B; E- d3 _# p        }
4 T8 u0 C; y, V1 i+ h' D& z

& s8 x4 Z( m! l+ B5 {        TRISTATE                        m_bUPnPDeviceConnected;


// Implementation
+ [, K3 B4 q* a- O$ |6 p        // API functions
4 Q8 J' `) G& l3 J( C        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
, E9 o6 f; ]3 U% ~1 X8 m) S0 a        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
( s* r+ ~9 L/ M5 D4 ?$ c* l        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


        TGetBestInterface                m_pfGetBestInterface;
8 s' a" z" E+ ?, \3 j0 r        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
  m8 R& v/ B% [( f) Y$ w
, ~* f) q+ v8 Z5 v
        static FinderPointer CreateFinderInstance();
4 R# [0 J& r9 m; W$ ^( V  }# P        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
5 E- R$ @# q( u! y' e                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
3 Z6 m0 y9 G- r4 ?! J' M                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
2 ?( r$ B) o4 ?$ [6 a0 L; K' J; Q0 G% K2 x  g
% J* E1 t8 M) D# b6 E. q" l
                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
  }. P- h5 r! S. d& {% y0 x- x- U; @" P, a

                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
% ]& X& d# f/ E, @! v        };
# F' m9 |! H% x; `       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
. ~0 C5 D: |3 s; }0 {% y        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
6 l" V" l; R& U" c! Z                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
) H$ g* M6 f2 N, a" r
& k6 p; M5 r, X0 o; N$ E5 }' J
" d3 \. k9 A% Z( G1 z        // Utility functions
( N9 _  _7 L& k$ ]$ v' H8 I2 w        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
! S. R- _5 j- C' V8 e$ M        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
6 o  W3 b; I0 K% _        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);


, b; G4 K& p4 S9 @9 k1 v0 H1 N4 T// Private members
6 z/ {% w$ V% O1 s5 Cprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
/ V$ Y! D* h2 w2 _  N. ~4 d        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
" h* x( k6 \$ E+ ^8 j9 _
" B! \$ V, F* k0 q
: g; j% n. i8 G/ E1 i/ \5 b" ]2 w  o        LONG        m_nAsyncFindHandle;
) d3 y  i$ b, ^7 x; T! e        bool        m_bCOM;
. O7 z; F" b  O  o% t. f: P6 {- f        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
& K$ @/ D) X1 F        bool        m_bADSL;                // Is the device ADSL?
/ k5 C, A1 t4 W0 x  |        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
9 F, Y6 U, t# i, Z9 v7 k) m2 O        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
/ r0 U9 X; f$ L3 U9 N2 t( |        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;


};

# H. s2 n+ T- ~$ L
// DeviceFinder Callback
. O& B; H. w; P7 ]  [/ Hclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
% W2 a( {5 p) y: ?- w{
5 B' S( l' r+ |/ k) h9 upublic:
. a2 z& L7 C, D2 G        CDeviceFinderCallback(CUPnPImplWinServ& instance)
- g/ V0 S3 X2 [0 G" j) E) u                : m_instance( instance )
' x2 b1 e( r, x5 K        { m_lRefCount = 0; }


8 r6 Z# y& f5 b   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

% v- K" [: F" U% X
9 }$ S8 R  g' n- @1 H# Q// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
! i, G/ P' T4 U: I6 O, O: Q% I7 C        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
7 @+ W+ ~- c* d% f8 M        HRESULT __stdcall SearchComplete(LONG nFindData);
. k" s% x  _4 `
+ }; X( W% C" `0 y' e$ o
private:
* P7 l  a3 X( r3 c6 k$ W: _( M8 q        CUPnPImplWinServ& m_instance;
( |2 o: i/ f# k( H' a" r2 m        LONG m_lRefCount;
. ?9 s4 E% K1 n5 x9 D4 t: v};

* t4 O) f. B4 [& I
8 N/ ]3 Y* o% a! B// Service Callback
9 L) q( Y; u8 C) b+ o: }& S7 \5 bclass CServiceCallback
        : public IUPnPServiceCallback
2 a. S4 F; O' c/ p7 z! q{
% g$ x, s) T. Y* r0 t  [  Ypublic:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
8 s+ {$ Z1 \6 |! X        { m_lRefCount = 0; }
5 `: B0 b" ^! Z& Q* p# Z   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

8 r, o! p- `3 U8 j* o
// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
" U, o( g! Y6 B, [+ v6 S! M        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
& V8 O* L  g7 R5 u* O- X

2 S& h9 W% P9 k1 F; ~$ E# S$ nprivate:
        CUPnPImplWinServ& m_instance;
3 ]# {6 Q) G3 E        LONG m_lRefCount;
+ Y+ |  V+ g4 b" y6 ^1 s% w5 z% }4 S* q};
9 Q) x4 `) A3 l+ s$ p8 \

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

# _- j$ w( u. M. g" c) B
$ s, i; W; y) N2 y: L* y- q/ ^使用时只需要使用抽象类的接口。
0 z  S" O& C% m" qCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
0 Z* D7 I! e* R1 C- A5 p% K7 c9 HCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
' e! [6 \! P2 u( \3 wCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.