UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   # b+ @5 c( b+ q! [& ?5 v' Z; h
3. 
   & t/ J3 l' J$ K( ^
4. #pragma   once
   / c/ e5 K) Y( h: F# A( f
5. 
   
6. typedef   unsigned   long   ulong;
   7 R9 v3 q: Q" f( {9 S
7. 
   
8. class   MyUPnP
   
9. {
   $ J' {* W9 O4 \1 f" A3 F
10. public:
    ) I: S1 L; o- E( q. p/ r8 K
11. typedef   enum{
    1 k% D' ?1 I6 S; e6 A8 G8 i4 z
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
      F( \3 Y) C; ^! C1 X
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    . x2 E# G- j: i& ?/ k# H& l
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
    ; q) e3 h  V1 d  j& W0 h
17. }   UPNPNAT_RETURN;
    
18. 
    ' j: z- ]1 P/ l3 D
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    & s1 ^' r! a; g3 p+ U6 P8 N: P
21. UNAT_UDP //   UDP   Protocol
    : l* f% W! Y; f" K
22. }   UPNPNAT_PROTOCOL;
    * e0 D$ A5 g) f7 k' I  [6 n* w$ z$ Y
23. 
    9 P4 f8 {% j) p  H9 O5 u
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    3 E' ~( G: `0 e
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    9 D: C. r8 u9 x  P6 m
28. CString   description; //   Port   mapping   description
    $ f( i! h7 E  T2 z
29. }   UPNPNAT_MAPPING;
    
30. 
    4 K; F& h$ J9 j' k8 e( e' ]
31. MyUPnP();
    
32. ~MyUPnP();
    + J& G# U" j5 N- h  H7 n2 r) K
33. 
    : k3 }5 c/ g$ Z5 M" y  d8 j
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    % W- c7 Y: z+ V4 V/ @, d
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    ' I0 r% B! m" o  y$ h9 _
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    & [6 A+ q7 v) W$ a+ w# N: r, V
43. protected:
    
44. void InitLocalIP();
    ( [; y2 ^/ V6 e% \, q, U
45. void SetLastError(CString   error);
    
46. 
    . X) c3 N! ?8 s
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
      m8 v; B" m* z$ _
48.       const   CString&   descri,   const   CString&   type);
    & I4 l( P3 F. @
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    3 G3 }. r1 F1 |. H' A4 P- B/ c' P
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    3 h+ D: j5 Z9 V8 Y3 t2 G# {5 c
52. 
    & A: k! F1 p6 g- r5 p
53. bool Search(int   version=1);
    , ]  E3 h0 Q# T
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());}
    
59. bool InternalSearch(int   version);
    . W  C4 u# M6 S$ x- ?1 }
60. CString m_devicename;
    % j5 f4 W5 ?0 b
61. CString m_name;
    4 R" U+ P8 B: d- s
62. CString m_description;
    
63. CString m_baseurl;
    ) Q, f; O  K* d" P
64. CString m_controlurl;
    ( z* x6 P1 g) g6 S; k. A; B. Y
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    . O) m, H8 @9 x  ^' l4 l
68. 
    
