UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   7 j3 o3 Z8 a, n9 F1 g# s4 C. K
3. 
   
4. #pragma   once
   : K6 m- ?. d) T9 P5 `9 B
5. 
   
6. typedef   unsigned   long   ulong;
   ; C, S% F# T8 V- G/ m3 R. ?2 _
7. 
   " [( H$ I  v" E2 u: ^/ o5 f
8. class   MyUPnP
   
9. {
   * a  j0 R$ V7 O
10. public:
    0 ~- H1 o7 \1 b* N$ r/ N
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    / _* N% P9 g0 g5 E. Q
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    * m6 \4 B- d# p: l  ^$ Q
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
    - h+ c" {- E: z# e* L
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    ; w4 Q8 k) r4 ]& c: a. }5 a
22. }   UPNPNAT_PROTOCOL;
    
23. 
    7 q. l4 v( x3 i' f. F7 e
24. typedef   struct{
    7 _9 N2 V% f' n# q0 x; Q
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    % h' }. F( a1 x& `' f/ ?
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    9 r6 ]# x, |. c$ m9 g
30. 
    4 g& L0 Y: a8 M& Z- `* [  U
31. MyUPnP();
    ! q0 C5 v+ I3 K" W+ `
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    # p8 W3 i; _+ I6 E
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    $ H: R% h2 o$ p# j$ o6 ^
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    ( {+ t, {8 Z8 G# |+ X$ p
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    9 ]$ B9 \% `1 R0 y' T5 E
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    / y4 ~  G% i3 l7 s  `" Z  `
48.       const   CString&   descri,   const   CString&   type);
    7 k: a9 y! h& o) }5 U
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    8 h0 B8 ]$ i/ G4 A1 r% u
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    8 s% _+ ^8 {1 s- B3 f- e
52. 
    7 ~+ y6 m- A: Z$ K% I, ^
53. bool Search(int   version=1);
    
54. bool GetDescription();
    9 T0 ~& w8 u; V) w' e
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    , E! d  v; t0 Q
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    : O) |& F+ B! ~& I* i2 z( I+ Q' [
59. bool InternalSearch(int   version);
    ) }& I, b  i. N9 `( M; J3 S
60. CString m_devicename;
    , M2 h' n& n- d# b, ?2 q
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    + D4 \5 E  [0 b; U2 q* ]  n
64. CString m_controlurl;
    
65. CString m_friendlyname;
    4 y# b7 F& a) [. B+ F% x
66. CString m_modelname;
    ) M# ]: Y6 x: E. W1 v$ L
67. int m_version;
    9 N' w! r- ]+ \6 g4 H
68. 
    6 N5 n  G4 @, K4 ^3 p
69. private:
    / m, f: N" u5 j. Z
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    ; `' h% s5 R$ n6 a. M
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    0 q5 B* i8 y9 |
75. 
    " {" F9 ?& t1 r
76. bool isSearched;
    - [* g8 W' M% G; R# |, b% y, u
77. };
    $ n' n! R" r0 V- j
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   + z) X# P1 T  R- p
3. 
   
4. #include   "upnp.h "
   
5. 
   9 x* x3 p, j  i5 O9 T
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   0 _) Y6 l! k& t: {3 y) b
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    5 U. N  K& k: V- d8 `9 I& ^
11. 
    : v# x9 Z1 W3 L, }( |
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    - n' B3 l  }* _) F
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    . [/ n$ o+ I2 G* @, K8 l
19. 
    9 V5 F( y0 O. R( M+ |# H
20. s.Format(_T( "%d "),   i);
    " m5 j6 S  ~- c) g7 c% U
21. 
    - E. `, x0 |, z
22. return   s;
    ; f4 e5 U* W: [: r5 u; t
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    6 H" Q% q1 z1 C! |6 `3 |2 N
26. {
    2 z+ {& o7 l8 M' p8 W
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    ; n5 E) d3 X7 i$ E, w8 r
30. const   CString   GetArgString(const   CString&   name,   int   value)
    / r6 m- N) S; ^5 l: I
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    ; S" Y2 c/ H6 b  l  z& g
33. }
    
