UPnP

天云上人

/*uPnP.h*/

1. 
   * s7 P2 r' g; }5 {! J; Z" \, ?
2. #ifndef   MYUPNP_H_
   ! ?8 u  b, M% `! ?9 b# \
3. 
   
4. #pragma   once
   # B, @- D6 P( A: @! F( B) Z9 [
5. 
   
6. typedef   unsigned   long   ulong;
   ) \- O* J, [$ E
7. 
   
8. class   MyUPnP
   , ]' K# t; X& L& T( S% b" j& K& V
9. {
   3 p% S- c' ?5 V- x, h
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    8 i- `. ]8 |$ L1 D+ R: Y
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    6 s( O2 M% {5 C: b& t
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    . A/ {* Z1 A, o
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    3 C& f+ `* Z+ }: u+ E* p5 b, W. u
17. }   UPNPNAT_RETURN;
    & P3 I, E( g# E3 b
18. 
    ; ~2 {0 C3 |% f& v9 R
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    % q. r5 r5 G& l+ `* r
21. UNAT_UDP //   UDP   Protocol
    + f) L7 ]# f1 S2 i
22. }   UPNPNAT_PROTOCOL;
    * |) V5 W4 c  L
23. 
    . G5 @  J9 N' n7 \7 B2 ]8 G) l0 @& P
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    / E1 t9 v8 e3 m0 b2 t; x/ K' ^8 p( U
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    , b1 m) {* i& [& i" v, \2 P
28. CString   description; //   Port   mapping   description
    ; Y$ K( E5 H+ u% ]" z# L
29. }   UPNPNAT_MAPPING;
    - V2 Q' D. ]! O5 v
30. 
    
31. MyUPnP();
    9 Y) K. K& S  U8 e+ Y5 d9 \
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    4 C/ L3 Z4 g. ^! K4 b" _
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    , m2 i& i% q( U: {2 f" |
36. void   clearNATPortMapping();
    " H6 t. C* R% Y: k6 f: ^/ d
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    / A! [, Z2 ~. b3 g4 q* l, L+ T" T
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    ' B6 [& D* q# i2 M/ F; ]! g9 D
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    . O6 Z8 L" O6 j. ^: |5 h8 [" _
46. 
    8 {2 e( ?* K3 q: ]2 A$ a
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    9 Y  f! e* @* V  V" ~4 a
48.       const   CString&   descri,   const   CString&   type);
    / V7 L; ]! A0 W% F. L( ~
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    - `; M; h/ G+ e
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    " \4 o; U( ~7 ~5 h! j$ D2 J2 _
53. bool Search(int   version=1);
    
54. bool GetDescription();
    " f- C" E2 t1 d* G) I' F
55. CString GetProperty(const   CString&   name,   CString&   response);
    2 q9 ^; B: H- K. ]& @
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    3 Z1 @5 B; d1 X+ x7 d: h& X. A
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    ( ^4 d  ?  T/ U# V: n/ ]7 G) X. L5 N
61. CString m_name;
    8 D, `) ~" ^& h: d
62. CString m_description;
    $ H4 _' a( Z4 x/ m4 k3 S" z
63. CString m_baseurl;
    
64. CString m_controlurl;
    % w7 f4 c2 r' Y' x
65. CString m_friendlyname;
    & g7 z: R9 @- u6 a- w1 s  E; O! e
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    $ J: R5 x8 Q' F
72. CString m_slocalIP;
    
73. CString m_slastError;
    0 \+ S9 U3 x1 X4 h/ Z3 d
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    6 @- b' {8 t, k) x$ V- r
77. };
    & b. ?& L7 t; F3 X: \: `% t
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   " ]" D" ^: V. {) \5 C, N
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   1 d# u% e- W0 }  Z+ z( ^" x
5. 
   4 p+ N# Z( I, t4 X7 V& S3 C3 l" B
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   ; {7 e3 A; Z0 q" D! y4 R4 S
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    ' X  F' G9 u1 g0 Q" v7 y
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    * o4 P: x) V3 P4 B- K; \! U
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    2 I+ I. X2 c3 @
15. 
    
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    : S1 i: y/ ^# k% T) D- [" y
22. return   s;
    $ R7 N! a* [3 J1 P7 t$ ?. l7 e# c4 a
23. }
    6 c( Y. j" u" v  N2 }8 U% v, }
24. 
    , z) W+ F8 C% y
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    % ^7 x7 T# m# P0 Z* f
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    . c( Q* e0 W* Y8 ~
31. {
    1 U5 m, _" t  N
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    ' X' I1 ~5 k- n: K7 G& g
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();
    4 l3 I4 z$ ~, q- ^% m! E
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    3 Q) A- x: ]$ D# y4 O' N
46. sockaddr.sin_family   =   AF_INET;
    * [; A5 a3 Z# ]* b) o. Q
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    & p* V9 ]7 O8 n6 {) ?+ U
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    2 K- p. v. t2 ]2 K" I
51. ioctlsocket(s,   FIONBIO,   &lv);
    9 C' o5 r7 j! r; J9 ?9 K- d9 `
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    4 `4 s6 P! w4 c  Q  k1 w. j( ]; l
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    ) o7 z. O" ^/ ^& ]/ C: i$ C4 p
63. return   true;
    + Q3 l- R- |& B% S
64. }
    8 G( H! W- k2 w% S
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    ) r8 ~; B, t0 M0 `. _7 @
67. {
    2 z9 T9 V0 Q- U$ v) ?* L
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    9 L9 K2 Q* u6 T
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    6 }8 u" K6 C' O& E& }; B& u' ^
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    / t" j) d+ b) k$ c5 P2 K! m
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    4 b. ?3 t1 W" [4 b1 U
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    * h7 g5 O& N9 ^; b/ I) ^) e# E0 C
81. }
    3 n1 s! g' Q' `
82. 
    ; p7 `8 B9 r# b8 h$ U$ J
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    ' p* Q, G% I# O* }3 }/ U
84. {
    7 N: v3 }$ }# |( u$ ~1 i" r/ o
