UPnP

天云上人

/*uPnP.h*/

1. 
   - e6 |% J, `* v
2. #ifndef   MYUPNP_H_
   3 n1 H# t- u1 s
3. 
   
4. #pragma   once
   ; V% p/ `5 D6 p9 {& Z; ^
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   4 W5 }3 |# g! N8 J! ?5 c* ~5 N
8. class   MyUPnP
   % Z! M, m# d) [/ o) y: _% X
9. {
   
10. public:
    
11. typedef   enum{
    , B4 V# @, N. ?4 J4 z- ~
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
    
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
    ( K/ r! d8 v# Y
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    0 \0 \5 g+ S+ Z1 d' x
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    $ e! j, ?) k8 Y& `8 d+ d9 o
22. }   UPNPNAT_PROTOCOL;
    * J" j* [1 r3 U. p, @
23. 
    , J6 P# I4 w$ R
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    ' l; w! `) [; n3 C+ a
26. WORD   externalPort; //   Port   mapping   external   port
    & g. ]7 B0 A9 ]5 X% Z) Q
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    ( F! g  @' O, x
30. 
    ) z3 x6 X* X; }; Z; @9 Q
31. MyUPnP();
    $ T8 Q; y6 ]- t& J5 t4 l4 d7 |
32. ~MyUPnP();
    - C$ F* d' M1 ]4 F* C
33. 
    2 C5 m5 b6 a5 }  b8 S
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    / ]; `/ T, K* n. q
36. void   clearNATPortMapping();
    4 ^! Y9 [* D9 V  b  e) n$ g
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    & B! H* h! K6 P7 N0 y
43. protected:
    , X+ i- F: x5 y" S) C
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
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. 
      O6 D. q4 @  B2 R9 r& A
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    / j0 a5 ]6 f2 {, t1 I# p) E5 D
52. 
    5 Q) b. J' \7 E; R
53. bool Search(int   version=1);
    
54. bool GetDescription();
    . \) Y+ ?9 Y+ o7 ^
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    1 |: _2 I9 }& [
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    2 y+ d4 u! P4 W/ {* P# m
59. bool InternalSearch(int   version);
    - W) ^0 f3 j; U( |5 |
60. CString m_devicename;
    5 q# y' I. i! E7 ]
61. CString m_name;
    
62. CString m_description;
    3 K: r0 u$ b% Z
63. CString m_baseurl;
    / w4 i( _' w3 g, t
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    / N  c& a# b  {2 E+ k: p- G
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    - I! }4 H9 P1 C  c; f$ H0 D' a
73. CString m_slastError;
    ! |4 K' U8 k+ p' {3 W+ p* u+ k1 P
74. WORD m_uLocalIP;
    3 b2 \6 ^* A- u4 E
75. 
    
76. bool isSearched;
    
77. };
    ' c" V$ ^% g7 s5 X" V
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   ! _$ g/ g7 _0 m: G
4. #include   "upnp.h "
   
5. 
   $ T1 E: J/ X, V# J: ^
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   ) K2 R! g9 _% @9 l7 @2 y3 b
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   : }: I, T0 A3 x
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    ) `" b1 W4 o8 P9 O1 h
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    ! I( S5 B0 o# _% o
13. static   const   int UPNPPORT   =   1900;
    ; v! n* o; J. `8 F, M$ `: a
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    7 u6 _# ^! N5 ~' j: X! b( b0 A+ n
15. 
    . L! ^2 O; F! `- W  Y; `
16. const   CString   getString(int   i)
    
17. {
    , U$ D2 F# t) ~7 c
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    , ?1 u# K- S" ?/ m- S. M
21. 
    
22. return   s;
    ! s: \- U3 k4 I! u5 l
23. }
    ' f5 ~! Z1 g! N: P( |
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    & C0 a: [0 a  k+ }2 X8 N
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    " X3 V" d: C" l" Y2 Z
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    8 a5 ]  y& v5 R0 t- C) R: _
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    # F+ D2 w7 {/ O% z/ e( e( y7 c  z0 ]
36. {
    
37. char   buffer[10240];
    6 l0 A# Z3 [1 b2 Q1 P: q2 w
38. 
    , n: Q( [8 J8 q
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    0 M2 M4 Q5 J. V8 S
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    # d" w  y; y4 ?" B: y8 S: Y
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    ; P* E# R+ e9 g/ f( k  U$ ]
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    # r6 x) ^  g4 G: A3 f
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    & L8 C0 I( z4 d7 r" t- n. G! L: v
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
      H/ @1 o* C$ G4 a: e
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    # W2 N# ?% J$ r1 I2 [
53. Sleep(20);
    8 C. ]* k- I  w; B- }
54. int   n   =   send(s,   buffer,   length,   0);
    , e' V1 {2 _2 y4 i
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    , t; z. _, y- J6 o
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    / V: v& r9 X/ G
61. response   =   CString(CStringA(buffer,   rlen));
    * z4 N5 a* \: M& G$ r) K& y
62. 
    
63. return   true;
    6 [$ n8 k+ g7 K8 K' C& y$ w- Q
64. }
    
65. 
    # r- T. Y( P" I3 a  A: q8 u
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    2 u0 }( ^) ?% z
69. 
    
