UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   / B3 X- b7 r* \, W
4. #pragma   once
   6 I' ~4 t$ p$ P
5. 
   8 ?! W# r2 C! `4 a* L& ^
6. typedef   unsigned   long   ulong;
   0 ?  o7 Z8 ]  h) R8 d! l
7. 
   
8. class   MyUPnP
   - Z1 o; S9 U; ^3 C" s) ]
9. {
   6 ?: Q3 a+ p  ?) T8 r8 w
10. public:
    $ ]0 s: T! d" \) {% B& C  {
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    7 ]0 n9 e3 V9 n* `! s
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    3 z& ]  \: U- [
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
    6 H8 ~5 \, K% B! d' [' D1 z
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    # I; w* s! Y5 t
17. }   UPNPNAT_RETURN;
    # a6 |9 W( K, I& O# m9 ?
18. 
    . [, |; R" w! H# A* x6 f( w
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    + \/ a9 b1 s4 v: L/ u/ q
22. }   UPNPNAT_PROTOCOL;
    * u) P. n, {! s( Z
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    - m/ j4 Z" B, m8 b2 P) e6 X
26. WORD   externalPort; //   Port   mapping   external   port
    $ x8 {3 i$ N4 e9 I
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    / T$ f, S& n' z0 l, [
29. }   UPNPNAT_MAPPING;
    # R$ e6 i% J2 [+ |& _: _
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    / _8 v" a  K& }) R# x0 E! p  B
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    & d( m" m. ^2 e- m
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    ; p( ^. f& h2 G, d1 H' w# \
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    $ Y0 Y" N$ ~$ Z& m! O4 Y! V
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    ! m% N1 J! R* K$ H4 c6 y
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    : e+ K4 Q  D9 y
46. 
      `& w8 H0 t) N8 z3 O% w5 ?; E
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    1 I7 l7 u* j* [1 y  j( I: M+ O
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    - y0 ~( |' H6 ?2 s  L
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    2 u3 j) b5 d& c0 }% y
52. 
    1 I$ t* x& D: M! o
53. bool Search(int   version=1);
    
54. bool GetDescription();
    + G/ w+ K8 X  V3 @
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    ) a! D0 Q0 ~3 O( E4 L, B' s
57. 
    ; z$ A; _* a' X0 ~* X5 {
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    ) M2 Y# }6 R% a* X
59. bool InternalSearch(int   version);
      i0 V) A& K- g0 m. g; k6 a: B
60. CString m_devicename;
    & F# Y4 ~+ Q' E5 c
61. CString m_name;
    ' Q( e6 H& c5 r
62. CString m_description;
    
63. CString m_baseurl;
    / p8 x: z$ o; f% r
64. CString m_controlurl;
    
65. CString m_friendlyname;
    : L2 D  q7 x' D. V
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    8 ^2 ]6 `9 k9 S0 c
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    $ l' h$ ^) V4 I0 Q& r0 [
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    + Q; Q9 d* W( x( @
74. WORD m_uLocalIP;
    % y( S' u, b; E2 C* K6 Z$ M
75. 
    ( I7 u& ~0 p( Y! Y5 @% G2 E( t
76. bool isSearched;
      ]- H7 U( ^" ^" H0 h1 k
77. };
    6 W( j. ^* b) u- z! r
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   1 V* N' \3 C* O1 v, Q+ m6 z9 N
2. #include   "stdafx.h "
   
3. 
   , q7 A- K% b+ F" P$ D; b
4. #include   "upnp.h "
   
5. 
   1 M/ b( V! {. W8 I: M; ^8 ?) G
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   3 U# H9 _; p3 x; r4 u! I* j0 h* u1 u
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    : Q! m0 J; O7 c2 Y' M3 f
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    2 `: @8 w/ h1 ?! g- ~$ Y
15. 
    $ F1 M0 T- O/ {  W
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    8 F3 s( l0 S3 W" h2 P1 R
19. 
    
20. s.Format(_T( "%d "),   i);
    ; I- l* v( H: j% y) G& b9 B
21. 
    
22. return   s;
    
23. }
    
24. 
    + i% k" |+ h7 y" R9 H
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    0 [3 `* `  x' R5 b' o! i& e
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    , K* o' b& m5 p8 K3 z
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    ! C- }' r  r) v; R1 N1 t
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    ; s3 y  P+ N% o' i1 m+ k6 `
33. }
      l* R- m' [* A" P; S! }$ Y" A: |
