UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   
9. {
   
10. public:
    , B, G2 n& u" g7 [
11. typedef   enum{
    9 g& x# g0 J8 K( A/ M; B
12. UNAT_OK, //   Successfull
    : E% r+ C& Z  o
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
    - h2 R# ~! A. V4 L& V# d+ A3 h
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
    / Y1 |* A( i& w' z* j
17. }   UPNPNAT_RETURN;
    
18. 
    0 w+ K0 C) U) Y' X0 D
19. typedef   enum{
    " x- f( z0 z. Y
20. UNAT_TCP, //   TCP   Protocol
    # b0 ~! A; N( s; j* X* x
21. UNAT_UDP //   UDP   Protocol
    6 {" E! m5 V. T- d2 [2 o; t
22. }   UPNPNAT_PROTOCOL;
    
23. 
    2 M" f; H4 A7 j7 P1 [
24. typedef   struct{
    9 ~9 G; W' r* i
25. WORD   internalPort; //   Port   mapping   internal   port
    % N% n5 a% a3 U0 e
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;
    
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
      ^0 J* s  W* R. `: F- U
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    - @2 S8 O( G8 d7 y! H
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    5 }" o: g: \- W& L4 B
36. void   clearNATPortMapping();
    
37. 
    5 g! ?: ^5 j8 |: K' P
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    - v) K2 ?" j2 [
40. WORD GetLocalIP();
    5 i/ O/ G9 D, q+ \5 c
41. bool IsLANIP(WORD   nIP);
    ) W* i7 a' M+ T# P( p0 o3 r" y: T
42. 
    
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    ! d9 \: K7 Q6 A1 y
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    5 p* ^( R* @$ U% d6 g9 x& l
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    & p; O5 c" t% e; M$ G' M. A
50. 
    3 d* u9 Q8 H0 E) A% r6 {  p" E
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    8 o; S  X1 j- r
52. 
    
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    . r) W" K0 }7 N' G% d9 i8 L' p* e
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    ; J! K( R3 f, ?+ m  l3 Y+ Q& c9 R
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
      e- ?; b6 b: X) P8 H- t
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    / ]$ p: \9 W1 c, Z
61. CString m_name;
    * T# P7 \# n9 a2 a) ~" r6 P
62. CString m_description;
    & m( B6 q9 c  h" m- c1 Y
63. CString m_baseurl;
    
64. CString m_controlurl;
    6 _* f; H* S. F' W7 P1 u) S
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    , O# P- j) y5 p6 }* }
68. 
    
69. private:
    # e. k8 q, M# I/ ?5 E
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    / N7 m4 R  U" K; |
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    . \+ u/ b# ~- L; C" h3 j
77. };
    : v/ u* e" B2 U. }! n
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   : B6 F; Q' D4 `8 L+ G6 |
4. #include   "upnp.h "
   # [6 K8 L) l; v5 f/ E
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   ( e- J" D& m; C
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   / u2 P2 Q: @1 ?( ~/ }) D! Q
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    " n. K$ J! H/ D& x
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    $ }( ]" R. X) D
13. static   const   int UPNPPORT   =   1900;
    # F5 \8 i+ }7 p/ q8 C* E& r4 y
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
      g3 Y) O3 N+ [3 W  F: ~( @: U/ E; e
15. 
    ; L5 o# q) S. p" X
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    & [# m9 L/ }/ ]. W
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    6 S7 ~% F. e5 y" L: o
22. return   s;
    
23. }
    1 v+ d* E. T4 d& p
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    4 o1 G; H$ P% C  S* g
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    # H# G1 k; Q3 q! C2 k* Z
28. }
    
29. 
    + w  b5 e8 Y/ J- D
30. const   CString   GetArgString(const   CString&   name,   int   value)
    5 ]' C" Z4 ?! C; k- U2 _
