UPnP

天云上人

/*uPnP.h*/

1. 
   6 S3 t4 l  [5 O" E+ l: F
2. #ifndef   MYUPNP_H_
   7 ?3 U& X0 @7 _
3. 
   
4. #pragma   once
   3 A, L, _$ r* I1 f  Q
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   ! j7 D3 T/ q: m+ u
8. class   MyUPnP
   
9. {
   % u- ]' q# ~" H# w
10. public:
    ! q' H2 I$ {; K; J" ^
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    , z) t# B3 I+ |  O$ b# x
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
    # h& ?5 {7 i; R2 T& [4 C" h
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    7 K5 U) W- r/ \* e( S
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    * e  T) M9 N5 N3 k. V
17. }   UPNPNAT_RETURN;
    
18. 
    0 f( T- ]7 V1 v& n. y$ A. J; D% a
19. typedef   enum{
    , z; x9 W& A+ O
20. UNAT_TCP, //   TCP   Protocol
    0 `, [3 V7 f+ Q
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    1 W: Q$ }8 v5 m1 i# h/ |
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
      ~3 e: Q. |0 t
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    ; z: h' S1 S* g- M, {6 w
29. }   UPNPNAT_MAPPING;
    " ]! f. H- v1 V
30. 
    2 T; Y) K) k# |  N( K
31. MyUPnP();
    
32. ~MyUPnP();
    : @  G( N8 h" ~. f$ z
33. 
    , w3 j% o9 I. M2 G6 {' x! A
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    7 T3 U* Z5 l+ `7 s- _
36. void   clearNATPortMapping();
    2 D; p9 p( E' ^/ i7 K# k
37. 
    7 j& l5 ]3 s4 ~+ I! ~) E
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    9 [+ u1 L8 `( p
41. bool IsLANIP(WORD   nIP);
    
42. 
    # H  ]1 u6 ^1 p- ?" B" J! _
43. protected:
    ' \0 z- k% O( K3 Y8 A1 Y; @
44. void InitLocalIP();
    $ f3 M) {( {* c8 u, n7 z& T1 {
45. void SetLastError(CString   error);
    ( V/ {8 n& L( C% Q" c3 D
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    8 n& i% G- v( W' n8 W( d5 x
48.       const   CString&   descri,   const   CString&   type);
    + J0 G0 z$ j" k* o9 B% Z
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    3 @7 T  U* P* \! Q
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    # u4 q& X/ `( x+ ^8 }
53. bool Search(int   version=1);
    3 d5 W  r; b2 [( ?0 Z6 M
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    # L7 Y) Z2 x0 y# w. M
57. 
    5 r( U# H, q2 b- j$ T7 Y
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    $ W8 L* G% X5 |, g/ R! M
60. CString m_devicename;
    4 A, _! F7 _1 }  ~! V
61. CString m_name;
    
62. CString m_description;
    ; o  A- g3 q- ^8 {2 I8 I, ^
63. CString m_baseurl;
    ! R& o- d1 q$ H  u- B) q) u
64. CString m_controlurl;
    ; Y, o; y& Z* {7 b- u
65. CString m_friendlyname;
    3 Y0 C1 I, Q" l6 c! ~1 r* w
66. CString m_modelname;
    
67. int m_version;
    
68. 
    3 \2 `2 ]8 X  Z
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    + m% V9 n& S. |8 M, ]3 E
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    9 ?, n; Q& d6 O$ q/ M3 d1 _
75. 
    
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   / s' ]$ C3 o/ E. e+ Y
4. #include   "upnp.h "
   2 v* ^3 A) z% f" @; [) g% w
5. 
   0 b* ]8 i# K  U2 U8 x
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   & @, k2 p8 D5 d5 n
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   - q- M. D" a$ f$ p9 W# t
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    ( ~! o' K7 L$ ~/ X, c* J
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
      _/ b0 w4 f4 k# Z
13. static   const   int UPNPPORT   =   1900;
    ; \. z2 l5 j& t+ F+ T; o
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    * g7 }$ j  A7 x- m
15. 
    
16. const   CString   getString(int   i)
    5 a8 r" b: r% M. u: N
17. {
    * j% v: @8 \5 ~' f
18. CString   s;
    " ?; o$ L& c4 m7 u7 r& O
19. 
    
20. s.Format(_T( "%d "),   i);
    # p9 u! Q: K0 U
21. 
    
22. return   s;
    ( K; {' s, }; Y
23. }
    5 R5 t  E# R3 l9 M0 ?
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    / i( Z- `  h1 z* ]4 k( o! N
26. {
    % F; k; v$ N/ z  {
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    " ]- D1 H& @) I& d0 P! A8 e$ \! o; Z
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    8 p* H% }3 c  K7 a, s" _/ \2 \  A
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    ; C3 V, l" x5 b. E
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    8 \: Z5 x+ c1 V8 S& A0 \
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    - ~7 g7 c$ \4 X$ w  _9 _3 N
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    - H) f  C0 P2 r% ]+ s
43. uint32   ip   =   inet_addr(CStringA(addr));
    + A% Y3 `' V& Q3 L" Q  E4 A. Z
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    % p* n* y) _8 Z
47. sockaddr.sin_port   =   htons(port);
    ) i) t) x6 s: ~
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ; w# Y* z2 `& G7 P: F) |* p
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
      E& n; d0 s0 p
51. ioctlsocket(s,   FIONBIO,   &lv);
    2 i5 X- \0 w5 D2 O
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ; Z6 ~( ?2 c3 k! b+ \: v
53. Sleep(20);
    6 L, y) e( I; j2 W' Q: [
54. int   n   =   send(s,   buffer,   length,   0);
    ! ?0 _) o8 B3 R9 a; [  f$ ^
55. Sleep(100);
    - Y8 [1 m6 |0 @9 v+ _
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    # P  S' l# i+ Q. Z
57. closesocket(s);
    9 h% _, h% Q; F
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    7 s( Z% T# `* Z6 J8 |* c- f. l0 D0 S
62. 
    
63. return   true;
    
64. }
    
65. 
    # Q; ]+ n* s0 N; @# v
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    1 G9 }- Y$ w1 D0 d
67. {
    
68. char   buffer[10240];
    * O/ {7 ?* y/ E3 k9 `# P7 P
69. 
    0 P4 p; w8 L4 v2 B9 f* S  [/ R
70. const   CStringA   sa(request);
    4 ~6 o6 u7 A: i! [
71. int   length   =   sa.GetLength();
    ' ~, Q$ F/ c. p
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    7 ]# _% d! P3 l7 d; w7 j
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    , c4 H  f6 p% `
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    4 x) U2 ^) W) Q4 `$ v' c9 X
84. {
    
85. int   pos   =   0;
    
86. 
    8 {$ I( W6 y0 }2 K- D  `4 f3 \
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    3 Z8 D% a/ J. r- D9 Z7 c
89. result   =   response;
    
90. result.Delete(0,   pos);
    ' o$ z/ H" j  \+ Z( H  f- t' @9 i
91. 
    4 @$ |5 s7 d9 N. K
92. pos   =   0;
    8 f* A$ S% s1 l% Y
93. status.Tokenize(_T( "   "),   pos);
    ( y) B7 t* \& ~+ k/ T/ c
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    6 ]* C9 y! b4 r1 k
97. }
    
