UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   / b$ Z4 {- M, e( [% t7 [
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   ; T2 r: W9 A! `& G& v5 ^
9. {
   ; x# I( H( c! {5 l2 X& x: ?
10. public:
    ) _3 K0 k' o& Z; L
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
      T+ U3 F& _+ s7 E& S# L
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    ; R2 U0 {  p% v7 l- V
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    7 f/ ^0 X2 C7 w: `. m; n
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    , y+ w/ @0 Q( Y+ o
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    9 K1 {* g2 @0 P) F. i
17. }   UPNPNAT_RETURN;
    3 h0 i& L& }# x6 H" h
18. 
      k! _& t# c' }6 b! J4 N$ @$ F: \- @
19. typedef   enum{
    " _4 w' i  W; R, k5 M% v! V3 v6 O
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    ! [3 V8 V! ]4 e6 n. g
23. 
    
24. typedef   struct{
    $ F; M1 _2 d* m) {
25. WORD   internalPort; //   Port   mapping   internal   port
    % v5 G; N) |  ]# G
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    0 b8 ]* y' X4 Q% f3 G. a
28. CString   description; //   Port   mapping   description
    9 Y- \9 h- Y' t/ l
29. }   UPNPNAT_MAPPING;
    
30. 
    6 \$ h( \7 e( Y
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    ( T6 D# o' P, P( N. \/ J
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    7 l* c5 y( c! |
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    ' s- J7 w  P$ f5 V1 ~
36. void   clearNATPortMapping();
    
37. 
    - K6 d2 D% e8 x: |, O4 L+ x
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    * ]- Y# V6 y# b8 E: V" [9 v
40. WORD GetLocalIP();
      j- k& \  ~; V
41. bool IsLANIP(WORD   nIP);
    0 a2 B, Q' s" O' D
42. 
    2 d4 P6 L& G% A" r. v. C
43. protected:
    / y6 G  l3 |* C4 J, I: t. S
44. void InitLocalIP();
    ! I2 ]. d& O) f, ^( N7 s4 t, y' S
45. void SetLastError(CString   error);
    9 A- _+ {- j2 ?- [
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    & y( s. t! Z* ^5 {2 i6 r$ A
52. 
    % J$ z1 r* R# `6 u: j- u
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
      ?" E0 S1 ?$ Z" d0 z, r
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    , u4 p4 J# g* o1 Z, G$ l
59. bool InternalSearch(int   version);
    * T1 o! g) Y& ?. P; ?  O
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    5 J3 a$ u+ w6 `! m- Z
63. CString m_baseurl;
    1 l9 ]8 S, ^9 A6 f, |
64. CString m_controlurl;
    " o' E* H, ^, V  k
65. CString m_friendlyname;
    9 ~/ X4 B9 Z7 z* N5 o
66. CString m_modelname;
    . _+ ~; O, x! H- x' c+ N" E8 }
67. int m_version;
    
68. 
    + n% Z3 W9 h3 {
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    4 C1 Q& C! ~5 e& E' N; i. z
72. CString m_slocalIP;
    " S' B( n+ n" C# i7 r7 ^& R  K7 i
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    1 b. r5 z3 T) z/ ^: L1 _
75. 
    
76. bool isSearched;
    
77. };
    / h  ~. [- d4 {% s
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   7 i; q# u  X! C
2. #include   "stdafx.h "
   0 l* E- X2 ?- H" [  l
3. 
   2 I* E. ^# W1 b' r& B% @6 T
4. #include   "upnp.h "
   2 c! ?# Q5 }8 ^7 M" D
5. 
     C( A% L/ L1 U7 Z9 a$ W3 q! J- V
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   2 g/ s( X; v! I6 M& ~
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    ! R/ K9 I' k% a0 n4 t: L
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
      t# \3 w: u# Y# [9 Q- Z
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    
17. {
    & e. i& m! `4 n
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    : j* Z# \  B, G9 R8 u. d2 k
21. 
    ' l5 P! Y+ q5 _: x
22. return   s;
    
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    % S& M  V; h  d* `8 o0 j2 c2 \
26. {
    # j4 P' }8 r9 |( D( @+ F4 ^
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    7 F% S$ _& I, ?3 l& r4 f, J
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    2 |, o& |" G. t- R: @% s8 [
36. {
    
37. char   buffer[10240];
    1 J$ g5 g  m  b4 i+ k0 @
38. 
    ) C' R0 `# K$ Y# O# F* F
39. const   CStringA   sa(request);
    - |3 Z+ {4 V1 N7 M
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    + f) o2 n# s5 h, E, V% A
42. 
    9 L; \+ Z$ `: h! \3 |" i
43. uint32   ip   =   inet_addr(CStringA(addr));
    2 z) @" I( h* k2 y. h
44. struct   sockaddr_in   sockaddr;
    - r" ?" D3 _- z* ^$ }+ C' d
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    ; M' W! O7 t, U) R; Z3 _
46. sockaddr.sin_family   =   AF_INET;
    4 \- O, g. T9 ^$ p
47. sockaddr.sin_port   =   htons(port);
    # V# g/ Z( p! L7 L4 i  G; `3 h
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);
    ) W  Y+ A; H3 d) C9 A  L2 H, e& b
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ! q2 ?7 P! E0 A# M$ C
53. Sleep(20);
    , O& H* d8 d: |
54. int   n   =   send(s,   buffer,   length,   0);
    * _, v2 b* D  S  i6 I) f" o4 u
55. Sleep(100);
    : ]: h1 p# c" a! q' i7 c0 h
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    3 p" V9 E9 o4 f9 Q& j
59. if   (!rlen)   return   false;
    
60. 
    , g+ W- Y* t5 X- u4 b- W9 n
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    / V2 W" W1 b$ S
63. return   true;
      z; g% w( g& f7 W/ N
64. }
    : E- z% n! s# W! z
65. 
    8 Z3 P* M+ s5 Y% X; Z! q
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    8 C+ b9 ~0 T' y# K* @7 N6 K9 a
69. 
    
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    0 K( x, n# H& ]( W  i2 U
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    . f1 ^3 @/ J) B* S9 T
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    8 T, _6 C% j! E- n' o) C* G
76. sockaddr.sin_family   =   AF_INET;
    0 F+ i1 A3 |: F
77. sockaddr.sin_port   =   htons(port);
    % s, T; S( t/ a8 g, @! h* {5 J
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    0 i0 e/ W- j7 k- h% a5 f, r$ M! d
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    - B3 p) V# ~5 x9 ^, |4 c! }3 j
82. 
    / _0 T( @5 }5 R+ W. h) _0 ^1 p( r
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    7 S$ S& s: b* I" I/ Y( a8 t
84. {
    ( Z2 e; U; }( w4 \. C. j$ J
85. int   pos   =   0;
    8 i0 F! }  V* T) ]4 V+ n( G( u
86. 
    + T  {1 h5 U# E) |
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    9 ~/ s* }/ h9 L' U, v3 D
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    , W+ U# s) X( U  M/ z6 L: a
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
      i" Z3 w2 Y$ K0 D% k* f) U
96. return   true;
    
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    ' f7 F3 J+ n3 ~
100. {
     / u) c' U) }+ c. t
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     8 }# w8 t0 g4 e* N1 T& f. {
104. 
     
105. int   posStart   =   all.Find(startTag);
     5 \+ h2 |& ?7 Z
106. if   (posStart <0)   return   CString();
     6 p! Q# J: D" {( w: t
107. 
     ! j& U: Q3 |% f% w- x
108. int   posEnd   =   all.Find(endTag,   posStart);
     ; ^3 M+ {- G& N' n
109. if   (posStart> =posEnd)   return   CString();
     ( a1 z" T' A& x
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     2 W7 I, s, o* y! S% `/ R
112. }
     ' Y8 U( o) \5 i9 u) N
113. 
     6 K/ f8 f' X' V" Q; u. A
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     8 o0 M5 H7 ?$ N# J/ \/ D& M' ?
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     9 m* X0 f- q1 P' s
120. 
     % _/ U$ u; C7 F. o, p% u
121. MyUPnP::~MyUPnP()
     6 _9 ]: m. g) b6 R2 i1 B3 f
122. {
     9 V0 T5 L' Y$ R" p6 K
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     / v* P2 i7 |! O2 ?6 ^, K
129. 
     6 N. ^3 t- t7 k0 d
130. m_Mappings.RemoveAll();
     # `  u5 R/ V. L' I) x: W" Z4 _: [
131. }
     
132. 
     " q5 J2 U2 n! I
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     8 J  Z; D  K2 O0 o
135. {
     , g2 ?! h: q1 F& ^! m; y  @" f
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     / [1 V$ p( ^! T% `
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     5 W/ a3 k: [/ h# z# Q0 \
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     . Y  i! {9 k# ]2 f0 ^% }0 I/ q) {
144. };
     $ F3 N" U% I' ~, h$ n
145. 
     $ k0 I$ j9 w- L. f- C* I1 E
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     - M, v/ @0 a6 ^
148. ioctlsocket(s,   FIONBIO,   &lv);
     ( ?: ]7 P! `% X2 D' a! E
149. 
     8 v4 j; z$ v- M' U. c; v6 R& ?* n
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     8 F3 h# E+ R- ]; e; q
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);
     
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. }
     3 r' D6 K4 G: `' A
160. }
     
161. 
     ) I/ X% l" F4 T6 {0 e! R5 n
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     ( P) z( m, g1 L" O
170. CString   result;
     * v* [) H$ m9 b% v; j# N  a7 p+ d# r
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     9 r: y8 w4 X/ q  \1 m' b
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)   {
       t; Q1 I1 z: B% }3 x( [+ v) \2 }
176. for   (int   pos   =   0;;)   {
     & u9 [8 ~" P( S2 x
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
       `9 x' \6 a- g- ^( t1 V
178. if   (line.IsEmpty())   return   false;
     " y8 M4 s9 p+ I5 c
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     + U& t# J/ h" h
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     1 A' t# P: o( B6 q5 e0 l, q
187. }
     
188. }
     
189. }
     
190. }
     2 i5 h! @& q' {0 |
191. closesocket(s);
     
192. 
     ) f3 P+ J' {  D% k5 u
193. return   false;
     
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
, U  k* Z$ p; `8 i
0 I5 v" u7 _" }9 n- Q. G) h
///////////////////////////////////////////
2 L" Z( n3 @5 K8 |) v//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


#pragma once
" B* s1 @4 o/ B) {, w: t. r#include <exception>

5 k9 \% ?9 c) j8 d  ~& v
  enum TRISTATE{
" E, Y% x* n6 r6 M  z) Q9 N( u        TRIS_FALSE,
        TRIS_UNKNOWN,
! W% f5 O4 h' y( j        TRIS_TRUE
% Y7 X/ e! |+ z9 u1 Q6 i};

- U1 A, G+ y! e6 B' s* b, S& ~& `
" l$ d( m2 ]: oenum UPNP_IMPLEMENTATION{
0 l$ Y6 e& {3 u4 T9 S2 C# k        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
, ]) _) r  ]0 n+ M        UPNP_IMPL_NONE /*last*/
$ A) y7 _( A, R};

