UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   " u2 d1 {- a: a, r2 D1 L
3. 
   
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   & w: e1 }4 z8 L; @$ D* e
8. class   MyUPnP
   
9. {
   : _+ F3 E- c# z! K$ x/ {5 t
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
      w. {9 L# E) \1 l7 ]8 H
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    2 ^- [* m. d% o
18. 
    % Q* h* e$ x- ^) v
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    ) _0 z0 t9 W- }
22. }   UPNPNAT_PROTOCOL;
    / w% V6 k7 \3 y# R2 E
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    0 ]  }0 m# c8 {" Q2 J8 O
26. WORD   externalPort; //   Port   mapping   external   port
    4 R+ C4 O+ y  T+ Y8 A
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    5 T4 ]" c% K, `
29. }   UPNPNAT_MAPPING;
    ' E' f% a  }2 q7 u
30. 
    & k1 s9 I- @* }7 a
31. MyUPnP();
    
32. ~MyUPnP();
    0 m8 H# v) k5 B+ A7 \
33. 
    6 `5 k8 q1 r* m9 @
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    3 W) F) z2 T- Z
36. void   clearNATPortMapping();
    , h, G- }9 Y/ ?  w# y) g
37. 
    % R8 X7 N2 O7 C8 U! E- Z
38. CString GetLastError();
    7 n/ w7 X. q+ T/ X# Y3 q4 a0 Z0 }
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    % d0 W5 A8 J( W$ v2 ]  ^- l) d
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    3 _$ X: |7 u# e' {' [
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    / u5 n, G+ i: X) u  d' Z6 k
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    - G, `8 Z1 c' H, X2 F
50. 
    3 C, B+ w9 l6 K( d. M8 K. @
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    # p& g3 X; w$ E! s- z$ ]
52. 
    
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    / G. o4 j6 ^$ X- J+ S6 b8 _! ~
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);
    
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    4 W( q- C  `* v; c! P; M% O
63. CString m_baseurl;
    + m( y- j: X: Z) h5 Z$ s
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    
68. 
      i. s7 L; w% j( `7 B0 d, Y
69. private:
    ; ~: `- _( V5 X/ T* g2 K
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    - Y+ D! S' w" J( W/ K6 ^1 x/ Q( [
71. 
    
72. CString m_slocalIP;
    9 L) f% `- Q5 C6 p" ?9 d+ n
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    , z' @; q! E  R  z9 J. B; ]: k
76. bool isSearched;
    + E) l2 u* H  O3 ?& @+ X( E. y$ x$ R
77. };
    - |; k/ @; m; C! n
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   / b* M9 `* L# Z4 |
4. #include   "upnp.h "
   
5. 
   9 F8 K8 K7 t6 _$ t# Z( A) w5 A1 N8 G  g
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   ' ~+ w7 o9 J# U
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   % L8 E2 Q' o+ x8 q
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    / Y3 Z, J5 a: [4 M1 h- ?& I8 I: F
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    . l' c2 _* j+ R( z6 m, f
16. const   CString   getString(int   i)
    & I+ {8 e8 J9 x* k& ?/ Q
17. {
    $ b- q. K& q) K% m
18. CString   s;
    4 k: Z& z# F1 u; O9 ?/ a+ _
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    1 W& p; _4 G+ b2 l" l! T' K& Y- P, v# E
23. }
    
24. 
    & q9 [% {5 r) t+ K, L" C# n
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    ! J: ^' R& L  V: d* g: q
26. {
    , i4 E- g  d' @
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    6 D: y1 m) Y" Z: `3 l5 Y
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    : F+ t' H7 I: _3 d7 L, m# a
33. }
    " V* f% E" P/ Z, l2 h/ H4 J  k1 l7 e! ]
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    - s$ E  n: d2 ]5 |5 ^& ]4 O
37. char   buffer[10240];
    * j3 e+ R) \+ i: \5 v0 Z
