UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   ; F: c! c5 d, i
4. #pragma   once
   
5. 
   3 I! h# m0 x/ Q+ M4 |2 D0 P
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   . q& Z6 b7 _* s9 W% |
9. {
   
10. public:
    
11. typedef   enum{
    
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
    5 ^: o9 m+ w& c1 o$ Z9 B
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
    / ~" D% a* j% k5 p8 I8 _
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    ) ?/ o4 ~5 h  \9 P
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    - C7 ?: J6 N$ L6 r# A$ O5 t3 ]
23. 
    , j$ ]( m; ]/ _0 p
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    , ^. N: q8 E% o0 R
26. WORD   externalPort; //   Port   mapping   external   port
    : _: E, m9 H! j8 s+ T; L2 n
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    - e4 d  r& \' d0 b) ~) T% P4 r0 n
30. 
    " }7 p2 J; y7 ]8 `3 n
31. MyUPnP();
    
32. ~MyUPnP();
    8 n8 b( c+ Y$ W+ Q' Z6 n2 B
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
      |8 D" z7 l' |7 N
37. 
    
38. CString GetLastError();
    3 c. D- {) M. M; B* Z# T9 U
39. CString GetLocalIPStr();
    0 [& g; X0 |+ G6 O) R" b
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    5 N; k/ v& T# u. J  D& e
42. 
    1 o2 |3 R- K' r5 d
43. protected:
    ( T& \; X! S( L5 g3 A
44. void InitLocalIP();
    5 q; r5 w& `9 ^& V0 Z
45. void SetLastError(CString   error);
    3 q, H. E, ?+ z5 j; t7 c
46. 
    " H; v: d7 V9 k* \) j3 U
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    " ]# }! c, c& Y: m9 j" }1 i5 O
49. bool   deletePortmap(int   eport,   const   CString&   type);
    0 J% m! F7 }$ R  ^: N
50. 
    4 _5 H& r% A/ h/ N
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    4 {3 s- w3 c- q' E1 U3 R
53. bool Search(int   version=1);
    7 l& U7 U; v! N( z0 N
54. bool GetDescription();
    " Q1 o! l+ J; d, `  ^+ F
55. CString GetProperty(const   CString&   name,   CString&   response);
    $ Q: Q; n' h+ B  k0 N
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    ; C3 G* ]* k2 s, k! |$ a
57. 
    0 V& ~5 O3 X8 p- Z$ H! h
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    7 {6 k5 r( v0 r" U8 c/ g: O' o
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
    ( a: w# a; S. ~: D5 |
65. CString m_friendlyname;
    
66. CString m_modelname;
    8 j$ o* H- M) x5 b
67. int m_version;
    % S7 L( S- C( A* N  J, s% a' o
68. 
    + T; C1 _( k7 s+ K! S1 D  W, ?
69. private:
    " p( X- K6 p% C8 r, |$ _4 x- \
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    , f: Z( c& n5 o+ ], M# w0 `* m! v& R
72. CString m_slocalIP;
    ) {3 ]( ]& }; `& L
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    " t, W, d( }1 M3 j2 i7 H* J
75. 
    9 F" S  b3 X- f: M0 D1 z3 E
76. bool isSearched;
    : c5 r, f# g/ e, R6 E
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   ! P7 T9 M, i& e; ~
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   $ [- j1 [2 X# y, V. l7 f  X# @! ?
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    $ U7 z1 Z. k: `9 e+ i9 q/ U
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    5 w- |! _# Z$ M  B; @- A
17. {
    ( M; l7 Z# e! I4 k
18. CString   s;
    * s- G1 b& p! A: W8 Q
19. 
    # J7 j0 P6 l% J# I
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    
23. }
    : Y' l- \7 ~" U# I5 n" a8 Q
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    8 t9 H! c) L# X8 H
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    3 P& |& M+ F( p- K0 p
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    0 K5 [3 X8 Q" A
33. }
    
34. 
    6 }3 ?% N8 Y( }7 R8 B# C$ u
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    ( L5 ?  k# j5 z/ r) w' S: O
36. {
    $ `" m, B' K! @1 u
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    # E" v6 a* ?9 R& q! C# v! _
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    1 C1 z8 N, @, L7 O: K# d' _; P1 {
42. 
    " H2 z- I0 ?% o
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    ! i8 R. x9 G( ^* h' ]2 R& h( \8 \
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    % {) G. ^' L2 A9 X' E7 z: I" z
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    5 q# i9 t* B7 f4 D6 Y* x8 \" T* v
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    + F. v1 `! p+ {  Q# a8 N$ x6 x% E6 K' Q
54. int   n   =   send(s,   buffer,   length,   0);
    " r( v$ @# j* K: t7 h
55. Sleep(100);
    % p4 r. ~  O0 A& ^
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    ' `: N( |( R. D3 ?! [
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    0 b$ t+ @- }4 p5 `7 {
59. if   (!rlen)   return   false;
    % Z# `& |# F" P5 v
60. 
      p* ]. T5 \5 q
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    ; F# K9 W9 O- \6 [5 s- T
63. return   true;
    
64. }
    8 W% W* L/ Y& Q! R' m
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    / a8 }* {1 v, E9 x# a8 p- h
69. 
    ; M0 o9 f9 Y& o8 |' `5 W; C
70. const   CStringA   sa(request);
    $ b8 p' g7 B9 V4 O
71. int   length   =   sa.GetLength();
    * {8 n0 _0 ]4 ]" _, ?  s
72. strcpy(buffer,   (const   char*)sa);
    ) l/ Y5 `4 o0 x  W. l# _
73. 
    0 {% M* C; Z; t- R6 s4 H
74. struct   sockaddr_in   sockaddr;
    8 |2 d  ]* @$ L& ^* q3 k4 A
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    9 e' F% l3 r& j* F9 t" S
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    . I- s8 ~; d+ h. U( }) s/ }
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    4 P' t' c( y  A, V( Y9 s+ M
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
      z5 r/ t7 X, {, o2 F5 r0 t8 c
81. }
      ]: s! _* F2 p, l# c: s; k7 F5 r
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    + i2 S" s& A' R
84. {
    
85. int   pos   =   0;
    
86. 
    ; D: z* L* n! d# e4 X7 y  Z
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    + C. W( c) q4 v9 ?5 U
89. result   =   response;
    ! ^3 ]' B' {8 P  W* z# ^  D% v
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    4 L, T7 s- L  c, c- j" b4 _
94. status   =   status.Tokenize(_T( "   "),   pos);
    % ?9 U1 x0 c2 w# J! x# h# S
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    . }7 J4 d/ ]. z3 V" O7 S2 y( P
97. }
    
98. 
    / m; c" l6 E5 I" d7 f
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     ; R/ @9 o* N( S, w5 N2 e
103. CString   property;
     
104. 
     & K% C, A% e% F, w9 I
105. int   posStart   =   all.Find(startTag);
     " ?) ?6 @' f! K5 r1 Z, O
106. if   (posStart <0)   return   CString();
     5 `" ]% Y* \& H3 h- q& c
107. 
     ! D+ \3 M/ v  @
108. int   posEnd   =   all.Find(endTag,   posStart);
     0 q2 f1 Y7 x1 P+ M8 B$ v. Z
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     ; A" i) o3 G2 T! l" }0 p' h2 [
113. 
     % N- l" P& I' }7 R: i6 f
114. MyUPnP::MyUPnP()
     2 `& n5 s& Q1 L4 [9 a
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     9 f0 w$ ~+ H' o8 E, P' p
118. isSearched   =   false;
     0 [% _1 v1 v# Q3 c- H' J  M: d" q' H
119. }
     % C4 u) s7 d2 l* `7 p2 F" Z
120. 
     
121. MyUPnP::~MyUPnP()
     1 c. E9 @0 ^+ ^  v, N! C
122. {
       H. r7 r8 X/ y6 O* A: b  K
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     8 s; O% `" }5 l1 b3 Q
125. while(pos){
     : g* J6 `8 K' p' `
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     1 g0 V. H5 C, S) D% ^
129. 
     
130. m_Mappings.RemoveAll();
     
131. }
     $ l. V4 \7 D3 ^0 Q3 h8 h, V2 o
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     % H! Z: |- _& {6 G
136. if(version <=0)version   =   1;
     5 c, z3 ]* S# x! O: A% O4 Y
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     0 r0 u+ b9 O0 y" _4 a6 S1 y* i0 `/ ^
141. {UPNPPORTMAP1,   _T( "service ")},
     " h+ |( c* e: W
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     $ z3 J6 g5 r* t7 t) o- q
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     & Z1 k+ e- a3 _) `1 L$ `+ J
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
       F3 R; f- O0 ]* T: o
