UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   6 ?% n) s! n) V- e
3. 
   
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   2 d" O+ b9 w- V% `( J
7. 
   
8. class   MyUPnP
   
9. {
   
10. public:
    ( X. a2 A# l$ O; q" G: y
11. typedef   enum{
    : [. ]0 K8 R% q: d# y$ `
12. UNAT_OK, //   Successfull
    6 C7 T  l3 g- y0 G5 Y- M! b
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
    9 D# q, j& K2 X( W/ y5 r
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    , V3 X; V9 o6 [) {8 `8 y
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    2 F( f5 @8 n9 J+ ?# j  n1 \2 L
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    : V8 E0 K& g' O7 l
22. }   UPNPNAT_PROTOCOL;
    * c4 X0 q1 Q& f* q  x8 f- U( ?9 `
23. 
    ) d( H" k+ Z- I: F
24. typedef   struct{
    4 S) n4 e% `8 ~% J
25. WORD   internalPort; //   Port   mapping   internal   port
    . T0 v0 b, ~5 [
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
    2 h" l) g1 G. _: f. V
29. }   UPNPNAT_MAPPING;
    
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    - S9 J% V7 |: ~& N' r9 g( X, `: E% m" T
37. 
    - R$ x: k4 d: k, @5 c7 N
38. CString GetLastError();
    2 r9 v. H% B' \% J6 A
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    2 P3 k' W# P7 ]- _7 r" u# s
41. bool IsLANIP(WORD   nIP);
    5 j! g) l7 Z- C5 X! G7 r2 q
42. 
    
43. protected:
    3 v3 ?0 |. z, ^  M, T2 Y' {
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    - B, S; U  ?! }  k8 W
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    & W! P* h% F3 O& I4 E0 x) x
48.       const   CString&   descri,   const   CString&   type);
    : q1 Z6 X) O1 a3 ]+ c
49. bool   deletePortmap(int   eport,   const   CString&   type);
    6 ~3 `- o% r4 z' Y
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    
54. bool GetDescription();
    2 t1 c  x4 S+ v& M
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    * `8 E/ A+ s9 j  [
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
      E8 W4 n0 I2 o, _. W; x
62. CString m_description;
    8 t) J* O5 I! z! {) u% g
63. CString m_baseurl;
    % N' }% J* {+ }4 P: ~
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    3 _' J9 a, F* z( l0 \) I- J# {7 q
71. 
    
