UPnP

天云上人

/*uPnP.h*/

1. 
   0 _& F9 I1 [4 o4 P8 D& ~3 c9 I
2. #ifndef   MYUPNP_H_
   
3. 
   3 i# M6 t; k9 q  Y5 A1 b) T- b, [
4. #pragma   once
   
5. 
   ( z. Y2 L2 M3 Y5 L' Q: D/ m+ r. K6 {
6. typedef   unsigned   long   ulong;
   
7. 
   ) u9 v' l  X) k! p  g4 r
8. class   MyUPnP
   
9. {
   
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    3 U$ O7 y: u6 \7 w6 z- C
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    9 A9 S" ^' J% u6 ~
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. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    0 @' J  r7 G! ^' U
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    ( X9 |9 q% K4 p4 ?# G; T
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    $ ~' w0 \) F7 V$ {+ j
28. CString   description; //   Port   mapping   description
    ; D, ~$ K! B- I6 s% {
29. }   UPNPNAT_MAPPING;
    3 Q$ |" B6 u7 I  y& D/ h5 T' @- D9 [
30. 
    1 Y/ v0 w- R' b, n, c! |8 E6 p
31. MyUPnP();
    - O9 p6 `' \  P( }
32. ~MyUPnP();
      U# i, p( p# ^5 F6 S
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    9 v6 H: a% U7 X% Q' }
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    " l2 u; d* k* [2 t9 A: B1 e# _1 S
36. void   clearNATPortMapping();
    / ~8 O4 j/ y/ u: c  K$ y; g3 l
37. 
    
38. CString GetLastError();
    9 ]  [8 h0 V+ N0 J0 Y( j
39. CString GetLocalIPStr();
      }8 G. x4 V; {0 e5 A, E5 ^. h* v
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    
44. void InitLocalIP();
    4 o3 f, C7 ~& f" w- L- Y
45. void SetLastError(CString   error);
    
46. 
    % j- X* G$ B# L& K* j. i. @. O
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    5 d; V# g. `; W. w6 R/ C$ F( E
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
      v5 c# r8 R) h$ U9 h7 H4 `
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    
54. bool GetDescription();
    # e" U9 K9 g/ H4 J" \
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    ( `! m2 P9 t; C3 a* q- f
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
    ! s- H# N  A8 F$ v! _
62. CString m_description;
    + b8 M0 @/ B0 u
63. CString m_baseurl;
    ) E% ?9 T- o1 v  h4 s) E* s0 W" K
64. CString m_controlurl;
    ' B1 S" C- q8 l+ h+ v" v7 T
65. CString m_friendlyname;
    - b+ @  n9 F! I; \8 r  h3 A
66. CString m_modelname;
    ( C# K2 O& e9 _7 q3 s. E, ^. C* X# N
67. int m_version;
    5 g+ q+ S- x' u. X; e" q
68. 
    
69. private:
    ) R5 r7 S# l3 B1 a4 ?
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    4 J3 o# }* f1 j; }
72. CString m_slocalIP;
    6 N1 s8 W8 A9 o# n; l; R
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    # x4 N5 W, s: d6 r9 j: w
75. 
    9 d! }% t* Z3 `
76. bool isSearched;
    
77. };
    6 K1 L7 `7 Y% w5 A/ w( i0 b4 M
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
     h7 g- r9 Y8 n, h( S
4. #include   "upnp.h "
   7 H" u3 h9 ?8 X. k4 k
5. 
   3 _" |8 a3 m* T) d/ f0 t8 S
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   6 i9 R* O7 E4 g6 `" m
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    + k7 z2 r7 O0 D' [
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
      F: o+ L1 R6 q% k) _
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    + q1 Y) ], Z& b: N* Y/ y
15. 
    
16. const   CString   getString(int   i)
    - z5 Z- |3 o* B* y# S
17. {
    
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    & U7 e& H- i0 u: Y& c
22. return   s;
    $ y3 r: I: V2 O8 A8 K3 `8 L
23. }
    $ y& ^$ h- [; @5 Y* W& X9 r
24. 
    2 h5 y/ n- M. B" `. q/ B
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    # c, s5 Q: _' o. R. k% V: ]
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    - H1 d" l+ K1 B* i% E  @; ]8 L4 m
30. const   CString   GetArgString(const   CString&   name,   int   value)
    4 X3 \; v. N* Z0 G& k9 ]
31. {
    5 {# q  l8 n: d6 K( I
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    & e$ _- m' P2 Z6 [
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    3 k) H! d6 o7 E* U# T4 G
36. {
    
37. char   buffer[10240];
    & B3 q# \# D1 S4 V7 _: k) X+ I
38. 
    
39. const   CStringA   sa(request);
    9 A$ y0 B, N- y( O8 A
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    ' `8 ]% Z& h3 j! l
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    3 B, s, F5 h4 q: u% {0 {1 l2 u
47. sockaddr.sin_port   =   htons(port);
    / w3 I9 u$ {4 q' c
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ( S- b8 v% x% {0 p+ c; C
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    6 q/ ^9 \( x. w6 {# Z8 h$ e
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ' k" o3 W- A5 U' m4 q" y& t
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    " q, k; {" N9 Z1 ?7 g( S
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    1 @, o5 l) A3 a& }6 @
57. closesocket(s);
    8 E* S/ O- C4 a% V$ u3 a( A4 t
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
      j6 j% }" [% M# I
59. if   (!rlen)   return   false;
    ; D2 p8 ?4 \' t
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    2 J( b( g7 S. C! T7 r8 q) m6 k( f
62. 
    
63. return   true;
    4 p' D& [* y$ J0 U
64. }
    
65. 
    8 I3 J( a7 l" e5 F3 }& f  x* }
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    * h+ v5 _( s# Z4 L) K& z7 t
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    ; c' z! ?5 i0 B. A5 C, K' b5 {
73. 
    - Y; d, q7 Z# Y5 K# U
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    4 J1 N% I2 {2 O; G" i- H
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    3 ^) w$ [* T9 V, s
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    1 T4 o  Q6 }+ y% Q4 }& H
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    - f) a! y/ [8 |% t4 A
85. int   pos   =   0;
    2 m0 C7 N8 Z! g
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    # f1 Y- T( L3 Q: y, O& M& l: h* U: Z
90. result.Delete(0,   pos);
    1 z5 _: H& f, Y4 m/ S2 F
91. 
    0 J4 [/ k" {+ q7 r# c2 [9 O6 S( v
92. pos   =   0;
    6 g3 k- v8 N0 C+ E. a+ \
93. status.Tokenize(_T( "   "),   pos);
    , ~! D7 V. m" J
94. status   =   status.Tokenize(_T( "   "),   pos);
    . m2 M6 i1 E) i$ t/ U( B/ p2 ?
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    3 e$ k0 \+ _. R2 E$ X
96. return   true;
    
97. }
    * Z- ]/ g3 p. j- X( ]
98. 
    . n9 V- b; J0 y$ s
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    ! f. \  @) t6 z" D( J0 I- ~  F% X" ^
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     / o, y4 L5 |0 V# M& `- s: Z: a
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     9 M0 s5 I) g8 T% d4 v$ m( E4 B( J0 |
103. CString   property;
     
104. 
     ( U  h: m5 `5 `0 T7 Y& O# q- V% K8 z
105. int   posStart   =   all.Find(startTag);
     ! c- W9 t# m! p+ b& z  a/ m$ ~
106. if   (posStart <0)   return   CString();
     0 Q: r6 t, M5 d" ?$ t1 w5 o1 N
107. 
     , S9 h/ i7 x: J3 C2 q5 l( c9 y
108. int   posEnd   =   all.Find(endTag,   posStart);
     , v- @7 g$ k0 M! v
109. if   (posStart> =posEnd)   return   CString();
     * c; y8 o  m3 Q( I# ]
110. 
     8 G5 L5 h' l$ t# ]- U- T
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     , f1 J8 P: \, Y: U! U/ j
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     9 c# ~* m; e! h
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     % P+ R% ~. e/ [9 C) j- _
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     0 T* t) d, h4 P3 w
128. }
     ) ]- U$ \3 x" H& f4 c
129. 
     
130. m_Mappings.RemoveAll();
     # q( N' O" i+ o3 Z4 y
131. }
     " @4 C; {( u( ]; b. d
132. 
     4 S* p8 z% ?- s9 X; L
133. 
     : H: f* n. k: c
134. bool   MyUPnP::InternalSearch(int   version)
       I9 H# v1 p  @/ m! V! p4 A
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     7 Y% p0 k( z# w1 ]5 J  l; ?6 x
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     , ~" s- j) m( P$ P
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     ( E( n3 e, T' X5 h
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     ( ~- W2 j* A) w: J
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     : S2 f; n/ F9 S2 ^, B$ d6 q
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     ' Y' C$ J( O' h5 Z
152. if   (!(i%100))   {
     * A! D9 H5 T: `
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     0 |) l; {7 k4 F. P7 F
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 "),
     ! d+ N( i7 l4 ~
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     # f% M4 w8 p8 s/ O* A# M
161. 
     / A8 H  ^6 S5 L/ |* E- B3 m( k
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     0 m- J* C' q5 s: n: [! m0 M1 w. a
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     / S3 i# t: t: |3 ^) S) ^( e( V8 U
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     + b8 H8 ?; w) H, c( O
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);
     ! x" P6 {1 y8 Y( u: M& z! x4 H, o
175. if   (result.Find(m_name)   > =   0)   {
     8 l, c* l: ?3 M% s& k, h4 I: A
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     9 G1 e9 W) S4 X
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     8 ]* y) L9 H7 t1 Q, p. G
186. }
     ' F& B- l' ]- u* q! _( b7 B
187. }
     3 \) h# J6 ~, d' [+ G8 |3 F) O) |, h
188. }
     
189. }
     6 J+ Q6 D- T4 F4 g* L
190. }
     
191. closesocket(s);
     ( }5 m) V  M; I0 M. z& g
192. 
     2 R2 B  m8 Z4 {: _8 _2 \* R
193. return   false;
     
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
5 O: W: a6 @/ ^& d4 a6 k
( S( Q, P! R6 n' M" ?; s
; v2 ]4 T3 d" `, q///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
* H& j3 S! o! a+ \2 m

, c" U. R# p8 w- _8 z/ A& ^#pragma once
+ _. {% T0 Y& W9 P6 G#include <exception>

- h7 m- D5 y  z3 \9 |" R  B
  enum TRISTATE{
0 `; S6 I# V' \        TRIS_FALSE,
2 C9 o2 G( t$ @        TRIS_UNKNOWN,
# S$ N8 a" K, a1 |: K% Q3 x        TRIS_TRUE
};


enum UPNP_IMPLEMENTATION{
7 S9 C1 P- a0 M& ~5 h- u( m% b5 ]        UPNP_IMPL_WINDOWSERVICE = 0,
0 s0 C% Y2 a4 x1 u% G6 _        UPNP_IMPL_MINIUPNPLIB,
  n( \& R8 a9 k        UPNP_IMPL_NONE /*last*/
7 h- `/ x$ I) k0 Q};

. ~, [1 a8 H9 \% ?0 [6 d


+ [# K& O1 f9 {0 s* y1 x7 Yclass CUPnPImpl
{
public:
        CUPnPImpl();
3 R$ d3 b0 j; w& F3 h        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
! Q5 b% e; K& W8 l: s5 p7 M        enum {
1 p3 c" ^) r3 w% E7 p                UPNP_OK,
8 ~! u7 U9 B: k$ G9 o  \4 l$ T                UPNP_FAILED,
                UPNP_TIMEOUT
+ y& x5 A  H0 a3 G$ B9 M        };
0 t- V, t3 Y% N: B9 |5 f

0 Q$ i. [. ?& n        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
  _! |) t" A4 t% J! ]+ c0 n, [8 |        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
+ C! t: E5 S, g2 i" J        virtual bool        IsReady() = 0;
6 F, c  @+ {( H6 F1 F0 k        virtual int                GetImplementationID() = 0;
       
3 z( f1 \& u: l1 C* N( d" `        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
: H. |4 h5 y$ T; d1 H
  j& C5 f, ^, N2 v
: F4 ~+ [' G, J, [+ r        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
* G# A! f3 a+ G8 D$ H7 C        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

7 S0 P/ T, [. Z: P- d6 P
// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
" l7 ]. u3 p  q! r, }        uint16                                m_nUDPPort;
1 T4 o7 V( g+ i3 d        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
5 V, X: S8 |7 ~, s- p        bool                                m_bCheckAndRefresh;
) [" G. v+ J: h/ E2 [- @6 Z) H7 z! w  B
3 O# s8 A5 p6 _3 c- d
% V" z2 x! h9 E- h9 Qprivate:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;

* X& |9 A  a/ A' a
};


// Dummy Implementation to be used when no other implementation is available
' v" m% o" ?0 D% X1 j2 Vclass CUPnPImplNone: public CUPnPImpl
; ]' d9 E* ?4 l* C# `{
6 e, W* m* n! R( b6 e9 n6 Opublic:
3 g0 x" P! W" M! |( \  _9 d) s        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
4 f5 t* ]" ?7 r9 x4 o        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
) _$ q# w; n: T# @6 d
7 N5 q& [7 J, j8 S7 G$ x( V* T/ _
" r3 U: q; F( n3 Q) U: @: b, N/////////////////////////////////////
  b8 C- l4 m% f//下面是使用windows操作系统自带的UPNP功能的子类
' N) M' G- _2 t* c
- k! H8 y: J) s% y
#pragma once
#pragma warning( disable: 4355 )
& K$ {2 r( A$ d6 g; v6 j3 q
+ a4 G% q, E1 x& ^8 s% S) _, U& {
/ z6 [0 n7 P* r# s#include "UPnPImpl.h"
#include <upnp.h>
3 D/ ?% Y% ]0 C) h* \0 t#include <iphlpapi.h>
% N; {9 H, W' _  }#include <comdef.h>
# W. Z# L6 ?3 d6 C8 d- _; D1 d#include <winsvc.h>
. q2 `: s* {9 T. X9 f0 D8 T1 S
$ B5 [6 G3 u1 g3 y( R* X) ?. ?
/ R% }9 D% C) Z#include <vector>
- e2 V2 ?" t& p8 E. s#include <exception>
: i( U4 F$ j# D3 I6 l* \* ~9 Q* d#include <functional>
2 A- w( s( B1 `! X1 P: V; K4 ~
$ ]: a! ?7 D7 s7 V8 H# g2 p
* r! F4 m" I! i& t3 q5 h# q

; h3 [3 [( Q8 ]; ?5 A; j+ R" }typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
1 A7 H4 f6 A; m/ V. L9 }( q) W1 {typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
. R7 C  n3 l# C/ |  wtypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
& d9 V5 L( N( E: _! \0 F% Stypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
" t* {' D; t- M. \" @typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
7 X, O5 B% [7 _, x* ltypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
+ m& d0 m- M$ H  L8 B);


typedef DWORD (WINAPI* TGetIpAddrTable) (
  K6 y* t! x: ?+ j  PMIB_IPADDRTABLE pIpAddrTable,
0 a4 Z  q, |% S9 J& B! b  PULONG pdwSize,
5 L2 T% d* @0 J3 r' f6 H/ s  BOOL bOrder
);


typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);


; B2 m# f7 }+ Q" \0 z2 v" t0 TCString translateUPnPResult(HRESULT hr);
5 {" m3 `  i; p$ x  M3 T$ w. G- u& yHRESULT UPnPMessage(HRESULT hr);


0 O, n# d! k8 `) i# g" jclass CUPnPImplWinServ: public CUPnPImpl
6 }  U5 t! |# ^" H+ G* G. M{
, f- i9 I" o' y0 `3 v5 T( B        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
, g8 U3 b+ V8 h& `0 L9 Bpublic:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
* s4 V( i% N8 Y/ G) g
( C( `3 i, U. h, E" }
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
9 W: G+ W/ C+ y  @; T) l        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


+ k& ?+ K$ p- v" }6 i3 [8 o6 o: v        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
, O7 K& X, k9 l+ y; Q; b
8 C) ?5 {6 f! z% L0 n* S
3 h& A+ h* t/ f8 K2 m1 Zprotected:
6 P) G$ ?; \. q2 A8 a        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
  M" L; R3 W; I8 v3 ^        void        RemoveDevice(CComBSTR bsUDN);
7 O5 [2 J9 l0 B3 b1 ]        bool        OnSearchComplete();
        void        Init();
1 f1 I5 }) x% b; z: P7 }( G
2 l+ L  ^+ \# w9 a2 ^9 a0 {  v
        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
2 _- m/ ?, J8 V* d                        m_bAsyncFindRunning = false;
                }
                MSG msg;
, t# c7 B# z& H% D* e                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
8 p9 `" T' G+ |$ f1 l. }2 L2 a        }
/ x  u' J- n* r# b2 _, S. `4 N
5 x1 o6 o3 I) t6 n% D9 k/ |* c
' D' n0 z5 S" k( ]1 z0 j        TRISTATE                        m_bUPnPDeviceConnected;


// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
$ c5 ^3 S* A, X9 ^) U8 g$ O2 w6 G        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*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
& X+ G. Q* n* o% W: _

        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
9 X7 g* w+ d. l5 H! }        TGetIfEntry                                m_pfGetIfEntry;


* I9 X# E5 Y# ?! d- ]4 M: _/ q$ J, m        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
, [' R. c" q% J  U: o        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
3 f/ f3 @0 i" k8 q7 e- g                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
- R2 X. R7 B- V
, G% W: e3 o. G# @  I& f
                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
/ ~% q2 a" `6 x$ H
' @; ]7 g) \. Q2 V5 F
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
2 V$ b# M+ c5 w                CComBSTR m_udn;
& V. }( P" l* p& `3 B* l        };
" Q! q5 j+ F; O7 Y2 V) t/ I' y8 g       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
' z% e* K7 D8 s9 R% }! M5 q4 h1 A        HRESULT        GetDeviceServices(DevicePointer pDevice);
* _5 O0 U) b2 Q4 A3 n        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
: ^6 b5 T1 @! X8 _        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
, U5 C1 ]3 X+ R0 F- Y6 a+ }6 k                LPCTSTR pszInArgString, CString& strResult);
$ _1 e9 P. w* j2 k) A        void        StopUPnPService();


        // Utility functions
6 y! b7 Q/ ~6 P* h$ @( }# a        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
- D3 ~8 k9 V. |1 t" H1 b8 [        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
/ f5 c2 o0 Z/ ]- s9 P, p        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
+ a; ^$ ]/ t0 j9 [( q( O, ^

8 M4 C& _: E2 v. ~) |9 v// Private members
) o" P7 R6 M" Nprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
6 B# k6 d4 \/ Q9 t! [        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
, Y  Q; M' s! ]  }        ServiceCallback                        m_pServiceCallback;
% C1 s% W- r* r/ y% N$ O4 M
; a0 w8 q% M8 G5 U
        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
0 l7 _- R! q. }+ K* n  B% e        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
: g2 t1 Y7 |/ \. }  t        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
) H' L1 |- @% C( P3 I        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
3 s$ r2 C" ~) G6 \$ s+ I$ z3 {* C! C        bool        m_bSecondTry;
; ^2 N- X: s( ?        bool        m_bServiceStartedByEmule;
9 P& f& ^- C+ V/ Q5 ^        bool        m_bDisableWANIPSetup;
( y/ f0 U! L% Y        bool        m_bDisableWANPPPSetup;


2 p8 j+ l( S8 J5 V+ ?$ d! g};
( K( \5 U/ f. T; P6 h9 ^" Q! N

& p! U3 @; k5 B// DeviceFinder Callback
7 ?# N+ \3 T# U+ l, [class CDeviceFinderCallback
$ H# G9 j) k8 A" |0 v        : public IUPnPDeviceFinderCallback
3 ]( a! ?) K7 ]4 w6 i* [{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
6 D; m8 J. t$ H- d                : m_instance( instance )
        { m_lRefCount = 0; }
# ]( {7 j2 ~9 Z5 l$ u3 f

; }" N, s6 P" B/ \   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
; i- K; e( X) R, A6 O   STDMETHODIMP_(ULONG) AddRef();
- g, R' P$ E+ s3 |7 }   STDMETHODIMP_(ULONG) Release();

0 y3 k( T1 g* C9 O9 P
// implementation
  Z9 v( s! p6 G* z; Y" W0 p, Gprivate:
& r# ]9 d8 B, r6 h! N% y9 a" ~3 I        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
$ Q8 v& \2 q2 ]' E% M" u2 Z        HRESULT __stdcall SearchComplete(LONG nFindData);
/ F7 Z' t, k7 K

private:
        CUPnPImplWinServ& m_instance;
* W1 \8 d/ }$ S, `        LONG m_lRefCount;
9 x8 ~# Q( b& H};
3 `* t- W) b: Y
- b, R+ s, H( @1 u
4 y$ |  x% B+ ^& O// Service Callback
! X* R$ w4 L2 v% jclass CServiceCallback
        : public IUPnPServiceCallback
{
- x  }' A& l6 K  K: apublic:
$ R' Y, I9 N, d, F# g( Z3 [        CServiceCallback(CUPnPImplWinServ& instance)
. H$ D- H, t1 j: F2 h: H                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
1 [+ n* @+ A) R  J$ q; C   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
9 S) n/ |% [% O

& \9 l& ?) z0 y0 I// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
. u! ]  ?! p* l8 s+ N& P6 P        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
# y& v! b" a+ [8 c7 f0 w
' ]: o4 K7 J4 e7 |
7 t9 G$ D; {) oprivate:
+ D7 S8 a- P  s2 \% N+ m0 B$ X        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};


/////////////////////////////////////////////////
& z# N, k% E2 F% n
( I2 s5 A3 o+ F1 |- n
使用时只需要使用抽象类的接口。
5 t4 q' Y, v( Q; p; @& M2 M$ wCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
: m7 z5 h4 f& Y" k  YCUPnPImpl::DeletePorts删除端口映射.