UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   
5. 
   1 @$ h5 _* U- ~- r
6. typedef   unsigned   long   ulong;
   + {# F5 V7 U( ^
7. 
   : o! I1 J* ~  h8 l
8. class   MyUPnP
   / S1 T/ u  I' p+ z
9. {
   4 [. C1 @( r& T4 }( L& v
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    # p9 M) E( X+ E# D4 d
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    1 L, T' A( K; {; B9 H4 }% e  X' x: H2 i
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    % Z6 g& X6 c; g
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    2 @, j2 S6 S6 k; {/ [( ]; n: p; N
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    % T: v' w7 N9 Y9 C! A6 [, @* `
17. }   UPNPNAT_RETURN;
    
18. 
    ( M2 ~: I' ~8 |! }% h4 y+ L
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    5 n. A$ z/ V7 T. z1 ]% J
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    
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)
    * O# b! J, T# o9 e% A$ j
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    # Z" }- s( ~, K- X- b& \* l: q
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();
    # T! Q8 J7 f' j
37. 
    
38. CString GetLastError();
    , @. H$ g/ y+ K. r6 d  O+ K
39. CString GetLocalIPStr();
    0 n" _+ d- O( c1 N
40. WORD GetLocalIP();
    - ?  C7 x- B4 x3 W
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    
44. void InitLocalIP();
    ) ?0 [0 q/ g# i4 y5 p3 q( Q
45. void SetLastError(CString   error);
    1 U0 D& h4 ]) L
46. 
    
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);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    . T$ N7 k* z+ D$ z+ `+ \
52. 
    / ^3 s6 {, r9 T
53. bool Search(int   version=1);
    8 \( \8 a7 u' e
54. bool GetDescription();
    : c0 l% d9 k: X% K
55. CString GetProperty(const   CString&   name,   CString&   response);
    " R) m; o/ F% l/ c( ^7 \
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    ' r3 P. B7 c2 T/ c/ N9 q6 b2 b
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    - h, k0 G3 f; D0 R
59. bool InternalSearch(int   version);
    , N, b- b  h1 K% O8 c3 `; M6 b
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    ! }) q/ W1 r6 {% U# c1 Y
66. CString m_modelname;
    5 q' P/ X6 p1 {: _& I0 I
67. int m_version;
      c' i. }/ I) e2 z
68. 
    " i1 R9 |- e! u; L" z9 d
69. private:
    ; e  `2 `; C+ N1 D
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    $ W7 O" f8 \' Z+ L$ S' a. G
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    ! ~7 p8 X, W# g9 j
75. 
    $ A+ Y- ?' f6 ~7 b, {' n# t) O
76. bool isSearched;
    4 K5 ]  f: P+ \. z
77. };
    ' w- M& X# ?$ M/ x( J
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   % C7 ^4 P3 f, s2 Z" L! i; L
2. #include   "stdafx.h "
   
3. 
   & S3 g/ K' Y# [+ R
