UPnP

天云上人

/*uPnP.h*/

1. 
   " \4 W2 R8 D/ ]/ I' z6 ?& f
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   ' C0 |. ~0 X, Q
5. 
   + r- R  r* e, B3 P# g; f
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   
9. {
   - k, ~' |( e( h' R" K; j
10. public:
    3 l, J# {7 {8 K+ ?' L1 ^% }
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    6 Q% M2 \0 p5 ?/ H
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
    * ^$ q3 Y# I1 W- J) p) V
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    # l. V7 |& Q, \. B8 D  s! G
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    , C% E% q% K$ M. R
17. }   UPNPNAT_RETURN;
    / K* I# e6 H; y+ J: i
18. 
    2 P# ~( j# U+ x* A0 [
19. typedef   enum{
      A% V$ c5 L# K5 T' o# M/ w
20. UNAT_TCP, //   TCP   Protocol
    2 U* i, j; [/ [
21. UNAT_UDP //   UDP   Protocol
    1 @* |+ Y" \0 ?8 r  n0 u1 }0 t
22. }   UPNPNAT_PROTOCOL;
    , [6 p( H( H. y/ U0 Z
23. 
    0 F+ L! R( D! `* S$ ^
24. typedef   struct{
    5 |9 Z3 y% d# g0 A- O: Z
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    0 N/ {2 u7 E. n
29. }   UPNPNAT_MAPPING;
    . U9 |! w4 }, d- m% }& j4 o
30. 
    7 O6 ?1 c7 i7 ~& I, _& }+ L
31. MyUPnP();
    ! m3 l+ s* y$ o' e: X  d2 |
32. ~MyUPnP();
    2 {! E0 |5 z9 i/ G1 F7 B# G
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
      n$ f' u/ x; i2 R" E0 {" f  E, Z
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    " i! X! U$ T) I9 v( s
36. void   clearNATPortMapping();
    
37. 
    - @5 ~+ |1 q" I: C! j
38. CString GetLastError();
    / N. t9 r1 B  e" r- U' @
39. CString GetLocalIPStr();
    3 J% z( l: E: J! E# o/ D
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    4 b$ N" |3 f" ?$ }6 r
46. 
    - f2 U4 @: b: y2 I
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
      f% b6 j0 z& o: e+ x$ E
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
      Z$ X& V  q) g3 ~; o
52. 
    
53. bool Search(int   version=1);
    / S, }: G6 p# l9 a9 [# g/ D
54. bool GetDescription();
    # v2 ?* S8 Z6 |# T5 i6 w
55. CString GetProperty(const   CString&   name,   CString&   response);
    + ~' |  J. T7 c7 g( u, {
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
      ~- P2 U5 }( `: \, E( A
60. CString m_devicename;
    
61. CString m_name;
    * G# |; P5 g6 |4 {
62. CString m_description;
    1 O/ L, o6 c- o+ Z, `
63. CString m_baseurl;
    
64. CString m_controlurl;
    2 h! P/ J- P3 }! M' M" b
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    1 v) i8 p, x4 W1 e/ @" ]9 N' I
68. 
    0 Y# S* i/ s: }
69. private:
    . J! i7 @  C  d  B9 L
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    + q! m2 |; `! T
76. bool isSearched;
    
77. };
    5 g% y& Z3 L( }. ]/ L+ w! U
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   ! n, B+ A. D0 e* i- \' x
2. #include   "stdafx.h "
   
3. 
   - e/ k1 ^  q$ j. R  i2 l2 J5 i  E
4. #include   "upnp.h "
   % F+ m0 C/ A) s. {' b- K) @% D; `
5. 
   ' w# v9 [. O" M2 I
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   $ l% t& N% y4 w2 e2 O+ I8 I! Q
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   / U8 [2 w& \- D: }# R9 R
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   7 W, m/ f$ N! b1 ]  N
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   - |8 R' A( ^# {+ e* N
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    7 r1 B1 j$ \7 `; V1 t. G0 w2 Y
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    5 |! L, `; N: V' i& h, E6 [
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    ; \7 a9 r1 M/ I5 }9 L7 ]
15. 
    3 [; }* Y* |6 M1 G% x% S/ O
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    
19. 
    ( R- a$ q6 h5 G+ m$ b" J& `6 j
20. s.Format(_T( "%d "),   i);
    ) q/ n7 A) Y, y" ]$ \1 K
21. 
    
22. return   s;
    
23. }
    1 \+ Z5 w# `- H: v9 T- R
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    ! _1 X' [! s7 |6 w2 ]% \
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    ( u/ l& M& h; o( {+ r
28. }
    3 Q% n0 E5 J) q( L" d: x
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    2 m1 e1 L- C& _- l; P5 u
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    7 l: Q( t* V8 N, Y' @1 b
33. }
    1 }. a  P* l& U$ k) [: s5 ]+ ~9 z
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    ) k, }9 v  P$ j& Q- \
37. char   buffer[10240];
    
