UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   , S# i- g, t1 D1 Z2 Q, R5 S; t
3. 
   
4. #pragma   once
   - j& [! |7 ^* }' @) ]; c6 U4 T
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   0 T& Q2 N  {4 l: g$ f/ X2 b" V
9. {
   0 A6 |6 o. i) ^+ X
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    6 h! U2 `9 C" e+ D3 q
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    " v. S  y3 r8 ], d5 f
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    " f0 w. |; A' R
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    9 G# V* F; W8 c  w: Q
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    / O. a  H8 s" W% }9 D, n2 p5 N
18. 
    * D$ ~5 V: K% l) ?8 m( k
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    # n/ G& l$ t! Q& p5 m0 q' ^
22. }   UPNPNAT_PROTOCOL;
    * e4 r% l( [) v% i0 l4 S, [8 t4 E
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    5 F+ ~, K* K5 F5 w. f( ]8 y5 O
26. WORD   externalPort; //   Port   mapping   external   port
    - |  }0 K, c- e! i% z
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    : O1 e7 c, V4 M* W
28. CString   description; //   Port   mapping   description
    - {' N% f) ^/ t* d" I" H
29. }   UPNPNAT_MAPPING;
    
30. 
    - i8 }. S0 J# G. g1 |. u
31. MyUPnP();
    2 o! c- x3 v( m  F
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    + F2 C2 b9 o) U; q8 W
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    2 Z6 q0 h5 W- M8 |: j
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    ) I) K* [$ [1 J
40. WORD GetLocalIP();
    * d3 z* x- _$ J+ S1 [8 ~
41. bool IsLANIP(WORD   nIP);
    ! X, J7 A* g9 W* e7 ~8 ^/ y6 C$ I
42. 
    % a) |8 ^9 a% q1 \! P
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    7 ^# |- T6 Y5 Q* D8 L
46. 
    5 q6 V. M0 c! ?3 T
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    * N% J* {) z7 g, M1 ~0 Y& s
48.       const   CString&   descri,   const   CString&   type);
    2 f+ R1 j/ ^; Z3 p0 V
49. bool   deletePortmap(int   eport,   const   CString&   type);
    ! Z$ o; Y4 {# U* a, I* ?7 A
50. 
    , L. ^" N8 t3 ]5 n* M
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    1 m  r. I1 r8 l# Q) B8 ^( j
53. bool Search(int   version=1);
    
54. bool GetDescription();
    ; _, ?4 W3 J' U3 c8 ]+ T- p3 C! I4 ~
55. CString GetProperty(const   CString&   name,   CString&   response);
    : a! p( y. `" f- ^* K% b2 h
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);
    , E  X4 s% @* N- _1 s, x& v. P
60. CString m_devicename;
    
61. CString m_name;
    - p# X+ M# C6 [: B0 c
62. CString m_description;
    6 p6 N. o; H. e; s4 ?) I1 W8 h4 ~' x. W
63. CString m_baseurl;
    
64. CString m_controlurl;
    % S* J6 d4 Z( c' _" G
65. CString m_friendlyname;
    ( i, b8 `1 B( b4 F0 T# a
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    , ?! H. d! d, }" G
74. WORD m_uLocalIP;
    
75. 
    7 S1 a. l9 H1 g3 O+ ^! p7 W
76. bool isSearched;
    " ^2 W0 v& W: B( x# ?9 R! r3 J
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   ! x- e, E+ ^8 ~
2. #include   "stdafx.h "
   + n% ?* g# K; g5 [% `! r# [( n+ }( k
3. 
   / ^5 s9 Q$ ~' M( X, h! q4 |  O
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   : {6 c7 U0 W* x3 I8 Q
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   * H: r: u5 v2 n- f4 W- u" L
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   6 F: @- S; i4 a. y5 ?
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    # D' H" F6 ~) w
11. 
    3 {. Z7 m5 \; l: z$ x7 c% @# R
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    / ?4 g5 I( q9 ^: E0 n6 J
13. static   const   int UPNPPORT   =   1900;
    8 j8 X6 X6 W; c) Y  Z+ P! J
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    % p( N* e3 r6 g) x5 a6 W1 E
19. 
    - _  Q; c; [! X) n: m
20. s.Format(_T( "%d "),   i);
    
21. 
    & [3 W- p* a' I; D+ Q
22. return   s;
    ) Q: Z' w/ U% X) p# M
23. }
    
24. 
    * G/ _: a0 ^9 D( j) }+ _
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    % L2 R8 C1 S0 U  x+ C
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    0 ~. f$ a6 ~' C& v
28. }
    
29. 
    % V: Q  Q) t0 j: P
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    : C( ^  z! Y, l% X+ H* ]# {, @
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    * O, l. e. f9 v/ f) h8 }, i) t
34. 
    1 v0 G2 U- }) m7 V% H% i2 O& R
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    2 N. {) b& f- H" B6 p3 n4 s
37. char   buffer[10240];
    