69. private:
      E% C3 B4 v: [# [" P4 a
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    ! O2 K3 @: `& D+ y1 p
71. 
    
72. CString m_slocalIP;
    $ k2 J2 Z$ q8 D$ d/ D8 a$ [
73. CString m_slastError;
    + W2 G' h" E/ p+ |0 T  ~0 G/ e6 u
74. WORD m_uLocalIP;
    
75. 
    8 F( j% V4 D; r! z
76. bool isSearched;
    / Z# B+ G5 K) L- N+ V+ l% L0 m
77. };
      W9 y" y, j' z5 O/ j
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
     H& g) Z% {4 @* F& o
2. #include   "stdafx.h "
   
3. 
   : W) j8 b. n$ a6 h) l0 m$ o9 |
4. #include   "upnp.h "
   
5. 
   . E/ g; \0 u  P) S! _' B
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   , l2 Q4 K( p  _- M! g7 l' M- R! Q& X) z
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   ! A9 |" V7 s3 e8 L4 t5 f
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   , `& @/ q+ i' M2 I$ P
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    - ^- U& W& h. o9 v$ b* @1 n
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    - W  r/ v( _+ f' c/ J7 E
19. 
    
20. s.Format(_T( "%d "),   i);
    , J9 k# R' z4 E+ j# F
21. 
    
22. return   s;
    0 F+ T; [' n; Y1 ^; L
23. }
    
24. 
    : ^7 ]1 c- c0 ?0 W* H
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    ( I; {1 G( j3 \. ]
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    : [5 z& L! @# P  e+ L, R
29. 
    ; b# L1 c5 C% d
30. const   CString   GetArgString(const   CString&   name,   int   value)
    7 _9 \) H5 h7 R
31. {
    & D6 F/ n; o) ~8 w5 v0 V3 J' x
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
      D6 E; D+ u2 D- m( R
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    " H% N8 |1 f+ H
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    # e* e! y" E6 o, `/ f+ P1 U
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    - \3 n! ~" t- \8 A. Q* g9 Q2 X$ D# q' E1 E
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    5 c5 e; S' p, y) Q9 r
46. sockaddr.sin_family   =   AF_INET;
    / \- c; g) F2 |$ y6 P# [
47. sockaddr.sin_port   =   htons(port);
    5 a, c) }4 R4 a8 o$ P! r
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);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    5 |0 ^* ^+ E8 n5 C* p0 X
53. Sleep(20);
    $ q9 C) i9 U1 y$ t- h9 j' p2 ]
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    . i( G$ Z3 D. W% ~; g& L
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    # T( i8 q! q: b& z
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    4 m3 a; \4 {9 L! Q8 Y
59. if   (!rlen)   return   false;
    
60. 
    1 H3 e7 ]6 c8 }: n
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    ; q# s6 X/ H3 l4 T7 p; z' y" N5 q1 Z
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    ) a; M* M  s( S8 G5 j/ B3 l
67. {
    
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
      s/ w( o& O. A3 b
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    0 g& z5 e( J  A+ S
73. 
    
74. struct   sockaddr_in   sockaddr;
    . w' g3 k* A( }9 f- n: j7 `: ?
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;
    " \; l5 }( z$ c0 j( y! l) _
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    $ f- R' Q) n' ?% [/ l: X  b
81. }
    / i1 C" U0 T: o: j- b
82. 
    9 {  R! g7 {, T. O/ X$ f8 H
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    - k4 u' F' L5 }( P' a4 H
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    6 V! Z& a" j/ i! A$ E
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    ' j* C- M9 V1 t( _7 O( }# F8 [
96. return   true;
    
97. }
    4 I8 z* N( j4 j
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    / n) |8 R0 B0 k
100. {
     4 Q3 @) q2 {) d
101. CString   startTag   =   ' < '   +   name   +   '> ';
     ) ^+ }8 H: @2 D1 L6 K
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     7 z. T/ m' S9 U7 h; z( O2 U$ q
104. 
     
105. int   posStart   =   all.Find(startTag);
     ( b. b: E3 r3 d* c+ T1 Z5 q# g7 e
106. if   (posStart <0)   return   CString();
     ' ?! q9 y, [/ W% E  l0 a; M* I( D
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     ! ^9 D1 \" ~; \: S) V* ~
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     2 E# J. e9 \' b3 z! g- I
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     : v) e3 G! }  O4 P" q, `
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     : R$ ^5 U3 q* ~# C
116. {
     " z1 R, E* ^% J6 m6 v8 z" \
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     6 }5 l# t1 X' Z/ s, T# b! a
120. 
     
121. MyUPnP::~MyUPnP()
     : t& @# k+ A, Z7 ~9 t3 c# z, b" }
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     8 ?, }9 m6 k* e4 c+ {! t
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     : g, z5 c  E2 k3 |- l. J) I
129. 
     $ L; o" s5 }6 I( e
130. m_Mappings.RemoveAll();
     1 W- C; p3 E( @1 `
131. }
     
132. 
     - y' p+ a8 ?! R$ N7 t% m9 B( G# u7 V
133. 
     3 g! z8 q& P4 s
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     : V" T; J2 `5 H  f
137. m_version   =   version;
     $ C" \' i. h$ D- O9 ]( v5 R3 o5 ^! ~
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     % E  Y2 ~- G! D* f( ?7 l& b" {
141. {UPNPPORTMAP1,   _T( "service ")},
     $ a: @, Z7 E- A" U8 R% ]4 Y; j% F
142. {UPNPPORTMAP0,   _T( "service ")},
     * {1 {) L! N( h( x* g; A
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     9 n% R3 B9 b  i+ U0 Y% X
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     5 @+ V. J6 [' h  j
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     + x) X7 R6 h$ e. S' y
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     ' l; {5 M' U1 [) Z* X
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     , _- }+ t) L( ?& \% {
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 "),
     ) `( s( V) r) k: X
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     , {$ s" @5 T! [& a1 [/ S/ x
160. }
     
161. 
     
162. Sleep(10);
     
163. 
     & u, Q! q, a% j1 T- ]& b
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     ( r  m' ?, ]# J$ g
166. if   (rlen   <=   0)   continue;
     / E% {" L  o+ }1 G# a$ r6 f
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     % i; ?6 S1 r# t
172. 
     + L& h! P) L1 e  \7 Y
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);
     
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;
     
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     2 x5 k' _) s' N9 x+ Z; K
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     
187. }
     
188. }
     5 A% g& o! K( c* H3 t; h; `& t
189. }
     & Z8 B: c* P% Y+ S* W( F6 Z0 B
190. }
     
191. closesocket(s);
     ' Y  F& @# F3 e) U8 c9 \; I
192. 
     
193. return   false;
     
194. }
     

复制代码

天云上人

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


% p1 {( v& H( l$ u& F: }! }///////////////////////////////////////////
. O3 F. Q7 M3 h! ~//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

* k7 J- ]9 R! k1 P; i- S
4 K$ i8 m( N/ n) h% w#pragma once
# {" U9 A, B6 t& p% j#include <exception>
. j- r2 @: x, G
' @  E; A; @1 q- P9 ?
  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
};
+ o+ a8 E8 o  ]# O
: h6 r" r1 Y. B
  _/ e5 L- z8 M* jenum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
3 w! l0 k4 G% G9 W+ m        UPNP_IMPL_MINIUPNPLIB,
- ~, S3 B& N' F% I/ R5 }        UPNP_IMPL_NONE /*last*/
};

