UPnP

天云上人

/*uPnP.h*/

1. 
   + h& b. G2 \6 l+ L7 `. Z
2. #ifndef   MYUPNP_H_
   / k2 e6 e: `2 c* Q9 F  v: w
3. 
   ; M+ x% Z6 C2 y" u9 V% {
4. #pragma   once
   : _0 e3 {  k, f7 [
5. 
   0 D" S5 ?0 Q/ ~2 a
6. typedef   unsigned   long   ulong;
   3 }7 _; B, t8 |; T6 c
7. 
   
8. class   MyUPnP
   ( m/ m' [2 M5 j6 q/ q
9. {
   
10. public:
    / b! X1 _& @: _8 |7 B3 U8 N
11. typedef   enum{
    6 h2 ?. v/ A+ F0 c+ V
12. UNAT_OK, //   Successfull
    
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
    + I- K$ F! K  W* P# o* n8 ^
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    ; V! z  R9 f" a3 }
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    5 ^3 ], n8 ], l/ G2 Q6 ^' X6 ^7 v
18. 
    5 X& J4 n, p* V% S6 I+ `" e
19. typedef   enum{
    & M, ^2 c0 u$ f; |0 T
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    ; S, o# J6 V7 D5 p( G3 Y
22. }   UPNPNAT_PROTOCOL;
    2 V( g1 T0 M( W& R/ _- p
23. 
    
24. typedef   struct{
    & |( G" U0 B! ]) V( S* W$ }
25. WORD   internalPort; //   Port   mapping   internal   port
    # @1 g) g1 T% [1 y% g* h% G
26. WORD   externalPort; //   Port   mapping   external   port
    0 c# ~1 b7 j8 A5 v. T( l3 r" ^
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    4 [9 e% c# p6 h! Y6 }4 ]7 e
29. }   UPNPNAT_MAPPING;
    
30. 
    % ]# U8 d& @0 N0 ?% ?
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    % F( [% R" n7 A
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    ) \2 r+ |- m0 J0 k! k2 w  y
37. 
    2 ?3 ?# z( ?. C7 e- @+ e5 L
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    7 w6 Y' B4 P( ?0 T# e, y
42. 
    
43. protected:
    % G8 j! t. W. f* `8 y; Y( g
44. void InitLocalIP();
    ) l: Z" K, f/ j. U
45. void SetLastError(CString   error);
    
46. 
    % ~' |  a. g! O; K" i/ b9 t4 P
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    2 T& L! \1 Z1 @' c% O
49. bool   deletePortmap(int   eport,   const   CString&   type);
    3 [- F- q2 T/ @# X! S
50. 
    & x" t5 ]3 ]7 o8 Y0 F4 c
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    ) {( Z+ L: i1 W
53. bool Search(int   version=1);
    6 R3 Q, H0 F+ H9 j, C/ U& g! y
54. bool GetDescription();
    3 E1 `1 o+ ~( _0 l$ R; }
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
      o) d/ \2 b/ m% r
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    5 u2 c" i3 W/ Q( v1 ]
59. bool InternalSearch(int   version);
    ) j0 e" n, E* J6 ~: o
60. CString m_devicename;
    9 D) I: o# L8 }& o5 P  Q3 B* I4 d
61. CString m_name;
    + a9 Z* D7 @  I6 ?. r* Y% q. j
62. CString m_description;
    
63. CString m_baseurl;
    * q- U& c( K! p3 S
64. CString m_controlurl;
    
65. CString m_friendlyname;
    3 e/ P; Z, \/ b, P
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    * R- A4 ^/ X3 d* a* t- E/ T; Y
71. 
    9 a3 B: h) i3 p# D7 a, @
72. CString m_slocalIP;
      n) ^8 p; {0 x& _" Z8 B
73. CString m_slastError;
    ; ^0 D( g# B' _( t- ~" M( ]
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    5 X, d% e4 D; a0 z( v
77. };
    4 j, F* ^5 }3 b# ]
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   5 M8 M4 }4 _1 n. r6 x
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   
5. 
   , G1 W! l7 i* u$ \8 t$ P
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   8 Z. \5 O0 O1 w: f3 @- Q
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   . o$ F; ^! I% d. C; ~2 u  p
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   # ~  s# U6 ~( D& ^( |% Z
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    . L& D% W% c9 w- ^, |! ^$ b
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    ; {! f4 c" F3 P& @$ m
16. const   CString   getString(int   i)
    2 I* A. M8 g) u3 L6 G
17. {
    ' T" v" P1 c9 j; \) H
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    4 w- H! I3 N8 S5 W1 f
21. 
    
22. return   s;
    " j$ }, B8 t& c4 N% R
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    ; O! f7 \- G# G& ~' l0 O
26. {
      Z6 e5 P1 Y2 ~: U7 w1 x9 G9 j
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    ! ?, O6 u) R' U) m( g
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    - R! U) \/ j: s2 ^3 w
31. {
    6 i7 |; q* U: w8 |" G2 T
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    5 Q8 ^* {/ D5 v+ d6 {" q! K0 H
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    : Y0 O2 m0 P3 o" Z) A  d
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    ( Y. ]6 U9 i4 ~; m( z
42. 
    / n, E/ s. a6 }# o5 U
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    ) B0 H: Q5 ?, i
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    , V* W% G$ V: K  j5 t; ^3 ^  _( l
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    " M/ Q) L7 Z3 Y
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    9 c2 F% I. f: v8 P) m2 ^' O
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    0 ?+ Z) M, Z5 w
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    / X: r3 X9 u- M. b) @$ I
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    : @8 L' d4 @: \2 b4 L& a: x
59. if   (!rlen)   return   false;
    + d% n9 B  ?5 X+ J8 f
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    , h, o6 Q/ b5 W- `' B) g
62. 
    
63. return   true;
    * i5 S6 {- {3 l) q: q9 b
64. }
    
65. 
    # {. W) w& _) }
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    * J. ~4 m0 [+ s. X; Y2 r8 I
67. {
    " w- k' B2 b( ?# V. A1 y& u
68. char   buffer[10240];
    # e6 R- @1 Q7 N
69. 
    
70. const   CStringA   sa(request);
    * g) w4 y& v% Y$ q
71. int   length   =   sa.GetLength();
    " Z6 u2 ^/ p: Y0 U$ [, E
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    : r$ L# i; f( h" R/ O% j4 t0 ?: y+ b
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    ; ?+ d7 [0 j* g9 x7 n7 E+ H
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    $ m- J$ J7 q3 C3 O, t6 Z6 \
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    4 N2 h+ d5 ]1 d- y4 I: _1 U1 e
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    - D' l+ M& K, m! ?
88. 
    5 r$ \/ F" c0 j/ F" N
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    9 X3 j  e2 S! x( r
96. return   true;
    $ g( \- U7 s2 m' B8 ?
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     4 ?0 E+ ?9 w( _3 X4 e2 W, y
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     # t# x9 X! C5 N: _+ _: C
104. 
     
105. int   posStart   =   all.Find(startTag);
     - x1 F+ P; i" ?: J
106. if   (posStart <0)   return   CString();
     
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. 
     
114. MyUPnP::MyUPnP()
     * f3 n( m; L; s7 }" @  W- ?8 l# K3 b
115. :   m_version(1)
     
116. {
     % \3 E2 d* O1 F$ j  s9 g
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     1 c/ }# N: d6 l. I1 j6 v
119. }
     
120. 
     . f  u& q5 j5 V, h* n4 }
121. MyUPnP::~MyUPnP()
     + r2 u0 q+ b& e2 ~  Q% R' b4 o3 U
122. {
     ' ~: y: i  ?0 ?( c
123. UPNPNAT_MAPPING   search;
     9 M. x; |; X. z  t1 h3 K' F; ~
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     1 a, g. p) V" f9 s
127. RemoveNATPortMapping(search,   false);
     ( i9 T' V0 F* l$ s; d
128. }
     1 Z0 A6 ~- m' c
129. 
     
130. m_Mappings.RemoveAll();
     1 q: E/ x" P$ f1 R0 x
131. }
     
132. 
     9 a& @# z& |! d$ a7 j) c! \: {3 C
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     + b% V' K7 g7 J$ V
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     8 \; E7 X# @; D+ Z# B
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     : W  ^- t" ?5 y( x$ F9 I$ ]
142. {UPNPPORTMAP0,   _T( "service ")},
     + U0 g" a6 z' Q$ I. L
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     , D" t9 w; x: p. m9 @0 b! U: u
145. 
     . G2 W, q$ w! X( \6 a  A( O
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     5 U5 g* _- q# i' Z
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     * g; U9 E8 P7 K
149. 
     + E+ R$ |: }; g! j! U
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     0 |+ d; b; y. m& i" X
152. if   (!(i%100))   {
     2 {; B  X8 T' u$ {  E
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     ) P" z$ U1 o+ @' O2 c& S$ X
155. CString   request;
     8 j+ z, O$ h! |
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  e( D; ^0 x! ]; N
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     - d0 I; d" i, j1 T
160. }
     5 T. h9 F8 H  `1 g
161. 
     & A1 U/ g5 ^! E* ]
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     3 Q* ^: e- C/ o- O6 C2 _
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     - o& p! j7 r: Q! E1 A$ T, Z5 ^  b
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     , a. ]3 }/ h3 ~7 \% U
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     ; ]/ }+ t& U, z4 y, Z) Y
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     : o1 z4 j/ {/ P6 \; i# D6 k
175. if   (result.Find(m_name)   > =   0)   {
     # M1 b* U6 K2 r0 _
176. for   (int   pos   =   0;;)   {
     ( D9 a& C  R7 [& }$ ^3 t, Y) F
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     8 q! Q% v6 x2 K$ ~5 S2 a2 S
178. if   (line.IsEmpty())   return   false;
     + ^3 k, f. z8 t
179. CString   name   =   line.Mid(0,   9);
     : ?' d) K4 S) @
180. name.MakeUpper();
     , n3 T2 Q+ _, C6 H
181. if   (name   ==   _T( "LOCATION: "))   {
     - w: Z2 U+ W9 r4 y* a& P/ `
182. line.Delete(0,   9);
     ) U4 A, P8 O4 ]- q5 H3 n
183. m_description   =   line;
     & p6 N8 A& `3 q, j6 u; l) `
184. m_description.Trim();
     % m) ]2 C6 ^, K2 j0 V3 k
185. return   GetDescription();
     
186. }
     
187. }
     
