UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   2 z6 H# j2 Z( N& [- p% z9 L7 ~
3. 
   5 n% E8 u5 l& Q! \% D: a
4. #pragma   once
   $ d+ g( [+ N* H1 R% Q- ]
5. 
   
6. typedef   unsigned   long   ulong;
   ! M0 Z( ~0 O8 D8 Q: C
7. 
   ( o2 i' |1 F# j! m6 W1 W
8. class   MyUPnP
   - C: r4 u" W7 h' d3 w0 ~1 Z: {- u
9. {
   8 _) E$ \& H! z8 w
10. public:
    
11. typedef   enum{
    ; w/ W) G# S& h& G" E
12. UNAT_OK, //   Successfull
    3 p/ t0 u5 ~1 t( i; J
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    3 `! J/ q+ N, F& l9 S$ ?
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
    ) u! o- E' u' P7 E, x- [
17. }   UPNPNAT_RETURN;
    4 C4 b# H# A  l8 k. O' s  ?7 b
18. 
    ! E/ \# m( Z: B7 h
19. typedef   enum{
    ) A5 A$ T0 v" P: Q6 j, ], V5 g7 O( H
20. UNAT_TCP, //   TCP   Protocol
    * g7 j& ~$ m# f: r1 @- l/ t
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    7 c1 T/ w, |/ N! X/ ?4 Z
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    7 p; a3 P) M1 B8 ]3 @+ q% `3 X
26. WORD   externalPort; //   Port   mapping   external   port
    1 m& K6 y* y: o- V* u$ U+ n* m. }8 [
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    9 B" ], {# T0 b- s% l
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);
    0 l9 b" I! h% O& a$ u0 A
36. void   clearNATPortMapping();
    4 o6 F( X2 b( k& l& }
37. 
      S3 S7 \2 F% r3 Y  t: A# h
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    & ?$ N' n& ?$ o# _
40. WORD GetLocalIP();
    3 L, I8 Q9 e6 h$ {  J& w
41. bool IsLANIP(WORD   nIP);
    1 @  w9 W4 W& R' W! [7 h( L- _( L
42. 
    # T9 Z5 l+ p6 q2 `; d7 I
43. protected:
    
44. void InitLocalIP();
    ) b- U$ g+ G5 u7 Y' f' N0 p; [
45. void SetLastError(CString   error);
    1 ]# n7 _( Q. \! f
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    " X' v; @* [) A
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    ' K$ j5 C6 m# G. L- a
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    * M# g$ G+ t; ^6 l% e
52. 
    
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    ; p! A) S) q1 y% }
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    6 n6 ?* m$ U3 D
60. CString m_devicename;
    
61. CString m_name;
    3 l8 H! J2 o0 ^1 _% L
62. CString m_description;
    ! |* a* {) P4 c5 Z; g( k
63. CString m_baseurl;
    ( k6 x2 c: V6 t
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    5 B; ?! x  M7 D, P: m; G% `3 ]
67. int m_version;
    
68. 
    
69. private:
    ' H& Q; k9 m% G1 u6 ]; H! Q
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    8 Y4 U- ?( J- u! l  q
72. CString m_slocalIP;
    $ |& ]) `+ c5 P) u
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    ) k2 @/ D; t# H; M
75. 
    
76. bool isSearched;
    
77. };
    # |  U9 |% Q2 R
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   , L: ]2 _2 `9 f) c5 g+ e
4. #include   "upnp.h "
   
5. 
   . W! v. ?! s% z  j  w5 {# ?# D
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   : ?9 n; O; c* t2 B5 j' P! C1 s
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   ' t/ F) d6 y# M  Y
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   / S6 P# t9 h" O. V8 B
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    7 ~2 i5 L/ ]" H$ e' i
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    " ]# n/ I/ m8 K# b; W% z0 m
19. 
    ( F5 w& f  B/ m; ?1 Z
20. s.Format(_T( "%d "),   i);
    
21. 
    ; z8 P& v! t# `! n/ ~! M3 v' z$ v6 y" r
22. return   s;
    2 P4 `' o/ p$ j0 O% U
23. }
    
24. 
    % D% y8 w% \2 A1 z; @' D
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    ' e* E0 @* E% N5 k
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    , s8 t% c; S5 v3 Y) ~' `1 M& l! p) @6 A4 t
29. 
    1 @; [, D) F# m( M7 e2 F, V
