UPnP

天云上人

/*uPnP.h*/

1. 
     h% m( H) s$ _; Q$ {
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   
5. 
   # \2 A' H- Y9 S5 y% K  n+ r
6. typedef   unsigned   long   ulong;
   1 h& a) g5 R. T' _
7. 
   
8. class   MyUPnP
   * D' v) @8 u  b% N
9. {
   . S# n( ]7 N( `4 J, a
10. public:
    
11. typedef   enum{
    0 R  I1 k# j" S1 l# n2 J3 b
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    / ]# ~1 |% {4 Z- W0 g
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    3 U, D" P# {! \0 L) C
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    $ T" `2 p/ G# m* I
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    - @2 ~( o. [, t+ R* s1 r7 b
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    5 W% t) ~0 r* }, v
24. typedef   struct{
    8 m7 Z/ M  J- H7 Y7 `
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    5 t  a+ n! T, `& C: x
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    1 X1 C6 p. |# j/ P$ ^
28. CString   description; //   Port   mapping   description
    5 p2 ~9 ]6 j6 K  n3 ?# p
29. }   UPNPNAT_MAPPING;
    - b8 w4 M# P( Q5 F( p  _
30. 
    / v5 z1 E6 J4 ~, Y
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    4 D5 \+ R( t  V5 `
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    % z6 S* Q) T; j- J  M4 q
37. 
    
38. CString GetLastError();
    " J) E: q, e; l0 |3 S
39. CString GetLocalIPStr();
    6 t0 f: C! s% i) Q+ b! n* t" n
40. WORD GetLocalIP();
    6 ~' `5 `4 d+ L: k5 O$ g, X. q
41. bool IsLANIP(WORD   nIP);
    2 i7 J" [* C+ o7 |
42. 
    
43. protected:
    : T& c5 U- O7 k+ X6 ?, y$ t$ z
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    1 M# y/ |+ ~# D& K; v- t1 n) Q8 S
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    : [7 q5 V+ j% [+ s
48.       const   CString&   descri,   const   CString&   type);
    6 x) v4 b- @4 D9 t, L; J; U
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    3 W% X/ W% }: O3 G1 X
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    2 S" d" r- d9 X9 R* S
52. 
    8 w; s! c6 @) U' s9 A; i/ o+ p
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);
    & X0 X# ?) K- G: d4 P* x
57. 
    6 T7 X, J2 t( @8 g$ \
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    . v: ~6 R' L9 w1 r
59. bool InternalSearch(int   version);
    6 n" L# Y" J6 h4 [0 C( ]8 }, U% u
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    : D" o3 w" ^6 |& k9 k( T
64. CString m_controlurl;
    7 `1 h5 {+ |% h$ e9 X
65. CString m_friendlyname;
    5 E) I: _. Y6 H
66. CString m_modelname;
    
67. int m_version;
    # t, b5 n* c" J, h  K  J
68. 
    : N! p/ ^8 f; f0 r* L" e
69. private:
    - I. u: m$ \8 |) }# O; ~4 c
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    $ W4 {& q5 n; k& X
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    & N8 {$ \+ S7 b1 F
75. 
    7 U+ X3 ~5 R! R
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   * b+ Y- }& z% b/ h2 @; M
2. #include   "stdafx.h "
   6 m1 t2 o3 |  m: Y3 ^  E4 i) l8 H
3. 
   5 R  q8 T) o# C) }
4. #include   "upnp.h "
   
5. 
   7 j4 M. Q  R5 f
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    ( X; [* n, e6 r) L7 b
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    7 a5 v& U9 J9 r& m5 L* Q3 }
13. static   const   int UPNPPORT   =   1900;
    ; @+ L6 @0 z0 b
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    2 }4 A9 j  }1 H) y. r2 M) ]
15. 
    
16. const   CString   getString(int   i)
    # L1 D2 X+ n8 T+ _" V! Z( n
17. {
    ; Y' q: l, f/ H
18. CString   s;
    5 d+ q, _+ Y3 T- I5 q
19. 
    
20. s.Format(_T( "%d "),   i);
    9 {- q' r* m2 f) k' t/ t1 N8 T
21. 
    + T8 h' ]) o# ]- ^
22. return   s;
    
23. }
    3 I, \. g. t! ]8 O2 N3 W# {