31. {
    ' b! q7 [( i, F  }+ @$ X
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    $ g1 b7 \; Y0 }( P9 @# `# L
36. {
    0 _2 ]  o1 p- ~" m% K
37. char   buffer[10240];
    ; \  \2 g2 e6 d4 c" c3 E( c
38. 
    4 u7 c( x' E2 r' c, H. E8 q4 \3 E
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
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;
    " M8 q+ L$ R2 e; f7 z
47. sockaddr.sin_port   =   htons(port);
    4 W4 Q2 P; [. H: w+ g; G
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);
    * E: g) R2 y( U! F
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    % ~4 K9 X& R( ]. R  U7 T+ T0 ^# N
53. Sleep(20);
    ! ^' [$ ^% s& W  f
54. int   n   =   send(s,   buffer,   length,   0);
    # @( J8 }+ s& f7 a: F! g5 R9 q
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    * Y* C: a$ B1 a# c+ y6 w8 J  A2 r
57. closesocket(s);
    - ?3 E6 A! l7 `0 B: {; |
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    & X7 W5 t, }8 @
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    0 K1 F7 G& Q6 ]! ^# a& [
62. 
    4 s2 e* d9 {$ _+ T) H& v- e4 x/ y
63. return   true;
    
64. }
    4 f' D. r- ?# ^" q. |5 l
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    % v5 d% b9 V5 X0 U6 p* X) s' }
73. 
    7 i: z& f9 c* t2 W" e) r- \6 c, Q# v, T
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    . J. B9 }/ c& a' z* {4 l- W. V
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    1 v6 E/ W' [! I4 S7 M
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    ! a5 I( `! P7 A' W: }
82. 
    + g: a& b) \! ^( J% j
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    * M9 w6 I& D. V& A) @
84. {
    
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    " a; \/ h+ Q. F! k$ x" {
89. result   =   response;
    , \) P# V  n, D$ I# }
90. result.Delete(0,   pos);
    0 l. m4 T& y2 z1 @
91. 
    7 e# v) k5 A/ P
92. pos   =   0;
    2 i" T, n0 o! }; T. u5 i
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    . U8 Z7 U$ S( s, c" G) w. F% b1 {
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    2 X& G" j& S9 E' o* S
97. }
    + F& f/ t& L! l) y: {: I# R
98. 
    7 h' y" h! w- Z6 I+ w* P
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
      w/ I7 P: i+ S+ t# x
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     & j- f2 K  o, Y8 D
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     : I$ v8 }1 \" [. T" I1 i
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     ; e) J+ I, y0 I1 c1 p
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     1 X6 f5 L. b& {, S
109. if   (posStart> =posEnd)   return   CString();
     % M2 @/ s' @8 C$ b: S
110. 
     3 T0 m% p- B  S5 ^/ x0 m3 ]
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     ! g  x0 n7 N! M% E
112. }
     5 N* r  u8 g: V4 x6 w. |
113. 
     
114. MyUPnP::MyUPnP()
     , i- s8 g; d  |' H0 L/ Y8 f
115. :   m_version(1)
     ( A5 B0 C$ F0 v. i8 w5 s
116. {
     5 A- s- U* u1 h; n
117. m_uLocalIP   =   0;
     ! L: d7 w! W) k. f% R# W6 W  m
118. isSearched   =   false;
     
119. }
     ' [. w8 g6 a) q5 c0 Q2 i1 a/ R" |
120. 
       s! u& u1 w2 v
121. MyUPnP::~MyUPnP()
       v/ O. ~6 v+ k3 B
122. {
     
123. UPNPNAT_MAPPING   search;
     9 }$ B- m9 o+ j$ a  H# }0 H
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     ; l6 x) }# v) {3 b0 @; v
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     
131. }
       `. e& k. X2 K, C+ e
132. 
       O+ r1 {$ W% q; ?: Y$ E
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     ' O+ a0 b6 R: f! N0 X5 ^) T) g
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     $ i3 n6 D- _9 p, X& m; L& ~
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     7 \+ Q8 u& }) T% t
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     / M7 K# |/ |/ Q6 s: L7 M
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
       t& k% @+ m, ~) ]' x8 x3 s% ]% n
147. u_long   lv   =   1;
     " v8 l+ v! V+ k4 w8 v3 a5 P, }
148. ioctlsocket(s,   FIONBIO,   &lv);
     * ]6 R* Q2 `. T. c# w