30. const   CString   GetArgString(const   CString&   name,   int   value)
    & T) W( f$ O! H4 f9 v7 |
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    " Q" ]$ n7 |* `3 t' \
36. {
    ! \* p6 N$ E' V+ ]0 U
37. char   buffer[10240];
    ) s9 Q& k2 o8 ]& ^6 s/ H
38. 
    % o0 y  J) o! {
39. const   CStringA   sa(request);
    3 y6 M! J, |. O9 }9 c9 R
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    ; w* l* x' i1 s1 x" B
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);
    ' G, s: l& w  Q# g8 v
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    ! F( n6 M9 x/ B8 L8 h3 O
50. u_long   lv   =   1;
    $ @/ }) f6 J9 y4 q7 t
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    * i) g, q9 O& u$ ^. `8 Y
55. Sleep(100);
    4 J# ^9 U& o/ ~+ w
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    0 M' j" |* ?7 V* \6 S& v9 {! t5 G3 p
62. 
    % C5 b, X9 N. l% e
63. return   true;
    
64. }
    - @; S( W: \# O$ o6 V6 S1 G
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    ; }: f* Q3 D( e: k" _6 z9 x
67. {
    5 I" y' D# [' x; C; \
68. char   buffer[10240];
    
69. 
    5 B+ ?! J( o& l- A# J! ~
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    5 Q  b, O2 }- `1 ]3 U
72. strcpy(buffer,   (const   char*)sa);
    4 |" S7 Q" f/ h& ]7 P4 D
73. 
    
74. struct   sockaddr_in   sockaddr;
    , q( P& r" b& L7 S( Q
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    ; Q% |& L$ |  T+ Q
77. sockaddr.sin_port   =   htons(port);
    % f- g: D7 k2 f1 f0 K" t) ?
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ( a/ Z+ x% h+ |  y  a
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    " Y% `, a& L: M$ i7 w2 U
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    3 m3 x$ {( {9 `9 i( _' J$ N
84. {
    
85. int   pos   =   0;
    
86. 
    ' F8 [& x# s" C+ R+ `
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    & @3 U6 k* A" S1 k7 }8 d  U4 T
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    - {- Z6 d6 L0 F0 g& a, h
91. 
    1 m" p+ s5 j0 i! Z: s1 e
92. pos   =   0;
    4 ?0 X0 }& |, K; H& l. E6 Y
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    9 W* `* A$ s# X, w. X, L
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    % w5 K- F* n+ D6 q& P% z
97. }
    
98. 
    " b% I- f6 t3 p9 n# c! m
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     8 f2 F$ w/ n3 |9 q$ P
101. CString   startTag   =   ' < '   +   name   +   '> ';
     ( R9 [8 L" R: z/ w. Q; `
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
       o" v9 Y' I4 I* ]' g* K; x* g
103. CString   property;
     : H$ v4 ]( H' x! V, Y+ c! G
104. 
     % R! \9 J' ~  P9 o- L* x/ a
105. int   posStart   =   all.Find(startTag);
     ; x* l$ r# |& n, u' z) l% b
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     $ z! @# l5 O8 F+ `
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     + W3 v* F: _' K* d
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     / U) k1 F# x: v, J8 m( q
112. }
     
113. 
     9 F& a1 M5 x3 T6 I) o
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     6 t; D! ~: C# {+ [
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     $ X6 J1 k! }" A0 c) X3 k. b
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     0 x  Z0 s- p0 m6 y" j, Z
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     $ {  O. R" y) T+ @$ K4 C
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     ! v2 d. ?5 |' |0 C$ B$ P0 {
130. m_Mappings.RemoveAll();
     5 M+ R2 n6 C" I) d7 T! G! R0 o
131. }
     
132. 
       `: L) n8 S' L! s* r  A
133. 
     + G& u( u! p: D3 b; i, c3 b
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     3 Y; j% ?4 n0 R+ ]0 ?
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     9 W) [: k9 t9 ?
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     4 R* |; P' e" N
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     ( G% _- P5 Z- V# W
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     ( L0 t( C1 z9 U7 [: E5 \) N
150. int   rlen   =   0;
     " X% V, W' ]! L# H
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     " B( y" K% b! u/ L3 g( [
152. if   (!(i%100))   {
     " j% Q3 e9 ]" j, M. j  W
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;
     4 A& z1 j3 Q$ T; k) E. D+ r# G9 u3 I
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 @  a1 w+ D& {: @0 r
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     ) }4 j7 g1 P  ?6 T: z! c; J
159. }
     
160. }
     
161. 
     
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     . ~5 }" X4 M8 H0 z
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     + [% N% Y, w3 N6 m5 _9 H7 F4 g5 ^9 I& g; P
166. if   (rlen   <=   0)   continue;
     5 ~& v' d1 u. j
167. closesocket(s);
     4 \. f  S+ v! N8 a
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     ; F( b7 S( r2 X* V
170. CString   result;
     0 o& R. \2 m- M% x" y1 `/ w0 n
171. if   (!parseHTTPResponse(response,   result))   return   false;
     8 s3 q" z, b$ ]5 B9 e7 \* M3 W6 {
172. 
     . E; n1 G0 g- _
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     5 A  v; ~7 W$ C
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     ( }6 Q7 {+ O& H
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     $ G; [( _5 Y8 m& P8 B# _/ d
178. if   (line.IsEmpty())   return   false;
     * _  ]; e" I- M; G" s
179. CString   name   =   line.Mid(0,   9);
     ! G5 @4 n& g$ B: z. \: y
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     - X, M# {% o* q6 J# B
182. line.Delete(0,   9);
     
183. m_description   =   line;
     : s1 r  P' i  m
184. m_description.Trim();
     ( c7 u# \; o( d% V9 @
185. return   GetDescription();
     2 B# A/ j" ^2 K9 A, R
186. }
     ' g5 n2 u, s3 n( ~7 E/ o
187. }
     5 o& H- c* G( K8 H6 q
188. }
     * u$ E" B1 ~; s+ M3 \
189. }
     
190. }
     
191. closesocket(s);
     / {3 P+ ]' l* W" s0 I# t, `
192. 
     - g" X; k; @3 A& ]* a
193. return   false;
     ' a8 I! {# |; Y6 u
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
$ J$ f; X# `1 i' t0 m( y

7 C' X. P/ _( L  N///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
' o6 \, ?% \& E
3 l% o# N0 O3 o9 e+ Y
5 Z/ N9 I4 l- H/ v#pragma once
6 }+ `0 G4 Z, L/ N, o#include <exception>
1 E9 _: }9 ?8 {, h% i3 d9 U5 i
' w$ G: Z7 ~" @5 f( L. }
! d9 h' P! @7 M2 X( Z1 ?  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
};


( [) J" G! ~/ \- Kenum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
7 E: S* i9 @7 d2 K        UPNP_IMPL_NONE /*last*/
};


  M5 `' p, y: G0 D
