UPnP

天云上人

/*uPnP.h*/

1. 
   9 B" @, t3 o0 i) A0 d7 X6 |/ k
2. #ifndef   MYUPNP_H_
   + M# w8 p6 y, c, M
3. 
   
4. #pragma   once
   2 G8 r6 O* w' \  D" E" y2 F! ]4 h# M1 W
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   , w' u7 s6 s: l/ G
9. {
   8 ?7 D; W% O5 N
10. public:
    ' w) k8 `4 r# o0 s
11. typedef   enum{
    9 T- d2 z$ ]0 n7 o
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    . o# v8 t5 t% E2 p. J; g
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
    8 y$ q" M; P4 M0 j
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    " d3 X! l% i- B( p
21. UNAT_UDP //   UDP   Protocol
    ) O. V0 l$ D+ |/ M
22. }   UPNPNAT_PROTOCOL;
    9 ]/ [4 J  a0 {- q" I
23. 
    ; h2 X7 `, V  e( S' r7 K
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    7 a( e- m' \% [4 e) w( T
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    * n) ?, ]/ M7 k1 U
29. }   UPNPNAT_MAPPING;
    
30. 
    6 |- l- a7 m- \  w' d: l) {& P
31. MyUPnP();
    
32. ~MyUPnP();
    - ]& J0 L4 g! H6 Q5 K
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    ' K. ~% U0 m( I
39. CString GetLocalIPStr();
    ' T6 n/ R$ S& ~9 B
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    / `5 X7 @+ K) e. G- M
43. protected:
    
44. void InitLocalIP();
    5 \3 S, E, J; m9 V% Z7 Z( f
45. void SetLastError(CString   error);
    1 D' {! f. B) w8 d6 L
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    * B4 z; D* I, K7 K/ v+ H
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    8 l0 a7 r2 o+ s/ x, Y6 h9 y  E
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    : C; a5 _( C9 `; V" E; N8 z, Y: H
52. 
    1 c- \5 l! P, }+ r  K  Z4 v, Y- j
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    ' j% s0 E+ u% p
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    " j2 v# r# M" I1 C/ m9 Y
60. CString m_devicename;
    : X6 h* f8 ^/ K, ^
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
    3 M9 c0 [4 {9 S9 B$ k. V
65. CString m_friendlyname;
    # m; j+ Q* Q( R* A$ O
66. CString m_modelname;
    
67. int m_version;
    
68. 
    3 W# V5 \2 ~( v$ r6 S. o
69. private:
    $ Y+ O' F0 M4 @0 P* |7 S* N  \4 N
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    ! ?. T$ @' Q$ w& {
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    " i+ O4 I2 P. @$ I% H! ]4 v" Q7 D
75. 
    9 k" G" l, N0 Y2 j- B2 C) Z
76. bool isSearched;
    % \/ J) g( R" A2 L$ K
77. };
    9 ?! I; I* \$ `; @9 G
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   . S% n8 }# R& a" x6 E2 l( I- B4 [
2. #include   "stdafx.h "
   % l! A, b. b/ G/ |+ f
3. 
   
4. #include   "upnp.h "
   $ r3 j: v4 r" n) f4 ^
5. 
   / D3 n4 ]' z$ S8 c" \7 e
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   $ R3 V! b+ p% }- T3 G) L
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    9 a: h$ b& W5 Y% [" ?- p
11. 
    5 \% |6 D- v+ b( ~  L
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    ' D- C4 v7 {) x( P+ H6 g" \
19. 
    " [  _! a* K# u, z" e& x5 b
20. s.Format(_T( "%d "),   i);
    8 V" m! s( W/ F4 F( I4 l' c' H5 N
21. 
    
22. return   s;
    
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    . C. H% O6 r0 G# H0 G
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    2 i% y5 M/ U  q
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    9 c9 n3 C0 Y2 }
34. 
    + K! a/ ?4 U  Y9 m' S
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    ) X+ ?& |0 l8 k7 V
40. int   length   =   sa.GetLength();
    " {0 ~9 J! G& j; b8 Y
41. strcpy(buffer,   (const   char*)sa);
    2 t$ W1 ]/ K5 c" p3 l
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    0 Y: L  N7 I/ l+ Q' L5 I+ C+ {
44. struct   sockaddr_in   sockaddr;
    : a2 }, {/ }/ R0 c; T7 l
45. memset(&sockaddr,   0,   sizeof(sockaddr));
      |* p$ M. m1 N- H) J
