UPnP

天云上人

/*uPnP.h*/

1. 
   3 S* x. V; M9 ^" o
2. #ifndef   MYUPNP_H_
     ^- S2 G  Q, t7 y( V+ \
3. 
   
4. #pragma   once
   
5. 
   1 O5 v5 y0 Z8 s
6. typedef   unsigned   long   ulong;
   
7. 
   ! Q) ~; ]; R8 ]  R
8. class   MyUPnP
   % N% x; L  i' ]" H( u0 W7 }
9. {
   
10. public:
    
11. typedef   enum{
      V; @0 P* m3 W+ A" M4 M0 s* [; h
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
    
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    5 e2 [# }4 [  \; V& D0 z! j6 x* \: X
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    : t' o5 b  I: f
17. }   UPNPNAT_RETURN;
    ( D/ ^6 [9 i9 z/ S
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    # S; ]' L+ l/ I
21. UNAT_UDP //   UDP   Protocol
    7 [5 w9 `. ?6 ]  ?( \
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    7 v# ?0 D/ @( M5 ^
25. WORD   internalPort; //   Port   mapping   internal   port
    1 d  L8 `7 ^( f  Y) c
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;
      l  }! \- J' R
30. 
    
31. MyUPnP();
    # d" d# p4 g! [/ ?' b# X0 l
32. ~MyUPnP();
    8 K( q$ D6 B5 ^: u& F& j5 b
33. 
    * T, n; s3 T3 {  t& B  V  b
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    - S- I* D  m8 Y$ x/ T
37. 
    4 @4 W  f) [$ o3 {# R( L: T/ c4 A
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    , L1 z  o! {  A" }9 B1 o! W7 G5 ^0 e
40. WORD GetLocalIP();
    7 Y7 O: a3 ~" o( d3 h6 x
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    4 x) l0 y3 z5 Y! G; o
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    7 j: n6 y  d. u& s
46. 
    : c7 c4 |: D  s  s9 X" p
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    2 e1 T- L/ X) F
48.       const   CString&   descri,   const   CString&   type);
    % _) g6 _# `' B, z* ]
49. bool   deletePortmap(int   eport,   const   CString&   type);
    + U: ?8 D( m/ ^9 k6 k
50. 
    8 d! w9 f4 N1 f
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    ; G+ T2 u8 S8 i9 T; {
54. bool GetDescription();
    1 ]. B2 h4 j) q$ S/ W4 t5 n
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    ( s! h, Q5 g. S# O
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    0 G5 c) [- Q6 B( q: @9 m  z5 y
59. bool InternalSearch(int   version);
    % D: l. F4 x% y) k; I# o
60. CString m_devicename;
    2 I: H9 l" ?* \4 O" j' q9 S
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
    7 W. _4 t( n, K; i: s
65. CString m_friendlyname;
    1 I. g& g/ I* R/ R  z2 x
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    8 G* H* z7 H7 I
71. 
    , ?# `/ f. K& M1 U
72. CString m_slocalIP;
    ! z: O% h1 k! l  c1 K7 _
73. CString m_slastError;
    + D1 P: }* Z5 v8 k& }
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
      d! B( L" Q1 R4 @- g
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   + r0 X3 k8 C0 v
2. #include   "stdafx.h "
   
3. 
   8 c& x- u! n, |5 _9 G5 ?, A& c
4. #include   "upnp.h "
   
5. 
   ! [( K( X! F4 [. D5 ?6 Y/ l% G
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   % v* c, U  g- Z, }; T* B
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   7 |6 Z# Y& N& d  I# G( ]
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    ' m! P0 G" O/ g
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    5 j% P/ e2 \) Z2 h5 v! C
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    3 C, \* L% B. F6 ^+ E7 o; D. ?( j
15. 
    
16. const   CString   getString(int   i)
    ( G: D2 r8 b7 i  U3 r, ^
17. {
      k4 E5 c% D* A: n* `2 ^. g
18. CString   s;
    
19. 
    $ p6 T6 F% j) c( ]' d
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    # H: p9 }& x$ h+ Z1 F  a6 F
23. }
    
24. 
    ' y* W0 k4 E; J! f6 i
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    9 k0 ?+ l0 M2 |
26. {
    : s6 S( O5 R; y) }% e: U
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    7 ~) V6 ?1 U6 R
29. 
    . R' Y2 q3 ~; f3 u# w8 B/ J
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    : O+ C+ \; q$ z& l% N
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    ) Q$ l+ M; a* R& j
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    , B/ j' F- @: t  ?; i+ ^3 q2 K
42. 
    2 P! Z* c7 w6 h$ ]& i0 C% _
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    % Y: H2 `9 W5 g9 Z! U+ q& l
46. sockaddr.sin_family   =   AF_INET;
    6 @' P% F" ?" x. _  j
47. sockaddr.sin_port   =   htons(port);
    & x3 f  ]; ?  u" W0 _
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);
    # @* J: g2 Y) G1 K
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    : c( I3 E5 ]9 K- N7 |- G* j
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    . L% V& P9 j% Y5 g( E
55. Sleep(100);
    2 l% q% G4 Y6 L9 k7 w
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    - z+ N. W/ J( R9 T
57. closesocket(s);
    7 P6 \# V# @, `% Z2 J
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    . r; p, K4 T; _- F1 w' Z' _
59. if   (!rlen)   return   false;
    7 G& O# Z7 _1 B# A$ a- i  I* Q
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    ) j) Q; z& M- ~, A9 t
67. {
    8 e! \3 `+ b! F- F; _$ r9 A" u
68. char   buffer[10240];
    ' m! j  Z( l5 x' \* ?
69. 
    
70. const   CStringA   sa(request);
    6 {3 B# H- ?/ x- |1 |( I3 x) C
71. int   length   =   sa.GetLength();
    2 s2 d( Q+ W0 a- S2 @% ]0 Q
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    ! {; P; I, m+ A4 v
74. struct   sockaddr_in   sockaddr;
    6 M& b5 V1 r( v- ?  \# Z1 U1 P7 O
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
      {' _  x; C8 F! k. L! n0 \0 d
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    0 m; X' v3 @. Q4 p
79. 
    ) G2 u( {' R. A; W* r
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    5 K& Y. O/ o' z. {2 R5 \6 g1 R
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    8 [) }- y* n8 \4 ^1 L! R
84. {
    % K  \4 ~- n( H) o- \" g; ?
85. int   pos   =   0;
    
86. 
    9 T% @' _8 D. M0 b& y8 p9 c8 I
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    $ u$ y! B8 b& ]& o9 l+ w  `
89. result   =   response;
    7 C  e) |6 K* C& ^/ Z  z
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    6 d7 |/ q  q% N# m( ~. r5 X! m" Z  O
93. status.Tokenize(_T( "   "),   pos);
    % O* ?& \5 L- j7 {0 M# g2 a
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    ! z0 F3 X# O. R% B1 X# b
97. }
    . d, b4 {) E# Q  g* d
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    . P0 `, |- N0 X( V) g
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     ( ?  }. i8 |2 T4 K7 s$ {
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     0 ^. d) [$ O- Z' z+ L
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     4 y4 v, x+ Q& `9 Z& M3 ?
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     ) _, @6 ^  l' P3 V
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     . I  B% p: X0 h4 u6 U3 e: K& R
113. 
     4 b" I6 t/ x( a
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     / L4 c7 ?5 O+ n8 P
122. {
     
123. UPNPNAT_MAPPING   search;
     8 }5 h. D; r) _
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     - t2 `* J" d2 V' ^4 f/ w1 l
127. RemoveNATPortMapping(search,   false);
     
128. }
     " Q; \, |1 J/ M4 W- a0 z5 h
129. 
     
130. m_Mappings.RemoveAll();
     7 u8 ]6 p# G% w/ [, Q
131. }
     
132. 
     + {& Y' ?. [9 N2 G6 S& x
133. 
     2 |) f( B. y* t. Y6 h. ]9 V2 I; I
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
       D# d# \7 U- E
136. if(version <=0)version   =   1;
     * `# c* n6 D! [4 a3 ~
137. m_version   =   version;
     
138. 
     9 ~3 }  [$ y, n$ P7 n9 s  y
139. #define   NUMBEROFDEVICES 2
     6 o( M) A1 J4 m8 n/ N
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     : x4 c+ _3 ~5 f4 n# S
142. {UPNPPORTMAP0,   _T( "service ")},
     " J% g0 ]6 C/ o3 t8 N" s2 X6 i0 D; ^
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     5 G5 f2 q( W& n% v( g5 K- q/ P8 T
144. };
     9 [# g4 C# Y9 k" P
145. 
     3 u. U# r) B& Q7 m- m2 S
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     . k9 N1 E' q% r( W7 i8 p! `
147. u_long   lv   =   1;
     " x6 J0 [/ i6 }! z7 Y
148. ioctlsocket(s,   FIONBIO,   &lv);
     8 C: t/ A, ~) c
149. 
       W7 W% v& ~3 d) y: h
150. int   rlen   =   0;
       j: Z0 `+ W9 U0 J, L
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     9 j7 i* c6 X4 n2 D- X1 I6 j. _
152. if   (!(i%100))   {
     . F/ W# t. H4 @# b& K
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     " l3 _' P' A% R( g' A1 t, `6 F  X
155. CString   request;
     + v& X4 R) l  d2 n" l- d2 b6 G+ L
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);
     2 E+ _/ k4 i$ A, O' a2 L
159. }
     ' x/ V. l8 k6 S0 }& Y1 E
160. }
     0 F; O3 t2 ]( Y
161. 
     ! V) g% w4 |0 H  o1 ]
