UPnP

天云上人

/*uPnP.h*/

1. 
   7 B- D% Y0 G- a/ r5 e' R% C
2. #ifndef   MYUPNP_H_
   
3. 
   " ?0 l$ e2 D3 @" _. ]) E! k; I3 E
4. #pragma   once
   
5. 
   - Y" d: P; F  e
6. typedef   unsigned   long   ulong;
   : R' r0 b3 J3 @- F# J0 U5 F7 |1 h% J
7. 
   
8. class   MyUPnP
   - W; h! x( R. x+ u' i8 h8 _# u1 p5 Q
9. {
   ) e" u' R1 ~7 o3 c- |
10. public:
    8 N1 ~# a5 Q4 W3 F; L1 |; y
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
      m& p) {; W, B3 s8 w$ [7 D" `
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    ; w+ ^, P% a" @8 c  i# [" d# B
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
    ( E2 d% O) A. u
17. }   UPNPNAT_RETURN;
    
18. 
    + W0 F+ z- @3 Q; e( @# v
19. typedef   enum{
    ( J( B4 L8 j5 r6 V7 o) p
20. UNAT_TCP, //   TCP   Protocol
    8 ]8 g( y0 h* L* J7 \
21. UNAT_UDP //   UDP   Protocol
    % j& B. O, A( R; M9 ^3 m
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    : D3 I. \3 _" L9 R1 s! I7 p7 y" |* p
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    0 _0 V) h, K4 n" ?6 w, ?7 Y" b; {
28. CString   description; //   Port   mapping   description
    % {% x$ R. }6 k, s
29. }   UPNPNAT_MAPPING;
    
30. 
    ( v6 _/ V1 N/ Z1 n2 l
31. MyUPnP();
    ( o/ y" j* R3 [5 z) Y, u
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    7 D0 _# g2 x  `4 q$ |; a
36. void   clearNATPortMapping();
    # Y  q7 E! E4 b# _# s
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    1 Y/ Z/ f6 L5 E" s3 C- O. H; r- ?
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    
44. void InitLocalIP();
    5 c4 y7 B0 h. J) N6 b3 L
45. void SetLastError(CString   error);
    0 _! x* O  U- f. |8 ]7 m
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    + J+ V' R  T* O1 w6 \# m
48.       const   CString&   descri,   const   CString&   type);
    / R+ O( r. @" Q. d
49. bool   deletePortmap(int   eport,   const   CString&   type);
    " q* {2 V6 U0 G" s0 I! p$ F
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    " p' j: C! w. j  Q$ J
52. 
    0 Q" c5 W" r( s
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    # O' [0 O( M4 j" f
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    4 j" Q: F$ a8 }! V
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    ; b$ P0 r$ X4 d
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    - G6 w7 U) E# T( L+ R' s
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    * B+ ^3 A, Y+ X( q/ ?+ o- _
64. CString m_controlurl;
    / C) i2 t! h- e# Y, P0 l0 r
65. CString m_friendlyname;
    2 @9 h+ ?5 u( ^9 h: Z5 c* d8 m
66. CString m_modelname;
    : t0 s- T4 K* e: k
67. int m_version;
    ' k& G4 f5 s' }: z- s
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    / D: i- z0 \/ o+ H/ I' J9 ~
76. bool isSearched;
    ; B% D* A: K4 r
77. };
    * f! r- `& p/ F/ f5 k3 N
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   1 q0 _6 q  z6 m, V3 X
3. 
   
4. #include   "upnp.h "
   
5. 
   ; z! Q; z8 E* O/ h# |) T* @- q
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   ( G1 {" s" t+ U6 C
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   % O1 e: ]% N! b" ^  g" a
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    6 {* v! n! W' `5 s& d0 r5 T; F
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    + [' R% f& ]( S! U9 S
13. static   const   int UPNPPORT   =   1900;
    $ K9 A9 T. |* X0 \2 x9 G% s
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    3 y; O/ v% f1 S0 L
15. 
    
16. const   CString   getString(int   i)
    
17. {
    , P0 P3 l* {- x4 q/ _" N- U+ A
18. CString   s;
    + s) u1 c6 ^% i( T
19. 
    - w9 V! `6 m% Z0 }/ I
20. s.Format(_T( "%d "),   i);
    ( E! c1 [, z6 y
21. 
    
22. return   s;
    
23. }
    : p8 O9 [2 ~6 B, s! L
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    7 p0 F- a6 ~5 W7 a4 K
26. {
    7 @3 E, T1 _6 S/ l
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    + P& u9 }: T! o8 f4 g8 E* R
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    ( Z$ Z+ B) I8 Z- f
31. {
    - T, c7 p& e7 O
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    3 w7 ~, J+ w. `* u  e% l0 y% v' L
33. }
    4 \7 l/ R) f: u" `4 B
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    4 u  e; t& C. S" d" ?# L6 V9 X8 U% p
38. 
    + l' b4 X2 J# F/ y8 i. v
39. const   CStringA   sa(request);
    ; Q0 d7 q! J' F+ l
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    6 m, x7 @3 D7 J- m
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    & v0 J. h% F/ J* }) r& j9 y( Q
44. struct   sockaddr_in   sockaddr;
    2 K5 d7 a" [1 S# u$ p" e
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    7 g( B' ?7 |8 M% B2 K
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    2 t: l0 {, v6 E* m* V2 I8 h: f
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    # ~9 P: R/ S2 o& a, M
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    # N% F8 |3 W7 b8 h; o4 f! q
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    ( f) G" Y3 q- n( W! c% I
59. if   (!rlen)   return   false;
    , R* u! }8 W: m8 O" A; F7 @$ w+ b
60. 
    . b( ?6 i* R  p
61. response   =   CString(CStringA(buffer,   rlen));
    1 c( M7 m0 d4 u- P) {2 t7 m  M
62. 
    
63. return   true;
    
64. }
    1 Q( T  W7 H6 _! S7 p
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    3 n  E! X% ~& f) Z7 S: N  B
67. {
    0 o# G% c7 @$ B! s4 d
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    & X6 h  v: X3 Z# }
72. strcpy(buffer,   (const   char*)sa);
    ) M! k$ Z( V8 S# N3 w/ h
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    # o& B  _9 C- i  g9 q& v
76. sockaddr.sin_family   =   AF_INET;
    2 O6 Y* m5 i' |  N4 D' R
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));
    
81. }
    # q# s& x0 T" _# ^
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    1 o& p8 K1 v/ t5 R1 {% u. X1 h
84. {
    
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    8 M2 F1 s# b: r$ ~4 p7 [* P; U
88. 
    6 c' E: _( [+ M& r& Y3 ?) h
89. result   =   response;
    " z8 }& P" Q; y& B6 }& H+ `
90. result.Delete(0,   pos);
    ) L, I. R$ |; `5 c% C3 L' M
91. 
    - |8 A" ^* z9 J
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    0 K: h0 _+ `9 w7 y3 r
94. status   =   status.Tokenize(_T( "   "),   pos);
    ! R4 h% E6 X7 g) W# p+ S0 R1 j
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    . Q1 s' U4 P5 A6 `6 b
96. return   true;
    + Z, w+ q5 f/ I+ s9 x) J# u9 H( e0 l
97. }
    
98. 
    ; p2 p, L0 I7 k5 F/ g
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    " K4 d8 i4 g& m8 {5 P
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     8 m2 t; H# |8 `' w; i0 k
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     & N% Z0 A. J( w* }. K
103. CString   property;
     1 K5 f5 x! G0 Q+ G$ [
104. 
     
105. int   posStart   =   all.Find(startTag);
     # S$ v# m, C  Y6 Z% W" \+ J
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
       ^7 m& X( |- ~& ?& p  B" C
110. 
     , [2 j7 P, m7 d( q" m
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     - n0 L. _+ z. P- G6 H4 F8 Z9 C8 s7 U
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     8 g1 P) ^5 h9 M' u
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     1 b) h: ~3 X2 L
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     5 r: K! U! X! i( I, N6 E# r6 I
125. while(pos){
     9 F( [+ X  W. g: g, P, K
126. search   =   m_Mappings.GetNext(pos);
     $ G5 h& h! b6 O
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     7 C7 c9 K8 H5 K
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     & O0 Z' f5 D; ]6 }3 u
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     6 t8 F6 W) a5 q1 C% E
139. #define   NUMBEROFDEVICES 2
     7 s( F: i4 p" W/ t
140. CString   devices[][2]   =   {
     6 V/ H, j- a" U( v+ ]# W
141. {UPNPPORTMAP1,   _T( "service ")},
     ! S: n) ^- X3 S" p4 {  b
142. {UPNPPORTMAP0,   _T( "service ")},
     6 J& q9 w- E9 d$ @
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     / f/ B  g: N6 d) O0 \1 B( E
144. };
     1 L6 a4 b6 b* }+ L/ \- H6 K
145. 
     2 V3 a4 l: ?" ?* Y4 {. I6 A
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     2 }/ f  `2 C$ \
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     9 u& s2 P9 s9 I' V
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     5 A2 f5 Q' k6 q' G) P9 s+ h
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
       y" `0 g1 V8 K% X; Q; u; p
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     & \7 R% t1 ]5 b# U) S0 P' j
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);
     7 Y3 t* J% n8 R1 Z4 Y( Y
159. }
     
160. }
     4 u7 V" g% z3 D9 Q2 ]
161. 
     
162. Sleep(10);
     
163. 
     * B& ~+ @* k  m3 G- t) m
164. char   buffer[10240];
     : R; [: Z2 j+ w- `$ o
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     : s6 J8 T' \3 K
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     # w: w  @* m5 _* ^) [2 Y# o0 ~
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     2 w: z  Z' e2 d& K1 a, E
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);
     5 o8 v2 B3 H' ]: W! r5 Z( ?
175. if   (result.Find(m_name)   > =   0)   {
       c- [! n2 y6 G3 |3 v
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     6 E2 Q* q) N  }7 f3 q
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);
     * Q! ~, I# n! b4 g" \- ^
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     ) }+ O) C4 T; Q. m7 P
186. }
     
187. }
     
188. }
     
189. }
     
190. }
     
191. closesocket(s);
     
192. 
     
193. return   false;
     - |& K' h9 U8 U2 E0 j6 M
194. }
     

复制代码

天云上人

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


///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
6 \+ N8 Y3 f# v0 e) J5 L) H

6 U. f! H" I8 b( f7 i- c- ]5 f/ A' @#pragma once
% ]# `/ J* q/ A, K. r#include <exception>

' D5 p# ^, k7 _; x
  enum TRISTATE{
        TRIS_FALSE,
% c0 N2 `/ v+ b/ V. |        TRIS_UNKNOWN,
6 R. z) @. `3 {: P- y1 S9 ~        TRIS_TRUE
9 I7 _& E" D7 u6 D};

& _  u9 ?) v9 w
0 ^* ^, \+ W# l* p- ^enum UPNP_IMPLEMENTATION{
4 e& a6 N0 u! w/ e" m/ y+ @        UPNP_IMPL_WINDOWSERVICE = 0,
6 O9 \5 [) S  j! U, Q3 B        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
; ~4 z, C9 e2 g; E# m& H2 l& `2 R};
9 }, W, V, R- v! T( }' r& Y  A

, t/ b# T" o& P' Q* t
  X& Y3 w' K7 z! T+ }% G& H; M+ {8 W
! I. E* u4 @8 `* I% D) ~class CUPnPImpl
8 @2 c6 |: j" S7 J. T0 S& p( h{
8 v) Q' W5 q/ jpublic:
8 U" u$ z. V. k, Y5 S) C        CUPnPImpl();
        virtual ~CUPnPImpl();
. s) u9 I) }8 V/ O0 v/ x        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
) I- `$ ]! d0 A6 O7 x; m& s                UPNP_FAILED,
$ O4 t4 r, |/ D( R                UPNP_TIMEOUT
        };


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
/ f0 j+ C$ n! P! E& O8 ^- F+ J        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
" V7 |# a9 [/ _! V1 C       
$ T, s$ g0 j5 z, q        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
" V  [, m1 S; v0 n
2 ?1 s0 o  i& S( Z. _; J7 D9 t& ^- V
" H) |+ G' A( v8 {% z        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
5 H8 |/ D' J" q" }4 s: W        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
3 W% l2 B0 `6 v0 ^        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
% ?. X4 x. O8 [; v$ F8 r' ]        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
8 S4 z& B2 B# ?. o7 K% T* W/ v+ Q

// Implementation
  f/ o' @# ^5 G; ?+ C& F# E. v, Tprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
# A' W0 L9 L- W" R/ A! J/ f        void                                SendResultMessage();
2 U7 T* u0 P+ K; d4 v        uint16                                m_nUDPPort;
* T6 n* S, c2 H        uint16                                m_nTCPPort;
; i. O% Q* v! X2 Q5 Y, h8 h        uint16                                m_nTCPWebPort;
6 d5 F) h8 i4 b4 P2 A! f( I        bool                                m_bCheckAndRefresh;


) d3 c" ?: H/ K3 Rprivate:
        HWND        m_hResultMessageWindow;
# |3 K5 |' A3 a! _# q9 ~0 M        UINT        m_nResultMessageID;
0 b9 o, e; x1 c1 W, v+ y

, m! J, z( k9 e0 O4 l};
; @2 N) `5 E) L- o. x: R9 ]
9 @; w8 l$ ~- H& @- A* A7 ?
// Dummy Implementation to be used when no other implementation is available
$ Q: c/ n" U# U: e/ X" ~class CUPnPImplNone: public CUPnPImpl
{
. V+ b8 q/ i+ apublic:
4 ~0 P1 [1 [* x3 s- w        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
. i/ q- R& V0 E        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
1 j# z# K$ C3 @$ H        virtual void        DeletePorts()                                                                                        { }
8 d( u+ t3 y' v4 J        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
! s& O3 ]3 T  ~3 O: s};
4 q# {8 }3 t' d1 q, }, s" @
) t, E) ]9 R4 }* L) n
# p" {* `7 j* ?, A& j. d/////////////////////////////////////
" v( g6 `/ Y1 H. i& i6 m//下面是使用windows操作系统自带的UPNP功能的子类


