UPnP

天云上人

/*uPnP.h*/

1. 
     r2 D% c( L! K/ ?4 ^( b5 J6 k8 @# s
2. #ifndef   MYUPNP_H_
   0 s. ^7 w5 B0 {2 U: N% ~; s- G* o
3. 
   5 C& @& W: z& o& g7 g
4. #pragma   once
   4 U* ~7 W( O0 z) i( {" f
5. 
   & S. D, }& B% N$ l7 ^
6. typedef   unsigned   long   ulong;
   " z* e" T! w9 n7 p5 m. z
7. 
   
8. class   MyUPnP
   
9. {
   
10. public:
    , Y3 z5 b" ]' v% T5 I/ l* L4 G$ b$ U
11. typedef   enum{
    . N2 i0 J5 T/ e: N- B9 u' G
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    3 K  v5 A3 {) q$ a$ L" v
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    / n/ f: I, a. j) D! ?/ x. h/ G. _
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    9 L+ L* O3 Q, V: Z$ x* z0 u0 M
17. }   UPNPNAT_RETURN;
    + g  `' x: e* [
18. 
    ' E/ j/ M# K2 k' }
19. typedef   enum{
    # t1 |1 P* _! m* s5 D0 r" U
20. UNAT_TCP, //   TCP   Protocol
    - s3 |! ?2 H* k, R) |6 n/ p7 w
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    0 B' X( m! l; k% c+ j2 U9 T# R* A
24. typedef   struct{
    ; G# @( M  T! ^6 Q% B
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)
    8 B/ w. Y. [: A: L( i) V
28. CString   description; //   Port   mapping   description
    1 ]- s" y: o2 `+ w9 v6 L# Q
29. }   UPNPNAT_MAPPING;
    
30. 
    9 g: ~! G" c# e0 d; ?1 N4 |" W
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
      \8 \$ {& h# c) }! B
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    5 w5 S$ _8 J$ ~8 s
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    ( o" Z  s$ v. w" w  k
37. 
    
38. CString GetLastError();
    * j1 W2 P* Z  H4 b0 N- J+ |
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    7 m/ O5 s9 y1 I) n) O* I
41. bool IsLANIP(WORD   nIP);
    6 m- V3 b0 }- \% U. ?  u; Q
42. 
    & o9 _8 `; E" A2 |, ~2 u: Q
43. protected:
    * o9 j  a$ b0 {; {3 D& Q" P$ c- w
44. void InitLocalIP();
    2 `* J  Y0 g% M. |# F
45. void SetLastError(CString   error);
    
46. 
    . l7 r. T0 f, g  i* j& \1 ]
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    & [8 J1 m- E9 {, l7 d- F! e
50. 
    . ~. w1 u% U9 G" y! W$ T+ C  q
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    9 y6 H, D9 b9 ^* N- z3 ^
52. 
    , _; l/ T3 p6 v! _/ ~( M
53. bool Search(int   version=1);
    
54. bool GetDescription();
    5 m( S! b! g  R! ^- ]% u0 z
55. CString GetProperty(const   CString&   name,   CString&   response);
    2 H9 E; x7 \$ \& G6 g
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    6 _% u) q5 C9 S9 ~( t& j) r
57. 
    - T, S3 T! x; _
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    4 Y0 Y$ H3 m% P$ P: C: M5 O
59. bool InternalSearch(int   version);
    # y1 Y1 ~$ a5 w7 ~( N) {. o
60. CString m_devicename;
    
61. CString m_name;
    . s  K# R. s3 v
62. CString m_description;
    
63. CString m_baseurl;
    : [$ {- l# G/ S* O
64. CString m_controlurl;
    # v! C6 o- C+ X7 k" r+ R
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    3 n" Y  J/ @3 I9 i
68. 
    
69. private:
    5 @; ]5 Y: S, ]9 h; M, V) I+ W3 E
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    + p7 L7 ^$ e& W! p8 X2 ?* F; |
74. WORD m_uLocalIP;
    
75. 
    - R' d# R* u; _0 U# ~6 ?$ r
76. bool isSearched;
    
77. };
    ( k& J3 T' h5 ~6 A* S$ d* j& c% ^: X
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   ) e8 x% N0 x7 m, y$ K9 Q, d$ F
3. 
   ! [9 R; O$ Z) S3 Q$ G
4. #include   "upnp.h "
   
5. 
   8 m9 Z: p2 U, m
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   8 {8 |8 P) N/ f$ P
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   : N/ r" x$ _& g$ `
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   4 X! n: A; I9 D0 ]& `! }' s. u
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    6 q4 N3 {# ]( @6 P
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    ( j/ ]: B, V7 E2 A9 F+ [3 B! G' N
15. 
    
16. const   CString   getString(int   i)
    9 R" |* ]% c" A9 A* ?# I! k9 ]
17. {
    0 Q3 K  ^! b; R& d* k) J
18. CString   s;
    8 |  t& g$ s% E+ @& e. z+ |
19. 
    % F- l) Y, z( V" r6 N5 C
20. s.Format(_T( "%d "),   i);
    
21. 
    : S% t" g1 H: T2 I) P
22. return   s;
    1 _% B  o: S9 @. z, w
23. }
    8 n7 v5 g6 @9 j* _! `' N- ]
