UPnP

天云上人

/*uPnP.h*/

1. 
   $ L: w" c% r2 p. x
2. #ifndef   MYUPNP_H_
   ; _7 M, a, F: n! t+ _& f
3. 
   + h# e0 {8 b1 A( Z9 l0 U! }
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   , x; M; G& Z+ e  I9 b% I3 M
7. 
   
8. class   MyUPnP
   
9. {
   
10. public:
    
11. typedef   enum{
    : s/ x' W5 l: }0 t3 H2 e
12. UNAT_OK, //   Successfull
    3 T# A2 T+ G& G& A
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    7 t; I, @- v+ Q, X) S
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    # T7 C* W+ [& s  q9 u4 I! P
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;
    9 w5 a! |% v  P1 ]2 f8 i
18. 
    : B* N3 X: E% B& v& ^
19. typedef   enum{
    $ u: d; m5 b. B4 U! d8 l6 x# X
20. UNAT_TCP, //   TCP   Protocol
    4 n9 w( m! o5 S9 I
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    2 O2 V# U' {0 S/ G) Y
24. typedef   struct{
    ( E4 c" o  l+ q% E
25. WORD   internalPort; //   Port   mapping   internal   port
    ( X! K1 d; M( r* v: J& u5 m+ }
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    . q: P. z* D# ^) o& X# d; l
31. MyUPnP();
    , ~# I+ I$ Y6 K# |% x
32. ~MyUPnP();
    
33. 
    9 L9 }4 N6 Z/ I3 Y
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    9 U( y* Y& l& `, [. v$ c
36. void   clearNATPortMapping();
    3 i5 {$ @- G0 Y0 D# x0 a/ ~; P
37. 
    8 A% `( A- F  F) z3 \: G8 r' T
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    ) I! F) ], ?& i& n- g% q% K) M
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    & r: X: {5 c" @0 [  [+ v3 Q
46. 
    6 V$ u, F, ~# J' v( H. z4 |, a& L
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    ) z3 ?" B4 z1 x, C
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
      }  K4 f- z; |9 ?. {6 S7 ?
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
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);
    ' ]$ r# O1 ~$ U6 i
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    ! k$ D# W; w, [4 \/ I# E) H
59. bool InternalSearch(int   version);
    ( j; {# ]0 Q$ i4 K" S" L
60. CString m_devicename;
    2 Y4 a- w& s  [0 z9 E* s& p" U
61. CString m_name;
    $ X" E2 B7 K3 a( N  ^
62. CString m_description;
    8 h' V5 C( F: a0 b: j  v
63. CString m_baseurl;
    7 V% h8 Y- T6 G
64. CString m_controlurl;
    / B# b4 R/ z& x! Q- \: s, c
65. CString m_friendlyname;
    
66. CString m_modelname;
    & M* e/ @: L" R) d1 V* S
67. int m_version;
    
68. 
    0 D9 ^! u# C: d/ Y9 {
69. private:
    1 j0 `0 F4 u1 f% a  ]* f
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    , @" z  N2 a: N5 i; ?
71. 
    6 L- k# t" L7 z5 }9 K9 n
72. CString m_slocalIP;
    " c& z9 i8 O( F& ^+ s
73. CString m_slastError;
    9 e( [& ?, ~+ D4 E% i9 r- m
74. WORD m_uLocalIP;
    
75. 
    ) ?$ [; a! K# A! `& V( m% d
76. bool isSearched;
    + X( |) s$ ^; W0 z: f
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   ) b9 U% [; A% V
4. #include   "upnp.h "
   0 }4 |3 \$ a1 x0 O/ M$ }
5. 
   3 Y8 s" w4 u1 Q" S1 C
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   + W5 W! O4 r. s
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   ; ~5 M8 c, o4 t, p/ r
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   ; \: [' n' S, h: Z# t# V  a
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    $ K0 e; [; f1 p! g- l; Z
11. 
    ) M! a1 P1 [6 f; `: T! |
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    / g9 [' f& {, B( i6 w
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    1 n1 d; G* v- n
16. const   CString   getString(int   i)
    
17. {
    ; v! h/ @5 Y: c7 c
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    % W: z! g# x2 o( s+ c
21. 
    
22. return   s;
    , `& y0 v7 C. a. g
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    7 \. s$ v5 e" V" }% ?
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    : R, m8 |* S2 f
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    8 y3 G$ D! ]$ u2 N# u
33. }
    
34. 
    2 O4 d3 p3 S" S# h
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    
38. 
    , d* s0 @  L1 D! T
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    1 t5 t- o' {" I
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    4 ?( n$ L  a  R5 i* ]6 D) L
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    2 B  ]" c- ]& K; t" ?# ?2 z* {
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    4 w+ T/ `+ Y" r/ h0 W9 D2 R6 P
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ' H$ D* ?0 W7 L' i6 ~9 v8 |- a
53. Sleep(20);
    : ^) B* l) Q4 c! G$ n
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    8 v" J) I+ P# @
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    - c- a  }) q  V( ~4 D8 e/ G' w
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    . M) |; P: {+ @2 d/ I0 z0 ]. T
62. 
    ; F& T1 F; }% U1 ~
63. return   true;
    . g3 s0 \8 i& e& D+ N3 l" Z
64. }
    
65. 
    ' n( ?" B) {( j0 ?
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    ; ?/ y- s7 M4 J$ h% M( w: w9 p3 i/ ^
67. {
    
68. char   buffer[10240];
    + g$ F! M$ i' R- i/ s1 }6 {5 Y, P
69. 
    7 E5 d  k! q2 C: w' z7 E6 H
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    ! [6 p& |- I% q% _; C! E
72. strcpy(buffer,   (const   char*)sa);
    
73. 
      [$ L- G! u5 w4 h( v/ p  x1 m
74. struct   sockaddr_in   sockaddr;
    . U" O$ j. I9 w# e- X; F- X- Z
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    - A8 P9 F  f7 h
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    - |: }3 O9 z, L- z$ G# ~# t
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ) x" Q1 A) {. d! |/ Y4 N! L' o
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    ! d2 r' ?8 |0 ^3 A8 |% x
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    0 O9 V( f! b. ^( l" ]! H7 z8 T
85. int   pos   =   0;
    8 N, i3 s; C' q9 k2 {
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    4 o6 P8 Y8 G4 ~
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    $ f9 f0 f1 ?* O9 N& ~& g" l% a5 B  s
94. status   =   status.Tokenize(_T( "   "),   pos);
    9 P9 |( s5 ]; I: K* k. e
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    5 F- d' a' e% W  g
96. return   true;
    ! ^" ^% b0 p. D3 c# ~
97. }
    ) h) g7 n- G7 @( z$ L% L
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     8 ~. t9 p" _* r& K: w$ R5 W
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     0 z& }- R8 ]3 \: e, Q" l
104. 
     
105. int   posStart   =   all.Find(startTag);
     1 a+ W" i' d. ?  |  B. ]; X
106. if   (posStart <0)   return   CString();
     
107. 
     8 P5 e* k3 h: A/ B
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. }
     ) v8 n8 N5 t/ Z- _: `
