UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   : z( _9 w' K% h, h
4. #pragma   once
   5 c, h4 f& `1 b! J  z- ]# `
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   
9. {
   + @6 _3 n! B3 H4 o
10. public:
    # J0 b/ o3 Q' D  O; N
11. typedef   enum{
    0 Y$ o5 r  ^5 s0 ~  K. C
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    # U7 F' C. |5 m1 M8 I6 H2 X' }
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    . k2 n9 \2 W( f
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
      n0 g! c& Z  S$ T! D' r
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    / F  M, j0 h$ J
18. 
    - A+ h" x# a3 m  I3 @0 v
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    5 ]# M/ l5 L* P0 E- `, o
21. UNAT_UDP //   UDP   Protocol
    + P6 [: m! g0 ]3 L5 p3 S, r: H# ?
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    ) u/ Z8 A( ?0 @* j. c; w
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
    4 P' S4 Q2 V2 V9 c
29. }   UPNPNAT_MAPPING;
    - L: Y) {8 @7 W' p. V
30. 
    * ]. u- z2 G# Q: w9 [& D
31. MyUPnP();
    
32. ~MyUPnP();
    . ^3 \; }( O( \2 c$ C
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    * ]" o! c0 f# H/ J, u* ~7 a' I3 A
37. 
    / O% R% n# w4 h5 q8 \2 o. B
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    $ Y* v$ j7 ?8 c1 i! H
42. 
    ' u" i9 {4 u1 Q( v# a
43. protected:
    & t4 ]1 ~. d" T- \
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    1 ]- u" e. \( J  }, q
46. 
      v! R3 r9 p- A# f, J
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    " g# y+ r* Y. E7 P1 T
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    7 Q0 W9 ~6 G. r. C' n
52. 
    