34. 
    & i2 y! F+ T5 P$ E9 ^* B4 Q
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    ) D% s0 X5 ~8 |# g. G! e
36. {
    
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    5 ], f4 ~" I0 b5 `  C
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    0 p! e9 _' k7 c: ?
42. 
    5 m/ X4 ^  \' E" J+ Q) g( \
43. uint32   ip   =   inet_addr(CStringA(addr));
    4 @( v+ K) H. U. B" l
44. struct   sockaddr_in   sockaddr;
    , o/ ~0 E8 {  l3 Q5 _+ H# T$ G
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    & @' k9 n. x$ M/ _$ T' d
46. sockaddr.sin_family   =   AF_INET;
    % R/ }4 g  r1 j1 C
47. sockaddr.sin_port   =   htons(port);
    ' j) N: o) p5 Z. z) ^0 c: D
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    8 ~( s3 D& W9 K( B+ z
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
      ]0 C0 V5 q' e
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    * z$ x* Z+ `/ z
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    , B- \- `: K# r, b
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
      n/ W' S( U0 R( x- `4 H& q! u
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    / H& g3 U) {# h
60. 
    ' V# L0 Z9 U0 d( `1 x' P4 [
61. response   =   CString(CStringA(buffer,   rlen));
    . X/ L, g: i. |/ C4 r3 @, ?
62. 
    
63. return   true;
    
64. }
    
65. 
    9 X) G1 ~. K2 E) z! F$ K( l; Z
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
      ]/ L% \, q" F; p8 a( u