; U1 k# \% g. l6 m
$ B1 U1 }+ i$ O

class CUPnPImpl
& ?& w. f$ U- J3 o7 {{
public:
        CUPnPImpl();
        virtual ~CUPnPImpl();
! n4 S$ J- b1 o2 R& q( [8 Q- W        struct UPnPError : std::exception {};
0 s6 D) m# O3 V- `# h        enum {
                UPNP_OK,
' z% m8 y- ]6 [! ~4 X, U# D* O                UPNP_FAILED,
) L, \7 l# ~* C5 x' M+ `3 G8 Q0 a6 L                UPNP_TIMEOUT
        };


. v5 W7 g! f9 N! T0 H4 d; g* K        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
7 F0 s6 e; S! O. ]! w5 x7 Y        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
$ x( @# D) c4 y* W4 T        virtual int                GetImplementationID() = 0;
( v, S, K! Z' \5 ]       
! V$ p  n/ T$ V0 [        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
1 c6 @; N4 A! T! p+ C1 @% f
9 l8 ^- j! G/ m& G# c
- D/ E0 k5 {( n5 z; U; n) ~1 _        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
2 r' o# o) k' r# M& w. C. w6 o$ U        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
2 h5 u: Q% y" ?3 L  E

// Implementation
protected:
" G0 d1 q! M4 e# L/ j/ t  e        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
; ~: ^3 o/ v. \1 F  Q        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
# }1 i+ W! J. z" n4 e5 w
" @, ?4 ?+ \9 H. d! Q
private:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
7 Y6 g$ ~7 W0 w

  l! ]+ L+ ~7 w0 x% M$ P' V};


// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
0 W/ t2 e- c5 \) K" a' t  D{
8 q& N! X/ Q& t$ n' H. l" i& L" Epublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
" H& P; d- W0 v" r) f1 ]        virtual void        DeletePorts()                                                                                        { }
& q1 I5 r9 ?6 s% U8 C. z2 g        virtual bool        IsReady()                                                                                                { return false; }
' o" X! |7 F5 W! u7 {        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
& b% S! y  E7 e% c+ l7 A+ l};
5 A; k$ _) m" B& W

