UPnP

天云上人

/*uPnP.h*/

1. 
   5 P& b  j9 \' r7 f6 |8 M& B* r
2. #ifndef   MYUPNP_H_
   
3. 
   - I; _: b  L2 q: `% D0 Z
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   6 Z/ h. K5 r! [
7. 
   : K1 B4 A$ R& t* P
8. class   MyUPnP
   1 q7 R; A% D" h5 D. H; {6 l
9. {
   
10. public:
    8 F7 }. B/ h" ^8 b+ {. l+ N: |
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    
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
    * N! S1 h3 ?4 r5 C
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    + d  J& Y% x7 w# ]
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    0 h, t8 {4 W: Y' v. Z
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    . H3 X/ F, @& ~; o+ V" v; I0 V
29. }   UPNPNAT_MAPPING;
    
30. 
    
31. MyUPnP();
    " L- M3 o& V$ x; G, P7 e
32. ~MyUPnP();
    
33. 
    & K) m' A$ j4 r( [* J5 s& @/ W
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    - s0 f# ]. T8 J# x+ ?
37. 
      F- g2 N2 J: r7 p$ M5 J
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    5 _) d3 f# ]. D: f0 w' P* H+ ]
40. WORD GetLocalIP();
    5 e. c& p8 W! w% e; a
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
      ?1 A7 P+ I# L7 w  v
44. void InitLocalIP();
    3 J( I3 p. b6 k' y  U
45. void SetLastError(CString   error);
    
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    ( n4 z8 ?9 P( ]7 I3 ?) E
49. bool   deletePortmap(int   eport,   const   CString&   type);
    $ g- `/ V, P' e+ L4 c  L9 m6 U
50. 
    
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);
    
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    4 R$ T3 h7 k5 |
60. CString m_devicename;
    
61. CString m_name;
    6 i3 R5 b4 @* k" e" \- W; x; B* j, s
62. CString m_description;
    # y2 [) Z5 _& V$ ^( ]* F' o
63. CString m_baseurl;
    ; g: t! U" F1 z
64. CString m_controlurl;
    
65. CString m_friendlyname;
    $ W+ t) W, W1 H  j7 h
66. CString m_modelname;
    
67. int m_version;
    + A5 G7 f/ q% ]' A+ H1 ]2 B+ X" D
68. 
    / P3 e2 q" z) @) v& B' a3 o; `- j- \
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    4 Y& F5 v1 G- G5 K
72. CString m_slocalIP;
    
73. CString m_slastError;
    1 L- U  R! v1 k, p. z
74. WORD m_uLocalIP;
    $ V6 f5 D) A$ T; i  d
75. 
    1 m, U" }+ B: V+ t
76. bool isSearched;
    
77. };
    & j7 l4 ]4 I$ f8 l9 {, y& h
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   , N  L: x4 n' _
3. 
   
4. #include   "upnp.h "
   
5. 
   4 l+ ]0 _  J7 R- \. P" i- F
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   % w" N" d" Y& N# n9 ]
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   " ^6 l* N' C. K5 Z* [
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
     r7 M$ m9 Y$ d: X- z+ L- a
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    0 m; J6 O3 }/ |, p8 o- }. s
11. 
    * B) l% {4 b: r5 S; X" g
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    , Z; J1 a/ s3 P: l
15. 
    # n: Q* C6 K  }' b) c. `0 @8 H. ~
16. const   CString   getString(int   i)
    & p2 y) l! Y# i! b/ ^$ C" j* x& K6 z* Z
