UPnP

天云上人

/*uPnP.h*/

1. 
   . f& D8 Z; n6 N3 o+ j* S! B
2. #ifndef   MYUPNP_H_
   
3. 
   ; x2 f8 v6 p/ p  m
4. #pragma   once
   
5. 
   : D( L: p* y6 j, s$ G0 U
6. typedef   unsigned   long   ulong;
   ' J2 f& p0 G1 R+ Z" d4 k
7. 
   0 m/ d' \' w$ I. w( }
8. class   MyUPnP
   - x* n2 u  a$ a" k. ^' i
9. {
   & I. V! U6 [# D, l
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    ; y& e- a. }* Y6 Z$ u
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    2 c3 m( K& m6 }2 B
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    . y# j1 w1 v/ m
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    $ |+ c( q4 K+ v9 W4 x+ I
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    $ O5 t1 k1 j1 `) [% H; Y  X
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    / j) G8 M, k. y% F
24. typedef   struct{
    - m1 B# }  Z9 o5 a0 V
25. WORD   internalPort; //   Port   mapping   internal   port
    7 X5 ]& V; X$ }. O  m$ U
26. WORD   externalPort; //   Port   mapping   external   port
    : y, D* z8 j9 L- _" W: B. Y1 @
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    - \2 l; R& {5 x, P# h
29. }   UPNPNAT_MAPPING;
    
30. 
    
31. MyUPnP();
    4 J6 w& {, I* g+ T8 F3 C
32. ~MyUPnP();
    
33. 
    & G  h/ y" W9 g
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    ( l8 z3 C( d2 a( d7 T- m& ]! K/ f
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    : {& G, |) Y) ?$ U9 Z. e
43. protected:
    , Z4 z8 w) K/ b2 V
44. void InitLocalIP();
      t8 j( p) Q% r1 x
45. void SetLastError(CString   error);
    
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. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    1 i8 D7 ~. s' _5 V1 u* j& `
52. 
    
53. bool Search(int   version=1);
    
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);
    
60. CString m_devicename;
    
61. CString m_name;
    3 p+ J6 D6 `- L) G" _0 U5 `
62. CString m_description;
    
63. CString m_baseurl;
    6 R0 Y6 F+ V2 c
64. CString m_controlurl;
    + C: F$ \0 J7 {, A9 A$ S* R
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    $ b% c9 C% N9 y4 A* e
68. 
    / `; @' U; z5 ~8 ~9 z/ X5 _% X
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    - N3 e- X+ u8 c; [3 k+ Z) P/ I6 w
72. CString m_slocalIP;
    
73. CString m_slastError;
    % M6 y( C. p. i- Q5 ?/ _
74. WORD m_uLocalIP;
    
75. 
    8 Z3 d+ ]. S0 B& X( j4 I, J
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   6 B0 w9 E" E$ {. B
2. #include   "stdafx.h "
   4 v% ]1 V  i8 g1 X0 P2 E
3. 
   
4. #include   "upnp.h "
   5 B+ m- V+ r: b& x
5. 
   3 {* |8 \/ u$ P/ F8 C! q3 b
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   & g, Q7 L$ n% @9 v  k
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   5 g& v8 k$ P2 y6 b/ t, n
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   6 Q  u( L8 ^1 I% e$ E3 F1 m/ u) ~
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    - r8 H; [; V1 c  O+ q
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    1 b7 X* h7 x/ Y- m6 X2 k
16. const   CString   getString(int   i)
    
17. {
    9 Z' e0 |8 R; c, f
18. CString   s;
    
19. 
    - H9 y* @8 o; d% g
20. s.Format(_T( "%d "),   i);
    6 [/ n9 o: d3 T, J- u! U1 B
21. 
    
22. return   s;
    
23. }
    ' c/ `! a+ k1 M
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    . }2 |  Y* o# Z3 Y( s7 T. U9 y
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    , J2 ~; U& O: v4 E" f
30. const   CString   GetArgString(const   CString&   name,   int   value)
    5 Q" m! n6 V$ |- L7 ?4 `$ b
31. {
    3 ?8 R# z  B0 y+ {) Z3 v/ ^+ W3 U
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    $ o. J9 U8 b& Q4 [* B4 c( f! y
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    . b- l7 a& g* [6 R
37. char   buffer[10240];
    " y& C' ?! o4 e0 |, i
38. 
    8 r3 B% [4 `+ @" ^& M- T2 S
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    ! s& R: O1 u0 m! z
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    ' ]  P6 D/ x1 C9 n
44. struct   sockaddr_in   sockaddr;
    / D1 |7 W4 U* o; b9 _+ x# N& l
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    9 D8 G& h# ]( r1 ~# K" `
47. sockaddr.sin_port   =   htons(port);
    
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);
    9 k' z  j# u4 j; R0 j& J( X  M6 ~; H
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    1 a: P" U5 q5 ]6 ]
54. int   n   =   send(s,   buffer,   length,   0);
    # K+ J' w. Q. Z8 C) \' n+ s' r
55. Sleep(100);
    2 G) W7 H8 j3 u7 X" P3 |
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    % W, u: `+ G4 G& T+ T# V
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    ( W+ M6 C5 |0 c/ H: S* n
59. if   (!rlen)   return   false;
    
60. 
    % Y/ B% ?: {7 v% Y# _
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    
64. }
    / P. m" r$ L5 F9 G7 W
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    - k* P+ Z- i% E3 S
68. char   buffer[10240];
    6 n; D( m( W- r% i
69. 
    * a  A9 v7 _& T* x# z3 Y$ e
70. const   CStringA   sa(request);
    0 R. s" P2 A9 ]. w* `/ H
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    ! d1 F$ v9 `: ?" R5 }" h
76. sockaddr.sin_family   =   AF_INET;
    * ]' }8 d- _: B, c
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    1 @2 |) }' [2 T1 ?8 I4 y  j; U
79. 
    0 [& O, x+ l) [. X) u! h! F
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    9 [0 o( e9 c' z$ F# c& ^
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    : t; s0 p: W- {+ s3 |! i9 r- _" m5 N
85. int   pos   =   0;
    / w+ V, n7 q- Y' P3 e
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    & [$ X: {. ~  V( H" Q
90. result.Delete(0,   pos);
      r# l# i: A: N9 X: _
91. 
    
92. pos   =   0;
    7 U5 |7 w3 G5 F% r/ D6 S; ?, z, r
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
      J; N' v  k' h  v8 X( O
96. return   true;
    7 v5 {- X9 m7 W7 y2 |
97. }
    1 t. k5 W' z) j  `% T# e
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    * V4 r/ M: c* ~
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     + b' v! \# {$ e! @4 c
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     2 u' R) T% B: y6 o6 W8 P
106. if   (posStart <0)   return   CString();
     " \/ c- E: Z3 b3 @$ T: c% r
107. 
     ' ]% \1 p7 b, y5 z) k
108. int   posEnd   =   all.Find(endTag,   posStart);
     " \; n+ }& _+ e7 Y7 B
109. if   (posStart> =posEnd)   return   CString();
     : S0 }: a: S$ P* N$ p7 b( D
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     8 u5 B5 `7 [* g3 \0 O+ m- \
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     + r3 E% Z0 ~3 h4 z6 K
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     ( H5 w% Z8 I0 }  L1 O
118. isSearched   =   false;
     3 E# j+ N" k" @" ^
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     + x, c5 t1 n7 p+ n
122. {
     8 B( S$ |3 {2 `+ b% E: O' |8 Z
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     ( \9 I$ [$ Z* }4 E4 T7 c
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     . Q9 E  }. I/ [& D8 C; \
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     0 Q3 D7 f6 m% ~, t4 i
131. }
     ( M% ~* S, T" M2 R4 g9 D7 |  f- |
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     ; n+ K9 p: W5 r& V
137. m_version   =   version;
     
138. 
     9 g5 g6 m8 S. g/ w
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     , e# j% f# W( \$ r% x# R* l6 |
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     - U0 K! ?) r0 O$ Y- P
148. ioctlsocket(s,   FIONBIO,   &lv);
     4 v2 \& n: t0 o7 K' {$ n
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     3 W  D- p; O" m/ R5 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);
     + r6 {/ @& H7 T( y8 `8 h3 `
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);
     $ Z' w$ T& l# x5 E+ Z
159. }
     3 E. `/ r; P$ C. z; [7 R
160. }
     - j3 _6 ]% ?6 {
161. 
     
162. Sleep(10);
     7 v# C& t/ w1 l) O1 e; P& m6 N
163. 
     0 L9 M- y- W) z" i) H. R" M$ g
164. char   buffer[10240];
     + Q' x( R1 _3 Y
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     0 }4 @0 \) ?  ]  {& c& V8 g" y9 \
172. 
     # T. O" G, [0 N5 d5 _1 o4 _  A
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     / {: ^* C7 z. j/ B8 Q& f. P; V
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     $ W9 V; A2 a- f
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     # M5 G9 K$ U1 z2 g: G
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     + @2 j8 P: Q; E2 m2 l: N
178. if   (line.IsEmpty())   return   false;
     % F9 M: P- `; C
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     ; f& W9 \( m% E: R, u0 w5 b+ @) J
181. if   (name   ==   _T( "LOCATION: "))   {
     " G% N5 W- v7 G, [1 s9 {
182. line.Delete(0,   9);
     0 K4 d! ^$ z. v9 f6 ^
183. m_description   =   line;
     
184. m_description.Trim();
     ) Q+ n9 G7 |  ?; q! O0 q/ v
185. return   GetDescription();
     
186. }
     1 I1 |& {" U8 x( a
187. }
     + F/ P" E4 t6 m' `  u9 V
188. }
     
189. }
     $ C. Q  _8 z9 c
190. }
     
191. closesocket(s);
     
192. 
     
193. return   false;
     
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
* q* y3 J6 b" k) V9 |7 w! g
) x" B" Y+ e( E
///////////////////////////////////////////
; c' [: V+ v1 T- h8 \7 D/ R8 U//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

& K" J  S7 V; e3 Y' i' y
#pragma once
#include <exception>


  enum TRISTATE{
8 X( X2 |& h/ v        TRIS_FALSE,
" R* a* h) |. L6 z        TRIS_UNKNOWN,
# d8 _+ l6 i% @7 M9 z& y5 S6 U        TRIS_TRUE
0 l" w9 U5 O, s/ p};
. ^$ ~$ M, I+ t6 ^
# l$ h! b# _6 I
enum UPNP_IMPLEMENTATION{
2 R2 b8 D. x0 R3 e6 d/ }8 o        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};



/ l# B5 \0 K  h% Q  L! W$ C
class CUPnPImpl
{
* ~0 p: d- m4 b2 Q$ @public:
* }* d8 j0 C4 p        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
# w  }5 j) B' x  P        enum {
0 N( s. X, u) i' y                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
        };

4 S0 S/ j. v: T  s2 Q
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
5 ~) k* W# |7 Z" t        virtual bool        IsReady() = 0;
! \% i' j2 n3 ~" x/ Y( P        virtual int                GetImplementationID() = 0;
* M! E9 i7 ]4 t! r       
, j: O- Z; w# D/ [        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
* _8 A. f  a' Z! g        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
. k$ u3 x& k8 w+ C7 ?6 j        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