/////////////////////////////////////
1 ~" j1 V$ j# ?+ l  X8 ]9 @//下面是使用windows操作系统自带的UPNP功能的子类
! a- n: |- E* u" a4 w# ^! A
( c) u, Z$ b4 K' }, C3 p& [
#pragma once
% i- E( n, P& r6 K#pragma warning( disable: 4355 )
4 S- b; f+ T$ T  t) D, G! M

#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
- j9 m$ O0 c' A/ U8 N#include <comdef.h>
#include <winsvc.h>

8 a" |2 i* |8 Y, n) e  c3 D
7 y; T/ T4 }( e1 e) j#include <vector>
#include <exception>
" \2 g5 S  b5 `2 G3 m/ ~#include <functional>
- S3 U% m( H4 G5 n( `

9 [/ }" A+ P5 Q! F2 o6 p+ B6 g
- h% Z/ O* U6 ^- `0 @
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
; M# D+ X6 \6 Q! }4 l# d* Qtypedef _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;
0 p+ i5 ^) m- V: R2 f+ \* btypedef        _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;

, z! `! Z1 w: ?; H# w+ ^
/ n. T% J, Q# G! [$ O. S; wtypedef DWORD (WINAPI* TGetBestInterface) (
7 l$ i$ I  t, X' \% Q+ k5 E$ b  V  IPAddr dwDestAddr,
- ?  c% B( U& F) d# b  PDWORD pdwBestIfIndex
);
$ L6 }  c3 B2 X

typedef DWORD (WINAPI* TGetIpAddrTable) (
  @6 t8 C; v% D! c. V  PMIB_IPADDRTABLE pIpAddrTable,
8 ?8 A3 T* m- z  m  PULONG pdwSize,
7 x* Y% G" d" X  t: v1 ~( [3 L" m  BOOL bOrder
! J/ e6 q8 x2 Z0 ]( Q8 @( P);

% G2 I1 d4 ]7 m! [. b) k& D
typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);
5 m# X7 o) p& G; d$ y* F& @
: k) g* J( m# _$ A6 Q, m% C* m
CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
5 S. e% a# n+ s

' `3 j9 }* f5 K7 [' Q* L" `" Vclass CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
9 ?$ ]3 [0 ~% F* L, j        friend class CServiceCallback;
$ x; S2 l! N1 N. t0 A) x/ h1 M$ [// Construction
( x: r" v' A& `' ]" b8 I! Npublic:
3 a0 Z4 i' o6 d* b        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
8 a1 x9 q! V2 W) r4 N# m

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
) T( t6 V: S2 [& n) M' w$ F        virtual void        DeletePorts();
# y9 O. ~* ~) b' d8 c% T# I        virtual bool        IsReady();
        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)
        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };

