UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   * b6 {6 D- k1 Z2 u( _6 W7 I
3. 
   
4. #pragma   once
   
5. 
   ( ]2 M: n/ L+ x+ I7 \& K/ ^
6. typedef   unsigned   long   ulong;
   1 K  E7 u+ u3 i" j/ |- J
7. 
   
8. class   MyUPnP
   $ G# B5 |1 G* c+ v- ^$ d
9. {
   
10. public:
    3 \" _$ V) f/ a) o) D
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
    3 w: X$ Q+ |9 h! x! W7 D" A+ r2 ]
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    " e! ?6 n5 q- l/ N
19. typedef   enum{
    ) `" T# x3 a) r5 u7 Y& d4 G
20. UNAT_TCP, //   TCP   Protocol
    6 u8 A0 ^% P( q9 F5 W: k' V. ^: m
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    . y. M5 A, O. `& H  o. u" H
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
    , N/ F: N* n2 `: o
29. }   UPNPNAT_MAPPING;
    , ?. k4 [6 x' P$ z) d
30. 
    
31. MyUPnP();
    . `) i$ {- }% [
32. ~MyUPnP();
    / X1 U+ Z6 }: y4 U9 X/ a
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    ! m, f6 L1 x; N" I; U. A
36. void   clearNATPortMapping();
    $ K" d4 L1 S* K: o8 l
37. 
    
38. CString GetLastError();
    / P: |" @* f# h. _7 N" h6 G% C: ^
39. CString GetLocalIPStr();
    # }/ b6 n  p+ G8 c2 ?
40. WORD GetLocalIP();
    1 ^( [8 O$ |! \5 [
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    
44. void InitLocalIP();
    / n+ E+ w! ~: `; ?% f) J2 [* \8 W5 i% R
45. void SetLastError(CString   error);
    
46. 
      }, \% ^8 Q! j" c8 E
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    0 E0 Y/ R) _/ Z! p7 d9 ^
49. bool   deletePortmap(int   eport,   const   CString&   type);
    ; e) [' X! O& ]) v) {' r1 z
50. 
    ) u# X; h" @6 U
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    - ?" {1 R# `( V8 [& `
52. 
    
53. bool Search(int   version=1);
    
54. bool GetDescription();
      y" v, R% l! [, |8 k0 M% @: Z( U
55. CString GetProperty(const   CString&   name,   CString&   response);
    + s9 ]  G& t' U
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    ( Q, o  F9 ~5 E+ z: H
57. 
    ( U' [, I5 s/ q. T
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    : \. J4 c7 |% W, t9 m
63. CString m_baseurl;
    
64. CString m_controlurl;
    % f" a8 Z" M8 m# a5 x) Q3 a
65. CString m_friendlyname;
    " O) J# Y% |8 ^) E8 R
66. CString m_modelname;
    & x( ^. J! K3 l' \) ^- ~
67. int m_version;
    & R0 t+ X0 @+ d' G% ?
68. 
    : L0 D& r$ C# P
69. private:
    $ w% l/ c0 r) H6 ~7 |
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    & P0 v$ D& g8 q7 ~$ _
71. 
    
72. CString m_slocalIP;
    0 ~, Y, |' s3 O  j6 T6 H/ B5 M
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    & Q8 F/ I+ O/ A
75. 
    9 Q: Q4 A: z* |/ J/ v+ R
76. bool isSearched;
    1 h" c, m: U. N4 @
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   # l; n# o' N. {1 g' w, E
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   % |& h" _5 ?4 X! p) p
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    1 [2 `. f& D% Y3 R8 N4 `2 h9 z
11. 
    ; x' u! n$ u" M- l2 f& @/ Z3 _7 M
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
      M/ u4 C/ v8 i5 B3 z; V( @4 |/ ]2 [
13. static   const   int UPNPPORT   =   1900;
    , m8 l5 I6 R- B/ H
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    + A9 x: P9 \& {3 N! R
15. 
    
16. const   CString   getString(int   i)
    
17. {
    . D- p4 v, @+ W4 Y
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    9 J% l! `" w( B! i9 v
21. 
    
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. }
    5 g3 z$ Z5 Z. j. s8 h1 S" [" j
29. 
    ' u( m: ~$ Q! M- y3 T. W5 f4 p# D
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    . k% K7 U% @) K8 Y
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    / Y7 w! F" F6 W7 ^. ]+ A9 G1 Q: `
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    4 k% S  {1 |: M1 U/ Y" E
42. 
    . J# D+ I' b6 d6 J& _7 g
43. uint32   ip   =   inet_addr(CStringA(addr));
    & w$ l' G6 \, |# Q% p
44. struct   sockaddr_in   sockaddr;
    8 }/ w- g' R7 @$ a
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    - R9 x7 }" F5 H2 k9 P
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    ! Y+ v  R# c, Y0 L1 C2 q
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    " K- u3 X  @; ^; z3 s( L
54. int   n   =   send(s,   buffer,   length,   0);
    . E  u% v" e: ?
55. Sleep(100);
    $ s, }( ]& N7 N
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    " T4 g& a- W4 G" `- Q- b/ A+ c
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    7 J9 b; L0 ^2 i# C  C! H
60. 
    9 p$ X1 J; r& x0 M- ]
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    
64. }
    8 a6 ~. {8 c2 N& Q- ?! W; ?
65. 
    3 A1 t+ L  o2 T( b. A8 ]
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    " c: c/ j% R' g' J5 |# D/ K
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    3 R4 d: j5 k* ]$ {& P4 P5 R( q
71. int   length   =   sa.GetLength();
    6 k( ~; {9 F% S9 m! t
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    ) O( F' ]- `8 F( L8 W4 S/ U
74. struct   sockaddr_in   sockaddr;
    # I: _9 X0 u3 T5 d) {; g1 Q% T
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    " J, J1 F# l% Y; Q+ i& T
76. sockaddr.sin_family   =   AF_INET;
    ( I: O) n6 |. `# K
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
      g, L' q4 Q, E, w# e! q
79. 
    8 e' Y" c0 V: I
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    % d* ?  P" E% ^! P
85. int   pos   =   0;
    8 W7 X; `  e9 s& K# q
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    4 W9 t1 ^2 E! ~( M0 S
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
      o! m1 p$ d  }# j# g" B$ f
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    : D# w' [0 W" x8 W; }
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    ; Y, R  |, W' t0 {5 T3 }
96. return   true;
    
97. }
    9 A. g/ y1 s# I' ?
98. 
    3 n3 }+ C* \* j" ~" d
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    & M0 X  C3 J9 f) ~$ G5 e
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     3 M; A( _8 O6 i. E9 C+ {
103. CString   property;
     ' R3 ]( a: n% ]+ N3 `" h
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     9 \$ ^- Z, y; j4 U5 j
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     ) N/ Q0 a/ L7 u: w  u; L# X/ Q
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     2 y- U4 Z2 }/ F, G7 u- h
112. }
     
