UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   + e" z# A" y8 i! O: ?- }- `3 y- l
3. 
   
4. #pragma   once
   ! S' I( @/ s- e4 l# k2 d( U
5. 
   0 B7 S3 S1 [. ~. E7 l2 Y
6. typedef   unsigned   long   ulong;
   
7. 
   / ~6 F1 H$ R& m. {5 V
8. class   MyUPnP
   # H6 A5 U' e. H% B
9. {
   
10. public:
    & I  M  g8 F" }  w& x$ p9 Z) I  t
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
    : M1 Y# _2 [7 U' F( ?4 N* O2 }% I  A
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    ( {1 g9 c& a' e: \; i* |
18. 
    ( o, p* [3 h$ v/ |9 ~
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
      E  n+ X/ \' i
22. }   UPNPNAT_PROTOCOL;
      n7 g8 P+ D* l0 y
23. 
    
24. typedef   struct{
    ) Y) U* R( D$ u: K' G
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    , B9 T' ]. M  k2 W& ]
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    # e; `+ n- [6 Q% V
31. MyUPnP();
    
32. ~MyUPnP();
    # d0 X! I/ E4 x, x3 V0 }, I  I
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    $ X5 C1 D& R( n
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    + h" h2 l4 c; |% ^4 R5 G  e
46. 
    * S: V4 E$ J& \8 E9 p
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    $ a) T$ f+ E5 {8 E* N4 \
48.       const   CString&   descri,   const   CString&   type);
    4 V( k; o$ `/ c# R
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    0 u, @3 t8 J4 K2 H
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    9 \/ L  t( o3 ]! v% c2 |
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    7 q( j8 X; M* N  c$ E9 Z
59. bool InternalSearch(int   version);
    " f% l0 N4 W; @% S3 u7 E% I* E+ @
60. CString m_devicename;
    0 i+ M1 ^  x0 [$ ^6 `: H, a
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
    " [+ b1 f! w& d0 ?
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    % t/ t& |+ F* l
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    / X$ J- H7 ]* m+ k* e* O0 H
71. 
    
72. CString m_slocalIP;
      y6 P% \# [' X" X) Y, N
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    % T! u/ W( @: b1 Z) C) |5 Y
75. 
    2 ~5 b6 T9 P/ N2 u2 ^) }4 |
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   8 Y: C4 |' k* h0 T9 I2 N0 U9 s) f
2. #include   "stdafx.h "
   
