UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   - F" D, I; E% ?. M3 P
4. #pragma   once
   2 U* w( h; X2 r( F: g. ^! ]
5. 
   ! _% N' b- w' F. |, O5 }) O
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   0 {7 I7 h. I! y
9. {
     y9 A( h& Q' D0 A- ]9 G6 T
10. public:
    
11. typedef   enum{
    : b* k& Z. B/ N' O9 @4 a
12. UNAT_OK, //   Successfull
    
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
    ' X2 i' X& h; b; C- L& K+ l% u) e- M
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
    
17. }   UPNPNAT_RETURN;
    # B& ]" y. v; f
18. 
    
19. typedef   enum{
    + y/ C: C: L, ]/ [- ~
20. UNAT_TCP, //   TCP   Protocol
    : O! a! H/ \/ e1 ]" g/ _
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    . h+ l, L; Q: j4 }- U0 l- v% z
23. 
    6 b# \& `& K! \8 M+ T
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)
    
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    ! S% @4 o1 R1 X4 B  ]" w8 e3 Z
30. 
    
31. MyUPnP();
    : d; m) M! @% R* B( U' \' i7 Q4 H
32. ~MyUPnP();
    ; m2 `; W( ]6 c, t% [
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    . j) R5 V4 W4 g, h& |4 q; \9 a
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    8 K9 a, b/ C9 z1 q' m; i0 ]' {: G
36. void   clearNATPortMapping();
      S$ N! ^9 p0 @' D6 V" ^4 L
37. 
    0 ?- \1 G4 ?, i1 C, E. g9 J% G
38. CString GetLastError();
    0 i" R" J) u  V  b
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    ' `# b, L1 ^% \6 g: ~
41. bool IsLANIP(WORD   nIP);
    9 z4 H, r, k; b3 ?% `' ~* J1 q  h
42. 
    ; H* w4 U5 a9 z$ k  K
43. protected:
    8 [  i, r" C$ b+ q
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    : r: n2 O. \- H. ]  g9 Z" W/ w
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    1 n7 A2 G: u! E5 c
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    
54. bool GetDescription();
    . |$ _" _+ U" m1 x* ?" u
55. CString GetProperty(const   CString&   name,   CString&   response);
    / J; g" V- ~+ f/ [
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    7 f! ]3 m6 k* d, c& g
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    5 K8 p0 S% y- {' l9 Q- Q
59. bool InternalSearch(int   version);
    , ]  W) v3 F( R9 P- s  q: @3 c) C
60. CString m_devicename;
    : F; ~- {; h' H, z) E. S6 x+ {
61. CString m_name;
    5 o$ N$ O, ^, B: k% {
62. CString m_description;
    
63. CString m_baseurl;
    0 c2 F+ R: t( ]
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    0 F) O2 L# j1 R5 o0 E/ H
68. 
      n1 Y% e% P& U
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    : m% s) A' T8 _. Z  \2 Q0 h
72. CString m_slocalIP;
    0 }+ o# T9 N/ t+ G$ t
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   % Y* F, R' b; X' g' i
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   . g; U& {1 Y8 b" z
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   0 }  F1 N# u9 C' S0 @* q
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   : y# F+ @% u) R
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    - j& E: [& C3 r/ b0 j( g
11. 
    : }  x0 W  P: Y7 f
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    / E. |1 C* O5 [- R- K2 U
13. static   const   int UPNPPORT   =   1900;
    3 `7 o: F6 T7 m5 j4 H6 A
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    * f7 |: o8 W* Q8 F" H: j) T7 k
15. 
    
16. const   CString   getString(int   i)
    + {: Q& j& A- R, a: n6 R3 q
17. {
    
18. CString   s;
    ; O) S0 e. ]$ z# _# b
19. 
    ! S3 p; f. B2 [, ^: ]
20. s.Format(_T( "%d "),   i);
    + u+ K0 V6 m  y# N
21. 
    
22. return   s;
    
23. }
    + `# t. o; F: @" w7 O
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    % |2 Q* F' W6 @8 I
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    7 P- `8 {: g2 {0 I
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    - [7 Z/ s' O8 P9 k5 I9 }
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    + D' k: l9 q8 G8 C
33. }
    & ^5 p$ `) |* H% b' [6 X, O$ k
34. 
    0 N/ L. j, Y7 P. R
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    
38. 
    # i# C( a5 g; L
39. const   CStringA   sa(request);
    $ H3 s6 u1 w, J7 @
40. int   length   =   sa.GetLength();
    9 M2 ?8 C9 g- p- ^/ r
41. strcpy(buffer,   (const   char*)sa);
    : j/ F* I, G$ W# M
42. 
      v* Y+ k& W4 ]( U
43. uint32   ip   =   inet_addr(CStringA(addr));
    + ~$ G* P% c* h( j- i
44. struct   sockaddr_in   sockaddr;
    6 @. R' d% |! t* i) P7 r
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    / L: I6 j. w6 l6 n
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    " {% J- z6 k$ q; M7 O" h2 Y; z
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    0 E: P1 F& v6 P* z
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    0 U4 @$ ~- [: E, I% `0 K! K
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;
    / g9 |# a0 y7 S* Z* [" P- N
60. 
    $ X/ y0 b: S! S
61. response   =   CString(CStringA(buffer,   rlen));
    ; A2 g9 d' V; r; C2 h: Z
62. 
    2 V/ v" \( t/ A& V' V% W5 l% G
63. return   true;
    / ~/ R* `- T9 Z: g# C
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    
69. 
    ( t# _; ^( y+ x, M
70. const   CStringA   sa(request);
    * a( J/ N, |2 k) i, U
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    - q( f, f2 `% M6 ~$ z  J
73. 
    
74. struct   sockaddr_in   sockaddr;
    ( q' p; T2 h1 H: T" v7 }' s( P+ t1 P$ k; W
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    ; g, ?# C5 ]* ~* c' t$ T+ N+ L
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    4 t. D6 G& F5 b$ L' N
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    3 e, I( d& H$ I0 A
82. 
    3 }# A5 Q7 ?6 J% }  t
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    8 b$ R6 `2 L$ v4 M2 I
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    $ V, l1 H4 U: m: X
92. pos   =   0;
    " ~( v) e5 k' W+ p; r
93. status.Tokenize(_T( "   "),   pos);
    4 N: H4 A+ n' g" v' T+ M9 o
94. status   =   status.Tokenize(_T( "   "),   pos);
    $ G! `% q& D- B( }: {8 d( ]% m9 v
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    1 X+ X" A! B* }) W# E
97. }
    
98. 
    ; A9 H* |+ I, j1 B) O" ?1 v3 O
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     5 g$ f2 |5 h8 O% _# b# E3 ~
101. CString   startTag   =   ' < '   +   name   +   '> ';
     # o0 S! Y$ u7 `: D( @5 `
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     * i8 G5 h; \3 h  D$ G
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     4 h# O0 h1 i* c. y
109. if   (posStart> =posEnd)   return   CString();
     3 G2 w% r9 p7 ]. h) G
110. 
     + H: c+ G6 K+ Y; A! _9 ^
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     1 x4 L! y" J& z$ a! `$ f8 ?
113. 
     
114. MyUPnP::MyUPnP()
     ) X5 W& W9 s- S$ X' L9 J7 O
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     # _2 c" o* Q: E4 u: Y7 D: ?/ U
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     " S( ^: n6 g+ `# F
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
       G  r' T* b: d- ^- l) g, _1 K) y
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     
133. 
     - j" w6 k! z# a