38. 
    # z: i) B) P" K4 ?6 @3 M; R
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    $ T0 q) W+ L0 D) T
44. struct   sockaddr_in   sockaddr;
    3 @$ |  c$ j" u1 z- Q* G
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    : f: [) x+ K: N3 U: L2 l. d8 s; {
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    7 L4 ]6 r% \/ v5 b; A
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    % I4 H0 I( J9 Q9 p, ^
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    7 W. |$ N* u$ A3 B: _1 [# l( X
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    $ `$ m5 d. ?% ^/ B1 Q, d7 m
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    1 Y! D% n# c9 p
57. closesocket(s);
    " \- G6 l7 Q- m2 e  E% x# p' @
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    3 G% ~- [+ I- s: [
59. if   (!rlen)   return   false;
    7 p# X) ^/ O/ K& d
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    $ N9 X/ P. n! x3 \$ {& G
62. 
    
63. return   true;
    
64. }
    
65. 
    # R+ V  X2 C$ y; ?' y( w- V: Z
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    * t8 u2 m8 W+ ~9 s  Q
67. {
    * j% j6 n- d" H* a- e! \( }- `) W
68. char   buffer[10240];
    & l0 r2 C) G4 H. p9 E; C2 s$ W
69. 
    8 }# T5 R9 @# a6 T" @
70. const   CStringA   sa(request);
    7 W/ d. w. n' V, z/ F9 A* V5 H
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    % e1 N# o5 b& ~! F2 V8 `" `
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    $ N1 V8 R& j5 u3 Q
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    & {) @8 p: B- }3 T0 h$ F1 J( T: N
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    3 L9 \" x1 i& x( @& T3 i
81. }
    6 M) k7 y) w  G& \# B1 Y7 C) a
82. 
      k! z+ m5 p& g# j0 o
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    4 `! U9 m7 ~. ^4 k, U
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
      u, h) x; @. u, j% y3 ~$ i$ p0 v4 t
88. 
    
89. result   =   response;
    0 _3 L1 A, T+ J: V
90. result.Delete(0,   pos);
    - W- _* i! p! R  H% K, k* A
91. 
    , i1 s* n; s8 A+ ]* |
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    $ d  d' H2 i# y( M# ~
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    
98. 
    ! e* G- Z* u& |
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    % y$ k$ k! h' c0 R* W2 F+ e
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     , }/ n$ K" F/ I" y# I
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     : d* D' U: h. {
103. CString   property;
     3 e3 [, d6 ?: g( J% i; A
104. 
     4 @# p2 [) U. D6 d6 H( W
105. int   posStart   =   all.Find(startTag);
     ; [/ L9 m6 \8 t
106. if   (posStart <0)   return   CString();
     # `5 B# P* y4 a1 f0 A* o