53. bool Search(int   version=1);
    . q, ~1 H6 ^: {6 M. J% E* t8 W3 t
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    + O* {: C7 [0 z/ @
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    + B2 n4 a; g; P2 p/ w7 S
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    , T0 R  L; D7 X6 h8 E* t
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    ) E4 T. O5 P. b2 {
61. CString m_name;
    5 t. j! z4 |6 I( r0 ~: I- u2 Z) a( a
62. CString m_description;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
    / ^: S0 O1 B/ k1 q
65. CString m_friendlyname;
    / ^+ }4 t$ Z: S* Q, S  `
66. CString m_modelname;
    
67. int m_version;
    ' ?+ }) @4 v% h  G
68. 
    # N& @/ e7 g1 S2 `. E5 F) S4 S
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    5 L, W8 w3 F9 P* E' S
77. };
    ' o: j+ K+ H2 g+ Q. X1 r8 q0 B
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   ; [1 B3 `, z. }: ?* n
4. #include   "upnp.h "
   0 G2 q' j7 u( E9 ^& e
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   / p0 h! D) ?2 }1 y6 w
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   ( E8 D* `" @5 k0 z4 d2 K
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    ) L3 p: B; }1 ~* {4 O9 Q
11. 
    # o. A( j8 G; i
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    8 g2 T1 n/ e  ^3 C0 s, h7 h- B0 O
13. static   const   int UPNPPORT   =   1900;
    ( F( v( p- Y! q! B( _; F
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    3 C6 z' q( d+ U3 E* K7 _
15. 
    8 l3 L( P5 I2 R0 F  Y; v2 I- K
16. const   CString   getString(int   i)
    ! G4 T/ \8 A" T1 W7 Z
17. {
    9 v7 d( \% x2 ^" \
18. CString   s;
    , J( w2 p; V+ Y
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    ' t" l; |" v1 U/ e/ |
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    - \# j! z( D3 n1 R) N
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    & g% a6 u8 f( l) f+ }% g8 O+ @
28. }
    
29. 
    2 f+ @4 L1 t$ ~
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    , t4 _; `  V5 u& C( X, [8 l5 a  l
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    4 s# S/ `6 _/ _1 Q4 b$ t. T
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    ; Z8 p: ]* C, G  Y" ~
36. {
    
37. char   buffer[10240];
    : [/ z4 S. d2 r) I, b, E* k2 {
38. 
    ! ~% F9 [0 ^% l; Y: Q: E6 C% n
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    $ v; W/ R, m, Z: B$ E8 i
41. strcpy(buffer,   (const   char*)sa);
    3 l2 w: m8 j+ O0 I% I0 k, ]
42. 
    5 B  m$ d" H, i6 w  ~) g
43. uint32   ip   =   inet_addr(CStringA(addr));
    ) O+ F! K1 b. {
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    8 {/ l6 A9 l, J! Q: F% D4 |/ F
47. sockaddr.sin_port   =   htons(port);
    , H2 S4 O) C3 L# q! [
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    6 X% K9 {8 ]8 [. q/ P3 W) ^: u
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    ' R1 L! j/ {6 {( p( s
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ! k2 H/ o2 d8 h6 C( }. d
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    7 p1 h, D2 ?( o. x9 }! w$ R
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    ' n* p9 L: U4 u
57. closesocket(s);
    ! ~2 M7 s( Q) p7 ?0 o
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    ; a' e6 N& m0 @* q3 V
59. if   (!rlen)   return   false;
    
60. 
    " y  h; W3 _2 t9 P7 |
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    ! C) U$ i6 R7 D/ @) D
63. return   true;
    
64. }
    6 G$ u9 {$ o' W0 J) X% v
65. 
    2 n; L# _' z" u8 M- M3 j
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    . y( Y/ R0 J' \# I! x2 O
67. {
    ; Y! L  k3 i9 \9 X: Z6 F
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    4 _( a7 i2 V3 U9 j! c
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    . J7 J4 V/ \8 n
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    / c7 V* b1 ?- D% S7 u
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    ! v, T+ J4 X7 K* q, o5 L% A
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    1 F$ |2 C9 }5 H
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
      O) Y8 F$ z) c1 @# s% N: h
85. int   pos   =   0;
    
86. 
    ! n2 y8 V$ }" b0 `0 Z/ f+ ^
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    % I" a9 u/ L8 M4 U( F6 e; ]
88. 
    
89. result   =   response;
    % @: ~6 q  T/ k0 L6 E
90. result.Delete(0,   pos);
    
91. 
    7 z8 m6 v$ I3 G8 e- J
92. pos   =   0;
    ' M: `! |- |" B7 K  N; B
93. status.Tokenize(_T( "   "),   pos);
    % G1 b: t& |/ j( ~
94. status   =   status.Tokenize(_T( "   "),   pos);
    & u, V" E! e. O1 x
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    ) |% G' K9 V6 h: }9 `
97. }
    3 a! B$ {% R: v4 I5 T, Z
98. 
    6 M* M% {, s9 ^5 T3 k; h
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     + R/ d9 C# r  W- C% Y3 X
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     3 m6 A3 R* ^& ~$ C3 K5 |
106. if   (posStart <0)   return   CString();
     
107. 
     . N" P" |1 z+ M: O% {- G  b0 u
108. int   posEnd   =   all.Find(endTag,   posStart);
     # R6 D) W9 o5 E. `$ O; r& G
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     ! b' D9 f2 s- l) [* w8 h
113. 
     / k# N; e2 ~' {; I+ K/ r
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     & g: f+ i+ w3 S5 j) N3 z4 G
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     # B. b( P1 W) A3 J  u
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     " Q$ P9 K  P' C" e  C5 p
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     0 V0 `& d1 u, m/ ^
127. RemoveNATPortMapping(search,   false);
     9 x5 l" B! C) V
128. }
     $ @, [8 _$ k0 s! m1 a. L; c+ ]  |2 b
129. 
     
130. m_Mappings.RemoveAll();
     
131. }
     7 J6 l" K; l2 |6 L& ~
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     3 b# s* x" G& W3 e6 u1 x
136. if(version <=0)version   =   1;
     + S: ?1 U5 B& Q2 \! [6 j3 [1 i) D; `
137. m_version   =   version;
     
138. 
     1 S% e# l, H. R1 k; r4 k
139. #define   NUMBEROFDEVICES 2
     + B( r4 E1 v1 c' O; D: F
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     * C! }1 F7 h+ m+ I1 a$ F
142. {UPNPPORTMAP0,   _T( "service ")},
     : u5 N/ R# S# e! z( _; y; N+ U
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     ! X( c) R- \- N, K4 `6 p7 y
144. };
     ) w: |% K: g3 }4 W9 H  Z5 m6 J5 w# j5 v
145. 
     . C1 x! C* Y+ o
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     $ W" N, a8 u( F0 b, u6 B. Z
147. u_long   lv   =   1;
     4 `1 T, Q+ T: t9 D
148. ioctlsocket(s,   FIONBIO,   &lv);
     ; Y, Y# w5 z" s+ Z* l" M* c: a
149. 
     
150. int   rlen   =   0;
     ( z: r+ E' f! e5 \+ ]
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     ) s1 a4 d  w! `. y' @
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     7 O& z# G0 K0 h0 o8 v6 o
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     2 s& Y$ D$ |. o# J0 y1 O0 D
155. CString   request;
     : C( p- a- }% d6 H$ b9 ]3 q
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);
     . q9 N( w( `5 c& N, u8 L) n
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     8 n7 t5 G' v+ W' y
159. }
       @. u! z& H2 g& Y
160. }
     ) l  D  R) {; M0 K+ u$ j
161. 
     
162. Sleep(10);
     5 c3 O6 K  O0 i, T0 G$ V
163. 
     8 ~+ s$ p! P6 F6 Y- T
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     " H' u2 M0 F; j7 k( R( t
166. if   (rlen   <=   0)   continue;
     / e% v! e* t4 R: ]3 @, b
167. closesocket(s);
     ! m) K0 ?% N0 f+ S4 ~, A
