UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   + E5 O1 x" l* S, ]  P" d% V# n* H
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
     r; p  k- I/ V: ?& a9 C
8. class   MyUPnP
   ( Q- P/ h/ m% m; @! G  a( V8 ?3 k
9. {
   6 T3 S" Y' M) a' k
10. public:
    ' X/ k' ]5 e, T
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    ) @- C9 h- C1 [1 m1 }
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
    - x5 }# m- Z3 x3 v& o
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    4 u" B) C8 U; {' Y8 q6 g  [" f
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
    
21. UNAT_UDP //   UDP   Protocol
    1 [* J* g7 n2 H; A
22. }   UPNPNAT_PROTOCOL;
    ( Y; d6 M4 ~# P
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    . L/ A- t! f- G' G# Z0 ^- H: l
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    ' @7 z& x; W. p1 O" p+ H  A
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    ' \, ]( v5 b+ D7 ~+ [  \2 p
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    , S5 s6 o. q+ N' J/ r
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    2 M5 L  W( p3 ?) i$ `. Q
36. void   clearNATPortMapping();
    : p- ]  F' i5 n3 ~$ b) |9 i  `
37. 
    
38. CString GetLastError();
    " I' Q! S8 E: g! D/ p' p
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    3 u5 B: E0 d) d" v& I. q
42. 
      E; ~0 ~* M9 z: ~# m& @
43. protected:
    : S5 p. ?3 Q4 E# ]" `# a
44. void InitLocalIP();
    . o- ^9 k' F& z# C
45. void SetLastError(CString   error);
    5 Q" I9 f' Z) E+ E! U( c! F, s; C% y
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    ) R, J+ M1 o7 I8 A" k" m+ ^
49. bool   deletePortmap(int   eport,   const   CString&   type);
    7 {& m( Z" J7 u" y' ?: c' t7 s
50. 
    : G. K/ a5 T9 X( O* f4 o, f
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    & I% f6 I' M3 w  l" c, J
52. 
    
53. bool Search(int   version=1);
    + d7 A0 p& p, o& K
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    * v7 R5 j! r, U. E8 j" o
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    4 n# o7 p: x+ e& F* ^' V
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    / @8 W# e/ V# f9 b% ]* ?( S7 _
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
    7 Y7 e4 q! U& L
62. CString m_description;
    , ~. f% R$ p# e7 |% r) c9 }+ \
63. CString m_baseurl;
    
64. CString m_controlurl;
    . I3 u) W. R1 Y( A7 e! R3 ]9 w8 ?
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    
68. 
    2 P% z$ D4 J7 e. Z
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    9 C) I" i2 o$ s8 x! O/ r
75. 
    * G$ g5 A9 b9 c0 g4 o# Q7 I
76. bool isSearched;
    
77. };
    . C$ K# }# U5 [) _$ F- `( }% U+ A
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   " [: `$ ^% R& k( O
2. #include   "stdafx.h "
   
3. 
   , O+ V. f5 s' n5 S! A6 R+ Y  d
4. #include   "upnp.h "
   & }. r) b! K/ [5 P% S5 ]4 q  g