149. 
     / q) K8 A- f" Q2 t  V& s  x" k/ `( t5 z
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     2 D$ w0 W5 l4 V
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     ' V3 k' T4 J3 p4 M4 C) E
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);
     
159. }
     
160. }
     
161. 
     
162. Sleep(10);
     5 P1 |& W6 M% F5 e
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     / Z2 ?3 n/ j+ @
167. closesocket(s);
     & u3 }  D) a& M
168. 
     0 r( f  O" _% Z  \
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     3 h3 |: A5 B; z5 J4 s/ `
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     8 K* ^6 |* H0 W' l9 |/ i9 y0 m
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     ! o# I; z; K3 j3 ]
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: "))   {
     4 g$ {, y7 Y# s* g  O" @0 E
182. line.Delete(0,   9);
     0 Z/ r% q" x7 l, g3 T
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     
187. }
     
188. }
     
189. }
     ; d5 v( S2 X+ }! \  r# m
190. }
     
191. closesocket(s);
     & G# ]- m8 C. ~. a( \0 }
192. 
     - {- x: w2 B1 u2 j
193. return   false;
     2 u. L/ h: _- c: c+ @6 }
194. }
     4 j# a3 }( y" a5 w- a+ M

复制代码

天云上人

以下有关upnp的接口来自emule，
( R3 G# S& R/ p" A. x

///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

3 z7 O$ j0 k; \. u: `# O  ~* w" S
) n" a" Z0 [9 V" a9 T: v#pragma once
#include <exception>


  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
' D/ b" }' U  S: g6 I# L5 h5 u};
/ @8 N2 m3 M* p  u
! q4 H! d0 X# L9 L, R
3 \( D) M9 P* J/ |enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};
# B8 `' r& W3 p$ {

( M6 \# m5 g  ?( V& l' M" `
4 {, ]9 S- j, D, ^0 p  h" U
class CUPnPImpl
; R% K; }' E- [. H; {{
public:
        CUPnPImpl();
4 ^8 E, ^* i0 j. f) J        virtual ~CUPnPImpl();
8 {% ?' k3 W% B, R        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
  f' `: G; X0 h5 F7 C* N: X& J/ x) s        };
. W* Z, X  k0 [) D
7 ?3 p* o6 Q, W( }
0 b7 n' I: k4 x1 K- u+ H( b        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
+ m& r' z+ n9 F/ P        virtual bool        CheckAndRefresh() = 0;
0 v, K( _) n! h2 m: P5 j9 r        virtual void        StopAsyncFind() = 0;
- ]$ Z2 e- P+ z7 \        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
0 h( g! h+ \, r' T) ]9 ^        virtual int                GetImplementationID() = 0;
/ ^7 c" i8 m# e3 f2 h! f' Y       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
  |0 M( e) T0 a0 e3 P

* A6 j) F5 @* K        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


// Implementation
protected:
0 C! R; d1 M9 ^+ O3 f- U- s        volatile TRISTATE        m_bUPnPPortsForwarded;
5 z0 V3 D0 z: G7 A/ L+ U        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;


private:
7 T/ L% M5 J% U0 p9 k# i        HWND        m_hResultMessageWindow;
5 T* ^. B+ Q+ [$ x: R" P        UINT        m_nResultMessageID;


};


// Dummy Implementation to be used when no other implementation is available
' P6 Y5 B) g, v; F( x( Qclass CUPnPImplNone: public CUPnPImpl
. v7 a% @# j! ~, [# t0 A{
" d6 P' K" O# g8 n% z; p/ ^, n8 Fpublic:
( G) b+ s% U$ D+ |# B7 Q' S        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
: b( L4 V5 {) T7 j" v        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
; [8 F* z. t& h5 |        virtual bool        IsReady()                                                                                                { return false; }
1 F9 U- L9 c% I        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
$ a; G) x- G. B; p9 T};