188. }
     
189. }
     
190. }
     
191. closesocket(s);
     ( o2 x1 ]3 A+ s3 \. k+ r4 G
192. 
     * w* `" }! J. [$ B6 p+ T* p
193. return   false;
     
194. }
     , n0 j  }6 j+ F- E2 T

复制代码

天云上人

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

( k0 s/ C& p+ b( T( K
, l$ G0 ^4 k4 M7 v' a% K& T///////////////////////////////////////////
& ], I' I7 E1 D1 {, `//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
; F0 G, V, p6 i; Y6 ?3 z8 y% [1 t
0 ~6 }& g9 G) N& T" N" _
#pragma once
$ A; B0 g  g- t6 n, F6 ]" H. r#include <exception>

; N3 H! M  s8 q2 O& Q6 n- |
  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
! W" d, j# m' C) y0 Y        TRIS_TRUE
};
& ~/ H0 N& k* ^# b) M
  l5 o# U4 f# s% ~* l& |2 l! B- d
enum UPNP_IMPLEMENTATION{
- t! j# h6 g, _$ s        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
6 i+ j5 p6 V) D+ ]) O7 `: j* J4 a        UPNP_IMPL_NONE /*last*/
2 E; {+ `* Q5 k};




# N8 g! x' p1 aclass CUPnPImpl
{
- @: C8 s$ K' @4 R8 y5 {public:
4 ^8 ]" a& X* l" E' c5 H        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
9 j9 g% @2 k8 W& w; s                UPNP_FAILED,
3 W5 i5 ?' ^, P/ d  W9 `                UPNP_TIMEOUT
& r" D( W+ t% x  j! S' q        };


