UPnP

天云上人

/*uPnP.h*/

1. 
   $ m, S- W! o  H1 |; R: f
2. #ifndef   MYUPNP_H_
   
3. 
   & E( S, Z6 H6 L: ]3 X% D4 R
4. #pragma   once
   & c4 i, Z& S: ?& B/ c: Y
5. 
   
6. typedef   unsigned   long   ulong;
   7 {) g2 X' K( s4 p% H
7. 
     h) S- ^" r  R9 C, x$ O' D( b
8. class   MyUPnP
   
9. {
   
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    5 P: ~! M" y5 S  X5 |. E
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
    , Q' A/ R4 U9 l2 O- y
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    / J$ Z( U0 a6 |# [$ @0 O/ }7 x
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    0 ?/ g% {" F6 v2 D# f
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    8 {2 x% L% x3 ]6 I4 Y% x
25. WORD   internalPort; //   Port   mapping   internal   port
    # N) k( S, h* J3 x
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    5 C* n* S; Q: c  T2 D4 R
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    ! d. i+ {6 a  w3 i
31. MyUPnP();
    7 _) b& P3 j/ T" q# Y% n/ l2 G% F
32. ~MyUPnP();
    ( c9 i' F  O5 h0 t
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    2 Z, Z2 y- _" r
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    - Q# z( P! L2 ]8 ~
36. void   clearNATPortMapping();
    
37. 
    # x6 S3 H& A5 @% ~
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    / B3 L- }: G8 ]- Y5 m, r8 }& k
40. WORD GetLocalIP();
    ; M' y# t% Z- Z. j* Z
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    7 R3 `( I8 W2 z0 @, C+ ?5 u, n# F1 W
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    ) l7 b  D! f+ Q6 I; j4 r# z/ R6 U
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    / [4 \9 J: |$ M8 H5 x1 ?  A
48.       const   CString&   descri,   const   CString&   type);
    , J# x+ Y) b* J+ q) p4 f9 t
49. bool   deletePortmap(int   eport,   const   CString&   type);
    : T% K- F* R0 v7 k
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    + q- l7 o+ m, V/ m2 k# `- V
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    ' S+ Z! L! z8 S; u# J% {8 i
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    " M7 w" w4 X4 ~' M) D: s
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
      H/ S* V. ]- m  U! d( M- q: R9 d% a
59. bool InternalSearch(int   version);
    4 N# u  c  {* N3 I8 M5 y
60. CString m_devicename;
    0 Y6 D( K6 {  Z$ J: l" ?: `
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    # n* A$ w$ P0 w' S
64. CString m_controlurl;
    
65. CString m_friendlyname;
    + A" z  J5 k, f( [8 P
66. CString m_modelname;
    
67. int m_version;
    2 g- ]7 T- D3 l  d0 ~$ |
68. 
    
69. private:
    2 w: y+ X/ ^& p3 s# m/ H
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    7 D( F) O8 {5 @$ F8 x' o* b
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    9 |! @. @4 _9 j
75. 
    2 r& Y$ Y- `  ?
76. bool isSearched;
    
77. };
    # A. d" B; w/ j
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   ! C" w  y6 [# @
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   
5. 
   ! D: T( N' @- q  _; G, z; b
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   1 e% U0 F/ t" |
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   % H/ U  Z# \7 Z; M
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    ) ]2 t0 n+ L2 A& H
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    # ?5 W( n; r  k. I+ r
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    0 o- c3 [; l: I; p; c8 d- t
15. 
      J$ {# L/ ?6 R  x! Z. X: `
16. const   CString   getString(int   i)
    " b" W& y- o6 o
17. {
    + c+ {( k- p8 P* V/ O8 Y( G# F
18. CString   s;
    3 \, f8 R9 R! l. J/ Z! S
19. 
    8 N; K' X; _: P7 ^
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    
23. }
    3 a7 @. t6 x6 Q$ V* d1 X
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    7 \4 p- x+ }4 l
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    & Z, H: }7 a3 o( q0 R
28. }
      ^; P' ]1 i" {1 R
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    ; H7 P9 ?" B' _: @
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    / z; z. U" ~' ?( n) Z) r0 A
34. 
    ) v+ v) _: h# f8 }# h
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    7 ]( W* ]0 f& ]" R* r. b
36. {
    
37. char   buffer[10240];
    
38. 
    5 i, g5 y- m8 C: ^8 Z* M" f3 o
39. const   CStringA   sa(request);
    ; f: d% i! R3 L# _; g2 G$ ~
40. int   length   =   sa.GetLength();
    6 a5 K7 Q# R3 I9 t; P9 e
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    4 e. j7 a4 d0 G, ~. ~
43. uint32   ip   =   inet_addr(CStringA(addr));
    $ ]$ ]: ~8 D. E3 Q: X7 k
