UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   & K: M" |& g, W; @8 u
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   $ B" D. J- F$ L9 V4 R( a
7. 
   
8. class   MyUPnP
   1 B$ v8 D- Z% T/ F; T" N8 {( I
9. {
   : p; q& y& K+ v, |* Q; a5 T
10. public:
    + l7 E$ I2 ]5 B" u8 W7 c
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    6 P; R& H# i0 {* M$ Q
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
    ! g$ H$ }0 @0 i' w, F
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 ^1 l# q- J% @' S& P4 n8 b$ M
17. }   UPNPNAT_RETURN;
    ) B* f, p$ c. S* r# Z0 k( M( [
18. 
    / }2 l& |! J0 `. ^
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    ! q. |+ F0 j2 _2 |8 }8 m
21. UNAT_UDP //   UDP   Protocol
    / l. `* u/ S6 i6 ~6 j4 ^. K% L
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    ' K" h. g+ D6 y( W
28. CString   description; //   Port   mapping   description
    8 f% u8 |4 P9 R7 `' X" @
29. }   UPNPNAT_MAPPING;
    . f" |7 H: A9 K3 B
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    - n, [6 P4 r' m- U0 J
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    ( q8 q' w* z- Y: M- M7 M# f4 z# t- o+ d
38. CString GetLastError();
    % J) S4 A( V+ u8 A
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    6 ~  ~; f" _8 [. z+ P
41. bool IsLANIP(WORD   nIP);
    
42. 
    7 D. J+ M. ^4 O* t& x7 R
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    ! _" i' n8 [6 q- Q
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    - O: a0 j3 K& h& o/ p% k: \8 R% M
48.       const   CString&   descri,   const   CString&   type);
    # B# ^$ i% f/ Q
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    2 \3 M$ J4 X, F# Y) A
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    " ^, L9 D6 V9 ?8 F, \1 `9 Y
52. 
    . n4 N# a- a$ V& A
53. bool Search(int   version=1);
    
54. bool GetDescription();
    / y  }4 d7 S% @# ]8 `
