UPnP

天云上人

/*uPnP.h*/

1. 
   ! i) Q2 h1 _9 G) G* a5 M
2. #ifndef   MYUPNP_H_
   % h2 M( R3 Q5 {
3. 
   
4. #pragma   once
   
5. 
   ; c$ m  }/ ]; X
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   
9. {
   
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    # C7 C1 U3 [/ s9 M, r& h
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
      a1 q: X+ `1 {7 C
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    + |, H& Y/ h. u5 L
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
      R" \/ t. g9 [0 Y4 r( B5 O
17. }   UPNPNAT_RETURN;
    
18. 
    $ z5 h- H- W5 z$ p9 D6 c8 f0 b6 T
19. typedef   enum{
    ; [* J( o; [* }4 Y3 s
20. UNAT_TCP, //   TCP   Protocol
    ! r9 E) c2 c7 _8 H9 b; O
21. UNAT_UDP //   UDP   Protocol
    ' x, m5 b& T0 G: P
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
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)
    
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    5 C7 ^1 l3 a8 u# l4 R1 G" E
31. MyUPnP();
    5 l. L. t' g: q* p6 L0 z+ q
32. ~MyUPnP();
    1 n/ z. c5 z8 L2 U3 |& M
33. 
    & }" }3 N7 V+ |5 G9 {9 g$ |/ w' K8 W
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    . Z+ p, L  }1 D; @4 S' u
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    3 _. H! \/ ^& ^* Z5 [" \/ r
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    . c4 n8 Z$ S) \
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    ; l" \, B. F9 {$ L
43. protected:
    & D9 H1 h6 B+ R; [
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    # {2 ^1 @. n, ?% |5 W4 {
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    # w/ [% I# i7 f) _
48.       const   CString&   descri,   const   CString&   type);
    8 M# l& G4 @& @6 H* R
49. bool   deletePortmap(int   eport,   const   CString&   type);
    & ~4 r. R: ]1 P' ?9 P1 d
50. 
    " N7 p9 e6 d( N6 M
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    * F2 `- [7 L9 ?2 U4 ~1 @9 k
52. 
    
53. bool Search(int   version=1);
    6 I$ ]$ W9 y$ `& s
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    % E" Q( m! Q3 ^8 W
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    $ Z  T7 c3 X4 a9 t$ t3 k8 j
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    8 }1 e$ w1 A, e
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    1 A) \" y) ^) K  n. ]! W
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    * ^% N- x: T. D( h& w0 I
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    5 j' ]' V/ z" G
74. WORD m_uLocalIP;
    $ b4 @/ K6 R8 |3 l' C0 G
75. 
    
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   0 l# t% a& f; n4 e. V8 ]0 w
4. #include   "upnp.h "
   . s3 |  N* j6 r: C7 ^7 X
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   ( \+ {4 a& v4 j
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   " j; n+ W1 w4 C) i& V& ~
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    8 J, @& S+ B7 M1 y0 x8 G
11. 
    ; a% ?4 x+ K6 R/ V" G  Z/ y5 ]
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    5 C' w& X% T; \. d7 b3 n% c
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    . n  b) t) s, o3 M2 P6 m2 [
15. 
    
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    
19. 
    # D. Z* [" L) v1 d' r
20. s.Format(_T( "%d "),   i);
    
21. 
    ) k2 m. n, N  K* D  [3 Q/ l
22. return   s;
    $ ~* W- X4 ~& `& O4 l, [2 ?
23. }
      r- B, ~) h( T  x4 a7 x8 W
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    ! o2 q* g) n8 w6 I$ P( ]
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    % t6 `2 n; Y, g! f+ R
28. }
    
29. 
    : Q% w' L9 q, h( S
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    ! M0 O% ]( |& C& W
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    ( `8 i7 R# J/ M% s  E1 x) g: s
38. 
    