85. int   pos   =   0;
    ) `; P2 A: M3 O- B' `$ Z  N
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    : R7 {% x/ S, [
90. result.Delete(0,   pos);
    
91. 
    : r! l2 b5 V8 {8 [9 ^# S2 c/ Y$ Y
92. pos   =   0;
    1 Y/ S0 x& V; B0 C+ Z4 T
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    - ^/ S5 T) B9 ]( ^) X& P" N8 q& n1 ~
96. return   true;
    3 [8 ?! a% }$ g$ u( _8 v* v
97. }
    ) i4 ^7 B, s  Q! j( X; |! q& E
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     9 d: N$ w$ ]9 k9 {
101. CString   startTag   =   ' < '   +   name   +   '> ';
       i' T5 ~8 B7 @& j
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     1 f7 j! D) h  p. Y6 \0 Y2 f" h+ V
106. if   (posStart <0)   return   CString();
     & F6 F" M/ e6 i. Y3 S( X+ \5 A- M
107. 
     % C/ T& z( D. R% J+ d& D
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     & @# N* M5 j7 X7 s
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     2 |1 I1 b3 u1 y5 s7 e3 V
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     : |1 i" C0 \" u9 Z% a& @+ [
120. 
     ' s; ^, q- y) M6 z- w
121. MyUPnP::~MyUPnP()
     
122. {
       l6 Y% U  i2 x' c
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     , b8 U* ^1 ~/ s! p3 _6 s& n$ C1 j
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     , q& g! [" n* x1 P' P/ t
130. m_Mappings.RemoveAll();
     & C4 _3 K$ a1 }5 t
131. }
     
132. 
     : h1 w# r: ^& d/ @
133. 
     5 c3 e) ^& S3 U' Z% T- l! k
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     6 j6 @7 w5 w+ G3 |, d
137. m_version   =   version;
     . m2 G; c, l! Z" H6 [5 p
138. 
     
139. #define   NUMBEROFDEVICES 2
     " R3 o) Y/ w+ m. v
140. CString   devices[][2]   =   {
     ) V1 `, h) Y% w- X9 {9 i
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     ' I' |/ Q; b+ ^1 W
147. u_long   lv   =   1;
     6 U# b, ^& `! N" N
148. ioctlsocket(s,   FIONBIO,   &lv);
     ( U8 h' W- S6 i% Q. i* u
149. 
     / m) C; `5 J) M* k" [
150. int   rlen   =   0;
     8 c1 u. X$ f& _! Y9 C$ O5 C8 T
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     9 |2 |, U# |/ I9 O  Z" q1 F2 M5 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);
     
155. CString   request;
     ) b. p. G" [% B; q& ^9 @& Z2 w
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 "),
     & {$ \' Q! {8 B* h  W
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     # B0 W0 L6 b. X
159. }
     . k1 z2 O6 K" O) S5 q9 {
160. }
     9 ]! M2 T* Z1 j: Y' A
161. 
     
162. Sleep(10);
     / c. S$ {5 y1 K
163. 
     ( t5 p# Q* g) S# X1 z9 H) z% V
164. char   buffer[10240];
     3 h) C: E8 t' G* u% @7 `" U2 F2 e5 C
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     ! [% _, @* q4 O4 A6 ?0 v; Y: K
167. closesocket(s);
     4 V( Z) z/ r2 x' }5 G2 A! g
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     6 \$ R* c! s0 o( e: V3 G# v: [4 s
170. CString   result;
       D; |( G1 o; N7 @( L# t
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);
     
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     5 x0 w1 x. U* r
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     8 `9 ?3 `0 B5 J: e
178. if   (line.IsEmpty())   return   false;
     7 H$ ?, e  ?2 h" U, s) U
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     1 {: }: U8 A! ^" {  \9 w& i
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     , V2 Z# F; }2 e: z/ V# a
183. m_description   =   line;
     ( Q: p2 p# |2 w
184. m_description.Trim();
     
185. return   GetDescription();
     ! ]9 i% i$ r( e% z
186. }
     0 t4 }# S7 s" r( Q2 @
187. }
     
188. }
     
189. }
     
