UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   / R7 f  k9 {: ~% d% P, L
5. 
   7 Y- d  H3 V; l
6. typedef   unsigned   long   ulong;
   ) p! ^) t% J0 G/ D' |
7. 
   4 u/ z9 h# D8 W8 e
8. class   MyUPnP
   
9. {
   
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    0 L. ~  f- p6 t6 E
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    : Y3 F! l8 d: ~" }/ h# V5 a
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    $ v& F0 e) m# P
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    0 s% F5 d( G% b/ O% E
18. 
    
19. typedef   enum{
    # q6 Q  C( d- s: R6 `% S: o
20. UNAT_TCP, //   TCP   Protocol
    ! ?+ x/ s$ l, R$ T8 y
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    ! U% x" H& A$ z0 r. d8 r% k
24. typedef   struct{
    ; E% _6 l( X( p; ?4 `( Z6 ^6 M; _
25. WORD   internalPort; //   Port   mapping   internal   port
    % E& @% A1 Q. W6 C
26. WORD   externalPort; //   Port   mapping   external   port
    * {4 {1 Q) h/ T+ M
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    0 S9 h' p4 L2 |& H. z, _2 R# U
28. CString   description; //   Port   mapping   description
    % |* V3 z' p7 n
29. }   UPNPNAT_MAPPING;
    
30. 
    & @. O, c+ q" P: ]; i1 r
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    - x: H, B/ O2 V2 n, i
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    4 ~4 B7 _% B8 K6 r
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    . J/ Z) ^, A/ \8 i  S' f+ b& L
36. void   clearNATPortMapping();
    
37. 
    7 _3 H/ m) ~0 R2 D' ]0 \
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    - T0 U4 m1 ~* Q/ H8 H8 q
42. 
    7 V9 e' `* a" p4 S2 @, }
43. protected:
    8 U  _; v" |& h  k
44. void InitLocalIP();
    , j) e/ d1 Q( z* U" T, @
45. void SetLastError(CString   error);
    
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    - s' x' C2 h9 t" {4 R
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    ! B1 ]3 z0 G4 |: x* ~& B5 h
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    6 ]! N% f, t8 n, p$ a7 h& y
54. bool GetDescription();
    : \! u( {1 _3 k# `
55. CString GetProperty(const   CString&   name,   CString&   response);
    ! V/ Q# Q8 y8 w
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    ! Q9 C7 Y& D; ]' P; {
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    6 S5 t2 b( M5 h! G! t0 L! B' o
60. CString m_devicename;
    , i( Q' ^( a; `6 {0 p
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    2 h9 e0 P. ^) t& Y1 M! @
64. CString m_controlurl;
    
65. CString m_friendlyname;
    8 y  `3 \+ N% F5 F* `
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    1 y) j3 N- b, y. @
71. 
    ! P' N- K; s( n8 h  {  h2 V
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    ; p. P1 D/ d+ f( N# T
75. 
    
76. bool isSearched;
    # \) i# S0 z2 c# \: ~2 f9 k
77. };
    - u2 M: _; j7 D( R
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   - D7 t7 Q& {( e& [6 x& ]
5. 
   % b- u9 b$ k4 n# a# k! z
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   2 }1 y  B) @4 d. x
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   % m: @7 M1 Z. j; U
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    0 K, r; n& C# T7 i* l! j6 v
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    " k$ W3 y6 L" F& c. \) `
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    6 e+ r: J) ?: R6 @, p
15. 
    
16. const   CString   getString(int   i)
    ) b: V* n' E7 R# c2 i# t8 q6 ?
17. {
    9 p6 R; Z3 u5 u; U2 h
18. CString   s;
    
19. 
    - K1 \6 C1 T! ^  Q
20. s.Format(_T( "%d "),   i);
    . l+ U& g3 s9 P+ D7 @8 x
21. 
    
22. return   s;
    
23. }
    
24. 
    6 z3 _& d/ D" {
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    * K  Z: t) {8 ^5 A
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
      E) a+ x7 t) a5 s/ ]
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    ( L, d$ g9 [+ i- G
37. char   buffer[10240];
    - q9 O5 w2 s) {! N0 f
38. 
    4 ~' R( J9 n/ b7 {
39. const   CStringA   sa(request);
    5 y: Q9 [1 @% S! Z  Z/ r
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    & u1 ]1 o6 ^: j) k# R. D4 m0 s
43. uint32   ip   =   inet_addr(CStringA(addr));
    & J" |5 X  x. C# q
44. struct   sockaddr_in   sockaddr;
    * b* H  T) O$ U; B
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    % f5 x( r0 o  m% [
46. sockaddr.sin_family   =   AF_INET;
    1 q1 e9 I1 q# X: P  f5 b1 ]& p
47. sockaddr.sin_port   =   htons(port);
    2 f$ M* C4 k0 I2 S* T. Z
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    . s" T& o% K8 @
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    6 [/ Z% C' s7 v: w+ f; I" D% q
59. if   (!rlen)   return   false;
    
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    / e4 P# P0 S2 [. _1 ^6 y
63. return   true;
    
64. }
    
65. 
    1 ?( @* k0 Q4 @: D
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    ( M$ ?* X6 O0 a2 x$ A
68. char   buffer[10240];
    % z1 Y2 ~0 |4 i4 z: P
69. 
    3 _- e0 q5 K; z
70. const   CStringA   sa(request);
    # Y6 o1 h% l/ w, }. r
71. int   length   =   sa.GetLength();
    , W& W7 M: J" Y* M4 R# a
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    * R( x* v0 [4 c( o" E. X. Y
76. sockaddr.sin_family   =   AF_INET;
    ( j) M. Z5 e& e( y% G4 r
77. sockaddr.sin_port   =   htons(port);
    / y4 t! O4 y' @2 x/ j* i. Y
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    : n' f3 E8 `  E& x' T, A2 H
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    . w$ v  G- o/ o9 h. s
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    + A2 v; M  y0 P$ D! d4 g! ?
84. {
    1 R% ^& [9 G- s* X$ N1 T
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    1 l7 ^5 f; e$ k% J8 g: M) F) ]
90. result.Delete(0,   pos);
    / J, f9 b# Q$ R4 b, r7 P6 I
91. 
    ( C5 J  ^; @- B* v* T) R
92. pos   =   0;
    , p' Q) I9 M7 P. i
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    , k( P# ^- j1 o5 a, `  Q, @5 Q
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    9 A3 P! Y# [; d0 B. W- v6 u0 t
96. return   true;
    
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     $ Z$ \& O) o4 F1 q! B7 L( u
101. CString   startTag   =   ' < '   +   name   +   '> ';
     2 q! N0 |1 j" |" w( e
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     * d- I  J5 @3 Q2 X. c
103. CString   property;
     
104. 
     ( {8 Q4 J5 z. ~* c/ V% t  W
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     6 ]$ ?* l0 {6 n5 f3 Q/ n9 G
109. if   (posStart> =posEnd)   return   CString();
     4 B  Q$ X- w# v! L/ f
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     1 r% s$ }! q3 ?+ [2 N8 ]
112. }
     
113. 
     $ h% ^( h; y& W7 L
114. MyUPnP::MyUPnP()
     9 |, ~  ?! |; g' a, d; }
115. :   m_version(1)
     : [* d: Q0 ~8 j5 g
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     + {7 V6 ^7 P( S
120. 
     
121. MyUPnP::~MyUPnP()
     1 j# A1 a" q1 f$ u. _
122. {
     # q3 V- z7 z; R  L) @$ n$ Y
123. UPNPNAT_MAPPING   search;
     8 C" W6 B, h; U/ t) t. M2 a
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     # r9 L9 K5 g' e* K
126. search   =   m_Mappings.GetNext(pos);
     ; H, u' R' h' d9 L1 I
127. RemoveNATPortMapping(search,   false);
     3 s& o6 N+ i: Z4 ]* T% F- I' ]0 M
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     
133. 
     / I/ A: H. ~$ n3 A" q8 A6 h& x
134. bool   MyUPnP::InternalSearch(int   version)
     ! r2 X; O0 P* V+ R
135. {
     
136. if(version <=0)version   =   1;
     9 C- B  j+ W+ e$ ~1 m! N1 x- H
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
       T* [2 Q% F8 j% {; A) ~' p
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     ( Z% f* \( ^$ y' {, b" P7 O  K
148. ioctlsocket(s,   FIONBIO,   &lv);
     . T; c0 F7 @0 v( C; e4 `) Y
149. 
     4 w' k* P6 e" T' ~. \: ?$ v
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     ! s# F* C7 @5 y' q# E- |
152. if   (!(i%100))   {
     ; f! m3 f( p( ]) F# q: Z. i
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     ' g, W, N* n% B( n
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     ( a1 |; F/ ?% \2 Q
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);
     2 s0 u/ y/ U+ |1 u6 `3 w
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     
161. 
     $ o; [0 T1 B0 ~) U4 P& c4 @
162. Sleep(10);
     / J. S7 b) p$ E! e1 y$ `& o
163. 
     : |# A9 X& C2 Z. W
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     $ z- u" F9 U# I. n6 t
167. closesocket(s);
     
168. 
     " @  r3 A, O+ G! p3 @
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     ; [  F* P; H5 l& J2 v
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);
     
175. if   (result.Find(m_name)   > =   0)   {
     6 v5 f/ d! |, }1 e/ ?
176. for   (int   pos   =   0;;)   {
     5 Z1 p7 e6 C/ |; E% j* b  R2 g7 V
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();
     3 F/ c+ h9 f8 `% {8 [
181. if   (name   ==   _T( "LOCATION: "))   {
     " x* B" w! w9 I% D  z5 H
182. line.Delete(0,   9);
     # b; \* d; A3 n- l
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     1 G: M, I0 c; T, @6 r( q7 u/ Y6 b
186. }
     
187. }
     ! m. X% A" c* f% y
188. }
     5 X( i: r4 |; X; ]$ a
189. }
     " t3 |2 _+ |6 e4 p; A
190. }
     
191. closesocket(s);
     : @0 Z. y6 R/ H- J. O
192. 
     
193. return   false;
     
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
9 Q+ a& I0 v7 }6 S8 f( }, e( O
& I5 _2 I7 V% f' G7 F* ?
' \, ~. H! s3 w: S9 B///////////////////////////////////////////
+ g' K* D$ d0 `//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
2 D3 |9 J7 F/ n7 d$ ^5 i+ _! K$ F
, Z+ ?, |  P: y7 T) h: y5 P: E/ `
#pragma once
#include <exception>
2 {$ d! h: C8 N
! L1 T* p6 W# s! T
  enum TRISTATE{
4 v7 W. D/ s1 E5 s; R        TRIS_FALSE,
. P8 @9 Y. H) @6 e2 K        TRIS_UNKNOWN,
; q; z) D) I: \  c6 R        TRIS_TRUE
};


1 I& G+ Y1 t" \3 P/ y3 Z, Denum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
2 ?- G' G3 ~: b. I, h        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
4 [+ C8 F6 R2 U0 g, S8 N' S5 @+ W" ?( c5 H};
. k1 C/ Y' B# T3 B; w
, }' U% J" c6 m' ?  K( h4 R

+ N& ]4 R- e4 Z: A6 @4 _
# K/ t7 p# \- H6 Q$ ~5 \2 q; Y2 d2 yclass CUPnPImpl
{
+ h" D- V' l1 h. ~$ p& P- ]public:
; Z! |, B, z8 w        CUPnPImpl();
        virtual ~CUPnPImpl();
4 o% m/ o- Y$ d! d& J6 {        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
                UPNP_FAILED,
5 e  v6 o5 W8 j! c+ j9 c9 \" d                UPNP_TIMEOUT
        };
8 O" a# B2 V4 g2 v  w3 a& ^
( e! V8 N9 r( Z  V& {( ^+ Z% K
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
  i( E* n% A  \  q: H% m        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
( l* M8 z# M6 y1 m        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
) G) W1 Z: x- p, |" [/ B5 Z3 J        virtual int                GetImplementationID() = 0;
       
1 h# p% E# V3 M; Z        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
, R: C# C8 m$ L9 j- Y. k9 b
" D4 P" Z* f: `& t+ b  [; U6 ]9 @
$ D; o- l! p6 J4 L2 `        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
" G. w$ ?* k+ U" b( I/ }        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
, k1 d# i, r: f( ~

// Implementation
+ a/ j, Y0 }. }5 I, l7 Hprotected:
, T1 e" u. h) G0 f& E4 d        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
# ], S' d& a- Y' ?        uint16                                m_nUDPPort;
+ v* z6 f7 O2 A  z6 D5 a+ o        uint16                                m_nTCPPort;
1 g3 a5 z  B( m2 A, ~! w        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;


private:
: q  `8 `6 u+ Q+ c, c/ u, m+ I, H5 f- k        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
4 x! d3 Q! D7 H+ r# u2 M# ^$ S

};


// Dummy Implementation to be used when no other implementation is available
  }8 P+ J. ~) ?; q5 H" I  o6 wclass CUPnPImplNone: public CUPnPImpl
& e# I" x! h6 F* B0 I# B6 _{
public:
3 h9 M8 c' U8 M8 \        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
; C. g' o* T. o) L0 V8 a3 d        virtual bool        CheckAndRefresh()                                                                                { return false; }
( v* Y1 H- l. ~( G+ x% d$ [7 H        virtual void        StopAsyncFind()                                                                                        { }
- o/ h* o# o7 m/ `: M" X        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
- o- _) g; \% n; K3 ]};

  w8 |3 X1 {. M
/////////////////////////////////////
# J( ^6 j( h8 j//下面是使用windows操作系统自带的UPNP功能的子类


4 O: {' l: C: F#pragma once
4 o$ a8 ]& W6 X# {. q' m; E+ U#pragma warning( disable: 4355 )
# x, {- {$ \& J* |# ~' c3 S9 G

' a) d1 ~: D* D+ ^#include "UPnPImpl.h"
#include <upnp.h>
- b5 t9 G+ ]: w- M3 N#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>
" i) n5 W% Z0 u2 F
8 w' a" u, n' l2 p+ T) n
#include <vector>
, u4 C* ]! p' P3 q: X. ~#include <exception>
. R% ]6 A9 N4 i4 p#include <functional>


0 t& E9 D' g! @: K+ _. i

* c1 Y( i( V" ltypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
8 `( j! I4 @! K; ttypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
& e6 }( v( _; y( Dtypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
8 s7 N- a4 Y2 x& Ftypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
& s1 X7 M+ z8 c3 z  b7 gtypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
+ G. g4 K8 Y5 V9 `: i- v" O* ]3 ~typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


8 X9 _$ B0 n  w( E7 Ctypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
% F% {" O# W0 ~, G  x: o0 R2 s  PDWORD pdwBestIfIndex
5 I, [4 Y) t8 W: J1 G% o);
" u. t" [' z4 t4 H

2 |% P/ I; j) c) p2 U- htypedef DWORD (WINAPI* TGetIpAddrTable) (
; Q' O3 c( |4 N  s; d$ v  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
6 C  g  i- R; F& w: o);
2 r, h) T1 B9 b5 q% D) Q* H( m
5 R! f3 d1 ~0 Y+ h: r. ~
typedef DWORD (WINAPI* TGetIfEntry) (
! L6 z: ?. ], \0 Z$ N2 Z% J$ e, i  PMIB_IFROW pIfRow
);

( {5 t* _" S6 f. m; W  `5 D
CString translateUPnPResult(HRESULT hr);
  d4 n6 p  s' [& |) |HRESULT UPnPMessage(HRESULT hr);


1 }6 n7 |# b+ q/ jclass CUPnPImplWinServ: public CUPnPImpl
{
' x& g! y5 B! s9 |! }) P: u        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
4 l2 q* z3 h' W3 j6 k  Epublic:
/ y( m8 G% F8 ~4 g# e        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
2 f' w( l# x$ N( h        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
" h7 [+ S2 A% t2 |% X        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
, M% U- T# U  J' |8 e# G

% M: Q  C, `& Z% T        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
7 B( z. d6 f8 c/ M1 n- y! ^        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
2 H& U7 i1 X( f8 q4 J8 g) S        virtual bool        CheckAndRefresh()                                                                                { return false; };

: r. q* E. Z, `  ~4 E( k! a" b
9 \4 C( T/ k6 A. A) `; }8 {protected:
+ x8 K+ z  j' B        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
( G' ^5 E* d8 H# [% g. I        bool        OnSearchComplete();
        void        Init();


% B2 H0 _/ A0 E) G+ R" ^1 M  m        inline bool IsAsyncFindRunning()
/ J% Q2 a+ {: l6 K4 r8 P        {
  o) N  J- I; A4 F+ `2 k                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
1 t1 {! J' ~) {/ F" j                        m_bAsyncFindRunning = false;
, \* p# Z% ]" j# k  g                }
                MSG msg;
- d& B2 W* Y9 C! V1 M! O  N2 p7 T# H& c                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
' o* p& s9 R& D6 o$ p( ?$ J' x                        TranslateMessage( &msg );
7 u' D# \+ z" [  [# t; k                        DispatchMessage( &msg );
* i$ n$ R, @7 c0 \- B8 ~                }
7 F$ C! W$ c! j                return m_bAsyncFindRunning;
        }
* n. r* x. k$ _6 N

        TRISTATE                        m_bUPnPDeviceConnected;

6 ~8 d  f' \) g1 a; z
2 a4 v/ F8 T9 I& Z5 w// Implementation
1 h% b+ J( N' C7 C+ [3 H5 K. r        // API functions
5 r" P8 m. W0 x& c4 H# v        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
% \% h  f- s6 i; N) @4 W        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
: I( w; M3 h5 }7 V7 |8 j& p        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
3 O  Y1 o, t, H0 ?6 d, \2 @, W9 `        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

' z) M( v6 W; o* i  ]9 F
( f: d* A" U9 P0 Q        TGetBestInterface                m_pfGetBestInterface;
" _" T/ d% z% i! r3 h  A0 H3 J        TGetIpAddrTable                        m_pfGetIpAddrTable;
  ~# w& z  ]' m- W- M# x        TGetIfEntry                                m_pfGetIfEntry;
5 B* S8 o0 d( B
2 S& N( [0 w* O1 a2 j. `
        static FinderPointer CreateFinderInstance();
% D4 m% a1 b! c. ^- f+ ]/ K& C        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
. y0 Q( U3 U( W' W0 C+ Z( h- M                {
                        CComBSTR deviceName;
& F4 f0 L' A5 ?+ J                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
* s  Z6 [' V( ~
- q& {+ L+ j; ^& _" ?
$ o' R) M# u2 l, S: `                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;

& K4 M% W* F! r
, f( S& w2 W3 i/ i" a                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
% X# A' O/ Q: c- l' E                CComBSTR m_udn;
        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
% @  D) ~" v  N        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
& Q! Z: `. ]- i        void        CreatePortMappings(ServicePointer pService);
" V1 y% Y7 J! }& @        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
* X% n& d* ^% L# r/ s  Q& O        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
- q! o$ O: g: L/ j' E        void        StopUPnPService();
4 Y2 i0 j1 Z+ o+ H3 a: Q1 D" h
( G( @2 j. m+ J& w
4 X: r2 S' N4 j" U. ^5 ]        // Utility functions
. A. S2 C! k# N) @) |. ~        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
7 }) F% a" t, n! q3 n; i4 l        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
# N3 |9 K! L/ c/ }; M4 X" q2 H        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
2 w/ |1 K. y& R. p1 a        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);

8 f" p/ b! F2 p  \
! w7 D& ^9 z  _0 n% Q4 g! j// Private members
9 Z2 u) E* b: a8 t5 A# j6 U! iprivate:
+ c8 V) C" o, V0 w        DWORD        m_tLastEvent;        // When the last event was received?
3 l3 e/ H5 }; |        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
' w. A5 ^$ Q5 g' Q# Z        ServiceCallback                        m_pServiceCallback;
! ^) Y: C/ j* C8 ~) e
( x  R0 P- u- O) n
        LONG        m_nAsyncFindHandle;
8 W. V' X5 I# h        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
5 K0 w( Y5 {! D# x: @; w  n        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
1 q% b5 s, _- Q0 |6 b        bool        m_bInited;
4 T) `& w! b  J/ G+ ~1 o9 c        bool        m_bAsyncFindRunning;
9 N( |- U% I, r$ Q' R3 v- v+ f        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
; v2 C/ \- ^- i/ P" \, X        bool        m_bDisableWANIPSetup;
, x$ U' K+ p! Z1 [1 _4 d        bool        m_bDisableWANPPPSetup;


};
9 \6 O/ L% l. d5 [4 r' i% a# X
7 W% d, b# i) t* I
// DeviceFinder Callback
class CDeviceFinderCallback
: [, p- b% }1 b! V$ ]4 r0 e( T4 p        : public IUPnPDeviceFinderCallback
% g4 e3 T' k. ^; j0 N, o{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
* J- z* G: P( B6 f/ w5 q                : m_instance( instance )
        { m_lRefCount = 0; }

- h' C+ b  o* Z8 q3 u+ }# Z$ ^3 e
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
( @3 R# K' p  N+ |, w   STDMETHODIMP_(ULONG) Release();


// implementation
private:
/ u! i  Z  N/ q8 i. O        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
# _7 _# S1 m/ S/ F2 v$ u
& l, d( t! g7 X7 w8 G" ^3 _4 _
8 ^) z$ @2 T. h* T9 [  wprivate:
! w- g: p& V- O1 i( S        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};


// Service Callback
( P0 f3 K: a$ `- ~class CServiceCallback
, J# s' Q: B+ I  ?        : public IUPnPServiceCallback
{
* b. M5 \8 N' `: n* |9 I( }8 W# @( @public:
2 x7 m: t! S  G        CServiceCallback(CUPnPImplWinServ& instance)
& U( [/ J5 Y( E& D                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
. k$ M4 p- R2 V   STDMETHODIMP_(ULONG) AddRef();
- U4 w( M; @' N: B   STDMETHODIMP_(ULONG) Release();

0 K' l1 F2 A: w2 Y! A" @0 ^) B
// implementation
, r2 K6 N. Y4 Q: W0 R) `private:
* X& o0 F; j: C) w9 ~/ r        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
  J+ X% X" f8 |: s        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


* a# g# q) E1 ]  s  jprivate:
        CUPnPImplWinServ& m_instance;
. ?* U4 R% @- r2 o        LONG m_lRefCount;
};

! B& k% w+ Y& m0 e4 C& N. J) ]
+ V; w. X" M9 W; z1 }: k/////////////////////////////////////////////////


使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
% {' J: O9 \2 s7 V' h/ H' q4 Z* ?/ |CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.