UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   7 [6 s' q  V) D
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   7 U/ I* w1 O' ]/ |( D2 k+ A( w
8. class   MyUPnP
   * c  ^1 h3 S6 {* Z
9. {
   
10. public:
      P  U1 s2 }( W; d, d' W
11. typedef   enum{
      P/ y7 H& |( L0 n4 ^
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
    
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    $ t" P( F% R- |* l0 b) F& {4 n
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    8 s$ A8 w7 S7 ?: p- X
22. }   UPNPNAT_PROTOCOL;
    3 ~0 F" `. w+ e
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    8 `& U3 h$ s. r; x- w! s
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    / f6 M! ]2 B, o2 Z. @, E+ s4 p
30. 
    ; v& ~! F. p" R" @
31. MyUPnP();
    * |7 a1 \& W8 u5 K
32. ~MyUPnP();
    # |# Y. d' k7 W8 j' \( y" O
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    4 O: m  }1 L% |$ K* d6 t
36. void   clearNATPortMapping();
    
37. 
    % B0 B: c) F' \" K0 h
38. CString GetLastError();
    & [; i5 L* h0 \& h* @" L' K& e* X
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    % j  ~2 V' Q1 u7 ^
41. bool IsLANIP(WORD   nIP);
    & t- R% |0 J+ A" I/ f
42. 
    
43. protected:
    2 A' P! h6 K  L1 \7 N" i6 K
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    / n1 i% S  a/ O" c% e# W1 u4 q
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    % c" U, W( R, g& q) i$ {; U! [6 h
49. bool   deletePortmap(int   eport,   const   CString&   type);
    ; r3 ]' C( z3 ~- b- P
50. 
    9 e$ A. v& |/ @! b6 c& [) C3 r
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    4 S$ J0 Z* t, D! z* ^; U
53. bool Search(int   version=1);
    ; c& _) z+ L( Q  V) ~3 v
54. bool GetDescription();
    2 x; j4 f; A- X/ @
55. CString GetProperty(const   CString&   name,   CString&   response);
    + z9 L, d$ A' [
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;
    # F) h4 ^4 M3 t& P" R  S
62. CString m_description;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    ) o3 |8 Q, U- w4 c4 j2 K7 m( D
66. CString m_modelname;
    
67. int m_version;
    . b: S$ Q* D" C1 H  _1 c4 V. T' ]
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    5 B& p/ e, ^! C4 u4 @
71. 
    1 X7 T+ q8 ~2 {! @' a
72. CString m_slocalIP;
    $ q8 Z# b2 w7 s
73. CString m_slastError;
    + y' r3 O! u2 \+ p0 [; @
74. WORD m_uLocalIP;
    / j. e% H: t5 C( h7 Q
75. 
    3 G9 W# w7 Y0 M( e4 u  B5 V
76. bool isSearched;
    + J6 `" \: r! n" D( x2 U- q
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   ! N. c- Q# J" Y! W
2. #include   "stdafx.h "
   . r  O+ h( x/ M' }# `- S& Y
3. 
   : w1 u1 x. k; b
4. #include   "upnp.h "
   8 w8 `7 P7 ?$ o- b6 x# x7 U! q8 F
5. 
   6 o, w4 Q+ H# R+ n( \5 i
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   * R% L7 }! v) n, j
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   . j2 n7 Y$ n3 _& V- W0 C3 ~7 x
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   3 J" Y8 d! m( v
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   , \2 G# \! X' P! i& P
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    8 a* A. |4 Y9 @+ R
11. 
    ( h  F# `/ s; H9 z
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    2 h5 J5 _. m5 ]8 F$ b. V
13. static   const   int UPNPPORT   =   1900;
    4 D% U" s) G9 u0 a- N
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    
17. {
    . j0 y$ z. s' I6 p* i
18. CString   s;
    ( O0 y8 @. b. M% k/ `( C8 k
19. 
    
20. s.Format(_T( "%d "),   i);
    0 P( G1 ?; h) v! J/ D# P
21. 
    0 U+ d/ j; ?; u: z. T( y
22. return   s;
    
23. }
    . i0 t# I: {. B# K
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    0 e) L0 B! d; k  ~; n! x
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    3 }; Z! |  T* ]
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    * h( e0 }# P) {7 p
31. {
    $ u9 J* z" ^) C% x/ u5 n- p7 k
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    9 B% D: O5 r" O  }' x- }! l: t
33. }
    + N4 `; Q$ W, J# e4 ]7 j
34. 
    & K% b4 Y( n. i9 m% T' U2 t' Q
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    
38. 
    0 |6 F# F4 k) H. v5 O. X
39. const   CStringA   sa(request);
    3 X$ _1 g$ k; N
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    9 I8 S: \) D+ H& ^4 n
42. 
    - h6 m0 R. j' \: Z6 ~0 y& W3 Q
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    $ i* Q  x+ l7 z* r
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    & A+ }- c5 U% C- i
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    ) i1 h, s4 A/ r8 `% R& H* H% n
50. u_long   lv   =   1;
    2 m' _/ R- ^1 P0 U9 J7 U
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    ; }! t! q7 j) T0 ?( v$ Y2 |% P( `
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
      O/ S9 r! y1 s5 b
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    9 V3 A7 g0 h" S8 c6 C, s+ ~8 y5 q
57. closesocket(s);
    * _! B, m$ }; B( H
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
      a9 @* q* R, N7 O3 K. _' ^
60. 
    & S- Y4 e! p( U; y8 z8 L
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    ' t! }& f5 x% s% T/ O' B, Q
64. }
    
65. 
    / r; L7 J8 ~! O  a# r" ]' [4 ]2 L
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    , N2 Q* n1 s* ~
69. 
    
70. const   CStringA   sa(request);
    6 i- H# p2 M4 u+ u; y
71. int   length   =   sa.GetLength();
    $ a4 E/ Y0 L9 x# P+ k
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    0 V1 y; H- b# Z$ r7 i
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    7 L$ C+ x2 A1 S7 f
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    0 r9 L# \  a# @( g% U+ |1 F
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    : N' I9 o' }/ e$ p4 H) N* e
81. }
    ) |' |( @& S! G# D3 n
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    5 N! }$ V4 _: B& n7 U% i* x
84. {
    
85. int   pos   =   0;
    
86. 
    1 _  H1 P0 y/ B5 F+ _
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    2 @4 X' d0 j* B8 ?
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    
98. 
    - N3 H5 h8 N6 y* b
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    9 b6 C# a# x2 h- @4 C9 {* w
100. {
     & x: [5 P+ b: D+ H
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     $ o7 p; k! T0 j  s+ d
105. int   posStart   =   all.Find(startTag);
     5 @8 h0 o( j6 N' B- l2 }
106. if   (posStart <0)   return   CString();
     & a4 P# S9 S- t% o) P
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     " ~9 c4 }, B, ^- K& Z- Q
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     3 ^8 p2 F# l$ T$ G; X
114. MyUPnP::MyUPnP()
     & ?: V7 C8 r+ u6 o, C# z+ f  `3 k
115. :   m_version(1)
     & W/ I$ `* z6 b( J% l
116. {
     7 B+ u0 R0 L& C* k+ _
117. m_uLocalIP   =   0;
     . f4 ^, V/ m' B$ D) R
118. isSearched   =   false;
     ! i/ C& x/ ?/ ]
119. }
     
120. 
     ) Y2 g! }% D. k% [
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     4 b6 x2 A+ W4 p1 l( A/ X
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     # e  G. q) S; k" E! X. R) ~. |) l
128. }
     6 r2 |0 L' V: F
129. 
     
130. m_Mappings.RemoveAll();
     * X3 N2 I) l$ J8 d$ S/ A$ d* x
131. }
     4 G6 \% E. @$ X% Q" r0 `5 ]( M
132. 
     
133. 
     " p: R) ^4 m1 E& m0 M8 g" f
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     9 k6 r$ Y9 g$ V1 _
136. if(version <=0)version   =   1;
     7 U( G$ n/ s: G% s
137. m_version   =   version;
       @3 o: e/ c' R/ n
138. 
     
139. #define   NUMBEROFDEVICES 2
     1 U- l; N; M6 e7 m
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     # v4 l0 `" j7 W7 ]9 G0 ^# G% S
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     + D# t0 D/ C' a* S( b' L
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     1 i+ t, U* s+ R
147. u_long   lv   =   1;
     % F2 r" i& G7 G# e- T  O
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     , L( ]# s5 F$ f, V5 j/ e6 L. p4 y$ }
150. int   rlen   =   0;
     . E- \4 k7 l: D! t; ]8 x( R2 b
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     1 t& [) y0 m! D
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     ; G( R3 w# l% y8 O- }( Y5 ?
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     * o$ ^$ P( G2 P6 ?8 y8 }- w% I8 C
155. CString   request;
     
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);
     9 N5 [, @$ }! I- A
159. }
     . d7 m* h0 \9 F' @4 d. J9 m/ q9 |0 }; g
160. }
     
161. 
     
162. Sleep(10);
     : f# U; r% l1 J; X! @* Q3 W' H
163. 
     2 l0 s/ D: i5 \2 s1 x
164. char   buffer[10240];
     - p- `. T5 z9 o" w' I( ?" J' @
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     6 d5 ?/ {+ z+ C9 R
166. if   (rlen   <=   0)   continue;
     + b. G3 I- E8 d+ c1 i( j& Q) K# f- w9 x) \
167. closesocket(s);
     9 u; T0 q: C0 I! u! H$ t
168. 
     - T% M- |5 U9 M2 v- C
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     & k: w- G" I# \7 s- z
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     / z# ]4 ]0 s! {' o% l. m& q
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     0 s  W. o& |4 W* l7 A& B  |% t8 _
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     ) g9 _2 b" `: K% ]
178. if   (line.IsEmpty())   return   false;
     5 V; n5 c: ~% Y% }
179. CString   name   =   line.Mid(0,   9);
     ) u/ V7 b5 n+ z
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     + ]7 c% h; h. Z2 j; \, ]* N
182. line.Delete(0,   9);
     / w+ h  u% m- R( u! w" D. e# H
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
       n5 z7 D( P2 S1 K% r
186. }
     , i' {, o1 p4 i2 r7 t/ }% `
187. }
       _' ]& c# }- D2 |