149. 
     
150. int   rlen   =   0;
     + F4 W2 b8 e0 U# B6 [, K
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     + Y2 l6 f& Z/ y  y, y) Y
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;
     3 y: P) g2 B3 z' E, o) i8 P) L
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 "),
     # P/ k! Z+ y" F4 ~. D* F) K8 p
157. 6,   m_name);
     4 r2 f* `6 O. A/ s, z
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     5 a+ @! C" _5 ?- F5 c3 {8 o) u% @
160. }
     7 f# J0 ^; E* ]' \
161. 
     
162. Sleep(10);
     
163. 
     + j0 e* |) q2 Q: a( P" P" C) ~9 f
164. char   buffer[10240];
     ( A4 |' z0 A, b! n
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     1 f$ L9 T% s) m$ j' [# p$ G. z
166. if   (rlen   <=   0)   continue;
     ) N- I) a1 [; p5 J1 y
167. closesocket(s);
     7 C9 Y7 H5 E- F; C# Z
168. 
       V# k7 u2 @2 v9 `. V6 o+ b1 q% v( i4 |
169. CString   response   =   CString(CStringA(buffer,   rlen));
     ; a8 A; x" R+ }+ a2 `7 q) I
170. CString   result;
     5 x+ B" ~1 a3 V: R9 `0 c+ L
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     $ t- r: K$ G7 j) R6 A
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
       p1 Y; N1 e6 }6 m' F& n0 g3 m" b
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     ' I( d* q" v- ?0 ^
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     ! o: Q" O2 r; N0 p2 i5 M. B8 {
178. if   (line.IsEmpty())   return   false;
     + F  l$ u" I5 L9 [. O
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     / x7 v! T; j' D7 N( }6 g7 y
181. if   (name   ==   _T( "LOCATION: "))   {
     8 G& J9 c5 n. h; Y' ]6 A6 I
182. line.Delete(0,   9);
     5 O+ n. v* V: H6 Y& T5 M
183. m_description   =   line;
     0 }5 D  d* j/ h2 o
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     
187. }
     
