UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   % o0 L2 O8 w$ d; F0 k' X% }( J
3. 
   3 ?" ]" x* [  [! {1 W
4. #pragma   once
   
5. 
   8 S/ q/ Z% c2 X) U. k7 y$ w# {
6. typedef   unsigned   long   ulong;
   7 I4 q8 U( H7 i& x  k$ `& @
7. 
   
8. class   MyUPnP
   " P; u, Z9 n/ C& ~5 T
9. {
   2 R  q/ `1 ]+ l& U
10. public:
    , Y7 Z8 Y0 c/ n/ r% `+ F
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    # y8 P/ F) o, B3 Z
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
    + Q; K" y) g7 u5 `8 s- X2 N* j
17. }   UPNPNAT_RETURN;
    - D! z, \3 V- N& w
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    8 p  N- H' H) t- Z- h* U; B/ a' k* b
23. 
    
24. typedef   struct{
    ) q# T' |: Y9 {
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    : a9 `2 ], t1 E
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    ) k7 j6 h: F, K5 F
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    ) W# X/ Z9 q* F+ B; K' O
31. MyUPnP();
    : k. B: g, M* L- r. V. a# t3 O
32. ~MyUPnP();
    
33. 
    ' V0 x! V$ i6 V0 A5 H4 r; H; v. S
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    " L! X% }2 R$ N  O0 S: N4 z
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    + K7 e' c# e; l) u1 N
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    $ j) {& T# R; F
41. bool IsLANIP(WORD   nIP);
    ! Z- E, t/ x1 x$ {: k$ d; o5 g6 X! @
42. 
    % d7 j7 c+ E$ A9 t
43. protected:
    / b0 t* R; W( {
44. void InitLocalIP();
    8 W& M7 |3 t, c+ ^! J. I3 h. q/ m
45. void SetLastError(CString   error);
    
46. 
      Y5 p8 k4 K4 m) U9 M
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
      ~) y: f3 o9 I( U. C- X6 G
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    
54. bool GetDescription();
    1 S2 z5 C  F, ]1 x1 P% u. n" N  M- e( g
55. CString GetProperty(const   CString&   name,   CString&   response);
    / p4 j; G9 c7 R0 w7 \/ \$ D
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    ! x# i7 u( w2 f' i% s
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    & s1 \# C! R% F0 X) P$ j
59. bool InternalSearch(int   version);
    + F& n6 ?( T+ E( g( [- o
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    0 o& N1 Z# v/ {3 `* n; l
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    * S% @: c) ?% {; ?. M; z4 b# m) R2 X
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    # J9 O8 A- b7 B1 D, v( ]
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    % v  D% c5 k% \. P; x7 Y
74. WORD m_uLocalIP;
    / T0 P4 |2 [' n6 w
75. 
    & _$ t2 S& D1 \% L
76. bool isSearched;
    8 h7 k3 U9 @) Q
77. };
    * _& ^# Z9 n6 z% s
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   . U  k! m: K  G" R1 F4 C# M/ w
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   / v- }3 ^  p, |4 H' w' e
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   & v0 x* W( X! g- |' K
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    , M' Y! x% [! Z0 e  o
13. static   const   int UPNPPORT   =   1900;
    $ N' e% t8 a7 B: i6 S
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    6 ?) E0 s: p2 P0 n7 [1 p) |* ^$ a
15. 
    $ N2 T/ A( C/ D/ m' C! B) i
16. const   CString   getString(int   i)
    3 q8 R' Y7 T6 V
17. {
    2 U: y; e* d& |3 I: E* ~) |8 _
18. CString   s;
    + k8 `" F3 T3 {/ L/ P
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    ) h0 ?- v: P" q5 T% M: H# t
22. return   s;
    
23. }
    , A' m$ Y+ i* D' m& P
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    - |3 c0 Q# z7 X, W3 O! ]
26. {
    % @% r! b# n7 j8 ~+ o
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    5 ?, F1 C8 Z# K4 o( H. K8 e: X3 Q2 z
28. }
    
29. 
    ) ?$ i& m' X- q( @/ e! B
30. const   CString   GetArgString(const   CString&   name,   int   value)
    8 _) r8 s+ l& X# x& n  K% q6 ^2 j: ~* f( U1 Z
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    # ?6 H6 d$ t4 s4 [) x( s- u% Z
33. }
    , Q( p7 _% C- W+ G3 S
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    5 Q4 j5 x( J9 W' i
36. {
    ' w5 v5 \# R( W( t- B7 F" ]
37. char   buffer[10240];
    / ?2 g4 _# q' r- z7 a% x
38. 
    # e( E7 C( A# m+ B% z
39. const   CStringA   sa(request);
    2 G5 B* z+ B% v, w1 |7 L
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    ; u2 Q, S: B3 D  p0 o
42. 
    " _! s% x; \( Q9 }4 _9 k2 d4 ~3 h
43. uint32   ip   =   inet_addr(CStringA(addr));
    ( m- |4 c7 r( A0 F. j. j
44. struct   sockaddr_in   sockaddr;
    8 t; n" m+ B9 q1 H1 R4 H
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    9 y* R2 ]% l, s
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    $ V# Q2 z: _, O% g9 {
50. u_long   lv   =   1;
    6 i" c) `* Y; Z
51. ioctlsocket(s,   FIONBIO,   &lv);
    2 S! M- h. b! t: [
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    & ^0 ~: I$ q5 A8 N
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    ; E& }5 r6 J$ p9 U) ^% r
55. Sleep(100);
    7 S* z7 e0 c5 Q! v5 R
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    ( |( P/ }4 D/ s' m, @* M' p4 Z! _* d6 m
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    ' a  j  K% Z3 O
59. if   (!rlen)   return   false;
    
60. 
    # A* X0 E+ I$ B9 e4 g8 |+ k
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
      w# g" _$ q1 Y6 I0 G
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
      d" ^$ T/ R% D- h
67. {
    ' p# R  [6 m: @; J) S; v
68. char   buffer[10240];
    % c: @! g4 R% ?8 y+ N! k- z
69. 
    ; ?+ }1 J( l+ w4 j2 z( D! w3 x& P
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    - D) @* h1 N% o4 K; ~
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    ' Q. e( V( u- i7 _0 G3 W
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    & Z* f" t1 [& g
79. 
    4 x$ ~' K5 Y: a3 g0 I
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    9 V2 t+ G' s3 @
81. }
    6 M" c' n8 S* w" D* f# c8 }( W
82. 
    6 u2 U$ r0 x8 l, |4 C0 g( t  ?
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    3 [6 Z' D/ O- y$ N
84. {
    ! {( D7 v+ W" u) Y' Q) Q( _
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    6 A" d! j9 V3 ?1 H: n9 G7 n
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
      Z9 A9 n) h6 V5 g; L
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    ) {1 r5 b1 D6 {6 P; d& R
96. return   true;
    * b$ g3 k( {4 f, A6 N
97. }
    : p, m+ V! B  ~9 G
98. 
    ! Z# n: g- E7 V% ~% h. \
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    ) Y# F& M5 x! Y- E) a/ U8 t
100. {
     # b, I! ^+ O$ c% L6 H4 q' n0 V
101. CString   startTag   =   ' < '   +   name   +   '> ';
     & i5 F" M* |$ D9 p) m
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     ! I+ C# j2 J) h; j. D% k6 z
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     6 W1 i4 Z" R6 {1 Q, l
107. 
     7 y( ?9 E8 P2 a
108. int   posEnd   =   all.Find(endTag,   posStart);
     7 e; y9 m: n! f9 r
109. if   (posStart> =posEnd)   return   CString();
     1 Y9 k. @. c) h# O0 A
110. 
     4 {, Y, a( c5 c7 C7 s) F
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     ' x; q8 ?) f4 ]& a& ~) S
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     ) S: U' G. h' f+ R- W
117. m_uLocalIP   =   0;
     " k. F: q6 K- f' r) ?
118. isSearched   =   false;
     
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     2 T! K3 [9 @9 X$ A
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     : ?! Q' J3 x5 ]: Y3 O( a1 q
128. }
     " m3 y3 M- G8 d2 w9 |3 l$ W9 E
129. 
     # A* K$ x1 ]- S) p
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     , a9 _3 _4 @  [/ [% p
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     2 S3 p6 ~, D* B9 h. Y
135. {
     * J3 b9 k4 N7 s1 l1 X) q
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     " c, C* S  `0 m6 i3 ]
138. 
     
139. #define   NUMBEROFDEVICES 2
       q4 Y5 C( n' R# t4 \1 W
140. CString   devices[][2]   =   {
     7 ^; h. Q/ X) D( X
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
       k+ ~7 B0 s: t$ {  Q/ t' x, n4 m
145. 
     $ I! m& e0 E4 B1 G& v
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     & j9 o3 @0 \3 _+ a
147. u_long   lv   =   1;
     4 c- c1 r& g3 t- g& |0 D
148. ioctlsocket(s,   FIONBIO,   &lv);
     ' h/ T. g2 A, a* J4 {' m! K" l
149. 
     7 k1 @3 F: h* l6 L9 h# E- |9 C2 o
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     8 |1 S, Z1 c1 \# J1 L+ L  }
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);
     ) c: ^1 u8 j/ C4 ~* }7 Y( `
155. CString   request;
       _. _! z% M* L5 t2 j
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);
     2 j( q, F# s; c, w* j; F. S
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     5 q% u, M" U! S9 [/ C; ^+ I8 ?. p- q! t
159. }
     : N. G6 N7 j  ?) [
160. }
     
161. 
     
162. Sleep(10);
     ) q+ s; d( r3 W& P% P
163. 
     ) }8 S, Y6 A1 P
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     3 ]4 Z' |( m, _% p
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     + e# G/ p# }" \) D2 i
169. CString   response   =   CString(CStringA(buffer,   rlen));
     " |! `5 |. C' W  x$ N  d2 [
170. CString   result;
     " N' r. i1 N; L& Q$ m
171. if   (!parseHTTPResponse(response,   result))   return   false;
     + F: [: U4 \, u# G( w
172. 
     / R: D. E& G. o6 z* M+ C9 O* B7 {6 y
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);
     ' r: x9 I: a' @  _- \2 |* B. @
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     + W3 f/ q$ l; m& U+ x; Q
178. if   (line.IsEmpty())   return   false;
     , @# v+ u0 i" p9 G
179. CString   name   =   line.Mid(0,   9);
     2 B. D- Q3 y3 G% j: L7 M
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     : Z( T2 P+ O1 z# f* b7 J: F
182. line.Delete(0,   9);
     # f1 q; A' s3 X# J5 K
183. m_description   =   line;
     - U3 G& O: ]( ?/ D  {8 w$ R
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     
187. }
     
188. }
     7 i3 n; o' G# O* W0 f9 g
189. }
     
190. }
     
191. closesocket(s);
     
192. 
     
193. return   false;
     
194. }
     

复制代码

天云上人

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


3 h; G( F7 c0 l( [) k///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
1 g3 `! F7 u3 X2 E/ ~1 }# T4 J

#pragma once
#include <exception>

" Z4 t. a4 w, x9 t/ Z" n' X0 h
  enum TRISTATE{
        TRIS_FALSE,
: u" j5 Y9 Y4 t  C. B3 S        TRIS_UNKNOWN,
) ?0 k4 M7 {' P+ H        TRIS_TRUE
: P& ?6 S  P. C$ q};


enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
- U5 D8 u  ^& M% N        UPNP_IMPL_NONE /*last*/
};


0 I! Q0 [1 B3 q/ T' B# b

class CUPnPImpl
{
public:
        CUPnPImpl();
        virtual ~CUPnPImpl();
  N  {8 |2 Q1 W) E- X% w, w1 a        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
                UPNP_FAILED,
( P% n8 a0 L( X% P& u6 x0 U2 D                UPNP_TIMEOUT
) Q; W; h2 l$ j2 S1 {        };
: b) e" n0 f1 d8 @. S/ E

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
9 ^2 Y# f% E2 m        virtual bool        CheckAndRefresh() = 0;
3 b  X9 h% p. ~9 ~1 t! [        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
% P; k! @' j3 P$ F# N) B  ?/ p        virtual int                GetImplementationID() = 0;
( z: s. {% n! m& f       
8 `, {. n% e  q  H        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
8 S/ t4 m8 K; X1 \' e" p

        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
- V$ v, D  L) z        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

1 z7 W! r' H% i0 c. `$ ]% T1 t, ~
# q& C& I# v& P5 Q! w// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
* C' ?' b; H2 W& \5 z( L        void                                SendResultMessage();
0 S4 O# ?! b$ }2 m% Z7 [: h        uint16                                m_nUDPPort;
: F+ I& q6 ^4 ^" ?. p# ]& Z        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
* C9 @2 Q5 B% A5 y# Z2 I- v        bool                                m_bCheckAndRefresh;
. `" _/ G6 ?1 w5 P; |3 c0 i
& ?% K/ c4 |, l6 K' h/ m$ {+ A
7 E7 b* a7 @5 U7 G+ E1 Oprivate:
, e3 t% s  G- p" s        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
7 b+ v- _' g% W' P. @

( s4 ?/ J7 |" l$ m/ x& ~& Z. Y: h};
- Z9 j5 I; j( w- a4 p
" l. ]) [& ~: B* \
+ x& M# H- S7 M& U- {; u& V7 k// Dummy Implementation to be used when no other implementation is available
% ~$ f7 Z1 X6 \+ Rclass CUPnPImplNone: public CUPnPImpl
{
public:
9 z2 J) y& ?2 z7 P& g        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
8 l' e. [+ m: S* H# ]" V        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
8 y7 J. d3 H/ n5 q0 z9 B        virtual void        DeletePorts()                                                                                        { }
: B8 F, K- i( u6 D2 m3 A        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
9 r4 j6 s! l) G4 z};


7 d9 v1 T/ g- @4 ?+ ?& B4 F1 p/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类

3 k. P5 T. x7 r0 [8 j+ v
#pragma once
0 `$ n% a5 h/ j0 p, A#pragma warning( disable: 4355 )


4 B( [, s. G9 c4 H) W( w- ^( J#include "UPnPImpl.h"
#include <upnp.h>
( v, z1 n7 n/ a* K8 z0 S#include <iphlpapi.h>
: Y, n- l5 B! X: x0 Z. u8 q#include <comdef.h>
' D8 M2 A  V4 s) w  K( H; S3 Q#include <winsvc.h>
: L0 r$ x: v7 H2 [  Y% u

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




typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
$ x1 X& d; q: l8 d4 _7 t, G. k1 y6 Ttypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
0 [$ k$ p! u- I. N+ B1 }  Btypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
/ j3 J8 {1 W+ V' Dtypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
8 S3 ~% T! d& G1 b& T  M! z9 K) @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;


typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
, M" r  t( q, V/ l6 A4 ~) s" L  PDWORD pdwBestIfIndex
);
* H) p0 t' ^5 I& m0 z* N
) y3 d2 a" |$ @# e/ U/ ], n
4 ?$ y5 ?! I  e2 t9 stypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
+ f6 z% h$ M5 ~0 s% Z  PULONG pdwSize,
& T& q- j$ Y0 A' v  BOOL bOrder
);
4 d3 M: E4 `( l9 F
6 U! h( n  E$ ^7 N9 w8 C0 g
typedef DWORD (WINAPI* TGetIfEntry) (
3 C& N- S: F: t  PMIB_IFROW pIfRow
);
" E5 v0 M4 @: K* q% E

CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
+ R1 o0 g7 v4 R' h1 a0 X# \  C
: {0 _  l. p: E4 @% g1 [+ |
class CUPnPImplWinServ: public CUPnPImpl
, h7 R9 j9 T% o8 q9 W# t6 P{
0 V: A3 b! F% [        friend class CDeviceFinderCallback;
) ~) R2 p) m4 q2 d1 x        friend class CServiceCallback;
// Construction
$ p- m. v! t+ G: W7 Bpublic:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();


1 R7 d- c! Z7 @7 x        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
5 J; {3 `9 |+ M6 I  P/ d* G9 S        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
6 J" c1 P, L( e# x: M        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


! A2 I* L8 R6 M4 o1 Z6 R7 }. 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
" S- {) {0 Z, }1 k( d6 ]        virtual bool        CheckAndRefresh()                                                                                { return false; };
& |! n. ~3 V: t; W+ o; p  i4 |
3 e8 r- n; A- [/ z. W# _! W3 c
protected:
  g) u+ o6 i  |8 C        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
/ Q5 ?; e/ A2 T4 F4 E. ~        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();

9 v1 E5 w7 n+ {& t' i# u& o
        inline bool IsAsyncFindRunning()
4 o8 F  F- {- P+ J7 d7 V        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
: ]4 ?5 V9 y3 W6 \* A+ j% H                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
                MSG msg;
; z) ]2 H3 f5 d9 p# D                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
' b' d% Q/ f+ m) ~* n                {
' `/ I9 X  {3 m& M9 [% X$ t( _( J                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
- b) u4 O" K- y/ O; q) M, J! n                }
                return m_bAsyncFindRunning;
        }
1 O$ F9 {" {3 F

        TRISTATE                        m_bUPnPDeviceConnected;
* Y! E9 P- }; T3 U# z! H3 [/ @

4 {5 j- C# y. b// Implementation
- c4 {7 U- H7 O; B" ?6 K8 l        // API functions
: @6 x- S9 ?$ J! W3 I        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
, f2 x$ Z+ h& g5 C9 b/ A        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
! z- S* s8 u4 V* b) p7 |* G) W        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
3 x, L! Y5 s- R0 z5 [        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
0 |" R! m7 K  g1 b: Z4 C4 I, ^! Z
# ?9 O0 Z0 _4 }
        TGetBestInterface                m_pfGetBestInterface;
. E& J3 N! k9 R  S) J6 k% j        TGetIpAddrTable                        m_pfGetIpAddrTable;
* x* Z6 C! z) A% X4 D0 a        TGetIfEntry                                m_pfGetIfEntry;

: ?% }  V- x' t. {' h
2 X; S/ k: R9 V* H  n) ~$ C9 C        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
& R" K, r% m6 M$ H. ]& t: S3 K2 Q/ j                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
; A; }5 Y, f, w* S9 B                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
( R; j1 V# n2 m/ J; z8 N
2 |2 E; X, B2 i' S. ^
. v. g: F9 N) q                        if ( FAILED( hr ) )
) j! F" c" ~6 n                                return UPnPMessage( hr ), false;


- f5 r! H" `6 v' |                        return wcscmp( deviceName.m_str, m_udn ) == 0;
5 `- X9 L& [" S6 ?# Z                }
                CComBSTR m_udn;
' V$ V' T' i8 N        };
       
5 l- g: w+ W+ @        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
" g+ I- D+ Y0 \7 U- t; c        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
+ V( V4 u2 O" K% j        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();

" W3 r4 B; J# e
& w9 s, R& S( M5 K" b7 U/ P4 t3 ]( r        // Utility functions
" E' K( U$ A/ b0 Z" p* f+ j        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
2 I4 \; M7 X9 o; r& W3 u6 P* a        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
, S4 R! m9 g2 u' Y/ s        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
! G3 F2 T3 M7 C* K% E        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);

3 {( C- f# e' h  Q5 U
// Private members
" q- D7 t9 X& C, z+ n: [private:
2 k6 n$ x2 y/ m        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
" T, g! g$ k6 K: L  Y        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
+ F: s% e* ~0 S7 O: E; }7 c  I

        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
3 u( f) T1 x% k* F! B        bool        m_bPortIsFree;
; N0 n3 t) b( z; X        CString m_sLocalIP;
: Q& H, H/ f/ Q        CString m_sExternalIP;
5 _. p4 v1 q+ N        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
, a6 `5 l9 a. [        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
8 s6 B) x; z/ ^        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
$ Z: l# G4 @4 j9 @! m: ~# g/ y        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
7 x( `( v3 Q5 M  E/ |

};


* `  a9 C. I; ?; |3 `. v# z// DeviceFinder Callback
0 ?. P& U' K: c  ]% [' Uclass CDeviceFinderCallback
4 E- L# r! t+ Z; Z* J5 C        : public IUPnPDeviceFinderCallback
' v1 D' c7 n: D' ^# R0 `3 e5 |{
1 \8 [3 [9 r) E- e% Dpublic:
9 e4 y6 b* v4 q! P4 u! H        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
  }4 q; Y1 d- l. s& ?- |        { m_lRefCount = 0; }
' C8 F1 H" t: R- e$ C+ H0 X& x

, \/ R% I. h' z2 Z# |   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
1 m4 N2 v# l% g9 K- D  }' k% _. e/ [   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

7 g$ n# b' `: l" A3 K
8 Q. p' e/ p. ]' e// implementation
3 e/ \: T. t' v( D$ Kprivate:
: ^9 }4 j2 f) _. J& ?* A6 S  J        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
6 F- I, t) t7 m/ T3 y        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
) C' J) s. o, P2 R! ]% s- {3 C7 J
4 W6 y. ?( v( Z0 x4 N' `
private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
* \' u- k5 r) ?  K8 N( S' S

; H8 M# [8 {6 s8 ~2 [// Service Callback
class CServiceCallback
8 B& U) c! Y4 ~& R7 [: y+ q        : public IUPnPServiceCallback
{
4 ~9 w  L  ]! E, Npublic:
& v$ A+ ]$ ^1 o' b  N        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
$ {, I0 G8 ~6 y# Q7 E- B! U        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
: l) f! |; E: P6 o4 O- N   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
8 r$ H- u0 v1 ]5 X

// implementation
( O9 l# d8 F/ E; c+ y7 s6 Iprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
8 I8 r6 E5 O6 e
5 o0 d4 j" q/ E9 X
( \/ i; S; A9 f9 @) U: Z; U/ kprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
9 X" p0 \, m  w+ m};
) j/ D/ m8 t& s
$ X$ W. m' n6 H4 a  e/ b
# k% _$ k9 e$ D, U1 z/////////////////////////////////////////////////

6 `( A( n; D$ x; @- L) L% R+ u
. X& _+ A0 A. Y6 u; J+ v/ e0 z3 u, s使用时只需要使用抽象类的接口。
$ ^5 y! j* C$ @4 |6 P5 LCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.