UPnP

天云上人

/*uPnP.h*/

1. 
   3 M9 H: y: R* |7 Y2 T) m6 d# S
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   * Q, _% d! N* t+ c. T) R
7. 
   / _7 F4 h' C$ g5 ?
8. class   MyUPnP
   
9. {
   
10. public:
    ; w! y& |* r6 z5 ]+ N' d
11. typedef   enum{
    2 x7 u/ ?' C* f  P
12. UNAT_OK, //   Successfull
    2 u( ^; N9 Y0 Z
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    ' |/ x0 N" f5 T
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
    
17. }   UPNPNAT_RETURN;
    & \- j( \6 e" s* t; Z
18. 
      I- U6 b) ^* t' W; K
19. typedef   enum{
    ' U& B3 |! @" f! f" u% b% x% \# c# G
20. UNAT_TCP, //   TCP   Protocol
    ) V( t! n) s/ K$ T$ I
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    2 d  }+ {' W- R5 n; A6 W; r
23. 
    
24. typedef   struct{
    ) p% N5 c" r% w. g8 i# z4 E1 B
25. WORD   internalPort; //   Port   mapping   internal   port
    * f( U5 d# l( y: q  l. \3 ]
26. WORD   externalPort; //   Port   mapping   external   port
    8 I* J  l" R- m' j
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    & z5 z$ [: S4 Q- `$ n! p
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    . d$ Q3 u- g7 z' a; \8 H( G
30. 
    - ^1 Q/ \$ d7 g4 f* E5 p
31. MyUPnP();
    
32. ~MyUPnP();
    5 `) J% n: }# H  C$ ~/ i% B
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    % k( G8 P- m. m% m0 X0 G
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    " _; x- Y7 U  w: M7 t+ ^# E' p
36. void   clearNATPortMapping();
    
37. 
    $ F2 V$ x* {% }# V) L
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    9 \! ?$ \: m5 w2 U0 I$ g
40. WORD GetLocalIP();
    # Y- V. J3 }4 k
41. bool IsLANIP(WORD   nIP);
    # y  r! Y- r3 b9 L) Y2 s& {6 f
42. 
    : }* f2 J; p% q5 M: G
43. protected:
    9 U$ R, M% ]7 c2 r! [$ [3 b9 Y: }
44. void InitLocalIP();
    9 [6 A- T2 k+ I: Q$ G$ m
45. void SetLastError(CString   error);
    
46. 
    % g: R8 Z  i0 F' z" c6 N
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    + t, `: i  H6 ]
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    + @8 P" i* h* ~( ~  F* P/ X
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    8 ^& x- c- I% z0 l- t
52. 
    - [" Q# |# Q  E( U
53. bool Search(int   version=1);
    4 w* d5 r3 U" i, Q0 x& O1 c
54. bool GetDescription();
    " t' I8 m$ K& O
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    . X0 g9 i& u5 U
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
    / g7 D" H9 q$ T0 ]  r/ L4 f
65. CString m_friendlyname;
    8 ?6 j; }& q  p0 ]. N
66. CString m_modelname;
    
67. int m_version;
    
68. 
    0 F  y2 S2 C7 _- e
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    4 {  }  r; A! g9 ?
71. 
    6 }4 d7 D$ W9 H- b/ V2 ^- W' W
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    5 \& X  e/ k' c8 s- S+ T
75. 
    
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   8 ^0 L" F4 l5 Z9 F5 T! Q5 V& N
2. #include   "stdafx.h "
   8 b: @+ K6 H2 f8 |
3. 
   + H7 K( v5 _+ v# I. }. A4 n
4. #include   "upnp.h "
   / A: {/ ^& f( ~6 h4 ~1 E0 ?
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   , X: X  i% G5 u: \" T2 w3 q
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   9 p3 Q, ?& t. m- ?0 P2 M% y
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    0 t0 q, U4 r# {8 x  O
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    6 `0 j7 }* b+ z  Q; I6 O
16. const   CString   getString(int   i)
    4 o% v* `! _! U: r5 B1 ~
17. {
    . ?; [' ?0 c: u+ h( z) B( _% j
18. CString   s;
    3 [, k4 X7 a" G. h0 a$ n$ f* }& n1 p
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    * P  b$ i8 i( o$ `+ I
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    $ p; a( y: Y# Y; M6 V6 z
26. {
    ( ^  ]/ w' r4 x  ?0 Y
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    , N4 Y- i/ f) k7 a" b8 g
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    : c, n& v7 Y. {
33. }
      G) l- W& F1 R; l
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    $ V3 ^, F. f5 e) v5 C
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    ! \0 I$ q; c7 o" S1 D6 ?' C
40. int   length   =   sa.GetLength();
    * ^2 J, M7 x) |" ^+ H1 r
