UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   $ M6 ]3 g( n: p- \
3. 
   / |7 T7 L( i  L6 K) Y( O5 x
4. #pragma   once
   , _4 c" q! O+ q
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   + ~4 G5 m. R! W; G0 Y
8. class   MyUPnP
   " k$ X8 ~! w3 T+ d9 A6 d$ {
9. {
   
10. public:
    0 m) q; B( c7 C2 [5 ]
11. typedef   enum{
      @' n5 ^: M) c- ^4 L0 x" U
12. UNAT_OK, //   Successfull
    
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
    ! G, J4 W% [3 }) E1 D9 k
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    * a* B& M- o2 p, w4 c6 E
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    6 A& R0 c0 R: x4 @6 |  Z6 M0 Z, ^
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    ( s- E% i; T0 U: d
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    - s! K* Q: G% `% l( E, o7 Z
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    / S1 x) }# [1 @0 a
29. }   UPNPNAT_MAPPING;
    & A' X0 J) M6 B; U+ B3 E
30. 
    / R8 h# R  X9 ~% r
31. MyUPnP();
    
32. ~MyUPnP();
    - T9 v, L$ O3 \0 d
33. 
    4 r* L( C8 N7 g8 s7 {
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    ! E" \% m2 I$ w1 @
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    3 P/ S$ F( R& n* X, Z
37. 
    : Z" `' Y5 w- I  @2 j
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    
44. void InitLocalIP();
    : J- q8 V3 u! q/ z. G+ H) x
45. void SetLastError(CString   error);
    ' ?3 \7 t6 \9 A
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    % K0 W3 E0 Y% U% u/ D- W' U
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    0 P' M, c, w; P) y8 @4 \1 `
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    % s& @+ g3 b( k" P
52. 
    
53. bool Search(int   version=1);
    7 h& J# F$ H4 i& j! v3 @
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    * K9 k. {2 y9 f. H; ~3 A
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
      t) T  c+ J# `6 K
57. 
    ; W% H' j" ?& ]3 p
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    * |' R( b6 u% a7 g7 j! n
59. bool InternalSearch(int   version);
    5 M" l( g- G+ c" c. L- W
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    & M  e- C; ]# u& G& |3 L* `
63. CString m_baseurl;
    9 [$ e  J4 k% c- N
64. CString m_controlurl;
    8 V! z; n% ~0 L# J
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    ' c' h* g4 L$ I8 h+ W4 \) ~
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    - `# ~( Z/ Z9 Z4 H9 P' e
72. CString m_slocalIP;
    
73. CString m_slastError;
    4 i7 w  K' j7 X0 }/ p$ |
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    . M$ o0 I7 ]7 M5 u
77. };
    5 v* o& k& H$ P+ L* x9 s$ x
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   " e: I( l) u$ x# n  w1 U: x
2. #include   "stdafx.h "
   * O" F% F2 |6 E5 m
3. 
   
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   5 W5 j$ q( @/ a
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   # _/ c0 ?/ l+ o2 B( H4 G' L
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    9 @7 `0 J- K' X: T& J
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    ( W+ K6 k$ q- X& J6 }; M# Q
13. static   const   int UPNPPORT   =   1900;
    , T& `4 |! J( g* @" B" g& X
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
      Z. n) o+ R' g: }
16. const   CString   getString(int   i)
    5 J: j" u# b' k! F( V
17. {
    6 }, |! v9 u4 }, R+ S4 x
18. CString   s;
    , p2 @8 q8 b6 y, B& u
19. 
    
20. s.Format(_T( "%d "),   i);
    0 {. t% X4 h* ?8 x/ D" {, g0 n  q
21. 
    
22. return   s;
    / b! Z1 c1 j2 B0 }$ _# P
23. }
    $ U  o, A( A) \1 w9 }
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    # O' v# E( N( q) V& j
28. }
    ) Z8 }% I8 f( e5 ]
29. 
    9 S2 j. X2 r+ C% g  X- U6 y+ d
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    ' `$ ~5 a1 L' M. L9 m/ Q
33. }
    & e- r9 j+ w+ R2 o
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    ) ^7 g' R5 M/ q( X2 `; b, S& s) ]  Q
36. {
    3 ?! h* t0 s0 v" U* ~6 h, `7 ?1 h& x! `
37. char   buffer[10240];
      A( `% p8 c$ n# |( b8 V$ V9 p6 j
38. 
    ) o( U2 f2 O& {: L  o/ f! Y9 g
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;
    ; J5 @; z  ^6 |/ J3 N0 `1 ?6 q
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    - H2 h& ^' q$ W6 H2 l
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);
    
50. u_long   lv   =   1;
    * t4 }/ z9 i! C/ B5 [6 |" p
51. ioctlsocket(s,   FIONBIO,   &lv);
    8 b2 _, N3 ~9 |9 I& c
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    5 Q6 ]" y$ z& ?* N" i
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    3 e- h# K6 ^! D" W1 f
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    
64. }
    9 O& j0 _7 l( H" G( [8 G
65. 
    4 I. r* X  d: F9 R+ a+ {! S# @
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    & q6 `2 N2 x2 n) b& g
67. {
    0 m# c( \- S( h6 w$ ^
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    ( S# h" ^5 m* a  F
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    6 D* r' P. G, v1 P3 n$ F9 G, a5 H3 v; t
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    9 J! a6 Z' c% d
76. sockaddr.sin_family   =   AF_INET;
    . b  ~! e* q" l
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    1 m1 M. i- M9 b6 D6 m
81. }
    0 D& r+ W4 O$ N3 i, v
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    / Q7 j; Z2 J4 q( x( n: o/ o% }2 n
84. {
    
85. int   pos   =   0;
    ( M- f  q1 C8 u
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    0 j5 V; Z( D8 w5 x8 n
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
      P+ I, ?0 l2 j0 ^: [1 E
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    5 `8 S% l3 B6 [1 i' c) o) z1 C
97. }
    7 W* |* H$ ?) }
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    3 Q2 ?5 B" n: R1 w- I& \( t
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     6 {* Q6 x! }% ]' q6 s1 F
103. CString   property;
     