72. CString m_slocalIP;
    : P6 g4 I& L# N2 B
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    
77. };
    - ?% Z+ ?  M( _! M
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   9 v1 u6 S" b  o' ?4 F
2. #include   "stdafx.h "
   
3. 
   , V$ f8 \5 ?1 i# f7 s
4. #include   "upnp.h "
   % K8 `% w! h4 z4 q* S8 I
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   4 a  \1 V, w/ I! x* M. r
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
     e" w! \9 ]% D9 |' D, x
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    ' m$ m( T; E' [1 u+ X% B/ f7 \! y
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    2 C7 d2 z) K! w* J6 J
15. 
    - h+ n, Q3 x9 Z  {8 R6 D6 ~+ A
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    + E+ {% e' f* t5 {) J
21. 
    & }, \8 g+ ~; W3 {" W/ x, t1 {
22. return   s;
    7 h* i/ J" T7 o* `! E( u
23. }
    % G" b$ |# W# Z  ]0 C0 t( k9 j
24. 
    " u1 I2 R% J- }7 h9 u; G) B
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    0 c5 R; P2 t8 w2 `9 z; G; b( s
26. {
    9 ]% U( Q$ c$ `" g+ I* a
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    . z4 M% @2 X+ J6 d
29. 
    . y1 n8 k# s: n& r$ I# u  w! U
30. const   CString   GetArgString(const   CString&   name,   int   value)
    ; b" b: u8 `/ J- i& P1 A) e( @
31. {
    $ A6 X$ Y1 Y6 k6 s* [  G9 N. g
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    4 F! X/ }( |. Y; z
33. }
    
34. 
    % W+ `. _% h" E* J% P
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    : ^9 D, P; M4 R2 N$ p
36. {
    9 I; o# O' P4 J% q* |
37. char   buffer[10240];
    $ {4 z2 N3 [" E4 \6 b' d) r
38. 
    " `  m2 x" y1 L6 r5 T0 o$ J
39. const   CStringA   sa(request);
    / O! Q6 w9 _" n8 j+ S; j
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    2 |1 z2 A0 K' X$ U" r# B3 S
42. 
    . c5 P  F: Y+ t$ \0 ]; K3 \  a' U
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    ! [! b$ S% A& `- }
46. sockaddr.sin_family   =   AF_INET;
    8 Y* |! J; V& H  a
47. sockaddr.sin_port   =   htons(port);
    " `1 z/ g' j0 {9 \9 J
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
      \8 e, J& y8 `! [+ c
51. ioctlsocket(s,   FIONBIO,   &lv);
    9 q8 s' D% k& D0 n
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    $ Y4 ?" t1 c$ P* l
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    ) ~5 @. I1 f, e: M# H+ P5 {
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    & i- L8 t( E  |( K6 E
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    3 `9 Q4 s" k% z: u' j. a
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    7 m+ |/ d9 r4 s8 |2 R9 G; S6 F
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    / T' n: V$ a9 i0 C  R
69. 
    
70. const   CStringA   sa(request);
    & M% r4 W/ N7 ~
71. int   length   =   sa.GetLength();
    $ r7 X6 P6 m" e0 d% m) N* w7 E
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    , x& o& ?1 a0 @% E0 w
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    0 R0 [# a  f) n- N4 i" a
76. sockaddr.sin_family   =   AF_INET;
    6 Y2 \7 i/ s$ M. n( M& w1 M
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    . e6 B, U# i. n$ }0 S/ Y
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    : l! i; V1 `6 g4 `- E7 Q& `
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    3 l- D2 b7 S8 c3 @8 _& l
84. {
    
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    ! U7 [6 \, Y6 {% J5 {
90. result.Delete(0,   pos);
    
91. 
    $ Y& B% u/ H" J5 i" H
92. pos   =   0;
    # j. c' K( h& o% R" N) \& x4 r2 t% _
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    ; T2 u3 K& b) Z$ e% ~0 `- p
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    ; |6 F' T0 m1 `. k( v4 b' Q
97. }
    0 h! ]/ B' S" t1 z/ X; j
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    # D7 c% t+ J6 I" B4 @  C& k
100. {
     - s5 j" Y+ A: n* D0 t& d
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     3 ?( v; }. j& z: q9 O6 t! i
104. 
     # Z$ g4 e0 j; v, \" V5 n  q  T
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     4 U: u0 S4 X0 i& Z3 G) V. K
107. 
     ' N  E( b# ]; n. h' ^/ N0 k7 L
108. int   posEnd   =   all.Find(endTag,   posStart);
     2 g- L( \0 c7 o/ p8 O0 q
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     8 P' f  }! H3 `* H+ a  u' c' L; Y3 i9 V
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
       Q8 e. ~1 F+ E: G! f  M! [
112. }
     : S6 F; a  a" @
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     , T, t+ T) j% k, m
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     ! h( B% {  v8 U: {5 _, y
120. 
     
121. MyUPnP::~MyUPnP()
     ( C5 h0 k) N+ S% R3 U% [( o% T
122. {
     
123. UPNPNAT_MAPPING   search;
     8 W& J! ]2 d& j7 K( h7 k6 T
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     7 w4 \/ T6 X( r4 e- J
125. while(pos){
     0 a( r" Y5 m9 ~
126. search   =   m_Mappings.GetNext(pos);
     ' V# I7 d/ r3 @* {% h( J% U
127. RemoveNATPortMapping(search,   false);
     6 p" @% ~/ }" I& q8 a2 |0 E
128. }
     4 f( D% t- I! t+ e* c" ]
129. 
     ' w! t2 v+ t& q& e& C( [$ H- G' K% o0 E
130. m_Mappings.RemoveAll();
     
131. }
     1 ?( o9 Q$ z. X5 N* A  e# X
132. 
     
133. 
     ) H2 E: l9 A; L  o6 J8 P
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     9 A* n' }6 D1 c: n
137. m_version   =   version;
     ; o  \+ u5 D$ g& {4 T' C! E
138. 
     ( P) y# X+ o* f: e  g& Q( L
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     # @/ o6 O4 [% B& i6 s2 {
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     ; x% X  p4 @' r% ?) u
144. };
     
145. 
     6 g8 p0 f: m. y' L( W5 |
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     ' [9 n' c  C: w9 A
147. u_long   lv   =   1;
     # W- B, o0 o9 S
148. ioctlsocket(s,   FIONBIO,   &lv);
     6 j9 k' @0 m( M, W, C# T% p% \
149. 
     : r* k+ ^1 L& K, B! I3 U
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     , z8 @) }/ d! J- P5 ~& S8 ]. x6 N. T- G
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);
     , o3 Z# m3 y6 L  b
155. CString   request;
     # m+ I* p6 L1 Q3 ?  }
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 "),
     " E0 p; z: \6 v* k
157. 6,   m_name);
     / ^8 y# [. L, C5 J4 P
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     5 k+ B# B; e! v( R& f
159. }
     : N4 N4 k" r5 E! r: T
160. }
     
161. 
     
162. Sleep(10);
     
163. 
     4 c$ ?  }# s* ^. h4 z5 X& ^" f4 Z
164. char   buffer[10240];
     5 M) @; _7 `- m% L+ K. j3 o
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     , `; Q0 v% m- X, W' o/ n
166. if   (rlen   <=   0)   continue;
     8 ^6 p$ {/ u' w4 @7 C" J- Z2 M1 A
167. closesocket(s);
     2 n8 U! O3 L+ h+ f' x
168. 
     ! X- t( {% |5 O( W# L5 |
169. CString   response   =   CString(CStringA(buffer,   rlen));
     ' t$ F0 D! e8 V+ a4 E5 b9 i& x: g7 Z0 A/ d
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     1 {) u! ^/ Z0 {& Y
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     + C4 g! ?. ]: X& L& Q
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     / k. D& U6 f) `3 R' b7 X  J" W
176. for   (int   pos   =   0;;)   {
     ) C9 O+ ?' D6 `2 w. k
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     9 u6 O' ~* {) b$ D+ E2 z: y$ D$ \
181. if   (name   ==   _T( "LOCATION: "))   {
     % L  Y) K( c! t9 s% U; P
182. line.Delete(0,   9);
     
183. m_description   =   line;
     9 K# P; l- K, e/ \' Z3 ~* n
184. m_description.Trim();
     ( d4 ]' p+ m' Q6 c
185. return   GetDescription();
     
186. }
     
187. }
     
188. }
     
189. }
     : h6 \& f0 N+ p2 _