41. strcpy(buffer,   (const   char*)sa);
    + n1 d( s5 F( i1 c4 c) h3 I& ~5 k/ x
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    4 z$ m' m: T! }5 w( R  ?+ ]% l' S
44. struct   sockaddr_in   sockaddr;
    2 m  w  P6 V: t7 g2 H, v% C8 s
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    1 ?' a. y7 V" s- r
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    9 F2 o3 a" r& I; ^1 }, n  X
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    % P/ f' O! w5 H2 t( h
50. u_long   lv   =   1;
    - u! n+ ^$ z$ B1 x; d( _' T+ w3 M
51. ioctlsocket(s,   FIONBIO,   &lv);
    % ~) p! X3 _. V% W, Q' W& u
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ) \3 q5 z5 n  p0 C/ M2 D, V
53. Sleep(20);
    + y8 Z; G: G& V2 X% U+ c  ]5 K
54. int   n   =   send(s,   buffer,   length,   0);
    1 {' y- j* A% N0 ?
55. Sleep(100);
    / {4 G' y! Q: U  k" f2 n. y: e
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    8 W) ^4 d2 I& x9 h* `6 Y7 f
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    ; j! J/ Z+ t. q7 h2 n3 |/ Q% C( D
59. if   (!rlen)   return   false;
    ! o# t1 f) A8 K; l
60. 
    % i" w- f, g( _* U2 j( V! `6 L
61. response   =   CString(CStringA(buffer,   rlen));
    % ?$ h6 V% C& S" x" t& `* \
62. 
    3 L* A! D6 Q8 u" v/ }4 c
63. return   true;
    . `7 r# e8 N3 `3 X+ g( Y; `8 T
64. }
    1 c; ]' n+ i) |