5 c! n! b2 U2 a0 e8 b
2 g3 |2 j2 b2 i$ a: i8 dclass CUPnPImpl
{
- f' I* g% V5 r* v# R* b4 _public:
        CUPnPImpl();
        virtual ~CUPnPImpl();
$ b( \4 Q$ r2 ~        struct UPnPError : std::exception {};
        enum {
  s; s# k. n6 I* T* r6 ^                UPNP_OK,
                UPNP_FAILED,
' \- s; d  F2 x, }& E! z$ z                UPNP_TIMEOUT
$ l: W7 g+ Z: z+ E$ m' g        };

. {' R; j' Y$ y4 E
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
# @1 \0 `0 e* h, n" _        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
8 @2 w! t" o$ a" w

8 j5 ?$ u. L5 N        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


6 n) z: m7 w9 W( o* _& w// Implementation
protected:
3 u% a( I4 H' Y9 N        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
" ~4 X$ j4 R& [2 W* y% w* l        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
. Y$ k5 |. J8 [9 U4 j        uint16                                m_nTCPWebPort;
4 z. O5 {% ?# u        bool                                m_bCheckAndRefresh;

( g  N; f4 c# H, g' Q
' X2 P1 l+ }6 T! iprivate:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;


2 r' L* [9 x& D( P6 ]% n};


" ~5 T6 c2 m: S5 d8 c; |' l// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
  i; }  |" A. ^, x) \public:
& d' y% V- C5 D  T% }1 M        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
5 V' I* @& G, D/ w9 e        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
3 S9 D" b7 }% Q, D6 [        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
0 ]% ~5 @6 E! o7 _  ~; f% K$ c};
+ p/ ~5 I" C  W: L+ h! i

/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类

# u7 D$ d/ }. o, e- v  H$ f
#pragma once
#pragma warning( disable: 4355 )
& w$ t& }$ H6 o4 ]: u
) m" R1 D- x& G  e  h1 g
) Y! Q- a6 b1 w8 h# H" s! S#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>


#include <vector>
#include <exception>
#include <functional>
  Q+ a' q7 `. ]* o' C! }