162. Sleep(10);
     
163. 
     5 L; [8 j8 N! M& Q2 H
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     8 J7 o. P2 |5 y, F- j6 h  C: S
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     ; J0 I0 c# D, W" r2 f  P
171. if   (!parseHTTPResponse(response,   result))   return   false;
     7 K1 L% t+ {" C+ T, w
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     3 r8 w4 ?/ x  t
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     ! W" p& A. g- y. R! P
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);
     
180. name.MakeUpper();
     ( D3 p* Z7 A! h! A( N6 _/ Q
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     * `! w6 J+ {  ^$ H6 P4 l# I
183. m_description   =   line;
     
184. m_description.Trim();
     . S1 b7 R+ a+ k5 o" }
185. return   GetDescription();
     8 |! T7 }6 t6 y- p0 {+ \
186. }
     0 X' y& s0 ~$ n& _- `
187. }
     
188. }
     
189. }
     
190. }
     
191. closesocket(s);
     
192. 
     
193. return   false;
     ! C" Y5 c' P: B- }7 H3 }
194. }
     

复制代码

天云上人

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


///////////////////////////////////////////
0 x' ]6 N2 V$ w. }3 ]//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
: Q4 v% t3 \) {" u/ T: `7 m2 ^: z' d
$ Z7 o% f7 x$ N' `3 G, H% M
#pragma once
- L1 \% k+ O+ J& T# D8 \2 L& i: q#include <exception>
! t. G- I4 B8 k; o$ C) T0 q/ F) Y9 Q1 ?
8 y' \7 D2 v; l) t' @6 d
  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
};