113. 
     ' r# l; s- o! X
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
       C3 V. ~  N' S# R
118. isSearched   =   false;
     ; ~* r9 u; f. m
119. }
       H8 X) w: W( o% X8 R
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     1 E% ]3 m! `6 B. i. W
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     3 M4 i: l% E: F0 G8 P, z2 e
128. }
     ( A0 p4 L* H/ M  j+ i# ?6 _
129. 
     
130. m_Mappings.RemoveAll();
     
131. }
     2 z! N; L) C: _- w/ q! `: x
132. 
     
133. 
     ; r; x6 l) @. r- @" Q
134. bool   MyUPnP::InternalSearch(int   version)
     % _1 m, M, |/ q6 }6 t- S
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     + x' m' l! c* ]8 E! V# `
138. 
     ; n/ E) m" F5 X# G
139. #define   NUMBEROFDEVICES 2
     / a+ c1 \3 a$ I: G* q8 z% n
140. CString   devices[][2]   =   {
     + I4 F7 j/ H% C' E( b! C9 t
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     - b5 }) }/ w  b+ ?" _$ ~
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     : A+ [+ B; {$ K
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     ) `  q7 v/ p. {. S; w. X
147. u_long   lv   =   1;
     , C: g$ s9 J) D+ A6 p
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     3 n# e7 O1 \4 Z5 U1 K0 Q
152. if   (!(i%100))   {
     $ A: e: P: N+ l' ]+ z- l0 i6 [
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);
     . p9 L) t. K5 f* N8 y
155. CString   request;
     . f, N! N1 [% L  I. |2 K, D
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);
     + ?$ z0 F3 W$ H- X9 C8 F
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     7 o8 [9 K, U/ C& c7 H& H2 Z& V
159. }
     : H0 U+ \) {& {" x/ z! v% U' B
160. }
     ; S$ L0 e8 E9 y" D' \2 X
161. 
     " b/ n) |+ J9 B: d
162. Sleep(10);
       {5 c# V5 f: ~& Y, ?  g- @
163. 
     . s  {* G: [( _4 s
164. char   buffer[10240];
     ) n+ Q9 [) h0 l. A' G
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     + `2 Y' A* D* D+ K5 W
167. closesocket(s);
     
168. 
     2 L6 \, I+ Y( Z& A, i
169. CString   response   =   CString(CStringA(buffer,   rlen));
     + n$ v; D. j+ z* c$ C; y
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     3 U7 {8 H9 m5 @7 B& G7 ~( F
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);
     
175. if   (result.Find(m_name)   > =   0)   {
     * w" l' \; M2 [  J2 A3 ^7 w/ K
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     8 Q, w( f/ c8 N0 i' E3 [1 B$ u
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     ' Z  H, i7 ~- M, H4 J
182. line.Delete(0,   9);
     : I1 o" o! N6 C8 b
183. m_description   =   line;
     5 m8 Q0 e: d. i
184. m_description.Trim();
     
185. return   GetDescription();
     5 e6 R5 k% V8 _7 W3 Z
186. }
     
187. }
     
188. }
     7 N, |1 |6 E' {4 K+ ~
189. }
     ' t! W4 ]3 _7 t% d4 X' [2 p
190. }
     
191. closesocket(s);
     / a4 W5 R/ H% M
192. 
     3 {. _" V+ c8 P9 f
193. return   false;
     ) V( h  M3 |3 e' B9 P- j7 B9 M( R
194. }
     + Q- R  J& v0 N8 V8 ^/ J1 t0 _0 v! W

复制代码

天云上人

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


///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


: q; f0 y; N3 x" K9 n- T#pragma once
' m7 x& B( m1 o0 B4 W#include <exception>
6 Y3 \5 y7 }2 w! }+ e

  enum TRISTATE{
        TRIS_FALSE,
+ ]0 D  C! x) A8 V8 [3 \        TRIS_UNKNOWN,
4 X4 Q3 [0 y. r; g& b+ m3 I        TRIS_TRUE
};
' {/ A6 e4 K1 b8 v; F% t% w$ p
, g+ n& a0 G) b$ U% s$ v2 \9 i
% w5 r; j* t' ?* H+ F! F+ Z% benum UPNP_IMPLEMENTATION{
: b. e! u( `! B        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};

" x- U& B; z( c/ ~8 d7 \# V
! w$ A, ]0 D( J4 V

class CUPnPImpl
{
public:
        CUPnPImpl();
" h# j7 R, ^6 K* G" ^/ \        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
6 L% w4 b8 _, _; G4 |# p        enum {
! ?; n' o/ T3 `4 B& _9 f                UPNP_OK,
( A3 K& Z" S9 r: }: U7 W  X( d                UPNP_FAILED,
                UPNP_TIMEOUT
& ^, Y9 X3 I. C' z; o, g; K        };

8 X0 n# u8 A( A$ P1 ]9 @* Q& L" A$ h
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
& h4 m  o' Y& W7 Y6 T( Y: [        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
6 k! y/ A6 \* g; Z) a) K        virtual int                GetImplementationID() = 0;
       
) l/ q, a( b6 \2 Q; h  v7 h        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

1 m. a& O) y5 b7 p. d# q1 X
& `+ U' Q- U% q- _2 u/ X: B8 Z' _5 R        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
6 m; K. M( W  }( P! J( }        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

/ Q( q' u+ D  s( G3 \) m" K% g! ]8 o
// Implementation
protected:
4 o; h4 }5 M* d        volatile TRISTATE        m_bUPnPPortsForwarded;
$ A# Q4 r) o2 l, {1 n4 T9 k( a* I        void                                SendResultMessage();
        uint16                                m_nUDPPort;
7 j2 Y. Q  {9 h8 Z* ?        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;


private:
* L, U4 N! m! I        HWND        m_hResultMessageWindow;
# J  s9 k8 |: [4 @$ m        UINT        m_nResultMessageID;


};


// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
: L8 {( {6 Y+ j{
public:
, S2 _/ T( u0 K( V' @; `4 ~        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
" H% {5 K& }  W; z' Q        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
) B0 B/ s+ q1 \" {( ]3 A( |* F};
- `& |1 e; |% k/ q( Q$ l
. z$ H' v+ H% `& y2 |1 i% q: Q/ \
/////////////////////////////////////
8 \) X: s5 F+ E4 |9 V//下面是使用windows操作系统自带的UPNP功能的子类
1 J' i) m( F* d. {- N8 r
4 D4 x+ g8 F3 N8 ~/ M
7 ^- G2 z, y# P' v; l8 `#pragma once
#pragma warning( disable: 4355 )
8 }/ t$ }0 g2 ^

