UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   / D) Y* O# _2 T4 Q( e( X
3. 
   
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   
9. {
   
10. public:
    * M9 q% P  f; m1 j
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    ) g/ A/ t" G( e( W5 v8 ?
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
    ( \7 ?9 s5 i- ^# W8 s
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    ; X! l8 j% w# m( X8 ]/ N8 z% U! j
17. }   UPNPNAT_RETURN;
    0 f6 P, n  R: L- t& ^
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    " d6 n9 `) e; F8 s- P" E" {
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    ; b% k, @! M: V
26. WORD   externalPort; //   Port   mapping   external   port
    8 ]3 l0 E, ]% W( w9 D# \+ t
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. 
    " H# M+ \! j* ?
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    $ R# m+ B( C" L1 d; Y/ T( s
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    1 ^, r- J- i% N
37. 
    7 p" F- t& e/ v8 q
38. CString GetLastError();
    9 {& X  p4 U3 J' O6 E2 k
39. CString GetLocalIPStr();
    ! N1 G! Y9 s+ ^2 p  r$ m0 x
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    # N  t) @0 d; D1 H' Q
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    6 B1 P0 M, \$ x- F
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    ( R8 y' m3 V1 c, P5 M" q
52. 
    + p3 ~1 _( E' p3 m. G8 X  c
53. bool Search(int   version=1);
    
54. bool GetDescription();
    0 [6 V1 N  ?% N, r
55. CString GetProperty(const   CString&   name,   CString&   response);
    ! \2 j! V; G( `4 A. Q% c+ T
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    3 U/ B: y- {: c7 x1 }" k
57. 
    # ~0 n4 q3 J7 A! I' o8 W/ e
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    ; z. s- L+ b6 P3 h+ D
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
    - X+ m/ W" O2 ]
62. CString m_description;
    
63. CString m_baseurl;
    , u1 k5 i0 y2 p. Q$ |; v
64. CString m_controlurl;
    
65. CString m_friendlyname;
    $ I0 E# l" |4 m  h# a; u7 D" u4 l
66. CString m_modelname;
    + y& |1 V8 e& C2 e* g
67. int m_version;
    - g3 ?) P' C+ j+ l+ Y; }! _
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    1 L1 V( h1 g% r
71. 
    # U& b, a/ E' e- @8 w
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    # K2 G; w7 F, E3 T. O- A
76. bool isSearched;
    9 [  a3 _! ~) J( p' w" L; d
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   ' N3 H( g. ?1 S+ W1 I
2. #include   "stdafx.h "
   / I" {7 k$ P9 m$ y
3. 
   ' W% T: T. J  A5 }* P
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   3 Y1 _. y; K4 Q
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   8 y; k* ?7 _) q* |4 G0 V" I/ E
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   4 v# X6 L4 l7 K* L6 {
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   2 ?1 D1 }  ]' b- T9 h6 V  E
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    2 M0 p$ ?0 k: L$ |: Y9 `0 t
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    $ d) Z( Y9 Y( \% U  \+ S( @; Y) \
13. static   const   int UPNPPORT   =   1900;
    ; W: w! x& x/ D7 X9 ]
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    ) L. R$ q) C/ B3 E9 w( F, S
15. 
    * V+ i0 s9 j# q! p4 `
16. const   CString   getString(int   i)
    
17. {
    / a# G5 {% C9 [% E1 I( E
18. CString   s;
    $ c7 d/ P7 B8 c8 D8 `5 _' X( F
19. 
    
20. s.Format(_T( "%d "),   i);
    . p, F. R. R! g2 V! m6 D9 x
21. 
    ! j3 w: y" L' t8 a4 ^$ c0 R
22. return   s;
    : f1 ?8 X) ^" A& Q" Z+ S
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    6 g8 }6 h$ S% l! _
28. }
    # B- m' C6 m+ f2 d
29. 
    * G; Z: J- p+ O, V' i% z* T
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    ) C4 v: F! b# o& L1 t# `7 H" N
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    4 {" q; n" J/ i8 q
33. }
    
34. 
    ' |8 h% l: ], V0 j  l7 [
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    
38. 
    ( e; N3 x) [5 ~. X1 ]
39. const   CStringA   sa(request);
    + j2 C; T9 H- X( G5 A( j. L
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    % T) b4 [5 d, M9 o
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    + d' C9 e  @8 C
47. sockaddr.sin_port   =   htons(port);
    6 Y1 m) M, r! u4 e, @# J
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ' b% G: G& O- ~% A6 i7 ^, w. |
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    . s. ]/ I/ l+ H" F
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    7 j/ F, g4 X, v6 p7 E8 `; G
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    * J8 S" z$ V* a0 `% v/ C* Q
59. if   (!rlen)   return   false;
    
60. 
    # y' @1 {) v* S0 j" r9 g. F  f( U
61. response   =   CString(CStringA(buffer,   rlen));
    7 u* F* d/ a+ B2 L
62. 
    & p; z( S+ m5 N8 l2 X
63. return   true;
    
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    5 E5 K' I; \" w1 }' Z
69. 
    4 n5 Y$ f2 x8 e
70. const   CStringA   sa(request);
    ' ?3 [$ x( m6 j4 r8 ]
71. int   length   =   sa.GetLength();
    8 {  R/ O- U' s/ E( V& F
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    * W8 n3 `/ c2 i8 m# ~4 h/ q) Q' p
74. struct   sockaddr_in   sockaddr;
    " ?' y5 d; B8 X$ Y# ]
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    1 `' ~  x* l1 x  {& \2 U& P* B0 V
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    
86. 
    & E1 B' O0 ^4 X. m0 I( H
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    , p' d0 z1 V9 p) ]6 V+ H
91. 
    , Y, M  q9 I) h# Q$ e
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    $ ?$ c3 ]  B1 H4 f
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    + a% W8 J: W, N3 Q; z* M3 ^
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     . s. [* D6 U9 o$ i" p- U% d) a
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     8 x( s, W* I9 c' r& _
103. CString   property;
     
104. 
     9 e# o/ L; V2 I4 a1 k$ H7 g+ _
105. int   posStart   =   all.Find(startTag);
     ; O; S: ]2 k) r9 Y+ o) _$ q
106. if   (posStart <0)   return   CString();
     / ~+ y3 w, X: m* d
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     * m5 }. F3 u  D1 b5 \0 _0 z
110. 
     ! M- C& z: i7 U$ \2 z* D7 a
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     4 j1 N$ B( b' Z- t0 t' L9 O
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     7 H, B. j) e+ r+ ?" ?* a
116. {
     # R' `- |" P$ I& n# r
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     9 x! M0 ^' Z% w, Y, O$ f
119. }
     2 O8 z: r9 a) h8 W$ ~7 G3 B
120. 
     0 _5 F& W' l6 K8 w7 Z1 _$ ~
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. }
     - [; J5 h0 g' I8 X7 t) T
129. 
     " g7 H% V7 }4 M6 w1 s3 G
130. m_Mappings.RemoveAll();
     # H( ]; e8 X# x
131. }
     
132. 
     1 Y$ d" a8 _! G- x# J
133. 
       s. d8 G$ Q5 J! y& z2 h, z; [! I6 L
134. bool   MyUPnP::InternalSearch(int   version)
     ; L0 w! o) `' E9 b
135. {
     - B  T7 @7 y8 x3 c  }* D
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     3 @4 e& e$ J0 U- }: Q
138. 
     
139. #define   NUMBEROFDEVICES 2
     5 Z# c! _5 f6 j3 a8 v  S
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     + O! K+ f, A( k$ l  S
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     6 V3 f3 y" ~/ `
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     ) J. A" `( t) V7 o5 p2 y# B8 r
147. u_long   lv   =   1;
     / n# `6 c& s) w9 f6 p) C9 x, y
148. ioctlsocket(s,   FIONBIO,   &lv);
     + q& a" o! R& b9 m
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     " {) s1 ]4 D/ Q& ~7 t& J
152. if   (!(i%100))   {
     & k7 R9 H+ B" u' i4 n
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     0 S- M7 a  y5 \) W" d
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     * q5 \- K  i8 t  q5 x0 t( K! v' P9 u2 h5 G
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);
     2 \2 `* L$ K+ B/ M( z9 R* O
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     3 n5 A% [8 W1 |% {% M1 u# M5 Q
159. }
     ( B# u* @3 W5 w9 P: ^* b# B
160. }
     
161. 
       V5 B% Q3 X- v) V
162. Sleep(10);
     - _6 l" o) B. C( n# c7 F# T+ I8 b# c
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     * K( D9 ?9 E( l2 n
167. closesocket(s);
     2 o; Z6 H! L5 r1 S! G% e' }
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     + ^! L2 T# i8 E& V5 Q. J0 ]
170. CString   result;
     3 k# }2 b# X- V' k- n
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     
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);
     ; O4 T9 y, V( h5 S4 D
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;
     8 x7 d% Q: l6 e/ B6 X8 f
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     ; a2 }' U: U. p5 y3 a, s/ C
184. m_description.Trim();
     
185. return   GetDescription();
     * `9 L# L' d1 C" {
186. }
     9 H/ T' p4 I) N3 I, k