#pragma once
# `: ~4 r% Y# }+ j! w4 N6 o6 i#pragma warning( disable: 4355 )

$ S% k' ~) h: C: A* A
% G5 {% W/ S# G; O1 n/ c; l$ W#include "UPnPImpl.h"
% `/ g0 ^0 x% H+ P8 c2 K1 \#include <upnp.h>
#include <iphlpapi.h>
+ e+ D4 o( w' \# e  [% j  J- D* L#include <comdef.h>
5 R; P* O$ S' \0 E" y& D#include <winsvc.h>


#include <vector>
3 b! |% `5 D! q% K#include <exception>
#include <functional>

$ G  V9 T4 L6 q. }9 W. Z9 ~
8 o+ j# @% C3 e7 R# [* I+ I! ^3 S

typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
, Z2 ?' D6 p! Ttypedef _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;
3 l* k/ @0 |; t* _# `5 xtypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
( _( d% h; n# i

. h# [5 |1 z# l: Mtypedef DWORD (WINAPI* TGetBestInterface) (
9 u' W& p) \( r9 G: A7 x  IPAddr dwDestAddr,
3 d+ u  r/ f. t9 K$ C; X' w  PDWORD pdwBestIfIndex
);
% j8 W5 c' `- u7 s
; ]/ B9 @+ L/ @" h& B* Z
0 g7 t" ~  C% e2 Z" B" a2 q6 itypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
5 {. T: c; z/ L) W8 T" V7 `8 t  j  PULONG pdwSize,
  BOOL bOrder
);

