UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   5 k1 v5 d1 `; e4 ^! h7 R
3. 
   
4. #pragma   once
   1 h. j& X+ w: N9 m2 A3 o' M
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   $ Y4 W8 h$ O0 c9 K: e1 S
9. {
   
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    & u3 g/ }4 @3 y& O+ F- B* I5 B
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    : g' v( x3 x6 k
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    ( U- h+ x3 Z8 J  c. @
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
    % u8 R9 r% e* f: G* v
17. }   UPNPNAT_RETURN;
    
18. 
    5 U/ r. Z4 k* a0 _6 i* p
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    " M/ R) b! j. `' r% T
21. UNAT_UDP //   UDP   Protocol
    6 d% ]  z. C1 I: b( U2 E
22. }   UPNPNAT_PROTOCOL;
    $ E  q$ p( \+ V' A0 B
23. 
    
24. typedef   struct{
    ( d+ j2 H, {$ r1 v% Z3 H& j, e7 \- e
25. WORD   internalPort; //   Port   mapping   internal   port
    ( V- J# j$ u* u( ]: ^5 k0 u( _  \# Z% t2 a
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
    3 Q: t; O; n) C: k- `$ X
29. }   UPNPNAT_MAPPING;
    6 P# X9 E4 ?* |8 y: s( V! v
30. 
    
31. MyUPnP();
    ( B  D  E- x+ w
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    4 D) ^9 l9 Y7 }3 w
36. void   clearNATPortMapping();
    3 X& r8 Z$ l3 Y
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    - a6 _/ u8 h2 [/ \6 X7 z
40. WORD GetLocalIP();
    . L+ @( t# e  l& d
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    * e0 i. j  W& d! |* r
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    6 S% b$ P& O) H; R
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    ' V, E1 G1 L( b( i" n- Y% O- P. [! `
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);
    & h5 n% r1 j# X1 E* a
60. CString m_devicename;
    
61. CString m_name;
    4 y; [* V  x% p" Z$ T* M
62. CString m_description;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    " _- w5 d: m5 J8 A6 y
66. CString m_modelname;
    7 D, d; P% f' w
67. int m_version;
    
68. 
    ; x7 P- }; b: b. W
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    9 k. i" Z' }# H% w8 C5 x
76. bool isSearched;
    8 f6 a6 G' h* M0 _3 o
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   2 R* ^4 ]4 ^4 u, O5 L$ G
4. #include   "upnp.h "
   * A' @! f' k' P: y% O# R; L
5. 
   $ a. F# O1 s9 k* w" c
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   ) m. @1 v8 I9 ]5 v+ l2 x! V3 e. o3 i
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   * o) n  G. f, C) S, [. U9 |
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    * i( l7 ?9 X" x1 N2 _
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    7 }, }0 d) E3 X3 t1 G
13. static   const   int UPNPPORT   =   1900;
    6 K# Y4 O5 z5 q" R
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    % Y1 q% k" J3 l6 Z  K$ k( C
15. 
    
16. const   CString   getString(int   i)
    5 F. p5 N, U. v  l
17. {
    
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    & C& T9 Z/ s4 r) e0 |# ?0 Q, v+ y
21. 
    