39. const   CStringA   sa(request);
    4 i; R( M+ Q$ e. P" \+ f
40. int   length   =   sa.GetLength();
    " a, x. W9 j) L4 n4 x; [3 R
41. strcpy(buffer,   (const   char*)sa);
      j8 Q! A; J/ E8 U* |& ?
42. 
    + c7 G6 d, N0 G! Z3 |
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    0 z. C8 q" q# e' k0 |
47. sockaddr.sin_port   =   htons(port);
    . T+ S/ |% K1 x+ U. h2 D  B3 `) S
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    5 g# W, p/ O2 v: L$ D
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));
    0 g8 H& S2 T9 k: f1 T$ u
53. Sleep(20);
    $ |* k# L! D" w1 ~
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    ' Z. M7 C. F+ p
57. closesocket(s);
    & ^; j" U0 z3 k2 L' l
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    " h" R" P; Y% m
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    5 Q' w, s5 V& p4 w: V
63. return   true;
    3 d" E& z  t  ?( g2 F) n& @6 T
64. }
    3 ]7 M  ^: P$ l% O- w
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    # J  x/ z3 f# J
67. {
    " P) w9 L  h: O
68. char   buffer[10240];
    $ a3 h/ Y& `9 a* |7 v# Q
69. 
    
70. const   CStringA   sa(request);
      j  }( U. X, H9 s  k3 q# D
71. int   length   =   sa.GetLength();
    + G) o8 g6 p7 Y1 c. Y( g: s
72. strcpy(buffer,   (const   char*)sa);
    : L8 l  |/ s4 }- e+ v
73. 
    7 j* h4 @7 k% E+ B5 J
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    & `& i6 E7 E. q0 g! l7 T7 B# f
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    . ^0 ]( |3 T1 F, m; C; H
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    ) Q4 d+ d) C& e# k3 f* Y
84. {
    
85. int   pos   =   0;
    1 v) v7 {- u  j4 k0 e! \
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    ) I9 b9 o1 K3 G9 |
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    , [8 b! A; s4 K7 Q; w0 E
91. 
    
92. pos   =   0;
    , I8 D$ b- h! B# O
93. status.Tokenize(_T( "   "),   pos);
    5 ]( j# l% f9 N* I
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    , j3 z( a2 d* H: Q" v0 ~6 F
97. }
    
98. 
    5 l) U  L4 P" D5 `  ~. q
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     : @$ ^' r/ o! H/ _5 @0 W
101. CString   startTag   =   ' < '   +   name   +   '> ';
     8 S6 d. `/ M/ {/ ~! }
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     2 d0 V3 H* ^+ b6 A+ h
104. 
     2 T. ^" E7 N" d( j2 u6 Y
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
       h9 d: l1 t% }& q& @1 y
107. 
     & ^& G3 v  u( c. {7 F( c, d4 B+ n
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     , E, L9 e- O* A5 |8 K2 H
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     1 \; f' R, k, m* I1 `
112. }
     * D/ a( _- s; t) d4 v
113. 
     . t6 a2 q1 Q1 s! {& z  k- `
114. MyUPnP::MyUPnP()
     0 C' `1 `2 C0 d
115. :   m_version(1)
     - r  M$ ~. v: @# c. @
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     / u0 R3 g. j) K# `% j) r
119. }
     % D5 u& m8 n9 ], v4 O6 m
120. 
     
121. MyUPnP::~MyUPnP()
     , P8 T. I4 @* T& v( f
122. {
     0 }9 K$ w, ^) e, z