34. 
    & w3 [8 R6 n1 G: h5 L9 p7 @& Z* a  f! [
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    6 s6 y( A( F  ?# R% T
36. {
    5 n. j4 P7 X! q2 s# S2 }
37. char   buffer[10240];
    
38. 
      W" M/ l+ h: w: C. f2 x
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    2 J8 [+ o" B  m3 {( T8 @
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    7 P5 i% k# x% r) ^# S- K2 ^& L, H
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    5 G9 C' A3 ~- B) H5 j
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    . F$ H2 u+ a; f. ]; G- ^
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    4 }1 ?  s/ s0 o
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ' t" v/ c( z! e5 V+ D% q+ l4 Y
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);
    & v* x, Y1 I$ A& S8 _1 I% M' n
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    1 [4 ~( C5 J; i0 n( u
60. 
    7 T7 s8 i4 v2 B" M
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    2 f; ]5 B' j) f& M4 {
63. return   true;
    - ]4 e& C9 q! I8 Y4 e
64. }
    # J" D* R2 R6 ^/ `8 @
65. 
      T" m# x3 |" {1 [9 G
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    ' V1 p/ H; Z6 b! k1 Q7 f- t
67. {
    
68. char   buffer[10240];
    2 N" d+ }5 ]- f
69. 
    
70. const   CStringA   sa(request);
    8 \) c- ]8 i; ]" E1 ]
71. int   length   =   sa.GetLength();
    + i) _0 Q2 @2 c" S$ I' L
72. strcpy(buffer,   (const   char*)sa);
    , e" a6 u) e* Q6 `3 g) [
73. 
    / Q! o. v, e5 m' N. k5 M  q
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    2 [% Y* l( m4 U
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    " \4 G2 P6 b% @3 H
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    3 j6 ^$ N& _# c$ ~" V. ?2 P
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    ! I3 T" {+ d8 }3 I. N+ G+ O6 \* F
89. result   =   response;
    # K5 L1 x! b4 H. L7 {
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    % y4 @; l. X4 z  N+ `8 g/ L
93. status.Tokenize(_T( "   "),   pos);
    3 `( @4 l7 O/ b
94. status   =   status.Tokenize(_T( "   "),   pos);
    8 }7 r8 x  w8 `( ]
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    . G3 n. N6 r* f5 u; _) }
96. return   true;
    , A" e, V  k' [4 Y
97. }
    + G  S( d% E3 K& w( F/ X* G
98. 
    , O1 J  O  [; o: f% A
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     1 q# C  P! z4 d; `1 L$ F
101. CString   startTag   =   ' < '   +   name   +   '> ';
     $ S+ G2 c/ e9 C/ }  M
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     7 J, h: A+ T1 u
106. if   (posStart <0)   return   CString();
     8 V6 _. Y5 e* P- @
107. 
     7 {* t2 U) D  M- O# D: A, ^4 {
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     / A1 N) R/ v/ P8 [3 p
110. 
     $ T  R& ^4 T3 h( @1 U7 l
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     8 H& T4 L  K, z3 P5 w% l
112. }
     6 g% m7 Z9 J0 Y$ K
113. 
     6 v* m* r5 {! V- p4 b
114. MyUPnP::MyUPnP()
     - k2 q$ H( t5 j+ M  t
115. :   m_version(1)
     - G1 D1 S& _; u
116. {
     
117. m_uLocalIP   =   0;
     " s; p$ ?# y/ n1 D' f
118. isSearched   =   false;
     / j0 G  E. D# Z
119. }
     3 q' P3 o" \- C7 ^/ S1 o( S' @9 s
120. 
     
121. MyUPnP::~MyUPnP()
     : Y' X! s8 J% H# s! h$ Q7 K% h9 i
122. {
     
123. UPNPNAT_MAPPING   search;
     5 R& F! m) b8 o: N# K  [2 c
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     ; m0 s% h1 e- e" M, b) @
126. search   =   m_Mappings.GetNext(pos);
     ! u- O9 N# w) p7 {
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     0 y  }) u+ b2 @! A( X
130. m_Mappings.RemoveAll();
     
131. }
     . b! c2 ?* h8 j$ U: }! @5 |& K( O
132. 
     9 {: p* }/ |, W9 q3 h
133. 
     ; e1 F! q- n9 V( T
134. bool   MyUPnP::InternalSearch(int   version)
     . Y: y5 c9 v$ u0 z$ F9 b* @% a
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
       X4 ]- D" D% ]( P* M7 U
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     3 m& j- d: M' O4 i; Q% [& k! }5 u0 n
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     6 W+ S/ X6 r' D. A& A% D
147. u_long   lv   =   1;
     ' m' v" f$ p: w5 v
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     0 _. \# p. _5 g  q/ T  V- L7 a7 G' A' c
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     0 N: H  D# p+ |$ \- W: y
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 "),
     2 x! w$ ?% t$ O2 u! B
157. 6,   m_name);
     ( z" D2 K0 W8 F8 z2 s% U+ q
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     7 c# E1 j1 P7 x( X& I2 s- [
161. 
     ( d7 T! ]. x1 l4 k
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     - s" }1 p" r& a+ ~1 `. I" X
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     & @# J, X, b& D6 G) b
166. if   (rlen   <=   0)   continue;
     1 }& i$ r+ ~4 g, C$ o! L0 P& [
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     9 V# _3 d: g  O
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     ! Y2 ^9 ~6 _' a! T0 {8 K
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     ! b' v7 [! M% m+ r5 F+ T3 V
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
       A, K6 o7 n- p7 y( d* L5 s
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     * m/ ^8 T( f- s. [( M" R7 P
180. name.MakeUpper();
     / F% X8 y  o8 {! _4 y- p  ~: d
181. if   (name   ==   _T( "LOCATION: "))   {
     . W' }: r1 \9 R) o: N) [9 U
182. line.Delete(0,   9);
     
183. m_description   =   line;
     " o% L6 f: E6 |4 Y" C0 L( ^
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     0 o- U( y* T' C. B2 i
187. }
     9 o8 @3 I" p: ~
188. }
     6 `- K1 ?  D+ V" S# S
189. }
     - o4 U2 A: y3 y+ N  e: i
190. }
     % z5 M# C! g& r- H9 S  b
191. closesocket(s);
     ; |& ~. u4 V! n: w3 s! a- r+ n2 F! V
192. 
     2 d  |$ y1 `5 B& D* G
193. return   false;
     2 _, \. {) v( q2 q. \
194. }
     ; c% x( I, M  e

复制代码

天云上人

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


///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
- }' s- K; G8 H! G! w: T5 R3 C
4 u+ H! T6 J! E2 g: K, O
) ]2 D" L  X0 E: [2 v#pragma once
5 e9 T  ~& F1 G; `# K3 ^#include <exception>

. n  T: ?8 ~$ G6 N% _
  enum TRISTATE{
1 e9 O9 N6 i  M' @% g% u        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
1 A( q# I9 d, m, d! _, M};

# f) e6 h3 C) x
enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
: u6 j7 f2 t* ?: @& r, T0 x) S  y+ n' f        UPNP_IMPL_NONE /*last*/
};
' h  ~& g: d$ [$ R" ~$ g$ S
5 C& U+ j7 x' h, D5 S/ I
" Q5 S) @" U8 W4 r, O" ?; [

7 k2 o7 {$ C4 S$ J: M  Xclass CUPnPImpl
{
public:
        CUPnPImpl();
4 |! R; P+ r" |9 @        virtual ~CUPnPImpl();
: s/ L6 J3 m' [; J+ B" R        struct UPnPError : std::exception {};
$ P+ s/ P, ~: |+ H3 W        enum {
                UPNP_OK,
$ ]3 d. @2 T0 t( T+ s                UPNP_FAILED,
! z. k( O; I5 M) ~2 v" A3 u                UPNP_TIMEOUT
        };
  C7 ]1 C9 ~; Y

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
; Z6 \2 F4 k( p" X9 E6 g' N$ P        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
, Z& C. D8 e* \4 N# n5 D, y       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
0 u, j. W* ^% `: ~3 [$ e' c

) k7 ?2 P" |5 E        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
/ W1 C; j7 |, [+ {3 U" F        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
& z! S: Z+ u) S$ a8 L/ e: b; e, x( {        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
! r1 w. @$ n/ e6 F        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
7 v& e; U% V$ v( h/ ?: h% d7 n7 q

// Implementation
: P* q' B: ~$ E3 jprotected:
; N' L7 z- o2 e% }, l0 G        volatile TRISTATE        m_bUPnPPortsForwarded;
" `) C7 A2 y# Q; W        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
* s7 d# ?. p! {1 g& ?0 `+ i        uint16                                m_nTCPWebPort;
, u" ^8 }! W; n$ s4 h' d        bool                                m_bCheckAndRefresh;

5 }5 N: O. p1 M: M1 f
private:
! b- F# x, y1 [+ J! \        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;

) \9 ]  f& @" w7 X
. T1 ?' g" e: c- z% }. z0 \7 d};
2 {- J& E+ O- ]0 C& n" G) J
3 e/ T, K; Y- F; L( [
) a6 ]5 ?' w) O$ L% W' e// Dummy Implementation to be used when no other implementation is available
1 j' \6 c8 C$ S7 s' gclass CUPnPImplNone: public CUPnPImpl
7 T6 i4 a! K5 q' X{
8 K7 n' [( N  A4 `public:
; ~' k' M8 l6 c' z6 H        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
. U  v( `/ x( d0 }5 I& l        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};

7 r! y# U! E& ~4 B
* F# W! R  r- t; S5 k  Z0 a3 {; w/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类

4 ^5 U% V& k* ^, M( Q- |7 {
#pragma once
4 @) L0 _/ c, E#pragma warning( disable: 4355 )

7 ]8 j6 p' T" W1 f) a
5 {- K# q1 s9 O, o0 x7 I#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>
2 F4 F: Q( P; G* k' L3 A1 |
3 E  \. l, t. v1 o' t- z# U
' V; D. _: i6 |+ }5 V#include <vector>
#include <exception>
#include <functional>


- p3 D6 d; D; h% B7 [3 F3 ^

typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
( T* l9 V1 r, _$ O* B, Ltypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
  q! w4 `& F# c/ ~% p1 A+ e3 Qtypedef        _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;


typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
- ~2 t, w9 Y7 j7 J, G( v+ r  PDWORD pdwBestIfIndex
);
" ]5 F7 y/ |. m/ G
5 w4 G- _7 s2 Q4 G' P$ q7 r/ [
" A0 Q; k2 m$ {0 O/ ~/ Etypedef DWORD (WINAPI* TGetIpAddrTable) (
, \9 v; `: J( C& {( ^3 c( Q  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
);

. \- v$ b$ g% s
typedef DWORD (WINAPI* TGetIfEntry) (
" _4 v, [, t. Q: Q, H  z! r  PMIB_IFROW pIfRow
);


CString translateUPnPResult(HRESULT hr);
- v2 z& Y  ~( VHRESULT UPnPMessage(HRESULT hr);


class CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
/ g( B& g1 M. y+ F/ m        friend class CServiceCallback;
// Construction
" T; p, |. h+ G0 N* [5 fpublic:
1 Q: t( `& M0 X' m) T" p) }        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
  i1 S. J* s, F/ N" ]

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
; E# g/ p! q  h5 h) n. Q        virtual void        StopAsyncFind();
! H: J; ^- T- ^0 P% @6 n# M4 v        virtual void        DeletePorts();
! h/ ~+ J6 P; H$ ~/ S        virtual bool        IsReady();
. K  j8 V8 {5 _' u8 J        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

% E! ]  |; {) ]6 O8 \
        // 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; };