" Y5 J0 q( |; t+ O
/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类


#pragma once
#pragma warning( disable: 4355 )

4 K+ }1 S3 Y; L
#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>


#include <vector>
#include <exception>
8 z+ I9 V) ^' k6 F9 J#include <functional>
0 ?( n( t; U# N% B* N+ v; `


# X; S/ R0 e( @6 Y) k6 H; A9 H
typedef _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;
1 @" Y( t' I0 B& d0 E8 _typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
7 v% L% j4 d2 `" k/ Gtypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


7 Y" q1 r$ K# |+ F1 V3 c- c# stypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);
$ l( u& k! f* _
5 f% e+ O7 l) u7 {! ^# [& L* G
6 A- [( Q; p4 O  ctypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
+ @0 y1 `$ n; R( f+ [) R);
% P* H* Q. J7 |# a

( Y8 `/ X3 F0 ]0 m( {typedef DWORD (WINAPI* TGetIfEntry) (
. D& z" T* q* y& L, C  PMIB_IFROW pIfRow
);


CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
. j1 `( F, p# [  ?% w

class CUPnPImplWinServ: public CUPnPImpl
{
0 w1 h, L6 H9 c5 [: T8 u$ l: N3 X' h2 S        friend class CDeviceFinderCallback;
& y7 u2 l+ V* \; x( F2 u        friend class CServiceCallback;
* H7 \# I7 R* D" j' r// Construction
public:
. O6 }1 z1 o* K; k$ }% w# \4 Z        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
* U, |/ P* e5 @9 K+ z, I; @5 y: i

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
) [' ~' j8 O, Q/ E* i; w        virtual void        DeletePorts();
        virtual bool        IsReady();
7 E, A+ P; S6 t1 r) C" o" |        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
+ A! T0 A7 @' U3 C

        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
6 @4 K, |7 g, \9 V. r        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
( O1 d: b+ F% t( S8 r1 A        virtual bool        CheckAndRefresh()                                                                                { return false; };
7 f$ x1 i( t/ }- |8 ?: l, M7 n8 E

protected:
/ i& H" b+ o- @" z/ L        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
7 y2 r* r; S9 w8 ?, S7 q        bool        OnSearchComplete();
        void        Init();
& U3 h3 O/ ?& g- H* R+ S
( z9 [( C8 |9 _( W
; r/ M# D( C8 h5 W  l: |        inline bool IsAsyncFindRunning()
  H  Z4 x' f  Y9 w3 R        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
# n: W9 ^/ s, d6 ]6 O# |1 \* D# i                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
( Y$ n  I7 c/ p- Y$ z" j- U& X                        m_bAsyncFindRunning = false;
$ w5 u, Z; {! H& _: z                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
. q' N( E' o, f1 d                        DispatchMessage( &msg );
                }
- `- C8 e; v) d+ K3 I                return m_bAsyncFindRunning;
        }
8 }$ Q% |) S6 T' M
0 P6 |/ }/ E0 P) S
* C& T2 I: X9 \: L' y; _        TRISTATE                        m_bUPnPDeviceConnected;
! s# e% w5 ~! s1 c: }

8 h4 T1 Z+ o. T# g// Implementation
' S4 R; `& y( l6 @( [& ~! S6 |        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
. ?( G0 L* S% @5 ?& F9 y" o        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);
- @% i& G7 I8 Q2 `( S( V5 Z
0 T4 d  _8 S. [. P& {0 c
7 X% ]9 U; \1 m4 r; L        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
1 I+ ?! C7 ^' g* x
8 ~! T9 q  e7 q8 j* O
2 V9 j* p7 Q8 S7 \+ r& }! ?  Z        static FinderPointer CreateFinderInstance();
/ i% p4 ]$ [3 I( A+ ]5 [* _        struct FindDevice : std::unary_function< DevicePointer, bool >
. O9 S8 _2 K9 {, Q/ W4 S        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
: f& f8 _/ Z6 r: v8 C* [- |! Q& R                        CComBSTR deviceName;
. @* C+ r+ s8 E4 @& N                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
) ~$ O1 r' N( g+ V. O/ i
/ d7 b9 F7 F: ?# K! w
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
  g8 A3 k0 v* R7 G7 j                }
                CComBSTR m_udn;
        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
- d. a5 ]; V8 a, |1 \( r7 K        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
2 `+ |$ p# d3 \5 m7 a8 @        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
- p9 j& |3 c3 Z4 y1 e                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();


        // Utility functions
7 y; z1 b  q5 Q( t# c+ P        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
, ^, Y0 `2 H3 ?, u6 n        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
8 ^) W, K( J6 g6 a+ ?7 m        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
* K* `; z! b( x6 a1 X
$ ]! I) X( g) [  j) f
5 \) ^5 W. p% \6 Q5 D- f9 R+ q// Private members
private:
, }% }8 r& `: U# M! R5 c* H+ G        DWORD        m_tLastEvent;        // When the last event was received?
& z0 E! _' C6 V8 o" B8 W        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
6 m9 y) o) U: j9 k2 P7 J        ServiceCallback                        m_pServiceCallback;

. m+ q/ m$ i7 a# l
        LONG        m_nAsyncFindHandle;
" R( c; D! w2 U        bool        m_bCOM;
        bool        m_bPortIsFree;
4 |1 \; |" ?* S        CString m_sLocalIP;
        CString m_sExternalIP;
5 D# K) t1 O, t: u! q+ f+ @        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
8 F. Q; n  {5 y6 h# V' u" @        bool        m_bDisableWANIPSetup;
3 q4 D$ F2 h. q* s        bool        m_bDisableWANPPPSetup;
7 \; Y$ d: n0 b# ~; l

2 l& t# C4 _. N" J/ [+ K};
' z( Y4 `6 y3 |! ^' N  }( I

, Z0 |. q$ u6 S8 U5 ]9 k, @// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
; C' |$ d5 d8 C& W: l7 @$ E        { m_lRefCount = 0; }
5 w: R7 A  W8 f! s5 [

   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
8 W! P) y7 j) a+ H   STDMETHODIMP_(ULONG) Release();


// implementation
private:
* z. D9 k% B0 l# W, F0 ?        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
8 `2 ~; i% x( i' Y+ F, r; ]        HRESULT __stdcall SearchComplete(LONG nFindData);
8 d! i* N0 i$ L; J; S. B6 X: x
( ~) x4 g% r6 v) i$ k% S1 U: [5 |
, o; B% j/ C& ^( H8 }( i  Cprivate:
        CUPnPImplWinServ& m_instance;
% F7 b* a9 ]* i" }% o        LONG m_lRefCount;
+ A) e, b. t) w- o5 C, ?  K2 c};
, l) m2 T# o- L% g3 \
( d+ o  B; C! F- [( x
0 O2 `/ n4 m4 r% F// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
) }3 N- S# v! ^0 N) m. a{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
& V3 p( Z% `7 }6 U$ ]* J        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
# T" b0 Z2 i8 Z1 r   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
" Y5 e) T- k1 p: I
  D" _1 |4 _6 v  t) `
" e9 ^9 n3 R5 s- J- e) v// implementation
private:
* F( J  R2 r/ o: J* T        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
- @$ H! v) H' @0 y- k        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
, T. A5 m$ T, Z& \: f: n  e* w+ c
% w0 U! a1 c  _3 E8 u
; ^& [" f1 Y7 }# v1 l" `0 Tprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
! l1 k& P  k6 E2 G9 |/ a

( `- C8 `7 v7 y; V; c) {4 ~/////////////////////////////////////////////////
% O* w/ W- w. H8 R, }6 m% }

使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
- h1 P% R1 _' zCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
9 s2 o0 o6 Z9 CCUPnPImpl::StopAsyncFind停止设备查找.
3 G) u7 l2 g+ c& W+ l- SCUPnPImpl::DeletePorts删除端口映射.