98. 
    $ i6 F; b! i, C# g+ D, s+ E- L4 R  f
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    * M: v2 R- f+ e- f; N
100. {
     ' D" u. i0 k0 P9 P  J0 |; ~: ^* _
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     4 E3 H; @7 C4 |  M* h
103. CString   property;
     3 M* |% M9 |, x% t9 O$ n! D( s
104. 
     
105. int   posStart   =   all.Find(startTag);
     ( A' M7 u/ b8 O8 O: p
106. if   (posStart <0)   return   CString();
     ( k5 c% D# q/ m/ H5 J7 A$ L
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());
     
112. }
     
113. 
     6 D% a7 P! l8 a# z
114. MyUPnP::MyUPnP()
     " P3 B5 h* ^% O- [
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     ) Q' a; Y$ O; y
120. 
     $ [! ~$ P4 i5 u' O( ]4 E" l% H. ?
121. MyUPnP::~MyUPnP()
     % b" [" `# K4 N# Q7 Q" A6 P# c
122. {
     
123. UPNPNAT_MAPPING   search;
     0 j8 _* q* E9 H9 R, }: Q
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     - U/ Z. O: [7 ]- n, O. |
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     0 C) n0 b' I& h3 V  c
128. }
     7 p2 M+ f" Y  L$ d/ I( P6 N
129. 
     * s: Y; H& W$ B( l/ m9 m4 j4 g
130. m_Mappings.RemoveAll();
     
131. }
     . R) U) N. a- M" r* N
132. 
     / R9 x* \; K1 x! v: k  Y. ~9 j
133. 
     2 j5 H  x, z. h# i- D9 n3 j: ]
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     ) {( \# L& k5 W# }  n9 |. F
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     * F% q9 a/ _) ^9 U  S# m' m
141. {UPNPPORTMAP1,   _T( "service ")},
     : {" M) `+ J4 V+ V
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     : y* E" D* M2 A5 q, Q
144. };
     
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;
     2 o4 H2 m' Q+ ~
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
       A/ W4 T# k! t2 U* t1 g
152. if   (!(i%100))   {
     * U, O3 l+ L1 Q( K
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     " l6 ~; z0 j4 q( W7 R7 `
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     - _1 \2 j, \: W1 F
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, ~) r2 z1 h3 l% J  `! R  t
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     0 Q" Z" f5 _2 [% p
159. }
       G- Z% L& g6 Y% y3 N  _
160. }
     
161. 
     , A' x! I* r% `9 Z% T# l% b& _  X
162. Sleep(10);
     
163. 
     ) D" N: Z/ L9 O& r4 y
164. char   buffer[10240];
     : H- W' y* K) I8 Y: U0 L, P
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     1 _. i, Z. \" J, i$ ~
169. CString   response   =   CString(CStringA(buffer,   rlen));
     6 }7 M, o+ [) R( [4 s
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     # y, @/ n; n, C; 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);
     
175. if   (result.Find(m_name)   > =   0)   {
     # [: R8 D6 j, R0 H$ l
176. for   (int   pos   =   0;;)   {
     8 T1 Y8 |: y+ F, P! I$ R6 ~0 D
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     7 y9 S% K) E% y# l: a. v* k
180. name.MakeUpper();
     9 V8 E1 J9 `2 h9 `
181. if   (name   ==   _T( "LOCATION: "))   {
     " }; K( f  a5 e
182. line.Delete(0,   9);
     % u6 b* K! @3 e7 a8 C; v
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     # L4 k5 s, O& H; @% ~7 u
187. }
     
188. }
     
189. }
     1 q) ]% r+ L+ o; A& k, }