44. struct   sockaddr_in   sockaddr;
    ! g# r* T9 M7 t4 x' \- M
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    2 `8 P( y0 M3 R( f. J' @) J
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    6 I" F4 |6 f/ C* y
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    % f( G5 \4 O4 g8 P6 X
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    % X% p: e  C' I+ L4 J8 N9 \8 T
53. Sleep(20);
    1 W/ l' k' m8 x/ K5 z  Y
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    ' h) B0 B1 m3 _: L  W5 M
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    7 B$ J8 N# s$ h! [7 s2 G
60. 
    ( t, `# L: {% i0 Y
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    / s/ o& F, T* a# m
63. return   true;
    6 @  ^- a9 e$ E- }
64. }
    % Z0 E1 U& x  Z, P
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    8 ~% _' t: t7 ^5 Q. H
67. {
    
68. char   buffer[10240];
    
69. 
    $ F+ T' v+ V* U9 G$ G9 I! b/ i
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    8 c9 |# l- V( L) w: e% \
73. 
    0 {! t; `4 m2 Q. @+ X1 S
74. struct   sockaddr_in   sockaddr;
    0 ^0 D9 I$ o: Z3 A& ^
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    . u5 D* @& _5 K( g. E% Z8 f
76. sockaddr.sin_family   =   AF_INET;
    & J) ]" z8 P  P3 ^% }
77. sockaddr.sin_port   =   htons(port);
    ; M. |. W/ y1 [' N" E' ^: j/ T
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    $ A) z0 Z- ^+ V& Z" b9 t5 m4 F
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    & e  F$ ^/ N' ^' r4 N; T- u8 v3 t  w
85. int   pos   =   0;
    # P7 W  o. Z, Z7 @) u6 S' H  C% N8 V
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    ! w6 K1 y( _6 T. `2 v) |  G
88. 
    # ^, P& l0 y8 f4 X6 P! r% Q0 w
89. result   =   response;
    + J, \& g! z  y, D
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    ; i% d1 v* d0 Q
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     * V4 |4 X/ G' v& n+ ]- W
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     " t/ y2 X, D# k% ^: f7 r: i2 U9 e' |
106. if   (posStart <0)   return   CString();
     $ J  ]& p7 ^; {( r1 ~0 L+ C. H
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     : D) ~; w8 y6 b; A% S
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     ( _" x* K( e& M5 O0 d
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     * \1 s! g  @* L& q
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     * t1 r" O; x  N1 V
118. isSearched   =   false;
     5 L3 X' d! a- Z" P
119. }
     ) d- Z! t3 p* i/ U5 x% U
120. 
     . a. z, w- L6 C; l/ k
121. MyUPnP::~MyUPnP()
     # n+ M5 b9 x" s5 Z& L8 s5 ^# d7 C  J
122. {
     
123. UPNPNAT_MAPPING   search;
     * f/ V, S4 z$ U- B  E1 `$ A
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     ; j: `( q- l; \- f) [: V
125. while(pos){
     8 P9 V( ?6 W+ L$ G& \" I5 z
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     , C+ L% W- D: r6 |+ S
128. }
     
129. 
     0 i' K; \7 V) U: W& [
130. m_Mappings.RemoveAll();
     
131. }
     7 v2 A/ o; D/ j% ~% i; h+ E* k
132. 
     " ]7 r! _5 ]: T. w: E- J4 [- i
133. 
     * k1 Y; S! S1 I4 n; t. Z
134. bool   MyUPnP::InternalSearch(int   version)
     7 U6 n  D! H8 f. {( v) E! A+ I) g
135. {
       y6 z- G) z" P6 \+ d1 c
136. if(version <=0)version   =   1;
     ; M% o# F" f0 f3 r+ F. r9 ~' W
137. m_version   =   version;
     
138. 
     , l( W5 I4 B- ?* D/ B
139. #define   NUMBEROFDEVICES 2
     2 J6 o; `4 e- Z4 d
140. CString   devices[][2]   =   {
     6 n' H& j" q7 C3 D0 a
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     4 b  T7 L1 W( A- z! W6 F4 R
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     * ]% U4 C# m, B5 ~+ [
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     & S  y+ m; W! n- A$ X8 C
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);
     & v+ D3 [9 B: m. [+ W& c
155. CString   request;
     5 G) Q$ d$ ]' g$ S" C& T- T4 |
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);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     8 q) [; [( H" h, ~* Q+ K
159. }
     
160. }
     
161. 
     
162. Sleep(10);
     3 D" J. `4 k. g& D
163. 
     6 f- N/ P- P' i* f* S
164. char   buffer[10240];
     ) n" q9 v% C& A) R; W' Q
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     8 q# _& D) y3 |% q/ {1 N4 C
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
       m5 l! e1 ~9 m; ~3 c" }) T- E) O
169. CString   response   =   CString(CStringA(buffer,   rlen));
     2 f! X5 ~- r$ I* B! B- U
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     $ U  d8 z& u+ V7 T  C' ]( L
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     7 b2 Z  X0 z3 \1 }! v5 ]7 `5 f2 G
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     ' |7 T/ q* V$ L6 u, G: N+ H+ H# E
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     $ S. K) u0 Q1 z- C  n/ [' o( [% f" w
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     6 g2 ~% K6 a8 e6 I$ m6 p+ a
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     
187. }
     * [6 m0 b, B' u% G7 C4 x9 s: ^
188. }
     
189. }
     
190. }
     
191. closesocket(s);
     
192. 
     / H6 }1 t9 S5 o9 `