22. return   s;
    2 m3 d4 @6 M- ]9 \1 {4 v& I
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    : \- V" N* j3 {6 w( J% i4 R# {
26. {
    ! j/ B& X: i% d% _2 D
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    , ^& g: u  D% J3 r! ?- p
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
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)
    + W( _0 @7 g1 I( @" w; z( S
36. {
    
37. char   buffer[10240];
    
38. 
    1 o, I- N) U) y
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    ) g3 d6 y" m- q/ X0 {+ T' @$ }
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    $ D. A2 c( u' U, A" O5 _
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    2 l4 p+ F# h, o* R3 n
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ) l7 k% z* q! P2 ~/ g" A
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    " [4 c' ?" V+ _+ J1 f* x' c
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    ; Y1 W6 C9 d3 D0 I/ P* W
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    ' S6 f' H7 c- }: b/ u
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    ; y) N% H$ t: E* S: m0 W
57. closesocket(s);
    8 s! C& A5 U, [3 M1 K( \
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    ( i1 p6 y. F' I, e, o
59. if   (!rlen)   return   false;
    # k( @. M6 G" I, |2 d
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    * ]0 T, b: s% S+ T5 [- F
63. return   true;
    $ P. f8 B& |. |. ]( L) a
64. }
    % h! u# D. c2 a& E4 [* B
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    / B& }1 z) h6 ]" J6 _' R
67. {
    9 L; E- g; G9 K1 u5 I5 ~6 d: U: R  T: I
68. char   buffer[10240];
    
69. 
    : z# {- ~8 D1 R7 h7 S2 A& D# m0 g
70. const   CStringA   sa(request);
    ! i% q" j, W. l
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    1 h& u3 X) V$ ?" o
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    8 _3 ?# U- I' p& Z0 e
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    . X8 G! H8 n8 v4 r
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    ( {0 _) q- d9 q% A" ]
82. 
    6 r. X; p- V4 B" M
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    # u4 R; `$ D- T
84. {
    * H/ i! [& T* ^9 @1 U/ X
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    , Z1 t) @1 Z$ L+ I. o6 ~+ k/ ~
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    ( t2 t' G& q1 z6 t/ a
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    ( X+ i' K9 ]% s$ i8 ~- _
96. return   true;
    
97. }
    
98. 
    5 k' H+ o: K% J1 r# Q9 s. d) j
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     # T# o6 y8 y! d5 w% H8 G% q( M
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     # j. H% o. Q2 q7 C+ |, I
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     % _( y: ^7 T5 A+ G
107. 
     8 F" s  T( @3 K
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     1 }8 H4 r, M- W# g) n6 R5 D+ b, s
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     8 W# O# w% Q: o5 p" d& V$ }2 c
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     . F- W. J" d0 K
116. {
     
117. m_uLocalIP   =   0;
     " [* d0 g9 f5 m8 O& A$ x( E6 Q4 B3 o4 |
118. isSearched   =   false;
     8 N6 N( l$ @) B, I; J. Z1 X
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     6 [5 X' ]8 e; o. P% Z5 e! K  P& @) f
122. {
     # ^/ g7 Y; m9 A7 ^1 ?
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     , l: I  H$ ?: I; U3 @5 ^1 Z
125. while(pos){
     5 |1 q$ n, h8 M( n8 J$ @, i1 G* S5 J( l
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     
131. }
     8 }2 L2 b) h* L
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     1 T* X4 V; b& H' ^2 Y: G7 ]
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     + U5 F2 J) j5 _8 [  g3 q( T: G( f
138. 
     " K- v9 H! J( g4 F+ i8 j
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     : V- ?; ]5 l; k. Q. ]- f! t' [
141. {UPNPPORTMAP1,   _T( "service ")},
     % e& }  }9 v7 c- g: U9 p: }% N$ I% u: W
142. {UPNPPORTMAP0,   _T( "service ")},
     1 p9 k* q  }( t. U! b- s. F
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     0 @  ]1 H: I3 l* H! [! q
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     : _  v7 X9 `& d, i, l0 d! _) A
147. u_long   lv   =   1;
     : q* N1 L" l+ |1 @; t) Q
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     # Z2 q: V, T7 U% [
150. int   rlen   =   0;
     $ t- o2 c$ f: T0 b& [, ?
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     . J7 ^5 f9 `$ n1 [$ e
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);
     0 U6 \0 |) W9 n' G
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 "),
     " E9 f& v  \( U& f- g' E
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     # S) E1 |' O' O3 m& @! l. [
159. }
     9 Q" R& b' G- T& g3 k( t
160. }
     : b8 S: Q$ g' |, E8 s
161. 
     
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     2 v+ t6 r  {6 Y$ h
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     2 C" g4 d0 {# `
168. 
     9 R6 s4 {3 J4 A. t2 V
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     / F  ]; K+ B) N& e" d# o4 k7 J
171. if   (!parseHTTPResponse(response,   result))   return   false;
     # y( G+ n9 m& t$ m" o: }
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);
     ! b& {: o! M9 \' }0 _; q
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     4 \2 X: b9 b7 U, |. v
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     5 m' w; i2 H8 n' ^% V
178. if   (line.IsEmpty())   return   false;
     4 {' O+ K3 j4 p
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     , G/ }9 ^! W5 |2 H" m! y9 u. p
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     ) c6 Q, q3 A% w- M8 T% P$ S
186. }
     ) J1 g- T" u/ N, j1 ?1 N
