UPnP

天云上人

/*uPnP.h*/

1. 
   ( b1 g$ U. r2 M. R6 A
2. #ifndef   MYUPNP_H_
   
3. 
   4 {- j& \& w, F4 |  p( ~, s, ^. s
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   : |' O# e: S' E( f) K% x) d6 D
8. class   MyUPnP
   
9. {
   # P$ ?, p1 _0 W2 `
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    ( P) D  \! L+ O) @! M; H# C/ o
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
    3 H: m; C' S* u  f: B" a0 ]) L
17. }   UPNPNAT_RETURN;
    : O  X% G8 S( v/ N9 k  f/ B
18. 
    
19. typedef   enum{
    1 W( A& [/ Q) M: q4 P
20. UNAT_TCP, //   TCP   Protocol
    5 \3 h) l7 F# s9 d) w$ r
21. UNAT_UDP //   UDP   Protocol
    # @/ H5 e) ]! G
22. }   UPNPNAT_PROTOCOL;
    % n2 t6 R- V3 [. z; j& j
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    : g  |' ~0 U. B2 o  D; ?
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    ' I6 v0 A' o+ F
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    0 f( a) j. N7 ]
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    ' ]( d9 m' m  {8 A4 K. E2 V
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    6 ]; s. b- \+ M# d# `% v5 c
40. WORD GetLocalIP();
    4 m9 J, R) P& U3 n  r. K/ I
41. bool IsLANIP(WORD   nIP);
    $ r1 `2 |" o1 G% I  Y
42. 
    : o% ?" c+ Y- m6 W
43. protected:
    ! \* R( a! j# {3 a2 k. \  ?9 Z
44. void InitLocalIP();
    * m( v% C7 ^* O
45. void SetLastError(CString   error);
    . C' I! d, k$ J
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    1 j: b0 R0 y9 [2 T6 \# q& A6 T9 K$ r
49. bool   deletePortmap(int   eport,   const   CString&   type);
    1 C/ f9 z3 S, E0 k5 S( b, j# _& x
50. 
    , n" _, T, T! ?2 m
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    & e% x, L% e9 F
52. 
    % [8 F6 m+ {$ h
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    - j+ d: l/ K6 }9 e; V9 ^
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    " h& w- n" Y$ m" I  b# i$ f' l
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    . n7 \9 {- o* w0 m/ C% b
59. bool InternalSearch(int   version);
    ( E1 i- g" {4 b. ~$ f
60. CString m_devicename;
    
61. CString m_name;
    , [, o5 V9 g7 P  k
62. CString m_description;
    6 j9 m* r# ]8 I) M. Y1 n- v
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    : c  Y- [: A6 Q/ i0 ?& ?5 h
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    & {1 Z2 n2 D9 B# c
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    ; {, [: l; w/ f0 J
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   3 Y3 O+ L5 M/ d2 R6 @& g* ]
3. 
   8 R$ H5 N$ J! H# ^2 `/ H
4. #include   "upnp.h "
   - }4 b( i3 q0 ^( R2 `4 D
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   ; R/ m2 y+ p) `# q) z3 y& A% h
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   ! I- w, S7 W1 O; ]8 Z
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    ' I) h* _) t8 H. A3 h8 u* L
11. 
    * i+ W# m+ T2 z: L# w1 T6 x. Z
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    ' s! G1 |2 n+ n) E& P7 ]- B, l
13. static   const   int UPNPPORT   =   1900;
    9 l9 x8 z4 C" Y9 c6 u8 V
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    ( }+ S7 s( p6 @3 q1 a
17. {
    0 R! |6 W- i2 p5 L/ f( A
18. CString   s;
    : \' P! g6 b/ i
19. 
    + A. l$ N1 M6 x) e/ ^
20. s.Format(_T( "%d "),   i);
    
21. 
    6 }, f( X7 r/ e  X/ y, Y
22. return   s;
    
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    3 j  x7 _- g  [& ^
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    9 @% i6 l- e+ x9 A2 _6 y& x( M+ f7 f
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    4 g% K, _5 b, ^$ }
33. }
    ' `4 l0 ?8 D& E9 m
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    9 l5 e! h) D' f! l
37. char   buffer[10240];
    4 u8 y0 @5 k' g$ n0 y
38. 
    / {" ?  z: J# \) _, r6 c1 B
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    ( x: |% [2 G  R
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    4 s6 t% }5 w2 ]5 T
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    ( g/ N; Y/ m$ R; N6 ^8 X
46. sockaddr.sin_family   =   AF_INET;
    & f; R. g! s7 R+ @
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;
    " t2 h0 g  w4 _/ C
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    8 E6 _  f* M& Y/ \
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    ' N0 w6 }6 c) C4 I( p. _
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    4 P  N! t$ g" S% C0 ^9 I# g
59. if   (!rlen)   return   false;
    
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    ! s9 N7 ^8 R3 t0 t# ^$ [" ]3 g+ s
63. return   true;
    5 @1 @5 D+ E' M$ W9 l
64. }
    8 D+ S  V. `$ ~2 B
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    6 T$ [. W4 l/ f+ M# E5 M
69. 
    * y$ q( N4 ~3 |$ N& G
70. const   CStringA   sa(request);
    ) g% T2 J1 r- b/ \% d3 r9 C" n. @1 B$ C
71. int   length   =   sa.GetLength();
    3 R+ \  d' A5 n1 J
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    " |  y# ~+ w  }7 D% [$ @
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    3 y# X! l/ G" k  q' O( b5 t% A% e
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    1 E7 B+ K4 E- T( S  F0 ]
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    ) ^: q) I" m0 E3 S% N3 Y9 `
88. 
    
89. result   =   response;
    ( r2 |; W' K& G. _
90. result.Delete(0,   pos);
    & q" P$ }9 k$ q. `
91. 
    , E+ B$ j; o( s1 }$ L8 i
92. pos   =   0;
    / \, h& P" ]& U0 K
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    " h3 L. g5 e; N  `2 X
96. return   true;
    . |9 ^( A9 p' i1 n4 |9 B
97. }
    8 X1 E) f* E" M6 i; i. a
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    & B6 K! x1 Z; y" G0 g) u! t% M
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     7 ~) {  f- O/ [  J- V
103. CString   property;
     
104. 
     $ ]8 O3 l$ ^' j! I* Q
105. int   posStart   =   all.Find(startTag);
     % |6 e+ @- a, k. ^% _
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     / ^6 q5 W& m  e% q$ E% ~+ V
110. 
     1 g' L. d% c  r' _# |
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     7 i$ H* x" Q$ ]; P) p3 ]
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     9 b8 u+ m& l* w% o# E5 N
115. :   m_version(1)
     9 m+ s+ _% G) ~& c9 ~$ ]
116. {
     
117. m_uLocalIP   =   0;
     / q; t! f. O1 L) r! R
118. isSearched   =   false;
     : L- _! n) H$ F( D/ h
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     $ V$ i# K; y9 O  Q& J
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     4 ^9 e. e" X$ x% e/ F: A
128. }
     5 @/ j5 r  S8 q+ G( Y# `
129. 
     + H# R6 }( E0 T$ g7 T. [9 Y- E
130. m_Mappings.RemoveAll();
     
131. }
     / S9 `" D  D- n/ Y0 F
132. 
     # [" m0 ^0 r$ M' J
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     9 P& L' F- j* L9 Y
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     ; q! s- Y2 C4 m* u5 ], Q% K
140. CString   devices[][2]   =   {
     : }$ ~0 b* x3 l: b! w2 n$ r
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     1 O/ ]7 o) Y; v4 m3 ]% q
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     - K# }; Z7 s/ M. Z
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     * `- L% I; w+ _' Z# G# b' {+ v# A8 s
147. u_long   lv   =   1;
       R% M. b9 w7 S4 u
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     $ L5 |! u& B% d' T+ C
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     + [9 f  k7 n1 |3 m
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     ; ~( s8 M% E9 v" y  S  v1 `
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     * j4 X. p1 f/ S) y% h3 a* w
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);
     / `. u4 L1 _* K: `1 H+ e
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     * ^$ @' D0 \4 h+ t+ K
159. }
     5 A+ V* O2 a* A* v. k& W