190. }
     9 V" e2 G* h3 f+ R( x) a& n
191. closesocket(s);
     / _; g9 x7 Y" x" `4 f+ g
192. 
     5 C  p5 |( ^1 T7 W  |5 k& ^
193. return   false;
     
194. }
     $ C  A5 r" F' G- a: J& B

复制代码

天云上人

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


$ L# w0 g9 s' k$ N; }; V///////////////////////////////////////////
' j8 n" Y$ h8 Q1 ^' E. O# }+ R% K//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
9 W+ O6 q% \, y. D
% m9 e5 u# @( M) v
#pragma once
# Y* f! R' @6 c# T. D4 P#include <exception>

, }! B+ ~% z; [* W) q) R1 H+ t
  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
; j8 ?3 L( ^/ r5 f) }+ ?        TRIS_TRUE
( [# W# [! V! H};
! a8 a. _! ]# F$ V* Z
" F# W6 Q; L( Z
" Y1 ~6 X" S' f) A4 {7 X: P' venum UPNP_IMPLEMENTATION{
3 O( T8 J$ S5 ]( o& f' |3 s        UPNP_IMPL_WINDOWSERVICE = 0,
, v" }, R+ c' I0 q        UPNP_IMPL_MINIUPNPLIB,
7 B  H; ~1 Z3 D4 Q9 {6 `3 O1 H        UPNP_IMPL_NONE /*last*/
};
2 h/ B2 T! C1 A1 {
/ T2 Z- y4 X8 b$ a
) b; l/ T$ l  T; L% x5 g5 E. ?
2 ]8 m" z9 A6 J* H$ }% y. g* w
, Z  k8 H, G8 u, |; G. v% Hclass CUPnPImpl
{
public:
        CUPnPImpl();
3 F' Z2 F( O+ w# G( N        virtual ~CUPnPImpl();
: m% Y% M3 r  w8 G( Z9 [# F        struct UPnPError : std::exception {};
. u3 c) n; l( \2 K% t  ]        enum {
3 J) `0 j2 [4 E                UPNP_OK,
, S8 x* W: w. ?2 c& e                UPNP_FAILED,
                UPNP_TIMEOUT
6 `+ p9 v$ G$ q) |        };