3. 
   ! F% P; U3 d  `6 _/ ?
4. #include   "upnp.h "
   $ Y1 c- |# ]- a
5. 
   5 l% K1 O! v2 r
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   5 v/ j4 ~% U) |! {+ ^" p
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   * k! X2 \7 d7 p6 K% Y& p9 x8 g
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   ( A: g" u' o; \4 f3 I$ ~( ?: t: c* X
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   3 y4 _. c4 y. H% @4 n* p
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    ( r- }- X# d. [$ Q
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    , L- C/ H  \) K; N! K& ?: `
13. static   const   int UPNPPORT   =   1900;
    ; V1 E% O/ i3 x5 |. M8 B
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    4 H5 Q3 z" f' `/ h3 n
15. 
    
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    5 h: [9 J9 @* ?, L2 O
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    : o5 L% w( `' n1 p, X% y9 g: C
23. }
    2 k# p& N. y# G" K) Q, I' V& v
24. 
    0 L5 l. F; W( d# X5 ~
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    & j" z/ R, }$ [0 m0 C
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    , ~; C4 f, h& m* q4 W7 z
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    $ w9 d) O" F2 l0 a% s
33. }
    . T  f, I' L& h& O! g5 Q( J* Z
34. 
    7 Q6 V: c9 M6 P! j  M
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    ' w9 L% f- H7 R4 @/ n
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    0 A7 z) e  f9 }/ }* w' T4 ?& |0 Y3 {: N
42. 
    ) R. I! m+ T+ B7 m& Q- W% f' b6 o; |
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    * k- ?# B1 Z0 J4 \1 q/ m" a: g/ P
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    6 u# j5 _7 R8 l+ Q. e0 v
46. sockaddr.sin_family   =   AF_INET;
    . o+ c* q; C: N9 r0 g# _/ c/ I
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;
    8 g2 y" ?. t5 t% ]# t
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    6 {! u! ?4 j8 E; ^: _
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    0 H4 D; b1 o% @+ s8 F
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    2 A& ]0 p0 d3 ~# ?8 \
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    # e' F9 W' \1 w) Z7 T8 \
62. 
    0 c& I- k3 j2 K; P* ^, J
63. return   true;
    
64. }
    $ o4 W5 V' D, u4 |. m
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    * x5 w5 ]! |9 u3 f! [
69. 
    4 M, S6 q, t( x( h8 N/ Z6 ?/ j
70. const   CStringA   sa(request);
    , K" t' Z1 N/ A- h
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    % l" _6 r1 R, J8 y* n9 e: b
76. sockaddr.sin_family   =   AF_INET;
    2 |: n7 b4 e) n! q! H5 y0 u
77. sockaddr.sin_port   =   htons(port);
    ) E3 |. }. E9 y3 N- R# X
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    , ?) n& g9 J0 l( ]0 c8 {8 `
79. 
    1 u1 ~; D4 h" l+ \$ U! k; \
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    5 o. i; |( D4 k4 c
81. }
    ' \( U) n" e" j  q- M/ r. a, [
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    , {3 X$ d# Y  q3 L8 T! }
84. {
    
85. int   pos   =   0;
    
86. 
    * s7 P! ~# ~: [- [. T
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    , D( e  A  z5 c3 [& F" T
88. 
    
89. result   =   response;
    ; `$ W. O+ d# z" E. q
90. result.Delete(0,   pos);
    9 N8 u, |) J( x5 z4 u# V
91. 
    
92. pos   =   0;
    : z8 B! s- l! T' A# G
93. status.Tokenize(_T( "   "),   pos);
    $ {0 `2 a: C8 \( u* E' C
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    $ C, ^# T( p. |0 x  u3 T
97. }
    
98. 
    $ n; L) C+ z7 Z+ n5 b8 R7 S* S* T2 V
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    5 k) ~2 @7 V4 E, L0 V6 ~
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     : [. {3 ]2 |- ^( E, C5 Z+ g! J
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     0 Y# I' S6 r* s: C" T
109. if   (posStart> =posEnd)   return   CString();
     + `( U2 ]+ w* @& e* a& U# H7 b' S3 R* g
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     2 v: {6 ^' o3 v! K2 J2 v( n6 Q0 k
112. }
     2 ~6 a7 v# |1 R5 |- ^% s0 ~+ ~
113. 
     
114. MyUPnP::MyUPnP()
     ! F5 z: y8 m0 H6 `* P) _% J- d
115. :   m_version(1)
     3 J: f4 W& }5 v1 |
116. {
     
117. m_uLocalIP   =   0;
     . n/ U3 C5 A, p3 o9 n+ V. I1 C1 d
118. isSearched   =   false;
     
119. }
     1 D. H0 L5 }7 ?' _
120. 
     & A1 ]  I7 i  J! P* h7 j
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);
     
128. }
     
129. 
     & `, W4 |) r# L( F+ T6 u
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     % G* c! [. b: U
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     ' U; l# a3 ^$ J4 k
144. };
     : E% I3 q, i9 L+ I
145. 
     ' {# n3 u9 h: l6 p
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     7 S+ `2 u4 f+ e2 z2 m
149. 
     
150. int   rlen   =   0;
     3 D/ V9 q3 e5 i+ E- Z
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;
     : H$ x( x: y+ P7 N9 W7 o
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 "),
     $ V+ Y' ^0 j, J; O* I1 Q$ ^+ c
157. 6,   m_name);
     ! Y3 t, m; f/ \& H( ^! x  j, g8 d2 o
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     * S8 A0 X6 L% I" b& G
159. }
     
160. }
     