24. 
    + H! X$ c; W( v, q( U+ w+ Y* \4 r
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    * U( \7 f- T; s5 B
26. {
    / v8 N2 o! S! ?( v" T7 g
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    / O. N( d* x. f4 t
29. 
    3 J+ h9 E% O/ r* B8 E0 [8 c' s; k
30. const   CString   GetArgString(const   CString&   name,   int   value)
    0 C' X. {3 l" ^9 o  c6 }1 w* h# X+ F
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    6 e, o; t; ]. n" h5 F& f, p0 j
33. }
    " q; c# x" {, h+ L$ s5 o6 U
34. 
      k2 e% i7 ]- Y/ h/ P8 s
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    % K. W3 E! a4 Q; d& B7 g
37. char   buffer[10240];
    ; N, r; S! ~, N# s( o- }
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    0 l8 N: O; e7 Y) G. u; `- I  b
42. 
    " U3 M. k2 T6 P' ?% ?$ ^
43. uint32   ip   =   inet_addr(CStringA(addr));
    0 R/ b3 ]" P' b# k: I! I
44. struct   sockaddr_in   sockaddr;
    - Y! q. G6 l, e1 I% H& g
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    0 u# }! w  p8 r  J
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    ' B- o( r7 D/ H( x% A1 }( t
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    " h4 M/ t! Y. G4 i8 M- c: T$ T
53. Sleep(20);
    7 g" E* ^& x  Z3 C9 w" X4 H7 w
54. int   n   =   send(s,   buffer,   length,   0);
    ' A$ r* U# Z& g* {0 v; B
55. Sleep(100);
    ) Z( E; o" z8 B; C$ k" i- L
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    - v, C5 ^7 p8 O8 Z
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    6 q1 M$ \8 V" }
60. 
    ( c# c+ X! A: b; j# C
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    : D% d9 A- Q6 Z- ?& G
63. return   true;
    & l% f9 f$ b6 C% d+ [
64. }
    
65. 
    ; S1 u1 {/ X& H2 Q5 r
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    $ g4 v/ z4 M3 ^: G. D
67. {
    ) i' ^. ^6 j: \% \" n1 M
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    8 \7 [. L- X: E/ e1 j
71. int   length   =   sa.GetLength();
    * T2 {9 ]0 P* v* x7 h2 W- p
72. strcpy(buffer,   (const   char*)sa);
    " z. i5 u6 F& k# T: s
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    : p& D2 ^/ e( T: u
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    8 R9 @* c" ?. T: ]
79. 
    9 Z- E! f6 _; r4 @3 F( f: A
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    9 t% L, Y% ]) X0 J6 t
85. int   pos   =   0;
    , c+ v$ A- y; o  N
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    ( P1 U3 [: A* [; H( k
89. result   =   response;
    
90. result.Delete(0,   pos);
    4 S  Y2 e2 b) I: x
91. 
    0 I' a, ^' j, `3 z
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    ) d: L5 {& u( v. w6 |  B
94. status   =   status.Tokenize(_T( "   "),   pos);
    4 D: _, n7 v' j0 j8 K( n3 @3 T
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    & O  i* _+ v; N3 w. T' W( {% g
97. }
    ) k! A2 M( U$ \9 U. w* i
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    6 H2 \; D( A' o/ e4 m' e
100. {
     ) x7 G9 l3 Y8 |
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     / a4 L: g) J& B, @2 j/ Z& I' N
106. if   (posStart <0)   return   CString();
     
107. 
     8 i9 s9 L- b! A, r
108. int   posEnd   =   all.Find(endTag,   posStart);
     / ]# U1 ^0 j% K: j7 S
109. if   (posStart> =posEnd)   return   CString();
     7 s) f6 j. k- i8 r0 n
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     8 `' L8 P/ `: c8 x; E
112. }
     & Q8 j; `, G( }3 o: T, b! [5 N
113. 
     7 w  n3 U1 q* l# v: ]3 c
114. MyUPnP::MyUPnP()
     - x3 b6 s; K% ^5 a6 R! ^0 i
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     6 I3 A8 o* q: Y5 q4 N
120. 
     5 `8 Q, X' T: A, @7 Z) {5 `  Z
121. MyUPnP::~MyUPnP()
     
122. {
     / B" A/ _6 ^$ e3 G( M( R. P
123. UPNPNAT_MAPPING   search;
     ( L8 w  T' b3 }/ n  s4 s
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     3 C' B2 s; p) R2 s- a
125. while(pos){
       ]% Q3 |! E# ~3 Z
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     / C) t& b7 A1 s/ u) p
128. }
     ( o% ~1 f) ~/ \9 {( t
129. 
     % k! m# J# P( N1 C3 \2 v5 c
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     5 a' v2 ?( w2 g( l7 h3 F* C
135. {
     ! I/ `0 q1 f4 ^( K% |1 ^( |
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     & R5 H$ K$ g& x8 ?
138. 
     $ s% H; P4 v% s/ ]5 v% g
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     , E" ]8 b: x$ _
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     . i- G1 i8 s; K: \) l
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     ( W* y6 R9 {4 m9 D9 e% X8 E
149. 
     
150. int   rlen   =   0;
     : z$ a9 r0 f. A$ M
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     8 a! e. A( R2 I' p) u
152. if   (!(i%100))   {
     5 \: m6 T$ ]& t+ q
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     , ^. z; e" g% |9 p) n( O
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     ; M. [0 I# r  X1 v# B  w8 I
155. CString   request;
     : U, p1 u7 ^% \( S6 k: X% _
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);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     ; p( S7 c$ H7 L$ t' ]) ^; G
161. 
     ; M  F% ]( ^, H4 H5 l
162. Sleep(10);
     % Z% H5 G4 [- Q& e( j- o
163. 
     , a4 E( l& V8 r: _/ `4 V
164. char   buffer[10240];
     3 ~8 s& s: E1 a. E( }8 N
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     4 A, G! f- D+ v- r4 [
166. if   (rlen   <=   0)   continue;
     ( y) J; @0 e' j% V0 L
167. closesocket(s);
     , Q7 d" ?4 j& |$ A
168. 
     ) J# i4 Q5 m4 J( m0 P. b/ L/ [+ ]+ n
169. CString   response   =   CString(CStringA(buffer,   rlen));
     7 W6 E& @  C' P1 J. u1 c
170. CString   result;
       J& Y8 Q1 n( b) w; j
171. if   (!parseHTTPResponse(response,   result))   return   false;
     , [/ W+ t( |6 R
172. 
     & G/ [: g9 r% L. H
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     " j, k, Q" q* a6 M' m
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     ! S. e9 r8 G( |; W
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     $ ?4 T- B$ X' ?  r% w4 H  I# k
178. if   (line.IsEmpty())   return   false;
     7 B$ W: f2 b* ?9 o
179. CString   name   =   line.Mid(0,   9);
     1 s/ o) a  e+ V" }
180. name.MakeUpper();
     * K8 p8 j) W6 Z8 P  R
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     / F1 `: ?/ ]* ?) f2 c) D
183. m_description   =   line;
     
184. m_description.Trim();
     5 g0 |; i4 \2 V$ h
185. return   GetDescription();
     
186. }
     , B2 Y- K$ T2 L. N
187. }
     % C+ I1 Z5 \, X+ n, y% |
188. }
     
189. }
     
190. }
     / K0 s3 X1 e' t9 g1 e+ C+ H
191. closesocket(s);
     ' R- O+ S3 l4 _( V; M- [# Y6 F
192. 
     " ?4 t" I' j) S* e) V
193. return   false;
     
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
, `' B0 g( }! N

///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
9 `: O/ W# L/ }( Z4 X3 Z
. m3 z/ h  U6 K) f. ^! m! D, D
#pragma once
#include <exception>
! D- M! r( I; j% c  v* w5 e  O9 L  L) N

7 B; ^7 M+ B# m  enum TRISTATE{
        TRIS_FALSE,
9 B/ D* y) M5 ~9 U0 b" v' d        TRIS_UNKNOWN,
        TRIS_TRUE
};


# C4 J4 h6 Y* ~4 O% A* T% aenum UPNP_IMPLEMENTATION{
7 c" y* _; q' p3 K0 N( \' ~        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};


: _: r* v) w  W$ X

: y4 e) k& W# Z7 Wclass CUPnPImpl
{
public:
        CUPnPImpl();
        virtual ~CUPnPImpl();
8 v& ~$ f) o5 t5 d4 T        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
# h! v" q% B" n. n                UPNP_FAILED,
6 L! N2 `& K0 F1 ]8 q                UPNP_TIMEOUT
& m8 v2 f. u. G! \        };
- t3 h$ I# f! V8 K

2 L0 O% I/ }9 R/ X' v* ^        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
* b% P0 ?/ _( \: J, M" N        virtual int                GetImplementationID() = 0;
- M% l' d' o3 B3 c$ y" T2 N$ T       
+ `; J( m: t7 l0 w; p$ W        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
1 ?* |; I+ Q9 u- k3 B# I! G
& \- x5 {6 n9 ]* U) r* b  b
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
9 ]2 O1 I  u' c0 j        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
1 C# V' D1 J# z/ _+ U5 x7 h4 C- y        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
+ l' V3 R0 S+ J3 q0 [        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
; C9 x5 V7 d. v( C6 {
* ^) W( L) @- d+ A; l: N' \
5 Y* W5 H: I0 O' `1 W( Z// Implementation
protected:
, b: R  E7 K. m        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
% y7 U# c* v2 T* z        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;


private:
: g& u) y  u# K& p6 B' x        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
" ]/ h; J0 ~4 J. E& x& C( k( k7 y

};
* B; Y) G* k' n0 \$ o% k
# h$ s  z0 {4 i* m$ A( E4 ^1 @
2 a/ f& Y9 Y* X' f% I: H- V" T7 @- g// Dummy Implementation to be used when no other implementation is available
8 G2 f, R& O; R( i* ^5 }- N: ?7 Fclass CUPnPImplNone: public CUPnPImpl
) ^; A5 w: V1 J4 b, J{
: H2 H0 p. v% C1 _3 jpublic:
" x' C# j! ^! k7 a/ d( @        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
! H2 }+ r$ C% Y1 Z& n  X* s        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
5 d4 u9 Q# Q* s$ S. V, G: p        virtual bool        IsReady()                                                                                                { return false; }
# @2 D( s3 U$ A2 d- o- L        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
* k: f8 Z' e) `$ U8 Y7 W9 R" f};
- B! o# {2 r: G8 c
$ H9 f' q- ]5 F- ], [/ q3 @- b6 A
% ~4 Q' p8 {' s4 H) W6 Z2 A/////////////////////////////////////
) n0 Y# y( Q8 U9 _//下面是使用windows操作系统自带的UPNP功能的子类
0 J  I3 R* e( t7 y6 W- N  [& n
; Y1 [& n5 c+ z( F
$ n$ G% g5 B. f5 z#pragma once
' `& i1 a0 x1 B8 i+ S#pragma warning( disable: 4355 )

2 Y4 c$ g$ z2 d! E7 q* d) I; x) y
#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>


' b. g" n) _- S#include <vector>
#include <exception>
: e* w- N) \! D* H" X+ i0 d0 T#include <functional>

! ?' _' M8 L  U. p. R  @8 g+ b/ w

9 j0 Z0 w' u. C6 I+ C
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
' |, H# y+ @5 e5 _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;
) V) Y+ l+ X" L5 Q7 Qtypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