% Y8 \0 b1 P3 H! F: Z' R
: D0 z0 T* I  E, G( o3 htypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);
& M* b  {' m- `, l0 J
& k1 N1 c+ d! E* h: e1 Z
- r9 ]/ u+ X& u7 x2 T/ P3 LCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
- i( e) _) B. B+ g: U( d, u) c* w
: b4 U4 M* x, x( g9 \2 w
9 y  w9 U: N3 X% t& uclass CUPnPImplWinServ: public CUPnPImpl
9 T$ {+ _: K1 \{
, ^& _0 Z% a. N2 n5 {7 b1 m8 P        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
public:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();

/ B: e9 Y; b& \8 L. X+ O0 H
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
0 f1 P& r9 _4 f2 x& y6 @) F* S! [0 ^7 Z        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

( W% p& {5 R3 L) s4 N! c
8 p; @0 R( Y6 P+ M  g        // 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
        virtual bool        CheckAndRefresh()                                                                                { return false; };
9 U( U( S9 H. }, S1 Y- o

protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
4 n: P. T  b3 x) a* z1 g        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();


        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
4 ^' _) p( ]* P& @5 T  E                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
! M1 e/ V' \2 m                        m_bAsyncFindRunning = false;
+ }" T' I; |  O8 g' y0 {                }
. `6 Q' R% ~$ J4 O                MSG msg;
9 B, ?' J9 O5 C7 w3 [  o. _4 p4 ]. _3 o                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
0 p) k$ V* [3 x7 a4 C                {
0 O& r4 ]7 M* ~) z                        TranslateMessage( &msg );
# b! ~/ I. R# }( r; w5 Z3 F# U. e                        DispatchMessage( &msg );
" F  T- c/ x4 c9 S                }
                return m_bAsyncFindRunning;
        }
) c1 ], M% W4 i  `
% c% c  Q% f( M8 y$ U% [
/ Q) W' J" W; M, h0 O* t        TRISTATE                        m_bUPnPDeviceConnected;
7 s& h( \4 T  x& u5 ?

// Implementation
        // API functions
+ b6 C9 o, `. q$ {! \# F8 Z9 t        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
4 z: W+ H( O  ~/ G        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
8 x  ~6 j" N5 V) [
+ e! q7 U- \9 |% x* Z
        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
( {% o3 f) t$ G8 l; t% W! I
6 a, @; d0 K, {+ s9 x6 \8 @
        static FinderPointer CreateFinderInstance();
  n) k* Y0 \4 a; |        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
; |/ R' M; @0 z6 A: q                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
8 P" G& R- a8 X: t7 D3 \                        CComBSTR deviceName;
2 v: `  m' t6 ]3 _- K' X, t/ J                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

2 V, u+ I# D4 |4 u, J) `( u
                        if ( FAILED( hr ) )
; R0 ]! X% w4 e6 ]+ c                                return UPnPMessage( hr ), false;
3 h# K- ^$ {3 A6 w

6 U4 u, q2 s, _3 b! V5 X3 B                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
+ ]: i/ B8 n5 T8 C        void        StartPortMapping();
0 k6 c* H& C8 W: c! r        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
: L4 c: _  ^7 \# u2 V$ M" L        void        CreatePortMappings(ServicePointer pService);
/ m, L3 L1 w7 p3 V8 {* F        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
, D0 Q* c* g, W: X: u                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();


9 {4 I& A3 o+ `        // Utility functions
) q, }3 v8 P$ D, ]; N        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
; l" w8 s$ D0 G) o7 y7 Y        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
1 D  \; U# F% ?        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
* h; \& Q7 |3 r        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
8 p# A, r" H. i2 V3 P/ t3 k" j6 J) _% Y        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
8 Z: T6 t2 }$ `. F. J        CString        GetLocalRoutableIP(ServicePointer pService);

2 A. S3 u0 x7 g) \  k5 k/ U: I
. P4 A. M! i, a) `6 _$ |* p1 ]3 U// Private members
/ V. B; N; |+ G! U7 a2 Q" T  fprivate:
: }0 E- t" l7 U# m1 F  V2 l. K( B1 b        DWORD        m_tLastEvent;        // When the last event was received?
5 H0 j9 V* i. F: r6 m        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
/ h7 x% L( q0 s3 W% y' ~9 s7 F        DeviceFinderCallback        m_pDeviceFinderCallback;
* M4 m+ K- }- h' k, T        ServiceCallback                        m_pServiceCallback;
$ d5 n/ y5 T) P  X0 e
+ P$ {; c5 D. j0 E
        LONG        m_nAsyncFindHandle;
3 c$ s2 ^0 B; _        bool        m_bCOM;
        bool        m_bPortIsFree;
3 w4 T$ g! S: q7 T) ]0 i        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
2 |) c. D7 C4 {! g- z) u        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
  C7 X/ @; U  K9 f        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
, u9 j! Y1 B3 [2 m/ L: c# f        bool        m_bDisableWANPPPSetup;
4 G. K- |1 B, _
7 d" |. D( S* I/ s* G# Y1 K# @
};
1 x! r0 K2 O  j; i3 x% m

// DeviceFinder Callback
! }; x5 u5 A( G. zclass CDeviceFinderCallback
5 w2 ~- \' _7 t: k( f2 ], t$ J        : public IUPnPDeviceFinderCallback
1 ^; B! _2 c+ u8 V0 }% K{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }

) r$ _3 p# C3 [! s
6 |) [9 O' b% W   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
2 ?5 g* {, u4 X7 s7 u) _$ q8 V$ A   STDMETHODIMP_(ULONG) AddRef();
8 m. `2 a+ r$ T- e2 @( |, K5 T   STDMETHODIMP_(ULONG) Release();

& w' z% F* J7 |6 Z$ l
) ?2 A/ o: t& X. J8 p// implementation
& S! z9 `' V' `; S7 lprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);

) v/ u" t5 p" ~  A
/ y2 ~; \, y9 H1 g' bprivate:
' v8 M& T( b7 E1 z+ W# \        CUPnPImplWinServ& m_instance;
2 X. Y* z/ }5 z$ D2 \2 ^        LONG m_lRefCount;
};
" j4 T. B/ `, d) ^
# B; E) J) A# X0 V' I
3 \. x3 \9 @& n3 ?0 B. a- A: B3 _8 J// Service Callback
% u; @" K4 j% V+ D9 o  A* kclass CServiceCallback
        : public IUPnPServiceCallback