67. {
    : r2 L  w, G" _6 R; F
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    * d( x0 j4 y' u* [
71. int   length   =   sa.GetLength();
    ! O- g! z" w! G
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    ( V! ^0 I: v3 i
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    2 D4 f* b" p4 `9 `+ c* G8 |: L  U
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    - z4 F" ^! `! u3 b" W3 |0 S
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    9 t8 N8 }4 E) o; z6 U4 H
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    1 l4 J1 c" x* v
84. {
    
85. int   pos   =   0;
    ( |' R- I# p* s9 o6 B
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    $ @( M6 A8 U3 y6 e1 `4 b1 S0 u9 E7 c
89. result   =   response;
    * W4 C! [2 U8 ^3 @" w) u$ x& J
90. result.Delete(0,   pos);
    ) j) F4 I* r8 a5 K
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    8 V9 G$ Z: N8 l: W) T* c
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    
98. 
    ! [+ s- R% }6 k9 I% [( N) p
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     9 n( d" o- Y" ~. V- T! Y
101. CString   startTag   =   ' < '   +   name   +   '> ';
     ) X  S9 o7 E1 y3 U* c
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     * [$ R3 [1 B- ~
103. CString   property;
     # w9 M; I0 \. A4 d% N5 P. S
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     - k3 D2 I. P1 A& i0 }+ S3 _
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     : F" Y7 i% C6 i% \) l
109. if   (posStart> =posEnd)   return   CString();
     8 T6 R1 N  {; {: y8 y5 x8 |
110. 
     1 a/ f3 Q" q4 Q6 O! k
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     % q' _$ U& K. w) S: d4 v$ `
113. 
     * `+ A# h3 C" c2 I& g9 H. E
114. MyUPnP::MyUPnP()
     % h8 h1 w5 y- v  ^
115. :   m_version(1)
     & E0 w' g% r  e7 |
116. {
     % d& r$ m1 t  w  D9 ^
117. m_uLocalIP   =   0;
     " u  Z, r8 v) T0 b# k% {/ _
118. isSearched   =   false;
     3 c% J7 ~+ b+ E+ M- S7 g
119. }
     
120. 
     ! {. a& N, e# ?3 ^# `& A; t
121. MyUPnP::~MyUPnP()
     4 O% g6 {" m. W
122. {
     0 K: P. w) h+ C! Y0 z- @
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     " T1 h/ h: u4 h) a8 c' I
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     7 Y1 Z* x0 [& d9 y& I
127. RemoveNATPortMapping(search,   false);
       j$ ^0 j4 h& J. |+ K" M
128. }
     4 P, I& z- `, M5 I/ I, V$ M
129. 
     + j& G" Y& o4 s6 u% ^, P9 X9 p
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     
133. 
     5 A8 M2 q" _( |: Q  z' r9 ?" O6 _
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     / d5 k$ R; }3 m
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     . T' |1 u5 N1 U/ [) I( n2 f6 b+ N
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     ; o9 ?5 p/ }' k* R' a7 i
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     7 F" `% w) i. I9 S
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     ! N" H- L& \' q9 U
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     / m; P- E% p7 k6 V
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 "),
     
157. 6,   m_name);
     . Y' a" Z% I2 z& V& ?" e
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     ( H' R2 `8 @, i' @, F4 F! d$ W, F' s
159. }
     ( D- n2 Z" Z, b, _: l5 h. R
160. }
     4 N5 t8 m; d( b( m, u
161. 
     
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     / C1 h0 t) _6 {7 F
171. if   (!parseHTTPResponse(response,   result))   return   false;
     5 y3 e% P: ]8 Y" O5 L
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     ) K4 C3 A9 Y4 u; P& g5 l; g
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     + f1 l+ I9 j' c  G. F
175. if   (result.Find(m_name)   > =   0)   {
     - s/ j& d+ }! G2 z" s  m" |' s  i
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     7 l0 z" ^( i/ s1 }- \
178. if   (line.IsEmpty())   return   false;
     # O& S  Z) H5 U. P
179. CString   name   =   line.Mid(0,   9);
       R3 X$ A) l. \* y% }2 |9 Y
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     ; m( {, u! j, s% N
182. line.Delete(0,   9);
     , a6 d0 H4 h4 q5 b( k! r
183. m_description   =   line;
     
184. m_description.Trim();
     % p4 {  K% f3 m8 R5 |
185. return   GetDescription();
     . ~- z+ C. v: m' D1 i* S
186. }
     9 v$ i( ~/ f/ B7 ]/ e3 N7 l
187. }
     & E) }4 [" u: P/ _* m* {
188. }
     
189. }
     ) W- U9 K" G  F2 {2 @  w7 [: S  a
190. }
     
191. closesocket(s);
       z- u, O  P4 Q' b. q% O' ?+ T, A
192. 
     
193. return   false;
     $ c& p# I/ Y" b* A6 M
194. }
     

复制代码

天云上人

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

+ ~9 o4 O5 I; K" I
///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


