UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   ( Z8 X; N- c% |- [1 n
3. 
   2 Y- x, |5 d. S5 j4 {
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   ) Z$ s4 a$ `  Y$ h8 I
7. 
   
8. class   MyUPnP
   
9. {
   
10. public:
    * c% V) F0 S; _) a7 k
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    " h5 S4 {, e6 k; h
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
    # C& ^7 X1 c7 l; F  {
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    ( }5 Q  ^& u4 i2 `! K
18. 
    & b% d# ~- I" m% X1 n4 d: @4 p
19. typedef   enum{
    4 J+ P3 K  V5 ?, b+ x( \* h; @0 t) x
20. UNAT_TCP, //   TCP   Protocol
    , P7 m7 c7 M9 J  p8 R) ^; w- M
21. UNAT_UDP //   UDP   Protocol
    ; a2 @( y% T( `! o  n
22. }   UPNPNAT_PROTOCOL;
    
23. 
    , |; p6 _6 [$ B
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
      o/ M0 ~$ ]! L8 i3 B& e% c, z
28. CString   description; //   Port   mapping   description
    6 V. L! z8 |; ?
29. }   UPNPNAT_MAPPING;
    . j0 i' }5 \) p6 X
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    ! a( ~" E/ u& `: D7 S3 R
33. 
    + e2 \+ A# x2 V1 D& a  ]
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    . f; o7 ?- g- E; p* }" [
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    $ Q! G" Q) M$ c& X
40. WORD GetLocalIP();
    & q2 g+ O1 Y: P! A
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    ' @% q( Z, h& K$ k
44. void InitLocalIP();
    ! O: s) W! _5 A" D# K5 _- U% I
45. void SetLastError(CString   error);
    ' y7 |3 p; c, s) o
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    / ^6 {& |; o4 ]( U% T, V
48.       const   CString&   descri,   const   CString&   type);
    ; U- c# V0 A* P! f" d: \
49. bool   deletePortmap(int   eport,   const   CString&   type);
      `7 b& `* R# ^
50. 
    ) y& h" s/ f+ p; K, d4 Y# X( o% L
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    1 H/ U3 W# G5 E$ _5 H
52. 
    
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    8 l% X8 o  K- D/ y
59. bool InternalSearch(int   version);
    - b* B: s) L3 ^; _/ `1 J" b
60. CString m_devicename;
    
61. CString m_name;
    - g6 `- |% g/ [1 s
62. CString m_description;
    
63. CString m_baseurl;
    ! A8 l' X! ?) n2 v1 ~5 W1 y. x
64. CString m_controlurl;
    $ R2 K0 k1 n1 \/ o9 ]' x
65. CString m_friendlyname;
    + T# i2 Z" |, \$ l, s/ r
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    1 u" h# K% W& q, i" N! W
72. CString m_slocalIP;
    . }! |$ B- \( s, a/ v8 K
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    
77. };
    6 d. k3 w% v5 w$ D& m
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   , _/ Q# C7 n; `  {; J* p) I
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   2 O1 \+ \7 q. j
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
      e3 R" V1 s) g& X! x
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    & e9 A8 G" u" _! W
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
      j  g  Z% y: O  |+ J
15. 
    
16. const   CString   getString(int   i)
    