190. }
     
191. closesocket(s);
     9 [- d; ^$ A- G3 b, u  q
192. 
     6 p) A+ o9 s+ U1 T+ J
193. return   false;
     # ]7 ?% s; }. f& Z/ J' s% E
194. }
     

复制代码

天云上人

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


1 d9 e) }5 Q, F5 H$ `3 t7 z) V///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
% z2 [- I. i3 ^/ {5 U1 B
7 z6 l: @, w2 p: s- J/ o
2 ^0 g$ d7 D  m8 y# o2 O#pragma once
0 E0 C7 N3 C" f- l; @( n4 \#include <exception>


  enum TRISTATE{
        TRIS_FALSE,
" W0 k5 K, X( f0 L        TRIS_UNKNOWN,
! Q( s+ c" {$ f) O4 Y" J" D4 J        TRIS_TRUE
};
- O& f$ }- ?( U% W: G  a* F
. P: f8 e0 y; K  k; q0 T5 |
$ g+ }( [0 t2 ~3 f) [5 Cenum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};
2 i/ [* U2 Z9 ~$ U  o



' @7 E2 P5 s" \6 _class CUPnPImpl
# @- s% h4 U7 z4 L' K8 Y{
6 b5 p) [4 J, |( Q! l+ Q6 ipublic:
  z, G* i5 G/ u8 [/ a# P1 I" Z        CUPnPImpl();
        virtual ~CUPnPImpl();
) k7 [& R6 z2 {% Y# o/ f5 Z, T. m: j        struct UPnPError : std::exception {};
        enum {
- K& ]( q4 y" T# X) u5 q                UPNP_OK,
% C! |  V0 |1 |- H$ x( C                UPNP_FAILED,
                UPNP_TIMEOUT
        };


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
! v! P% y) W2 _2 S) w: s        virtual bool        CheckAndRefresh() = 0;
% j1 [  S% Y1 p4 v' L. k5 t& S        virtual void        StopAsyncFind() = 0;
" N3 A' S% \2 z! B/ a2 b! ~        virtual void        DeletePorts() = 0;
! [( l3 U# f+ P/ f+ q, j& n        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
- p1 j+ B8 Z5 ^. H       
% c4 w$ e% Y5 A0 x( G, i: v( G! \, `        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
, A2 I' _5 F0 z; v8 v" A' a# @2 V
: n" H$ S( {* U8 t0 ?( f
9 }) w' K/ V, y+ p" U        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
1 u1 N/ }$ S. y6 g! T. S, j        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
% g4 w! Z1 R+ w$ {        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
# f2 l! [% Q% A* q5 T3 T: y
( H6 N0 S5 N) C/ _
// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
/ P' K  m* ?0 q9 }$ V$ b: [/ L        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
6 [& _7 k4 |% G" M. @! g
  P" v. S: d0 U& ^2 e
7 Y1 n# t) a) X0 G- z- ?private:
        HWND        m_hResultMessageWindow;
+ X) c" X$ {2 M3 a0 H" N        UINT        m_nResultMessageID;
% I/ E; t$ G0 |' q1 w# A% |

7 W0 D: `& }& ]9 g* H. H" k; O};
# V; |! v' }' ?( v% Y3 t! u: }# R
' ]$ r# o9 D# M+ e* j) b7 i
% c- w* n* O- B( o/ }// Dummy Implementation to be used when no other implementation is available
6 s1 \$ O( _2 S( S. R* n& C. tclass CUPnPImplNone: public CUPnPImpl
# }8 a0 f8 R, Z0 |# [3 s{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
- O, T! W( j. @' o. h7 @. d        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
' |3 @" R# K) W};
5 E! Q* J% s5 v* \- F, i7 A6 l

/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
- _. W% I( `) q9 V

