UPnP

天云上人

/*uPnP.h*/

1. 
   $ H1 J* d3 [% p3 K
2. #ifndef   MYUPNP_H_
   " i: b8 |; ?' J& X1 T2 s
3. 
   
4. #pragma   once
   6 C. M7 Q: q3 {( u$ _# ?' ]' a2 g
5. 
   
6. typedef   unsigned   long   ulong;
   ( T9 }- F/ s! p" c" X9 Q& `; X6 e! @
7. 
   
8. class   MyUPnP
   
9. {
   
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    # M( x' o+ l, L
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
    . i! O) c* O- }/ V1 \
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
      f" U) Z- t' ]( R2 ~2 f
17. }   UPNPNAT_RETURN;
    " M/ \' U) M) b# ?5 r
18. 
    + Y' U" J) p% a
19. typedef   enum{
    , C. k0 C2 x) R6 u& }6 S% b4 W
20. UNAT_TCP, //   TCP   Protocol
    5 Q; ^; e2 F0 }# Y( f
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    8 S2 c) ]* O6 ^) D+ X
23. 
    / h; f8 a5 ~  {. |) {
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    9 p' Q5 K5 a8 C* H6 o
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    8 S; ?5 t' a/ Y
29. }   UPNPNAT_MAPPING;
    
30. 
    ; N; U/ E! A$ G# v- C
31. MyUPnP();
    
32. ~MyUPnP();
    4 g$ ^* O2 ^# F! n2 F4 D& \% ^
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    , s9 \" D- u2 ?. t
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    3 X3 _3 x5 ?" ^! t- E4 H8 \
36. void   clearNATPortMapping();
    - ?- S. [6 F8 A1 W3 i  z
37. 
    + M% X8 Y) R8 u% j9 g) p. S
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    5 i" ^6 p; Q5 f0 E; n7 N
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    + u$ H: [3 |* v
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    $ A( {5 e& N; u
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. 
    & Q( x0 B" V, E# u+ v. N0 _2 K
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    7 d8 M, [1 h' A. P6 h& ^
52. 
    
53. bool Search(int   version=1);
    2 s9 h+ E0 Y. A) M% _
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    + I' ~" j5 q; |; u
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    + g1 c; _/ W" V* P& A  u% P
60. CString m_devicename;
      A. r! }7 o7 g# T  ~  M
61. CString m_name;
    8 H' [" |! T/ Z% a. I
62. CString m_description;
    " h" v6 I- r) Q7 V: g
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    7 W8 l$ H- f3 f" W& O) K8 n  Y  ]
66. CString m_modelname;
    
67. int m_version;
    2 I1 D* b6 X" l3 Z3 X3 m
68. 
    " Z) k# s: H  g3 Z- j5 {! S. O( m1 y# f
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    ( D! S* d2 P& i" M1 b* Z2 i
73. CString m_slastError;
    % {; g" G5 f$ \$ U5 U1 Q
74. WORD m_uLocalIP;
    2 F3 T3 I' o0 G# b
75. 
    . A$ {* L: ~8 K$ z/ O, s
76. bool isSearched;
    
77. };
    * u4 P! H# e' Y1 d2 `* |4 R
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   : e8 X9 J3 c( T' |
5. 
   - Y( Q, g; a( R" m8 F9 t  E# A
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   ' s% K+ d" i  l$ d" C. U
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   . {2 K$ F$ r( S! T
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    " @. e8 t+ k3 A  S  Y( W
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    + v+ i8 K1 V% H5 F7 P* y
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    3 n* N+ _9 u% W& F" e: K
16. const   CString   getString(int   i)
    ! t! m( S# [. S; Y. k) Y* O0 n
17. {
    ! j7 S  v  k  j
18. CString   s;
    3 T, c* R. X3 M- `2 s" t) }
19. 
    & M/ Z: E/ G; Q9 f
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    
23. }
    
24. 
    8 o9 J0 E: @6 H# S" c5 d
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    ' f4 W, F& Y$ d8 K  W; A7 e* G
26. {
    5 _$ y# ?9 z+ s7 V- D
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    : [. j( t" F, Z: i2 k
28. }
    / w6 B" k" M% y3 C+ H8 k- {
29. 
    9 g, R  c3 K+ x, s
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    2 t1 t2 W; @3 G! k) p
33. }
    
34. 
    $ m5 }& o" h+ h
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    1 u! n/ l, H1 J; W
36. {
    7 F2 U$ a' L9 r) v& P
37. char   buffer[10240];
    
38. 
    8 N% P+ j! q; ^. D
39. const   CStringA   sa(request);
    1 a7 W3 X* U9 V  I$ N
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    3 s: z2 X4 u* W
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    ' V; p6 T4 D  o; p  d
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    6 O( V) m% F2 i: r2 c
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    1 R  d6 u% @. F/ n$ A% w- `
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    , O5 q, C  z  I  O$ O* w4 K
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    1 |8 G9 t+ o+ F5 ]1 I+ ^# U
53. Sleep(20);
    + l' b- |0 Q* ]" S
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
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. 
    6 z+ }0 I4 S: p: X
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    : [# F% _- E) I9 }' b
64. }
    6 N+ S9 M+ u6 }! d# @$ S
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    ' }+ o: u- Y7 Z  _, z1 a
67. {
    
68. char   buffer[10240];
    4 P5 M7 K( }2 h0 E* ^+ r
69. 
    7 K; k, M! L9 P  z
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    + D4 z2 c- |" F: D9 L
73. 
    
74. struct   sockaddr_in   sockaddr;
    7 I3 A! \& m9 b" W
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    $ ?4 E( g5 E% {$ `/ d" I% W
76. sockaddr.sin_family   =   AF_INET;
    1 w3 I, X3 j: T! ?5 x4 D9 M( ?( C
77. sockaddr.sin_port   =   htons(port);
    + l7 y7 v9 w1 N
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    , O6 {$ J. J' n3 x% @3 r* j# q
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    , i" V! _4 S" T* r, b
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    3 e$ W$ u1 }8 ]- h. P& ]' }, J
84. {
    ( q) \. g+ T" k3 a7 d" P
85. int   pos   =   0;
    
86. 
    ; `7 l+ `, w0 D
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    . j# u- }* i6 u7 j( j' B
89. result   =   response;
    $ ?! C) F6 o& n) S9 P! T
90. result.Delete(0,   pos);
    0 o0 \& M) O% k# u
91. 
    8 }, @9 g1 G! p& I- C$ c
92. pos   =   0;
    ' f* {6 F/ N2 f7 r
93. status.Tokenize(_T( "   "),   pos);
    1 Y2 p( D( c8 Z0 |' a
94. status   =   status.Tokenize(_T( "   "),   pos);
    0 s- y- O* O! y. V0 y; T. u
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    6 g; V( Q( \3 e) {( n& c0 {6 Y
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     ( o% n1 ^' t& `+ `
101. CString   startTag   =   ' < '   +   name   +   '> ';
     1 W+ C) c# Q1 T4 R- c/ S
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     # j! C. s( P, x: \
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     # I' p- L3 E& V/ m/ i" `+ p( i6 o
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     ; O  O- `2 Y& o2 o4 N
112. }
     
113. 
     * i! E& l% }4 g* r
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     2 R4 O6 h* U. m: F7 y
119. }
     ; g$ `( }5 M' f& d0 Q: C
120. 
     
121. MyUPnP::~MyUPnP()
     0 D4 Z# U+ }4 t# d
122. {
     
123. UPNPNAT_MAPPING   search;
     ! \- d, b6 L& o
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     ( }- \2 [* \: S' \/ A6 s. g. h5 o
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     0 @7 Z3 T, A+ k5 ~3 k0 P
127. RemoveNATPortMapping(search,   false);
     " h2 X+ Z9 y; T/ e
128. }
     
129. 
     5 \$ f1 s2 d" ^" I
130. m_Mappings.RemoveAll();
     1 {1 x1 s3 R- J4 r- N
131. }
     " Z' d8 ^- ?5 T" k
132. 
     
133. 
     1 S: m, \# y5 _7 K. S, W' e
134. bool   MyUPnP::InternalSearch(int   version)
     9 _* R6 i* y( s. y) Y  r8 R
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     2 X; ?7 @( T7 ~' _
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     # o- e$ U/ ~4 J' _" U( J
144. };
     8 W  q5 ~6 |3 r$ [9 Q
145. 
     ' N0 D8 ^4 f  |0 i6 N+ }
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     4 t! g4 k( P# ]2 ^! ~( W+ ~
147. u_long   lv   =   1;
     ( H, O7 L5 @; j: b% Y: j
148. ioctlsocket(s,   FIONBIO,   &lv);
     ' y3 ]% e2 x0 \. v- b5 w& I* J' A/ C
149. 
     / f% B7 `3 K' h8 W; o5 V$ D7 U1 ?& ~) O
150. int   rlen   =   0;
     % P! k5 S9 c8 h
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     , {( R# G% c  R+ [
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);
     ' W: t/ e% n; B' q. G9 h
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);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     % T5 r# f, r# g& \- Y$ q
159. }
     
160. }
     8 Q- c$ I1 N% O. O/ S3 w
161. 
     
162. Sleep(10);
     5 Y) e* c8 s9 \5 i0 B
163. 
     6 X  D$ q2 I7 w
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     & v) d6 m6 b: ~3 X* b
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     % [; U8 x5 Q" e
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     6 E& d5 T% o) V5 {6 l% T
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;;)   {
     : [) R. }5 @2 _4 y. T
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();
     ! K5 T, F8 l: e/ D1 S; c
181. if   (name   ==   _T( "LOCATION: "))   {
     : A+ J3 s/ t: ?: L
182. line.Delete(0,   9);
     2 A$ Y; i$ R. G' q
183. m_description   =   line;
     3 S9 X; m3 C2 S* g- Y. y
184. m_description.Trim();
     1 s7 z+ f! Z' B+ m* Y4 u7 u
185. return   GetDescription();
     + g: w0 a( S$ n% O, M7 Q3 V
186. }
     " T0 U/ @) O3 u' J, e) x
187. }
     
188. }
     
189. }
     
190. }
     5 k, {+ \+ @7 f& p. N8 L
191. closesocket(s);
     6 R( o  W6 x2 Z( s# T: P/ r
192. 
     ) c5 I) t! p6 M" f
193. return   false;
     
194. }
       ]( d; F1 i5 N, g! w

复制代码

天云上人

以下有关upnp的接口来自emule，
! q) _. x( |: Q  E3 v! C7 \
+ P1 C3 m: e2 ?. x8 F) O7 U
7 N, U6 j' G0 F///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

5 l! M& Z, N" i! R5 }
#pragma once
9 \- Y, g2 {9 p1 V#include <exception>


  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
& ^, T0 H5 I$ @% a' P7 E9 }6 W, M! x};

4 ~& q2 B% e. S  i- d6 J
enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
( ~; K; o2 ]* e- P# Y        UPNP_IMPL_MINIUPNPLIB,
1 j0 X# V. U# p3 D% K* k9 v4 b        UPNP_IMPL_NONE /*last*/
};

) q9 h5 o, M' d$ V


class CUPnPImpl
+ c: P% {( i- n& M- O8 e0 X; W{
public:
7 b' d% \) d) e$ y1 U5 |        CUPnPImpl();
        virtual ~CUPnPImpl();
& ]1 I; C( p7 P* \9 G, ^: U        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
                UPNP_FAILED,
1 p7 s8 h' n0 V* ^6 G                UPNP_TIMEOUT
        };
" n: n- _! N% p3 J8 w6 J& Q- r1 o

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
$ q5 t) l# d$ ^, T. k        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
, ]: M5 s$ H" s        virtual bool        IsReady() = 0;
% A: w0 o% @: L        virtual int                GetImplementationID() = 0;
+ x' z% P& p: p       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
# @5 c$ w6 J+ R, l7 b9 J

        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
. H8 ?& J. A# G        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
" P% ^. `8 S. J/ v0 v6 E        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


; j$ f) }8 n! l( |. M1 ^" b& l// Implementation
protected:
8 x' T+ {2 ~3 a- i        volatile TRISTATE        m_bUPnPPortsForwarded;
  [# ]  ~' y/ }; o$ S, R- Y        void                                SendResultMessage();
( |. @0 K5 `- t' ]! r) ^' `* T        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
& C$ C# ]& o$ [6 ^        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;

' A( d# g2 A' s/ n
private:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;


};
7 a/ [" k$ \- Q% L

7 L9 q3 K6 j, Y" t) _// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
% e3 x7 w+ z) Vpublic:
4 @# ~' ^- {9 Q7 O$ S, o0 a        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
% Q: N% T! ?+ o! z* Z: L- D- v        virtual bool        IsReady()                                                                                                { return false; }
$ B3 }7 B& w+ E7 [/ [2 E        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};

7 v% l3 j( F: q7 y3 t
/////////////////////////////////////
, n% e3 q# i3 b0 v8 e2 T9 k//下面是使用windows操作系统自带的UPNP功能的子类


#pragma once
#pragma warning( disable: 4355 )
- L& P" V) D$ A
# U$ u2 Q* s$ s1 R
2 |- Q6 a) K& K- d" G; u#include "UPnPImpl.h"
: O+ ?- ?/ _# d9 \6 u" b#include <upnp.h>
#include <iphlpapi.h>
2 j- ^% c1 c' m; y. |#include <comdef.h>
#include <winsvc.h>


#include <vector>
#include <exception>
#include <functional>


/ X( M' W: [5 M0 C8 ~& C1 w3 |$ t
: z5 m' A0 ]; S4 h9 d
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
) w, T! k& W8 g5 W8 j' }typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
" B6 ]5 l/ D5 A. {2 ~; w( Ttypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
9 u1 L+ g3 z+ V- m) }; Wtypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

: j: g1 R3 W+ X" M' t
typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);
2 a1 o, O4 _2 A# e# N3 ]/ o

typedef DWORD (WINAPI* TGetIpAddrTable) (
4 D4 w4 [, Y2 d7 u: @/ M* j  PMIB_IPADDRTABLE pIpAddrTable,
. T+ A* n1 g! u( L1 I* b, B2 z  PULONG pdwSize,
  BOOL bOrder
);
7 I$ v7 C3 @0 H  }

4 `# o4 }' h# I% etypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
7 f) U* X* y' D  q& Z$ G);


CString translateUPnPResult(HRESULT hr);
( r: W# X5 L7 N& b8 S& aHRESULT UPnPMessage(HRESULT hr);

+ Z- c' m' D, a# W3 B
class CUPnPImplWinServ: public CUPnPImpl
4 a, D! G5 @% ~{
; U2 Y1 h, {/ a* u4 j7 A  n' ^4 U        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
  z* O0 y; a& ?4 ~( w% a) A// Construction
. C9 P' K, A) |4 |public:
        virtual ~CUPnPImplWinServ();
. g7 u: t# U! I* A        CUPnPImplWinServ();
( L. A: M, `) I0 D1 z/ G; Z! }* i
3 f  G( g4 B: a
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
/ S3 h' H1 u: ~0 P& t        virtual void        StopAsyncFind();
) g* k+ Z! o# i0 K        virtual void        DeletePorts();
/ g+ ?5 K" G$ s4 i. S; M        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
: R8 Y1 O9 I* l( S' L* H1 y! v$ w
9 S% l: ]; X; u( K0 u# w# I7 G. D% t
6 a8 B6 U3 V+ L& E" o5 N6 I8 Y) u- g        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
9 {8 v" x3 i* \+ J- t        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
6 B2 h, ]7 V6 g) x  b* \: b+ n        virtual bool        CheckAndRefresh()                                                                                { return false; };

* F" r2 S  N/ l  y
protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
+ u) P2 K. n" e  n+ ^4 A4 h        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
$ `/ ^. q) p4 i1 R        void        Init();


% N$ F8 I; ^/ ~/ x& A4 j' k        inline bool IsAsyncFindRunning()
        {
, ^8 e& t+ q" e; o& K                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
) U+ B1 D3 K& \                        m_bAsyncFindRunning = false;
                }
& T9 {) ]! m1 c' t                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
% W5 g- }& _  v6 D0 V& O                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
; J  d' ~! ^8 D/ C                }
                return m_bAsyncFindRunning;
4 ~2 n; y2 c; u/ I% V( j0 R        }

& J0 [) c( [8 f) n% v! x- e* o
/ Z  S7 c% ]4 ?        TRISTATE                        m_bUPnPDeviceConnected;
! F2 I! n+ v# I& v  N
. q  i! u' E: m+ A4 Z& K1 g  h
// Implementation
        // API functions
. f: _3 Y" Q' O$ X4 M1 f        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        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);
, m9 ?7 c0 w) q% B& U        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
; ~: z  }2 G, e3 l9 M3 _/ c  s        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
4 m; u1 L" K$ X+ G% L. m% p

        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
3 S6 T9 k2 g5 d9 g% G+ N! E

* G: S& A9 g4 X; e$ o        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
) }3 B2 j- [  b) N5 }& {5 A' U$ a        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


/ ]6 C3 R5 {2 h                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;


7 c$ Q& G" k3 P9 k                        return wcscmp( deviceName.m_str, m_udn ) == 0;
" i; B& f, h- }3 b9 D, O                }
! e, I# k- L- @% u                CComBSTR m_udn;
        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
0 G7 N) r( G; I8 u# h        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
# s2 w8 g6 ?5 Z6 U7 Z! w2 t) E4 b" I        void        DeleteExistingPortMappings(ServicePointer pService);
6 \8 K, Q) C/ J* d- Y1 p8 L, G# g        void        CreatePortMappings(ServicePointer pService);
% |4 E7 H' V$ p2 n2 z. @) I        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
* Y( H' u* j, F3 ?/ J5 m        void        StopUPnPService();
1 ^, U+ d: c4 e2 S# }+ K7 E

        // Utility functions
        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);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
, q6 n( ?5 e, z+ h6 U$ H5 C        CString        GetLocalRoutableIP(ServicePointer pService);
* b$ g6 ^5 e/ O* \2 s0 g) R3 c

0 {% N' v1 B9 F+ m- g9 X, L// Private members
- V4 S& Y& M% T: g! \' r, N$ f; bprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
8 k  H, |2 ~# B        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
6 W6 g7 b: D9 r' M  z! N4 ^* }$ _        FinderPointer                        m_pDeviceFinder;
2 Y1 c% M" W: ~. o* o        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
/ ^0 @: `) n3 w5 y4 P! [" G

' g- O; p, ]  t7 t6 j  m        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
/ T, h0 Q! P8 h( Y5 j        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
% z  {! A; \& a2 J2 Y        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
4 D- r7 r' j# v9 |        bool        m_bInited;
! p+ n* r/ b# t( I        bool        m_bAsyncFindRunning;
3 ^/ s$ o( i' @" W/ M$ c        HMODULE m_hADVAPI32_DLL;
1 v  W4 S: L! Z7 j# m: j; Y        HMODULE        m_hIPHLPAPI_DLL;
+ C: r$ s) m8 G3 V$ p5 F) p        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
& W3 b7 V7 g! ]3 c+ x! p5 g

# {4 ?! x$ v" m/ e& l8 h};

( n! E/ k3 K5 f# Y, S
1 }+ F& B/ t# |, d9 @2 P// DeviceFinder Callback
class CDeviceFinderCallback
" a7 t7 R; z! y( ~$ s( L" M3 W) t        : public IUPnPDeviceFinderCallback
& D" u$ M6 L- B2 i. `1 l/ U5 Q1 N% _{
2 u3 ^, H; S8 K3 M6 c2 Wpublic:
9 Z9 k( {; d9 S' I; w& s, X        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
: r; B  J4 F, I  ~6 s- o) v0 o

" C1 R, @2 D# J4 p& I   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
4 k! _- r5 U( Z5 {8 v; A9 b   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

2 U/ L. O5 D0 a& h) P4 j& V4 i9 U+ p
9 K+ L* s0 S; M7 \3 z// implementation
6 _8 V6 j  v3 N% e5 T/ pprivate:
) O% ~8 K2 {( E        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
' }3 Q+ r  f% a6 q, p7 i        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
( |: G/ ]) C# s3 N* d4 A4 Z) H

5 i: |3 K- ^: o. ~% t) }+ y: Aprivate:
* s4 B5 x! ^" A+ L- Q# j        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
& k5 m* f5 t; |  Q1 g- M3 \};
' y# h) ?" e0 H  R" c) d- e6 n
: @% ^( r, A9 O4 V$ b
// Service Callback
class CServiceCallback
4 u* j$ o, ?& i  W, t  }5 q& F        : public IUPnPServiceCallback
( K' {1 E* g. r, n, m3 K{
public:
. v4 C6 z) {, S7 \% Q$ O        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
5 t0 w% r0 M) m3 Y/ W( {        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
  q8 }+ @. i/ l3 u$ X% |   STDMETHODIMP_(ULONG) Release();
9 N% u% ?  f- a4 `- t5 E
  v* v1 \. P4 n) f/ _
( I+ O( Z* I1 Z// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

$ H3 l8 o) h  [/ D
) Y! ~/ N( Z) g$ \# W: oprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
0 \& z0 ^3 q0 z* c/ I4 X" c6 c" U};
6 W+ c5 p* {. Y$ W. @3 K
5 y4 ?: ]( q% ]& K: }$ M; q3 H
1 N* O+ V8 i9 \  P/////////////////////////////////////////////////


0 j* v  @; Y4 h/ |# X; Z5 O使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
0 Q  y! Y: o2 B4 WCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
0 s  w% q: }8 f# FCUPnPImpl::DeletePorts删除端口映射.