- y1 E  P( }5 z- d/ P/ _


typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
. i0 T) J' G% e" _7 Mtypedef _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;
4 u- O* D+ g1 ?typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
/ o' b2 H; N# }/ Q4 Jtypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
: f' J6 r" V4 p( P, X+ c
; w( Y' s7 s- I+ o
typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);

, K& E- l5 I: \$ n; i
" o4 |# r3 s. n& ~) k4 ktypedef DWORD (WINAPI* TGetIpAddrTable) (
) {6 `# R! g* N" I/ W, Q, W, y* M  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
4 c# i; a3 i3 l. U- i  BOOL bOrder
);
  u5 u4 b. i5 n- F- W% R
# K5 P7 j4 u, W# d+ q4 s( z; r
typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);


, T, l, |5 Y9 T5 B3 ?+ kCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
2 P/ B1 k( y. A3 F4 d$ M) t5 o
) \2 {/ K$ D5 c, Q& A
class CUPnPImplWinServ: public CUPnPImpl
{
5 c+ y8 D- L& t        friend class CDeviceFinderCallback;
, c+ c$ c' u# A' n) }% u& q        friend class CServiceCallback;
+ @9 I% w: a4 v- [// Construction
+ Z7 A' [  I3 h5 _public:
        virtual ~CUPnPImplWinServ();
5 [! [* f& A1 T: K: M( w' X        CUPnPImplWinServ();
5 s7 h2 |# v9 {( C

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
; j$ g8 m: ^9 u" p        virtual void        StopAsyncFind();
- A3 C% l# H6 Z+ l* j4 [( O: y        virtual void        DeletePorts();
& J. L/ O& Q" _$ a- [" V' \        virtual bool        IsReady();
/ r3 o/ o% R7 q7 c: f# o7 s        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
        virtual bool        CheckAndRefresh()                                                                                { return false; };
5 b+ M( c- {; f0 X$ `; F/ u5 c+ c8 }
  }) P" d$ j+ ?- z
protected:
. @6 ^' E* ]; Z* j        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
  P/ D+ T2 R! u" g/ U( J' N' {        bool        OnSearchComplete();
0 B9 h- G: p& @! K5 }$ T) G( u        void        Init();
6 a1 b  `! E8 j5 h

6 t, |' O1 K7 T: G        inline bool IsAsyncFindRunning()
        {
) d. m/ e2 L& U9 V/ r1 W2 I! t                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
( [& B% @  v  T                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
9 y5 J& \+ B: u' z3 ~  t                return m_bAsyncFindRunning;
3 a: O) O9 ]" c% {+ s% B        }
3 Q4 U! _! ]: j1 N2 i  F" e

; D- e1 d6 N- j4 T5 Q+ b: A2 r        TRISTATE                        m_bUPnPDeviceConnected;
, }6 G$ D! [" E" t* h1 v6 }6 l( }

// Implementation
2 X/ K3 }# V. n        // API functions
# v/ w9 P. y2 K/ J2 v        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
" k, Y  k1 S9 S# L        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
) u+ z0 o/ }% \        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
; X+ L# c; P" A' M( R: j

+ s0 M7 f! ]7 b: t2 S0 {        TGetBestInterface                m_pfGetBestInterface;
2 S1 I7 r* L- J# R" N6 S$ h        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
+ W( V$ d/ Q# H$ W: T

: E4 [" E* u3 _1 Y* U        static FinderPointer CreateFinderInstance();
0 q+ U( M1 p  N0 v* u0 e        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
7 C+ j# m9 G, W3 Q$ L, u                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
6 y0 F9 T+ h; `/ O                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
6 p) q9 p+ A9 {3 |

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

- G; W! X* o8 s) r& |* W
" R- s+ H% t+ ?8 Y& F9 t                        return wcscmp( deviceName.m_str, m_udn ) == 0;
% d# d& Q( {9 h: {1 m* m                }
                CComBSTR m_udn;
        };