38. 
    0 M2 m4 w- U8 r7 p8 g
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    / h& E  d5 e/ b5 R+ a
42. 
    6 u: K) h( ]6 d
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    1 L" l' O! x' I
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    # ?4 l# ?" v0 P* t( u
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    5 L  m+ [& G  P: v8 a7 e
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    ( Y" s! E. M, q  Q: M
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    / u5 C- T6 J" B. [6 ~
57. closesocket(s);
    % c* Y  r7 _8 ^0 z( X4 X
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    # G3 T0 d# b% i3 a; U: {% a) w
60. 
    ! ?7 x* m8 N0 S5 G3 |1 ]: U
61. response   =   CString(CStringA(buffer,   rlen));
      Q. N6 z. N% I* _1 R
62. 
    
63. return   true;
    
64. }
    ; w2 t* |9 y9 i; Q
65. 
    7 i  w& y3 f7 g$ L2 j) f
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    8 v9 K2 E( P* }, Y9 A6 V# N, I
67. {
    # K7 U, O) b1 D( y5 R, T
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    : |& A5 M6 S" n( `$ p1 |
73. 
    . ^, G, {9 ]* J( j% Y
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    * G" ~2 B: ^7 ]. o( k" r8 K/ m
77. sockaddr.sin_port   =   htons(port);
      ^5 M! ~9 U8 n/ J3 n
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ! j* _; W# Q5 G$ g/ S
79. 
    0 n' \7 ~( ]  H6 N: I# k; C
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    / M2 W6 _- u5 Q8 H
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    / z7 t, F9 O; @0 i" i. ^
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    : I$ O- l8 T" ^' u! h+ s+ N
88. 
    9 i6 \/ Z4 e1 B4 X  e
89. result   =   response;
    0 L0 T. D: m, K
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    3 R* ~( X: s* U3 l2 v6 e% ^
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    * ?( Y4 a0 M* x# U" q; |8 J4 o
96. return   true;
    
97. }
    # z& u7 T* H) i9 a5 K& T  X5 U
98. 
      G+ V. P3 ?6 \3 [; k
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     ' V* L# H  V$ ?3 i7 e
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     # }5 G- X. B9 ?& c+ Z
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     " H1 u! u7 g$ O( K$ L
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     1 U  Z$ h9 z# S9 {4 m3 V0 J. E
110. 
     , L3 k$ ~' z+ ]
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
       u+ b3 n, p9 G1 v2 Z
114. MyUPnP::MyUPnP()
     " q$ f! h; M& x& v+ @! I% }$ ~) y
115. :   m_version(1)
     
116. {
     ! o, s8 n+ m6 Y
117. m_uLocalIP   =   0;
     ; ?- X  Z3 T9 M
118. isSearched   =   false;
     7 [' A( q( u* e
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     ) s; H& c( q# A6 m+ Q2 ~
122. {
     - d  A% n7 u4 ?3 P9 e% D5 f
123. UPNPNAT_MAPPING   search;
     4 V3 Z! p7 a" n4 a, h% b
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     0 z1 r5 A  q8 K0 d, p, P
126. search   =   m_Mappings.GetNext(pos);
     , B" N- u3 l4 q/ n7 z
127. RemoveNATPortMapping(search,   false);
     1 o6 \" P5 X. b6 @
128. }
     
129. 
     0 v; D! O; u. q8 u
130. m_Mappings.RemoveAll();
     0 R9 m- Y7 R+ S* D4 Y( }
131. }
     
132. 
     
133. 
     5 W) u( K6 `4 J* m8 Y
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     ; D; y/ H4 q, c- _- K
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     : o: r) o; }' B3 \* L" p
142. {UPNPPORTMAP0,   _T( "service ")},
     % C* F( t, k! c$ e1 E
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     7 M, K- |, a3 |. P3 ], L- g. Z% n
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     2 \  W# h! X; L/ c3 W1 N# Z
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;
     # c4 j9 K1 ^0 ~
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 "),
     ! ]6 U; ~9 y- l7 W# D; ^1 }
157. 6,   m_name);
     0 a: v4 p, Q* _" g1 ^
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     . e/ ?% i+ c, B: i) ~) m0 e
160. }
     7 _/ J. ~# F) H# v+ G
161. 
     
162. Sleep(10);
     
163. 
     8 Z1 g% U1 B) `) f( l
164. char   buffer[10240];
       p* }5 K- q: w! ]+ V2 W
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     , B# y8 W9 `- K& W  C6 L
167. closesocket(s);
       w9 a6 C/ P. t- H) j0 i0 |& g' `
168. 
     ' G: r5 H3 Y! I9 {& f$ H
169. CString   response   =   CString(CStringA(buffer,   rlen));
     5 G) K3 s# B$ w  Y$ @
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     * H7 G0 _- {8 L- m4 m
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     ; M/ @. }9 a1 I. D+ H( w5 V/ o: r
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     4 G) l7 ~2 @' {3 f4 W% ]
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 [- H+ D# q* U/ P
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     , S9 X( j# w; F# \- O
182. line.Delete(0,   9);
     
183. m_description   =   line;
     ; h. F3 V) R1 H8 a( t& v. o) `
184. m_description.Trim();
     1 |3 w& L% n/ Q! L$ X
185. return   GetDescription();
     
186. }
     
187. }
     & E. Z  f' S' N9 k) K
188. }
     5 Q. L9 d# j( f/ `
189. }
     
190. }
     
