UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   . ?% V# }& Q$ l6 S1 k
3. 
   ; y- G4 r" f) f0 H, z# O  I3 p7 i8 i
4. #pragma   once
   ( }+ s9 c1 _* ^: R; c
5. 
   - v7 B5 f* y# Q: i4 M
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   " S# F0 J5 O: P+ s; k3 T
9. {
   / f0 C) U+ S; O, v# {
10. public:
    
11. typedef   enum{
    5 j1 F1 w) J, f
12. UNAT_OK, //   Successfull
    # H: r; B$ z6 f3 C, W; k
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    / c3 [1 q1 f! @3 J, k2 N. H
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    2 S& t/ [6 O- m0 n) |, ^/ L( j
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;
    
18. 
    4 d/ b; C; ?# _8 j1 w" ]9 g
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    + w3 w# |9 H: H6 u; _2 s
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    # x0 N# A1 |. `7 _: c& w0 O/ A% L
26. WORD   externalPort; //   Port   mapping   external   port
    2 p: w* Q9 y/ X( R! r8 k0 {; H
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    % a* r8 R" x4 b6 D6 E9 L2 G/ M# O
30. 
    8 }0 c1 I' r1 ^, r3 ~& L
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    0 W3 B% S. M# [' q3 g& E0 m. W
36. void   clearNATPortMapping();
    - [/ H+ s5 N0 O4 ~; {
37. 
    % S: u# S- d$ A8 Q. C
38. CString GetLastError();
    - {+ v) {8 k. v: E* T
39. CString GetLocalIPStr();
    0 E, ]+ w- o7 e
40. WORD GetLocalIP();
    3 j8 ]* F* k/ i1 O, y6 W
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    
44. void InitLocalIP();
    & y9 \7 d3 T! }0 {8 g
45. void SetLastError(CString   error);
    
46. 
    ( O& |" s5 H6 H
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);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    + l: K/ G) n1 w. T1 Y
52. 
    8 D4 M( w. V4 u9 s, e3 V' Y
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    4 h* b  y; C, ?7 J2 s1 b# \
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    ; _% y& |$ S( m/ }# J& G
57. 
      S1 X* R) H% S# x% H
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    4 |! K- C5 [) ]: p/ K
59. bool InternalSearch(int   version);
    , R5 H: s" T, x( h0 q) F
60. CString m_devicename;
    
61. CString m_name;
    6 E1 Q0 y4 ^# x- A6 h
62. CString m_description;
    ! Q3 ?6 i% ~1 C) G" a  @5 b2 f& d
63. CString m_baseurl;
    , c4 O, R( _% j6 e
64. CString m_controlurl;
    6 H$ N4 d# w7 v$ j$ a
65. CString m_friendlyname;
    
66. CString m_modelname;
    9 [! w0 M3 s0 A5 M4 J
67. int m_version;
    6 N- @* }+ j4 |9 z  F. t3 p
68. 
    * [& g3 K, _% I
69. private:
    . ^, Z9 N, i8 L& C$ h
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    0 j0 [( O. i" L$ z9 B' E( i1 Z
71. 
    - i: v& _; p) K$ g( U0 E$ A2 [
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    . c% b7 z0 d! _  @; ]
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
     X# v! J. ^/ q8 j+ v/ o# }! M
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   & m9 N7 {5 g* u" T- N
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   ' q- C$ a4 `* q  j# Z) f
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
      o2 K4 d4 Q/ m, j0 R
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    ( V$ A$ i: P) P8 c# t# B
16. const   CString   getString(int   i)
    
17. {
    ' G5 y. A2 U5 _( o  {9 P, e/ f0 K
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    
23. }
    ; _2 V+ _1 L. p6 j
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    4 b" P1 G( O3 V3 _+ p, D" S
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
      n7 {* \; M( U0 U. t, e
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    . b. J" g( W, d2 b% g9 T
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    3 c% ?! H9 w; Y" L) A
33. }
    
34. 
    / f  l8 Q# ~4 C4 C
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    
38. 
    + b5 i4 f7 s/ S! }% V
39. const   CStringA   sa(request);
    0 X/ ^9 t: x& g. M1 j
40. int   length   =   sa.GetLength();
    , ~1 X* v+ ]  M) |0 `
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    " v$ P# f0 U% s+ b3 g8 D/ \
43. uint32   ip   =   inet_addr(CStringA(addr));
    ' d3 {5 ^' K, R
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    . u( P5 b) Z/ d* J
46. sockaddr.sin_family   =   AF_INET;
    " H% m; H" y7 b7 c, N# V$ B
47. sockaddr.sin_port   =   htons(port);
    " y& x' L3 W2 i/ S/ C5 d) @
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    ' r3 a0 \& P" S# J9 h+ u- G
51. ioctlsocket(s,   FIONBIO,   &lv);
    0 r1 p  i, M+ t0 j0 x6 c3 o
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    0 w" s2 l, w! Z2 T' L9 ~& q& I9 C( \3 _
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    $ g! l$ [4 B8 t5 @! W4 f
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    + s$ j- s! E* ^2 ]( f
59. if   (!rlen)   return   false;
    , J5 s1 V, \0 j/ R1 K2 X# n( X
60. 
      @) g) y# ^! Z- Z; ~8 V
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    3 e8 m* i( S: y; L( u
63. return   true;
    
64. }
    ' x7 G: D) D/ {% v8 e) `# l' J
65. 
    1 H! v' j0 p* T& ^" m
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    $ p- @" H5 h9 w! i7 I& X
67. {
    9 G4 J- Y+ u4 I9 k( Z3 Z. f5 t
68. char   buffer[10240];
    
69. 
    # L6 o3 f+ G8 D9 f4 L1 E7 X% f
70. const   CStringA   sa(request);
    : S- ]5 c# P) W* X
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    ' T  A  Z# @7 w
74. struct   sockaddr_in   sockaddr;
    3 _8 f# k8 i' A* Q$ _
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    ) b# [! R+ x9 Q( q  z$ x, {9 V9 ]
76. sockaddr.sin_family   =   AF_INET;
    # R6 X& N4 W. [, a/ p
77. sockaddr.sin_port   =   htons(port);
    , H& M2 L' i/ G' [9 X# y, l
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    6 t8 y) p1 w3 [( H* f) K' }5 N% }
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. 
    8 P( y; _8 T' W  M3 U% ~
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    $ h4 x* ^( R( ^. ]$ _- A
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    ! T* C# M$ k- ~, ~
93. status.Tokenize(_T( "   "),   pos);
    2 _$ S7 b$ W# F# X! K+ t
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    0 u. Q" `; S% @) c" ]
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     ( I. t  i7 o( k% E8 |' j% O' y
101. CString   startTag   =   ' < '   +   name   +   '> ';
     7 q1 F9 T6 N" `, O
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     . U; ^2 V9 x4 O
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     7 `7 t1 S. A6 Q+ L
106. if   (posStart <0)   return   CString();
     
107. 
       B- Z# o! k' b4 |
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     / R  C5 ^7 q. u$ @. o8 O
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     # i; i3 Q1 w1 }$ Y2 Y
112. }
     $ q- E: Y. m: k, T4 |8 A7 r7 A
113. 
     : R2 @5 x6 i. k) }4 K
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     . ?# O7 J& p: X: g7 [% }. v
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     / x: r. T  B4 u0 [
119. }
     
120. 
     3 m* n8 P  B, v. k( a( V3 _
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
       F; E, X% v2 N2 E: `
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     ) w, P7 N8 F4 e' V  r, U8 \$ c
128. }
     8 |5 n0 r" A8 {4 S. n  H
129. 
     - f" @4 R9 v+ Q. D& b' i& a
130. m_Mappings.RemoveAll();
     : M) e# S) l8 t+ D
131. }
     4 a3 s/ s! W/ m1 D
132. 
     # @4 O2 [" H8 t9 F! x5 M9 A' ^$ ]1 X
133. 
     # Y5 x0 \. |: v$ L8 L" R
134. bool   MyUPnP::InternalSearch(int   version)
     9 n# d  ]) W3 }% g3 ?+ g
135. {
     
136. if(version <=0)version   =   1;
       e4 x& O: e/ h0 J
137. m_version   =   version;
     7 V4 _, k" z5 Z1 v
138. 
     / @9 J0 Y" w/ n, H1 I
139. #define   NUMBEROFDEVICES 2
     3 f3 c$ V% Q! y3 r
140. CString   devices[][2]   =   {
     # O& x7 C% _3 e! b) N
141. {UPNPPORTMAP1,   _T( "service ")},
     4 X2 C* q. g( d* K: G
142. {UPNPPORTMAP0,   _T( "service ")},
     / |0 ], o6 w$ T; [. {+ E, t+ u
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     3 ]! Y( u) D; x4 n: F  l
144. };
     
145. 
     7 D& c# u% M4 O7 g$ c
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     2 e1 v; N' Q( W# W
147. u_long   lv   =   1;
     ) {# ]0 t! S) D, h
148. ioctlsocket(s,   FIONBIO,   &lv);
     4 f. y5 _' S4 {& v2 @1 Y
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     8 S0 B1 K5 J* n4 E. D( J) H$ Y9 c2 x: P( 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);
     3 v% y+ @  }; o- @' ?  c
155. CString   request;
     " M3 L2 ]1 I% W( h/ B+ Q
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);
     4 U1 L& ~* ~2 o7 q6 y7 U; F
159. }
     
160. }
     ! C5 h7 [- U: O8 y& ^
161. 
     
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     0 E& k) A- c' ?$ c2 |: G
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
       Z6 B" I  V: K0 J, ]3 t' j
172. 
     ; T  s6 F- S% Y) x( y, C
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     ' k# J+ P: y' Q* _! d% {  E% ~
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     / M4 n$ P* T3 i- Q
175. if   (result.Find(m_name)   > =   0)   {
     & O% \1 q! t7 N$ S0 ?: K0 n8 f
176. for   (int   pos   =   0;;)   {
     - p+ L5 C  M7 [( c* T3 g
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     0 _8 g. [0 z' Y8 @, j$ `# u
178. if   (line.IsEmpty())   return   false;
     0 R  {: Y& U* d
179. CString   name   =   line.Mid(0,   9);
     / R4 ^$ f/ M1 _/ ?/ a8 p: n
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     % z" t' ^6 `2 J+ Z& x! m
182. line.Delete(0,   9);
     
183. m_description   =   line;
     2 S8 l, i! ~! y5 D8 v% A
184. m_description.Trim();
     % a/ p( X1 v( m
185. return   GetDescription();
     
186. }
     : m+ L; ^. J# D+ p
187. }
     ' k2 s* q8 b' r7 f% u
188. }
     & c& t- p; w; L8 G
189. }
     / U; w. y- g/ a/ w4 L
190. }
     , {! g" Q% N3 q( t. q' w
191. closesocket(s);
     , t  t5 l+ M! b0 \; ~
192. 
     
193. return   false;
     2 D  `& j2 }! K) V) S8 G/ l
194. }
     2 [2 \# d- t' c% ^4 m2 I

复制代码

天云上人

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

% g! @8 J9 ~, X. c2 `; T
: T# _! e" l. H: ]  _4 z  u///////////////////////////////////////////
9 s/ r0 Z+ j; U//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
8 z! P5 Z5 }1 @( G0 a# T

- ]/ ~$ @5 `1 N1 b1 p+ \#pragma once
#include <exception>
; d# Z. P- i5 @, w9 L. [

! B# ~1 x" Y/ Y8 |  enum TRISTATE{
        TRIS_FALSE,
6 ^9 P$ Y$ O2 d        TRIS_UNKNOWN,
& R5 X9 \& H. D# Q+ G2 w        TRIS_TRUE
; D- r7 x7 [- T% z5 k# I# R};
; ~+ Z! Z: O# r! Y( e
/ `& G) {9 Q# _% E3 b6 D
enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
) z: _8 r+ E( _9 o3 C( j        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};

* b3 i# w3 z, @. N$ v


. C' D' \' D- k" P; ?9 S% U2 ^class CUPnPImpl
{
public:
8 D1 e( ~0 i  j" F9 F$ o/ H        CUPnPImpl();
5 H# P6 b7 x* r! `9 c: L7 ]& d        virtual ~CUPnPImpl();
2 E& T% F9 u% H/ f4 d& T        struct UPnPError : std::exception {};
        enum {
/ ~" ~& ?% y( ~, J& x9 f                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
        };

5 ~! i# s' r8 ?8 C9 J: Y  H% S2 V
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
' o0 e: c3 A3 r2 T, k        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
! ~, t( x+ M0 j" [        virtual bool        IsReady() = 0;
- x/ F2 p2 n8 F        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
" j6 p5 n$ D; h( @: k. k8 d, z# R* n        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
1 w/ r" `# o5 K6 P7 V! x# O) X4 R

// Implementation
5 R* Y& J% Z7 x  V2 Gprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
3 W, k% p( L' Y- N        void                                SendResultMessage();
# ]% Q5 @4 `8 |. E, S7 {/ p: h  u9 I        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
4 V& I6 H/ A9 _, ?# w) _8 Q        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;


! ^/ y0 z! }& W3 M8 a1 kprivate:
& W1 h( T1 E8 p% U        HWND        m_hResultMessageWindow;
; O- f4 z$ D5 c  s9 d        UINT        m_nResultMessageID;


};
: |0 c8 C! u' y( z: ]; e& o
1 C, \9 I  ^6 ]. M6 V( S
  {, t/ C) T2 H3 S1 h! J// Dummy Implementation to be used when no other implementation is available
: Z: A: Y. b9 u  mclass CUPnPImplNone: public CUPnPImpl
, l' A+ j* ~; a+ u. j; x" o# Y{
public:
6 ]; ]& x) [  m* i        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
7 @3 Y  V1 v4 ]1 ~/ U9 H( ?, L  Q        virtual bool        CheckAndRefresh()                                                                                { return false; }
/ j+ ?$ ^7 ~$ s; f        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
2 A3 s* v4 B* ~- ~  H9 ]* U8 e        virtual bool        IsReady()                                                                                                { return false; }
8 i! b3 |5 K6 q8 x        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};


$ H" S6 G5 W' L5 a% _7 }$ K/////////////////////////////////////
6 o& E4 o/ Q/ Z% J0 i6 _//下面是使用windows操作系统自带的UPNP功能的子类

4 p6 U. W  U& c8 I+ o# U
7 V. G8 A) y% c, h1 @#pragma once
% l8 ^3 _5 H0 A6 p; H2 ^" h#pragma warning( disable: 4355 )

8 F6 M- }6 ~1 {8 ?, Q9 F7 Q
0 W: n6 V$ \6 N9 Z#include "UPnPImpl.h"
$ Z8 x( d$ f  [7 i# y! Q#include <upnp.h>
#include <iphlpapi.h>
' ~' c6 Q5 j% h6 F9 i#include <comdef.h>
! _* z4 y4 g8 s0 T#include <winsvc.h>
* h/ G6 c7 D2 u6 }& |- d

+ f& {4 }8 |9 E* ?+ I# h5 u% C#include <vector>
#include <exception>
  x( N  e6 v3 W8 w#include <functional>
: y) n& [# `$ b- @1 \" u8 j
9 ^: K0 s* G$ g/ @  ]
6 }& g: _& A7 n

& S; ?3 d% C/ P* Ptypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
3 ~: ~- |6 \* h1 v( t- h, ctypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
* g; S* O2 a  r" utypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
( G  N( j7 n  {. Itypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

, _1 [' u& L4 M
, k; P! |3 R& ]# f* C4 i6 A3 Wtypedef DWORD (WINAPI* TGetBestInterface) (
2 M( N$ t' j% X) @& h$ B" M  IPAddr dwDestAddr,
1 I! S. R6 I% j# ~  PDWORD pdwBestIfIndex
& Y$ t5 L& u5 b) i);


typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
( z8 R) T! I+ Z6 Z" x5 t3 u* _  PULONG pdwSize,
& N* U+ X! r; `) i5 w  BOOL bOrder
) `, ]: g0 s1 U: D);
6 ?' q7 Y' D1 G( V, [

typedef DWORD (WINAPI* TGetIfEntry) (
# _# v7 w- D$ ]; m0 h  PMIB_IFROW pIfRow
);


CString translateUPnPResult(HRESULT hr);
% Q! @; }& M1 B! w  e  q1 `5 dHRESULT UPnPMessage(HRESULT hr);

, M; y1 D7 J& r  Z; k+ F
0 `3 |' d0 o2 Y2 J* ^" Z5 w) Xclass CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
0 w9 H# v! |( g7 d        friend class CServiceCallback;
// Construction
8 g, O6 d7 b& L  O8 npublic:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();


; ^8 q' c0 N4 i! O: s; ~        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
7 j/ H# ~3 _( v) d        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
6 w# K; n7 _* p8 h        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
4 R* y0 i+ U% L: n

8 J# ?: N9 L# D$ \! o' i* j        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
% c2 z8 b9 ], k6 A* @) |8 n        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
2 L& v* Z9 p9 }: s+ h3 s        virtual bool        CheckAndRefresh()                                                                                { return false; };
3 p$ F  E% h2 X% g) b

protected:
, z) P6 G+ x; e3 v3 K( V        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
1 i2 Z8 [# t0 I* A0 y        bool        OnSearchComplete();
        void        Init();
5 S8 _* z; \6 w. e1 f( y  o& @7 |

        inline bool IsAsyncFindRunning()
        {
3 n0 e8 n4 Y/ m# Z4 A                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
: x/ S- y  Q. T0 R% O* Z9 v                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
. ~# B' g/ k  s6 L                        m_bAsyncFindRunning = false;
                }
                MSG msg;
% ^3 Z5 G% O0 q/ _. j                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
' q  G5 B& s4 a. M2 Z6 y                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
3 b, l1 @  Z, p; y# {: E/ M        }
& A. Z. z# ]2 \$ u) P

8 D0 g5 g  _$ z8 Q7 a        TRISTATE                        m_bUPnPDeviceConnected;

! H- r+ w! B5 P$ x/ n
4 O; P* R2 T9 E4 ^// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        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);
1 h* y, M2 q3 k+ T        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
. }+ Y, u* w0 z( C! v. a0 G

        TGetBestInterface                m_pfGetBestInterface;
- P: B: s5 ^9 G+ z2 G8 ]        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;

1 L) f" x% ~6 g% i. m
  Y- E- O, |6 m/ U4 R+ ~        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
8 K# K5 _) w* `2 N% g                result_type operator()(argument_type device) const
+ t1 Q# n# j0 h3 @                {
. K9 U& @' P, B# |( T                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

4 v: T; O$ V" b- \% n
                        if ( FAILED( hr ) )
7 e* m. m+ Q. x: _8 q5 E                                return UPnPMessage( hr ), false;
! I/ z+ ^- O) t6 X4 d

                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
* V9 I- {5 D% S6 c. P                CComBSTR m_udn;
/ d: p3 \- j( s; \5 M        };
/ Q+ ^$ C4 e+ P- e- V  W       
9 N; W; ]; P: K3 k        void        ProcessAsyncFind(CComBSTR bsSearchType);
5 |! j; `( v- E  E        HRESULT        GetDeviceServices(DevicePointer pDevice);
2 j* M2 U! L$ H1 l        void        StartPortMapping();
0 r( a& l7 T8 p- n% Z" n" e8 U        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
+ T* S1 d. l* _& s9 V        void        CreatePortMappings(ServicePointer pService);
& n5 H# M% p5 P! h+ g+ k6 U1 G        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
; `# X  m  u/ }        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
4 b8 H7 H9 `$ |* f1 g                LPCTSTR pszInArgString, CString& strResult);
6 ]* T& S5 ]. p: l" R+ E5 D) T1 H        void        StopUPnPService();
% R' n1 _9 Z' E; T1 @0 `$ v

7 ^% A5 M& q! }7 v! ]" w0 P8 M        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
8 S! K# f+ Q1 j& \        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
( e( Q8 B" I2 f! a  B        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
: X2 P  |: d; D8 V2 }        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);