187. }
     
188. }
       p. I; K; x9 [5 D, ~
189. }
     
190. }
     
191. closesocket(s);
     % M5 ]% c+ q9 n* W5 a2 j( [" V, O( \
192. 
     
193. return   false;
     
194. }
     $ t  c3 G4 f& \! J2 P; A" \  F

复制代码

天云上人

以下有关upnp的接口来自emule，
  R, w9 c. F5 q" j5 e( G2 S3 A! Q" v
8 D7 l) p9 b$ c7 A
///////////////////////////////////////////
) h$ Q; B1 ?" `) R* _  j5 _  v8 ?//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


#pragma once
#include <exception>
" Y7 b: }5 S8 E! H1 T8 E( K
' e, n- k) j) n% K
3 c: |+ N1 u$ G! v  enum TRISTATE{
1 f% Y& J8 J& W  c! F        TRIS_FALSE,
, h9 T, }# I3 m0 x        TRIS_UNKNOWN,
        TRIS_TRUE
};

" u. L/ U5 R6 |  c$ D( o0 _7 V* {
! N' h) W& R8 P5 Benum UPNP_IMPLEMENTATION{
3 ^2 t( E% i; u2 k! g        UPNP_IMPL_WINDOWSERVICE = 0,
. y/ U. t9 \9 N+ I5 Q        UPNP_IMPL_MINIUPNPLIB,
$ \. z: R% C% V. C! ^        UPNP_IMPL_NONE /*last*/
0 X0 y" p# D# G3 C  S  e  Z};

" |3 s( b/ m8 m1 h6 r+ r8 E+ Z
  H* }9 Y, X( }8 f+ w) A
$ o: C" a2 \9 Y% P/ Q# ?0 A) ?2 b
class CUPnPImpl
{
public:
5 w1 l7 E3 k+ i        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
  h& v5 I7 G2 V. y8 F8 G6 w, J) ^        enum {
                UPNP_OK,
( Q- O" U4 b1 l: V* i                UPNP_FAILED,
                UPNP_TIMEOUT
        };
0 `7 Y0 a2 S9 _  V. {9 `% N, S

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
' Z! W9 ^: k! y% L. y        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
: E2 o# E2 V; R4 i4 H3 J

6 K8 M) ~4 n7 I8 D3 E3 ]        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
( F7 q( O, @  ~        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
$ e' @( I% k- ~$ ~) _        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
+ N! z: x3 q4 @, l* E$ G
- E* M8 d6 u3 k5 I  ]: v4 S
// Implementation
( I4 |+ Q. ~: _4 ]4 W% Oprotected:
/ A; s/ y# R$ ~% l8 V; i; @( q6 j        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
/ R$ d: ^" \. x& n- U        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
, z4 t, S3 g, ~- @% w* p5 w        uint16                                m_nTCPWebPort;
* \+ s+ v1 T+ ]2 E9 w7 g6 j        bool                                m_bCheckAndRefresh;
0 \4 a9 c7 `- ]6 _

private:
        HWND        m_hResultMessageWindow;
, O5 U5 d4 X* v& o        UINT        m_nResultMessageID;
  w  `: v' S0 b5 c

* Z- ], N: `" i% O: d3 v};


1 E( A% j% b6 X+ `7 ~9 Y/ }# Z4 o// Dummy Implementation to be used when no other implementation is available
# k5 x) q0 s0 B) d5 Z" Iclass CUPnPImplNone: public CUPnPImpl
5 \& B! W( p1 X0 g{
) [* A  ]- G' B. }* D+ ppublic:
4 C% b+ l8 `3 w' J+ Q+ C        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
* M, X  b  D7 ]7 M        virtual bool        CheckAndRefresh()                                                                                { return false; }
8 j( z, Z, Y1 V* `& u4 {        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
/ t/ d6 ]" K9 e1 M7 b% j        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
& h2 i- V6 R* W4 U( S) r};