17. {
    
18. CString   s;
    $ `! i8 O, A9 {% b4 h
19. 
    
20. s.Format(_T( "%d "),   i);
    ) ~  h' F$ B; p, ]# Y
21. 
    
22. return   s;
    
23. }
    
24. 
    # C1 P# Z9 w; \2 d& P
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    3 y6 H! n. b7 {0 K
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    8 B9 W& y0 J9 ^/ Z! c/ {
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    ) M+ O) p7 A0 U8 k3 }
33. }
    * Z& N+ J3 i$ m) C: j# z2 R
34. 
    - A; Y  Y4 \8 n7 _( w6 \
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    $ A* Y( p) T  q- |
37. char   buffer[10240];
    ! U1 h& m4 B. B! G
38. 
    
39. const   CStringA   sa(request);
    / G4 T1 Y0 }. t% ?  f3 O/ q
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    " e# i5 k. D, l& t8 {* w% |
43. uint32   ip   =   inet_addr(CStringA(addr));
    * i( g$ i( U9 F" g! [
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ! Q1 ^0 H! x" _! C
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    5 Q* I: d+ n1 u3 }2 A, [1 w9 o+ t
53. Sleep(20);
    5 _. q4 W& x3 w
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    ) q6 a2 f: Z+ g9 i  [$ o" _$ o* a
59. if   (!rlen)   return   false;
    7 v0 N: y& K. b" g
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    % H$ J0 _- m# t
62. 
    
63. return   true;
    
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    0 W1 x  X; K% \2 a
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    4 S0 q' l" r( I- @; {0 W
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    8 H( [/ k; N; j/ e
76. sockaddr.sin_family   =   AF_INET;
    % W8 V4 W* y$ G+ V! v
77. sockaddr.sin_port   =   htons(port);
    * x6 n$ @# b$ t8 @- h6 X, j
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    , T) A) o& Z' O! ^1 l, Z
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    0 C1 h1 m6 Y5 |" Y" o% E8 E1 q% C
85. int   pos   =   0;
    3 a0 E, z2 M, |7 i, B% [
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
      R8 ?4 U# Y9 a* t4 S. {
88. 
    5 [8 S' x4 v) `5 F4 U- y- y
89. result   =   response;
    2 Y% ^& x$ G) B( }+ }- ^
90. result.Delete(0,   pos);
    
91. 
    6 j# Y# s- {" }$ T/ n+ F
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    9 W" D; Z) m! o( \/ Q
94. status   =   status.Tokenize(_T( "   "),   pos);
    5 w# {' b) \  ^7 t/ x  H7 E7 D& H
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    0 l- }$ N+ a1 W  c
96. return   true;
    , d5 T! x* U9 |( x# q/ ~
97. }
    ; l  r0 @# l, c
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     & J& S' e0 h  ~; U+ H; Q6 a% Z! k
103. CString   property;
     ; _8 p7 _& ~' J. H3 ~
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     ' Z, D4 W% Q# j( J
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     % B" [- _. W5 O8 J, p8 j7 T
112. }
     0 ?' P5 h/ A5 {: H8 g: C+ s9 r
113. 
     2 R: C$ Y# R+ p) d2 B- c
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     $ A( W& A9 ]9 {2 g9 _! @
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     7 w0 o; _7 l8 A$ U
123. UPNPNAT_MAPPING   search;
     + W1 z1 p. L+ s: P! M
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     + q5 S5 r" C( C; z0 ]0 w# M  N+ r7 ~
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     2 `, t# ?6 H  ?' a% A
129. 
     4 K3 F) [: w4 j8 Y4 P' _: `& N
130. m_Mappings.RemoveAll();
     
131. }
     . P' J: t3 V* f# o) R0 ?) _