24. 
    ) z1 ]9 B$ V& E: l: f
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    3 _! S( z$ B7 h( G& k
26. {
    - g% p3 d3 ]. X1 G/ t8 `% b4 n
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    " I- Z1 J* J9 f4 R5 U) s; ?$ G+ a
29. 
    ( T7 @1 c3 a+ U0 ]
30. const   CString   GetArgString(const   CString&   name,   int   value)
    $ s0 s8 o3 }! R# `! H
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    $ ?( V7 `# g. f% I% W
34. 
    2 v, I# \4 y/ q; c2 j
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    - d5 a5 [0 c7 o6 V: f; I, L
36. {
    % ]% V; W: j9 o/ |
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    8 l/ y5 h" k0 O6 u6 ]
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    & C  O! w2 @. V
44. struct   sockaddr_in   sockaddr;
    * d$ Y" V0 A. n
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    / I: {+ M! t4 w4 ?+ T- p
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ) e) S: x6 d  Y
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    # J5 I4 t4 o/ j# G* a/ I* F& w7 D& Q
51. ioctlsocket(s,   FIONBIO,   &lv);
    9 ~# Z6 H- h& U+ d9 {
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    / x/ m! h" C& Z; l( K1 d5 H
53. Sleep(20);
    * {, P: |4 A  P6 r7 |* R+ I% r3 M9 H
54. int   n   =   send(s,   buffer,   length,   0);
      ^" d! b9 g( @$ w9 g2 K
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    7 s( L' o# |3 y
57. closesocket(s);
    0 H2 I. Q8 ]/ W% h
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    " s' j" N: b1 w. N0 H5 ?1 V1 H
59. if   (!rlen)   return   false;
    ' e9 c! K. Z- m) S! M, o
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    
64. }
      I+ T8 b3 v6 J5 ?: m, t: S
65. 
    - \8 i4 W! \) c7 [: ~6 z6 i
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    7 c# k4 J+ h# }6 B6 s5 D" b
67. {
    % Y8 _2 b7 q9 n+ h! K
68. char   buffer[10240];
    1 i/ ^1 [% H" I- R& w% o0 E
69. 
    
70. const   CStringA   sa(request);
    5 r: [% L4 h: z8 t  q+ |  k
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    ! ^( g! f+ X$ a; p' D% A
74. struct   sockaddr_in   sockaddr;
    8 ]' Y8 ?  f6 ^- s
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    3 }  D  X' k% y/ t' _8 X, p: N
76. sockaddr.sin_family   =   AF_INET;
    # T0 ?4 v7 V4 n6 k$ L6 R# t- z; X
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    0 l  y6 G$ w3 |5 X* u5 H9 Q# z
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    0 {, Y6 [; j( r/ F2 a0 P  \0 y( `4 {
84. {
    0 `# `, ~; L/ x1 Z! M& j
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    . }/ `' v9 T4 n8 S$ E
91. 
    - \5 R. p* b3 g" L7 Z
92. pos   =   0;
      [2 B0 q. [5 Z0 E/ i% U' `5 f5 C1 ~
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    2 \  w3 @2 V' V: a! ?6 `
97. }
    
98. 
    4 X. }/ Q1 M! N6 q' f8 i6 C8 O
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    : d5 k- Y/ X2 q! a0 t  N
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     6 N; r* i# r" T" x5 L
103. CString   property;
     4 h) p0 C0 Y- _
104. 
     
105. int   posStart   =   all.Find(startTag);
     " G' s0 ]* r: [' o' g' s: A) E
106. if   (posStart <0)   return   CString();
     / l$ D1 n- f2 W0 i  e
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     + k3 M( u1 e, K9 E% J; L: ]
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     0 Y* \8 [( ?: y: {  c" n
113. 
     ! ~+ x1 h0 {9 C8 P
114. MyUPnP::MyUPnP()
     0 I; m# v8 g- ^+ x$ l/ u
115. :   m_version(1)
     4 t: p& [" `$ E
116. {
     
117. m_uLocalIP   =   0;
     9 D. g, v6 W/ T$ _" ~) K& o
118. isSearched   =   false;
     
119. }
     ; T: U( A0 i* D0 k6 f
120. 
     
121. MyUPnP::~MyUPnP()
     : `0 @7 Y; k" Q
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     , R. k! @% U1 q9 R
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     & H  x- ]& b0 ]* S6 G9 [) `
129. 
     
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     ! I- h& L) w: V" ^
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     $ J/ b5 k  {* f8 F1 `
135. {
     % b9 |: l+ w9 Q% ^8 h+ x$ ]
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     & n/ |; @% @7 s* L
138. 
     1 ]  s; }8 U( k$ Z2 c) z9 p
139. #define   NUMBEROFDEVICES 2
     : Q% Q6 w! [# @9 w! g# p+ k9 v
140. CString   devices[][2]   =   {
     * }2 x$ M& r9 P8 k6 b7 u
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     0 x; p: U2 c( a1 i# V
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     9 j: D0 G* k6 K/ F$ t0 F
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++)   {
     # P! D% J+ ^4 D; y4 _
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
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 "),
     * j2 x0 U7 e  u
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     , F6 {4 o1 `1 ?" ?( o# F. f
159. }
     
160. }
     9 R6 k4 _, |) @9 f" R  z+ p
161. 
     
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     4 `3 F% a- J" ]$ C* _- i
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     9 u+ B( ?5 x* K2 L
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     8 B' e7 G( D: R. \
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     " e" J; V! f; M. X% w8 K1 }+ H' X
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     4 K/ A6 i; |5 M# _& P% B
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     6 r- H) Z% a! F5 i5 `3 x
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     ' F- h" W8 x" ?! n* U- b, V4 J
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     6 y" V5 H  S) Q" O2 @* ?1 Q
186. }
     
187. }
     ! p  `( n" n! h# v9 }
188. }
     ( A# J0 _; f0 E3 P, f0 D# Y4 n
189. }
     , N3 t3 U  }! v5 d% P9 w, X
190. }
     
191. closesocket(s);
     
192. 
     
193. return   false;
       F/ J# C. _+ m2 `& A
194. }
     + J* t# j* ?- h0 {& E! z

复制代码

天云上人

以下有关upnp的接口来自emule，
/ f3 M/ p  b# V' Z, ^
( `  ^3 K% A* @+ C9 j
. K+ R; V9 |1 Y///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
- Z$ b* x& x$ o; u- F

#pragma once
% o' f0 F- f2 r5 B# K  C9 S3 T#include <exception>
3 {. u+ {0 l2 S# N2 _

  enum TRISTATE{
6 u+ ?" D  L5 D9 y        TRIS_FALSE,
; A. D5 v/ _6 n" y        TRIS_UNKNOWN,
/ k6 C! [7 Y  N, o. }1 W* o        TRIS_TRUE
};

% S$ ^+ x# \8 G5 V  B! v9 j# q
& [. o8 Z$ e: V9 p4 @enum UPNP_IMPLEMENTATION{
) n* V% A- w$ N6 S        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
2 [$ \5 d) _* \7 e! t$ x: z) r$ k        UPNP_IMPL_NONE /*last*/
};


6 v+ v3 [+ n* @6 y7 V2 ~8 a
! A* H, L! u8 y4 y0 G1 L0 x3 {
; Z! C% a  r4 {- {class CUPnPImpl
{
public:
! `& v. R, J: E  U9 }8 k        CUPnPImpl();
; Q. Y  d. e: f  \        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
        };