' W& w+ Q" Y" Q# a# i) c4 D
: b6 B$ b! ]" O6 V0 b  ~

class CUPnPImpl
- |- e7 a7 ^* V7 N4 K{
# G6 j- b/ @6 ?public:
! r2 y1 Y$ `3 k0 }) ~) H        CUPnPImpl();
, h2 p1 a) M4 d        virtual ~CUPnPImpl();
* h; h& {% o1 e7 i: j% `5 ]3 L        struct UPnPError : std::exception {};
; C0 y4 \; F( o        enum {
+ E- N. J5 \- I7 n; t5 z                UPNP_OK,
                UPNP_FAILED,
6 N+ U3 g0 ]' C2 W" o7 |                UPNP_TIMEOUT
% U& @$ c/ ^) O+ i6 M* Q$ p+ F- E) y: u        };
9 T( G7 [  o2 ^4 O$ f

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
1 [- f( p0 R; m        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
4 r  W0 W2 i* O; w; `7 N, S3 C7 Q        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
, D. ?+ I6 b" }/ K% v! d: r
3 }+ P  C! `+ I& b; o* ?
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
& q0 s2 D( P8 [8 J1 l) m. r        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
" `/ O; \& L6 z; ^

/ k3 K4 s/ r1 i, ?/ h# w// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
# j# S) ]5 q6 h) r        uint16                                m_nTCPPort;
; a, S% ^4 k& `6 T) J  m        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
9 m, ?8 x6 E+ m) _

private:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
7 Q( P5 @% U- a+ u9 _0 i# v

1 S: n# q3 e# N};

( Z  N8 R- e" a
6 B9 X& y; k7 p8 T// Dummy Implementation to be used when no other implementation is available
, X9 e( u% h- X5 A* D1 y4 {class CUPnPImplNone: public CUPnPImpl
{
+ J0 X4 v4 {0 U; S. w' Qpublic:
: @' I) X# A6 A/ C" U        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
( d8 a. r. [! p$ Z7 p0 K# `        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
6 K) W0 C$ U  f        virtual bool        IsReady()                                                                                                { return false; }
; H8 Q1 S8 P" E        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
8 U# d( _/ a' Q, X6 h. x" D5 W
/ h5 q' F  c4 G3 o$ a; n
8 X9 C3 w, N1 D  X: ?/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类


#pragma once
#pragma warning( disable: 4355 )
7 D, h) T. ^4 Q1 y3 \. ~/ q5 c
: ]* \  D2 z$ R8 s" `# u: d
' k; R! e5 o& e% Y1 C( w) x#include "UPnPImpl.h"
  ~( a' Q/ `4 J#include <upnp.h>
# v- e7 m9 Y. ?3 k7 S( H8 T#include <iphlpapi.h>
#include <comdef.h>
/ l% Y; ]' s9 l/ Q#include <winsvc.h>


' o1 J+ L. y8 E( s#include <vector>
#include <exception>
#include <functional>

+ b" A* n( I. A( \

0 I% a) Q+ B) B5 \: U
, H3 R+ J* s, w1 A9 N. Stypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
  S" B7 B5 c  M% Q' c8 V* Etypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