- t- q8 h) C8 {* P* g0 g" v#include "UPnPImpl.h"
% ?0 P8 l& z( n- _3 x. [8 ~#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
0 l- s9 Q, \( [6 K# G#include <winsvc.h>
* z8 N( k% g% H* A+ U+ c/ i& A; Z
7 ^& o* N2 I. _+ l
. Q) c( n/ H1 }. s! S#include <vector>
#include <exception>
#include <functional>




, w+ J% }, h4 S: N- ]# O$ ^% c, e; qtypedef _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;
) E+ t: ?+ l! M9 s* G3 atypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
8 B9 ^; A- U9 j- otypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
6 }) B5 F- v5 S& G7 _typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

, o  u$ N% G: r) K( T# `
$ I* J( f5 F5 Ltypedef DWORD (WINAPI* TGetBestInterface) (
4 F6 `( i; \; f( l6 X' F/ F: j! c! u  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
$ N5 f$ t  p8 @, W* Z4 a2 r( M);


typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  i+ C6 ?- Z) d  PULONG pdwSize,
  BOOL bOrder
$ M& H+ `7 _3 r: G8 _: S);
$ j1 ~- e- W  w+ R- g" s, ?0 H# P1 p
3 T) T+ z7 H+ X
; n1 J5 O5 O. otypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);
9 i  r8 V  K2 U3 E