113. 
     ; q% ]9 A% c/ [$ j7 e6 A
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     5 p. J! l0 w0 ]( J- j2 a, T
116. {
     + _# m" P- |, a
117. m_uLocalIP   =   0;
     : V: u4 A* U/ {* a
118. isSearched   =   false;
     
119. }
     3 @2 `: H/ n* s7 |
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     / I7 v( U7 n8 L
123. UPNPNAT_MAPPING   search;
     2 K$ K' Z# S5 P0 R
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     / o! ^2 G: V0 j* U  W
125. while(pos){
     3 h( T( C6 L! ~
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     
131. }
     $ H" h& D# S0 V5 G
132. 
     ; e& y$ g) I9 w* w  s
133. 
     ! E1 ]) ~1 m2 J9 G
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     0 O: z  g2 ^5 C: J
136. if(version <=0)version   =   1;
     1 H, _- U, J& N* D2 y  W
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     6 b9 ~& _6 J; d+ W5 b) B
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     3 d% @1 M' V  v* c& N3 v. f
145. 
     / _. N! G# E6 l1 {: H1 O2 ]# Z
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     ; ]  g3 j2 l9 V6 V4 @( z+ B
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     + y  F( w+ B( q- Z, p+ A; @
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 "),
     ! }' u4 k! \$ A
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     ! e$ @( \0 U& V3 R
160. }
     # e" @4 B0 {* q9 N8 C! K
161. 
     : n* X6 x" F4 O% k9 i8 Q
162. Sleep(10);
     0 f5 |8 L! O9 n' z
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     " p! J8 t4 a" l
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     / L9 b, r1 m* W/ B7 \& T7 ~! E
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);
     ; ^3 z; t2 m3 W' I2 D/ }; I
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     1 X4 K1 \# c- [; {# ^& q
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     : Y# d: Y/ D' Q, c
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     + q/ z1 N, v% c3 d9 C% U
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     4 Q6 {& K* z- E  D
182. line.Delete(0,   9);
     
183. m_description   =   line;
     ! ]% U7 @' }; i, _0 A( w& l
184. m_description.Trim();
     3 D5 P9 X# X8 g# j$ l* f+ ]
185. return   GetDescription();
     , Z$ Y9 \9 R% L6 |4 r6 F0 |
186. }
     
187. }
     - X3 u0 ?* }4 w
188. }
     / q9 h1 u8 M% `' `
189. }
     
190. }
     8 m  j2 s8 N4 \$ N+ d