104. 
     * ]9 h4 S# w- e( m6 D
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     7 \$ \6 T+ ?% j
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     $ {) V7 H. H: G' b
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     ) s9 B# L# v1 r" R5 d. \5 Z
115. :   m_version(1)
     
116. {
     6 h; W# Y: h, m
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     9 _0 H& ^; o7 x" H' i( c
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     . G4 n+ Z1 L4 K( D
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     7 [/ B$ o1 p' x3 X9 o+ s5 w/ H
129. 
     6 N, U3 g* h5 y2 L: j4 _( m8 U
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     * g* K6 S' q1 Z8 N  X; @$ V
133. 
     0 J+ ~: d0 R. c& P2 y
134. bool   MyUPnP::InternalSearch(int   version)
     9 Z- P; X9 U' [0 O9 x$ t2 N* a1 B
135. {
     ; l$ j: z  G) Z8 @; I' I
136. if(version <=0)version   =   1;
     ' E+ U1 L7 _( p  ?, j, L8 y
137. m_version   =   version;
     
138. 
     & }4 X. W# K4 `. H6 K0 F7 t0 k
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     2 m* i0 k* J! a  _6 S6 u: P; U; a
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     ) ~/ W( @0 _) e" p; `* k- O
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     2 X* X9 Q( U' Y
152. if   (!(i%100))   {
     8 \/ `4 w3 g9 c4 @2 l# k" f( e
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);
     
155. CString   request;
     . t, S, E1 C2 O( D
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 "),
     9 M. x4 l2 O) D. x) H7 t+ s
157. 6,   m_name);
     , W% ^+ Q; P& Q! B; V
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     5 p) _; I4 R9 ^* _0 h# @7 ~  x6 E
160. }
     ) {$ F, `0 j* E: B' J& C1 W
161. 
     
162. Sleep(10);
     
163. 
     ' e: H2 ]9 q' y5 k8 W' C2 U3 ^
164. char   buffer[10240];
     " ]: b% D7 {8 Y
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     ! {, a" s, G  ^0 O
166. if   (rlen   <=   0)   continue;
     ; v' E1 u2 d9 E9 X" t; I$ I4 f
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
       I' j7 Z2 z: x5 i6 t
171. if   (!parseHTTPResponse(response,   result))   return   false;
     8 f+ z7 M  W+ P4 g1 r4 t& f) d: I" i' K
172. 
     5 s: A$ R) ]' R" K* ]
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     # M2 Z8 y' k$ ~0 H
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     + x2 W) A- o" H# q/ |3 {( i  G
175. if   (result.Find(m_name)   > =   0)   {
     ( n4 f9 f$ C; C9 A0 A4 Q
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     9 ]3 u( ?+ j$ d% ?0 I5 ^
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     " R! ]! B! Y7 `+ G: i
180. name.MakeUpper();
     6 K9 ^: Q4 ?# V
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     * ^; X1 e) X# [2 y3 q* ^2 b: a: o* N. t
187. }
     
188. }
     $ w9 m2 l+ o6 F+ Q( _3 ?
189. }
     
