UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   - r4 V( i9 \/ ^% N, }
3. 
   
4. #pragma   once
   & m" P( v: `" e) Z
5. 
   , `, K. [; W7 Y" x7 v: N
6. typedef   unsigned   long   ulong;
   
7. 
   ! J& `& Y: {, g% a) u; ?* T) s- Z
8. class   MyUPnP
   ( |$ w& |: [  D7 _
9. {
   
10. public:
    ! t3 `; A6 ~2 l/ P  D; g
11. typedef   enum{
    * r' K/ _1 |. i9 A
12. UNAT_OK, //   Successfull
    , }5 m* |( c% M! B, p8 ?$ p
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
    
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    8 m  h3 ~- M0 @8 L" u9 p+ M
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    / k( a8 g" k4 y* L& m
21. UNAT_UDP //   UDP   Protocol
    # K9 ^& T  j' b/ z$ t* A
22. }   UPNPNAT_PROTOCOL;
    
23. 
    ; |' [* r$ ?* m% h9 x! ^
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    9 b9 ?/ }1 N. E
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    4 J+ [( z) \$ k6 M4 _
30. 
    8 k+ ]. b7 v% L) }
31. MyUPnP();
    % Z) W; _9 A2 a0 z0 E- L  _
32. ~MyUPnP();
    . X9 t+ l7 m8 {7 B! o
33. 
    ' ~- v0 d* [) j$ [2 q4 y9 C
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    & u7 A3 m' a8 ?3 C0 P/ O
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    
44. void InitLocalIP();
    ( |% S+ ^, ]4 R* C
45. void SetLastError(CString   error);
    ) Q0 O, s, K& m" F; H  {& i
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    4 d2 U$ X+ _, S' v, q
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    / P0 L' Q5 {+ L; S7 G
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    9 C- `9 f# n2 {5 e1 |7 w) Q$ r7 a
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    $ j; d! f( n! a
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    8 c- {2 G. i$ M
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    ! `/ Q. {2 o! w0 e
64. CString m_controlurl;
    ; x, ?* g+ m" q. a7 {
65. CString m_friendlyname;
    1 J$ n7 w7 @* ^& X- w
66. CString m_modelname;
    7 t" Y3 \9 U$ y4 U
67. int m_version;
    # c3 U+ e/ E* @+ o4 k
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    3 T) B& @% h: ]" `- q
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    5 O' v2 m. }4 Z9 a- S* F0 H7 F
74. WORD m_uLocalIP;
    
75. 
    , Z6 L) M+ Q  p! c$ r; U' e
76. bool isSearched;
    + z% p4 l) ]; O7 h% U
77. };
    7 a1 W; |6 ]* I+ j8 q+ N# {
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   ; c* _# y0 p: E/ R0 m
3. 
   
4. #include   "upnp.h "
   
5. 
   " i4 x  |7 ^. M, B% U8 G# e; m
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   # l8 y; ?! h3 n: g( d
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   4 H# k, n+ t! V' [
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   - d! C) N- B( d3 w+ b
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    " R5 t2 S2 N1 W& t2 n
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    0 J+ _8 Y1 g. X3 F  T. R% d+ ]7 S
15. 
    
16. const   CString   getString(int   i)
    ) U/ S& \4 w+ f$ G  w- O
17. {
    
18. CString   s;
    
19. 
    # k, W1 A9 z# ?& P9 Y, K& g5 O; u
20. s.Format(_T( "%d "),   i);
    ! }; j% u; l6 h
21. 
    . V* N$ w6 N4 N0 }& s# l" ?
22. return   s;
    
23. }
    $ E- w* y) P) s  ?0 ^& v4 z8 }0 d
24. 
    # p- g7 B5 R- J7 c# Q
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    # ]# f8 [; _: c, b0 C/ L; ?
26. {
    + I- g2 S3 C1 e, D- J" L+ o
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    0 a/ d8 o$ F3 ]# X: b/ x/ g
28. }
      \0 Z% ~6 Y6 C  ?( _
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    6 h6 d/ R7 Z3 z3 C0 r/ w
33. }
    
