UPnP

天云上人

/*uPnP.h*/

1. 
   ) `  ?3 w: @( w
2. #ifndef   MYUPNP_H_
   
3. 
   8 b; L2 E6 r6 }0 `# e2 E
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   . G) G' i& a0 l
9. {
   # H( t, \9 n- j; K
10. public:
    
11. typedef   enum{
    2 \( i3 ]7 F3 |" C
12. UNAT_OK, //   Successfull
    ) H1 h1 o  P' X: k; R/ }
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    ; w7 }. i0 z* Z' J
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    
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;
    ' W1 z2 H, {6 q& Z- _7 z1 ?
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    + p& Q  k  b1 r* F) B% I, F
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    " F9 O" o! S$ z9 o( X6 P; B
23. 
    " N) y2 U7 j3 V, g
24. typedef   struct{
    3 l+ V4 M; x4 n
25. WORD   internalPort; //   Port   mapping   internal   port
    & ?6 f. Z5 @5 l8 B6 H# u) O- `
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    # ]+ Z8 t& Z0 ~
28. CString   description; //   Port   mapping   description
    9 w% T: N: t) V! D* I% M" d! Y
29. }   UPNPNAT_MAPPING;
    
30. 
    9 D9 H$ s5 _- d. {" @# t
31. MyUPnP();
    0 ]+ Q5 H7 ]6 t. L4 f7 z" T4 q
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    4 c8 k, Z/ e2 G- w" L& O
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    3 p4 A& G+ p  M0 o+ \! l6 R* b( w. ^
37. 
    . X4 Q& s# m3 a5 }
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    9 U3 s) p' a$ Q2 p; ?& J% n+ C
41. bool IsLANIP(WORD   nIP);
      K, |# Q) Q% w7 D7 c4 B6 L. F' a
42. 
    9 O2 X- A# M* B4 W3 m* J
43. protected:
    
44. void InitLocalIP();
    ) a' Y! Y! X3 c2 {
45. void SetLastError(CString   error);
    ) \, N! H1 a; S2 i" B" j
46. 
    ; c) M* y' ?$ i; ?
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    3 F) a/ m- A; q4 x+ H
48.       const   CString&   descri,   const   CString&   type);
    , w" b% k( Y9 F, ]* u
49. bool   deletePortmap(int   eport,   const   CString&   type);
    6 u( O5 ]1 V0 n* j6 f' d- ?7 ?
50. 
    * v1 p; F3 O9 e! g' B: d
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    + n" h. ]( }0 ]9 C9 ^
52. 
    / o) S( W( u0 a4 t* A. a2 h' u9 d4 G
53. bool Search(int   version=1);
    ( M7 {3 Q; I5 m8 k! I
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    - d5 m' X: l; c8 Q# S. I( Z
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    ' _8 l& H/ a3 g) r
57. 
    + ]) e' p8 x0 l3 @
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    , b- p: b3 w6 W' r- w- R+ {# Q1 r
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    " m. i* O: S/ w" Y  _* I$ {' A
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    , O' K6 ~0 c2 g; ~- f
67. int m_version;
    
68. 
    
69. private:
    $ |6 ^! U) ]% i& y  s5 A( v$ h% T
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    % ]/ }6 \7 }! U* R8 g
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    / A' |) d8 g5 K) X9 _! `! a4 I& v" q
75. 
    
76. bool isSearched;
    ; u, g. L' X( |9 p$ ^
77. };
    $ H6 _( x+ @/ d1 R$ W
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   1 a. u: c; \# f1 q, P# w
2. #include   "stdafx.h "
   
3. 
   3 {: Q  k1 d/ _( Z/ H5 a
4. #include   "upnp.h "
   8 P3 i% U+ D5 N) T( `
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
     [, W2 G9 U! q$ j
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   % w" y7 x; }4 x* Q& B: H" c1 k, G8 V
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   ! u2 v- [( P+ Q4 c8 [8 ]; ~
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    ; G" s& c3 W4 o
11. 
    + q2 I5 a  B, i0 H0 p
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    6 X, W' A0 \3 ]  M
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    & B6 V, y9 q6 V  V* b2 a
15. 
    
16. const   CString   getString(int   i)
    * u6 C# K9 E3 K
17. {
    2 h9 @' q4 j2 L+ G8 r
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    % D5 ]) K* R! a/ u1 A. j" j4 f! l
23. }
    $ P# c: d! z  \3 j9 o+ C
24. 
      s1 m/ y' J# e9 w5 u
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    5 x) \. q* }7 p' O& a) ]3 K
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    5 p5 p7 Y! ]/ `6 b' p2 F: ]) K* i
28. }
    
29. 
    1 p0 H6 N* k; w- \% |
30. const   CString   GetArgString(const   CString&   name,   int   value)
      L/ h7 ~0 l' ?6 r1 N
31. {
    & w: e% @/ }) R% N0 @
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    6 X7 o: ]0 u( V$ U
34. 
    / J$ S" N" T/ E3 N0 S) S$ l( ~
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    ) V3 T! V& U' h4 q5 f% w  B6 z
38. 
    