+ R( L' I( }6 y; I/ q1 @typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
2 n5 F+ Q% p6 ^2 ~# Q) f% G( Q$ utypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

5 x6 m& ^2 x. `6 M: \: _( n
typedef DWORD (WINAPI* TGetBestInterface) (
1 {% {, b) L# E+ Y: s2 X  IPAddr dwDestAddr,
4 i$ R3 R$ {2 d0 s4 q  PDWORD pdwBestIfIndex
/ m/ _& n9 `1 z- s' i! z4 @& \);


typedef DWORD (WINAPI* TGetIpAddrTable) (
& W" ]) ~9 {! I' o9 a, }5 k  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
# j9 J  t% j& T1 p) [  BOOL bOrder
. R' U" T& t% W* z4 E; ^+ A);
" {0 M& @4 {. r8 f* t
9 \5 y3 ~0 b# O6 p
typedef DWORD (WINAPI* TGetIfEntry) (
8 A) ?% l9 q) B5 m  PMIB_IFROW pIfRow
);

* y/ p; X7 Y* M2 |, `, O2 p
CString translateUPnPResult(HRESULT hr);
' P4 l( S+ g  S. x0 fHRESULT UPnPMessage(HRESULT hr);


class CUPnPImplWinServ: public CUPnPImpl
; O, L4 _0 ^( T9 q+ c/ R' V{
: O, p6 D  m& E7 |4 M        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
3 I( \, {- S! ?' X// Construction
public:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();


0 Z. Y. y5 ~1 e8 b3 x6 e1 N        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
0 R: z, y6 @( P- O4 g1 V        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


3 B4 E6 t- R' m& v8 n3 Q        // 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; };
' [6 n8 [6 f% b. c4 T% @

3 t, W7 O& ^, A" Iprotected:
, @' `  R# [0 @- F% ?( y! X        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
  v2 U3 ]  }5 }; S) E. n* Z        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();


