UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   ; [, Q8 t9 E3 z6 u' o. n9 g/ T
3. 
   7 @$ Z! k6 @) f' g! Q* D% h) i
4. #pragma   once
   1 \0 ~7 ~4 p' p( T- y. J& Y
5. 
   ! }( B$ W, E) C" g7 H3 e2 |8 ~: }
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   
9. {
   
10. public:
    
11. typedef   enum{
    
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
    
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    5 q' i* ?) y* v: y' U
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    + H3 s& F8 B0 d
17. }   UPNPNAT_RETURN;
    
18. 
    3 l2 [3 e3 o; Q5 V; X4 z
19. typedef   enum{
    . H) m/ n. R% m, L: O* U
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    6 q/ ]" W1 X! G+ T5 a
24. typedef   struct{
    5 E) n" O- v0 x1 E. t. a
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    7 x- h* d/ r) ]' F7 p- r" F
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    ) h1 d$ B; b% t, N& @
28. CString   description; //   Port   mapping   description
    : B5 R0 K% X* m
29. }   UPNPNAT_MAPPING;
    ) q/ i' l+ Z% g9 R! u6 v( A; G7 ?
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    ) ]) ]* E! ]. b# l
36. void   clearNATPortMapping();
    
37. 
    ) Q) j+ k& h4 I* B) `4 P
38. CString GetLastError();
    / O+ S! }+ i% p2 g( M
39. CString GetLocalIPStr();
    * A, T3 k" c9 T8 R. G; j
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    8 Q- f0 p! C+ l* ]2 p
42. 
    2 m# o8 v* u7 K
43. protected:
    
44. void InitLocalIP();
    % M9 @  o6 u% P& ~
45. void SetLastError(CString   error);
    
46. 
    & Y7 t% q: r; {& T
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    + w; P$ ]% t9 ~8 Q8 M0 i/ {
48.       const   CString&   descri,   const   CString&   type);
    * h# r9 T0 A6 ^- H3 {
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    9 i( F3 _  p5 J7 H3 [9 b
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    , a, l; M/ g  E+ u+ v
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    6 R# k' }; F$ e4 S7 ?
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    9 K6 \- Y1 E5 g' g, z
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    8 `6 g9 F  c- g9 Q0 V
61. CString m_name;
    
62. CString m_description;
    ) W% |! X& B, W% T' ?) P0 W* Z
63. CString m_baseurl;
    4 |* n" }* v5 W- Z4 W/ O  a
64. CString m_controlurl;
    + G( |4 t2 r: s9 l" C/ ~7 w
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    ! x" R* [9 @3 t- H9 d
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    ) J3 P- O2 ^, ?  W. y6 j
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    / `" m( k  H3 x3 J
76. bool isSearched;
    
77. };
    " b% l. i$ `3 S  B6 B8 J8 L
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   ) R- I3 O  q* b7 a+ N1 q: d2 [
2. #include   "stdafx.h "
   % i$ I) u+ t4 B1 q9 o2 Z3 ]
3. 
   3 X5 p+ b( D+ _- w: u6 F  b) P  N
4. #include   "upnp.h "
   
5. 
   * d! R, W' T& Y0 _& f, G
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   6 o) h0 A+ G. y7 Z
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   ' e( M+ f  I8 M' [$ P
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   6 y- D( L) U( e5 b. _) g' \0 p2 v
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    & k! l, Q% j! O/ i% D1 T
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    6 z0 q) a6 s$ C6 w
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    ! z! a2 ~2 Z. U4 v6 q
16. const   CString   getString(int   i)
    : _7 z1 y9 d. s
17. {
    
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    6 a6 a9 Y9 u& N
21. 
    
22. return   s;
    
23. }
    
24. 
    " N5 x* y4 ]+ G# t! n
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    6 ?. Z  E4 B) `) \: D$ M; D
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    9 Z2 T% ?2 I1 B+ D- s3 p8 j, y& T1 X
30. const   CString   GetArgString(const   CString&   name,   int   value)
    3 H: ^  {6 X" s6 Q2 m
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    0 h! l- w1 S' b% l
33. }
    4 k5 n8 @* z% F7 A
34. 
    
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();
    
41. strcpy(buffer,   (const   char*)sa);
      Z/ c. ^, r& E4 ^. k7 c
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    1 N+ b7 b0 P+ Z. |# v& J
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    3 d# U' e; u$ P1 g  B/ F' J
47. sockaddr.sin_port   =   htons(port);
    8 a# p9 b) i0 x, ?% Q
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ! g& K& g+ ]; K0 i" V
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));
    * v9 K1 [6 `4 s0 t/ L5 V
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    + W# Q& X1 |! p3 G- h$ B- Q+ l
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    + n% h' }8 {9 Q5 [( E4 O0 |
59. if   (!rlen)   return   false;
    , f" D7 g6 P  C7 t: A) V2 U