160. }
     & ]3 Y" \2 z" c; U: ]: R$ y/ R7 Z0 y
161. 
     ( g6 s, T! o; o- P) ?# Q
162. Sleep(10);
     , b6 @' z9 ]+ h% j
163. 
     
164. char   buffer[10240];
     + m1 o, J2 ?0 m( ?( O, S: n# h; ~
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     + \" a1 B5 {# J, w0 g% {
166. if   (rlen   <=   0)   continue;
     4 \. a# ]" y; j' j$ b
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     . v) N4 x' n5 R9 q9 L; G2 j7 j( S
170. CString   result;
     : o% z. \2 O: k
171. if   (!parseHTTPResponse(response,   result))   return   false;
     4 q3 ?6 O4 A* i7 d6 e$ \% c
172. 
     6 v% c0 B9 R3 v; P5 [* k+ _
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);
     8 D* k( s" L; H: W0 a3 ?
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     ; r0 i! D+ ?- C0 o% n! o
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     % H" H: k1 f$ e9 B8 Q
182. line.Delete(0,   9);
     
183. m_description   =   line;
     5 {4 Z5 ]" y+ z8 Q
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     4 z+ i! u0 `; K9 J3 o; l
187. }
     9 |' s4 ]; }; E* ~2 r
188. }
     : x/ S! Y9 ?  {: T. u) A
189. }
     - O) g4 j& D" l7 K9 z2 V: V
190. }
     
191. closesocket(s);
     
192. 
     , F/ p$ i5 f  y, B" z1 h0 }+ S
193. return   false;
     
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
$ T6 {, j, D9 P( q1 z/ O$ x

8 l4 e7 T4 p, N4 V1 \///////////////////////////////////////////
6 W7 [9 n# b, o" M; }8 W9 d+ F+ N//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


( C* T. a" I, O; P% E1 a8 v1 Z+ V( c#pragma once
#include <exception>

. e1 j( Q" b( W" Z3 u! F
  enum TRISTATE{
        TRIS_FALSE,
) m$ Z$ M8 ]$ C% Q) S& |        TRIS_UNKNOWN,
        TRIS_TRUE
3 g. S! s& b+ r- U% ^& p& ~4 w};

4 k. ~" I% |! v/ o# ]
enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
* n: z- k' h8 A+ c        UPNP_IMPL_NONE /*last*/
% a' t; p- s2 ~};
) b8 K# E- J# l; I4 Z: p* T# K

9 j0 I  K' b) d& z

class CUPnPImpl
{
public:
9 Q% V" W+ p( `8 F# X0 n        CUPnPImpl();
. T6 F1 L$ a  @. y        virtual ~CUPnPImpl();
9 f1 Y) C5 E$ |        struct UPnPError : std::exception {};
6 ]: V# h! T/ M$ `, U        enum {
4 a. Z7 J2 ~/ `  e( O( K, N* Y                UPNP_OK,
! e. h* I9 _, Y  h- E                UPNP_FAILED,
                UPNP_TIMEOUT
        };
$ H$ F0 D' a, s2 v" j
4 J1 E8 B# Y/ T$ {7 U" E
/ a' K6 H$ w+ T8 b+ D# w+ {9 ]        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
  ]3 }; Z$ c% W7 @9 n9 M, K        virtual void        StopAsyncFind() = 0;
- Z& s# Q& r; c0 J( T1 Q- ?        virtual void        DeletePorts() = 0;
( S. y/ H- f3 Y2 F, O1 I, q$ o7 s        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
% [+ d: k3 }. u' Z& A" k' M
- i. m7 q1 t' E+ T0 _; _! |; Y
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
! B% Q! ^4 U& M8 |+ h' t        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
) u5 p# D7 B/ _' i7 ]' y        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
: J% p# u3 M) f& G& Z% I/ T! D        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
2 d+ o0 ~7 M1 h$ [. `0 {* u' {
* d, i: h0 C) S
// Implementation
; Y/ Q7 P* S0 C4 m* ]; |0 {protected:
2 m/ X* N- V1 V1 O# f        volatile TRISTATE        m_bUPnPPortsForwarded;
) k; u6 a8 A6 l, z4 G        void                                SendResultMessage();
% Q$ @" `4 V/ R, ]0 \( s" B; `" j        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
1 U5 q, a$ b# }- _. ^        uint16                                m_nTCPWebPort;
' q: z9 K2 e7 u$ u* B        bool                                m_bCheckAndRefresh;
0 Q& m% I1 ^0 [. ]

private:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
9 d7 p0 E8 k, E0 Q! B% o  P1 L
) G/ T* y6 n5 F/ ~
' {. t8 m; B, Y2 h, M) @; L};
  R6 q) C, P$ I5 b5 @( t( [
: t7 g1 @. s) k, W. A$ N
// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
7 g& ~5 P! b# Y2 W, P{
public:
6 Z( X, V3 J/ r$ h% R        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
! s7 s. Z- C& _/ X  J        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
/ P- @- f0 J: E" i* {7 g+ p' ?        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};


/////////////////////////////////////
+ v, s- G% M4 S//下面是使用windows操作系统自带的UPNP功能的子类


#pragma once
3 N, D5 ^) k( p#pragma warning( disable: 4355 )
. C3 W2 s6 k, U' C; r
% G; _- B) K/ g# z% I/ _% s
#include "UPnPImpl.h"
#include <upnp.h>
, i) G3 I6 _( z* Y' [6 Y#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>
. K0 c1 W; F+ \% W

6 V: C4 O5 m, q% f$ Q#include <vector>
$ ]/ G: ~2 @9 k+ \3 z9 l9 V1 k#include <exception>
#include <functional>
7 b/ T' d3 Q8 @9 P3 y



typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
0 T' c0 ]: T! ?, itypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
  t6 N6 }7 ?1 y" q+ mtypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
! U! l& I+ Z. x% Wtypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
; P; P4 G2 ]8 N) y1 ptypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
. ]; E2 K1 Y# s% n, ]

) j+ ?' X4 W+ h$ W% Stypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
( Z: p# O0 M7 {  PDWORD pdwBestIfIndex
);

" ^3 b) ?) U. `2 B$ \
3 C. n1 W" Q. F+ `typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
: u$ \& G! E4 p/ I( T+ z, I  BOOL bOrder
" ]4 |# h" |0 Q% {8 N1 C);
4 [4 a" @8 B4 Y

8 X% K: m- H+ b' htypedef DWORD (WINAPI* TGetIfEntry) (
( a5 ~' g( S- N0 w- \. P( f7 u  PMIB_IFROW pIfRow
9 E; |- F$ M: {  n4 W# c2 T+ @! C);
9 Z) w4 x( r. P) q" k7 J8 V( C

CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
# l6 D$ A- g" A- r
% H: W# v8 e( }& G8 y( t+ Y% R" T
! H* W" |8 f2 v/ g: D% \, dclass CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
: z! m- c& p) h" C7 ^' y6 N// Construction
9 C1 p6 D9 d' ^0 U2 d- tpublic:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
# ^8 E6 Q- B7 a( W# [

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
& ]5 _4 N3 a5 t4 K- s( G5 q        virtual void        StopAsyncFind();
) u. n0 ]/ U! q( z  y9 x        virtual void        DeletePorts();
7 j3 T+ T& P# E        virtual bool        IsReady();
: v# w8 m. p2 i% D9 U7 Y        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
. C0 w) F, `% p
& Q! X% }3 L9 Z1 o, \
, G$ }5 V6 ]8 `) V- e- P        // 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; };
, C+ W! Z. y; C; ^" }! h; h