6 b; e5 O. r) n, }- w3 u: e* G% f
5 n0 E" a+ A% D4 q/ Genum UPNP_IMPLEMENTATION{
" C$ Z3 H7 o6 O+ E/ e4 }# v' w        UPNP_IMPL_WINDOWSERVICE = 0,
0 X: B- c8 w4 j9 [( G9 b        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};
" ^' a; @2 e9 z4 v, O; k* c7 ^% L

! w# z, d) f9 w$ V6 V$ z

  s  ~3 g: D( b% {8 o( A' n4 oclass CUPnPImpl
{
public:
        CUPnPImpl();
7 w2 c" {; S$ S6 L7 @4 \# p2 [        virtual ~CUPnPImpl();
, a. j6 A% T& u9 }6 c& j' W        struct UPnPError : std::exception {};
        enum {
$ a1 O) A1 Y# s. F7 M                UPNP_OK,
% E( y2 v$ o9 @                UPNP_FAILED,
) N. b5 y1 a2 e5 U) ~; ]6 f$ }                UPNP_TIMEOUT
; o7 E( r. H: j( y  [; X        };


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
1 y' B4 V; i0 c# l" W, X        virtual bool        CheckAndRefresh() = 0;
! s7 I. \: S9 ]3 Q        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
4 X2 q6 Z7 F) m& L        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
" P8 d. \1 E/ H8 n" T' V7 `6 G
" p! b( l4 S2 V
: j  U' w' Q+ u- O% k8 B- k        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
9 \/ W$ x8 h/ N7 l1 E4 ^: S' o% V        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
! J! r2 _; ^- k! `/ k( [        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
( _! T' W" J! \. F        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
8 |: `% c5 f4 I2 F% u( _        void                                SendResultMessage();
        uint16                                m_nUDPPort;
5 X: \+ [$ v, y8 t3 J: t- @/ d        uint16                                m_nTCPPort;
7 m. q( @5 J/ _' \% t) r1 M        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
, q3 ]" w! W* R: a% R# ?
+ k9 L; G. U- `$ E+ I2 A+ `
4 d; y* @0 f. d( |) c2 I$ Hprivate:
) n5 _9 L5 M0 r7 N        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
9 d% m  \+ l3 V. @6 o2 [& x, B
  ^# E  K, U% g3 d4 Z* |4 @5 P
6 T1 V' a8 Y( C) y6 O; R7 e1 E' r};
5 A; m4 d9 o8 k; k' s; }
& n% ^' N2 V+ r$ C  P( Y1 U
* p; v. }  x! {" ^4 s// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
# X, L* g% v) J' o& q* N{
9 _/ f& @4 W1 h0 P) m9 \& L2 q( fpublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
* b  K9 \' K/ a0 h/ A9 R        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
9 H0 H. m% s- [6 G2 N};
0 R( D8 s& D& a: R: a$ S

/////////////////////////////////////
' X" q# S( K2 e; j1 {% d% X//下面是使用windows操作系统自带的UPNP功能的子类


& `! ?6 X  J" D- q#pragma once
#pragma warning( disable: 4355 )
1 |  D' s9 b8 U8 l" P

/ [" Q& w' B* M6 }1 ~8 M#include "UPnPImpl.h"
/ h7 ^2 P- v) w. v#include <upnp.h>
#include <iphlpapi.h>
  ?( l$ T+ K* L( l7 D1 [! h& A; T#include <comdef.h>
#include <winsvc.h>
5 F& \: t8 A- V3 s* Z1 p" i

#include <vector>
" Q% R6 q3 B- T+ l) d. ]  Z#include <exception>
#include <functional>

; {' c+ ~* j8 @! c
1 g* ^* X3 B8 \; H4 n" Z

typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
1 w/ R  P4 @9 `  z+ i: ntypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
8 l9 W+ M/ q3 T7 K  ]typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
% |1 ^, m- n2 j) P, C7 ?/ htypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
, C5 L) C. l; r  Y' E; j  PDWORD pdwBestIfIndex
1 @4 ]8 J6 e' d$ A& z);