168. 
     . F5 e: m$ P" [7 O
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     $ |# b; Z9 J6 y" T/ a. q7 O9 y
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     ( L' ^+ n+ @! D/ N5 g( ?
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     ' ?) s1 N7 F& R
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     - e3 Q4 U" p; [* Y$ r2 ]6 q# e0 L
178. if   (line.IsEmpty())   return   false;
       K, `4 `) e. r$ S
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();
     ) {3 F% {8 \& W$ n) c
186. }
     3 Y, Z& K6 f  d. V7 ^$ N, U
187. }
     & \4 a3 `* u/ t- W) f" v
188. }
     
189. }
     
190. }
     , d: x7 `. d) z
191. closesocket(s);
     * Y7 {4 z, k3 d
192. 
     
193. return   false;
     & m- m+ a" w- B3 z
194. }
     7 {# t0 `5 w9 n$ o

复制代码

天云上人

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

1 C( v' ^$ O5 b8 {1 h
" s  [3 A* M. v///////////////////////////////////////////
5 m4 ?2 n2 P9 H8 R2 H5 K//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
) F+ i8 C: z$ g; M, b- l

#pragma once
#include <exception>
2 j. S7 @+ \: [3 s! n

# r( B) a; I4 i6 x: ]# C  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
: T( O! J- F( Y( r- R1 c! X        TRIS_TRUE
};
# {; m" M/ H2 `% Q
7 j$ u. Q( y- @4 o+ m* g+ T
enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
$ D! P1 ?! E3 c% `$ T        UPNP_IMPL_NONE /*last*/
' Y7 P6 Z  e) ^# n};

6 X* d4 J3 ?8 W" y+ \


class CUPnPImpl
{
public:
/ C: D/ q" s  H        CUPnPImpl();
0 z" p4 n# C+ v$ N( b! [, e- p        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
# U2 |, Z  F1 T4 [' K; D        enum {
2 r7 q, X$ X6 Z2 G% u; R8 {                UPNP_OK,
                UPNP_FAILED,
! L( Y* P" l, k1 j                UPNP_TIMEOUT
8 Z, g- f  F. H2 O! }        };
6 \+ W) w6 m0 W: M

9 c0 k) k. L+ |: N/ @6 a' m, A" J        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
! P" L7 r7 |- _        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
( `( ]  L" V" H0 S& O1 W2 e# C2 u$ m        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
3 W6 i8 w6 m. R- E0 P5 K        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
% J1 k' E. _2 v3 t5 p

' N  @$ Z9 r  ?        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
- q# L% r* V+ [6 X+ h        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
( C6 u. T. _. g" c1 A# q) Q; {
0 {6 F, M+ z$ j7 W
4 H' \$ P: V. B// Implementation
2 Y( i4 Z: z/ }7 bprotected:
* \9 S& g! W8 Q/ ^        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
2 j" `# T; V% w6 J, F5 x        bool                                m_bCheckAndRefresh;
/ H! l& O( ~2 A6 J6 X3 W
# ^# i9 I8 y8 ]2 o- X% s! l; D
0 O& L2 O4 X' Z: Y6 X3 ~private:
; a+ ]; u0 z1 ~! r* V0 g7 x/ X        HWND        m_hResultMessageWindow;
' c& e; o4 ]3 g# v+ s        UINT        m_nResultMessageID;
) C5 ~. E1 g, m

};

7 e- w+ H5 T# @. H7 \
// Dummy Implementation to be used when no other implementation is available
* n( X# r6 B" {class CUPnPImplNone: public CUPnPImpl
$ E% H/ n6 w" T- _{
; V- u: h5 }. X6 }; [8 Dpublic:
$ n( q* |5 Z: @4 ^2 f        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
1 p5 J3 ]+ F2 J# h7 j        virtual bool        CheckAndRefresh()                                                                                { return false; }
4 g! ~4 S1 u6 p% c# b. j6 `        virtual void        StopAsyncFind()                                                                                        { }
( p6 L% ]8 \+ b  o        virtual void        DeletePorts()                                                                                        { }
$ Q0 D2 O$ x7 p' M        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};

2 r2 J1 a6 w5 m1 n1 d6 _7 k; o
5 N  o, k0 S# {: ]. g( c2 y/////////////////////////////////////
1 B7 P$ Y2 G  b( v# A. ~( p7 K- Y! u//下面是使用windows操作系统自带的UPNP功能的子类
* G" ^# M( j9 Z7 o2 b- b

$ m  o) R( O. ?/ I2 `8 E( D#pragma once
#pragma warning( disable: 4355 )


#include "UPnPImpl.h"
7 D( s1 N: P3 m" {  c( R, U5 L#include <upnp.h>
#include <iphlpapi.h>
% A& ~4 q1 I& L9 t8 v#include <comdef.h>
& B: s. e* N: y8 J% w#include <winsvc.h>
' e( Y5 K/ f! o9 A

# ?' [* D6 j& L2 x* R4 Q6 N" W#include <vector>
4 S9 l5 t1 L0 p/ V& P#include <exception>
#include <functional>
" [; I: l6 ?( h/ p* c7 E5 N



typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
" \$ U% I; x4 r$ ttypedef _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;
  ~# {* l4 p  Qtypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


) ?& Y, u4 L$ S6 O% Ttypedef DWORD (WINAPI* TGetBestInterface) (
9 ~1 u9 z" j5 f" \6 H/ c  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);
6 w8 R' W) H* Z2 g, \2 N: l) A
2 m% ?5 {: ?6 m. [7 P' @) E" ^
typedef DWORD (WINAPI* TGetIpAddrTable) (
2 V7 V" h' h; H" P0 Q0 j  {  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
& L# W: n7 G& ]+ v);


4 v$ W: E2 r6 R" w. r# Z: n* A1 Ntypedef DWORD (WINAPI* TGetIfEntry) (
4 d% m; j3 _0 L$ a3 v7 n  PMIB_IFROW pIfRow
! M3 r# p/ V7 F9 O: h  b);
, b5 [  O$ {/ S) K  Z* b

% ^' u; j3 |% V$ lCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
9 i1 z$ E0 V: W1 T4 k$ c/ b+ D
# o, _' F# n: X2 E
$ U0 n1 g' c9 f, s8 w) S4 B2 vclass CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
/ u( [- Y' x" ]% s0 Q4 L        friend class CServiceCallback;
, Z" J* [6 {4 \( H// Construction
% j# q0 F* ~# G/ b- L1 q. ~public:
        virtual ~CUPnPImplWinServ();
- u6 C) E& X5 N( @4 n  c4 [: R        CUPnPImplWinServ();
5 T. ~- }1 z) I6 A! L1 s
. b1 h" n3 |6 |% L# B1 e$ K2 L# w
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
3 |6 o) j+ }/ A, Q- W
" Y) n( r! C1 ^
        // 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
. @) ~( {% a% A9 S2 O0 ^9 h4 K6 K        virtual bool        CheckAndRefresh()                                                                                { return false; };


" |9 q- Z5 J% H2 \; ~* Yprotected:
: b3 o$ a* @, U3 {0 ~1 _+ M1 U        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();


        inline bool IsAsyncFindRunning()
' n( k* g+ e1 A# Y        {
) M) S& q& s! u# E  ?                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
% c* x" E2 B' q) b" I4 x                        m_bAsyncFindRunning = false;
$ o$ \# M" Q& @9 h                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
1 Z/ }- L' C$ M. J* g% r                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
0 o, ~9 z. _$ u  V. b! f        }


        TRISTATE                        m_bUPnPDeviceConnected;
: S+ z6 h7 Q; c9 v( e
6 m/ c8 C3 o4 ^$ U6 E' `
// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
2 O2 ~# d' m$ m; [- h6 n/ t        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
) o+ J$ a  {# t9 t6 q        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
9 C7 u& ?% [4 }* P, S: \: I        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
( a; k$ |" D! L$ {3 _        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
6 h* b6 [: _% x$ |. S8 ~

! Q3 n: e- f& m) y' g% D        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;

( O. u4 U+ S3 r8 Z8 Q, T
        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
& X2 f4 h5 X+ _7 T2 }                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
( s5 I; X6 T# @& V1 N  J+ L                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
+ [; c" O+ m1 z" Y  R1 m" I

                        if ( FAILED( hr ) )
, f: A  x0 s/ l( d6 h( f                                return UPnPMessage( hr ), false;

  x0 j  n3 R2 M# R) q
$ {; c- \0 k+ }1 |3 f" o' @! w/ D7 D                        return wcscmp( deviceName.m_str, m_udn ) == 0;
$ q+ R7 G6 t& N, \$ `- `& v                }
                CComBSTR m_udn;
        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
' u! C( M* f! G1 }8 Y/ X        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
) k8 n! h; R; k. l) O- l        void        StopUPnPService();

# F$ |* u6 G9 I* ?0 y& ], O$ X; n
        // Utility functions
. @3 i% U  b8 p$ u, x        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
+ S6 a- o* a; X1 g$ R        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
# [  O& N( d+ {$ d- N        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
3 `1 m) Y- b  R- C7 v0 |        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
& x( D) l. F- V9 O+ R8 }        CString        GetLocalRoutableIP(ServicePointer pService);

9 Q$ u% ?$ C& `& T* t' P
1 Q' B! X6 C! M! n( e8 w8 m// Private members
; w- o+ U7 F7 k6 t. h8 d. yprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
1 r9 V' @" v) C. c2 w) l" b1 A  d        std::vector< ServicePointer > m_pServices;
3 S' w% N. f8 p        FinderPointer                        m_pDeviceFinder;
8 F9 a) j' {# b& x4 t        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;


4 e' r% n( E7 m# f        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
' [7 ~2 r) |9 q. g) b        CString m_sLocalIP;
5 }# i, X7 r- b1 @/ r        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
% v6 d# \& c! ]9 s  o( N' V' q( m1 M        bool        m_bInited;
        bool        m_bAsyncFindRunning;
6 g/ a: ~+ \) O, t  |$ t! \4 P+ p- s+ n        HMODULE m_hADVAPI32_DLL;
" W3 f  _0 h" r        HMODULE        m_hIPHLPAPI_DLL;
6 t  q, Y. E0 S/ R+ ~: H: s        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
2 Q; M7 d3 Z8 [% D* Q& J        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
/ u( e( X# u& v# Z+ J4 q

};


// DeviceFinder Callback
class CDeviceFinderCallback
1 |. q% j' h2 w4 s        : public IUPnPDeviceFinderCallback
{
& O) p7 F6 W& Ipublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
3 {- ]/ r: K+ P2 ]( j& j! {        { m_lRefCount = 0; }
8 v+ }9 P& `) T2 M' B) Q8 e
! f+ t4 J1 ?; G& S$ o- u" ~
2 y4 W. H( W- Y. R4 R1 t5 A   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
) d$ @0 ?- |/ V" M  n   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
# P- ]6 f* x& a: m0 {! B' X! U

/ L( i) z3 w7 `5 P" b. K// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
4 o% {* F  m4 k8 D5 o! S0 y        HRESULT __stdcall SearchComplete(LONG nFindData);


0 @4 [& t+ ~$ f* O9 Bprivate:
        CUPnPImplWinServ& m_instance;
. n) b. X' U9 w/ F1 I2 J" v) ~        LONG m_lRefCount;
$ V/ p( e* F. a) M};

0 [; S& J3 v) l4 \. o: E. N
// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
7 }% [+ t! u: R0 S: X3 G{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
6 u1 {0 K9 C+ F" ?. X  M; J        { m_lRefCount = 0; }
   
/ N+ \6 J" b/ w; K9 S: g( P   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
4 M  c$ @- J1 \" q) L' T   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
% M# Q8 B4 Y" k1 G; P
/ e9 B, [' s: F5 c. h
% I% E6 E+ L* y6 t! j; r// implementation
private:
0 k+ D; Q- Y# s  @" B        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

( |; B: I1 |! a8 }: e* A3 l
private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
8 h! C. \' W; T5 y1 S0 k};
- P4 e3 \* W; M6 W% Q" ~

/////////////////////////////////////////////////
9 f" U' n- |0 ]+ H7 n7 ^; R
0 _/ F1 S& T4 ]& O/ E- _
使用时只需要使用抽象类的接口。
1 P1 _# M- K, {; U% L; |! w' a0 ]CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
- V5 g% R; ]% @' ?. O& _CUPnPImpl::DeletePorts删除端口映射.