191. closesocket(s);
     
192. 
     
193. return   false;
     : a6 L! m2 _+ m; H% l
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
' @4 \  A0 b5 p  F# d$ K: y! S' g

9 U. ~$ Y4 J: l% ~///////////////////////////////////////////
) v6 |( \  G) Q/ g//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
2 f' M& G9 C- g$ I( C) i5 I3 {; s
0 t: {4 P0 Z% o* D/ r) f
#pragma once
6 H, ^# x( N7 m' F) s8 x9 }#include <exception>
% Y- F& S# O1 E7 _9 R
$ j+ ^; c# M8 Y! [, @
  enum TRISTATE{
  |8 E: h& S- k( r8 X$ S        TRIS_FALSE,
        TRIS_UNKNOWN,
6 c+ @" Z/ `2 m5 w! F        TRIS_TRUE
};


5 L0 W! U# H# [/ Eenum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
+ Q& w3 X% I+ Z0 j+ M* z  Y$ _% w        UPNP_IMPL_MINIUPNPLIB,
7 g+ T& I7 J" [7 u" a        UPNP_IMPL_NONE /*last*/
};




# j6 g" q8 ]8 D6 R6 E6 t0 ?/ bclass CUPnPImpl
5 q$ O4 n2 _) w7 L- ~{
public:
        CUPnPImpl();
$ o. [$ p" n5 R# a        virtual ~CUPnPImpl();
. |% P) d8 q9 r) ?6 y        struct UPnPError : std::exception {};
        enum {
' D  B2 t4 `8 F                UPNP_OK,
4 N( N& r) s$ Q4 m5 }                UPNP_FAILED,
! o" @. d0 V/ i9 k1 @$ P. _9 r, Q                UPNP_TIMEOUT
; V" h3 p) L: ]( _- `. r" Z        };
# t  S* C. a3 B- v: e3 @* e

! R3 ]$ v6 @( M2 {# Y7 q        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
" {* C/ R' l/ ]/ ?+ ~' x9 @        virtual void        StopAsyncFind() = 0;
/ |% a; K/ A0 K, ]8 h        virtual void        DeletePorts() = 0;
% {6 Y6 F9 x* t0 F        virtual bool        IsReady() = 0;
  j$ j& F. k" j5 e3 a* e6 K" @  u        virtual int                GetImplementationID() = 0;
       
! l4 a" O6 L& j# \8 I" ^        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

# Y6 L& S8 @! x! V2 ]
1 I3 ]$ W6 ?9 p6 k        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

$ q8 U. B0 D& x: x7 i2 D& J; d7 u
// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
9 |- q0 ^2 v  |        void                                SendResultMessage();
. z: L$ A" `# s2 Z7 ?2 Y        uint16                                m_nUDPPort;
9 |; l9 l( A1 C; u        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;


private:
        HWND        m_hResultMessageWindow;
/ x# `/ `9 K& Q0 ~5 \. ^- x0 K        UINT        m_nResultMessageID;
2 n7 I0 `5 g" V$ W4 T- Q: X0 a

};


% [* s5 S, p% R5 ~9 c4 m// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
& \% p! D/ _7 F{
& Z4 c* u, Y- bpublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
( }3 X/ w6 D% z        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
: l; s" @7 n" S1 m) @5 }        virtual void        DeletePorts()                                                                                        { }
0 g- p5 ?- ?/ i        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
9 G& Q8 f3 Z7 Y3 Z% \( }1 ?
' I9 u6 ^1 v# H- k: |
( }7 ^' ?0 w7 P% |' x3 @/////////////////////////////////////
, u4 g/ H- [  |9 ~//下面是使用windows操作系统自带的UPNP功能的子类
: U1 H' X1 e0 ~: o* B( X
* F4 V0 r! a- W# |) Q
2 k, g0 b4 M# C: w& t- X0 B7 r9 P4 Z#pragma once
#pragma warning( disable: 4355 )


9 V4 C. `# s8 [- v; E+ }4 d#include "UPnPImpl.h"
2 Z# E3 k4 }, W2 X#include <upnp.h>
2 r0 l3 }3 ?% f- @2 y#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>


#include <vector>
#include <exception>
#include <functional>




3 R7 U. w1 ]0 l4 J& C4 jtypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
( h" ]7 C3 K2 i/ }, ?, u8 K5 Btypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _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;
" g7 c, `- P- z3 t8 f9 T  ^, ctypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


& _3 d; v' h+ p& C8 a* jtypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);
& U. ]2 L  g9 [; ?

typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  }- F9 W% z& B) \. E% a  PULONG pdwSize,
; W  j5 J1 K0 K$ `6 N/ f  BOOL bOrder
);
4 I- X5 M. i8 J  K: v5 x( e
2 y: `7 t* r5 d3 t
typedef DWORD (WINAPI* TGetIfEntry) (
2 E& U) u6 s# a2 p- N$ Y1 ?  PMIB_IFROW pIfRow
);
$ R) D- J# F  w  R: i  o- g' H* ^