134. bool   MyUPnP::InternalSearch(int   version)
     ' m% ?# r4 |, [$ P
135. {
     ; m' G, x$ D, j4 m
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     7 d+ O% Q! h( W# q0 y
140. CString   devices[][2]   =   {
     $ W) T: K- D- A' C: b  K* d( e
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     6 v2 L! u+ z7 v; R8 B
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     1 @8 Z( N. W+ H% m# g
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     # R1 y' R/ o; T9 P: w6 X
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     * C. K3 p: J1 q9 U
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     ! Z9 \* g5 s. C8 p, T+ S
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 "),
     $ N# J8 ?6 j7 M$ R4 Y! T7 \
157. 6,   m_name);
     : ~$ E% h' e/ J* R/ B0 _- }
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     . [. R, l7 h; Q
160. }
     7 v5 n2 I+ q3 Z4 t; J
161. 
     4 O1 {6 j' v/ v+ O
162. Sleep(10);
     , `1 m: ?, [, a7 Z' @9 L( ?
163. 
     % N# ?3 ^. E$ ~5 s! Z$ q4 y$ l
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     : b* {' \# L! a, {
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     & g% f. [) b; y+ |7 L
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     $ K; ?; d, c0 P  o1 B
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     0 ?# y  L% q: Q
175. if   (result.Find(m_name)   > =   0)   {
     0 g8 a& p1 @7 A6 d
176. for   (int   pos   =   0;;)   {
     8 s/ @; ~1 F) q, |4 S7 x
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
       s( O& ~. v9 H$ j+ z
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     5 k4 \$ f+ t+ k) Y9 J. Q
180. name.MakeUpper();
     " P6 A" w' j" _- D9 J9 e
181. if   (name   ==   _T( "LOCATION: "))   {
     : T9 C8 K) @. y
182. line.Delete(0,   9);
     
183. m_description   =   line;
     % Z; U7 R: E1 y0 }3 O$ X3 b
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     * I1 W' b, ^8 ^" {
187. }
     / u0 R% r+ n& j- u3 j" k" e
188. }
     
189. }
     
190. }
     + H/ }' l) {+ J! W: c" _
191. closesocket(s);
     8 r/ c- g9 I2 _* y
192. 
     9 \! q! d' T2 a& \& e* W
193. return   false;
     0 o3 y1 y- ]' k
194. }
     

复制代码

天云上人

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


///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
# u( X3 l2 g: \0 N  t

#pragma once
" I5 s; W# Z- {) R#include <exception>
: s  }& p: q" y; U
5 H1 B# l3 U$ h/ _/ g1 l6 S/ O6 H( _: W  [
  enum TRISTATE{
        TRIS_FALSE,
* o4 I( J+ b  G5 J$ ^  s        TRIS_UNKNOWN,
        TRIS_TRUE
# w# R+ ^. u; f/ L2 a};
1 p0 Z2 }# `; M

enum UPNP_IMPLEMENTATION{
# [6 R( t+ ?! L& A( }  t( h, {        UPNP_IMPL_WINDOWSERVICE = 0,
6 G# F& }+ i2 b( _% o8 `+ z  P        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
: H5 D+ T, \# E};
4 @! u- e  N: i& B6 H. l( T0 J
0 K- t3 b1 Y' Y7 C" H( Z


class CUPnPImpl
{
4 ]& n! _) S8 ?; g. \/ ipublic:
( H1 G. Z  f6 h' ^+ n9 H        CUPnPImpl();
        virtual ~CUPnPImpl();
' y! K* g& `. ]" c        struct UPnPError : std::exception {};
' ?+ {% i) D- H' o  U+ f        enum {
* D: u" a2 ~$ q+ ^4 ?                UPNP_OK,
) {) ]6 K# l  o& t8 l                UPNP_FAILED,
                UPNP_TIMEOUT
' v/ }+ i6 |  s. r3 b" E        };


* j3 j4 s" L( W) Y8 K0 u. _        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
( L9 N" V9 Q0 q        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
2 H+ u, x  b4 H: ]2 N5 S# o- B6 _       
: T7 O. ?1 \+ G. J1 y6 H6 L        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
3 s6 R: i* W; f        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
! q1 h% w) ~, n4 L) o        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
9 Q6 _9 u& x+ p9 Q; F( r        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


3 Y( j8 n4 k7 ~  B& h# p// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
9 c( \# }: k5 J        void                                SendResultMessage();
2 O( n1 E/ m. p: {/ f6 e; A0 |        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
" w  p% V7 B% J  P0 ]3 }        uint16                                m_nTCPWebPort;
2 z' I7 {9 M9 i- W9 `* r2 q        bool                                m_bCheckAndRefresh;
) D% G% _& C0 p0 M
: T' \. F$ b3 D5 A# F* {- Q% B
% E; W4 P! @  W  z# j4 w" }private:
5 X$ x" t7 J, z8 X        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;


6 [) e# n3 B& g" y};
- b3 [6 w/ U6 G6 G) r

" v1 [( \1 s$ q% k7 J2 d// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
5 M# Z, h. H8 }/ Spublic:
  u9 W' Y0 d% T& O        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
0 X  A4 G9 u1 e; x, E* N        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};


# _( n# ~. }; c9 ~6 A4 G$ C8 t3 L$ e( Q/////////////////////////////////////
0 ]8 W3 }2 n2 W: G- Z8 L% ~//下面是使用windows操作系统自带的UPNP功能的子类


' q3 }$ N3 E) p; Y% N#pragma once
" U) {# T5 }; O# F#pragma warning( disable: 4355 )
6 M) ^1 e/ K3 x* m/ g0 G

#include "UPnPImpl.h"
# F0 j- ^2 C; ^% m& s#include <upnp.h>
2 {" m- r; `: R% x& @0 C#include <iphlpapi.h>
0 I4 e7 U/ V9 ~* ^. [1 L1 e#include <comdef.h>
#include <winsvc.h>
# k( a" M+ J3 D( I/ ^# r. ]
; n+ F: G+ E4 K
2 ]4 {$ Q, j3 m3 i" q8 Y#include <vector>
; i! y2 g3 x$ A1 A4 Z#include <exception>
( P3 Z  S& S4 @2 e( ^#include <functional>


- X- H, t8 W* Z" d' s7 k6 @

2 V' g- c6 l& P1 ^typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
) C' L" S: q6 H8 h# W' A; M6 Vtypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
- C2 w9 O1 D( Q2 Otypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
4 }3 R0 I2 `9 f# Y+ o+ Xtypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


* X' w5 g9 S% {% Q. {5 ]' }0 Q6 ttypedef DWORD (WINAPI* TGetBestInterface) (
: P* X5 Q. _6 i  K. [/ _1 \  IPAddr dwDestAddr,
, p0 ~% k" B6 C3 x# }7 y4 ^  PDWORD pdwBestIfIndex
8 G/ L/ }3 [4 m& T);
( t6 n7 H! O# K$ h
- Q7 I! C2 ?# Q( M3 O- ?
typedef DWORD (WINAPI* TGetIpAddrTable) (
( G2 Y1 Z, n% m  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
" d; I" Q1 T* n  BOOL bOrder
);


typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);
! c6 h* ?. g: @% |  C# C
5 B0 d7 I2 ^8 L$ P& Y# w% h
' A6 w4 \  [6 UCString translateUPnPResult(HRESULT hr);
4 a0 g* {* c1 w7 GHRESULT UPnPMessage(HRESULT hr);


; N( G/ i8 k% X- q. Xclass CUPnPImplWinServ: public CUPnPImpl
{
2 H; R  m+ f& D* S        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
6 {) P0 F8 a& O3 p# U// Construction
public:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();


! u  v9 K  y0 k5 n$ c6 X        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
" N" f" c( ]) |" |. N        virtual void        DeletePorts();
/ Y1 z+ ?& D# V- e$ d4 F4 W        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


  J$ G( O$ P: V/ x' c) p        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
7 T1 [* _' P3 C7 [, n4 I        virtual bool        CheckAndRefresh()                                                                                { return false; };
* {- c3 c8 a& h5 v& N

+ A3 L. X8 G. ^( u2 B- T6 ~# lprotected:
* t% `( `4 A' N7 O        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
* z0 ~% ?3 @# |% @$ ^! ]6 C) C        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
; p4 y* ~* W5 s9 w# n' r        void        Init();
/ f+ ~- v. g- w* c; x6 o/ K8 w

        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
% f, |7 b5 L$ [: ^+ y. S6 i' _                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
" x3 V9 R4 w9 g2 m0 o                        m_bAsyncFindRunning = false;
  m" \. v* d9 \8 l- v                }
8 r* Q' P% N$ \1 a- }. \; C                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
  I5 c$ V- N" w" M. N( V                {
                        TranslateMessage( &msg );
8 I( A, [) W. V8 ^6 ^2 s0 N                        DispatchMessage( &msg );
% ]7 R: i; K) V2 U* T                }
( `5 u3 {. u4 q' t                return m_bAsyncFindRunning;
* v  i& R6 o0 {2 A7 ]* V        }

, Q# m$ i* T) G2 _- f9 ~6 {+ ~
' I/ D$ s3 y" @        TRISTATE                        m_bUPnPDeviceConnected;
: V* Q5 r' `/ a* p6 e

// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
( L' E! A( `2 q( f3 m. X& G, M) ^        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
1 m+ p  t6 ?% M6 @% l) O        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
- Z$ n- v/ l; w6 {        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

# @* h8 s2 L! C8 y; p
        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
; w4 h9 x6 O2 z$ O        TGetIfEntry                                m_pfGetIfEntry;
/ j, @# t- A0 f9 u) {% e- n3 g
; M! d- W) Y) T
        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
9 M6 ?' h* }0 p/ Z& [        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
/ c( g" I: H9 j1 `) O* S+ q                result_type operator()(argument_type device) const
  r! F0 Z3 @& e- P                {
                        CComBSTR deviceName;
+ ?0 x% T9 p0 q                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
) h1 h3 K; \. }

8 ]6 V. j5 E' P' \  j4 |: w8 y                        if ( FAILED( hr ) )
$ Z3 M. Z2 k7 C6 g. J  j" L                                return UPnPMessage( hr ), false;
- J0 @5 M; a& X3 r$ W- z
" [/ w- ], m% ]# d- @3 h* x
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
, M5 k$ q( E/ z  [8 I- ?; v        };
* _- j- D4 x1 ?       
+ [% K8 G& ~5 e        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
) A. y6 q8 Y' J. i, L  i3 A2 h! K        HRESULT        MapPort(const ServicePointer& service);
$ n& V5 ~) ]- U3 r. n        void        DeleteExistingPortMappings(ServicePointer pService);
  z: G2 n0 s# s# H- v$ `" z        void        CreatePortMappings(ServicePointer pService);
$ S$ q3 e5 j/ e( L' P        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
& Z% `4 p$ R  ?/ {% l1 q0 [        void        StopUPnPService();
7 h, c: }2 _& U
2 o2 x2 B* D3 e- E
        // Utility functions
2 @2 x0 C: K. E        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
% a, O" E$ a$ m. a) a7 @* P4 n! L5 T        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
4 B: x9 s6 v- b; i5 y5 h        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
+ L, w& ?, r$ q        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
: w" g, h, B, X        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);

4 p; L7 `( \" |& V4 d
/ q2 A7 ^+ ^0 o// Private members
/ \8 K( p6 v4 D8 H; H: hprivate:
- _9 f- G( c; q% H8 V5 h+ F3 A2 D" c        DWORD        m_tLastEvent;        // When the last event was received?
) K1 M+ d4 t( @( s% M6 [8 w0 o        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
7 Z+ @0 x1 g7 ^& p7 s8 Y& L        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
  w0 U: |% N9 b: ~! B

        LONG        m_nAsyncFindHandle;
7 t  {6 }  L4 B' D+ j2 I# k9 g        bool        m_bCOM;
        bool        m_bPortIsFree;
1 b- J4 f) I2 e2 R        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
' L0 Q3 f0 ~' ?' Y7 z6 J1 J3 g) F0 W- D        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
( d& v- {! P# S# n        bool        m_bDisableWANPPPSetup;


$ v8 o8 Y$ s* Q& }, Z! ~};
" T/ y$ J' ~" O& y
# j/ B% W, S" g: t  V& J2 t
// DeviceFinder Callback
1 P: O6 u# i& A8 T& u2 {class CDeviceFinderCallback
8 g! S0 N" z; C2 g+ ?* B$ N        : public IUPnPDeviceFinderCallback
8 N- {. R, L' J- v5 p{
1 f8 B  _; H& c8 Gpublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
$ b& Q: J  F$ j7 }! }        { m_lRefCount = 0; }
8 Y: t: {( H' E) e- T
4 `; V- |& Q  {" r0 U1 `
0 K7 c! n0 x, a: B   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
2 Z7 i: \, @2 _) o& t3 y   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
- Q* J" b) w1 w/ H% A
6 |, u9 J2 u$ {3 G0 P; |. ~9 h' K
$ D9 x, y  Z: s: e6 V3 X# h// implementation
3 w' F1 m) ]3 o, Oprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
" f, M: `, `7 K3 D" ?        HRESULT __stdcall SearchComplete(LONG nFindData);

% z& |! P/ |6 u/ j1 S; Z
private:
% X0 a- x4 |7 p% Z, g& @        CUPnPImplWinServ& m_instance;
' m, U; R  [3 [) \; m        LONG m_lRefCount;
, c) m5 q* v( a  `};
6 g! ~: w' U" M

  i* \4 n" Y7 q) P2 Y* o: d// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
* b4 l6 @, h0 X& E- G{
* ~+ c4 o* Y: t- ^6 a9 npublic:
        CServiceCallback(CUPnPImplWinServ& instance)
; `* v) z  V! G1 c+ J                : m_instance( instance )
        { m_lRefCount = 0; }
3 }- }( U! p5 `* \: e+ ~2 V% [* r5 n   
0 V# k; I5 A0 u/ x4 U1 }! G, y   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
/ [3 f8 H& Q& o' a* C" X
/ C+ W8 m& S5 a: b
// implementation
. c" c, L9 W4 Z* B% cprivate:
( a3 c0 R& ^6 o' T, u  U        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


+ {, Y# E9 j3 ]' ?+ `6 q) ?private:
        CUPnPImplWinServ& m_instance;
& @0 ^) Z+ C3 Z9 o  s6 M7 C        LONG m_lRefCount;
, ~' B2 c8 C3 j* N$ C: B6 B};
7 [+ E: _2 A% d9 s" p7 G

/////////////////////////////////////////////////
+ N, a- ?# G; R9 V1 H9 g# w3 p

% P% U& j+ J4 |" n8 C4 Z使用时只需要使用抽象类的接口。
4 b3 k+ W; Y4 ~6 @$ \8 P8 p7 V) ICUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
) Q& g! D5 q' [5 uCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
7 T+ K8 }' _: d9 U% [5 mCUPnPImpl::StopAsyncFind停止设备查找.
& y6 h) W3 H3 Y! i. s" r* n2 dCUPnPImpl::DeletePorts删除端口映射.