UPnP

天云上人

/*uPnP.h*/

1. 
     N' j+ x0 p" k  |
2. #ifndef   MYUPNP_H_
   
3. 
   ) l, g4 C- f2 N) n
4. #pragma   once
   : r4 A, d$ g7 O2 v) I# }, r" ^
5. 
   * y) R, V# u5 v8 ~+ M9 V
6. typedef   unsigned   long   ulong;
   
7. 
   ' Q4 ~: w. ~& S
8. class   MyUPnP
   
9. {
   3 S" R& x2 s* a$ P: R' N/ R6 z
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    ' O9 q: b, z- b/ v
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    0 r0 i0 r0 X0 |
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    - e* r& }/ O! j% `2 y( N# f
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    ; n+ z2 c- l' I; `
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    ; i) ]. Y# u! T2 U& d2 D
24. typedef   struct{
    1 W& B( F9 f  @% A; g' I0 b' K  p( `
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    0 D. \" k/ p, C6 C8 R
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    ' I6 D( N- j+ }
29. }   UPNPNAT_MAPPING;
    
30. 
    3 _% @( D2 c0 H4 B
31. MyUPnP();
    ( h0 j4 z' ?$ X& B
32. ~MyUPnP();
    
33. 
    $ J; W: h4 ^5 E' J
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    6 t( L: W6 J8 t; }9 z. m
36. void   clearNATPortMapping();
    9 D* @9 z3 l# k
37. 
    ; T, ~2 w/ H% n6 v2 B" R/ G
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    # @% V0 y6 d; e  j, V9 E' T
41. bool IsLANIP(WORD   nIP);
    
42. 
    ' I5 f  Y$ T* J# s1 Y3 l
43. protected:
    6 P9 ~( ^$ R7 p$ n
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    " O8 Z2 h8 L) r7 c; `- i9 M/ D# j
48.       const   CString&   descri,   const   CString&   type);
    * h8 u1 V8 s- ]- @# m& q: |9 K; T
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    ! ~, n5 Y. t9 [3 E3 Z- Z5 v2 K, ^
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    9 d- q, H& k' c& B
59. bool InternalSearch(int   version);
    / P$ Z( ~; d) g; f5 L6 }
60. CString m_devicename;
    , r. |4 M! [, t
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
    ' j* b' z3 p4 N& `( q7 s
65. CString m_friendlyname;
    
66. CString m_modelname;
    + c! n7 h" @  V8 F9 G
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    ( l+ x2 e, t" {6 D; m
71. 
    - r+ X. ~8 l5 _  b' C8 [5 a$ X) V
72. CString m_slocalIP;
    
73. CString m_slastError;
    . o# O! F  v$ d: f; P
74. WORD m_uLocalIP;
    - b5 A) J% ^- W  f. f- [7 U0 C
75. 
    
76. bool isSearched;
    
77. };
    9 w1 j8 F: b8 T
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   ! X+ v9 Y1 A. y& V
2. #include   "stdafx.h "
   
3. 
   % m  N4 G0 L4 x$ o* R  k5 M
4. #include   "upnp.h "
   3 }  C" c6 ]8 m- _. v8 H
5. 
   - y2 i; l. F! x9 M# j
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   8 O2 e5 N( u/ P. N. A+ m/ k" T7 R
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   6 y2 O8 D' @* S) M6 Q
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    + h5 I+ P- r% z- t5 E. s/ v
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    $ t" N* W% Y: _4 J
13. static   const   int UPNPPORT   =   1900;
    5 d, Z( f" C, E  ?
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    7 z! u# @; V7 G1 H& N: ~( j. r
21. 
    # v# s- Y; K1 Y/ o* O
22. return   s;
    
23. }
    - g, ^1 F) m4 p' E5 I7 f+ ]
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    6 o2 l1 R' y; K" a( b5 X6 o
26. {
    7 V& a6 Z4 g' C2 }: v' u
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    / Q! Q' ]# p. ^, j! j3 m
28. }
    
29. 
    & i) B4 t2 B" K, C
30. const   CString   GetArgString(const   CString&   name,   int   value)
    , k+ {% F: M3 h0 C1 x; n+ d) l# G
31. {
    6 G% i1 Q. K/ i# [3 V
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    7 }* l" s1 y' R* t% D+ K0 W
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    9 I. {8 w; f3 T7 h* r1 O/ {
37. char   buffer[10240];
    ! {3 n) ]' G  I0 g. U5 \7 S! s: p
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    0 Z' _* n7 H' v0 k
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    6 @1 ]5 U$ B) a. q' L# D' z
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    " H" c  S# A/ f3 K' n$ ^  {1 H7 P8 h
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    # B4 ^' F' m' N: ]: w4 ]
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    ) s  \1 t9 @; ]( G, ]* V& f, O
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    . ?* M" ?" I# V; `* @. |( u% a
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    : T# A9 L& \( i' d1 ]
62. 
    8 U  ?) u: L/ A+ c
63. return   true;
    ! d9 K  F, l( l& b! C
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    . E  v: _9 g8 V* H4 R; k
67. {
    
68. char   buffer[10240];
    , A9 D% g' J7 F: j& h. H
69. 
    
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    2 Q* K& c9 l, t: o. [; c1 p
72. strcpy(buffer,   (const   char*)sa);
    . V) B% q2 L) _1 R5 J1 ?* k
73. 
    ! Q& p( v1 j" M! p
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    - \0 u7 l/ v9 P% i/ X0 N$ p2 e4 Z
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    6 [# ^& P( n2 Y) c3 |
84. {
    
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    5 O4 [# k6 u% a& L
91. 
    : G5 C, |& a% e! I9 x
92. pos   =   0;
    2 ?8 @) Z& r6 I  F
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    * c) E( A$ a7 O7 w+ c) w
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    6 S& G3 d  v" s0 H: ~# p
97. }
    5 W2 r1 s. i3 z- ?8 n  u
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     , V4 `$ e4 k9 ]6 }
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     , u5 X: F6 ~) k* T$ I
103. CString   property;
     
104. 
     6 x. {( {! U# [& T' h
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     & k! f* O5 g* n+ {9 C) a
107. 
     8 q) k) e+ n2 M
108. int   posEnd   =   all.Find(endTag,   posStart);
     4 M- T1 E- s8 m( Z4 h
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     0 m7 k" \& Y3 N1 h) h$ L& O
112. }
       G$ c. I8 d& T6 c# |. [
113. 
     ! j, `0 m8 I* h) V) E
114. MyUPnP::MyUPnP()
     & O% o  b+ K4 A% s$ Z
115. :   m_version(1)
     4 {% d9 E- `2 m* j
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     ' ]/ Y. l  L" r. I& Y; u8 F+ `
119. }
     ; h4 e; l( |% d( g9 \- o9 H
120. 
     # ^, G3 ^+ o3 U4 }0 V
121. MyUPnP::~MyUPnP()
     4 Q8 ~3 k! c+ ~. e
122. {
     ! K+ m3 |/ r1 [0 O$ _
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     - U( X4 N/ ?% z! a9 w/ ^5 H
128. }
     
129. 
     8 w1 a8 j% S3 l& ~2 c# e2 G3 w
130. m_Mappings.RemoveAll();
     ; T" }8 R' h1 c) ]5 R
131. }
     
132. 
     & b' T+ E9 F7 F4 w4 j( a% q8 u
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     1 z0 r# F5 o% X' @5 u& b
135. {
     2 s, G6 }1 K4 r  n0 ^3 Q4 H, F& [
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     . V& w* t* d4 l/ d/ V
138. 
     ; ^  q  p3 `' j$ V) S5 I( f
139. #define   NUMBEROFDEVICES 2
     9 s& V. @$ U5 p$ U4 h5 `+ z" j- K
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     , L* F. [  b" b) l. a# h
142. {UPNPPORTMAP0,   _T( "service ")},
     " e, X/ m4 f4 A; s  m
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     ; Q& W- s% I! a; E7 `
144. };
     ) j2 Y" S0 e2 h2 L& i
145. 
     & e) A( j% s: x+ n% }8 Q
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     0 S8 _  B+ j9 K$ {8 b  o# S
148. ioctlsocket(s,   FIONBIO,   &lv);
     # O6 b1 h- g4 f2 i3 f
149. 
     
150. int   rlen   =   0;
     # k4 C8 @+ e$ u. \
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     5 a( {" Q) }% S( H$ x' W
152. if   (!(i%100))   {
     + F9 N! m' v; H! I8 k' _1 a! _5 ~6 x  A
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     ; v% B( K7 k/ s3 \& k3 u) c
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     8 f# f2 n5 V2 x$ G" i' g7 K
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 "),
     4 _2 E% W+ M' ]7 a9 g5 [& e% o
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     / c8 d& {( r# @& A& u
159. }
     
160. }
     
161. 
     9 j5 a- H* ?  F+ u5 f
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     4 G. E* W/ l9 T  R3 Z; Q
167. closesocket(s);
     
168. 
     8 P5 z& d+ r% \5 I- [2 Q
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     $ l/ C! w9 r. H/ v0 A
171. if   (!parseHTTPResponse(response,   result))   return   false;
     ( }! f& \1 u& {2 K7 W+ ~
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     ) Y" W8 P7 c6 A! a  j+ ^6 Q
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     % ]! t" Z" Z4 s& ]# e
179. CString   name   =   line.Mid(0,   9);
     ! \6 h$ p) e! ]% q  @0 M
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     ! I* b; V& p' J# ^# ~8 _
182. line.Delete(0,   9);
     7 w: D3 L1 ^7 k6 `$ v9 X7 b+ J9 z
183. m_description   =   line;
     1 [; x  U! R. y# X
184. m_description.Trim();
     
185. return   GetDescription();
     9 u: L$ F% \' d" S5 h* H3 z, l* r
186. }
     
187. }
     / g8 K, h' ]* v, u' j3 B
188. }
     
189. }
     
190. }
     
191. closesocket(s);
     
192. 
     
193. return   false;
     
194. }
     * ^1 g  s0 T( e8 j$ o

复制代码

天云上人

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

1 Q; n( ]/ S. W. I, O6 v( \% ~1 r
///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
: ?& X% b% Q! k* S4 r* W; L

#pragma once
#include <exception>
/ z0 M8 n( M5 A( d  V
0 I0 d. r* L# V0 |
  enum TRISTATE{
. X1 x2 A& S' V( y9 b2 ^        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
};
0 G, j' j3 W5 `! `9 h
" c2 }) H% \9 ~  c8 c/ q
) i& a4 C3 f" u; S* benum UPNP_IMPLEMENTATION{
; Q5 r- \8 X6 ]5 d% Y; T$ i& r        UPNP_IMPL_WINDOWSERVICE = 0,
3 a, Q/ p. a6 a, ~$ ]        UPNP_IMPL_MINIUPNPLIB,
) P: _( z8 g( E$ j8 m0 C, c        UPNP_IMPL_NONE /*last*/
9 }3 g& Y( l' e$ G};

  F3 {, h) Y: T3 |/ S1 Q
6 ?; f( g/ G9 s4 |' ~) v6 \# u3 h& e$ \

- C, g2 B& t6 _) F& S/ Tclass CUPnPImpl
# v( [$ \4 l) {2 n{
public:
        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
                UPNP_FAILED,
6 l2 [" m+ L0 X2 V; x8 H                UPNP_TIMEOUT
7 G8 V* g8 m; K" u/ x        };


- V' Q$ n( j  A$ G        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
2 r! }3 ]5 f+ `8 K8 v/ S        virtual bool        CheckAndRefresh() = 0;
& e/ N4 a' K* C! _" `, Z        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
% I/ K3 V4 j+ G, O: u8 ~7 P9 m; W       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

! a! s" c- O: {% V5 ?
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
. B* t" h2 ?. H/ v. @        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
* D1 s) V- c9 P3 g9 @; I        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
9 O" C( W2 x9 c* ~6 h8 b' f) H        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
# R* f7 q' _. Y# M( \+ s5 ]
! d8 S; T7 r2 ^
// Implementation
9 d0 G1 F0 b( o( K: {8 w/ d6 aprotected:
7 P6 l+ v0 k* Z        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
2 V7 @. t: w% d( {3 W' I# g+ X/ V        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
! `: l) O$ [; n$ \        bool                                m_bCheckAndRefresh;

( `4 }7 ]5 c5 L. M* J; r
- D, j. L& q1 `7 Oprivate:
# R' d- G' _: a- T6 k        HWND        m_hResultMessageWindow;
; G* N( b8 e% H: v% r/ m- x0 b        UINT        m_nResultMessageID;


};
5 B9 S  j2 j. E/ v7 k% n& x" e
( o% Z' @& U- P$ H! r# T
. ]- n2 I  c4 \  P; o: n2 ~5 j// Dummy Implementation to be used when no other implementation is available
' ?! x) F4 c& t" s/ Y* Jclass CUPnPImplNone: public CUPnPImpl
4 A( w, M, g+ Q) S* k3 g{
* X+ J. z" M9 y/ y# G& ^public:
4 T: ^  \" x! O' S  R8 h2 s        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
; o' C, ^6 g- F2 r8 m9 _0 X- `        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
: N! E" b% @' q0 j' O6 W* A        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};

$ @+ d, P# F. R; J4 E
/////////////////////////////////////
* E$ q% |( N) Q! i# U7 b//下面是使用windows操作系统自带的UPNP功能的子类


#pragma once
" l8 S& }- I1 c* `% |#pragma warning( disable: 4355 )
, S9 S7 @: t/ Z7 J* Z" k4 g+ z- R2 V

8 p, t) [; j3 G% @, A#include "UPnPImpl.h"
, E. J8 O) n* E  k#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
2 q7 z! g8 e% }7 a#include <winsvc.h>


#include <vector>
#include <exception>
# Q* Q* T. a" G' c; N1 |#include <functional>


3 v, c+ o+ J  M; l
; ^1 y$ y% F) D- B/ {( ]
$ C+ ^" w8 D) ~typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
. w: n8 j" ?2 W4 Wtypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
; i( j% J: ~5 l' n( K+ `4 Htypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
$ r3 ~# a8 w: f' Ztypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
4 X# G7 ]5 x4 C$ ~0 mtypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
: `  W! u2 w- x
" l2 |7 M7 v  |) G& o( _2 c
+ M- }. ]: m$ X" Jtypedef DWORD (WINAPI* TGetBestInterface) (
  j7 h& l% ?1 s- _8 P  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);


" T" y1 _/ ^; U; ?, o( u$ Ytypedef DWORD (WINAPI* TGetIpAddrTable) (
; H) e+ ]( y4 o/ s: p  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
0 h' j8 O9 Z% l4 X) X);
; V) f5 E3 J& @) p9 d

* ~: t" l- S7 u6 `/ x' i, atypedef DWORD (WINAPI* TGetIfEntry) (
+ A  H5 o; O5 F' l$ @  PMIB_IFROW pIfRow
" S$ k7 y" {# j$ v6 E" s0 h) T, n);
- G4 h; p8 l' m" e% ~" f
; o4 n' l/ B8 H1 h6 U' S& j
# E8 A3 c, K* e  T7 ^- _0 o  eCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
3 j2 c; Y! i( }& z: R$ R
4 u! w, o! q# c$ c( V
class CUPnPImplWinServ: public CUPnPImpl
& J% C$ N& ^, A) U( }+ j{
        friend class CDeviceFinderCallback;
1 t5 @+ N  k  [/ s' l9 ~2 ^        friend class CServiceCallback;
. `0 S: P4 p8 B- \9 L3 ~5 j// Construction
) N9 e: `- O) E! V2 K: A& Bpublic:
/ B6 P* f. o/ H; Z5 D: Q1 S        virtual ~CUPnPImplWinServ();
- E2 E: @' y8 q& a        CUPnPImplWinServ();

7 X3 X: A9 }& v$ |, C9 V$ _& Z0 c
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
8 N  }# E3 `" H5 O6 l) n        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
6 q& o, |! G# ~; V3 ?; A( F

        // 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; };
  _1 ^, y. L6 T, b2 ^4 z. Z
# ^5 @: F3 q8 ~
protected:
( s) X$ W" m8 A6 U% A+ k5 k2 C        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
) H$ {; F% B" i4 ^0 n/ g0 d/ d        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
  B& S2 a4 u3 Q, L1 W        void        Init();


/ K  N, j. \" d: o% o) }2 P        inline bool IsAsyncFindRunning()
  C! f2 ~# Q2 V1 P2 m        {
: W- Y9 N3 O7 l* ]* F                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
4 v- w0 ]  d% L                        m_bAsyncFindRunning = false;
                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
; {! c. R+ u$ o8 u) i                {
% ?. F3 K3 V: d2 E% w( f                        TranslateMessage( &msg );
# w5 t0 |2 K# h; B  _' w                        DispatchMessage( &msg );
                }
$ ^; X; W! u3 f: S* j: x6 U3 _                return m_bAsyncFindRunning;
        }

) Y, x# ~( a+ g+ B: G! i. K9 \/ a
        TRISTATE                        m_bUPnPDeviceConnected;


