UPnP

天云上人

/*uPnP.h*/

1. 
   ' M$ X  T2 s, Z4 V
2. #ifndef   MYUPNP_H_
   0 t- \$ A7 ]1 V4 H0 [" v7 V
3. 
   * ~: j3 i9 k/ G( k1 E9 T+ M4 z. i3 z1 h
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   
9. {
   
10. public:
    
11. typedef   enum{
    " I0 \; n; _# R4 v. ~, }
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
    % d5 F3 A8 z! H: t8 c5 s2 C# D/ 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
    % u* o3 W$ ~( Q! N
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    % y" U* P5 m% X
20. UNAT_TCP, //   TCP   Protocol
    : ~0 U0 ~; N3 p5 f( R& ^
21. UNAT_UDP //   UDP   Protocol
    . s8 _: z6 _* x7 z% M
22. }   UPNPNAT_PROTOCOL;
    6 A3 V0 V( `( F- C/ F" J7 `5 T
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    4 [, m0 I3 R- l4 b4 l" I% J9 A
26. WORD   externalPort; //   Port   mapping   external   port
    1 n5 j3 S/ ~3 z
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    . a6 j: @1 E1 n
29. }   UPNPNAT_MAPPING;
    : `/ w& g5 r/ b. L' i7 [3 t9 J8 G
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    3 G4 G2 S  w( b5 S) T1 I
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    0 ?$ l0 a4 F+ B, w6 ~6 d
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    # n% v0 W9 f+ T. X4 w
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    6 D( N9 y! Y4 n1 p
42. 
    * U# h5 h' t4 i# S- f* \: q9 ^
43. protected:
    
44. void InitLocalIP();
    3 E* J" |8 g: D1 B
45. void SetLastError(CString   error);
    0 L; H) E1 u/ s9 h) i
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);
      y9 H5 t8 B$ F6 i
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    
54. bool GetDescription();
    / I. s3 m! T, V5 A9 V
55. CString GetProperty(const   CString&   name,   CString&   response);
    ; n5 d8 I& v) f5 o3 |# @
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    3 x, I; n( V1 `& {( o' w
57. 
    + L7 f5 k2 x/ k2 r  J8 P1 F
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    , W$ q. l8 P1 T5 [+ }4 O2 Q6 @- P4 [
59. bool InternalSearch(int   version);
    " ^2 c/ G' v/ X" B5 v% o
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    ) G+ l1 }/ t6 n/ e5 ~0 \5 X
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    2 G! \+ w: b/ H* Y/ k/ d3 k3 v" ^
67. int m_version;
    # f% R. m! \3 L8 [4 \. K2 F
68. 
    
69. private:
    $ x# t4 Z( j8 @8 t4 e! [
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    6 b2 a* p! |# E9 F2 Z* j
71. 
    + }! p) |' r3 n) G
72. CString m_slocalIP;
    3 b9 z# _- v2 T
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    6 S& u3 I( B! P/ h2 u
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   $ _: y) J3 e5 _9 v1 E' p
3. 
   
4. #include   "upnp.h "
   
5. 
   + W0 m6 @. e2 Q# c) \
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   . d  w/ F* m7 N2 B4 C
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   & D! P1 w* n" q* G/ G. @
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   0 d& U7 q' z$ {
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    4 k, `+ F7 e$ h: ~
13. static   const   int UPNPPORT   =   1900;
      @& n4 {2 H: X" g# U7 \
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    5 }# y; J" j: N; G8 R
15. 
    
16. const   CString   getString(int   i)
    
17. {
    8 ?; \9 r& i, l& a3 N1 i
18. CString   s;
    " F7 J2 w/ s% l
19. 
    & j/ _1 V- X1 _4 b- |1 T
20. s.Format(_T( "%d "),   i);
    + A7 Y' V; \$ n/ R
21. 
    
22. return   s;
    
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    $ H. u1 F: W3 f  f& ~+ E# s
26. {
    : `9 \3 x% K' o" o9 \
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    ; `! c, \0 X3 U" I" p
28. }
    
29. 
    ( u, U$ ?7 m: A/ \( P
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    , k% `& \" e- J! v8 Y6 b9 U5 O
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    ' e; \' {( e* U; @
34. 
    3 s! d; [! A' X- E6 ]5 r; n
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    6 d8 R, F& o$ p  Q
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    4 Q* H0 S/ K( P- G' _! v
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    " p* L& a( [. [% Y8 e% r0 Z
44. struct   sockaddr_in   sockaddr;
    ' f/ k, d1 |& ^  n3 m
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    4 ~5 |+ K8 f% F! v; A: R8 l! R; {
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
      i/ }# a6 N/ w0 p
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    ) l. U: G7 }( a4 Q
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    $ Y8 [* a0 z6 V
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    & ]3 `3 v# m/ `: d  @
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    & R% Y2 y/ d) q2 v: |) X# X- n8 b
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    * C" s: e' X" l% k
60. 
    " L2 z1 h0 Z) l% o( J
61. response   =   CString(CStringA(buffer,   rlen));
    $ L1 ~9 L; C) C. }
62. 
    
63. return   true;
    " H% b+ \, L. @, h1 [
64. }
    ! m: ~# M  N$ w. h' ]% t) b
65. 
    5 R" p9 V& d. k2 w1 I" @5 @7 r* U$ f
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    
69. 
    8 d4 s6 s7 S/ X1 N5 u
70. const   CStringA   sa(request);
      Z9 M; y- N  p6 p$ T" U
71. int   length   =   sa.GetLength();
    ' R2 K% {! _2 J4 e4 k/ L( t# o
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    ! M) L- S5 x9 f) p; f
74. struct   sockaddr_in   sockaddr;
    2 v7 d' @! z( u# B/ e
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    7 H1 b  u. b) C* ]2 w1 w) \' K
76. sockaddr.sin_family   =   AF_INET;
    # c: D# [* [) d0 ], c2 \7 `3 F
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    " x, j) ^8 x$ @5 {+ F/ E
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    2 X1 ~- E$ O# Q% y) s% I" K
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    1 Z8 [: d+ S& Z
88. 
    
89. result   =   response;
    ( X" P' ^! r# d
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    3 X2 k& N# {& H+ k
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    % s1 A. n1 W4 l7 x8 r
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    8 y3 v& d# U( Z" K' v
97. }
    
98. 
    5 k" ?# P" H, v2 r
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    ) U8 B  a: n: N$ m* G0 _
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     6 ?* H) \( Q' U, X( G1 d
104. 
     
105. int   posStart   =   all.Find(startTag);
     * {6 p# l3 y  \, R: A
106. if   (posStart <0)   return   CString();
     & D" r0 F& `2 R
107. 
     , _# H% {* |* M: T2 j
108. int   posEnd   =   all.Find(endTag,   posStart);
     ' K7 q9 c3 o- A* N
109. if   (posStart> =posEnd)   return   CString();
     1 {  j+ ]# P6 t
110. 
     " U1 d! g$ M0 v$ @# W+ R
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     8 u! f" q5 F3 g2 t* G* U
112. }
     2 y/ I2 D+ a0 v
113. 
     3 D. O7 M( R4 T9 P. f$ h
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     / y% ~0 E  @1 ]9 f7 }" H
116. {
     # k: ?0 R0 B+ N" x
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     2 o) W: O4 B* O0 e: Y/ q3 w
119. }
     6 q: C0 J) L# d
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     1 P: ]8 C( h6 c3 W
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     ! `" i6 ~3 h& ]+ i
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     4 J5 u, x0 o. C4 k
129. 
     
130. m_Mappings.RemoveAll();
     
131. }
     ' I+ o( x2 A6 q' T  K# d7 E/ F
132. 
     
133. 
     5 o+ d+ n, C0 A; U) V6 _
134. bool   MyUPnP::InternalSearch(int   version)
     * k: K3 C- B" {9 O5 z" l+ `3 P
135. {
     " I$ v7 F2 X3 u1 B9 }( N2 {. p& J5 F
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     , o: T) F7 B- V; l, v5 \$ L2 J9 v/ W
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     6 v5 ?: B3 ~5 n  j
145. 
     3 f) H2 ?- F3 o, h& p
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     6 G, m, V$ ?0 q" i
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
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++)   {
     $ l' |, ^/ I7 C9 q7 X
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     ; B' t# ?; L5 q; `% X
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 "),
     - z% U7 F6 s8 C/ x9 \! `6 j
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     * t# R! A* R4 [8 J
159. }
     
160. }
     
161. 
     6 |% _8 F. {/ C* G
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     7 g$ L/ T* j3 d
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     4 |$ {: ]' p1 d# D3 O( ]' P- P, _6 _
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     8 X, D4 ~) \, @3 V' M# y
169. CString   response   =   CString(CStringA(buffer,   rlen));
       L  ^+ ]/ `# r
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     " f( p" L4 M5 O  N) d8 c
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     # ~2 p" O7 b$ F! c9 C% M
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     ; B  G! P, F$ C4 w1 D0 L: K
180. name.MakeUpper();
     ; H/ r) N% p% y