55. CString GetProperty(const   CString&   name,   CString&   response);
    - b: n  S; R7 Q$ |4 J6 z
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    . @2 Q8 X! j( h/ y# U$ e
57. 
    5 w$ \6 n4 u1 \; |* E
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    % S# S! C/ r3 a2 J8 g0 d; B/ U
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    . x5 x. r  `- f9 t
61. CString m_name;
    : C- d7 A7 o: j
62. CString m_description;
    ; T0 M+ K8 p1 l5 k% i9 X
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    & i% {" f  m5 z( L1 }# c( s% }4 t
66. CString m_modelname;
    
67. int m_version;
    6 R+ U3 }( c  n! \& M. q
68. 
    
69. private:
    # [! e' r6 Y0 I& l/ }8 k3 A
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    ; D  M( R# }& X! S& |8 m' J
72. CString m_slocalIP;
    $ ?0 V- c7 D6 X* B% I
73. CString m_slastError;
    " Z6 J' u" m  y+ k8 W6 X; J5 r4 o/ _
74. WORD m_uLocalIP;
    
75. 
      Y1 q2 j$ e+ _7 c/ F8 H& n
76. bool isSearched;
    
77. };
    ) o& B2 M. ]+ y1 a( I! x
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   * Q( o5 m  y' k8 f
2. #include   "stdafx.h "
   1 y( H( I# @, o% q! S! e' d5 K
3. 
   
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   0 P- G4 q* u" r/ q
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   ' d' d4 ]$ j/ a6 b6 A3 J: O
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    4 _1 _% w, x/ J0 g. ]* ?. e3 _
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    $ ?& q* K8 K# H6 f- m/ d
17. {
    
18. CString   s;
    0 N! B* ~, P2 d; l( i5 A2 Q
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    1 P; J% `5 `  j6 p/ g, u. R) c
23. }
    
24. 
    # ?* S. s' ?( n2 r4 ~
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    , W6 K* F5 i  D/ w7 F! k# g3 ~
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    0 @: N* R, H0 L2 A& a3 Y& h, H
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    , ], t, n, m+ J  }
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    " I! [% a, L) z/ Z7 W0 m! B) y
38. 
    5 v0 X! N1 p' e& s; ~
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    9 k+ V* b' ]+ p6 p: q5 F
42. 
    * v5 u$ S9 n% p; P
43. uint32   ip   =   inet_addr(CStringA(addr));
    ; X- j% p% G, h/ v# g4 c
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
      i$ B7 Q! e) W& D' w1 J
46. sockaddr.sin_family   =   AF_INET;
    6 A  T/ R) `7 @5 {: }' w
47. sockaddr.sin_port   =   htons(port);
    1 b; ]$ p1 G. `8 W
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    7 m" ~, k& @7 i# Y4 a7 p+ W
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    3 w2 Y/ a* B6 i8 y: O
53. Sleep(20);
    1 a; }! k7 g5 D
54. int   n   =   send(s,   buffer,   length,   0);
    * U  G: X2 F$ p: V3 ?7 d8 w
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    : S% N/ \/ u9 J1 \: c" K
57. closesocket(s);
    7 S- {) }& e8 }$ }( K; J
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    1 d4 U, I& p7 X& X
61. response   =   CString(CStringA(buffer,   rlen));
    6 p' i* W* ^" D# H- e
62. 
    
63. return   true;
    , A5 S2 T1 e/ _! {5 A
64. }
    
65. 
    # A6 v2 K+ [/ ~+ Y, G0 x7 [; }
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    2 Y2 g( u7 e# W0 z( j8 l
68. char   buffer[10240];
      y6 ^+ U9 W( ^1 K) q
69. 
    4 l* \; o  B( V& s" P  m; e+ k
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    $ b( W  E6 Q( F6 m( H
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    0 H( c; ^' U$ t  H" f
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    6 z" g- [8 }# ~7 r
84. {
    ' @# w$ p7 l! h8 _4 b  L6 J2 t
85. int   pos   =   0;
    , X4 L7 Z/ m5 }' R0 m
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    " G$ ~4 s6 p: `, |1 `
88. 
    
89. result   =   response;
    5 W# o( K+ \" f+ \6 e) x
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    ! b  k* m9 y( {& u
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    : w$ f6 c0 Z4 {4 s' \
97. }
    $ z2 E, L6 S6 w, J, j+ V
98. 
    , J( c2 n0 h5 i  ]# k/ F% T
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    7 \' a/ E  _- r, F0 w
100. {
     3 u6 F/ w8 X; K! w; O
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     2 K: w3 |3 b# W% y6 s
106. if   (posStart <0)   return   CString();
     , W6 M" [$ X) T! Y
107. 
     3 |) z1 ?, W+ M5 n" T
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     8 h3 [. {8 B) j+ m: s% e8 C: A
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     ! o9 f& R4 Y) I" |5 o
116. {
     
117. m_uLocalIP   =   0;
     8 o& o" v9 ^7 @# @6 F0 i2 {* t
118. isSearched   =   false;
     
119. }
     
120. 
     ; _  Q6 Y0 T4 d
121. MyUPnP::~MyUPnP()
     
122. {
     ' s4 k9 e9 A* b# k! k+ n6 s# u
123. UPNPNAT_MAPPING   search;
     " Z" Z! k' A$ v2 C. |$ w. ~
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     * U3 a- C9 C& x  D
125. while(pos){
     ( B) I0 B7 s) b8 W0 e
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     . e. c) d9 y' K8 M
129. 
     
130. m_Mappings.RemoveAll();
     & ^, e0 e/ v/ g9 ^. G
131. }
     
132. 
     
133. 
     0 X: y5 u2 W8 Q- e- U
134. bool   MyUPnP::InternalSearch(int   version)
     6 r- a/ k& g6 `' ^( F( ]7 @$ `