188. }
     ) l$ Z+ v4 a' Q8 l% `7 p1 q% Z
189. }
     5 z5 \. T4 d+ ?- U$ P( q  |8 }1 z! B
190. }
     
191. closesocket(s);
     ! i+ k3 L; d/ B- B5 [: ^) k) K
192. 
     
193. return   false;
     
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
2 i7 {; D/ w2 H8 n( Y$ d
, l" {0 \3 l0 m
///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

- O+ w' M8 S' q+ u
9 T, q4 W& ~9 ?& d* c; |* f( V#pragma once
4 J: ]4 G7 J8 b" H& q#include <exception>


# P, t5 R# E. Z$ v4 N3 b/ c  U# o; T  enum TRISTATE{
        TRIS_FALSE,
/ k' R) z$ G  K* z  o        TRIS_UNKNOWN,
        TRIS_TRUE
1 ^0 ]0 ^# u# J# x* ?0 {6 E};
+ Z( I2 ]0 [* @
! i) Y' _  a) _% b2 ?1 w5 G
enum UPNP_IMPLEMENTATION{
6 p" l( f$ w! s7 i        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
* Q/ l/ B7 C3 s; Z        UPNP_IMPL_NONE /*last*/
/ V- G' M' V8 Z* H9 i1 |0 D  I# s! Z5 V};

& g0 c; j3 j$ Z
4 f2 v& a% Y+ p) N% K1 l
! x, H3 U. }& @2 f  C9 T
class CUPnPImpl
+ h$ T  }: \, `2 P{
& L1 `$ |/ I* P/ A0 x* `: xpublic:
, ?1 ~/ e" w7 v: j, h* s& T        CUPnPImpl();
, ~$ J3 i7 C, n6 u$ g        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
7 a# ]. q; l/ |( ?' v, D7 k% W                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
        };

" N& D  ?9 }- ~3 d1 `
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
! I2 Q, b& d4 j5 L% @        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
: C9 U7 L+ Y( P1 A: ]        virtual int                GetImplementationID() = 0;
4 r: `3 j) Z6 x, q8 I* x0 k/ K       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
4 n/ ^! c- ?6 R% T        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
) e/ _3 F1 T8 m6 {/ c

// Implementation
8 m( o3 G2 o* G# ^' c) C$ [' `protected:
& o2 K* N/ f& x( K7 S        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
. `6 q$ x+ H: \# i  ]7 E        uint16                                m_nTCPWebPort;
7 o" G8 ~2 J4 S, {% K        bool                                m_bCheckAndRefresh;
$ C7 v! [5 K2 |8 X- \

private:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;

. P  S" r7 i6 |% Q
};


// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
3 f$ t: p9 P! T; \1 G- epublic:
, {' t* L2 k& F( L        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
+ G, ?' F0 A3 M7 |" Q, l* t# h, ~        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
% c% `- ?# w4 ^- g. `: \7 V        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};


* b' A) d7 N: t8 f, i* x1 @- v/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类


#pragma once
#pragma warning( disable: 4355 )


/ t1 q8 \6 i0 `  z/ G( Y#include "UPnPImpl.h"
#include <upnp.h>
. k, M7 p9 Z/ {5 q% o5 A#include <iphlpapi.h>
#include <comdef.h>
! b6 {% Z' ~' U#include <winsvc.h>

7 V8 W& _( j$ l7 K  J( c
$ g3 Q5 w/ S9 j% u0 }( N#include <vector>
( O5 X: c" |' S, Z" m% y  M#include <exception>
' k9 N% x( r4 n#include <functional>



( d0 M7 ^8 L1 N$ `, K; I* N& L
typedef _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;
; i" F; j0 F7 y4 a6 f( Vtypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
/ o2 K5 r$ N6 Y# rtypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
" q" C$ E' D2 \% t" t' p/ k  `typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
. _! V8 e: }6 o2 b4 ^! ltypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
* Z) g' G; l' Z/ q) Z4 E$ D
% @1 G" D. O. y0 t' B# ]5 V- i0 p% p
typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
- r9 M; B$ K: v; A$ Z3 y4 [  PDWORD pdwBestIfIndex
);
- f9 V8 }* n+ e0 ?
0 q" X% i5 j. M& |
typedef DWORD (WINAPI* TGetIpAddrTable) (
" K8 @  I( m/ a4 Z' L" m; Q/ m  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
* d% p. g, W4 W7 E8 n4 U3 F6 Y  BOOL bOrder
% W, _0 ]9 l; D$ X& k) i0 Q);
  @6 A# f' W; W+ @' W' z9 w

0 E: }2 k3 \+ w5 }3 O) p% `typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
) @! k/ h+ o  k, y" |6 i. i);
# T1 K% ~* R, |
' L- t7 q; I6 c
CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
  s9 C  q7 }) Q" I2 e" u