5. 
   4 n9 V: d" f" }, q& i7 {
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   ! R9 A, d  k1 r
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   / [& r' k/ V* j
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   2 m! ^8 m# O) X# _
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    4 Q) M8 B5 k# M8 w" ~
11. 
    , m% E, i! F( @- L
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    ! s: ?) `' J& w# H* Y: h  Z% L7 _
17. {
    & ^' g+ H* W% d, r" h
18. CString   s;
    
19. 
    8 F3 G" V2 k/ {/ X6 ^; z! Z! X; V
20. s.Format(_T( "%d "),   i);
    
21. 
    / J7 ]7 Z5 s  K1 p: _! l8 o
22. return   s;
    
23. }
    
24. 
    5 l7 |# Q% u% }" n/ w2 b7 l
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    2 m  O4 ?, F+ Q0 B' ?
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    / V' W9 k9 F8 k- g6 y, p) E1 n
30. const   CString   GetArgString(const   CString&   name,   int   value)
    9 i2 x' \7 s% x1 y) H. G, g3 p
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    4 [1 ?- Z, L% a; V) Z4 e; u
33. }
    
34. 
    0 t+ A2 W: C# _% ?5 I% q& U3 V* I
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    3 ~6 d4 U: U. H0 Q, D1 E. F. N
36. {
    
37. char   buffer[10240];
    8 k( @* }  i- n5 K; h
38. 
    7 b  F3 L9 }6 \% c  W
39. const   CStringA   sa(request);
    ' }3 L6 [& ]1 y& I# ~- z5 N
40. int   length   =   sa.GetLength();
    ' @' o. |1 a( V' p& O
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    ( O$ p) w  A2 l: f& [# c' W
43. uint32   ip   =   inet_addr(CStringA(addr));
    4 I2 R- H. W' Q! \" r( j
44. struct   sockaddr_in   sockaddr;
    6 p% R+ J! |3 S: ]6 p, I" z9 Z' u% {
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    , h1 O1 E5 i9 O+ ]/ U
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
      Q/ m2 H3 a- d6 V. o1 r- |3 C, t
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    1 g$ o7 ~$ E$ @0 a: L" c+ o$ b7 s( |
53. Sleep(20);
    8 L- W2 r7 M/ z9 B
54. int   n   =   send(s,   buffer,   length,   0);
      R% |4 g& V. N$ l% ~
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    8 I0 U* F+ m) ?4 ?) y1 o: o$ K5 C& A; T
57. closesocket(s);
    9 g9 `4 G4 ]5 A3 ]* Z
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    : v1 s+ q9 H/ g
59. if   (!rlen)   return   false;
    * @% C' g# u6 }9 O3 v
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    4 C$ N  j1 k! d1 @9 g
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    7 \, g9 C, h7 ~  z8 `. M2 {6 [
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    9 K- S* R! x! D) ^/ A  C7 R
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    " {# P/ Y+ }3 h. {, N8 ~
73. 
    ! l- L; n8 }" O- {2 N; \$ [* `
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    % s7 T4 D2 g5 m) q8 e3 N. M5 F; m
76. sockaddr.sin_family   =   AF_INET;
    # P9 R! t/ U4 U4 t
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    & t& S3 N5 B0 \( f+ D6 r  u+ j6 y2 |
79. 
    4 y* r1 P7 [$ ^& `7 {. A3 j8 I
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    5 I5 i5 P6 \4 j) |
81. }
    ; q* Y. L- r1 W6 U7 f, @4 ^
82. 
    - V. c- C2 n! x# }6 d9 S6 \2 k# E; B
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    7 B4 J+ R9 J6 A
84. {
    7 ]7 E& d/ \1 j5 M
85. int   pos   =   0;
    
86. 
    ' B0 r2 W2 M2 ]
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    6 ?, J, J% d) W  ?5 D2 K
89. result   =   response;
    6 `! ]" t  Y: \) G7 C' K2 l4 ~
90. result.Delete(0,   pos);
    ) K6 ]2 R1 Z; [& J
91. 
    
92. pos   =   0;
    ) Z# _/ n5 c6 U7 h/ h) U# T# ^; ?+ J
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    & V& f: p6 K: Q8 Q3 S
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
      @0 P0 S) T9 U* x6 r; P8 j