38. 
    0 p2 T9 ^9 _; S
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    % w) O+ D( \8 u
41. strcpy(buffer,   (const   char*)sa);
    2 ~$ B1 K, ^) K% @, R
42. 
    , P+ B5 g+ Y: x. `! q1 |. M
43. uint32   ip   =   inet_addr(CStringA(addr));
    . x; r- j9 T% m- t
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    8 S$ N9 v$ [8 l+ [9 |  v2 Y
46. sockaddr.sin_family   =   AF_INET;
    & c: ~/ ^; _- T; \6 }
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    5 t2 {1 @# b0 C  Z
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    # v( E; J" F4 |
54. int   n   =   send(s,   buffer,   length,   0);
      y; u+ S) x3 V2 l7 R
55. Sleep(100);
    4 L+ [$ c1 K& _" |' d
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    ) f* c8 W! B2 F4 N' L
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    $ G& [1 z9 Z, w( x! ]: ^7 I& J
61. response   =   CString(CStringA(buffer,   rlen));
    / `3 J! |1 G7 X- Q# s% w
62. 
    1 ]' d" _" Z! }9 l9 ~0 a
63. return   true;
    
64. }
    - w: J) \/ F- H5 a
65. 
    . r0 ]6 r! ~/ g3 w
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    8 U% n3 {9 }) }$ v6 c
67. {
    0 ]+ A  v5 J1 ?! {; @2 d5 S0 A
68. char   buffer[10240];
    ) N, H2 ?4 E# C: P  R. l; t2 t
69. 
    ; l/ ^+ E2 S: o7 `
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    ! t6 A& y# E( P* o
73. 
    
74. struct   sockaddr_in   sockaddr;
    5 u7 B: N3 }; n! ]6 J8 x3 N
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    + F6 `% u0 Z1 R( d5 L: m
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    # I5 I* @5 {7 x2 h& J/ C
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    6 l4 L4 A/ e# j( N" A7 |
86. 
    + y8 b; m9 g3 S5 c
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    + u& H% l3 e$ z0 D
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    & ^) M1 R4 W, n; [
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    5 d9 n1 k# P2 \# p
97. }
    % F! U- @; c! b; H
98. 
    - W, U0 H' D8 j( U+ K
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     $ ]/ W, ~; i  N+ J& \8 T! Q
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     , m0 c  Z$ I  u4 p- c
105. int   posStart   =   all.Find(startTag);
     2 Q: o* m4 d6 i$ y; h3 N% D- l
106. if   (posStart <0)   return   CString();
     & \: O) x" g# c
107. 
     $ j% U* h: [5 g
108. int   posEnd   =   all.Find(endTag,   posStart);
     , |3 b. ~( K' J2 l! P+ n9 ^
109. if   (posStart> =posEnd)   return   CString();
     5 r5 m, d6 R; n
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     0 m1 i: ^, H9 R
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     ' ~" W7 L  p4 ~* a
128. }
     
129. 
     . X8 c. h$ q% O# _/ Z
130. m_Mappings.RemoveAll();
     2 w* Q0 X) f4 N2 f
131. }
     
132. 
     