135. {
     % U2 c2 }) R4 _, \8 m$ f
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     * T  l3 m. B' J( N4 V3 A! e) C
138. 
     
139. #define   NUMBEROFDEVICES 2
     2 I" u3 O6 o, b, L
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
       q0 q8 G% _  P( s' N+ k% b
142. {UPNPPORTMAP0,   _T( "service ")},
     8 |& a1 |5 O+ z" y, A, v$ O
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     . g, y# l, l0 \% U% j
144. };
     . O6 `- Q' {( Z4 P7 P2 v3 e) Q9 w
145. 
     5 U3 }, }, N+ Y$ V
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     1 w- _5 U  f- w9 `
152. if   (!(i%100))   {
     9 k" z! h; [0 ~' l* V
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     1 i" p! S: V7 E0 v$ N( G1 Y3 x
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     . O& U% b; s) J* O. S  p  {* e0 X
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 "),
     : J3 W; g  _2 X' E0 Q
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     : z+ b- }# y! Q) r
159. }
     / b' O/ B' w, Q0 b; p
160. }
     
161. 
     
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     3 H0 g% @! W" @. q
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     & ]9 f5 s" H. K! k
166. if   (rlen   <=   0)   continue;
     ! T9 J' b. u8 ~: k4 x# p) ?
167. closesocket(s);
     2 a$ A! m8 C0 q8 ^4 o
168. 
     2 m7 L/ n6 v% m, ~+ x
169. CString   response   =   CString(CStringA(buffer,   rlen));
     ! Y* }7 w/ p$ k7 N) Z* f+ \
170. CString   result;
     " V" ^5 n: n& u: m
171. if   (!parseHTTPResponse(response,   result))   return   false;
     0 @5 f9 W8 `# \0 ?2 z  k7 K
172. 
     % a0 K7 `9 Z3 ?1 r7 X6 F, S- e2 u
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     : ]6 C2 @/ G- f
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     3 }1 G9 e! B  w3 C. j
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     & m1 n/ `/ n3 t* i
178. if   (line.IsEmpty())   return   false;
     & a1 o, Q0 J  G4 q7 G
179. CString   name   =   line.Mid(0,   9);
     : i: {" p. i% z# e( F
180. name.MakeUpper();
     ! m; A2 {$ \" z  |  n! R3 o% `
181. if   (name   ==   _T( "LOCATION: "))   {
     8 n9 |2 {1 J$ ]6 c6 h0 F0 @
182. line.Delete(0,   9);
     ) L" i3 a$ t# V# W# R, V. X3 v3 t9 V
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     ( Z" S; T2 @( _! {, k
187. }
     
188. }
     1 d, l! r0 c, m: r, {
189. }
     8 F2 I8 O( c# J3 T6 z, S% [
190. }
     
191. closesocket(s);
     * i0 {: d0 P: n! O8 v
192. 
     
193. return   false;
     6 a& n; M# ?3 Y( s. ?  \& k
194. }
     6 {: b# f+ ~+ Z6 t# x

复制代码

天云上人

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

+ r7 Y. s5 F3 ?0 v& }+ @# E; P6 g: M
///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
  j) \# U$ d& Q2 M7 v
6 y/ R. n3 h* e# Q8 c9 l
/ F/ Z! y) r' ^/ k- u#pragma once
#include <exception>
* g8 H7 A5 c" H  V, a6 |% x

  enum TRISTATE{
' C4 s7 m( }( i4 V) P8 m5 V# y        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
};


enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
0 ^- z4 x) {6 k: u0 i7 ~        UPNP_IMPL_MINIUPNPLIB,
0 d* J  p& y: P4 x        UPNP_IMPL_NONE /*last*/
};