193. return   false;
     8 y+ w9 a2 h; |; p
194. }
     

复制代码

天云上人

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

# b# n2 w8 s( g3 [- Q% V
///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
+ w# d* f2 i" h7 X3 _  W

#pragma once
) J, w$ [# A7 I2 N6 t% @# D#include <exception>


8 D. B+ ^% K; \  enum TRISTATE{
( [' v  s/ r# q% Q" c; {        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
};
5 V3 b  a7 i2 y+ D$ l/ i+ V
: M5 ^( U  P- R& Y+ A  D4 {
) ^0 j" M+ _- D  @/ N' @enum UPNP_IMPLEMENTATION{
( |7 l) `* [& ?$ j        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};
! Y. h; v3 P* u. p+ B( O9 ~

- d1 \* U( w/ U9 z0 H/ }* Z3 X* w
9 {) s4 q# f% B( y6 z
class CUPnPImpl
{
public:
  T3 B6 O! ~9 `- U        CUPnPImpl();
/ S* k5 A. U! `6 T$ M( m2 z& |& N        virtual ~CUPnPImpl();
# N) b9 a4 d; d" b0 n5 u. G/ h        struct UPnPError : std::exception {};
        enum {
2 C/ G! N7 z1 L                UPNP_OK,
$ j, C1 R8 ^& f- [/ m                UPNP_FAILED,
                UPNP_TIMEOUT
        };
* h4 ]! L  p# T- v

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
% M7 Z# _7 b6 q. A        virtual bool        CheckAndRefresh() = 0;
) _5 {) d# `( t- `        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
/ _6 x9 k1 {+ d- h( u8 H+ n7 a$ I, c        virtual int                GetImplementationID() = 0;
8 k4 t" ^; N8 Y! z8 e& M       
7 @5 F0 K1 S0 n        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
6 w" ~- w/ \5 w" Z$ d0 o
; ~) ^1 ?& K: f2 k- ], p
5 X7 w  V1 G  e+ R* Z( l* k+ f        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
0 l8 }9 c5 [8 \9 t, ~        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
# a6 R5 Q# _, W! w/ y

// Implementation
* O% R4 G) [. W" nprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
2 e3 K% h( f% d  c        void                                SendResultMessage();
        uint16                                m_nUDPPort;
# o, i/ z& _9 b) L, }0 ]7 N  d5 i        uint16                                m_nTCPPort;
' Z; w0 N- u, d; J6 ]; x        uint16                                m_nTCPWebPort;
0 D9 U$ {) l, u9 |- m% e* x        bool                                m_bCheckAndRefresh;
* t$ c1 |+ ^4 [0 F7 R7 Q: D0 c  H. W, a
! {, l6 S, p& V2 I' p& X4 A+ x3 o
" @1 [" X+ K+ A! ~5 A1 Jprivate:
  e1 a2 x, M& i% o, U: g        HWND        m_hResultMessageWindow;
% Y" I6 H) c6 W' @8 T        UINT        m_nResultMessageID;
' U+ [$ y" M2 ^
( k8 R: J/ a( R5 X+ F3 y0 ~2 R: a
3 l) C; E& f0 |6 }0 Z& x};

4 L8 n& m" d( N  y2 L2 x
// Dummy Implementation to be used when no other implementation is available
1 Y/ X% P/ C9 Vclass CUPnPImplNone: public CUPnPImpl
4 ~) Q1 v  R2 s# u" Z6 c9 }+ w{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
. t8 |7 Z8 J/ o5 X. D, p        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
5 _  V+ h3 }7 C! G# N        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
+ B2 W3 p1 g1 s' p" A) N' T};
! ]5 X" r1 {: i% f. n7 E
5 K2 e' N2 e$ D1 {* V
1 O* v, O! t; T/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
/ [, s9 A9 @6 s  H+ L& P7 ^
5 k$ D3 C' x$ H& P( C1 a( O  z
' H  D# ]! b. ]& F, R' d, f* h, \#pragma once
#pragma warning( disable: 4355 )
, v- I" X' u1 M0 n( n5 A. W

#include "UPnPImpl.h"
#include <upnp.h>
" U9 K/ v5 [) Y#include <iphlpapi.h>
" {3 ?' a+ ?/ t7 H. j0 X#include <comdef.h>
5 h3 r1 H: C: `0 m: D  f+ J& o#include <winsvc.h>
$ g: L- \3 i. S/ h7 ?5 P" h

4 k4 R1 A/ q9 V% P$ N#include <vector>
- e1 |- G+ w6 b8 {& z4 x#include <exception>
#include <functional>
9 F* x' u. l0 `9 G) U


+ o1 J) C( e4 e; b3 v
$ |4 b  }  M; R3 W/ p% h) ^5 ~4 ~typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
, _/ F. I3 i8 f0 w6 R0 O4 ktypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
5 B8 p# \9 l; {4 U9 K& w# I5 htypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
, X  p! b9 T: Otypedef        _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;
5 Q7 l5 p' N* U8 E; Z
+ R4 `8 p4 j! }( V7 M4 p' p
typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
, N' o, ^; r$ H0 I' m3 G  q- d  PDWORD pdwBestIfIndex
; p" Q( R- Q! I% l$ Q);

" j4 R2 ?  N3 n# ~0 s0 K
3 r6 B0 L- J3 b. Y- C3 stypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
4 y0 G7 H& c9 e4 T! E; \4 i/ R  PULONG pdwSize,
: }& Y# O, Z7 X0 X  BOOL bOrder
);


typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);
& r5 L  T3 O- t

CString translateUPnPResult(HRESULT hr);
( k- Y; v6 h& i- p: {HRESULT UPnPMessage(HRESULT hr);

* |4 j9 Y6 t3 y4 W
class CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
1 e: W1 x& y# b4 @7 P& w# ?        friend class CServiceCallback;
7 f) L2 x2 @7 n// Construction
public:
. Y3 }4 p) E& t$ ?        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();


+ h" e0 ^# b: L9 s# ~5 F" ?) Q        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
4 K; q9 T& [! O0 ]8 f. O1 `        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
0 x+ h+ L# P1 X8 _0 l0 v9 E# h

; w: z0 P- U+ S2 n) C4 [0 E        // 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
1 {) G1 D" R1 X7 ]* p" a" {        virtual bool        CheckAndRefresh()                                                                                { return false; };
8 }" N% H' A. h1 q/ I* ~
$ N$ o9 f2 D5 p! ^$ R: d2 x+ |. p4 z
protected:
7 m$ S( d* r1 F7 Y5 h( ^! _3 y        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
( D/ e9 w( B& q9 R        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
& i, f7 ~- S2 V; S) f- H% m        void        RemoveDevice(CComBSTR bsUDN);
! `. |5 r4 ~2 X- G. Y4 I6 h: E        bool        OnSearchComplete();
        void        Init();
# G  j$ }" Z  F9 {5 x9 J3 g+ A
' V& X. N& t- r) a6 P
        inline bool IsAsyncFindRunning()
        {
* V' ~& U4 I7 Y, Q. P                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
6 x# t, J) W7 l2 P1 Y: J) V% \                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
0 H& G. e1 J8 _. S, \- ]& h* q5 U: G                        m_bAsyncFindRunning = false;
  `: M$ t: l. K& u/ q  l                }
) b& v7 X9 r" W) @' B6 m' `                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
. D: V( A5 x7 l+ K  O5 p                {
                        TranslateMessage( &msg );
8 K: F7 }4 i" N" P& {0 V9 `5 C                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
        }


* h! E6 G( d2 Z8 m7 S( Q& f$ {        TRISTATE                        m_bUPnPDeviceConnected;

, D2 a9 o- a, u7 ?2 _3 M( ^* k
" k8 E  D. `3 V) B" R7 ]// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
4 a3 x" c9 K% L+ M( m, s        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
& O1 B! O7 n+ p0 u! M        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
9 \( b4 H8 P; Z7 o2 v. Z        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
' b1 w) ]. h$ ?. Z9 U" k' n) N        TGetIfEntry                                m_pfGetIfEntry;
/ H8 d4 _9 D4 C+ Y+ o- g+ f# x7 s

        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
( U6 F3 v  ^/ n1 M; e; z                result_type operator()(argument_type device) const
/ w! l) y' F8 ]                {
/ d- {0 K7 T5 b                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


; E8 k0 [1 Q7 s8 d6 g( l6 ?                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;


; S9 }( E! }: o0 J+ x- m) e                        return wcscmp( deviceName.m_str, m_udn ) == 0;
) @8 X  C. ?) J0 K3 _9 \                }
                CComBSTR m_udn;
        };
       
' C4 X- i- g8 C6 R3 m        void        ProcessAsyncFind(CComBSTR bsSearchType);
( ~9 k- U/ L. _% y+ r: n        HRESULT        GetDeviceServices(DevicePointer pDevice);
* C: `8 K. N: P# P        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
$ c/ Y! ]) v, P) E! O9 a! w% B. @        void        DeleteExistingPortMappings(ServicePointer pService);
- L) a' d( A' E1 j- C6 N        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
- v3 e6 R* X: H' Y        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
! g8 P1 Q4 n9 t7 I9 x& k+ a        void        StopUPnPService();
; m# t- P6 q7 y# N+ f! i
7 ^2 F  ?4 r% m- u7 p: d0 B- ^
        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
# _6 ?+ S: z& ^$ N) R, n( Y: Y6 H        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
. S+ U* M  Z" x2 r4 R        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
; q9 q" ?3 t& Z# s8 t( |6 [        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);

9 Z" z! f* Z5 n6 D$ A# Q
. q, ]0 a7 Z$ f// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
& s. g; k+ M' H% `

        LONG        m_nAsyncFindHandle;
+ b  z8 H! {* ]/ ?3 V) V        bool        m_bCOM;
        bool        m_bPortIsFree;
% c; z; f, G$ L& }, D- \% l+ I' Z% b        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
  l6 Q$ N" r3 X        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
5 \5 q9 b' u; X0 X# a5 q2 o        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
/ r5 [/ o) @: y% r, d. V        bool        m_bSecondTry;
4 r  s1 q5 J, D: c        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;


: j# s; D) i. F" }! J/ P};


4 v4 N" U& `- A+ m9 w// DeviceFinder Callback
class CDeviceFinderCallback
7 V' t4 t9 r% v- }        : public IUPnPDeviceFinderCallback
, C5 y+ ]. s- J/ p* D/ C{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
. X5 Q1 [$ K/ R0 v                : m_instance( instance )
        { m_lRefCount = 0; }

* d  \  s0 p" b- U. I7 _% q4 R
7 ~9 h9 y2 I) S. f/ u" \   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


$ n5 T+ v/ [( X6 }5 I// implementation
private:
, R, i* |8 ?1 C1 s$ y/ h9 K: O        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
: L/ Z' E1 z5 H" Y6 t/ l        HRESULT __stdcall SearchComplete(LONG nFindData);
1 G. e6 x* b& @7 x9 D0 N& [

6 L; G6 D$ d; i! ^5 aprivate:
- @* g2 L( F( {& S        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
, a! v, N9 J. Y/ F2 q};

- I2 P) k# u8 M5 `
// Service Callback
2 C$ k/ S/ c. [( Nclass CServiceCallback
        : public IUPnPServiceCallback
{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
6 O: x7 ?& k2 G0 i& f                : m_instance( instance )
  A# N5 l' u( W& Z" G" }        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
, S8 g& l$ X, W; ]2 j   STDMETHODIMP_(ULONG) Release();
/ V+ K% a% `# M5 P0 B! l' [

8 i) U" E5 n' J! K// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
& c' A5 v% M) _$ f0 N        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
1 g; y  g6 W* d. b- S. Q6 F
+ x* ^" ~* c& a6 {$ d& I
4 m9 `6 ]& |+ q1 M$ k# S# y7 |+ pprivate:
6 G. K# x$ Y5 c% Y- C7 K& y) q        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
  L8 ]) C; n' t' a/ x};

# i! e6 ?+ c4 F* K" w" }
. o& _- H4 R5 U. h# L3 P, v( l/////////////////////////////////////////////////

% d& X" U4 d' D; M! r/ A
使用时只需要使用抽象类的接口。
% t2 l% |8 s( R! k- l; d2 ]8 CCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
" E. u; J) i8 z1 K) G5 \2 H  vCUPnPImpl::DeletePorts删除端口映射.