UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   # F) y/ o4 g; g, k5 y3 f# ]
3. 
   
4. #pragma   once
   0 ]3 I( U, n) I: F- ^+ d- b
5. 
   : o1 e8 E- T) s4 a" ~2 l; W
6. typedef   unsigned   long   ulong;
   3 a8 P8 h/ }& |, V& C
7. 
   
8. class   MyUPnP
   * P* q+ u! T# Y4 [" L* i0 t9 i7 b/ C
9. {
     ~* F( b' e$ z' I# X0 r) k; @
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    6 ^3 D; j7 [6 q0 `$ U
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    0 X% W+ {3 J& X
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    3 X8 A7 T* a5 B% ]5 j0 Q- |
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    ( Y: t, R! k6 O) ~: ]" `0 |
18. 
    & o! K0 ~$ h7 t( _! C! q3 q
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    ! n1 m  L/ m" |  ^& a
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    7 b4 i- z6 W4 X2 A) s+ m
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    ; u5 M, h& x' W% j
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
      B2 y2 w% ^3 O1 a8 |
31. MyUPnP();
    5 x( K* k8 Z6 G" V1 @0 |+ C
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    5 C" j7 F6 W. a9 }# V4 Z
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    ( N. J& A1 \9 H: B  k7 K
36. void   clearNATPortMapping();
    ! c/ f  h1 B9 x4 p/ m9 N4 @, X
37. 
    ' g8 F7 b  v. a4 ?. |( ]3 `
38. CString GetLastError();
    ; n/ W5 @6 G) t9 _
39. CString GetLocalIPStr();
    + O; K3 O9 n/ J3 A+ ?# a
40. WORD GetLocalIP();
    ( ^7 d0 p' p7 b! D5 i
41. bool IsLANIP(WORD   nIP);
    # S$ u: e- a. \# s
42. 
    
43. protected:
    + G3 u0 K2 }) f+ t# F2 A+ V
44. void InitLocalIP();
    , p8 u& K' q! R: z6 M& K
45. void SetLastError(CString   error);
    $ `3 K" c3 ]5 k1 ?" Y. U4 R6 x
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    1 C% y, S* ]2 f: d8 C. r- S% ?
49. bool   deletePortmap(int   eport,   const   CString&   type);
    , H: b6 r! k- q# [# a: V
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    7 w0 h  Y& S& y# y5 b1 [
52. 
    , H9 |0 P$ k( [( G% \+ s. M
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    ! \" e4 }) P% E
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    4 `/ u! L5 j) {4 J
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
      P  Z% u* F, z7 W1 P
59. bool InternalSearch(int   version);
    8 z8 q4 S1 R, g- u9 s5 n  a4 J
60. CString m_devicename;
    
61. CString m_name;
    % C' Q( ^3 ]1 D( |1 B( g2 Y  K
62. CString m_description;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
    . a0 f6 ~2 k: m$ G' m5 `
65. CString m_friendlyname;
    
66. CString m_modelname;
    3 b# `, O- R' O2 F/ J
67. int m_version;
    
68. 
    8 e* A. t4 a5 u& B7 S- ^5 j
69. private:
    . Y1 r9 g1 Y5 m0 b) O$ }
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    - n2 _# l" r' o
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    2 k: H, b! K, e" c) m7 M
74. WORD m_uLocalIP;
    8 k9 c* ^% O$ M( N& ~
75. 
    - c$ }4 D, R/ e  S
76. bool isSearched;
    2 \  n0 y8 d/ c
77. };
    ! U7 i3 r$ [2 }# _
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   5 G% B- o8 u4 d* ^) B  c
2. #include   "stdafx.h "
   1 W: G: y: x+ B* \4 |5 H2 N5 t
3. 
   