/ Y: K% [9 _5 b# d+ Y

- ]2 F0 c# _2 R/ w5 C4 }. j
8 e: h1 b+ U$ x0 A: C1 k  ~class CUPnPImpl
5 O4 D* [0 V% N5 D6 }5 W( G, T{
public:
" y: h0 V+ ^% q/ ^" x        CUPnPImpl();
: {+ [5 I  H& H8 e+ }        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
& D! a9 X" O* ~# j8 [6 A* L7 H        enum {
                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
        };

$ [; D% C$ }9 N# c) [2 K) L  I
( C; ~: f: j" {; K6 O        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
. i  ^5 o+ @/ D        virtual bool        CheckAndRefresh() = 0;
, n, y' d$ m! [) w        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
1 }. m8 H) K) c# K        virtual int                GetImplementationID() = 0;
7 a' C8 C# ~; i' n3 I$ s3 o       
. z/ S2 P: c2 M8 X        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

/ q3 U$ v1 j4 ^/ C9 p
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
/ W. \( _" Z" N) B        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
  S4 s' B2 H& f. c! R# ~) ]        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
" ^$ }" M: s6 N
2 `2 w/ D8 ?) S, [( v7 _4 I( X
// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
; v/ |% @2 m4 T0 w6 t% y. e7 s        uint16                                m_nUDPPort;
3 S. c% z- o; ^9 P( j1 Q5 C! M7 Y        uint16                                m_nTCPPort;
* u# V) Q) {$ p        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;

% T0 K" H" T. }: R! D
: @, c5 c- I9 f& t, Rprivate:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;


};
, G$ Z7 h5 c' D
) N5 X! D$ Y1 ~6 I
// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
. y" F- V0 d# p3 @9 b6 L{
public:
8 F/ r  L9 c. _8 q2 a: D        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
5 o' S4 }* P6 N. l1 m2 L5 b        virtual void        StopAsyncFind()                                                                                        { }
; X; X1 [1 I: a3 ~$ w% [        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
# F7 B/ m# @# u# h" S5 u3 S

' s) Z$ K! |. r& Q: p/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类

3 y) L8 e, M8 e! Y$ f4 s
#pragma once
#pragma warning( disable: 4355 )
9 h; h! [2 ]9 n! @! w$ M( }
: G% j8 w3 D( @3 C5 F
3 ]8 x$ L+ H" k; N* l#include "UPnPImpl.h"
4 f5 \& o& m! A/ |#include <upnp.h>
) l2 t' C  B$ `% }- ?! W7 \2 Y#include <iphlpapi.h>
2 N& Z6 j4 h& `0 R8 H* Y+ C6 G#include <comdef.h>
#include <winsvc.h>

1 v4 ?" F. u1 c& ], f: n  h+ Y
#include <vector>
  `/ |! T( H* {3 }$ f' f& I#include <exception>
#include <functional>
; D- P3 c9 B! p1 `' U- s/ U



* D9 A' B! ~# Otypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
5 ]( ~( |  `* Y2 v" a( Btypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
1 i# k- O" p8 k& ptypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
/ Q( Q' `1 }" c6 @5 w7 htypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

! k6 l. `- T5 t& b; N* l2 j0 I
" N: N& S: l0 `$ C6 m  Ntypedef DWORD (WINAPI* TGetBestInterface) (
% E  J6 h& y3 Q0 p  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
: s7 M4 ?& t* V& p);
8 k& x- ~* w& Y9 H) y6 J- ]/ z# R! A

typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
6 l# A' ~) F  K  PULONG pdwSize,
" S' \$ ^( W& u  D  BOOL bOrder
& T5 _. s; ~! z);