132. 
       k& |+ F3 S. P2 g% j1 |- t
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     3 ~2 u3 Q( v+ w+ b8 g9 T
140. CString   devices[][2]   =   {
     ( P1 l8 c6 U$ v% r+ U
141. {UPNPPORTMAP1,   _T( "service ")},
     0 U+ u( N8 j0 R; i5 ~7 u
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     $ D( Z/ W4 M) ^" J
144. };
       p( z4 M5 X$ z6 m( f
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     + _! n' s) J: i5 j$ }! v
149. 
     ' X* X: U1 c% J5 J
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     9 B6 F. Y& N5 B
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);
     2 ^/ U+ D6 l2 t  K0 {$ ?
155. CString   request;
     & ?$ a% z: B$ I$ ?  M9 p/ |
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);
     ' N4 w% M, t2 y; }
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     2 p1 r4 K9 ]7 y' ]
159. }
     $ b: F% S+ r$ I$ e
160. }
     
161. 
     
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     , |. ^  P# o+ q% t2 a' ^
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     6 G0 r# f9 F" ?  \$ V
166. if   (rlen   <=   0)   continue;
     2 ~8 S( O& G2 ~+ u. a- Q
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     & y# O' P) E& B! ?- t- s
170. CString   result;
     " {& p& E# Q3 I5 @8 v
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     : E- y/ p- E' R) T9 s% g  M
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     * j) [) f8 w6 F7 l6 P
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     # B& A, D2 X( p* J- L
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     3 `0 x( S. c; \- ~, D
181. if   (name   ==   _T( "LOCATION: "))   {
     7 H0 [* i7 H& H" F
182. line.Delete(0,   9);
     " Y: N" [/ F* z( x5 _- ^$ r
183. m_description   =   line;
     5 w5 l! [- y# e4 M1 I4 k- U
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     : k& ]+ C3 h) D
187. }
     1 e6 d0 o( k5 ?1 o2 b4 h- w
188. }
     
189. }
     ; Y) v2 L" V- J. @# _/ c
190. }
     6 w, _- O5 k# @' `% i, P
191. closesocket(s);
     7 F$ {: G6 {4 D2 U; x; Y
192. 
     8 f" A3 f5 }7 o' ?$ @; A! u
193. return   false;
     
194. }
     + m- G  s( _- h' E. X' p

复制代码

天云上人

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


///////////////////////////////////////////
. B1 r4 h, _& q0 x* ]$ R//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
7 Z2 F" W7 g7 C3 T' U) Z- W
- ^1 a( E: l" B) m% E( m& [* u% x  G
( B( z7 [3 E( O. \, ^1 H8 [#pragma once
8 w: x3 Z  B& |2 `- ?#include <exception>

5 k/ u6 F& E9 {9 r9 `1 o0 q6 F
7 @9 \# w! g3 Y  ?5 @5 z  enum TRISTATE{
4 U1 w; |" d( _  `6 I: l        TRIS_FALSE,
4 ]* d2 I/ w% s. }, v8 H4 C0 v' }        TRIS_UNKNOWN,
5 _( }( s4 N! }5 [1 ]0 Q        TRIS_TRUE
; j; z+ j8 |; _; \! u};
9 A( v  k- k; _
8 n/ \6 e, g  O5 \! x% j
enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
5 n2 l2 y5 _* P. i! \. z( P& `        UPNP_IMPL_NONE /*last*/
};

+ v7 V( q, h3 r7 w4 U

( v. {" j3 w3 u' ?3 f
; S0 |& G: U- @  h( K) `/ h8 mclass CUPnPImpl
2 L5 K! x; K" L{
9 ~8 c. K* w8 O: Fpublic:
        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
2 M2 M( R/ X0 l; ^$ N                UPNP_FAILED,
                UPNP_TIMEOUT
        };

$ I; L" E/ E8 p7 T- l
, E5 \% ^% d$ G6 b, P9 w        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
3 Y( C& b! d+ p- h7 z        virtual bool        CheckAndRefresh() = 0;
4 L! ~5 v: c' s; R7 h8 i" I        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
' r1 J+ J! R! |  G0 n        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
/ ~/ s: R# u) a$ Y* C        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
7 P$ w9 u5 r  r7 P' P3 O- e5 w

        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
: o8 [: N' T6 D6 f( t' B        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
+ |, V6 b# T' z: ^+ T2 r2 h, P

: [  `  ?4 |1 Z// Implementation
8 z' Z) s2 N5 ^! K6 Sprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
9 _! p3 s" H. V. r! S# q        void                                SendResultMessage();
. I2 ~4 |4 [2 }3 H9 G        uint16                                m_nUDPPort;
& E8 N7 r& l* g' v3 e( P        uint16                                m_nTCPPort;
6 [8 |/ E4 W  p4 b) d; g        uint16                                m_nTCPWebPort;
- a9 J! Y3 Q& A" B; ^. m% \        bool                                m_bCheckAndRefresh;

; B+ o6 q& A+ C* U6 G( t3 L/ H4 I/ q1 w
/ v' f% n, I8 x& z1 T, Z8 K3 Hprivate:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;

4 |3 Q# g$ [; u# H& b+ A
+ a5 u9 |/ R# z: ~};

- q  \0 X2 J, S4 \+ y
// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
: ~) e6 y% D; q4 X! S{
public:
6 x" U$ t( U% {        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
4 P1 n/ X: f3 b' n        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
) l; [7 [6 L. a        virtual void        DeletePorts()                                                                                        { }
& ?2 R  y. \' r1 X$ L        virtual bool        IsReady()                                                                                                { return false; }
- N, e1 V: @: M& I( N4 A        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
8 b" q" \2 V+ m) ^4 M5 N$ J};
- ^$ u! j" t0 Q$ L+ I( w
) c3 H4 f" V: A. V  [
/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
) o2 S: k) X+ W  s2 W+ F" B
# m# @5 H% I3 s6 [  }! ^
* ^& [- i: P4 U9 I1 C#pragma once
- P" J/ g6 j9 E3 F( g#pragma warning( disable: 4355 )


#include "UPnPImpl.h"
" _  p6 W, b7 i) J. w+ h$ K#include <upnp.h>
#include <iphlpapi.h>
# Q0 R  a- D0 P; I! l9 l#include <comdef.h>
( H; t- H0 s2 k+ Q#include <winsvc.h>


#include <vector>
% V4 p  ^" \& N#include <exception>
#include <functional>
& M' y9 I& e3 @

- J7 \! }' ?5 t1 r3 n0 K( A

2 D2 h# L3 M/ f% c, `2 Z0 Z) p7 Vtypedef _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;
typedef        _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;


typedef DWORD (WINAPI* TGetBestInterface) (
$ \1 w: r- D4 {# W8 V. n) m  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);
8 F* |. F, Y, f4 d( M: v+ H

  ?2 Y/ `) H1 stypedef DWORD (WINAPI* TGetIpAddrTable) (
) F) g" v+ T* U  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
9 J2 f6 J+ l5 z* ^7 s, a);
; c/ T6 `' k9 ]7 @5 m

typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
! b: r1 X, u- p2 ]);

% A9 o' R: [: g+ {
CString translateUPnPResult(HRESULT hr);
( L6 l/ {; y: f2 x& GHRESULT UPnPMessage(HRESULT hr);
* m9 g2 O% w/ L3 [( g7 R

9 H- X# i+ u$ ~( n+ l* rclass CUPnPImplWinServ: public CUPnPImpl
6 _$ w: @0 U3 M" n' \# t* A7 B2 @3 }{
: Z) V5 J2 r% ^        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
/ f6 E- C8 E5 }: ]8 u9 n# e) N// Construction
public:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();

% o* I7 [! s) H2 p& a
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
1 b" h* `8 {+ e/ P        virtual void        StopAsyncFind();
) z& w6 Q# t: ~7 M; L        virtual void        DeletePorts();
        virtual bool        IsReady();
" Y9 k2 v$ K9 S8 V        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
; o; U) n1 }  g, W- {; S$ n+ r
1 g# y. i9 B% @+ \  N4 \
        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
9 ]/ Q& a/ n2 e9 P3 v" N4 g        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
7 a2 z& x. X5 ]
: P/ y& [' F! B
protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
& |5 _* S% O& p  E$ Q        void        RemoveDevice(CComBSTR bsUDN);
) Z4 O( T; v% m* m        bool        OnSearchComplete();
! d" _4 U$ V$ q* n        void        Init();
6 Q  n& k! J3 |' R% J5 n! `7 Q, S

! P( a  V( o* I        inline bool IsAsyncFindRunning()
- C5 i& p3 Q& G& X        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
  v* t- k+ s+ v                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
" p. H* t7 Y# T$ Z                        m_bAsyncFindRunning = false;
                }
                MSG msg;
7 k. F/ B$ D5 y8 l* x' J                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
: p+ J$ Z5 P& R+ K5 K                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
3 X+ I' A2 T* \! v1 |8 ?, q                return m_bAsyncFindRunning;
        }
1 y' }" ^/ n. O5 }
2 }0 \. z, U" N  @
        TRISTATE                        m_bUPnPDeviceConnected;

. Y# i' ?; N( h$ [4 r+ J# |
// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
" N% L7 L( X; F" I5 A5 g, k6 ~        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);
5 Z, F) |' M, \2 I' y        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
. d7 z6 S* l+ u) q; }* i! c

+ r7 U  S- {5 t' I' h  I        TGetBestInterface                m_pfGetBestInterface;
$ V- ^# d1 U, x+ O" E$ x8 Q+ q        TGetIpAddrTable                        m_pfGetIpAddrTable;
) m  t/ {3 E" U  K5 E; E( A        TGetIfEntry                                m_pfGetIfEntry;
" d% Y% J5 N* R. f4 x4 o
* m: M4 U8 r; N: C, _
        static FinderPointer CreateFinderInstance();
! I2 I$ P( G& m% J, K6 g7 |3 [        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
" w; g, i! F+ f4 w& }1 s                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
# L$ C, |( p4 c9 C0 @0 N                {
& u* t; [1 U7 a% q6 J: o                        CComBSTR deviceName;
5 j/ z( n8 ?8 m  \  _. y$ f                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
, i- c0 z6 q3 _
* _: v0 B  D, W6 G- e
                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;

" `8 |& d* V% r
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
8 l' U) [4 H5 j+ l& }& b4 n# M                CComBSTR m_udn;
        };
       
; p; \4 O9 h' L& @/ J9 ?        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
4 e  F' E% B0 P0 U7 Y2 O        void        StartPortMapping();
: S. X/ P. Y2 K9 K6 q  Z' I        HRESULT        MapPort(const ServicePointer& service);
6 c# N- [+ J' ?6 _# m% b4 r        void        DeleteExistingPortMappings(ServicePointer pService);
8 n' g8 D% B% a& h* i& x% Q" H& b        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
# s- O! w% N* E        void        StopUPnPService();
, @9 \- m$ W9 C3 _( o. e( W2 y

        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
1 j5 J0 y% M% u% x: Q/ t9 l+ V        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
) E9 w- O4 x) i8 G        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
# l  G" X4 G) A& F1 I. V' ]        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
! ]. P" G: h3 T' W# S# V: Z- j        CString        GetLocalRoutableIP(ServicePointer pService);
' T3 I) V( c# X. K! Q* Q
/ v5 c8 l- f  G7 K( o. j
. [4 q  _: _0 K! |: @// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;


        LONG        m_nAsyncFindHandle;
9 A1 o! B- R, I# ]        bool        m_bCOM;
7 P+ T' s6 Y1 _* a        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
1 O7 \' {4 B* A/ M- b1 B        bool        m_bADSL;                // Is the device ADSL?
2 }( r9 R: Q7 O2 a6 I$ T        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
5 T! I- z+ a& N, B- d        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
% v% H; ^+ T" C  u5 W7 k        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
/ p, P* n  Z4 h0 k        bool        m_bServiceStartedByEmule;
2 K0 K/ ^3 c) [        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
1 @) q% S1 J' \5 K2 W$ i
' [/ F9 B8 s% J' S7 i9 S2 d& }
8 n" v: G( }( V. I% [" u, Q};
& z6 k1 Z& }; s" E: g- z* R: C

4 _) u% g$ P# K* p// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
& M# o: W( ^) q3 z9 d0 r* @  ]* r" [{
# o& W( d+ |$ N5 |' f3 Gpublic:
6 i3 G9 ]9 }: c7 I: ?; k/ H' X  J        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
( W! ~! e8 V& l+ O
/ f& |$ {; w. m" L+ p
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
7 V& `8 i4 g* ~! F   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
9 R1 l0 u6 u( ~
1 s' Z# V0 r9 b' k- h
// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
" d1 @, A- B' D, z0 z        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
0 a; u/ D1 X6 L0 }$ F9 ^9 x5 f- o        HRESULT __stdcall SearchComplete(LONG nFindData);


private:
6 Z$ e1 [/ M0 A( I  V1 C# A        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
* Z+ N% D8 C0 M1 D! u

% S8 M6 Y! D+ G) }9 `// Service Callback
class CServiceCallback
; B8 ?. Q, l2 S2 J6 i: j        : public IUPnPServiceCallback
/ I( S1 G2 z1 R% K& S0 {{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
9 x- i% e. e% \: S        { m_lRefCount = 0; }
! z3 r7 M1 m/ P- n1 P3 }; \   
8 r- z9 ^& }- ?  A0 w   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
  ]. \4 g  C7 [   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
  v* T& S$ Q1 s- ?6 V

// implementation
" m, M; N4 h5 C# g) n% L8 iprivate:
+ \4 f/ u) \3 i- G. r0 _        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


' U% J/ k% q% Rprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
8 ]9 e  b) t8 x. a
( l- r8 f4 R/ Z" s. g! d+ F
7 @& j4 q9 q- j3 }/ N9 l/////////////////////////////////////////////////


, X; I! r+ r% M; p# A使用时只需要使用抽象类的接口。
2 p( i" I4 K/ R! V0 KCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
/ r* X7 I0 }  s) |* RCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
- o+ c3 K% ~$ ]& O1 |( R5 LCUPnPImpl::StopAsyncFind停止设备查找.
- R& V7 s' B6 W7 M! W' [7 BCUPnPImpl::DeletePorts删除端口映射.