4 q2 e% g9 K3 Y        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
7 F0 }4 I0 A+ x1 m/ S  k( ^                {
3 A, |6 p0 M" [% X                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
# }) S1 J% p9 q- ]$ {                        m_bAsyncFindRunning = false;
                }
                MSG msg;
" ?4 Q) _+ c3 M3 I, |9 j                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
( b8 L" T+ ]% G4 ~  p- s                {
                        TranslateMessage( &msg );
, [5 D' w: D# l                        DispatchMessage( &msg );
                }
" K6 |) _; {: g, E; ]6 s, f                return m_bAsyncFindRunning;
+ {3 m; A: {: W+ g! P" p3 W        }
  j6 ?5 v+ u5 A: n' N( k
* J7 b" x4 q8 W* u& Z: {# P* z
. Z- V2 t* g1 f8 D- d; g        TRISTATE                        m_bUPnPDeviceConnected;


( f  Z& s- Z+ O& K% o// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
5 P. Q* i* P$ f        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
6 g: a- h: v8 W3 G2 t; V! Q2 b        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
- v7 ~. ^' P& N) @0 }$ J        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

: _: E( |9 ]$ v: v
        TGetBestInterface                m_pfGetBestInterface;
' i. n6 J9 _. }3 E        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;


        static FinderPointer CreateFinderInstance();
" a2 I+ Y0 a' p4 T  L: ?        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
7 @- w1 e% |9 w% v% X                result_type operator()(argument_type device) const
                {
5 r8 c* `$ R5 `( B0 h                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
& \+ \6 W# l) k+ ~6 T

                        if ( FAILED( hr ) )
. n' o- [' s% f                                return UPnPMessage( hr ), false;


+ X0 j, E4 q- [1 J/ \                        return wcscmp( deviceName.m_str, m_udn ) == 0;
$ ^" A/ j  O8 x8 m" H/ T& S( ~                }
6 {' r; g( w- O" |/ R& p/ Q! Y: I                CComBSTR m_udn;
2 C" a7 S0 b1 d9 `4 R/ u+ ~        };
       
1 m6 L4 Q4 R6 {' P, C! @) ]$ m7 }        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
* s& J4 D: f: Z        HRESULT        MapPort(const ServicePointer& service);
/ f: I2 H) I9 Q/ Q; l/ m( r' r        void        DeleteExistingPortMappings(ServicePointer pService);
2 b5 u" E3 ?6 C2 B5 f( N        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
; f( ~# Y5 A# J: \: N                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
, P. P- P7 U. Y6 ~% E; G

, k( Y# `- {7 H+ s- ?        // Utility functions
3 G# A1 g) ?. @, b+ U: A  t        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
% j7 f/ ^) D3 d3 d        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
" N  a5 P+ L5 L- b% t' t( z        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
- M* Q* ?' d4 E/ ?; j7 `& {        CString        GetLocalRoutableIP(ServicePointer pService);

2 z1 ]7 H" t; P% j! F( R
// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
. K: R% m1 n' E0 c5 @        std::vector< ServicePointer > m_pServices;
4 I9 N, l) I2 c* o        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
8 J/ z, a+ X% W" @0 K
3 ~/ a* f8 R, X& X$ M
        LONG        m_nAsyncFindHandle;
0 N% O1 b& W* V# K' |# [/ t        bool        m_bCOM;
& S$ F9 [. Z8 i* _        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
0 W6 z( Y  q0 W9 Q) M        bool        m_bADSL;                // Is the device ADSL?
5 N' E8 e* }- t* h9 K! {/ T        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
; F0 Z( @- z5 |1 |: D        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
2 z4 O# W9 W2 E: _1 G- P$ ^, g        HMODULE        m_hIPHLPAPI_DLL;
2 W/ n6 P3 I, E, C3 P* c        bool        m_bSecondTry;
- f) C# P& w1 ]        bool        m_bServiceStartedByEmule;
! w  v9 X, f9 w* q8 X        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
; E0 i6 y% }. a1 v7 |

};
* K( I' e" z$ c( v+ O1 X
; M$ G4 u$ K$ T6 q$ c; A9 a
// DeviceFinder Callback
! W5 E! h7 M, q) Eclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
6 ?' m6 I& w' I{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
9 V# `8 o2 Z* E5 j# Z+ d% a

   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
" e9 Y" E' b: z* j. J   STDMETHODIMP_(ULONG) AddRef();
0 W: I) [! s4 a- m0 r9 R( h- n9 u   STDMETHODIMP_(ULONG) Release();
' b  b% i% w  Q: @: l4 J$ G7 u! t
) b- Y8 v- S# s! a. T
- T8 [' r: G. O// implementation
) d: z3 T6 H0 R3 t& p+ Fprivate:
' {+ R) M1 B* e( v: e7 q: K        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
) U( r6 [+ r( E, w- R        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
9 q' f! c( z. _0 q" C. R        HRESULT __stdcall SearchComplete(LONG nFindData);
" K) _( h2 ]" d( y  [

private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
" @3 k6 E  l9 e, z" Y: h+ k
7 p* D- h! J) J3 u$ x9 A& a2 B
( U/ \* U+ s. n- ~) O// Service Callback
( N( X! M) o1 t* |/ p/ _: z  U* y6 D  wclass CServiceCallback
7 v: Z2 s' {" i) _8 D+ f4 V3 g( F# D        : public IUPnPServiceCallback
  x( N  ?: _5 z" R% i2 a' ~{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
3 y$ B0 F4 G) }) n; t   STDMETHODIMP_(ULONG) AddRef();
9 C. h" Z0 f, Q6 s# p2 A# z   STDMETHODIMP_(ULONG) Release();


// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
. H% U; A. d+ l2 _7 h: M

4 @: l6 H1 u, y# Z# ^private:
        CUPnPImplWinServ& m_instance;
/ |1 |) {7 t: z/ y3 H        LONG m_lRefCount;
};
" H) ]: J/ w5 u
" P3 M/ n( J% q& V
/////////////////////////////////////////////////

# ?5 i7 \5 `4 j; ?4 N+ w* T; J
+ G1 [# i; ~) y6 P- f! @4 A使用时只需要使用抽象类的接口。
0 m8 x! u8 W& [$ y1 t) M8 WCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
8 c/ S! j; m% XCUPnPImpl::StopAsyncFind停止设备查找.
4 _2 ?' H) [3 _( B, A6 d) HCUPnPImpl::DeletePorts删除端口映射.