+ V; `2 ^. l+ q* b$ ?% Mtypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
* c) J7 E' \2 G2 i5 O) @/ o  PDWORD pdwBestIfIndex
);

- @9 A; [( j. b3 d" W
typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
# v- F( Y4 l- A5 d+ O  PULONG pdwSize,
+ t9 |9 u3 R6 R8 R* x0 |  BOOL bOrder
);


& [" D% n$ ^7 ctypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
5 w/ X% }: y  K7 I) \);
- ]  `8 K; i0 o' |4 m

CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);


6 U4 R! w* w' p! }/ {class CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
" f3 C" t7 s1 a3 V+ G3 r# Apublic:
8 I1 K6 ~# {, e* S9 t        virtual ~CUPnPImplWinServ();
. j: |' e6 o. B( p        CUPnPImplWinServ();
1 k4 X4 n' N( y/ \

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
6 o; f3 p* ~+ i/ f4 X2 @        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
' t; O/ b8 L% u1 J9 I; y
, B  d! \7 t4 Z' h: c; J
        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
+ Q! e0 k1 ?/ |! r        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };


protected:
0 Z  p) G% e2 c* T- v        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
4 X: A9 \/ L: t        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
8 D& T/ ?6 C( Z4 V- t* D1 R        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();


' y; d* Y2 f9 t9 r& m7 f' l7 b        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
7 V: J/ n+ f  [* R                {
# v$ M* g  m4 V0 f* b                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
% f- q4 m' O0 F/ W: ]3 |5 c7 I2 C2 S                return m_bAsyncFindRunning;
        }

1 k* q% h+ d7 F# u' W0 b- I, E. ^
        TRISTATE                        m_bUPnPDeviceConnected;


// Implementation
3 D9 V5 n( }& X5 i! x        // API functions
, e# \1 f+ `9 @        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
3 `5 Z9 g! V7 y5 B# V5 _2 }2 h        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*);
% i2 R9 K0 A* [( a( y' L" Q9 M        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


* Q$ E0 ?$ v: T5 T$ e5 M        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
$ w- t; s/ \2 \2 B, Z# T+ f

        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
* e8 ?) P2 S7 ~0 K0 G2 W        {
7 L6 z# b- [0 D5 Y7 {7 q5 n" R                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
9 ?% Q. J: S/ D; W9 I                        CComBSTR deviceName;
5 L& G4 y" }& J& B( O  S                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
9 X' |% T/ ^; V4 `( z1 q5 t# D

% x- P& R  }- D' J                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
5 f4 `6 f- A! w* W" P& b/ i

                        return wcscmp( deviceName.m_str, m_udn ) == 0;
/ S5 @% x% F  G! N                }
, d5 u! T* `2 s! K                CComBSTR m_udn;
        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
1 ~0 c7 C: O, R7 b3 D) i        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
1 u9 z" a" }) J! }        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
5 v+ L; r: o' Y/ C, X4 k/ h) p; x                LPCTSTR pszInArgString, CString& strResult);
: x5 b+ t; ^/ l  j' `        void        StopUPnPService();
* p( d: w8 @7 I8 U! f0 P3 J

+ Z$ z, N1 X& p# k        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
, Y  `+ {" m0 {/ v        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);

$ P3 W( d7 U( l
// Private members
private:
" ]% F5 X4 [/ n9 ?        DWORD        m_tLastEvent;        // When the last event was received?
' b4 K$ E- E% C! Y        std::vector< DevicePointer >  m_pDevices;
# [! L/ \  J+ U4 |4 s9 ~! x        std::vector< ServicePointer > m_pServices;
2 z0 |* S+ [' r) E# j        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
$ y: j$ J- G4 p' k# B  N0 P
% a1 y8 n0 D  I, H5 G, ^/ s8 w. r# s: ^
        LONG        m_nAsyncFindHandle;
( o: P3 b/ W7 k$ E5 ~3 a  D        bool        m_bCOM;
        bool        m_bPortIsFree;
% n& |5 v( Q3 n3 f) ^5 `        CString m_sLocalIP;
# {3 H& P0 b- I1 d  `" F        CString m_sExternalIP;
. K) j- W$ y; ~( R" R  G        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
) }) W8 c# t* _2 l+ G. ~% c5 H; Y; ?        bool        m_bInited;
& T# }; @9 J" [  }2 i' ^, R        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
# N' a- y% ~  y4 b$ Z. z+ q        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
2 }+ v6 p6 w+ L& E3 [# g        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
6 v) A) z0 _7 f% t0 {

};