39. const   CStringA   sa(request);
    ; X4 ?2 Y7 s) ?* h  T2 x0 Q) X
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    5 V& x# p0 T, Q" Q1 Z  f
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    ; w4 Z* Q6 Y/ j" g; z
46. sockaddr.sin_family   =   AF_INET;
    : }9 g$ y- d' M) n
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ' |0 H1 k% L* X6 [: U4 ~. m6 d
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    6 @: R) I) K! B* c7 ~9 J
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    & R; L* U; E4 ^; F
55. Sleep(100);
    ; c! J# I# m$ W8 C6 [
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    ! i9 C1 `$ `# p" l1 j% n
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    4 z) W1 |7 m; e
59. if   (!rlen)   return   false;
      q5 I3 o& ~+ n/ D2 B& M
60. 
    2 [% p' O5 i( d" W! b2 z# Y" N
61. response   =   CString(CStringA(buffer,   rlen));
    8 q& X* \% z) l# j$ X5 a, |, e" m5 i
62. 
    / h% \0 P9 \% S" `
63. return   true;
    % \9 {6 Z9 \& O+ M
64. }
    $ \7 o7 s4 P% D  q
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    0 ~. j5 [: T. I, @. o
67. {
    
68. char   buffer[10240];
    ( r' L4 w4 n6 f
69. 
    6 q) B0 W" A$ w
70. const   CStringA   sa(request);
    9 K9 M! c% z4 j$ z
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    / L( r0 b/ y) O& Y* K
73. 
    
74. struct   sockaddr_in   sockaddr;
    7 Y( C) U! X" J8 h( Q
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
      `2 U4 X1 s) i) `
77. sockaddr.sin_port   =   htons(port);
    . [2 I3 J4 d3 J# Q% ~8 K9 {
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    4 U. m! F  ^( l( I, W
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    6 P. T3 w; }/ d- X: U
81. }
    
82. 
    0 D0 D3 ^- v9 M/ h$ G3 L% c
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    / }0 O# D/ P1 o
84. {
    
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    + @) @# a+ r0 [" o0 b9 H
91. 
    
92. pos   =   0;
    % I" C/ M4 I* M' g! ~; m( W1 o
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    5 E6 ]' @" S2 T' H! R2 {
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    1 c* P% J  W/ I) B5 L
96. return   true;
    
97. }
    & e6 d0 B/ \- J9 |/ ~& k
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     1 m# ^- M) M  l* [8 v1 Y
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     - J# H9 x# @  k/ ]
108. int   posEnd   =   all.Find(endTag,   posStart);
     $ G: C6 H! l9 u6 o' N
109. if   (posStart> =posEnd)   return   CString();
     + _+ a5 f  H- J
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     ) R, A/ J9 o8 L
112. }
       J4 Q, Y7 l# k4 {+ C$ D& E
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     ) l# R% \# I% P! C0 z* ]
116. {
     - f8 t" @# D- W* X7 b
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     ( R) ~3 x; O7 ~
120. 
     $ {6 V3 D6 n& b+ s) j
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     : l/ ]3 h9 F8 J: x+ j8 e% V
127. RemoveNATPortMapping(search,   false);
     7 m: U5 ]# L9 c' C( R9 s& L! x8 e% A6 O
128. }
     * S; c6 p4 s, C4 d  q- v2 c