188. }
     ( E( Z9 Q! v2 q/ r; l
189. }
     
190. }
     
191. closesocket(s);
       \, x9 j- B  f/ \# Q
192. 
     
193. return   false;
     
194. }
     

复制代码

天云上人

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


% @" i% F+ b3 `: J///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
" r: E3 s+ k% ?' k6 }

#pragma once
#include <exception>
; W+ e) w/ L$ P- J" Z' z
+ D  C5 t; d% F7 C
  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
% F2 O5 U" k4 ?; x. ^9 t6 L  o+ N        TRIS_TRUE
};
& n# X1 p  O, y7 p& f

enum UPNP_IMPLEMENTATION{
$ S8 `+ f* v$ f5 S        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
  L6 k/ T8 P9 z: q        UPNP_IMPL_NONE /*last*/
/ O/ j1 R; ^, q, ~  Z};


3 I5 s7 \* i- Z7 e

class CUPnPImpl
{
public:
        CUPnPImpl();
5 u8 ?8 L. [0 ]        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
! [! y  u  f" q        enum {
: Z. h. J+ Q, B$ r+ _2 D4 z8 {$ k                UPNP_OK,
                UPNP_FAILED,
* }# N9 V7 O( S7 q/ t                UPNP_TIMEOUT
        };


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
9 K8 ~, y# x9 n* l7 [: X9 l        virtual bool        CheckAndRefresh() = 0;
9 `4 ^2 N/ t) y9 c5 ]$ {5 N        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
$ d8 R+ [  v& X" m8 ~  T$ P- m9 T% l# C        virtual int                GetImplementationID() = 0;
       
% x+ E1 d& i/ ~$ B+ R7 B4 q9 j        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
) c+ B' N- B" {, i( D/ X

0 k+ Q3 r7 j* X2 J7 x        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
3 P* x. H3 [" S; _- W7 q2 k
4 @3 Q: L* ^9 V) g
; d" M1 p! A1 F# [  G' V" @// Implementation
* C$ s8 Y+ K6 T( z9 o/ Iprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
+ _" y- C& d9 [) {6 H# l8 a" h        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
# N+ [& w- G3 z6 k$ ~6 q# P1 x5 u" h% `        uint16                                m_nTCPWebPort;
$ g, R) ~% I! X+ K6 B9 j        bool                                m_bCheckAndRefresh;
# n8 j2 M& T. {
- ^2 V! H8 p, W7 U" A
' L+ k4 H! S/ y5 S0 t/ Pprivate:
        HWND        m_hResultMessageWindow;
8 P8 R1 c0 @) g& Y0 A# k' h        UINT        m_nResultMessageID;
6 u, c  C3 _1 i% w
1 `# R0 n+ Y- `/ c1 L: u" g
9 S$ P  _3 c, A! P};


) d( J* M5 R# }" Y& o$ s// Dummy Implementation to be used when no other implementation is available
4 L. _7 F1 n* q1 a- x- Uclass CUPnPImplNone: public CUPnPImpl
{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
( S) s+ j( ~" n* R4 w3 U4 _        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
8 U* |  u9 S: s6 P        virtual bool        IsReady()                                                                                                { return false; }
3 e& A# B. X; y5 V' z        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
0 S8 W* [* }  m# x/ f) C/ P! q, h9 m  ^

. |( o4 e4 M4 W. _( ^/////////////////////////////////////
( ]7 u5 {+ z5 U" B//下面是使用windows操作系统自带的UPNP功能的子类


* C# Q& `+ m9 i8 e$ V3 f#pragma once
, v2 o0 S1 t6 v#pragma warning( disable: 4355 )
6 l) s) Q% {9 L" j3 J

#include "UPnPImpl.h"
6 A& ^2 D6 `, S; `6 `# ^#include <upnp.h>
$ e! E6 H7 p! T2 ~2 d) Y; V' \#include <iphlpapi.h>
% q* @% M. n- r- i: x1 N8 u#include <comdef.h>
2 g+ A! A( Z6 U9 \#include <winsvc.h>


#include <vector>
  p7 L" c" J5 S! J6 d#include <exception>
#include <functional>

4 |7 q. T% M  A4 O1 g7 d

) Q2 \9 O- z/ _' k5 p  M/ k
& [, i! h3 a- E8 ?2 [5 ^6 [: W$ Wtypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _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;
' e; X$ i5 m. [4 Otypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
! _! c  }" f1 K" w) Qtypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
; E3 g$ i+ r+ V- v0 }

typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
6 E* P) ]0 ]7 |. n  a);

) f1 I( {6 _2 I
typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
: r, m2 G8 s; r1 r6 K  PULONG pdwSize,
  BOOL bOrder
4 |( ?+ G0 O3 R$ ^8 \& v, g);
! I, Z  C  m1 `. o2 n6 M

# ~: b: N( Q" q' d9 `typedef DWORD (WINAPI* TGetIfEntry) (
6 @: j% f4 l5 X& H  PMIB_IFROW pIfRow
" X1 e" j4 z$ V% C" e" ?! F);

0 w5 J9 ~8 k6 K+ ]
CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
( |* r3 |# Q2 @, u& {4 p1 p
4 J! s1 u# U7 u3 z! s
class CUPnPImplWinServ: public CUPnPImpl
* x# Y$ Z, n+ c' @% R) x0 V( e{
* r7 ~3 ]! D- P1 n        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
! }5 `5 u5 ^0 D8 U- G: V// Construction
/ C% N/ L% X7 \+ ^" i- o5 B4 Z8 `public:
" p+ R* K9 Y1 M( R        virtual ~CUPnPImplWinServ();
! U* j6 ]1 u, C2 A" I5 D        CUPnPImplWinServ();

/ g/ M) ?& {* ^/ \
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
$ L% \3 s4 i4 x  E: O' \! s        virtual void        DeletePorts();
        virtual bool        IsReady();
+ [1 k1 w; a/ @. {/ |" v; H6 K        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
: b1 t( Y+ K: R, t8 B, I6 ?
& F. W" a% Q7 Y$ W+ b' p
# n  V% K# k: m8 a        // 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
  K" N: d; n2 r        virtual bool        CheckAndRefresh()                                                                                { return false; };
6 K/ ?5 I: T- v! e: T

protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
+ I1 T* \9 H5 Z! @6 b        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
8 t& c) [) w& l, d/ `& O; w& {        void        Init();

- u0 c- m. q4 _# F
        inline bool IsAsyncFindRunning()
        {
# U6 S$ b' _/ o  c0 W' s4 J                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
. @: h7 M0 u' l                {
: p' B. M0 l( n1 u% d. e. `. ]                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
9 V. ]$ S; [/ ]                        m_bAsyncFindRunning = false;
# T! Q8 b0 a' ]$ e8 w                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
! w) O+ W* {; l6 f1 k; l1 j                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
7 n; h) y- j1 U' c3 v+ q                }
                return m_bAsyncFindRunning;