4. #include   "upnp.h "
   / d9 P. x& C* {
5. 
   ' B2 O& U1 r( _! i( C- H% F' T
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   + I- r. `; X0 r7 E  P. m# v
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    4 {2 J& j" K  O9 Q: M
11. 
    # {/ a3 i% D: }$ B; g
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    ) [) A( z; i1 q; G8 f. t
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    1 q, ~9 O% q$ A3 P8 D
15. 
    
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    9 T; f9 W* r7 M+ g. `
19. 
    " d; X8 o( Q; z% U1 G- z
20. s.Format(_T( "%d "),   i);
    $ y7 v( Y: T9 c$ Z2 i8 s+ s
21. 
    $ A. T, ~( k) e
22. return   s;
    
23. }
    # E! \! D" ?. ]3 R% e
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    - r; ?: n4 u$ }7 @3 Y* l# ]
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    # u/ H! w  }' V6 A  Y- E0 M1 P
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    3 a: ]/ ^8 }, [: L5 G7 }; s# u
37. char   buffer[10240];
    1 Y9 ?- g+ }4 X% F+ U( T
38. 
    
39. const   CStringA   sa(request);
    7 j) {7 f3 \- j+ A' w5 B
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    " K1 r! L7 N6 \0 T4 n% o0 ^; t
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    % l/ k9 n! b1 c) v# c% e  D) q
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ; c2 w# X3 b6 ^3 B5 c8 j4 o
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    % o) ]. N& i: A& V8 G. U
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    & g; ~) S, {6 _8 g1 f$ W: B6 Q
54. int   n   =   send(s,   buffer,   length,   0);
    0 K( ?4 q4 q. [% h) A
55. Sleep(100);
    7 a" l% I  h2 e, Q4 s
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    0 {: X# a/ ]) d
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    5 g7 w$ N% e7 |- [
59. if   (!rlen)   return   false;
    3 [+ e( e: Z6 {  B' Q3 I. }
60. 
    ) P1 l0 `- b/ L5 I6 y; _3 k
61. response   =   CString(CStringA(buffer,   rlen));
    : F  T4 G& O# b. T8 X& r
62. 
    3 C' F9 @$ f+ @& `( C
63. return   true;
    & X" B7 Y9 E+ [9 y9 N  b' }9 g* l; m
64. }
    
65. 
    0 E- E; H" M$ G1 E8 V0 {( U
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    + E% ~1 a6 C. J
67. {
    / r3 x2 X' _1 |; E' j; E5 w$ T
68. char   buffer[10240];
      \7 e/ ^1 _6 z" T; f) e% b7 M
69. 
    0 T3 {# p! ^# ?. w  s; I' G
70. const   CStringA   sa(request);
    4 H0 l* ~. \+ C3 k  Z
71. int   length   =   sa.GetLength();
    9 f9 U2 b/ e. L( l6 _  \2 a  B
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    , W6 w+ d- ]3 z
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    7 C5 f3 l# d3 T  l2 i+ A! p6 ?
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    . l  p6 e+ `" U$ I4 D
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    1 u' r3 O4 @  r: j4 k" _
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    9 n2 q. R% u4 L% c6 O
84. {
    7 B/ N* b; F0 y" W5 @' \5 }3 S
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    % X$ d/ c5 }1 ?* t5 W
90. result.Delete(0,   pos);
    1 T7 `# \5 |2 ?4 B. D7 d, l6 h
91. 
    3 ?/ ^: K1 e) }# S
92. pos   =   0;
    ! S# K) D6 h! M. g3 z  Y5 Q
93. status.Tokenize(_T( "   "),   pos);
    / l& E! b+ C& v* \$ L1 ^
94. status   =   status.Tokenize(_T( "   "),   pos);
    . @2 j- r' U1 L+ n  K/ M
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    7 x+ w  S% ~8 k" h, v
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     ( q+ o. `" b* D! a( r) I" A& I: r9 Z
103. CString   property;
     
104. 
     - d, ?* S' d  E
105. int   posStart   =   all.Find(startTag);
     ' ?, T9 m; q# F( t5 j/ p  N6 Z
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     * R% S0 {8 G: [" U' z- b
109. if   (posStart> =posEnd)   return   CString();
     6 Z  J# M& P1 u3 k! y8 {/ W: a
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     # @) _4 Q9 e0 k7 b/ B
112. }
     
113. 
     1 h& O8 D: X# ?1 j4 @* I
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     , \. s& b  d$ h$ W1 x0 s  r
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     ) b4 f4 m- W) ~5 ]6 L9 Y! o
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     5 S8 j; J6 q3 c7 b7 `: E
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     . Q, y7 h1 L+ f* g
126. search   =   m_Mappings.GetNext(pos);
     ' l5 U: u% x( s5 N0 g4 q
127. RemoveNATPortMapping(search,   false);
     
128. }
     1 J7 a7 t* z- Y7 ~) e% ]
129. 
     
130. m_Mappings.RemoveAll();
     ! @. G+ ~; B, Q, N) t, I
131. }
     
132. 
       M! a/ x) i$ ?% C" L
133. 
     ; k* M+ g$ I) B9 D' ]/ I, n$ B
