UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   ( z# C( v9 l" U4 P0 k, {( ?
3. 
   
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   * b* y6 f) v( }. D* K* P5 d( V
7. 
   
8. class   MyUPnP
   9 c. z4 N6 M! J+ u
9. {
   
10. public:
    
11. typedef   enum{
    # P, c- }1 O  h9 ^1 L% ^- f
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
    ( P( s) T) O% D: U! d
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    * f6 b+ e) L6 J
18. 
    
19. typedef   enum{
      ~3 E# @# T8 @. N: K
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    5 X& C3 N! P, H
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    $ r9 P. Z7 G' r; U# h
30. 
    + O, a, ?8 o* H3 g1 a  n+ p0 C$ a% u
31. MyUPnP();
    ( d; v, V& s, @# h* l
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    9 `3 y7 D/ ^! v8 r
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    1 C3 S+ q# V8 K
36. void   clearNATPortMapping();
    ; b: ?' v# q. m  M+ g
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    2 W0 G& y; s) w% s  x2 y+ y
40. WORD GetLocalIP();
    , t' @# J) H- B9 @
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    0 N5 i& ^7 T: u( v8 |9 b
44. void InitLocalIP();
    9 w+ N3 ^, z  U9 V* S; Q* W
45. void SetLastError(CString   error);
    
46. 
    6 }1 S4 ^5 z0 s2 L1 r! E/ z
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    ; `( u4 N6 y/ B( [3 J/ S
48.       const   CString&   descri,   const   CString&   type);
    4 H0 j: r4 ?, N; x6 D* H
49. bool   deletePortmap(int   eport,   const   CString&   type);
    4 K2 J% a: _, y* i8 E
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    # ?) d% {. {: Q. v
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    2 e8 n; Y9 f( U) c$ c- Y  L1 o
57. 
    " h* j, j9 A- ^8 E! ]
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    , |8 e  @: X/ N- d1 I
61. CString m_name;
    
62. CString m_description;
    2 Y5 o1 S6 h( Z) M. S$ h# i
63. CString m_baseurl;
    1 H- L2 V+ ~7 }) v/ a
64. CString m_controlurl;
    0 Y; o, O3 e: F$ @* b
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    + Y: B: Q, c3 x: n2 K) e, m
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    ' P) f( F7 r: q3 w
75. 
    
76. bool isSearched;
    ( L6 j$ N" q+ _# w0 h7 I: C
77. };
    1 H0 p7 V$ X/ f% t/ d
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   2 i# }7 }4 T' b
3. 
   & W" K# @( N+ [% n. }
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   ; S" s9 V9 l  o4 Z' u* z
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   ( j- u6 ]9 s% r3 X! k1 I, O- Y
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    + O: m- q. G+ ^+ x; k+ ^/ N9 I, O
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    7 F; k+ r6 [1 Z# Z1 G
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    " J: U& C0 }3 w6 Z, N
15. 
    7 z3 n" i) Y7 N) `4 ?
16. const   CString   getString(int   i)
    " k2 V- S; E; w) u& C
17. {
    
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    & h! n, @. R0 W4 ^! L& v/ u5 X
21. 
    . V# V7 v, a4 e! M5 @0 U, R4 d1 [
22. return   s;
    
23. }
    $ s7 [: q, z+ l" w1 z
24. 
    ' i# S# I! N1 h* V
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    3 b% b/ X9 E: |3 h7 ~0 u3 I+ q
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    $ H9 N) Q" X* G* \$ t2 n
34. 
    * j" P- |/ u; b, Y: p' F
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    ; K- G, |( p& o+ ~9 I- B& A
36. {
    
37. char   buffer[10240];
    6 E7 p. l! L" g# P& Q$ G8 {2 X
38. 
    7 n, R9 f3 z$ ]' C! A& l! u) K* U8 u
39. const   CStringA   sa(request);
    5 V  |- y; @  G5 W. y
40. int   length   =   sa.GetLength();
    " L# G: m* v4 N2 v, Y9 N
41. strcpy(buffer,   (const   char*)sa);
    
42. 
      n2 \9 l" J) `
43. uint32   ip   =   inet_addr(CStringA(addr));
    : B, C/ x' P+ D0 i! z; `1 @
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    4 V  {7 w9 `" C$ }
46. sockaddr.sin_family   =   AF_INET;
    : ?  z! B- S4 X8 s! Z
47. sockaddr.sin_port   =   htons(port);
    ' {: i7 D  x, t- [) N( \! Q
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    1 t9 }1 s/ |! v! c6 d4 d1 o
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    & Z- @- Y/ Q- U$ j
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    1 E- O9 H' ^# z: `5 {6 ?& L, n
57. closesocket(s);
    : C. e( b7 N6 ^1 D+ `1 \
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
      \5 g7 a) f% G: x3 o5 {
59. if   (!rlen)   return   false;
    
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    ! _2 D' j( ^# S7 e! h$ p# [
69. 
    
70. const   CStringA   sa(request);
    5 W4 Q& A+ h; X
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
      c0 U, y8 n" W0 R. O3 ?
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    " a" C% h6 z: Z! c# w# o% k- I
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    " D# [; s/ n+ p$ j/ K
81. }
    
82. 
    : _3 W$ b3 u0 e
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    ! Y  n/ ?% k1 L
84. {
    0 }% o9 }5 C1 v( n; B  y
85. int   pos   =   0;
    + T- f2 `/ R% ^7 J; f8 {, R- n' O
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    5 B9 p) H# }5 Y& o6 a" i. A+ J
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    & c$ U' M1 Y3 l
92. pos   =   0;
    ' A2 u& _" F, H: \! Y
93. status.Tokenize(_T( "   "),   pos);
    % q1 A8 N. i- L+ Z
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    3 f2 V. m8 g  f) X7 _; J% Q
96. return   true;
    ; X7 ^( M7 W' @. T. L
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     : b; b% W6 C# L) Y* g; I) A+ Y) f
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     1 N( l* Y' ]; S1 q* h, l; F+ Y9 X
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     0 b& e5 w7 C, A- {& x/ K/ p' S
109. if   (posStart> =posEnd)   return   CString();
       t7 ~- ^, r& ?0 R, Q# D* W# s
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     + x3 p  u) y: @/ x: p
112. }
     ( s+ \- y8 {$ a; t4 J; U/ `
113. 
     ) `, w- L9 K5 o1 k) b9 c) I3 a
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     - ]# d# m6 i) V( `2 S% |, t
118. isSearched   =   false;
     
119. }
     9 N/ {( T$ Q0 @* d  ^; T
120. 
     / T5 W5 O% i( q/ F& R; c: ?
121. MyUPnP::~MyUPnP()
     & a8 R" g1 ]- O: B, g' }
122. {
     
123. UPNPNAT_MAPPING   search;
     9 H/ K' f% @' P
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     ' d0 W0 s& o+ `) v0 d( @) O7 R
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     / ?4 ?$ l+ J2 h' W' e
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     6 D) y, Z0 r) R: {3 n! O) V! T
137. m_version   =   version;
     # \9 m/ T. V1 ?9 J! K, D
138. 
     
139. #define   NUMBEROFDEVICES 2
     . [; e6 I$ ]8 p* F# U) t7 |- J
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     % M. x1 [. U6 D- L
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     + g3 D8 H8 Q8 Y+ f' G8 V) X
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     $ b0 C, h" [+ I, Y. N( |) I3 W" u7 N
152. if   (!(i%100))   {
     5 Z" c2 c3 `! D: v$ ~/ g( }
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 "),
     
157. 6,   m_name);
     * j; z6 _. @* j) C/ K2 `3 _6 l( y
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
       }& O$ H6 g* c/ a6 Q' M
160. }
     
161. 
     
162. Sleep(10);
     9 o$ N* W" c# ^# g; W' Q% l
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     % {) E* R  a4 \# h/ ?3 S
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     & M# V' j9 J' L# x: O0 r
168. 
       F* q: d& P% T7 J! G
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     
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);
     ; @9 r# J! e9 n1 l4 t8 H3 W
175. if   (result.Find(m_name)   > =   0)   {
     5 w6 K5 j# \2 j
176. for   (int   pos   =   0;;)   {
     + a, }" A; V2 U8 H& s( F
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     5 e2 s. x8 E* G4 @/ a' I
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     ' Q$ Y1 L6 l6 c0 Q/ i' b
181. if   (name   ==   _T( "LOCATION: "))   {
     ' a+ N5 i9 C: V' F. D1 B6 @
182. line.Delete(0,   9);
     7 @5 }5 H; ?/ F! U: O6 c3 T
183. m_description   =   line;
     : r9 H# d, J7 S2 }4 H
184. m_description.Trim();
     . N! b3 h* f* b5 S6 J2 G
185. return   GetDescription();
     6 I- b7 C$ y; R' T5 V  |! T  _1 Y
186. }
     9 c( x6 @( @/ r4 L6 G
187. }
     
188. }
     0 F$ e: O$ h" b' F2 |$ r1 d2 J; |
189. }
     0 J$ a5 M8 j' _6 |$ u! v8 c
190. }
     
191. closesocket(s);
     
192. 
     
193. return   false;
     
194. }
     3 \; C% X5 k9 e: h# p: ]

复制代码

天云上人

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


4 c+ p1 Q4 q) h4 H3 ^5 B///////////////////////////////////////////
* v* ]' ^0 b$ E" f//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
7 Q; \: v) o% n; Y2 \6 o( ?  b8 A

#pragma once
* y; b$ B5 Q  @7 S#include <exception>


  enum TRISTATE{
+ v! s7 M- i& q- j8 r/ N- `        TRIS_FALSE,
0 q) W, Y( V# q/ J# g3 |        TRIS_UNKNOWN,
5 r# Z. l  H+ A) W0 g        TRIS_TRUE
8 ^. \  ^, i% L; _5 j( X# s9 [};
; z( l, V. \4 I% R, G% U' g
8 j% K# D8 B! S- N' D  i+ s4 I
enum UPNP_IMPLEMENTATION{
5 ]) N9 |3 O7 {7 M2 J$ G$ c- A/ X        UPNP_IMPL_WINDOWSERVICE = 0,
- {, f' _# G2 Q4 h2 I: z5 s        UPNP_IMPL_MINIUPNPLIB,
. A4 A6 j( ]2 P2 s; \9 k" c        UPNP_IMPL_NONE /*last*/
' J! @! X- Z( m  U) N9 o};
2 l* \$ v/ p0 D' Z/ F


6 h( [7 H. X: o8 K% e$ ?# X4 _
& I& Q) Q, J+ w, _+ Y; N; x+ o9 Tclass CUPnPImpl
& o( E2 r/ W0 e1 f2 V8 L& n! s5 V; K{
# t, r' K7 Y5 H: b3 k! |& O$ gpublic:
/ N1 }0 y- ^2 R( }! R; k* m) }- Z        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
                UPNP_FAILED,
/ K5 b! T- b+ p                UPNP_TIMEOUT
        };

9 B1 I. N: h" Z* T* t8 e
; v: u0 `- Y7 O        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
0 ^  m5 a& F; o1 d# ]. g& J        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
- ]& |2 X& b& K; U        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
% @! c9 e9 F# o6 T, F' D% ]+ B

        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
$ Z/ H9 e5 r# |9 H! Y+ _* z        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
* r4 E; G  ~: R1 |        void                                SendResultMessage();
8 s, t0 E. F# @$ u3 m( b        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
6 t# J, @4 r# C        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;


5 z2 z8 }4 N1 V& i; dprivate:
! v* h: b9 P2 O        HWND        m_hResultMessageWindow;
" K9 n- c- C5 V1 }+ @        UINT        m_nResultMessageID;

5 d% i2 v! A3 l/ \0 M' \
};
! o9 X& H; r( j

6 H4 S$ @4 X  e$ b% B// Dummy Implementation to be used when no other implementation is available
, q* P# f  u  Y) F1 f" I- [6 O7 _class CUPnPImplNone: public CUPnPImpl
, I4 h5 a2 X4 X% F5 w# X{
) |. y" {2 L9 A) e8 Q6 [) g2 [public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
: ~1 T, Z, i* p+ U& U& m, d$ y- z        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
/ i5 W0 k3 \: ^        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
, ~8 K1 Q5 g4 i/ f  i};
" ]' J' Z, I! c+ B
) g5 @: {& b! n5 N
/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类


#pragma once
. {' e& e5 ]  e! e: M2 `2 W7 u#pragma warning( disable: 4355 )


#include "UPnPImpl.h"
#include <upnp.h>
8 |  C8 u. ?; i* h7 R# {0 w#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>
0 s4 @  i% }0 j  `1 M' o

2 _- T) ^8 a: F2 q* ~" b8 k" t#include <vector>
#include <exception>
8 z1 I6 k9 ]1 O1 R( r( F1 ~& k+ m#include <functional>




4 `) H% i" l: p7 U) Qtypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
. z: L3 u. r" s% h# ttypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
) f" G& q' L3 A6 ~0 U% Mtypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
3 Q% ?6 c1 Z; E- |( N* Ntypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
* ]" _8 A& Y5 t8 Q' w! |8 M

typedef DWORD (WINAPI* TGetBestInterface) (
- h  K$ o" y6 ?3 V  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);


typedef DWORD (WINAPI* TGetIpAddrTable) (
( d- [* V: ]+ k/ }+ K4 t  PMIB_IPADDRTABLE pIpAddrTable,
  C& p- Q( T  U( z  PULONG pdwSize,
  BOOL bOrder
);
1 W) Y( I9 E. [
! m+ x0 N) c) T* d9 O" C  j6 z# R# R
4 K# S) _. @6 r6 p7 k6 Otypedef DWORD (WINAPI* TGetIfEntry) (
0 p$ G4 a$ z$ r3 m5 Z0 h: u/ R  PMIB_IFROW pIfRow
);

& D$ u4 s. X/ X$ K1 M
CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);

0 q8 c4 t$ f' ~/ t
* x8 g/ x7 ^; ^+ mclass CUPnPImplWinServ: public CUPnPImpl
$ E& G$ c2 d/ E/ o4 ^{
5 h  x( t& I. x        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
# D! y0 L$ a: D6 ]3 G// Construction
public:
        virtual ~CUPnPImplWinServ();
+ i8 k% G5 m: n4 y! q: E9 f        CUPnPImplWinServ();
( A+ V" q7 e- M1 W4 O

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
: ~* y/ Y5 k$ c0 t$ w        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
3 @9 R4 Q! s) P' _  S0 U8 D& h        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
1 F# p# t4 {$ D7 `  b* L9 h7 n

        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
  d; [) y3 Y- P3 U& V        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };

# [( j8 G2 `$ P, x0 X2 \$ n7 z2 |
protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
4 p+ U: u' L3 J& D        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
$ b- J/ Z" ?9 P5 Y0 l        bool        OnSearchComplete();
. A2 K; v9 d/ c( {7 W        void        Init();
: ~- C* P) V4 [1 P" a. d  W; X

6 I8 [& C/ W6 \% G        inline bool IsAsyncFindRunning()
4 z7 Z3 @; W7 s  M# e( k        {
3 l" W. z+ u. L  o3 i& w5 Y                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
: g3 L" q& q- D7 A7 l( b                {
3 K) s0 P, G8 G  W  h/ B2 W. T7 u                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
# Y& a9 q; }1 n' @                }
5 T* ?: }. ^# X& v, o$ }* N                MSG msg;
% E1 \5 E( r3 d/ Q  v$ v                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
0 @1 o& r5 f# z  R6 `) o$ b( O                {
                        TranslateMessage( &msg );
3 g; f/ G% U- Z9 ^) p5 |1 J                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
; m, w: p4 S% t+ h7 J        }
+ K! ]  l  t+ m" `
0 J* L3 L% x( t
        TRISTATE                        m_bUPnPDeviceConnected;

+ |  l- N  ~, v
4 `5 y% `" K# T1 g* w$ W// Implementation
3 N$ A1 _" K5 y" W. C3 s/ h+ s8 F' {        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
5 e; }4 h) W0 m8 d- R$ I        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
" W6 E0 `7 F6 O; ~* q+ l, G        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
5 `7 N8 T$ P  ], s- O- H& h        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


& _! q4 I1 f/ v        TGetBestInterface                m_pfGetBestInterface;
  _" q' W& D& p, b$ x, y        TGetIpAddrTable                        m_pfGetIpAddrTable;
) \, h7 ~" M/ _+ ^  x$ U        TGetIfEntry                                m_pfGetIfEntry;
- [$ [1 i2 F. v% J# n
2 \5 o7 S: j) Z- c: m! G
        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
1 z# [: l. v  X                {
6 Z+ i6 E( {- t% w6 _                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


* g8 k: q& M% V: }% n' m                        if ( FAILED( hr ) )
& n" h/ T% c8 Y( w: _                                return UPnPMessage( hr ), false;


                        return wcscmp( deviceName.m_str, m_udn ) == 0;
1 A% i/ j7 Y7 J9 K. B- d                }
                CComBSTR m_udn;
; Q) X# B% V) L) U        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
* m0 \- }. R2 R1 ?1 {- W        HRESULT        GetDeviceServices(DevicePointer pDevice);
' f9 r2 f. ]+ U- W        void        StartPortMapping();
# w2 t' h8 X0 R( ?& F& _        HRESULT        MapPort(const ServicePointer& service);
* q2 r% P, f8 I3 f$ v- D5 {        void        DeleteExistingPortMappings(ServicePointer pService);
- U7 X% m$ d, t9 a  i' _2 n/ _% M        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
8 e  U3 d5 s4 X! @8 E        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
' i' ?" O/ y! o6 ?- U

5 ?& s8 ]: t3 q  L  [2 f        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
5 l0 e% e, ]1 g/ I7 a4 u        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
2 [4 z, e: E) j" r        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);
$ }# E/ j, u* r2 k$ D- c
, h4 H2 z7 p. t% G( U6 y" a
6 c* r' v8 I# ?" k2 L5 O// Private members
0 ^" ]$ P% i/ Dprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
- \0 [6 C( n; m. B3 R; n7 [& i( R        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
/ ?; a+ C: R7 e  }) _* ^        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;


        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
) ^1 j* b0 V9 r. s- f" J        bool        m_bPortIsFree;
        CString m_sLocalIP;
. b; |' I( ?4 H        CString m_sExternalIP;
( d; F; b2 J4 x. X) O        bool        m_bADSL;                // Is the device ADSL?
8 d& j! |- r: _4 b        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
- e+ t  J! U8 C        bool        m_bAsyncFindRunning;
" N& n6 B" H- b9 P" g/ c2 g        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
3 J  r: v+ t( W! ?/ B        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
2 A: C* r9 y8 Z$ o2 B( |3 Y        bool        m_bDisableWANIPSetup;
. ~8 N  S3 l7 L4 W        bool        m_bDisableWANPPPSetup;
/ h' ^; k5 X, W8 {6 {" B

};
4 J& O: k$ A) a# \( }
/ b  m7 R0 q1 u
// DeviceFinder Callback
class CDeviceFinderCallback
3 ^8 v: w6 j" b* k. V, N% w        : public IUPnPDeviceFinderCallback
% ^& j& j: w; `8 \/ m{
' f* |- a, Q7 D0 N' `1 p' Opublic:
5 H, w$ R8 B5 [, p5 s/ ]1 f        CDeviceFinderCallback(CUPnPImplWinServ& instance)
* X4 A2 t& {, r" `, s. |                : m_instance( instance )
$ ?' l& s4 _& D1 U1 o+ y) |' I3 y        { m_lRefCount = 0; }
% y  s. a& H. S' h( P; C

& B5 a; E2 I+ ?" Q. b- V4 b- r   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
8 Z0 h) T+ A" g7 U" l9 }. g   STDMETHODIMP_(ULONG) Release();

- y7 z2 C! k: o0 t7 M# p. ^: B) V
4 L* n7 h. U- E* R// implementation
* K& i! V, j( x2 [private:
& l5 K& I+ W( ^& M/ \        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
" M  g0 L- Q) o3 ]        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
+ p3 ]# }/ g* g3 k. ~
3 v3 _6 A( K# U1 a' P
) y1 {( K) m- }) Y' I6 z9 Zprivate:
3 H; x0 r( u$ i; T& |7 Y        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
& `2 f4 w' J( H1 a$ n1 O
% L0 f6 \2 d; G7 n( R+ [! H( M
8 P, \* F! Y- l  U// Service Callback
2 @6 v5 R% I% X6 t' H+ r6 oclass CServiceCallback
        : public IUPnPServiceCallback
{
5 R/ G& ^( L; B- u0 Upublic:
        CServiceCallback(CUPnPImplWinServ& instance)
! U' m' [0 _! P- r/ d: r                : m_instance( instance )
        { m_lRefCount = 0; }
  i+ b( U& j" K. F5 v$ D: _   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
& i+ n* {6 t! l9 o  w5 V  x   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
" ?: c6 l: o1 Z) U' g% T( H, q
- W8 g) I3 o5 T( |
// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


! H0 S' K1 U& vprivate:
' ]9 e+ ?3 O  A7 E8 ^/ _; A; y        CUPnPImplWinServ& m_instance;
9 B+ D; I1 u/ O% y: o! k6 g        LONG m_lRefCount;
};
$ A2 `1 S% D. K0 s& X  O$ a" ~

/////////////////////////////////////////////////

7 H# ^* X. D" n" z
" p: L0 A; Z+ D, O; H使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.