UPnP

天云上人

/*uPnP.h*/

1. 
   ! P4 K( @6 E. N% {0 B6 ^
2. #ifndef   MYUPNP_H_
   
3. 
   * k' @3 [* ~: H) K" r
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   " M6 E$ I- G" |+ [2 H# C/ G
8. class   MyUPnP
   
9. {
   
10. public:
    9 j8 p1 y8 U/ {  a) ?
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    % ]  f0 W1 r! l1 t4 A/ w
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    ( |; s# m# `2 o: ]- |- J1 f6 j
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    * S/ e9 a: b3 {2 Y/ E1 \# j
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    & a/ V8 S9 B6 _# p7 u) o* k
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    1 S: ^: v# h) r9 Z% `8 {
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    ; p. o. R8 t3 G  j9 D! M% y
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    0 j) O5 k: \- H) g# l1 ~/ E' R1 d
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    ! g% U6 R" }: |. ^  ?! C
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    7 l) L. Y. F5 A4 t5 X0 P
30. 
    
31. MyUPnP();
    0 ~) o3 ?: `6 O# U. l% Z; g
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    : Y) N3 u' S6 ^1 H, P# G) h# C' u
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    . P7 F* e, a9 z
38. CString GetLastError();
      ~/ r* d6 W& ]' ]  H
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    ( F+ D; r" x; O5 B( _8 N
42. 
    
43. protected:
    
44. void InitLocalIP();
    
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. 
    ; Y/ k! J8 {8 J: @2 D% a
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    2 R1 u& G" g+ b9 @
52. 
    
53. bool Search(int   version=1);
    5 ?) I* M0 N" j+ }+ Q/ m+ }2 m
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    ( S6 X  B9 P6 J$ Y" _/ Q
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
      Z% i$ T, f# {' ~$ a
60. CString m_devicename;
    2 c$ l: f: R4 T$ v/ l/ f6 n: ~
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
    ; ~& d' l% G! }2 c4 b
65. CString m_friendlyname;
    
66. CString m_modelname;
    + @2 V: b! ^% y4 h2 p+ E
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    6 F' n& M3 h5 F2 f: g* Q
74. WORD m_uLocalIP;
    " }5 O3 ?8 @' W; p$ {- j
75. 
    
76. bool isSearched;
    2 N+ `/ P1 u* K
77. };
    * L# P! ]4 X! k" b9 ?
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   ) X9 A3 S. L1 ?6 f( e- ]
4. #include   "upnp.h "
   - T& G& K6 x% M0 G9 [  u8 t+ m
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   0 V+ P# p1 O9 P0 |! x( o
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   8 r6 a; J. h( N8 C$ A
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    6 R/ a( q1 S0 o+ g* x: s: L
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    . D* q! d- ^* Y7 b9 B1 e
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    $ F( w, g: J3 I1 B1 a
21. 
    
22. return   s;
    
23. }
    , \% K" c0 F: }' S! i
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    & a% Y  O' D$ L& }& X" E+ a
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    ) m1 P) \2 `4 \% i+ H  l; d+ P7 u
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    " M/ X8 f' f: Y) g- r: t
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    1 t$ P: l  Y, S4 y; f8 K4 C+ ?
37. char   buffer[10240];
    / u5 c9 r% x- }0 v  p& c
38. 
    & q$ k, n! Y: D
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    0 l1 t; A9 y" e' _# i" j
41. strcpy(buffer,   (const   char*)sa);
    - E' @# V0 B+ ~
42. 
    - g( T& D5 s, o) w  \4 _
43. uint32   ip   =   inet_addr(CStringA(addr));
    4 L' E3 o( O- U! f7 A
44. struct   sockaddr_in   sockaddr;
    9 E2 T  m- t* i
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    % h0 }. [% x2 x6 n
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    + p7 |' s( k2 H: U
50. u_long   lv   =   1;
    ' y& v: f; r5 E. L" H" n7 [( _
51. ioctlsocket(s,   FIONBIO,   &lv);
    6 }3 f  b! K1 ~' }$ ?2 \3 U
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    & n# I0 L  Z( ]# _  e
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    + u( _& U* b. `& ^
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    4 w- f9 ?$ v0 }: W2 @
57. closesocket(s);
    9 f+ R# i9 `5 o% c7 u9 J! v
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    " K2 I2 B9 d8 A; q0 K
61. response   =   CString(CStringA(buffer,   rlen));
    : T( R' X" b) q- I9 s
62. 
    ) v1 r% h8 Y: p% u& y8 h4 |7 L