187. }
     & \* _/ l7 ^3 {  `) \  w  t
188. }
     
189. }
     
190. }
     3 a3 a# k7 X" Q5 w
191. closesocket(s);
     
192. 
     3 G$ z% s2 {. @& v, D
193. return   false;
       E, C* m, q3 k' V& e
194. }
     " F0 [: g0 T3 c6 h4 j$ Z( _/ {

复制代码

天云上人

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


0 J, `: a5 T+ J2 z/ t3 g; F///////////////////////////////////////////
; b9 {$ H  k' W) f2 Z//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
+ ]3 F. Z+ C! {
. {3 D9 }1 M( ^
3 k6 Q( p* `# _. h#pragma once
#include <exception>
! z" ?7 _; q5 T' B/ X+ `) U/ k6 h0 q
) @+ Q1 R9 u+ s6 |5 B% R+ b
  enum TRISTATE{
' ?; P6 e% W( R/ S$ B1 O$ h; r        TRIS_FALSE,
" d$ H2 |/ \* _! N/ \, e1 `5 u7 J" C        TRIS_UNKNOWN,
7 [, j% w- M1 {! z) B        TRIS_TRUE
% L9 N: o, T! K# P7 n8 K* y};


* o& T: H7 g; d% R! w3 ?* P  Fenum UPNP_IMPLEMENTATION{
" I, }- T# w  `6 X- ]: ?        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
5 Z! \7 o$ J2 K" m. B2 i: W};
9 `$ d) m! F1 b7 f


: u- i6 n$ F" k+ w7 F3 K2 B% E
class CUPnPImpl
{
1 f0 }9 T' u' M4 U1 I6 {public:
/ `) ]: c& d7 b! n8 t' h' T        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
0 T+ [1 t; F3 X$ r8 c        enum {
' b( [4 \- O" w5 i4 D8 U                UPNP_OK,
$ g+ d" u. H& X4 a# t3 E; t5 |: G                UPNP_FAILED,
                UPNP_TIMEOUT
        };

- e# ?$ K! L) t3 E6 T6 w1 x
% v1 q! R; W. D' U        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
$ s  f5 W# Q6 _: |- j1 V        virtual bool        CheckAndRefresh() = 0;
9 j1 b" n) B* L3 |  h9 b        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
5 \1 O4 w( h+ Q2 e! C        virtual bool        IsReady() = 0;
/ e- B/ P+ m; N3 ^9 A* `* \) O3 {        virtual int                GetImplementationID() = 0;
: x; V( Q- q3 Z4 l; s* u       
, p' i/ f8 f' X( g, H        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
) U: v  I; L' t; B) H) Z5 R- P

        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
( M5 z, Q# o6 B; ]) |2 ^! R5 f        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


+ i" |8 P/ V8 ?// Implementation
0 d5 X) |) _& K4 I. r- qprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
2 j) |9 e, ~3 h* G8 j/ w# y        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
: F$ Z" P& ?) M1 s$ j1 }- C+ m# ^) i% @        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;


private:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;


! {! x$ f) F8 W( w};

& ~" k- Q" m. `* `
1 \, \: W5 ^! K# ^  _// Dummy Implementation to be used when no other implementation is available
7 S( q& a  }$ M2 s6 Aclass CUPnPImplNone: public CUPnPImpl
4 C4 |" h, f& z3 u  T- t* U{
! y: l# H& z1 y# ~0 qpublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
) |1 f' h+ B( K! M4 A) l        virtual void        StopAsyncFind()                                                                                        { }
& z1 J' K, F) A3 o) X        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
- E: {: H% T2 \, @* H, z

/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
2 W# j+ r8 O* V7 D( x: X- ?2 i3 M2 F

#pragma once
; V' ?$ {& j+ D; E3 O#pragma warning( disable: 4355 )
) p6 W$ g0 B' R0 ]5 W- r9 W  [0 c: @- u3 Z
7 y' M% E. i% A' T
#include "UPnPImpl.h"
& m8 X" H) F+ G2 J( Z& G#include <upnp.h>
" C. q+ m6 [2 V# F7 g#include <iphlpapi.h>
#include <comdef.h>
+ @6 Q' i7 n1 ~9 D#include <winsvc.h>
2 N+ N$ L9 g, r3 {

#include <vector>
#include <exception>
, {) `; X0 Y; s* C7 @% `0 T0 j9 N#include <functional>

. T  A: p" H' O. o# H
7 ?! I$ F6 H; Z
9 \& w* L) l' D! M9 i
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
0 H3 _& L0 Z7 j% u6 y% |9 ]: gtypedef _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;
5 ^: Z- B+ C" U$ {, Q. Ztypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


typedef DWORD (WINAPI* TGetBestInterface) (
* R0 @6 x2 @0 X, M" Y  IPAddr dwDestAddr,
- Y& S6 I( n2 a  v. w9 ]3 d  PDWORD pdwBestIfIndex
);


typedef DWORD (WINAPI* TGetIpAddrTable) (
2 s) E- h$ v; f  ~; M0 Y  PMIB_IPADDRTABLE pIpAddrTable,
5 Q  Z# N9 v8 ?5 Q  PULONG pdwSize,
  BOOL bOrder
);

% _" t9 a1 `1 U% |4 g
, ?( i4 ^& @, E0 b: ntypedef DWORD (WINAPI* TGetIfEntry) (
3 a1 S. `2 t7 F* ?, Z  PMIB_IFROW pIfRow
);
' g. k% Z2 o7 B- }7 b! a3 r4 G
+ }& I. C1 }; b4 i5 ]8 x: C8 r7 t
* `5 I1 |/ Q, P  V& A1 _0 jCString translateUPnPResult(HRESULT hr);
8 Q5 X: `3 K8 e6 o5 f4 IHRESULT UPnPMessage(HRESULT hr);
* o9 ~! y' Z  C. a$ G' ^8 c. ?7 i1 x1 i

1 e' Y( e$ |2 \; W! m% uclass CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
# x8 ~6 Z6 }) f' ~& i& k        friend class CServiceCallback;
  ^0 L6 L( e) V: ^& x6 Q( O// Construction
( c. z2 N9 M2 {7 M- ], ypublic:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
3 x  H2 C7 f, N, ]# m; d& E
/ ]+ j& G5 d. I' F1 ?3 Z8 v
+ w' O% [' }& R3 V. f8 Z: l; Z        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
/ i  K+ l0 O1 b( A9 _/ [3 d2 Z

        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
- L, [8 f% }' B' b
7 g8 H4 u% B7 X" D: ^* u
protected:
) A8 r/ D9 F" }        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
9 Z* S( z1 u  x        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
* i* n7 \+ M3 r9 s; u" o  U: E        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
. \- `1 Z$ m' o        void        Init();
2 u! s' ?4 @8 O3 p8 l) Z& |2 l9 F
+ O& p. O6 M% u6 c+ T
/ h8 s' _1 ^) t- |- Q        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
- e) l- S$ F" E2 X2 R8 e                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
& ?2 X( a5 C% I8 f! K! X" c                        m_bAsyncFindRunning = false;
                }
9 h% M: j) S3 \1 [$ A  p                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
" L7 q! P9 H/ U' z                }
2 i; A6 D. |- E' }7 C9 E' U- w8 ]+ o                return m_bAsyncFindRunning;
. e; n5 }9 a& V4 C, T& t! m        }
$ T9 d' p0 H* W5 l4 U  I

: f. \" T) \% c! l' w        TRISTATE                        m_bUPnPDeviceConnected;
+ Z% z8 ~+ r* ~4 a1 u
; Y- @9 L! b2 ^  B! }  P# S8 Q7 l
// Implementation
        // API functions
0 j: I: _7 @% M, C, ?        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
) A3 W0 `' D' ~$ \5 a! B        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
1 g$ b6 ^0 \4 [/ s        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
  l) }5 p/ U9 K4 r7 P. m1 x4 ^        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
8 F6 J9 A% L8 p( B" ~        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


( v7 C, @! N( D" `1 r, r5 f' _! K        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
% l9 g% r) U0 l% i! A6 a        TGetIfEntry                                m_pfGetIfEntry;