161. 
     
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     " g4 f4 W. ?/ k9 B
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
       g6 G( Q# Z- Y1 g
168. 
     , c* s9 {! W$ \9 q
169. CString   response   =   CString(CStringA(buffer,   rlen));
     & r; d5 a/ w8 B! c5 }/ Y; _& @. Y( x
170. CString   result;
     * q: Y8 t/ X. @" e- n
171. if   (!parseHTTPResponse(response,   result))   return   false;
     # f/ J5 _# d5 B
172. 
     $ W+ U( U7 D8 o) L  H6 {
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     3 \( U/ q' n$ g; V
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     + z2 @4 X( I) N8 q, |9 @
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;
     * U: ?  K' A1 f, E+ t
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     # ^0 z6 u  I% v7 w
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     0 Q( B) K- a( g; _9 d1 p
187. }
     7 k* j; A9 g+ D
188. }
     " F- h2 d3 K& O  ?8 s  k
189. }
     3 O$ m/ C' i+ ]/ i  n
190. }
     
191. closesocket(s);
     
192. 
     
193. return   false;
     " w* o# E% t. q  }) U) T
194. }
     

复制代码

天云上人

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

: H8 p9 A' s1 A  l  W# y
5 V, Z  B4 N% \# b1 @) ^///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


#pragma once
& e% I1 i# \* y+ o7 p8 t( \5 x" D#include <exception>