63. return   true;
    $ G. ~# z- @* \5 ]& T
64. }
    9 B' X7 y, h0 p+ ]1 D
65. 
    " S; b2 S& F: \& Y
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    
69. 
    & C$ v  o5 J7 V- E  v
70. const   CStringA   sa(request);
    ( j  q% ^; s/ |# Y8 G
71. int   length   =   sa.GetLength();
    4 s+ p! b) I7 c4 S
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    8 X% {( g3 _( q* p1 }7 K$ I5 }
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    : w' x& f/ G* `" s0 L7 ?
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ' j7 g* S2 N  v- e* M
81. }
    
82. 
    ' V  Y) a% T7 j: l* d: |
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    - H- x1 N7 a: o
84. {
      B7 `0 W8 S* ?; U0 N! U
85. int   pos   =   0;
    % t' K2 z7 O) x' W; t
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    - d3 a* f: Z" A' K5 X& T
88. 
    
89. result   =   response;
    + P/ |! p( k) m2 q, j' q$ o  T
90. result.Delete(0,   pos);
    
91. 
    / A  }2 P) m) G3 M! f
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    9 b. }. M3 k* W1 P4 T
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    ( X6 N  i; i, K
96. return   true;
    + M8 L# Y2 `* l! t0 X: l2 H
97. }
    ! k+ I/ n- M  w3 L9 E+ f
98. 
    & P( w2 J& `" q( ?( R2 ~
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     0 t& U! T; x4 g% V# F: S
103. CString   property;
     ) Y: s+ v9 s9 A( R
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     4 ]2 i& i/ M! g  s; H% V, K
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     " W4 H5 h) @: D: j! _5 I
110. 
     0 {' G% U' L5 z3 o; v
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     $ A# F) g. z' S- o
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     6 T" o5 ~, n4 l3 x( b
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     $ j! Q1 X, T+ \) S; ~/ x
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     4 P, j# K! a7 |: R
122. {
     0 P1 C; q% S6 N$ c# w0 \; L  ]& h* r
123. UPNPNAT_MAPPING   search;
     ; s  v9 o. R) {
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     ( j7 j: y* O: x& ^1 N
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     ! A& [. D% R# F, d' v( v: O. v
129. 
     9 u, C3 e4 P) `; y, @, g
130. m_Mappings.RemoveAll();
     7 p  j9 R, y3 i3 z& [" l9 F+ W
131. }
     
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     $ I1 n2 ?( U5 B; q$ h3 y' g
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
       S6 p$ b' ^% \1 Q9 e6 ~% Y' w
138. 
     ' o* s- @1 l* v4 F/ D! l- J
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     % U+ N' H# G9 T' w" H  m. e  \7 V# e
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;
       A# o+ M, G, c. h
148. ioctlsocket(s,   FIONBIO,   &lv);
     6 P1 d7 L2 m1 e, K8 [
149. 
     9 ]7 V& W/ z, S- Y
150. int   rlen   =   0;
     + E; J9 b' z6 D# q. ~
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
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);
     - n! U( S6 O/ b- R
155. CString   request;
     0 T, X: G% \1 L6 J" w, g
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 "),
     : X) t0 p! b  Z  n# T0 i3 b, K
157. 6,   m_name);
     + y5 t& W9 w( K0 K
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     
161. 
     ; S: Q+ J: ?. k. U/ w, h( K! t1 a
162. Sleep(10);
     & R, g; l' N+ @- g
163. 
     1 @: M+ C, g* q. b/ @9 C
164. char   buffer[10240];
     2 u6 a) b2 ~' N. `# |; P
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     7 c% z1 Y5 E" b( x
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     * k$ |! m& S! E- k: r. c
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     # _: u2 c6 ~7 v! Q$ N
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     ' J( x+ V/ A! x' V9 ?! p
175. if   (result.Find(m_name)   > =   0)   {
     * _8 T% ^$ j& u1 P+ T( v
176. for   (int   pos   =   0;;)   {
     0 g7 f) l  _+ ^& f1 Z7 B! G
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();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     3 F& q, N% `; [
183. m_description   =   line;
     
184. m_description.Trim();
     5 \, W- I4 A" R4 ?/ t- f) t5 ~
185. return   GetDescription();
     
186. }
     9 Q& a* Z: A3 {) T; I& L
187. }
     
188. }
     3 O3 G0 i% g" H. l" L0 S: N) K
189. }
     4 q7 K- J6 @3 u' E0 @1 ~  u, t
190. }
     
191. closesocket(s);
     7 L- `) i* I  a4 X& M
192. 
     
193. return   false;
     * I  n0 [% d9 q. @2 |
194. }
     * c/ b, H- l+ e" x

复制代码

天云上人

以下有关upnp的接口来自emule，
$ y) N; L6 b; i7 }6 G7 m4 g

