UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   ; N& Z# Z. l9 u% Z
4. #pragma   once
   ! \+ R$ E# Z  F( x4 V! r
5. 
   ' S; b1 `* X- s" |
6. typedef   unsigned   long   ulong;
   8 u/ i) G! P" Q' X
7. 
   
8. class   MyUPnP
   
9. {
   % s  w3 V" i7 i& f  D! S
10. public:
    
11. typedef   enum{
    4 M# M' C. R2 p& Q  i9 e" t
12. UNAT_OK, //   Successfull
    % T6 \4 Y! O+ c8 [6 x; [. i& I4 [, E
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    : Y: E( ]  Y3 N  X$ o0 N) R
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    % E6 d( ]3 H7 h( g4 P5 d/ K
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    6 b; H2 l0 K8 G! D
18. 
    
19. typedef   enum{
    + y& a; f7 F. F4 L: W
20. UNAT_TCP, //   TCP   Protocol
    ! k/ I) ]2 b8 x1 Z' 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
    ( C6 j* g# [) `$ E
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    5 ]5 j, E$ s. n1 ^& y
29. }   UPNPNAT_MAPPING;
    
30. 
    & `2 Z! x5 p$ |9 L: e
31. MyUPnP();
    
32. ~MyUPnP();
    ! m# W4 j) z" t# L( C
33. 
    + S/ w8 a* G: C, I
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    2 ~( b" M6 v/ H6 ~! k0 S
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    - {5 R7 d' q, b; o' Q% N* w/ N
36. void   clearNATPortMapping();
    7 a0 O' U) x! Q- C/ G- h
37. 
    
38. CString GetLastError();
    9 t+ s/ ~& @7 _( z1 i' Y
39. CString GetLocalIPStr();
    % H. N# V" P0 N8 Z. E2 ]* \
40. WORD GetLocalIP();
    ' z8 C. E& x! P" k: y1 A9 t0 F% S! P+ ^
41. bool IsLANIP(WORD   nIP);
    
42. 
    ; s5 c$ N6 S* {; s
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    ! H! G0 }/ D( a
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    3 r' v+ |. X* Q4 [4 [
50. 
    . F5 W* o( i9 a( `2 E& \# w6 u
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
      B8 P7 S1 N) R# ~( h
52. 
    $ V# S9 l1 L# c# D) ^7 c: f
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    3 N3 X, D. w+ k
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    - O0 v0 v7 F9 D
59. bool InternalSearch(int   version);
    " {' s4 d7 ?/ u% u9 ?; m
60. CString m_devicename;
    4 ?; U& Z( O0 }  n+ w' _. {
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    & w% |3 T, L. `
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    ' R3 c  S5 h4 t
67. int m_version;
    
68. 
    
69. private:
    8 U% G9 @, Q% |
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
      W1 z) m) r0 I8 P' i
71. 
    + b+ K  u& C, q0 b1 M; ^
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    ! v. k$ R) y0 ?: y5 @, {4 l
75. 
    : ^, Y" a) b& Z0 Q; n
76. bool isSearched;
    / J7 |4 @6 c* U. \/ `- x/ n
77. };
    / N2 h# M& ~& n) s% K
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   4 n# v/ C5 r# _+ Y- U# ^3 M
3. 
   7 D, Q; B, W. d: ^/ }