' t0 W# W4 ^6 D/ a
3 y: |) P" b* I8 |- N        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
5 m$ ~. k" Q; ]6 `' M        virtual int                GetImplementationID() = 0;
( `2 N0 h% c7 ~6 L0 D- Y       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

+ X8 r0 ^- S5 ~" g3 f/ V
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
8 K, d' J" v' S        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
- X9 X& T5 n2 l; N: |        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
$ y% r: J% A6 ]- i6 h- ?        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

6 o: M& \8 g9 U7 U
// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
1 e) X$ [/ V% ]/ R6 e        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
# `% \$ T; Z# U        bool                                m_bCheckAndRefresh;

3 W$ W' R( q! \9 N' {
private:
2 W: @2 N  W9 |; A& D/ _3 d/ P+ a& k  r        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;


" o2 Q) B. U6 L; O2 V1 V/ p};
9 X* o2 L& X7 c" h& ?! ^

// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
4 p1 _2 v: J, f) \& k% Z' p0 p{
/ i& A3 W& A" Y- g( Y4 [public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
. g: b* j4 s- |: R        virtual void        StopAsyncFind()                                                                                        { }
  W4 k; I* _# m& a2 ?( `: w/ y        virtual void        DeletePorts()                                                                                        { }
# p9 \* X. V  ]+ p* }5 q        virtual bool        IsReady()                                                                                                { return false; }
4 t- ?/ r5 j8 Y$ `- j3 A        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
5 a' T# h7 w# f' Q3 S$ P& }};
/ C; u9 f" k0 N! t9 B8 J* g
7 W: l6 b" W! G5 Q4 ?' F  D: z
/////////////////////////////////////
  `9 ~6 Q6 f, y; Q- `//下面是使用windows操作系统自带的UPNP功能的子类
( L) Q% h2 g4 {; O
; t2 E$ X0 W0 ?  d5 s3 n
' c" Y: D7 S  {4 p, W: J. n#pragma once
$ S+ a: l/ H3 D3 u, z5 {( Q#pragma warning( disable: 4355 )
( j  f9 B6 I5 p2 M: A

#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>
; r* u! L8 L3 Q) U

4 ?; y- I. g6 |$ [#include <vector>
#include <exception>
# R0 q% v) _; l8 @  M#include <functional>
1 H1 I: e. z2 y, V
! U/ ^7 u' x" g
1 X* t) j4 c- q

, z8 z$ L3 t, \/ G$ ~/ ttypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
' L; b( Y; B6 o2 V* I  y$ Utypedef _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;
% R2 k' v! S% S' ntypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


2 s! m! a8 U* ^typedef DWORD (WINAPI* TGetBestInterface) (
- [5 y# A- z0 e" G" y  IPAddr dwDestAddr,
$ g5 }- @: G" a/ P  PDWORD pdwBestIfIndex
. ]1 v8 e3 h! ?0 p+ S7 p& K" y' b);

7 Z) K% {# z# ^- I7 O  O3 ]
1 q5 q' U4 n4 i$ q! {. Q7 y  Otypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
);
0 H( E/ |! A2 ?

typedef DWORD (WINAPI* TGetIfEntry) (
5 ~: S, z- \' V+ I* {6 m3 n  PMIB_IFROW pIfRow
- G' B. z) |" b- v; j9 n7 j);
0 t5 G7 V7 g5 K! E" S. K
% X2 |/ ~! R- p  {' b5 L
CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
( d1 Z; A; |+ o/ ~0 C1 a

class CUPnPImplWinServ: public CUPnPImpl
* v; c- E+ z* V4 l5 G{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
public:
( n6 E, p3 d: c6 L5 v4 o# q        virtual ~CUPnPImplWinServ();
9 x4 ~! ^) R0 W( z        CUPnPImplWinServ();

, O& p8 f0 ~. X, @6 [/ L- J
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
7 K$ ~' N  M5 E; {7 _        virtual void        StopAsyncFind();
" ?9 a% h5 v& z6 Y; D  m; N( P        virtual void        DeletePorts();
- c! ]* a- V* l% t; Y        virtual bool        IsReady();
% L6 Z6 Z9 E1 h- J8 |+ _' n        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
/ y8 x- F1 e7 o; q4 w8 O7 |

        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
" ^1 `4 r4 x: r& I- z' y        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
# A/ E6 }; W9 j, E        virtual bool        CheckAndRefresh()                                                                                { return false; };
8 t% G0 d% r' a4 ?: ^5 H) I
) @" o- _! k8 x. n
protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
" A& v! x0 g  {/ O        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
* Q: }  o: M/ K, w/ e3 N        bool        OnSearchComplete();
7 ]3 M, ~' ~1 h& x$ q        void        Init();
- L, l" w" p1 i4 x) @8 K
% g+ d) ~! l% K' J$ i; ]
        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