46. sockaddr.sin_family   =   AF_INET;
    , H/ a8 O( z8 P
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    ( u* m. h! l& E- m, _8 ~5 e
51. ioctlsocket(s,   FIONBIO,   &lv);
    5 i) Y/ y7 G/ l$ b( H2 Q5 @
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
      t' y9 {1 ?5 u5 C6 L1 f4 ?
54. int   n   =   send(s,   buffer,   length,   0);
      q7 b  {) E) o; |3 F
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    / H, r9 u* l. A: t( j
57. closesocket(s);
    - @/ d5 `, n- O1 I9 F6 i+ [: v
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    * j& `& B3 ]0 O2 s1 m( e
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    . Z% R; b4 \7 X/ f& i
63. return   true;
    
64. }
    8 I2 ^6 l. v" M4 v) c- k' t' A' C
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    1 p# P, |& L3 S+ {/ u( f+ [
67. {
    
68. char   buffer[10240];
    $ e% b( J8 b6 B2 ?0 s+ R
69. 
    
70. const   CStringA   sa(request);
    ) `) F: ^6 _9 ~; f0 Y- U
71. int   length   =   sa.GetLength();
    2 W! D& L$ ^: E* K4 t
72. strcpy(buffer,   (const   char*)sa);
    8 ^, o- O0 O" K1 n) ^  i$ P
73. 
    6 |* q6 g. c5 F) H
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    . q3 I7 T. g5 @: Y. R( `8 C
77. sockaddr.sin_port   =   htons(port);
    4 F% c5 P* V$ x* E' Y  J
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
      a3 N( W: r8 ^8 Y. O. g3 c- {4 k
79. 
    8 e6 ]. M' G- T1 m# m: j3 g; r
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ! O: h& \; X5 c8 S1 I
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    5 W; n8 K* O6 `+ x  n( l3 ~. V3 K
86. 
    5 J) ~+ x8 P4 e, ~
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    ' U  g. l9 p* Z! j) ^, w7 E: `, `# Z
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    7 V3 \) L, n; e, f5 x
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    ; z* D7 W: I6 |) i; y4 j) ]
96. return   true;
    
97. }
    * M7 g' c" p; K
98. 
    4 q% i# @  C7 F1 @3 y; V/ a5 ]& L
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    / x& [4 p2 d- b8 ^8 E' y
100. {
     5 A# Z/ F" n" I$ I4 {
101. CString   startTag   =   ' < '   +   name   +   '> ';
     8 \8 u9 g9 J- b  f, X
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     ' ^5 a5 s9 q: M
104. 
     
105. int   posStart   =   all.Find(startTag);
     , P7 h9 ?$ s! [( g
106. if   (posStart <0)   return   CString();
     9 F4 J2 {! T% z" m5 j  w
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     ) {. @% F' J, g! C: m3 b" i) R
109. if   (posStart> =posEnd)   return   CString();
     4 [- p, I8 F* m( b+ H% v
110. 
     7 ?# |2 G3 j  k8 t
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     + n9 z- e7 l! R' y4 M" h
114. MyUPnP::MyUPnP()
     , C; ~4 ^4 ^' T
115. :   m_version(1)
     2 w) Q! R8 W. g3 x3 o
116. {
     
117. m_uLocalIP   =   0;
     # L% H4 k  \! ^0 u* Y, j
118. isSearched   =   false;
     
119. }
     8 h  h' `; {( P6 I1 E
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     - h' Q+ E& w% F+ u4 D$ m/ K
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     # u  |8 b. n$ j
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     5 [1 `/ x4 ?# Y& F
127. RemoveNATPortMapping(search,   false);
     
128. }
     8 X# K# h/ ]: Z) O
129. 
     8 n' Y- t- V" I$ a1 z
130. m_Mappings.RemoveAll();
     4 G3 U, [3 d$ |3 w* H
131. }
     
132. 
     ( G' {9 `9 C& W$ R9 e
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     " I1 k! C( r- v8 j9 [  M! p
135. {
     + z+ X1 O( y  k, F6 j1 A& n7 r( O
136. if(version <=0)version   =   1;
     6 {% y2 ]; K; O% k$ x, ?
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     5 l$ n$ [( f: w2 f! R
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     $ z( \9 D9 k9 R# h; z
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     5 L  k3 m1 n2 E
150. int   rlen   =   0;
     # ]% ]& t. A# H- k1 L
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     ) j9 e! I% {' L+ m- J7 f
152. if   (!(i%100))   {
     ' F4 A) l! Y/ b# s
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);
     9 m/ B$ o2 `0 K1 H$ R% X
155. CString   request;
     6 o  v$ ]( F; h- e
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 "),
     # N6 Y; d6 @2 K9 f: }& ]
157. 6,   m_name);
     $ L# R$ O+ j* w* Y2 Z! {
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     7 N1 \1 E; q% F# M  n
160. }
     
161. 
     
162. Sleep(10);
     
163. 
     5 G, e6 t' l2 ~3 ]% u
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     ( Q. M, |: A, X* o) A) h5 c
167. closesocket(s);
     
168. 
     : |$ k' f: S3 }8 n6 K, d
169. CString   response   =   CString(CStringA(buffer,   rlen));
     & `7 u5 d$ O+ R3 l: _& ^6 @
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     7 i6 [8 c0 t3 P9 e# {
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     " A, U) _/ i9 o' ?! \/ L  v
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     ; `/ t- {$ V9 o. D( c) P" F' U
176. for   (int   pos   =   0;;)   {
     $ m9 d5 S, a' n1 }8 [, J6 ~
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     : E6 [( R; O) i
179. CString   name   =   line.Mid(0,   9);
     # ]5 s# u, M$ J8 O3 o
180. name.MakeUpper();
     1 k+ u6 I  g3 V" n) k/ x
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     & y  F: x# L; g5 t+ a  u% u1 @
184. m_description.Trim();
     ; @9 M. G* H. A
185. return   GetDescription();
     
186. }
     & `8 q$ a- i1 J: O! w
187. }
     
188. }
     
189. }
     
190. }
     : X) \8 m+ H# a) j  l: E" i) m
191. closesocket(s);
     
192. 
     
193. return   false;
     , J# L" a4 ~  R) L6 \
194. }
     

复制代码

天云上人

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


! |/ j7 n: j# H3 f6 n///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

# _. |' Z8 f4 E
4 L3 T7 b8 v3 B& b#pragma once
! N% V- x/ `; H2 b7 R#include <exception>
, h& P/ P( O- h  T. {
  V$ n/ s3 Y6 y5 }6 V
% h$ u) B/ {9 ^  enum TRISTATE{
2 }; X/ C* J) `: T8 `* x        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
};
" a* e. }* C+ W& L

enum UPNP_IMPLEMENTATION{
! M. s1 c; v2 j3 ?        UPNP_IMPL_WINDOWSERVICE = 0,
- F% ?- g; {3 J" ?2 E* [        UPNP_IMPL_MINIUPNPLIB,
) e& B. b2 `2 {; h# A: j        UPNP_IMPL_NONE /*last*/
};


. p- x) H) ]) L. s8 d

class CUPnPImpl
{
# b: }+ T$ Z  U- z$ x  P# M0 N- _$ e- Ppublic:
( O8 l7 Z/ f1 r2 v' s, [! B        CUPnPImpl();
) r) B* a. h( y. k  c        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
: }' ?, G6 [/ ]7 V        enum {
                UPNP_OK,
                UPNP_FAILED,
% N) _4 k# `0 Z- ?0 Q                UPNP_TIMEOUT
        };
1 p, z7 w0 z- I5 Z- \2 S
; c- O; G, y# D7 ^; w
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
- F' c( J& j5 l) H$ R8 z        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
! `1 R3 r, U$ @1 ]5 R        virtual bool        IsReady() = 0;
% p/ D- |  G5 S9 o9 U5 z        virtual int                GetImplementationID() = 0;
       
6 ~/ N- N' m6 s$ r' D; D  n        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
9 j: o6 e. ~$ L( y4 T# B& t

        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


+ q: [' k6 s" B// Implementation
protected:
# X( ?7 h2 O" l8 E        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
4 w; Z6 T( n. D2 G0 _        uint16                                m_nUDPPort;
5 d( }! }" ?1 n. q/ n. P, \        uint16                                m_nTCPPort;
- W+ u5 x. E. b- P0 H* A6 K        uint16                                m_nTCPWebPort;
- T1 t; C  x. }6 \, }# w        bool                                m_bCheckAndRefresh;

) ^- O/ `# S; W
private:
1 Z" l  l. M2 \- L4 L6 |        HWND        m_hResultMessageWindow;
) W4 ~& C0 p. H7 s9 q7 g$ E        UINT        m_nResultMessageID;
2 Q, ^  T/ i1 l1 {
4 s. v/ `2 `; n5 r3 q& ?/ @
( E4 `2 N5 I( _4 c7 Q' p7 V};


// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
/ B3 {8 b" ?" A; O; T{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
* X  d5 [) F2 @) S        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
: k! C/ z: `# |        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
! E7 h6 N* [" ]" C) C' ], j};


/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类


#pragma once
#pragma warning( disable: 4355 )

* W/ r, m, r9 d! T0 C. [% O1 |
$ _8 G- L6 c& n& U7 ?, c" `#include "UPnPImpl.h"
) Z3 V% t( @/ }#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>
, B3 l4 I2 @* P4 s- R7 U

; c8 }( }9 |1 T/ B* b#include <vector>
#include <exception>
+ H+ U  A* n- H' R0 n#include <functional>
; o! i7 _, Y0 x' W6 \. W7 E


  [: D: w  J7 G8 H# H
+ _5 I" W1 X  r& a7 ptypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
8 _9 V7 }8 J; j6 \( Itypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
2 D4 W4 l6 G; {' D  M6 C3 H9 Otypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
" V. W/ f7 i% Q4 ?" x* K6 ytypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
  D& x& k* G$ n$ g2 Ztypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

" P! A! u4 s1 P
8 y& s8 U0 k$ h$ N& f0 E8 T# wtypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);


typedef DWORD (WINAPI* TGetIpAddrTable) (
5 `3 |8 ?, G6 r3 p  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
);
; @  [# @# f- M9 w: k0 b: x
6 t* t8 d, \0 B3 f$ g
9 ~3 M6 W5 P6 O% a, @5 stypedef DWORD (WINAPI* TGetIfEntry) (
7 O. R# v' v, b) v+ N- L  PMIB_IFROW pIfRow
$ w" U4 ^& D/ |- a; F# C$ S& x# O+ w);


# w' Y) C, E" hCString translateUPnPResult(HRESULT hr);
3 `0 |% N' v/ l6 {+ G  b( U( v8 lHRESULT UPnPMessage(HRESULT hr);
& ~% @8 ~6 Q( t# J1 j6 w# E) [1 e1 h

+ C" ?, o0 R+ \& Y1 l0 R. B+ |class CUPnPImplWinServ: public CUPnPImpl
$ f9 J* j5 N  G. y, ?+ y# n{
( Z5 _- U6 B2 M% Z5 s        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
: @1 I$ O' u. J& D( N% j: Opublic:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
, c. p0 b8 F% f% o. n* i3 n

# b: _* f9 C- R( C        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
& V( y+ [) u6 k4 O1 a3 j7 u        virtual void        StopAsyncFind();
! S1 d' z5 T; S        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


* [9 g1 k$ P) _* r( u; F        // 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; };

! E# C) a4 d5 C. Q/ ?( d6 z5 `' O9 @, j
protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
3 q  b4 n) O. Y$ M, H5 Z: o* H' k5 D        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
, o4 A: C( A% j+ V5 M1 c8 }        void        RemoveDevice(CComBSTR bsUDN);
, L% `8 n! S& y; Z7 D: m, X$ m0 C        bool        OnSearchComplete();
        void        Init();
" t; ?7 ~, c5 \4 s! e5 U( r

        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
6 O: |. e( [( _8 R                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
$ n! ?! j- M& h, }                {
* S" Z; ~) o- W5 C* Q                        TranslateMessage( &msg );
1 a8 h. @9 c2 ]* Q; p7 K$ B                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
        }
3 F+ {3 Q2 ^0 `. `3 V

        TRISTATE                        m_bUPnPDeviceConnected;

' x# I8 V- _, G0 a  V" }! T
4 t# t- T+ x* x% h/ Q# y5 b5 p// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
" d$ _: v0 c6 I        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*);
: I+ U; m6 o- W$ q5 f+ M, l        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

" z( A& A" r& ~
, ?; ]5 D& C) ~) m- R! C3 O) {. _1 e        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
0 K  O' |3 w9 l  i; x

. I) T& C  T/ h; V! p/ z0 g        static FinderPointer CreateFinderInstance();
- W2 c; [) Y! [        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
5 p6 B6 p" d) V/ W0 g& n8 |                result_type operator()(argument_type device) const
                {
- `. u0 V. n5 E" w2 N. k, [& E                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
4 y) R- [& w) a, p( {9 P" k

" C! g! U$ @4 Q; i5 b/ f( o                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;

/ A5 \1 u& a' V- h4 f
" o. C2 _# H; s! H6 F( ^                        return wcscmp( deviceName.m_str, m_udn ) == 0;
) }, \. K( T3 G                }
( G; B! V7 I6 r                CComBSTR m_udn;
        };
2 I; J4 e2 u0 o) c1 a1 _% W% a       
5 b- }, @! N* ]% h0 l; v7 }, b        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
0 m* Z3 a% w9 P& k, V  G        void        StartPortMapping();
& u: ?1 k7 t) e# U6 N# G5 o        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
+ _" q1 X$ t/ y: D        void        CreatePortMappings(ServicePointer pService);
5 w9 @3 G  Q. c  Q- R        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
: p6 X; P# G( n. w0 k4 D        void        StopUPnPService();
+ A/ t. S* l( ]  @
1 ^; _  K# G- r! k/ V+ ?
        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
+ C9 n1 d1 d# d1 |+ B8 u6 `        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
# @. k7 x3 P# K) ]        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
8 M; q, d# U# k  U7 D        CString        GetLocalRoutableIP(ServicePointer pService);
9 l+ l: C* v7 S$ N
" z* R7 X$ [9 ~! a- ~* D( Y
2 F4 S# _' N: O+ q* T// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
4 {0 }# E! A" \9 _# Z, g0 |, [        std::vector< DevicePointer >  m_pDevices;
. z3 Q3 S/ J, B. Y+ b9 f, C        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
4 r' n" n) t( ?  W

        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
6 b- g8 g1 N% |. c7 T1 _" b& m        bool        m_bPortIsFree;
5 B/ V/ g" \1 j) c" w3 b0 `        CString m_sLocalIP;
        CString m_sExternalIP;
- P  G$ j; |) n! I7 }: I        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
* r: y/ N) K1 n! z" G        bool        m_bAsyncFindRunning;
  c  q+ s/ K) `1 D5 _% |5 B. C        HMODULE m_hADVAPI32_DLL;
; }. \; C7 F! X; N        HMODULE        m_hIPHLPAPI_DLL;
( \. O. m  O/ {        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
1 e* |# [( T( T6 t        bool        m_bDisableWANIPSetup;
% T/ c) R4 o) a% {) Q( n        bool        m_bDisableWANPPPSetup;
' ^: [7 s; }2 @; D* T# T

3 p* @4 L9 S! E! `' p9 a};

8 V$ e8 F( L6 E! C% y3 C1 q9 U5 i
7 f, }  |9 w; K8 Y- I// DeviceFinder Callback
class CDeviceFinderCallback
* h7 i1 `4 a1 R. r        : public IUPnPDeviceFinderCallback
, c$ k3 Z! w3 d! c6 x{
3 l. u! u. y8 _1 Kpublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
5 H) M& Y# ~% _6 |( w* p        { m_lRefCount = 0; }

& _' h5 f2 t" ^1 X4 }$ M
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
3 F6 @( q( p. X6 R8 [! O9 Q  H   STDMETHODIMP_(ULONG) AddRef();
& U9 J2 x; x# v' V6 v   STDMETHODIMP_(ULONG) Release();

9 s1 t5 A4 `: _2 y8 B# c! E5 V% q3 y6 g5 C
' E2 K7 W8 e& C3 b// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
) |4 y7 k% I% C( T        HRESULT __stdcall SearchComplete(LONG nFindData);
/ l) ?, N4 C  s1 R

private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};


// Service Callback
, N* u0 i9 Y6 ^- o/ y' H2 fclass CServiceCallback
! Z. G- ?: O5 K, F4 j: c$ T        : public IUPnPServiceCallback
{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
) A0 I& o! ^# o9 A/ P                : m_instance( instance )
8 I# y5 b4 N2 D* a        { m_lRefCount = 0; }
1 e/ U2 z1 h. O& C# ~! M, B   
3 n! y( @3 z* M5 q   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
# p$ K, Y. z/ s+ n8 o. e   STDMETHODIMP_(ULONG) Release();
& R& z: K8 X$ h4 a% \+ M) s

8 Y" n- i  U) G6 D. @7 ^$ f// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
6 @, ]+ P% l7 M& U& r7 L        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

" ]0 @4 B9 b9 A
private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
+ Q4 C! f$ S0 ^5 H" t0 S* P5 j, N};
% S, P, [$ c( B# d1 Y
  E8 c* K  ^: B/ n! ^8 e
( ~3 W: J. h/ f. h8 V/////////////////////////////////////////////////
, K, @5 \  Y6 m6 h
4 N9 R' P* Z! a1 T+ n: R" Z) Q" U2 f7 ^
8 k) |9 c8 K7 c2 s- L5 P/ @* J使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
( G$ A+ I6 n! p# p( R( S3 MCUPnPImpl::StopAsyncFind停止设备查找.
0 d( d' Q* X9 f0 k- zCUPnPImpl::DeletePorts删除端口映射.