4. #include   "upnp.h "
   2 {+ }1 }- W4 Q/ {; ?
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   : E* ?2 Y, H' I9 _' B
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   , K. k9 \/ u0 n
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   + a5 M3 v: ?0 M
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    - x3 j# U) H; j* S
11. 
    ; K. y0 D+ h' I  P5 z0 C. O3 a) d
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    3 i3 h7 S1 s: Y1 o0 I$ l
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    % E: J4 T( J+ T8 s% g; N
15. 
    
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    . o7 }" X: m  x2 j: |
19. 
    
20. s.Format(_T( "%d "),   i);
    * `$ }: w$ w9 C6 y, g
21. 
    
22. return   s;
    1 |7 Z3 l$ @( M" x; C7 y
23. }
      o; X0 o0 B# R
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    % Y3 _7 Q. c* i6 y
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    . j# O) x% x9 X
31. {
    * O1 U# c) H9 S4 T2 C
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    & @% h3 [) k3 W+ r' s
33. }
    * Q! C  \- W( |
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    - ^# v: w- u2 L% ?' g
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    # c/ d  X+ p' n  z8 ^
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    0 I3 t; y; q: v/ t) B1 H4 ^* i
42. 
    9 E1 y4 X( l- g
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    4 f- ]/ }8 w: i0 x( I  _' N
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    / [3 i- g: x  o
50. u_long   lv   =   1;
    ' g  n! E% N  p0 X4 E' |
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    3 i- u$ d- _2 k% D4 d# }' U- i
55. Sleep(100);
    , m( o! b# D! C" x% G7 R3 }. R
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    5 u* I+ `( i+ K7 w0 [2 J. k9 ]& [- t
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    . J- U& m- S+ v7 L. [7 N
63. return   true;
    
64. }
    
65. 
    $ l% Q) o& ?8 G  x4 K8 u9 G% R
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    5 ~1 T% M  V/ a
68. char   buffer[10240];
    
69. 
    ; N3 I5 ?5 v$ z4 T0 x, r/ ?. B
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    # J: W- O9 S; G" \! G5 l
74. struct   sockaddr_in   sockaddr;
    ; i' B' z3 z+ z$ f6 M
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    6 j5 ~! p4 _% `. w
76. sockaddr.sin_family   =   AF_INET;
    ' H: L2 y, R0 I& F& J! U4 C0 k
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    - [) n" _! @% Q% l! i7 `; B& s. _6 n7 |
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    & Q: |$ ~) O* Q- L
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    : a( o; A) b- |8 O* v* d* n
85. int   pos   =   0;
    - C) e" ]9 [4 K- I  T* s* S! K
86. 
    : u& w5 J" n' M- j
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    3 a& J& U! T, ?( d
88. 
    
89. result   =   response;
    % V4 a5 M( |2 L# O/ C
90. result.Delete(0,   pos);
    
91. 
    ) {% |& D: f% S1 Q- M: @
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    " L- \. N0 D8 N* [
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    - Z) R5 h1 K( A
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     0 k+ y! B, ?" W8 W! T5 m8 e
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     6 ^9 @; C( e, r8 _1 r4 k7 \7 a
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     2 z% m1 d; R) l. P+ S' q$ a( F
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     ; q6 }6 q3 X# X7 @' @! g+ L) C4 M
112. }
     
113. 
     ' ?& a1 ]% t( _- v( `9 t1 c
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     # o2 z8 |' U7 M, N& a& O( [
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     ! E7 d' k. D# {( D0 X; r0 y
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     ' G; j8 ~. G9 V* t0 {
123. UPNPNAT_MAPPING   search;
     1 F3 y* e" M. v/ {* n1 s& {
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     
131. }
     " O. U3 m. R! `
132. 
     ; r* u, U7 ?1 L# K0 o+ W! I
133. 
     / l# V2 }* w/ X; o" a
134. bool   MyUPnP::InternalSearch(int   version)
     2 F+ E' {% H( \2 t% E5 Z
135. {
     7 ?' i1 t$ X( x9 a7 S* v& s$ P! v
136. if(version <=0)version   =   1;
     5 A/ F: T6 ^! \) ~, S" b& ?, e$ u
137. m_version   =   version;
     
138. 
     $ k4 P/ M0 _3 h$ A$ w5 ~
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     $ N0 R: M* U5 B) i# J( w, r. o
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     ; }- n8 L. q7 x0 ]
147. u_long   lv   =   1;
     9 r" }, j4 j0 H- D5 w7 o
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     # N* B  Y9 c" l8 S
150. int   rlen   =   0;
     . t' I" H+ W: I, X, o! u% o6 q7 d
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     2 W3 I$ Z# Q; I: i
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     ) |$ ?& X( s+ ^# c& `4 G6 r
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     $ v4 @" e& ~" r/ P8 d6 L9 G  B/ w
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 "),
     & v2 I, [' J0 v: N* {" P
157. 6,   m_name);
     0 R! m. ?$ [( Y) P/ \+ ?9 l3 S! _
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     ) \2 T. Z7 D8 z) j3 K
161. 
     
162. Sleep(10);
     
163. 
     ! Q, ?6 C  ?6 o8 R! f
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     ) @: T6 {' W" `9 U$ P  Q
167. closesocket(s);
     1 G! X- ^6 o) L' P6 v8 w* L