% d8 c& Q( M) a0 v( H+ D* `! cprotected:
+ ]! q* Z3 g4 v6 k        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
4 T1 C4 {2 X. l# P2 r        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
# m- {+ C" E4 K+ ?, i; J        void        Init();

$ g+ v6 Y+ C8 O$ r+ G
1 {* M( i: Y( X        inline bool IsAsyncFindRunning()
        {
3 F$ y  a+ R. x$ n5 N3 [                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
7 J. n4 j! c# |/ p' `$ z+ _: u* R3 |  J                {
! B4 E3 P6 J4 |6 [                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
8 l' v0 A  M+ O$ x1 P8 t: I. a                }
  y; t9 {+ @( K, m/ s" X1 \                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
; b3 v0 p8 i* `$ g# ^                return m_bAsyncFindRunning;
$ R# A5 r2 O" q' T- T2 J        }
- h2 Z, ~) y$ _1 t9 U. h# Q
# ]' |, x2 V1 [  H8 j
/ l1 T* Q5 y  o. i7 }: i% x6 L( d        TRISTATE                        m_bUPnPDeviceConnected;
8 ^4 O) Z0 W1 X' F$ a5 O" A# k

// Implementation
        // API functions
1 |3 J' I+ a8 K: K/ @0 T5 Z' G        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
, W- c! g7 I9 d8 W; P6 ~        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);
. j' b5 l; u  ?3 y- W( U: z        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

( g% B: m+ l7 j6 F9 e
        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
- Y$ I2 g, ]! O( L        TGetIfEntry                                m_pfGetIfEntry;
5 ?2 g" l5 h3 U: ^8 D1 K
1 z) ~" [4 b, U9 ^5 g& J7 W
        static FinderPointer CreateFinderInstance();
4 }0 i+ H4 S% i. N& A; o6 P6 E" n        struct FindDevice : std::unary_function< DevicePointer, bool >
5 |& u0 u% m% f        {
- C, B; w8 U3 D                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
/ [, o: R/ h  h3 C* H" I& T                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
- X/ Z+ p1 Q( n- B+ k

                        if ( FAILED( hr ) )
) n/ X5 @; [$ r/ h2 G, j: n% @                                return UPnPMessage( hr ), false;
/ F3 h+ [; R* h" h& l
! n% q4 d3 F; I/ ]; q' ^2 G
) w' H5 z& a, ]$ E2 N$ V                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
  ?/ ?' R: a& E8 r9 m$ V        };
9 h- g4 D5 ]1 `! B       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
* v$ I; X; o4 M/ f3 D- u        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
1 u, A& E& ~% i2 f- i1 L4 O. f        HRESULT        MapPort(const ServicePointer& service);
& i" F) H! ]1 r# u: _$ |) Z        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
& r5 V9 C/ G0 ~% X* f- N% A                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
# b6 r: c& g" v/ s4 S9 n# K3 Z* Z

6 W0 b9 X6 |0 P. u3 f2 j% t        // Utility functions
) C9 v8 T/ E' B) V: K        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);
$ P% E7 M9 J' D* V        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
& c6 Z' j) o) }% `3 v, i0 A3 J. e        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
; Z3 J: }/ {+ g3 y        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
8 @5 t7 e: _9 [! c2 P5 F

// Private members
private:
; m7 w7 [( I) H6 d        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
: _( u! o7 r! w( O1 x: ?        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;

% E! M2 C6 \6 K6 g% M: k* K
        LONG        m_nAsyncFindHandle;
$ |2 s1 [5 D6 M        bool        m_bCOM;
        bool        m_bPortIsFree;
3 H) ]$ y  `1 x# L. b        CString m_sLocalIP;
' Q# z/ J( U$ \! n/ j        CString m_sExternalIP;
/ T# _2 Y+ H# O        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
$ w) ~4 ^5 {) n        bool        m_bAsyncFindRunning;
$ g: ?0 s' h/ Q1 I        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
+ H, q0 k5 h$ V$ E4 ?        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;


: w+ K" x& y) C) l" d% }};


$ x$ k4 @7 ?( V! e5 U8 g: s// DeviceFinder Callback
1 v( T0 N+ _* @# ?: v9 D$ Gclass CDeviceFinderCallback
3 T0 P' [+ C& c4 C, v        : public IUPnPDeviceFinderCallback
$ F) G- V4 ~# ]/ R{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
% p* ], y+ |( j. Q1 G                : m_instance( instance )
        { m_lRefCount = 0; }

3 ?) K( q4 g2 i
% l0 j, X& X7 d/ u% r, d2 m- B3 C1 m   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
  s* h, w" ~( Q2 V2 m' g   STDMETHODIMP_(ULONG) Release();
5 m" I5 x& e1 @& n# m3 d6 ^
; _& P: o2 O/ z$ [5 P- V
% G- K( C: `* H// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
) W- \( y& P6 @$ _9 D5 D- w! I* ]        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
8 l# J# o& q# [, J( q
8 K& g2 A; i" c" q. g3 |3 u/ W
/ n. r' T1 a, g+ q7 r/ lprivate:
  g$ R& j" w" H; l        CUPnPImplWinServ& m_instance;
0 O* U) A" S0 ]6 T: K        LONG m_lRefCount;
};

( |% W- Y- j* E& P6 R; L: J3 c
// Service Callback
' Y4 I8 A& w  `3 Iclass CServiceCallback
" b$ ^) z8 o& W# _        : public IUPnPServiceCallback
{
$ T7 |3 E3 b1 d1 ?3 ]public:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
! Y3 \3 Q1 X# n6 _   
0 M6 J  @; q3 X! p   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
0 S' V; O4 Q2 T" o3 s   STDMETHODIMP_(ULONG) AddRef();
( J* d* b/ l9 B7 z" W   STDMETHODIMP_(ULONG) Release();

2 p2 u9 W+ [% ?8 w8 Z. @
// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
5 f, o) m  s" f# [$ q* w        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
, F4 r/ ]6 U) V

private:
% T( X# }8 t# Z0 M* q0 A        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
, O. B( @1 Y8 e  R# o  J% w};
; p/ m/ }' ]( F+ x7 P6 |  O2 g& ?6 K
/ [2 r+ Y" y- I9 X4 y
/////////////////////////////////////////////////
  ?, Z/ W/ g0 m8 j
3 X7 m, {2 Z" o
2 [! P  W: v4 v! y- ]使用时只需要使用抽象类的接口。
+ I1 i  T1 P$ s/ XCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
7 M2 z4 F# S! a( U4 C1 y, D  J4 uCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
* ?$ s' [" N* u: m" Q5 R$ ]9 z0 D  TCUPnPImpl::DeletePorts删除端口映射.