CString translateUPnPResult(HRESULT hr);
* m: Y& }* V7 K0 o2 }& cHRESULT UPnPMessage(HRESULT hr);
7 m! D* z  A- B/ E

class CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
public:
        virtual ~CUPnPImplWinServ();
; a: j5 P4 T2 ?  I/ P$ v        CUPnPImplWinServ();

8 w/ q* U0 o9 F! W& }2 g
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
0 Z6 ~6 ^1 ]( o4 _) t        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
) Y# q8 |  Z4 }# x) d
2 Z! v2 X$ N, G. S- v
        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
% q: _. ~5 q9 {' t8 l        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
4 \, n4 ~, r$ R

protected:
, I! E/ s# y3 O& [        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
" a! c1 F( s- n" m8 @; ^        void        Init();

  \5 q) _- d6 `
        inline bool IsAsyncFindRunning()
. [) ^; e% Y0 g) |& _, L        {
, x2 L' \7 z" O  L5 A                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
: Y6 A# ~# b9 ]) n! E' I                {
* ?7 N/ N2 x7 q1 p. Y, Q$ ^                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
' J- {7 c! \2 f2 y/ G5 C                }
                MSG msg;
5 c. C2 P3 s7 m1 c                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
) B6 C* I7 N* N5 \; o                {
                        TranslateMessage( &msg );
1 W; B; u7 I9 C( u0 j( @+ I                        DispatchMessage( &msg );
8 c9 `8 p( A; J% w) M( w( I                }
3 |/ U2 n- z$ X                return m_bAsyncFindRunning;
        }
) z" W& M; P1 x" ^* M8 t* Q3 R

! k* B+ S1 k$ ~! a5 `# |        TRISTATE                        m_bUPnPDeviceConnected;

7 h! u" B+ l1 A8 L
# l7 |3 z7 P9 V1 f2 i  w// Implementation
        // API functions
6 t& l5 X/ m9 c9 B8 o: A2 u        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
. }2 T/ ?" C1 B+ I* M+ z        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
  G2 n2 R/ u/ y- K2 X5 X- j        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
# l3 u% \: h8 u3 z; P) m        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
. z; B( _8 f. n0 E4 Z        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
$ n& {- V2 {, d7 `1 {1 P        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
) [. g! v8 C; ~- c1 U) c

! H0 x9 f0 N! @5 M        TGetBestInterface                m_pfGetBestInterface;
: m1 j- t4 i: G! Z' R        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;