107. 
     & p) G4 g" V! z. `" b
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     6 O2 L; z0 Y6 G$ F+ h/ v; M
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     7 ]0 V; R8 M, \# B
112. }
     
113. 
     6 i, `5 Q2 i% O- g% d# \
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
       g6 z& l6 U" n3 E% @; Q0 |4 n
116. {
     
117. m_uLocalIP   =   0;
       k- [% u1 j6 R+ x8 A+ g& `
118. isSearched   =   false;
     ; O( A1 W& {4 d" b& N* ?
119. }
     ' D, Y8 T0 u' k' t5 B1 x; C/ ]
120. 
     
121. MyUPnP::~MyUPnP()
     1 N; [; M6 P/ k$ i6 p3 T7 ^
122. {
     0 f5 l) @, h: a/ t9 W; h% ], x
123. UPNPNAT_MAPPING   search;
     % n- Y8 z+ @/ i+ Y& X6 A
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     ! \( Y9 ]+ g5 k4 ?: {( U) v7 p
126. search   =   m_Mappings.GetNext(pos);
       g( a2 F" {& G
127. RemoveNATPortMapping(search,   false);
     ! i4 N. T) K4 N) {
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     . L) m" ~: L5 O- _, B! W
131. }
     
132. 
     
133. 
     " c; I) k! n1 }% D/ s& [9 X4 P
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     * E9 H' A( p7 n
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     2 ~0 H* n0 d2 i+ u8 |" d$ D4 }$ z* `
138. 
     & {3 k; T* c# Y0 L5 N& b# h
139. #define   NUMBEROFDEVICES 2
     ' t5 M4 g& _4 ]7 b1 z
140. CString   devices[][2]   =   {
     8 V& }/ o7 z5 e# X) f. ]
141. {UPNPPORTMAP1,   _T( "service ")},
     . Q' k) N- w3 \
142. {UPNPPORTMAP0,   _T( "service ")},
     . f0 q9 F+ q+ [% M# F* o3 O4 N
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     : h2 {: w  m0 ~0 e! {* ], ^$ Z
152. if   (!(i%100))   {
     - A# U8 A. `  H- v+ F, H
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);
       E% P3 n4 s: k; s. b& Q
155. CString   request;
     
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 "),
     7 C  ]4 ?/ i( v5 X6 U
157. 6,   m_name);
     ; s' b. z# X+ H, z9 A
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     " ~5 B8 |8 \2 T: I) y) Q$ X# J
160. }
     2 [$ a! s! w+ b5 X; ?, O
161. 
     
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     4 T7 R3 @! j) {2 F
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     ' \; x8 R) q% [' A
167. closesocket(s);
       T0 |& h. x5 ]) t5 M" i3 o! J
168. 
     * [! ]7 h9 V' b* Y* _3 r1 Y
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     9 e' `# i: w* Z; C: `
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     + K" V5 v; Q; N. k- U& S5 C
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     - g9 D( ?! H) B0 T$ p1 J: Q
175. if   (result.Find(m_name)   > =   0)   {
       ~, d0 E3 S1 }) ?, p
176. for   (int   pos   =   0;;)   {
     * g9 H0 b3 L, c0 r$ L( H* @7 u3 @9 j# J
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     , P( ^& ]" t' |, l: ~: _) ~7 H
178. if   (line.IsEmpty())   return   false;
     ' W% Q. s2 u8 l" k2 _. v$ c- k
179. CString   name   =   line.Mid(0,   9);
     ' b  Z* x7 H3 N" t3 C  W6 w
180. name.MakeUpper();
     ! u/ S- k7 W% z5 r& c6 t
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     $ U0 Y& }9 C7 ?& n- ~/ P: W/ j7 Z
185. return   GetDescription();
     & P8 F" w7 h1 |5 l2 B: `- ?1 M
186. }
     
187. }
     ! Z0 S7 x  {) d) n% U$ Y
188. }
     . G4 s9 J4 q% a7 l+ h