190. }
     2 L/ C$ v  f  z6 L
191. closesocket(s);
     % {% j* C4 V6 x* j
192. 
     ) a9 {6 C) ^" t9 R
193. return   false;
     ' I0 |" u! E0 x% j' |
194. }
     . y7 X/ e+ [0 G+ }* L5 h% a

复制代码

天云上人

以下有关upnp的接口来自emule，
/ _( `, V  B. Y; ]. W: |
/ V3 U/ ?# W% g9 i' Q
8 H- A4 ^' ]+ d9 w///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

) `" P- N$ Q. Q1 q' K7 |
#pragma once
# G. e+ z5 p% C3 L3 t6 [" G#include <exception>
2 M8 _1 @7 I. r1 o, h6 F& J

  enum TRISTATE{
7 p: p8 G' h2 u8 m* }        TRIS_FALSE,
7 G0 e2 f9 H+ ~5 H6 {        TRIS_UNKNOWN,
        TRIS_TRUE
; [, U$ j8 B" o};
  I/ z2 a& Q/ s0 X" ^
1 e; z' \5 C) z0 m
enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
; ^1 [; H! A- K+ }        UPNP_IMPL_MINIUPNPLIB,
& [1 y+ [6 x3 O  I+ W$ M        UPNP_IMPL_NONE /*last*/
7 l+ J8 Q3 c" h" E9 L5 M4 f};



4 k# Q; p$ i9 S7 _. u% W) P( o
class CUPnPImpl
{
public:
  o9 N/ _- p. k- r% k; P+ o: E4 z        CUPnPImpl();
" x2 E1 R# Q1 O: P. p! j( ?: C        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
( a8 |0 y. X# e  b. B        enum {
                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
) v8 L% j- ]) u. @0 \* D( [- L        };
  f5 X1 l; P5 C. q" H
* m6 ]0 X& D  ?5 u+ ], e) `3 L
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
8 l' J4 g) r7 C& B8 c/ x        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
6 V& v* M7 l1 {, e& I4 V        virtual bool        IsReady() = 0;
/ M6 @. I' e- R( |9 Q! U; o        virtual int                GetImplementationID() = 0;
, t5 W" V# i6 I       
5 k3 k" \6 [9 p. P6 M        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


- a: b5 V! X) i. l" b* ]        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
" Z  ~7 w' M5 _/ ~        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
% H0 h$ D4 J% m; h  U        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
2 B6 H2 V6 e, J        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
  S' ~5 g1 H. B* |; e* |8 N! B5 [

// Implementation
protected:
" v) I: s# J2 q+ U- h! d        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
1 }+ b, H7 R+ s        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
4 R% L& j+ X# F9 ]
$ n3 {: d" A7 w' q5 _1 w9 {* x" ^4 B
private:
        HWND        m_hResultMessageWindow;
' H$ x. V) E9 C        UINT        m_nResultMessageID;

, X/ g. `+ L0 M
};

% ^5 i, b/ c% |! a+ h
// Dummy Implementation to be used when no other implementation is available
/ @* D8 }* x$ S8 t- g- I! C% i; Q* T* l: Jclass CUPnPImplNone: public CUPnPImpl
{
& ?# D$ n+ n; a9 t4 M* qpublic:
* n* E2 I3 k! [2 h        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
6 W/ C8 X& ]* L3 A+ C) U        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
( K- E7 e; s% S3 D' w- l! H( O};


/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
6 o, U; B" R$ t* F& S5 q2 _

. [, [4 w$ O+ Q+ M/ _0 O#pragma once
#pragma warning( disable: 4355 )
5 x" U( o6 y5 ]4 ?8 A
, W- t# t! k! N' \
2 R4 r& S& k$ f#include "UPnPImpl.h"
% M# N  G* _6 _$ b+ g& b#include <upnp.h>
#include <iphlpapi.h>
2 ?/ ~! q2 v5 q- X9 F8 h% w1 J#include <comdef.h>
, n) Z7 S; |) t* _# [/ e#include <winsvc.h>
: G+ D6 ?( h1 s, [5 P, W

+ b* Y2 o  N$ A  b8 @6 F% d, Z8 Z0 b#include <vector>
( g! i, a: h9 N! t( s- @#include <exception>
4 `' ^% |! A) @$ J) F#include <functional>
. [$ \0 w0 I3 v( \# ^2 z



typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
* X  k0 t# @3 b! I: [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;
- K- i2 j- U- G# G" btypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
$ `" L; Q# ~; ptypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
- C9 h% C8 j9 K- l* t+ e: T7 a
) d. V1 ^& }. S5 s7 b, W+ F$ E# y- z' c
+ w4 w& ?* ^6 z/ jtypedef DWORD (WINAPI* TGetBestInterface) (
2 i: S8 T" X4 B7 o3 I0 V, i& T" ?% m  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);


0 J0 m# N( _# x  d" t7 c: }typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
8 P# |0 Z8 Q, ]; ?  PULONG pdwSize,
) u# U" F$ E4 R& M, T  BOOL bOrder
);


typedef DWORD (WINAPI* TGetIfEntry) (
; P) }8 z6 @) t6 s  PMIB_IFROW pIfRow
, R8 H. X' a8 q9 J8 @);
7 r1 a& X. a( n- m+ u6 J
+ Z9 l8 b; t, N( K: ~9 X1 \
CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);

# G6 v; D: K4 O% I: s+ _) v( U
$ D: U1 `  m+ l: r2 C( rclass CUPnPImplWinServ: public CUPnPImpl
# b& U0 f' s  W* N{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
+ ^) _6 \. A' Z% G// Construction
public:
  K2 P4 l7 K; n$ m9 l        virtual ~CUPnPImplWinServ();
/ L% c8 F8 O$ N+ |        CUPnPImplWinServ();
6 M- m7 |8 Z/ s% |) o
. C  I/ [, w5 v: \# [: D
0 _# r( K. X6 d0 V& q/ q6 t        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
6 P, A& P$ }  [! @  r7 L        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
% F8 N: g* U8 a8 G* p0 w' [1 U9 Y        virtual bool        IsReady();
# o# V& R9 P7 {# ~! g4 J+ T  Z        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
3 Y, Y- b) L7 `! m1 P
# @1 Y5 K, j6 }2 p) b& C
        // 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
& a$ J1 A# |) g2 K8 A: z) O        virtual bool        CheckAndRefresh()                                                                                { return false; };
; R) p, Z! v9 p' ^' a# _! V( C