6 }6 C) m6 X, [" {" `$ d        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
4 [* O, J+ r; q; Z! p                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
; t5 u9 N# @* y0 z8 p% E, ~

1 u5 Q/ Z" b% c& m7 L                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;


& N# d- \0 g' ]0 |' M                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
, @/ w) ~: e7 l* ?2 r8 H                CComBSTR m_udn;
; D. G1 U- y  [$ n9 ^. i' m6 L: W        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
; b; O; {. N8 D- z* n        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
# ]0 V. e) Y9 K" I' U        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
, U1 T5 x' z& D6 S3 X3 o) L, ?        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
4 P7 n* `" R3 Q7 j  N  t        void        StopUPnPService();
- L, d& x9 u8 N& w* [0 Q1 C" [

. o) x. w7 w4 T        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
9 s$ S" I  {/ A& w* X1 H        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
: f: Y1 Q  q# }) U$ p. f+ H9 \        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
* e% s& m; p* r, w* B. p        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);


// Private members
6 F; C4 B' O! l5 v* z; E& Dprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
9 d" |4 {& I' i        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
6 ]' X( J4 P# A- L) h$ U1 ~        ServiceCallback                        m_pServiceCallback;

' A* S  P' D) }, l1 H
0 E* c, T1 b: B4 A; h2 Q4 b& y0 |3 }        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
" j3 a+ l, t0 E+ _" B4 A( {9 U# s4 E        CString m_sLocalIP;
! b' w) |# l6 B* Z, e8 g        CString m_sExternalIP;
/ F; e$ u/ _$ I% u3 Z        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
$ a2 R3 D% N  s% E& b6 D* H        bool        m_bAsyncFindRunning;
4 V% l) \6 C- N3 |% c+ M) J3 f        HMODULE m_hADVAPI32_DLL;
: K0 G4 c& s4 X3 U6 a4 N        HMODULE        m_hIPHLPAPI_DLL;
( Y3 i* ^. g7 i* i$ B        bool        m_bSecondTry;
9 T' `# U: D( x% S8 n        bool        m_bServiceStartedByEmule;
9 T3 d) D5 y. M! X  _( i! ]        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
, O8 f2 y8 B; b8 z. a, o
- D, I3 u- M" P
};
. S+ u% k1 |: K
' n" e: u/ t$ `( t* f
2 ]# {2 t+ J8 r5 V" L4 x// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
* d* O" Z- i; ^7 B0 q9 j* d6 h3 a, ~                : m_instance( instance )
: c! g, O" j9 f0 }2 g6 m5 C        { m_lRefCount = 0; }
- M) Q! X+ m% W/ j6 i4 Q
1 z8 N& @: _" G0 F1 X
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
, o5 p/ Q' Y# a2 N+ E   STDMETHODIMP_(ULONG) AddRef();
+ [0 ?& M# A9 b; V+ [9 S7 E# }   STDMETHODIMP_(ULONG) Release();

0 l6 K/ A0 i+ \  ~0 F
; [( f( z' W, o5 Z9 T, a! L2 i" k// implementation
2 a! _( Y  ?$ ^7 a. a& Y% Oprivate:
/ k5 }1 A- k" y, L* w        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
. G8 t$ r: M4 ], \$ B3 O" I        HRESULT __stdcall SearchComplete(LONG nFindData);

: j# G6 w- _+ z8 M* M  j
  Z  M: y5 n0 Q% f, d+ E& Wprivate:
9 N! ~+ i; W9 A7 ^: S! c        CUPnPImplWinServ& m_instance;
3 K* `( t2 F% Q9 I# d1 [% m3 p        LONG m_lRefCount;
};
) ~. N& Q) y% L7 g( e: L
& H# P2 x$ t. K2 H9 T$ u: v; \
// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
{
3 a- i- h, m, v  l% _7 Gpublic:
$ {# {0 o6 z( D6 w% h- D        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
  `$ Y- F: r+ h( d2 ~: O5 W" W        { m_lRefCount = 0; }
# q1 F3 m/ v0 U8 B' l- s! C! v   
8 _; i& N9 T, k& H8 d   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
. U6 Y0 w: y. `   STDMETHODIMP_(ULONG) Release();


! A7 X0 o/ F1 I) m: T. [' t// implementation
; i" ?) l$ Z5 z3 r' @# r7 m3 Qprivate:
+ ^4 q! l' m. A        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

9 B, G( x" ?; Q  m; K
private:
        CUPnPImplWinServ& m_instance;
3 M' z! t0 f1 ]7 X        LONG m_lRefCount;
" J( ^5 p) w8 w6 ]3 L6 e# l};


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

- U$ {* k; {' P1 ]- l9 {
- m6 k; o+ i" D使用时只需要使用抽象类的接口。
% K" ?! H9 C$ }7 r- |CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
8 u+ [+ m& f9 A' L( SCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.