17. {
    6 Q' t* B+ I0 j& F
18. CString   s;
    & r. f% M; W9 ~* B
19. 
    6 j) {$ W/ v. {4 A2 M0 S
20. s.Format(_T( "%d "),   i);
    / b" z' g6 {9 E% t# L. |, B
21. 
    
22. return   s;
    * z- e! Y/ N( t- [; I
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    : F' x7 e. a' u% W3 [
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    6 G9 w) {- I- i3 s5 b
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    " r2 U6 s7 {8 f' a( R% V
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    ; W( g$ j/ k, x7 @6 d5 K6 ?6 L
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    ( F; H  m! q; c$ Q( L
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    , `* \3 M1 Y- J; Z/ h2 x
46. sockaddr.sin_family   =   AF_INET;
    , n, @, |- ^1 k- r
47. sockaddr.sin_port   =   htons(port);
    5 F4 n; B. M' \
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    7 A9 R) ?; d$ a: M5 t' V, J/ w
50. u_long   lv   =   1;
    4 v4 j( n( p2 R% l) z9 V) w
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    0 y) I! l7 o/ c3 c
55. Sleep(100);
    # v5 ?. }, m8 [; L( h6 _
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    - [; [2 U& j( ]* r; q7 @
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    * T, n6 A" h* q' k2 E
59. if   (!rlen)   return   false;
    ! z0 S! a! ]4 `) }: j9 a1 c
60. 
    9 ^+ }! s2 [/ j$ U. i
61. response   =   CString(CStringA(buffer,   rlen));
    1 ]1 ]3 a' G" h0 p# E
62. 
    # z1 n4 v: S0 Y1 b- h; B% H
63. return   true;
    , T+ ~! B# e& V& |2 g4 @. C" H* v5 I
64. }
    
65. 
    9 G9 l/ r; _% \. [& I. _  `5 L0 o
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    ' S8 X  W4 d& Z+ k0 N1 Z$ t
69. 
    4 W" }# X2 v' x" @; C$ R
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    ' B: Z+ Y& R/ R
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    & I6 }+ z- ], d, E8 u
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    0 d2 A. A; i, f7 I! N
76. sockaddr.sin_family   =   AF_INET;
    " N- Y% g  Q& E6 n+ T2 e- B* [
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    - b1 Q  P& V7 z6 a  I- F, G
82. 
    7 t* l3 o5 B0 U5 V2 i) J; ^0 P
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    ! n" N  w$ Y0 e8 `5 r6 W
86. 
    1 d# q4 }* C/ }% h" ]. I
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    5 `9 d3 }9 ?% B. S  v5 g
92. pos   =   0;
    7 y$ J2 G. o( b& o. n
93. status.Tokenize(_T( "   "),   pos);
    - t- D1 s  c& L6 i, w
94. status   =   status.Tokenize(_T( "   "),   pos);
    % ?3 }( P$ Z9 L+ j7 n. N4 X) h
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    * q2 k; R7 D! o  u' N4 ~
96. return   true;
    
97. }
    
98. 
    7 }# R7 d- F: f: A( u
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    8 B  c7 n- I9 i- `3 F/ `4 f9 v
100. {
       q/ s$ X7 b$ v. @# R9 \
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     ' I/ S! {" }$ k9 N
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     7 p% x: V: m' ^, K) I0 W' K
110. 
     0 ~6 d% z6 x2 c9 ~; Z9 f  T3 o
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     . ^# y2 @% r3 |4 f3 X
112. }
     
113. 
     ; `$ C9 ^# J5 u
114. MyUPnP::MyUPnP()
     & J8 o9 K' I) \$ Y) N0 _
115. :   m_version(1)
     
116. {
     ! K# ~5 T) J( m
117. m_uLocalIP   =   0;
     # H$ Q  ~0 R, c' O; m
118. isSearched   =   false;
     + S; Z# I/ U4 M! Y* n7 ~
119. }
     * m, D5 R+ _9 ~; O/ U) r' h- C# A
120. 
     8 V, o+ J) c$ R& @8 f9 m6 N
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     6 G3 N$ |/ \/ D: v# {- d
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     ( E9 U, g0 N% r# L# y( \0 q8 I
127. RemoveNATPortMapping(search,   false);
     
128. }
     . z/ ]( c' S9 @, N5 o
129. 
     ; {6 n7 e1 b6 f, H
130. m_Mappings.RemoveAll();
     * n) x: M1 ]5 E, I
131. }
     $ U; W. {2 @4 v2 c
132. 
     6 n0 u, Y# _. s2 A" i
133. 
     . ~! u! p* T" v' {) p8 v% }
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     " ?6 F! g* l7 O* g
137. m_version   =   version;
       y! K- h+ [+ X/ w; G8 [4 K
138. 
     4 \5 Y. E& c. X6 F9 O
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     3 `' B7 @  ?$ t* p+ Z
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     $ X1 ^4 B$ y* y% b
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     5 L0 k& M/ s8 R
150. int   rlen   =   0;
     4 I" W3 E# B/ J; k; d
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;
     6 g2 T- z  `$ U- V$ k
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);
     1 R# k! h2 f3 v- O
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     6 r! a  Q: |  {) ^( v
160. }
     6 Q5 O4 k' _* Z0 G# R5 B/ w
161. 
     
162. Sleep(10);
     4 b: j- U  j! F  F2 }& g- e( a* Z
163. 
     # q9 I3 t$ ^, p! f, k  i
164. char   buffer[10240];
     5 U6 t7 ]1 l" z" b
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     $ ^3 `0 g: l8 h1 S
167. closesocket(s);
     
168. 
     6 G1 t2 j& ]1 |0 q. d1 K
169. CString   response   =   CString(CStringA(buffer,   rlen));
     1 u; X. l2 }/ I" `: n
170. CString   result;
     4 [* r( o# H2 v' y
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     $ H* s+ H$ F1 a( s2 P) ~/ U
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     ! q/ Y4 z, {+ I9 C9 M+ }7 x
176. for   (int   pos   =   0;;)   {
     , j' ~8 i6 X" d, m
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     + }# M& R, G  v% k$ T5 u# Y' y( h
178. if   (line.IsEmpty())   return   false;
     6 ?8 E! j4 b0 L3 p+ {/ ]
179. CString   name   =   line.Mid(0,   9);
     ! n* T/ @3 G0 P  s/ I, r) a( P; `
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     3 D+ {0 O) U% F- s. J6 c. f
182. line.Delete(0,   9);
     - C9 `2 n7 W+ J" `' }8 }/ G
183. m_description   =   line;
     * `1 e* Y. B* ^$ d+ G
184. m_description.Trim();
     
185. return   GetDescription();
     6 Z; F3 Z& z# R* o! `. |
186. }
     5 ]8 r0 _# Q  i  d- Q8 h; p
187. }
     ) z2 S+ J5 x% D  a! J