protected:
( u; |% A) W# D        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
2 ~0 L$ [5 s9 p  N% B9 f        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
, n8 Q8 {6 Q! [8 _% B; r        void        Init();

" N! I, v) z: N4 W$ U
        inline bool IsAsyncFindRunning()
; Q2 b2 E4 x, \, T: q2 }        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
4 ^6 d; ^% V% `                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
0 a$ ]6 s# g7 d6 b                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
- g7 m$ f6 w2 I5 O0 V$ M0 K. _                {
                        TranslateMessage( &msg );
1 m& H) H7 D$ j" D+ s8 r" I                        DispatchMessage( &msg );
, }8 f, ]" V; z9 A5 F                }
                return m_bAsyncFindRunning;
3 T" b5 G: ]$ ]+ T        }
1 W: g8 O# c- C: _" c
% n) _( {) m% P) ]
        TRISTATE                        m_bUPnPDeviceConnected;


// Implementation
        // API functions
1 X+ U( H; W& F9 w2 T" f2 F6 {& g        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
! ~3 V! R1 U! A- O8 M+ K! P        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
; v1 e& _6 d2 h  W8 y! A9 n        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
# {( Q' [& v) P( W        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
$ H, N/ p* a* F% H3 L        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


. R) @1 C7 I$ M2 b- L        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
; Q# U8 e6 V" u1 |* p" T% j9 o

2 i9 S) U; i. q  f2 T        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
) H% j7 t7 B5 _+ R) Z        {
$ _/ j9 E" N9 u4 Y6 U                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
3 ]) l4 X3 O5 q7 D) \: c' n                result_type operator()(argument_type device) const
* j2 x% y( P: e+ y; m& t4 ?$ _                {
% [9 j6 V& ]2 F" `/ T                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
4 O7 Y- J* A8 [0 P

: \- l1 o5 q/ r                        if ( FAILED( hr ) )
+ f+ e) o& o! y; D' z                                return UPnPMessage( hr ), false;
  `+ r  |) r8 c2 }2 b3 t
5 m3 B% d9 A- w! v0 |' c- {. _. Q
$ f0 A0 ]8 Q- Y* C                        return wcscmp( deviceName.m_str, m_udn ) == 0;
  k1 I% p6 x: ]. j9 \! P                }
( c; N0 m' m8 _: |                CComBSTR m_udn;
        };
. b% j7 N4 A1 d9 s3 q7 `       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
- U$ R" G* q% w5 ^+ o7 _" w        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
$ d" K  t; p) h. g        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
7 ?5 r" E/ X- m3 Y7 U5 P
) N$ u! L4 c+ ~5 k0 w
$ T1 ^% D( Y0 x5 f& z' d' H        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
1 ]- f* B: l6 x/ \7 r        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);
& Y. F  Q/ ~: Q8 N% `% a+ t% g
+ X9 u6 T) m+ e- b  {; j
// Private members
( c! j) B8 f( d, W3 Hprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
7 X- k3 [8 Z- C        std::vector< ServicePointer > m_pServices;
9 k9 q* h, B) h$ N+ c: C% |/ L        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
% E( U4 I- j- R& e0 q: w6 x% Z1 Y
5 A. N. z" a# _0 ?) ?. `% _1 B0 b
2 N" i' Y! W% e: ~+ b$ ?' ^4 z        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
% f0 r7 H& v" C* ]4 k; r6 u        bool        m_bPortIsFree;
        CString m_sLocalIP;
