UPnP

天云上人

/*uPnP.h*/

1. 
   3 h$ B$ U$ w! n$ }' y9 b5 {0 n% C
2. #ifndef   MYUPNP_H_
   . j. Z. _$ H( L6 r( B, ]+ S4 b4 K
3. 
   # o3 n/ r7 G) {4 ]! B4 F, k
4. #pragma   once
   ) |' l: ]) V- {* M9 b
5. 
   6 @0 S0 j! G# B8 t- G# Q
6. typedef   unsigned   long   ulong;
   6 r. _7 F3 ]/ _1 M2 G5 Q
7. 
   0 \8 _. D4 N2 s/ j2 ]+ g# U2 X
8. class   MyUPnP
     t; k) [% w5 G/ j6 q& A
9. {
   
10. public:
    8 q: E* v) U5 S" J1 \0 Y
11. typedef   enum{
    8 A4 _  b$ b/ m3 r$ z2 G2 V! I
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
    ) d  y5 k3 f3 Z1 t
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    $ i2 t( ?5 Y6 @. V. r6 e9 W
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    , B/ Z% @4 g' G
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    - i) M1 F. U& M: y6 D2 ]
24. typedef   struct{
    " u6 h  w: z( l  g$ W
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    0 Z: H1 d; _( V0 n2 a& r9 X
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    & U; B% Y" m# ~+ Y6 ~0 Q' S- ]
33. 
    * Y8 f' o% E  j/ W: S" C
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
      G, G4 T8 i4 ]! s! n- z* L
36. void   clearNATPortMapping();
    % D; B( P" }) v* X: E
37. 
    
38. CString GetLastError();
    8 s3 y* X, Q) p! R& v/ P3 ^
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    8 d! Z) o" v; K+ _. F$ ?$ i& t
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    8 ~% m# e' }- G: {4 f. W: g2 {8 M
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    ' P- i; B1 `+ |0 e* G
48.       const   CString&   descri,   const   CString&   type);
      {" X5 z- n! m6 K
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    / n2 o  d' B( I9 y6 \8 ~6 ?; e
52. 
    
53. bool Search(int   version=1);
    , S3 x. L: l6 |( ]
54. bool GetDescription();
    ) O* s. x& H' [& l+ q; }
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    ) m0 X2 T0 |, I5 X% b
57. 
    / k' x5 g+ j2 g5 D# W. u) q
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    0 E$ n9 _1 z- B! p
60. CString m_devicename;
    * w+ @" J" J/ l0 p! K% o! c
61. CString m_name;
    
62. CString m_description;
    1 ?& l; Z% N- {  f) ^
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    9 I) a! l1 G& L/ Q  l& I9 M! @
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    ' u" {3 y: S, F. @$ `
74. WORD m_uLocalIP;
    
75. 
    3 U) T9 p5 F6 g
76. bool isSearched;
    