3 y+ T6 d, \# }; ?# O" ]' v# z( k1 Y///////////////////////////////////////////
$ E9 t, @8 ~* Z. \5 E6 j6 n//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
# \; k( R: m$ w: H2 Z
3 z4 X7 f0 G! {! {
#pragma once
+ G6 c: d6 _+ e3 W- I& S2 q#include <exception>


, n% U. r1 k; P  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
};


enum UPNP_IMPLEMENTATION{
! r) Y1 W* \2 X" q* j+ a        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};

4 o3 J5 n* T* |

* H, b2 E6 F: n5 a/ Y
class CUPnPImpl
{
% }0 A! K3 v) o3 ?  qpublic:
        CUPnPImpl();
; j. i# g2 N1 l2 G0 P, \        virtual ~CUPnPImpl();
' o- L0 _& C; I: `) m9 P5 P        struct UPnPError : std::exception {};
        enum {
5 y, X( S7 V0 _+ M) C) G) L- Y. S                UPNP_OK,
; w3 a; f; D6 R! _                UPNP_FAILED,
; t# }6 q1 e5 Y/ H+ g: X) K                UPNP_TIMEOUT
' A# a! x1 n) a8 v1 H        };
' p* \3 S' M! c% q; F& }

$ y) E5 M: T9 N1 S4 ?+ K        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
" {/ c7 \5 c$ }0 e) i3 x: I$ Q        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
+ U0 C" G$ h7 [) u       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

4 {* o7 l3 e9 b" F. n8 c/ Y- e
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
8 A: o1 ~. k6 |- M2 F        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
) U1 I; z0 f0 i% d' M1 W5 _        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

  X8 _* K; _: c' O' z! T
// Implementation
2 d6 l& A0 L  w9 p5 s7 Iprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
$ m6 Z9 d7 B7 y3 v) m, ?* e        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
$ p" w2 {4 j2 p, L5 z        bool                                m_bCheckAndRefresh;


( _' }+ G2 \/ U! }6 U- Kprivate:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
' i. m5 _' c4 Q$ K
8 w4 w* J# H+ [1 ]: l) H
% g9 y% U' E; q# u# W};

1 }; F" V0 {( w- ^
// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
# u8 G- O+ {5 M' Y9 c+ V{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
) t, O( ]$ u; ]7 K, q( |        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
  u1 w! A3 c: ~        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};


/////////////////////////////////////
: f. t! Y- ?1 T; H: e/ m//下面是使用windows操作系统自带的UPNP功能的子类


1 E3 A8 ~8 E8 [#pragma once
#pragma warning( disable: 4355 )


  k0 X/ M0 h) X/ `: l9 y#include "UPnPImpl.h"
: K# f( `* \) _1 n. b' W/ y) E#include <upnp.h>
. s8 b, D# _6 m9 r#include <iphlpapi.h>
9 o1 c; i1 G4 \8 z% }#include <comdef.h>
#include <winsvc.h>
' p6 T6 j* f! q6 f

/ |0 s" J" p4 y; _#include <vector>
#include <exception>
#include <functional>
" F4 B2 E, }  |& F  n+ m6 `
7 d  I1 C- x2 Z