3 l$ @* O1 m+ O+ c) S! S3 `3 i{
' E; `' E' J! T7 }* y5 ypublic:
( o$ Z5 `2 p. x  k9 i# D        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
( B0 c; `0 m* d9 H        { m_lRefCount = 0; }
   
: \; E3 C* F1 \- L3 }( y1 @- X6 L   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
9 R! w* F5 ~! S   STDMETHODIMP_(ULONG) AddRef();
" N6 {' s! s7 y: ]! O/ Z   STDMETHODIMP_(ULONG) Release();
2 K* U; q$ I3 R; a

// implementation
/ W1 J: d* L2 K9 Q) @: q5 {6 kprivate:
( G; ?( b, `8 I8 @) x        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
9 P! ]1 G& J% b) E# k! z        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

6 d( F3 D) n. `  P: p
private:
        CUPnPImplWinServ& m_instance;
+ E9 n! g8 M- I        LONG m_lRefCount;
};
# w4 u+ G3 |/ O4 q

( C. v: U9 j' S& r/////////////////////////////////////////////////


使用时只需要使用抽象类的接口。
( }- p. r) A/ w1 G2 eCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
# m$ _$ _; L* V$ }4 \CUPnPImpl::StopAsyncFind停止设备查找.
. S- b2 H! g, Y! q+ b( B* oCUPnPImpl::DeletePorts删除端口映射.