34. 
    5 v# P" T* Q3 T) Z; d# X) w- Q
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    1 l# W) T$ Y# x- T$ b% `1 X- W- {
36. {
    6 R2 R6 p+ }6 G( D2 Q
37. char   buffer[10240];
    # k' t" j4 T! b3 u
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    " O- {: R( t. ^$ k; T
41. strcpy(buffer,   (const   char*)sa);
    ' @; e# S# _) a9 v& X; p5 ]8 {6 S
42. 
    0 K# G- i/ D$ g+ f
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    ; c- ^, I5 z* ]$ N4 G: L! J. K
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    % x! u7 I0 i/ t
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    * l9 F7 d+ C$ s$ R7 {: A, z$ f
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    * Z" L1 U5 h5 n5 V* X
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    9 B6 C" H! a6 k1 w$ T. @. ?1 x) V2 R
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    - N, s5 H; l# L$ P- g4 r
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    / I( N3 p( C  |5 G# |
57. closesocket(s);
    : y; ?" o& f1 F$ j& |. ]* f, i
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    8 n. _& P4 O, ^0 v9 s
63. return   true;
    
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    . R# u3 m+ ~: b4 u
68. char   buffer[10240];
    - L( K+ F- s4 F
69. 
    $ f  x  E6 ^. s3 U/ W& _
70. const   CStringA   sa(request);
    ! v  W/ R) X4 h# s; z% a/ ]3 A
71. int   length   =   sa.GetLength();
    + ^9 j: g3 m; I& X
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
      ]9 u# s/ c" I3 ~7 }  D* x
77. sockaddr.sin_port   =   htons(port);
    " t2 c( }4 e% n! Y/ t
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    ( ^" M( {8 L3 b4 ]% p# z
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    * o2 ?$ B6 @. i! D" c5 N
81. }
    
82. 
    5 H; ~, |" W/ m; c8 C2 q
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
      e9 H2 H* i& X0 d, E
85. int   pos   =   0;
    . L% [) Y, }+ ~6 ^% S
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    6 |5 k; O* B: K
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    % I4 f8 Y9 V  P- G) u+ r
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    # g( {+ \6 j- O, N% {0 B) G" t
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    * }! a* W4 N- i
98. 
    ! C, }  U( \/ ?
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    ( d, o3 M8 T' g$ E+ L: p. {
100. {
     , `. X3 I% |4 Y4 y4 A
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     8 a! U& @/ o* N: r( G
106. if   (posStart <0)   return   CString();
     3 h2 R" u* O3 d# m$ Z9 v
107. 
       v. ~; Y% A" `( Z
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     , P4 ~7 @+ V' r% l. ^
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     9 n0 S/ L  S6 F2 D( N; [
112. }
     
113. 
     0 U* J; g2 L$ k- b6 t7 o
114. MyUPnP::MyUPnP()
     7 u$ T" T% w/ T& N$ S; t
115. :   m_version(1)
     3 {6 p7 ?1 p6 O6 Q1 g- S
116. {
     ' \: F( R* @9 j7 [& b2 X7 `1 i3 H
117. m_uLocalIP   =   0;
     # |; @9 ]4 ~* Z- R
118. isSearched   =   false;
     2 y* }2 K7 F5 j3 y
119. }
     5 M, q9 N3 X7 t" F6 u
120. 
     
121. MyUPnP::~MyUPnP()
     / R( X' }4 d$ O8 o, w: x
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     + `. D) W2 c3 I8 W+ K' w% g/ J: p) B
125. while(pos){
     8 i2 s# `: e; Y- G& ^
126. search   =   m_Mappings.GetNext(pos);
     . m2 i4 w8 s1 `0 A
127. RemoveNATPortMapping(search,   false);
     3 _% C, B. l: T3 F2 S" \
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     4 t; o. E% C3 U4 D+ v* k; [! n% i) B
131. }
     $ L' o. Z( P  I( `  Q% C! e% w
132. 
     1 d9 y. T7 c  @; q
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     ' K/ P3 g& F) ?& l2 P
135. {
     / m# j( J( X* f" A! b. F+ Z
136. if(version <=0)version   =   1;
     , A) k7 A6 s% D8 d$ y
137. m_version   =   version;
     9 G: F/ ?$ j8 w8 \
138. 
     
139. #define   NUMBEROFDEVICES 2
     ' k$ V7 P# z; E, p: p3 R
140. CString   devices[][2]   =   {
     % H4 c" k9 I1 J
141. {UPNPPORTMAP1,   _T( "service ")},
     8 H8 |' p* [  p, \% i: b
142. {UPNPPORTMAP0,   _T( "service ")},
     $ t4 |/ r3 e2 m; {, x2 ~
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     9 y; O% W" W9 s% Y1 v
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     : _9 h4 Z' F; e& i% b$ U# {
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     / X* y0 g$ K$ Z7 n0 f5 u
149. 
     
150. int   rlen   =   0;
     7 u2 o4 J. L: S' Z+ N' l
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     4 l& ?, I9 p. p) ^$ u# p2 N. j
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     ( V- t6 h5 Q1 c; v
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     9 _* [2 {* m- B! e6 k$ P
155. CString   request;
     
156. request.Format(_T( "M-SEARCH   *   HTTP/1.1\r\nHOST:   239.255.255.250:1900\r\nMAN:   \ "ssdp:discover\ "\r\nMX:   %d\r\nST:   %s\r\n\r\n "),
     
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     , k/ P  O3 r1 W: D# K! O
160. }
     
161. 
     
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     " t8 T3 I  V) p0 j
166. if   (rlen   <=   0)   continue;
     + P! e6 u  f0 q. t$ j" ]3 J4 M
167. closesocket(s);
     " r2 m& @2 N9 ^- I0 ~3 x
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     & I0 Y+ G) d  ~  ^
170. CString   result;
     9 L+ h/ Y+ E2 C* ^# z
171. if   (!parseHTTPResponse(response,   result))   return   false;
     ( Z; R: A8 W; q
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);
     
175. if   (result.Find(m_name)   > =   0)   {
     / z, {1 ^8 F8 n% |$ l( Y
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     + k" C9 _# C0 N5 p& ?, m  J
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     0 Z" H2 r7 k7 K9 p( v# K
184. m_description.Trim();
     4 X% K; q5 |2 }0 Q7 i
185. return   GetDescription();
     
186. }
       |! [. V0 A' F( y, d: [8 X: `
187. }
     8 y8 R8 w7 x* l1 W* C
188. }
     
189. }
     ! Z+ C  e! }$ g" @- b) h: \: S, H
190. }
     
191. closesocket(s);
     
192. 
     
193. return   false;
     
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
0 r6 C/ w# O1 F
, m8 @1 x% D4 S3 J' J" S1 S
+ Y8 N5 A2 c/ @, p. w///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

3 @% Y* F/ B8 ]
7 `* p& u& t" c+ T( Q) Y#pragma once
; H1 C, b7 A/ ^9 W& |#include <exception>

% M- Z, L7 S  S6 l( c6 }! u
  enum TRISTATE{
        TRIS_FALSE,
* X6 x1 ]0 U* C8 G        TRIS_UNKNOWN,
        TRIS_TRUE
! F' d: W$ C$ b, Q' F};
& f$ N  E' H& D: Y! c# h. j
) T  w+ y: |! e  M9 e* G/ m. r
enum UPNP_IMPLEMENTATION{
/ S( v/ L9 q9 a- Q$ [        UPNP_IMPL_WINDOWSERVICE = 0,
  H8 @1 c) ]4 t, W% x        UPNP_IMPL_MINIUPNPLIB,
0 u& x/ _/ L: J9 C0 Y( Q        UPNP_IMPL_NONE /*last*/
$ @8 ]/ j( p9 X4 M4 f9 r: u};
3 j# @4 m9 P, t5 M' H6 O9 O

1 q3 ]' {$ C* ^5 v( Q

8 j2 ^3 |+ \2 i  i  [! \class CUPnPImpl
9 b+ `) i, B6 [{
public:
        CUPnPImpl();
/ n7 d& o5 [; E1 D        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
! @" ]' G' F% q( e7 s        enum {
* Y% e7 ?. d* b$ B                UPNP_OK,
  |2 }9 `/ F9 e$ ?4 b6 m                UPNP_FAILED,
                UPNP_TIMEOUT
- H( M8 Q! g- c. A  v        };
( k: v4 ?% @7 M
- z6 k3 O  ~  |  a
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
& x/ e) T+ `% r- U9 J+ V        virtual int                GetImplementationID() = 0;
       
5 Z# T; M5 w4 ~        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
, C0 s- B2 I" X% \

; U2 s$ ?) t% k9 r, X        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
0 v" q5 f+ u2 u: l2 q% [. s3 u' K& Q; U6 D        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
* [6 M4 a8 I0 q8 l- P        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
2 i0 c% C# f3 m3 t  k- c; G* d& v( Y        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
# v2 E7 H$ o+ i" s- E
- U1 ^( @' C* I+ d4 j" B7 x) j' t( d
// Implementation
- U* l6 E# x! P# Lprotected:
/ f7 v/ [1 B* j: g6 y9 }" A# u* U# ?        volatile TRISTATE        m_bUPnPPortsForwarded;
' z( A. m9 A% B$ H$ I4 s. M        void                                SendResultMessage();
6 y4 \6 G" P$ A7 e6 c' p        uint16                                m_nUDPPort;
+ z1 F+ [. R5 x$ X        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
: L: _3 ^6 G; k# n/ F        bool                                m_bCheckAndRefresh;


private:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;


9 G" f' Y4 b1 p. w};
/ M8 k1 m% [* L+ J& L% ?
0 }6 h( g+ [/ t9 V4 y  r
// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
public:
! R6 ]) W: Z* c& O& H. S        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
5 Q( K  A, z- Q2 [: Y        virtual void        DeletePorts()                                                                                        { }
; u- K  {- z$ o- s! C7 a        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};


8 t3 a+ o8 e% F. k/////////////////////////////////////
3 d) t2 I2 `' C( N4 j//下面是使用windows操作系统自带的UPNP功能的子类
& X+ \$ b! K7 D0 j% H

#pragma once
4 L: l( |( g- [1 X9 w#pragma warning( disable: 4355 )
. t. L0 Z8 F- w( H9 W7 g! M* I

/ L+ K% o6 g& ~# i' z  s#include "UPnPImpl.h"
0 n; G# B% Y6 Q3 r/ b' z#include <upnp.h>
#include <iphlpapi.h>
' U1 W8 s  v. q6 n$ }#include <comdef.h>
#include <winsvc.h>
# e( ]4 Z% ~& t8 D
) m/ j* j- R" i" b: [+ I5 C
4 L5 J4 S' i6 m, x9 ?5 q#include <vector>
" z7 r% [% I. F) @8 A( x#include <exception>
#include <functional>
3 J& l3 N3 L* O
& H2 U5 [( `5 n% ^+ j5 a8 d
% q" g& C/ U3 c" b- J
% l: g0 T5 H) I! f$ V1 M; T
* Z& ]* d4 G& G' wtypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
4 o; @" m9 [; {  btypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
. Z) N/ |/ s3 V/ B  |8 otypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
9 b/ ]5 @4 a% z7 r! @* k/ Ctypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
& z7 G. u4 J) t/ {0 w: {typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
, x# k9 U3 ~5 b

8 k2 a1 h3 a9 f! \typedef DWORD (WINAPI* TGetBestInterface) (
" {1 S2 {6 w4 _+ a1 T7 W. `  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
* I" U  J9 x1 M);


* }" X; }5 o0 a: ^7 v+ ]typedef DWORD (WINAPI* TGetIpAddrTable) (
5 @' m% n; N$ w8 a9 O% P) l* x  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
);
2 ?! `' o2 Z! z+ e  L$ y2 p  m
6 S  t1 M2 u& Q
) C1 M( n* \" D1 o9 y( X7 G6 |" [4 jtypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);

8 J. K% X& o" Z3 ~1 i
$ Y7 T- T5 a# S7 l% L. \7 ]# YCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);

) R, |) C: f6 f: n) O8 g( g
0 d- p4 H  ^! L  o; q' nclass CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
; O! D4 C! L7 `- }) Epublic:
2 l" b. t9 e5 j) ~+ z: O4 Y        virtual ~CUPnPImplWinServ();
2 s9 I* |. o7 H* [9 W" \1 I) d; \        CUPnPImplWinServ();
2 Y4 l, y6 {8 T2 P

9 Y/ Q; ?4 {9 j/ b& ?' q        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
6 |1 k0 L; }1 Y' ~5 v" n        virtual void        StopAsyncFind();
' [4 a6 @' @" T: C( V1 F7 s, N        virtual void        DeletePorts();
        virtual bool        IsReady();
& F, r; s- ^0 H9 [4 x3 K3 `        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
! ~/ n6 n& `1 F* V# m

        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
. a- J9 |. o2 Q: k        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };


5 B5 ?) ~( w" b2 F' E  z) v0 lprotected:
1 ~- O, d7 b& H( a1 {! s3 A        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
+ a$ s( D& k: G# m  X  A        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();
$ G3 g: ?3 V) J1 t

        inline bool IsAsyncFindRunning()
        {
, g7 r' S( X! k. g1 e8 ]% y, p                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
# c$ l" B+ V/ M6 u                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
1 }: |7 c2 [8 ]# Y                        m_bAsyncFindRunning = false;
                }
                MSG msg;
9 j- G5 U' N7 G. n! `% J( {                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
% e0 {& H2 b5 Z1 @                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
( Q5 V4 ~) T8 U, A                }
2 w+ A: L2 ^6 d% M                return m_bAsyncFindRunning;
/ q! z5 K" [; d: _        }
( _* i( G8 f4 a; N( z7 V+ Y

        TRISTATE                        m_bUPnPDeviceConnected;
% C, F! p6 D% q1 [% N

+ R( q, u6 D1 B$ ~// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
2 n5 W. \+ Y# U        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
# _4 K* u2 {* U5 B        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
& I  c4 |4 Q" u* E9 P        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

- i' z+ ~% c6 P2 z$ o
        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
& b  r: n, ?" F        TGetIfEntry                                m_pfGetIfEntry;
" p6 D' w* D1 g8 B

        static FinderPointer CreateFinderInstance();
) M  C4 m* G8 C( H( B# Q        struct FindDevice : std::unary_function< DevicePointer, bool >
' J. C1 \' _1 l& F# h/ \, t5 g# ~        {
1 v# F4 U( W  |. W& ]5 c: z$ D                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
3 T3 a# Y! l0 o8 B+ ~! V                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
: ?/ |! Y& t# y: `' r3 a

                        if ( FAILED( hr ) )
* {+ X$ s+ F3 F                                return UPnPMessage( hr ), false;
8 n% L! r, m0 A9 Q

3 b4 ?3 w$ D& u                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
+ A! G3 P. V6 j( N; H( u. c; C# h        };
* N. S. z# y8 q0 k  s6 _$ U4 w4 U! O       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
2 f* x. m& A- `1 D        HRESULT        GetDeviceServices(DevicePointer pDevice);
/ C& T6 s1 R( w- U* Y        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
, u+ s& o$ G* a+ J% C7 Y$ J        void        CreatePortMappings(ServicePointer pService);
8 i6 f2 s4 O( q" o6 \( h& d        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
* `2 H5 Q' {# h9 J3 s        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();


  b% ~" u4 h- G. J4 m$ ^( H0 o        // Utility functions
) X6 O: f0 M: G( D+ t. |        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
9 G- M9 {8 V$ k6 R' B; G        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
5 I; f. S+ U8 ?4 ^% O2 Z2 ^        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
/ u+ x. n* ?4 `) v- i( y: y        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
! a7 |0 n' L* K( q' x7 {
6 v  A. h. ^% V. g  \- B
// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
: Q( z3 `; Z, Q3 ^        std::vector< ServicePointer > m_pServices;
; l4 p' O5 a8 V        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
8 U3 H! e8 H: R- B7 T) y

5 ~/ b2 X% W% W        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
) x( {: A0 ~- z# Z# B: g        CString m_sLocalIP;
" k8 a( N" e0 U, @, Y- E7 f        CString m_sExternalIP;
9 \2 b# O) j/ D+ O- ?" P( k        bool        m_bADSL;                // Is the device ADSL?
. {7 o3 ]) L  G" A/ I! Z        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
+ J, T5 d. _# c; @! t        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
" T4 b. R' ~3 V- V        HMODULE        m_hIPHLPAPI_DLL;
: ~) g% w$ \* f! t' Z7 G# O        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;


: j, g7 o* S0 B' R3 O1 x};


// DeviceFinder Callback
' n" h' P, C5 F% H7 ~class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
# N+ ?! ?* }0 b0 }public:
& _, @& l% C+ S5 T; ^* A        CDeviceFinderCallback(CUPnPImplWinServ& instance)
/ F1 `0 w: Z! m$ t8 {. Z) S                : m_instance( instance )
        { m_lRefCount = 0; }
0 U, Y8 A; F( B& B/ a  w
3 u% x: u5 V# \2 N6 l9 X5 b
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
3 Q! H: H( u6 `5 b6 r* m9 C; v   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
5 |6 Z9 X& O5 d- J* a

9 M. e4 j6 a+ K2 U// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
( a" Q! `# L1 H9 y        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);


5 K. r) D* I9 P6 F, L& lprivate:
+ p  q5 C/ l% G8 ^* a: h        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
+ T# y8 T  W0 `0 M

// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
{
6 H* d  }" v9 I4 r' I6 H/ ipublic:
        CServiceCallback(CUPnPImplWinServ& instance)
- G& s0 e) {7 K6 P- x6 A6 M4 x! U                : m_instance( instance )
        { m_lRefCount = 0; }
0 b- `2 J/ @! d7 E6 B   
% e0 a4 A( F' j* d, r   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
  u, A* j% K# L+ C6 X2 x   STDMETHODIMP_(ULONG) AddRef();
4 O; x$ `5 r2 ?  n* P! R/ {1 o" A   STDMETHODIMP_(ULONG) Release();

/ Q3 [4 d! l' b$ Z$ F
4 S+ }- c/ f9 F: x// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


- e# S5 k( V! G" a* [8 P& j) hprivate:
0 A( a/ A0 _5 F* Y6 j) o        CUPnPImplWinServ& m_instance;
* \) H% H4 \: a/ q9 O0 Q        LONG m_lRefCount;
1 P) R) l2 y! U% V/ G; E};
3 O; S- R+ a7 M  l! O; v

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


使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
8 q, Q$ B5 N9 x4 R, \2 I, s1 _CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
2 y0 W& Y. [4 w8 {9 p5 fCUPnPImpl::DeletePorts删除端口映射.