70. const   CStringA   sa(request);
    5 `. v: y+ C* H, ^7 A
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    & M2 W; E+ F  w2 c% N% u9 d
73. 
      X$ Q3 Y# p/ q& Q' C
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    ' E# a* S( I) f, \/ h2 v9 i
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ( W: ]. i7 s% p9 f4 R7 w) u9 ], C
79. 
    . ^/ Q0 t7 p; z' E. v/ y% ?
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    2 N) X" V6 I% B" \
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    8 e; Y8 l, D# e) |5 n" c8 j
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    - D- ]: q" x& b9 ?2 R' ~" a
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    2 i# ^6 [. b+ e
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    ' o" y% m! u7 F$ H
97. }
    
98. 
    2 L9 K) D- k) u2 {5 U2 s
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     ' ~* G! H: D' o$ [; D; H: v
101. CString   startTag   =   ' < '   +   name   +   '> ';
     , t9 t; K1 g8 D# Z( v' u
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     5 m# r' T4 u9 |& k9 v( g  |
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
       ~8 Y$ `+ C4 R% j! u
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     ( J8 a; i) }" u5 m/ e6 U! i
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     . c9 A2 c9 F4 m
113. 
     5 \: T$ U6 \5 o1 r, j
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     + Q1 t" c+ @5 x8 n+ x: t, C! Q
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     3 [, q' W0 r3 U: F- p  ^
119. }
     . \/ G, k  P: @) x3 s, T
120. 
     % ~: {* o* z6 @
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     . U& @' z9 o$ P  O% m. H
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     / U8 |$ @1 ?+ X
127. RemoveNATPortMapping(search,   false);
     % X: V1 `& B1 I* S
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     ; [2 v) R! ?" L2 E  f( l
135. {
     $ b% m6 V3 ?0 p$ q6 j8 ]
136. if(version <=0)version   =   1;
     % v! c4 j8 Z) [7 \5 L
137. m_version   =   version;
     
138. 
     ) _$ s& u% L. w4 _  |; U0 Q
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     0 f" U9 Y9 C' v
142. {UPNPPORTMAP0,   _T( "service ")},
       B7 a4 o9 h$ n3 g# F4 v5 E
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     " {/ X3 C0 J- E: f) M7 k) y
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     % h. U) q  j4 o4 x
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     8 b: O9 ~+ h( W8 i4 T; a
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     3 E2 l1 T" O0 t. h
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
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 "),
       q% I( ]' i; S- T( i$ G
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     
161. 
     
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     7 ^3 I! c- b7 T1 m, t
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     ; @7 L7 R7 s- S. t/ D  K/ h
170. CString   result;
     5 p$ F# n; r, G: W% V- r2 y
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     4 T3 i8 R( K( S4 v
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     ( u. {8 G+ P$ R, r4 z
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     5 Y" x. c. Y% _. d4 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;
     / c" l+ A( g. n! s
179. CString   name   =   line.Mid(0,   9);
     $ {6 W' c, ?& W' a8 `2 S
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     3 ~! u3 Z9 }& ^) m& z9 J' I2 d
182. line.Delete(0,   9);
     / {% R# ]1 F, _- `
183. m_description   =   line;
     * s: w: L& e* Y% h, g( Q0 z
184. m_description.Trim();
     4 _5 x! Q! j" d) y4 f  y
185. return   GetDescription();
     ( v$ W# I  U4 z3 {: ]. @! o: I( p: g
186. }
     * f: ?% F! w3 b. _0 B. A
187. }
     ' m" [+ i9 o& h5 d. `& K
188. }
     ! g) a1 |, A& w3 k, A  c8 G