1 R+ F2 N* a# ^, f5 {- C# C- M/ Vtypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _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;
. M3 H1 V  z, l( [5 Y" ftypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

2 W. g; C! l3 Y7 o+ B8 D' A* V: X$ C
( K! _3 N8 ?9 ^# o; F; L+ Ftypedef DWORD (WINAPI* TGetBestInterface) (
* |* s$ \$ ~& C: F  IPAddr dwDestAddr,
+ e+ z, X  B5 C7 P+ j/ {# G3 q  PDWORD pdwBestIfIndex
);

% N- |; F! S$ l  I: L) F" ^: s- \7 ~9 M
$ H; `! t3 c) V, f% J! ctypedef DWORD (WINAPI* TGetIpAddrTable) (
. H5 ^& M. }& Y) f6 c9 }$ B) X  PMIB_IPADDRTABLE pIpAddrTable,
0 @5 [/ |- w0 Z9 x( i: e  PULONG pdwSize,
7 E% `, @; D5 M8 d  W. d/ L) L  BOOL bOrder
);


+ L4 X9 Z9 U' T4 o2 _9 Jtypedef DWORD (WINAPI* TGetIfEntry) (
& o) k; N# i" X6 w5 O  PMIB_IFROW pIfRow
) x2 O6 ^! ]9 n- l1 r3 L2 U& c);


3 C9 U4 l3 r) x( Q- @  t' b4 [CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
9 q  Z( o/ O# G: }4 ?9 M* x# K

) X* T8 u; b' h0 z9 wclass CUPnPImplWinServ: public CUPnPImpl
" W- V, K' f7 {; J* D8 A{
: v5 s6 T  y, I" m% D9 [! v0 y8 S        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
1 y7 m9 v! L+ G+ x) S+ d: v. apublic:
% Q' J( ]: c' R5 }  h        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
. e0 e: z$ t( g
) T1 p/ Z! \+ G4 S/ w. A8 G/ \
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
1 x" N1 u- z) ]6 w8 D  `        virtual void        StopAsyncFind();
) t1 Q. c4 k& Z9 e9 b( r        virtual void        DeletePorts();
4 L/ j9 P/ b- Q% ~5 p        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

( U7 P4 ]6 B' G2 t; ?
1 x1 q! V- @8 l. u        // 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
# D4 L7 V; r- Z) _$ F        virtual bool        CheckAndRefresh()                                                                                { return false; };


protected:
' @# M. H/ z5 W: y( ^  C        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
! V6 g! n. ?1 }1 l% d, Z        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
) S* ~4 S6 @: e. n        bool        OnSearchComplete();
( t1 f6 O+ D6 ]; P% p% [$ I8 s% f+ P5 K        void        Init();

9 v2 y& i/ Q3 M+ B7 z# y1 |6 ?
        inline bool IsAsyncFindRunning()
' I4 j: {' S1 o4 W; Y9 m        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
$ D5 k6 {- z7 }                }
                MSG msg;
0 q- S3 _( V# j! w3 b                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
$ @, N0 B0 k) g+ Y$ J) L5 k                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
9 O+ r: a) o7 V% S# I  [        }

& w5 D& q# s& H5 K7 b/ t' b7 W
        TRISTATE                        m_bUPnPDeviceConnected;
. p0 C- }' Q8 V" [4 C  x: |

: |/ K2 S/ h% W6 p/ c0 g) e7 P// Implementation
) y2 S# }; _) Z5 \        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
& U3 c2 W2 Z) F        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
/ c' Z$ {5 |6 u
% C4 X9 W$ K5 G- b3 Y8 P* m# i
' U  X8 B7 Q- ~( @9 A2 n        TGetBestInterface                m_pfGetBestInterface;
' t5 A. {) _4 R* n' _2 l# B        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;