/ Y. r! @# n( vclass CUPnPImplWinServ: public CUPnPImpl
5 _, d, t/ }. L2 U/ @, I4 q- m{
        friend class CDeviceFinderCallback;
( J7 Y! Q5 k; [. C$ v        friend class CServiceCallback;
// Construction
2 o3 I* S5 m2 l$ |public:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();

/ }$ m& ?: M8 e  ^: Z6 j0 R2 V
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
) z, F+ j/ z. y5 j2 A4 c        virtual void        StopAsyncFind();
5 y' `1 A; l5 p5 T9 E        virtual void        DeletePorts();
5 X. w4 [' S  |9 p* P        virtual bool        IsReady();
1 c/ A4 e/ X- V# i4 }2 ?        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
6 O' z8 h8 }  c9 K
( R* q. P* H. j' [( Q& |+ [
        // 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
        virtual bool        CheckAndRefresh()                                                                                { return false; };

% _7 V! ]0 A+ H4 ], ?0 Y0 ~1 u
protected:
3 d5 f# X% x3 q; `# M6 d* J        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
* T' }8 Y' L# A% e* u0 L; H1 t. v* j        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
9 h0 e9 ?, D, ~# U1 _* Y5 w        void        RemoveDevice(CComBSTR bsUDN);
" x6 ]! i* e( w- b' d        bool        OnSearchComplete();
        void        Init();
3 C3 r: g6 V! l; B" |2 r+ C& M5 t

        inline bool IsAsyncFindRunning()
        {
; B: V; {$ U! \9 j. r                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
  U4 ~# K9 D+ e9 s( R3 X& i( Y2 \3 w                {
$ |% k0 E7 d/ M! Y                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
, N) Q; g8 f, I6 F                        m_bAsyncFindRunning = false;
7 ]4 t  V+ I" g+ x- [                }
                MSG msg;
8 M1 D6 ?6 j8 ~0 t5 |% \& Q                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
  [, Y' x! s1 y                }
                return m_bAsyncFindRunning;
* I8 q: ]5 ]" [2 q        }
+ ^" i. j* W) i0 w

        TRISTATE                        m_bUPnPDeviceConnected;