3 z) f6 @2 i; V# K' _' D7 n
- G1 K- k) z; n) X3 ]3 n6 ]protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
1 A2 C) P2 }# y4 R' o        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
2 Q2 _0 h* F$ I$ a        void        RemoveDevice(CComBSTR bsUDN);
( H! O0 O2 n; N0 n        bool        OnSearchComplete();
        void        Init();
, l% m+ h4 A1 r  |
# e* S$ d. C; Q8 ?; r) E
        inline bool IsAsyncFindRunning()
2 Z+ h% |" ^* E1 H% k7 M7 G        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
) J  S, A  e0 w5 t3 \                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
  ]0 P, l) ]& G! f1 ]0 [& S                        m_bAsyncFindRunning = false;
0 l$ Q2 S/ Z* k& S2 c; m                }
                MSG msg;
! h$ q! S: r  z/ _! V0 }& P                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
3 E0 [% E6 A; l  G7 C                        TranslateMessage( &msg );
8 q! S$ q' w1 Y* T! C( M+ a# t                        DispatchMessage( &msg );
- h3 A$ S, ], B+ m, X) ]8 g( o                }
" C% p3 p) d7 X                return m_bAsyncFindRunning;
2 \6 x, \( k# P1 Y7 f        }


& z, `2 l& K) L  k' R" b        TRISTATE                        m_bUPnPDeviceConnected;
( p) e. o+ B. t+ `# S7 N& B
" |4 _/ t0 @1 g, M# m
// Implementation
) Q7 B% L  w$ ?4 ?  W( t: H  Y7 @2 p        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
( P0 v8 b$ [) X6 q        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*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
0 n. z) s6 O1 s+ e- |' U

* K$ @$ `+ B5 S" M- W  Z6 I4 Q8 m        TGetBestInterface                m_pfGetBestInterface;
& g# |& P# K2 r$ B        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
) G" n) J* t3 g* @6 s' L# X

* M; `! I8 J# b  o7 w9 u        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
4 D8 n+ C9 J% F' _; l4 A! b* {
8 T0 z% O) Y. H2 }
$ J, q" G; O7 D8 m  D7 a* Q  d                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;