CString translateUPnPResult(HRESULT hr);
$ P7 p9 j8 h- U2 l/ B2 H  ]1 T/ @, gHRESULT UPnPMessage(HRESULT hr);

6 [7 |4 ~% ]5 s2 `0 x2 S
class CUPnPImplWinServ: public CUPnPImpl
5 V1 m5 U) M+ N" q/ Z{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
# E0 Y2 E8 X" ^1 X5 b% ?" P// Construction
$ g0 ~  M# v- R- E3 Ppublic:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();

$ V( ?# U( [" O. u4 X# w% E
1 ~% _( p% L4 o$ B. ?8 f- C  k        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
' m+ d4 i: L1 X$ S# m) y. s/ [        virtual void        StopAsyncFind();
: i6 j! o5 |. c7 J& j& ^2 x% I4 R        virtual void        DeletePorts();
        virtual bool        IsReady();
9 u, J$ G( z4 r$ J0 s5 F* B  y        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

- v) F/ n$ d( @
! e; w- t2 r& ]+ Y        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
& c5 m$ g$ j, w; r# P        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
' P6 b7 u$ F' L5 c8 [! M        virtual bool        CheckAndRefresh()                                                                                { return false; };


' o% \& C! M$ _protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
5 D- J6 z0 y* r8 ?, F4 x        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
2 }6 F! q7 i9 N) F/ @) B        void        RemoveDevice(CComBSTR bsUDN);
: s7 P& G9 Z1 a" W2 n$ N        bool        OnSearchComplete();
; t" b( {" w% G) U        void        Init();


* v& g' c1 N0 {; g9 B& J5 o        inline bool IsAsyncFindRunning()
        {
% `' A( V/ G5 v9 V                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
: _% K7 `4 ^) g# B2 Q5 _& p* p                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
                MSG msg;
% D# H8 J1 w6 m+ g3 C                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
1 L) |4 a" m: y6 P                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
: u/ f4 \5 b/ c. r                }
                return m_bAsyncFindRunning;
) U3 H3 ?8 Y9 ]/ C        }


        TRISTATE                        m_bUPnPDeviceConnected;
8 C( H! E+ Z  B" }) s7 t
3 O$ s3 E- t$ f7 Z
// Implementation
        // API functions
& i: g0 P# A2 l& ?& w1 P; a        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
! Z6 t2 Q" L7 [1 B        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        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);


* ?% X2 C" `' c* G+ q        TGetBestInterface                m_pfGetBestInterface;
3 K! y) V% C  j        TGetIpAddrTable                        m_pfGetIpAddrTable;
+ T3 E" u" q0 Z. {, i# b7 s% }        TGetIfEntry                                m_pfGetIfEntry;
" \, f+ Y9 k/ H* ~' l9 O+ [3 O& c
5 \, s  ~5 {9 q- j/ N/ g2 m6 M
        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
& G7 E9 _# n# n6 I1 `% x, N. |# _                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
6 a: Y7 T- z7 \4 h8 ]" X# y1 U! f                        CComBSTR deviceName;
, _' P3 U) c/ K9 f7 u0 a                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

: H$ S' C+ Z9 z; E1 x% s
                        if ( FAILED( hr ) )
6 ]% ?6 k% C3 V# R5 E1 {                                return UPnPMessage( hr ), false;


9 j' J1 ]+ G, g0 N1 m: @; ?                        return wcscmp( deviceName.m_str, m_udn ) == 0;
" S, t/ o. o- U0 k1 B/ A5 o1 H                }
0 n/ ?: L% [5 L& J# ]* n# Q                CComBSTR m_udn;
. S" @- x& f+ q* q. H/ Y        };
       
/ c0 F, L) B; c/ P        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
" J) f6 m. C. K* Y        void        StartPortMapping();
& N7 p7 p) ], `- R4 r9 C2 |        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
  M9 p$ q, E; C" X" R( B- `                LPCTSTR pszInArgString, CString& strResult);