! X& g8 p! `1 [; E0 c. S' ^; l) Z' r// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
  h0 v! Y( n# r8 |* y# _7 }6 y        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*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
6 Y" H4 I( t5 X" ?' ^
: q. x" g; G: a, W- e6 M6 n
. q2 |# d) F! P, a- k" L4 f        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
2 ?8 u3 r( A) _$ K, G$ {        {
6 n/ s$ G( J" p+ C                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
; [; ]. z, @& p, Y                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

. M3 j: S4 w. z! w' a
! ~+ Z# f5 g, t$ n& ^  ?& e                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;

5 X/ y1 X9 ]* W7 t
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
% G" J8 M) Z5 n- D                }
                CComBSTR m_udn;
        };
       
" X0 p& R( |' J/ r        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
5 B0 _4 r7 [% z- q        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
# `8 R; D& w% U6 Z; k/ `. s        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
6 W- Q4 n" L0 Q( ~        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
; h9 f1 Q3 A2 Z7 {! }                LPCTSTR pszInArgString, CString& strResult);
! E# M- O7 g# }  n9 d$ T        void        StopUPnPService();

6 m1 U. ?9 F+ M- n2 R7 P' I: I9 ?1 D
        // Utility functions
- {0 {+ g# j9 C7 `        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
0 N: D: r+ X* Z        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
; [) H5 H4 N% C. o        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);


0 J, ~& X9 w& }7 Z1 _// Private members
5 L( d8 K* {0 {9 J# sprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
" X( o) g2 |7 K. x/ F1 O+ B2 l        std::vector< DevicePointer >  m_pDevices;
* N  J7 D- |' ?! e2 Y3 x3 @0 b- _% _        std::vector< ServicePointer > m_pServices;
7 [9 K3 ^1 V9 k" H* \, y! Q        FinderPointer                        m_pDeviceFinder;
. s6 \0 M6 `  |, Y* }' ?1 k& h1 O0 ]        DeviceFinderCallback        m_pDeviceFinderCallback;
* j3 }% s6 Z4 P9 [8 B8 W# ]        ServiceCallback                        m_pServiceCallback;

, m. ]! A+ y1 N! g( g' }! A4 m
& K; y9 N7 L) W        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
% \! \7 v  y6 [& ~9 }( ^        bool        m_bPortIsFree;
& s- L2 _7 F0 U! @0 D! B: R! q        CString m_sLocalIP;
) ^* P0 O" D5 u3 T; q        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
# r1 d5 ^4 M, A, u# R" x        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
5 o" p3 c3 Z; |6 k! _        bool        m_bSecondTry;
, s# Q# u, E5 X2 T# S, N        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
' I* R; I! o) `: Z' \) B        bool        m_bDisableWANPPPSetup;
- f5 p! G2 D) L" A1 u7 |+ V

  v1 d9 x7 Y1 H; {};
* x/ c" B) g8 D' G5 R4 O% ]
5 o( q4 |1 o: M+ R8 _- M
- f- Q7 ~7 |, ^9 v$ X( c; f// DeviceFinder Callback
3 m3 g2 {. {. N0 {% M& Vclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
  o  Y1 z& S7 h        { m_lRefCount = 0; }

  t3 I! r5 U* O  ?6 @& ^. H
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

0 G$ O, `$ @5 q+ S) ~! \
+ x# t1 E+ m; c! l% M( ^4 ^// implementation
* H( B( D  n3 m2 d, @& ?, a: C; zprivate:
! e+ I5 ?1 I5 n        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
: V% f. |4 {2 B$ u        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);

4 m4 _4 B1 f# D1 X- b
private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
9 b. x" Y6 t9 I8 T- i};


3 [  p, Q( e$ N% s1 E$ S// Service Callback
! E$ }3 ]4 D) H1 Cclass CServiceCallback
        : public IUPnPServiceCallback
( {; E4 }& \0 K6 @: S{
public:
0 T  L& O) @% ]7 r" D/ g% h0 }7 Y        CServiceCallback(CUPnPImplWinServ& instance)
' n7 A' O1 V2 _0 _                : m_instance( instance )
7 `6 O# A* o0 d* G5 f        { m_lRefCount = 0; }
   
# z! |7 |7 f0 i& J( v# v9 b   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
# Z) y. `1 Z1 F% B6 u
% A( c; T/ ?. I3 |. q
// implementation
private:
$ H& p9 w$ w# c8 _6 z3 X' `        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
1 s3 K) J6 ~% K# l0 D        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
, g3 S; d( a7 O' }

private:
/ i' d& i7 S9 E' @7 L        CUPnPImplWinServ& m_instance;
$ @' e: m/ e& N$ _2 K# Q        LONG m_lRefCount;
};
. g, [1 m% ^5 k9 f; ]
/ r6 n$ [) I$ C0 k% D7 F! g+ ^
) }0 s3 y9 d) a1 p0 ?' h  i) w/////////////////////////////////////////////////
+ z  x2 I/ W  w- `( V: w
6 e5 k, {+ q( S2 @+ K0 E. s2 d
. u4 R$ w" @/ l2 D! T+ m使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
( U6 L5 V& x2 I8 t* r! H" qCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.