typedef DWORD (WINAPI* TGetIpAddrTable) (
$ f% P; r9 z9 b$ ]  PMIB_IPADDRTABLE pIpAddrTable,
# v. K" D) w$ g' W  PULONG pdwSize,
  BOOL bOrder
);
1 c: B' R4 i* |4 w, X9 A8 v

+ s/ ?+ ]3 n: C8 V& xtypedef DWORD (WINAPI* TGetIfEntry) (
( Q: ]$ e6 D0 {4 r  PMIB_IFROW pIfRow
4 |2 z. U3 C1 Q, k);
& z3 H4 C2 u9 B/ G# s1 E0 `

- L6 h  U- |# ]3 ?CString translateUPnPResult(HRESULT hr);
$ e) [% ]: S8 p( R7 @8 bHRESULT UPnPMessage(HRESULT hr);
0 F& o' l+ |6 A5 a# N& o4 Z

class CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
$ S, d3 A, ]9 }# P  q/ s. Q+ o// Construction
public:
- ?& X9 p1 G) H/ R. _        virtual ~CUPnPImplWinServ();
* L) M5 G; U* Q        CUPnPImplWinServ();
% x8 l) U: c" J1 f9 \" E% _2 A0 Z$ Y
* }: }" R- A+ W1 d0 i
2 {; n4 Q3 S) E. `# |$ ]        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
! m: ^, C- `5 U) K        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