* }( ]4 {$ \9 u( d( D; O7 {1 {
2 ~3 x$ P# A1 \; ?2 c  U8 O- P- nprotected:
% V2 r/ k+ L2 G( _7 \, t, Y        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
- }# o7 u+ h3 B  i! T        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
6 Z/ o, U+ O8 H1 L3 X3 z        void        RemoveDevice(CComBSTR bsUDN);
1 N$ X6 n4 Z$ m& j( \' }$ }8 |2 a        bool        OnSearchComplete();
        void        Init();

' K7 T" i/ q3 M+ p; E) _/ v
! C" o( [2 ^3 D% [        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
4 Y4 S6 u2 K+ R- L' H: d+ d/ W# j                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
) W: Y/ W9 B; c- a                        m_bAsyncFindRunning = false;
9 E; M0 ]; l$ x                }
/ ^. ?" C5 L! Y                MSG msg;
6 w! j0 ~2 ^# G' P                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
' s, {* S& i  B2 a) R; h" V7 E! }                {
- g  Y* m4 }6 Y4 c                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
# {1 t3 |& Z7 z0 o! c                return m_bAsyncFindRunning;
        }
; s/ q) t  c& V1 Y9 Q

# a2 @$ p7 y8 T+ r( A- \7 N        TRISTATE                        m_bUPnPDeviceConnected;