" l% N) w7 I9 `4 z        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
) Q$ k/ C, G3 `# ?+ o" i0 `: o$ l        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
# f8 u1 H. `2 g        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
* D. A6 K  W1 ~" }        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

4 p! s  Q' M( @
9 f, P. b; H) ]$ W* n6 Z        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
8 Q' D) ?* e: m( p5 z        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
$ W: \9 @4 s8 r9 a3 T! d: t        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


) u9 v( t) q8 {, ?// Implementation
protected:
8 I/ e: Y& m  w6 i1 Y* e        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
' e2 m. ?/ B9 z% L$ g0 W        uint16                                m_nUDPPort;
4 X7 |9 }! ]  Z" e        uint16                                m_nTCPPort;
9 v* D' [7 R% y$ ]+ {) U9 d( D        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
9 H; i% l, P7 d1 ~* L1 M0 ~, V
" U% F4 r% `8 D) T1 T2 U
- m* k2 X8 m( G+ X0 I- }3 M& aprivate:
- a- x3 b; v, \4 c( m  c        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
% |+ e( R, f7 `

};

2 {/ K+ `7 r- }/ p
// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
- B- P( g8 d* ]" W6 |# ]- X3 K" `{
4 O8 r2 S- {/ u' G: j% i& Kpublic:
( W6 Q( O, v2 q3 a# E        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
5 D5 K: {3 [' r- A" y        virtual bool        CheckAndRefresh()                                                                                { return false; }
" u* u" N. o8 m        virtual void        StopAsyncFind()                                                                                        { }
8 M- Y0 J# Y( a0 x; t        virtual void        DeletePorts()                                                                                        { }
8 ~2 i3 g; F% J. d0 T        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
& p0 d1 D5 W% E! W};
4 d3 x8 ]( Y7 v$ ]& K1 E  g7 ?
$ k/ D9 @6 ]' |
/////////////////////////////////////
" f1 G6 N  ?1 {" T6 [- M7 q//下面是使用windows操作系统自带的UPNP功能的子类


& Y; N: v1 q! E3 s) c#pragma once
9 k! ~/ }" F  ]2 |) h, [#pragma warning( disable: 4355 )
4 m9 w7 D; ]7 `& ~2 Q; M

#include "UPnPImpl.h"
#include <upnp.h>
$ Z8 D( v1 z0 P" L; L#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>

6 w. U% M, ~$ Z8 F' s; a4 Y
/ Y, h5 F  S' O8 ]# R#include <vector>
#include <exception>
#include <functional>
4 _8 y" u, ?+ S; p4 w5 a

/ Q$ N$ x: k. \$ U9 a& w

* w8 A* a. h7 |& p5 \3 j( qtypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
6 n. |/ ~! V- etypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
3 \) R" _" T1 j+ Ctypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
6 `5 ]! @- x) `, H5 n% Z  W0 Ptypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
) l% c- W3 A$ d: N  `! ~typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
8 G6 b2 m* f5 ytypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
; @% |* p1 W7 [1 J$ M+ {
& `  h' V3 [7 B: t
typedef DWORD (WINAPI* TGetBestInterface) (
6 M- M6 {6 b" D' R9 L7 g& q  IPAddr dwDestAddr,
7 T. L1 [  L  i% u: D% l  PDWORD pdwBestIfIndex
);

9 |" x% M# k  p( ^3 L1 B
typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
);
0 c9 N; G4 i" G
" ^7 {* B1 {  l: G! Z, V
typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);


$ m# m( y' i3 `4 S5 QCString translateUPnPResult(HRESULT hr);
- J  |- {! x9 n% E+ hHRESULT UPnPMessage(HRESULT hr);
8 m) l* M) O/ I$ M% w  r4 \