% I* K, n' l6 Y' e& v5 @0 K
* k2 s2 [" t- Stypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);

5 ~2 g  i) }1 G6 _9 Z
CString translateUPnPResult(HRESULT hr);
( b& N% O( n2 Y, [7 JHRESULT UPnPMessage(HRESULT hr);

4 _& X, S! J( ]( s% n8 U
class CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
) U% Q) \) S, N  P# }        friend class CServiceCallback;
: o4 f3 E3 \/ N( h7 S. _6 U0 ~  m  s' j// Construction
public:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
8 U- a5 T, X4 ?- \  a+ U7 k& R

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
. [# v0 U7 f7 f) p' w: K) r/ U( E        virtual void        StopAsyncFind();
- N3 }4 r/ t* p5 d% n' U, g        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
# S* ~5 [0 G- b1 o7 V* z2 z

3 T8 j2 _2 J. w        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
) p) h8 k% D; N0 P. L        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
* Z! V( v; I/ e) w7 S* r3 [5 L5 d        virtual bool        CheckAndRefresh()                                                                                { return false; };

! x, u6 y* y/ B- l, u" e# L- C
protected:
$ L+ d# m- `0 T8 e( o        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
' [9 x( r- y+ @5 r' `$ E        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
& F6 T3 P! j0 l7 C# m; N, R6 H# W) z" Z        void        RemoveDevice(CComBSTR bsUDN);
0 E* z: M+ C4 ~: O        bool        OnSearchComplete();
8 s- t; `4 @6 d7 [: B        void        Init();

- X$ U  X" D6 M; j2 l
        inline bool IsAsyncFindRunning()
7 u5 R! ?6 |5 O( I. i6 G8 g        {
4 G# ~7 Q4 I5 K4 [                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
3 a& o% \8 D* ~) L                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
( h1 K8 y: C5 J, [                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
7 H; T0 n; D! D5 _/ t# \5 L) v                }
+ z2 D0 H5 p5 n) U                return m_bAsyncFindRunning;
8 ]& L7 d2 l, _2 A        }
+ e) ?& |4 K: y& o9 C' \. ?; @
, |8 a( h! K" U, V4 G9 b' ]0 i
        TRISTATE                        m_bUPnPDeviceConnected;
! Z3 O7 B' l" K. _/ e# ?

// Implementation
- w9 M: M; h: U8 P# l* \8 T! l" K        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
- P. Y! _) q0 b3 G7 _4 B# D; I, W& e        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*);
1 G* z7 Y2 g( g5 H* M0 c# b2 J        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
1 v7 F% g1 N' k7 U5 ?

8 A4 |  X: Y% q7 ?9 }4 A$ i8 n% x        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
5 l3 L* w3 M( |0 \$ u        TGetIfEntry                                m_pfGetIfEntry;
# m" T" J- E- e4 [

) A7 _1 h" }7 P/ _3 W" I        static FinderPointer CreateFinderInstance();
+ v( [7 P+ w" \3 [6 w8 z        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
- i  L8 S* U3 J5 i# V                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
" J' @5 Y0 }, g6 s- b$ [5 D/ j- v3 I                result_type operator()(argument_type device) const
& X8 q6 }+ D' Z# D                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
' j5 K' ~/ n8 z- e5 ^0 M& l! k+ M
7 F  |9 T+ O: g  a8 R
                        if ( FAILED( hr ) )
. [  F5 W( k/ ~/ E( [8 a                                return UPnPMessage( hr ), false;
, \+ c% F. O% T$ w. ~

                        return wcscmp( deviceName.m_str, m_udn ) == 0;
$ P5 A4 _0 C6 ~, V                }
                CComBSTR m_udn;
+ i5 M( [, I! C# ^3 l2 @7 ]        };
% w% u; W5 x4 ?2 M       
9 q4 y9 H# G* j        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
1 r1 q+ }) s% A. i# o* X        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
3 e  l5 k4 k6 ]4 k! S1 Y6 I+ R        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
, V# |4 A& Y; a. G9 I        void        StopUPnPService();
7 I( n2 D- @3 j8 ^3 w% O2 i  `

+ E& W4 c: H# p* p+ s/ |        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
% e, h7 B8 N: e+ p7 X- l0 Z        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);
        CString        GetLocalRoutableIP(ServicePointer pService);


// Private members
. k+ ]$ `: j7 e# s: b4 Mprivate:
4 y' b! ?$ `* X* C        DWORD        m_tLastEvent;        // When the last event was received?
6 G& l. d  i8 N7 b) B  O        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
2 J0 Z' ]! l, ^; b" o0 `        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;


8 ~1 t5 @, j$ X8 K( t        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
% `  y' H  a2 C  w# W( ^  f        bool        m_bPortIsFree;
        CString m_sLocalIP;
+ T9 x& ^' U2 f( V) v/ l& ]        CString m_sExternalIP;
8 D- H) |! X% ^2 v* b        bool        m_bADSL;                // Is the device ADSL?
2 E# v2 ~' s& Y. l* ]& x        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
8 c% j( ]+ `8 s+ X3 X8 I6 K        bool        m_bInited;
" H# }! W+ x: K# s$ j4 {        bool        m_bAsyncFindRunning;
9 D7 [3 |. w- P+ _9 z8 p! f5 F        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
3 q1 e+ x, s" o& D7 d! b, i  [        bool        m_bSecondTry;
2 M- _0 W( F( A  d( w, e3 h        bool        m_bServiceStartedByEmule;
" \' c+ r( F8 c5 `' t; Z( g: f* z        bool        m_bDisableWANIPSetup;
1 p& m6 @4 l8 v$ G1 [. W3 z9 z        bool        m_bDisableWANPPPSetup;
4 {" Z6 Y8 W+ Z  L

+ K. @6 z& D" S) _/ {};


5 B& O$ C! H2 _5 O2 q9 U1 Q// DeviceFinder Callback
, a- i! }' X. H& |( Q+ L/ e5 {2 w3 Mclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
public:
. o: A" g* |6 Z- E- ^6 ~        CDeviceFinderCallback(CUPnPImplWinServ& instance)
: W4 m! f) @6 Q' _2 Y& l5 G8 N# Z                : m_instance( instance )
/ ~/ |' }) l4 n        { m_lRefCount = 0; }


+ o: O. p' d; w4 j( x   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
# I9 c( F( u2 D) F6 S4 z# f9 ^1 \   STDMETHODIMP_(ULONG) AddRef();
* h4 i9 ]) q+ u4 r$ Z- k  `   STDMETHODIMP_(ULONG) Release();
! d1 b7 X9 Z0 l  R& D; Y" }

& \8 y" {! }6 r; {4 t// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
* L  J* C" O; Z5 e6 C+ `: x        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
+ R* v3 P9 x  `  F0 ^

private:
9 h( W) x: y" W* f: G        CUPnPImplWinServ& m_instance;
8 v1 r- u6 l3 S$ y  @        LONG m_lRefCount;
- Y0 K) J4 w- f. s, Q7 Q) R  \- Y% x};


// Service Callback
- H& F9 r5 G$ ]) K3 N6 S$ k8 ~class CServiceCallback
        : public IUPnPServiceCallback
{
public:
& B. t# }5 C$ B6 s0 V/ I' a  S8 [        CServiceCallback(CUPnPImplWinServ& instance)
4 U7 C+ }4 w) u3 I+ d                : m_instance( instance )
        { m_lRefCount = 0; }
   
6 \" a; D4 Z  ^  t/ U1 D# r; u; V: Z   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
( o6 u1 P& n& [8 x/ Z9 l

// implementation
private:
* O. [- S3 g3 X' |1 Y% P& |9 T        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

) f3 ^1 e1 y8 c
5 |5 B; A+ b9 U0 c7 X/ [private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};


# |. ~: s$ H+ S+ v% g- L/////////////////////////////////////////////////

: C/ v' ?) {4 Q, l
使用时只需要使用抽象类的接口。
0 ^$ J& n. K5 O. ]  M+ X8 CCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
2 o1 _3 J( j# U9 F4 JCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.