; A; C( u0 T! E% s' d// Implementation
! j) g) S' \: ?1 A, I3 x. W        // API functions
- w# |; y' k' Y0 T$ v        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
, e: S+ z: S9 _6 }7 k        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
3 R( W6 S* ?$ `0 Q* I        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

9 V0 D% ]5 D$ k3 O& ?$ U
        TGetBestInterface                m_pfGetBestInterface;
2 A# v! K! [* P! I# u' `* S" L1 I1 G        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
) W- d; ~. I: @$ E$ i+ h. H8 E& m
( t) e: [- |" k9 A: |) t: }4 i& D
) p- M' L& J& N; r! g7 J  {3 i        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
) I0 u$ Z5 p0 M/ \8 a                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
' t  O  N) ?' Q/ }  J+ Q                result_type operator()(argument_type device) const
                {
' i% p% O8 f/ B' F% ?% i( n                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
, Z5 L" e. ~7 H5 B. ?" s: |: ^! {

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

* H5 U$ N' Z  \9 E9 W, B
* f8 `. T9 l6 C2 T, r5 E                        return wcscmp( deviceName.m_str, m_udn ) == 0;
+ {2 d! u: {, n                }
                CComBSTR m_udn;
1 j- U/ f7 Z+ A; W5 D        };
       
8 A0 t/ `8 k0 Z% [2 M4 n        void        ProcessAsyncFind(CComBSTR bsSearchType);
: O" t! a+ l, T6 N        HRESULT        GetDeviceServices(DevicePointer pDevice);
' _5 E$ E3 w% \' A- l        void        StartPortMapping();
7 q$ J- I9 i+ n% e' Z. _6 W3 W        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
, W% ]6 R; V( S* x        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
3 q# l6 G* O/ P- o4 _% ~2 D# f( P                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
* V( ], }6 u* C, g
( ?" ?( A  p/ W. [; V& ]" B
' i; L, N2 p. R        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
8 A) O' O) l) i8 ^4 J        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
: o- T9 X* e1 q6 i( t  E        CString        GetLocalRoutableIP(ServicePointer pService);
. f# b3 ]9 [1 l- E2 _" z/ U
! t2 l0 E/ c9 Q: i' A. R. E& r
( o' b. _' N5 R! _( k: Z// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
2 {% g# r! @; u  {$ m8 S' T4 S        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
8 N3 X0 ^& b4 z! E, S0 g        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
  w& \# {% i- ~) W

, e: P0 ]8 J8 U        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
  P6 E2 U+ A9 O        bool        m_bPortIsFree;
        CString m_sLocalIP;
* H, z$ ?+ v$ Y5 U6 f  y! W        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
8 H5 R# j  b$ ?6 Z        bool        m_bInited;
1 D5 u3 l6 e1 p, \! H; E3 ?6 |        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
0 x% r+ A1 x( H- ]$ t! `& j        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
1 `  l0 D. P/ a% v0 ?( p# L        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
# F% K) h& I/ {
3 i) K9 I* K. ~8 K2 j8 l
};
; u7 C9 n* r0 p: F2 B

// DeviceFinder Callback
class CDeviceFinderCallback
- F4 ~9 r6 {( H( n- g        : public IUPnPDeviceFinderCallback
! q+ @: w/ K$ Y{
3 G4 X, @! f3 Qpublic:
1 B/ s+ t2 h' G. K" N9 O/ j        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
# D, E6 i0 \% K9 E* A        { m_lRefCount = 0; }
$ H) q( `5 g' |  F

   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
# o! i, h) {3 b

3 D/ B4 k' B% u2 a) n6 R// implementation
private:
0 m/ ]9 W: r  G/ K6 _        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
( ~- a. V% k! p2 C* X4 w        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
4 o! v! ~. O! u        HRESULT __stdcall SearchComplete(LONG nFindData);
, y7 s+ Q4 {/ X8 ]* f% ~

private:
        CUPnPImplWinServ& m_instance;
5 q( G6 _$ H* O0 B; L        LONG m_lRefCount;
};

; p  A0 d8 O) H
// Service Callback
class CServiceCallback
1 {$ c, ?" r1 s; [# F$ _$ `        : public IUPnPServiceCallback
3 ^; h; {. _) c* n; u{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
: q& m4 ?9 F. v# s* }, P        { m_lRefCount = 0; }
/ q% J0 d/ E" B& q. [   
- R; `5 [. `8 ~  h4 m6 d, o7 d0 I% T   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
- w. u8 d" j/ ]; E5 |% W2 O   STDMETHODIMP_(ULONG) Release();
; o" J7 B" c+ a3 A/ x
% I0 i$ r& d! H5 K0 f
// implementation
private:
+ Q8 ?6 i1 e1 Z- I8 \8 l7 e$ J        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
5 C. W* g- N0 x1 R% {8 N! ]0 }- X$ h
/ J+ J8 |* {, o% i7 I9 r
private:
        CUPnPImplWinServ& m_instance;
+ @) S" l) f# l2 Z' `1 |        LONG m_lRefCount;
};
- f0 Z$ x4 W$ K$ f% V

6 _# d) [2 A5 U4 W/////////////////////////////////////////////////
, u! s3 N6 f5 u) N+ {. \

使用时只需要使用抽象类的接口。
. r, u+ M: w+ f3 \CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
& d6 J" J, i2 ]CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.