60. 
    ' ?  W' d* c* G  a- A
61. response   =   CString(CStringA(buffer,   rlen));
    8 }8 ~1 n( N; ?5 h* q
62. 
    
63. return   true;
    
64. }
    / |9 F9 e# N5 z* {" f9 G
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    
69. 
    4 h- q8 r, ~  g; [5 l; {
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    0 r4 |' x# I1 D! P2 m$ W! }
73. 
    
74. struct   sockaddr_in   sockaddr;
    4 m) w6 o& ^1 X" k1 D2 a7 D. M
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    - R- \# f# r# {7 ^) q
76. sockaddr.sin_family   =   AF_INET;
    1 k+ Q: V/ {+ o0 A( M! f$ ^
77. sockaddr.sin_port   =   htons(port);
    * h' B) p" v0 d, ?& W) j% E7 v. T
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    - r0 f! Z- s$ Y/ T( w2 Q/ K
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    . \  p% h, w5 P  y  Y8 |# a, |
84. {
    ' Z0 l) W5 z) J
85. int   pos   =   0;
    % ^: R" w' _$ r1 |7 j2 u
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    ! r" Y8 n% p% O9 ^+ _4 s+ G
88. 
    2 q0 l3 N- W5 L6 W7 N
89. result   =   response;
    
90. result.Delete(0,   pos);
    " Q  S) f/ w' b: ?
91. 
    $ q# H3 T/ L, w" k
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    : e8 [+ J/ A# n+ d3 Z* ~: f
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    4 W" P5 [8 Z& f9 A
96. return   true;
    
97. }
    5 z5 F. @* w1 D7 [
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     ( ^  _6 |3 J0 f
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     * H1 Y' W. y2 T+ d3 }
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     0 _+ U6 E$ T. t0 S# L' p
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     6 \+ ^7 O5 Y' T4 _, n
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     7 O) m7 p, |/ N2 `1 h! \
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     
120. 
     4 ?$ j/ Z/ i1 A( w3 E
121. MyUPnP::~MyUPnP()
       m6 ?  Q$ W6 f6 A  d1 O  C: N# E
122. {
     : I) {- D% |8 M! I* ^
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     + [2 d5 X- d( p" e  Q
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     " z" J( T9 L5 P) M& u# p
128. }
     * }  @8 f3 b* W+ b. u) e6 B
129. 
     
130. m_Mappings.RemoveAll();
     0 E) p8 k( ^$ @3 ^
131. }
     
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     8 s+ Y% y, Z) h: b( K0 e
135. {
     % J+ }. y- R: E
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     ) F  ~( j2 Z+ [4 G4 B/ \
141. {UPNPPORTMAP1,   _T( "service ")},
     ( c2 h/ b/ [, K* ]; c5 h
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
       m" M6 L1 i  H* b* l
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     + M: p# u5 f( D6 ~( X
149. 
     
150. int   rlen   =   0;
     * d2 i) f0 Z* l2 c
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     
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);
     $ _. {( |* c4 s* X0 {
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 "),
     
157. 6,   m_name);
     1 L$ l, H& [# P4 h% i
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     " T' W  j/ Q1 S9 H
159. }
     9 S; ]( ]* @" W" Z2 ]
160. }
     
161. 
     
162. Sleep(10);
     6 q* k# g4 O4 m$ N/ A
163. 
     1 `! }/ D) u5 J0 _6 H) {
164. char   buffer[10240];
     1 x8 s- T+ G; E; o! o
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     " u( ^6 I1 o: b- i6 B3 b
167. closesocket(s);
     
168. 
     5 f: j. X) [# y+ F
169. CString   response   =   CString(CStringA(buffer,   rlen));
     4 Z' u9 i. j1 |. \* v7 \. b
170. CString   result;
     1 I$ U7 w* M1 ^  ~0 O7 G( B
171. if   (!parseHTTPResponse(response,   result))   return   false;
     ! C1 e2 M8 _1 m4 w. q2 [0 K
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     0 |6 n5 b4 q% Z  L5 T  o
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
       U9 W7 x) h$ r3 J( ]/ {' C
175. if   (result.Find(m_name)   > =   0)   {
     : g2 r1 S& A" n# _) q! c/ t5 R
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     " q: n- v1 x6 s/ [; S
179. CString   name   =   line.Mid(0,   9);
     0 X5 |$ C. a* _
180. name.MakeUpper();
     + A" ~, R+ _# I7 U0 y8 Z: k, C7 i" a
181. if   (name   ==   _T( "LOCATION: "))   {
     # I/ L; A: k! h0 J$ F" h
182. line.Delete(0,   9);
     
183. m_description   =   line;
     * Y4 _, A3 l! E* }. ?3 f! y. I
184. m_description.Trim();
     8 h8 s& [+ r8 p# Q, o5 e1 e/ z
185. return   GetDescription();
     / y5 q; C7 f8 Z* C, B& z2 t' G
186. }
     
187. }
     
188. }
     . A: x2 r& ?5 e: n
189. }
     
190. }
     " q" v0 _& E, ?# a9 B
191. closesocket(s);
     3 n% A6 J3 g/ y
192. 
     3 q! R4 B7 d2 H0 a! ~
193. return   false;
     7 ]( i; `1 P4 D, P. M
194. }
     

复制代码

天云上人

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


: o: l# b5 P- |7 }& i///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
- p7 R: }) O. L$ H9 t

( y- S2 o, V5 u8 t  f+ I% G#pragma once
#include <exception>

* V- P; x) w5 e! y  J: d8 P) q
  enum TRISTATE{
1 g! Y+ W8 Z6 r; m        TRIS_FALSE,
/ P8 L8 E* I* [8 d- {        TRIS_UNKNOWN,
; [" F+ c$ s, V, W6 p! U        TRIS_TRUE
};


0 {6 Q$ V! ]6 _- B: b1 n9 Eenum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
. Q, q6 Z& o0 S5 [% S        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
- e0 k/ J4 }5 n# s) b};




class CUPnPImpl
{
) i& I' {) k  ~! P' {3 Npublic:
        CUPnPImpl();
8 e  c7 @* i1 l$ g        virtual ~CUPnPImpl();
) d" U" V5 _( b* G1 u; P( d% ^        struct UPnPError : std::exception {};
' X- n$ `  L) Q* I6 w9 Y7 A" I9 s        enum {
                UPNP_OK,
                UPNP_FAILED,
9 ?/ |1 D% M& F$ }                UPNP_TIMEOUT
+ G3 g9 c+ t7 C( O! [  B; q        };

8 d- u. a% ]8 C3 {; z5 O
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
2 ~7 b+ F" i3 k( P' l8 t1 ]0 |6 b        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
; w" p$ f+ o# O7 E% r       
$ z/ K, R  ^" \( O        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
. ~4 e9 z8 q+ {7 H+ ]

; V( r( s6 c0 f) J- s        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
) r0 l5 Q9 D% M+ ~# Q2 |6 @+ U- J5 I        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
+ Q3 o& \! _  H( R9 A" T4 `: s4 E

' G) ?3 _' V/ ], `: z% `  w// Implementation
protected:
; {; l1 _& u( x        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
( R7 d3 o) j5 b  o% Z6 L        uint16                                m_nTCPPort;
5 @! I% k; W+ U        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;


, f1 P' i' J: v  p7 Gprivate:
0 T& P+ q% C% y; q. i! P( L        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
3 G: v  v- D& s% f

};


  a: n, y( Q6 c; R2 k// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
1 C2 c& F$ d" @- S3 x/ p{
2 d  S; h( z) fpublic:
' S5 }1 ?- z- i' j        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
( J# c2 }- f3 H: Q5 W        virtual bool        CheckAndRefresh()                                                                                { return false; }
9 Y# i7 z- t& o/ q/ H; Y, W  g        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
4 f2 N+ r+ g2 R6 F1 t        virtual bool        IsReady()                                                                                                { return false; }
# i( Y$ y* U% D; U3 F( a, r        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
7 S& }; ?- {1 z};
' R7 @" v  z9 M0 `8 K
" [$ `8 G5 K9 Y) ^" W* I" q+ W
3 A; X: ], q$ y" R4 E* c* o6 `  E/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类


#pragma once
#pragma warning( disable: 4355 )


1 g+ ?2 T8 S, L* P" }! W#include "UPnPImpl.h"
#include <upnp.h>
& v# U6 i) [2 {+ n" c#include <iphlpapi.h>
4 {& j  k' j9 }' q3 z2 u* T#include <comdef.h>
( D7 w( U7 ~# {9 d2 F% F#include <winsvc.h>
7 U  U) |; S, R% h( x( U% u1 C5 p

#include <vector>
#include <exception>
$ B0 R4 N! N. U. Q# Y#include <functional>
' [8 a8 j& u0 _$ K7 `" d
; b' N/ E4 E+ o# e4 V
* z  {! z, t) I# ?- K2 G
( E$ r" p1 T5 J; r0 L  T3 |$ M
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
# z$ M  g+ m5 p, Ytypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
& d) l, q" \: ^7 Wtypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
* D2 E  `* x  a! q7 g9 ]typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


: F/ o' o+ e( Xtypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
# F3 ]" \5 c- m; s$ t% {, n$ f$ Z  PDWORD pdwBestIfIndex
+ D- O5 V6 C  @% x);
6 a% y& _4 z0 i6 U' z; l
8 f2 C7 m& Y6 f5 P9 w7 z  B$ e
typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
9 n. H, q, }3 {/ Y  PULONG pdwSize,
! A1 [0 c% G* z) ^; o& S' R2 J* p  BOOL bOrder
6 [8 n/ U6 R2 g+ Z* ^, \3 t! e);

$ k$ _% }+ Y, {$ y$ c" Z
typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);
$ t8 z- _( c3 V/ }
: @+ j1 `2 [7 W; [
CString translateUPnPResult(HRESULT hr);
# `. _7 g# f: R+ x3 Q7 cHRESULT UPnPMessage(HRESULT hr);


4 b6 v) s6 t2 w' F( A; \1 @class CUPnPImplWinServ: public CUPnPImpl
{
- h$ a# o, r; j        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
- s" q0 ]- ~& a* C6 ?% S( S6 x: s// Construction
* |7 E8 x  N5 Y2 \public:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();

9 j, k7 p$ h4 U8 i
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
8 ]3 ]" P, _& O9 y. u$ d2 }3 \4 B9 ]* I        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
' K- h' W6 r: M* W0 }        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
' S7 A  o" e) r        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };


protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
' m* \: }4 Y9 A; L7 C7 A        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();
& k1 W0 F4 G* }& e3 b

        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
, b( H. N' U( e7 [4 Q  o4 y& F                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
" A0 T7 \7 R5 N5 L                MSG msg;
2 s( t3 H) H7 j, n- Y3 ?# q- S                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
, L8 }. a  a4 _$ ?# n5 G3 z5 h                {
                        TranslateMessage( &msg );
# d+ Q9 e+ A* R                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
2 p0 ]0 S7 i9 @$ s- o. \; p        }


        TRISTATE                        m_bUPnPDeviceConnected;


5 l9 @1 N* ]# _# b8 ~# t# K' [7 c6 v// Implementation
        // API functions
7 I+ u: g" _( v* k" C) ]        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
9 X5 c5 M! D0 [2 B/ W5 G        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);
0 Y0 R+ _% K* j! L        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
2 ^# c+ l0 i! A( J( P: w) F        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
3 F, a& G( C  v! X9 X

2 {- E9 X- B+ i) D& x1 l        TGetBestInterface                m_pfGetBestInterface;
- R# U! d. I. n4 s3 l# h        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;


, a$ R  ~+ J6 W. H% O4 u2 s; D        static FinderPointer CreateFinderInstance();
( e- J+ }* A$ d4 h9 W" a& D        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
9 ^" \3 g1 v! N: E" r                result_type operator()(argument_type device) const
/ y, J1 [1 x  U, W                {
: W: \6 k* t8 X                        CComBSTR deviceName;
& X" F# m/ [+ ~% b% N. u                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
( J* R' A  j# U  u" A
, H4 i" ]" Q; s+ b+ g# M0 G
                        if ( FAILED( hr ) )
. b- P3 c4 j( |& [- A" u                                return UPnPMessage( hr ), false;
0 G' d9 Q) ]) P8 K" i1 {
% k7 K8 n8 A3 ?) |$ Q
( c. i% e; n; r# ?' I3 Y* W* s$ F                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
8 Z* z1 g  v2 }! X8 s        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
" s* m8 m, q0 M$ d- `/ e$ F        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
: g- }& D! B% c) Y3 ]; V5 Q" y        void        DeleteExistingPortMappings(ServicePointer pService);
& u2 g) m2 h1 L" H        void        CreatePortMappings(ServicePointer pService);
2 D3 y* a* b% p! v, e9 w        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
+ r1 u' {8 W7 b' t        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
% z# R3 B. Q" c* r6 K2 X
5 f" d# P' ]  Q+ z3 u( {! o! y# t0 [
        // Utility functions
' m6 B" O1 G6 |$ D3 z2 c0 W3 ]        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
4 a% {. ]8 E. V! d& g4 j        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);
) A* P. k7 W$ U+ W6 T6 Q8 }        CString        GetLocalRoutableIP(ServicePointer pService);

) F* Z. B3 s9 h1 V/ F4 s
( t0 q) s5 X+ W" \& X// Private members
, S* Q0 D% j6 J4 W7 S+ ?private:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
; k5 }9 Z+ x/ T* A6 W        std::vector< ServicePointer > m_pServices;
+ W# X) H- r/ U- x        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
, m  g4 `4 S, n1 m: p4 T/ _( n8 r2 v

% A7 o$ A) ?& X( t        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
7 C# s4 {# P4 Y0 h5 i( {6 D: m1 \        bool        m_bPortIsFree;
        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;
  L9 U! q& C" @8 q& Y# J9 U        bool        m_bSecondTry;
. V( A6 @/ N& F2 H. U        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
: F" X  h( {3 H. G" [; w        bool        m_bDisableWANPPPSetup;
: w; ~: h5 w# P) L$ m" Y7 G

) W0 o" U5 v' Z; P# U" T1 E- x- j, U};


5 H! U5 v6 X. \7 r, w: U: w9 S// DeviceFinder Callback
class CDeviceFinderCallback
; m" x9 A- u6 r4 x8 M& S        : public IUPnPDeviceFinderCallback
{
9 Q5 [& w+ y$ E0 Z: o& Apublic:
7 U3 b8 O! o2 m1 s* U2 U: V        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }

/ m8 Q& S+ |" I; \( K% v9 F/ Q
- e. c+ _, H  ?- h4 M7 C* ?   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
% h/ q$ V# r8 c: Q$ j   STDMETHODIMP_(ULONG) Release();
6 v; E/ I1 j- [- E
- v7 [# |$ k6 }' Q% w5 |/ O
// implementation
private:
* ?1 \7 I! S- Q1 A' L8 ?: j        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
0 d7 m6 t7 x' U. s  w0 A        HRESULT __stdcall SearchComplete(LONG nFindData);

1 ?+ `( d6 x+ g1 V
% {* E4 ]8 s' ?9 g, E5 P1 Iprivate:
- J6 P! D& S) M' ?        CUPnPImplWinServ& m_instance;
: T' C6 J! m8 s5 ]7 T: Q        LONG m_lRefCount;
$ F2 N) A# F) L5 U1 e};

) x0 d, Z, Y% L5 x
// Service Callback
% T7 U6 G1 ~! ?- b5 }# W; y0 Zclass CServiceCallback
        : public IUPnPServiceCallback