& t9 F; I" A  P# }7 c. L        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
) `; {  R( z7 X% q        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
; }0 A3 x1 c8 ?- }& H8 Y3 M, q        virtual bool        CheckAndRefresh()                                                                                { return false; };

: @; k) o$ E4 J8 p+ ?" l
0 C* I5 a& i) cprotected:
5 I4 S$ Z: g& X$ [$ `/ Y( e" g        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
; }5 `/ z2 S) f1 x        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
/ M8 K* |- M% s; M/ I/ }        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
2 S, @# s6 o- n        void        Init();
( g: V& G0 }) o) |1 [
& S  ^" |5 ^- ?5 E- ~6 I8 P6 y
+ ~. S. I! ~( G  L* I8 ^5 w5 o1 }        inline bool IsAsyncFindRunning()
        {
0 K" `0 R7 ^: O, F: X                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
/ P0 N& U+ N: ^! j7 \4 _3 O                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
% [- A* s. G) h3 @1 c, V4 I0 Q                        m_bAsyncFindRunning = false;
, U. C1 |5 d) G2 ^% n                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
; A# ^5 n3 X+ L& ~2 P                        TranslateMessage( &msg );
. }7 I- m- Y& j. |1 c+ l( n                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
        }
+ F8 K$ ~/ K! _3 ~# w  v

        TRISTATE                        m_bUPnPDeviceConnected;