0 ?4 \/ g9 a0 ~
// Implementation
+ n- M8 C, d! f  C5 P1 [        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
2 y+ k+ J! f; Q0 y% c. V        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
3 K9 R4 \/ l1 T

        TGetBestInterface                m_pfGetBestInterface;
1 J) ^) q$ h" A( ]: E' X        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
5 F$ P) J. P8 p- k
$ L! `9 I# B# F% R- ~
        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
- G% X6 n" R0 c* X                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
/ V7 f3 ^3 m0 j) w3 V1 p/ H                result_type operator()(argument_type device) const
! \5 Z1 w' N$ ~$ E* o                {
' f, H, f4 V3 a3 U                        CComBSTR deviceName;
7 H7 O5 y' j+ v+ z                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

1 E3 M4 i0 U- V8 g0 O- {7 G, O
                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;


) \  u& _% v* Z' \* A4 `% X8 Q                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
        };
. _' I' B8 S) r. F. ~. \( f       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
4 w1 t5 g* Q3 O  R, G' D        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
/ b/ e6 |; S% w% Q2 T% N" ^        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
9 Z# Q" @! u6 D4 w' q+ m" {
9 a4 t4 a5 C7 I$ ~
        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
0 X8 k9 r$ v4 v9 U        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
; t( f$ r4 J9 l( A5 h3 z6 L! {7 p        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
  }( H% o5 _( @4 k3 j" i- B2 F; G        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
% r2 c; \( C/ v7 d' T        CString        GetLocalRoutableIP(ServicePointer pService);
* r# Z+ R0 A: O4 T) Y

// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
- C9 o' {, U4 P2 D6 v/ _        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
3 i/ \: {- Q: z        FinderPointer                        m_pDeviceFinder;
% W- I7 C: f* f% h& \        DeviceFinderCallback        m_pDeviceFinderCallback;
9 W; R  Q4 K: t2 F4 W        ServiceCallback                        m_pServiceCallback;
: y6 E. j. r5 i6 l8 y2 @# e7 Q
2 E6 R! i- Q  t
4 s, E- O2 x; H        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
, ~, R/ U; |6 M. c3 F0 \$ J* H8 d$ H        bool        m_bPortIsFree;
        CString m_sLocalIP;
8 |$ I+ R' t9 h4 ]  T( K        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
% Z& r' a- y. S% X+ l        bool        m_bInited;
9 o( l& G7 k* q' y3 I        bool        m_bAsyncFindRunning;
( C, @# V% k/ ~# L  B0 N/ y        HMODULE m_hADVAPI32_DLL;
" c9 x& G% u5 D0 r        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
2 y  G5 e5 l9 g2 u6 [        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
$ A5 J) u# ^! R3 B! T        bool        m_bDisableWANPPPSetup;

5 s* n; [5 d; Q  K: |. x8 E
: ^/ M$ d" @+ q, P};
& F8 G: _+ a9 o1 ?6 [! P& u
! F7 Y4 z& u+ a' j& x2 |& W
// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
; g- e' \: H4 i6 i0 J/ }  h{
0 p$ L* Q' H5 `7 d7 l, `) Dpublic:
+ Q( R" N) p7 G4 }+ H% l0 V        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
/ G* j' z; ?# K- M/ |0 j8 @# b3 n2 E9 ?

   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
. M" x& V* k- `, R3 g   STDMETHODIMP_(ULONG) AddRef();
- O' m% Q6 K6 u   STDMETHODIMP_(ULONG) Release();
/ s$ d- x( S! ~# P2 s" [' S* l
* \- u1 v. r8 I
5 y4 M; o: z* d& Q1 |// implementation
6 G! G7 W& G8 D# L/ u2 u1 Mprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
. o; N0 W9 z$ [1 i+ {* D' v

. x' s1 v0 ]1 D; u0 h* w4 C+ p) Cprivate:
: M) [. R$ ]9 a. X" J& n        CUPnPImplWinServ& m_instance;
' s, ~5 d. P* b' q% V8 o5 ~- Y; }        LONG m_lRefCount;
};
. Z, [" g* b0 S# J! r
3 D  r5 K( g7 K- h
+ n9 Q. F" {4 b% [* h2 I6 [. O: q$ N// Service Callback
* m. r1 i6 e" g0 Q  qclass CServiceCallback
        : public IUPnPServiceCallback
{
( J1 J" @$ x" }public:
: l7 {+ \3 I; R+ m' i        CServiceCallback(CUPnPImplWinServ& instance)
3 c2 `( D" S3 A' r                : m_instance( instance )
        { m_lRefCount = 0; }
   
4 p7 ~4 a! A4 f2 y   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
+ A( F2 b9 d! }( X; E. r   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


% O. k5 h2 V' b// implementation
private:
# C5 k1 t: i3 }        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

0 g8 n* X  U( W& I1 m: J1 \, E3 f
# Y: g2 S: M9 Wprivate:
6 m. t. ~. U1 h        CUPnPImplWinServ& m_instance;
! ~/ A* `8 U, O. g3 Q+ }1 ?, l        LONG m_lRefCount;
};


/////////////////////////////////////////////////
  f$ B7 Q) N2 C2 ^* k6 ]2 C. Z) t

, q' k3 E1 ]/ A0 s- h# a使用时只需要使用抽象类的接口。
/ I! o7 ~; b& c2 C, @CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
8 R8 A% A7 u8 ^; H+ j# SCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.