# v7 ]& f( @1 {6 E" s. t- \{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
$ L, U* B. Z; }) y! T2 [2 \        { m_lRefCount = 0; }
   
& F! J. [, o0 L  |/ ?0 O   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
3 E4 l- B) t  x' k" r1 |$ K   STDMETHODIMP_(ULONG) AddRef();
) [+ Z# N+ ?4 {8 B" g) u, h   STDMETHODIMP_(ULONG) Release();
' o" ~# v. m6 u/ q; i: R% d4 R5 S6 r- R
4 ^) _% }7 K6 \* Z" N
// implementation
$ @1 Z* A1 m6 h/ O* c( bprivate:
% V: L0 `9 h4 v; n, `# }2 L        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
3 U3 `! q0 ?. w0 y7 d$ w        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

7 O: x4 v% J, Q) C% G- P
2 C2 U: l( k5 S, k" |! T! Eprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
8 x; r7 V, j# O% `, G};

! e4 |9 P& U. V7 J8 ?
3 h0 ^* N  y& |: C7 d2 ~/////////////////////////////////////////////////
: T1 `7 v& L: T' p2 H& R
/ c  h( S6 M0 O7 T" f+ F7 M# e
& ^3 _* q% i* a( k使用时只需要使用抽象类的接口。
4 B* o8 q; @( d; [CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
/ Q' ]5 ?4 u' A* L7 `. N% mCUPnPImpl::StopAsyncFind停止设备查找.
2 B- r: E! u4 v; |3 b9 P/ SCUPnPImpl::DeletePorts删除端口映射.