100. {
     8 J7 c2 \2 A% S1 y$ h
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     3 U9 [& e3 B/ B7 v5 K5 f
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     7 `6 G$ L4 P8 ^* |
108. int   posEnd   =   all.Find(endTag,   posStart);
     3 H8 G) J8 d" m% E4 ^: l
109. if   (posStart> =posEnd)   return   CString();
     5 A! w7 H. y" N0 i
110. 
     6 Q0 v! p% `5 o" x8 D0 O
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     / {4 m$ w' ~8 I2 K1 Y+ E9 d
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     # Z  l+ K0 t: O# \8 X; {
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     : m. B3 n7 L( i8 T. n! t
123. UPNPNAT_MAPPING   search;
     / g  k6 z$ T1 [' @8 }3 Y
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     % x( V% j& w+ K' w- \3 c& p
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     ' N% G: N  N; w/ q6 a/ i
127. RemoveNATPortMapping(search,   false);
     + K% @* Q: [% l$ O  E
128. }
     2 k; j2 o4 {/ f
129. 
     8 b5 ?' R/ U3 D; i0 @
130. m_Mappings.RemoveAll();
     . Q; j5 w4 p+ ?6 {: t
131. }
     
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     ! Y( H: M* ^+ f
135. {
     $ M, \! G' G& l- E/ T/ [4 q. r
136. if(version <=0)version   =   1;
     7 y7 z, h& l8 y  O% x" V/ y
137. m_version   =   version;
     
138. 
     0 S) Y# L: T5 j: P( m
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     - t: T3 J* I9 E8 M- d3 j( g
141. {UPNPPORTMAP1,   _T( "service ")},
     6 W: l$ u1 }+ R# M. {7 F/ q
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     & y; `" ~# C2 e
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     6 v, v0 M) |; i. J  g# ^- U, {
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);
     # d( K1 y+ Q) q$ V" x) C
155. CString   request;
     # b9 D  n3 _# r, N
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 "),
     2 g, m* f" i, U1 d9 @/ T
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     - x7 u0 B' I4 W( L6 V  i1 [2 ?
159. }
     ( k7 G/ t+ {1 ~$ R; g
160. }
     " C; _, Z4 d" W) {& _
161. 
     
162. Sleep(10);
     
163. 
     , v( l+ N" @- ?8 S. s: q
164. char   buffer[10240];
     $ z+ q) F" b+ }& @' n
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     , }* c" ?8 n% C, J
166. if   (rlen   <=   0)   continue;
     : ?4 l7 n0 p/ M
167. closesocket(s);
     
168. 
     $ t2 e3 I8 f5 b0 B
169. CString   response   =   CString(CStringA(buffer,   rlen));
     ' |) [9 Z3 [( b
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
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);
     : S7 ?: g9 _( }
175. if   (result.Find(m_name)   > =   0)   {
     9 U1 O' [: a2 Y" Z
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
       I1 _+ p3 M  P* X# X' K1 N: \
179. CString   name   =   line.Mid(0,   9);
     3 r  H: g, p# q" {) I4 U& D
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     ' [7 S9 q  N4 k
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
       b* s. ]( X" D
186. }
     
187. }
     
188. }
     
189. }
     / p5 _3 g/ d! h
190. }
     . V, v' ~7 L( j1 Q
191. closesocket(s);
     ! v# \& i& R8 Q5 m; A
192. 
     1 R9 O' Z( \) m
193. return   false;
     
194. }
     5 I9 M0 L5 K* T: s5 X4 U& ]! v

复制代码

天云上人

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


# w& T# \1 q3 H( w///////////////////////////////////////////
/ M0 C7 f) j" \: C# x4 w5 b' X//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
$ I% A! H: C2 O1 s
  \4 ?/ T* Z& S, J2 g0 J8 _+ L
% h5 Y' W7 [5 [; n+ @#pragma once
#include <exception>
: |) D+ C' m5 g3 T$ e, r" F

  enum TRISTATE{
) [/ B# s+ j. ?& t# n  P        TRIS_FALSE,
7 A$ X3 J& M) p9 C  L        TRIS_UNKNOWN,
- O+ N" B  K/ R& V# x+ D        TRIS_TRUE
};
' W* J7 y* W& L. J# l9 f
0 z$ e: R) c, j& X
enum UPNP_IMPLEMENTATION{
$ M, H5 X  x1 M( U1 m        UPNP_IMPL_WINDOWSERVICE = 0,
3 `" l4 [9 e7 v% a. I; _        UPNP_IMPL_MINIUPNPLIB,
+ H* c8 l5 L& S" W: w( c% P        UPNP_IMPL_NONE /*last*/
};

! i- R# g& R( F, \
. c6 H6 A: p' Q+ d! V0 w

class CUPnPImpl
, g2 d1 k2 L, B8 v& \{
public:
- F% H3 E5 d( D4 ^        CUPnPImpl();
  J4 m- N# X3 ]% p2 x0 m! f        virtual ~CUPnPImpl();
* G/ O5 D& J( k& k9 A        struct UPnPError : std::exception {};
6 K: D4 T, W. c# w& R        enum {
                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
; H" U! X) I3 B$ Y        };
' L! {# T$ m! x. l# t, t- n

9 a! u2 Q, k: S        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
( t- u0 r. U  T. p& A- O* t' z        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
% D6 j0 I  a" A8 f        virtual bool        IsReady() = 0;
' t3 r9 R8 H2 |! u# f        virtual int                GetImplementationID() = 0;
' K% H8 c6 X3 l. m       
! h/ \' ^% \: m, s5 D        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

3 t1 ^& n0 n3 i& P+ E
" F" t' D3 n& R. @+ ]8 T! r8 L4 d        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
" @8 X% \# Y7 T/ N6 w        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
! Z+ S, o+ Q2 M6 u4 t( \; v" e

// Implementation
0 Z3 L5 |/ i0 u) V: k) C1 T/ ?protected:
* @) I) s$ M8 G  i$ z  S' t        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
' e0 _8 f' M' W& _6 w; a* r        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;


private:
4 e+ q) Q7 {3 y$ u        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
' O& t! C( l+ a" ^, P; Z) o
: A0 j  E8 p4 q. g& ~, t2 f
};


// Dummy Implementation to be used when no other implementation is available
. U1 M2 |& {. k3 Mclass CUPnPImplNone: public CUPnPImpl
{
1 Q6 q, G, M6 E9 |$ u' }+ ]$ Zpublic:
6 O' o; q  O" U* o/ r# j        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
' @' [4 b3 z: w4 E        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
0 s) q' l0 i% {+ C1 l        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
! J8 `) L0 ^$ n$ Z        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
, o1 s' O7 @$ b  c' K  \$ C# e};
* J- w# F9 H, t& }- N$ A5 n
& ?8 h2 H5 w( z; L5 R" u
/////////////////////////////////////
' ^+ u9 P7 _: I: }. H# L//下面是使用windows操作系统自带的UPNP功能的子类
7 b: o3 W/ m) N% g- a
* I' N: s8 }8 Y
#pragma once
' h( {$ s6 S) N" @6 Q#pragma warning( disable: 4355 )
9 D0 ?5 {* `) i" R: W8 v
: l+ D& n! b9 {& w
#include "UPnPImpl.h"
#include <upnp.h>
  ~* T+ _- S) ~& E#include <iphlpapi.h>
#include <comdef.h>
9 V9 N" a9 T% j# }8 U#include <winsvc.h>
5 B  i$ h# S5 a2 s- R; w  y

; _$ L. h% n5 l7 A#include <vector>
#include <exception>
9 o8 G5 u# t2 c) W#include <functional>




typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
7 i: t" `+ q/ E. P9 I3 z' s5 {typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
* P* ~: O9 [" f) jtypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
, u+ X8 L% J5 n+ b5 n) Atypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

2 q  {& D) _* y5 X; M
typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
9 {  K- u7 C! g1 F& H  PDWORD pdwBestIfIndex
, H  }# H; r4 S( o. `5 k);

  g& j3 F, J  u! V" w
2 _, x1 H6 C, q- m' g+ t9 itypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
. P/ E& M$ a( g1 t);
  e# f+ h9 w( K' w) a! w

  ~1 i& b' e8 `) S& ^typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);

* N# b2 a* C; d) r( i( w) ~( i
5 h4 ?. S$ |" {: K# V  t# ~* JCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
& A4 L1 s# W3 A2 C0 ~- s) F7 [
- F! b, d4 M( [3 W3 u6 i
class CUPnPImplWinServ: public CUPnPImpl
{
) g) l9 k- g" X; p8 @2 T$ k        friend class CDeviceFinderCallback;
! o8 F" a4 d9 `5 v  a- z" L        friend class CServiceCallback;
// Construction
0 p' R+ l: r8 ?2 E$ ~& G( dpublic:
7 e/ o- F7 {! ]        virtual ~CUPnPImplWinServ();
  F& z& s8 x0 ^        CUPnPImplWinServ();

5 q1 B5 d" B7 ]  \9 C! W
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
# ]; m7 a  o* P: k        virtual void        StopAsyncFind();
+ _- `. e# Q, @+ T& [/ B* J" [) \        virtual void        DeletePorts();
( @  H' Q8 i  V        virtual bool        IsReady();
) @& L$ Z2 j( }+ k5 N        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
0 c. a9 r2 ?1 w  @: p        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
  h$ F$ E) q* h, E8 D* D2 _
' \$ @. d" ~  D/ B9 D9 ?6 W2 ^% K
7 M' Q5 M( B7 xprotected:
2 s8 q9 D3 D" k. j# A4 |        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
8 b  t/ N' J- L6 H        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
5 ]6 Y2 H2 j1 i        void        RemoveDevice(CComBSTR bsUDN);
" |8 I+ e. G& I6 Q+ w8 R: x4 {6 o        bool        OnSearchComplete();
, L, W" O8 i' \3 t: s        void        Init();


8 P& F3 a$ h" K! D        inline bool IsAsyncFindRunning()
* q2 B$ t( `  i2 j9 c# ~        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
                MSG msg;
* J# ?5 \* @* |7 p+ D+ O                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
7 C% Z+ J4 ~" q                        TranslateMessage( &msg );
6 U, }+ c2 O' f4 G9 ~- d3 A! ^                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
, V8 @" @% Y5 c. t9 z& j        }
( `7 `/ }2 [8 `+ v! R7 |  Q
4 N" S3 ?1 v7 y; G) d
: Y% V4 V8 j& y: S# D( t; v        TRISTATE                        m_bUPnPDeviceConnected;
4 U6 s7 O: Z4 H- `5 ]3 P5 V

// Implementation
3 F7 V  g, a3 R# l: Y        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
2 g! r# \# h. g        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
: ?; V, {- L/ ^  L$ i        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


        TGetBestInterface                m_pfGetBestInterface;
4 p& L' w* t- k& F        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;


7 q6 U4 `" Y5 w) A8 J8 Z        static FinderPointer CreateFinderInstance();
% j0 m4 l1 a* e2 O# t4 M        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
* H. X, [8 j5 S                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
7 o8 u$ [* }2 r9 P4 C                {
                        CComBSTR deviceName;
: z1 i; X- |/ p! l2 q                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


1 h$ i+ E! T# J! X8 \                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
  d! D( u( P5 S! A
) |9 p' v: e+ z) k* @$ ]
; [( r0 F. y6 P$ k% P2 x& ]                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
+ T+ T1 D0 e5 s, O$ X2 i/ v, }                CComBSTR m_udn;
2 E8 ~9 J# h  e  f( j        };
4 T, u  d) Q0 b+ @& }. ?) R       
/ Z6 l* O4 |! q! U& G( s: S5 V        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
! o. f3 ^9 e7 o. E. W+ i        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
& ~' g; A( k( C        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();


        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
1 |. |. _5 x' {4 U* v        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
; u; h$ L2 F. X2 X. @        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
3 Q0 `7 J  c6 \2 l! C! r        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
* J4 `' H- |- Z8 y% h' S$ H% ^

) d& P+ {, C' V, [// Private members
! C# J% l, y, M* e" yprivate:
+ e+ R3 N) E, Z0 N# |6 ], ^: m        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
! Q1 h8 }; J3 Q4 `        FinderPointer                        m_pDeviceFinder;
* X( i  I; B+ f: V; T        DeviceFinderCallback        m_pDeviceFinderCallback;
9 Y6 E7 O, h: B6 P6 ~/ Z+ Z        ServiceCallback                        m_pServiceCallback;
) @* R! s9 E; v" b- P

% D$ U# l/ e9 K        LONG        m_nAsyncFindHandle;
2 k" r2 N  D; l9 G) T; U$ S+ l        bool        m_bCOM;
* Z. R& o; ?' f  ^3 [        bool        m_bPortIsFree;
        CString m_sLocalIP;
( w5 f9 e( h  ]8 ^        CString m_sExternalIP;
* T3 b' z9 z- g( L1 H/ f        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
  ]* b! k6 R( m. k& g        bool        m_bInited;
        bool        m_bAsyncFindRunning;
8 d6 E& M) g: D" A7 v, i" f  V        HMODULE m_hADVAPI32_DLL;
4 S; A2 j3 @: g" ?1 _* v! [        HMODULE        m_hIPHLPAPI_DLL;
/ G% W9 I  l: T) u5 [4 x  k        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;


  x& D+ f. `$ ?};


. n' J$ z. w7 t4 v( x// DeviceFinder Callback
. v+ {: z* q, yclass CDeviceFinderCallback
, I$ G8 A( |% C9 |2 S        : public IUPnPDeviceFinderCallback
{
public:
* @' e$ C/ i% T: q8 b; Z        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
8 l1 L$ H  q6 X5 j        { m_lRefCount = 0; }
: P9 E6 Q6 K& i9 N: M) E: @' ~
' V9 O' J- c# ]* a7 @' o
8 p7 H$ j( L+ M: f/ K. R; k5 n   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
/ d6 x8 j' X% w& n9 p+ B+ |% p   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
$ F1 i% C( z9 {$ r2 Q  V

- W$ _$ O; [6 i' x* r// implementation
" m5 t9 V! d/ aprivate:
6 A5 r9 P3 i& ?7 X        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
' d8 j9 ]- i! S9 p' u* s- W        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);


private:
  H( C: y+ D* I2 U$ ?7 h6 M        CUPnPImplWinServ& m_instance;
1 `0 \( E: Z/ u7 z        LONG m_lRefCount;
$ f: y& Y- t  W1 U};


3 s  Z' ?2 o% j" y  O8 M// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
0 C6 G. A- ]3 O* s   STDMETHODIMP_(ULONG) Release();
. e' E- @9 z* `6 I8 D

// implementation
, @5 ^# Z  J8 q- z" g: [private:
; l5 a& {: p! y0 y6 v8 f. q        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
  g4 I: {, N! J. ~& M

  j  t7 X' R3 W! Rprivate:
        CUPnPImplWinServ& m_instance;
8 x/ s/ {& L: P' e: D        LONG m_lRefCount;
& m- a$ o+ o+ E8 [/ C9 v};

) {# H6 k1 B- f6 B$ N
/////////////////////////////////////////////////
* [1 q+ V, Y3 `' u7 U

使用时只需要使用抽象类的接口。
+ `( \- u# C; a+ b, @0 {0 ^1 bCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
$ p: ]3 i! U( w) I# x) tCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
. x. Q9 a) p0 M' x1 RCUPnPImpl::StopAsyncFind停止设备查找.
# ~( x( t& ]  W0 U$ _5 FCUPnPImpl::DeletePorts删除端口映射.