7 N' {$ z5 D) h#pragma once
0 w- B1 a( S) H: c8 Z2 A  \$ K& U#include <exception>
7 T/ g- @. `# c( U' R8 U
0 K* Y! h; {$ Y' ]
8 \! D8 t2 t: G% P  enum TRISTATE{
  d# j( d" l8 L7 A        TRIS_FALSE,
5 X+ _& e% ]% t# E, A        TRIS_UNKNOWN,
        TRIS_TRUE
5 d7 K. |8 k8 t# e! @' z2 C};

: t5 T7 q. b4 J0 ^
enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
& X! j+ R7 s0 f9 E. ~        UPNP_IMPL_MINIUPNPLIB,
% U+ w( g" ~3 M        UPNP_IMPL_NONE /*last*/
};

' d& ~& b5 J0 Q+ z, x# S- u$ c4 v
9 j9 y; T- f" W: t

' c) \+ y0 c4 i6 @8 w4 L. oclass CUPnPImpl
{
3 y/ c; k3 a* D/ i5 ~3 bpublic:
2 ]$ _9 g, F* N6 W( F        CUPnPImpl();
7 b1 a& L7 A% l- B# ?' B        virtual ~CUPnPImpl();
1 }9 r9 ~# Z5 d7 _+ v        struct UPnPError : std::exception {};
5 E  W: X; i5 ~, w9 ]5 M        enum {
                UPNP_OK,
; z3 Q6 I$ m* {" d7 B                UPNP_FAILED,
                UPNP_TIMEOUT
        };

  k( S- a& f2 t1 b
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
8 \& \- Y3 a% v  E7 d. b6 J# n- S        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
# i5 M" y/ b! \( B9 y1 s        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

0 n) C$ z6 b3 ^9 f
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
: i7 X2 E& W1 ^7 @        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
$ _0 E9 p+ F5 ?1 J1 S/ z. k        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


; B! E- d& e5 I% F0 |0 y// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
- f5 a: g$ e. q, R" O& F" y2 b) j        void                                SendResultMessage();
% q( S! p+ {7 `        uint16                                m_nUDPPort;
0 [2 s% ^  \# d/ ^' T        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
; w% B' j4 E9 R8 Q( i" T5 R3 |( X- `
" k! _: ~* D# a9 B2 Y
9 J, M, F5 O; `% ~, u$ \& Jprivate:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
2 z- o8 \0 a+ _) N7 h

$ z5 L) R! x; J" n};
; c* e5 T+ d5 Z. X; p+ Q& V3 J
# e6 ~# N1 [& `; y) p( K
3 _, A! N+ P5 E2 i% j// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
1 y' d* a! }1 R# w- D9 J6 lpublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
4 x' N+ S$ q# w8 \, G        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};


/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类

$ w/ R" t* d1 ^& _
+ z/ t4 V4 c( G) V! ?& K# \#pragma once
, P0 H$ P% ?) X3 p2 ^, t$ |#pragma warning( disable: 4355 )


$ y# b; \2 v4 R#include "UPnPImpl.h"
! }0 \9 L* z. W3 a% u3 v#include <upnp.h>
#include <iphlpapi.h>
9 ?* @6 d+ Q& R#include <comdef.h>
#include <winsvc.h>


* Z1 h) y) T8 k2 J/ i! ?! g#include <vector>
#include <exception>
+ E5 |) }, d0 C, ?4 q#include <functional>

* J4 r$ b- Q$ h/ C! q) L


5 b0 P8 @1 {( \/ e7 }) z8 G! Htypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
/ s& s+ O- n- k( s7 Y' Ytypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
2 |  R0 V7 @3 C# Gtypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
- \1 l% p% @$ p8 W* Htypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
: q' T" b  E' g, ^typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
2 t! m- M% n4 ?$ y

/ E, R: G$ i7 K5 E! Ctypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);

, l& v$ I* j! v6 j4 J
typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
# [! r, F1 W  Q. L8 _  PULONG pdwSize,
# O6 E; ~- H; C1 {4 `& M/ t9 {  BOOL bOrder
);
/ v/ a+ I+ m$ q5 K5 {$ ?
, Z5 b# _- T& F
2 f/ l- N! r& X1 _typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);