% l$ H+ u+ G: {        }
* ?4 r' X* a3 U& f

  ~& e2 u/ V- m) V6 Q4 F' }4 s        TRISTATE                        m_bUPnPDeviceConnected;


3 x. k$ Q' T6 s: I8 J4 r) m" l( Z+ j// Implementation
1 G: g: m# o  s/ ^% X2 J        // API functions
/ H+ o7 V0 d3 q        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
/ D7 M+ u" P8 ]4 i/ m- _        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*);
1 G& S3 y: z" U, N7 l2 N        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


! G) K5 r' e. I        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;


7 p9 s1 s0 k: m) J4 B) w        static FinderPointer CreateFinderInstance();
' i- u8 B% z* z" D        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
2 ~6 @6 C6 V! }2 e! P! e6 X0 X                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
9 N) I. ^! z2 p& E6 w                result_type operator()(argument_type device) const
) B$ f2 w* f# O2 ?4 z) B8 X- F                {
4 e5 [! D. J# \# `1 c                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

) l- K# d3 z& Y: X- W1 p
2 Z, E3 Q5 W$ N0 @. `" h1 S                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
1 o" ~0 s6 y- v7 q2 w2 J
2 F+ O! X) `6 ^% ?! q: ~
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
7 h, |' v! m$ l5 t, j0 h6 ^1 k                CComBSTR m_udn;
        };