123. UPNPNAT_MAPPING   search;
     ' J" ~: J5 k! u0 Q* s7 c( l  T
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     7 x, W( {& K$ H! M; \- H
125. while(pos){
     ; \- G  t0 ?+ v- k7 m2 a
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     - X' @) v8 i- C
129. 
     * D: S* ]+ h! s
130. m_Mappings.RemoveAll();
     / G4 Y2 [3 i4 @: H% e  U) L4 U. I
131. }
     
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     + }, b6 @; |8 o& [' Z! s
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     ' v4 p% {8 b- U
140. CString   devices[][2]   =   {
     8 }1 {9 w& y# ^$ C; S$ G4 l
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     ; h' x+ ~6 I9 m4 n; i2 a$ |
145. 
     , ~0 P2 R$ E0 F- L! Z
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     % c6 \7 [5 y( m  S  L: X
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     / a8 O2 |( ^' i( b
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     9 @  ^9 B# G) o2 @1 D5 B6 i
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     & A) M1 w& P/ l
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     / z- f, S, h( B9 n
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);
     " C! V: E, O" E) @
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     # [1 ~' J% Q& D
159. }
     
160. }
     
161. 
     
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     * C% B/ W3 k" p. j/ h
166. if   (rlen   <=   0)   continue;
     * J- _0 m6 i( l" G4 [
167. closesocket(s);
     . E8 w# I- h8 r! n, _& x2 N
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     : g6 |5 q, K) O: c9 U
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     . q4 _4 g" h5 ^) S6 k; G1 B  l
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);
     : z2 M5 S7 j% I( f2 Y4 i
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     - p% p4 \* @) C- L
178. if   (line.IsEmpty())   return   false;
     - x, n  ?* `6 {  ]9 ]6 ~- g
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     7 p& M% w; h' X+ d; p  l* k9 j
185. return   GetDescription();
     
186. }
     
187. }
     
188. }
     3 y' d1 ]3 T3 Z9 p" M
189. }
     + M  A2 B; j' D4 w4 S
190. }
     $ y% P3 Y! W" \
191. closesocket(s);
     
192. 
     
193. return   false;
     
194. }
     4 p, R+ B9 q% _& s$ G! F, x

复制代码

天云上人

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


///////////////////////////////////////////
, }3 X5 O% k- E; P9 \8 I0 c: ^8 |$ n# Q//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
/ u- e# d3 h: o' c8 F
$ K+ y# H% Q! E; m# B
#pragma once
#include <exception>
* d4 W& H8 A% M9 b, G  l; C) a  m
) n! ?3 o5 v+ k$ J& o3 I0 {0 C
# t, b, k3 T9 s5 G' Y; o  enum TRISTATE{
3 W: e3 U% @( L        TRIS_FALSE,
4 }9 \# ?6 u( W' O, \        TRIS_UNKNOWN,
# P  j& V5 ]8 }+ H( b2 D6 Z        TRIS_TRUE
5 Z5 g6 [; K6 {1 N- K};
& e3 d) r4 N2 J
# M  m' }1 k; S3 [
enum UPNP_IMPLEMENTATION{
0 d0 O% c0 s( k+ V" |+ D1 S$ O8 q2 Z        UPNP_IMPL_WINDOWSERVICE = 0,
9 c* m7 g  k5 x3 b& \% Q        UPNP_IMPL_MINIUPNPLIB,
( V" M7 v0 z, G6 N        UPNP_IMPL_NONE /*last*/
+ ^* o5 E  Y! Y" A3 s4 I; n  P" u};
0 V" v% o; q  @$ H! I/ X: B6 r

! L3 Y" K% h8 i) _! e9 p' J" w

+ G7 z! t& L7 J2 Z# h% Xclass CUPnPImpl
; ^! h0 _$ z+ o& r& r6 k& o{
public:
        CUPnPImpl();
. N" u* D# P% C9 Z  S* t        virtual ~CUPnPImpl();
1 K5 {, h- K- h4 X/ |7 m        struct UPnPError : std::exception {};
        enum {
3 j+ l. S# A# i                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
5 V. Y4 k/ ^6 w: y+ G        };
7 N0 t% M2 Q. f+ E$ O+ e3 e) U
" i, X/ h+ s6 T* X0 U( w
) ?3 B( D$ @4 g* R$ I/ c% o/ t        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
9 z0 o# P# F6 T. I6 {        virtual bool        CheckAndRefresh() = 0;
" Q! u$ c. V( E3 O- k1 c        virtual void        StopAsyncFind() = 0;
: ~% M; l, k1 N9 N( b        virtual void        DeletePorts() = 0;
4 F. H, h: a  ]4 K- S% |3 O        virtual bool        IsReady() = 0;
5 g1 P& B. C* ?        virtual int                GetImplementationID() = 0;
       
' r1 ^. `( c  B) f+ S& u        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
8 v3 G! R: ]9 F6 o