4. #include   "upnp.h "
   1 d( F/ `; s. e/ `! l3 q
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   ; n. |6 m/ s8 C! O/ P& Z
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    0 C* J8 q6 c9 Q# j; Q( t# ~
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    " u2 v3 q% P- _8 u; Z  t( s
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    3 e8 X$ [, {+ K, [8 o; E$ J
16. const   CString   getString(int   i)
    / m9 A. C* c9 O7 N8 [& ]! p
17. {
    3 e" @5 f: X$ u1 S$ T
18. CString   s;
    7 q+ E% X" ]7 _- v
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    
23. }
    6 `+ H; v1 G# }% D  y, O
24. 
    9 Z6 I; r9 o! r% p6 ]
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    ! b; W; \1 v% J8 }
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    ' b* @7 _) M4 w/ |& [- d  G' L
31. {
    ; q" \6 n5 N: S4 E
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    # _0 |, F' \- @. [# H$ D8 u
33. }
    
34. 
      h* p, f8 Y% l5 U( w  V
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    
38. 
    ) u0 u: j$ S* Q9 Y* m
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    7 d! @/ ~, V" b) N1 |1 U$ J; Q( R$ \
42. 
    # g# j: s& Y2 U, j
43. uint32   ip   =   inet_addr(CStringA(addr));
    3 z6 s5 c7 Y3 v* Y
44. struct   sockaddr_in   sockaddr;
    5 \5 ^: \. p3 T) t
45. memset(&sockaddr,   0,   sizeof(sockaddr));
      l4 [3 a4 r1 D
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);
    % C8 J! {- u7 B
50. u_long   lv   =   1;
    % `) q! `1 e" }/ E" w
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    2 y" c( f+ Q" }. o/ U& {
55. Sleep(100);
    & l; Z7 `# b% G
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    , `, ^: J: Z4 G; E- X
59. if   (!rlen)   return   false;
    * R% ^* Y5 {: I7 j' ?
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    6 J- ~+ C+ v: Q. A, r
62. 
    
63. return   true;
    
64. }
    . D  D2 l: {5 y, E% y) I
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    # b2 _, Q9 Z8 ]' B4 b  Z* ^
69. 
    
70. const   CStringA   sa(request);
    9 W, Z0 _' `+ A
71. int   length   =   sa.GetLength();
    & n2 ?, h1 a! A$ n
72. strcpy(buffer,   (const   char*)sa);
    7 i% {8 R: T' f; b3 c* n
73. 
    ' t4 X' ?( x& c* V' W
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;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    8 E$ o: f$ ^# F: I2 l# c
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    1 s4 a6 \- ^7 b' b2 k9 E5 r& E! g
85. int   pos   =   0;
    
86. 
    , |6 E% x% O+ `" |" U
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    7 Y) D; j' ?( Q+ G5 G$ j5 S
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    : V+ }. o5 @3 K, Q- j* g
94. status   =   status.Tokenize(_T( "   "),   pos);
    . c& L% I; `$ @+ _4 _
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    , h9 U" x) Z, O! b. D$ a7 t
96. return   true;
    * E$ p5 a2 E( V/ i  \8 G
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     " a/ ^& F: A0 [8 U/ g
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     # |+ W+ r8 D- O2 T& I6 L) v: t
103. CString   property;
     ( G+ D4 g0 ^. ?8 K
104. 
     5 A; n  S1 X' f
105. int   posStart   =   all.Find(startTag);
     ( ?3 ^# n  k% `% R
106. if   (posStart <0)   return   CString();
     
107. 
     . Z5 P. t9 p* l5 b% n  D& C( t0 d: d$ v
108. int   posEnd   =   all.Find(endTag,   posStart);
     $ P2 d) x! Q% \6 X* y8 U0 V
109. if   (posStart> =posEnd)   return   CString();
     # F3 q& {5 C/ e) a7 Z* c
110. 
     * v7 C7 P4 W5 S
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
       c, I( ~! d+ O2 I; ]
112. }
     6 u4 s; J0 l5 R: i
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     8 q2 N6 X- ]/ M8 z0 C( Q7 x3 Z% d
118. isSearched   =   false;
     
119. }
     
120. 
     ; N' D! k9 P( f" h$ ]2 x
121. MyUPnP::~MyUPnP()
     1 W( `0 j4 v# N, M/ t0 j" V
122. {
     - J* Q! G' S/ ?. p# z6 _) W) y
123. UPNPNAT_MAPPING   search;
     ! y! S& X3 h* S) \% z- q
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     ' ^1 _7 [) k" r1 k0 D3 |# {! z
126. search   =   m_Mappings.GetNext(pos);
     ( [. k. `8 `; b
127. RemoveNATPortMapping(search,   false);
     . s# F( u: v: a" X& a8 ^' d
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     / j# {& s; x) O
131. }
     ; d  E! u8 t3 Y1 _; B+ ]9 p
132. 
     
133. 
     ) C8 c; u) s& t7 |6 f
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     + @* o6 i! n( b5 \/ n* L
136. if(version <=0)version   =   1;
     0 v8 h& ^( m) y0 ]* x9 ]7 a, d' W
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     6 g  \& o$ V! ]5 @3 n) u
142. {UPNPPORTMAP0,   _T( "service ")},
     / }# T8 N1 U$ n' l
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     9 O9 D9 |1 `6 p; @
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     . n' {: T  ^% i7 z- I2 h
147. u_long   lv   =   1;
     ( d  A* M5 G+ n# ?
148. ioctlsocket(s,   FIONBIO,   &lv);
     ) i2 A  q% u% d( I2 J1 P$ D  q
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     * w2 P% k- R/ B$ l. O$ R0 ^
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
       D3 K) g/ I+ i* A
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
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 "),
     
157. 6,   m_name);
     $ }* t+ n0 g% W. y( C8 I1 S
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     ) T) F6 f! P& O: U6 n
159. }
     
160. }
     
161. 
     " J& |$ F1 m7 C  ?
162. Sleep(10);
     8 k6 K5 E$ v( |( [8 m
163. 
     
164. char   buffer[10240];
     2 I5 v  R7 z% d6 t3 g
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     0 G, j# x) T& L6 Q
166. if   (rlen   <=   0)   continue;
     . A6 T4 g/ ]$ m, }: z
167. closesocket(s);
     # b& H3 N( O: N+ C: s
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
       ]* |, L4 z" j7 U, e( W7 i; f
171. if   (!parseHTTPResponse(response,   result))   return   false;
     ' \5 [# v2 N7 P4 P' _
172. 
     3 d" c+ R, y* b
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);
     + L0 [. T# C$ A; F% Q( b
175. if   (result.Find(m_name)   > =   0)   {
     : ^; V4 c( n3 w& C& Y
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     & H5 f0 P9 e+ V7 K" x* e# R) [. L
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     1 h8 |" k  q* I6 z
184. m_description.Trim();
     8 j" {4 h; a, l+ \7 ?* r1 T
185. return   GetDescription();
     + p5 C2 d: c& m% N
186. }
     
187. }
     
188. }
     
189. }
     