190. }
     
191. closesocket(s);
     
192. 
     + t1 d1 P/ T4 W
193. return   false;
     2 b; f( @( ^0 g" @
194. }
     # _0 g$ X6 h, X* i5 T8 l

复制代码

天云上人

以下有关upnp的接口来自emule，
+ g% _. s' h! E
: q! E7 I' r, Q. [) g& g
; `' C, ?' a3 a$ ^* |///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

) A# m0 y( [+ v4 |& c0 X
7 B5 d1 ?) D" e  t: T. ]' g, J6 e! s#pragma once
#include <exception>


  enum TRISTATE{
        TRIS_FALSE,
4 N0 t: v( W' T. |8 E        TRIS_UNKNOWN,
7 c5 m: n/ Y: [% A" d        TRIS_TRUE
};
7 l) c$ [* w0 E6 o  N9 D  v2 r6 v
) s# d: d7 b: P6 c! B  j7 v
1 B+ w' y4 D( e2 j" b' renum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
% ?) X" V* a  ?2 }* l        UPNP_IMPL_MINIUPNPLIB,
. I% p! X9 l* K1 r5 u        UPNP_IMPL_NONE /*last*/
; @5 k; A9 x8 C  e$ m};
9 |  P1 i* }& S4 f. |


4 u' T7 q9 D7 `5 v- h0 K( y% d
3 d% H+ ]: r  K( @; {$ i3 T0 x" ^class CUPnPImpl
{
, I( C+ y. f4 dpublic:
; ^, G$ y8 A1 w: d4 |1 k2 s        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
7 K8 s, S! J4 A8 a        enum {
                UPNP_OK,
2 |7 W; B( Q, Y                UPNP_FAILED,
                UPNP_TIMEOUT
7 h  k  R2 g& c" P7 }2 S        };

+ h& j( |) {  i
' w9 f" @0 Z) _        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
; q( q! R( C% b' ~! L( H        virtual void        DeletePorts() = 0;
/ q( g' U0 X" J! P- U" [- H        virtual bool        IsReady() = 0;
1 V8 `3 r! N) n' }9 M1 X        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
* W5 K8 C+ ^/ R3 ]: i1 V: _* X8 g) H

$ A: T5 m4 p9 V. \% c# Z        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
. B+ f, v3 v! @6 h( Y1 W        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


9 M0 e& N4 }: b7 `8 L// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
+ A# c) d, S) d0 M, `! P* u# W        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
9 f  D9 q* b8 G4 i( W& r

; M* O1 S; j' A: kprivate:
/ Z7 X! W) t4 S+ ?1 A        HWND        m_hResultMessageWindow;
+ r) i; l4 Q* x4 E" Y. d" i        UINT        m_nResultMessageID;

4 n. G0 f9 J- R9 L. i% @
2 A' _; s# ^& G};
! i. W7 F+ w5 G! ~/ l) F1 B
# {' w6 ?( x+ z0 h( U7 Q
1 a  n+ \6 q6 Y9 s2 G3 q$ E// Dummy Implementation to be used when no other implementation is available
7 z0 d7 N2 B( v6 Xclass CUPnPImplNone: public CUPnPImpl
8 e1 S1 J* |/ I$ l5 B2 k{
, _+ e: K0 f3 F) C, vpublic:
: I4 I: ]* c# B        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
1 Q& ]4 F8 U9 ^* ?/ W6 ?+ u        virtual bool        CheckAndRefresh()                                                                                { return false; }
$ g/ t3 P) z* B! U8 O  Z        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
0 h0 \& l) m6 {) m- z" r/ k+ H% _        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
; a6 G9 W: R$ S% }9 c$ d- b4 [};

8 T7 G/ P) F2 x, Z
1 {/ l& h0 ?; {* W6 t$ u1 _* E# K/////////////////////////////////////
; z4 M8 o. q3 J( u/ i//下面是使用windows操作系统自带的UPNP功能的子类

# Q. d3 r6 c9 f; t; X
#pragma once
2 e- A' G# H$ {2 N$ X- ~#pragma warning( disable: 4355 )
4 t- ^) W5 l1 h  w, L: i# Z7 [: c
7 R, V0 t* E$ e( d# F
#include "UPnPImpl.h"
#include <upnp.h>
. S1 B, s  d# Y* l# h) C- P' B#include <iphlpapi.h>
#include <comdef.h>
' b: i8 R( t6 H( S0 F, G! H8 y#include <winsvc.h>
, b- Q, u3 m6 Q3 s/ V8 Q5 v- ]; d7 u
, q0 g: i! y! \
#include <vector>
6 o4 o  N- p1 E0 P; Z) K/ ?# Y#include <exception>
#include <functional>

8 b: V6 N8 O! G7 K" p! @


7 @$ G1 X* y1 w8 J/ Mtypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _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;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
! u0 y$ e/ L0 ]* Vtypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
- I: A4 M' ?0 h4 `9 qtypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
# y; B* e' k1 B

typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  H7 N! d  p- O1 v" }  PDWORD pdwBestIfIndex
& T" `  D- ]/ r2 u; q/ G; r);


typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
- b4 @# Y" ]4 `3 f" `# Q. G  PULONG pdwSize,
  BOOL bOrder
2 C. p! V# ^; g);


typedef DWORD (WINAPI* TGetIfEntry) (
9 k+ J* [/ ?1 x" U$ o" i  PMIB_IFROW pIfRow
" f5 [( D4 A, g  Q, S/ t$ x);
5 w/ w9 K0 k0 M0 `8 f  Y# y2 i

CString translateUPnPResult(HRESULT hr);
* m& V* @- p" qHRESULT UPnPMessage(HRESULT hr);
& n3 s8 F) a8 U8 h& E$ [

5 V- z1 z0 A2 C3 ?; S9 ]1 rclass CUPnPImplWinServ: public CUPnPImpl
{
, ^$ G0 a3 H* t7 ^% y- l        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
) {! Z7 f) K1 G; O% ~// Construction
public:
( Z% a; p! ~0 f* {3 x7 _        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
: j2 Q( m6 {. ~
) V+ @& U1 H4 r$ N5 A4 @) X
% Y$ I7 S3 e6 u2 G8 }) x# M        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
+ K7 F* e9 a! M4 r( T' X7 r        virtual bool        IsReady();
+ j1 |; z0 V# p9 J% I9 W$ j        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)
$ ]' A! K) |) o1 |        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
" k- Z: \/ t9 F% }
0 Q+ A" Z, Y* `  e+ v4 U- o4 w
protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
8 @: h- Q' U$ Y1 ]& Q( e' a        bool        OnSearchComplete();
0 ?3 a3 ]; Z$ @: c$ J! _: Q5 h        void        Init();
8 J" s  ?, Y- @( T0 W; z* k/ n

5 L9 v+ a( Q' [: F0 N        inline bool IsAsyncFindRunning()
2 J' e, I2 @3 p7 Q0 E; f4 @        {
; K  }8 G6 P7 |                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
+ P5 b3 O& _; H. a                {
: \( |9 _" `% ^2 N! Q                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
' g  \5 n- S6 u2 l0 u                }
. g4 [: I  W! K# L" h& d& S                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
. O: M# T4 S: @$ p9 q/ L3 s                        TranslateMessage( &msg );
  o# `0 ~7 O8 v( ]9 h- z                        DispatchMessage( &msg );
/ l9 p0 A* i7 Z8 L                }
                return m_bAsyncFindRunning;
  W* V6 p0 y# C1 Q8 s  z( [        }


* n: B' g+ `5 \$ M        TRISTATE                        m_bUPnPDeviceConnected;

" _# e' e$ k$ h
7 y* n% _* }# H) l/ f, h& E// Implementation
        // 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);
) M/ K: D) Q) O7 Q  W+ h        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
- y* N8 e* Z0 E9 T/ ?        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


7 q, j) b4 i$ b        TGetBestInterface                m_pfGetBestInterface;
+ L; Q; k% U  W        TGetIpAddrTable                        m_pfGetIpAddrTable;
  l  a9 A0 X( |# S( y; o+ W        TGetIfEntry                                m_pfGetIfEntry;

$ }6 e& e: |* ]7 S& {& J/ N
5 }4 z( K* f2 @; J) ]$ Q- j- d        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
2 p2 C% K3 r3 P6 f& U/ g                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
, [" k9 C/ v! f                {
                        CComBSTR deviceName;
3 L. U6 g* ~. h7 _/ f5 p                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;

" G) \5 X  |4 E" z
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
! J" a6 p9 U/ M, Y        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
+ Q; H2 k$ h4 G6 B        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
2 U. r9 s3 a- ~        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
5 ~) H3 @7 @: r$ u                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();