/ L7 n5 ^' N) S3 [        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
- I2 R& L0 G2 q8 k+ @        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


. w# R4 w6 ~" j5 n6 ^0 j: M. V// Implementation
7 D5 \& N' Y4 E  Aprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
# J" E( I) P/ Y; k; S        void                                SendResultMessage();
        uint16                                m_nUDPPort;
  s% l7 M8 K, [5 l        uint16                                m_nTCPPort;
4 m! y$ ?0 Z5 v4 |/ B        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;

0 \' F# B5 o6 u+ _8 w/ j  A
' T$ J! I3 ^2 W! a3 qprivate:
1 v1 |4 i0 p; b% O/ h        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;

; z) }- }* Z6 m4 _6 R3 T1 c, H
};


// Dummy Implementation to be used when no other implementation is available
& ]! s- f7 ]; h% B$ a1 Yclass CUPnPImplNone: public CUPnPImpl
. m! F5 Y! D; \# D  E- @{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
/ a: d7 S+ z" d: A. L2 s1 e        virtual bool        CheckAndRefresh()                                                                                { return false; }
9 O) F  H2 l* t- L+ x: {5 i; T        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
9 s' y+ b5 \- G8 a5 D0 ~8 N% C        virtual bool        IsReady()                                                                                                { return false; }
6 s9 ~' I: P( @* X, l! d" I3 ^  y5 S        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
0 U$ P. L" p6 P! ?/ s};


7 T+ g* R- [7 L9 ?+ \/////////////////////////////////////
- V7 ?' g* N% \( Z+ K//下面是使用windows操作系统自带的UPNP功能的子类


0 d9 ^6 E  ~; k& ~' G6 z#pragma once
. Q6 Q7 G9 t( B#pragma warning( disable: 4355 )

& h: v7 m2 j3 Z& J/ b2 }1 R, J
: R7 W2 d- y/ i  e#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
( n( q+ w& x, z#include <comdef.h>
( R3 H) B, M) N0 s; ?- s#include <winsvc.h>


/ a3 ]! Y% R) _" w" F#include <vector>
9 ^1 H4 z1 [% {* a* R3 E0 E#include <exception>
#include <functional>
: ^. h+ V+ F& `+ ~- [) f
7 B* T+ E  r0 b3 B: a
5 S3 u, N1 s% f0 _

5 d1 q7 t7 V# j. g3 d2 @typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
$ l# |0 z/ K" z5 s" h  C( }typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
) t8 n! J1 }4 Rtypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
& e$ u$ z& J8 `" t. P) mtypedef        _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;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


( [! n$ B4 _# ]% Y; l2 n3 Utypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
1 ~  d; b4 B+ \9 g$ l1 t' `);
% @, L) I5 E, t
5 g' G( Z* m1 R. Y, z& ]" V: ]
typedef DWORD (WINAPI* TGetIpAddrTable) (
+ ~+ O7 H! j5 O' h! s  PMIB_IPADDRTABLE pIpAddrTable,
$ f: K$ B1 S3 H! v1 L- b) a  PULONG pdwSize,
" [; c0 g5 s8 X% i; ^  BOOL bOrder
);
$ b+ C* o; v2 X* H! V1 F$ }
; w% B, v. X3 y4 ~  b' }
6 U+ v/ v0 D* Y* b. c. K- Ztypedef DWORD (WINAPI* TGetIfEntry) (
/ d. s  W- z2 Y# ?' K$ v3 j8 c  PMIB_IFROW pIfRow
, z% Q% _3 T. C5 m8 c);
( n1 P, U+ Y5 s, \' x2 ~6 x

  w! E/ I8 w  N% M0 j# WCString translateUPnPResult(HRESULT hr);
4 L! y  z, O6 F( e* a2 sHRESULT UPnPMessage(HRESULT hr);

2 ?& d7 @2 H: d8 o4 {2 J  h
# F# ~- `& F3 V8 k: ]class CUPnPImplWinServ: public CUPnPImpl
{
% Z! k' \  }' |5 O        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
% ~& A% r" m5 H// Construction
4 a0 R, n$ Z6 l7 K: Dpublic:
  |$ Z) D$ F/ ~6 F% p* s' O$ P        virtual ~CUPnPImplWinServ();
$ F/ Z+ ]* W8 i) |' y' |4 D5 h        CUPnPImplWinServ();

% d1 t( Y3 I; n# T
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
( J1 i. Z9 ~- F9 y) X% U3 q        virtual void        StopAsyncFind();
& ?8 q3 B) d8 l        virtual void        DeletePorts();
# p% Z% @" C8 s& W" f0 r$ e        virtual bool        IsReady();
) B7 Z8 u& M, I. V        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
4 b2 G6 [% k$ a. A5 Y! a/ Q
9 |1 `& P) Y& W) g
' F' u8 ~/ y: x4 T0 C- }        // 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; };
$ C# l% s% J8 L: E6 N
# D( l0 g5 L$ Y' c( C
protected:
" j" [0 G# ~) v1 v% X        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
' x- i* r  k) F) x8 ]        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
9 n- O8 q& Y1 P1 I        bool        OnSearchComplete();
  y) u8 Y1 h( o5 B        void        Init();
+ n0 ^, r  I( h$ o1 A1 T9 `: e
) W0 Z# R, n4 N6 b5 T' s9 f
3 Q! Z" ~& G, \$ B* Z% ]2 a        inline bool IsAsyncFindRunning()
  n( d# r; u0 f  q$ P: x- L        {
3 g% Y7 x& a( T. t                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
3 t. r3 b- l9 H, u5 b                }
                MSG msg;
4 G4 E5 X" j$ ~8 o                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
. b) w. Y+ B# f4 E0 x6 t1 {4 E                {
                        TranslateMessage( &msg );
$ t4 x5 ?( H& H) ?" V# z                        DispatchMessage( &msg );
- O$ l4 r7 P: k! X6 ?' S                }
                return m_bAsyncFindRunning;
* z+ C# N* p0 o6 g8 w  R1 L1 T& j" e        }


! Z6 Q( d  {' V/ f3 K        TRISTATE                        m_bUPnPDeviceConnected;


// Implementation
6 X1 ~# f9 E3 w% c' h9 I$ N        // API functions
' V! w9 i' f9 c3 k& F4 @9 d        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
$ S3 y" v4 @9 S. r9 m        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
& J& n% O$ o# M$ ]: q        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
' R7 X! J" H4 W% b/ G( ^  N& d; S        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
6 M, c$ d: z; `* {) ]
. ~: h- U+ e- v! A- J' v8 G
        TGetBestInterface                m_pfGetBestInterface;
  _4 i$ x! D& N0 u, z9 B        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
6 B4 Z, z# c6 t
5 K) C' A% Z; `
        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
- z, o( H# Y+ f3 A5 N* v        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
' n. F, b5 l- I& v2 m                {
                        CComBSTR deviceName;
+ B" V  p2 v- C/ i1 h                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
6 ]* B7 X7 \2 O* Z- x. R- _
" f1 l5 Y$ H( R7 d
! t6 [( S: q- J                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
2 U! z( V3 a7 n1 u' P
& E$ @6 R' D  d/ n
9 S3 v6 ?- d4 ~/ V. y. r3 \                        return wcscmp( deviceName.m_str, m_udn ) == 0;
$ [- k1 R9 S( z* ~                }
                CComBSTR m_udn;
        };
       
" Q" a% A7 J$ r' _* [        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
7 S8 n7 L. p$ v! h: u" P        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
5 ]& Y7 o* m, z( C3 N7 R+ k. y& N        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
7 E6 R, E2 L/ W" E& Q                LPCTSTR pszInArgString, CString& strResult);
1 ~+ a! [( p9 c7 R* @- F4 N        void        StopUPnPService();


. f) t" X7 d5 _4 |+ d4 h0 x        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
  n. L9 T: a: t  w; V( |, ^# S        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
) j# s! I* A8 W- M0 F5 `        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
$ I4 K8 z3 t! L2 N, B        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
- q6 J7 z/ U, e( N- R2 x7 l( K        CString        GetLocalRoutableIP(ServicePointer pService);
% v  J" K6 V. {) }+ h
8 I8 `( j" H" r- n  k" i0 e1 I
" a! s2 u8 ^( l" W5 c0 m// Private members
' q  B; X7 v& m$ c; aprivate:
! m& t, m+ j' Z1 c5 A# z& |8 t        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
; `6 ?3 f; v' u6 F7 G        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
- I2 V: |5 k( ]3 f# G" @

5 K( M; V1 n" k/ m( q" t% ]        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
2 H+ z7 S7 E" u% E: I5 K$ R0 N        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
1 u- Z& j/ b0 E# L( e        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
0 T- @' s* c! M/ l        bool        m_bInited;
        bool        m_bAsyncFindRunning;
4 t8 @% O; B3 e. P        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
" }) w9 G8 }0 l        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
2 U. v. ^' d7 z8 z' Y        bool        m_bDisableWANIPSetup;
# b' i  Y: K5 J" }/ B( Q/ ?        bool        m_bDisableWANPPPSetup;

1 E* c! v& \5 R( h8 }
};

3 n( n  K/ Y8 q7 c' I, _2 ?
; u9 f" a$ o+ Z% e( O1 T5 T// DeviceFinder Callback
! K7 x3 }# s6 b6 O9 ]- \class CDeviceFinderCallback
9 N' K( R  W5 l" Q        : public IUPnPDeviceFinderCallback
{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
; ]& N0 J5 ]1 R  O% y                : m_instance( instance )
/ F9 G" m( j% @        { m_lRefCount = 0; }
% E% {& k! d2 Z: x" w0 h% g

   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
  q- U  {% K$ [- \0 O3 H, U0 ?        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
' ~, }! Q- V$ \, {        HRESULT __stdcall SearchComplete(LONG nFindData);


2 o& D, t5 X" c/ Eprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
: o1 T6 K, ~" ~) ?  G9 R/ h4 Q2 k};

7 @. ^7 K- P4 d6 e* L: e4 ]) D0 @
// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
: t: W$ O  `1 K8 g0 }{
8 y8 y4 Q7 }* M( b4 o+ u0 U/ y' Tpublic:
        CServiceCallback(CUPnPImplWinServ& instance)
7 W4 L: `! `! N  D( y  a                : m_instance( instance )
        { m_lRefCount = 0; }
- j$ {( K5 n7 [( w- O. A2 t4 \   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
& M1 `9 q/ b3 b: X, s% S   STDMETHODIMP_(ULONG) AddRef();
6 a* o  g$ K, ]: g1 h) D   STDMETHODIMP_(ULONG) Release();
$ g/ a: v/ X  F9 a% R# e

5 x/ `" m9 ~  @4 ~/ @; _7 \3 y// implementation
( `, ^9 I; O6 ^  f) j4 Y% [private:
" S" K, G+ c  u  u7 o* u        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
* L" i( \/ Z9 E        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

& K1 R7 j% F7 |, Z% `1 }- K
9 r2 P$ M7 o& U" S! Vprivate:
0 l( F0 e0 e+ p7 Q' v        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
7 E% p& S5 o. @  O' P$ w};
" V# E- }. j1 B* ]" @, R

( p- d' Q& r/ W- A/////////////////////////////////////////////////
; X& [/ j: p9 D
8 t  \/ |2 |' y$ ~# Y" h0 o
4 @5 S* u) H7 l. e使用时只需要使用抽象类的接口。
; o/ ?8 W: h* r: ^CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
; @4 I4 y: e& U- \CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.