UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   / T& e' h& W6 N) ^
4. #pragma   once
   
5. 
   2 D* q7 A, Z  M
6. typedef   unsigned   long   ulong;
   
7. 
   / v3 p" I! r, [2 w4 [. s
8. class   MyUPnP
   
9. {
   7 d' D# \: _- ?; \! H' ^1 W) K
10. public:
    ; z9 f2 v7 S: i+ N+ s
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    % }% n- j" |# n
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
    + L' o( W) h# f( e9 v
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
    , M9 G. c& E/ s$ @
17. }   UPNPNAT_RETURN;
    + l: |) @3 F% W" v; U
18. 
    
19. typedef   enum{
    3 ?2 G3 ~' e2 D8 K
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    ' Z2 ?$ R+ `* k! k
23. 
    ; R4 o9 U3 ?+ u( r* R( o6 f" U# z
24. typedef   struct{
    " A5 c: c+ L( w# z
25. WORD   internalPort; //   Port   mapping   internal   port
    - n: F$ B: j# p8 X2 U+ l& ?
26. WORD   externalPort; //   Port   mapping   external   port
    0 U9 H" t  e, C4 d( Q
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    8 _' `7 o' o3 N
28. CString   description; //   Port   mapping   description
    0 B! X5 O6 A9 `" _1 g% P
29. }   UPNPNAT_MAPPING;
    . z' ]$ I: j) e3 h3 {- a' p
30. 
    
31. MyUPnP();
    , e3 t: u' c; L
32. ~MyUPnP();
    
33. 
    " p3 K1 ~& T- _* |8 f
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    9 v0 K' p8 U( C+ s( `; E: w& E
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    $ p3 U* b% v& h
36. void   clearNATPortMapping();
    9 O; e6 f6 Z4 b0 F
37. 
    
38. CString GetLastError();
    7 l5 j1 d! z, j6 v$ r6 H- O8 Y. M
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    + F0 E1 I# _5 a& D- _' n
41. bool IsLANIP(WORD   nIP);
    / S& g4 F2 d" O3 `3 O+ h  i2 M
42. 
    ( T, p1 ^7 ^) D, ]: P0 \
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    1 l- q! ], b1 {& t
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    & F5 {& g" `" g8 I
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    , g% G7 ]  \" _9 x0 A& D2 U
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    / q- Q! O- K' @* I& Z$ i$ F/ f
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    9 [& b1 x; L% _: W
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    # o6 p2 h) T( l9 L+ k$ i3 E
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    ) K" P2 h  d3 G6 Q
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    ( C$ ]3 {' A0 P; u& |
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    % n  I2 U2 `" c  b2 D# @
74. WORD m_uLocalIP;
    : z; w( R! }, ^3 ~, f
75. 
    
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   % i: x3 ~1 |8 c4 s- m( n9 Q% a
3. 
   
4. #include   "upnp.h "
   7 S; f. t' ]  `0 X( _4 {
5. 
   ; _7 x+ s* a" E0 L/ y' H
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
     k* I( p5 T( \. q
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   / d2 S1 c5 v0 P; s* {; _4 ^- o
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   8 c. a; t3 d  d/ T3 }8 S' W
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    & Q8 z3 y" T- B6 y- Z3 |6 Q5 @
11. 
    % j  m7 t% U, O: n
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    : z4 c- J/ B% {+ L4 J' G
13. static   const   int UPNPPORT   =   1900;
    " f2 l: R; q) }+ J& Q2 {6 Y3 @
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    9 j  a; q  u& k1 M/ B) {
19. 
    1 |5 z# o8 C7 A( R6 k7 ^5 S' {
20. s.Format(_T( "%d "),   i);
    $ I. N; H3 }0 d5 C) l" ]: _7 L
21. 
    
22. return   s;
    7 e, _; _' X6 o
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    4 J4 d) o# L) l, Y' n+ m( S. J0 \
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    2 f5 q6 L5 k8 Q- C; \: ]
28. }
    $ _7 w& G' h! z9 K4 ]3 [
29. 
    " `/ y' `+ _, |' Q! v8 Y9 x
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    1 ]( n- Q6 N. A/ U
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    7 b$ @* U& a, d! J0 q
41. strcpy(buffer,   (const   char*)sa);
    ) {; L: d! U" P" @
42. 
    0 X; f$ J+ b/ X, M
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    % b3 Y  Z  l) w9 W/ J2 \" Z+ j
46. sockaddr.sin_family   =   AF_INET;
    8 D' i2 W) _6 K% B
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    7 w/ j4 `  q8 c8 Y, C  \8 Q1 e+ z! y% g
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    5 ?- f: y6 j3 }
50. u_long   lv   =   1;
    ( f8 X" E& I1 l$ D5 }& `
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    5 I5 R- I0 e: w
53. Sleep(20);
    7 ?+ |* S" G& e: y) _, f$ I# [
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    & g7 A5 E4 Q. y" ~5 s6 F
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    & ?/ x& q. B. z
57. closesocket(s);
    ) {8 N& w: h- G" q/ J6 v- j
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    " o& D( y) H7 S4 X
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    $ f7 F5 C. ]% x$ d8 V9 N: P( n: s4 ~
64. }
    
65. 
    ; F" l9 k$ i+ x- @7 f
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    9 T8 r3 ?3 w/ _, }0 W) W
68. char   buffer[10240];
    & R0 R9 J7 z3 ?% g( Q( j
69. 
    ' l# o2 r$ ~3 t5 h& ]1 D/ l' M
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    ( ~$ _: f- b, Q: `4 Y
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    2 j7 F, c* c4 G' M
77. sockaddr.sin_port   =   htons(port);
    * `0 G1 f9 Z& {7 c6 [  e2 b% S, ]/ _
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    + l7 g3 e4 A, a( }1 H
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    1 q- t; Y' E% B% K" A+ n* N" {
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    0 C2 l& Q5 B0 P7 X
84. {
    - G9 O* }; A" R
85. int   pos   =   0;
      o' O9 X0 c( c$ a/ z* ^
86. 
    " Q+ q4 Y% P* [
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    - S' w6 ~) T: h5 V* {
91. 
    , p' Z& _5 T/ A/ W: x! {
92. pos   =   0;
    & B+ b+ b+ c- J& g% O8 n% m# d
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    & C$ b! Y) k1 P7 s8 p5 a$ i
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    ' }, e# h1 @8 l; |8 m% ?/ k
98. 
    * x8 H1 _9 p! _2 i: U: k
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    + K& l1 H: m8 h: R+ `5 M/ S/ ^2 B
100. {
     & l4 |6 M7 c" p
101. CString   startTag   =   ' < '   +   name   +   '> ';
     : Z% P- ^+ ^$ k/ G
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     * w; ~+ O- Y) d: X# ?
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     $ @* M- Z' t  ]9 L5 X
113. 
     : y% {* A# b" K: n
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     ( Y2 F7 D* O+ y2 |1 I/ H' J4 d8 e, L5 V
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     - c7 Z! i  r# r
119. }
     , M) I: W, f) n8 i- G! T: Q$ v3 ~
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     , v0 Z5 O" |8 W" w  r) q
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     9 T$ B* s! Z; ]( F$ F3 J3 \
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     % Q; Z6 E  f7 p0 j
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     1 d6 w7 f8 h6 l% p* V$ J9 K( J
130. m_Mappings.RemoveAll();
     2 B  ?9 d' v' s  S7 G
131. }
     - Y: b/ e2 Z* F" o
132. 
     
133. 
     6 z  q' @4 X- c1 {; w1 ^
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     + t6 Z& c6 l3 d; {' L7 e7 B
137. m_version   =   version;
     ( T2 Q! R' a  Y: E
138. 
     
139. #define   NUMBEROFDEVICES 2
     8 u0 E9 Q3 t+ P% s
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     " F! R9 x( T) `- h# }+ I) Q6 q; t
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     " r" h# O' w1 Z* d( m
149. 
     - q+ @4 F; m2 u# w
150. int   rlen   =   0;
     " I1 R, }' P6 ?3 P
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     
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);
     
155. CString   request;
     
156. request.Format(_T( "M-SEARCH   *   HTTP/1.1\r\nHOST:   239.255.255.250:1900\r\nMAN:   \ "ssdp:discover\ "\r\nMX:   %d\r\nST:   %s\r\n\r\n "),
       s5 m+ e! }" Q5 t* Z) Q
157. 6,   m_name);
     % R5 A. N# [! K2 V
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     
161. 
     
162. Sleep(10);
     4 \# g$ X1 \9 v3 r4 \/ x
163. 
     
164. char   buffer[10240];
     " F6 k3 Z3 X6 T+ H: m8 s
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     ! E$ U1 J9 \# a% ^
167. closesocket(s);
     ' S9 Q# _9 w) U) v
168. 
     7 B+ U% x# E. h, B. r- A* |
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     * [2 G& b, n7 C4 L% K
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     & v" d5 g8 t& O
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     : P+ T' E4 q% _& N$ G
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     0 d4 n6 N  X/ O, Y: R! n
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     + L9 D5 L* \7 M
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     ) [/ ?$ \& ?: G- P
180. name.MakeUpper();
     ! }7 d( R' `8 X, q9 w3 \$ p7 D/ M" w/ x- o8 d
181. if   (name   ==   _T( "LOCATION: "))   {
     4 ?( Q$ P1 a! g
182. line.Delete(0,   9);
     $ U9 i+ D' @2 I3 I9 ]% I" Z! ~9 I
183. m_description   =   line;
     , A3 A2 C& p% Z5 l
184. m_description.Trim();
     
185. return   GetDescription();
     & a/ _; q5 T! D# O* d
186. }
     
187. }
     ! v8 }+ O+ u- w3 G7 o  ~
188. }
     6 ^5 L0 o3 ]- r, ~& `7 Q" h
189. }
     
190. }
     
191. closesocket(s);
     
192. 
       B! ^; K1 o9 ]1 |: E: S1 F
193. return   false;
     3 [3 m$ u$ F" m7 R% t/ y2 ^# v- w3 w
194. }
     7 y, M# ?6 U, u( }; F/ Q* b

复制代码

天云上人

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

# l4 h5 k. m9 R. S1 t5 U
///////////////////////////////////////////
3 S6 J- w* |  t* M& J7 \8 A; P" [//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
9 K& f1 M! {  h/ a- ]3 d6 e% z

#pragma once
+ T, @2 |) N3 J#include <exception>

  t$ \$ U2 F- a' i7 `' ]: t