191. closesocket(s);
     
192. 
     
193. return   false;
     
194. }
     * G5 V- s) `8 Q% a

复制代码

天云上人

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


! T1 }& q3 M" p( f) t///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
  A# I% x) d8 y4 x' ~
# n* X2 u8 i+ {
#pragma once
#include <exception>
! o; t+ [" C0 Q3 q

0 {! Y# @- D' i: `4 J# v( w  enum TRISTATE{
# P1 F6 l0 E0 w        TRIS_FALSE,
% q, u; w. e7 I# U        TRIS_UNKNOWN,
        TRIS_TRUE
7 E' s. B1 J& y7 A};
, ?) t$ d, T! D2 W+ |3 e9 x

enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
/ f' |4 j8 A5 v5 n& V+ ]        UPNP_IMPL_MINIUPNPLIB,
0 q) }0 L6 p. p& m) c, `        UPNP_IMPL_NONE /*last*/
6 l' F% J- t8 L0 M3 G2 v7 M};
& M: h  d2 s# a

3 O  [3 p3 [0 h- t( ^
+ l  r7 W. G' G, }" `
class CUPnPImpl
+ J" o/ R- V8 K0 Z% N( Y  k& ]{
% ~3 n) u- \- Y: s: Fpublic:
        CUPnPImpl();
: c! m4 ^, Q1 b0 O, h        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
7 x. l2 H6 F1 C3 X& }: c. W        enum {
                UPNP_OK,
                UPNP_FAILED,
, u" @' {* s: g! b, ]+ z3 n: B6 _                UPNP_TIMEOUT
; p& H8 x" }) ^0 q4 Y        };


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
# v1 o9 c4 }1 ^( d7 u' U7 n+ q        virtual bool        CheckAndRefresh() = 0;
, @) ~# B1 s' K8 S& }/ k6 a8 i        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
6 }' w# w0 V! [8 a. ]( y: |        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
! ~0 A* q' p- E" U6 ~       
- l% C3 Q7 s0 L( s  Y        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

$ P* D2 ?$ w8 H! E2 _# ]
! G9 O* s/ d3 d4 [( z        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
, u/ m& |" T2 X        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
" ?8 K+ e2 H1 P8 {        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
7 l. b  G1 ^6 g9 b7 J$ G  \3 d
$ F: i# o) B9 E  n4 u2 W
// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
) C4 ^" B% K1 B4 p        void                                SendResultMessage();
2 H6 |9 w& ^  t- f5 U        uint16                                m_nUDPPort;
+ v% X/ m) C+ f& x5 U4 n        uint16                                m_nTCPPort;
7 ^" p5 i* _6 V        uint16                                m_nTCPWebPort;
3 B: Q) Z. ^, x* i        bool                                m_bCheckAndRefresh;


private:
. z4 n& a! S2 S" [        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;

1 P8 F3 T/ `/ K- u2 B6 x& N% P4 X3 L
6 @3 c, N# @1 H" F0 w. {};
, j1 P* a1 H& j# F6 V9 W7 `5 @
: W6 c0 N# U8 S3 h
// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
. u4 }9 w3 k. {* Xpublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
$ [, G$ ]9 o3 @! o        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
' |% I5 t' b! w" z        virtual bool        IsReady()                                                                                                { return false; }
5 m$ u, O( L  H) p5 Y" E        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
$ G. _- a6 h  x2 f* i

7 |. l9 h4 j( Y8 l# [: a& |2 t/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
1 t+ k1 }2 ]9 h7 I0 Y9 A0 Q
. n4 e  n) T# g5 O# s. x
#pragma once
2 v( A  _; \+ J+ e; g9 P#pragma warning( disable: 4355 )
# x- j; _+ X- A' P3 F- f

#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
1 b. |) T# K4 _% c2 z, i, w#include <comdef.h>
#include <winsvc.h>
. v7 |) ?+ g: I8 S7 `0 v( @
' b- X, `6 O, ], z$ J
#include <vector>
1 H8 X) L; B, z#include <exception>
#include <functional>