2 }8 D7 F; x/ h. [) j3 X8 E+ k7 B% v) Y        CString m_sExternalIP;
% H% q- `& F6 B/ h        bool        m_bADSL;                // Is the device ADSL?
! [* k' O+ \* X5 m& R& L5 A' W2 `        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
3 g2 E5 j, j! o0 r        bool        m_bAsyncFindRunning;
# M; a: ^9 g8 O8 @        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
5 p4 M* h5 M, G# C/ H* U  f        bool        m_bSecondTry;
" u6 |# Y0 r% _) d# [        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
# f3 g; y! c6 X3 i# D2 ?$ D: R        bool        m_bDisableWANPPPSetup;
- j, Y; `1 h, l

};


// DeviceFinder Callback
  c  R* L- {1 z4 }- u; l9 ^; j$ uclass CDeviceFinderCallback
& D" X5 ^1 @- ~+ K        : public IUPnPDeviceFinderCallback
  ^. i3 ?& C3 c, h1 ^' O5 I! f8 O{
public:
$ [/ E" v: g1 z( M, d) G) o0 r% a        CDeviceFinderCallback(CUPnPImplWinServ& instance)
4 B/ l5 {4 n6 _4 i# u5 h                : m_instance( instance )
        { m_lRefCount = 0; }


   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
1 s, k$ z7 ~3 P4 U   STDMETHODIMP_(ULONG) Release();
9 Y' c" j) \) C4 p! [- I
6 h2 M. {! W  e$ g
// implementation
+ r6 E. ~. C0 f6 ~( N+ pprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
& b- l! M* N7 s: b3 u        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);


private:
2 K/ M' x3 m7 Y2 U3 s. c( _        CUPnPImplWinServ& m_instance;
1 S% w; }- O, e& m4 Q8 j, n- G        LONG m_lRefCount;
  |! }) v, c4 ?4 c) ]& M0 `};
: h5 P9 u! n" i% `& u" N* c1 l4 ~4 F

2 d+ r8 }  T! F% n/ G// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
# K5 q7 t8 h8 f' I{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
( C, o5 Z- B1 r        { m_lRefCount = 0; }
   
+ M( A0 b) x5 }' a6 F   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
: n+ c* H) ]: z: e/ x   STDMETHODIMP_(ULONG) AddRef();
5 m* j: F, X2 m  h6 B/ H' P   STDMETHODIMP_(ULONG) Release();
7 g/ q! g  l3 C# D* N# i. ^1 n
# N, O/ s2 _! M4 v2 O( z0 Y7 e
// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

% I5 U6 r1 Q7 K8 z+ l- k5 j/ z5 E
+ `5 J3 h, B# y  m4 bprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};

* h: w# u7 `; N) `2 `* a
2 j/ k, m( _( l6 ]6 p/////////////////////////////////////////////////

9 g# y, k! X& o5 ~1 Y) ]
使用时只需要使用抽象类的接口。
8 e, @! M9 g0 G: lCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
1 [6 q: y; Q  Q8 }CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.