* d! g9 R/ y" B; F+ a9 a* z: W6 p
" u; y, d5 r% X+ Q5 j' Q        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
/ b2 h6 R6 P2 m6 e+ n        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
' D0 r' \# d: D) e. B  ]$ |        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
" r6 \: E% k, c9 f5 q! A# \/ b% B       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


1 d7 c5 Q$ T2 W8 p4 |$ H: Y: ~        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
/ R# m/ X, `5 s( P- @7 H8 [        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


% G' x+ k, Q- V1 R8 U3 z  X2 }// Implementation
protected:
9 h8 q* F# S1 t        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
9 j" o6 g) w/ n* {0 T        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
3 z- O, K2 l$ p" {7 }        uint16                                m_nTCPWebPort;
7 C2 e7 G# G3 J8 D% }: ~- s        bool                                m_bCheckAndRefresh;
7 w' C% N* E) m7 y. C( M3 o
- u4 O+ r. f, v5 l. c
$ R# F1 a% n: F6 mprivate:
0 U. s0 n' S& W+ Z: Q8 {  o  T        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
6 n5 i4 q" |& @8 D' c4 u

. v  m1 e8 e( G) d; `3 |};


// Dummy Implementation to be used when no other implementation is available
0 k5 T: G9 f/ e, u) g) E$ m! h5 Y! eclass CUPnPImplNone: public CUPnPImpl
{
public:
2 x2 [% u* z" \5 n7 I        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
' o4 [. t. b2 q$ Q; }8 n. Z9 P        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
3 n) s& _8 ]/ M. p. ]$ P9 s. {        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
8 S  R2 S% w  j" p};

- _9 C; a0 r6 m
, h+ U0 s: s$ ^$ ]/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类

( w- ]2 p+ S9 g/ `* u" I# |
#pragma once
#pragma warning( disable: 4355 )
. D, J/ K3 }- J, [5 `4 b# ^9 ~7 I
5 Z* |; i8 L1 N% P  F4 D  M
#include "UPnPImpl.h"
7 B( F! {1 w  d9 ~/ a( J9 c5 Q#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
, c2 r% ~9 p+ d% |# J# L#include <winsvc.h>


+ \: n9 G, F; n( W1 @+ I/ A0 t#include <vector>
#include <exception>
#include <functional>
' P% `' F( G9 a* P& K; g( w

, x3 C; W$ m# E
% [, V1 C' _: E# r8 Q6 l
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
* m+ l+ y% |, X6 `) xtypedef _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;
: A  {! m/ [4 ^5 @" P  M3 xtypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
- s# t, ]8 R+ p) Z" D6 ltypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

0 f  g# ~+ x' }/ E
! c0 M, @0 b% C* K! Mtypedef DWORD (WINAPI* TGetBestInterface) (
" x& b5 d, \' y" K, M7 q# {5 c  IPAddr dwDestAddr,
* {: ]. f& S# b; [  PDWORD pdwBestIfIndex
);

. _; Z( _# k2 T1 Q9 j& _8 A& i/ C1 ^
typedef DWORD (WINAPI* TGetIpAddrTable) (
% \2 J+ R& R% T0 y  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
, d" H' Q, I2 u3 M- f);


typedef DWORD (WINAPI* TGetIfEntry) (
; M. f: V6 U0 W/ s+ K/ [6 t3 _  PMIB_IFROW pIfRow
2 R# l/ z! j. \  t, b/ z$ l8 {" B);

7 ?% h. b' h6 d' m4 m
CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);


class CUPnPImplWinServ: public CUPnPImpl
9 L0 [, j: e1 h  p+ j' X: x3 |4 ]; X{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
- A6 ^  Q! _3 ~- {// Construction
public:
+ g, i  i- N; ^4 D* |! r        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();

2 ?% l+ t$ a; m) X- c8 G9 o
% `1 G$ t! v1 U! l! E- o4 d' |% w; D- P        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
" ]) y* S# }) _: u        virtual void        DeletePorts();
! S4 E4 x/ h+ [" g        virtual bool        IsReady();
3 ~4 ]! X6 @3 J- d        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
  q$ U  _# v* t

: k+ P( A/ V( J) w( }& m        // 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
2 z; E$ J( S3 R3 D& }        virtual bool        CheckAndRefresh()                                                                                { return false; };
! @2 c: B' \' F3 [; J

protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
+ M9 j$ m+ O2 c3 r* R& C3 y. g$ N        bool        OnSearchComplete();
  D( }5 [3 |2 \& T9 b; y. W        void        Init();


" o1 B( M; u9 y* D* T1 ]/ b# `0 C        inline bool IsAsyncFindRunning()
7 \9 q6 w9 Y8 B  T        {
1 z1 X3 T5 v  Q  I( C8 b& j                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
% g5 ]& a4 J5 R8 s# V3 }# X8 u5 R                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
- |% K/ t, v* `5 t  L; z                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
" S6 G: S4 R% \: [5 ?, H( B6 u/ d                        DispatchMessage( &msg );
# q5 p6 E" Z# b. n) G- v0 L1 a; d+ F                }
9 ~  g$ E  `7 E" |                return m_bAsyncFindRunning;
        }
4 O9 p2 u+ g* n# Q! K: [8 E

' `/ \4 d$ N2 ~- {        TRISTATE                        m_bUPnPDeviceConnected;

# k: y5 T* B+ p9 b& z  U7 Z+ o
// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
; \2 ~9 ?0 q, ?2 Y: L6 _( Y4 T% C, y        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
6 a( d" r: L  q! v) N, C        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
+ N3 ?( C$ [  {$ l3 p& u        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
, s' K& e/ m! b# V3 Q( v        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
" Q) L3 K: S$ r

( `# f7 i1 k  N& X: |        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
  d4 ~, Y! F9 {4 |3 k5 n        TGetIfEntry                                m_pfGetIfEntry;


        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
4 U" |/ S: S9 k4 k% n9 X7 }: P. p                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
+ p" [4 y) t- l1 E& y                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
+ u6 f3 R. M( U9 F                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


- j9 X$ `& O+ P$ @6 F8 g9 n  ]6 f                        if ( FAILED( hr ) )
6 ~3 P7 V& N  U6 m" T$ Z( N                                return UPnPMessage( hr ), false;
5 ^4 T, W% W9 [9 U* E+ a: |7 q" ~
9 H4 Y& {& O) z. S
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
+ v9 T: f: ^! i" T; d/ e                }
' m: n# t( |3 O# P0 y" U+ O                CComBSTR m_udn;
' z( L. L: ]" r        };
4 C% L5 y( Z% ^, d& `  N       
& S' u% H' `2 ]5 [        void        ProcessAsyncFind(CComBSTR bsSearchType);
+ y& D& k. U$ R/ A4 Z; h: Z& K. ~        HRESULT        GetDeviceServices(DevicePointer pDevice);
# x1 y7 a4 F/ a2 R& c' U        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
) L$ c3 |+ H& h  o$ A: Y        void        DeleteExistingPortMappings(ServicePointer pService);
, {! \4 s  L$ s! h& c  c1 i        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();


        // Utility functions
0 e5 O( e1 ]- r8 E        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
; H5 V( j6 X9 M. }' M5 c  S        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
$ W$ W. G7 T0 n1 X9 ]0 r, h        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
1 J, a$ k. K3 I7 J8 `; h1 i! D$ H4 ^        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
+ c' o5 y2 n  J9 a        CString        GetLocalRoutableIP(ServicePointer pService);
) E* `- k7 u/ q. e) s  J( _

2 {4 J/ `! X7 l& v- B// Private members
private:
  t" n+ s+ c' i# d        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
  C# F( r$ |9 X6 d/ w+ W+ w        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
+ [. b- |5 A  O) ?! }! V& W' L        ServiceCallback                        m_pServiceCallback;
* w0 V/ ^( A* S! T

3 H: _: b/ n: P: }        LONG        m_nAsyncFindHandle;
9 u  k# n& }1 ?3 Q3 D5 \& I$ a        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
6 C1 a( J1 H0 _        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
1 \  E* D0 e2 v3 K5 ?7 U2 i        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;

8 }  X$ S% f2 B- D  o
};
4 T7 T3 H0 `+ v# y) i( a

8 p. g2 k) u/ l- L3 s% [4 e// DeviceFinder Callback
class CDeviceFinderCallback
8 Z: q% e4 A( K! u: ?  f        : public IUPnPDeviceFinderCallback
{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
6 M, w/ r4 E0 `% z1 n7 K        { m_lRefCount = 0; }
8 s" A( h1 m6 f; e: k# E7 Y
- h  ?0 v8 D. Q
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
) e$ X6 c5 e: G2 B- l8 E8 @   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

- d, y$ C% |" L: ~7 d
// implementation
' W2 t0 w! U+ ?' E2 G  s0 Rprivate:
( X9 }$ l  g  g# w( o. x9 _        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
, O; S" y0 u. M        HRESULT __stdcall SearchComplete(LONG nFindData);
( i" O4 V. R! u8 V

private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
0 F+ i( S0 d* [' D/ }
: z$ l+ u% J8 F7 c+ X3 D/ r% V) V+ d
// Service Callback
1 @9 Q9 p: @; c: M8 ^5 h' O$ U0 dclass CServiceCallback
- h9 ^* D, t4 W' M5 X, B2 I9 e        : public IUPnPServiceCallback
{
( N; n3 y& z) t  G$ u: N* Apublic:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
! F& C+ C% M$ x4 c' S: {( r' A        { m_lRefCount = 0; }
8 k* W& v$ Z' L$ P% z6 S   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
, z, Z8 h% p4 Q0 R" O; `$ B! s   STDMETHODIMP_(ULONG) Release();

) L1 g- d, d/ h5 G& u
// implementation
private:
. E+ B# e# g4 N; k        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
; I$ d7 {" ]$ _        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

0 ~2 O: B8 X; K
private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
% d! H' U" K3 {8 F6 f

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


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