& Z+ l, a8 M: y       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
$ ^  ?3 {6 B4 Z! p        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
8 x/ j% r3 o, x8 A3 ?        void        DeleteExistingPortMappings(ServicePointer pService);
" V, }4 D2 l7 i- W, N        void        CreatePortMappings(ServicePointer pService);
( b$ y( i! j( h% T4 n        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
3 @1 n% _: a) K+ q                LPCTSTR pszInArgString, CString& strResult);
$ T; H6 W$ t% }' |& v5 C        void        StopUPnPService();
, h% j  G2 _$ G

        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
( Q! i; K4 @9 l2 S8 S2 V        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
  I/ O- d: f% A        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
& H5 I2 Q& U$ c8 X        CString        GetLocalRoutableIP(ServicePointer pService);
3 L, E3 l8 n1 y  |
% o2 |4 R5 _' H) l% @$ Y3 @
// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
$ R" w% E8 L+ z, d( d        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
3 U5 H. z" n+ M- Y& ]' b5 U        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
* S8 J* x- y% t& B$ g" a8 s2 @( _
! A0 o3 ?( w3 b
        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
  E) q# B& y) o+ v) _2 k4 r+ W        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
4 ^, x' C2 K& U5 ~2 C3 @2 |5 Y        bool        m_bInited;
8 L6 ?8 D3 b9 {1 B6 O# n# o        bool        m_bAsyncFindRunning;
5 o9 t. [) o2 g        HMODULE m_hADVAPI32_DLL;
8 @7 W; v/ Q7 o        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
7 k8 Q% @* m+ {7 P' c# Q        bool        m_bDisableWANPPPSetup;
0 _0 x5 @6 f4 ?9 o! \

. B- B; C1 Y. Z% y};
8 u1 R2 C$ c& p1 \5 R# q3 U  ]

// DeviceFinder Callback
7 [, y, r5 v: Qclass CDeviceFinderCallback
# ?" S$ {7 m3 N        : public IUPnPDeviceFinderCallback
7 _6 W% k& p/ c1 {{
public:
- ]+ v3 U; C! \0 Y2 d  w5 `" c        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
, |. z2 i3 L7 h0 c  P& B1 j' o2 ]
1 f) A. q) ~! F1 [+ c# m- ?7 J
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
4 g4 e% V# M8 {  ?6 w

* L0 ]' T8 h+ D: {3 ~# d// implementation
private:
1 b& w1 y4 |6 u3 p& y) ]+ ~        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
, v. m3 d  S- ~3 T        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
) {# w! y, I* p. ]- b. h2 U" w
0 l! f# _3 A% Y" u
) L: {: t6 I6 B9 cprivate:
" c5 G* K+ v, X        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
' h. I* e0 _' v7 S: e3 v

7 f/ o+ _' [7 n8 `$ @) u9 U$ U// Service Callback
% h( \; l/ R6 d$ W9 @' ^  Yclass CServiceCallback
        : public IUPnPServiceCallback
" d4 b# x9 }0 u) J5 l4 t{
public:
/ ~& C2 m; h/ j1 @. l        CServiceCallback(CUPnPImplWinServ& instance)
3 M: Y. u7 D- k1 C# ^3 O                : m_instance( instance )
; j: B. z- X& Y& z        { m_lRefCount = 0; }
   
; t6 e5 u/ A9 t3 o6 l0 W   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
: Y  j+ T8 `" F: J, x$ h   STDMETHODIMP_(ULONG) AddRef();
$ `- x6 _3 S( w5 [1 Y: q   STDMETHODIMP_(ULONG) Release();


// implementation
5 j. l$ H2 ^8 w8 @4 _private:
" @9 `7 {- ?0 C8 D1 l6 X4 F; d' i9 n        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
: S. g$ i5 x3 n# c+ _, x        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

4 y$ ~$ V$ _  m
private:
3 j6 j1 `  }* S  P2 e7 d        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
4 x. K6 ~2 s/ f3 }& E5 ^};
- V" [9 w/ i- B+ x9 I
- h9 R& e+ N6 H+ M! D
/////////////////////////////////////////////////
: M7 ^& v+ }7 R/ t* O/ y* t$ R

  B4 t: r5 H4 M. w使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
7 b2 d' a6 B' N/ {1 d; C. nCUPnPImpl::DeletePorts删除端口映射.