; K1 _. n% k# V8 G/ z( e


3 M7 ]- _/ e6 P2 i0 y& Btypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
' \" o; ]1 x, otypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
9 l4 m& [$ i( f# j8 e% Ztypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
4 U8 D  k. R5 c: V& g8 z- `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) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);

( F) [6 V, N' H2 `$ e
typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
' h9 q7 L& V# S4 q1 W* Z( g7 Y  PULONG pdwSize,
0 j0 b- |% k1 \( s2 S/ j5 h9 S" T, L. T  BOOL bOrder
- p3 G9 X, E: a! _. j3 \$ });


* P7 p/ y. f1 S/ Qtypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);
" [: z+ j) X' E/ F$ c
7 X& G" a1 g- k& ]5 r
- B% j3 t0 {$ J1 PCString translateUPnPResult(HRESULT hr);
" @; ^  Y# N* Z& T' l4 ^HRESULT UPnPMessage(HRESULT hr);
& V# d& L" `* t3 [
! d( ?( P0 u; p( O8 @
class CUPnPImplWinServ: public CUPnPImpl
  x4 Y# ?& L4 d5 \{
7 @/ n% h9 ?. i        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
/ B& {  b7 L8 l/ D% x+ ^3 N// Construction
: @+ e" j( J/ i4 {! s! g# z1 cpublic:
( Y/ A  }) m) D. U: M        virtual ~CUPnPImplWinServ();
! n# |1 F1 X  f# c( s9 T* \& T        CUPnPImplWinServ();
5 o  o2 u( c& ~! \. K! `8 {

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
# R* x2 N4 T0 o) s# t9 d6 m, w        virtual bool        IsReady();
% R3 i3 v; o8 f* L# B5 k6 ]        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
; @% F7 i' _: Y" w" x! T/ R
; l# I+ e8 L, K1 M7 `
: C4 j* l" ^  e) Z        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
/ T8 N" A+ d% _! R- R        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
( x- M" W: ^2 D, J/ \
! V' D7 p6 h' h0 K! }# @: B
) J5 Y, v, V1 G1 d4 Cprotected:
: ]+ I7 `$ k& Z0 }" G( W        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
6 E5 h% w/ ~: j: ^" q4 n% D$ H8 B# u        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
7 t, I+ _) ^0 z) s; d3 Y& V        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();


        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
& V: K! j/ G. |+ E8 U                {
- P3 M9 |: b# c% d) g# o8 L3 p                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
4 j8 ?/ i* I4 y/ T) F2 v7 S$ C                        m_bAsyncFindRunning = false;
; q( i  W/ ^- x6 z+ c                }
; a5 {( ~8 r% N$ d( A                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
* T/ U- y9 k! t+ v2 b                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
7 S4 |4 d  N6 b2 ?        }
4 {# C2 A& n" n2 N- Q
6 ~! i- U. w# T& `. b3 J  @
        TRISTATE                        m_bUPnPDeviceConnected;

1 P* U5 s4 k3 o3 k
" k0 c$ t; K5 r1 n# L// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
7 h4 q. M0 |% o. }) V        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
3 u) U6 y* E( @: P/ Z- C" Q        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
+ c: _1 |* Q0 u( z        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
  W4 I" D  o7 [8 H% E        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

4 Q& ?3 Q* K+ w: B
        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
: x! z. v3 g" h6 I$ o# f3 K, z        TGetIfEntry                                m_pfGetIfEntry;
! I! d& _: d/ c, ]/ p

        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
5 w$ c7 G8 w* A) s* ^                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
+ F( i% x4 @3 k                {
                        CComBSTR deviceName;
- E, {' B) d6 M                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
6 P- @" R! F; S* E4 `" }+ S
2 w, k9 O+ |& ?9 X6 B  [  N! _8 L
+ `% h1 V- J6 q3 P6 {                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
/ |; E% @" b, [$ A3 Q

7 [% c. S  Y* W" `( x; K                        return wcscmp( deviceName.m_str, m_udn ) == 0;
* W6 ~9 t+ N* ~7 g" G& ?6 F                }
, ?: e- u% h( }+ o9 F$ T                CComBSTR m_udn;
        };
/ k# ?. e& K! N5 T9 w; b7 A. ?       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
9 w( |" L3 S: J; j& j% U        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
" \: \+ X- i7 D3 b& N        void        DeleteExistingPortMappings(ServicePointer pService);
% n4 G7 {' g3 |0 S3 d" t4 W        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
. I) @+ [0 H6 x7 B5 N        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
/ I' K) f4 G, F2 X: h: E                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();


        // Utility functions
& A; [, F  i. }# C2 P        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
# g; Q6 I6 q: \9 J  W6 k0 D. I( G8 S        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
2 i) F* U* l: w) Z& d8 A% ~        CString        GetLocalRoutableIP(ServicePointer pService);
% _, b; j" Y6 W

// Private members
private:
  U9 \" O% w7 t" ^        DWORD        m_tLastEvent;        // When the last event was received?
% d  }  ^- ~$ h& F1 f        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
( i5 A( i0 D  S: R& q2 I        FinderPointer                        m_pDeviceFinder;
! j2 a' {8 Q( V! @. Y: a' ^        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
4 {1 J& N* ~5 Y, W) y

& b# C+ `# M% M        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
# g' Y$ ~% o- n$ p! s        bool        m_bInited;
# a- R2 ~* m7 y2 \1 i        bool        m_bAsyncFindRunning;
; H" S6 r4 u$ t1 G        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
3 j: |2 X/ {/ s0 y6 D. Q% U* |        bool        m_bServiceStartedByEmule;
+ P6 M. e* k  J/ P+ a- O        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
8 M- e) D8 Z- o+ Z4 A5 ^* S
1 Y; ^* P# A, G
. H3 F, B# ~" y) u7 v};


// DeviceFinder Callback
& W2 ~3 D1 ]% v# Q4 w( ?class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
public:
; f2 u: }4 t( w5 O; g0 R        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }


  C6 }( y! Y0 N% L7 ?( t   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
! I4 L, k/ Z% Y) t+ O   STDMETHODIMP_(ULONG) AddRef();
0 r; ~7 {& \" e# l* M, q   STDMETHODIMP_(ULONG) Release();
/ Z2 ^2 W2 R- U
, O5 M( C8 m# h4 ]
// implementation
" o! X; C  j  I2 q1 hprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
! H. X# }# ^! X8 A" N1 B        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);


4 ]/ c# \) G: L% ~private:
" M- ]) q2 n# E; P% N& o6 b        CUPnPImplWinServ& m_instance;
* b# [5 h7 ]/ j$ f5 L3 ]8 B        LONG m_lRefCount;
};


// Service Callback
0 H! u. L6 x% n# Nclass CServiceCallback
        : public IUPnPServiceCallback
/ }+ @2 r+ k0 V- F2 ]{
2 }: U( x* Z% j* p# G/ x- cpublic:
6 j/ F9 @( M5 V        CServiceCallback(CUPnPImplWinServ& instance)
/ f# A. R& o' h3 j                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
8 s7 T$ g% j! m9 H! F1 ^, f3 i   STDMETHODIMP_(ULONG) AddRef();
+ b2 T' y% x2 Z   STDMETHODIMP_(ULONG) Release();
5 F/ u8 p% r% X

// implementation
" S4 g0 a6 |6 x4 p+ j" M% T# ~4 uprivate:
: e4 S6 s0 r2 |4 L: ^        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
  O8 [( e" C& z        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


private:
        CUPnPImplWinServ& m_instance;
9 }; a# U6 W. T2 \* i        LONG m_lRefCount;
: m" u7 W2 F9 [0 E};


/////////////////////////////////////////////////
+ J) f( O$ ?7 }1 y
4 j' n3 ^5 X9 B3 g( d
使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
( s" i$ S8 Y4 e9 Z0 H2 gCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
; X. q! e% {  p. OCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.