133. 
     ; V3 U5 m4 _/ B# e* J
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     5 D2 W$ e3 f1 N0 z0 w6 l
139. #define   NUMBEROFDEVICES 2
     : G: k( s3 j/ n# r
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     " z% p7 u8 o3 n& \% `' x! Q) D
142. {UPNPPORTMAP0,   _T( "service ")},
     # e8 E1 ^% n0 |8 ^$ L  Y
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     1 [6 s# F3 c6 a4 ]
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     2 U3 e4 Y4 l; S# g$ Z* t2 I! P
148. ioctlsocket(s,   FIONBIO,   &lv);
     ) D( F! I8 a7 K5 k# M- f: z
149. 
     
150. int   rlen   =   0;
     9 A- J+ b; y4 t; V4 i" y0 q7 I
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     9 p! P# ~9 P6 k0 `1 b4 O
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 "),
     
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     4 t4 P7 _5 K: P, g
161. 
     
162. Sleep(10);
     ( B: Z) _& w8 ]7 C
163. 
     " O. z$ u# ?  }. T
164. char   buffer[10240];
     9 B" ^; {) M$ v( }
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     2 b+ S! F3 ~6 Z. p/ f
170. CString   result;
     & V1 F' z& y. [) ?+ i6 l7 ]
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     + y" m4 l, z! [* q1 t/ G9 t
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     ( W6 I/ F9 x6 ?% J
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     ' n2 J& X! J8 g! T9 V5 O
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     ( M; x" w2 X! S1 y/ c
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     % t% ?, X& \6 J
180. name.MakeUpper();
     # [; |- C- _% R/ c. }2 A
181. if   (name   ==   _T( "LOCATION: "))   {
     & L& K! n9 A3 T9 f9 ?2 z8 i
182. line.Delete(0,   9);
     
183. m_description   =   line;
     8 g" l6 ]' S% C
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     
187. }
     5 u) X8 K5 C6 M$ c, V2 F8 ^
188. }
       O6 @# s9 l3 @' m- M6 V! h
189. }
     
190. }
     
191. closesocket(s);
     
192. 
     ( T* i( d. b) b9 D+ G6 n) c6 n- B
193. return   false;
     . M3 ^* F- t8 X
194. }
     

复制代码

天云上人

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

) c& M1 D8 |/ b; s" t! E
///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


3 b0 z& P, X, S9 e- R* P) V#pragma once
#include <exception>


