UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   & I" L1 c- ^( U9 r3 F
7. 
   6 n, }) n: c) {
8. class   MyUPnP
   $ J1 z' E5 n( D& i  O' v
9. {
   
10. public:
    ' i( c- t# g( z  |
11. typedef   enum{
    & W+ l3 h6 I, Y: c- _
12. UNAT_OK, //   Successfull
    1 l1 S& j4 Q1 N, ?5 H# n7 X
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    $ a5 N6 L0 _' X6 l$ m: l3 J
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    
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
    / K* R( I6 [5 D
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    & M1 I. [- P3 W4 |7 B
20. UNAT_TCP, //   TCP   Protocol
    + N; c/ U% G5 V& \
21. UNAT_UDP //   UDP   Protocol
    $ C, P  }+ _0 A" u$ ?3 `
22. }   UPNPNAT_PROTOCOL;
    ! q- u  a: c: \4 x: J
23. 
    
24. typedef   struct{
    2 ]9 q& s! D4 M( _8 R. [
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    ) C: u, u" W/ S" y( P0 Q1 y
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. 
    $ U5 \3 \5 Q" F+ k9 r' G; x
31. MyUPnP();
    / g- s2 v. I) ]& r7 I" i( _; Y1 z
32. ~MyUPnP();
    , l$ V4 b& E; n
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    - q5 Z- W; Z8 X3 y2 U# r& O
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    - q2 _, q- N! n2 |1 W
36. void   clearNATPortMapping();
    ; K1 i9 @4 g1 b9 \
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    % ]4 g+ U+ s; R$ }' K- D
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    ! O  p: X  _* x' `- A' L" ]  n
42. 
    4 x+ r. ~' U5 e2 V
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    " C+ M+ |1 \3 l! F- B  d4 U
46. 
    : f  K7 [1 K4 v1 L4 Y" n
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    $ Y2 |: K$ @/ ?) K  X/ o
49. bool   deletePortmap(int   eport,   const   CString&   type);
    9 |8 G$ T! u3 ^- p- x
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    % U& u1 d# L" N! t' q
52. 
    
53. bool Search(int   version=1);
    ! S/ Z7 f. _4 Y& h/ w8 @7 v3 y
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    : b( p7 V) g& y# M- m. x
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    " h/ K5 e) q/ M: p
57. 
    8 l, C  l1 T& j3 k8 w3 t
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    # t9 x6 O$ n/ P, T! ~% u* {
60. CString m_devicename;
    + s. ]! k+ h7 ?. ?
61. CString m_name;
    - n# t# i' Z2 D& @* _
62. CString m_description;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    # F, B% }: x! u+ d% _$ x8 r5 M% Q
66. CString m_modelname;
    
67. int m_version;
    
68. 
    6 D" v+ w/ x- ~5 m- H4 f, Q5 \
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    9 C& Q; K  k- X, g: U  ?. ?; `+ S
71. 
    
72. CString m_slocalIP;
    * e9 \# g/ b; r7 o* j* Y- n! s& l
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    5 ~8 k2 x( F7 E6 C$ [' u$ K
75. 
    
76. bool isSearched;
    ; p% L; |4 P2 W) M1 ?
77. };
    4 E8 J. G" S- q  _/ y3 V
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   4 k2 x8 o, ?( z8 Z
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   * ]( P6 m/ N- \$ j# M- m
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   4 T4 I# R; {# V. L; Z0 h5 u. D
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    5 @; O" G: @, C" I# r1 b: e/ ?
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    1 o: b$ B- `+ N- D) l
13. static   const   int UPNPPORT   =   1900;
    , r  r% {2 m0 w3 m+ v9 L
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    2 l# U2 X) C1 q
19. 
    5 _* f$ ~& l" o+ _: S5 d+ U
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    
23. }
    
24. 
    + c9 j  \$ U. {6 O+ b
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    $ h7 F% ?/ z, k& R, o9 g1 \
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    6 o( G6 V( G- f7 i& d1 k1 u3 R
28. }
    - I! Q2 t  u$ ^& G4 l  Q
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    % O; I1 N( m! C; O/ C; ^% `
31. {
    ) L4 z- x8 @: m5 `  ~. ]) G2 Y
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    $ u# C8 @# S* E5 h$ V* n5 a; H
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    : _* e; P1 F  \4 W; c
36. {
    : u. I, b8 u$ y' s) `) s
37. char   buffer[10240];
    $ h% c3 C* K+ B5 U7 X: z* Q0 _$ J
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    , A9 D2 J/ m# v9 K
42. 
    ' ^7 i- P$ \! E0 e! F
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
      K: d- L" `0 e- f- C
46. sockaddr.sin_family   =   AF_INET;
    6 R, Y8 m- {! G' M
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    6 W- l# S+ N8 [% \, q/ ?, y% U
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    9 E6 z+ B  D* l$ {+ S/ E" b4 T# g
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    0 g6 l' ]( m) {. [7 C
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    , `+ i9 U- A$ d6 B1 f7 E# X
55. Sleep(100);
    / D8 Z) a6 |) j! p5 W
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    $ I/ D  p/ F2 M2 R" a( B
57. closesocket(s);
    - z* G& m; m2 k4 w9 ~
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    9 ?/ _  L) ?) A+ Y; P5 n
59. if   (!rlen)   return   false;
    
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    , Q& I. N4 I8 D4 _' s& e- Z* W
63. return   true;
    
64. }
    , {- A% h3 A7 z) _  V2 T
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    , x6 z7 H, q2 ^  t+ a) v) }) b) g* m
69. 
    
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    ! l0 D) f6 X' H  c/ P( r9 J5 S4 \
72. strcpy(buffer,   (const   char*)sa);
    % |) \- I/ h; a6 g) F# O  X+ T
73. 
    ) J0 U% m" `' j2 N: d7 ?0 d
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    + c3 V" r- B8 s
77. sockaddr.sin_port   =   htons(port);
    9 x: O1 j9 j  P2 B  b5 W! o, @! |) H
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    7 N$ D4 u  ~/ v" k6 I, [( |
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ! M4 X8 ^; N) ~; W
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    3 M% ~4 K$ X' G* R* T0 F
84. {
    
85. int   pos   =   0;
    % O+ }. `. n+ U4 v; \" Y8 R3 g
86. 
    ) W& \5 s) }# h  d; `
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    # I  L% i0 m3 U% p
89. result   =   response;
    / G. }" ^& c% j; L9 W* d0 \! v8 c