% r; r/ K! w6 y# s& k" ^! Q( B) a        static FinderPointer CreateFinderInstance();
/ o  J$ r& m- r5 E3 r* j  K- p# E        struct FindDevice : std::unary_function< DevicePointer, bool >
0 u# a% j- i- a! y% F        {
# U0 s+ I6 \8 D5 l                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
8 u5 h/ \9 M6 S; l' M( Z9 O; ~( _                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


# z" P* f' f) r+ v: G/ r( X                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;

1 [* F! l- n) D9 {0 Y0 Z* r
9 ~+ K1 L) S* t$ ?, Q: i! c                        return wcscmp( deviceName.m_str, m_udn ) == 0;
" p, y) v; ]: h6 N- r- j                }
                CComBSTR m_udn;
        };
       
' V' p7 ?: H5 Q9 d* {1 V        void        ProcessAsyncFind(CComBSTR bsSearchType);
. a  e- g& e4 B# x6 I" t        HRESULT        GetDeviceServices(DevicePointer pDevice);
  u, E1 R  G0 {8 o/ V        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
- X9 {7 S6 {) V# ]5 z; J) b        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
/ {" r: H' ?: r" s        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
9 x! O, h9 P7 \0 W, D; R        void        StopUPnPService();


        // Utility functions
- P7 q0 }( u0 I: p! v& W! K( Z4 s% i        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
* G! B7 m1 Y, I        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
; Z3 \, F1 A' \' O% ?  M        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
6 U. t" Z! n! ]9 X2 z3 r        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
. p* x/ D, h  ^: j        CString        GetLocalRoutableIP(ServicePointer pService);

7 x: C7 s0 C! l+ d$ ?* n3 N
// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
3 |" j! ^/ f( a- K        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
. {0 e9 u8 b  Y: U
/ u3 s. m( I" \, v( l9 {6 W
  g2 W$ a" s$ y* k, |. F        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?
        bool        m_bInited;
+ ?* z& K9 q3 [- E( u; B        bool        m_bAsyncFindRunning;
: \0 `0 Y$ K2 ^2 i' `% P& G        HMODULE m_hADVAPI32_DLL;
, x. b* g0 l* \. o* q$ X9 u4 ?# N        HMODULE        m_hIPHLPAPI_DLL;
' T4 N0 T- q( Q# h3 i% f( s, p" e        bool        m_bSecondTry;
0 N. q5 E4 i8 P  o6 }        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;

2 F6 s) n/ A' G% T- z$ {
" \% W: r6 j1 b* ~7 L& g};
4 H- i; b8 D& E2 z! D* k1 M( l) `
: S: i; [( q% K; y
7 W3 M+ \2 u$ z4 e, w8 L' l$ g// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
& k! _- j! i  [$ Apublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }


$ v7 l; @. |; ]" c# P4 t3 W   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
% ^. w1 R9 p4 s0 i' G6 d   STDMETHODIMP_(ULONG) AddRef();
, U$ v5 W* s2 _   STDMETHODIMP_(ULONG) Release();


) P* H7 ?4 L1 O9 I// implementation
private:
3 |0 W+ u1 Y0 K; P) |: d) g9 y4 D& y        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
# x% ]% ]3 u* f7 ], K- G. f# U        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
. Y! Y$ c+ r( z- ]

- H! I# V! Z+ {% Cprivate:
: G6 h; V& @) m        CUPnPImplWinServ& m_instance;
! D/ n% Y: s9 x: @9 `0 g" K        LONG m_lRefCount;
5 A* B. F2 {- g6 w+ s- b, y};


& |& W) {1 @7 T1 c' g: Q7 w, c// Service Callback
0 _: |. m& a: l7 Q- ~% H0 K/ Wclass CServiceCallback
        : public IUPnPServiceCallback
{
public:
! m9 A! Y5 E/ C$ ?$ y$ I4 o        CServiceCallback(CUPnPImplWinServ& instance)
( {2 A% \# P& w                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
5 V; h  C4 {% I" X% `) x: r: H' o   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


! |) H+ |5 s; C! `// implementation
- ?* v& k  X" q, Xprivate:
9 e" A& X0 u1 K, X        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
* [& ~/ i0 \  l# |
/ C$ p& S* k5 t5 B, x
# P( }/ F4 Y. [! m( h  lprivate:
# L& T: M, s4 ]% G1 b' f( N        CUPnPImplWinServ& m_instance;
6 f  z1 |* O. |4 V7 D        LONG m_lRefCount;
7 h9 W3 _6 S) Q; Q/ V* \};
& D& f3 E) T7 ]) J: \' U

& j" Z8 ~- f* y2 l2 g0 J* X& z: i8 U/////////////////////////////////////////////////

4 @* s! h, g) g" C* ^' n
使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
3 U+ `& O8 V; Z" r5 f' ZCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.