! ?2 b  Q8 R3 C8 Q/ `* J
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
6 W1 z9 n* _, r+ F8 d$ z" n                }
$ z, \- l% y' G1 @4 Q4 U- ?- ^                CComBSTR m_udn;
5 Q, j3 b+ i. X5 r        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
3 M- m2 b: j" o) }. J  ?! R        HRESULT        GetDeviceServices(DevicePointer pDevice);
! x7 Q: E' Q2 W& D' Z, [" T        void        StartPortMapping();
2 B5 l5 y6 }( j0 O" p2 P8 e- m        HRESULT        MapPort(const ServicePointer& service);
4 ~" A5 T, Q. ^$ X& i        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
4 E0 E3 i1 K1 b& d& O  v3 D        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
) {2 K1 N# `4 m6 B* s                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();


        // Utility functions
( Z# L0 @+ h* [& _# h6 i0 P        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
$ K( ]; y0 L+ h2 g4 q        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
, I/ ^8 T+ {1 y0 p% N' I        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
9 I3 j. x( y' ?8 K        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
2 u" ?7 _* N8 i- U' }        CString        GetLocalRoutableIP(ServicePointer pService);
6 K  ?5 `; F5 a9 a& |4 x. N+ C
# ^; a/ R6 y. q' v* k
/ a$ D0 {3 ^" T7 }/ v// Private members
' ?7 r& L6 k# i; m0 a$ vprivate:
5 g8 f8 N+ w! B/ x: |        DWORD        m_tLastEvent;        // When the last event was received?
. K4 Z0 \4 v+ K        std::vector< DevicePointer >  m_pDevices;
" u6 h# E- `* p        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
  X$ B8 h5 J, Q8 J2 E+ C        DeviceFinderCallback        m_pDeviceFinderCallback;
9 x. N8 A' r! O" z3 O        ServiceCallback                        m_pServiceCallback;
% D; g- u! [! a: m& s

# o; E5 M  u/ G& W        LONG        m_nAsyncFindHandle;
! S$ A: w2 t8 i5 I- c5 n/ `        bool        m_bCOM;
+ I4 x# b: T2 ^  W/ H9 ^) Z' Z        bool        m_bPortIsFree;
' U% y' Z- `+ w/ e' J6 L' ]        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;
% ?/ Z) G' P; u7 z1 J        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
: ^+ ~% q' V& D; U8 \        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
2 E# S3 g3 R& K7 R/ M5 o

};
; h7 }8 `6 ^1 b# Z, Q
$ v' ]- o" y. o2 b( Y- o- ^
, n7 G! _: A" ^* P' Z3 g8 r/ _// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
8 x# ]! D* f0 U, O* G6 Dpublic:
" ^  k! B) c& m        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
& m/ T5 G* i: L! ^/ s4 [

; D% }2 n- q* f" i" H- ]6 |   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
# d; N) R; i' j) [   STDMETHODIMP_(ULONG) Release();

% C+ \( }: [7 ~( w+ T" T6 c' E6 w. \
2 w4 z! s* `* W$ @0 y) n: G# s// implementation
private:
+ b! D! |& I: t+ p5 c' i        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);