189. }
     
190. }
     - k  ?" G0 Y9 d# N+ @" {, _
191. closesocket(s);
     $ A* J9 f+ ?- t
192. 
     
193. return   false;
     
194. }
     

复制代码

天云上人

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


///////////////////////////////////////////
0 T4 T9 l& N9 C5 u7 F) C" h0 A: V//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

4 S3 `; J* ~: o2 V7 K
#pragma once
. v9 A, q; L, R! i#include <exception>

  L0 ]7 Y3 w+ z
" O: ~8 N( f/ Z0 I; [; x  enum TRISTATE{
& {: N# L, R, I- ]& ?        TRIS_FALSE,
1 e( p7 x! u3 l        TRIS_UNKNOWN,
7 ]! s) h+ b. E, Q        TRIS_TRUE
};

/ {: E& @) w- y' S# j1 U& x
. Z: U5 T/ H4 {; J) m9 \4 _9 eenum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
# ^& @" ^0 @. E7 Z! K        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};
1 O1 y$ {8 w$ q7 Y6 F



! b5 v0 F# q/ r" w5 tclass CUPnPImpl
! F, g: ~( N: K' h{
1 ~( O  x$ m8 W; S" [) ^+ N% l  qpublic:
        CUPnPImpl();
5 Z& W/ Z" W$ V+ g0 X9 n: P$ }$ o        virtual ~CUPnPImpl();
' a" H! |' K+ ]$ {" Z        struct UPnPError : std::exception {};
6 |1 \: N0 j) g% @! U9 {% s$ I. T        enum {
& ?" D4 k/ m0 {2 F" j                UPNP_OK,
' ?7 s- ]4 X" c* @1 I                UPNP_FAILED,
                UPNP_TIMEOUT
        };
) t" D& `8 F  q5 v
& c9 G2 h% y4 b9 M6 w* d
' Y+ q+ |6 J' G! G        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
& d( j# i1 u$ E        virtual bool        CheckAndRefresh() = 0;
" t& u7 W0 }% h        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
( C3 ~3 b- ^' N( R0 f9 J3 ^, ^        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
* X  e7 L) M* ?1 K" n  Q1 r        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
. E3 z% t8 p, k7 F5 w! s        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

4 R! Z$ X2 `* ~" v
// Implementation
protected:
$ A5 R$ F% M) M( u2 N# }* Q        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
7 m$ ~' V2 h8 W

% x6 O( V/ ?( v, g% p  A: l. fprivate:
" B' u+ S8 d0 H9 Z        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
* ^) a9 w. k' S& t. H) E$ r4 w

: Z, [- |% {3 ^7 B- ~};
3 x& B4 g" w8 Z  _: r
- u8 @, T/ @1 n$ k$ E  t  Y: F
4 w5 S2 I: @" C8 E* S// Dummy Implementation to be used when no other implementation is available
" w; ^8 I( B$ F) I( Pclass CUPnPImplNone: public CUPnPImpl
{
* \# f' s4 J9 Lpublic:
- ?9 {( T. R$ U  b0 }( W& t, z        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
( W( {3 E; @# L( P- g! F0 ]        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
0 u- E) o' x; p6 k/ T- K        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
8 R1 |3 T2 C2 b* Y$ t$ v  v};
$ G: u, L) I; U. H2 Z2 S4 [
& W- D3 a3 y/ s6 S  f
/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类


#pragma once
$ N( Q) o2 x8 ~# Q- O#pragma warning( disable: 4355 )

1 s) N0 ~' K- s& q0 J
#include "UPnPImpl.h"
8 R# h  g4 E- E# e#include <upnp.h>
% R" P5 G" S3 M7 e' v$ f3 I+ c#include <iphlpapi.h>
6 y2 D) \) V( ?: x; E#include <comdef.h>
#include <winsvc.h>
  ?( N, T: |# B3 R/ x
) \; R, T2 A6 Y& `
+ Q8 P  k0 b6 b: m+ U- R#include <vector>
7 w, @6 P4 K) R/ ^1 \#include <exception>
#include <functional>