4 B/ n( _% p7 N3 x6 H  s
# G) _2 h* e) d+ c// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
+ F, N+ {  K. C9 M' ?' t        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
: C3 x& d$ B6 S        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
, \  i: S5 r7 ?3 s! s: `' K4 ?' l        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


, R- J" n) q, x5 }        TGetBestInterface                m_pfGetBestInterface;
9 A0 r7 ~% w% s6 I; R9 ^        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
' d' ?  f# [: g4 M" f6 P/ ?

% o9 W; o. R3 U) }! X' l        static FinderPointer CreateFinderInstance();
7 B5 p1 T7 C0 s0 L$ ~        struct FindDevice : std::unary_function< DevicePointer, bool >
/ K6 |5 M8 o) C# `4 i        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
  B6 @' x5 U! @9 B                result_type operator()(argument_type device) const
, C% A* _3 W4 \                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
  r. b# i' }$ _& u9 r3 q
8 t% ~* R- {( D+ ], \  u$ k; f
" |% \8 k/ v4 Z; O9 @                        return wcscmp( deviceName.m_str, m_udn ) == 0;
: K' R" m$ c; F$ m8 l                }
: O: J2 y8 s$ W7 X$ B                CComBSTR m_udn;
8 @( H4 E; F/ S; T        };
5 ]0 \! Y. n* {7 K/ O8 Q       
6 l9 f! M* {/ }7 C6 P4 h# y        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
6 F2 H" u/ U) U- F1 L; w& t        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
/ O$ R2 V8 X/ u/ @' o, q: V9 L% y8 V8 c        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
4 P0 n/ u* B$ r9 L        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
" M, ]2 Y2 [, v; s& {! E4 B                LPCTSTR pszInArgString, CString& strResult);
/ b0 W* M& f8 n" \( ?        void        StopUPnPService();


; g9 k! [/ y2 S* R' B3 E1 v; S        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
) A" G, d/ q/ j8 a$ A: C9 j8 g        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
3 m+ E2 L1 w! c) C7 e: u' G% y7 S# j        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
6 n' j6 E$ k, E# v* {! X        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
+ R* {- A7 J5 G3 K        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
/ [- \: D5 [5 v# H* k        CString        GetLocalRoutableIP(ServicePointer pService);
5 F" X8 Z7 D- O; [

// Private members
: B5 p) d- k% hprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
& C" m, x' S/ J  b3 K# p1 I        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
3 N+ L; P  Q6 T4 j! l" b# i

        LONG        m_nAsyncFindHandle;
- c8 n9 o' L5 }# E        bool        m_bCOM;
# }& {* c+ J6 l8 m7 _0 R5 P        bool        m_bPortIsFree;
* ~2 O: D* x- Z0 v        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
! K& `  B" M+ X6 V        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
5 A- }% Z# r% M2 _        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
/ Q- q  E, c  G" [        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;


};


* t3 u$ `* p. l8 h# m8 L// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
- o4 j( P% n5 J0 ]* a, [" e1 s{
/ F: S" a8 r# ^# b! i  [% Jpublic:
0 ^$ E; B. X, Y        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
& k1 |# m6 R8 k* S/ b8 Z, T- U2 {, g9 S6 w        { m_lRefCount = 0; }
4 ?) J3 {5 S8 C1 o
( Z; Q; t8 t' j
( ?4 b' i, n" ~" d  p' C! p   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
8 p$ C" W& @, o# X   STDMETHODIMP_(ULONG) Release();
. b7 Q' R1 K) M! c$ @4 p. r8 R
8 ~) h: p/ Q3 S3 i# W0 v
+ R3 |1 u9 J' Z4 G// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
5 b; {+ Q6 K2 g, ^" W6 y0 W. C: U        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);

7 ]6 A8 _& W& z) y/ b
private:
/ m1 A1 T. F- q  m4 c! E# a        CUPnPImplWinServ& m_instance;
: Y  x4 i/ x4 g2 b        LONG m_lRefCount;
};

9 T# E% L/ I! G; e3 `' Q  z! Q: Z
// Service Callback
class CServiceCallback
6 N7 F. O+ @; E8 L        : public IUPnPServiceCallback
% V6 S. Q4 f8 @{
( m3 N8 Z8 ~9 A6 U: @/ A. Spublic:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
1 h0 b, n3 W$ D$ k   
% j7 S" H3 r3 ^( z! S! D+ \   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
& \/ z7 @0 S6 \8 K   STDMETHODIMP_(ULONG) AddRef();
. k& v) H, N  h0 X1 ], g   STDMETHODIMP_(ULONG) Release();

9 ^  u" E3 I, Y' G
' \, U: X& W- C# l. D. [// implementation
! ^& m! J4 s, p3 r8 mprivate:
0 j5 t8 Z: l! c/ |        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
- d6 m4 W, \0 X" e- t( Z  ~! E
- Q: K8 S0 q7 i/ P' v( i& F
private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
4 x. n7 Q  y! Z: I4 |8 E};

- T! K& Y3 h# L! s
7 z$ m5 d' X9 ^4 B6 b/////////////////////////////////////////////////
& f' o9 X- `( G/ s/ {; B

6 b. h9 }# V1 p& ~- f1 D, h( r使用时只需要使用抽象类的接口。
; e5 [& {1 I9 X! yCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
' i6 s7 p& K- C6 @: _0 A1 SCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
5 s1 H! X# |# k0 O8 D/ i8 ACUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.