UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   + m% x8 n0 [: M5 ~
4. #pragma   once
   
5. 
   " s, f  t8 a9 y# o5 P( Y$ C
6. typedef   unsigned   long   ulong;
   % @% D) [" a0 k+ L. c6 y( j( ?
7. 
   
8. class   MyUPnP
   ! u! }4 P+ c8 ]1 d6 O8 p
9. {
   
10. public:
    + w# O5 S# X. S4 G
11. typedef   enum{
    ! u$ U0 ^/ [$ ]4 {  k; G7 ]
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
    5 P" t7 }( C6 i+ z1 e7 ~+ j5 ]
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
    ; y. A# \% g. g; |! K& C0 m+ p
17. }   UPNPNAT_RETURN;
    
18. 
    + C" N+ ?& c0 W6 T$ `
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    0 j$ Y0 {/ {# S. v( Q( B% ?! I
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    
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;
    ' I+ ]1 p5 ]5 F; C+ J
30. 
    
31. MyUPnP();
    # h* T9 ]% v7 }3 _" [( v
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    3 F% K, ~9 j" G) s! ~! A) G
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    ( W2 P7 a+ o$ e" v' O
37. 
    " c: j& D! D% N- s
38. CString GetLastError();
      x" e& y& s( P8 d& Y( p
39. CString GetLocalIPStr();
    9 B! h) C" u; l" w1 S* B
40. WORD GetLocalIP();
    ' _5 h5 l/ u( C& ?. @  s
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    9 X$ I4 T- x( q: h1 n
44. void InitLocalIP();
    + n: G7 w" [( `' J
45. void SetLastError(CString   error);
    ) u8 j, {1 K4 ]0 n1 _( d
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);
    * |' F" {0 y9 w
50. 
    # n2 \' L, Q9 h- c
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    " V2 L) T+ F- X6 u
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    , L) ^* D& X- [' g: w2 n6 B
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    ' t* X6 I( `' t: A$ i
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    - A) z+ I0 [$ |6 K7 K6 Z' [
61. CString m_name;
    
62. CString m_description;
    8 k0 {  i1 I0 ~7 \  J6 U
63. CString m_baseurl;
    & o& p$ y# ?7 O4 M
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    $ I+ j! ?% S% j
67. int m_version;
    6 ]% o" Q( s, e) R0 w% \
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    # [' N5 @9 n8 q3 g1 ?" |
75. 
    ! o6 g& |* e6 ]8 Y( M1 A4 ^" H
76. bool isSearched;
    
77. };
    1 Y2 W! v2 q* Y5 v: C% U" ]0 v# r
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   : A' G0 h2 L( j$ A. U  P: ~
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   $ s5 x3 r; K1 K
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    - t3 w6 y  i! d# K4 [8 ]& P
13. static   const   int UPNPPORT   =   1900;
    , J: C6 L1 J" P" w- N7 M3 W0 T: @6 }; T
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    ! S0 h& D* y8 X, K9 W% U
15. 
    
16. const   CString   getString(int   i)
    
17. {
    ; }6 q% l4 f7 j! p
18. CString   s;
    , b0 ]1 z8 l+ a  u
19. 
    