77. };
    ( o' ^0 C8 [+ }) t( y
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   3 {2 ]1 B( y1 T& C. L
4. #include   "upnp.h "
   ) l# ~6 K" o# v! H
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   ' v, p9 o$ Z! F7 B
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   6 c6 F: L7 }; }# f( P) q9 T* u$ o
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   * D8 K  W7 u5 A, N' T
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    
17. {
    , t) i: i% `9 V+ V! A3 o
18. CString   s;
    - Z, S, E% }% N, J4 l, _7 |
19. 
    2 s$ S0 }( [$ r7 o' [  o. Y
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    , N! G$ ]$ S: G" i1 [9 U" D
23. }
    ! {' T6 M$ x6 r1 o
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    % H8 k5 i, ~- T* ~6 \
28. }
    ( h( n# M7 T7 G$ I6 M
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    7 e5 G' o2 C" x6 y
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    % S( }( ^5 I( b) `& o8 y( N! I9 b! X; e
34. 
    1 K7 ?, [% P8 c3 r( f# A; r
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    ( c7 d4 R! q% \
37. char   buffer[10240];
    ( t& i1 ]# O6 x2 P% t
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    ) C* z* [5 }/ o; e
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    0 X! M3 B- t7 r& R
46. sockaddr.sin_family   =   AF_INET;
    6 I7 t8 R5 a* u' N2 u9 W+ \
47. sockaddr.sin_port   =   htons(port);
    & ?! t2 z; C6 e. V5 t! j
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    0 T. j: |1 o/ F8 N# P7 r. k% q6 ^
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    6 G/ J* M' W+ }, V
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    7 w5 u2 N3 l, E6 W9 @+ h! }
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    . ^+ Y$ X  Y5 b( [- l1 ]
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    9 b5 b4 x9 C; f4 I9 p5 _
60. 
    5 j- x% ?1 n8 d3 Z; [5 l0 C1 T2 i
61. response   =   CString(CStringA(buffer,   rlen));
    ; |( X/ L* ^: t, ~  \% w1 k
62. 
    / Q: m  m; q3 _2 _5 X
63. return   true;
    
64. }
    
65. 
    2 c  s* |/ ^* D
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    9 W' L" r" V% \
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    ) g. t% u6 A8 \; ?! P3 A% V
71. int   length   =   sa.GetLength();
    7 Y' R3 @+ K( L: K5 W' i. ^. L, S
72. strcpy(buffer,   (const   char*)sa);
      t5 }5 a* h, v: a
73. 
    + g- E& o  z7 H- u/ y1 F$ ^
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    ) u+ B0 c9 E' V: B" U
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    7 _  i; _' O) Z5 I4 Z- _7 l
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    & Q7 L- v9 ~# x2 e! R
81. }
    ( f/ k( B# r5 E* [5 U
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    * l, ^9 v- H0 G- }; Z" v
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    ' b$ R2 I# F7 g. A' ~* i4 x
89. result   =   response;
    
90. result.Delete(0,   pos);
    % ]! ~9 ^6 V: G0 i
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    % \+ k4 F, E/ J! P! d
96. return   true;
    
97. }
    * N. _/ z% j+ r- S  @* R: h
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    / F1 ?' i$ i$ v  A/ \  D2 f
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     $ g& V' D4 ^- K1 u8 \
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     9 Z* x+ Z* P! _: `2 c
106. if   (posStart <0)   return   CString();
     # m( P( g( ?" T# F