189. }
     6 Q/ f. e. E  e' |+ C( P/ f
190. }
     
191. closesocket(s);
     # X, I5 h( z, H/ B! h1 ?" v
192. 
     : ~6 [- Q- u( G& j1 W1 Z
193. return   false;
     
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
! Q$ y, b! C3 y3 n, _% h  d4 b
0 Y! L  \% Q# |2 i
! w6 v7 O4 w+ t3 g; T* l9 ]: }2 h///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


#pragma once
#include <exception>


  enum TRISTATE{
        TRIS_FALSE,
% a6 U; C' |% d; P. m+ k: Y" K        TRIS_UNKNOWN,
        TRIS_TRUE
};


enum UPNP_IMPLEMENTATION{
) P; M% q4 P9 M5 H        UPNP_IMPL_WINDOWSERVICE = 0,
; C/ y" I6 d! S! i        UPNP_IMPL_MINIUPNPLIB,
/ X- E1 e) J0 D$ b% L        UPNP_IMPL_NONE /*last*/
};




class CUPnPImpl
, A3 z% X% S/ B" s{
* @' w: c6 u# `4 T1 Z/ Ipublic:
        CUPnPImpl();
! s7 {# w' L  Y, ^; Q  N; i! b1 [        virtual ~CUPnPImpl();
$ }6 N7 b7 N# q" m3 [        struct UPnPError : std::exception {};
5 z% w9 n: X$ v' q/ O& K5 [        enum {
: U% `5 d/ h3 {6 C- D                UPNP_OK,
7 ?, Y6 d+ b& I                UPNP_FAILED,
                UPNP_TIMEOUT
        };

5 i. X0 b) u  a" y% \
* t4 p) B4 Y. X  C% p! r/ y, W! K; n        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
( m# S/ P8 D7 z; C, T# \" z* h        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
* x# `# D0 Z+ K4 q0 J+ a3 Z        virtual void        DeletePorts() = 0;
  }. J. y3 t  e3 ?        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
" _& @( h. y7 Y9 e& \& i       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
. K/ s, s9 u1 x7 E

/ L# V6 F8 u8 Y+ O$ |& _0 U" t' G        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
' D1 z! G# u7 I- [) ]9 g# h        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
- e6 \1 {, m) Z        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
5 f, b* G  W+ x6 e6 K

// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
1 ]0 c! ?( ~1 r" }: s9 J- x        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
  `4 o8 ?. P4 V) p" j$ a        bool                                m_bCheckAndRefresh;


private:
9 r5 w2 c# b: p8 z' ]$ @! ]        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;

+ M% w% F1 D" b  ]/ d
};
3 G1 e6 m. N; C4 W$ N: t0 p1 A+ l0 q
6 ?7 I* P" L3 d5 f* v
1 i" ~; L5 V, W* l  g// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
* o4 |: {1 [- w7 Y( h* |# h4 Y# Mpublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
7 ^7 Y/ P# J, ~* O5 m" T2 v        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
: |6 m9 a6 h; a        virtual bool        IsReady()                                                                                                { return false; }
2 o; Z: P, B$ u# e        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
+ z' L8 m5 x) [. D) |" h# y# `) e9 b; S
; p& ~7 [" y3 ^! b1 P- D
- V) ?  T' K: R/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
4 J; Q) N6 k# ]9 Y! S
) p8 y  \" v  B
#pragma once
#pragma warning( disable: 4355 )
8 f+ }5 V6 p6 X7 F
/ @. u% c9 n" Y; c
#include "UPnPImpl.h"
: y4 s  D, m  i% c+ s5 [: K1 e#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
! p) ^/ f  `* V. Q% ~#include <winsvc.h>


. A- E5 L+ l7 s8 E+ M. r# Z, b#include <vector>
#include <exception>
#include <functional>

2 s0 f( y  x' b5 m# p2 I9 b

" q( A8 P2 v! R& j; W: j
typedef _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;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
# W# q# k' M# ntypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
# S2 F$ d& a' H" Vtypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
" D  ]" [( L& e

typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);
. L3 h! Z, v! `; I. v  V

  Y1 C: @% E) Ptypedef DWORD (WINAPI* TGetIpAddrTable) (
9 Z# w  T" ?8 A, w+ V$ \" G2 l  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
, y2 e8 h! M' K1 L" [9 j; i  BOOL bOrder
; W7 D# R5 y2 ?  L$ j8 [: l/ j);


( }8 t* X5 ~* Mtypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
) z/ w7 W. p, D0 S& Z' t);
( }$ n. n1 @' _7 m) S9 x, \

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