. o7 g/ g  [, {/ T2 q        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
* z5 K  d$ ~' v+ p% _$ s) m                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

. ?( ^5 ]8 O+ B1 i% P% _$ C
6 I5 A: b- A- h5 B. p2 Q                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
5 o3 T7 `9 j; x- l% _

  K0 e0 p: W. l* c                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
8 d" s# Q1 _! l# f) ?! m9 u                CComBSTR m_udn;
% Y' W& s% c- g# N        };
6 |7 s' s0 T' G; y7 `4 h1 A: d       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
6 u; p8 `/ C" l4 ?: J4 [7 r" g2 w: Y        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
  F, s. H# F+ y. N: z( [  l        HRESULT        MapPort(const ServicePointer& service);
  W  o& \  X5 E- E* n1 n        void        DeleteExistingPortMappings(ServicePointer pService);
0 \+ n- d) H$ z5 {1 L/ B! E        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
! V3 {- x) r. r6 q, @2 @4 P' d. Y        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
/ ?0 ~* u/ W; m6 g4 R/ N* `                LPCTSTR pszInArgString, CString& strResult);
3 V" t/ S7 r4 H, {  {        void        StopUPnPService();
- B& n$ e# g( `4 ~" Y
" V0 i/ |! h7 l7 T3 D
        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
6 ]5 L* t/ O$ ^: M1 @3 z# e- x        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);
- I6 j! q+ {. u) l. J        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
4 @( f; e- i8 t4 ~8 V        CString        GetLocalRoutableIP(ServicePointer pService);
2 J) x; n$ W/ t& m' k
- \& o9 T1 g' @# F
( a. A8 q2 T( T* Y3 i" R% q  G// Private members
3 n  g2 H0 W  H! o" O$ k" Aprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
! b- m, F  H% B* Y- z        std::vector< DevicePointer >  m_pDevices;
. x5 ]8 N0 t7 m        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
- i* l0 k/ b! D9 `2 J8 J5 `
8 t% F0 z. o& s
, M, O6 y2 ?! A- ]! s1 m: u        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
+ B; B% I9 g# D% z# C9 A        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
  K) ^; Q& C2 W: c- D        bool        m_bInited;
        bool        m_bAsyncFindRunning;
' P5 C6 j9 _" }! G4 L% G3 F. U        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
# j  z  l! r+ a2 W( i2 o/ `3 C8 _        bool        m_bSecondTry;
" R) T6 p; ?7 s/ K( X! z/ t        bool        m_bServiceStartedByEmule;
! w4 G' i8 [  t+ K" C        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;


/ v7 d0 E, ~6 v8 r* g& N  d};

+ H5 v7 ]1 X6 U; W. j( T8 F" ?
2 _& V9 ]/ k% C# n5 S  L0 s// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
2 E7 K9 O3 w" V& K# Fpublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
+ I' A2 d/ \* W, ^. ~6 [        { m_lRefCount = 0; }

+ s- U* B& q& U2 J5 u
) ?9 K, g1 g4 `$ s" w   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
# L- x6 E3 e  N7 |: M8 w   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

6 C2 j' x* g2 P( K
// implementation
& G% |& ]1 g' c9 ~! {" ~private:
7 K+ F3 q8 g. u2 v        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
# M) U4 d: X( ^5 i3 i! ?5 Q        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
8 r6 ^  O0 I$ `. }% q

private:
2 U0 H) s0 O6 [1 T        CUPnPImplWinServ& m_instance;
1 f9 B! b$ S9 q5 x, i  s        LONG m_lRefCount;
};


// Service Callback
2 @0 Z7 k$ U/ S$ m5 dclass CServiceCallback
        : public IUPnPServiceCallback
# A* \! Z, @6 ?3 L9 h1 K8 k0 V{
  t2 n: A2 o1 e8 m9 k$ n/ J+ Npublic:
9 z0 m: {3 X& [. d5 a        CServiceCallback(CUPnPImplWinServ& instance)
4 l7 n7 r5 ~' b9 k! Y7 P                : m_instance( instance )
        { m_lRefCount = 0; }
+ @6 I# Y' k& ^0 x. f; c4 q' [* w   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
8 t* j/ X7 d" W* r+ ^% I! C3 z9 b   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
9 k' C& s8 M9 s

. I" x* a+ y5 h! A1 c$ G// implementation
; w# {9 ^* F  S3 Iprivate:
4 E% f+ w6 A, q6 B4 T8 s! P# Z        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
1 n! v$ Z: Y+ r0 M  |9 o  G/ }) }
8 ]( I% p/ e( y# R
- A0 Z9 P% c6 ^. b* R5 W5 |: y( Fprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
0 t2 d9 m# n- x* E1 {% u};
. e+ S8 S, U6 W
' r  O2 s3 O/ c
1 e7 q2 {8 }/ }3 I8 a/////////////////////////////////////////////////

* ~- `9 ^, j1 W" y; i
使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.