134. bool   MyUPnP::InternalSearch(int   version)
     4 h1 o$ c- \1 a6 W( F5 c
135. {
     ' X5 t9 ?8 ?2 M+ ], q
136. if(version <=0)version   =   1;
     - ]: Z. G: }7 _2 c' y) o
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     * F) i: Z6 n/ k8 _* J  n  }3 a8 z+ ?9 L
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     9 e: e9 k7 z: w0 N# q+ B1 l5 M, A
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     $ J' F: p( Q) a6 V6 `- y
148. ioctlsocket(s,   FIONBIO,   &lv);
     0 O% c, T3 L- X! I8 C" D
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     ( T% Y  L, Y' e+ l) x, e3 T, ]
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     9 n  K% F: G2 L9 M
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     4 x6 X9 `4 n5 V4 G+ C
155. CString   request;
     2 p5 V7 `  U6 _( V, l: {
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 A! C( j8 I, t: o
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     
161. 
     7 D! f4 l' |, o3 r* A) L3 B! p
162. Sleep(10);
     . f* d, x# I- ^: A/ N) K0 A" c/ B
163. 
     
164. char   buffer[10240];
     + ?4 @! q2 `7 T9 ?
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     7 X) U. n- i1 O
167. closesocket(s);
     4 D, D$ u4 y" ^( K' m
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     ) t. q* t  _# i) t: x/ P" m  f0 f& m
170. CString   result;
     : `7 V4 Y5 Y& m% m) j1 P" |
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     : w- z. a: ~: ?- f# ?
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     " U  U: }+ W9 j% S' U
176. for   (int   pos   =   0;;)   {
     8 v# }6 _) y0 a3 @; A
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     & |: b  B" I  P4 V
178. if   (line.IsEmpty())   return   false;
     " T9 \: R! L) p
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
       s( l1 P) O& P5 }% N4 U4 l
184. m_description.Trim();
     # q, W) }0 a' d: ?
185. return   GetDescription();
     & M2 V  t$ `+ T
186. }
     ' x$ w$ @) f5 ?- G
187. }
     
188. }
     ' e; ]+ Q6 D3 @! {& `  |8 T7 f
189. }
     4 @; M/ V1 P, Y- h9 `  x3 ~3 X
190. }
     
191. closesocket(s);
     0 n+ f- l: i$ G6 H% x
192. 
     1 I* a5 f  m/ Q5 Q) V
193. return   false;
     
194. }
     9 i3 f* S9 q. l# Y

复制代码

天云上人

以下有关upnp的接口来自emule，
$ ~, j8 b: Y, e1 @

///////////////////////////////////////////
2 u( {1 a5 [6 d' V2 o; b0 G//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
& P# h3 ^  G$ _; v" C7 A3 E
! B! Z( v4 E; R
  X& d4 [+ T3 k" p" Y* A* v9 j. t6 n#pragma once
#include <exception>
4 F/ }& Y+ u+ a& D* h

( Y, `7 ?6 C: }8 @$ r# B5 t  R  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
};
2 q- b/ q" L6 e2 |; o8 C
3 z6 m( J. H! B' X
enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
% x  R* b3 @- e0 a% [, a1 M        UPNP_IMPL_MINIUPNPLIB,
4 z3 `9 j& H3 w  ]5 x4 b7 [        UPNP_IMPL_NONE /*last*/
1 K( @) }. z: u/ v6 P) A( `};
" q( i! M- i7 X' t8 {( v, \


2 s0 b8 l3 x0 P! J' z* F
6 q. g+ I. g2 p6 d) @  d5 Mclass CUPnPImpl
{
public:
        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
/ Z5 }( D4 s0 d* d        enum {
: k1 a" p. F( k. C1 Q                UPNP_OK,
                UPNP_FAILED,
/ Z: R$ @6 _' V5 v' y                UPNP_TIMEOUT
8 k1 s2 H4 E  H, q        };
8 v8 m$ e; o. F. j$ Z; ?

3 W1 u& D% v8 E7 n  @        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
) G. a* s/ r) C; Y% w        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
' V7 L/ E: A' a5 s( u" o! t) ?& v        virtual void        DeletePorts() = 0;
/ X5 d% n" S$ p; m% j        virtual bool        IsReady() = 0;
0 T. F& Q8 |& |( q        virtual int                GetImplementationID() = 0;
       
, E7 d: _; Z! S' c3 Y; n/ }7 a" N        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

3 E) Q# j1 v  ^, f3 f4 D
& w' H8 `4 g' `) A8 S. }' l; Y// Implementation
protected:
$ l8 }* s2 R3 I( I- l5 x        volatile TRISTATE        m_bUPnPPortsForwarded;
: D, O& X5 ]* B) a        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
7 O5 t+ O* ~, N; ^& H0 M* g+ y        uint16                                m_nTCPWebPort;
) ]: M8 c7 I6 |        bool                                m_bCheckAndRefresh;
: k' p+ R4 ^/ v* y* D  j

5 i+ `5 J; T: b) e1 O; sprivate:
        HWND        m_hResultMessageWindow;
5 C1 v; S- r. m- V5 M; D        UINT        m_nResultMessageID;
- T5 {6 Z5 D4 P3 T

};


// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
5 `; p0 G" F" |% V: W8 r% C  b{
0 r6 \6 [  S: i" U& ?public:
* X9 G! N; z# x        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
9 {0 d9 z# M8 x$ W6 Y        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};

, h. }% }. x  P% t9 _; X, M& H
/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
& v( {4 y4 l0 A6 A& J
5 w+ v/ G: G2 U4 x% g5 u; B
#pragma once
#pragma warning( disable: 4355 )
* @- b/ c! b/ J: {

9 L5 y0 g! i' X" Z( x3 y& G% B, t#include "UPnPImpl.h"
#include <upnp.h>
& t% ^$ G0 u. o& d9 b6 D' C#include <iphlpapi.h>
7 N: u* {; ^; I7 W; l, X#include <comdef.h>
#include <winsvc.h>
0 Q" Z: W4 k: o7 }, X
! g# l7 P( Y- M+ ?* T
#include <vector>
! c' R* a$ }5 Y/ G1 M& U/ A#include <exception>
#include <functional>

+ y) R! D" X/ E. o$ N


typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
2 J* Z5 R. G$ p- ]. ttypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
" [) E* K+ V1 C% ztypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
5 Y! E8 p) E* Q9 ]0 gtypedef        _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;

& ?9 D9 U6 J/ M8 f
typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);
. y) Y" P" ^$ ?
# h% X7 i5 G3 g6 f% O
typedef DWORD (WINAPI* TGetIpAddrTable) (
' x  b# g- k: i  Z% F6 |  PMIB_IPADDRTABLE pIpAddrTable,
  [4 L# J$ l" T( O. x7 g  PULONG pdwSize,
5 s, k, w: ~& F; V/ F$ h6 B3 N9 @  BOOL bOrder
- Y9 W7 _# b* i3 n5 ?6 n2 Y);
  _' f) I9 g3 k& c

5 e* }/ x1 S+ [  P1 @typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
2 H* h1 D: h& Q: [9 Y# Y);


CString translateUPnPResult(HRESULT hr);
' E/ g" c! I) }: _( tHRESULT UPnPMessage(HRESULT hr);

7 ~; H8 }* b& x+ W3 p" ?0 L
3 c( o& X0 @0 h7 @1 [7 }% @class CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
4 F8 t, H+ _8 z: G        friend class CServiceCallback;
// Construction
# R( f; g) N  u1 i) @public:
7 S6 E0 u# g) u        virtual ~CUPnPImplWinServ();
- ?1 b' ]& W# }) `) j3 r3 D. K; G        CUPnPImplWinServ();
, L# w% V0 p( ?( C* @
- P- _# |# w& q2 m
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
% A8 `, h- E; [! M        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
' q3 K* z! `# ?+ R. i5 q; C/ h' L

        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
, c$ d4 I, y3 Y9 i; H$ r        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
1 @# f' N2 p/ i5 ^        virtual bool        CheckAndRefresh()                                                                                { return false; };


protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
( w' T8 ^+ ?! k. I        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
" Q* \9 G0 e  k" P( b4 f* `6 }        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();


        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
" G; G' g4 E" V                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
3 C: @. H+ p" w1 u                        DispatchMessage( &msg );
                }
$ R- k5 t5 H0 E$ J: I' q                return m_bAsyncFindRunning;
        }
) n1 k1 C( v3 m8 J/ M8 A) n+ M
! e8 L- \2 O$ k- B
5 A1 g) a0 ^7 b5 b        TRISTATE                        m_bUPnPDeviceConnected;

  f. E/ U& h3 }2 y! S" _
// Implementation
, L% P4 l" @7 O2 }# i, e        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
0 Y/ I% @) O( ~        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
, _4 a) b8 K5 h  d1 b7 t        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

+ Z) ]7 D) B' s5 j8 A9 U) g& i
        TGetBestInterface                m_pfGetBestInterface;
4 T  P( U4 I' ?) V        TGetIpAddrTable                        m_pfGetIpAddrTable;
& _+ _& V+ r" V        TGetIfEntry                                m_pfGetIfEntry;
, D; P; W. Y, G# i! {( o# j

0 [2 T( u" K* d- ?9 @. w6 c  D* ^        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
: l7 t7 ^) }! S" c- ?                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
8 B, ?) h( G* b0 }0 Z
7 |9 y: V! W; A4 K: @
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
( L( [4 {+ c* }" J# m* B3 L                CComBSTR m_udn;
" P( f' W0 H) D2 X3 F( T        };
; L3 i! I! Z$ r+ H1 Y* X       
% c+ e2 X3 P. d0 z+ j$ Z3 X  ^        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
0 E: D5 j9 Q) l, Y' _        HRESULT        MapPort(const ServicePointer& service);
- w0 ?' [4 K, d: ?2 e$ l        void        DeleteExistingPortMappings(ServicePointer pService);
. S. D5 e9 D7 g: q9 }, z        void        CreatePortMappings(ServicePointer pService);
8 o9 `7 \3 D0 P  Y9 k* [        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
1 {' R- P# J: O        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
2 D6 ~9 J4 i" d' \4 N' N2 n4 L                LPCTSTR pszInArgString, CString& strResult);
8 v5 l. B9 E1 ~$ m3 v        void        StopUPnPService();


2 O  _' M/ S! @6 F  P        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
4 o/ d$ ~7 b$ k/ c$ L        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
0 @2 a  o( w  j        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
, o4 L6 _0 {; c2 e8 p        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
3 ^6 {! \5 q9 d& v/ }* \        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);

$ `, @5 ?( N6 s: p. Q% q/ n  a
// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
7 E+ |) [8 U' Q        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
2 s1 V" d( H, k+ _- P8 `4 m        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
7 ?6 R' s: Q3 F" \. o( V

        LONG        m_nAsyncFindHandle;
- E7 Z% U* T2 V        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
% u5 K" m: Q8 s; J( T        CString m_sExternalIP;
/ z; M" t$ F/ z        bool        m_bADSL;                // Is the device ADSL?
0 j  X/ F0 f; o1 x$ H: [( v9 i        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
0 x! N( |# F! e6 h/ B        bool        m_bAsyncFindRunning;
: I1 e+ E7 ~, H        HMODULE m_hADVAPI32_DLL;
- a9 V5 X* t9 K! l; v        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
# F" u" t3 `6 T$ H/ _' |) i1 O

9 D0 g6 U2 Y! c8 Q' _: q};

% h7 e+ j( ~# k
8 ?; E1 N9 r2 A1 n; E; c// DeviceFinder Callback
4 x+ L( l9 r6 k- Y$ r$ bclass CDeviceFinderCallback
9 E* q- d" t+ h6 J! h# [0 D        : public IUPnPDeviceFinderCallback
{
1 ~: v$ l; ~9 F0 b6 `: e- `public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
$ f0 G4 Y  c: d3 m& X1 N                : m_instance( instance )
! V5 T; {7 M) B" |$ l$ L4 I        { m_lRefCount = 0; }

: q* l' O2 n( N- o
, b/ \+ U* C8 }" m, N   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
7 y9 ~4 o" A( c   STDMETHODIMP_(ULONG) AddRef();
) a& Y; a1 f! V% L2 ~+ a; Z   STDMETHODIMP_(ULONG) Release();
) I1 h4 I8 L3 [  D* A$ k; G

// implementation
$ G" }) G: J4 T. P" Dprivate:
8 [5 P7 O0 J7 i( f8 x$ o0 f( ]3 T4 q        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
( {5 o9 ~' b/ `5 N        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
& l% Y  B1 j9 P; T8 E        HRESULT __stdcall SearchComplete(LONG nFindData);

' M8 g1 p+ j6 M
/ Y. ~0 E" j4 B' N. K" l1 Cprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};


# }; F, R6 [0 s0 V// Service Callback
. o1 M6 C$ T7 X3 e4 P' rclass CServiceCallback
        : public IUPnPServiceCallback
{
public:
& w  b. Z2 D% e" F" G. j  e        CServiceCallback(CUPnPImplWinServ& instance)
, w: q: v$ N/ ^. h( A# U                : m_instance( instance )
2 E1 M+ ~8 S. j# Q2 `        { m_lRefCount = 0; }
   
. o; p7 M, x1 ]" v   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
" ]" o0 x9 F& w6 h0 E4 ~

// implementation
3 V1 z0 V% `) Y6 E* O5 z: rprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


# V* u, a# h6 q  G! U/ v0 z( u) ]private:
        CUPnPImplWinServ& m_instance;
; W6 p5 v0 d# x8 j' l        LONG m_lRefCount;
) l# n4 Y! [9 h% r# x" g. X6 E};
& S  S4 N) |6 O8 U' j5 r" B

8 y2 W" \4 B7 }/ z" x. |/////////////////////////////////////////////////

. O+ K: s' J7 L/ B7 C
使用时只需要使用抽象类的接口。
9 m7 W( L; N/ }8 i  D$ `CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
9 i7 c; `# |& s8 f+ RCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
, P" i0 R- o( a. {# eCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.