/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
  E  D' ]0 a* Z" O' X

8 N& n! Y! S1 B$ z! {  n( k#pragma once
% N0 ?  W4 [# K0 q#pragma warning( disable: 4355 )

1 m9 g( e) L' G. r' k
2 r& r$ ^; e  J* j- @& O& L2 f#include "UPnPImpl.h"
; a7 l4 Z* u! f& X#include <upnp.h>
#include <iphlpapi.h>
; r: S& e$ p' n9 h2 b#include <comdef.h>
9 `6 H, y6 N* i#include <winsvc.h>
: C. e1 X  h% c5 N1 q
* X* r' A, \% @. ?  H
. Y1 r4 O- @6 {) E! S) }#include <vector>
9 f2 h+ y0 d) c0 S' m9 _#include <exception>
: C+ A: Q( Y- i9 _#include <functional>

6 S9 Y" N. z& `( q! k, ]
0 Q" }* k" U; [6 k1 v

2 b% A8 L: N# \+ a% atypedef _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;
/ I1 _9 e3 X; C6 {typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
% B/ @* ^. L8 ?: ?% i7 Qtypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
9 `5 Y0 @6 K5 X1 J2 G: T* v8 _typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
( D# o+ t5 ~& K. otypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
/ Z. h+ N! g& y& o) {- [4 V
1 d- r4 |. f, n2 z' l1 j
typedef DWORD (WINAPI* TGetBestInterface) (
% s& H; N& @  m4 @' H4 l% s6 x  IPAddr dwDestAddr,
1 \+ y) x  K0 x" u( U  PDWORD pdwBestIfIndex
# p/ p* A0 s5 M8 [" |  g);
$ B8 Y* K* w% k8 E

% r4 p2 l' Z! h% B) D5 }4 E" \* otypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
- \$ y+ ]9 N( M! k& I  s  PULONG pdwSize,
: s$ V* _2 i" ~8 t' G) R: t  BOOL bOrder
);

5 I$ Z0 d2 p: K; C3 `5 _
/ \6 U2 u- ~# B' Q0 }typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);


% V: n4 I3 B5 E$ f8 I: jCString translateUPnPResult(HRESULT hr);
8 U$ b+ g5 x6 r; G8 H% n: gHRESULT UPnPMessage(HRESULT hr);
0 a; Q0 p$ w( c
' f4 ?* `7 H" B  l+ Z
3 N# }$ \. K/ V- ~6 c' Aclass CUPnPImplWinServ: public CUPnPImpl
7 ~+ U9 g8 k* w{
" J4 y1 ]  l' ^8 c        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
: E2 ?* E! B. H4 R6 K5 E& ?// Construction
public:
& G" B" `6 ]5 z! y        virtual ~CUPnPImplWinServ();
; N, y& {1 `% P8 r2 R        CUPnPImplWinServ();

3 [7 @2 ~: z$ n% a
0 u' v! v; @1 C* D# t        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
/ Z! C" @7 H7 l
# v8 E( @( I) [; W/ @
; |2 t6 M, k  j7 y1 A/ g        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
0 A- M* ~3 V) A        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
/ o7 d: I5 K2 S4 P1 M1 w

9 s2 j, r( |& e  X* q( n/ p% \4 A7 lprotected:
9 W# U$ S; v; z0 G, l' _  Q3 b        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
' v/ B; _1 c7 p( \( n% h        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();
8 D6 i5 ?' @) J% l$ r

        inline bool IsAsyncFindRunning()
        {
8 d" _; b1 j# Y* c( j0 z: N9 M                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
, r! g# v2 r) a9 w5 J" Z! C' w                }
! w$ Y+ J+ y$ z                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
8 {3 _8 t0 G5 }8 n, \                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
        }


        TRISTATE                        m_bUPnPDeviceConnected;

9 V! D. h: P2 U+ j, _8 z
// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
2 f1 s9 e! L+ z" A0 c7 u        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*);
+ [7 x5 `* u+ x$ j; R& Z        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

% q" z  i: ?1 M$ V. F& }
& b, c* M, i! H" c/ j9 \        TGetBestInterface                m_pfGetBestInterface;
" O, w0 M8 ]/ z6 t2 ?        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;


' o" S+ @  V" @$ v1 p5 [        static FinderPointer CreateFinderInstance();
* ?5 V$ k  G+ ?        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
0 e# Y# h$ t4 A; o& Y! a1 H                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
/ ^0 n2 I0 R& M/ x                {
' d, r  o# G, z                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
2 j4 Z+ v' s  [' ^0 D  T

                        if ( FAILED( hr ) )
+ [4 ^; o5 E2 K% N& A  i                                return UPnPMessage( hr ), false;

( Q6 U2 ]3 H3 ~4 z0 d
% ~& }2 j% L7 K+ e+ o                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
5 z) E. b$ s  E        HRESULT        GetDeviceServices(DevicePointer pDevice);
+ {& b% j' {6 g$ ~1 H3 x2 f- |        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
; p# s- J$ ^' }+ v        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
  N4 O8 G5 w; V& L                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();


        // Utility functions
& _8 p2 C: R$ P! A, D        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
, F  @8 A% W. Z  P        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
. G9 x$ e9 S- C7 W" ]0 g        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
" B: c6 n- f, H& u        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
7 k' b/ U) }: V$ [        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);


. e$ n0 p* I  Q/ c' M// Private members
private:
- v0 @: W6 E( N- X6 L& `6 L& o: g        DWORD        m_tLastEvent;        // When the last event was received?
( e# S/ G2 U5 s        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
* X% Y3 r+ X; x) u) [- {        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;

2 }; u' @: h) J/ W8 @1 j
        LONG        m_nAsyncFindHandle;
7 [7 t& X( g7 Z        bool        m_bCOM;
        bool        m_bPortIsFree;
* F' ~2 M5 P) |! l" k0 I% H        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
4 j! q1 m$ R6 x9 S/ @        bool        m_bInited;
  N' y! h  k9 S' j0 K        bool        m_bAsyncFindRunning;
, C& s7 g5 G/ w# G8 z) [! v  S        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
+ ?  ]2 O" J) [' f" R        bool        m_bDisableWANPPPSetup;


};
9 `0 I1 K2 h  ?+ ~; r
# C) I# i1 t/ r% b& q/ U3 h
// DeviceFinder Callback
9 P$ ?2 u/ x6 @  w5 ^. i+ x( G! [class CDeviceFinderCallback
0 k9 k6 l6 O& _' \' A: R        : public IUPnPDeviceFinderCallback
6 h% ^/ l+ v, M" w+ H1 H2 V{
public:
' b# `) D3 j6 @1 ?" t        CDeviceFinderCallback(CUPnPImplWinServ& instance)
1 P7 ~1 @# b- P6 L                : m_instance( instance )
4 ?3 v; O; J4 R  t: Y        { m_lRefCount = 0; }
: s/ o( O2 A9 U; Y1 n- k7 Z2 v

# ?: c9 L( `# Y/ G* T   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
1 Q* n; ?$ V9 h9 Q( q; u; x& X
3 X  v+ X9 V$ {( {7 t" i
( M" C+ q# m5 g( I) G: m5 ?) E// implementation
/ d; E6 t! s0 s" C, x1 h" l& u9 rprivate:
1 q% h* \2 F( [  [* K        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
1 m5 [* w% O# |, y        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);


private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
) ^, m  {& d8 s* U8 m
" I, K  s5 G, C# I: M/ t4 L/ F% M
// Service Callback
( t( n" r: B; @. [! Sclass CServiceCallback
        : public IUPnPServiceCallback
{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
4 M. I: c; n. ?9 `+ X, j! |   
) ?; }! l  j4 f   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
4 t8 F4 M, @0 j   STDMETHODIMP_(ULONG) AddRef();
! v* L+ C, u- ]" Y) H: \* z   STDMETHODIMP_(ULONG) Release();


  P, p' o. N) G# b5 R7 u// implementation
private:
% G4 y' @6 Y/ D" X* K! [        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

* X6 x- q1 l( ?% T- d
private:
9 M# R  H+ \& n$ ?        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};


/////////////////////////////////////////////////


使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
2 k3 c: R, e$ d. c1 A: Q3 J9 GCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
! n1 i& R% n0 P2 t; DCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.