129. 
     $ P% s- h* k& U1 \6 c
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     9 S& @/ J! z: @3 t) D6 A7 Z7 t8 ^
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     5 v1 F2 d! E" g6 r' k0 P
136. if(version <=0)version   =   1;
     " |, l$ Q) X6 @1 M+ s* w
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     + F! z4 [$ S9 ]
141. {UPNPPORTMAP1,   _T( "service ")},
     7 N( ^; l" T8 k( t! `
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     2 z1 s- g) Y* y
144. };
     7 q& Z9 O' j- K7 v
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     1 M, `, n! w( d( V' R, ~/ E* o
147. u_long   lv   =   1;
     + P) B4 q8 Y) p
148. ioctlsocket(s,   FIONBIO,   &lv);
     1 D8 p4 B- W: ~# J
149. 
     " G% [; I$ T" O6 z6 t" z0 ~
150. int   rlen   =   0;
     0 ]" F, P# R: l2 s
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     - a& E; B4 U  _5 Z1 B/ x
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 "),
     : J; y" W# ^3 o) h+ s( x! X
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     , L2 L" g5 x7 [: V) K- O6 I
161. 
     
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     * P3 @' ~) G0 q5 ?1 q+ K5 k, c0 @; \
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     ; M. k- g% h1 ~1 G" y
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     3 X3 I: P( N$ t
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     5 y& S5 H+ i5 b5 ~' n- f+ C( S
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     - i- D& `: z& Q) T: m) s
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     2 ~5 x, O' A0 m
175. if   (result.Find(m_name)   > =   0)   {
     : P0 `2 j( L0 L% g" K7 x
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     ! `3 M/ E; f4 x' O3 P- Z$ }
179. CString   name   =   line.Mid(0,   9);
     9 `. c$ {8 J0 I7 |; ?7 l
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     3 u4 U' V4 t" ?. z
182. line.Delete(0,   9);
     8 Q4 K# N8 _4 S/ @2 \8 J; s6 U
183. m_description   =   line;
     
184. m_description.Trim();
     " M- ~5 h9 x: a0 k8 B9 M0 P9 H
185. return   GetDescription();
     
186. }
     , c: U3 J0 |# o! d2 {! h$ h
187. }
     
188. }
     ( n" W2 N  }2 D3 a5 R% }; b4 ?5 X) o
189. }
     
190. }
     * D0 N$ B  @: p* }3 `$ g
191. closesocket(s);
     / |- @8 Z# o( l& D! ^+ m5 h4 I
192. 
     6 R: q& Y, O  O' Z! @  j. k, {$ o( K
193. return   false;
     5 X  u& I; ]# k9 w' K0 X- I! k
194. }
     $ x% t( Z. [* _" p" D' ^

复制代码

天云上人

以下有关upnp的接口来自emule，
9 ?" A+ l" W* \2 C+ d* g6 x

///////////////////////////////////////////
, `- [9 h3 N3 w6 \//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
* D* B# \* `4 b8 Z% r

8 Q$ y% M- ~% ~* s# n  Y#pragma once
7 s% Z. ^' R# n% d8 `* @8 j#include <exception>

. X  w  E! M2 Q; @5 W
  enum TRISTATE{
( u2 E( b  J. C        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
};
* V, t7 f9 x/ g
" ]" F9 N) m  n8 V8 t+ z1 T
enum UPNP_IMPLEMENTATION{
+ B) Q$ J+ ?$ u+ @( h        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};

$ x- q$ @  |& a( j


% [2 m+ O* |( R9 p8 m2 s' y8 Fclass CUPnPImpl
. r" X, q0 B$ `) Y: {{
" v+ g- E- t- upublic:
( K) o: Y. r  s6 x) ?        CUPnPImpl();
- O) e- y( K$ r4 v        virtual ~CUPnPImpl();
: R- A# w+ F$ y9 Z) P        struct UPnPError : std::exception {};
6 T6 ?7 ?, q- O; a0 H) h* J        enum {
. B  Y) Z2 v* }: c8 L7 }$ o                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
! W  u/ l& ]% D8 }& z: X) F2 e        };
0 K( r" P* T5 c( _9 v

% u. d" |5 `8 L8 M. n( C9 r        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
8 t  F* z. _0 |1 }        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
. t/ c- n8 z# Q! t" c0 d        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
9 R, O* L# ^6 e% r7 \, a6 M

        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