107. 
     . p$ [0 M% V: Z8 B6 U! j
108. int   posEnd   =   all.Find(endTag,   posStart);
     2 f5 O( k( F# ~" L$ `7 v; y
109. if   (posStart> =posEnd)   return   CString();
     8 [5 {. i7 L$ i( K" y
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     - c1 [6 m6 A* x( M4 O& z6 z" {/ G
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     ' O: J, u/ F, M8 u
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     
120. 
     6 x7 }: m3 T- ?$ O+ Q$ x
121. MyUPnP::~MyUPnP()
     6 m$ }4 k) n4 L8 K! u1 I' p
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     ' k) E7 A$ E1 g% m) z
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     % Q/ n5 d5 d5 C7 x2 \/ w' p
129. 
     + R! \5 c, h1 p, E6 Q
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     6 W- r" G! Z0 P9 Q$ D0 B1 K. W1 s
133. 
     4 L/ {4 @0 Z! j
134. bool   MyUPnP::InternalSearch(int   version)
     # {4 I) _7 T; N6 l
135. {
     1 m9 a6 O4 b* W. m$ ?% L
136. if(version <=0)version   =   1;
     . b9 {3 P  o: X4 e4 z7 n
137. m_version   =   version;
     6 D2 \1 S/ q% A7 t
138. 
     + R! }  |" \( J  d7 O9 G' I
139. #define   NUMBEROFDEVICES 2
     / g0 Y! z' @  v$ ^: i
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     * I1 [: K: G3 k9 o  B: ?
144. };
     
145. 
     7 }% q- A0 T- x! m9 W2 P
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     3 B! F  S( v. B# Y% h' T1 w
147. u_long   lv   =   1;
     ; ~, F% a9 F$ H7 _7 y. {6 k
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     * I8 w/ t0 {/ y4 \4 C3 N( y
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     , m# b7 ?% S+ \! `( @/ j
152. if   (!(i%100))   {
     : H* P! _6 D. ]. v
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
       r$ g/ s7 y% d0 v
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     0 ^. O! c1 i5 f8 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 "),
     ( D* K$ q& [' h0 J8 j
157. 6,   m_name);
     + Y3 r. K3 i2 u* `
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     / Z* a2 R2 K  l6 j
159. }
     $ s: |) Q" s5 v4 l# M
160. }
     + I' t' d. |# ^, W, T5 o
161. 
     2 a6 [- T6 b5 n
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     . P' A7 _1 v" f. v& M
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     & A0 f$ K, r* G
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     ! n- c* \( K3 T  R1 c5 [3 r
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     " A+ I) d" j, E+ X
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     7 A) {7 {; _% S/ n8 [# Y
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     4 U* R" F$ s, _
178. if   (line.IsEmpty())   return   false;
     : y. h5 n1 z, c/ |
179. CString   name   =   line.Mid(0,   9);
     # Q3 H& A, U) |0 `5 ~9 F
180. name.MakeUpper();
     2 N4 T8 G/ x  P0 l
181. if   (name   ==   _T( "LOCATION: "))   {
     6 e: I3 O( h6 T7 k9 ~5 h: }
182. line.Delete(0,   9);
     
183. m_description   =   line;
     7 \& H! k5 @: _/ s  A! n' ]% H
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     
187. }
     
188. }
     - {) I. J! {$ E3 ?* J& e
189. }
     0 p9 L9 }2 Q3 v! j% ]
190. }
     9 z+ V, C5 A0 g8 m
191. closesocket(s);
     
192. 
     ; [0 _  \3 S' k( h# K7 @
193. return   false;
     ; `- y9 u3 j8 g" y3 ^2 }4 {" p
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
" \+ W" I* ]+ n4 S
+ B  K6 w( p  j; F- r+ F
///////////////////////////////////////////
+ o% t8 L1 `: f0 Y//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


1 E; T/ Y6 \6 r# N0 B. L#pragma once
, [( y# E% t" T  b. l#include <exception>

& I6 {2 s8 T7 o: q) b
  enum TRISTATE{
0 b- u3 z, p, e% i3 B9 d# O, R        TRIS_FALSE,
; B$ ^4 J  a: @# `) {% X+ ~" A        TRIS_UNKNOWN,
  F, [# i- b4 e0 F  i3 v6 J% S        TRIS_TRUE
$ h/ x8 {2 i  Y+ W6 T};
% C' x; ~1 r8 ]* w, {: C
4 \6 J, x/ B- t" \! Z) b2 c
enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
& J# Y) e) w2 r* y4 q+ V- p4 f5 X        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
$ |" Y3 F& M& `% B; \) t};
$ P* p" `' ~! n, G" P& j, @



1 Q  G2 j/ D4 Z* Lclass CUPnPImpl
{
public:
6 r+ ^. h# u6 w( E3 t- ?        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
) G4 m7 [) J: B) t                UPNP_FAILED,
                UPNP_TIMEOUT
# q- h9 h, s! O! O( z6 J        };
; }7 G# j$ M/ j! W" A) I
6 ]$ [( t8 [8 ]1 ]% a  X
9 B! k2 X# z2 p        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
8 t( L- g7 G+ M" r/ I$ o3 C        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
% Y5 ]! o% @% D* @- \# [0 a        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
* b! \) R! k- {6 L" h5 y- A       
        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; }
; z( t) d& k* G+ R        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


5 H3 R  ~2 p& N, \+ I7 [; M// Implementation
protected:
5 W  N9 @) b5 z0 H  j( T        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
; U: C, T* ]1 V* E8 b/ V* n        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
9 m# T9 V, V" f$ r8 R- B8 [        uint16                                m_nTCPWebPort;
& k0 ^* T6 l! G1 ~/ A  g        bool                                m_bCheckAndRefresh;

' s" s1 K: T6 `: x( z
$ w7 Z$ D( t$ R) m# z9 L0 T6 K* pprivate:
, O4 q# O6 q% W  ]2 V* N$ S        HWND        m_hResultMessageWindow;
- t  L8 ]. g% g        UINT        m_nResultMessageID;
2 h9 C" z2 r0 A4 X0 B* `" C
( R  d: |4 `3 P* N
6 S, @: [/ x0 B2 [7 u};

  |* k% ^7 n  r( H( _" _9 P8 z
# ~, V- ^  `. \2 W* K0 t. @// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
' H+ N8 O" ~  K1 M! i% ~1 j{
6 @% s" u8 [! U" @4 T6 `; Dpublic:
' b7 l0 H! D- |2 o4 H  [7 c        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
3 M: V* X# Y* u3 Y+ d/ J  D        virtual void        DeletePorts()                                                                                        { }
9 }5 G3 S6 D/ K8 _        virtual bool        IsReady()                                                                                                { return false; }
% D' Y* ~. R! b        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};


/////////////////////////////////////
. T( R- W2 K0 r  \1 r2 r  k//下面是使用windows操作系统自带的UPNP功能的子类


#pragma once
- m9 k! L8 y; ?& x/ ?- k' c#pragma warning( disable: 4355 )
5 `6 d4 H* |5 r. J/ @$ k% L

#include "UPnPImpl.h"
" L. _) j* M0 |2 r- w8 p#include <upnp.h>
! X; e3 [/ j6 H) _2 P: ?2 t#include <iphlpapi.h>
( _5 ?9 t$ v& @) N& @#include <comdef.h>
3 `% c, a( O" h" B#include <winsvc.h>
5 H0 s9 a. j* I( d
- R) ~+ ]  J$ m& T  b* S
#include <vector>
9 O4 P" C, \& h+ d# x#include <exception>
" r7 O" t" x5 ~0 a8 B* K#include <functional>
7 G3 ^1 s( }+ m8 d$ a. j8 g
$ n8 J% J2 v* e; F* m( M; ?$ O0 M* y
' M3 A$ V" V3 E: i

% L8 U5 [# L9 ]3 [7 n& Stypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
' D! u1 Q3 C$ s6 }9 ktypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
; A4 Z* p% {4 v' x7 \' _typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
* ^, f+ ?- e. z' t4 I8 Mtypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
& j5 y5 j: p1 v# K
0 p# B' @0 m' @- ~
* [) e: P' F: N+ t. |8 c( Mtypedef DWORD (WINAPI* TGetBestInterface) (
  v. _* r0 o+ O9 U$ {+ c0 p  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);
0 ?9 `- C# }& A: O/ g

& Z5 p7 O1 C! S$ |typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
, ~  x4 E$ U6 W0 {/ ^0 J. |  PULONG pdwSize,
( H1 u* g4 z) G+ v; }) H' |. v  BOOL bOrder
2 ^9 ]: b. S7 i);
, T& u" f/ G  C+ m6 ^& J2 k  `+ ?( h

typedef DWORD (WINAPI* TGetIfEntry) (
" D1 K4 [0 o1 s2 r# c- m- d  PMIB_IFROW pIfRow
);


5 E4 ?7 M- |. A0 |& P* C2 |$ ECString translateUPnPResult(HRESULT hr);
. f) A& J0 O  \6 ^/ u6 ?1 ~. [HRESULT UPnPMessage(HRESULT hr);


5 u9 b/ G) `1 `; o3 I, Sclass CUPnPImplWinServ: public CUPnPImpl
: y& `9 v: l4 G4 Z2 Z{
$ {- D& o. X: E7 ?0 O0 s* _% H        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
3 ?/ F; t9 h0 {: y# [, I// Construction
. x, D) I( s+ ^2 |4 }5 p4 xpublic:
& o" d5 d" s0 _- X# v" e9 V        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();

+ x" K8 n, o- Q' t
8 _/ b' U: Q2 S! G% V        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
. C; _3 w8 W0 F        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
- A, B& I9 M9 |& o( J) D        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
) |4 A9 A  Q5 F
4 r- O5 }" L& K, o. [6 e
        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
; a/ @- r4 P. ?( |2 k$ B/ }9 b        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
7 \* u7 _7 Q* ~4 Y: m

protected:
; p( N) Y* T  _0 D9 ]( b* v7 r3 A+ |& Z        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
( S( E3 n0 M" T5 }, K  F        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();
$ w: s0 ~( n. ?) f' s' k$ m0 @
8 _" I* g* ]: w& X# K9 i
: v! d- H$ O7 i2 T0 S        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
% x; V/ A: }9 H1 X                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
+ G6 h7 a; I, D                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
5 i/ X* R0 }% T- @6 a        }


        TRISTATE                        m_bUPnPDeviceConnected;
4 o/ ~) N# ~2 Q. J6 u7 o

// Implementation
; i5 \, N& ^( X. k        // API functions
6 V/ b+ |; Z# d8 N; l) ]' z, H$ ]" K        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
# u8 F/ ~6 H' ~# l/ G9 X" y7 s3 Y        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
  ?4 n+ ?. u5 S1 O% J        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
6 t* J0 p5 l8 M( i  {        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
! t% A3 ^* R6 k* y        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
- \+ a: h3 E5 D( m        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
! u: p* ^+ s7 [- e! {
5 F/ V9 A: o" u2 @2 g
        TGetBestInterface                m_pfGetBestInterface;
& Y* |2 p" c. Y. o        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;

/ c/ w% x; k/ z/ _7 O
        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
: C! k( ?. F5 J        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
2 o- [$ G; o* L2 @                {
" I6 v% s% `3 l                        CComBSTR deviceName;
$ F: n0 o+ E% G, M% K5 j) ~                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
& q+ f$ W0 R# F% c  F

                        if ( FAILED( hr ) )
9 R3 w& y& H/ ^) x                                return UPnPMessage( hr ), false;


                        return wcscmp( deviceName.m_str, m_udn ) == 0;
  n. Q+ D8 B9 u& m                }
( J! m0 s8 ^: M8 _" P, c                CComBSTR m_udn;
3 Q$ ~' l8 D# E" ?% r6 l, Z+ G7 a        };
       
% l/ z2 ~5 o& Z! l  r2 `        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
# ~. G  _  K) i) G" r& b        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
( B2 _( i7 \4 I0 Z' W  E4 |8 K        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
2 C% G9 F; w. n7 X        void        StopUPnPService();


        // Utility functions
2 |. i/ m0 X) G        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
4 F2 ~# `" e- w  v* \$ E5 A5 g; u0 I+ y        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
# m$ b; J, A" w4 S1 l: D! B8 y        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
: Y' p% z# E6 c1 I- H
3 _3 q/ f# f  }0 C
) T  c  Y- n' d8 l5 [# f3 z, ]; n// Private members
9 x& s' X' ?2 m) xprivate:
0 E! Z$ F' A2 B        DWORD        m_tLastEvent;        // When the last event was received?
2 M% S" U& @: @' A' ]) S        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
7 u  ^# e8 g+ H8 \1 x        ServiceCallback                        m_pServiceCallback;

6 O; c( }8 y1 A5 T0 @
0 C9 \8 |6 r2 A        LONG        m_nAsyncFindHandle;
) g! B* _/ u7 R1 W        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
" U& f% I7 W: [/ |# E3 ]        CString m_sExternalIP;
& U  i" Y/ U% e  U' J$ e        bool        m_bADSL;                // Is the device ADSL?
3 Q1 ^8 K4 N( ~( F3 [7 D        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
6 g6 Z. ~, E* E        bool        m_bAsyncFindRunning;
1 g$ D; K* I) q4 A( i, j1 z  \        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
( l2 I, d+ K9 U/ v1 H! l3 q/ v+ k        bool        m_bDisableWANPPPSetup;


) Q, @* i! @, Y  z( u; i* M};
) T( _. `# }9 s8 q# l- }5 H5 u* ~

// DeviceFinder Callback
: y" _8 I: N5 e. i. tclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
: G8 X6 s/ b1 W0 hpublic:
: G+ x+ [* G6 \  n% }  j- H        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
% l( p+ T7 W! J, M        { m_lRefCount = 0; }
7 F% i- ]% W; J

5 D1 ~) M* P  _% b4 }. m   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
9 G( \. ?" S2 c$ K# }   STDMETHODIMP_(ULONG) AddRef();
- z9 ?. N! L- Q! }* [   STDMETHODIMP_(ULONG) Release();
* f+ R3 v& Z0 X6 @- F3 D0 l
5 n/ c, R6 D# E; u
// implementation
( V$ h" U/ a/ T( K' Vprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
% Z) g% t2 A7 D- ]6 |# M; T3 Z/ w        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);


9 D/ x* l- t8 e% ~/ Q2 Xprivate:
: s' X  Z# I' a% A, _% T+ U) i& K        CUPnPImplWinServ& m_instance;
( B. n  D% k9 c        LONG m_lRefCount;
};
0 S/ k9 O8 r4 M

, M: n6 X! u2 R5 L& x// Service Callback
( v" \/ e+ K5 E& _9 [class CServiceCallback
        : public IUPnPServiceCallback
2 O9 a9 M. [5 T1 r) H" p; p{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
   
+ M7 `" E4 q$ o2 M# T   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
$ G6 U+ Y' {& P. V% C* N   STDMETHODIMP_(ULONG) Release();
! x( Y* u4 {0 s4 o7 o
* H, J/ c& |* C, n; s- [
2 `& O0 O' I1 g$ [0 Q// implementation
/ F* ?: a  S' B, M( v/ K9 m0 e) u3 F6 aprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
$ Z& V8 a( y6 v+ h

private:
        CUPnPImplWinServ& m_instance;
! I" C3 r% q" }& E' a: T" K        LONG m_lRefCount;
};


2 X2 S8 R1 N% d9 }& n! c/////////////////////////////////////////////////
6 {5 C1 \( _: a# L
/ W* x. d. }& l+ Z9 \' O( [
2 d  ~6 l! g' {  T( d* l+ x$ @使用时只需要使用抽象类的接口。
, b, w; m# v, D7 r- iCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
  }+ g" Z/ |' e# E9 i$ o8 iCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.