/ B4 \" Q! G. E  x( {" E9 @& A) c) i
class CUPnPImplWinServ: public CUPnPImpl
{
' N" E* w+ C+ F; Z" T" Z2 e        friend class CDeviceFinderCallback;
3 D8 |7 }0 U8 q3 e, Z- w        friend class CServiceCallback;
; W7 Q. B7 n7 ~" m$ R0 g// Construction
public:
% ^9 w9 C; A- }* d, r' K        virtual ~CUPnPImplWinServ();
! p) G* S% P/ n4 j6 v        CUPnPImplWinServ();

; |* c8 _& a7 {0 {4 s
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
1 v; W  a3 Q* F/ W( U& Z        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


        // 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
% n& i7 }7 t9 ]- t+ C        virtual bool        CheckAndRefresh()                                                                                { return false; };
2 Z+ w. h- V( F" K; r- ^
* ^/ s4 e1 u& V( v# x6 q: i1 v# Z
% a4 ^9 E! U. J5 F6 Fprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
" A0 j0 I. t: l$ E# m        bool        OnSearchComplete();
        void        Init();

$ l% u9 {, v/ i$ g
        inline bool IsAsyncFindRunning()
2 p2 F& U5 P4 N6 E$ Y        {
+ k7 E8 o$ `+ i7 ~/ \+ }                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
- j! |- ~; Z1 I  o                {
$ d1 Y( m7 [5 q                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
% q4 y5 ]* [, C2 l: S) B8 B                        m_bAsyncFindRunning = false;
+ b3 a6 i. m; F. p) X3 y                }
/ @* C# F/ F4 T- w8 t5 |+ u* y$ v, w                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
( d! a% b8 M  N; f0 S                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
( `3 G, i1 Y1 x9 j                }
! F# P2 o$ @& A9 |/ q" a: V                return m_bAsyncFindRunning;
        }
' r) s: w, ?1 `$ \, p

$ |9 f# Z1 z/ w% w        TRISTATE                        m_bUPnPDeviceConnected;


3 g8 Y5 N2 n" Z, A- n/ g// Implementation
        // API functions
8 n, H' @; F. z  o. b4 `, R        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
5 ~; j) V0 g* R4 s        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);
) h) H# k$ A5 i, @/ @* i3 ~        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
0 ~$ k" T) T! \  d( q8 n' N) f4 s3 Z
% o4 G: O) g7 p3 K9 Z& ]3 Z
4 l/ P) q& V4 G        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
7 n2 F1 g0 A% K3 Q/ d4 u; m

) l- G! t7 T& i! Q% ~1 }        static FinderPointer CreateFinderInstance();
9 Y% F+ I+ G* f        struct FindDevice : std::unary_function< DevicePointer, bool >
' j8 z2 B( Y1 ?* v1 k( z4 \- t6 }5 t        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
/ ^& \: A& t6 K% y- E+ m" f                result_type operator()(argument_type device) const
                {
) l; p8 E3 [3 x                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
; d5 L9 V, T3 E) i6 c
# t$ [' @9 w9 m& {
                        if ( FAILED( hr ) )
2 R) A: K6 M& M) T% S  ^4 ?7 m                                return UPnPMessage( hr ), false;
$ g' u' N. J- P; P, V
1 R: ~: W8 A% X* u+ O+ u6 T1 e
) |0 F: S& t3 s/ \2 c                        return wcscmp( deviceName.m_str, m_udn ) == 0;
; J  Z" n) j6 H) \, T, x                }
" J: u/ s3 f% t3 u: k                CComBSTR m_udn;
4 Z2 q( }  V8 P, M) t( _4 U% U        };
% N  G8 E; C& J* C- ?3 c       
' D/ o8 B* e5 w/ d7 m7 M        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
+ c0 e, `: c: [0 Y' X7 a2 M        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
$ m* G# P+ u2 H. f/ o9 M        void        DeleteExistingPortMappings(ServicePointer pService);
% }5 k4 S, y9 s3 r0 {. k5 ?        void        CreatePortMappings(ServicePointer pService);
+ N* l/ q, J+ h9 @. t        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
+ a- b( m+ w1 c0 W+ x                LPCTSTR pszInArgString, CString& strResult);
0 I& `: w4 r7 R$ k6 r        void        StopUPnPService();


        // Utility functions