188. }
     
189. }
     ) Z# ]8 r) S* q) ], R
190. }
     
191. closesocket(s);
     
192. 
     6 v( W* z3 F: M
193. return   false;
     
194. }
     & c5 B; ^! o) N: k* r

复制代码

天云上人

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


///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


#pragma once
#include <exception>
3 g. t1 R7 j2 h2 T( u
$ e: [0 Q/ {2 q9 O7 ?
2 y/ U% }' j$ Q: y1 j- Y+ Y7 m% u  enum TRISTATE{
        TRIS_FALSE,
& ?9 O! Z5 d, _9 R3 z1 G        TRIS_UNKNOWN,
        TRIS_TRUE
};
/ s8 n! c8 l& e+ f" l! s1 n6 S
# J3 _: Q3 @% }
: ~2 J* w! z; I# L! L* }& Penum UPNP_IMPLEMENTATION{
: L: u. d8 C% _* @& Y; F% v        UPNP_IMPL_WINDOWSERVICE = 0,
" \0 S* z4 f5 r1 F        UPNP_IMPL_MINIUPNPLIB,
" o; j8 R( y4 w9 H# K1 F        UPNP_IMPL_NONE /*last*/
' S. J  H' P) V: p  W};

4 z1 {# T/ K$ s1 R$ A6 X/ M

, j$ o2 I* z1 [+ D0 a. I
class CUPnPImpl
& }1 e! x& ^* r: H; w{
public:
" o* P& t9 Y  Q/ P7 e; l        CUPnPImpl();
        virtual ~CUPnPImpl();
- z, U3 E' l8 v! y# P        struct UPnPError : std::exception {};
0 P( p( w* X8 ^% z' D5 Q8 D        enum {
                UPNP_OK,
; c( A: ^# h" W                UPNP_FAILED,
% ?6 q8 z3 g1 w, T( K                UPNP_TIMEOUT
* z. C: A. Q& p- [        };


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
, q$ T6 C# e% ~- m) W1 N        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
7 w3 e4 H6 A" z* C5 w$ ]        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
/ i8 A  \5 O3 o. O9 r        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
- F7 f1 x  ~2 O$ H
* n( V2 s. d1 @) Y2 x5 `& A8 y
* D0 v3 u& ], Z! l/ g9 W+ v        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
- ]* Y* W& R5 H: n& W* f/ t2 _: n        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
5 @, T% J! w' j5 _% {: b
1 e$ \# O* m3 x" m6 c
// Implementation
7 V6 _6 `( ^' ^# s9 k: Pprotected:
: [! U1 Z% ]! b        volatile TRISTATE        m_bUPnPPortsForwarded;
' F5 O+ I9 e* M. i8 k9 M( l        void                                SendResultMessage();
+ n; p# X. A2 I) N( H& K        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;


private:
        HWND        m_hResultMessageWindow;
" f; C; K. k3 k! e% P( F        UINT        m_nResultMessageID;
2 x, \, I$ w9 F5 p" ^. q

};
4 Z* C# g- `) z7 f& K3 a

// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
9 a6 N" L6 h9 h3 r{
. Y7 I* ?# b. s9 H( @0 Y$ w3 H7 opublic:
- B; E: \* M+ j        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
+ l& c7 U* `6 R; H* f        virtual bool        CheckAndRefresh()                                                                                { return false; }
9 n  @7 ~- ?, i( v! `5 j        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
* h, g$ \! P. S2 m2 k3 e- X0 R        virtual bool        IsReady()                                                                                                { return false; }
- c) g: m" d$ T2 C& F        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};


/////////////////////////////////////
2 \( N$ ?; I4 Y& n' D( Y8 G//下面是使用windows操作系统自带的UPNP功能的子类
5 P! W3 G8 l% Q, {/ h) z
; f/ L. n% X  P* C: q! W
& }+ z2 L% l  _' |4 E: q#pragma once
0 M7 y! v& E! [#pragma warning( disable: 4355 )

. p  b) y5 f" G. x  i  E; q
! @9 Z( T4 D1 F/ Z  l1 H# O: v#include "UPnPImpl.h"
#include <upnp.h>
- X' u3 O# `- N5 k#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>
; |7 ?) n+ B& d! f- j& S$ Q$ v, ~( p0 O

#include <vector>
2 f: ]$ h( E3 p1 A( B5 @% n#include <exception>
#include <functional>
! f  B; q# s+ z% F

% v) o) ?6 {5 E, ?! p

typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
( h$ j- A, G. A8 gtypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
, D" w, F# W+ \; f8 m9 s0 p6 ltypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
0 K  U. O+ r1 H% g  J7 Ntypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
2 f# y' S$ L+ ~  q5 wtypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
% Q" _7 z4 V+ G) Y: G! ?  o# s( Z
; X8 M2 ^, r1 O4 x/ _
typedef DWORD (WINAPI* TGetBestInterface) (
$ S4 ]/ C. i) P8 x" n' [$ E3 o" I2 v  IPAddr dwDestAddr,
0 h6 H$ z, W; C  PDWORD pdwBestIfIndex
);
: @% z7 m" G& q: R/ h( R' K

6 [! r5 W8 `! _- d# b2 ctypedef DWORD (WINAPI* TGetIpAddrTable) (
9 k4 x* B2 R8 I1 w# x# n; I8 k  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
);


typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);
9 q3 N" q6 C) }3 s5 m, [% Y
" \% m6 q8 u, c4 Y9 \. A7 ^! ~
CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);


# d. J. ^7 p2 ^9 Dclass CUPnPImplWinServ: public CUPnPImpl
6 x  q  v$ B# n  Y, b{
        friend class CDeviceFinderCallback;
: `( I# t1 m0 X        friend class CServiceCallback;
// Construction
public:
        virtual ~CUPnPImplWinServ();
; B4 v. R5 l7 k        CUPnPImplWinServ();


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
6 w# y9 M" c" d7 C, z+ l8 A; C! q        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
& P& J/ U" e& L& a# }, j) z; D        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

4 Z6 C, t! z' H5 r
, g& [* e, `2 J& l        // 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
$ s$ {2 o- W0 @! {5 l. W        virtual bool        CheckAndRefresh()                                                                                { return false; };


protected:
+ V, q% \, F4 x( }5 t# Q8 e: _8 P        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
5 u1 \' F$ S" W5 C$ L" M5 o        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
" s% U+ w! j# h0 K4 t$ s+ J        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
: I/ M: W7 F6 p, r" J- S9 }        void        Init();
8 t! G+ C: Z( B9 ?+ c( W
6 A# S6 c4 `- u1 J7 G7 j
        inline bool IsAsyncFindRunning()
2 t' G% Z# r) I: C9 r7 K        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
0 C' x# y; Q4 D$ f1 B                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
0 N# [, y% I  P# f: b                        m_bAsyncFindRunning = false;
- N# R: V5 I8 j" I* _1 q$ S! ^                }
" o' u, p- s* H( f                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
( i! g* B* N$ c6 g                        TranslateMessage( &msg );
3 e/ v" U- p: \                        DispatchMessage( &msg );
# V6 U' B8 u! e  x) ~& @# z/ ~                }
                return m_bAsyncFindRunning;
( k7 |/ d7 E4 ]        }


        TRISTATE                        m_bUPnPDeviceConnected;


  @' |3 t) _2 H2 V// Implementation
        // API functions
9 K. T% g( I$ \7 e% w$ V        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
4 i9 o6 l0 c- c% l# p6 }        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
: E2 v5 D2 b! X! ?- j: x        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
/ d% e/ W( m/ \% s        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
. X/ Q+ ~; W% s# j+ b# \
! u6 N4 U; ?( b0 ~* }. t" I
        TGetBestInterface                m_pfGetBestInterface;
3 q. a* e  L+ P0 s6 c  P( [; G' w        TGetIpAddrTable                        m_pfGetIpAddrTable;
. a4 M- S6 a8 a: K4 k; \9 e0 B6 a        TGetIfEntry                                m_pfGetIfEntry;
# P9 e( Z, l0 `* _
4 P% p1 C  l% l( |
        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
7 |  @. |: M; y4 s1 M                result_type operator()(argument_type device) const
5 R* M5 U. L/ d. z( r2 [                {
                        CComBSTR deviceName;
0 V8 w& b0 _, U+ C% n. F1 v( F                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


                        if ( FAILED( hr ) )
- {0 E1 Z. w! i! E- J+ J; }0 w) U6 B                                return UPnPMessage( hr ), false;

4 |) w1 N* ]' e+ V
* B/ z- p3 g; ?: M. b, e* [* W+ ]                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
. D' d4 K5 S  n        };
, |& H3 T, R% V2 U       
9 ~% h6 H! i) v# J8 Z2 ?1 S        void        ProcessAsyncFind(CComBSTR bsSearchType);
- X. D" g/ ]0 x+ Y1 X7 O        HRESULT        GetDeviceServices(DevicePointer pDevice);
. D% T9 Z6 l) {( H4 |& V. W8 |2 z        void        StartPortMapping();
* T& b  _# c; o, A: ]        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
& a& X% B4 Q6 w        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
. c/ `" m6 p1 J; S; N                LPCTSTR pszInArgString, CString& strResult);
) L* ?9 D- e4 \+ w+ a) q        void        StopUPnPService();
* y, d/ _( f/ V5 W* c5 Y
: v3 q( y, H+ F  v: z  B: L
' b) Y$ y: i5 {# |* I3 y+ K        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
6 ~) |' j; U* Y0 }" H/ b+ H* Q        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
5 K4 ^, X. A, C, G        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
) m+ J0 g- B' J) z9 f4 v        CString        GetLocalRoutableIP(ServicePointer pService);
4 S! F$ i$ `4 B, K+ u. k4 O1 T

// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
& S' P. o0 P8 k' q9 C# a$ d        FinderPointer                        m_pDeviceFinder;
4 z. B% M: |% D3 o& c' o% k- I/ |# v% I        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;

4 e/ q/ r. W8 B. ?0 j/ X6 G
        LONG        m_nAsyncFindHandle;
0 P. m/ I$ R9 @. v7 \        bool        m_bCOM;
; T- S" N9 ?- ?4 ^  w  Y        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;
! g) `& |2 F! s% P' t# k        bool        m_bAsyncFindRunning;
0 m9 [/ j2 u# O5 I1 V6 _        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
3 e* Q* |) c0 S- e: R        bool        m_bSecondTry;
8 a7 C$ R/ k2 X% ?; {: ^" v        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
: h6 C- [/ U+ m; T        bool        m_bDisableWANPPPSetup;

' B; U0 ~9 _9 q6 |8 ]
+ V% R+ t% r' p, p+ e( [# a8 b};
$ d9 t* K$ k0 C2 x% W

// DeviceFinder Callback
" x" J( U# k" u" u" _5 Uclass CDeviceFinderCallback
7 p: I8 F: }* E. d        : public IUPnPDeviceFinderCallback
& B. L3 q& C1 }3 O. m{
public:
4 Z$ n+ k% K4 N4 `  U+ ~5 u1 n9 b( [        CDeviceFinderCallback(CUPnPImplWinServ& instance)
: N0 T5 W, ^% A4 o. ^                : m_instance( instance )
3 L3 x4 ?% a1 ?; B# J+ Z; }4 l) _        { m_lRefCount = 0; }
1 _& B. a" s* A/ e
( F/ e, {4 V* X/ J4 D: M% [
% v* q0 C, [: M   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
: c* Z! t& ?# i$ ~& a3 C   STDMETHODIMP_(ULONG) AddRef();
9 h: U- t. v# o4 ], F3 Q! e  w   STDMETHODIMP_(ULONG) Release();
& v: l) k5 n' e
" z, \4 \* o) t& u; c  ~0 c+ f
) r: `: m+ a; l: g6 D+ }// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
0 [- Y2 [* U& y) r9 u        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
$ I- g' Z* F; l1 h' b, D. i! f        HRESULT __stdcall SearchComplete(LONG nFindData);
( [. Y8 _% S6 w0 Z" K
/ D/ E0 `8 I+ c4 b, t# c$ X; ~
private:
( Z' I' v: k) p# V# `) Y        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
: f6 v0 p" p* M* D# E7 c};

3 f3 y9 m; z4 \& M8 `, |
7 o) N$ j6 a) P// Service Callback
, v( `8 t, g( j' m# |% D' Lclass CServiceCallback
( \4 U' _/ F8 r9 R1 m; n        : public IUPnPServiceCallback
  W# Y9 o+ k5 F" c! }! i{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
: y+ j$ j% m; W( R4 l9 Z& B" O+ p        { m_lRefCount = 0; }
9 F7 z9 ~, m2 ?8 v3 `   
/ F# q9 Q( X3 ]9 E' n$ S   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
4 G/ O2 {& \5 M9 p2 a( p' W( z   STDMETHODIMP_(ULONG) Release();
$ U& b% v' {' U; f6 W# j& Q
" N# ~; l" u. u$ n
// implementation
private:
+ t; f: j! ?  h" V0 l# B        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
/ x" ~. \4 i/ s; q1 l4 ~        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
: M+ ?/ [8 o1 L: O: {0 Z
: B# z* F: Q% v
private:
- V* H- ^1 o9 ?        CUPnPImplWinServ& m_instance;
8 P, ], W8 c7 y        LONG m_lRefCount;
};

  v2 u9 l5 c2 A% t2 h% J
/////////////////////////////////////////////////

, h  M: C6 `) w+ ]. ?- r4 k
; Q" x" `+ m- w使用时只需要使用抽象类的接口。
4 a1 y4 S- _/ h. u/ z; O% `$ |CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
5 X7 O& B  i2 j" Y0 tCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
/ e, C2 ~3 ]' gCUPnPImpl::StopAsyncFind停止设备查找.
9 Q3 Z' y  u& D! O7 BCUPnPImpl::DeletePorts删除端口映射.