1 a5 x/ R# J2 n# ?$ T( i
CString translateUPnPResult(HRESULT hr);
# k  J2 z4 }1 x: |/ RHRESULT UPnPMessage(HRESULT hr);
5 {, v$ @# m+ J7 N1 ^3 ?+ r
7 |+ W& A8 \, q" |; K5 T# m2 K; m
class CUPnPImplWinServ: public CUPnPImpl
4 t7 \+ V0 h5 P7 e% [  y4 q{
! }  t) Z' J; D( V4 S        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
public:
        virtual ~CUPnPImplWinServ();
6 B+ }8 [9 _& I        CUPnPImplWinServ();


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
9 d0 l! r% p! N) ?/ G        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
  F- n/ y* S2 A4 A8 M        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

, m% E( N  Q' ?, h: W
$ L7 F! Y0 K3 X: U; l4 k        // 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( z9 V2 ]! Z$ d1 n5 {        virtual bool        CheckAndRefresh()                                                                                { return false; };
2 T! a5 F0 a% A+ U- B0 a/ V, r" q
! D/ f# g+ A' {8 P& O+ F
5 E, H' u( V! V9 S1 P9 Bprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
" F0 P! w6 m0 U: o* R' k0 Q2 z        void        RemoveDevice(CComBSTR bsUDN);
2 ^3 y0 {* r9 B& |% [. w        bool        OnSearchComplete();
        void        Init();
" |9 J" ^+ u* S

        inline bool IsAsyncFindRunning()
3 G3 o. w+ i" k: h/ ^        {
# e; L8 L" C$ {                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
9 F9 K& q0 l- ?+ c# r                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
! n: g% O5 k1 ~0 M- B/ h: c/ A( g                }
                MSG msg;
1 H. a7 u6 _3 c" f, J5 M( a' P                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
, O+ c3 K0 n1 E) i, y4 O  w/ J2 j                }
                return m_bAsyncFindRunning;
        }

  ]! f: Q1 G8 A7 L( L
* z7 C1 {6 y: ?8 e        TRISTATE                        m_bUPnPDeviceConnected;
5 U! M4 U: Q# [: y" S" ^
, ^2 ?7 Y! o3 ?+ [" I% o2 u
// Implementation
: ?1 k0 y, ~5 q% k        // API functions
& @! w$ I, r" C9 j* W! D* I. D        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*);
# [3 L4 K% y5 @* G( |        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
  ^+ Z) x( @( ?: T2 ~+ B
* h5 O4 l) c( ^9 Z. ^; g
        static FinderPointer CreateFinderInstance();
4 u" V  Q% e, I7 a" e2 h3 |/ @        struct FindDevice : std::unary_function< DevicePointer, bool >
9 f% S1 c7 M$ r5 y- C$ ?+ T' o* U        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
! m. }5 C3 v7 t; G6 R$ q                {
. D3 x' |& O, Y4 [                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
0 p* @" K: }* X1 w0 a

                        if ( FAILED( hr ) )
4 G$ R8 n5 B# O; \- p* @9 Q  G" @                                return UPnPMessage( hr ), false;

# K7 ~& C" ^$ R8 W" G; z9 B
2 O" ^9 i, z% A, b; S! t                        return wcscmp( deviceName.m_str, m_udn ) == 0;
& D( f$ e8 v7 ^1 V, x. l; Z2 v                }
                CComBSTR m_udn;
! r; O$ S  d8 ]& @3 F4 \( z        };
4 |8 ]* V8 q/ Q       
7 w" _- r! O/ t) H        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
: c# m' y  x  x% g1 ]! y) V        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
4 o4 O7 J5 Q# H+ n        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
- H/ i$ k8 q7 `        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();

6 g$ \9 E3 o  D8 X
        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
" D- [' c. @" i7 n2 J& V        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
$ S- m/ t3 s" Z# H        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
9 |1 y7 c$ v. ^# f        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
+ d- g  @; H3 ^4 f6 [        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
! I2 c4 @+ t- j) l6 ?        CString        GetLocalRoutableIP(ServicePointer pService);
% A- @: Q5 b; x& W4 E3 E

// Private members
) [( Z9 P' M4 f. jprivate:
6 V. J0 A( W  \6 A0 R: E        DWORD        m_tLastEvent;        // When the last event was received?
) S, x; D" h; o        std::vector< DevicePointer >  m_pDevices;
8 d3 S' e' c4 w5 v* A5 _1 j( s        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
/ X( `5 A5 x+ w        ServiceCallback                        m_pServiceCallback;


( t* {; A& ?- y5 F7 j( {4 X1 A        LONG        m_nAsyncFindHandle;
# e! a, t3 X5 h. X7 \& g* ?0 S        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
' f0 F8 P: {1 D8 B6 v        CString m_sExternalIP;
: F4 r. k8 K  ?        bool        m_bADSL;                // Is the device ADSL?
  @) C, O  @0 n5 n, {        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
9 m- p+ O9 h# B! C6 [; m6 P; c2 r. O        bool        m_bInited;
; j# `3 M9 s2 o5 V4 j        bool        m_bAsyncFindRunning;
1 z4 n1 X. c% U        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
1 N* ~' t. }2 ]

};

/ I8 s& u' {4 e; ?& B
* d) d6 S7 m4 z; r% ]/ v0 z// DeviceFinder Callback
0 G; [7 A0 f" k: p8 iclass CDeviceFinderCallback
" c8 L5 G+ O1 u: Y' `4 k( M6 [        : public IUPnPDeviceFinderCallback
' K/ ]2 W: e$ K  J. ?{
public:
4 b$ M) g4 Y0 b# y2 b        CDeviceFinderCallback(CUPnPImplWinServ& instance)
6 C3 _! P' C2 Q                : m_instance( instance )
        { m_lRefCount = 0; }

, h, [3 [( r  `4 L
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
6 X. a- ?- Z, V$ Y, p- \   STDMETHODIMP_(ULONG) AddRef();
* E7 M2 Q9 w$ J   STDMETHODIMP_(ULONG) Release();


// implementation
, k' Z+ x9 O8 V0 r* q2 ]private:
+ n' N, F  \6 w3 m        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
) ~1 s& Q# B9 @) ]: G7 O7 h" S
: h; t$ q7 ]/ C7 }
private:
        CUPnPImplWinServ& m_instance;
$ z5 v  z9 {9 ?. C7 z  }        LONG m_lRefCount;
2 Z6 o- |: K, G6 e7 t, K};
6 c+ R# X- L+ v
+ i' H# q% j4 F, G% I0 k
// Service Callback
class CServiceCallback
% d  |" s1 l: P# z4 u6 F! s$ l) o: }        : public IUPnPServiceCallback
" y( d, ?0 g& K" e9 @+ X  ?4 z{
public:
& L" r1 H1 r7 o  B- J        CServiceCallback(CUPnPImplWinServ& instance)
  @( f" g& t. w, L9 B                : m_instance( instance )
        { m_lRefCount = 0; }
) F- l% \  s  T8 p0 h4 }% I   
% w7 s4 {  V/ `$ p1 e- K5 Y   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
9 N5 X) L( F8 o2 j5 k/ _   STDMETHODIMP_(ULONG) AddRef();
0 S8 L3 U- b6 R   STDMETHODIMP_(ULONG) Release();


/ j& ^& J4 V2 t9 `0 z- B// implementation
8 g/ [' g6 X5 |3 K% D7 Y- l' ^( r' z- Bprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
6 H7 c) K1 x7 [4 B        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


private:
. o2 x5 }4 W: ?* s( d        CUPnPImplWinServ& m_instance;
8 X* v/ j" L$ a! E6 S& C9 ]        LONG m_lRefCount;
1 V2 `* G' g. V# U6 Q% B3 E};


/////////////////////////////////////////////////

  w0 `0 A0 {2 L
" A0 ~2 F: B4 J使用时只需要使用抽象类的接口。
& B* r: {3 u' L3 ZCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
0 X6 Q  ^7 Y; `  ACUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.