181. if   (name   ==   _T( "LOCATION: "))   {
     : ?( g$ K8 d6 ~5 Y
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     8 r0 c8 U5 I6 U' ]+ J6 p- ]
186. }
     
187. }
     + F% P5 x" ]0 {. w5 @! c
188. }
       m, B( X; z& z/ {- D& U/ Z
189. }
     0 f! G- N* g( B  A+ j* D$ s
190. }
     % g5 u' @! _: L+ Y, p3 ]9 {
191. closesocket(s);
     
192. 
     # B( v, Q6 N/ r# X* I
193. return   false;
     
194. }
     0 ~) y& V5 f" z; a: r+ d

复制代码

天云上人

以下有关upnp的接口来自emule，

# n2 G& g! A- t' F' d, N
///////////////////////////////////////////
3 U$ D: Y3 n. q$ c- Q5 `7 W' O//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
6 _* y+ W/ R3 U
% }; A% V8 J+ ~- \/ ~. a+ w$ `
% G5 L- i0 u' k2 @# v4 A+ N4 X* O7 c#pragma once
#include <exception>
- ~" [# W. I$ A4 j
- p$ b0 A: S0 n6 w5 x
7 p8 v6 L0 F) Y$ A  I( O4 b+ x* ~0 B  enum TRISTATE{
/ q* i% T6 q+ |& `5 T5 i8 z        TRIS_FALSE,
        TRIS_UNKNOWN,
( {& Q' `( \5 F- V        TRIS_TRUE
};

1 e* ]" b) z4 i' Q% ]
enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
! W0 h! v5 ]) m( `1 `: c4 M/ W; M  M) I# X        UPNP_IMPL_MINIUPNPLIB,
  y* P3 l& r% v3 M( L        UPNP_IMPL_NONE /*last*/
6 n( D. [  e3 T};

* z6 h" q9 d8 M, j7 R3 A
. f$ @  k5 |1 V% e( w
& A) s  B$ n! D3 x" q4 v% x
7 s9 w  f4 z; D$ H+ uclass CUPnPImpl
{
public:
        CUPnPImpl();
/ f$ G1 ~! j# M) t6 M        virtual ~CUPnPImpl();
5 L/ q# O: u: @& R+ H  U, X        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
                UPNP_FAILED,
, `$ m" q2 l" e: v0 X# e/ v, f                UPNP_TIMEOUT
        };

8 ?/ e" Z; o# K( x
3 [' ?4 z) U+ U' u! {1 ?        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
% Y1 Y2 y4 d* y9 c: |        virtual bool        CheckAndRefresh() = 0;
; l4 ^0 n) Y; }1 V7 h5 S, W        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
. |; f) @; O. n        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
: K7 a8 @6 G& ?+ s% z       
/ _7 V% }0 e" _- I8 B! n  b4 L        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
/ j$ N. }' M& X* I" [

        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
& c/ R  C9 j/ c' n" B3 s( K2 h        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
$ D1 f! i" j" v" [        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
) @" M/ n3 F1 M( ?

// Implementation
6 _2 E$ ?) I! Yprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
, U8 P+ P2 D2 g0 n$ o1 H4 F3 `) k$ ]        void                                SendResultMessage();
1 h+ I: X: _; `        uint16                                m_nUDPPort;
, }- Z7 I8 S1 T  i        uint16                                m_nTCPPort;
2 V* R9 W1 z+ B) r) w, Z4 h" t        uint16                                m_nTCPWebPort;
3 f" V# R  ~" G        bool                                m_bCheckAndRefresh;
" h: _' c) q8 ]8 I' j

) I2 W7 ]4 n9 q, p9 Oprivate:
        HWND        m_hResultMessageWindow;
! y0 D$ {7 i0 B8 t+ I% z) t( X        UINT        m_nResultMessageID;

9 J/ _& C9 F- C* B% I7 g
4 s8 ^* k* H! O$ b; E};


// Dummy Implementation to be used when no other implementation is available
# R, s% x# r3 H0 ~, d: lclass CUPnPImplNone: public CUPnPImpl
{
public:
5 {/ P; T9 o+ Z: K        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
6 v/ d, t. R4 h" J  t( S        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
3 @% c6 C+ }" W# W        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
7 T5 r& H/ ?* T) q' z/ E1 N4 h
9 o' O$ Y6 D' Y
+ {) }/ E  \9 T$ I! z6 v/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类


) v- g7 n/ }/ b/ K% i: b/ q#pragma once
7 [7 j7 w4 |# Y+ s8 Y/ f( S: D+ a+ Z#pragma warning( disable: 4355 )
* ?' v0 P2 F# }* a' L7 }5 h

* e6 t2 A) [6 B* l! W#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
6 r( T/ E) U5 {) R: b8 j#include <winsvc.h>
' H4 G' g& f# ^1 u1 o& a
* g8 j* z( B8 \% K6 i' n. o/ N
* A9 E0 h9 l" B' d& o$ A  s7 R#include <vector>
( T$ J$ E8 ~% J#include <exception>
#include <functional>
" @& E3 l  W8 {0 N! U* A# f: h


) x/ O) l4 x) G2 i
- V+ P; U4 g3 Y" g$ p% }3 ptypedef _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;
2 z* S% H8 Y6 M: [typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


. r& w" l2 ^- }2 z& R4 b1 Otypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
$ F/ R" k1 i' r+ x! X% T  PDWORD pdwBestIfIndex
);

% [* d6 }+ B  b, I! a! n
typedef DWORD (WINAPI* TGetIpAddrTable) (
# z3 x: I7 ~. |  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
6 }7 p2 o4 K6 W" y);
! I; W' ]- V% N! P+ S- p

/ {1 M; g* M& y4 e& }( ttypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
0 P$ N4 a: f9 F5 b- e);
# S* Z9 S' i# F; {( W
3 U4 x/ M+ o0 k! h8 s
CString translateUPnPResult(HRESULT hr);
# w% p- p* o# w/ c& }HRESULT UPnPMessage(HRESULT hr);

0 k3 a* m0 H; Q0 W% R. N# @7 s. [
class CUPnPImplWinServ: public CUPnPImpl
9 I/ O" ?6 c7 d8 \# B( |: E/ \{
. R" @& E  x" I1 C+ ^) N        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
public:
- z& l& E, N6 z. J        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
( W, A/ a, |  V1 p7 X' F; c/ w- P

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
" s6 b0 h3 P6 c% V7 D7 w        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
: m, B2 l# G* [" H* A4 P5 I        virtual bool        IsReady();
5 D+ S% A  i) R6 l7 S# r0 z$ p        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


, B8 e# O- d: A2 X$ J* `* T$ h& a) n        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
; Y) W, R4 s& n9 q2 T        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
9 A- `9 A) Q6 f5 i- N1 O        virtual bool        CheckAndRefresh()                                                                                { return false; };


protected:
/ N6 r* w, v0 m: r        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
7 X* K# W- p( h! Q" I        bool        OnSearchComplete();
        void        Init();


: `! w( m( W3 q  Y+ x# ~( c        inline bool IsAsyncFindRunning()
: W) |4 g  `/ ?/ c# A. d        {
5 u1 O7 _( F: ]6 Q) Q8 @/ K7 \                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
$ g2 l- s. D  `, f+ q: x% u" f& j                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
0 O, M1 M: i) b1 D  e" R+ J$ O                        m_bAsyncFindRunning = false;
# g/ R) M9 l* ^+ Z# S7 t2 t                }
; w4 p: l* u) J' G" Q: Z                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
/ ~" y  N4 B6 \* I# K                {
, X; n' Q7 w& I! P; d                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
, x! u9 I( i, E: {# n& ~, E+ T                return m_bAsyncFindRunning;
& r& W+ T' {2 ?: j) ~        }


        TRISTATE                        m_bUPnPDeviceConnected;

$ D& }# _: K& t' M
3 \# T9 C" d% b$ }// Implementation
: n$ i8 d6 c% v5 l! N        // API functions
; ~) F2 R- {7 X$ d$ I        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
2 Q( L- H" k- y        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
! F$ M$ r+ u* ]. Y6 n- x6 q        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
2 b4 M, z  t' z3 ^7 R/ z% s        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
% }9 g. W$ b& v3 Z        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
8 Q- }  R9 g+ e& C6 b( _2 I; M

        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
) {% @  k5 B. s" V
, O, `# F1 P& b; E6 H6 t
2 f( x' O+ E+ R        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
1 V5 i" B  L4 p, e/ F. g# B9 _, |! ?                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
- P3 K6 z0 g* `8 ^6 j- _; o4 Z
" t0 P, K* N8 @0 t1 W. e% p
3 C7 W5 @2 {, i/ w$ q                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;


' V. y9 L! J8 t! N% g% M                        return wcscmp( deviceName.m_str, m_udn ) == 0;
7 x* b: c# V' U* t1 D5 [3 ]/ ~& W                }
                CComBSTR m_udn;
& Q6 Y5 ~) R7 ]! U! W: [4 a        };
9 g' I8 L5 b8 A  N* ]       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
: i  U1 E4 k$ g/ U2 a! z        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
: ~" \4 A  _$ t; {, J0 `& m        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
: I4 F0 b8 w: t' t9 a7 v        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
* f3 b% p- W3 N/ c( m                LPCTSTR pszInArgString, CString& strResult);
4 a8 B- K9 w* {7 o% ^. \- b        void        StopUPnPService();


        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
8 ?, d; r# s- }/ ^5 b        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);


// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
) e% H( [2 h6 S+ b/ m        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
' |' w( E+ L2 q7 y        DeviceFinderCallback        m_pDeviceFinderCallback;
& l, O7 W3 k6 O% d) N" V. |        ServiceCallback                        m_pServiceCallback;
' b) _; S) Q$ y' x
1 G, e4 x5 F9 A8 Q2 }$ \
* T+ q3 F5 J6 x2 [# p) S        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
6 ^( d5 W2 _. ^4 c  N1 j, `        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
# f, O( F/ P6 l* e8 i" W5 w0 b        bool        m_bADSL;                // Is the device ADSL?
% r5 d8 d, f. N  l6 O        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
2 }8 c1 ?( k$ z% Q% M: C9 M        bool        m_bAsyncFindRunning;
7 [7 W$ r! S+ U7 T3 W        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
5 l* w) c! U8 Y; S4 o% O        bool        m_bDisableWANIPSetup;
* b% Q6 G; d2 A3 G0 k% f        bool        m_bDisableWANPPPSetup;
5 j( r& f; G" G, H$ p; _% A
7 Q/ m( {# K3 `7 E" ?
};


) z3 |* W' ~7 r" {. F0 f// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
% K/ W" V, d0 w  ]/ `public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
3 m3 S3 j/ l9 u5 ?        { m_lRefCount = 0; }


   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
6 `' w; O# V  F/ P   STDMETHODIMP_(ULONG) Release();
' L! m- G$ ]" d, b4 Y( S0 S1 J

5 T" m8 x8 b! \: E  a4 O, m9 G// implementation
: L. G# l, g; W' y3 P$ w# w1 }private:
: U% E/ o# m/ R0 p7 S& J6 q& Q3 S6 k        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);

2 u. i; L1 H9 t& y4 y/ ~
$ |3 Z! {0 g6 ]8 [! L' b6 h* Fprivate:
! q) K1 h$ a: O        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
  U; b6 \' B: z; Z/ j' m  x, y9 H};

; U$ @5 ~; R& m& `
) M, h' T3 J" _  i0 D// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
{
public:
* m0 ?# @) Z7 B9 r' q        CServiceCallback(CUPnPImplWinServ& instance)
6 z+ P. S( w8 h' u3 n! N5 x; R                : m_instance( instance )
1 {* e0 G, Z( r6 A+ a% g        { m_lRefCount = 0; }
) l2 \5 ]$ o; R7 m" c   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
: z' P: v6 Z2 V4 D0 G   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
) {8 D$ a' w* `% W. ~( W$ I

+ P( x: f  E* K  j// implementation
private:
, A9 ~8 S" }9 w' ^( ?% L+ a        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
0 @0 c1 V% F8 Q& |% J
0 F( x2 i4 S. i# @) g5 }
9 {4 t/ H; c" Y* G. X$ Eprivate:
/ d) j) y- m5 y        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};

: T/ y+ w9 }  V; i$ j; G2 K
3 p- a" {4 `. P" t/////////////////////////////////////////////////
* |3 ~& L- |- i( @$ p3 m" B! m4 y

使用时只需要使用抽象类的接口。
, I3 }# U, g& G1 I3 UCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.