7 v) q+ N8 G2 z6 Z       
: Q' g6 f. }% I% N        void        ProcessAsyncFind(CComBSTR bsSearchType);
, `8 _2 B+ d) [4 a* A8 X9 R        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
5 T  w" y7 v8 |- X! C        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
( {$ V$ ?; ^. E. u4 N) |3 V6 p" S        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
; s# w/ ?1 _$ t                LPCTSTR pszInArgString, CString& strResult);
( Z1 a6 M. L: M% @: K0 Y        void        StopUPnPService();
& J+ ^# v5 @8 R2 L% [0 k( k

        // Utility functions
9 O! m8 Y( w$ b        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
* d( w! i! G* e/ O. }7 s7 i) g, @        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
2 G4 h7 l' |, g9 E        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
0 G# X; L5 r* O" j) ]& N$ o+ M- S        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
# i+ a7 ^0 [$ K* H9 i, x

; y# x2 v/ ?8 ]" Q5 @; b( W// Private members
' a2 y9 ?7 V& l6 {. d1 N* tprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
0 _& d% ~' ?. f! b+ z) n7 h        std::vector< ServicePointer > m_pServices;
3 _4 P! w; T6 w; @        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
! ?6 R5 Y% P# G; a

: D! h4 |$ S5 W9 w1 c/ I5 a. M        LONG        m_nAsyncFindHandle;
$ X! l) G( E' B* p        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
! X% H* |: X" d. a2 {        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
) K' y+ w: ~0 v& y        bool        m_bInited;
        bool        m_bAsyncFindRunning;
+ j0 |3 u1 H" O6 g- y+ f        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
  b2 N% R6 m& \; t        bool        m_bServiceStartedByEmule;
' T6 s( x1 I1 K" {        bool        m_bDisableWANIPSetup;
4 ~( t& Z( c8 |3 P; E        bool        m_bDisableWANPPPSetup;
+ }- }' J( o: @* l0 W

" D, e$ f+ e0 U2 s: f) x: }};
% A& N3 V- v" c$ R3 ]7 B+ S* h

: i4 }1 _4 B( J5 }0 N// DeviceFinder Callback
+ i; c! G6 R( m( b0 Eclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
5 k2 H) y- _+ S- o) H5 R                : m_instance( instance )
        { m_lRefCount = 0; }
5 a5 p4 ]. `0 m4 v3 |4 V' x

, T5 h9 k, t" b   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
7 a- N1 U3 B1 S9 Z3 l2 {1 k   STDMETHODIMP_(ULONG) Release();
# U  h5 T1 C4 C; u

! Y. m! k& I0 f1 P6 f! S// implementation
, g. W; y% C) Lprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);

2 q9 n2 r* h) j* l
* k' ~4 l) W7 j+ B& Pprivate:
: S# m: U+ p) p4 e# t2 u: c        CUPnPImplWinServ& m_instance;
! f5 N$ w! g  j! t: L1 S        LONG m_lRefCount;
};
# h* s" l" V7 @2 b) N
& ^2 `$ t0 v2 @- \0 `
// Service Callback
class CServiceCallback
( Z- J" z7 W; j; a  y! ~, e        : public IUPnPServiceCallback
{
+ A. J9 ?  \: I8 G  }7 @public:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
. @( O% f: U) ]. D8 _# g   STDMETHODIMP_(ULONG) Release();


// implementation
private:
: A; n" w9 @& t) ^0 t8 y        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
! V: s+ s' z  M& d4 ~- z1 Z. s; w/ d        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
2 H# E+ U4 H  x* E
; V' {, W% @+ q
, F8 p5 z4 K0 f/ A; q- a1 qprivate:
        CUPnPImplWinServ& m_instance;
6 q2 F  I. H1 }8 q1 H8 J/ y7 g        LONG m_lRefCount;
1 R+ B# ~% I5 I2 ~};


- a  k) |1 y8 Y" z# h- e/////////////////////////////////////////////////


8 @  w, U8 h3 s% P% k5 F. R0 o8 Z5 g使用时只需要使用抽象类的接口。
2 ~+ B0 i% `$ kCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
7 C- j0 o  v9 UCUPnPImpl::StopAsyncFind停止设备查找.
/ n, K2 j5 J# v- vCUPnPImpl::DeletePorts删除端口映射.