+ W( a6 t( Q: U, Y# ~* x6 |// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
9 W% J' }. H5 e2 w! [" z        uint16                                m_nUDPPort;
; Y/ R$ w# Y# U; R8 C        uint16                                m_nTCPPort;
# @. q! ]6 d0 Y1 L4 x- x        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
8 |, s) Y( X" Q. R. a6 P3 e& E
& h# s& J. J& `+ L) Q4 J' k( F
& Q7 ~8 f4 j7 s8 iprivate:
        HWND        m_hResultMessageWindow;
$ u) ]/ X8 @; g9 O/ M* J6 q. O; m        UINT        m_nResultMessageID;
  w: T( E* c6 V
! Z1 |$ x4 L. }" B
# T" ~3 Q' n( J% M1 R. I4 H- N};
) H# b/ U3 Z7 B2 S# r8 d; x' _

. b% f  q3 d& f. c+ G! _0 j0 q// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
* `5 Y! @: x# [& Y        virtual void        DeletePorts()                                                                                        { }
  H( Z) c  V. k4 Z8 u        virtual bool        IsReady()                                                                                                { return false; }
8 J9 B4 ^& B6 a9 U+ i+ ]        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
$ D- r& @# h, K! i0 \( c" y5 }};

* U/ c1 Z- ~6 q9 c- a+ h: K' M
( v& o$ m% X7 \! |: @- j# z/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
+ \+ ~+ c- ]- I* Z
7 e1 x4 F- W3 _0 K9 ]. y
#pragma once
#pragma warning( disable: 4355 )


#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
  T8 g) B2 f$ R& M$ e  B3 C#include <comdef.h>
; p1 h+ E  |2 F) t' e#include <winsvc.h>

8 i" x$ C$ z9 }  s) t7 d, r6 ?
) h& B; u) t# N! Q#include <vector>
3 R* C+ f/ w: h! Y8 }! ]8 P#include <exception>
- X) q4 D7 |+ y* c* \. U#include <functional>

/ N2 o) R) B2 ]. x7 |% k" e
1 _. F# H. w* M8 _" N3 G
1 v% t8 t. Q; R7 o2 d( x! n
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
+ h, f8 F9 s7 p! u5 e% wtypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
! \7 G1 v/ L  x& H% `) f/ J$ etypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
9 u1 y3 N- v! D* u" ctypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


  v, z: [" f( j  p- b" Ztypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);