190. }
     7 A) c# P4 D. i+ g% ?# Z
191. closesocket(s);
     0 k/ @- ]; S* Y: l- n: }
192. 
     
193. return   false;
     
194. }
     

复制代码

天云上人

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


) D- n6 X$ ^6 v/ `* V///////////////////////////////////////////
8 G& i) z& i) |% n5 S//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
" m- u0 D& V$ _. \9 `
5 ~# u* `$ J) y
#pragma once
6 E4 N) Q- V! i$ _5 r8 q#include <exception>
4 [& o/ s: a# D; ]3 i0 W3 j

  enum TRISTATE{
/ [9 c* E* K; O. A/ b# S; E        TRIS_FALSE,
% M8 r5 O/ R/ F% w% B        TRIS_UNKNOWN,
9 l0 n  z- y  _  s/ Z+ t$ R, ~        TRIS_TRUE
};
! t2 `  K# T. `$ J3 {: q$ L$ p
) r! M( r7 [" G  |0 j$ q$ j3 a
( ^3 Z) p2 G1 _# a9 O# cenum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
0 h! o+ i! v2 I- ]: G/ i/ L$ u};

2 \/ [8 Y1 o3 C( _0 G9 t8 p+ e
1 n" i: a1 r# Z& N

class CUPnPImpl
/ r( E$ x  Q# z$ P3 l6 a{
3 W  W# d" A- M, Xpublic:
        CUPnPImpl();
) j4 G; X0 \3 Z, ^2 n; n4 G        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
( k: v& l( ^7 |/ ]8 \' V, ~        enum {
                UPNP_OK,
; {& \8 c4 S$ R                UPNP_FAILED,
/ R: Q9 O3 i8 c9 ~# u% k; F                UPNP_TIMEOUT
        };

6 T( d; ]9 Y/ K
& G# a4 c, O# |        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
5 U# s( u( m. d8 n- z. o, C9 V        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
! _, m/ [9 G: `; m2 U4 k$ s        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


9 D/ C* ?) u& Z6 x1 M        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
+ I; N8 p0 i" f2 Q# G) L        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
/ E6 L/ f4 U' {9 c- H: L        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
( @+ _1 v# y3 r

7 z; h5 [+ x4 ~2 o2 M& W9 x// Implementation
protected:
8 {+ d- }% T  \. V% }5 d* |        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
$ A9 G9 t' [; r9 ~' ^( R! a& L        uint16                                m_nTCPWebPort;
& k5 t1 [7 k$ r- z        bool                                m_bCheckAndRefresh;
/ U; z# W  |/ [/ I: t7 u- [/ F; o, R

! o0 x# D1 N2 q+ hprivate:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;

( m/ }) d, m8 ?3 T2 }3 U) X5 k
};

% A8 A  l8 G/ j6 ?
- D" ?7 x, V3 Y9 |) O% h5 w// Dummy Implementation to be used when no other implementation is available
$ N+ v. s, L3 w& K, Pclass CUPnPImplNone: public CUPnPImpl
{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
2 ?; j* }/ y* \& P5 G- L1 Q8 ]  ?        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
: C' L1 R+ t0 ^8 u! `/ }5 k        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
$ ?7 F8 P' b- X  m4 X; B        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
. R0 b6 ~2 |3 b4 F1 I};


/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
$ t; Z! v! U; v  S$ y0 A! B

#pragma once
; E  ~, Y$ ^2 h& T#pragma warning( disable: 4355 )

7 ]; f6 o5 G6 B: Q/ g
#include "UPnPImpl.h"
#include <upnp.h>
) c) T2 v! l4 d( B3 r7 S& y#include <iphlpapi.h>
2 E7 e. W# u8 a- G5 H4 E1 g#include <comdef.h>
7 y) k& Q- u1 J, C$ w& Z#include <winsvc.h>
$ [1 U8 X) N4 S# E
9 Q* I3 r8 P9 {2 G
6 D6 x$ }/ Z4 y& s2 }' S#include <vector>
#include <exception>
9 A" z$ U6 J+ P4 H#include <functional>