90. result.Delete(0,   pos);
    ( B; _& i( k+ ^- ]
91. 
    ) u- W8 f; K& \# g% s; t
92. pos   =   0;
    % I# a) q3 g! W6 ^  b- P
93. status.Tokenize(_T( "   "),   pos);
    2 Z2 c( W* I% H% H0 M
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    3 l- L6 m' i( y6 y
96. return   true;
    
97. }
    
98. 
    ) E% A" L; `% |
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     4 O4 i$ ^" [5 B* Z3 M0 `6 I
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     ! t' j) E7 Y- e2 R
104. 
     " V6 n! T+ Y, \& E7 m2 i) D8 F9 Y
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     0 }8 H! U) \) L( V. l$ F
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     + ?5 c9 [+ s( b5 P. e7 K
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     1 h0 j1 p  V5 h# n
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     3 n7 W4 b- V3 {$ q$ n6 b4 {- Y
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     & o8 H# H5 B4 i0 Q  m
116. {
     
117. m_uLocalIP   =   0;
     * q: L4 g( q' m& e. g9 r( z) d8 G
118. isSearched   =   false;
     6 n/ w: Y) Y6 e( b1 n( d
119. }
     
120. 
     ' M- p* \3 b+ O; P6 a1 r
121. MyUPnP::~MyUPnP()
     6 y' X8 N4 m% ?
122. {
     
123. UPNPNAT_MAPPING   search;
     & g6 B7 T0 W6 r& U" r
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     8 V7 O1 M/ Z* Z- t+ ?8 }3 E, ]3 o
127. RemoveNATPortMapping(search,   false);
     6 z& t+ v2 y7 ~& @  `  P/ {) T
128. }
     ' l7 f' c$ y0 ^, a0 a) B
129. 
     
130. m_Mappings.RemoveAll();
     
131. }
     6 q) @1 y, h) L3 G; u" ]$ d
132. 
     # x& M- R' t: r+ c' v4 Y' U8 g
133. 
     2 o6 X1 q0 M4 `0 Y1 _
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     4 w3 r$ }! N+ H1 w$ O" S
142. {UPNPPORTMAP0,   _T( "service ")},
     , d  o7 S# i) b) v- i$ z
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     ( R' m8 `% I2 O
144. };
     $ D0 f3 D9 N$ f6 c0 {9 j
145. 
     1 N" f/ Z# ^7 ?' W$ ^
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     $ z" W" g6 I1 r" U: Q5 G
149. 
     
150. int   rlen   =   0;
     2 \# \4 U5 R7 T% ^# Z/ e" `0 S1 L
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     9 ~5 `3 Z. v( H! w3 {
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);
     0 k8 |5 r# g2 E3 V
155. CString   request;
     6 p& T; z/ t2 B3 G, b- l& a- T
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);
     5 @/ [& z' L9 m# p0 e  B
159. }
     9 q9 Z& A4 }( ]8 G
160. }
     : @* a% \1 M5 u1 y/ V! Q) s
161. 
     , O& \. M5 S  N; G
162. Sleep(10);
     
163. 
     $ X3 ], b/ f2 C6 Y2 [" j1 z' `
164. char   buffer[10240];
     ! D* K$ z9 A( f+ Z
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     ) o0 c/ z7 m/ ^5 \
166. if   (rlen   <=   0)   continue;
     : \- I+ k7 A: j# x2 L( |
167. closesocket(s);
     
168. 
     - a8 ?! c  m  e* S, O2 m7 G  G
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     : K2 u$ M: H- s+ b  s
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);
     4 J% {9 n. R, N
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     & p% M- C7 V6 n9 y* E4 A
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();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     ' ^; K% X: u0 t6 m
182. line.Delete(0,   9);
     - i% [0 p( M' o3 |! Y+ o
183. m_description   =   line;
     
184. m_description.Trim();
     ) k9 \# L/ b  N4 J' b& S
185. return   GetDescription();
     
186. }
     ' B! R. [$ }  V2 M4 V' Z4 ?# Y: b
187. }
     # L7 ~3 n  p7 c0 E+ K
188. }
     . v+ o  V% w7 I
189. }
     # T- B' K# d# `' Z& Q9 M+ M
190. }
     ! F' k( p  h7 _$ ]+ ^, P" V
191. closesocket(s);
       c- _  ^" o* v4 L5 @
192. 
     
193. return   false;
     / G$ p/ V4 ^! d. x7 ]7 [# N
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
8 l4 m3 F1 Z* F1 {. \0 O1 u

( x7 M" O% t+ u+ w3 ]; I, i2 H///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
! j# K+ M$ X" |1 Y  X
  V( G" S- w# I8 c& \
#pragma once
#include <exception>


  enum TRISTATE{
6 r# O& K- v- G7 f) y* u        TRIS_FALSE,
        TRIS_UNKNOWN,
# Z. `8 o# b/ Z* Q7 G8 {. ~        TRIS_TRUE
};
2 j$ a0 S8 F& A5 u5 o
% Y- K: d4 p9 U# C/ o
5 z6 v5 a+ z8 r( O6 \$ Zenum UPNP_IMPLEMENTATION{
; ~& q3 @  ]- h5 U6 Z$ N, u+ p, g2 {        UPNP_IMPL_WINDOWSERVICE = 0,
. c7 z# |9 ^4 J9 s! `% N5 Q' W        UPNP_IMPL_MINIUPNPLIB,
0 V+ s% \2 o" f( J, B        UPNP_IMPL_NONE /*last*/
};


* f0 ~7 ~" _! n8 _* c' {* M/ E" S
. i. D) E* _# }/ H9 s, @
class CUPnPImpl
{
; ~3 t( s- P1 W+ fpublic:
6 y6 g. J! q9 j1 L: p8 A( q        CUPnPImpl();
        virtual ~CUPnPImpl();
0 B* O& L: ]; G# p2 x% y8 o        struct UPnPError : std::exception {};
        enum {
* ^% e' ?  e6 f8 c4 s$ B6 H                UPNP_OK,
3 Q5 U# W/ K9 V4 N5 v: Y3 E                UPNP_FAILED,
; D" U' |" i, Z+ }                UPNP_TIMEOUT
        };
7 \  J0 R7 V: I1 W4 {

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
$ P* y. ]) e( `. \& D* p9 f" ^        virtual bool        CheckAndRefresh() = 0;
4 `& r" n! S7 q9 F8 z        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
2 i: T1 s% D0 [% A$ Y: C; r        virtual int                GetImplementationID() = 0;
  |* A$ J! E2 s0 ^5 a8 C       
* o, U9 F5 E) O( q" b1 o4 Q( R        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
: O" ]$ ~7 j+ A7 O- R

        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
3 C; U- @; B7 _7 B. X3 Z        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

! T7 X: X! W. n4 S6 i
// Implementation
3 _) d7 m9 U" Y9 m8 Aprotected:
& z, M- ]- c2 D/ B; ^        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
7 }/ Q/ ^. y  [7 D) d8 [8 H        uint16                                m_nUDPPort;
, d4 ?0 Q( U: t& Y: B+ O* K0 G0 T        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
& T7 z( ]% q5 j$ s. A        bool                                m_bCheckAndRefresh;
4 k  L" Y1 W0 T4 D9 T
) S2 o5 w- E6 `0 |9 i* H
9 _3 C6 x) q/ Y: Aprivate:
! |" f' @4 J: ]* ?# Z( V$ z        HWND        m_hResultMessageWindow;
: G1 {1 I- U! y2 N1 l3 Z/ v; ?        UINT        m_nResultMessageID;


};
2 F; X8 }" |1 Q3 O& t+ z% n

// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
$ s/ h, s. B: j; U( `1 \1 Kpublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
# }0 _0 W4 v* _+ g2 V        virtual bool        CheckAndRefresh()                                                                                { return false; }
7 k4 M) [) D. ]        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
, C% E2 C* e, I5 N4 n0 l3 L3 Q        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
9 a% U+ {3 z8 ~1 i1 f
( F! q$ v' V! T, R! w
/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
- @+ f8 ^( |3 O2 h, x! O2 f
2 ]8 i# F$ C4 W
#pragma once
  J6 @7 P' R! D. K: ^7 R" T/ y#pragma warning( disable: 4355 )


* ]6 e: z5 P- W2 T7 x! U% |$ n( R#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>
: @) p" |$ j6 s  F

7 Y' h* j; j) l#include <vector>
#include <exception>
4 q% r  Y' l* H' P) g' v3 k6 T#include <functional>
# y, ?! d: x$ c' f4 w5 g
! t( K2 Q( S4 r, @

$ _. ]+ F  B& _2 G( t6 s
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
9 H7 h; v! m$ I/ [( w& Otypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
2 J  `$ h& V* Etypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
$ Y- _! ~$ `+ \' P" Xtypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

' D! e( o7 a' o; L& w
typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
6 |# `' X2 `7 h  G  PDWORD pdwBestIfIndex
: Q9 E  Y  S+ D* G6 P" I+ o" ?9 Q);

) N8 y$ m! I: Z7 U9 u& G9 L
typedef DWORD (WINAPI* TGetIpAddrTable) (
6 `! b$ M6 J' t  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
  X: d- `' G% n! x0 l" o);
5 W8 i& \" U$ m2 P# u2 Y$ z. \6 r

typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);

: g6 }6 L' N! H: N) @+ Y
) f1 ?$ C  v5 s8 A0 mCString translateUPnPResult(HRESULT hr);
4 }' J% |/ u# E3 H( n5 q1 YHRESULT UPnPMessage(HRESULT hr);
$ v" s/ o7 r+ v0 B
' e0 n! R- y0 B, T
class CUPnPImplWinServ: public CUPnPImpl
8 _  J0 ^, I" o1 s5 l) y" f{
& R/ a1 ?; ?0 \) w% y( B        friend class CDeviceFinderCallback;
! c4 T/ }% I( j) {8 o) P0 n; c        friend class CServiceCallback;
// Construction
& e9 q4 U7 M' C9 f! \, cpublic:
2 H, z+ O" i$ r: q5 w        virtual ~CUPnPImplWinServ();
9 x% I, d8 p+ x        CUPnPImplWinServ();

; s+ N, g5 ~6 F9 u+ t5 B
+ [0 u4 @" T! P0 s5 H+ f        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
) J; U  F; @0 {! I5 w5 t  z        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
# t% Y6 ?; ?% G+ `0 z! _) a3 \        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
- q1 V" }% `; N: [4 E6 n* S

1 }! X6 Z% W% m: ~% s0 _        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
, b6 `' l% d8 u        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
3 G& T7 K, R, u! y- p

protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
* U* w8 _& S' Q! k        void        Init();

# h+ o9 ^8 G3 M8 k! T; q8 D" |
        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
# A6 y& a5 x- K  n4 R/ A                {
7 P( B  G! A, {  L1 a                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
7 a( D. e) W- d                MSG msg;
6 D4 P, m' L4 F% ?- u                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
. y& X; J* {7 \4 s2 Q, I                {
                        TranslateMessage( &msg );
; [) m" O$ Q4 }/ t                        DispatchMessage( &msg );
                }
( d- L; B+ V' y, q- V                return m_bAsyncFindRunning;
' }. i7 L8 [  a" j, {, q2 D        }

5 l4 i/ W6 k" k6 M4 y& ^
6 Q9 t. U$ b! t        TRISTATE                        m_bUPnPDeviceConnected;

. o& G0 I: h* b# p1 g
// Implementation
# w# \+ v1 H/ a! D9 e* u$ v        // API functions
% Q, I/ M8 I2 ?& i5 E) ^- Q& |        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        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);
* Z3 N; A* E* f, |% \        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


& w) Y- [, S- U7 Q        TGetBestInterface                m_pfGetBestInterface;
* h3 N8 b% p. n0 D        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;

8 n8 ]3 ?8 }; T  R# l, n% K: x
; N$ {# b7 \' X6 o        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
. o8 ^; z8 E* C        {
: y/ `/ ~) T( q1 ~% Z! [                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
. V9 _6 |1 B) c& C- c                {
! P1 n; T+ {: S7 }  P                        CComBSTR deviceName;
. I% I! a) E+ v                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
' v3 b8 b3 m3 E) J* ]

2 M; J* K6 w# t1 D+ o                        if ( FAILED( hr ) )
4 o6 _( |1 ^0 }5 o1 @                                return UPnPMessage( hr ), false;
, t4 M. z' g( E/ v5 F
+ S' a$ k& ?+ D- I
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
% h4 u  R- u# U, p( y# C                }
                CComBSTR m_udn;
        };
- G& Z. k; _  N8 ^( G$ r       
8 S& Z% \9 O' M8 T        void        ProcessAsyncFind(CComBSTR bsSearchType);
. S5 G+ w6 J4 A1 K% s        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
* |. N, R: r2 K, `% [& L  A* s        void        DeleteExistingPortMappings(ServicePointer pService);
9 V, O5 W2 l" }" @        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
5 d1 z8 Z/ ?! ?3 i/ x2 a3 p9 y- U! t
7 |% Z# ?7 ^$ F2 ~
        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
/ s' j% l: ~3 y' d        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
5 u' L1 C9 A8 p        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);
  ]* F6 v/ F9 i' p2 l# }" D

0 @8 L7 G9 O6 N0 \, X0 X3 ~% d// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
0 k8 R+ ^6 n& u: o        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
% z9 f5 p" q! Y" Y% F
& L5 i* ^5 m- s' x% `' K2 m; `; O
# @2 D" T$ _+ C2 J3 w        LONG        m_nAsyncFindHandle;
' f- R0 c# Q6 g: H% _        bool        m_bCOM;
2 `, h/ N+ h% ~/ R        bool        m_bPortIsFree;
        CString m_sLocalIP;
4 p! ]9 f3 q1 p) ^  r7 W        CString m_sExternalIP;
% _+ b: \' F) O. v# S# D        bool        m_bADSL;                // Is the device ADSL?
4 M. h) F$ Z6 C8 U        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
/ Z7 E+ S2 _0 K* i% d. F        bool        m_bInited;
        bool        m_bAsyncFindRunning;
# X- t( o( Q" H, W        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
& @# m  G- p1 X8 J& S        bool        m_bDisableWANPPPSetup;
4 L0 x1 l: z5 k! Q# z; W; T0 S, H/ b; ]

: X0 `4 i5 y6 }; e# A};
3 {# V0 r, I6 m: Y9 A4 Y

// DeviceFinder Callback
( X+ W$ v5 T1 n$ S1 x2 L9 A( Iclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
5 U& x0 |& \/ ~0 _2 Npublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
) j% G" x/ k# h: ~3 P                : m_instance( instance )
        { m_lRefCount = 0; }

% D" \/ x/ \7 d' v0 e
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
' _1 ~. R. J9 k   STDMETHODIMP_(ULONG) AddRef();
; G8 d+ S" S; c) q6 q! }' `   STDMETHODIMP_(ULONG) Release();


// implementation
. j2 b9 K% M% N$ sprivate:
9 {# |/ \( ~$ ]; Q; [( i/ \4 m        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
( ^. @8 X6 O) q# `6 [' C0 k" V        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
* C. _; ?% l  {0 p* S        HRESULT __stdcall SearchComplete(LONG nFindData);

, B- X! X7 Z5 N
private:
        CUPnPImplWinServ& m_instance;
2 U" _* a* ~" I1 n; {        LONG m_lRefCount;
};

6 i# A" O+ d4 v* Z" _  S
// Service Callback
- T  Q8 c& g; O9 ?* p0 V. L  p9 Mclass CServiceCallback
        : public IUPnPServiceCallback
) C$ {4 W9 v: L& I{
3 @' i5 `) n* `) Gpublic:
$ C% D) A) k& i/ v- @0 G, x: Z, v        CServiceCallback(CUPnPImplWinServ& instance)
8 s- a7 E5 J! H3 e" t3 d" ~                : m_instance( instance )
& U7 ]  J# n8 S5 ^8 A5 k        { m_lRefCount = 0; }
   
: H; k0 h+ t4 L4 n   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
( s5 a% J2 K4 P/ z# t

9 }, M" z" u7 {3 h! H// implementation
private:
) N' A: Y0 K. c2 R3 O3 P        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
- }& Q$ s9 N  o2 B7 {7 }8 \
! H  z1 b7 |6 Z% }3 |
, v" f" X1 ]+ R! aprivate:
+ O" T/ ~8 B5 K, o, M        CUPnPImplWinServ& m_instance;
9 }- I# P# H  Y9 Q- b+ ]        LONG m_lRefCount;
+ a1 g: W/ X: b, I, m6 G7 Z};
2 x' J) o5 P& k

/////////////////////////////////////////////////
/ c5 y3 d% L3 Z' Z+ }; l
7 B( u. i% D" c
" z" X. T+ s, h1 F) A使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
2 Y) ?; v3 H" b! NCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
. b1 g8 K! o& b. g! ]CUPnPImpl::DeletePorts删除端口映射.