* \* i% l  b0 @/ o1 X& I, D# Fprivate:
6 {' O& l3 Q6 |6 N2 p  Y        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
% v& X# [% j5 \2 j6 R. t% N7 L};

8 |& N, Z4 T6 Y9 h$ S" J" ]
+ S; h- {+ E# ]3 P6 Y// Service Callback
) r4 p( ?& d* B% {0 J& n9 {class CServiceCallback
        : public IUPnPServiceCallback
: p  A. B- [; t% x8 J{
9 ^! w! X) M& ~/ f* U6 x; Z2 \public:
  M* p. D5 V5 P        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
# n. e; O, S5 t        { m_lRefCount = 0; }
" z; A* c+ v- j( u+ j- m   
$ H% Y* z' Z: \/ G   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
7 U, S* ?4 K# d: Z2 D- A  [   STDMETHODIMP_(ULONG) AddRef();
; W; q9 S" v4 w   STDMETHODIMP_(ULONG) Release();

1 Z' L9 @$ F$ h6 P, v1 G! S
// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

- e% }+ X/ p  s' Y) o
; M/ @. v. |9 ~) f6 q1 n2 pprivate:
5 B- c4 L' \8 \) Z9 h        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
; v9 W7 q3 E6 b* n+ K" ^$ n  |5 r};


/ v; s. X  b* i0 M8 F  p$ \2 {/////////////////////////////////////////////////
$ Y8 R0 Q+ Q- t/ B

6 X4 _( ?* s- ^2 B; b! {" x/ ?0 b使用时只需要使用抽象类的接口。
: b' f+ g( `9 bCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
! T/ P! T7 Y9 {$ `1 RCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.