168. 
     : v7 C/ t+ \6 p- S. d
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     0 j  p6 }, |  d$ {- n9 G
171. if   (!parseHTTPResponse(response,   result))   return   false;
     - U; |* e5 t5 t- G! q0 f+ S* o, V/ V
172. 
     & e6 F3 A: r: T* _+ g) S8 e
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     - s7 D: l2 ]+ F7 u; P6 a3 I0 S+ r
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     , E: l# f" ^" ]: k; D; i( `
176. for   (int   pos   =   0;;)   {
     ! P+ {4 a) V6 Y! C
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();
     1 m6 p6 j6 J& Y9 e! D) n
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     . c3 R" F: K  E
183. m_description   =   line;
     5 h# i& e: G! g1 d
184. m_description.Trim();
     
185. return   GetDescription();
     6 G) o& Y) t7 |
186. }
     / f! @9 p% H9 g/ H, q
187. }
     ' y% n, `0 G2 n1 p3 T1 V
188. }
     . _6 e; G) A: J/ z7 E( N
189. }
     / M* F" x, W* P; ?: I7 [" ~
190. }
     
191. closesocket(s);
     9 `  m  Y& h6 q9 b
192. 
     
193. return   false;
     2 Q- I9 G( L( R' p0 G! C
194. }
     : r4 Z  `; q* b: h6 r

复制代码

天云上人

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

4 T9 d7 T: }- }; ^
/ G; H8 r! @( _4 M" f///////////////////////////////////////////
* k- p. s- t  u6 g4 L. _$ k& D//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
" @3 u0 e( V" H- J8 D0 f

  M! M, K  O9 M" h  M#pragma once
#include <exception>


  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
% @- L6 D3 n& k};
; B+ D* W1 G) e

enum UPNP_IMPLEMENTATION{
2 m$ T1 U; L/ U& K1 `        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
; C( C+ u! F. C        UPNP_IMPL_NONE /*last*/
" `' P0 Z  k0 J& H; K* Q};
: o9 n) p% l- Z
: I) w0 i3 p0 i1 U/ g
; H' x3 _7 n% T; O5 ^
" c4 H, I) u; e: ?
, T# w8 s1 T. u5 w+ l) yclass CUPnPImpl
( [( z  I( e" v+ A6 K7 ^0 P. i{
% ]) Z2 r. K) d* P8 M% N9 Rpublic:
+ u* r7 ?: Q2 }2 X3 {        CUPnPImpl();
8 F$ D  ~, H2 B. ^, g9 ?        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
8 k# h( T+ \& a/ M& {' i1 f# S        enum {
2 |. i" H$ F5 p' f  z                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
        };


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
+ ^, X. w/ L9 G4 ?+ y        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
7 [/ @3 d5 ~* Y9 i8 U1 W) f        virtual int                GetImplementationID() = 0;
1 ^# v! }" M8 W9 M& Q- S       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

9 c0 Y- W4 V+ Z) W1 v+ ]+ Y
) z% V, w, D' b. G( H% H4 E$ j        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
2 \! ~' n3 E5 O( W$ Y" K0 |        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


// Implementation
1 Z4 y  j1 V- I# L  Xprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
( U* a5 a9 C( v        uint16                                m_nUDPPort;
3 X6 G6 E# C: Q2 q1 B, a  g7 g' H        uint16                                m_nTCPPort;
: y1 t2 H. ]# S, ?- _        uint16                                m_nTCPWebPort;
8 ]0 W2 l" G" F5 C4 c& a. F6 n  L" b        bool                                m_bCheckAndRefresh;

4 d  L. T+ I+ U/ g
private:
: P" j* P& ^. p2 i% X  @        HWND        m_hResultMessageWindow;
/ ]! u( K, Z' ^        UINT        m_nResultMessageID;

) x1 G8 B1 f4 F( i9 ^0 n. n5 X
};
0 V& H) U+ W7 w  B+ N5 {3 t0 L
# N3 \; ^' |  M
// Dummy Implementation to be used when no other implementation is available
6 U# W* n. W6 \& L4 b" Uclass CUPnPImplNone: public CUPnPImpl
0 z, U0 D( l9 Z* \, g$ ^{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
/ L0 F9 R; A: V% \        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
7 @' Z3 P0 O* X! D. n( P        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
6 x2 X+ G5 S; u7 E$ O        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
- ]: w2 M( q* Z2 u1 c3 q# Z8 O};


/////////////////////////////////////
- E6 h2 T, z1 I, g# S! |& J//下面是使用windows操作系统自带的UPNP功能的子类


#pragma once
- Z) o; A# l: f#pragma warning( disable: 4355 )
! r1 Z! \) D% n* n) l& O! }
/ ~% m8 b+ S, u2 A1 E" O
8 |3 C3 A. R) ~, J3 D$ a; Q#include "UPnPImpl.h"
! ]0 \& `8 i4 `' V* U#include <upnp.h>
#include <iphlpapi.h>
. K6 ~; D  y9 [" O- q- e% e; m#include <comdef.h>
#include <winsvc.h>
, _2 n9 C6 C0 |0 t! f2 X1 w8 e

6 `" @4 |& {! M+ D3 d#include <vector>
: A0 h1 y- `* j0 H% F#include <exception>
6 }* P, N! [, [#include <functional>


1 @. }8 @* Z& w& m
8 L& }% ]' b3 R& E
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
" C. `5 U) d9 E$ Xtypedef _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;
. @. M: R! E  t. ztypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
8 p- l; l+ s' z4 Ztypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