#pragma once
% Y0 Z3 ^, k* y' c% u9 B: x#pragma warning( disable: 4355 )


3 I/ A8 D/ q& d$ J# O  \' [% D#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>
0 D, u0 H& c3 q. c8 F1 u: |

8 c/ J+ \: `7 |$ `3 q#include <vector>
#include <exception>
#include <functional>

1 l; F  w, y, r* Q
3 y9 d) _& P0 g% p

, k' W" U- v0 }* \( U% x& Ztypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
; b9 V) `4 s: o9 Ctypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
+ ]- I% K9 q+ J# ]/ ~2 G+ Jtypedef _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;
7 a& \$ Y9 W8 G8 H' Mtypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


$ X7 ]& w& {( a; L2 vtypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
2 L* t* _: S; r* U  PDWORD pdwBestIfIndex
);


/ c$ h# g  o2 K' |4 l3 M5 vtypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
+ v4 i, M9 J9 [2 s) w0 a+ n% l  BOOL bOrder
);

* W7 y/ T1 F4 c* G* O9 l8 m
typedef DWORD (WINAPI* TGetIfEntry) (
2 f% g  y# U# C+ P8 f2 ^; f7 T5 i  PMIB_IFROW pIfRow
6 F# \! m/ C% r! q$ {);


CString translateUPnPResult(HRESULT hr);
4 R# i; R, K$ q$ l; p. L7 E2 D% JHRESULT UPnPMessage(HRESULT hr);

$ Q$ m( X0 Y" b; C% |1 l& @
class CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
0 H5 Q& R# i: s+ Y# p4 N% s& Y        friend class CServiceCallback;
// Construction
public:
1 Z, W& p+ [1 t  G7 @& V9 V        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
0 }3 }( G3 T5 H! W' X" K( g; P* b

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
. |  J& y' c! M: S, o        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
5 t& R* ?, Q  M& g$ @        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
& h  t, G( W% K0 F+ J! r5 L
  Y& v, v' Q  a% A) O$ v
0 v) }: ~$ f# x: j- t8 u        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
; E, s1 z0 P6 k) m        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
' w3 B# {3 q6 Y. R
. H) {' X& `& p6 y* n
protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
- `/ J/ w! g- B( i, [! I        void        RemoveDevice(CComBSTR bsUDN);
; p( R1 g# L- c2 f; x& F        bool        OnSearchComplete();
5 {5 P, ^3 ]! C; P0 ]% J+ H        void        Init();
! n! Z9 \1 e7 v& y4 O; K7 P* D, S

$ G8 B/ i. ], T6 W  B7 H        inline bool IsAsyncFindRunning()
# n& ]- w5 l; Y9 W        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
6 C4 c1 G) S) }1 z6 x, {                {
5 m  ~( Z3 J. _) o7 N# `% L                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
5 H) V/ o# K5 s: x, V2 N                        m_bAsyncFindRunning = false;
' Y' p. F( T2 F# t$ c                }
                MSG msg;
, P2 b, H0 O5 t0 ^) W                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
3 h2 w7 a' Q" ~! V1 k6 r                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
6 d" V: V0 @$ I7 Q                return m_bAsyncFindRunning;
9 p, ?' J# D) i. f        }

