UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   
5. 
   " d* E( h1 ~/ Q! i! k8 v  d
6. typedef   unsigned   long   ulong;
   5 b- G; h$ V/ H& E5 \& h
7. 
   6 v3 ~3 o. J; s6 z
8. class   MyUPnP
   6 x8 T4 I& R& Q! ]. B
9. {
   : U  Z7 m8 r7 ^* C
10. public:
    ' i! U8 P" `) Y) z. i0 d8 ^
11. typedef   enum{
    . B! D0 b8 i! E5 @$ |  Z
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    9 W0 p- W; A% ~" ]
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
    3 d. Z: t% {* x
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    - d0 ^* C. l$ J) |/ f
18. 
    ; j5 {  W2 Q' I" ]4 j+ k9 B
19. typedef   enum{
    6 R/ \8 o; t2 {8 `$ q* \
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    2 R% h6 p, q! a1 G0 R, `6 k
22. }   UPNPNAT_PROTOCOL;
    3 {" [8 L  M* n3 D6 C" R
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    9 }5 m$ r- _! l3 V3 ~' y2 H
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    0 M$ X( u5 N/ R! N5 D$ G$ r; |
29. }   UPNPNAT_MAPPING;
    
30. 
    % p# v% S; T5 s6 t/ q
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    3 j" _) R5 D% I7 h3 t  z
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    : B# h! I( k+ E; ]  s7 {4 i/ w
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    
44. void InitLocalIP();
    ) [0 o! t. c  O: @
45. void SetLastError(CString   error);
    " g5 ^) J2 W/ V" W. E" A
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    6 R% a. q5 J% j' f  }) K, P! J* p- Z
48.       const   CString&   descri,   const   CString&   type);
    $ d* k+ \/ c/ ^+ ?5 ?' l$ Q; O
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    ! j6 D% T& B% h( Y5 r! ^8 q
53. bool Search(int   version=1);
    3 [4 h; Y$ c1 B' X+ X
54. bool GetDescription();
    ! }2 M, Y& z! S6 ^* W
55. CString GetProperty(const   CString&   name,   CString&   response);
    ! b4 P# t1 k: x$ }
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    / y5 t# O$ S- B0 D  b9 f
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    $ Q; O' y- i7 `9 a- T, ~9 g# ]* v
61. CString m_name;
    ) d, c) t5 l" `- K
62. CString m_description;
    
63. CString m_baseurl;
    ( r  n0 {% V' G8 v* m& A  f/ B
64. CString m_controlurl;
    
65. CString m_friendlyname;
    8 c9 h! q2 r- Z# K  C/ T
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    % s/ U" t) q' z( U0 C
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    & N6 J$ J5 p4 W) U7 n
73. CString m_slastError;
    9 Q6 X5 `+ a8 r+ }5 x. {( ?  ?, O
74. WORD m_uLocalIP;
    - \8 M, `; P: {6 T' q9 u$ L5 r* r
75. 
    
76. bool isSearched;
    7 n! `- G4 `7 B6 V
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   * z" e; h: T& d8 p" m7 t  G
2. #include   "stdafx.h "
   
3. 
   5 A( m, A+ T- l4 K  E7 z% |
4. #include   "upnp.h "
   , j) D; E: [& f3 K5 D! _: d
5. 
   ) k/ P+ T3 p- ?4 T8 g9 W4 P- {8 M
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   ( f. r/ f0 w7 O2 M" i' {. A
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    7 X# d+ @  ~3 A: T  q) H4 ]
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    ! @9 z4 W0 \7 F) c0 \* ~
16. const   CString   getString(int   i)
    
17. {
    # R$ S# ^3 ~+ A  \9 S+ _6 N
18. CString   s;
    
19. 
    0 y1 L" ]% l8 ~$ k3 {
20. s.Format(_T( "%d "),   i);
    
21. 
    ; q; s& b' c1 G# T: I, S# u8 o! s* W
22. return   s;
    4 U# |' U9 X  D- k% t9 x7 `
23. }
    
24. 
    4 _; M. o! V( F- Z0 i
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    8 Y: H5 Z6 \  G' W3 n; ?
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    ) Y. J' c( d' B  [
29. 
    ( Z' }' R6 z+ i5 W8 ?- m
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    ' l/ q5 e! O1 l  k4 T: W
33. }
    
34. 
    ( i7 Y) u) \$ Q- y, p. z$ i
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    " [& d; X! H9 B0 ?0 k& r: k. f
37. char   buffer[10240];
    
38. 
    8 B1 x& {) ]3 |4 M# w" ], x
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    ! X+ _; F8 ^  F: J* T5 M
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
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);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    4 H4 k2 Q: T8 O; w+ g; d
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    $ g( H4 N# C5 d8 H+ k! t$ l6 Y
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    3 }  T; l2 H% O7 Z( v2 ?
53. Sleep(20);
    5 b3 R8 K# U1 D" @0 y; `4 X
54. int   n   =   send(s,   buffer,   length,   0);
    % k: R& V7 B& ?! h/ i* z( ^
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    6 n7 |, ^5 B& G$ `6 k0 b
59. if   (!rlen)   return   false;
    ( m$ }1 b2 ^9 p8 e1 f
60. 
    4 e5 T+ O$ u3 P5 k) }% _
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    7 \& s* ], n; [3 z: h% V) A. W
63. return   true;
    " g, z  K3 O, H) i. w) i' P2 t6 f
64. }
    6 a# A% O/ \: F  ]9 s0 `
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    0 s4 t3 T! @2 U  ]5 I7 O8 D
67. {
    
68. char   buffer[10240];
    
69. 
    ; Q  {/ N! e+ S# s
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    # S# p! j+ o, m; a( s5 [: q
72. strcpy(buffer,   (const   char*)sa);
    7 j! |2 C8 \$ `! p! |3 _- v/ J
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    - k" I* q* o0 i% L8 Q
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    9 ]" I7 K- H6 t2 h
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    + [: T5 K3 w; L: t8 w( f7 E) B
85. int   pos   =   0;
    2 D- W/ o% [; h( Z- M
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    ' `6 E3 K4 E, C% m; L8 L
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    8 ^, d  G2 B( l7 ^$ T+ ~
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    9 u/ l$ M+ v5 n2 o' z; X
94. status   =   status.Tokenize(_T( "   "),   pos);
    3 \1 X0 S8 I/ O( a
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    ' {0 ]. S2 I+ e' Z. ^: w* Q4 ^3 j- i
97. }
    
98. 
    $ j) u8 {  N. Q' |3 e; F5 ^
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    " ~/ [/ O/ b/ B4 S5 ?# K  t; t
100. {
     % C- B% ]/ N  R- A: _
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     $ [5 R9 I3 l4 B' `% f( t
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     ; @1 B2 y0 l  `2 a5 `" G8 ~
108. int   posEnd   =   all.Find(endTag,   posStart);
     . w; f: y0 d" c) ]
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     1 Z# W, U& {1 B6 e  q
116. {
     
117. m_uLocalIP   =   0;
     % x+ F2 E$ B! A2 I9 G, T
118. isSearched   =   false;
     
119. }
     " ~* c1 |, p3 s3 Z* O$ H$ N) i
120. 
     ' ^# s$ m, j4 n8 P+ x
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     1 A3 m5 X! D$ w- I
129. 
     ' E9 u" b; @3 Y, w' d- B
130. m_Mappings.RemoveAll();
     
131. }
     6 N! S2 M; p/ Q- x* d
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     0 S( h4 I8 c% r- H
135. {
     
136. if(version <=0)version   =   1;
     + a0 ^( N/ T/ ]* J
137. m_version   =   version;
     9 Q1 G: F  J" t. B; @4 l2 P
138. 
     
139. #define   NUMBEROFDEVICES 2
     ! q' L0 D+ y! U: k7 x$ F) G
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     9 c& T- s  L( J1 q
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     7 }' w& u' H" u8 S( x) P
144. };
     
145. 
     9 K% M! C8 w2 y  ~- S6 n; ~
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     ' ]1 K: P) n+ }, u. e. b
147. u_long   lv   =   1;
     $ Q3 o; t% p7 P+ p/ p2 K
148. ioctlsocket(s,   FIONBIO,   &lv);
     6 e/ |( h3 N* @1 `3 k& o
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     ' C( S) x/ L2 ]. y. q3 x" ]' l# v% _
152. if   (!(i%100))   {
     5 E* n* O+ ?" Z0 ^  l" m! D) M6 f
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     6 f2 h. @' k2 e& H& e
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     , f1 F( o( q6 O  F! L1 o2 a: Q  t
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 "),
     % t. U6 F% p, [8 a; B
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     4 U( [3 N1 I6 J& U+ G
160. }
     ( E9 _1 C' q7 b  Z" W3 x4 i; X
161. 
     5 x1 |/ T3 h0 V8 h& i
162. Sleep(10);
     4 O7 J- F( l+ o& r5 T. ~& U, v$ e' |
163. 
     6 b( L7 r/ S  V
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     8 }3 r  I  S3 A* ?- k6 |  K. i) @
167. closesocket(s);
     
168. 
     ( ~* f4 e) u0 E
169. CString   response   =   CString(CStringA(buffer,   rlen));
     4 x- U6 O% u: s- N6 j
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     2 [) P# T& F, |4 m: J) C- G" _
172. 
     : G6 h: F/ `9 @0 Q
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);
     
175. if   (result.Find(m_name)   > =   0)   {
     ' `3 O) K: P3 y& @1 _  C
176. for   (int   pos   =   0;;)   {
     : t9 d/ a/ y+ G
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     : I" p0 C3 s0 h7 H; _# n
178. if   (line.IsEmpty())   return   false;
     1 ?. B+ T2 B% F6 A+ B+ d* l, V
179. CString   name   =   line.Mid(0,   9);
     * w- h( ?( o" G! d& ?
180. name.MakeUpper();
     0 H  [' t) y7 L
181. if   (name   ==   _T( "LOCATION: "))   {
     + h8 D1 Q( n  u! X0 X
182. line.Delete(0,   9);
     
183. m_description   =   line;
     ! K5 L$ G7 d9 p# J& n
184. m_description.Trim();
     
185. return   GetDescription();
     5 {* C% w/ O, P* D% J3 q
186. }
     9 W) l9 k; l8 x9 p
187. }
     1 H5 T# e2 s8 e+ I5 r( v
188. }
     
189. }
     0 [9 }2 d7 h# J' N* l
190. }
     
191. closesocket(s);
     
192. 
     
193. return   false;
     
194. }
     

复制代码

天云上人

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

6 ^2 B5 K; X. Q1 s4 c
///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


$ \4 u0 i/ _0 T, D0 N6 w#pragma once
) N, D/ e8 {  F) O: b#include <exception>
& C( b' D' y, I$ h
; i+ y3 E0 O5 ~0 H- a% F" p; u& |' u
  enum TRISTATE{
7 d$ V/ i1 b$ Y6 s        TRIS_FALSE,
0 N+ Y" Q8 O6 B        TRIS_UNKNOWN,
        TRIS_TRUE
};
6 n' _2 w% A$ }! P
& L: B: [* h: [9 x- K
. ~* d" j$ ~, U( U! U4 m4 i) s5 \/ _enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
( ^9 ^, p3 X- h, \% j8 T# v3 I' [% E        UPNP_IMPL_MINIUPNPLIB,
% w. e9 }/ Z- Y        UPNP_IMPL_NONE /*last*/
* Y; G3 u3 Y5 L: b};
, o+ J8 l; A0 s1 I$ d+ [. M, Y
" I  J# [" w* H  O2 k' c/ E
% O* y" X- [6 O6 f; C

class CUPnPImpl
$ R, a' _& P0 U8 F, K4 ~5 M( n/ s{
public:
        CUPnPImpl();
5 U( ^# C' U/ U        virtual ~CUPnPImpl();
9 _: f- a7 H6 q. f4 n* {9 m        struct UPnPError : std::exception {};
        enum {
& b3 i& o6 a. O5 Z% G                UPNP_OK,
4 m) i! |# w/ p/ Y9 n                UPNP_FAILED,
                UPNP_TIMEOUT
        };
. i* Q4 T' h- i/ c( g; D

% z8 @6 b1 c; {7 n' c        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
9 }( H/ O0 x% H3 B+ G8 K        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
7 _( u7 s, X  F! u1 H        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
7 K* |( `9 a1 p/ K

        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
: t. Z9 P0 U( d0 Y1 C5 T7 A2 h

// Implementation
; d5 Z$ k+ `( b2 fprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
! l, _) \( m- O( o        uint16                                m_nTCPWebPort;
& {; L6 _3 j' K& ]; a& B- H* y" H4 u        bool                                m_bCheckAndRefresh;
- \2 j+ a; I( E& u
/ a. @: @  q# [- j5 b
private:
5 Y/ J! V8 T% @4 p        HWND        m_hResultMessageWindow;
8 U) s# g7 i3 Q/ [        UINT        m_nResultMessageID;
& j/ [/ l  m$ @

& m* S1 C# g3 t2 H/ }};


$ E" a* Y  |3 v" {5 v/ I+ p// Dummy Implementation to be used when no other implementation is available
" U4 y! v# J. H: Q3 K% O: ]class CUPnPImplNone: public CUPnPImpl
1 V( K2 K4 A3 j" j* ^* }4 ^{
public:
! E. s& ]% ]# i& S) k        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
5 Y  E8 n& I- T* e) w6 H& j        virtual void        StopAsyncFind()                                                                                        { }
; |$ K/ q5 z( R6 \) T$ I# ~        virtual void        DeletePorts()                                                                                        { }
# ^# \" m5 J! Q  J        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
2 ~. V" n. F3 s  L* @$ D};
9 v- S0 R$ X% X% }1 B: H
. u" g" e" z' f9 Y
7 s9 G8 G; E% r* @% k" i" R/////////////////////////////////////
+ F! w: Z( `* j9 c& O//下面是使用windows操作系统自带的UPNP功能的子类
0 B2 H0 K! L& L% k1 n
0 x: ~4 Q$ g, d* k1 N- v7 N6 K
( K# O" ^! Y* E/ A% J#pragma once
' y1 g4 }/ r. a0 h- m$ G1 y#pragma warning( disable: 4355 )


#include "UPnPImpl.h"
+ ?0 I  |7 {1 W% Y5 C( n4 w& S#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
% u) N  G- L! U! Z: T/ @#include <winsvc.h>
/ |9 E$ j! o4 l
& ?& E; L% [( \' `/ P! G
* P% p( U; `( ^3 P0 K#include <vector>
#include <exception>
9 s/ F' ?. f- L3 k5 S#include <functional>
! O; `! Z. ^! r. l. I8 B8 d

, Q' E- r* T% C. O  a+ ]. n8 H
1 W9 P5 q0 F. k5 c; Y
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
) T8 g5 X2 _, m  f% r7 itypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
- D  `# k7 ?3 P0 o- a. c. p$ jtypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
8 y7 s+ f0 L+ A: T7 s6 Atypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
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;
& l9 _* O; F# j% k9 o
% h; n! u( F; ~! \& |. ^* W
# ]  N7 U4 j1 [. ^0 x9 L" ptypedef DWORD (WINAPI* TGetBestInterface) (
9 F/ H/ D7 H# e  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
% X$ Y" p( \2 y$ [/ o( e);

! m1 N$ V0 S5 i, E" }; W$ D; h
& c1 X8 [+ Y# V( t, D1 j. s/ ~8 J- qtypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
0 ~, z0 j5 t/ F: g/ j! `  PULONG pdwSize,
# _4 X& y0 q* X9 c2 p/ B# O, r  BOOL bOrder
9 e* @2 l8 z  A1 q);

9 E2 Q3 `1 C; R, u* M2 N) O
typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);
1 ]1 Z! G. D8 n4 H

CString translateUPnPResult(HRESULT hr);
) b# o; a( k* P: a% _0 t$ J6 t3 a+ kHRESULT UPnPMessage(HRESULT hr);


) P- z$ B. l3 J+ @class CUPnPImplWinServ: public CUPnPImpl
: y0 m4 M+ z5 ^3 h- i6 u! ~{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
public:
4 v6 g: w6 b4 D# e  D6 H        virtual ~CUPnPImplWinServ();
: O5 o6 p$ s# P/ v0 U5 u        CUPnPImplWinServ();

7 w) T) l- b1 l' h1 I1 U& n% u% L& F
8 z, [/ a" |9 n7 K; }        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
/ ^$ @* B( j& m4 P- ?        virtual bool        IsReady();
; K9 Q0 w; j! q) Y9 y        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


5 z) u$ A0 _5 F! d4 M! F        // 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
$ T" S, {  g4 f% H2 v        virtual bool        CheckAndRefresh()                                                                                { return false; };
5 e5 ?: m4 k% F- `  k! y

& ~. `$ A: C6 F+ S, ?protected:
5 [" B, B6 T4 m# d+ I3 l! B; d        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
& ^8 }: a) |# G8 f        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
$ z0 H) W$ P  K2 J9 T* @        void        Init();

- L9 [. M; J& c% G  Q0 `
5 t2 F& [- ~& V        inline bool IsAsyncFindRunning()
; m* d$ l% {  s/ k$ N        {
; A) O( v. E0 f# i- Z                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
! N' q; X5 v- Q7 D                        m_bAsyncFindRunning = false;
( e' r& I3 \' f                }
                MSG msg;
4 y$ y* }8 K- b$ S  t7 N- y9 b                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
3 I/ K0 b* h6 {7 f4 D8 Z$ Z                {
                        TranslateMessage( &msg );
% o- f7 F6 V! b; @, u% t3 B" E                        DispatchMessage( &msg );
- `9 W! N: F2 L: ~- ^: k! x                }
                return m_bAsyncFindRunning;
' c$ B$ d2 a) X  G3 n6 C  L        }
( Q3 y% I& _9 R7 k

        TRISTATE                        m_bUPnPDeviceConnected;

) I) v) p2 ~& D1 E
" z' ]- D/ Z8 L1 h// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
6 ^% w& q. `/ |) b6 Y  l% i        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
/ E/ U9 h; m1 ^# f8 U        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
2 ^# n! l: u! L( i% `4 U        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
  l+ ~2 u" ^: Z5 {% g2 S        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

& \& h+ e) U, Z
4 k3 q; _" D+ l2 T        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
6 T* C/ F4 K1 k$ N. V% E- _        TGetIfEntry                                m_pfGetIfEntry;
( d4 n* C, w0 [( ~( L- C
, P- s; e+ V( Q- v; ^6 x
6 p& z2 ~! B  g" g2 ^' t        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
, m9 R2 P: D: t+ e# i        {
$ [% i" y! n4 T3 F+ f                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
- c1 z0 n) ~: c5 o                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


# u$ d3 m3 m+ d) Q                        if ( FAILED( hr ) )
3 S9 Y3 z: P6 @6 Z& i6 e' ?. }- A                                return UPnPMessage( hr ), false;
5 D4 _7 I: Q) }2 W+ Q3 X

. v0 y: s3 j* x2 E# D$ i4 _                        return wcscmp( deviceName.m_str, m_udn ) == 0;
( p/ ^5 m; v1 s" Q8 t& V                }
, ~& L, L- W2 A% i                CComBSTR m_udn;
        };
       
4 U5 G2 V9 G1 C! p# L: P        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
- q* }9 K" J7 i- ?2 b; I3 G        void        DeleteExistingPortMappings(ServicePointer pService);
( i' \, ^/ }7 T/ A9 f* e        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
: v1 h/ j, o- ^7 n7 ?8 E$ x+ L9 i        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
. C& L. l' d0 _+ F+ J  m3 V                LPCTSTR pszInArgString, CString& strResult);
- [2 d! r& n5 f1 ?        void        StopUPnPService();


6 ]7 ?- l* b- D        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
8 R7 @% B% f) R# l3 }+ o' j, N        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
( H8 P3 q/ _+ Z' D+ p# ~  O- [        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
+ b  B, _4 o: A        CString        GetLocalRoutableIP(ServicePointer pService);

  c9 O* I3 s' v* A% u/ ~
3 s8 x" i5 n7 s9 s3 b3 _// Private members
+ F+ q# U+ j" d. B: j5 \) {+ ?private:
) m6 L, ^+ E3 C( G        DWORD        m_tLastEvent;        // When the last event was received?
- }2 b  v& z4 L3 l' J& R        std::vector< DevicePointer >  m_pDevices;
6 G2 f9 q# u6 {& d$ a7 d        std::vector< ServicePointer > m_pServices;
8 \2 w- r2 g2 P        FinderPointer                        m_pDeviceFinder;
7 w6 W4 r$ [6 c, I: K7 G! V        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
7 \+ R" A$ s' H' d
$ \2 I9 b8 S3 {4 a+ c3 z5 ^
& p9 D( y: N0 y9 g$ o        LONG        m_nAsyncFindHandle;
6 X. [: {9 b; t2 {        bool        m_bCOM;
$ }' ~0 g( {. m        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
. i' S3 c3 u# u0 D: j3 V        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
8 K1 Y! n) p1 d' H7 c$ d- y8 @6 Q        bool        m_bAsyncFindRunning;
' N1 l1 b, u6 p- O/ _        HMODULE m_hADVAPI32_DLL;
. G. P1 Q  m& N/ u1 ~  [        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
0 U! {- b7 X& Z1 |5 x        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
6 h2 G' ^: ~; r! C" E        bool        m_bDisableWANPPPSetup;


3 N3 ?' ?) f# t1 R* P  D* v" O7 V};

* t' F' e. ^1 O/ G! w( K  G) k
6 z" w( |/ E/ T2 m+ O// DeviceFinder Callback
class CDeviceFinderCallback
# u4 A+ z2 ~% M5 F        : public IUPnPDeviceFinderCallback
3 {$ Q4 O% y' ~: C, _{
9 B6 ^: ]# S4 Qpublic:
3 G$ S& |1 C: X% T: J' E        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }


: b8 M' O$ N/ ]% s9 F( \% z! `   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
+ |* M6 i/ K! f8 \   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
  G, W4 n  n9 t* A# _# j9 Z

// implementation
private:
6 Y  r. C9 s7 |5 o2 L3 o7 h        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
7 g% P; U  T8 J        HRESULT __stdcall SearchComplete(LONG nFindData);
6 e5 }6 Z, j; n6 [0 k* C9 X
7 B1 h/ I. _1 R& }
5 X& r! M" S" Q" h9 r+ \private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};


" u8 o& \1 i5 L2 ^' k$ x// Service Callback
( t$ v. I7 B6 Y$ L# ^! h3 ]1 T- k# cclass CServiceCallback
' U$ H- `! {) y! V        : public IUPnPServiceCallback
) `: r; t. t; z/ Q: ^{
public:
' r9 W* M& E. c- j/ F4 k        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
( u/ n2 E1 j/ F- r- f( C" F/ e   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
) U/ A: L  D* `& J2 B        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
9 w  F1 B- t5 ^' o

private:
/ N( O! i( [8 {! k/ {, e        CUPnPImplWinServ& m_instance;
# }! b) w3 `  B( _! D, Y; R) \; y        LONG m_lRefCount;
};

6 k9 d0 S# U& e: _6 D; r7 u* Q
2 _0 A. G! w. }. P/////////////////////////////////////////////////

5 U7 V# Q5 x1 F$ O
) Y. `4 z) e' W1 t6 V4 {使用时只需要使用抽象类的接口。
9 t$ e" |8 ^) L: {  tCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.