# ?, ?% i  G! o$ etypedef DWORD (WINAPI* TGetBestInterface) (
5 l. k5 z6 h# M3 L  IPAddr dwDestAddr,
0 [" E2 ?. z% Y* D+ t) }  PDWORD pdwBestIfIndex
1 G: C* y# e, x3 c+ U);

; I( a  ], E6 J* {$ I; g9 T
6 o0 b- S4 u+ }+ m+ J% dtypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
# {6 m2 o1 K* b7 \) _6 g  PULONG pdwSize,
9 l/ G7 Q- B, g! Z. Q, ~  G2 p  BOOL bOrder
);
- X7 @6 [; f6 q; i
2 L8 M7 N# ^9 w  p2 o; K
typedef DWORD (WINAPI* TGetIfEntry) (
8 ]' s+ @' L- y% F. j! L' O  PMIB_IFROW pIfRow
- G& H& R7 w+ R7 \& w);
* _- v% z9 d" u" J, Q2 @

CString translateUPnPResult(HRESULT hr);
& A$ f: A0 D  Y$ z" BHRESULT UPnPMessage(HRESULT hr);

( ]4 R6 Z) ~3 o" Z1 U$ J
class CUPnPImplWinServ: public CUPnPImpl
/ R) A8 n8 d) U$ f# p6 c- u{
9 l3 w7 [; F8 w8 n0 h; F        friend class CDeviceFinderCallback;
& b, i( n8 G9 D' c* |        friend class CServiceCallback;
+ l5 ]" v; d+ s1 J// Construction
) }: A2 L" P; @" Q# gpublic:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
, f0 u+ P9 r3 s: a3 |/ g6 U( i! A
$ U6 k8 U  h6 }% g2 h
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
: }) B6 n7 X% j- j) j$ p4 C- I        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
4 S+ l! M# M+ Z4 Y
2 s8 B' {- n) D+ j
5 K8 }! D2 D& a/ A# H1 w        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
1 U! f: w- N5 _- Y- b' M        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
6 P$ v$ f* g7 ]9 T

protected:
8 }( x; `: U9 w: H. {! L1 b* [        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
. d% m* ]( d& E4 [        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
; X! m! j2 j, N3 E        bool        OnSearchComplete();
        void        Init();

6 d# R6 V* {- D* @/ C% X
% u% d4 {* S0 }  T+ z' [; E        inline bool IsAsyncFindRunning()
# G0 a0 ?: \! c  T        {
/ n& w. b+ Q* x: N                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
' [7 q6 u& u+ {2 S                        m_bAsyncFindRunning = false;
9 D1 b( y2 Q6 @( h( @! d                }
; F  s; b4 O1 ]2 r: N3 F7 G. ]3 w0 Z                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
0 k3 ~* Y, i9 G9 a                {
                        TranslateMessage( &msg );
- {! c8 _+ m( ?6 Z$ q                        DispatchMessage( &msg );
- G% C% P8 Q. W; L9 q                }
, l) u, x# O  ]4 Q8 X4 Z. h                return m_bAsyncFindRunning;
; \" G+ R$ x& J. W( l# d        }
' r3 A/ v2 [' t  k+ \5 W

" b7 {8 v! d# E! Q% E        TRISTATE                        m_bUPnPDeviceConnected;
+ J% f/ N5 O) K" e7 Y' z9 a/ W
# B" Y, a6 ?; |0 N  }8 H4 j
// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
) C7 p# R/ e/ t8 P        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
7 V1 N. `9 b6 X9 P: F        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
2 a8 Q" N, D" n        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
6 F. @. w5 r% J4 `+ M
( K" U! C7 O. I% C/ E6 F: t
3 y3 [: O$ Z) R; u        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
% }( G5 ^- Y# a        TGetIfEntry                                m_pfGetIfEntry;
4 X6 ^/ {9 \+ A& c

        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
/ m0 r7 G$ u  g$ B" ~# o        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
) N& y$ z. w5 P                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
/ V8 d3 K8 d/ h( \9 s$ ?3 H                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
& t0 Y9 X+ ^( m* w+ h! d2 B4 v3 J
8 F1 ?0 e1 Z8 y7 ^+ ?: E
5 r% f1 y1 d( @7 y4 z& U% A" \                        if ( FAILED( hr ) )
6 t, B5 S' r( [% a( ^5 }. R                                return UPnPMessage( hr ), false;

$ y" ?8 L2 O1 T- K4 R
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
, s. `4 G, @' E# c. s& J8 O                }
. G, v' k6 f: e- A% b7 N, \$ K                CComBSTR m_udn;
        };
       
) U( ^+ ]  x# t; q. u8 f        void        ProcessAsyncFind(CComBSTR bsSearchType);
2 Z: m1 ~# r2 b6 Z$ U( F        HRESULT        GetDeviceServices(DevicePointer pDevice);
9 O' A* ^6 Y9 s+ r        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
5 M; }1 m7 r4 T2 E- s2 E# J' v+ i        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
' k. W9 r3 W8 D$ M. [8 \1 L        void        StopUPnPService();
0 g% f/ B' J% W+ T7 n2 ?

        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