; l! ]! j% Z0 m3 d, n/ D' ]  enum TRISTATE{
        TRIS_FALSE,
; O( E# A0 R" Y& f        TRIS_UNKNOWN,
! L" z' L9 _9 w" [4 y; I9 k" d        TRIS_TRUE
};


enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};
. }6 E; c9 X) _2 J0 E3 k
: m4 K5 s, i; C) K6 t; q
( X6 [4 X( q! k8 u
3 \3 \" h9 C' v4 h: e
9 r  G' q& N$ `; d1 Hclass CUPnPImpl
: Y) x6 D' x1 @, e: C2 H{
, t% i4 G) u7 j6 Fpublic:
        CUPnPImpl();
        virtual ~CUPnPImpl();
2 h9 n. J# N# V& P3 x  n        struct UPnPError : std::exception {};
        enum {
- h0 b+ ?- c( N. H" V, @, q9 _                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
        };

6 c. F. t% B, R/ e
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
- v4 l) ~5 C3 I        virtual void        StopAsyncFind() = 0;
9 D3 O$ u9 W+ k        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
; y1 [- O# ?( n' `4 r        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

; o$ p( Y% \" i# Z1 A* u! A+ ]
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
! M& S* m$ g9 |2 v5 L0 o        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


9 i6 }$ m+ M5 R# q% g/ [% P// Implementation
  z% l$ D: e9 N5 \: W8 u! Zprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
* N5 [$ \, `2 A8 I9 j5 ~        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
3 ?0 R6 ]6 r) \+ j        uint16                                m_nTCPWebPort;
# E0 M! t! V5 t, d9 o: @! p        bool                                m_bCheckAndRefresh;
9 V! S0 w8 r/ [1 y0 `

private:
' ^& U2 \$ |  Y  e        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
2 }- w5 z+ k5 X& F! Z7 D% S6 \

};


// Dummy Implementation to be used when no other implementation is available
( N! O4 D/ B0 D+ A- mclass CUPnPImplNone: public CUPnPImpl
7 H; r- V) Y9 |! }( D  G! U{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
9 z' ^3 |5 s9 v9 L5 q1 p; `7 i7 {* d        virtual void        DeletePorts()                                                                                        { }
' ]9 m6 z9 V4 _        virtual bool        IsReady()                                                                                                { return false; }
5 Q; Y2 d! g# j# I: h  z- o4 r' B9 ?. F        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
+ V$ o2 f" k% n};

  ]+ g6 C% v0 B8 ~, }/ {
: F0 ?% f: r* R" m% x0 n* t/////////////////////////////////////
( Y% H- V3 n4 U6 M5 g; I  u//下面是使用windows操作系统自带的UPNP功能的子类
; m* }; ^0 F, q% x1 f, S( l% I6 r. L
* f, z" ~$ e: w
#pragma once
#pragma warning( disable: 4355 )
' s' P1 E/ Q* Q7 j8 w' p' Z
8 H4 \, Z- g' s& n- u' b" x
* U& \* i! H, P5 g) g#include "UPnPImpl.h"
9 Y; m* n- A3 h/ J7 c" W7 S% u#include <upnp.h>
3 i& [6 W+ ^( t& z& Q7 \#include <iphlpapi.h>
% [) M4 M" H6 P# J#include <comdef.h>
( |  V: l! D: z#include <winsvc.h>


#include <vector>
#include <exception>
#include <functional>
0 S" Z- ^! m; K$ [4 P; B+ c. k. _/ a
. k1 v0 i5 w. M( w, ?, w6 Y8 P4 H/ u

9 V1 ]8 _8 D( E8 G0 n# h/ w
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
8 I) L; V' ?1 a( D7 J, Btypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
# K* T5 F( Q* z# Y8 vtypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
0 ^( G# p% u& @3 g

typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
3 Z3 w7 {* G7 ?9 {);
1 v6 P# t2 E4 k( J
+ H& T) l9 j9 j3 f
5 ]/ {3 M7 J0 j( j! gtypedef DWORD (WINAPI* TGetIpAddrTable) (
5 R5 e& ?: y7 \$ c. y2 f- M  PMIB_IPADDRTABLE pIpAddrTable,
4 u/ `; g0 J0 r* O1 i7 e  PULONG pdwSize,
2 h  |9 p5 x; C  BOOL bOrder
);

: X2 I, q/ u% ~7 ^% {! B  l& v
typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);
/ H( U7 R! V4 U9 U1 n/ }3 ?# N+ U

CString translateUPnPResult(HRESULT hr);
  r5 `  C( }. l1 l  j  I: h. yHRESULT UPnPMessage(HRESULT hr);


class CUPnPImplWinServ: public CUPnPImpl
{
$ K8 Z1 ~0 h5 [% B/ ]        friend class CDeviceFinderCallback;
$ r( t& `. n$ A. y        friend class CServiceCallback;
& G* E$ I4 a2 ]' A, r// Construction
public:
$ e5 x# ]0 T* E  c: ?6 G! y        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
7 B2 p  @8 ?4 e8 Q1 j! r! p. R

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
! W3 k3 @7 m7 x6 v; l        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };


1 `  C4 Y/ _% T/ w+ B) p& y9 T6 t$ Fprotected:
: k8 u/ o, P0 H3 C. d8 u, w        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
  ^- I% C( L! c, v) U3 [& J        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
8 O6 d1 O9 Z* c5 s        void        Init();
) s# d6 s- l: X, _

        inline bool IsAsyncFindRunning()
        {
- ]) [" n& s- |0 ~8 a( I                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
6 z# d! U: \5 R3 b                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
/ s0 v- H; i$ i- Q                MSG msg;
9 o; J$ C' u$ G! }4 `                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
" D* O% W) O, @% J$ \2 x, ~7 D                {
5 ]0 o, U( P3 F6 T1 m( _, }% ]                        TranslateMessage( &msg );
' P! y" S, N, ~" V1 m5 |9 t7 I                        DispatchMessage( &msg );
+ g1 y  q7 n8 S1 j3 [" b: V$ x0 ~, J                }
8 B$ d/ c) Z1 }$ y- j                return m_bAsyncFindRunning;
1 F  b3 [. F7 y3 a7 M        }
: q+ {0 u0 }9 k! I' ?
" J# D/ \) [% e+ {' W
        TRISTATE                        m_bUPnPDeviceConnected;
5 t1 b& {* J: T0 \5 O
4 f! |6 c: `6 _  @- \  v+ O
// Implementation
        // API functions
) D% h( C2 D  b7 T        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
7 ~$ G, P" g7 R+ w        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
: l. G+ H+ Z+ a1 w: ]1 p        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
1 _& o8 a8 w% Z) o- D: ?        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
6 N  _- _* p5 K, v        TGetIfEntry                                m_pfGetIfEntry;
0 A1 P' ]% q, @8 _" @6 Z6 D7 N
. P! q  t  ]. H) z- D7 Z
        static FinderPointer CreateFinderInstance();
: C) c2 {5 ^9 V* V4 ]$ V        struct FindDevice : std::unary_function< DevicePointer, bool >
0 f- C( {. Z+ _8 D# g! r- `        {
0 u3 I. z% n' I+ f( Q; U                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
& R2 `; G# R) E, ]                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

% T* ~, S# ^# a) q6 \' f
                        if ( FAILED( hr ) )
# P, `2 b) O  ~9 h% U% I                                return UPnPMessage( hr ), false;
5 F9 `% g8 g9 f3 b# F1 P& Z8 D; O
  x, d% y% J3 P1 E3 ^
. c+ \& Y# b  n( S% D                        return wcscmp( deviceName.m_str, m_udn ) == 0;
4 V; g! u) k5 ^! M5 \# k) y                }
                CComBSTR m_udn;
7 k0 ]7 m# k' Y2 C$ ]8 e        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
9 J% a: G3 ~( j# e7 [# ?5 N        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
+ c+ Q8 `8 n' _: g% F+ f7 H3 s3 o        void        StopUPnPService();


        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
( w3 O: n7 D/ v, Y1 S' D! w        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
( l- X, V/ t# ^  l! N, L$ C5 @        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
: \9 _2 j3 v; ^- E. J2 u        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
" S3 Z5 t9 P& {+ H2 P        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);


+ a  s" g5 U% s, M, R// Private members
, l5 g7 _0 Y6 y8 G, V) [! yprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
) q6 v  V2 H6 v5 r8 u, K7 `        std::vector< DevicePointer >  m_pDevices;
6 B$ ~# E5 W, _* K7 L6 W        std::vector< ServicePointer > m_pServices;
( c- ?' h8 z+ q; ?: w' Y        FinderPointer                        m_pDeviceFinder;
: z; A  O9 q; O$ x        DeviceFinderCallback        m_pDeviceFinderCallback;
& F: C# f  v# B; A        ServiceCallback                        m_pServiceCallback;
5 C! z9 [. T7 G; s
8 \1 a% A, a2 e, |, q( g/ T( l
        LONG        m_nAsyncFindHandle;
. F  |+ U! B. B& ?6 j3 s; c        bool        m_bCOM;
) p' C# v, d- C; O        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
  @3 X# m( z' v2 U& [! k! z3 T1 N        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
& Q  @" w9 C2 M! l4 D  K5 h  g2 ^4 z        bool        m_bAsyncFindRunning;
9 B6 a8 N- W! G6 M) \        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
! u" V- u6 t6 [8 C% o1 U        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
6 v7 K0 ~9 u# Q% S8 R        bool        m_bDisableWANPPPSetup;
* E& C5 v9 C* V5 S+ ?: c2 z

};


// DeviceFinder Callback
$ O) I/ o' g! {class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
0 g  N- r) G0 h' Y4 C/ z; m                : m_instance( instance )
        { m_lRefCount = 0; }
0 ?/ t5 ?+ ~) E3 Y' a: q

, E! z9 j! \, S9 v   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
. P! B0 D0 R) j/ a2 G$ t9 V$ ?$ L   STDMETHODIMP_(ULONG) Release();
& ^' U0 B: O5 k' j% e

. J! w  K6 P7 c+ ~/ z1 h7 r// implementation
1 P, q1 ]8 G! ~' _  Hprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
4 L, `* S( f7 O: w; ]. n: P: W        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);


0 B% @8 {7 j/ ~& }0 |% Dprivate:
+ ~+ w- ^$ p3 L# k/ \        CUPnPImplWinServ& m_instance;
1 U. \3 |, L9 |. U        LONG m_lRefCount;
};


) @* a* t& R5 p4 \0 h8 g* E( k' v// Service Callback
; _! X6 H. h/ Q2 A$ Aclass CServiceCallback
& l$ k1 x6 u5 S( {9 y2 J: ?        : public IUPnPServiceCallback
$ P6 A/ k$ v0 \. y5 o" W* J) E{
' ]+ J& q8 S6 n4 I4 v) Rpublic:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
9 o. p* M7 |. c$ @        { m_lRefCount = 0; }
- x& W; a- Z' y- d# i   
7 M  J5 ]$ r1 r& f8 `- y6 r; X   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
9 E/ K3 Y& v. I5 {1 U   STDMETHODIMP_(ULONG) AddRef();
5 _9 x* w5 s2 ~( b4 H6 x" v   STDMETHODIMP_(ULONG) Release();


9 m& F) d) |# Z% }$ a4 h0 y// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


% a% c8 U( I/ w' bprivate:
  }$ |& P/ P" G! S$ B! i1 k        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
7 C; L. G" v% y3 u) Q};
5 L1 ^# s1 o+ }6 O1 `1 F1 K

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

- s: Y* C$ [4 M1 d; c# K6 h
5 [2 a  s% ~& [$ }' Y使用时只需要使用抽象类的接口。
2 i; J# X8 K( N0 |! n7 {4 K+ TCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.