+ S3 s6 n/ f( u" t$ g7 t( e! F$ E        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
9 g2 t' {1 @  l* ]% l; [2 i" Z        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
" e! y: I; c) L& b" E$ k& R  E        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);
        CString        GetLocalRoutableIP(ServicePointer pService);
7 _* N7 i* F: _0 s
3 |9 i# V# g5 v4 B, A/ k% M
$ K2 @0 {/ p7 i( R( x: n  U/ c// Private members
private:
6 m! }( ?5 g: Q9 I/ s1 h) r# v        DWORD        m_tLastEvent;        // When the last event was received?
4 p9 \, g: O, t4 I+ ?        std::vector< DevicePointer >  m_pDevices;
: q' c8 R4 k+ t2 I' L- E        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
  L. g% X$ B% u; B        DeviceFinderCallback        m_pDeviceFinderCallback;
5 O/ n8 d2 Y: A& x        ServiceCallback                        m_pServiceCallback;
  I0 @4 }" `- d* a
" N$ \$ U# Q8 e9 j- Y- a8 X% f
        LONG        m_nAsyncFindHandle;
1 S6 _6 u- |# A9 p) t        bool        m_bCOM;
        bool        m_bPortIsFree;
% A6 v5 R& b  W        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
1 w% Q. s$ L2 H  H2 b7 j  j7 G* u        bool        m_bAsyncFindRunning;
1 Q) A4 P" @! m( x) Z  i/ `        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
5 ^- W5 y. L- G" Q$ H        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
( h# l" q7 N8 s, r: f' Q! E        bool        m_bDisableWANPPPSetup;
- H0 G' I0 [  K: j5 V, J

};

% j; o) V$ w0 T* u
9 t$ h( R  Z- C% W5 v) ?  z// DeviceFinder Callback
class CDeviceFinderCallback
: u! B. L/ Y0 z        : public IUPnPDeviceFinderCallback
{
public:
+ M+ e$ M9 k; f. e! d- |$ r        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }

9 U: Y( k" M9 K; L
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
6 g6 r% \9 W! y3 W% c% f5 U   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
) B) G, }7 X- w$ f; e

// implementation
8 A3 \6 b6 u) p7 g; iprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
3 G) ]! R4 j. w3 P0 M1 U6 r        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
. [6 _! `1 b2 y4 q' G. c        HRESULT __stdcall SearchComplete(LONG nFindData);
" ~% ]2 A1 Y# M0 J

6 L; e9 I" {) H( W' N- yprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
& W7 ?8 h, K  k: m
4 l& T' C3 ~( h2 ?. R4 a
* ^3 H+ B( M- m3 q7 b// Service Callback
/ L+ r% t( T  N7 s1 ?0 U8 Zclass CServiceCallback
+ C% W( |& q. I, e" o, {: a        : public IUPnPServiceCallback
{
& p; U1 h, D' Z  l; Gpublic:
        CServiceCallback(CUPnPImplWinServ& instance)
. y. K! K( M, N3 `4 A5 s                : m_instance( instance )
3 o0 @" h+ K! f* k& }( i- m        { m_lRefCount = 0; }
   
- Q0 Y3 V# A  ^$ ~% W2 ^   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
$ L# R+ h* E4 \- W( @0 [   STDMETHODIMP_(ULONG) AddRef();
4 G+ p) U: V# G  s4 B1 {0 V" {   STDMETHODIMP_(ULONG) Release();


// implementation
' H7 T6 I1 w) T4 ~, s$ @/ uprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
( h# E: P, v0 Q9 N0 l, e        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

6 q, j  p, _0 G, G5 X
private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};

, q  k- J8 t' i+ r" G  l% v1 q$ I
/////////////////////////////////////////////////
& \7 Y: l# ]! e5 I
' e; r7 e2 u# M) H% U# v
使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
( B: s. c9 A( B7 jCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.