4 Z( S0 A+ ?' D  Y# ?! T6 m/ ~  z+ L
  enum TRISTATE{
9 @$ `& b% f$ d& n        TRIS_FALSE,
5 R) o+ b$ |- U        TRIS_UNKNOWN,
        TRIS_TRUE
};


5 B& m5 ~7 `: h& Senum UPNP_IMPLEMENTATION{
) f: A5 D* ~$ `, ]' |        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};

0 K7 u# Z3 Y0 g
$ V6 ?& O; _. b- M- C4 a
7 s- D3 O6 n" w: K
' {6 s7 d/ D2 S) O, C) m9 xclass CUPnPImpl
{
) R! E" |% T& _) o; upublic:
4 O$ A0 T6 _) k- Z9 e- i        CUPnPImpl();
, w& M6 v' Y8 H  t0 i        virtual ~CUPnPImpl();
* k; x. e  [, w" p1 N        struct UPnPError : std::exception {};
' ?, W" n  U0 m( O        enum {
: p+ U7 I* m2 L2 u& ^                UPNP_OK,
8 H) a7 A4 s$ v( i                UPNP_FAILED,
7 a& A: H7 L& U! b2 q0 N2 U1 q                UPNP_TIMEOUT
        };

! m9 i; `4 N# \6 ?' u/ ]# C
$ K0 g  [; l, J$ r, {. k        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
7 r% K# M. u5 }; R: T        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
6 y5 t: H  X4 _3 ?5 f- v$ d+ Q6 d        virtual int                GetImplementationID() = 0;
/ f) H8 f& ~/ ?. f( x       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

4 w) {( s" e( S- G3 ?+ x' C2 A  d
/ R. R' P( D5 X% H$ T' ]" O        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
7 e% g5 ^8 S7 ?$ x        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


: ^4 V$ `% n7 N) U$ f3 J. t// Implementation
protected:
/ j5 `) V9 {+ q# H0 z        volatile TRISTATE        m_bUPnPPortsForwarded;
6 F/ n1 y% r* U+ ]        void                                SendResultMessage();
% S- j+ Y" O: u        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
% j1 |4 M9 B9 M6 j* U3 u        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
( S+ c! i0 V/ ^3 x5 f
2 l& ]" O8 t/ @& M7 u5 ~# x
private:
        HWND        m_hResultMessageWindow;
  v9 I( {- |, }6 p& e; b: b$ J        UINT        m_nResultMessageID;
; R# }/ T: o$ L+ B
1 O1 {  u" l* o& M( j$ _/ e4 R- g
) F% b( _8 T# \! ?( f};


" Z: |4 U1 \" c// Dummy Implementation to be used when no other implementation is available
# f# L# ]* U  Gclass CUPnPImplNone: public CUPnPImpl
3 }2 y/ D# n4 X2 _. W, ~9 h{
1 i- c# w" r% L( y+ M  S/ M9 U9 @public:
2 ]9 N3 I( l, U! W9 ~7 U        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
9 U  s$ U5 Q2 w% F7 u9 R        virtual bool        IsReady()                                                                                                { return false; }
4 @" B6 K2 J" x" `! \" H# t; G        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
2 `" t7 I4 V- z5 j, G};
" @" E2 J9 a" S% J
0 Q3 I1 O6 w9 J( y
/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
- S0 G3 L0 R8 e" U( N
- q8 A( c, ^& H* p: E: V- E
#pragma once
#pragma warning( disable: 4355 )

" I% _, \! y% x' U
#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
5 w) f  r9 _9 W* Z#include <comdef.h>
+ @/ Q% w! e# @$ ?; h$ g4 H#include <winsvc.h>
. c! r, h- Z# d7 X
2 Y+ s& C( p2 b& v3 b. b5 E3 L
7 w; d* T- e% J5 k#include <vector>
#include <exception>
* `# p7 T6 J6 E+ t6 u4 N2 f+ H( q#include <functional>



/ t' ]; ~+ z5 v0 h: G/ N) ?
9 b( }- E' Z5 r/ ctypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
- Z+ ^- ?/ k4 M: ]+ Dtypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
. \' D& u4 g6 ]& |2 y2 [typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
( n0 T* z5 R* O: w2 _1 |4 l2 qtypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
! _* |1 u8 x: n3 `' wtypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
7 r/ E8 f) `2 ~( m

" s7 v6 E3 j$ q6 Ttypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
+ ~0 F( Y: H$ Q# Q  PDWORD pdwBestIfIndex
0 n9 s8 y5 R7 L2 @);
: E2 f, N  q5 \$ y7 L

6 `6 o; ^) ~& j1 H/ L9 l4 \typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
5 ^4 ]6 h1 I' h. L8 y& X& F  PULONG pdwSize,
  E& ^" O6 b) S- M# A- M  BOOL bOrder
);
& _! n, }6 K: m2 F4 H0 n- i

# h) ], ^( n* Ntypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
8 T5 N) s0 i) q8 ?7 E) o$ ?" F4 x);


3 Y6 v. X, v+ O* Q+ v3 b' Q. cCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);

1 v. l1 _. k# {
class CUPnPImplWinServ: public CUPnPImpl
5 e# \* d7 H0 Y6 x  Q: _) f{
        friend class CDeviceFinderCallback;
9 L: ^+ }8 f7 R7 q$ o# V; x5 f1 o        friend class CServiceCallback;
// Construction
+ f! N6 \# ^& s3 |) r: tpublic:
; b  u3 [) ~% _3 X) r        virtual ~CUPnPImplWinServ();
  l, i# O8 s/ m1 N& e        CUPnPImplWinServ();
9 W( s) X0 N# ^6 w7 ~1 P- R
: C, g/ ]! }% @$ k4 W4 i. ~
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
% a/ q5 f0 c1 ?' Y        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
2 m% T+ v1 i! G3 w3 l% i        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


. W% @/ j$ y4 \* O0 ]0 G, p8 e        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
5 U1 n7 w0 o5 k% p        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };

# D; V4 l6 u. Q, q( \1 K  W
$ v7 i1 c' @  K0 s$ tprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();
6 E0 U  U" q1 I9 B

        inline bool IsAsyncFindRunning()
        {
/ @1 w0 i5 W4 G. v3 L5 H                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
- p5 l7 Z% Z5 W. r7 h( }% L7 ?" K                }
                MSG msg;
5 `1 k9 J' L( r* A: f                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
. |2 }! N0 V3 b+ d9 k& u. H                return m_bAsyncFindRunning;
- U) w; k  x! b) [+ `% J        }


. n3 c3 p$ u( z* `; B        TRISTATE                        m_bUPnPDeviceConnected;
. s1 K+ r) A- S( \

// Implementation
! l) R1 o, G3 s$ \' k9 e4 L- [$ i        // API functions
( b: b; [6 z( B& {/ U8 O- T        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
6 v9 o' o( `, {. {3 g+ t+ V        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
9 P0 N& I% P4 i, i2 \

. K  r9 L0 F+ d$ P        static FinderPointer CreateFinderInstance();
+ N. N; s2 A# B& O! k' N        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
8 m2 ?6 w% s' i9 s3 k                        CComBSTR deviceName;
; K, d4 j5 G1 L( f                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

# R6 l. {$ |& s( }8 t
, B! X. s8 p4 T5 r                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
1 P1 [& T3 {% g' F8 f" X6 T+ K0 o
7 U" i* L1 K0 b3 R: T2 P7 c% i
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
  x) w& j, j! z- x* q; Y( S% t                }
                CComBSTR m_udn;
, F) ~( v3 f2 `        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
: O3 n% y# I0 W$ m        HRESULT        MapPort(const ServicePointer& service);
1 H. b1 j( D9 Z        void        DeleteExistingPortMappings(ServicePointer pService);
* S2 x( ]1 E) M7 m2 V. [" Z/ [        void        CreatePortMappings(ServicePointer pService);
0 C2 o/ S% f4 f( M5 A# b5 q        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
2 v! h6 [% x& i, u) }2 e" v        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
3 F/ U: I" \- [; o5 X( n        void        StopUPnPService();


        // Utility functions
' _) [  p# X& J+ v5 L' a        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
( L. s) |9 v  y& a        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
3 o7 B; F; h- V$ ^9 a5 w( W3 }        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
# \! {2 @  {8 O        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
% t; x) G3 @( q4 \) m% m        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
+ ?3 ]7 Q) z; v4 P6 u& p2 U        CString        GetLocalRoutableIP(ServicePointer pService);


9 C( d/ ^2 e/ G- C// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
- X1 V/ |& |0 n; u6 y        std::vector< ServicePointer > m_pServices;
+ C+ J0 S- r* r+ `: d- _2 h        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
: N0 t- U# F6 a* U2 W- J. F8 \        ServiceCallback                        m_pServiceCallback;
  I' [+ Z) c" i, X- x0 o, W( e& d
* u  d5 ]: o5 @0 e0 l
        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
9 f# u/ h& g; t- f  i, I        CString m_sExternalIP;
- \8 L% v+ H& c9 I; q% c' C, R+ [        bool        m_bADSL;                // Is the device ADSL?
- }! }/ Z! _% {& J; z' I$ G        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
8 Z. P1 s! Q( w        bool        m_bInited;
        bool        m_bAsyncFindRunning;
, O5 O+ T. C6 m( l5 i        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
5 |- _# g- w- t* n" o% r# C        bool        m_bServiceStartedByEmule;
# o: i6 g* I9 o4 r, V        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
" O$ D: X7 f* o+ R$ n$ B

6 f6 s0 u1 r$ A) X};
0 w5 @5 l8 E& x8 M
$ _5 D2 v- j4 m4 H
; X( ^4 T$ e0 i// DeviceFinder Callback
class CDeviceFinderCallback
1 A1 |; n8 C/ W' [( Y% u        : public IUPnPDeviceFinderCallback
{
public:
$ e0 r& J  I* I6 f9 B6 C* h        CDeviceFinderCallback(CUPnPImplWinServ& instance)
6 f* g& \% Q* I' J1 x. d                : m_instance( instance )
* C9 \! \9 N) x7 Q. a3 x" J        { m_lRefCount = 0; }


: [" s/ n% j3 |& W   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
9 ]. S' j# y9 L1 k- i   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
0 s' }$ W- p5 Q

' p; {4 T1 @' R% E- ^* Z- K// implementation
private:
) e2 L9 G+ y/ x, @5 x, N* Y        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
6 d9 Y4 I' H* p# f: y- U

% o: x+ C4 e4 q2 Qprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
) {; R  B4 V  t, Z. ]& k

// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
{
! Q8 W9 K& i& P; [; G: s1 M4 E3 Lpublic:
7 i) ?3 K5 K( k        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
   
8 J& J) B% I. D* b8 \' e   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
  h, ?1 J# b1 N: p! i( g   STDMETHODIMP_(ULONG) Release();
2 B: z8 M7 n! G8 L+ x

// implementation
private:
! t( j% M' [6 m: Y4 c        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
/ J# P. h$ n* m: s9 |' H$ l# U! T        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
$ v  g  l( @5 o" L. E
/ P" J3 q9 T( _
private:
        CUPnPImplWinServ& m_instance;
; c" r# G9 x; |/ `( o        LONG m_lRefCount;
4 x$ H: u( E6 N/ s% r};
# Y! h9 Z; a7 g
  F6 l: t1 X* x( {+ {9 N) |
% p, [7 `! J  }/ F/////////////////////////////////////////////////

3 C3 H" q3 |: d- l$ P
, R8 K+ |+ L1 \' @! b0 ~使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
0 L: K0 \  u1 |$ h+ C5 j' xCUPnPImpl::StopAsyncFind停止设备查找.
: ?" A+ \. D5 E9 H1 Z8 iCUPnPImpl::DeletePorts删除端口映射.