class CUPnPImplWinServ: public CUPnPImpl
5 e4 f- c1 b9 U1 C% Y9 D% V6 s{
/ W0 j! @% G# J5 o7 M( ~        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
. n+ J0 j- n5 Bpublic:
        virtual ~CUPnPImplWinServ();
# o5 |$ W- U8 B6 h* o9 r        CUPnPImplWinServ();
) B: M, j: o( O3 R  a
  z( q* T0 q- t4 }, Z8 T
% ]* V+ E+ G, e0 {/ @8 z- s# {        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
/ m/ x0 B; W/ q  T/ |- q9 W$ w        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


! w+ s8 P+ d( M2 E5 p8 Q        // 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; };


protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
0 m4 b" p6 X8 N1 e1 \% q        void        RemoveDevice(CComBSTR bsUDN);
) I$ z/ b& P% g* j        bool        OnSearchComplete();
  ~, q- z0 E5 A) I        void        Init();


        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
- \4 {) J3 L& M6 x7 V, v8 W! j                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
& b+ w! A9 Z; {9 M  l% {                        m_bAsyncFindRunning = false;
                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
% f1 x$ _- e! F8 S/ C1 I9 U/ p6 x                {
  F: ~3 z; f8 W: }7 C% m                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
5 i' E8 u6 @% S1 Z        }

  S5 ]* s2 z/ P, s+ H6 T
        TRISTATE                        m_bUPnPDeviceConnected;
7 B2 N3 \! {8 Z6 X4 @+ F, F' M

// Implementation
- b9 w: n3 Z  c8 I, {3 c        // API functions
# Z* R; d$ [1 n6 I# v        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
  E7 r! p+ M' }( a+ o( Z        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
! H! J( b/ L# M  f* v! P" L# Q        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
2 C; e1 i+ R5 [        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


        TGetBestInterface                m_pfGetBestInterface;
3 _6 T0 k2 w3 ^0 q# V        TGetIpAddrTable                        m_pfGetIpAddrTable;
1 R5 D/ x1 {  W        TGetIfEntry                                m_pfGetIfEntry;


. }) S5 ~; ?0 j* q        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
0 F5 \% `) B. d3 j& Z+ s- M        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
1 i  P! b0 B& q$ {: e                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


                        if ( FAILED( hr ) )
- X1 p. m" }( j& p                                return UPnPMessage( hr ), false;
) O1 }7 i5 e* z. [, X9 Y) j  k

' e9 C+ z7 K: H6 x& x                        return wcscmp( deviceName.m_str, m_udn ) == 0;
% P" d5 `% B& \                }
! C: Q- }: y5 C1 N                CComBSTR m_udn;
        };
       
8 B( j$ c0 [( ~$ p4 w4 S* A        void        ProcessAsyncFind(CComBSTR bsSearchType);
( q2 }! F" n! Q        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
  C* S; L+ \* _! a' H        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
9 v% e. D7 F/ C% L0 y; O' H        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
  p: V7 X8 v; m' ~                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();

8 z  t, s- `4 ?6 H& }" M. J' [
        // Utility functions
% t0 h# ^- A' N+ z" j) L        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
) D4 s1 v9 g" E  o, b+ X        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
8 S% r. V$ f/ O2 k9 R/ n% @' a        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
: i  |& b, P3 ^+ ?1 j        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);


/ D2 I0 C  `! l! V  ]4 m8 ^// Private members
private:
; |5 i. ^0 P4 E( Y  C' y: R3 f; ~        DWORD        m_tLastEvent;        // When the last event was received?
' h6 [" v$ n4 k" Z$ z' {7 N        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
8 M: }7 j( l# R; I$ P        DeviceFinderCallback        m_pDeviceFinderCallback;
6 |; V# j; _& s; k( V        ServiceCallback                        m_pServiceCallback;
+ }* ?6 q: b% [- e
. L# w/ U5 K6 Q' Q$ X0 ~( z, }6 D
        LONG        m_nAsyncFindHandle;
- ?) r* T/ P1 w, `. a2 \        bool        m_bCOM;
) _- b& S/ u/ t0 l9 G        bool        m_bPortIsFree;
9 a- e& g. |5 Q; I9 e        CString m_sLocalIP;
        CString m_sExternalIP;
, e9 b  \, D6 K6 ^; Y4 o8 ^/ y1 F        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
* f& c1 g# }( T  C- a- x$ B        HMODULE m_hADVAPI32_DLL;
' p2 O6 X6 v  H' U2 L( y) a        HMODULE        m_hIPHLPAPI_DLL;
/ t1 |5 H+ l+ W/ ~" h        bool        m_bSecondTry;
# D$ x7 M8 }0 M        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
  I- F6 ^6 e/ t
% v9 ]7 J; L' V- ~9 ?7 K' A4 O
$ K5 \$ P$ r  M2 k};
2 G" J. j! {* J9 m; ]7 ]# u- ~
! {/ Q" w% P. Q% F  F8 R
// DeviceFinder Callback
: w7 o: [+ E+ A! ?  M% ?( Yclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
4 p! e8 e# r4 _: R" k7 {' F' `public:
* i5 K' I$ s. a2 }9 ^        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
$ A7 v1 u' X; y7 I' u3 [7 N: M' X        { m_lRefCount = 0; }

* o! Z% u. V; ^& |. o% G) P. ?
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
1 y$ r6 O& h: B   STDMETHODIMP_(ULONG) AddRef();
0 Q3 H9 @6 ~! X4 Q1 p- N: K   STDMETHODIMP_(ULONG) Release();


) p* u$ k1 e: {# o& H6 ]+ j! ?, N& n// implementation
: i* j  o" f. sprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
/ x* r6 f- C/ N8 X        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
  X" o# O$ L4 c+ _, r% d& o  w        HRESULT __stdcall SearchComplete(LONG nFindData);
& ], @. t5 E4 K2 W" I3 D

private:
        CUPnPImplWinServ& m_instance;
7 u) z/ h0 q& |( v/ f        LONG m_lRefCount;
9 p9 _- B& f! x" r: [};

4 y% E, i9 [2 q6 K, ]
1 F/ @- g6 J' Z# A7 `// Service Callback
class CServiceCallback
% o2 ]9 k: c* F$ e: L1 W& z3 H        : public IUPnPServiceCallback
{
! @1 b' i" p4 I& W# zpublic:
% L! O5 F! ]+ X, L& z; V6 _        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
5 z6 l7 _+ [3 `   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


( ^* U; Z- U( t: `& l( |// implementation
; U8 o- l! }  p& m3 Iprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


  c( t- g+ E" [( Z& H9 _) j8 Dprivate:
        CUPnPImplWinServ& m_instance;
6 x. R8 c3 W* ]; C# S        LONG m_lRefCount;
};

; i6 g$ ?! S' V0 }% n* t
/////////////////////////////////////////////////

8 d7 L1 q& X0 o+ g2 V# z  i
7 P3 _* o& ]! h8 l3 ~, f9 k5 j, _使用时只需要使用抽象类的接口。
6 f7 W. {4 `1 B* C/ p4 a) @9 ?- ACUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
( t& y9 y0 c! ^  PCUPnPImpl::DeletePorts删除端口映射.