65. 
    . N% d* B' [& `8 J' D8 u- \# c
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    + U% G* k1 A/ G5 r+ Z  e/ l0 {( M& U
67. {
    
68. char   buffer[10240];
    ! @' j( p2 e% K7 ^0 q5 h: p6 r
69. 
    - d% m9 h. W6 N
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    ' ?4 p6 H. U" A: r0 g
73. 
    
74. struct   sockaddr_in   sockaddr;
    % q% w! x9 X4 n: R
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    : z5 P9 R" W! i8 h# [- W4 u
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    + z# ]0 q! H7 _2 ^
79. 
    $ s9 B; b2 {4 ^  v4 r- X- d& L
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    
86. 
    0 X4 z% p! N; I8 ~
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    & U+ O7 i7 F- U0 n: W6 a
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    5 f; I/ e: H- W- ~
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    , z6 v  @" v6 Q) `3 @6 u
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
      \( F, g$ v) L
97. }
    6 N2 l) j2 Y/ J" m! L6 H8 y
98. 
    0 b+ a, K; X; ~/ i6 c5 ~8 F" I2 Q
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    3 P( G2 Z# x0 I$ Y7 }. O5 v
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     , D6 m+ F! g  s8 u7 U; d1 j7 y
104. 
       b1 S# L$ v: q" |& S
105. int   posStart   =   all.Find(startTag);
     4 W) y% Y6 t- t( _
106. if   (posStart <0)   return   CString();
     / W7 a( {6 n& E  c2 C
107. 
     ( n' V# p' N6 g* g$ `( A. A
108. int   posEnd   =   all.Find(endTag,   posStart);
     + R. o$ T$ n& ~. H/ M, n8 l
109. if   (posStart> =posEnd)   return   CString();
     & K, m& ?/ p& n& ~( ]- A
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     . \5 w! W! }* M# P8 W! H
113. 
     . b5 {+ G: U$ X; q2 g0 l
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     , Q  w# M  _& P- j
118. isSearched   =   false;
     % l  v: y% a" p& B# I
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     9 }' t& t7 ?; W! @
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     6 g" u; B( v3 k% M0 l0 B, Q0 f
125. while(pos){
     8 [; M5 J9 Q- l2 n% M$ p; P+ [. s. E/ q
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     , R+ C( |) k& U7 ~9 v; y
130. m_Mappings.RemoveAll();
     : q: a7 A  g! y4 G, @$ d
131. }
     
132. 
     
133. 
     . m: Q6 R1 o  L+ P
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     3 _4 ^' S" O: a( Y3 C1 T
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     1 c  Q! {. d' R2 y5 I- D
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     " j: C. t& p# S* l% n6 }
141. {UPNPPORTMAP1,   _T( "service ")},
     . Q) {7 Y+ {% u. D
142. {UPNPPORTMAP0,   _T( "service ")},
     : z9 j8 m! p' \; X
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     % a# w2 t: |+ E6 o+ u& Y
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     9 t% R4 t% [+ N, D
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     
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 "),
     # O& a3 ]3 d! _
157. 6,   m_name);
     ' ?9 C' D8 D! j7 k* L& v
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     % x* W. B& O+ B' P7 v* k9 t
160. }
     
161. 
     ( j% \; P" ]0 g: A, a+ B! ?3 e
162. Sleep(10);
     # S4 \2 l, q4 V0 e0 w
163. 
     # x: D; Y6 a/ S' L& k
164. char   buffer[10240];
     7 }0 h; k4 z! q! b5 Z; S1 o
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     5 J2 p0 B5 z( Q7 {
166. if   (rlen   <=   0)   continue;
     1 C* ]% Y/ s7 p! L" Q
167. closesocket(s);
     , U% Y6 g1 Y2 `! B1 x% n; ~
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     & D. C  g. i, X3 V1 G8 @/ D* m) p
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     5 r2 H9 r; Q% C4 R& [$ d" j
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     ! t" {, z$ w! I
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;
     ; Q/ a, c$ f5 ^) T* Y5 ^
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();
     " A, J. y# Y. `6 ^
185. return   GetDescription();
     
186. }
     
187. }
     - T" v5 ]3 }( \5 V4 b
188. }
     
189. }
     ; Z1 i* P0 V# Z6 g$ D; B- L' g
190. }
     
191. closesocket(s);
     
192. 
     