20. s.Format(_T( "%d "),   i);
    * B; x9 p: c( {5 y2 J! V9 b
21. 
    
22. return   s;
    
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    * v. Y2 C1 D# Z
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    " s! Z2 Q* P5 j
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    5 ~. M  z8 q; m3 L0 y
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    , g+ R6 ~4 O, f1 m) l" |' |$ l
33. }
    # i% ]% O+ f8 x4 y! }! e4 H+ q
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    8 e% ?  P/ H! H7 W; ]3 x( C
38. 
    
39. const   CStringA   sa(request);
    ' h2 S& B7 p3 y7 i
40. int   length   =   sa.GetLength();
    6 a( S# [7 k  D. ~7 I
41. strcpy(buffer,   (const   char*)sa);
    2 Z4 f8 b6 }; R1 s
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    ) l4 D, x. s. l9 x4 f/ t2 A) Z
47. sockaddr.sin_port   =   htons(port);
    ; _6 B# W2 D" C9 w5 F4 W) m
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    , P: J: |  a' R+ E  n
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    , d$ j8 u8 `) a+ ?0 i/ q
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    : K6 p% D7 W6 x- y6 Z
55. Sleep(100);
    ( @7 }: g8 I/ T: r2 T1 b9 Y4 {
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
      G# p2 k. o8 _4 [" u' m: h
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    
64. }
    5 `+ t* f% o0 V: j
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    
69. 
    / \2 m, A" M8 e3 ?
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    6 e* F0 e% t1 v( w/ I8 M# s
72. strcpy(buffer,   (const   char*)sa);
    ( @9 G& ?+ p' ]5 `
73. 
    
74. struct   sockaddr_in   sockaddr;
    2 M3 a% F9 Z; J" ~/ R- D& O* j
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    . ]( N- T3 {3 i  a3 }' E9 k7 C
77. sockaddr.sin_port   =   htons(port);
    8 E% E2 a/ a& A8 a7 c
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    8 B; c1 V# I; g( \7 ^+ ]! g! U
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    " p' O) s$ G; ~
81. }
    % g4 k6 D; M3 Q- ^& p1 e, a
82. 
    , p6 k4 X1 r' H' X, t, G
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    ! S' Y+ E, y8 C' ]6 G
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    , `. W2 I7 W) x6 q4 [* l) E: b, V" e2 O
93. status.Tokenize(_T( "   "),   pos);
    / t1 ^- d* I4 q5 i
94. status   =   status.Tokenize(_T( "   "),   pos);
    - J) C  l/ g7 V' p
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    + x9 K' i1 _6 r
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    : X) ], w. A6 ?
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     + o& }1 A' K6 y
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     & e$ ]% f1 k7 v8 ^( {( G) V& m  Y
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     1 J8 Y; i7 ]' u
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     . Q8 W" `, D; o3 ]  g
112. }
     ( t7 P! ^9 c7 I" m5 `* Z) |2 W/ u
113. 
     
114. MyUPnP::MyUPnP()
     ( s& ?" s8 g8 U) [
115. :   m_version(1)
     
116. {
     * B8 o) ^* y; ~! @% b2 X  m
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
       o0 N9 s2 t7 [- i  x4 w& n3 I0 Z
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     ( D! b: Z4 k' _6 I( a9 q; U9 m
122. {
     
123. UPNPNAT_MAPPING   search;
     : a9 f6 P9 d' b
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     8 ^( H2 `: B3 t* i' x: ]& l7 d
125. while(pos){
     + H/ V. y2 w* n* B' F; x) L
126. search   =   m_Mappings.GetNext(pos);
     # {- V& c7 J" w5 h8 R
127. RemoveNATPortMapping(search,   false);
     ' _% l, E$ C" G
128. }
     ! s/ q7 @; w% u5 y  `" `* S
129. 
     ' k: z2 ~6 f4 ?. L
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     , b: f- n( ^2 E: a5 H$ M
133. 
     + R4 u8 ^" k; ^
134. bool   MyUPnP::InternalSearch(int   version)
       k+ h) I9 |9 H6 q2 D
135. {
     $ w. i/ Y" E* e1 H) U1 [% l
136. if(version <=0)version   =   1;
     4 p( W8 ^7 t1 q9 ?( r: h) d) d6 v
137. m_version   =   version;
     $ k) a: n& a$ y: q
138. 
     
139. #define   NUMBEROFDEVICES 2
     + Y4 {- s* }. V* c* T& t
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     - y& t/ R- Z3 X6 P0 Q  ?+ C& I
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     # V4 Q* j6 Q; v
144. };
     ' g! O, d0 Q4 ?$ \9 G$ T
145. 
     * `2 f& T2 s" \( `& Q
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     5 H9 |' t4 z1 H; C$ C
148. ioctlsocket(s,   FIONBIO,   &lv);
     2 W4 T  a1 E$ q$ T8 E
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     8 }( A: |3 \8 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);
     
155. CString   request;
     6 F0 Y, N% ]- d% o
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 "),
     1 g3 C& r4 W3 p+ o
157. 6,   m_name);
     ' O' z2 X/ p8 y! [. i
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     9 b9 J) H' d2 k5 X; y+ R
161. 
     
162. Sleep(10);
     
163. 
     8 V7 A; I' x4 W" X$ [
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     5 y" U% w/ h' ^  u8 X9 ^
166. if   (rlen   <=   0)   continue;
     & F3 D. l7 {* u+ l: x. N  E
167. closesocket(s);
       ?. }3 w: y, @$ e
168. 
     + W' j; h& b  f" O8 p: w
169. CString   response   =   CString(CStringA(buffer,   rlen));
     " o) f1 O8 l1 y; W/ r1 ]. s
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     8 U' T: w. u% s, {
172. 
       |8 j3 R7 x/ _2 i% c7 i+ k
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)   {
     . N, X8 E0 y, a# ^7 ^! A+ H/ r7 X' Z
176. for   (int   pos   =   0;;)   {
     ) g  i7 Z9 _. r  Q2 u$ {
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     4 R/ w8 v0 o2 A; q9 |8 a
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     4 {; B, C0 Z! r* s9 ^% Q
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
       f& O6 j$ j: O4 S% w6 L' h3 H) u
185. return   GetDescription();
     
186. }
     
187. }
     ) Z( v. N! C# K: v/ L3 F+ O
188. }
     - h& q% N% l# J- l! \! T
189. }
     ) A- V0 d1 q- }