. P* j& ]# i7 A2 Z7 r
; [5 V* Q% v/ ^" c" |  z        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
4 a5 n1 n+ C0 B4 O; H        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
+ G( Z; L: |! k7 x9 R# w8 _        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
0 N6 N: Z- k" d2 k9 Z4 h        CString        GetLocalRoutableIP(ServicePointer pService);


  l2 d7 T0 ~6 a0 e% d. {5 I: F4 @// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
. V5 _. Y; S& p7 V3 ?; @        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
+ I3 p& Z8 ^+ B/ R' b        ServiceCallback                        m_pServiceCallback;
6 P: i3 g$ @# u6 ~% |) Q
6 D" r" r( ^) @& k% l0 u. m( y8 k
  X4 k2 j8 z% o& t/ A  k        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
: n: [: ^9 h1 o& {        CString m_sLocalIP;
# v: I% B# b6 _. i2 c        CString m_sExternalIP;
) b$ G! e$ {& I, Z- A% t, T        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
/ a! |4 k" Q' _$ r        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
6 @4 d$ |* V7 A# r        bool        m_bServiceStartedByEmule;
. z+ d& z( @$ J0 s. U        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;


- v, E7 K, t9 b! _. r};

& @" O8 Q- G" _  x9 w  W2 d
/ F. l' `1 V( \8 J5 Z// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
. z* v, c& M  Q/ W8 s{
! `) S3 W( M  u% dpublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
, W: d0 g0 w; I6 z& u; e                : m_instance( instance )
        { m_lRefCount = 0; }


   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
2 r" q" v7 W: @   STDMETHODIMP_(ULONG) Release();
0 M3 h- x- K8 E* \) u/ I
8 A% o0 H$ X/ r; r; Z
- D0 l; e# u# p) I* A* f// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);


private:
        CUPnPImplWinServ& m_instance;
* I! v* M& k, Y6 i/ i        LONG m_lRefCount;
};
4 A4 {7 z" i" c( A

// Service Callback
3 d7 e% N3 i3 m' q7 W- yclass CServiceCallback
        : public IUPnPServiceCallback
{
" ^: u  e7 w7 {( h  k* ]public:
, P! G# J6 J; |, R" `/ L3 Z( `        CServiceCallback(CUPnPImplWinServ& instance)
2 ^% `# l/ j0 P                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
4 O! r* F) w- \
* q0 f3 p( m% k; Y8 h
' W5 X, g; J2 z$ X) V; S// implementation
private:
5 F2 m2 g; O" P1 P& S        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


  D/ P% e0 ?: U( Bprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
/ X* U- [4 m3 D  K};
2 q; {7 R1 @1 M; ?1 r( I
0 q4 t' ~. Y) q1 u
/////////////////////////////////////////////////

( T' J* ]8 m, O. ^/ o
使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
! k4 D  I: F3 V9 y2 Y$ |0 yCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
. A9 H- v1 @$ DCUPnPImpl::StopAsyncFind停止设备查找.
( K: d! B4 V) O% U$ ~9 k" o- bCUPnPImpl::DeletePorts删除端口映射.