* _0 K" u9 j( e; q        void        StopUPnPService();
/ A8 k, U4 ?; h5 s; s

' Q! U9 t+ I) H$ [% q, K" j; U        // Utility functions
: j+ G) N5 v0 `  M' u% U. D        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
: W, j( E: G' ^5 N; }) s        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
# l1 R3 s3 b1 [7 `8 N- w        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
6 `: m' B/ n* b0 r        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);

% G. O( F) [! ~4 l8 Q) l  X
8 A; i1 a6 ]$ D+ M9 D// Private members
& U) L! Q1 r" S6 s& ]0 Jprivate:
8 L& e1 Y6 X6 e% R  y) K        DWORD        m_tLastEvent;        // When the last event was received?
3 ]8 e+ \$ t9 _7 e/ S& m' O        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
& y" }8 M) j. j' v8 I        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
* P2 ~+ R3 o$ k5 G        ServiceCallback                        m_pServiceCallback;
) {9 Y! z% Z& V1 o+ E0 e) `; N
2 f* ?7 @3 Y" ~$ b+ {9 H9 Q
        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
6 ~' j% @% b' w. H5 V2 S3 X  U+ y        bool        m_bPortIsFree;
; l- |/ ~1 S, d* U# g4 `4 r5 h5 X) @) Y        CString m_sLocalIP;
        CString m_sExternalIP;
, U3 b$ e5 N0 n$ q2 |8 v        bool        m_bADSL;                // Is the device ADSL?
4 {; Y: T  }! r! \6 H" y$ ~) m6 {        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
8 N4 k* N7 K0 l9 A: Q3 Q8 K. M        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
- ~% Q, Y- l7 N) \# _$ n* J        HMODULE        m_hIPHLPAPI_DLL;
4 n, z- s% w0 c! h1 K        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
- ~; y- Q7 }6 A: s7 V+ j' f3 s: H( `        bool        m_bDisableWANPPPSetup;
5 y! B9 y* H% d8 N$ z
0 I) e" L1 ^: _, r( A2 B# q
};


// DeviceFinder Callback
/ y# e' |7 a' w5 w4 `* @class CDeviceFinderCallback
- Z' X  M1 E! J; {2 c8 U! A+ z8 w" s' q        : public IUPnPDeviceFinderCallback
{
' E1 [3 d6 F1 d4 W! @public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
" w6 U6 Z9 l( S% V" \. p        { m_lRefCount = 0; }

/ k+ O: D& A  Z. y! n! t. I1 N
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
( m$ r0 {  i3 ?' I   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

- L- w: A7 o# P/ ^0 w% B# C8 q, e
/ @0 [9 |7 w$ a/ y3 J// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
* Z* r4 W# U# `# a/ }5 U/ `0 _        HRESULT __stdcall SearchComplete(LONG nFindData);

3 A( G- @1 a/ x  y. i1 P4 ?
* M/ s' ]- T7 F( ^! z: j& Q8 aprivate:
        CUPnPImplWinServ& m_instance;
: a$ D6 ^. @5 @$ z, B7 o, n; @        LONG m_lRefCount;
};


// Service Callback
class CServiceCallback
# [. Z$ ?2 u. n( t3 n' K% F        : public IUPnPServiceCallback
5 B( b2 u( r/ }6 H1 O* D{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
  E6 ]$ g; Z/ b% L9 U   STDMETHODIMP_(ULONG) AddRef();
5 M8 R' [# ^" A  B: ?$ ?& {  H   STDMETHODIMP_(ULONG) Release();

3 H7 V# m* x# T8 h) d- X
8 f# ~; V9 l; A, g// implementation
private:
0 I* e6 D, |/ S1 }6 y0 F; U        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
2 `+ ^+ P2 g8 |& f; U) P. ~

7 ?" b" H! }; @6 t6 G- t4 a" Cprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
) y/ P, ?8 B- a};
# }4 x' H; k* U9 l

/////////////////////////////////////////////////


, J; I% {0 k5 @: P: {$ L' a使用时只需要使用抽象类的接口。
  |( t( U5 t# A+ Q1 i2 e! _CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
. q; t+ _) [9 [1 FCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
; R5 N, b' I0 l" PCUPnPImpl::DeletePorts删除端口映射.