190. }
     
191. closesocket(s);
     
192. 
     - m5 F& ]: R. g" Y: a
193. return   false;
     9 k' Z/ R1 N1 n
194. }
     . g1 `: V) C* p- K8 `" j

复制代码

天云上人

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


///////////////////////////////////////////
9 O3 ]& S7 P$ V5 _% B* ?//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
& F$ G& z* g1 ^7 H: A
) J, V; o# z6 B
: ]; x  i6 E, A% a2 B$ V( y#pragma once
#include <exception>
( K: b. N. y) Z# W2 N! v0 g; v  r
' C( b' @# w  f4 F
3 f( ~  O7 L! {; _3 |$ K  enum TRISTATE{
9 G( I8 z3 o, }# I( ^  w        TRIS_FALSE,
        TRIS_UNKNOWN,
# p4 y1 O/ a; J, E        TRIS_TRUE
};


enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
: t& M! j! M) ]- O$ b};


+ U0 ~  R8 L8 {; Z/ H
4 e5 s/ ]+ O7 t/ h6 g
class CUPnPImpl
. T# o' n1 M* F, N{
public:
        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
- D2 A. e) F* `7 `2 E) e# t, O1 @3 q                UPNP_OK,
                UPNP_FAILED,
# Q# D7 [) I. A& A- {                UPNP_TIMEOUT
+ p3 J7 {, B4 i, `4 _        };

# E# e- G' Y. v2 }: H
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
2 t: _  D5 s6 z/ D8 k9 q3 m+ Z        virtual void        DeletePorts() = 0;
( p# x  n8 N& }* F        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
6 R5 G* ]1 X' |( y( ^- b8 ?       
" f9 c( H& V% C" j        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
: _; D) K( j8 p

: |* o( W8 H0 j+ d% D6 t        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
7 T5 R% v- g- V' v& P) g; t        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
) L1 [8 I$ P3 n) `+ F% |        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
, Q: ~! n( A2 r( i6 l        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

, q* @8 f# k/ t+ |
// Implementation
/ p9 r* |3 z1 G1 @+ y/ fprotected:
* O% Y  g* O* }& b( M( @        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
( R9 B7 P7 H: @6 c, `' l* E; `        uint16                                m_nUDPPort;
& J  j& r, l0 Q7 ]- s        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
* ~% V+ p  A( l: K2 q3 G

private:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;


8 |  J4 g- Q0 E9 ?% T};

' x+ D2 K. |3 t' o
// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
public:
' o* _4 L# w& }- l, M! W0 n        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
8 \& D* s+ q# @        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};

6 x8 `+ o; `. q% Y
2 E' c, a* W* ^% n! s% l7 R/////////////////////////////////////
0 Q! H3 U2 P: ?7 V4 v: \) x, c//下面是使用windows操作系统自带的UPNP功能的子类
! F& X/ E+ {: }

5 a" h' u+ E8 n7 h1 g% S; Y#pragma once
4 D2 j( I& _0 {% m$ P0 W8 `#pragma warning( disable: 4355 )


#include "UPnPImpl.h"
9 q3 I  w; e8 I5 Z7 b#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>
6 r# {# h& W  Z, d6 D/ o1 s3 _
. C; i0 e4 {8 b" C
, f/ P6 w6 Z6 |" |  l4 S$ V#include <vector>
#include <exception>
9 J3 d+ [8 K% F0 }8 g+ L3 V  @#include <functional>

% O7 x# g  X! @" {2 ?8 y: d! x
3 c. {& r9 P7 o2 h- g( G* R# O& V
7 e) P9 w+ k  a9 T1 _* }6 b/ l; N
9 e2 j. b" u! l/ c1 O7 xtypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
2 O* o- M5 A" v! j/ Ztypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
- v3 _" y! n; J) A6 Ntypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
" y, h' _* R4 |0 ?+ Gtypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
6 b2 q+ h& O6 B0 j5 K; K- ]typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

( A  N- F5 [; N
typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
3 D3 L% S. t6 [4 e* b: x7 S);
. L7 u* J) m0 _+ c5 U
$ x; M& J& n0 o# g! q+ V
& w1 V6 y& m% ~! e4 r* P% ~$ {typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
+ S3 B: d" E& ^2 v5 }2 m, O  PULONG pdwSize,
  BOOL bOrder
);


typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);

& P+ w5 T  v' [9 }
7 b1 v0 b) q+ N# U0 g; ICString translateUPnPResult(HRESULT hr);
7 t& F" N/ N  m) qHRESULT UPnPMessage(HRESULT hr);
5 M3 V' u# t9 b& Y2 }" ~( g

class CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
7 z$ C3 |& @- v8 k. A* W! K// Construction
public:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();

3 ?& Y$ r. J. I. a& `6 G9 y
. q, a  o% `1 A4 A        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
% t1 E/ R9 L* V' c/ @        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
: J1 X/ k% O0 l; A        virtual bool        IsReady();
6 E# b1 a5 ~; a# ?1 k, s        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
: o) _8 X7 M; @/ @. ~
* \3 I: y, I* h$ O* }7 P- J3 e9 ^9 H
        // 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
# _3 y* D, e! t$ `3 r2 d; i! X9 i        virtual bool        CheckAndRefresh()                                                                                { return false; };


protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();
4 D) u" v9 U. }3 A

5 M7 H! ]6 |9 c" r        inline bool IsAsyncFindRunning()
' l! D4 R# B' q3 }2 u        {
' @$ f( Y0 D, R8 J& q" q                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
; D! ]; I, l- e+ c% k' z2 V; W  R- n                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
8 i, D5 @! O$ Z* ^# G                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
9 t' ^( M. ]  D: Z                        TranslateMessage( &msg );
4 V5 b/ \) |; F" J, ~6 q  w                        DispatchMessage( &msg );
; ]+ R3 m: w  T/ ]8 m7 Y                }
                return m_bAsyncFindRunning;