: q7 Y: a" P4 [& p$ Y
: P& A8 d0 U) \1 n
* ~5 t/ H* @" D% `* B$ [* c" htypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
( Y# z  D* `" K4 i! Q7 T  BOOL bOrder
! _$ p$ d2 s% b& ~: n2 r);


typedef DWORD (WINAPI* TGetIfEntry) (
# G4 k2 h2 b. y8 t6 p  PMIB_IFROW pIfRow
1 C! i* h: c( s& W# L. w1 @( Z);


  W0 s7 Z6 i. o2 ~: UCString translateUPnPResult(HRESULT hr);
" i, Q9 H5 w4 k, XHRESULT UPnPMessage(HRESULT hr);
; @- j2 p4 E. N; M! `" M! Q
8 j2 Z- \. ~1 U! L9 n& l4 V( f) B& a
* G6 M( p: ], L% ?class CUPnPImplWinServ: public CUPnPImpl
$ }) v! j7 v% ^% z0 @{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
public:
$ `& W: \/ q  C* O- B" W  W# t* O! ]        virtual ~CUPnPImplWinServ();
% b: w( {8 j8 F        CUPnPImplWinServ();
# G) a# o& C/ X5 b& v1 c; I) B1 {

, V7 @  _* k5 D8 }1 o0 _        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
1 q. b, p. l' ^6 s2 ^+ H3 d        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


! ~3 o4 I$ H$ A4 U        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
5 \: w4 S$ U& i+ k0 B1 E: `9 ?        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
9 P6 r' K3 K0 T# F4 [; H' j        virtual bool        CheckAndRefresh()                                                                                { return false; };
) K1 z0 d* F( n( S. a

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 }9 f5 T9 g4 C
5 k/ N4 J0 [8 o+ |& f3 F
& _8 a7 u/ ^9 Z! h& W4 W! r, ?9 L7 f  E        inline bool IsAsyncFindRunning()
5 f' t4 Q9 t3 V( q        {
, t3 C; [  `" m: r# e. E1 [                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
" X* L6 f  u3 l" y' G' k  P                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
  h4 x  ?% M1 `4 }& K                MSG msg;
1 f* I5 H2 E* {  r. c                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
' d& [( a; ?; _. ~. y" z3 T7 j. l                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
        }

( ~% J6 Q: Y4 {  Q
        TRISTATE                        m_bUPnPDeviceConnected;
7 I, S! B0 U+ @! D  C" O: y, z  n

// Implementation
9 T0 e7 @$ c. j6 \7 w; K/ N' {2 F        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
( _& ~" Z' b) a        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
* A+ ]( G+ e1 I  \' L8 `$ J        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
+ d2 X# L9 p- [4 _# T

- m; k- ]6 i( z+ |0 {        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
2 n, p/ J  v; d9 q        TGetIfEntry                                m_pfGetIfEntry;
# ?* K; T( N2 l& n2 y
6 h, j: m! z) j  C4 J) [; [# e
$ v& [6 c+ c! [. p& r  P        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
) h( x" G9 \' Z/ u) Q, _$ x        {
2 j2 {% B4 u9 A8 F                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
; l1 p; n0 L' A( k' _! a                result_type operator()(argument_type device) const
( K: }, p5 B: p3 B                {
  K: k: w. D/ q& T, x, M, z+ \                        CComBSTR deviceName;
( T  L+ O/ o& A' t- y5 n7 q! D                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


' K0 \8 k9 g6 K+ s2 c, Z                        if ( FAILED( hr ) )
! K, p/ ?' u2 N% r/ C  D$ U                                return UPnPMessage( hr ), false;

1 i" _' D! ^$ W
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
$ [- W- `* r) A( B. A0 I) a                }
                CComBSTR m_udn;
        };
2 x$ n9 o/ Y0 [; v0 E2 ?       
; p4 C' B: z. _$ \4 y( G        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
  T3 b3 x( s6 n9 s8 l        void        DeleteExistingPortMappings(ServicePointer pService);
% c' T: {1 f) I' _        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
$ O4 s! {# m3 |, D        void        StopUPnPService();


% c6 U2 b$ f! ^0 `        // 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);
! n) A' S# S1 H9 ~% v! R7 n        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
; m7 v9 j. T. G% [* {) D, |        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
" i+ f7 c: @+ ]        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
- X# H5 ?& T9 {) R6 m

// Private members
8 S4 w" g1 Y* ?3 N9 ], v  Kprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
6 h. L. |; u# P0 i        std::vector< DevicePointer >  m_pDevices;
- O! r* s/ m9 B2 ^  _        std::vector< ServicePointer > m_pServices;
0 q. I- a& _$ {4 p        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;


        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
. j1 ^# H3 P& |7 l+ X5 b; o# e' t  k+ e        CString m_sLocalIP;
# w. N" _. H' k7 k" r        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
  P8 }% x1 d$ ?8 f7 m+ o4 Y6 _# R. t        bool        m_bInited;
        bool        m_bAsyncFindRunning;
& a* ^* M; \% z- Z' A        HMODULE m_hADVAPI32_DLL;
5 E) l# X# R5 z# k- ^% o        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
4 p! x7 X0 m* j

$ L2 H7 D2 d- }# M- B0 R, M};

: p3 Z$ D, I$ A& Y
* A4 A' B( \/ H: H// DeviceFinder Callback
6 M- L; v3 E. u5 s( c; E9 Q7 \class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
public:
8 h& r! S/ V: s- G: H/ \        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }

+ [- z5 t- t3 w" q( [
/ k$ G1 ]& W6 t: _% {( a+ ?   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
( x& ^* D( {; t& p   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
8 i+ |! F  R& h* |6 H5 ?
+ }( _4 [+ A8 n; ~9 _! k! l
// implementation
0 u1 Z* S9 O2 c, B* E! a( Y& `/ xprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);


( r! D% @7 o' s/ y3 s, mprivate:
) `. p, }, v! s) K* D        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
3 j, H6 G. v3 f6 q};

: Q( X! a3 D1 D/ L1 g% q
/ w) k3 V) A) b7 i% }1 h% g// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
{
3 O1 k! h: I# R( A3 t& o: Ypublic:
        CServiceCallback(CUPnPImplWinServ& instance)
1 ~- C9 Z9 l- q* G0 }9 n3 v                : m_instance( instance )
1 n1 c' J, E# H3 s) K; [1 V        { m_lRefCount = 0; }
: z- v! j; w  p5 J1 a   
0 u/ E( I: y! [% j. l# m' c: w   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
3 d; S& P8 M2 {5 _+ G3 {7 d   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
2 W, @) M% v; H0 f7 X' G. k

, \2 E# {. |5 r. q/ {// implementation
private:
- k) K: Q# {6 \# Z' J) K9 M. N        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
0 n; ]) g+ ~5 z  k$ c8 y' l        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
% T5 P0 |$ i! Y
) B7 l& f  h* u2 @, I. M
; s: H2 r. O& c" y0 f/ A* X' n0 Qprivate:
8 F0 z2 x8 F1 V2 k2 @        CUPnPImplWinServ& m_instance;
: t/ a( M( w+ P/ X) l% q- {$ H, h        LONG m_lRefCount;
};

$ d4 T5 d/ @1 f
' Y5 B- X8 q& Y. k- X2 N) E  ~/////////////////////////////////////////////////
2 |4 z8 q  a/ K

使用时只需要使用抽象类的接口。
( J/ y. V4 o; uCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
2 @& U/ M: o8 c1 `$ iCUPnPImpl::DeletePorts删除端口映射.