193. return   false;
     * {# b8 A: B2 L! C$ _! S$ r
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
2 d7 N. k* A$ |* I( c
  D* S7 |3 I/ l0 c# R: a
5 F3 z, M* I. F* J, n+ i///////////////////////////////////////////
( {4 y  o2 B% x3 j. K3 F" M//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
/ s+ u5 @  C8 Z8 a) S
( V6 i' M4 m0 E8 Z1 R
: p% C$ t* e6 ?. q0 [& s#pragma once
- C8 i$ C# n1 e#include <exception>


7 M) z( S- a5 U4 l: F* |8 M  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
! }3 k5 K# H3 C. K/ I- N; S9 o% H+ O! _};


enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
6 T; v# b6 u+ s% h( ?6 b3 n7 K        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
* N- D( A1 `7 ~+ i% @};


% }8 u  H1 R- T: `! n

/ L- \$ X$ u9 f9 a" b) U& [8 gclass CUPnPImpl
{
. Y+ v8 @( q( B- C2 X2 Hpublic:
9 I% ~' \, |$ }5 z        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
  E, @, |# `) f: {1 M                UPNP_FAILED,
                UPNP_TIMEOUT
        };


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
4 Y: f& q' @* V. Q" c9 @$ M        virtual bool        CheckAndRefresh() = 0;
1 L. W  X* I5 v        virtual void        StopAsyncFind() = 0;
( K: R8 F  c0 C5 i+ A4 w' Q        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
8 C& }5 v. ~) M  Z4 x; }) q        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
8 B4 {# @# d1 i
' [  @2 Z$ H$ }4 V6 t% V8 D( u
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
7 Y' X4 F. D' B& v: L* b( D9 t2 h        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
( R" ]6 S. s, a0 C( A4 u        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
) {+ ?" o2 g- {6 h! u
2 ]6 R$ E0 ]$ S; {0 e# j' U
// Implementation
$ E, s% X# ~' M$ F& R1 u( H+ zprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
, }8 [0 @4 G  b9 E! e7 R5 a$ M        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
' F$ g/ q* {# G% Z; L6 {( i        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;


+ Q4 s7 v9 e4 t- \( r( x* k! qprivate:
8 [; S: K! O+ p        HWND        m_hResultMessageWindow;
7 [2 H+ b1 B, Q" D! F        UINT        m_nResultMessageID;

+ X: n/ U: j: x2 H" m9 a
) C8 p# R7 X5 W};
/ I; @: |; d3 [( _6 u; u; O6 W
$ ?. Y; ~  |! |+ q
// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
: E( d- G1 z0 p# w$ \* J        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
# r4 k' g, y* M, C};


/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
0 @: k) @' M% A  A, ~

#pragma once
#pragma warning( disable: 4355 )
6 ^7 ^: u- S, K7 L) n5 H
1 {. X5 u. |- n" }. w" P2 ^: V; p
- P$ u2 K! }7 c#include "UPnPImpl.h"
/ H3 [4 a! Z. @+ a#include <upnp.h>
$ J$ [& ?! p# e2 u6 q6 Z#include <iphlpapi.h>
9 j: l% d' M0 Y/ @0 L#include <comdef.h>
3 _8 T5 l2 h  D4 C7 s#include <winsvc.h>
+ S; j7 s" f: N( a
4 Y# d) q; Z7 W0 F# z3 y5 M8 z$ D
#include <vector>
#include <exception>
  h7 x8 d9 R2 ?, C; `6 I#include <functional>



, c& g% A" t' c3 u
7 M  Q( R: }' V! Etypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
4 t- H5 ^; {; t" B' {0 Ltypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
" d* _1 D7 E1 t, `$ e/ itypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
0 s$ n1 R+ ]- {' ttypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
+ B- @" n. w5 wtypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


, s0 `4 E# r  ?3 w1 }typedef DWORD (WINAPI* TGetBestInterface) (
: O7 r7 v( f6 I5 e0 ?- Z  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
# E6 Y6 H6 P2 M, r);

* v2 x5 Q* w9 p* y' Z  h
$ d5 T' W( E5 r  p  ]" dtypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
- I  C. ~. k1 u- H3 [2 e);

* [2 f) c# p* l+ D3 w6 e
- F& _9 g; _' G6 ?5 H. Gtypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
* D1 z7 }' K, t8 \$ F* q);
( N* q$ I5 O6 g% u% b. r$ {' `

: i( `# a$ O& n. t* ^CString translateUPnPResult(HRESULT hr);
5 ^/ p1 s' @1 Q3 }/ z% K2 F- zHRESULT UPnPMessage(HRESULT hr);
0 ~) Z4 S. f6 S$ o; b- T; |0 S/ s

class CUPnPImplWinServ: public CUPnPImpl
{
( `8 O: @- K5 X2 P4 J        friend class CDeviceFinderCallback;
; u' l8 o  d, B; x3 N- \: e        friend class CServiceCallback;
5 u2 S3 R4 }% h! F5 _# {// Construction
' U+ k9 t7 X* a* e2 epublic:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
7 @0 ?' Z6 |; g4 b& _+ J

" b& |/ S0 ]6 P3 k) Q% B        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
* [: a" n# w) ^: k. F        virtual void        StopAsyncFind();
* @% N0 ]# G+ h3 X! H. k" U2 f        virtual void        DeletePorts();
        virtual bool        IsReady();
7 w: P8 I4 o5 D; N8 k, ]& T; y        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
- o) i) E/ g# v7 a, @! {( p7 I4 Z

% q* F/ P- M+ h8 C+ x6 z        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
, w5 d. N; e- [8 {; @0 E8 S        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };


protected:
! @( T. m' y+ r) U0 ?6 K3 e2 ]; N        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
# D" y; y+ e4 E$ p        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
1 `  D! F0 N: O8 t2 m        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
- [+ Q. [  Q! V# X- t        void        Init();

- T' O& ~- n' F6 h/ W
: u/ e! Q- K; S! X1 a7 N# h' r        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
$ k; G$ _, z- s- I& f+ a" _- h                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
$ A8 q# P; R- K% U9 F                }
: j, |8 {& s- |, R$ q2 L                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
- \; R3 r! {' U# Y                        TranslateMessage( &msg );
- u$ q, S! H& c2 u3 E9 N( |- g                        DispatchMessage( &msg );
& n5 d. A5 S. t: o4 K% K* f1 m* J                }
( j: g/ C7 r3 I+ X; w                return m_bAsyncFindRunning;
# P1 p( E  s' b. P        }
, R- G/ `5 q/ F1 i) z, u7 V
( L$ n' ]  ~, g# H4 F. u
1 X# I1 {. K* H: [        TRISTATE                        m_bUPnPDeviceConnected;
! g4 E0 o# d1 ^5 E5 p4 a
8 U; }8 J3 B# h0 V. C0 U
// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
- ]  d: h( p0 h        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
! H3 n  C( e3 t, U5 \8 e" v        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
% Q5 c3 i  F7 @7 s; M. x7 Q! ?+ ^        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


% c+ J0 g& R& I* ^. n$ c        TGetBestInterface                m_pfGetBestInterface;
- u/ Y! J  ^# q( v9 l7 s* R: i        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
+ r4 H2 u$ ]5 B: D! X. _

        static FinderPointer CreateFinderInstance();
5 z% C; r8 X# H+ n' ]        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
0 @" p7 [- g+ Y                {
4 y) R# W* D; N* D                        CComBSTR deviceName;
7 }" u8 W1 r& q+ u/ ^                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

+ A5 t! V% v$ u7 F! [. @. ^
& y: i+ P' K  ?5 t+ c0 c                        if ( FAILED( hr ) )
! x( R# m6 S- K                                return UPnPMessage( hr ), false;
9 W& J5 k% a/ @$ r

                        return wcscmp( deviceName.m_str, m_udn ) == 0;
# F- K& x0 Q! E5 ]+ q% V0 K: G                }
4 D2 x: Y5 |3 Z3 u                CComBSTR m_udn;
) {. G" j% W1 Z/ T! R        };
! Z0 R  f& m1 @9 `6 a0 _- T       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
6 K( I0 d5 b6 c4 L; P! T( J        HRESULT        GetDeviceServices(DevicePointer pDevice);
( L" u( h# l- z* Q        void        StartPortMapping();
: u. A  S0 P# C1 _* @8 I' g        HRESULT        MapPort(const ServicePointer& service);
: B1 A" `$ b4 w* G4 y( Y        void        DeleteExistingPortMappings(ServicePointer pService);
: Y" v" L$ b1 B) Q        void        CreatePortMappings(ServicePointer pService);
  \) j  {; ]( c  }* t        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
: n: {5 D( e  Y$ U% O6 n4 t, [        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
3 X0 r# r* z6 Q( L5 w, j/ Q                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
: S$ J9 S( a1 b& T
* y" G$ D" l  C0 T$ B! M
        // Utility functions
# r. m; V& ~6 N& w1 _8 n: D        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
, i  _$ {5 b/ f  ]& G0 i6 h' B* E, @        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);


4 d) ^2 I1 W  f5 Y( t: R& ]/ e0 o// Private members
# |! A- D, ]3 y3 E) \private:
4 T1 Y) P& E; I2 H  h        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;
; {5 w4 J; K) F: x/ x' g; u$ M" Z        ServiceCallback                        m_pServiceCallback;
% L6 Z7 H' M: c6 K: q" A+ H- H
+ @) d. S1 D8 z" h3 E: S1 H, z
9 s. t1 f  v/ d4 C        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
. _( S$ v  g& V, U: e/ X! R, J: X! s        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
1 M9 i* X3 Z. X4 m        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
6 @( n& P- n1 K        bool        m_bInited;
1 }+ y$ N; R. D        bool        m_bAsyncFindRunning;
$ S# V8 r, a' @6 s5 e        HMODULE m_hADVAPI32_DLL;
  B9 [9 r, S4 W6 s( p* ^7 V        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
2 r) L- q9 Q& u3 \* T        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;

& z, ]! H  W- {2 Z3 C) r( y
};
" b( X  X1 e( l+ L
! c. n3 e0 A& ]' e- ?5 j
8 X" [+ z. y3 a- O' E: Y// DeviceFinder Callback
class CDeviceFinderCallback
$ z5 t  D: e. ]% h7 X        : public IUPnPDeviceFinderCallback
& x6 s# @! n- c. q: H0 p{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
4 {2 |7 H& ~& }3 a  J+ V$ G8 \                : m_instance( instance )
        { m_lRefCount = 0; }
: |# u* m9 U7 k9 [4 r# I

, Q* h, G' w) M% ~   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
$ w: D. N) b6 p' B( @$ F   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
+ f; {  \# R. T! _
- R9 [$ O/ E/ h% ^* a5 E
) k3 A" X, ^5 _* B4 F// implementation
9 a( r6 {: i( ^" O$ Qprivate:
. E+ h' ?/ F! n. c0 D        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
. K( `$ B+ d- f/ m

private:
. Y! {' m& B9 ?8 ^3 S, z, c/ d# O        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
/ f8 N1 A' Y( }* Y9 a};
* a9 f1 l* i8 |

// Service Callback
, r; L& ^# v$ X! Mclass CServiceCallback
% K# v& \  c. S8 }        : public IUPnPServiceCallback
' T2 v6 \% B$ F) F7 S{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
% i+ W" o& t3 |        { m_lRefCount = 0; }
- H& W# u% y, m3 Q   
4 d, C  u# U# @) P1 {   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
; u! U  [# j- ~  a; P7 W   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

& K: [6 y4 X& O' `
/ l2 {$ _# u) w$ d// implementation
private:
; r$ Z/ P! U! ?  l* ~! A& X        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
0 h' P& {7 F; [1 J7 a& W

3 u- V( d/ P6 sprivate:
' u  G8 N+ W( {6 {1 ~        CUPnPImplWinServ& m_instance;
0 q% k% V0 d6 q- W3 x# \        LONG m_lRefCount;
6 Y' f" x/ [$ T3 I; }! n& ]; l! J};


/////////////////////////////////////////////////
* z5 L7 p" J; Q

8 o4 l$ ^& R, y' b4 B8 r2 v使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
' V$ M6 v: V& m/ t: nCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
9 i4 t3 E$ V& P! b" ~8 h1 hCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.