2 M4 {! j6 y5 B$ {8 \

4 C, u0 W/ C% l2 i4 j
" l6 Y- Q; U9 h. h# Ttypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
8 I: W! z& H. v% b6 V0 Htypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
0 U  x3 ~3 @; r# y) htypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
4 t) ^3 `% q7 D2 A6 \2 \
! J, H+ `. {, P/ z
typedef DWORD (WINAPI* TGetBestInterface) (
1 ]- o! f2 b- a0 ~4 g7 g) B  p  IPAddr dwDestAddr,
* z1 M; w8 O& s$ v" m% d  PDWORD pdwBestIfIndex
);
$ |  v* }3 b9 _# X# X
; V' ~0 I  v% P7 ^  ?
) V9 a4 J. Z4 A( Atypedef DWORD (WINAPI* TGetIpAddrTable) (
; d/ \$ g7 a2 h8 }) @  PMIB_IPADDRTABLE pIpAddrTable,
7 d0 @$ T1 i' B  PULONG pdwSize,
/ P6 F+ R1 u/ ^$ c+ Q# [' [  BOOL bOrder
# l" ?/ p/ C/ i' c. ~- s2 @/ ^);
$ `  w/ Y* }" _: {

typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
3 F: }/ f) w6 F7 B);


0 @; _7 [* ]% k" m7 E% ICString translateUPnPResult(HRESULT hr);
# Y" j6 H7 U6 c/ O. U) B2 JHRESULT UPnPMessage(HRESULT hr);
6 l. d2 `3 U" A& ]: V: R- t. Q