$ n: {  u6 L& c, ^/ ^# F

6 _* B9 c/ y1 w
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
0 B1 t& `# b! Dtypedef _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;
8 @/ e% J: S, r$ H) Q9 e7 B2 itypedef        _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) (
% ^  _; i( I2 N  K& A$ b3 ~  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);

$ D1 `& ]; ~  E' J2 l7 q
typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
4 y4 `1 T( O5 p. b7 s' P  PULONG pdwSize,
  BOOL bOrder
5 d' Q7 c9 c* s' R; D);


typedef DWORD (WINAPI* TGetIfEntry) (
9 ~$ k' x5 J1 C/ c) a; Q0 I5 w  r$ R3 m  PMIB_IFROW pIfRow
/ r: H& H' A% c8 Y% E);
0 [' f9 |. [1 U* W3 N2 ]( {
6 ^* I; q9 I7 l! D5 F
CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
2 {0 |% f! j8 x* l1 h/ Z
. K3 K: T5 P5 H: B7 W. l
class CUPnPImplWinServ: public CUPnPImpl
5 Q4 ?1 I0 T* {8 K{
        friend class CDeviceFinderCallback;
" n2 C1 Z4 ^, U+ R        friend class CServiceCallback;
// Construction
public:
. l( b6 {! \- F& z        virtual ~CUPnPImplWinServ();
1 a! p+ K+ ]8 U7 f; J- P        CUPnPImplWinServ();
( q7 s) K  M% ^3 d

- I6 k7 V. k- H9 }. A9 q; c        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
! |/ q6 C6 {8 i        virtual bool        IsReady();
1 R3 q( a/ g- ~) M2 v9 J1 `; `        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

, H# p  b% y! V* i' I
+ M+ }2 O& c9 R        // 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; };


' b6 `8 p* M: Yprotected:
) _; n. e# j! S) \6 g; W9 j        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
6 w& h, C9 G% I0 l; r! d# S5 H        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
  N% P- M$ q& i# c  z/ K+ u! [        void        Init();

8 ]/ b- H0 z' i2 P  _
7 s6 [9 z- T$ {$ V  T/ T( x# J. B        inline bool IsAsyncFindRunning()
+ U$ c" V: X, N) s; |        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
1 ?: k  ?3 h6 M6 \# n/ |                }
; S! n0 t- J. ?$ q                MSG msg;
0 Y) }. a- R4 |# D                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
# x: z% o1 M) F3 l0 c" _                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
1 b% P5 ?" K& ~8 m  |* f                }
                return m_bAsyncFindRunning;
        }
* }) `. y9 }) F; R& M; C
1 {' ^; T, g. _8 r8 [
$ q% ~$ [  m( R        TRISTATE                        m_bUPnPDeviceConnected;
: T8 e2 a0 v3 e% x4 @: E
4 y2 t/ T- r/ r- G4 U- e
- C& T& R) Y- V% e! S, M# e- b& g// Implementation
  R* Z6 `  Y* a+ l) t        // API functions
( |7 a8 ~1 x# Q& l        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*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
) M9 G( C) y+ `" j( T% x: J! u

        TGetBestInterface                m_pfGetBestInterface;
3 Z" `) j. z& ~        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
: Z) c: C/ q0 y9 e& @* R$ g- ^

        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
# D0 W- G' n  C! M0 I" O" p                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
# S6 M, Y" y& {
9 c& z& j( |5 o! |' l9 d9 r
                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;

6 J, M: k5 @2 f: ?' K
" s- Z0 ?7 _* O" D0 v                        return wcscmp( deviceName.m_str, m_udn ) == 0;
, g) _. \7 c$ U+ w                }
                CComBSTR m_udn;