. A5 w9 L4 x( k0 @
9 C( k+ v- S$ y; v  ]// DeviceFinder Callback
, G: s7 E4 v6 T3 q# eclass CDeviceFinderCallback
, ]9 @, k/ c" ]2 p% A1 L7 ~        : public IUPnPDeviceFinderCallback
{
public:
  k1 W. M, e' H        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
5 r; G9 H5 F+ l- E  }

( U  w/ F- M9 L8 n! V   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
4 z  Q! v4 C9 J. p& q  [' u5 {   STDMETHODIMP_(ULONG) AddRef();
8 x$ B; \0 \2 h* }1 p   STDMETHODIMP_(ULONG) Release();


// implementation
private:
' F, A& l1 \# V; w* v        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
$ H" n* @+ M. m        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
  `8 S8 o, P( ?" _' x+ |# W        HRESULT __stdcall SearchComplete(LONG nFindData);
5 }- q+ z# E5 B6 V2 [9 q. a. ?" W

private:
! e& W1 W% J" V" Z' y. ~7 [3 x4 W2 b        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};


5 w! Q0 j5 x$ \- x// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
+ \  N4 s: ?& v% N( w{
public:
$ {2 D+ Q* s- b) I% f8 k+ b; s4 Z$ r6 [        CServiceCallback(CUPnPImplWinServ& instance)
7 N" t2 J0 S  b                : m_instance( instance )
* D2 @2 ^% {* ]4 q. v; A5 J        { m_lRefCount = 0; }
   
" `& @1 O9 N) L; \; ?9 ]6 q   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
* \& a8 l; _* d+ z8 E   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
* Y  u: z! j4 [
. y8 k/ @' W8 Q! g  e
// implementation
( n! [  H9 d1 p* f, B. \( t4 A( Lprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

" c6 R' g2 {! {. v$ `
) `. T- m' S7 s5 K8 ?; ?2 L9 xprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
& l8 \4 d. K; E" t6 _};


+ L% O- Y6 l" ]- Z9 b/////////////////////////////////////////////////

' m6 b' I) a: P0 v4 _$ n
使用时只需要使用抽象类的接口。
, E1 \" D3 R3 v: \. H7 oCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
) G# z$ y5 C! v0 U5 q6 R0 `/ JCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
  X  w2 V" p: w/ s+ GCUPnPImpl::StopAsyncFind停止设备查找.
" |/ S4 j2 @# q$ q4 UCUPnPImpl::DeletePorts删除端口映射.