+ K: g- A  J4 u5 m/ o5 J+ c8 z( ?
        TRISTATE                        m_bUPnPDeviceConnected;


( c# g8 F8 Q; ^& k3 @7 U8 b// Implementation
        // API functions
+ R, E- b! r1 s9 Q, I        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
8 z. J' Z, s+ w' O( k/ Y        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
0 H( A* C( S; A        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
8 S4 V5 V  R1 ~) {        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


        TGetBestInterface                m_pfGetBestInterface;
5 F% A5 U7 H  ]/ t9 G2 {        TGetIpAddrTable                        m_pfGetIpAddrTable;
( K& i: a# @& I. _' c3 C6 Z2 u        TGetIfEntry                                m_pfGetIfEntry;
% x+ h* p, \, p# k- c
" K! F/ x: a0 B$ p3 A5 T0 l( o
. S0 L1 e; R7 G. _$ V( `8 K        static FinderPointer CreateFinderInstance();
" e& ]2 l' E0 X4 n+ x        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
7 T- c9 C$ ?9 ]                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
, q, l0 M. ^$ l/ v" _                result_type operator()(argument_type device) const
- F5 ~/ a" K3 n4 |# _2 b  b1 \                {
                        CComBSTR deviceName;
, ]( c* i. f2 m                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


                        if ( FAILED( hr ) )
8 y* f+ M; \; W5 v                                return UPnPMessage( hr ), false;


& c1 A1 p! i, @5 i                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
1 {* D% K2 e+ x" X                CComBSTR m_udn;
$ V) @  j6 I# u2 E' a        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
% M* @/ y0 |; p$ H) b' s        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
& p) d+ S  F0 X1 m5 M$ X        void        CreatePortMappings(ServicePointer pService);
* c' u  b. c+ C) M' |$ E        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
) I: e) q: E1 C0 N9 Z3 }        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
; Q: d# Q  ^) ]- w' e! A. y" a                LPCTSTR pszInArgString, CString& strResult);
) D7 \9 ]3 E0 [9 A. ^' G3 Y- d        void        StopUPnPService();


        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
' y  ?" T+ F. T) z3 h        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
* [, x7 }+ k% L: `9 W" ]" F; W        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
9 V. V' A0 _1 g4 T7 c        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
9 ?) L, y# I) B! N2 |7 ?5 ~        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
6 C0 a9 u  z  l

5 o3 ]/ @; K; \, |; f, Y// Private members
/ K0 y! N* }4 E/ g0 a7 g0 ~" _private:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
" y* u& p2 S, f' S5 m        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
9 A# O% `- [0 j        DeviceFinderCallback        m_pDeviceFinderCallback;
6 J8 [! f/ k$ S) W8 |! V        ServiceCallback                        m_pServiceCallback;
& _" t4 \  \7 j# ~$ G* y

- [2 ]8 e( m. v1 w) l* }        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
& t( u  ?' p% x; ?' I- i        bool        m_bPortIsFree;
        CString m_sLocalIP;
1 d* V$ u5 ~2 j; t2 ~4 a        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
! S* f& _% m' H( ~, C1 K! K        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
8 f7 O4 D# d1 v7 S$ d+ ]        bool        m_bInited;
        bool        m_bAsyncFindRunning;
' e; E4 r8 c0 r/ S5 A        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
+ E" K3 i3 s, y$ L, Y0 Y        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
; c  K% k- P8 g

};


// DeviceFinder Callback
8 f' t6 P3 i  a( m* @, s* Dclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
+ \5 D1 w1 o0 N- o# \/ T" C{
) |' \1 O$ J  w& W' Y& u2 |+ opublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
$ {7 g% e. J/ m9 e/ J$ a+ H                : m_instance( instance )
. [+ {  _2 x. ~. @/ Q2 f% D9 Y        { m_lRefCount = 0; }

8 F: q. o& K0 `" a
1 W, W8 o8 S0 b% K$ G# T3 j   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
- [" M5 }3 W& T- B   STDMETHODIMP_(ULONG) AddRef();
6 W, W7 N( ?* [9 E' `4 _5 f   STDMETHODIMP_(ULONG) Release();
) q. V0 G" K1 G) s/ ^5 P5 D
4 D! C/ E  q# x+ b0 W- n
" G' g; j8 i0 {7 W" G& L// implementation
5 i* B2 b& ]- H0 A1 Pprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
1 x4 ]6 `, a: b        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
8 R% o! b1 O0 C3 N$ n5 F: m. C        HRESULT __stdcall SearchComplete(LONG nFindData);


$ d# S& m* W0 |# ]  Y; H4 Lprivate:
        CUPnPImplWinServ& m_instance;
6 z: d7 x4 V4 d, b        LONG m_lRefCount;
};

7 `' L* P2 ~) G1 q; i" ^; f8 Y2 a
% b7 K6 w, k5 o+ e4 M8 Q* T8 X// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
{
6 l. R# z) J  `+ U8 vpublic:
- j- ^" j8 `/ P/ k+ l% c' m        CServiceCallback(CUPnPImplWinServ& instance)
1 z, U; X9 m. m# w% X                : m_instance( instance )
; o$ K+ {: U# U4 Z) I  S: ~/ F        { m_lRefCount = 0; }
. I6 D) b" ^: I  A, V' r; A   
7 k2 c9 u6 m' e2 B5 f   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
9 Y* y! b0 N( }, S  Q   STDMETHODIMP_(ULONG) AddRef();
% _; e( _# D7 u- S, Q  z" F   STDMETHODIMP_(ULONG) Release();
- n+ P1 D- X2 T1 x
, E& g4 c2 U/ `. B6 {' a2 z
// implementation
* h- n2 k) j7 R  x# \private:
7 |# u$ r4 J; E        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
; Z& h* _3 v& a1 w

private:
2 `' j/ P) ~+ g% j        CUPnPImplWinServ& m_instance;
$ J! {3 k" V) r5 q        LONG m_lRefCount;
% J/ W0 \) T+ @: s2 P/ [) w};
5 z0 U# l1 {3 z- J& Y" F

5 N0 r. W- r" E/////////////////////////////////////////////////

) x) K+ \9 S) V" \: R4 ?2 o; T
. F# h  p% T  r) \- ?使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
9 G5 Q9 h( Q0 n. pCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
# b! _. j" L$ U! f! jCUPnPImpl::DeletePorts删除端口映射.