0 S1 ~7 [) {  p1 {                {
- m! r* k) ?/ Y+ H/ r                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
" e0 D$ E4 T8 m% Z4 H$ f/ i                }
; X) I. |8 k( \3 p' ^* J                MSG msg;
0 X4 p! F" w$ I2 l8 z5 G; A                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
2 _) |" e" U- i1 j# }- b                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
% c; O2 V6 }  b/ ^8 _, D8 @/ M                }
/ x/ `; D8 k- h6 q                return m_bAsyncFindRunning;
        }

4 _1 }# B& R- Q* m
        TRISTATE                        m_bUPnPDeviceConnected;
, J) _' K( z+ H  e
3 j, l" q& r! b2 |( r9 P3 x
& W" l: f, p0 s; \// Implementation
        // API functions
! T7 q0 o; B( k0 g        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
# o4 j0 @8 y/ Y9 f8 Z        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);


' G0 u5 \8 h; L( I( d# n        TGetBestInterface                m_pfGetBestInterface;
3 l# [( u' f- j0 q        TGetIpAddrTable                        m_pfGetIpAddrTable;
$ s( l+ D' ?1 F        TGetIfEntry                                m_pfGetIfEntry;


0 B. m3 `. R& |) N0 h4 r  }        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
6 d) i( t2 B0 F+ d                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
4 l0 G! z. H2 q. B: Z( I/ G                result_type operator()(argument_type device) const
                {
5 Z" @  y. z9 F0 y# y  x                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
, W0 Y5 q5 B. ]0 c% k0 I3 E

! D" ]/ Q5 x+ Q: u0 o; p% h                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;

" L3 S. V* k( p: x8 \4 B4 e4 Q
. w2 G: e6 d7 H5 w8 w7 n# R! }                        return wcscmp( deviceName.m_str, m_udn ) == 0;
- N$ p9 P: b1 Q& c                }
                CComBSTR m_udn;
        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
6 Y7 W$ |# y; V" n3 {7 |. v        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
6 ?; u5 j2 V" n8 t' P8 w2 Z' I        void        CreatePortMappings(ServicePointer pService);
  B, s9 b) A( K$ ?9 A: F8 |$ c8 |        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
- C$ h0 V6 M5 J7 R0 d  z% u3 U        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
" y' ^+ L& B4 h" C* R                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
5 R- L7 Z" i7 g: W
! r4 I" m/ w$ h0 [! f
        // Utility functions
; ^. y, D. V$ j, Z- T' R        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
" D  ]) b& `  L1 T9 B& j        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);
) p2 O% U$ c5 ^$ H, x, t3 d; U        CString        GetLocalRoutableIP(ServicePointer pService);
- B0 ^7 U: {' F$ m, W
1 ^1 Y/ X/ k9 @5 f  C
// Private members
& x# H# R( L& Qprivate:
' ~, t- s1 r: D% ?1 ^0 m        DWORD        m_tLastEvent;        // When the last event was received?
+ {. r1 t2 U& S2 O& z        std::vector< DevicePointer >  m_pDevices;
$ ~: j& x* q, z3 R( R& D6 e        std::vector< ServicePointer > m_pServices;
! a+ p& `! i) I( e$ P        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
& L8 J. a4 t/ e" [9 r: d2 }
: g/ Q. {* L1 p: n
5 x5 _& `; s7 N$ t4 ~6 H( P        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
. r& j% ?% Y4 z# B7 }+ e' l        CString m_sLocalIP;
        CString m_sExternalIP;
  l' O! r3 g! w/ _) U8 V. B5 }: x        bool        m_bADSL;                // Is the device ADSL?
/ x) [) g( X! t* l        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
; y; O' A) s7 S3 ]& d        bool        m_bInited;
" Q2 i3 K* b4 L5 t        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
- c8 O" T2 Z' }) G- _* [        HMODULE        m_hIPHLPAPI_DLL;
/ ~; ~! P1 v* z4 t2 R0 z8 y+ V, l+ i( v        bool        m_bSecondTry;
9 L! i8 m* n! M        bool        m_bServiceStartedByEmule;
) j: j; k3 B5 |        bool        m_bDisableWANIPSetup;
1 ?& _. m6 }( o" {6 q2 M        bool        m_bDisableWANPPPSetup;
3 B7 m( ^6 T" s; A3 w/ B7 U
+ z4 D$ X5 M8 V' }4 _
};
! |/ i2 U  ^$ x, _
; f  `' _. W  f( C( e* F$ [) u
$ E0 @' J  f3 A* Z) h// DeviceFinder Callback
class CDeviceFinderCallback
8 a/ c. [% S4 l        : public IUPnPDeviceFinderCallback
2 K3 t  F7 _4 o! H+ ~3 O{
9 d- t; n+ G. C- E' j0 xpublic:
, {) a8 Q' I( S' x        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
% e! ~& l, t  G0 f

4 i" E/ j! [2 H3 Q2 ?1 _; q, O1 ~   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
2 t- c+ {! |( z: `0 z" ^+ ^) n0 F   STDMETHODIMP_(ULONG) Release();

( U, h5 z, v; {+ r2 U
// implementation
) j3 d* k. {% `! }private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);

0 [  C' @2 U2 o
8 L: n0 w' n- _0 g& g( b# nprivate:
+ J2 \# E4 u4 {, l+ O/ X$ f5 F        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
. a1 P3 q4 C, I7 G6 x};


// Service Callback
4 V3 \! B3 C* J& {$ @1 [+ @class CServiceCallback
: d. B5 ]- o7 f) _        : public IUPnPServiceCallback
' v4 F, s# X  M; [/ y{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
6 Z& t0 M3 `( r" G: E                : m_instance( instance )
: U/ M5 ^% p2 p' b        { m_lRefCount = 0; }
& p8 `$ R( H7 W. S   
/ Q- o4 ?$ D) _  W. ?4 {   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

# P6 i& t7 y) w" |
# i7 P0 q: X0 v// implementation
6 [/ s8 U- |. _: K7 h: S2 N  ?& A& }private:
. I+ n3 I! Z$ h# _* i% i7 z& l% ^        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


private:
        CUPnPImplWinServ& m_instance;
$ M, U3 w0 s; v, k9 Q) S& e        LONG m_lRefCount;
2 Q5 V* j2 ~" [3 V};


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


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