# q" }( }8 s' f" u2 Q: q% s        }
* T4 K  L4 t3 f

4 K+ _" l7 L! l        TRISTATE                        m_bUPnPDeviceConnected;

1 o' }7 T+ ], F& ~. a5 D
+ c1 L! T9 g1 e9 y7 g/ d+ C// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
" R6 e1 _8 }6 w        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
" @0 }0 o, `2 h( N2 I        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
( u" w3 B- I& Y        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


6 v9 w$ E, E& I1 X- x9 ?0 k        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
- }9 z; |# Y, J- M' r        TGetIfEntry                                m_pfGetIfEntry;
0 Q- ^7 [- c/ ~) Q

% ~5 G# t# ]9 D+ ^# w        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
0 J6 m( n2 O8 k0 ]2 r; W2 r! Z) V/ @                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
/ A% R; W6 D4 s9 E# E                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
4 v; r$ @6 L7 p* z9 K' b- H

2 c7 z, N% A3 h( O                        if ( FAILED( hr ) )
2 H" l) y0 _. l& E                                return UPnPMessage( hr ), false;
# [3 d* ~1 |* w

                        return wcscmp( deviceName.m_str, m_udn ) == 0;
+ b4 {0 a7 B5 F* z                }
                CComBSTR m_udn;
        };
) C' a6 f) ?# I% c) H       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
. k! n; X. K7 H, d        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
! o) E: v3 y) U9 K8 _        void        DeleteExistingPortMappings(ServicePointer pService);
" T7 D5 i+ u6 P- V' P1 q        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
# u- l4 u7 `) Z        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
# v- E$ l( t7 k1 Z) P5 D# k9 Y                LPCTSTR pszInArgString, CString& strResult);
$ F6 R# ~* Q6 l. |; Q        void        StopUPnPService();
# j/ ^1 X2 v0 o( H. [$ {# h7 T

