UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   ( S( T) _# q9 A* a
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   / y* n2 u: V3 n+ S2 i5 }
8. class   MyUPnP
   4 a. ^: v! G3 e* X
9. {
   
10. public:
    % n- @; m& _9 q3 ~1 @0 c
11. typedef   enum{
    ! d; f3 \- D* E. R" Y4 R4 _3 ?
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
    
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    ! K' |- \1 f6 I3 X' i
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    # ?- ?, S' ^* r2 R, q
19. typedef   enum{
    $ T5 Z0 C- [1 X7 `6 b* l0 f# M
20. UNAT_TCP, //   TCP   Protocol
    4 X9 D6 t1 C% f" f5 }" }) @
21. UNAT_UDP //   UDP   Protocol
    / z: J& d8 k9 d! l' a' y# S
22. }   UPNPNAT_PROTOCOL;
    / }7 N* \2 ?9 q- `' f
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    8 F# ~: ]* x% ]9 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;
    
30. 
    
31. MyUPnP();
    ; ^/ t) t( p3 R5 }6 G% {% [% t7 ?
32. ~MyUPnP();
    
33. 
    ; \, K# s- `6 o# K
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    1 ~% Z, G: o6 F  ~0 F0 _4 x6 V
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    % \4 q7 `  `7 h7 f4 ^
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    : K- s* t& ?# X- @: C
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    0 g$ M. j" ^. q9 x
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    . A! O. O, H! U3 O+ D! N* X, c
46. 
    2 ?4 R( C9 w- V0 U
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    $ |$ O6 L  a+ N: e# f- {
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    6 K1 W1 ^  k" g! R" k% W
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    % L1 V0 _- S5 X9 d* m
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
      l4 u5 ]6 N# a
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    - K+ i4 ?8 {2 N. Y4 B5 _$ n
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
    ( U/ H9 Z& S. W5 E: E0 r4 ~
62. CString m_description;
    
63. CString m_baseurl;
    ! T7 f8 @  u( ~& y
64. CString m_controlurl;
    # j  v% v& a. x
65. CString m_friendlyname;
    $ I( U! v* s, a2 Q  \
66. CString m_modelname;
    6 b+ H- l$ j! L1 N7 t. H/ r) P
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    ' U1 ?7 [1 r+ l2 C
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    " `0 g4 F0 O( {& W7 {6 f
74. WORD m_uLocalIP;
    
75. 
    3 L% Z& _. D' l3 j" n
76. bool isSearched;
    
77. };
    . n0 v5 Z: ?' j7 r& y& W( C
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   ' H5 J  l1 s. A: Y* F% _, h- N) i
2. #include   "stdafx.h "
   2 ~7 y, O" j" f1 g& K
3. 
   1 l  h% v" S0 q) l' X( x
4. #include   "upnp.h "
   3 ?; P8 q: ~) _1 e1 u9 @
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   ( Z+ m  d4 }$ h( S* V2 w
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   9 k0 {$ z% t' q! A8 Q5 i' w# x8 `
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
      B* z! \& }! F
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    # y; \4 Z1 \6 x6 I8 q1 z/ @- G. U4 ?' [4 p
15. 
    
16. const   CString   getString(int   i)
    ' p8 V- Z3 ^" g7 o; ?1 c
17. {
    
18. CString   s;
    8 |7 ^) _4 q  I. `
19. 
    / T3 }. l2 h" s$ Z4 ~
20. s.Format(_T( "%d "),   i);
    5 c7 h9 L+ ]! Q5 Q1 `" H
21. 
    
22. return   s;
    1 f' B* M4 Y8 U* k  c, F/ W; Z
23. }
      m* ^+ X; E& k! }2 v" M3 W6 r2 ^- J
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    / L$ y! ]* i( S5 I, d  j1 k# D; x
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    + _8 [% E  R# K) E7 e* H
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    2 G  Y: T7 O* t; y! q
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    5 C5 Y! l) W) q
36. {
    ! Z8 B' d. w2 e8 j; `+ Q# y
37. char   buffer[10240];
    
38. 
    , y, {, C  `7 [0 T4 _! k( L
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    # T1 Z$ q/ h1 g  _
41. strcpy(buffer,   (const   char*)sa);
    9 b  L; _+ B: _* c
42. 
    : A/ g0 K5 r  [* g% B& \- G8 ]
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    ) j5 @" [# c- Z
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    * W# C# Y$ ]' o( t- X" h" T/ ]+ s1 A
46. sockaddr.sin_family   =   AF_INET;
    . D) c! v7 K5 H* E
47. sockaddr.sin_port   =   htons(port);
    9 X% c& a3 i2 Y& l1 n& p
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    & m0 \; V, r/ P
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    3 X  o4 _, r! S) y' r# O
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    ' W* B/ D5 k3 R- _) G# |# z
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    4 w3 J0 C3 u& ?
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    . L- p6 d* q7 T3 O) z1 z9 a
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    % K8 N5 P, x$ O: ~% h. n  g  ^
63. return   true;
    
64. }
    % E" Y4 P& {0 L+ f
65. 
    7 W# o3 }* V  [) @( e
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
      s- O! B" L2 J
67. {
    % m( d! U6 }1 a
68. char   buffer[10240];
    
69. 
    * ?; F" r( x9 x+ J* {. T
70. const   CStringA   sa(request);
    * `9 k# N; Y/ }) _+ I* i! @
71. int   length   =   sa.GetLength();
    4 O# b5 W" {8 g& P1 o8 F& `
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    & U+ X9 B9 Q" S% k0 S2 X  p9 O
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    6 b9 v( Q: N9 _5 y8 |, J2 @. h
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    - u7 x: V9 J: r+ I1 q) C  E
79. 
    . ?2 M0 W: e$ H1 ~) B3 v7 s
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    - S' e& Y4 ?9 B! w2 C
81. }
    
82. 
    : n8 S4 q$ [& o
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    . j  D  N" ~- D( _0 e# M$ N
86. 
    $ x' f+ i$ T$ M7 _
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
      S7 ?3 C  @  E+ C) U
91. 
    ) I3 w' {8 P% d8 ]) D! A
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    + p) Q+ {2 O0 ]
94. status   =   status.Tokenize(_T( "   "),   pos);
    & [7 b0 k# r% i( L
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    " r+ ?  O) v- {) y$ Z. ]
100. {
     0 M+ O( X8 e* A& Q- U, U! f
101. CString   startTag   =   ' < '   +   name   +   '> ';
     % @+ |0 X# d- w4 C2 k$ _. P' Y
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     . k: C: S6 J- g; h* F
103. CString   property;
     7 t* ~; S6 a2 w
104. 
     & A% c6 V0 i; w
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     - c& e; {* q" b5 N0 _- I2 d0 P
108. int   posEnd   =   all.Find(endTag,   posStart);
     ' }: p& K0 L/ o% u* n
109. if   (posStart> =posEnd)   return   CString();
     / z% I' V6 c+ A3 c1 y
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     4 q8 F0 [% K6 T! h
113. 
     
114. MyUPnP::MyUPnP()
     . q) U3 |8 U: Y8 _: f4 H
115. :   m_version(1)
     + q" h2 s0 g6 w7 ~" F- w( [
116. {
     
117. m_uLocalIP   =   0;
     2 A, R/ V% l' V0 w& E. z
118. isSearched   =   false;
     
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     3 V# M3 D# b9 b$ a- h7 O
122. {
     5 b6 G  e$ H- F. }- g) o2 E
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     % q* L8 f5 M& S$ |% g0 S$ _
126. search   =   m_Mappings.GetNext(pos);
     . |/ |- J6 E$ I
127. RemoveNATPortMapping(search,   false);
     6 ^1 t% P  K) e: [; {- T* W( C
128. }
     ) y, X* q, [. r& d+ I# r
129. 
     4 D3 P8 A( |" C% T; a7 T" A( A4 v
130. m_Mappings.RemoveAll();
     
131. }
     9 U* y0 ]/ T- j. O4 Q2 J
132. 
     ) u9 F5 {9 P' Y. Q
133. 
     / t; D, ~+ Z  r0 z
134. bool   MyUPnP::InternalSearch(int   version)
     5 ?" I$ Q) X$ c1 O7 ?" U
135. {
     5 ]+ }, Q1 t( @; j- i& K; A& u0 ?
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
       ~# A/ u. k; l' ?6 l9 I
139. #define   NUMBEROFDEVICES 2
     6 P3 j2 y) ]8 v8 p' l' Y: ]* X
140. CString   devices[][2]   =   {
     + ^" o' p5 B# ]6 t* J3 r! Y
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     1 Z% |1 y  D( r3 C+ P
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     7 V0 E5 h& X: m# w2 e! r( J6 R
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     ' A. |# t/ p/ m' N
149. 
       q) _: t( ?. {  g* ?: r" |" E
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     - o2 z2 U) h% g; D
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     % H" Z" z6 t, Z1 n+ c- M# t
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     . ~; s9 b5 m; |9 e" Y
155. CString   request;
     0 ~6 b# |- @8 x6 g
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 "),
     8 F: {" `! Z+ g4 C/ f  v' S
157. 6,   m_name);
     * D5 r$ t3 ^, F! s8 ^5 `
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     & f+ W, I0 {3 s8 U; N
161. 
     , D+ U( z  K4 u# r3 z
162. Sleep(10);
     0 Q, U- Q: }5 Q& ?/ ]
163. 
     
164. char   buffer[10240];
     ! O$ z  c# I3 o. b
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     . M) k0 i- k4 D) z/ {/ s* J
168. 
     1 @# t& W0 z; m: [. P/ x
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     0 A6 q- E' T- ~' H. M, t$ K$ c
172. 
     . M4 P4 f7 q5 E
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);
     0 b5 e7 r  L+ x% h2 B- t6 O3 m
175. if   (result.Find(m_name)   > =   0)   {
     4 i) T& ^) X: O2 F! V6 W2 w
176. for   (int   pos   =   0;;)   {
     7 x3 A2 b. d- u6 H  C# x
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     5 \) O$ l' h4 i2 J( E$ q* a& |
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     $ B3 t, K  J* m0 O3 K5 h" |% J
185. return   GetDescription();
     9 f. G# U& u* n/ @4 G2 m6 ]+ u0 p
186. }
     
187. }
     . G& ~: T: _3 K' I! v8 o+ N& }" g* m& u
188. }
     , A$ r" H# f; v. b) F
189. }
     ; F0 N  l9 W$ A
190. }
     0 v4 K2 k8 G8 V2 o& X; o0 s; p
191. closesocket(s);
     ' ?& y, Y& K- l* ?
192. 
     
193. return   false;
     1 o3 {3 o2 W# N
194. }
     + c5 s3 [7 s' o( {& Z1 y

复制代码

天云上人

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


, p* I8 T8 ]' r/ j///////////////////////////////////////////
% s0 A+ h( @7 w3 M//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

. l( V  M) |, w
#pragma once
1 A" L- t  a  O4 N#include <exception>

% K( Q) R9 F0 `: ?; O; R/ I
, x% X( @0 \- V  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
/ F/ x. u# l. B& D        TRIS_TRUE
  E4 y6 o8 X9 X. J( v};


enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
" b0 C) `1 {, T: h3 ~# X2 s        UPNP_IMPL_MINIUPNPLIB,
; G0 S/ T7 \7 [        UPNP_IMPL_NONE /*last*/
};

- v2 R  I1 J# ^( |6 Y% {# W( N$ b

3 C: k  D5 G; m( Z3 x% z; G* x
class CUPnPImpl
{
public:
3 X; o8 z( y7 R: d# W3 w) e& V9 A        CUPnPImpl();
4 p3 ~# _; s9 Y: F8 |+ a* C+ e) B        virtual ~CUPnPImpl();
) j3 Q2 `% ?8 |1 f& x: F        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
        };

5 B1 R& g6 T8 `0 V: q  U
- E$ G& O9 H- X0 y2 T, x% ^/ B        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
; }% ~1 b9 _. p6 o9 U+ q        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
6 H; A# X3 [" d$ W, G        virtual int                GetImplementationID() = 0;
       
0 k6 W3 L( [* T3 |$ C        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

, U& ^: _  S6 |2 b! R9 S
, |$ |) D" Z+ ~( ^; |6 z7 {/ W1 O        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

0 r: H  X; @. C7 j: C8 C$ t  k& y7 l
// Implementation
# f9 ]4 d& d' {" c. E- A% Mprotected:
" g) h  m" X! f        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
9 T: ^2 @- j! p" s1 L        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
% I4 e0 Q" m6 g6 p: f/ O        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
9 `& d& s' b9 ^) x

. M2 t- b3 @7 z6 x; \4 [: j, hprivate:
8 Z3 l% h8 r6 I* ^9 E5 k        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;

/ y" A, g- a/ b$ J% x8 U0 x! h0 X
0 t2 W! d0 Q7 m. b+ U};
9 d0 K; Z. {; ~3 \

  \) u9 P7 u: y$ t5 c// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
! V  g' s1 h6 w8 h2 ]6 K5 l' }! E{
7 X2 d+ ?$ g# jpublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
4 `  O! z/ Q9 S2 u% a        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
( d  _1 |  q+ R( y};

2 y" W5 T5 Z) ~2 A- F
/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类

3 R4 B: q9 `1 c3 b6 Q
#pragma once
#pragma warning( disable: 4355 )
; |7 X+ x* U/ u4 Y" n7 }. v2 ^
/ K0 K% |4 k" b) `! Z* b5 c0 Z; x
#include "UPnPImpl.h"
) g* l5 [4 I6 M3 e#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
% Y7 |+ c3 B+ i) q  n3 Q) N2 V#include <winsvc.h>
. B* w: ^& s# l5 G; z
7 y' w$ U1 ]- o+ b1 w7 K- A9 `
% w( Z8 |  P8 N* `' O#include <vector>
0 f7 F! j! B" [# W#include <exception>
8 k- V& G' x/ f' T  H#include <functional>


3 _$ m, F# x! e$ {- o
' x# c* [8 c8 Z5 X
. {5 e- [# I, q/ {7 H- Etypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
; v* _' x7 q1 ctypedef _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;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
' B5 s) U' d: o3 d" }typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
/ h3 w, S0 `6 O& P& P/ `, l+ P
3 n5 [) K' O# T4 K3 M1 j+ T9 x# U
typedef DWORD (WINAPI* TGetBestInterface) (
4 N" R' k  L/ Y  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);
1 ^; o6 t# R% h$ V7 S" o9 z
- W, _* n/ I" G2 O2 ~" ~4 q) F
typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
- q7 E0 S" m+ q/ y. W+ W% F  PULONG pdwSize,
$ q$ [) M9 ~2 U  BOOL bOrder
);

1 Z9 n% R6 b/ x6 N# ~) W, o
typedef DWORD (WINAPI* TGetIfEntry) (
- h6 V0 L% i; j, Q- e* E  PMIB_IFROW pIfRow
9 D2 n+ W2 D0 e& P);
8 e) N' P: H5 g0 P1 e8 z1 `
! p  u6 s" ^! W7 D8 v  j6 W6 Y9 ^
CString translateUPnPResult(HRESULT hr);
4 O& @$ ~8 v) _$ u% Z! W* V8 m! zHRESULT UPnPMessage(HRESULT hr);


8 s* ], X' t) j; _3 K$ mclass CUPnPImplWinServ: public CUPnPImpl
6 [% p8 y8 ]! T7 g( i$ e0 N2 I, {2 r{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
public:
: h4 v- h2 N2 O: b1 F$ |* C9 J        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();

9 }5 m6 }/ ]7 I' R$ ?& f& a& b1 \
1 C% F8 W0 u; D2 E$ g" o- H; m( ^        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
8 [/ T+ x2 D' E4 n6 J8 j& ~' u* D        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
$ _9 g, K- M# K0 }3 @        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
1 q7 Y9 V# T6 m9 K& g  j        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };


protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
; F3 k+ i8 V. T+ |) E' Y        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
- E; D! o& R/ I2 V: M        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();

7 ]8 s1 N& Z: V3 r
: }' h3 q8 O5 M5 u        inline bool IsAsyncFindRunning()
9 n- X, \# I, N4 b) ]! Y" y5 ~        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
( K7 D! l8 [$ o% t; R* U+ w                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
* G2 Y/ l1 y- d                }
                MSG msg;
2 n3 q4 q( s3 C. c/ A2 X  S1 m; j! f' V6 w                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
5 \; Y( N4 f- k! f* i* r                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
        }


: |: i8 _7 B1 p0 ?& H4 V1 i        TRISTATE                        m_bUPnPDeviceConnected;
' E" I- b) w, q  F- U- k" y/ G8 `' `

// Implementation
        // API functions
* }1 N$ W6 E- Y        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
% I  p* h- k; [8 d0 [        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
# ?# Q  K* _" ?( t7 U9 x7 G        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
' N* x8 V1 o! W! F9 m$ N& [% ~        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
/ ^# F, d& z9 h        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
+ K) \' u3 u% z7 \7 [3 x
& |  F  G7 X0 ^; S. q+ W  o
        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
( I$ [; a5 S1 R        TGetIfEntry                                m_pfGetIfEntry;
: }- |2 f; n/ ]* c, `' @; E: |0 M

        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
6 ]' n/ }4 Y8 e  C3 e' h  m% z                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
0 z7 I( P5 W' ]3 C1 F                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
* }  |: U/ o5 A. l
8 ?- o8 ?0 y  m- w" O0 P
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
0 f+ b- d) t  f                CComBSTR m_udn;
3 u3 D9 w- m' ~$ A5 W        };
       
. {9 N# R+ l9 J1 d0 H        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
- e1 m$ M5 n2 b) w        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
$ K7 J" A# t0 P. S: E) x1 P; ^                LPCTSTR pszInArgString, CString& strResult);
- Y, u3 c% o) K6 w" ^        void        StopUPnPService();


; g1 M5 ]( Y) m7 [- E. Z$ T        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
. ^- O6 [/ }8 k0 ^3 Q        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
7 O7 Z7 E+ ?5 X        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);

: W3 [' G4 ?) |9 R0 v5 c
// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
1 {9 U7 ]2 F! r& s# c        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
$ P+ R2 g3 q# Z8 R1 \, J5 r        FinderPointer                        m_pDeviceFinder;
) N4 V) v# a( t* Y        DeviceFinderCallback        m_pDeviceFinderCallback;
4 ^+ M9 u  \) \+ P        ServiceCallback                        m_pServiceCallback;


2 ~3 F, p/ Q. u, U        LONG        m_nAsyncFindHandle;
7 P+ f- y* I2 t) f: E6 G        bool        m_bCOM;
. ~  p; D6 b" W        bool        m_bPortIsFree;
        CString m_sLocalIP;
( w) H! T9 `! @' }" ]* p        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
2 ~! V. p: f( K1 w: }7 O# Z        bool        m_bInited;
/ K% b/ s# {: e- K  E9 ]$ Z! v( X        bool        m_bAsyncFindRunning;
/ T8 `' u) N3 r        HMODULE m_hADVAPI32_DLL;
/ i' y1 M8 V4 z        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
! o7 Q) ?$ W7 K" _0 `, s/ {        bool        m_bServiceStartedByEmule;
9 x, `. }4 H) L3 k9 }- L        bool        m_bDisableWANIPSetup;
( V, ~  Z% j) Z1 A' N( ?, e9 |        bool        m_bDisableWANPPPSetup;


};
) U' b* y0 i( ?/ l( F# k/ b$ `2 m
" A4 o1 s4 O6 A. p
// DeviceFinder Callback
# [& H# P2 |3 k4 {# P: g8 g" Z# cclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
public:
. |& ~; [( p, z6 c) |" N5 k# t6 c% u        CDeviceFinderCallback(CUPnPImplWinServ& instance)
" P3 u6 ?; ]8 V+ C( I                : m_instance( instance )
# M+ A- F  d; T% X" e        { m_lRefCount = 0; }
* k7 u( p5 e7 J

4 Z8 S, w' ^0 E3 t6 A   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


$ V2 m& w1 B, e1 V! r5 D% @( h// implementation
2 Y" ?2 B7 `6 i! V- s) {( L- ?1 C$ tprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
& K3 T# Z  F1 }- e        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
% f" Z7 Z: c: j8 E        HRESULT __stdcall SearchComplete(LONG nFindData);
/ u1 k% {; h$ y" N, L+ o
) d6 a" w; g7 W( @8 ^, b* K
private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
1 D. r7 |- j! Y& w# @

// Service Callback
) \8 w) S# C7 v( ~class CServiceCallback
        : public IUPnPServiceCallback
  z! S9 V* \' m" p5 N2 S, M{
* r) p1 [+ P: D8 Lpublic:
        CServiceCallback(CUPnPImplWinServ& instance)
: Y! W, T$ {% F- |                : m_instance( instance )
6 z; H: M/ y, }3 Z/ e% Q& u        { m_lRefCount = 0; }
- i: e& U& A8 w( B) q   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
; M' j" u% E& Z; \/ o   STDMETHODIMP_(ULONG) Release();
5 G2 F$ H$ \* O* r2 E. R4 R8 s2 w
# ?# u. \0 T5 F6 D: o/ o7 ]+ H( z
// implementation
private:
2 \* ~1 |3 H; q& f  V) [# F        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


6 g: V# M3 C( \4 hprivate:
        CUPnPImplWinServ& m_instance;
4 f8 E3 B) G. i2 Y. H0 t( E        LONG m_lRefCount;
" s! j9 x3 e: N+ F, A1 a) o};
4 v# R+ E# U4 o7 r3 s

* F" F. }( _* m" n7 D7 e) E+ J/ S/////////////////////////////////////////////////
, Z( |2 L8 o5 k* K* B6 h2 S2 O; r
# w) [% H) A4 u+ ]
8 q* ^7 ~/ W' I$ P; L# [使用时只需要使用抽象类的接口。
% d; z* z$ o6 i  O3 q% f8 ]. cCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
3 y5 t" a! Y& a0 U1 YCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
) I2 |# ]& L6 O' \! Z! s* [CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.