- v% X; a% W% ~( D$ v/ ]
9 P/ ^3 c% x2 N: q$ }// Private members
private:
: x6 y0 b% F3 `6 o/ Z+ n        DWORD        m_tLastEvent;        // When the last event was received?
7 L' W; V3 s9 w- x3 v        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
& R1 N( }* s7 O        FinderPointer                        m_pDeviceFinder;
4 \& R, S! d) t# G+ c2 w1 y1 s) y        DeviceFinderCallback        m_pDeviceFinderCallback;
3 W6 F8 ]- ?0 W5 d8 k$ f        ServiceCallback                        m_pServiceCallback;
: e" x( `' ], O! w$ I$ N

        LONG        m_nAsyncFindHandle;
+ X& [! l4 `' H) b6 p6 v* Z        bool        m_bCOM;
& K% j& r* v4 Z: f6 E( M        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
2 S; _% V1 Q0 N+ Z4 W- ?5 r7 `        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
$ s( B2 M3 ?' t        bool        m_bInited;
& `' q- p9 J# s* V/ t  x        bool        m_bAsyncFindRunning;
5 B; v4 d0 n3 k7 e2 d: h        HMODULE m_hADVAPI32_DLL;
& N' R) G$ |: k' S  W        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
( Y6 }# i3 `0 k) J7 v+ I        bool        m_bDisableWANPPPSetup;


};
5 J6 y+ @9 C! `* G+ J) z3 m

// DeviceFinder Callback
7 M$ |4 R/ h9 |5 fclass CDeviceFinderCallback
; ^- n& z, L1 |        : public IUPnPDeviceFinderCallback
+ J. h3 Z* [0 M# {  U! `8 z% ]1 o{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
* J: h5 f: H% |  D                : m_instance( instance )
# s: z+ N3 ?1 J        { m_lRefCount = 0; }

- M$ E) J3 F( ]# i! e% w3 W
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
" h/ h, P0 {- c% a; J

. ]4 v6 ]7 y1 c( T. J1 `/ e// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
, X1 d: g% O) [6 p: W

* W/ T" W; T5 _  e9 ^; S' c; vprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
* S6 S4 w. e0 j/ k};


// Service Callback
: g2 s8 E* X5 m  T6 z. xclass CServiceCallback
2 J1 I9 e% l3 i1 z7 U5 q        : public IUPnPServiceCallback
{
4 k0 `4 _$ N$ |' T) |" i; tpublic:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
6 h- J7 b0 v, z. p        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
# `. p' W$ t  L; k& _   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
* K# K& I5 A3 i5 |
7 _- o; t- l/ K/ v
// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
& j6 C) i0 c9 v/ z# }        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

$ n( |  C6 B5 c  c
8 R/ U+ [' G0 k% Q8 S3 oprivate:
% t3 `+ O, W+ A: Y7 g$ r; t( E        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};


/////////////////////////////////////////////////


( _- a' D" u; j* l2 U7 p2 B6 d' u使用时只需要使用抽象类的接口。
: z4 W2 T4 h- l" GCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
5 E) M' T1 {. m5 R; u$ b; wCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.