& j( p; N) v# t" ^        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
7 |2 }5 b, w9 B$ q! V        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
: h5 b% [7 P( C+ T# L- |        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
" Q4 r3 k1 V. b* W3 w7 H8 R8 R        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
& ^% d7 {0 X/ B$ Q; T' d" b% a) j        uint16                                m_nTCPWebPort;
( V6 w; p2 V7 q" K- j3 y( A7 Q' o        bool                                m_bCheckAndRefresh;
/ ]3 O7 U7 T* [1 N/ r/ l+ w2 s
3 V: d* o1 g# l9 r9 C! q! @
private:
& p  j) l7 Y8 f; o" K        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
% a% ?9 x. `& j8 n- a* s
8 s/ C* [$ G( C2 |: D# k
2 u( k7 c4 w# n+ ?/ q' q2 q% Q- v& P+ F};


. v6 E( A. E' L// Dummy Implementation to be used when no other implementation is available
5 D7 ~+ R/ z) L& ]3 Wclass CUPnPImplNone: public CUPnPImpl
7 h1 {4 u3 O0 r7 B{
: {* o2 }- ]; h0 Xpublic:
: G; F$ o2 Z$ h        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
8 b( i" _% l5 g' O8 [( }  D        virtual bool        CheckAndRefresh()                                                                                { return false; }
1 p& B  v$ g9 K6 M, r3 W        virtual void        StopAsyncFind()                                                                                        { }
& @/ }  l3 O! w- C        virtual void        DeletePorts()                                                                                        { }
1 [9 O& v" f/ v5 C1 P# ]# E; p        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};

' M% X9 H0 v% s; ^  g
/////////////////////////////////////
! i; f9 k& H( Z4 A# x. `8 g& B' R//下面是使用windows操作系统自带的UPNP功能的子类
+ b" y( t1 Q4 i4 H" w4 w

! A9 `7 s* W  a+ w#pragma once
#pragma warning( disable: 4355 )

9 x' \7 Z3 {+ ]& ~4 F/ q2 v
3 v, Y0 w7 z, v4 B: T$ O, O* S#include "UPnPImpl.h"
$ l  Q5 l* B" H0 _9 l' F#include <upnp.h>
% }1 q1 `+ Z9 J( [5 j; C' O#include <iphlpapi.h>
#include <comdef.h>
* L( m7 i6 ~( m. \9 `. b* }2 w2 {9 N#include <winsvc.h>
: Y5 u2 }5 m- d

$ }% M3 `0 P- P+ l- K8 m0 @$ ?1 G#include <vector>
#include <exception>
/ ^: N" P, O. G  ~; x: E#include <functional>

% ?+ h" S8 c5 \; {1 q' Y
7 n3 s% l  e- y7 g8 K

typedef _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;
) V. B# D6 q% A- O& ^% J0 v! [typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


0 p0 G$ b9 w2 g& O1 r# Xtypedef DWORD (WINAPI* TGetBestInterface) (
8 N. t' W" f  l  M, o* u! d  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
& M6 m& P; m, n  F% `, d);


0 C+ S+ N! G! ctypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
" r4 p0 ]+ I2 G' S5 N);
  C. u$ o, M: r$ T# K

3 {$ F! [! f+ W) `0 K$ ^typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);
5 {& _0 C$ n+ R( G" G+ g

* l0 t2 j' o- {! O7 mCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
5 p' a: E) m8 w

class CUPnPImplWinServ: public CUPnPImpl
4 X8 Q' [: p6 i+ l* r( r8 \% T; d+ ^* _{
$ H; [: a9 n( S        friend class CDeviceFinderCallback;
7 x( ]$ I3 C# h, p% L        friend class CServiceCallback;
% `. d. E$ i: T4 X# w// Construction
public:
' o' ]8 A5 _8 V; x0 c" t3 a( ~: I: |        virtual ~CUPnPImplWinServ();
3 t9 l* z5 E. j* [1 o8 l5 y& @        CUPnPImplWinServ();


5 i1 ^) d, g9 Z        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
. R1 l( F" C1 d6 `% N- I        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
+ \* I  I4 X' N# {" F. R        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
4 C- L+ h5 A6 ?, u+ I; f) Z

7 h; {* ]3 |8 ?3 x# l3 @; @        // 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
5 N& H/ Q+ K% u$ z/ P9 m        virtual bool        CheckAndRefresh()                                                                                { return false; };


protected:
) D6 o4 l3 a- z+ z* H        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
+ O0 `  O+ v  C: {' m5 o        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
+ u! a6 w$ ?$ V# b7 V        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
4 ?! ?! N9 q% m$ O3 Z  }& L% N        void        Init();
. I3 f% u/ ?5 j

6 p8 v- q  X# f9 e  y9 y1 ~" k        inline bool IsAsyncFindRunning()
2 \- H3 F( b3 a4 r6 O5 i- @        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
) @5 W' h# |  M+ ?                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
1 z0 Y( l, D# @$ l$ m) I                        m_bAsyncFindRunning = false;
& N6 H5 G0 C$ n" r7 S3 @                }
                MSG msg;
4 g5 ~2 g' M7 r: G                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
8 a0 S: f& c. ~5 j                return m_bAsyncFindRunning;
        }
6 _: c7 d5 q* J% A4 B9 w

        TRISTATE                        m_bUPnPDeviceConnected;
  Q2 J! D' n# X1 K$ V7 D
- m9 {+ }+ s5 H3 z- f( ~  q
// Implementation
        // API functions
$ v8 w4 _! F8 h* [. ?( g        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
6 \7 r- [% g- ~* x: \  d        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

4 [3 ]; R8 x: ^8 O# t8 ]
3 g2 ^6 H$ l4 ~) R% k* j4 Y1 _        TGetBestInterface                m_pfGetBestInterface;
7 F! D% }) W: O4 ^( b        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
" v! x' K. W5 }  P! r  J& z

        static FinderPointer CreateFinderInstance();
9 m& V6 {& {' M3 Q6 z4 o6 Z/ s        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
5 G5 d' K) w/ M* [* ]: ?                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
+ _# i' u" t: A8 X                {
$ k  f! K# Z3 m$ U) Z9 `' O' E; r* [                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

4 k( O% j1 J' B# ?* f! M/ s* D
                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
5 `) r+ m0 ^- `, y; t8 U. Z
+ D6 u% t1 k4 I, i8 h( `3 T1 O
4 D8 a% q) q$ v7 B5 Q3 G                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
0 w. g5 d( C) Z0 C5 p4 @                CComBSTR m_udn;
        };
4 V; a/ m$ f( t, x( @       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
) b+ a# W  B& p2 z        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
+ P. V/ [0 z- q# K- z) e        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
5 w8 P/ c% m5 x        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
- d! \! D: r+ N+ a* a9 @        void        StopUPnPService();

- |! S! f+ a1 o8 ^0 @' `) `
        // Utility functions
6 S- w. u6 H! y  B- q$ ]8 U7 W        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
0 a! a; r( A4 V3 J" \1 i/ c        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
, p! E" L1 c, `$ q        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
5 B9 M) S  E: Q2 z, U0 H- ?        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
2 {% w" Y- ~* R  _! K) j        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
- H* K7 s$ ^/ d0 \  p  h9 ?        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
4 R6 W$ w. \2 M9 n4 A        CString        GetLocalRoutableIP(ServicePointer pService);

% o2 w* o) m' B  _' L
& z$ V8 Y4 e' T* a' w// Private members
private:
0 n# M5 ]. g( g# e: J% [/ c' g        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
. A3 q# x5 f8 q) D: S/ W3 E3 I        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
3 K' L# r3 Z# s) g7 B% f        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;

* n1 g6 x$ D9 C3 V4 M2 r# c
        LONG        m_nAsyncFindHandle;
: S1 @$ g' E$ N! n9 W        bool        m_bCOM;
$ ]7 W4 i+ N4 a        bool        m_bPortIsFree;
0 J. R- i6 {1 [; f' i, b: p        CString m_sLocalIP;
' D! q6 X; T9 ?. Z        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
" F* Z" `% F1 h! N: f5 H( n# S        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;


};


# ^5 o7 ^( E. O/ r/ M: Q// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
" c$ r' Z# ?0 E6 u& s{
2 ^- E9 M* H" ]) P4 y" ypublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
3 W4 G( m2 T# @. N" W; U" c                : m_instance( instance )
        { m_lRefCount = 0; }
% u: n5 n- c4 F7 L

) l0 U- z# p& _  [   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
8 [- n! |6 c5 I  Y  t: ~9 ?   STDMETHODIMP_(ULONG) AddRef();
9 Y- w$ v& q5 Y4 l" c: E, ^   STDMETHODIMP_(ULONG) Release();


# [" ?4 F3 y1 ^% g) ^& r5 b// implementation
private:
, y8 ^3 r5 o. n( f7 W! ~        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
& {% G( M+ i% R        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);


" V( A$ M. j, i& y% tprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
0 R) [) ]3 }6 x0 a+ x! @};
+ k$ @& d3 z$ ^$ j& ~. L

! d# n' V" Z' o* u// Service Callback
class CServiceCallback
8 r6 @6 w' Y$ |/ M        : public IUPnPServiceCallback
. v# G7 m+ n& F3 L, |* D) c% Y{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
4 A+ [7 ~5 K8 O, C$ c                : m_instance( instance )
4 ?- ]. r- T2 U3 I6 i        { m_lRefCount = 0; }
/ Y# ^% M; V7 D! u* \" R7 _3 j5 g   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
0 l! }, j" C' p2 ^) J* |; k! r   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
! J) Y  ~# G, {) J- v3 _4 m
, M# ]5 c. }& }! I
// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


& ]9 t# o  h% f- T& nprivate:
( Q* S$ g8 @2 W1 k        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};


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

! k; Q7 {9 h/ v
7 n6 c! M  V* `3 G: I/ b0 U使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
: f$ K7 [/ L& }  F5 RCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
8 B3 j! D' a2 n/ h- hCUPnPImpl::DeletePorts删除端口映射.