* T) g  O: V- [/ O5 `        // Utility functions
8 M0 E5 C7 I; C# C, V        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);
* B. ^/ E- n( N. i        CString        GetLocalRoutableIP(ServicePointer pService);


/ |# V) x, u: A5 b6 X  ]! u// Private members
0 z3 C, s: i! k0 w/ d& nprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
$ ^/ E- P& J! T" c2 p+ I* l        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
7 A0 @3 B  J5 S, f( V/ L7 B5 h+ {        FinderPointer                        m_pDeviceFinder;
/ p5 O! X( V$ i& W2 V& F        DeviceFinderCallback        m_pDeviceFinderCallback;
* U8 k" _1 R& ]2 i* g. `; R        ServiceCallback                        m_pServiceCallback;
6 O! l3 Y, [: F7 v# ?

% [  S1 k; a$ Y1 F' ^( r        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
0 r9 A2 x" \1 R# L; e/ y        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
1 T7 r2 c2 B1 \. w        bool        m_bDisableWANPPPSetup;
7 q9 r) H) N1 l9 A# ~

};
, h+ q: ^! i6 @2 ^) J
  n( p$ O/ h  U- _
// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
5 d: m  a+ v9 u7 ]5 b{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
: U' ~. A, C: D( v) Q                : m_instance( instance )
9 ~. C  ^3 F7 H4 C5 z, Z        { m_lRefCount = 0; }


" D' U7 [3 P, p   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
& n. `: Y  m2 M9 a, o   STDMETHODIMP_(ULONG) AddRef();
! O. }3 j& F% T4 `   STDMETHODIMP_(ULONG) Release();

5 ]) |6 [' }+ i9 y$ e2 P8 u
" z$ @& w) ~% Z) o3 m  u; G// implementation
private:
3 J5 o3 u8 n: d$ w* N# ]        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
" H8 Y5 |: l  i- W% U, P4 e

5 c7 B7 X- C9 s3 z7 Vprivate:
0 Z( k' x5 l/ A% G& F        CUPnPImplWinServ& m_instance;
8 s6 B! V  h: O        LONG m_lRefCount;
( ^- N, j& }1 ]' O};

9 V0 ]- F1 u: G' Y1 x: ?; K7 g3 t
// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
{
public:
* [& k' ?9 E  l+ S        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
1 q% b- x- O. R' O' [0 b        { m_lRefCount = 0; }
   
4 h4 I+ ^, U4 A9 z% ?   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


! A9 R7 Y* V( J5 e- S( n+ E! _// implementation
/ R' V. K' u+ m" ]% K: dprivate:
* m* E* g! w! J: O        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
+ y/ V7 ~! S- Y7 \" E5 O: y
  N" G& \5 z! l) @; C. p2 z
1 U4 q: W5 D1 `5 Zprivate:
        CUPnPImplWinServ& m_instance;
4 V8 k/ `2 [( T# [        LONG m_lRefCount;
9 P$ G  J- Y  ^2 O' V! D};
; K6 w3 o( `" F! r3 S

: A" U; A4 b6 T# q# \$ Z9 C/////////////////////////////////////////////////

" v. c% m6 V1 z* X2 A: {- {
使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
0 l8 d0 e. u4 {1 z4 u1 K! B, u5 QCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.