8 l7 w3 {" T" c7 M, G! a/ O! f  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
' ?' |8 `7 j: P  h/ h- o        TRIS_TRUE
/ c3 Z3 E7 D6 v* W/ Y; U( G};
$ l" S8 c4 A2 J8 P- ?, X+ B0 x

# v9 u0 a# x3 M- q. e' m, Kenum UPNP_IMPLEMENTATION{
: c/ M2 u  x3 h" m; E        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};

  k4 |9 ^; `  g% M9 B
- q$ v: N0 S$ q- K

! S" }. {3 h3 X) h% C1 \! e8 zclass CUPnPImpl
{
+ h& N8 Z0 q& o, `1 t8 x- Hpublic:
        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
. x2 A% e' g0 Z+ q* p* q        enum {
                UPNP_OK,
" F" D$ C/ I4 u7 }+ T$ i                UPNP_FAILED,
3 a% G! K  [. |; ]                UPNP_TIMEOUT
        };


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
. v5 U5 C6 w4 L* f. d6 d        virtual bool        CheckAndRefresh() = 0;
+ ~  v; m. _9 @' W( P& x, q        virtual void        StopAsyncFind() = 0;
$ d4 L4 a# C/ b$ s- d8 P7 v# o) F        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
% l! b6 s# E4 y3 P* ~       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

$ V) N' r8 n9 N" Z+ C( H( a
$ f* P3 V7 }& b8 d+ l) m* W' P4 z        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
7 q6 m3 V- f6 G9 y; }0 S        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
' S' N7 w& ~! y        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
, X; q$ J4 i8 [' `  n' Q" c  @1 }

- k9 f& {( s! @7 V, i// Implementation
protected:
. {$ R3 y; T% {6 N3 ?; M6 H3 \7 v$ k        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
1 b. R" Z- S# F. Y/ E" y/ H        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
8 T( p' ]0 }3 j3 z        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;


7 `# M* d5 f' t. ^1 vprivate:
5 m* A. Y% f& ~7 A1 t* R' D        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
& s8 M2 S' V5 _. r& O

};
. D7 K" p! Y( ?2 b5 s0 C5 z7 d! Z9 G

// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
% ?" \5 s, W) ~  |" o{
5 x" Z2 `+ u, K2 }7 h+ Y* c" M- v" wpublic:
; C+ D1 I# [) H        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
. x- t4 `% P+ ]% c  Q        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
5 v1 W5 J( y) S};
$ b& U1 W* J" k2 O; r" a+ Y1 p
7 ~! C: L9 @3 J5 Y: K7 \
! r% N' T9 e( a2 o/ x/////////////////////////////////////
+ I) }0 d, h' _  r* G& W+ e//下面是使用windows操作系统自带的UPNP功能的子类
3 g( a. s7 R) w1 \/ h& m

- \5 p6 k7 e1 N' L#pragma once
! b" m! C6 m# n1 [& a7 q0 g#pragma warning( disable: 4355 )


#include "UPnPImpl.h"
#include <upnp.h>
+ U+ E2 x% d; u2 x#include <iphlpapi.h>
#include <comdef.h>
( `' ^- B+ ~9 y/ g#include <winsvc.h>
; H# R4 t/ `' V' r

8 b8 Z4 b. P( G6 f$ ^9 h! O#include <vector>
#include <exception>
: D1 h1 l  K! J/ N; y#include <functional>

# S* K5 l# {3 d' `! X
" q* G2 W9 G9 }" m& L. p
1 e% E0 s6 W/ @
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
7 r. q! S- z5 ?% j8 f% mtypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
- m3 k/ d1 K8 O3 n% x7 y0 ^, Ntypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
. `  _8 ~, M( T, Z9 ^typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
3 i$ z+ ?! D1 q3 c- ktypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


5 Q! P2 Y: o: C/ b) M8 l/ ]typedef DWORD (WINAPI* TGetBestInterface) (
8 N5 w: G3 b6 Z$ {- r! G  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
$ G) n2 d+ W2 j8 U- ]6 r4 [) z);
1 O2 d: a. t$ f: g

typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
( ], u  M8 N& p  BOOL bOrder
);
/ \0 ]7 Q0 H8 t% X/ v" y5 h! t! f

1 ]/ y  x3 A8 G0 h0 ztypedef DWORD (WINAPI* TGetIfEntry) (
0 q2 }  N& }/ t5 V  PMIB_IFROW pIfRow
);
5 Q* ?$ e% Z% K4 I" L
7 B0 n6 L+ g2 {  y( `( j. n& d
CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
+ g0 r$ c( p- T5 _: U" y

class CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
public:
        virtual ~CUPnPImplWinServ();
# F6 T( o+ X, d        CUPnPImplWinServ();


' ^. V9 j# ]4 P& b" D        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
: R6 y) _5 [0 g( z6 ^- l# `        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


3 S2 o9 g7 O' b! G  V        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
! j! Z& s: X5 Y( ~: a! _* ]        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
0 I4 x7 M; b' I- v- d3 a) c2 K9 ~        virtual bool        CheckAndRefresh()                                                                                { return false; };


protected:
5 E6 n7 d  k' `+ E) U8 j7 o8 l        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
9 @3 L& s. V9 r2 k5 E        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
1 k. i% d2 _3 c- e# L. m$ o4 G        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();
, c7 o- q1 A! X2 J5 o
4 i/ N& @; M/ q1 R
        inline bool IsAsyncFindRunning()
        {
8 X, E5 A7 ?" p0 @                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
. O% V$ Y' v1 q; @& h, z* j                        m_bAsyncFindRunning = false;
0 v( V3 d$ O& R) U                }
                MSG msg;
' O' T  s3 t9 i9 D                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
4 a6 r7 j+ X% P                }
                return m_bAsyncFindRunning;
+ [( Z2 z/ C2 t: S        }

' R' o7 h; [7 u0 Z0 L
        TRISTATE                        m_bUPnPDeviceConnected;
1 S# Q( T) H3 L4 K2 x$ J% b5 \

// Implementation
        // API functions
- p1 z# {' {, v        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
! O* d; r3 f* `8 Q        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
3 G' F. e0 j( }* F, [        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
2 O- d9 e' P3 C4 ?" j1 j        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
6 s( y/ N0 F8 c5 l& t% y+ q        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
' f! S8 a, U/ \( P+ f( w( d. i        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


        TGetBestInterface                m_pfGetBestInterface;
: y6 P- ?2 P# h! w( P; g/ g+ W        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;

' ?" Y" z1 ]" o, |! Z' p# ?
, ~& v) p  C5 ^4 ~) L3 w        static FinderPointer CreateFinderInstance();
  d! e0 L! X/ D% c% z# P& |        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
% b2 t# |" B4 x$ C( o; k) e                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
9 ^- @; m( p9 Z! b: z/ ]                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
3 b7 f# z6 L3 Z
9 r: m4 |$ _1 @! A- D) X
" ^0 V- S% J/ D; B$ S- w9 e6 H                        if ( FAILED( hr ) )
) x; ~+ |. K0 \                                return UPnPMessage( hr ), false;
0 I% M- ?" u# q  e& u( N( F, u
9 @2 a5 Q9 L) B# v
# A6 ]+ ?( j. P6 x                        return wcscmp( deviceName.m_str, m_udn ) == 0;
% |. q6 f8 V0 [/ z% h: f7 _" x                }
, N# C9 e, Y6 D                CComBSTR m_udn;
        };
       
0 ~4 P9 }* o+ }. e$ g0 s        void        ProcessAsyncFind(CComBSTR bsSearchType);
) I! T" x) z3 P  [1 ~        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
0 h, M; `4 U: ~# k        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
; T. c6 C% P+ m" Y# M/ x        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
+ @. ], U& Z0 L4 S+ B; S& _        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
* _+ |0 O& F" N/ F        void        StopUPnPService();


        // Utility functions
" X, }- I( D& g9 |" D) |( g2 Y6 w        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
+ m4 ^' K7 G& u' b        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
$ h! W2 ~' v8 {        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);
: v6 ~, K  F( A# ]; A5 {        CString        GetLocalRoutableIP(ServicePointer pService);


; O4 A# E( l, Q// Private members
5 P! z5 J1 ^! V, Xprivate:
5 Q5 M* N: K8 }7 [        DWORD        m_tLastEvent;        // When the last event was received?
/ z, n  B3 ~& v        std::vector< DevicePointer >  m_pDevices;
6 [# B# J, H/ t6 w3 Y        std::vector< ServicePointer > m_pServices;
* i, F6 C5 F0 H# Q7 H        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
. e, N2 g' p& H) N  a5 F! V
+ ~9 r+ k, t& e1 j0 ]- {# r& s
        LONG        m_nAsyncFindHandle;
: y9 V, k8 \6 T( U4 l2 L0 a        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
0 n- M% O' `$ B. M8 ?        CString m_sExternalIP;
6 e% H( }+ ^) j/ l2 @# s        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
8 B" }. K& I% W! P) I) p% e        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
8 D4 B0 r7 h% |* q5 U        bool        m_bDisableWANIPSetup;
& W0 C( G/ T& p* d: s3 j        bool        m_bDisableWANPPPSetup;
  {1 ?+ ^1 v# |3 ]6 @

};
9 V' d: U/ W- E6 u' v5 V  A+ M  G
0 P, \/ |0 R% ^6 V
7 X5 A# B5 D. Y% K; r// DeviceFinder Callback
/ ^$ I$ X: H+ g, mclass CDeviceFinderCallback
/ Q! r1 l/ G) L/ ~        : public IUPnPDeviceFinderCallback
+ e7 h2 T! I; ^# j{
6 [" {* ]4 Z6 F" N( H5 [public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
: D+ @+ m) [& i0 ]        { m_lRefCount = 0; }

' h1 R* Z- V* ~% }8 V
  [+ u  j1 _) V8 I   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
9 f) g) m2 h+ Z   STDMETHODIMP_(ULONG) Release();
2 g5 ?6 x0 q( f# i

& n5 ?" N! {& A! ?; O- m// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
+ c; z5 B& {9 D* N        HRESULT __stdcall SearchComplete(LONG nFindData);


private:
7 J3 m% j9 J6 ^7 x0 m/ L0 l        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};


2 @  H% o# u5 j5 J// Service Callback
: B+ H& r/ Q! ~! A4 {7 ]3 kclass CServiceCallback
% P1 G/ L% i9 l/ l2 Z. H! U/ s* C        : public IUPnPServiceCallback
! M$ J9 ]- }/ y5 l{
public:
& {' ~, V# B- M2 Y6 q' @8 j0 A        CServiceCallback(CUPnPImplWinServ& instance)
2 r, R0 B7 x  Y9 t1 O, \2 d                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
; H) L/ b! N; W  w) t6 ~2 T1 @, I0 B& W   STDMETHODIMP_(ULONG) Release();
+ P1 C  n9 Z4 V

; v3 W+ E( P" w1 W2 {& M) w( t// implementation
+ j2 X' {& q- ^8 K) |private:
% `4 n( V" ^: Z8 |        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
+ r, K* ~/ ]3 T0 E+ ~        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

' {+ H7 R* G) e2 G. {: J" y
  G8 [( h" \( O! q! X4 ^9 Lprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
+ d% j3 |  t7 Y) H0 E- h" Z7 H};
0 Y& g& Q( G% K0 I, h, l

# I( f9 J4 T2 Z) Z2 Z1 |- c4 `5 G/////////////////////////////////////////////////
, `; w. Y! p9 y3 _) G% x

使用时只需要使用抽象类的接口。
+ ?$ _; ?: V1 W) y$ Y1 W' vCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.