( X5 b4 o' r. s8 Y9 Iclass CUPnPImplWinServ: public CUPnPImpl
' i! t( c, A8 b" ^, v{
7 h+ q' M0 C- `3 T3 R        friend class CDeviceFinderCallback;
# {% [: W% K1 B7 }        friend class CServiceCallback;
// Construction
) m0 ]1 [, t- ~& Apublic:
9 D  j5 c0 ]' L- ~, w: N6 E        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
7 F' }8 q) y9 j. h/ D

% f1 i( A! v( ?/ o        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
7 F+ U% A5 q& f: z' t        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
( Z! v2 Z) c3 o2 g! S# U
' W, T- B0 S9 X& G" T
        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
% K$ P/ m" z6 e$ I; f2 c" ]. l        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
5 \7 v! Y. }6 z: W$ c        virtual bool        CheckAndRefresh()                                                                                { return false; };

+ b" S/ ]3 X( q2 b0 G% y" q, m
protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
  Z9 e  I0 Q( X2 ]5 e" H) ]: ]        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
0 g: W! d3 u% F% `5 L+ a( B' Q' i        void        RemoveDevice(CComBSTR bsUDN);
1 t  V$ n+ m  w5 z9 Z        bool        OnSearchComplete();
+ G% R" H9 a1 \* u' L+ H* |        void        Init();

6 u% D3 `& V3 g+ v8 L
        inline bool IsAsyncFindRunning()
- J# ^: U2 a/ l( E) j4 ]# V1 l        {
8 R- n) X2 N: a( c                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
; f  o4 ^. H; U1 U1 p                }
                MSG msg;
& g/ ?6 j% w2 h( E% ~0 T0 I                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
1 y" z2 k  k* [8 @+ H' U# T- R                {
                        TranslateMessage( &msg );
9 q$ `+ U  F, W/ c                        DispatchMessage( &msg );
, w0 |% B! }- X( [+ g* k                }
                return m_bAsyncFindRunning;
. R! J* S% W# Y4 _" f; ~& T5 S. v        }


        TRISTATE                        m_bUPnPDeviceConnected;

: |9 J* v* l" M- x" I/ m6 Z1 h. A
7 `0 I" b) P, \% h// Implementation
6 }7 C! e2 O! \" Y* G        // API functions
* ]6 [* s" T+ ?8 m( W        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
1 r% A3 n; H- i3 d' ~) `        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
' h4 q8 n6 u3 V) X0 }        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
: g  M2 {4 N6 ^& Q; T
! w( ^. n$ Q% d- f
% b$ M+ @9 k& P/ I* U. @" d* r* b        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
0 o, h4 j' Z, r. _5 l        TGetIfEntry                                m_pfGetIfEntry;
, H. P8 |( @9 C4 V
; z; ^& Q& W, ], x: {
$ R0 q+ Q# }  `. J. i$ [6 l$ C; }        static FinderPointer CreateFinderInstance();
4 q! j1 }+ ^  S! w# ^% ~        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
, }4 {9 x4 X+ c. i# M/ y3 L                {
+ {. Y* D: H3 D3 S. H& t                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


6 D6 {  K+ A+ W- ]                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;


                        return wcscmp( deviceName.m_str, m_udn ) == 0;
) A+ L1 x  T* \0 O" ]% v( r                }
                CComBSTR m_udn;
: E8 I& p6 U' }% u        };
4 k* F6 }; n* u# i* S6 ?$ Y       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
- i9 J/ Q5 y0 Z& k9 _& l/ D3 J6 t        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
4 X/ H1 P; [9 }' \0 N0 L" l        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
) i6 |+ z: }. k$ ~, j                LPCTSTR pszInArgString, CString& strResult);
/ [. F2 |) G$ y; B        void        StopUPnPService();
; [% c* y( A% Y/ J

        // Utility functions
; E; ], s6 G5 x) y% Q' r+ J        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
+ I. t$ O* A6 `. \# r8 e9 y        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
# I' T3 _. v) ]2 i        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
9 x& u2 V8 @$ N; m3 ]/ }        CString        GetLocalRoutableIP(ServicePointer pService);


// Private members
) W& i5 F6 o4 t0 ~. }private:
        DWORD        m_tLastEvent;        // When the last event was received?
- {7 E) ^5 c2 k5 E8 j1 t2 L! L        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
7 R4 d. |" M% Z5 x8 n        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
) `3 k% K* j' W$ K1 H

        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