6 G3 f4 p, ?; m& X+ B! L2 `        };
: M* v# i) U1 l3 I: I+ g$ U       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
+ h% x$ K1 z2 t4 V7 B        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
1 n2 ]! n0 D7 v9 Q        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();


        // Utility functions
( X' a( a% ^' _. J, d        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
- {$ D9 ~+ [2 l. K" \1 y) Y        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
9 f8 W! r/ s) x/ h$ ], C7 t        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
/ b7 B* @( ^; ^2 p3 m3 S/ R        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
4 }) m: ^3 O# h/ |; u        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
0 G  }$ u- ?& S8 v; ^" O$ r

  y' j: c3 B) B# D// Private members
: t& G  p  R$ M! `) wprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
4 K( |: M3 [% c) |1 j        std::vector< DevicePointer >  m_pDevices;
+ |$ }" E! f. i* D4 ^        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
" ?; ]1 R2 z# h4 s, t6 n) J        ServiceCallback                        m_pServiceCallback;

: r. a- K, t( a
, y( x' {* ~! J9 q1 A        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
( b& ]* f) I2 x' X% a+ O3 t* s        CString m_sLocalIP;
4 r" f. D" h5 }6 x4 e+ v        CString m_sExternalIP;
( K7 p; _- c5 c+ B! J        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
" x) V9 V1 B6 C6 }        bool        m_bInited;
. t1 G6 f; b0 r  R. q. F        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
) D# S1 G% e7 A, E# K0 @        HMODULE        m_hIPHLPAPI_DLL;
8 _1 Y4 i& T$ |        bool        m_bSecondTry;
6 Q/ C6 t& H6 A# S        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
! i! L( K4 ~5 n; q* d6 F

};
6 m/ r: h! L+ q3 x; R

// DeviceFinder Callback
class CDeviceFinderCallback
* [6 W7 B; _' x* @        : public IUPnPDeviceFinderCallback
{
5 {3 U' q! V  f: r2 r. q1 ^public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
# m6 D4 G. n# }                : m_instance( instance )
        { m_lRefCount = 0; }
; l9 j1 q( _- _! I# v: f4 {* v

   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
# T9 {) a# n5 i/ p2 S   STDMETHODIMP_(ULONG) AddRef();
5 `) H* j7 P- q$ g  ~9 \7 c7 i   STDMETHODIMP_(ULONG) Release();
$ F; |- S; v% j" [) s

// implementation
private:
1 q% Q  v1 H3 o) }8 b5 d        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
1 C: H+ B; {5 a- W5 d, X        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
/ _2 h: y( _  e9 G1 b  Y        HRESULT __stdcall SearchComplete(LONG nFindData);
$ s9 ?5 I. P3 I
1 G5 g  v( u. ~- x! I) p
/ a+ S# I0 s! g. r) d4 E' mprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
. w5 g; X; ?) W. R0 {
3 O- D0 r- [5 i4 g" ^
// Service Callback
, U' U8 R* h0 H5 fclass CServiceCallback
: l7 q/ G! L8 U8 h        : public IUPnPServiceCallback
{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
- C  B2 D0 f; I                : m_instance( instance )
& |% r1 A+ {+ ~        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
9 K4 n4 t& }/ w0 D# }1 ^1 X1 q   STDMETHODIMP_(ULONG) Release();


// implementation
private:
6 H: W( m- T% N1 V        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

3 R" h, P: v) t; N
private:
        CUPnPImplWinServ& m_instance;
5 |1 g# ]" M3 C2 {        LONG m_lRefCount;
! ?- C- Q, c! H! B( x+ [# P};


/////////////////////////////////////////////////
1 i  S# m- w& H" G$ E* p3 p
. D" D$ \- O, t* m
使用时只需要使用抽象类的接口。
3 s! z% r0 @8 F( PCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
0 W" T* M6 R) J- U5 P" a! u" ?) T2 [CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.