8 U  u6 l( k; l# k7 ?        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
/ E" h: ?3 t5 t: L6 P8 ?& g        CString        GetLocalRoutableIP(ServicePointer pService);
1 ]5 a( R! d! q; V* e

// Private members
private:
/ G7 J" M& H9 Z( T3 @        DWORD        m_tLastEvent;        // When the last event was received?
9 g) j# ?: Q+ r, N) X/ n- o% y        std::vector< DevicePointer >  m_pDevices;
% W& d! i$ z  s/ r        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
3 }/ q: X4 p0 g8 g: \; v        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
8 f/ J* u0 ^- w& [) i
. {5 Y& K  R2 i: h& O) O
        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
) ~7 v: w5 k+ Q/ w8 B3 T6 n        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
' f- ?8 y9 t6 x7 V' q; v        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
" D) O7 v) X, r+ u. U+ e        bool        m_bInited;
3 b! q% S' i5 c7 p. w        bool        m_bAsyncFindRunning;
9 p9 `4 P1 ^' I( d+ D        HMODULE m_hADVAPI32_DLL;
% e7 y% H( t% W        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
. u$ e3 F7 \- U7 `( O' j8 b- ?# _        bool        m_bDisableWANPPPSetup;
! P; ]; V. w& E4 a# P
8 V" }# d! |$ e) T+ D$ h0 E, L
- h9 M4 y2 O$ s7 j1 |% [  m' T};
" a4 l  |* o3 X

// DeviceFinder Callback
class CDeviceFinderCallback
% G5 U0 y. G7 L3 C1 ~8 }7 w        : public IUPnPDeviceFinderCallback
/ C$ r: T7 m2 ]$ R' P{
: ?7 d; I/ g, t$ c; A$ c& R1 mpublic:
9 `# q9 w# M; k' M) q        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
: r4 u- _6 P) A$ M7 L" c7 j/ `6 o
- b. V/ ?' p1 S8 V% l6 E
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
/ t/ i' k* h+ j8 J* c   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


* d8 W& G& Y( R// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);


0 B( n# P0 ^  {* j( b( F  _; S6 t- {( qprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
, k4 M' A) F- _
2 s5 `' h1 z/ X  e. M0 r
( N9 c# H$ u7 Y// Service Callback
$ f; C1 W0 q; [class CServiceCallback
        : public IUPnPServiceCallback
{
6 P  o( n  t3 Ipublic:
+ o" t; @' D  W% a7 m. h        CServiceCallback(CUPnPImplWinServ& instance)
% Z- k5 Z3 c0 l: l! U; S( Z                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
/ V8 n  V# }% O, [6 E   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

" G+ D7 m' u4 l& W2 j4 X
# U  q2 x) y# \* Y5 m6 s// implementation
! E% D3 N! l. K/ Mprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
+ x& e( I' m2 `7 |) i0 Q! A        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


! I/ \4 X( x! q: {, T/ uprivate:
        CUPnPImplWinServ& m_instance;
# R+ \( o% K" \  z, l) Y        LONG m_lRefCount;
};


1 V$ K3 y& k( p* U/////////////////////////////////////////////////

7 j0 U1 ]0 a+ O9 `
, B) V2 F4 I* n. i) {5 k1 v& t. r6 U使用时只需要使用抽象类的接口。
" B2 f: k) U5 U* OCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.