8 V! M3 `% R: P/ b! B  Z4 B" n( Z        CString m_sExternalIP;
3 U* Y# q1 Z- F6 m  n1 V        bool        m_bADSL;                // Is the device ADSL?
  j3 ?* {7 ^5 z2 |        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
/ A- H+ @. d# w" a: X  e- h        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
7 Q7 ^! E& \- k        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
! k3 {- Q. Y* @; W* L& X; k        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;

" |- t& \# x- L* N; |% D
* w7 ~" u/ X: C/ L& d' t9 n  ?};


9 l% {, {% Q9 N% F6 k// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
  H  n% D. t7 ^$ ?8 m{
. Q1 N( Z/ H1 R" K0 k' Apublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
/ r4 D! j4 P5 a' o: v, J' |" J                : m_instance( instance )
' j4 t+ u% `$ _& e; B7 t2 c        { m_lRefCount = 0; }

) m: c6 @8 k" }: ]/ C3 a9 P# n* c
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
/ }& u( ?8 f6 O: p- k8 B; p   STDMETHODIMP_(ULONG) Release();
) ~0 M9 ^  p- }2 x

2 v( l/ C& ?5 `# q* L// implementation
private:
/ T% _- u6 l0 C7 x        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
& y0 i* A1 V7 Y1 D% U: K* I4 L" D

' q' [) K2 }# L3 ?private:
        CUPnPImplWinServ& m_instance;
% U5 S9 {& l4 M- _! d        LONG m_lRefCount;
};


// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
' x( `7 z; ?) s7 ^8 X{
public:
4 @- p' r  s6 @& \! H; \. Q        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
& u( f8 q: Z! m: Z! _0 f$ g- G        { m_lRefCount = 0; }
6 z! f8 l$ N6 E9 U: v& F6 X6 ?   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
9 {# \" |2 l4 Q7 L1 f# @   STDMETHODIMP_(ULONG) AddRef();
: \3 o: O7 ^& ?. \6 e   STDMETHODIMP_(ULONG) Release();
" D2 U# W  N$ H7 q! u% a7 I
, r, D2 K( ^3 m" h
! s, r0 ?) B/ S) V0 g& m8 ^$ ]// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
# v+ v: k$ e" @0 V% q; ?! T
0 X& e8 N! [: n, }) r- E
: ^6 h9 E' J4 B$ W+ f% Fprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};


3 S$ t! b; t/ U- @% t- s/////////////////////////////////////////////////
& R. K! ^7 X/ V& O8 N
3 p% `  u7 I) i  Z' X' x1 G
使用时只需要使用抽象类的接口。
' J3 j$ `! w' C$ r& M/ ECUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
5 @+ w1 y! A7 ?CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.